boost/boost/multi_array/extent_gen.hpp

   1 #ifndef BOOST_EXTENT_GEN_RG071801_HPP
   2 #define BOOST_EXTENT_GEN_RG071801_HPP
   3
   4 #include "boost/multi_array/extent_range.hpp"
   5 #include "boost/multi_array/range_list.hpp"
   6 #include "boost/multi_array/types.hpp"
   7 #include "boost/array.hpp"
   8 #include <algorithm>
   9
  10 namespace boost {
  11 namespace detail {
  12 namespace multi_array {
  13
  14
  15 template <std::size_t NumRanges>
  16 class extent_gen {
  17 public:
  18   typedef boost::detail::multi_array::index index;
  19   typedef boost::detail::multi_array::size_type size_type;
  20 private:
  21   typedef extent_range<index,size_type> range;
  22   typedef typename range_list_generator<range,NumRanges>::type range_list;
  23 public:
  24   template <std::size_t Ranges>
  25   struct gen_type {
  26     typedef extent_gen<Ranges> type;
  27   };
  28
  29   range_list ranges_;
  30
  31   extent_gen() { }
  32
  33   // Used by operator[] to expand extent_gens
  34   extent_gen(const extent_gen<NumRanges-1>& rhs,
  35             const range& a_range)
  36   {
  37     std::copy(rhs.ranges_.begin(),rhs.ranges_.end(),ranges_.begin());
  38     *ranges_.rbegin() = a_range;
  39   }
  40
  41   extent_gen<NumRanges+1>
  42   operator[](const range& a_range)
  43   {
  44     return extent_gen<NumRanges+1>(*this,a_range);
  45   }
  46
  47   extent_gen<NumRanges+1>
  48   operator[](index idx)
  49   {
  50     return extent_gen<NumRanges+1>(*this,range(0,idx));
  51   }
  52
  53   static extent_gen<0> extents() {
  54     return extent_gen<0>();
  55   }
  56 };
  57
  58 } // namespace multi_array
  59 } // namespace detail
  60 } // namespace boost
  61
  62
  63 #endif // BOOST_EXTENT_GEN_RG071801_HPP