index_bases.cpp 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153
  1. // Copyright 2002 The Trustees of Indiana University.
  2. // Use, modification and distribution is subject to the Boost Software
  3. // License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
  4. // http://www.boost.org/LICENSE_1_0.txt)
  5. // Boost.MultiArray Library
  6. // Authors: Ronald Garcia
  7. // Jeremy Siek
  8. // Andrew Lumsdaine
  9. // See http://www.boost.org/libs/multi_array for documentation.
  10. //
  11. // index_bases - test of the index_base modifying facilities.
  12. //
  13. #include <boost/multi_array.hpp>
  14. #include <boost/core/lightweight_test.hpp>
  15. #include <boost/array.hpp>
  16. #include <vector>
  17. #include <iostream>
  18. int
  19. main()
  20. {
  21. typedef boost::multi_array<double, 3> array;
  22. typedef boost::multi_array_ref<double, 3> array_ref;
  23. typedef boost::const_multi_array_ref<double, 3> const_array_ref;
  24. typedef array::array_view<3>::type array_view;
  25. typedef array::size_type size_type;
  26. typedef array::extent_range range;
  27. typedef array::index_range irange;
  28. array::extent_gen extents;
  29. array::index_gen indices;
  30. // Construct with nonzero bases
  31. {
  32. array A(extents[range(1,4)][range(2,5)][range(3,6)]);
  33. array B(extents[3][3][3]);
  34. double ptr[27];
  35. array_ref
  36. C(ptr,extents[range(1,4)][range(2,5)][range(3,6)]);
  37. const_array_ref
  38. D(ptr,extents[range(1,4)][range(2,5)][range(3,6)]);
  39. array_view E = A[indices[irange()][irange()][irange()]];
  40. std::vector<double> vals;
  41. for (int i = 0; i < 27; ++i)
  42. vals.push_back(i);
  43. A.assign(vals.begin(),vals.end());
  44. B.assign(vals.begin(),vals.end());
  45. C.assign(vals.begin(),vals.end());
  46. boost::array<int,3> bases = { { 1, 2, 3 } };
  47. for (size_type a = 0; a < A.shape()[0]; ++a)
  48. for (size_type b = 0; b < A.shape()[1]; ++b)
  49. for (size_type c = 0; c < A.shape()[2]; ++c) {
  50. BOOST_TEST(A[a+bases[0]][b+bases[1]][c+bases[2]] == B[a][b][c]);
  51. BOOST_TEST(C[a+bases[0]][b+bases[1]][c+bases[2]] == B[a][b][c]);
  52. BOOST_TEST(D[a+bases[0]][b+bases[1]][c+bases[2]] == B[a][b][c]);
  53. // Test that E does not inherit A's index_base
  54. BOOST_TEST(E[a][b][c] == B[a][b][c]);
  55. }
  56. }
  57. // Reindex
  58. {
  59. typedef array::size_type size_type;
  60. array A(extents[3][3][3]), B(extents[3][3][3]);
  61. double ptr[27];
  62. array_ref C(ptr,extents[3][3][3]);
  63. const_array_ref D(ptr,extents[3][3][3]);
  64. array_view E = B[indices[irange()][irange()][irange()]];
  65. std::vector<double> vals;
  66. for (int i = 0; i < 27; ++i)
  67. vals.push_back(i);
  68. A.assign(vals.begin(),vals.end());
  69. B.assign(vals.begin(),vals.end());
  70. C.assign(vals.begin(),vals.end());
  71. boost::array<int,3> bases = { { 1, 2, 3 } };
  72. A.reindex(bases);
  73. C.reindex(bases);
  74. D.reindex(bases);
  75. E.reindex(bases);
  76. for (size_type a = 0; a < A.shape()[0]; ++a)
  77. for (size_type b = 0; b < A.shape()[1]; ++b)
  78. for (size_type c = 0; c < A.shape()[2]; ++c) {
  79. BOOST_TEST(A[a+bases[0]][b+bases[1]][c+bases[2]] == B[a][b][c]);
  80. BOOST_TEST(C[a+bases[0]][b+bases[1]][c+bases[2]] == B[a][b][c]);
  81. BOOST_TEST(D[a+bases[0]][b+bases[1]][c+bases[2]] == B[a][b][c]);
  82. BOOST_TEST(E[a+bases[0]][b+bases[1]][c+bases[2]] == B[a][b][c]);
  83. }
  84. }
  85. // Set Index Base
  86. {
  87. typedef array::size_type size_type;
  88. array A(extents[3][3][3]), B(extents[3][3][3]);
  89. double ptr[27];
  90. array_ref C(ptr,extents[3][3][3]);
  91. const_array_ref D(ptr,extents[3][3][3]);
  92. array_view E = B[indices[irange()][irange()][irange()]];
  93. std::vector<double> vals;
  94. for (int i = 0; i < 27; ++i)
  95. vals.push_back(i);
  96. A.assign(vals.begin(),vals.end());
  97. B.assign(vals.begin(),vals.end());
  98. C.assign(vals.begin(),vals.end());
  99. #ifdef BOOST_NO_SFINAE
  100. typedef boost::multi_array_types::index index;
  101. A.reindex(index(1));
  102. C.reindex(index(1));
  103. D.reindex(index(1));
  104. E.reindex(index(1));
  105. #else
  106. A.reindex(1);
  107. C.reindex(1);
  108. D.reindex(1);
  109. E.reindex(1);
  110. #endif
  111. for (size_type a = 0; a < A.shape()[0]; ++a)
  112. for (size_type b = 0; b < A.shape()[1]; ++b)
  113. for (size_type c = 0; c < A.shape()[2]; ++c) {
  114. BOOST_TEST(A[a+1][b+1][c+1] == B[a][b][c]);
  115. BOOST_TEST(C[a+1][b+1][c+1] == B[a][b][c]);
  116. BOOST_TEST(D[a+1][b+1][c+1] == B[a][b][c]);
  117. BOOST_TEST(E[a+1][b+1][c+1] == B[a][b][c]);
  118. }
  119. }
  120. return boost::report_errors();
  121. }