subarray.h

Go to the documentation of this file.

// include guard
#ifndef AFF_SUBARRAY_H_VERSION

#define AFF_SUBARRAY_H_VERSION \
  "AFF_SUBARRAY_H   V1.3"

#include<aff/array.h>

namespace aff {
namespace util {

  template<class C>
    class Subarray
    {
      public:
        typedef C Tarray;
        typedef C Tcontainer;
        typedef typename Tarray::Tconst_value Tconst_value;
        typedef typename Tarray::Tconst_reference Tconst_reference;
        typedef typename Tarray::Tconst_pointer Tconst_pointer;
        typedef typename Tarray::Tshape Tshape;
        typedef typename Tarray::Trepresentation Trepresentation;
        Subarray(const Tarray& array)
        {
          // notice: a const Array does not return its representation
          // thus we create a local (non-const) copy
          // that's truely inefficient (copying the shape twice)
          Tarray copy=array;
          Mshape=copy.shape();
          Mrepresentation=copy.representation();
          Mdim=0;
        }
        Subarray& operator()() 
        { ++Mdim; return(*this); }
        Subarray& operator()(const Tsubscript& last) 
        { 
          this->check_dim();
          Mshape.shrink(Mdim, last);
          ++Mdim;
          return(*this); 
        }
        Subarray& operator()(const Tsubscript& first,
                             const Tsubscript& last) 
        { 
          this->check_dim();
          Mshape.shrink(Mdim, first, last);
          ++Mdim;
          return(*this); 
        }
        Subarray& operator=(Tconst_reference value)
        {
          typename Tshape::Tstepper st(Mshape);
          for(st.tofirst(); st.valid(); st.incr())
          { Mrepresentation[st.current()]=value; } 
          return(*this);
        }
        Tarray array() const
        { return(Tarray(Mshape,Mrepresentation)); }
        operator Tarray() const
        { return(Tarray(Mshape,Mrepresentation)); }
      private:
        void check_dim() const
        {
          AFF_assert((Mdim<Tshape::Mmax_dimen),
                     "ERROR (Subarray): you use too many dimension-brackets!");
        }
        Tshape Mshape;
        Trepresentation Mrepresentation;
        Tdim Mdim;
    }; // class Subarray

} // namespace util

/*----------------------------------------------------------------------*/

template<class C>
inline
aff::util::Subarray<C> subarray(const C& c) 
{ return(aff::util::Subarray<C>(c)); }

} // namespace aff

#endif // AFF_SUBARRAY_H_VERSION (includeguard)

/* ----- END OF subarray.h ----- */