diff --git a/extras/Hadrons/Modules/MScalarSUN/TwoPoint.hpp b/extras/Hadrons/Modules/MScalarSUN/TwoPoint.hpp index 5227e95a..5ec56714 100644 --- a/extras/Hadrons/Modules/MScalarSUN/TwoPoint.hpp +++ b/extras/Hadrons/Modules/MScalarSUN/TwoPoint.hpp @@ -45,6 +45,7 @@ class TwoPointPar: Serializable public: GRID_SERIALIZABLE_CLASS_MEMBERS(TwoPointPar, std::vector, op, + std::vector, mom, std::string, output); }; @@ -52,14 +53,17 @@ template class TTwoPoint: public Module { public: - typedef typename SImpl::Field Field; - typedef typename SImpl::ComplexField ComplexField; + typedef typename SImpl::Field Field; + typedef typename SImpl::ComplexField ComplexField; + typedef std::vector SlicedOp; + class Result: Serializable { public: GRID_SERIALIZABLE_CLASS_MEMBERS(Result, std::string, sink, std::string, source, + std::vector, mom, std::vector, data); }; public: @@ -79,6 +83,8 @@ private: template std::vector makeTwoPoint(const std::vector &sink, const std::vector &source); +private: + std::vector> mom_; }; MODULE_REGISTER_NS(TwoPointSU2, TTwoPoint>, MScalarSUN); @@ -115,9 +121,19 @@ std::vector TTwoPoint::getOutput(void) template void TTwoPoint::setup(void) { - const unsigned int nt = env().getDim().back(); - envTmp(std::vector>, "slicedOp", 1, par().op.size(), - std::vector(nt)); + const unsigned int nd = env().getDim().size(); + + mom_.resize(par().mom.size()); + for (unsigned int i = 0; i < mom_.size(); ++i) + { + mom_[i] = strToVec(par().mom[i]); + if (mom_[i].size() != nd - 1) + { + HADRON_ERROR(Size, "momentum number of components different from " + + std::to_string(nd-1)); + } + } + envTmpLat(ComplexField, "ftBuf"); } // execution /////////////////////////////////////////////////////////////////// @@ -130,24 +146,48 @@ void TTwoPoint::execute(void) LOG(Message) << " '" << o << "'" << std::endl; } - const unsigned int nd = env().getDim().size(); - std::vector result; - - envGetTmp(std::vector>, slicedOp); - for (unsigned int i = 0; i < par().op.size(); ++i) + const unsigned int nd = env().getDim().size(); + const unsigned int nt = env().getDim().back(); + const unsigned int nop = par().op.size(); + const unsigned int nmom = mom_.size(); + std::vector dMask(nd, 1); + std::vector result; + std::vector> slicedOp(nop); + FFT fft(env().getGrid()); + + envGetTmp(ComplexField, ftBuf); + dMask[nd - 1] = 0; + for (unsigned int i = 0; i < nop; ++i) { auto &op = envGet(ComplexField, par().op[i]); - sliceSum(op, slicedOp[i], nd - 1); + slicedOp[i].resize(nmom); + LOG(Message) << "Operator '" << par().op[i] << "' FFT" << std::endl; + fft.FFT_dim_mask(ftBuf, op, dMask, FFT::forward); + for (unsigned int m = 0; m < nmom; ++m) + { + auto qt = mom_[m]; + + qt.resize(nd); + slicedOp[i][m].resize(nt); + for (unsigned int t = 0; t < nt; ++t) + { + qt[nd - 1] = t; + peekSite(slicedOp[i][m][t], ftBuf, qt); + } + } } - for (unsigned int i = 0; i < par().op.size(); ++i) - for (unsigned int j = 0; j < par().op.size(); ++j) + LOG(Message) << "Making contractions" << std::endl; + for (unsigned int m = 0; m < nmom; ++m) + for (unsigned int i = 0; i < nop; ++i) + for (unsigned int j = 0; j < nop; ++j) { Result r; r.sink = par().op[i]; r.source = par().op[j]; - r.data = makeTwoPoint(slicedOp[i], slicedOp[j]); + r.mom = mom_[m]; + r.data = makeTwoPoint(slicedOp[i][m], slicedOp[j][m]); result.push_back(r); } saveResult(par().output, "twopt", result); @@ -169,7 +209,7 @@ std::vector TTwoPoint::makeTwoPoint( { for (unsigned int t = 0; t < nt; ++t) { - res[dt] += TensorRemove(trace(sink[(t+dt)%nt]*source[t])); + res[dt] += TensorRemove(trace(sink[(t+dt)%nt]*adj(source[t]))); } res[dt] *= 1./static_cast(nt); }