multi_array/test/reshape.cpp

// 
// reshape.cpp - testing reshaping functionality
//

#include "boost/multi_array.hpp"

#define BOOST_INCLUDE_MAIN
#include "boost/test/test_tools.hpp"

#include "boost/array.hpp"
#include "boost/type.hpp"

int
test_main(int,char*[])
{
  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::exit_success;
}