mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-14 01:35:36 +00:00
152 lines
4.8 KiB
C++
152 lines
4.8 KiB
C++
#ifndef Hadrons_MScalarSUN_StochFreeField_hpp_
|
|
#define Hadrons_MScalarSUN_StochFreeField_hpp_
|
|
|
|
#include <Grid/Hadrons/Global.hpp>
|
|
#include <Grid/Hadrons/Module.hpp>
|
|
#include <Grid/Hadrons/ModuleFactory.hpp>
|
|
|
|
BEGIN_HADRONS_NAMESPACE
|
|
|
|
/******************************************************************************
|
|
* stochastic free SU(N) scalar field *
|
|
******************************************************************************/
|
|
BEGIN_MODULE_NAMESPACE(MScalarSUN)
|
|
|
|
class StochFreeFieldPar: Serializable
|
|
{
|
|
public:
|
|
GRID_SERIALIZABLE_CLASS_MEMBERS(StochFreeFieldPar,
|
|
double, m2,
|
|
double, g);
|
|
};
|
|
|
|
template <typename SImpl>
|
|
class TStochFreeField: public Module<StochFreeFieldPar>
|
|
{
|
|
public:
|
|
typedef typename SImpl::Field Field;
|
|
typedef typename SImpl::ComplexField ComplexField;
|
|
typedef typename SImpl::Group Group;
|
|
typedef typename SImpl::SiteField::scalar_object Site;
|
|
public:
|
|
// constructor
|
|
TStochFreeField(const std::string name);
|
|
// destructor
|
|
virtual ~TStochFreeField(void) {};
|
|
// 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);
|
|
private:
|
|
bool create_weight;
|
|
};
|
|
|
|
MODULE_REGISTER_TMP(StochFreeFieldSU2, TStochFreeField<ScalarNxNAdjImplR<2>>, MScalarSUN);
|
|
MODULE_REGISTER_TMP(StochFreeFieldSU3, TStochFreeField<ScalarNxNAdjImplR<3>>, MScalarSUN);
|
|
MODULE_REGISTER_TMP(StochFreeFieldSU4, TStochFreeField<ScalarNxNAdjImplR<4>>, MScalarSUN);
|
|
MODULE_REGISTER_TMP(StochFreeFieldSU5, TStochFreeField<ScalarNxNAdjImplR<5>>, MScalarSUN);
|
|
MODULE_REGISTER_TMP(StochFreeFieldSU6, TStochFreeField<ScalarNxNAdjImplR<6>>, MScalarSUN);
|
|
|
|
/******************************************************************************
|
|
* TStochFreeField implementation *
|
|
******************************************************************************/
|
|
// constructor /////////////////////////////////////////////////////////////////
|
|
template <typename SImpl>
|
|
TStochFreeField<SImpl>::TStochFreeField(const std::string name)
|
|
: Module<StochFreeFieldPar>(name)
|
|
{}
|
|
|
|
// dependencies/products ///////////////////////////////////////////////////////
|
|
template <typename SImpl>
|
|
std::vector<std::string> TStochFreeField<SImpl>::getInput(void)
|
|
{
|
|
std::vector<std::string> in;
|
|
|
|
return in;
|
|
}
|
|
|
|
template <typename SImpl>
|
|
std::vector<std::string> TStochFreeField<SImpl>::getOutput(void)
|
|
{
|
|
std::vector<std::string> out = {getName()};
|
|
|
|
return out;
|
|
}
|
|
|
|
// setup ///////////////////////////////////////////////////////////////////////
|
|
template <typename SImpl>
|
|
void TStochFreeField<SImpl>::setup(void)
|
|
{
|
|
create_weight = false;
|
|
if (!env().hasCreatedObject("_" + getName() + "_weight"))
|
|
{
|
|
envCacheLat(ComplexField, "_" + getName() + "_weight");
|
|
create_weight = true;
|
|
}
|
|
envTmpLat(Field, "phift");
|
|
envTmpLat(ComplexField, "ca");
|
|
envCreateLat(Field, getName());
|
|
}
|
|
|
|
// execution ///////////////////////////////////////////////////////////////////
|
|
template <typename SImpl>
|
|
void TStochFreeField<SImpl>::execute(void)
|
|
{
|
|
LOG(Message) << "Generating stochastic scalar field" << std::endl;
|
|
|
|
const unsigned int N = Group::Dimension;
|
|
const unsigned int Nadj = Group::AdjointDimension;
|
|
auto &phi = envGet(Field, getName());
|
|
auto &w = envGet(ComplexField, "_" + getName() + "_weight");
|
|
auto &rng = *env().get4dRng();
|
|
double trphi2;
|
|
FFT fft(env().getGrid());
|
|
Integer vol;
|
|
|
|
vol = 1;
|
|
for(int d = 0; d < env().getNd(); d++)
|
|
{
|
|
vol = vol*env().getDim(d);
|
|
}
|
|
if (create_weight)
|
|
{
|
|
LOG(Message) << "Caching momentum-space scalar action" << std::endl;
|
|
|
|
SImpl::MomentumSpacePropagator(w, sqrt(par().m2));
|
|
w *= par().g/N;
|
|
w = sqrt(vol)*sqrt(w);
|
|
}
|
|
LOG(Message) << "Generating random momentum-space field" << std::endl;
|
|
envGetTmp(Field, phift);
|
|
envGetTmp(ComplexField, ca);
|
|
phift = zero;
|
|
for (int a = 0; a < Nadj; ++a)
|
|
{
|
|
Site ta;
|
|
|
|
gaussian(rng, ca);
|
|
Group::generator(a, ta);
|
|
phift += ca*ta;
|
|
}
|
|
phift *= w;
|
|
LOG(Message) << "Field Fourier transform" << std::endl;
|
|
fft.FFT_all_dim(phi, phift, FFT::backward);
|
|
phi = 0.5*(phi - adj(phi));
|
|
trphi2 = -TensorRemove(sum(trace(phi*phi))).real()/vol;
|
|
LOG(Message) << "tr(phi^2)= " << trphi2 << std::endl;
|
|
|
|
// ComplexField phi2(env().getGrid());
|
|
|
|
// phi2=trace(phi*phi);
|
|
// std::cout << phi2 << std::endl;
|
|
}
|
|
|
|
END_MODULE_NAMESPACE
|
|
|
|
END_HADRONS_NAMESPACE
|
|
|
|
#endif // Hadrons_MScalarSUN_StochFreeField_hpp_
|