From a13106da0cb40469cfecb281ee619186bf7f95fc Mon Sep 17 00:00:00 2001 From: Antonin Portelli Date: Tue, 24 Apr 2018 19:49:51 +0100 Subject: [PATCH] Hadrons: scalar SU(N) gradient --- extras/Hadrons/Modules.hpp | 1 + extras/Hadrons/Modules/MScalarSUN/Grad.cc | 38 +++++ extras/Hadrons/Modules/MScalarSUN/Grad.hpp | 165 +++++++++++++++++++++ extras/Hadrons/modules.inc | 2 + 4 files changed, 206 insertions(+) create mode 100644 extras/Hadrons/Modules/MScalarSUN/Grad.cc create mode 100644 extras/Hadrons/Modules/MScalarSUN/Grad.hpp diff --git a/extras/Hadrons/Modules.hpp b/extras/Hadrons/Modules.hpp index 528faecd..fc536393 100644 --- a/extras/Hadrons/Modules.hpp +++ b/extras/Hadrons/Modules.hpp @@ -69,6 +69,7 @@ See the full license in the file "LICENSE" in the top level distribution directo #include #include #include +#include #include #include #include diff --git a/extras/Hadrons/Modules/MScalarSUN/Grad.cc b/extras/Hadrons/Modules/MScalarSUN/Grad.cc new file mode 100644 index 00000000..be03b4a7 --- /dev/null +++ b/extras/Hadrons/Modules/MScalarSUN/Grad.cc @@ -0,0 +1,38 @@ +/************************************************************************************* + +Grid physics library, www.github.com/paboyle/Grid + +Source file: extras/Hadrons/Modules/MScalarSUN/Grad.cc + +Copyright (C) 2015-2018 + +Author: Antonin Portelli + +This program is free software; you can redistribute it and/or modify +it under the terms of the GNU General Public License as published by +the Free Software Foundation; either version 2 of the License, or +(at your option) any later version. + +This program is distributed in the hope that it will be useful, +but WITHOUT ANY WARRANTY; without even the implied warranty of +MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +GNU General Public License for more details. + +You should have received a copy of the GNU General Public License along +with this program; if not, write to the Free Software Foundation, Inc., +51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + +See the full license in the file "LICENSE" in the top level distribution directory +*************************************************************************************/ +/* END LEGAL */ +#include + +using namespace Grid; +using namespace Hadrons; +using namespace MScalarSUN; + +template class Grid::Hadrons::MScalarSUN::TGrad>; +template class Grid::Hadrons::MScalarSUN::TGrad>; +template class Grid::Hadrons::MScalarSUN::TGrad>; +template class Grid::Hadrons::MScalarSUN::TGrad>; +template class Grid::Hadrons::MScalarSUN::TGrad>; diff --git a/extras/Hadrons/Modules/MScalarSUN/Grad.hpp b/extras/Hadrons/Modules/MScalarSUN/Grad.hpp new file mode 100644 index 00000000..7718fbf2 --- /dev/null +++ b/extras/Hadrons/Modules/MScalarSUN/Grad.hpp @@ -0,0 +1,165 @@ +/************************************************************************************* + +Grid physics library, www.github.com/paboyle/Grid + +Source file: extras/Hadrons/Modules/MScalarSUN/Grad.hpp + +Copyright (C) 2015-2018 + +Author: Antonin Portelli + +This program is free software; you can redistribute it and/or modify +it under the terms of the GNU General Public License as published by +the Free Software Foundation; either version 2 of the License, or +(at your option) any later version. + +This program is distributed in the hope that it will be useful, +but WITHOUT ANY WARRANTY; without even the implied warranty of +MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +GNU General Public License for more details. + +You should have received a copy of the GNU General Public License along +with this program; if not, write to the Free Software Foundation, Inc., +51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + +See the full license in the file "LICENSE" in the top level distribution directory +*************************************************************************************/ +/* END LEGAL */ +#ifndef Hadrons_MScalarSUN_Grad_hpp_ +#define Hadrons_MScalarSUN_Grad_hpp_ + +#include +#include +#include +#include + +BEGIN_HADRONS_NAMESPACE + +/****************************************************************************** + * Gradient of a complex field * + ******************************************************************************/ +BEGIN_MODULE_NAMESPACE(MScalarSUN) + +class GradPar: Serializable +{ +public: + GRID_SERIALIZABLE_CLASS_MEMBERS(GradPar, + std::string, op, + DiffType, type, + std::string, output); +}; + +template +class TGrad: public Module +{ +public: + typedef typename SImpl::Field Field; + typedef typename SImpl::ComplexField ComplexField; + class Result: Serializable + { + public: + GRID_SERIALIZABLE_CLASS_MEMBERS(Result, + DiffType, type, + Complex, value); + }; +public: + // constructor + TGrad(const std::string name); + // destructor + virtual ~TGrad(void) {}; + // dependency relation + virtual std::vector getInput(void); + virtual std::vector getOutput(void); + // setup + virtual void setup(void); + // execution + virtual void execute(void); +}; + +MODULE_REGISTER_TMP(GradSU2, TGrad>, MScalarSUN); +MODULE_REGISTER_TMP(GradSU3, TGrad>, MScalarSUN); +MODULE_REGISTER_TMP(GradSU4, TGrad>, MScalarSUN); +MODULE_REGISTER_TMP(GradSU5, TGrad>, MScalarSUN); +MODULE_REGISTER_TMP(GradSU6, TGrad>, MScalarSUN); + +/****************************************************************************** + * TGrad implementation * + ******************************************************************************/ +// constructor ///////////////////////////////////////////////////////////////// +template +TGrad::TGrad(const std::string name) +: Module(name) +{} + +// dependencies/products /////////////////////////////////////////////////////// +template +std::vector TGrad::getInput(void) +{ + std::vector in = {par().op}; + + return in; +} + +template +std::vector TGrad::getOutput(void) +{ + std::vector out; + const auto nd = env().getNd(); + + for (unsigned int mu = 0; mu < nd; ++mu) + { + out.push_back(varName(getName(), mu)); + } + + return out; +} + +// setup /////////////////////////////////////////////////////////////////////// +template +void TGrad::setup(void) +{ + const auto nd = env().getNd(); + + for (unsigned int mu = 0; mu < nd; ++mu) + { + envCreateLat(ComplexField, varName(getName(), mu)); + } +} + +// execution /////////////////////////////////////////////////////////////////// +template +void TGrad::execute(void) +{ + const auto nd = env().getNd(); + + LOG(Message) << "Computing the " << par().type << " gradient of '" + << par().op << "'" << std::endl; + + std::vector result; + auto &op = envGet(ComplexField, par().op); + + for (unsigned int mu = 0; mu < nd; ++mu) + { + auto &der = envGet(ComplexField, varName(getName(), mu)); + + dmu(der, op, mu, par().type); + if (!par().output.empty()) + { + Result r; + + r.type = par().type; + r.value = TensorRemove(sum(der)); + result.push_back(r); + } + } + if (result.size() > 0) + { + saveResult(par().output, "grad", result); + } +} + +END_MODULE_NAMESPACE + +END_HADRONS_NAMESPACE + +#endif // Hadrons_MScalarSUN_Grad_hpp_ diff --git a/extras/Hadrons/modules.inc b/extras/Hadrons/modules.inc index ec954bbe..ad3a8727 100644 --- a/extras/Hadrons/modules.inc +++ b/extras/Hadrons/modules.inc @@ -31,6 +31,7 @@ modules_cc =\ Modules/MAction/WilsonClover.cc \ Modules/MAction/DWF.cc \ Modules/MScalarSUN/TrPhi.cc \ + Modules/MScalarSUN/Grad.cc \ Modules/MScalarSUN/TrMag.cc \ Modules/MScalarSUN/TrKinetic.cc \ Modules/MScalarSUN/EMT.cc \ @@ -85,6 +86,7 @@ modules_hpp =\ Modules/MScalarSUN/TrPhi.hpp \ Modules/MScalarSUN/Utils.hpp \ Modules/MScalarSUN/TransProj.hpp \ + Modules/MScalarSUN/Grad.hpp \ Modules/MScalarSUN/TrKinetic.hpp \ Modules/MIO/LoadEigenPack.hpp \ Modules/MIO/LoadNersc.hpp \