Chris@16: // Copyright 2002 The Trustees of Indiana University.
Chris@16: 
Chris@16: // Use, modification and distribution is subject to the Boost Software 
Chris@16: // License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
Chris@16: // http://www.boost.org/LICENSE_1_0.txt)
Chris@16: 
Chris@16: //  Boost.MultiArray Library
Chris@16: //  Authors: Ronald Garcia
Chris@16: //           Jeremy Siek
Chris@16: //           Andrew Lumsdaine
Chris@16: //  See http://www.boost.org/libs/multi_array for documentation.
Chris@16: 
Chris@16: #ifndef BOOST_MULTI_ARRAY_VIEW_RG071301_HPP
Chris@16: #define BOOST_MULTI_ARRAY_VIEW_RG071301_HPP
Chris@16: 
Chris@16: //
Chris@16: // view.hpp - code for creating "views" of array data.
Chris@16: //
Chris@16: 
Chris@16: #include "boost/multi_array/base.hpp"
Chris@16: #include "boost/multi_array/concept_checks.hpp"
Chris@16: #include "boost/multi_array/iterator.hpp"
Chris@16: #include "boost/multi_array/storage_order.hpp"
Chris@16: #include "boost/multi_array/subarray.hpp"
Chris@16: #include "boost/multi_array/algorithm.hpp"
Chris@16: #include "boost/type_traits/is_integral.hpp"
Chris@16: #include "boost/utility/enable_if.hpp"
Chris@16: #include "boost/array.hpp"
Chris@16: #include "boost/limits.hpp"
Chris@16: #include <algorithm>
Chris@16: #include <cstddef>
Chris@16: #include <functional>
Chris@16: #include <numeric>
Chris@16: 
Chris@16: namespace boost {
Chris@16: namespace detail {
Chris@16: namespace multi_array {
Chris@16: 
Chris@16: // TPtr = const T* defaulted in base.hpp
Chris@16: template <typename T, std::size_t NumDims, typename TPtr>
Chris@16: class const_multi_array_view :
Chris@16:     public boost::detail::multi_array::multi_array_impl_base<T,NumDims>
Chris@16: {
Chris@16:   typedef boost::detail::multi_array::multi_array_impl_base<T,NumDims> super_type;
Chris@16: public: 
Chris@16:   typedef typename super_type::value_type value_type;
Chris@16:   typedef typename super_type::const_reference const_reference;
Chris@16:   typedef typename super_type::const_iterator const_iterator;
Chris@16:   typedef typename super_type::const_reverse_iterator const_reverse_iterator;
Chris@16:   typedef typename super_type::element element;
Chris@16:   typedef typename super_type::size_type size_type;
Chris@16:   typedef typename super_type::difference_type difference_type;
Chris@16:   typedef typename super_type::index index;
Chris@16:   typedef typename super_type::extent_range extent_range;
Chris@16: 
Chris@16:   // template typedefs
Chris@16:   template <std::size_t NDims>
Chris@16:   struct const_array_view {
Chris@16:     typedef boost::detail::multi_array::const_multi_array_view<T,NDims> type;
Chris@16:   };
Chris@16: 
Chris@16:   template <std::size_t NDims>
Chris@16:   struct array_view {
Chris@16:     typedef boost::detail::multi_array::multi_array_view<T,NDims> type;
Chris@16:   };
Chris@16: 
Chris@16:   template <typename OPtr>
Chris@16:   const_multi_array_view(const 
Chris@16:                          const_multi_array_view<T,NumDims,OPtr>& other) :
Chris@16:     base_(other.base_), origin_offset_(other.origin_offset_),
Chris@16:     num_elements_(other.num_elements_), extent_list_(other.extent_list_),
Chris@16:     stride_list_(other.stride_list_), index_base_list_(other.index_base_list_)
Chris@16:   { }
Chris@16: 
Chris@16: 
Chris@16:   template <class BaseList>
Chris@16: #ifdef BOOST_NO_SFINAE
Chris@16:   void
Chris@16: #else
Chris@16:   typename
Chris@16:   disable_if<typename boost::is_integral<BaseList>::type,void >::type
Chris@16: #endif
Chris@16:   reindex(const BaseList& values) {
Chris@16:     boost::function_requires<
Chris@16:       CollectionConcept<BaseList> >();
Chris@16:     boost::detail::multi_array::
Chris@16:       copy_n(values.begin(),num_dimensions(),index_base_list_.begin());
Chris@16:     origin_offset_ =
Chris@16:       this->calculate_indexing_offset(stride_list_,index_base_list_);
Chris@16:   }
Chris@16: 
Chris@16:   void reindex(index value) {
Chris@16:     index_base_list_.assign(value);
Chris@16:     origin_offset_ =
Chris@16:       this->calculate_indexing_offset(stride_list_,index_base_list_);
Chris@16:   }
Chris@16: 
Chris@16:   size_type num_dimensions() const { return NumDims; }
Chris@16: 
Chris@16:   size_type size() const { return extent_list_.front(); }
Chris@16:   size_type max_size() const { return num_elements(); }
Chris@16:   bool empty() const { return size() == 0; }
Chris@16: 
Chris@16:   const size_type* shape() const {
Chris@16:     return extent_list_.data();
Chris@16:   }
Chris@16: 
Chris@16:   const index* strides() const {
Chris@16:     return stride_list_.data();
Chris@16:   }
Chris@16: 
Chris@16:   const T* origin() const { return base_+origin_offset_; }
Chris@16: 
Chris@16:   size_type num_elements() const { return num_elements_; }
Chris@16: 
Chris@16:   const index* index_bases() const {
Chris@16:     return index_base_list_.data();
Chris@16:   }
Chris@16: 
Chris@16:   template <typename IndexList>
Chris@16:   const element& operator()(IndexList indices) const {
Chris@16:     boost::function_requires<
Chris@16:       CollectionConcept<IndexList> >();
Chris@16:     return super_type::access_element(boost::type<const element&>(),
Chris@16:                                       indices,origin(),
Chris@16:                                       shape(),strides(),index_bases());
Chris@16:   }
Chris@16: 
Chris@16:   // Only allow const element access
Chris@16:   const_reference operator[](index idx) const {
Chris@16:     return super_type::access(boost::type<const_reference>(),
Chris@16:                               idx,origin(),
Chris@16:                               shape(),strides(),
Chris@16:                               index_bases());
Chris@16:   }
Chris@16: 
Chris@16:   // see generate_array_view in base.hpp
Chris@16:   template <int NDims>
Chris@16:   typename const_array_view<NDims>::type 
Chris@16:   operator[](const boost::detail::multi_array::
Chris@16:              index_gen<NumDims,NDims>& indices)
Chris@16:     const {
Chris@16:     typedef typename const_array_view<NDims>::type return_type;
Chris@16:     return
Chris@16:       super_type::generate_array_view(boost::type<return_type>(),
Chris@16:                                       indices,
Chris@16:                                       shape(),
Chris@16:                                       strides(),
Chris@16:                                       index_bases(),
Chris@16:                                       origin());
Chris@16:   }
Chris@16:   const_iterator begin() const {
Chris@16:     return const_iterator(*index_bases(),origin(),
Chris@16:                           shape(),strides(),index_bases());
Chris@16:   }
Chris@16: 
Chris@16:   const_iterator end() const {
Chris@16:     return const_iterator(*index_bases()+(index)*shape(),origin(),
Chris@16:                           shape(),strides(),index_bases());
Chris@16:   }
Chris@16:   
Chris@16:   const_reverse_iterator rbegin() const {
Chris@16:     return const_reverse_iterator(end());
Chris@16:   }
Chris@16: 
Chris@16:   const_reverse_iterator rend() const {
Chris@16:     return const_reverse_iterator(begin());
Chris@16:   }
Chris@16: 
Chris@16: 
Chris@16:   template <typename OPtr>
Chris@16:   bool operator==(const
Chris@16:                   const_multi_array_view<T,NumDims,OPtr>& rhs)
Chris@16:     const {
Chris@16:     if(std::equal(extent_list_.begin(),
Chris@16:                   extent_list_.end(),
Chris@16:                   rhs.extent_list_.begin()))
Chris@16:       return std::equal(begin(),end(),rhs.begin());
Chris@16:     else return false;
Chris@16:   }
Chris@16: 
Chris@16:   template <typename OPtr>
Chris@16:   bool operator<(const
Chris@16:                  const_multi_array_view<T,NumDims,OPtr>& rhs)
Chris@16:     const {
Chris@16:     return std::lexicographical_compare(begin(),end(),rhs.begin(),rhs.end());
Chris@16:   }
Chris@16: 
Chris@16:   template <typename OPtr>
Chris@16:   bool operator!=(const
Chris@16:                   const_multi_array_view<T,NumDims,OPtr>& rhs)
Chris@16:     const {
Chris@16:     return !(*this == rhs);
Chris@16:   }
Chris@16: 
Chris@16:   template <typename OPtr>
Chris@16:   bool operator>(const
Chris@16:                  const_multi_array_view<T,NumDims,OPtr>& rhs)
Chris@16:     const {
Chris@16:     return rhs < *this;
Chris@16:   }
Chris@16: 
Chris@16:   template <typename OPtr>
Chris@16:   bool operator<=(const
Chris@16:                  const_multi_array_view<T,NumDims,OPtr>& rhs)
Chris@16:     const {
Chris@16:     return !(*this > rhs);
Chris@16:   }
Chris@16: 
Chris@16:   template <typename OPtr>
Chris@16:   bool operator>=(const
Chris@16:                  const_multi_array_view<T,NumDims,OPtr>& rhs)
Chris@16:     const {
Chris@16:     return !(*this < rhs);
Chris@16:   }
Chris@16: 
Chris@16: 
Chris@16: #ifndef BOOST_NO_MEMBER_TEMPLATE_FRIENDS
Chris@16: protected:
Chris@16:   template <typename,std::size_t> friend class multi_array_impl_base;
Chris@16:   template <typename,std::size_t,typename> friend class const_multi_array_view;
Chris@16: #else
Chris@16: public: // should be protected
Chris@16: #endif
Chris@16: 
Chris@16:   // This constructor is used by multi_array_impl_base::generate_array_view
Chris@16:   // to create strides  
Chris@16:   template <typename ExtentList, typename Index>
Chris@16:   explicit const_multi_array_view(TPtr base,
Chris@16:                            const ExtentList& extents,
Chris@16:                            const boost::array<Index,NumDims>& strides): 
Chris@16:     base_(base), origin_offset_(0) {
Chris@16: 
Chris@16:     index_base_list_.assign(0);
Chris@16: 
Chris@16:     // Get the extents and strides
Chris@16:     boost::detail::multi_array::
Chris@16:       copy_n(extents.begin(),NumDims,extent_list_.begin());
Chris@16:     boost::detail::multi_array::
Chris@16:       copy_n(strides.begin(),NumDims,stride_list_.begin());
Chris@16: 
Chris@16:     // Calculate the array size
Chris@16:     num_elements_ = std::accumulate(extent_list_.begin(),extent_list_.end(),
Chris@16:                                     size_type(1),std::multiplies<size_type>());
Chris@16:   }
Chris@16: 
Chris@16:   typedef boost::array<size_type,NumDims> size_list;
Chris@16:   typedef boost::array<index,NumDims> index_list;
Chris@16: 
Chris@16:   TPtr base_;
Chris@16:   index origin_offset_;
Chris@16:   size_type num_elements_;
Chris@16:   size_list extent_list_;
Chris@16:   index_list stride_list_;
Chris@16:   index_list index_base_list_;
Chris@16: 
Chris@16: private:
Chris@16:   // const_multi_array_view cannot be assigned to (no deep copies!)
Chris@16:   const_multi_array_view& operator=(const const_multi_array_view& other);
Chris@16: };
Chris@16: 
Chris@16: 
Chris@16: template <typename T, std::size_t NumDims>
Chris@16: class multi_array_view :
Chris@16:   public const_multi_array_view<T,NumDims,T*>
Chris@16: {
Chris@16:   typedef const_multi_array_view<T,NumDims,T*> super_type;
Chris@16: public: 
Chris@16:   typedef typename super_type::value_type value_type;
Chris@16:   typedef typename super_type::reference reference;
Chris@16:   typedef typename super_type::iterator iterator;
Chris@16:   typedef typename super_type::reverse_iterator reverse_iterator;
Chris@16:   typedef typename super_type::const_reference const_reference;
Chris@16:   typedef typename super_type::const_iterator const_iterator;
Chris@16:   typedef typename super_type::const_reverse_iterator const_reverse_iterator;
Chris@16:   typedef typename super_type::element element;
Chris@16:   typedef typename super_type::size_type size_type;
Chris@16:   typedef typename super_type::difference_type difference_type;
Chris@16:   typedef typename super_type::index index;
Chris@16:   typedef typename super_type::extent_range extent_range;
Chris@16: 
Chris@16:   // template typedefs
Chris@16:   template <std::size_t NDims>
Chris@16:   struct const_array_view {
Chris@16:     typedef boost::detail::multi_array::const_multi_array_view<T,NDims> type;
Chris@16:   };
Chris@16: 
Chris@16:   template <std::size_t NDims>
Chris@16:   struct array_view {
Chris@16:     typedef boost::detail::multi_array::multi_array_view<T,NDims> type;
Chris@16:   };
Chris@16: 
Chris@16:   // Assignment from other ConstMultiArray types.
Chris@16:   template <typename ConstMultiArray>
Chris@16:   multi_array_view& operator=(const ConstMultiArray& other) {
Chris@16:     function_requires< 
Chris@16:       boost::multi_array_concepts::
Chris@16:       ConstMultiArrayConcept<ConstMultiArray,NumDims> >();
Chris@16: 
Chris@16:     // make sure the dimensions agree
Chris@16:     BOOST_ASSERT(other.num_dimensions() == this->num_dimensions());
Chris@16:     BOOST_ASSERT(std::equal(other.shape(),other.shape()+this->num_dimensions(),
Chris@16:                             this->shape()));
Chris@16:     // iterator-based copy
Chris@16:     std::copy(other.begin(),other.end(),begin());
Chris@16:     return *this;
Chris@16:   }
Chris@16: 
Chris@16: 
Chris@16:   multi_array_view& operator=(const multi_array_view& other) {
Chris@16:     if (&other != this) {
Chris@16:       // make sure the dimensions agree
Chris@16:       BOOST_ASSERT(other.num_dimensions() == this->num_dimensions());
Chris@16:       BOOST_ASSERT(std::equal(other.shape(),
Chris@16:                               other.shape()+this->num_dimensions(),
Chris@16:                               this->shape()));
Chris@16:       // iterator-based copy
Chris@16:       std::copy(other.begin(),other.end(),begin());
Chris@16:     }
Chris@16:     return *this;
Chris@16:   }
Chris@16: 
Chris@16:   element* origin() { return this->base_+this->origin_offset_; }
Chris@16: 
Chris@16:   template <class IndexList>
Chris@16:   element& operator()(const IndexList& indices) {
Chris@16:     boost::function_requires<
Chris@16:       CollectionConcept<IndexList> >();
Chris@16:     return super_type::access_element(boost::type<element&>(),
Chris@16:                                       indices,origin(),
Chris@16:                                       this->shape(),this->strides(),
Chris@16:                                       this->index_bases());
Chris@16:   }
Chris@16: 
Chris@16: 
Chris@16:   reference operator[](index idx) {
Chris@16:     return super_type::access(boost::type<reference>(),
Chris@16:                               idx,origin(),
Chris@16:                               this->shape(),this->strides(),
Chris@16:                               this->index_bases());
Chris@16:   }
Chris@16: 
Chris@16: 
Chris@16:   // see generate_array_view in base.hpp
Chris@16:   template <int NDims>
Chris@16:   typename array_view<NDims>::type 
Chris@16:   operator[](const boost::detail::multi_array::
Chris@16:              index_gen<NumDims,NDims>& indices) {
Chris@16:     typedef typename array_view<NDims>::type return_type;
Chris@16:     return
Chris@16:       super_type::generate_array_view(boost::type<return_type>(),
Chris@16:                                       indices,
Chris@16:                                       this->shape(),
Chris@16:                                       this->strides(),
Chris@16:                                       this->index_bases(),
Chris@16:                                       origin());
Chris@16:   }
Chris@16:   
Chris@16:   
Chris@16:   iterator begin() {
Chris@16:     return iterator(*this->index_bases(),origin(),
Chris@16:                     this->shape(),this->strides(),
Chris@16:                     this->index_bases());
Chris@16:   }
Chris@16: 
Chris@16:   iterator end() {
Chris@16:     return iterator(*this->index_bases()+(index)*this->shape(),origin(),
Chris@16:                     this->shape(),this->strides(),
Chris@16:                     this->index_bases());
Chris@16:   }
Chris@16: 
Chris@16:   reverse_iterator rbegin() {
Chris@16:     return reverse_iterator(end());
Chris@16:   }
Chris@16: 
Chris@16:   reverse_iterator rend() {
Chris@16:     return reverse_iterator(begin());
Chris@16:   }
Chris@16: 
Chris@16:   // Using declarations don't seem to work for g++
Chris@16:   // These are the proxies to work around this.
Chris@16: 
Chris@16:   const element* origin() const { return super_type::origin(); }
Chris@16: 
Chris@16:   template <class IndexList>
Chris@16:   const element& operator()(const IndexList& indices) const {
Chris@16:     boost::function_requires<
Chris@16:       CollectionConcept<IndexList> >();
Chris@16:     return super_type::operator()(indices);
Chris@16:   }
Chris@16: 
Chris@16:   const_reference operator[](index idx) const {
Chris@16:     return super_type::operator[](idx);
Chris@16:   }
Chris@16: 
Chris@16:   // see generate_array_view in base.hpp
Chris@16:   template <int NDims>
Chris@16:   typename const_array_view<NDims>::type 
Chris@16:   operator[](const boost::detail::multi_array::
Chris@16:              index_gen<NumDims,NDims>& indices)
Chris@16:     const {
Chris@16:     return super_type::operator[](indices);
Chris@16:   }
Chris@16:   
Chris@16:   const_iterator begin() const {
Chris@16:     return super_type::begin();
Chris@16:   }
Chris@16: 
Chris@16:   const_iterator end() const {
Chris@16:     return super_type::end();
Chris@16:   }
Chris@16: 
Chris@16:   const_reverse_iterator rbegin() const {
Chris@16:     return super_type::rbegin();
Chris@16:   }
Chris@16: 
Chris@16:   const_reverse_iterator rend() const {
Chris@16:     return super_type::rend();
Chris@16:   }
Chris@16: 
Chris@16: #ifndef BOOST_NO_MEMBER_TEMPLATE_FRIENDS
Chris@16: private:
Chris@16:   template <typename,std::size_t> friend class multi_array_impl_base;
Chris@16: #else
Chris@16: public: // should be private
Chris@16: #endif
Chris@16: 
Chris@16:   // constructor used by multi_array_impl_base::generate_array_view to
Chris@16:   // generate array views
Chris@16:   template <typename ExtentList, typename Index>
Chris@16:   explicit multi_array_view(T* base,
Chris@16:                             const ExtentList& extents,
Chris@16:                             const boost::array<Index,NumDims>& strides) :
Chris@16:     super_type(base,extents,strides) { }
Chris@16: 
Chris@16: };
Chris@16: 
Chris@16: } // namespace multi_array
Chris@16: } // namespace detail
Chris@16: 
Chris@16: //
Chris@16: // traits classes to get array_view types
Chris@16: //
Chris@16: template <typename Array, int N>
Chris@16: class array_view_gen {
Chris@16:   typedef typename Array::element element;
Chris@16: public:
Chris@16:   typedef boost::detail::multi_array::multi_array_view<element,N> type;
Chris@16: };
Chris@16: 
Chris@16: template <typename Array, int N>
Chris@16: class const_array_view_gen {
Chris@16:   typedef typename Array::element element;
Chris@16: public:
Chris@16:   typedef boost::detail::multi_array::const_multi_array_view<element,N> type;  
Chris@16: };
Chris@16: 
Chris@16: } // namespace boost
Chris@16: 
Chris@16: #endif // BOOST_MULTI_ARRAY_VIEW_RG071301_HPP
Chris@16: