stridedstepper.h

Go to the documentation of this file.

// include guard
#ifndef AFF_STRIDEDSTEPPER_H_VERSION

#define AFF_STRIDEDSTEPPER_H_VERSION \
  "AFF_STRIDEDSTEPPER,H_H   V1.3"

#include<aff/lib/strided.h>

namespace aff {
namespace util {

class StridedStepper 
{
  public:
    typedef aff::Strided Tshape;
    typedef Tshape::TIndexVec TIndexVec;
    typedef Tshape::TSizeVec TSizeVec;
    //
    StridedStepper(const Strided& strided);
    const Tsubscript& current() const { return(Mcurrent); }
    const TIndexVec& index() const { return(Mindex); }
    const Tsubscript& index(const Tdim& i) const { return(Mindex[i]); }
    StridedStepper& incr();
    StridedStepper& decr();
    bool more() const
    { return(Mcurrent<Mlast_offset); }
    bool less() const
    { return(Mcurrent>Mfirst_offset); }
    const bool& valid() const
    { return(Mvalid); }
    StridedStepper& tofirst()
    {
      Mindex=Mshape.first();
      Mcurrent=Mshape.first_offset();
      Mvalid=true;
      return(*this);
    }
    StridedStepper& tolast()
    {
      Mindex=Mshape.last();
      Mcurrent=Mshape.last_offset();
      Mvalid=true;
      return(*this);
    }
  private:
    const Tshape& Mshape;
    Tsubscript Mcurrent;
    TIndexVec Mindex;
    TSizeVec Mcarry;
    bool Mvalid;
    Tsubscript Mfirst_offset, Mlast_offset;

}; // StridedStepper

} // namespace util

} // namespace aff

#endif // AFF_STRIDEDSTEPPER_H_VERSION (includeguard)

/* ----- END OF stridedstepper.h ----- */