12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697 |
- // Copyright 2002 The Trustees of Indiana University.
- // Use, modification and distribution is subject to the Boost Software
- // License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
- // http://www.boost.org/LICENSE_1_0.txt)
- // Boost.MultiArray Library
- // Authors: Ronald Garcia
- // Jeremy Siek
- // Andrew Lumsdaine
- // See http://www.boost.org/libs/multi_array for documentation.
- //
- // reshape.cpp - testing reshaping functionality
- //
- #include <boost/multi_array.hpp>
- #include <boost/core/lightweight_test.hpp>
- #include <boost/array.hpp>
- #include <boost/type.hpp>
- int
- main()
- {
- const int ndims=3;
- typedef boost::multi_array<int,ndims> array;
- typedef boost::multi_array_ref<int,ndims> array_ref;
- typedef boost::const_multi_array_ref<int,ndims> const_array_ref;
- boost::array<array::size_type,ndims> dims = {{2,3,4}};
- boost::array<array::size_type,ndims> new_dims = {{4,3,2}};
- int data[] = {0,1,2,3,4,5,6,7,8,9,10,11,12,13,
- 14,15,16,17,18,19,20,21,22,23};
- const int data_size=24;
- // Basic reshape test
- {
- array A(dims);
- A.assign(data,data+data_size);
- array_ref B(data,dims);
- const_array_ref C(data,dims);
- A.reshape(new_dims);
- B.reshape(new_dims);
- C.reshape(new_dims);
- int* ptr = data;
- for (array::index i = 0; i != 4; ++i)
- for (array::index j = 0; j != 3; ++j)
- for (array::index k = 0; k != 2; ++k) {
- BOOST_TEST(A[i][j][k] == *ptr);
- BOOST_TEST(B[i][j][k] == *ptr);
- BOOST_TEST(C[i][j][k] == *ptr++);
- }
- }
- // Ensure that index bases are preserved over reshape
- {
- boost::array<array::index,ndims> bases = {{0, 1, -1}};
- array A(dims);
- A.assign(data,data+data_size);
- array_ref B(data,dims);
- const_array_ref C(data,dims);
- A.reindex(bases);
- B.reindex(bases);
- C.reindex(bases);
- A.reshape(new_dims);
- B.reshape(new_dims);
- C.reshape(new_dims);
- int* ptr = data;
- for (array::index i = 0; i != 4; ++i)
- for (array::index j = 1; j != 4; ++j)
- for (array::index k = -1; k != 1; ++k) {
- BOOST_TEST(A[i][j][k] == *ptr);
- BOOST_TEST(B[i][j][k] == *ptr);
- BOOST_TEST(C[i][j][k] == *ptr++);
- }
- }
-
- return boost::report_errors();
- }
|