dimtest.cpp 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312
  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. // Trying to diagnose problems under visual
  12. #include <boost/config.hpp>
  13. #include <boost/array.hpp>
  14. #include <boost/limits.hpp>
  15. #include <algorithm>
  16. #include <utility>
  17. typedef int index;
  18. typedef std::size_t size_type;
  19. template <typename Index,typename SizeType>
  20. class index_range {
  21. public:
  22. index_range()
  23. {
  24. start_ = from_start();
  25. finish_ = to_end();
  26. stride_ = 1;
  27. degenerate_ = false;
  28. }
  29. explicit index_range(Index pos)
  30. {
  31. start_ = pos;
  32. finish_ = pos;
  33. stride_ = 1;
  34. degenerate_ = true;
  35. }
  36. explicit index_range(Index start, Index finish, Index stride=1)
  37. : start_(start), finish_(finish), stride_(stride),
  38. degenerate_(start_ == finish_)
  39. { }
  40. // These are for chaining assignments to an index_range
  41. index_range& start(Index s) {
  42. start_ = s;
  43. degenerate_ = (start_ == finish_);
  44. return *this;
  45. }
  46. index_range& finish(Index f) {
  47. finish_ = f;
  48. degenerate_ = (start_ == finish_);
  49. return *this;
  50. }
  51. index_range& stride(Index s) { stride_ = s; return *this; }
  52. Index start() const
  53. {
  54. return start_;
  55. }
  56. Index get_start(Index low_index_range = 0) const
  57. {
  58. if (start_ == from_start())
  59. return low_index_range;
  60. return start_;
  61. }
  62. Index finish() const
  63. {
  64. return finish_;
  65. }
  66. Index get_finish(Index high_index_range = 0) const
  67. {
  68. if (finish_ == to_end())
  69. return high_index_range;
  70. return finish_;
  71. }
  72. unsigned int size(Index recommended_length = 0) const
  73. {
  74. if ((start_ == from_start()) || (finish_ == to_end()))
  75. return recommended_length;
  76. else
  77. return (finish_ - start_) / stride_;
  78. }
  79. Index stride() const { return stride_; }
  80. bool is_ascending_contiguous() const
  81. {
  82. return (start_ < finish_) && is_unit_stride();
  83. }
  84. void set_index_range(Index start, Index finish, Index stride=1)
  85. {
  86. start_ = start;
  87. finish_ = finish;
  88. stride_ = stride;
  89. }
  90. static index_range all()
  91. { return index_range(from_start(), to_end(), 1); }
  92. bool is_unit_stride() const
  93. { return stride_ == 1; }
  94. bool is_degenerate() const { return degenerate_; }
  95. index_range operator-(Index shift) const
  96. {
  97. return index_range(start_ - shift, finish_ - shift, stride_);
  98. }
  99. index_range operator+(Index shift) const
  100. {
  101. return index_range(start_ + shift, finish_ + shift, stride_);
  102. }
  103. Index operator[](unsigned i) const
  104. {
  105. return start_ + i * stride_;
  106. }
  107. Index operator()(unsigned i) const
  108. {
  109. return start_ + i * stride_;
  110. }
  111. // add conversion to std::slice?
  112. private:
  113. static Index from_start()
  114. { return (std::numeric_limits<Index>::min)(); }
  115. static Index to_end()
  116. { return (std::numeric_limits<Index>::max)(); }
  117. public:
  118. Index start_, finish_, stride_;
  119. bool degenerate_;
  120. };
  121. // Express open and closed interval end-points using the comparison
  122. // operators.
  123. // left closed
  124. template <typename Index, typename SizeType>
  125. inline index_range<Index,SizeType>
  126. operator<=(Index s, const index_range<Index,SizeType>& r)
  127. {
  128. return index_range<Index,SizeType>(s, r.finish(), r.stride());
  129. }
  130. // left open
  131. template <typename Index, typename SizeType>
  132. inline index_range<Index,SizeType>
  133. operator<(Index s, const index_range<Index,SizeType>& r)
  134. {
  135. return index_range<Index,SizeType>(s + 1, r.finish(), r.stride());
  136. }
  137. // right open
  138. template <typename Index, typename SizeType>
  139. inline index_range<Index,SizeType>
  140. operator<(const index_range<Index,SizeType>& r, Index f)
  141. {
  142. return index_range<Index,SizeType>(r.start(), f, r.stride());
  143. }
  144. // right closed
  145. template <typename Index, typename SizeType>
  146. inline index_range<Index,SizeType>
  147. operator<=(const index_range<Index,SizeType>& r, Index f)
  148. {
  149. return index_range<Index,SizeType>(r.start(), f + 1, r.stride());
  150. }
  151. //
  152. // range_list.hpp - helper to build boost::arrays for *_set types
  153. //
  154. /////////////////////////////////////////////////////////////////////////
  155. // choose range list begins
  156. //
  157. struct choose_range_list_n {
  158. template <typename T, std::size_t NumRanges>
  159. struct bind {
  160. typedef boost::array<T,NumRanges> type;
  161. };
  162. };
  163. struct choose_range_list_zero {
  164. template <typename T, std::size_t NumRanges>
  165. struct bind {
  166. typedef boost::array<T,1> type;
  167. };
  168. };
  169. template <std::size_t NumRanges>
  170. struct range_list_gen_helper {
  171. typedef choose_range_list_n choice;
  172. };
  173. template <>
  174. struct range_list_gen_helper<0> {
  175. typedef choose_range_list_zero choice;
  176. };
  177. template <typename T, std::size_t NumRanges>
  178. struct range_list_generator {
  179. private:
  180. typedef typename range_list_gen_helper<NumRanges>::choice Choice;
  181. public:
  182. typedef typename Choice::template bind<T,NumRanges>::type type;
  183. };
  184. //
  185. // choose range list ends
  186. /////////////////////////////////////////////////////////////////////////
  187. //
  188. // Index_gen.hpp stuff
  189. //
  190. template <int NumRanges, int NumDims>
  191. struct index_gen {
  192. private:
  193. typedef index Index;
  194. typedef size_type SizeType;
  195. typedef index_range<Index,SizeType> range;
  196. public:
  197. typedef typename range_list_generator<range,NumRanges>::type range_list;
  198. range_list ranges_;
  199. index_gen() { }
  200. template <int ND>
  201. explicit index_gen(const index_gen<NumRanges-1,ND>& rhs,
  202. const index_range<Index,SizeType>& range)
  203. {
  204. std::copy(rhs.ranges_.begin(),rhs.ranges_.end(),ranges_.begin());
  205. *ranges_.rbegin() = range;
  206. }
  207. index_gen<NumRanges+1,NumDims+1>
  208. operator[](const index_range<Index,SizeType>& range) const
  209. {
  210. index_gen<NumRanges+1,NumDims+1> tmp;
  211. std::copy(ranges_.begin(),ranges_.end(),tmp.ranges_.begin());
  212. *tmp.ranges_.rbegin() = range;
  213. return tmp;
  214. }
  215. index_gen<NumRanges+1,NumDims>
  216. operator[](Index idx) const
  217. {
  218. index_gen<NumRanges+1,NumDims> tmp;
  219. std::copy(ranges_.begin(),ranges_.end(),tmp.ranges_.begin());
  220. *tmp.ranges_.rbegin() = index_range<Index,SizeType>(idx);
  221. return tmp;
  222. }
  223. };
  224. template <int NDims, int NRanges>
  225. void accept_gen(index_gen<NRanges,NDims>& indices) {
  226. // do nothing
  227. }
  228. template <typename X, typename Y, int A, int B>
  229. class foo { };
  230. class boo {
  231. template <int NDims, int NRanges>
  232. void operator[](index_gen<NRanges,NDims>& indices) {
  233. }
  234. };
  235. template <typename X, typename Y, int A1, int A2>
  236. void take_foo(foo<X,Y,A1,A2>& f) { }
  237. int main() {
  238. index_gen<0,0> indices;
  239. typedef index_range<index,size_type> range;
  240. take_foo(foo<int,std::size_t,1,2>());
  241. indices[range()][range()][range()];
  242. accept_gen(indices);
  243. accept_gen(index_gen<0,0>());
  244. accept_gen(indices[range()][range()][range()]);
  245. boo b;
  246. b[indices[range()][range()][range()]];
  247. return 0;
  248. }