add modules MSource::Gauss and MSource::Convolution

Hadrons/Modules/MSource/Convolution.cc

@@ -0,0 +1,7 @@
+#include <Hadrons/Modules/MSource/Convolution.hpp>
+
+using namespace Grid;
+using namespace Hadrons;
+using namespace MSource;
+
+template class Grid::Hadrons::MSource::TConvolution<FIMPL>;

Hadrons/Modules/MSource/Convolution.hpp

@@ -0,0 +1,110 @@
+#ifndef Hadrons_MSource_Convolution_hpp_
+#define Hadrons_MSource_Convolution_hpp_
+
+#include <Hadrons/Global.hpp>
+#include <Hadrons/Module.hpp>
+#include <Hadrons/ModuleFactory.hpp>
+
+BEGIN_HADRONS_NAMESPACE
+
+/******************************************************************************
+ *                         Convolution                                 *
+ ******************************************************************************/
+BEGIN_MODULE_NAMESPACE(MSource)
+
+class ConvolutionPar: Serializable
+{
+public:
+    GRID_SERIALIZABLE_CLASS_MEMBERS(ConvolutionPar,
+                                    std::string, field1,
+                                    std::string, field2);
+};
+
+template <typename FImpl>
+class TConvolution: public Module<ConvolutionPar>
+{
+public:
+    BASIC_TYPE_ALIASES(FImpl,);
+public:
+    // constructor
+    TConvolution(const std::string name);
+    // destructor
+    virtual ~TConvolution(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);
+};
+
+MODULE_REGISTER_TMP(Convolution, TConvolution<FIMPL>, MSource);
+
+/******************************************************************************
+ *                 TConvolution implementation                             *
+ ******************************************************************************/
+// constructor /////////////////////////////////////////////////////////////////
+template <typename FImpl>
+TConvolution<FImpl>::TConvolution(const std::string name)
+: Module<ConvolutionPar>(name)
+{}
+
+// dependencies/products ///////////////////////////////////////////////////////
+template <typename FImpl>
+std::vector<std::string> TConvolution<FImpl>::getInput(void)
+{
+    std::vector<std::string> in;
+    
+    return in;
+}
+
+template <typename FImpl>
+std::vector<std::string> TConvolution<FImpl>::getOutput(void)
+{
+    std::vector<std::string> out = {getName()};
+    
+    return out;
+}
+
+// setup ///////////////////////////////////////////////////////////////////////
+template <typename FImpl>
+void TConvolution<FImpl>::setup(void)
+{
+    envCreateLat(LatticeFermion, getName());
+    envTmpLat(LatticeComplex, "momfield1");
+    envTmp(FFT, "fft", 1, env().getGrid());
+}
+
+// execution ///////////////////////////////////////////////////////////////////
+template <typename FImpl>
+void TConvolution<FImpl>::execute(void)
+{
+    auto &field1 = envGet(LatticeComplex, par().field1);
+    auto &field2 = envGet(LatticeFermion, par().field2);
+    auto &out    = envGet(LatticeFermion, getName());
+    envGetTmp(LatticeComplex, momfield1);
+    envGetTmp(FFT, fft);
+
+    std::vector<int> mask(env().getNd(), 1);
+    mask.back()=0; //transform only the spatial dimensions
+
+    startTimer("Fourier transform");
+    fft.FFT_dim_mask(momfield1, field1, mask, FFT::forward);
+    fft.FFT_dim_mask(out,       field2, mask, FFT::forward);
+    stopTimer("Fourier transform");
+
+    startTimer("momentum-space multiplication");
+    out=momfield1*out;
+    stopTimer("momentum-space multiplication");
+
+    startTimer("Fourier transform");
+    fft.FFT_dim_mask(out, out, mask, FFT::backward);
+    stopTimer("Fourier transform");
+}
+
+END_MODULE_NAMESPACE
+
+END_HADRONS_NAMESPACE
+
+#endif // Hadrons_MSource_Convolution_hpp_

Hadrons/Modules/MSource/Gauss.cc

@@ -0,0 +1,7 @@
+#include <Hadrons/Modules/MSource/Gauss.hpp>
+
+using namespace Grid;
+using namespace Hadrons;
+using namespace MSource;
+
+template class Grid::Hadrons::MSource::TGauss<FIMPL>;

Hadrons/Modules/MSource/Gauss.hpp

@@ -0,0 +1,111 @@
+#ifndef Hadrons_MSource_Gauss_hpp_
+#define Hadrons_MSource_Gauss_hpp_
+
+#include <Hadrons/Global.hpp>
+#include <Hadrons/Module.hpp>
+#include <Hadrons/ModuleFactory.hpp>
+
+BEGIN_HADRONS_NAMESPACE
+
+/******************************************************************************
+ *                         Gauss                                              *
+ * result[n] = 1/(sqrt(2*pi)*width)^dim*exp(-|n|^2/(2*width^2))               *
+ * where:                                                                     *
+ *   n=(n[0],n[1],...,n[dim-1])  (lattice coordinate)                         *
+ *   dim=Nd-1                                                                 *
+ ******************************************************************************/
+BEGIN_MODULE_NAMESPACE(MSource)
+
+class GaussPar: Serializable
+{
+public:
+    GRID_SERIALIZABLE_CLASS_MEMBERS(GaussPar,
+                                    std::string, position,
+                                    double,      width);
+};
+
+template <typename FImpl>
+class TGauss: public Module<GaussPar>
+{
+public:
+    // constructor
+    TGauss(const std::string name);
+    // destructor
+    virtual ~TGauss(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:
+    std::vector<int> position_;
+};
+
+MODULE_REGISTER_TMP(Gauss, TGauss<FIMPL>, MSource);
+
+/******************************************************************************
+ *                 TGauss implementation                             *
+ ******************************************************************************/
+// constructor /////////////////////////////////////////////////////////////////
+template <typename FImpl>
+TGauss<FImpl>::TGauss(const std::string name)
+: Module<GaussPar>(name)
+{}
+
+// dependencies/products ///////////////////////////////////////////////////////
+template <typename FImpl>
+std::vector<std::string> TGauss<FImpl>::getInput(void)
+{
+    std::vector<std::string> in;
+    
+    return in;
+}
+
+template <typename FImpl>
+std::vector<std::string> TGauss<FImpl>::getOutput(void)
+{
+    std::vector<std::string> out = {getName()};
+    
+    return out;
+}
+
+// setup ///////////////////////////////////////////////////////////////////////
+template <typename FImpl>
+void TGauss<FImpl>::setup(void)
+{
+    
+}
+
+// execution ///////////////////////////////////////////////////////////////////
+template <typename FImpl>
+void TGauss<FImpl>::execute(void)
+{
+    auto &rho = envGet(LatticeComplex, getName());
+    envGetTmp(LatticeComplex, component);
+    const int dim=env().getNd()-1;
+    const double fact=-0.5/std::pow(par().width,2);
+    const std::vector<int> latt_size { env().getGrid()->FullDimensions() };
+
+    //exp(fact*|n|^2)
+    rho=zero;
+    for(int mu=0; mu<dim; mu++) {
+        LatticeCoordinate(component, mu);
+        //FIXME: the next three lines are very inefficient...
+        //       should not need any communication (Cshift) here
+        assert(latt_size[mu]%2==0);
+        component-=Complex(latt_size[mu]/2-1);
+        component=Cshift(component, mu, latt_size[mu]/2-1 -position_[mu]);
+        rho+=component*component*fact;
+    }
+    rho=exp(rho);
+
+    rho*=static_cast<Complex>(std::pow(sqrt(2*M_PI)*par().width,dim));
+}
+
+END_MODULE_NAMESPACE
+
+END_HADRONS_NAMESPACE
+
+#endif // Hadrons_MSource_Gauss_hpp_