mirror of
https://github.com/paboyle/Grid.git
synced 2024-09-20 09:15:38 +01:00
8f5281563e
assignment. LatticeCoordinate helper to get global (reduced) coordinate. Some more work of similar type perhaps needed, but the bulk of the required structure for masked array assignment is now in place.
123 lines
4.2 KiB
C++
123 lines
4.2 KiB
C++
#ifndef GRID_QCD_H
|
|
#define GRID_QCD_H
|
|
namespace Grid{
|
|
namespace QCD {
|
|
|
|
static const int Nc=3;
|
|
static const int Ns=4;
|
|
static const int CbRed =0;
|
|
static const int CbBlack=1;
|
|
|
|
// QCD iMatrix types
|
|
template<typename vtype> using iSinglet = iScalar<iScalar<vtype> > ;
|
|
template<typename vtype> using iSpinMatrix = iMatrix<iScalar<vtype>, Ns>;
|
|
template<typename vtype> using iSpinColourMatrix = iMatrix<iMatrix<vtype, Nc>, Ns>;
|
|
|
|
template<typename vtype> using iColourMatrix = iScalar<iMatrix<vtype, Nc>> ;
|
|
|
|
template<typename vtype> using iSpinVector = iVector<iScalar<vtype>, Ns>;
|
|
template<typename vtype> using iColourVector = iScalar<iVector<vtype, Nc> >;
|
|
template<typename vtype> using iSpinColourVector = iVector<iVector<vtype, Nc>, Ns>;
|
|
|
|
typedef iSinglet<Complex > TComplex; // This is painful. Tensor singlet complex type.
|
|
typedef iSinglet<vComplex > vTComplex;
|
|
typedef iSinglet<Real > TReal; // This is painful. Tensor singlet complex type.
|
|
|
|
|
|
typedef iSinglet<vInteger > vTInteger;
|
|
|
|
typedef iSpinMatrix<Complex > SpinMatrix;
|
|
typedef iColourMatrix<Complex > ColourMatrix;
|
|
typedef iSpinColourMatrix<Complex > SpinColourMatrix;
|
|
|
|
typedef iSpinVector<Complex > SpinVector;
|
|
typedef iColourVector<Complex > ColourVector;
|
|
typedef iSpinColourVector<Complex > SpinColourVector;
|
|
|
|
|
|
typedef iSpinMatrix<vComplex > vSpinMatrix;
|
|
typedef iColourMatrix<vComplex > vColourMatrix;
|
|
typedef iSpinColourMatrix<vComplex > vSpinColourMatrix;
|
|
|
|
typedef iSpinVector<vComplex > vSpinVector;
|
|
typedef iColourVector<vComplex > vColourVector;
|
|
typedef iSpinColourVector<vComplex > vSpinColourVector;
|
|
|
|
typedef Lattice<vTComplex> LatticeComplex;
|
|
|
|
typedef Lattice<vInteger> LatticeInteger; // Predicates for "where"
|
|
|
|
typedef Lattice<vColourMatrix> LatticeColourMatrix;
|
|
typedef Lattice<vSpinMatrix> LatticeSpinMatrix;
|
|
typedef Lattice<vSpinColourMatrix> LatticePropagator;
|
|
typedef LatticePropagator LatticeSpinColourMatrix;
|
|
|
|
typedef Lattice<vSpinColourVector> LatticeFermion;
|
|
typedef Lattice<vSpinColourVector> LatticeSpinColourVector;
|
|
typedef Lattice<vSpinVector> LatticeSpinVector;
|
|
typedef Lattice<vColourVector> LatticeColourVector;
|
|
|
|
// localNorm2,
|
|
template<class tt>
|
|
inline LatticeComplex localNorm2 (const Lattice<tt> &rhs)
|
|
{
|
|
LatticeComplex ret(rhs._grid);
|
|
#pragma omp parallel for
|
|
for(int ss=0;ss<rhs._grid->oSites(); ss++){
|
|
ret._odata[ss]=trace(adj(rhs)*rhs);
|
|
}
|
|
return ret;
|
|
}
|
|
// localInnerProduct
|
|
template<class tt>
|
|
inline LatticeComplex localInnerProduct (const Lattice<tt> &lhs,const Lattice<tt> &rhs)
|
|
{
|
|
LatticeComplex ret(rhs._grid);
|
|
#pragma omp parallel for
|
|
for(int ss=0;ss<rhs._grid->oSites(); ss++){
|
|
ret._odata[ss]=localInnerProduct(lhs._odata[ss],rhs._odata[ss]);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
// outerProduct Scalar x Scalar -> Scalar
|
|
// Vector x Vector -> Matrix
|
|
template<class ll,class rr>
|
|
inline auto outerProduct (const Lattice<ll> &lhs,const Lattice<rr> &rhs) -> Lattice<decltype(outerProduct(lhs._odata[0],rhs._odata[0]))>
|
|
{
|
|
Lattice<decltype(outerProduct(lhs._odata[0],rhs._odata[0]))> ret(rhs._grid);
|
|
#pragma omp parallel for
|
|
for(int ss=0;ss<rhs._grid->oSites(); ss++){
|
|
ret._odata[ss]=outerProduct(lhs._odata[ss],rhs._odata[ss]);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
// FIXME for debug; deprecate this
|
|
inline void LatticeCoordinate(LatticeInteger &l,int mu){
|
|
GridBase *grid = l._grid;
|
|
int Nsimd = grid->iSites();
|
|
std::vector<int> gcoor;
|
|
std::vector<Integer> mergebuf(Nsimd);
|
|
std::vector<Integer *> mergeptr(Nsimd);
|
|
for(int o=0;o<grid->oSites();o++){
|
|
for(int i=0;i<grid->iSites();i++){
|
|
// RankIndexToGlobalCoor(grid->ThisRank(),o,i,gcoor);
|
|
grid->RankIndexToGlobalCoor(0,o,i,gcoor);
|
|
mergebuf[i]=gcoor[mu];
|
|
mergeptr[i]=&mergebuf[i];
|
|
}
|
|
merge(l._odata[o],mergeptr);
|
|
}
|
|
};
|
|
|
|
#include <Grid_predicated.h>
|
|
|
|
#if 0
|
|
|
|
#endif
|
|
|
|
} //namespace QCD
|
|
} // Grid
|
|
#endif
|