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

QedFVol: Change format of scalar VP output files, and save diagrams without charge factors for consistency with ChargedProp module.

Browse Source
This commit is contained in:
James Harrison 2018-02-07 20:31:50 +00:00
parent 594a262dcc
commit 9f202782c5
2 changed files with 129 additions and 73 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -236,22 +236,38 @@ void TScalarVP::execute(void)
vpTensor.push_back(vpTensor_mu); vpTensor.push_back(vpTensor_mu);
} }
// Open output files if necessary // Prepare output data structure if necessary
std::vector<ResultWriter *> writer; Result outputData;
if (!par().output.empty()) if (!par().output.empty())
{ {
LOG(Message) << "Preparing output files..." << std::endl; outputData.projection.resize(par().outputMom.size());
outputData.lattice_size = env().getGrid()->_fdimensions;
outputData.mass = static_cast<TChargedProp *>(vm().getModule(par().scalarProp))->par().mass;
outputData.charge = q;
for (unsigned int i_p = 0; i_p < par().outputMom.size(); ++i_p) for (unsigned int i_p = 0; i_p < par().outputMom.size(); ++i_p)
{ {
std::vector<int> mom = strToVec<int>(par().outputMom[i_p]); outputData.projection[i_p].momentum = strToVec<int>(par().outputMom[i_p]);
std::string filename = par().output + "_" outputData.projection[i_p].pi.resize(env().getNd());
+ std::to_string(mom[0]) outputData.projection[i_p].pi_free.resize(env().getNd());
+ std::to_string(mom[1]) outputData.projection[i_p].pi_2E.resize(env().getNd());
+ std::to_string(mom[2]); outputData.projection[i_p].pi_2T.resize(env().getNd());
ResultWriter *writer_i = new ResultWriter(RESULT_FILE_NAME(filename)); outputData.projection[i_p].pi_S.resize(env().getNd());
writer.push_back(writer_i); outputData.projection[i_p].pi_4C.resize(env().getNd());
write(*writer_i, "charge", q); outputData.projection[i_p].pi_X.resize(env().getNd());
write(*writer_i, "mass", static_cast<TChargedProp *>(vm().getModule(par().scalarProp))->par().mass); outputData.projection[i_p].pi_srcT.resize(env().getNd());
outputData.projection[i_p].pi_snkT.resize(env().getNd());
for (unsigned int nu = 0; nu < env().getNd(); ++nu)
{
outputData.projection[i_p].pi[nu].resize(env().getNd());
outputData.projection[i_p].pi_free[nu].resize(env().getNd());
outputData.projection[i_p].pi_2E[nu].resize(env().getNd());
outputData.projection[i_p].pi_2T[nu].resize(env().getNd());
outputData.projection[i_p].pi_S[nu].resize(env().getNd());
outputData.projection[i_p].pi_4C[nu].resize(env().getNd());
outputData.projection[i_p].pi_X[nu].resize(env().getNd());
outputData.projection[i_p].pi_srcT[nu].resize(env().getNd());
outputData.projection[i_p].pi_snkT[nu].resize(env().getNd());
}
} }
} }
@ -272,33 +288,42 @@ void TScalarVP::execute(void)
Usrc = Complex(1.0,0.0); Usrc = Complex(1.0,0.0);
vpContraction(result, prop0, tmpProp, Usrc, mu); vpContraction(result, prop0, tmpProp, Usrc, mu);
*vpTensor[mu][nu] = result; *vpTensor[mu][nu] = result;
// Output if necessary // Do momentum projections if necessary
if (!par().output.empty()) if (!par().output.empty())
{ {
writeVP(writer, result, for (unsigned int i_p = 0; i_p < par().outputMom.size(); ++i_p)
"Pi_free_"+std::to_string(mu)+"_"+std::to_string(nu)); {
project(outputData.projection[i_p].pi_free[mu][nu], result,
i_p);
}
} }
tmpProp = result; // Just using tmpProp as a temporary ScalarField tmpProp = result; // Just using tmpProp as a temporary ScalarField
// here (buf is modified by calls to writeVP()) // here (buf is modified by calls to writeVP())
// srcT // srcT
result = tmpProp * (-0.5)*q*q*Anu0*Anu0; result = tmpProp * (-0.5)*Anu0*Anu0;
*vpTensor[mu][nu] += result; *vpTensor[mu][nu] += q*q*result;
// Output if necessary // Do momentum projections if necessary
if (!par().output.empty()) if (!par().output.empty())
{ {
writeVP(writer, result, for (unsigned int i_p = 0; i_p < par().outputMom.size(); ++i_p)
"Pi_srcT_"+std::to_string(mu)+"_"+std::to_string(nu)); {
project(outputData.projection[i_p].pi_srcT[mu][nu], result,
i_p);
}
} }
// snkT // snkT
result = tmpProp * (-0.5)*q*q*Amu*Amu; result = tmpProp * (-0.5)*Amu*Amu;
*vpTensor[mu][nu] += result; *vpTensor[mu][nu] += q*q*result;
// Output if necessary // Do momentum projections if necessary
if (!par().output.empty()) if (!par().output.empty())
{ {
writeVP(writer, result, for (unsigned int i_p = 0; i_p < par().outputMom.size(); ++i_p)
"Pi_snkT_"+std::to_string(mu)+"_"+std::to_string(nu)); {
project(outputData.projection[i_p].pi_snkT[mu][nu], result,
i_p);
}
} }
// S // S
@ -306,13 +331,15 @@ void TScalarVP::execute(void)
Usrc = ci*Anu0; Usrc = ci*Anu0;
Usnk = ci*Amu; Usnk = ci*Amu;
vpContraction(result, prop0, tmpProp, Usrc, Usnk, mu); vpContraction(result, prop0, tmpProp, Usrc, Usnk, mu);
result = q*q*result; *vpTensor[mu][nu] += q*q*result;
*vpTensor[mu][nu] += result; // Do momentum projections if necessary
// Output if necessary
if (!par().output.empty()) if (!par().output.empty())
{ {
writeVP(writer, result, for (unsigned int i_p = 0; i_p < par().outputMom.size(); ++i_p)
"Pi_S_"+std::to_string(mu)+"_"+std::to_string(nu)); {
project(outputData.projection[i_p].pi_S[mu][nu], result,
i_p);
}
} }
// 4C // 4C
@ -329,25 +356,29 @@ void TScalarVP::execute(void)
Usnk = ci*Amu; Usnk = ci*Amu;
vpContraction(buf, prop0, *muPropQ[nu], Usrc, Usnk, mu); vpContraction(buf, prop0, *muPropQ[nu], Usrc, Usnk, mu);
result += buf; result += buf;
result = q*q*result; *vpTensor[mu][nu] += q*q*result;
*vpTensor[mu][nu] += result; // Do momentum projections if necessary
// Output if necessary
if (!par().output.empty()) if (!par().output.empty())
{ {
writeVP(writer, result, for (unsigned int i_p = 0; i_p < par().outputMom.size(); ++i_p)
"Pi_4C_"+std::to_string(mu)+"_"+std::to_string(nu)); {
project(outputData.projection[i_p].pi_4C[mu][nu], result,
i_p);
}
} }
// X // X
Usrc = Complex(1.0,0.0); Usrc = Complex(1.0,0.0);
vpContraction(result, propQ, *muPropQ[nu], Usrc, mu); vpContraction(result, propQ, *muPropQ[nu], Usrc, mu);
result = q*q*result; *vpTensor[mu][nu] += q*q*result;
*vpTensor[mu][nu] += result; // Do momentum projections if necessary
// Output if necessary
if (!par().output.empty()) if (!par().output.empty())
{ {
writeVP(writer, result, for (unsigned int i_p = 0; i_p < par().outputMom.size(); ++i_p)
"Pi_X_"+std::to_string(mu)+"_"+std::to_string(nu)); {
project(outputData.projection[i_p].pi_X[mu][nu], result,
i_p);
}
} }
// 2E // 2E
@ -358,13 +389,15 @@ void TScalarVP::execute(void)
//(Note: <S(0|x-a\hat{\nu})> = <S(a\hat{\nu}|x)>) //(Note: <S(0|x-a\hat{\nu})> = <S(a\hat{\nu}|x)>)
vpContraction(buf, prop0, tmpProp, Usrc, mu); vpContraction(buf, prop0, tmpProp, Usrc, mu);
result += buf; result += buf;
result = q*q*result; *vpTensor[mu][nu] += q*q*result;
*vpTensor[mu][nu] += result; // Do momentum projections if necessary
// Output if necessary
if (!par().output.empty()) if (!par().output.empty())
{ {
writeVP(writer, result, for (unsigned int i_p = 0; i_p < par().outputMom.size(); ++i_p)
"Pi_2E_"+std::to_string(mu)+"_"+std::to_string(nu)); {
project(outputData.projection[i_p].pi_2E[mu][nu], result,
i_p);
}
} }
// 2T // 2T
@ -374,29 +407,36 @@ void TScalarVP::execute(void)
tmpProp = Cshift(propTad, nu, -1); // S_T(0|x-a\hat{\nu}) tmpProp = Cshift(propTad, nu, -1); // S_T(0|x-a\hat{\nu})
vpContraction(buf, prop0, tmpProp, Usrc, mu); vpContraction(buf, prop0, tmpProp, Usrc, mu);
result += buf; result += buf;
result = q*q*result; *vpTensor[mu][nu] += q*q*result;
*vpTensor[mu][nu] += result; // Do momentum projections if necessary
// Output if necessary
if (!par().output.empty()) if (!par().output.empty())
{ {
writeVP(writer, result, for (unsigned int i_p = 0; i_p < par().outputMom.size(); ++i_p)
"Pi_2T_"+std::to_string(mu)+"_"+std::to_string(nu)); {
project(outputData.projection[i_p].pi_2T[mu][nu], result,
i_p);
}
} }
// Output full VP if necessary // Do momentum projections of full VP if necessary
if (!par().output.empty()) if (!par().output.empty())
{ {
writeVP(writer, *vpTensor[mu][nu], for (unsigned int i_p = 0; i_p < par().outputMom.size(); ++i_p)
"Pi_"+std::to_string(mu)+"_"+std::to_string(nu)); {
project(outputData.projection[i_p].pi[mu][nu],
*vpTensor[mu][nu], i_p);
}
} }
} }
} }
// OUTPUT IF NECESSARY
if (!par().output.empty()) if (!par().output.empty())
{ {
for (unsigned int i_p = 0; i_p < par().outputMom.size(); ++i_p) LOG(Message) << "Saving momentum-projected HVP to '"
{ << RESULT_FILE_NAME(par().output) << "'..."
delete writer[i_p]; << std::endl;
} saveResult(par().output, "HVP", outputData);
} }
} }
@ -450,22 +490,17 @@ void TScalarVP::vpContraction(ScalarField &vp,
vp = 2.0*real(vp); vp = 2.0*real(vp);
} }
void TScalarVP::writeVP(std::vector<ResultWriter *> &writer, const ScalarField &vp, std::string dsetName) void TScalarVP::project(std::vector<Complex> &projection, const ScalarField &vp, int i_p)
{ {
std::vector<TComplex> vecBuf; std::vector<TComplex> vecBuf;
std::vector<Complex> result;
envGetTmp(ScalarField, buf); envGetTmp(ScalarField, buf);
for (unsigned int i_p = 0; i_p < par().outputMom.size(); ++i_p) buf = vp*(*momPhase_[i_p]);
sliceSum(buf, vecBuf, Tp);
projection.resize(vecBuf.size());
for (unsigned int t = 0; t < vecBuf.size(); ++t)
{ {
buf = vp*(*momPhase_[i_p]); projection[t] = TensorRemove(vecBuf[t]);
sliceSum(buf, vecBuf, Tp);
result.resize(vecBuf.size());
for (unsigned int t = 0; t < vecBuf.size(); ++t)
{
result[t] = TensorRemove(vecBuf[t]);
}
write(*writer[i_p], dsetName, result);
} }
} }

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

@ -28,6 +28,30 @@ public:
SCALAR_TYPE_ALIASES(SIMPL,); SCALAR_TYPE_ALIASES(SIMPL,);
typedef PhotonR::GaugeField EmField; typedef PhotonR::GaugeField EmField;
typedef PhotonR::GaugeLinkField EmComp; typedef PhotonR::GaugeLinkField EmComp;
class Result: Serializable
{
public:
class Projection: Serializable
{
public:
GRID_SERIALIZABLE_CLASS_MEMBERS(Projection,
std::vector<int>, momentum,
std::vector<std::vector<std::vector<Complex>>>, pi,
std::vector<std::vector<std::vector<Complex>>>, pi_free,
std::vector<std::vector<std::vector<Complex>>>, pi_2E,
std::vector<std::vector<std::vector<Complex>>>, pi_2T,
std::vector<std::vector<std::vector<Complex>>>, pi_S,
std::vector<std::vector<std::vector<Complex>>>, pi_4C,
std::vector<std::vector<std::vector<Complex>>>, pi_X,
std::vector<std::vector<std::vector<Complex>>>, pi_srcT,
std::vector<std::vector<std::vector<Complex>>>, pi_snkT);
};
GRID_SERIALIZABLE_CLASS_MEMBERS(Result,
std::vector<int>, lattice_size,
double, mass,
double, charge,
std::vector<Projection>, projection);
};
public: public:
// constructor // constructor
TScalarVP(const std::string name); TScalarVP(const std::string name);
@ -52,8 +76,8 @@ private:
ScalarField &prop_0_x, ScalarField &prop_nu_x, ScalarField &prop_0_x, ScalarField &prop_nu_x,
TComplex u_src, int mu); TComplex u_src, int mu);
// write momentum-projected vacuum polarisation to file(s) // write momentum-projected vacuum polarisation to file(s)
void writeVP(std::vector<ResultWriter *> &writer, const ScalarField &vp, void project(std::vector<Complex> &projection, const ScalarField &vp,
std::string dsetName); int i_p);
// momentum-space Delta_1 insertion // momentum-space Delta_1 insertion
void momD1(ScalarField &s, FFT &fft); void momD1(ScalarField &s, FFT &fft);
private: private:
@ -65,10 +89,7 @@ private:
std::vector<std::string> phaseName_, muPropQName_, std::vector<std::string> phaseName_, muPropQName_,
momPhaseName_; momPhaseName_;
std::vector<std::vector<std::string> > vpTensorName_; std::vector<std::vector<std::string> > vpTensorName_;
// ScalarField *freeMomProp_, *GFSrc_,
// *prop0_;
std::vector<ScalarField *> phase_, momPhase_; std::vector<ScalarField *> phase_, momPhase_;
// EmField *A;
}; };
MODULE_REGISTER_NS(ScalarVP, TScalarVP, MScalar); MODULE_REGISTER_NS(ScalarVP, TScalarVP, MScalar);