---

TensorArraySliceIterator.H

Go to the documentation of this file.

#ifndef _TENSORARRAYRECURSIVEITERATOR_H
#define _TENSORARRAYRECURSIVEITERATOR_H

#include <TensorArray.H>

namespace xchen
{
  template<typename T, int dims>
  class TensorArraySliceIterator
  {
    int d, s, e, i;                         // iterate each slice along 'd' starting at 's' and ending right before 'e'. 'i' keeps current index.
    TensorArray<T,dims>*ar;                 // the array to be recursively iterated.

    typedef TensorArraySliceIterator<T,dims>  MyType;
      

  public:
    TensorArraySliceIterator(TensorArray<T,dims>*array, int dir=0, int start=0, int end=0) : 
      d(dir), s(start), e(end? end:array->GetSize(dir)), ar(array) { i = start; } 

    bool PastEnd() const                         { return i >= e; } 

    MyType& operator++()                         { i++; return *this; }
    MyType& operator+=(int inc)                  { i += inc; return *this; }
    MyType  operator+(int inc)                   { MyType ret(*this); ret += inc; return ret; }
    MyType& operator--()                         { i--; return *this; }
    MyType& operator-=(int inc)                  { i -= inc; return *this; }
    MyType  operator-(int inc)                   { MyType ret(*this); ret -= inc; return ret; }

    MyType& operator=(MyType const& rhs )        { copy(rhs.Begin(), rhs.End(), Begin()); return *this; }
    MyType& operator*=(double scale)             { transform(Begin(), End(), Begin(), scale_fun(scale)); return *this; }
    MyType& operator/=(double denom)             { transform(Begin(), End(), Begin(), scale_fun(1/denom)); return *this; }
    MyType& operator+=(MyType const& rhs)        { transform(rhs.Begin(), rhs.End(), Begin(), Begin(), plus<T>()); return *this; }
    MyType& operator-=(MyType const& rhs)        { transform(rhs.Begin(), rhs.End(), Begin(), Begin(), minus<T>()); return *this; }

    TensorArray1DIterator<T,dims> Begin() const  { return ((TensorArray<T,dims>*)ar)->LinearIterate(d, i, 1).Begin(); }
    TensorArray1DIterator<T,dims> End() const    { return ((TensorArray<T,dims>*)ar)->LinearIterate(d, i, 1).End(); }

  private:
    struct scale_fun : public unary_function<const T&, double>
    {
      scale_fun(double s) : scale(s) { }
      double scale;
      T operator()(const T& x) { return x*scale; }
    };
  };


}//end namespace xchen


#endif

---