portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-09 21:50:45 +01:00
Code Releases Activity

Add functions momD1 and momD2 to ScalarVP

Browse Source
This commit is contained in:
James Harrison 2017-03-27 16:49:18 +01:00
parent 4512dbdf58
commit 9f755e0379
2 changed files with 72 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
extras/Hadrons/Modules/MScalar/ScalarVP.cc
View File

@ -15,7 +15,7 @@ TScalarVP::TScalarVP(const std::string name)
// dependencies/products /////////////////////////////////////////////////////// // dependencies/products ///////////////////////////////////////////////////////
std::vector<std::string> TScalarVP::getInput(void) std::vector<std::string> TScalarVP::getInput(void)
{ {
prop0Name_ = par().scalarProp + "_0"; prop0Name_ = par().scalarProp + "_0";
propD1Name_ = par().scalarProp + "_D1"; propD1Name_ = par().scalarProp + "_D1";
propD1D1Name_ = par().scalarProp + "_D1D1"; propD1D1Name_ = par().scalarProp + "_D1D1";
propD2Name_ = par().scalarProp + "_D2"; propD2Name_ = par().scalarProp + "_D2";
@ -36,11 +36,77 @@ std::vector<std::string> TScalarVP::getOutput(void)
// setup /////////////////////////////////////////////////////////////////////// // setup ///////////////////////////////////////////////////////////////////////
void TScalarVP::setup(void) void TScalarVP::setup(void)
{ {
phaseName_.clear();
for (unsigned int mu = 0; mu < env().getNd(); ++mu)
{
phaseName_.push_back("_shiftphase_" + std::to_string(mu));
}
} }
// execution /////////////////////////////////////////////////////////////////// // execution ///////////////////////////////////////////////////////////////////
void TScalarVP::execute(void) void TScalarVP::execute(void)
{ {
for (unsigned int mu = 0; mu < env().getNd(); ++mu)
{
phase_.push_back(env().getObject<ScalarField>(phaseName_[mu]));
}
} }
void TScalarVP::momD1(ScalarField &s, EmField &A, FFT &fft)
{
ScalarField buf(env().getGrid()), result(env().getGrid()),
Amu(env().getGrid());
Complex ci(0.0,1.0);
result = zero;
for (unsigned int mu = 0; mu < env().getNd(); ++mu)
{
Amu = peekLorentz(A, mu);
buf = (*phase_[mu])*s;
fft.FFT_all_dim(buf, buf, FFT::backward);
buf = Amu*buf;
fft.FFT_all_dim(buf, buf, FFT::forward);
result = result - ci*buf;
}
fft.FFT_all_dim(s, s, FFT::backward);
for (unsigned int mu = 0; mu < env().getNd(); ++mu)
{
Amu = peekLorentz(A, mu);
buf = Amu*s;
fft.FFT_all_dim(buf, buf, FFT::forward);
result = result + ci*adj(*phase_[mu])*buf;
}
s = result;
}
void TScalarVP::momD2(ScalarField &s, EmField &Asquared, FFT &fft)
{
ScalarField buf(env().getGrid()), result(env().getGrid()),
A2mu(env().getGrid());
result = zero;
for (unsigned int mu = 0; mu < env().getNd(); ++mu)
{
A2mu = peekLorentz(Asquared, mu);
buf = (*phase_[mu])*s;
fft.FFT_all_dim(buf, buf, FFT::backward);
buf = A2mu*buf;
fft.FFT_all_dim(buf, buf, FFT::forward);
result = result + .5*buf;
}
fft.FFT_all_dim(s, s, FFT::backward);
for (unsigned int mu = 0; mu < env().getNd(); ++mu)
{
A2mu = peekLorentz(Asquared, mu);
buf = A2mu*s;
fft.FFT_all_dim(buf, buf, FFT::forward);
result = result + .5*adj(*phase_[mu])*buf;
}
s = result;
}

5
extras/Hadrons/Modules/MScalar/ScalarVP.hpp
View File

@ -41,8 +41,13 @@ public:
virtual void setup(void); virtual void setup(void);
// execution // execution
virtual void execute(void); virtual void execute(void);
private:
void momD1(ScalarField &s, EmField &A, FFT &fft);
void momD2(ScalarField &s, EmField &Asquared, FFT &fft);
private: private:
std::string prop0Name_, propD1Name_, propD1D1Name_, propD2Name_; std::string prop0Name_, propD1Name_, propD1D1Name_, propD2Name_;
std::vector<std::string> phaseName_;
std::vector<ScalarField *> phase_;
}; };
MODULE_REGISTER_NS(ScalarVP, TScalarVP, MScalar); MODULE_REGISTER_NS(ScalarVP, TScalarVP, MScalar);