Large scale change to support 5d fermion formulations.

Have 5d replicated wilson with 4d gauge working and matrix regressing
to Ls copies of wilson.

lib/qcd/action/fermion/WilsonFermion.cc

@@ -0,0 +1,163 @@
+#include <Grid.h>
+
+namespace Grid {
+namespace QCD {
+
+const std::vector<int> WilsonFermion::directions   ({0,1,2,3, 0, 1, 2, 3});
+const std::vector<int> WilsonFermion::displacements({1,1,1,1,-1,-1,-1,-1});
+
+int WilsonFermion::HandOptDslash;
+
+WilsonFermion::WilsonFermion(LatticeGaugeField &_Umu,
+			   GridCartesian         &Fgrid,
+			   GridRedBlackCartesian &Hgrid, 
+			   double _mass) :
+  _grid(&Fgrid),
+  _cbgrid(&Hgrid),
+  Stencil    (&Fgrid,npoint,Even,directions,displacements),
+  StencilEven(&Hgrid,npoint,Even,directions,displacements), // source is Even
+  StencilOdd (&Hgrid,npoint,Odd ,directions,displacements), // source is Odd
+  mass(_mass),
+  Umu(&Fgrid),
+  UmuEven(&Hgrid),
+  UmuOdd (&Hgrid)
+{
+  // Allocate the required comms buffer
+  comm_buf.resize(Stencil._unified_buffer_size); // this is always big enough to contain EO
+  DoubleStore(Umu,_Umu);
+  pickCheckerboard(Even,UmuEven,Umu);
+  pickCheckerboard(Odd ,UmuOdd,Umu);
+}
+      
+void WilsonFermion::DoubleStore(LatticeDoubledGaugeField &Uds,const LatticeGaugeField &Umu)
+{
+  conformable(Uds._grid,GaugeGrid());
+  conformable(Umu._grid,GaugeGrid());
+  LatticeColourMatrix U(GaugeGrid());
+  for(int mu=0;mu<Nd;mu++){
+    U = peekIndex<LorentzIndex>(Umu,mu);
+    pokeIndex<LorentzIndex>(Uds,U,mu);
+    U = adj(Cshift(U,mu,-1));
+    pokeIndex<LorentzIndex>(Uds,U,mu+4);
+  }
+}
+
+RealD WilsonFermion::M(const LatticeFermion &in, LatticeFermion &out)
+{
+  out.checkerboard=in.checkerboard;
+  Dhop(in,out,DaggerNo);
+  return axpy_norm(out,4+mass,in,out);
+}
+RealD WilsonFermion::Mdag(const LatticeFermion &in, LatticeFermion &out)
+{
+  out.checkerboard=in.checkerboard;
+  Dhop(in,out,DaggerYes);
+  return axpy_norm(out,4+mass,in,out);
+}
+
+void WilsonFermion::Meooe(const LatticeFermion &in, LatticeFermion &out)
+{
+  if ( in.checkerboard == Odd ) {
+    DhopEO(in,out,DaggerNo);
+  } else {
+    DhopOE(in,out,DaggerNo);
+  }
+}
+void WilsonFermion::MeooeDag(const LatticeFermion &in, LatticeFermion &out)
+{
+  if ( in.checkerboard == Odd ) {
+    DhopEO(in,out,DaggerYes);
+  } else {
+    DhopOE(in,out,DaggerYes);
+  }
+}
+void WilsonFermion::Mooee(const LatticeFermion &in, LatticeFermion &out)
+{
+  out.checkerboard = in.checkerboard;
+  out = (4.0+mass)*in;
+  return ;
+}
+void WilsonFermion::MooeeDag(const LatticeFermion &in, LatticeFermion &out)
+{
+  out.checkerboard = in.checkerboard;
+  Mooee(in,out);
+}
+void WilsonFermion::MooeeInv(const LatticeFermion &in, LatticeFermion &out)
+{
+  out.checkerboard = in.checkerboard;
+  out = (1.0/(4.0+mass))*in;
+  return ;
+}
+void WilsonFermion::MooeeInvDag(const LatticeFermion &in, LatticeFermion &out)
+{
+  out.checkerboard = in.checkerboard;
+  MooeeInv(in,out);
+}
+
+void WilsonFermion::DhopInternal(CartesianStencil & st,LatticeDoubledGaugeField & U,
+				const LatticeFermion &in, LatticeFermion &out,int dag)
+{
+  assert((dag==DaggerNo) ||(dag==DaggerYes));
+  WilsonCompressor compressor(dag);
+  st.HaloExchange<vSpinColourVector,vHalfSpinColourVector,WilsonCompressor>(in,comm_buf,compressor);
+
+  if ( dag == DaggerYes ) {
+    if( HandOptDslash ) {
+PARALLEL_FOR_LOOP
+      for(int sss=0;sss<in._grid->oSites();sss++){
+        DiracOptHand::DhopSiteDag(st,U,comm_buf,sss,sss,in,out);
+      }
+    } else { 
+PARALLEL_FOR_LOOP
+      for(int sss=0;sss<in._grid->oSites();sss++){
+        DiracOpt::DhopSiteDag(st,U,comm_buf,sss,sss,in,out);
+      }
+    }
+  } else {
+    if( HandOptDslash ) {
+PARALLEL_FOR_LOOP
+      for(int sss=0;sss<in._grid->oSites();sss++){
+        DiracOptHand::DhopSite(st,U,comm_buf,sss,sss,in,out);
+      }
+    } else { 
+PARALLEL_FOR_LOOP
+      for(int sss=0;sss<in._grid->oSites();sss++){
+        DiracOpt::DhopSite(st,U,comm_buf,sss,sss,in,out);
+      }
+    }
+  }
+}
+void WilsonFermion::DhopOE(const LatticeFermion &in, LatticeFermion &out,int dag)
+{
+  conformable(in._grid,_cbgrid);    // verifies half grid
+  conformable(in._grid,out._grid); // drops the cb check
+
+  assert(in.checkerboard==Even);
+  out.checkerboard = Odd;
+
+  DhopInternal(StencilEven,UmuOdd,in,out,dag);
+}
+void WilsonFermion::DhopEO(const LatticeFermion &in, LatticeFermion &out,int dag)
+{
+  conformable(in._grid,_cbgrid);    // verifies half grid
+  conformable(in._grid,out._grid); // drops the cb check
+
+  assert(in.checkerboard==Odd);
+  out.checkerboard = Even;
+
+  DhopInternal(StencilOdd,UmuEven,in,out,dag);
+}
+void WilsonFermion::Dhop(const LatticeFermion &in, LatticeFermion &out,int dag)
+{
+  conformable(in._grid,_grid); // verifies full grid
+  conformable(in._grid,out._grid);
+
+  out.checkerboard = in.checkerboard;
+
+  DhopInternal(Stencil,Umu,in,out,dag);
+}
+
+}}
+
+
+