BrickExpression.hpp

Go to the documentation of this file.

/***************************************************************************
 *
 * The IPPL Framework
 * 
 * This program was prepared by PSI. 
 * All rights in the program are reserved by PSI.
 * Neither PSI nor the author(s)
 * makes any warranty, express or implied, or assumes any liability or
 * responsibility for the use of this software
 *
 ***************************************************************************/
 
 
// BrickExpLoop::apply member functions
 
template<class LHS, class RHS, class OP, unsigned Dim>
class BrickExpLoop
{
public: 
  void
  apply(LHS& __restrict__ Lhs, RHS& __restrict__ Rhs, OP Op)
    {
      int n0 = Lhs.size(0);
      int n1 = Lhs.size(1);
      int n2 = Lhs.size(2);
      if ( (n0>0)&&(n1>0)&&(n2>0) )
        {
          unsigned d;
          do
            {
              for (int i2=0; i2<n2; ++i2)
                for (int i1=0; i1<n1; ++i1)
                  for (int i0=0; i0<n0; ++i0)
                    PETE_apply(Op,Lhs.offset(i0,i1,i2),
                               for_each(Rhs,EvalFunctor_3(i0,i1,i2)));
 
              for (d=3; d<Dim; ++d)
                {
                  Lhs.step(d);
                  for_each(Rhs,StepFunctor(d),PETE_NullCombiner());
                  if ( ! Lhs.done(d) ) 
                    break;
                  Lhs.rewind(d);
                  for_each(Rhs,RewindFunctor(d),PETE_NullCombiner());
                } 
            } while (d<Dim);
        }
    }
};
 
 
//a specialization of BrickExpLoop::apply for a 1D loop evaluation
template<class LHS, class RHS, class OP>
class BrickExpLoop<LHS,RHS,OP,1U>
{
public:
  static inline void
  apply(LHS& Lhs, RHS& Rhs, OP Op)
    {
      int n0 = Lhs.size(0);
      for (int i0=0; i0<n0; ++i0) {
        PETE_apply(Op,Lhs.offset(i0),for_each(Rhs,EvalFunctor_1(i0)));
      }
    }
};
 
//a specialization of BrickExpLoops::apply for a 2D loop evaluation
template<class LHS, class RHS, class OP>
class BrickExpLoop<LHS,RHS,OP,2U>
{
public:
  static inline void
  apply(LHS& __restrict__ Lhs, RHS& __restrict__ Rhs, OP Op)
    {
      int n0 = Lhs.size(0);
      int n1 = Lhs.size(1);
      for (int i1=0; i1<n1; ++i1)
        for (int i0=0; i0<n0; ++i0)
          PETE_apply(Op,Lhs.offset(i0,i1),
                     for_each(Rhs,EvalFunctor_2(i0,i1)));
    }
};
 
 
//a specialization of BrickExpLoops::apply for a 3D loop evaluation
template<class LHS, class RHS, class OP>
class BrickExpLoop<LHS,RHS,OP,3U>
{
public:
    static inline void
    apply(LHS& Lhs, RHS& Rhs, OP Op)
    {
        int n0 = Lhs.size(0);
        int n1 = Lhs.size(1);
        int n2 = Lhs.size(2);
        for (int i2=0; i2<n2; ++i2) 
            for (int i1=0; i1<n1; ++i1) 
                for (int i0=0; i0<n0; ++i0) 
                    PETE_apply(Op,
                               Lhs.offset(i0,i1,i2),
                               for_each(Rhs,EvalFunctor_3(i0,i1,i2)));
    }
};
 
// BrickExpression::apply - just use BrickExpLoop
// Legacy code attempted to call apply() as a static method.
// This is invalid since apply() is a non-static member function.
template<unsigned Dim, class LHS, class RHS, class OP>
void BrickExpression<Dim,LHS,RHS,OP>::apply()
{
    BrickExpLoop<LHS,RHS,OP,Dim> loop;
    loop.apply(Lhs, Rhs, Op);
}