mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-10 07:55:35 +00:00
First cut (non functional NPR code) developed by Julia Kettle
This commit is contained in:
Peter Boyle
parent
49f25e08e8
commit
03c3d495a2
93
Hadrons/Modules/MGauge/GaugeFix.cc
Normal file
93
Hadrons/Modules/MGauge/GaugeFix.cc
Normal file
@ -0,0 +1,93 @@
|
|||||||
|
/*************************************************************************************
|
||||||
|
|
||||||
|
Grid physics library, www.github.com/paboyle/Grid
|
||||||
|
|
||||||
|
Source file: extras/Hadrons/Modules/MGauge/Fix.cc
|
||||||
|
|
||||||
|
Copyright (C) 2015
|
||||||
|
Copyright (C) 2016
|
||||||
|
|
||||||
|
Author: Antonin Portelli <antonin.portelli@me.com>
|
||||||
|
|
||||||
|
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 */
|
||||||
|
|
||||||
|
#include <Hadrons/Modules/MGauge/GaugeFix.hpp>
|
||||||
|
|
||||||
|
using namespace Grid;
|
||||||
|
using namespace Hadrons;
|
||||||
|
using namespace MGauge;
|
||||||
|
|
||||||
|
|
||||||
|
/******************************************************************************
|
||||||
|
* TGaugeFix implementation *
|
||||||
|
******************************************************************************/
|
||||||
|
// constructor /////////////////////////////////////////////////////////////////
|
||||||
|
template <typename GImpl>
|
||||||
|
TGaugeFix<GImpl>::TGaugeFix(const std::string name)
|
||||||
|
: Module<GaugeFixPar>(name)
|
||||||
|
{}
|
||||||
|
|
||||||
|
// dependencies/products ///////////////////////////////////////////////////////
|
||||||
|
template <typename GImpl>
|
||||||
|
std::vector<std::string> TGaugeFix<GImpl>::getInput(void)
|
||||||
|
{
|
||||||
|
std::vector<std::string> in = {par().gauge};
|
||||||
|
return in;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename GImpl>
|
||||||
|
std::vector<std::string> TGaugeFix<GImpl>::getOutput(void)
|
||||||
|
{
|
||||||
|
std::vector<std::string> out = {getName()};
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
// setup ///////////////////////////////////////////////////////////////////////
|
||||||
|
template <typename GImpl>
|
||||||
|
void TGaugeFix<GImpl>::setup(void)
|
||||||
|
{
|
||||||
|
envCreateLat(GaugeField, getName());
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// execution ///////////////////////////////////////////////////////////////////
|
||||||
|
template <typename GImpl>
|
||||||
|
void TGaugeFix<GImpl>::execute(void)
|
||||||
|
//Loads the gauge and fixes it
|
||||||
|
{
|
||||||
|
|
||||||
|
std::cout << "executing" << std::endl;
|
||||||
|
LOG(Message) << "Fixing the Gauge" << std::endl;
|
||||||
|
LOG(Message) << par().gauge << std::endl;
|
||||||
|
auto &U = envGet(GaugeField, par().gauge);
|
||||||
|
auto &Umu = envGet(GaugeField, getName());
|
||||||
|
LOG(Message) << "Gauge Field fetched" << std::endl;
|
||||||
|
//do we allow maxiter etc to be user set?
|
||||||
|
Real alpha = par().alpha;
|
||||||
|
int maxiter = par().maxiter;
|
||||||
|
Real Omega_tol = par().Omega_tol;
|
||||||
|
Real Phi_tol = par().Phi_tol;
|
||||||
|
bool Fourier = par().Fourier;
|
||||||
|
FourierAcceleratedGaugeFixer<PeriodicGimplR>::SteepestDescentGaugeFix(U,alpha,maxiter,Omega_tol,Phi_tol,Fourier);
|
||||||
|
Umu = U;
|
||||||
|
LOG(Message) << "Gauge Fixed" << std::endl;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
template class Grid::Hadrons::MGauge::TGaugeFix<GIMPL>;
|
||||||
81
Hadrons/Modules/MGauge/GaugeFix.hpp
Normal file
81
Hadrons/Modules/MGauge/GaugeFix.hpp
Normal file
@ -0,0 +1,81 @@
|
|||||||
|
/*************************************************************************************
|
||||||
|
|
||||||
|
Grid physics library, www.github.com/paboyle/Grid
|
||||||
|
|
||||||
|
Source file: extras/Hadrons/Modules/MGauge/Fix.hpp
|
||||||
|
|
||||||
|
Copyright (C) 2015
|
||||||
|
Copyright (C) 2016
|
||||||
|
|
||||||
|
Author: Antonin Portelli <antonin.portelli@me.com>
|
||||||
|
|
||||||
|
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 */
|
||||||
|
|
||||||
|
#ifndef Hadrons_MGaugeFix_hpp_
|
||||||
|
#define Hadrons_MGaugeFix_hpp_
|
||||||
|
|
||||||
|
#include <Hadrons/Global.hpp>
|
||||||
|
#include <Hadrons/Module.hpp>
|
||||||
|
#include <Hadrons/ModuleFactory.hpp>
|
||||||
|
#include <Grid/qcd/utils/GaugeFix.h>
|
||||||
|
BEGIN_HADRONS_NAMESPACE
|
||||||
|
|
||||||
|
/******************************************************************************
|
||||||
|
* Fix gauge *
|
||||||
|
******************************************************************************/
|
||||||
|
BEGIN_MODULE_NAMESPACE(MGauge)
|
||||||
|
|
||||||
|
class GaugeFixPar: Serializable
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
GRID_SERIALIZABLE_CLASS_MEMBERS(GaugeFixPar,
|
||||||
|
std::string, gauge,
|
||||||
|
Real, alpha,
|
||||||
|
int, maxiter,
|
||||||
|
Real, Omega_tol,
|
||||||
|
Real, Phi_tol,
|
||||||
|
bool, Fourier);
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename GImpl>
|
||||||
|
class TGaugeFix: public Module<GaugeFixPar>
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
GAUGE_TYPE_ALIASES(GImpl,);
|
||||||
|
public:
|
||||||
|
// constructor
|
||||||
|
TGaugeFix(const std::string name);
|
||||||
|
// destructor
|
||||||
|
virtual ~TGaugeFix(void) = default;
|
||||||
|
// dependencies/products
|
||||||
|
virtual std::vector<std::string> getInput(void);
|
||||||
|
virtual std::vector<std::string> getOutput(void);
|
||||||
|
// setup
|
||||||
|
virtual void setup(void);
|
||||||
|
// execution
|
||||||
|
virtual void execute(void);
|
||||||
|
};
|
||||||
|
|
||||||
|
MODULE_REGISTER_TMP(GaugeFix, TGaugeFix<GIMPL>, MGauge);
|
||||||
|
|
||||||
|
END_MODULE_NAMESPACE
|
||||||
|
|
||||||
|
END_HADRONS_NAMESPACE
|
||||||
|
|
||||||
|
#endif // Hadrons_MGaugeFix_hpp_
|
||||||
8
Hadrons/Modules/MNPR/Amputate.cc
Normal file
8
Hadrons/Modules/MNPR/Amputate.cc
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
#include <Hadrons/Modules/MNPR/Amputate.hpp>
|
||||||
|
|
||||||
|
using namespace Grid;
|
||||||
|
using namespace Hadrons;
|
||||||
|
using namespace MNPR;
|
||||||
|
|
||||||
|
template class Grid::Hadrons::MNPR::TAmputate<FIMPL,FIMPL>;
|
||||||
|
|
||||||
199
Hadrons/Modules/MNPR/Amputate.hpp
Normal file
199
Hadrons/Modules/MNPR/Amputate.hpp
Normal file
@ -0,0 +1,199 @@
|
|||||||
|
/*************************************************************************************
|
||||||
|
|
||||||
|
Grid physics library, www.github.com/paboyle/Grid
|
||||||
|
|
||||||
|
Source file: extras/Hadrons/Modules/MNpr/Amputate.hpp
|
||||||
|
|
||||||
|
Copyright (C) 2015
|
||||||
|
Copyright (C) 2016
|
||||||
|
|
||||||
|
Author: Julia Kettle J.R.Kettle-2@sms.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 */
|
||||||
|
|
||||||
|
#ifndef Hadrons_Amputate_hpp_
|
||||||
|
#define Hadrons_Amputate_hpp_
|
||||||
|
|
||||||
|
#include <Hadrons/Global.hpp>
|
||||||
|
#include <Hadrons/Module.hpp>
|
||||||
|
#include <Hadrons/ModuleFactory.hpp>
|
||||||
|
#include <Grid/Eigen/LU>
|
||||||
|
//#include <Grid/qcd/utils/PropagatorUtils.h>
|
||||||
|
//#include <Grid/serialisation/Serialisation.h>
|
||||||
|
BEGIN_HADRONS_NAMESPACE
|
||||||
|
|
||||||
|
/******************************************************************************
|
||||||
|
* TAmputate *
|
||||||
|
Performs bilinear contractions of the type tr[g5*adj(Sout)*g5*G*Sin]
|
||||||
|
Suitable for non exceptional momenta
|
||||||
|
******************************************************************************/
|
||||||
|
BEGIN_MODULE_NAMESPACE(MNPR)
|
||||||
|
|
||||||
|
class AmputatePar: Serializable
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
GRID_SERIALIZABLE_CLASS_MEMBERS(AmputatePar,
|
||||||
|
std::string, Sin, //need to make this a propogator type?
|
||||||
|
std::string, Sout, //same
|
||||||
|
std::string, vertex,
|
||||||
|
std::string, pin,
|
||||||
|
std::string, pout,
|
||||||
|
std::string, output,
|
||||||
|
std::string, input);
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
class TAmputate: public Module<AmputatePar>
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
FERM_TYPE_ALIASES(FImpl1, 1);
|
||||||
|
FERM_TYPE_ALIASES(FImpl2, 2);
|
||||||
|
class Result: Serializable
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
GRID_SERIALIZABLE_CLASS_MEMBERS(Result,
|
||||||
|
std::vector<Complex>, Vamp,
|
||||||
|
);
|
||||||
|
};
|
||||||
|
public:
|
||||||
|
// constructor
|
||||||
|
TAmputate(const std::string name);
|
||||||
|
// destructor
|
||||||
|
virtual ~TAmputate(void) = default;
|
||||||
|
// dependencies/products
|
||||||
|
virtual std::vector<std::string> getInput(void);
|
||||||
|
virtual std::vector<std::string> getOutput(void);
|
||||||
|
virtual SpinColourMatrix invertspincolmat(SpinColourMatrix &scmat);
|
||||||
|
// execution
|
||||||
|
virtual void execute(void);
|
||||||
|
};
|
||||||
|
|
||||||
|
MODULE_REGISTER_TMP(Amputate, ARG(TAmputate<FIMPL, FIMPL>), MNPR);
|
||||||
|
|
||||||
|
/******************************************************************************
|
||||||
|
* TAmputate implementation *
|
||||||
|
******************************************************************************/
|
||||||
|
// constructor /////////////////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
TAmputate<FImpl1, FImpl2>::TAmputate(const std::string name)
|
||||||
|
: Module<AmputatePar>(name)
|
||||||
|
{}
|
||||||
|
|
||||||
|
// dependencies/products ///////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
std::vector<std::string> TAmputate<FImpl1, FImpl2>::getInput(void)
|
||||||
|
{
|
||||||
|
std::vector<std::string> input = {par().Sin, par().Sout, par().vertex};
|
||||||
|
|
||||||
|
return input;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
std::vector<std::string> TAmputate<FImpl1, FImpl2>::getOutput(void)
|
||||||
|
{
|
||||||
|
std::vector<std::string> output = {getName()};
|
||||||
|
|
||||||
|
|
||||||
|
return output;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Invert spin colour matrix using Eigen
|
||||||
|
template <typename Fimpl1, typename Fimpl2>
|
||||||
|
SpinColourMatrix TAmputate<Fimpl1, Fimpl2>::invertspincolmat(SpinColourMatrix &scmat)
|
||||||
|
{
|
||||||
|
Eigen::MatrixXcf scmat_2d(Ns*Nc,Ns*Nc);
|
||||||
|
for(int ic=0; ic<Nc; ic++){
|
||||||
|
for(int jc=0; jc<Nc; jc++){
|
||||||
|
for(int is=0; is<Ns; is++){
|
||||||
|
for(int js=0; js<Ns; js++){
|
||||||
|
scmat_2d(Ns*ic+is,Ns*jc+js) = scmat()(is,js)(ic,jc);
|
||||||
|
}}
|
||||||
|
}}
|
||||||
|
Eigen::MatrixXcf scmat_2d_inv = scmat_2d.inverse();
|
||||||
|
SpinColourMatrix scmat_inv;
|
||||||
|
for(int ic=0; ic<Nc; ic++){
|
||||||
|
for(int jc=0; jc<Nc; jc++){
|
||||||
|
for(int is=0; is<Ns; is++){
|
||||||
|
for(int js=0; js<Ns; js++){
|
||||||
|
scmat_inv()(is,js)(ic,jc) = scmat_2d_inv(Ns*ic+is,Ns*jc+js);
|
||||||
|
}}
|
||||||
|
}}
|
||||||
|
return scmat_inv;
|
||||||
|
}
|
||||||
|
|
||||||
|
// execution ///////////////////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
void TAmputate<FImpl1, FImpl2>::execute(void)
|
||||||
|
{
|
||||||
|
LOG(Message) << "Computing bilinear amputations '" << getName() << "' using"
|
||||||
|
<< " momentum '" << par().Sin << "' and '" << par().Sout << "'"
|
||||||
|
<< std::endl;
|
||||||
|
BinaryWriter writer(par().output);
|
||||||
|
PropagatorField1 &Sin = *env().template getObject<PropagatorField1>(par().Sin); //Do these have the phases taken into account?? Don't think so. FIX
|
||||||
|
PropagatorField2 &Sout = *env().template getObject<PropagatorField2>(par().Sout);
|
||||||
|
std::vector<int> pin = strToVec<int>(par().pin), pout = strToVec<int>(par().pout);
|
||||||
|
std::vector<Real> latt_size(pin.begin(), pin.end());
|
||||||
|
LatticeComplex pdotxin(env().getGrid()), pdotxout(env().getGrid()), coor(env().getGrid());
|
||||||
|
LOG(Message) << "Propagators set up " << std::endl;
|
||||||
|
std::vector<SpinColourMatrix> vertex; // Let's read from file here
|
||||||
|
Gamma g5(Gamma::Algebra::Gamma5);
|
||||||
|
Result result;
|
||||||
|
LOG(Message) << "reading file - " << par().input << std::endl;
|
||||||
|
BinaryReader reader(par().input);
|
||||||
|
Complex Ci(0.0,1.0);
|
||||||
|
|
||||||
|
std::string svertex;
|
||||||
|
read(reader,"vertex", vertex);
|
||||||
|
LOG(Message) << "vertex read" << std::endl;
|
||||||
|
|
||||||
|
pdotxin=zero;
|
||||||
|
pdotxout=zero;
|
||||||
|
for (unsigned int mu = 0; mu < 4; ++mu)
|
||||||
|
{
|
||||||
|
Real TwoPiL = M_PI * 2.0/ latt_size[mu];
|
||||||
|
LatticeCoordinate(coor,mu);
|
||||||
|
pdotxin = pdotxin +(TwoPiL * pin[mu]) * coor;
|
||||||
|
pdotxout= pdotxout +(TwoPiL * pout[mu]) * coor;
|
||||||
|
}
|
||||||
|
Sin = Sin*exp(-Ci*pdotxin); //phase corrections
|
||||||
|
Sout = Sout*exp(-Ci*pdotxout);
|
||||||
|
|
||||||
|
SpinColourMatrix Sin_mom = sum(Sin);
|
||||||
|
SpinColourMatrix Sout_mom = sum(Sout);
|
||||||
|
LOG(Message) << "summed over lattice" << std::endl;
|
||||||
|
|
||||||
|
LOG(Message) << "Lattice -> spincolourmatrix conversion" << std::endl;
|
||||||
|
|
||||||
|
SpinColourMatrix Sin_inv = invertspincolmat(Sin_mom);
|
||||||
|
SpinColourMatrix Sout_inv = invertspincolmat(Sout_mom);
|
||||||
|
LOG(Message) << "Inversions done" << std::endl;
|
||||||
|
|
||||||
|
result.Vamp.resize(Gamma::nGamma/2);
|
||||||
|
for( int mu=0; mu < Gamma::nGamma/2; mu++){
|
||||||
|
Gamma::Algebra gam = mu;
|
||||||
|
result.Vamp[mu] = 1/12.0*trace(adj(Gamma(mu*2+1))*g5*Sout_inv*g5*vertex[mu]*Sin_inv);
|
||||||
|
LOG(Message) << "Vamp[" << mu << "] - " << result.Vamp[mu] << std::endl;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
END_MODULE_NAMESPACE
|
||||||
|
|
||||||
|
END_HADRONS_NAMESPACE
|
||||||
|
|
||||||
|
#endif // Hadrons_Amputate_hpp_
|
||||||
8
Hadrons/Modules/MNPR/Bilinear.cc
Normal file
8
Hadrons/Modules/MNPR/Bilinear.cc
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
#include <Hadrons/Modules/MNPR/Bilinear.hpp>
|
||||||
|
|
||||||
|
using namespace Grid;
|
||||||
|
using namespace Hadrons;
|
||||||
|
using namespace MNPR;
|
||||||
|
|
||||||
|
template class Grid::Hadrons::MNPR::TBilinear<FIMPL,FIMPL>;
|
||||||
|
|
||||||
224
Hadrons/Modules/MNPR/Bilinear.hpp
Normal file
224
Hadrons/Modules/MNPR/Bilinear.hpp
Normal file
@ -0,0 +1,224 @@
|
|||||||
|
/*************************************************************************************
|
||||||
|
|
||||||
|
Grid physics library, www.github.com/paboyle/Grid
|
||||||
|
|
||||||
|
Source file: extras/Hadrons/Modules/MContraction/Bilinear.hpp
|
||||||
|
|
||||||
|
Copyright (C) 2015
|
||||||
|
Copyright (C) 2016
|
||||||
|
|
||||||
|
Author: Julia Kettle J.R.Kettle-2@sms.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 */
|
||||||
|
|
||||||
|
#ifndef Hadrons_Bilinear_hpp_
|
||||||
|
#define Hadrons_Bilinear_hpp_
|
||||||
|
|
||||||
|
#include <Hadrons/Global.hpp>
|
||||||
|
#include <Hadrons/Module.hpp>
|
||||||
|
#include <Hadrons/ModuleFactory.hpp>
|
||||||
|
#include <Hadrons/ModuleFactory.hpp>
|
||||||
|
//#include <Grid/qcd/utils/PropagatorUtils.h>
|
||||||
|
|
||||||
|
BEGIN_HADRONS_NAMESPACE
|
||||||
|
|
||||||
|
/******************************************************************************
|
||||||
|
* TBilinear *
|
||||||
|
Performs bilinear contractions of the type tr[g5*adj(Sout)*g5*G*Sin]
|
||||||
|
Suitable for non exceptional momenta in Rome-Southampton NPR
|
||||||
|
******************************************************************************/
|
||||||
|
BEGIN_MODULE_NAMESPACE(MNPR)
|
||||||
|
|
||||||
|
class BilinearPar: Serializable
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
GRID_SERIALIZABLE_CLASS_MEMBERS(BilinearPar,
|
||||||
|
std::string, Sin,
|
||||||
|
std::string, Sout,
|
||||||
|
std::string, pin,
|
||||||
|
std::string, pout,
|
||||||
|
std::string, output);
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
class TBilinear: public Module<BilinearPar>
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
FERM_TYPE_ALIASES(FImpl1, 1);
|
||||||
|
FERM_TYPE_ALIASES(FImpl2, 2);
|
||||||
|
class Result: Serializable
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
GRID_SERIALIZABLE_CLASS_MEMBERS(Result,
|
||||||
|
std::vector<SpinColourMatrix>, bilinear);
|
||||||
|
};
|
||||||
|
public:
|
||||||
|
// constructor
|
||||||
|
TBilinear(const std::string name);
|
||||||
|
// destructor
|
||||||
|
virtual ~TBilinear(void) = default;
|
||||||
|
// dependencies/products
|
||||||
|
virtual std::vector<std::string> getInput(void);
|
||||||
|
virtual std::vector<std::string> getOutput(void);
|
||||||
|
//LatticeSpinColourMatrix PhaseProps(LatticeSpinColourMatrix S, std::vector<Real> p);
|
||||||
|
// setup
|
||||||
|
virtual void setup(void);
|
||||||
|
// execution
|
||||||
|
virtual void execute(void);
|
||||||
|
};
|
||||||
|
|
||||||
|
MODULE_REGISTER_TMP(Bilinear, ARG(TBilinear<FIMPL, FIMPL>), MNPR);
|
||||||
|
|
||||||
|
/******************************************************************************
|
||||||
|
* TBilinear implementation *
|
||||||
|
******************************************************************************/
|
||||||
|
// constructor /////////////////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
TBilinear<FImpl1, FImpl2>::TBilinear(const std::string name)
|
||||||
|
: Module<BilinearPar>(name)
|
||||||
|
{}
|
||||||
|
|
||||||
|
// setup ///////////////////////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
void TBilinear<FImpl1, FImpl2>::setup(void)
|
||||||
|
{
|
||||||
|
//env().template registerLattice<LatticeSpinColourMatrix>(getName());
|
||||||
|
//env().template registerObject<SpinColourMatrix>(getName());
|
||||||
|
}
|
||||||
|
|
||||||
|
// dependencies/products ///////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
std::vector<std::string> TBilinear<FImpl1, FImpl2>::getInput(void)
|
||||||
|
{
|
||||||
|
std::vector<std::string> input = {par().Sin, par().Sout};
|
||||||
|
|
||||||
|
return input;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
std::vector<std::string> TBilinear<FImpl1, FImpl2>::getOutput(void)
|
||||||
|
{
|
||||||
|
std::vector<std::string> out = {getName()};
|
||||||
|
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
/////Phase propagators//////////////////////////
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
LatticeSpinColourMatrix TBilinear<FImpl1, FImpl2>::PhaseProps(LatticeSpinColourMatrix S, std::vector<Real> p)
|
||||||
|
{
|
||||||
|
GridBase *grid = S._grid;
|
||||||
|
LatticeComplex pdotx(grid), coor(grid);
|
||||||
|
std::vector<int> latt_size = grid->_fdimensions;
|
||||||
|
Complex Ci(0.0,1.0);
|
||||||
|
pdotx=zero;
|
||||||
|
for (unsigned int mu = 0; mu < 4; ++mu)
|
||||||
|
{
|
||||||
|
Real TwoPiL = M_PI * 2.0/ latt_size[mu];
|
||||||
|
LatticeCoordinate(coor,mu);
|
||||||
|
pdotx = pdotx +(TwoPiL * p[mu]) * coor;
|
||||||
|
}
|
||||||
|
S = S*exp(-Ci*pdotx);
|
||||||
|
return S;
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
// execution ///////////////////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
void TBilinear<FImpl1, FImpl2>::execute(void)
|
||||||
|
{
|
||||||
|
/**************************************************************************
|
||||||
|
|
||||||
|
Compute the bilinear vertex needed for the NPR.
|
||||||
|
V(G) = sum_x [ g5 * adj(S'(x,p2)) * g5 * G * S'(x,p1) ]_{si,sj,ci,cj}
|
||||||
|
G is one of the 16 gamma vertices [I,gmu,g5,g5gmu,sig(mu,nu)]
|
||||||
|
|
||||||
|
* G
|
||||||
|
/ \
|
||||||
|
p1/ \p2
|
||||||
|
/ \
|
||||||
|
/ \
|
||||||
|
|
||||||
|
Returns a spin-colour matrix, with indices si,sj, ci,cj
|
||||||
|
|
||||||
|
Conventions:
|
||||||
|
p1 - incoming momenta
|
||||||
|
p2 - outgoing momenta
|
||||||
|
q = (p1-p2)
|
||||||
|
**************************************************************************/
|
||||||
|
|
||||||
|
LOG(Message) << "Computing bilinear contractions '" << getName() << "' using"
|
||||||
|
<< " momentum '" << par().Sin << "' and '" << par().Sout << "'"
|
||||||
|
<< std::endl;
|
||||||
|
|
||||||
|
BinaryWriter writer(par().output);
|
||||||
|
|
||||||
|
|
||||||
|
// Propogators
|
||||||
|
LatticeSpinColourMatrix &Sin = *env().template getObject<LatticeSpinColourMatrix>(par().Sin);
|
||||||
|
LatticeSpinColourMatrix &Sout = *env().template getObject<LatticeSpinColourMatrix>(par().Sout);
|
||||||
|
LatticeComplex pdotxin(env().getGrid()), pdotxout(env().getGrid()), coor(env().getGrid());
|
||||||
|
// momentum on legs
|
||||||
|
std::vector<Real> pin = strToVec<Real>(par().pin), pout = strToVec<Real>(par().pout);
|
||||||
|
std::vector<Real> latt_size(pin.begin(), pin.end());
|
||||||
|
//bilinears
|
||||||
|
LatticeSpinColourMatrix bilinear_x(env().getGrid());
|
||||||
|
SpinColourMatrix bilinear;
|
||||||
|
Gamma g5(Gamma::Algebra::Gamma5);
|
||||||
|
Result result;
|
||||||
|
Complex Ci(0.0,1.0);
|
||||||
|
|
||||||
|
//
|
||||||
|
|
||||||
|
pdotxin=zero;
|
||||||
|
pdotxout=zero;
|
||||||
|
for (unsigned int mu = 0; mu < 4; ++mu)
|
||||||
|
{
|
||||||
|
Real TwoPiL = M_PI * 2.0/ latt_size[mu];
|
||||||
|
LatticeCoordinate(coor,mu);
|
||||||
|
pdotxin = pdotxin +(TwoPiL * pin[mu]) * coor;
|
||||||
|
pdotxout= pdotxout +(TwoPiL * pout[mu]) * coor;
|
||||||
|
}
|
||||||
|
Sin = Sin*exp(-Ci*pdotxin); //phase corrections
|
||||||
|
Sout = Sout*exp(-Ci*pdotxout);
|
||||||
|
|
||||||
|
////Set up gamma vector//////////////////////////
|
||||||
|
std::vector<Gamma> gammavector;
|
||||||
|
for( int i=0; i<Gamma::nGamma; i++){
|
||||||
|
Gamma::Algebra gam = i;
|
||||||
|
gammavector.push_back(Gamma(gam));
|
||||||
|
}
|
||||||
|
result.bilinear.resize(Gamma::nGamma);
|
||||||
|
/////////////////////////////////////////////////
|
||||||
|
//LatticeSpinMatrix temp = g5*Sout;
|
||||||
|
////////Form Vertex//////////////////////////////
|
||||||
|
for (int i=0; i < Gamma::nGamma; i++){
|
||||||
|
bilinear_x = g5*adj(Sout)*g5*gammavector[i]*Sin;
|
||||||
|
result.bilinear[i] = sum(bilinear_x); //sum over lattice sites
|
||||||
|
}
|
||||||
|
//////////////////////////////////////////////////
|
||||||
|
write(writer, par().output, result.bilinear);
|
||||||
|
LOG(Message) << "Complete. Writing results to " << par().output << std:: endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
END_MODULE_NAMESPACE
|
||||||
|
|
||||||
|
END_HADRONS_NAMESPACE
|
||||||
|
|
||||||
|
#endif // Hadrons_Bilinear_hpp_
|
||||||
8
Hadrons/Modules/MNPR/FourQuark.cc
Normal file
8
Hadrons/Modules/MNPR/FourQuark.cc
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
#include <Hadrons/Modules/MNPR/FourQuark.hpp>
|
||||||
|
|
||||||
|
using namespace Grid;
|
||||||
|
using namespace Hadrons;
|
||||||
|
using namespace MNPR;
|
||||||
|
|
||||||
|
template class Grid::Hadrons::MNPR::TFourQuark<FIMPL,FIMPL>;
|
||||||
|
|
||||||
273
Hadrons/Modules/MNPR/FourQuark.hpp
Normal file
273
Hadrons/Modules/MNPR/FourQuark.hpp
Normal file
@ -0,0 +1,273 @@
|
|||||||
|
/*************************************************************************************
|
||||||
|
|
||||||
|
Grid physics library, www.github.com/paboyle/Grid
|
||||||
|
|
||||||
|
Source file: extras/Hadrons/Modules/MNPR/FourQuark.hpp
|
||||||
|
|
||||||
|
Copyright (C) 2015
|
||||||
|
Copyright (C) 2016
|
||||||
|
|
||||||
|
Author: Julia Kettle J.R.Kettle-2@sms.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 */
|
||||||
|
|
||||||
|
#ifndef Hadrons_FourQuark_hpp_
|
||||||
|
#define Hadrons_FourQuark_hpp_
|
||||||
|
|
||||||
|
#include <typeinfo>
|
||||||
|
#include <Hadrons/Global.hpp>
|
||||||
|
#include <Hadrons/Module.hpp>
|
||||||
|
#include <Hadrons/ModuleFactory.hpp>
|
||||||
|
#include <Grid/serialisation/Serialisation.h>
|
||||||
|
BEGIN_HADRONS_NAMESPACE
|
||||||
|
|
||||||
|
/******************************************************************************
|
||||||
|
* TFourQuark *
|
||||||
|
Performs fourquark contractions of the type tr[g5*adj(Sout)*g5*G*Sin]
|
||||||
|
Suitable for non exceptional momenta
|
||||||
|
******************************************************************************/
|
||||||
|
BEGIN_MODULE_NAMESPACE(MNPR)
|
||||||
|
|
||||||
|
class FourQuarkPar: Serializable
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
GRID_SERIALIZABLE_CLASS_MEMBERS(FourQuarkPar,
|
||||||
|
std::string, Sin, //need to make this a propogator type?
|
||||||
|
std::string, Sout, //same
|
||||||
|
std::string, pin,
|
||||||
|
std::string, pout,
|
||||||
|
bool, fullbasis,
|
||||||
|
std::string, output);
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
class TFourQuark: public Module<FourQuarkPar>
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
FERM_TYPE_ALIASES(FImpl1, 1);
|
||||||
|
FERM_TYPE_ALIASES(FImpl2, 2);
|
||||||
|
class Result: Serializable
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
GRID_SERIALIZABLE_CLASS_MEMBERS(Result,
|
||||||
|
std::vector<SpinColourSpinColourMatrix>, fourquark);
|
||||||
|
};
|
||||||
|
public:
|
||||||
|
// constructor
|
||||||
|
TFourQuark(const std::string name);
|
||||||
|
// destructor
|
||||||
|
virtual ~TFourQuark(void) = default;
|
||||||
|
// dependencies/products
|
||||||
|
virtual std::vector<std::string> getInput(void);
|
||||||
|
virtual std::vector<std::string> getOutput(void);
|
||||||
|
// setup
|
||||||
|
virtual void tensorprod(LatticeSpinColourSpinColourMatrix &lret, LatticeSpinColourMatrix a, LatticeSpinColourMatrix b);
|
||||||
|
virtual void setup(void);
|
||||||
|
// execution
|
||||||
|
virtual void execute(void);
|
||||||
|
};
|
||||||
|
|
||||||
|
MODULE_REGISTER_TMP(FourQuark, ARG(TFourQuark<FIMPL, FIMPL>), MNPR);
|
||||||
|
|
||||||
|
/******************************************************************************
|
||||||
|
* TFourQuark implementation *
|
||||||
|
******************************************************************************/
|
||||||
|
// constructor /////////////////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
TFourQuark<FImpl1, FImpl2>::TFourQuark(const std::string name)
|
||||||
|
: Module<FourQuarkPar>(name)
|
||||||
|
{}
|
||||||
|
|
||||||
|
// dependencies/products ///////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
std::vector<std::string> TFourQuark<FImpl1, FImpl2>::getInput(void)
|
||||||
|
{
|
||||||
|
std::vector<std::string> input = {par().Sin, par().Sout};
|
||||||
|
|
||||||
|
return input;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
std::vector<std::string> TFourQuark<FImpl1, FImpl2>::getOutput(void)
|
||||||
|
{
|
||||||
|
std::vector<std::string> output = {getName()};
|
||||||
|
|
||||||
|
return output;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
void TFourQuark<FImpl1, FImpl2>::tensorprod(LatticeSpinColourSpinColourMatrix &lret, LatticeSpinColourMatrix a, LatticeSpinColourMatrix b)
|
||||||
|
{
|
||||||
|
#if 0
|
||||||
|
parallel_for(auto site=lret.begin();site<lret.end();site++) {
|
||||||
|
for (int si; si < 4; ++si){
|
||||||
|
for(int sj; sj <4; ++sj){
|
||||||
|
for (int ci; ci < 3; ++ci){
|
||||||
|
for (int cj; cj < 3; ++cj){
|
||||||
|
for (int sk; sk < 4; ++sk){
|
||||||
|
for(int sl; sl <4; ++sl){
|
||||||
|
for (int ck; ck < 3; ++ck){
|
||||||
|
for (int cl; cl < 3; ++cl){
|
||||||
|
lret[site]()(si,sj)(ci,cj)(sk,sl)(ck,cl)=a[site]()(si,sj)(ci,cj)*b[site]()(sk,sl)(ck,cl);
|
||||||
|
}}
|
||||||
|
}}
|
||||||
|
}}
|
||||||
|
}}
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
// FIXME ; is there a general need for this construct ? In which case we should encapsulate the
|
||||||
|
// below loops in a helper function.
|
||||||
|
//LOG(Message) << "sp co mat a is - " << a << std::endl;
|
||||||
|
//LOG(Message) << "sp co mat b is - " << b << std::endl;
|
||||||
|
parallel_for(auto site=lret.begin();site<lret.end();site++) {
|
||||||
|
vTComplex left;
|
||||||
|
for(int si=0; si < Ns; ++si){
|
||||||
|
for(int sj=0; sj < Ns; ++sj){
|
||||||
|
for (int ci=0; ci < Nc; ++ci){
|
||||||
|
for (int cj=0; cj < Nc; ++cj){
|
||||||
|
//LOG(Message) << "si, sj, ci, cj - " << si << ", " << sj << ", "<< ci << ", "<< cj << std::endl;
|
||||||
|
left()()() = a[site]()(si,sj)(ci,cj);
|
||||||
|
//LOG(Message) << left << std::endl;
|
||||||
|
lret[site]()(si,sj)(ci,cj)=left()*b[site]();
|
||||||
|
}}
|
||||||
|
}}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
// setup ///////////////////////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
void TFourQuark<FImpl1, FImpl2>::setup(void)
|
||||||
|
{
|
||||||
|
envCreateLat(LatticeSpinColourMatrix, getName());
|
||||||
|
}
|
||||||
|
|
||||||
|
// execution ///////////////////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl1, typename FImpl2>
|
||||||
|
void TFourQuark<FImpl1, FImpl2>::execute(void)
|
||||||
|
{
|
||||||
|
|
||||||
|
/*********************************************************************************
|
||||||
|
|
||||||
|
TFourQuark : Creates the four quark vertex required for the NPR of four-quark ops
|
||||||
|
|
||||||
|
V_{Gamma_1,Gamma_2} = sum_x [ ( g5 * adj(S'(x,p2)) * g5 * G1 * S'(x,p1) )_ci,cj;si,sj x ( g5 * adj(S'(x,p2)) * g5 * G2 S'(x,p1) )_ck,cl;sk,cl ]
|
||||||
|
|
||||||
|
Create a bilinear vertex for G1 and G2 the spin and colour indices are kept free. Where there are 16 potential Gs.
|
||||||
|
We then find the outer product of V1 and V2, keeping the spin and colour indices uncontracted
|
||||||
|
Then this is summed over the lattice coordinate
|
||||||
|
Result is a SpinColourSpinColourMatrix - with 4 colour and 4 spin indices.
|
||||||
|
We have up to 256 of these including the offdiag (G1 != G2).
|
||||||
|
|
||||||
|
\ /
|
||||||
|
\p1 p1/
|
||||||
|
\ /
|
||||||
|
\ /
|
||||||
|
G1 * * G2
|
||||||
|
/ \
|
||||||
|
/ \
|
||||||
|
/p2 p2\
|
||||||
|
/ \
|
||||||
|
|
||||||
|
*********************************************************************************/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
LOG(Message) << "Computing fourquark contractions '" << getName() << "' using"
|
||||||
|
<< " momentum '" << par().Sin << "' and '" << par().Sout << "'"
|
||||||
|
<< std::endl;
|
||||||
|
|
||||||
|
BinaryWriter writer(par().output);
|
||||||
|
|
||||||
|
PropagatorField1 &Sin = *env().template getObject<PropagatorField1>(par().Sin);
|
||||||
|
PropagatorField2 &Sout = *env().template getObject<PropagatorField2>(par().Sout);
|
||||||
|
std::vector<Real> pin = strToVec<Real>(par().pin), pout = strToVec<Real>(par().pout);
|
||||||
|
bool fullbasis = par().fullbasis;
|
||||||
|
Gamma g5(Gamma::Algebra::Gamma5);
|
||||||
|
Result result;
|
||||||
|
std::vector<Real> latt_size(pin.begin(), pin.end());
|
||||||
|
LatticeComplex pdotxin(env().getGrid()), pdotxout(env().getGrid()), coor(env().getGrid());
|
||||||
|
LatticeSpinColourMatrix bilinear_mu(env().getGrid()), bilinear_nu(env().getGrid());
|
||||||
|
LatticeSpinColourSpinColourMatrix lret(env().getGrid());
|
||||||
|
Complex Ci(0.0,1.0);
|
||||||
|
|
||||||
|
//Phase propagators
|
||||||
|
//Sin = Grid::QCD::PropUtils::PhaseProps(Sin,pin);
|
||||||
|
//Sout = Grid::QCD::PropUtils::PhaseProps(Sout,pout);
|
||||||
|
|
||||||
|
//find p.x for in and out so phase can be accounted for in propagators
|
||||||
|
pdotxin=zero;
|
||||||
|
pdotxout=zero;
|
||||||
|
for (unsigned int mu = 0; mu < 4; ++mu)
|
||||||
|
{
|
||||||
|
Real TwoPiL = M_PI * 2.0/ latt_size[mu];
|
||||||
|
LatticeCoordinate(coor,mu);
|
||||||
|
pdotxin = pdotxin +(TwoPiL * pin[mu]) * coor;
|
||||||
|
pdotxout= pdotxout +(TwoPiL * pout[mu]) * coor;
|
||||||
|
}
|
||||||
|
Sin = Sin*exp(-Ci*pdotxin); //phase corrections
|
||||||
|
Sout = Sout*exp(-Ci*pdotxout);
|
||||||
|
|
||||||
|
|
||||||
|
//Set up Gammas
|
||||||
|
std::vector<Gamma> gammavector;
|
||||||
|
for( int i=1; i<Gamma::nGamma; i+=2){
|
||||||
|
Gamma::Algebra gam = i;
|
||||||
|
gammavector.push_back(Gamma(gam));
|
||||||
|
}
|
||||||
|
|
||||||
|
lret = zero;
|
||||||
|
if (fullbasis == true){ // all combinations of mu and nu
|
||||||
|
result.fourquark.resize(Gamma::nGamma/2*Gamma::nGamma/2);
|
||||||
|
for( int mu=0; mu<Gamma::nGamma/2; mu++){
|
||||||
|
bilinear_mu = g5*adj(Sout)*g5*gammavector[mu]*Sin;
|
||||||
|
for ( int nu=0; nu<Gamma::nGamma; nu++){
|
||||||
|
LatticeSpinColourMatrix bilinear_nu(env().getGrid());
|
||||||
|
bilinear_nu = g5*adj(Sout)*g5*gammavector[nu]*Sin;
|
||||||
|
LOG(Message) << "bilinear_nu for nu = " << nu << " is - " << bilinear_mu << std::endl;
|
||||||
|
result.fourquark[mu*Gamma::nGamma/2 + nu] = zero;
|
||||||
|
tensorprod(lret,bilinear_mu,bilinear_nu);
|
||||||
|
result.fourquark[mu*Gamma::nGamma/2 + nu] = sum(lret);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
result.fourquark.resize(Gamma::nGamma/2);
|
||||||
|
for ( int mu=0; mu<1; mu++){
|
||||||
|
//for( int mu=0; mu<Gamma::nGamma/2; mu++ ){
|
||||||
|
bilinear_mu = g5*adj(Sout)*g5*gammavector[mu]*Sin;
|
||||||
|
//LOG(Message) << "bilinear_mu for mu = " << mu << " is - " << bilinear_mu << std::endl;
|
||||||
|
result.fourquark[mu] = zero;
|
||||||
|
tensorprod(lret,bilinear_mu,bilinear_mu); //tensor outer product
|
||||||
|
result.fourquark[mu] = sum(lret);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
write(writer, "fourquark", result.fourquark);
|
||||||
|
}
|
||||||
|
|
||||||
|
END_MODULE_NAMESPACE
|
||||||
|
|
||||||
|
END_HADRONS_NAMESPACE
|
||||||
|
|
||||||
|
#endif // Hadrons_FourQuark_hpp_
|
||||||
8
Hadrons/Modules/MSource/Momentum.cc
Normal file
8
Hadrons/Modules/MSource/Momentum.cc
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
#include <Hadrons/Modules/MSource/Momentum.hpp>
|
||||||
|
|
||||||
|
using namespace Grid;
|
||||||
|
using namespace Hadrons;
|
||||||
|
using namespace MSource;
|
||||||
|
|
||||||
|
template class Grid::Hadrons::MSource::TMomSource<FIMPL>;
|
||||||
|
|
||||||
121
Hadrons/Modules/MSource/Momentum.hpp
Normal file
121
Hadrons/Modules/MSource/Momentum.hpp
Normal file
@ -0,0 +1,121 @@
|
|||||||
|
#ifndef Hadrons_MomSource_hpp_
|
||||||
|
#define Hadrons_MomSource_hpp_
|
||||||
|
|
||||||
|
#include <Hadrons/Global.hpp>
|
||||||
|
#include <Hadrons/Module.hpp>
|
||||||
|
#include <Hadrons/ModuleFactory.hpp>
|
||||||
|
|
||||||
|
BEGIN_HADRONS_NAMESPACE
|
||||||
|
|
||||||
|
/*
|
||||||
|
Plane Wave source
|
||||||
|
-----------------
|
||||||
|
src_x = e^i2pi/L * p *position
|
||||||
|
*/
|
||||||
|
|
||||||
|
/******************************************************************************
|
||||||
|
* TPlane *
|
||||||
|
******************************************************************************/
|
||||||
|
BEGIN_MODULE_NAMESPACE(MSource)
|
||||||
|
|
||||||
|
class MomSourcePar: Serializable
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
//What is meant by serializable in this context
|
||||||
|
GRID_SERIALIZABLE_CLASS_MEMBERS(MomSourcePar,
|
||||||
|
std::string, mom);
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
template <typename FImpl>
|
||||||
|
class TMomSource: public Module<MomSourcePar>
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
FERM_TYPE_ALIASES(FImpl,);
|
||||||
|
public:
|
||||||
|
// constructor
|
||||||
|
TMomSource(const std::string name);
|
||||||
|
// destructor
|
||||||
|
virtual ~TMomSource(void) = default;
|
||||||
|
// dependency relation
|
||||||
|
virtual std::vector<std::string> getInput(void);
|
||||||
|
virtual std::vector<std::string> getOutput(void);
|
||||||
|
// setup
|
||||||
|
virtual void setup(void);
|
||||||
|
// execution
|
||||||
|
virtual void execute(void);
|
||||||
|
};
|
||||||
|
|
||||||
|
MODULE_REGISTER_TMP(MomSource, TMomSource<FIMPL>, MSource);
|
||||||
|
//MODULE_REGISTER_NS(MomSource, TMomSource, MSource);
|
||||||
|
|
||||||
|
/******************************************************************************
|
||||||
|
* TMomSource template implementation *
|
||||||
|
******************************************************************************/
|
||||||
|
// constructor /////////////////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl>
|
||||||
|
TMomSource<FImpl>::TMomSource(const std::string name)
|
||||||
|
: Module<MomSourcePar>(name)
|
||||||
|
{}
|
||||||
|
|
||||||
|
// dependencies/products ///////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl>
|
||||||
|
std::vector<std::string> TMomSource<FImpl>::getInput(void)
|
||||||
|
{
|
||||||
|
std::vector<std::string> in;
|
||||||
|
return in;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename FImpl>
|
||||||
|
std::vector<std::string> TMomSource<FImpl>::getOutput(void)
|
||||||
|
{
|
||||||
|
std::vector<std::string> out = {getName()};
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// setup ///////////////////////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl>
|
||||||
|
void TMomSource<FImpl>::setup(void)
|
||||||
|
{
|
||||||
|
envCreateLat(PropagatorField, getName());
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
//execution//////////////////////////////////////////////////////////////////
|
||||||
|
template <typename FImpl>
|
||||||
|
void TMomSource<FImpl>::execute(void)
|
||||||
|
{
|
||||||
|
LOG(Message) << "Generating planewave momentum source with momentum " << par().mom << std::endl;
|
||||||
|
//what does this env do?
|
||||||
|
PropagatorField &src = envGet(PropagatorField, getName());
|
||||||
|
Lattice<iScalar<vInteger>> t(env().getGrid());
|
||||||
|
LatticeComplex C(env().getGrid()), coor(env().getGrid());
|
||||||
|
std::vector<Real> p;
|
||||||
|
std::vector<Real> latt_size(GridDefaultLatt().begin(), GridDefaultLatt().end());
|
||||||
|
Complex i(0.0,1.0);
|
||||||
|
|
||||||
|
LOG(Message) << " " << std::endl;
|
||||||
|
//get the momentum from parameters
|
||||||
|
p = strToVec<Real>(par().mom);
|
||||||
|
C = zero;
|
||||||
|
LOG(Message) << "momentum converted from string - " << std::to_string(p[0]) <<std::to_string(p[1]) <<std::to_string(p[2]) << std::to_string(p[3]) << std::endl;
|
||||||
|
for(int mu=0;mu<4;mu++){
|
||||||
|
Real TwoPiL = M_PI * 2.0/ latt_size[mu];
|
||||||
|
LatticeCoordinate(coor,mu);
|
||||||
|
C = C +(TwoPiL * p[mu]) * coor;
|
||||||
|
}
|
||||||
|
C = exp(C*i);
|
||||||
|
LOG(Message) << "exponential of pdotx taken " << std::endl;
|
||||||
|
src = src + C;
|
||||||
|
LOG(Message) << "source created" << std::endl;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
END_MODULE_NAMESPACE
|
||||||
|
|
||||||
|
END_HADRONS_NAMESPACE
|
||||||
|
|
||||||
|
#endif // Hadrons_MomSource_hpp_
|
||||||
Loading…
Reference in New Issue
Block a user