williamr@2: // Copyright 2002 The Trustees of Indiana University.
williamr@2: 
williamr@2: // Use, modification and distribution is subject to the Boost Software 
williamr@2: // License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
williamr@2: // http://www.boost.org/LICENSE_1_0.txt)
williamr@2: 
williamr@2: //  Boost.MultiArray Library
williamr@2: //  Authors: Ronald Garcia
williamr@2: //           Jeremy Siek
williamr@2: //           Andrew Lumsdaine
williamr@2: //  See http://www.boost.org/libs/multi_array for documentation.
williamr@2: 
williamr@2: #ifndef BOOST_INDEX_GEN_RG071801_HPP
williamr@2: #define BOOST_INDEX_GEN_RG071801_HPP
williamr@2: 
williamr@2: #include "boost/array.hpp"
williamr@2: #include "boost/multi_array/index_range.hpp"
williamr@2: #include "boost/multi_array/range_list.hpp"
williamr@2: #include "boost/multi_array/types.hpp"
williamr@2: #include <algorithm> 
williamr@2: #include <cstddef>
williamr@2: 
williamr@2: namespace boost {
williamr@2: namespace detail {
williamr@2: namespace multi_array {
williamr@2: 
williamr@2: 
williamr@2: template <int NumRanges, int NumDims>
williamr@2: struct index_gen {
williamr@2: private:
williamr@2:   typedef ::boost::detail::multi_array::index index;
williamr@2:   typedef ::boost::detail::multi_array::size_type size_type;
williamr@2:   typedef index_range<index,size_type> range;
williamr@2: public:
williamr@2:   template <int Dims, int Ranges>
williamr@2:   struct gen_type {
williamr@2:     typedef index_gen<Ranges,Dims> type;
williamr@2:   };
williamr@2: 
williamr@2:   typedef typename range_list_generator<range,NumRanges>::type range_list;
williamr@2:   range_list ranges_;
williamr@2: 
williamr@2:   index_gen() { }
williamr@2: 
williamr@2:   template <int ND>
williamr@2:   explicit index_gen(const index_gen<NumRanges-1,ND>& rhs,
williamr@2:             const range& r)
williamr@2:   {
williamr@2:     std::copy(rhs.ranges_.begin(),rhs.ranges_.end(),ranges_.begin());
williamr@2:     *ranges_.rbegin() = r;
williamr@2:   }
williamr@2: 
williamr@2:   index_gen<NumRanges+1,NumDims+1>
williamr@2:   operator[](const range& r) const
williamr@2:   {
williamr@2:     index_gen<NumRanges+1,NumDims+1> tmp;
williamr@2:     std::copy(ranges_.begin(),ranges_.end(),tmp.ranges_.begin());
williamr@2:     *tmp.ranges_.rbegin() = r;
williamr@2:     return tmp;
williamr@2:   }
williamr@2: 
williamr@2:   index_gen<NumRanges+1,NumDims>
williamr@2:   operator[](index idx) const
williamr@2:   {
williamr@2:     index_gen<NumRanges+1,NumDims> tmp;
williamr@2:     std::copy(ranges_.begin(),ranges_.end(),tmp.ranges_.begin());
williamr@2:     *tmp.ranges_.rbegin() = range(idx);
williamr@2:     return tmp;
williamr@2:   }    
williamr@2: 
williamr@2:   static index_gen<0,0> indices() {
williamr@2:     return index_gen<0,0>();
williamr@2:   }
williamr@2: };
williamr@2: 
williamr@2: } // namespace multi_array
williamr@2: } // namespace detail
williamr@2: } // namespace boost
williamr@2: 
williamr@2: 
williamr@2: #endif // BOOST_INDEX_GEN_RG071801_HPP