mirror of
https://github.com/paboyle/Grid.git
synced 2025-04-04 19:25:56 +01:00
211 lines
7.9 KiB
C++
211 lines
7.9 KiB
C++
/*************************************************************************************
|
|
|
|
Grid physics library, www.github.com/paboyle/Grid
|
|
|
|
Source file: ./lib/qcd/action/fermion/CayleyFermion5D.h
|
|
|
|
Copyright (C) 2015
|
|
|
|
Author: Peter Boyle <pabobyle@ph.ed.ac.uk>
|
|
Author: Peter Boyle <paboyle@ph.ed.ac.uk>
|
|
|
|
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 <Grid/qcd/action/fermion/WilsonFermion5D.h>
|
|
|
|
NAMESPACE_BEGIN(Grid);
|
|
|
|
template<class Impl>
|
|
class CayleyFermion5D : public WilsonFermion5D<Impl>
|
|
{
|
|
public:
|
|
INHERIT_IMPL_TYPES(Impl);
|
|
public:
|
|
|
|
// override multiply
|
|
virtual void M (const FermionField &in, FermionField &out);
|
|
virtual void 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_plus + mass_minus) / 2.0; };
|
|
RealD MassPlus(void) { return mass_plus; };
|
|
RealD MassMinus(void) { return mass_minus; };
|
|
|
|
void SetMass(RealD _mass) {
|
|
mass_plus=mass_minus=_mass;
|
|
SetCoefficientsInternal(_zolo_hi,_gamma,_b,_c); // Reset coeffs
|
|
} ;
|
|
void SetMass(RealD _mass_plus, RealD _mass_minus) {
|
|
mass_plus=_mass_plus;
|
|
mass_minus=_mass_minus;
|
|
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,
|
|
std::vector<Coeff_t> &lower,
|
|
std::vector<Coeff_t> &diag,
|
|
std::vector<Coeff_t> &upper);
|
|
|
|
void M5Ddag(const FermionField &psi,
|
|
const FermionField &phi,
|
|
FermionField &chi,
|
|
std::vector<Coeff_t> &lower,
|
|
std::vector<Coeff_t> &diag,
|
|
std::vector<Coeff_t> &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<FermionField> &out);
|
|
|
|
void Meooe5D (const FermionField &in, FermionField &out);
|
|
void MeooeDag5D (const FermionField &in, FermionField &out);
|
|
|
|
// protected:
|
|
RealD mass_plus, mass_minus;
|
|
|
|
// Save arguments to SetCoefficientsInternal
|
|
std::vector<Coeff_t> _gamma;
|
|
RealD _zolo_hi;
|
|
RealD _b;
|
|
RealD _c;
|
|
|
|
// possible boost
|
|
std::vector<ComplexD> qmu;
|
|
void set_qmu(std::vector<ComplexD> _qmu) { qmu=_qmu; assert(qmu.size()==Nd);};
|
|
void addQmu(const FermionField &in, FermionField &out, int dag);
|
|
|
|
// Cayley form Moebius (tanh and zolotarev)
|
|
std::vector<Coeff_t> omega;
|
|
std::vector<Coeff_t> bs; // S dependent coeffs
|
|
std::vector<Coeff_t> cs;
|
|
std::vector<Coeff_t> as;
|
|
// For preconditioning Cayley form
|
|
std::vector<Coeff_t> bee;
|
|
std::vector<Coeff_t> cee;
|
|
std::vector<Coeff_t> aee;
|
|
std::vector<Coeff_t> beo;
|
|
std::vector<Coeff_t> ceo;
|
|
std::vector<Coeff_t> aeo;
|
|
// LDU factorisation of the eeoo matrix
|
|
std::vector<Coeff_t> lee;
|
|
std::vector<Coeff_t> leem;
|
|
std::vector<Coeff_t> uee;
|
|
std::vector<Coeff_t> ueem;
|
|
std::vector<Coeff_t> dee;
|
|
|
|
// Device memory
|
|
deviceVector<Coeff_t> d_diag;
|
|
deviceVector<Coeff_t> d_upper;
|
|
deviceVector<Coeff_t> d_lower;
|
|
|
|
deviceVector<Coeff_t> d_lee;
|
|
deviceVector<Coeff_t> d_dee;
|
|
deviceVector<Coeff_t> d_uee;
|
|
deviceVector<Coeff_t> d_leem;
|
|
deviceVector<Coeff_t> d_ueem;
|
|
|
|
// Matrices of 5d ee inverse params
|
|
// std::vector<iSinglet<Simd> > MatpInv;
|
|
// std::vector<iSinglet<Simd> > MatmInv;
|
|
// std::vector<iSinglet<Simd> > MatpInvDag;
|
|
// std::vector<iSinglet<Simd> > 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());
|
|
|
|
|
|
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,std::vector<Coeff_t> & gamma,RealD b,RealD c);
|
|
};
|
|
|
|
NAMESPACE_END(Grid);
|
|
|