/************************************************************************************* Grid physics library, www.github.com/paboyle/Grid Source file: ./lib/qcd/action/fermion/CayleyFermion5D.h Copyright (C) 2015 Author: Peter Boyle Author: Peter Boyle This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. See the full license in the file "LICENSE" in the top level distribution directory *************************************************************************************/ /* END LEGAL */ #pragma once #include NAMESPACE_BEGIN(Grid); template class CayleyFermion5D : public WilsonFermion5D { public: INHERIT_IMPL_TYPES(Impl); public: // override multiply virtual RealD M (const FermionField &in, FermionField &out); virtual RealD Mdag (const FermionField &in, FermionField &out); // half checkerboard operations virtual void Meooe (const FermionField &in, FermionField &out); virtual void MeooeDag (const FermionField &in, FermionField &out); virtual void Mooee (const FermionField &in, FermionField &out); virtual void MooeeDag (const FermionField &in, FermionField &out); virtual void MooeeInv (const FermionField &in, FermionField &out); virtual void MooeeInvDag (const FermionField &in, FermionField &out); virtual void Meo5D (const FermionField &psi, FermionField &chi); virtual void M5D (const FermionField &psi, FermionField &chi); virtual void M5Ddag(const FermionField &psi, FermionField &chi); /////////////////////////////////////////////////////////////// // Physical surface field utilities /////////////////////////////////////////////////////////////// virtual void Dminus(const FermionField &psi, FermionField &chi); virtual void DminusDag(const FermionField &psi, FermionField &chi); virtual void ExportPhysicalFermionSolution(const FermionField &solution5d,FermionField &exported4d); virtual void ExportPhysicalFermionSource(const FermionField &solution5d, FermionField &exported4d); virtual void ImportPhysicalFermionSource(const FermionField &input4d,FermionField &imported5d); virtual void ImportUnphysicalFermion(const FermionField &solution5d, FermionField &exported4d); /////////////////////////////////////////////////////////////// // Support for MADWF tricks /////////////////////////////////////////////////////////////// RealD Mass(void) { return mass; }; void SetMass(RealD _mass) { mass=_mass; SetCoefficientsInternal(_zolo_hi,_gamma,_b,_c); // Reset coeffs } ; void P(const FermionField &psi, FermionField &chi); void Pdag(const FermionField &psi, FermionField &chi); ///////////////////////////////////////////////////// // Instantiate different versions depending on Impl ///////////////////////////////////////////////////// void M5D(const FermionField &psi, const FermionField &phi, FermionField &chi, Vector &lower, Vector &diag, Vector &upper); void M5Ddag(const FermionField &psi, const FermionField &phi, FermionField &chi, Vector &lower, Vector &diag, Vector &upper); virtual void Instantiatable(void)=0; // force terms; five routines; default to Dhop on diagonal virtual void MDeriv (GaugeField &mat,const FermionField &U,const FermionField &V,int dag); virtual void MoeDeriv(GaugeField &mat,const FermionField &U,const FermionField &V,int dag); virtual void MeoDeriv(GaugeField &mat,const FermionField &U,const FermionField &V,int dag); // Efficient support for multigrid coarsening virtual void Mdir (const FermionField &in, FermionField &out,int dir,int disp); virtual void MdirAll(const FermionField &in, std::vector &out); void Meooe5D (const FermionField &in, FermionField &out); void MeooeDag5D (const FermionField &in, FermionField &out); // protected: RealD mass; // Save arguments to SetCoefficientsInternal Vector _gamma; RealD _zolo_hi; RealD _b; RealD _c; // Cayley form Moebius (tanh and zolotarev) Vector omega; Vector bs; // S dependent coeffs Vector cs; Vector as; // For preconditioning Cayley form Vector bee; Vector cee; Vector aee; Vector beo; Vector ceo; Vector aeo; // LDU factorisation of the eeoo matrix Vector lee; Vector leem; Vector uee; Vector ueem; Vector dee; // Matrices of 5d ee inverse params Vector > MatpInv; Vector > MatmInv; Vector > MatpInvDag; Vector > MatmInvDag; /////////////////////////////////////////////////////////////// // Conserved current utilities /////////////////////////////////////////////////////////////// // Virtual can't template void ContractConservedCurrent(PropagatorField &q_in_1, PropagatorField &q_in_2, PropagatorField &q_out, PropagatorField &phys_src, Current curr_type, unsigned int mu); void SeqConservedCurrent(PropagatorField &q_in, PropagatorField &q_out, PropagatorField &phys_src, Current curr_type, unsigned int mu, unsigned int tmin, unsigned int tmax, ComplexField &lattice_cmplx); void ContractJ5q(PropagatorField &q_in,ComplexField &J5q); void ContractJ5q(FermionField &q_in,ComplexField &J5q); /////////////////////////////////////////////////////////////// // Constructors /////////////////////////////////////////////////////////////// CayleyFermion5D(GaugeField &_Umu, GridCartesian &FiveDimGrid, GridRedBlackCartesian &FiveDimRedBlackGrid, GridCartesian &FourDimGrid, GridRedBlackCartesian &FourDimRedBlackGrid, RealD _mass,RealD _M5,const ImplParams &p= ImplParams()); void CayleyReport(void); void CayleyZeroCounters(void); double M5Dflops; double M5Dcalls; double M5Dtime; double MooeeInvFlops; double MooeeInvCalls; double MooeeInvTime; protected: virtual void SetCoefficientsZolotarev(RealD zolohi,Approx::zolotarev_data *zdata,RealD b,RealD c); virtual void SetCoefficientsTanh(Approx::zolotarev_data *zdata,RealD b,RealD c); virtual void SetCoefficientsInternal(RealD zolo_hi,Vector & gamma,RealD b,RealD c); }; NAMESPACE_END(Grid);