shaper.h

Go to the documentation of this file.

// include guard
#ifndef AFF_SHAPER_H_VERSION

#define AFF_SHAPER_H_VERSION \
  "AFF_SHAPER_H   V1.6"

#include<aff/lib/strided.h>
#include<aff/lib/error.h>

namespace aff {

class Shaper 
{
  public:
    typedef Strided Tshape;
    Shaper(const Tsubscript& last): 
      Mfirst(1), Mlast(1), Mmaxlast(1), Mdim(1)
    {
      AFF_assert((last>=1), "ERROR (shaper): invalid limit");
      Mfirst[0]=1;
      Mlast[0]=last;
      Mmaxlast[0]=last;
    }
    Shaper(const Tsubscript& first, const Tsubscript& last):
      Mfirst(1), Mlast(1), Mmaxlast(1), Mdim(1)
    {
      AFF_assert((last>=first), "ERROR (shaper): invalid index range");
      Mfirst[0]=first;
      Mlast[0]=last;
      Mmaxlast[0]=last;
    }
    Shaper(const Tsubscript& first, const Tsubscript& last,
           const Tsubscript& maxlast):
      Mfirst(1), Mlast(1), Mmaxlast(1), Mdim(1)
    {
      AFF_assert((last>=first), "ERROR (shaper): invalid index range");
      AFF_assert((maxlast>=last), "ERROR (shaper): invalid alloc size");
      Mfirst[0]=first;
      Mlast[0]=last;
      Mmaxlast[0]=maxlast;
    }
    Shaper& operator() (const Tsubscript& last)
    {
      AFF_assert((last>=1), "ERROR (shaper): invalid limit");
      check_dim();
      Mfirst[Mdim]=1;
      Mlast[Mdim]=last;
      Mmaxlast[Mdim]=last;
      Mdim++;
      return(*this);
    }
    Shaper& operator() (const Tsubscript& first, const Tsubscript& last)
    {
      AFF_assert((last>=first), "ERROR (shaper): invalid index range");
      check_dim();
      Mfirst[Mdim]=first;
      Mlast[Mdim]=last;
      Mmaxlast[Mdim]=last;
      Mdim++;
      return(*this);
    }
    Shaper& operator() (const Tsubscript& first, const Tsubscript& last,
                        const Tsubscript& maxlast)
    {
      AFF_assert((last>=first), "ERROR (shaper): invalid index range");
      AFF_assert((maxlast>=last), "ERROR (shaper): invalid alloc size");
      check_dim();
      Mfirst[Mdim]=first;
      Mlast[Mdim]=last;
      Mmaxlast[Mdim]=maxlast;
      Mdim++;
      return(*this);
    }
    operator Tshape () const
    {
      Tshape shape(Mfirst, Mmaxlast);
      shape.shrink(Mlast);
      return(shape);
    }
  private:
    void check_dim() const
    {
      AFF_assert((Mdim<Tshape::Mmax_dimen),
                 "ERROR (Shaper): you use too many dimensions!");
    }
    Tshape::TIndexVec Mfirst, Mlast, Mmaxlast;
    Tdim Mdim;
}; // Shaper

} // namespace aff

#endif // AFF_SHAPER_H_VERSION (includeguard)

/* ----- END OF shaper.h ----- */