Grid/lib/qcd/action/scalar/ScalarImpl.h

#ifndef SCALAR_IMPL
#define SCALAR_IMPL


namespace Grid {
  //namespace QCD {

template <class S>
class ScalarImplTypes {
 public:
    typedef S Simd;

    template <typename vtype>
    using iImplField = iScalar<iScalar<iScalar<vtype> > >;

    typedef iImplField<Simd> SiteField;

    typedef Lattice<SiteField> Field;

    static inline void generate_momenta(Field& P, GridParallelRNG& pRNG) {
      gaussian(pRNG, P);
    }

    static inline Field projectForce(Field& P){return P;}

    static inline void update_field(Field& P, Field& U, double ep) {
      U += P*ep;
    }

    static inline RealD FieldSquareNorm(Field& U) {
      return (- sum(trace(U*U))/2.0);
    }

    static inline void HotConfiguration(GridParallelRNG &pRNG, Field &U) {
      gaussian(pRNG, U);
    }

    static inline void TepidConfiguration(GridParallelRNG &pRNG, Field &U) {
      gaussian(pRNG, U);
    }

    static inline void ColdConfiguration(GridParallelRNG &pRNG, Field &U) {
      U = 1.0;
    }

  };

  template <class S, unsigned int N>
  class ScalarAdjMatrixImplTypes {
  public:
    typedef S Simd;
    template <typename vtype>
    using iImplField = iScalar<iScalar<iMatrix<vtype, N> > >;

    typedef iImplField<Simd> SiteField;

    typedef Lattice<SiteField> Field;

    static inline void generate_momenta(Field& P, GridParallelRNG& pRNG) {
      QCD::SU<N>::GaussianFundamentalLieAlgebraMatrix(pRNG, P);
    }

    static inline Field projectForce(Field& P) {return P;}

    static inline void update_field(Field& P, Field& U, double ep) {
      U += P*ep;
    }

    static inline RealD FieldSquareNorm(Field& U) {
      return (TensorRemove(sum(trace(U*U))).real());
    }

    static inline void HotConfiguration(GridParallelRNG &pRNG, Field &U) {
      QCD::SU<N>::LieRandomize(pRNG, U);
    }

    static inline void TepidConfiguration(GridParallelRNG &pRNG, Field &U) {
      QCD::SU<N>::LieRandomize(pRNG, U, 0.01);
    }

    static inline void ColdConfiguration(GridParallelRNG &pRNG, Field &U) {
      U = zero;
    }

  };




  typedef ScalarImplTypes<vReal> ScalarImplR;
  typedef ScalarImplTypes<vRealF> ScalarImplF;
  typedef ScalarImplTypes<vRealD> ScalarImplD;

  // Hardcoding here the size of the matrices
  typedef ScalarAdjMatrixImplTypes<vComplex,  QCD::Nc> ScalarAdjImplR;
  typedef ScalarAdjMatrixImplTypes<vComplexF, QCD::Nc> ScalarAdjImplF;
  typedef ScalarAdjMatrixImplTypes<vComplexD, QCD::Nc> ScalarAdjImplD;

  template <int Colours > using ScalarNxNAdjImplR = ScalarAdjMatrixImplTypes<vComplex,   Colours >;
  template <int Colours > using ScalarNxNAdjImplF = ScalarAdjMatrixImplTypes<vComplexF,  Colours >;
  template <int Colours > using ScalarNxNAdjImplD = ScalarAdjMatrixImplTypes<vComplexD,  Colours >;
  
  //}
}

#endif