mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-14 01:35:36 +00:00
369 lines
16 KiB
C++
369 lines
16 KiB
C++
/*******************************************************************************
|
|
Grid physics library, www.github.com/paboyle/Grid
|
|
|
|
Source file: tests/hadrons/Test_hadrons.hpp
|
|
|
|
Copyright (C) 2017
|
|
|
|
Author: Andrew Lawson <andrew.lawson1991@gmail.com>
|
|
|
|
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.
|
|
*******************************************************************************/
|
|
|
|
#include <Grid/Hadrons/Application.hpp>
|
|
|
|
using namespace Grid;
|
|
using namespace Hadrons;
|
|
|
|
/*******************************************************************************
|
|
* Macros to reduce code duplication.
|
|
******************************************************************************/
|
|
// Useful definitions
|
|
#define ZERO_MOM "0. 0. 0. 0."
|
|
#define INIT_INDEX(s, n) (std::string(s) + "_" + std::to_string(n))
|
|
#define ADD_INDEX(s, n) (s + "_" + std::to_string(n))
|
|
#define LABEL_3PT(s, t1, t2) ADD_INDEX(INIT_INDEX(s, t1), t2)
|
|
#define LABEL_4PT(s, t1, t2, t3) ADD_INDEX(ADD_INDEX(INIT_INDEX(s, t1), t2), t3)
|
|
#define LABEL_4PT_NOISE(s, t1, t2, t3, nn) ADD_INDEX(ADD_INDEX(ADD_INDEX(INIT_INDEX(s, t1), t2), t3), nn)
|
|
|
|
// Wall source/sink macros
|
|
#define NAME_3MOM_WALL_SOURCE(t, mom) ("wall_" + std::to_string(t) + "_" + mom)
|
|
#define NAME_WALL_SOURCE(t) NAME_3MOM_WALL_SOURCE(t, ZERO_MOM)
|
|
#define NAME_POINT_SOURCE(pos) ("point_" + pos)
|
|
|
|
#define MAKE_3MOM_WALL_PROP(tW, mom, propName, solver)\
|
|
{\
|
|
std::string srcName = NAME_3MOM_WALL_SOURCE(tW, mom);\
|
|
makeWallSource(application, srcName, tW, mom);\
|
|
makePropagator(application, propName, srcName, solver);\
|
|
}
|
|
|
|
#define MAKE_WALL_PROP(tW, propName, solver)\
|
|
MAKE_3MOM_WALL_PROP(tW, ZERO_MOM, propName, solver)
|
|
|
|
// Sequential source macros
|
|
#define MAKE_SEQUENTIAL_PROP(tS, qSrc, mom, propName, solver)\
|
|
{\
|
|
std::string srcName = ADD_INDEX(qSrc + "_seq", tS);\
|
|
makeSequentialSource(application, srcName, qSrc, tS, mom);\
|
|
makePropagator(application, propName, srcName, solver);\
|
|
}
|
|
|
|
// Point source macros
|
|
#define MAKE_POINT_PROP(pos, propName, solver)\
|
|
{\
|
|
std::string srcName = NAME_POINT_SOURCE(pos);\
|
|
makePointSource(application, srcName, pos);\
|
|
makePropagator(application, propName, srcName, solver);\
|
|
}
|
|
|
|
/*******************************************************************************
|
|
* Functions for propagator construction.
|
|
******************************************************************************/
|
|
|
|
/*******************************************************************************
|
|
* Name: makePointSource
|
|
* Purpose: Construct point source and add to application module.
|
|
* Parameters: application - main application that stores modules.
|
|
* srcName - name of source module to create.
|
|
* pos - Position of point source.
|
|
* Returns: None.
|
|
******************************************************************************/
|
|
inline void makePointSource(Application &application, std::string srcName,
|
|
std::string pos)
|
|
{
|
|
// If the source already exists, don't make the module again.
|
|
if (!(Environment::getInstance().hasModule(srcName)))
|
|
{
|
|
MSource::Point::Par pointPar;
|
|
pointPar.position = pos;
|
|
application.createModule<MSource::Point>(srcName, pointPar);
|
|
}
|
|
}
|
|
|
|
/*******************************************************************************
|
|
* Name: makeSequentialSource
|
|
* Purpose: Construct sequential source and add to application module.
|
|
* Parameters: application - main application that stores modules.
|
|
* srcName - name of source module to create.
|
|
* qSrc - Input quark for sequential inversion.
|
|
* tS - sequential source timeslice.
|
|
* mom - momentum insertion (default is zero).
|
|
* Returns: None.
|
|
******************************************************************************/
|
|
inline void makeSequentialSource(Application &application, std::string srcName,
|
|
std::string qSrc, unsigned int tS,
|
|
std::string mom = ZERO_MOM)
|
|
{
|
|
// If the source already exists, don't make the module again.
|
|
if (!(Environment::getInstance().hasModule(srcName)))
|
|
{
|
|
MSource::SeqGamma::Par seqPar;
|
|
seqPar.q = qSrc;
|
|
seqPar.tA = tS;
|
|
seqPar.tB = tS;
|
|
seqPar.mom = mom;
|
|
seqPar.gamma = Gamma::Algebra::GammaT;
|
|
application.createModule<MSource::SeqGamma>(srcName, seqPar);
|
|
}
|
|
}
|
|
|
|
/*******************************************************************************
|
|
* Name: makeWallSource
|
|
* Purpose: Construct wall source and add to application module.
|
|
* Parameters: application - main application that stores modules.
|
|
* srcName - name of source module to create.
|
|
* tW - wall source timeslice.
|
|
* mom - momentum insertion (default is zero).
|
|
* Returns: None.
|
|
******************************************************************************/
|
|
inline void makeWallSource(Application &application, std::string srcName,
|
|
unsigned int tW, std::string mom = ZERO_MOM)
|
|
{
|
|
// If the source already exists, don't make the module again.
|
|
if (!(Environment::getInstance().hasModule(srcName)))
|
|
{
|
|
MSource::Wall::Par wallPar;
|
|
wallPar.tW = tW;
|
|
wallPar.mom = mom;
|
|
application.createModule<MSource::Wall>(srcName, wallPar);
|
|
}
|
|
}
|
|
|
|
/*******************************************************************************
|
|
* Name: makeWallSink
|
|
* Purpose: Wall sink smearing of a propagator.
|
|
* Parameters: application - main application that stores modules.
|
|
* propName - name of input propagator.
|
|
* wallName - name of smeared propagator.
|
|
* mom - momentum insertion (default is zero).
|
|
* Returns: None.
|
|
******************************************************************************/
|
|
inline void makeWallSink(Application &application, std::string propName,
|
|
std::string wallName, std::string mom = ZERO_MOM)
|
|
{
|
|
// If the propagator has already been smeared, don't smear it again.
|
|
// Temporarily removed, strategy for sink smearing likely to change.
|
|
/*if (!(Environment::getInstance().hasModule(wallName)))
|
|
{
|
|
MSink::Wall::Par wallPar;
|
|
wallPar.q = propName;
|
|
wallPar.mom = mom;
|
|
application.createModule<MSink::Wall>(wallName, wallPar);
|
|
}*/
|
|
}
|
|
|
|
/*******************************************************************************
|
|
* Name: makePropagator
|
|
* Purpose: Construct source and propagator then add to application module.
|
|
* Parameters: application - main application that stores modules.
|
|
* propName - name of propagator module to create.
|
|
* srcName - name of source module to use.
|
|
* solver - solver to use (default is CG).
|
|
* Returns: None.
|
|
******************************************************************************/
|
|
inline void makePropagator(Application &application, std::string &propName,
|
|
std::string &srcName, std::string &solver)
|
|
{
|
|
// If the propagator already exists, don't make the module again.
|
|
if (!(Environment::getInstance().hasModule(propName)))
|
|
{
|
|
Quark::Par quarkPar;
|
|
quarkPar.source = srcName;
|
|
quarkPar.solver = solver;
|
|
application.createModule<Quark>(propName, quarkPar);
|
|
}
|
|
}
|
|
|
|
/*******************************************************************************
|
|
* Name: makeLoop
|
|
* Purpose: Use noise source and inversion result to make loop propagator, then
|
|
* add to application module.
|
|
* Parameters: application - main application that stores modules.
|
|
* propName - name of propagator module to create.
|
|
* srcName - name of noise source module to use.
|
|
* resName - name of inversion result on given noise source.
|
|
* Returns: None.
|
|
******************************************************************************/
|
|
inline void makeLoop(Application &application, std::string &propName,
|
|
std::string &srcName, std::string &resName)
|
|
{
|
|
// If the loop propagator already exists, don't make the module again.
|
|
if (!(Environment::getInstance().hasModule(propName)))
|
|
{
|
|
MLoop::NoiseLoop::Par loopPar;
|
|
loopPar.q = resName;
|
|
loopPar.eta = srcName;
|
|
application.createModule<MLoop::NoiseLoop>(propName, loopPar);
|
|
}
|
|
}
|
|
|
|
/*******************************************************************************
|
|
* Contraction module creation.
|
|
******************************************************************************/
|
|
|
|
/*******************************************************************************
|
|
* Name: mesonContraction
|
|
* Purpose: Create meson contraction module and add to application module.
|
|
* Parameters: application - main application that stores modules.
|
|
* npt - specify n-point correlator (for labelling).
|
|
* q1 - quark propagator 1.
|
|
* q2 - quark propagator 2.
|
|
* label - unique label to construct module name.
|
|
* mom - momentum to project (default is zero)
|
|
* gammas - gamma insertions at source and sink.
|
|
* Returns: None.
|
|
******************************************************************************/
|
|
inline void mesonContraction(Application &application, unsigned int npt,
|
|
std::string &q1, std::string &q2,
|
|
std::string &label,
|
|
std::string mom = ZERO_MOM,
|
|
std::string gammas = "<Gamma5 Gamma5>")
|
|
{
|
|
std::string modName = std::to_string(npt) + "pt_" + label;
|
|
if (!(Environment::getInstance().hasModule(modName)))
|
|
{
|
|
MContraction::Meson::Par mesPar;
|
|
mesPar.output = std::to_string(npt) + "pt/" + label;
|
|
mesPar.q1 = q1;
|
|
mesPar.q2 = q2;
|
|
mesPar.mom = mom;
|
|
mesPar.gammas = gammas;
|
|
application.createModule<MContraction::Meson>(modName, mesPar);
|
|
}
|
|
}
|
|
|
|
/*******************************************************************************
|
|
* Name: gamma3ptContraction
|
|
* Purpose: Create gamma3pt contraction module and add to application module.
|
|
* Parameters: application - main application that stores modules.
|
|
* npt - specify n-point correlator (for labelling).
|
|
* q1 - quark propagator 1.
|
|
* q2 - quark propagator 2.
|
|
* q3 - quark propagator 3.
|
|
* label - unique label to construct module name.
|
|
* gamma - gamma insertions between q2 and q3.
|
|
* Returns: None.
|
|
******************************************************************************/
|
|
inline void gamma3ptContraction(Application &application, unsigned int npt,
|
|
std::string &q1, std::string &q2,
|
|
std::string &q3, std::string &label,
|
|
Gamma::Algebra gamma = Gamma::Algebra::Identity)
|
|
{
|
|
std::string modName = std::to_string(npt) + "pt_" + label;
|
|
if (!(Environment::getInstance().hasModule(modName)))
|
|
{
|
|
MContraction::Gamma3pt::Par gamma3ptPar;
|
|
gamma3ptPar.output = std::to_string(npt) + "pt/" + label;
|
|
gamma3ptPar.q1 = q1;
|
|
gamma3ptPar.q2 = q2;
|
|
gamma3ptPar.q3 = q3;
|
|
gamma3ptPar.gamma = gamma;
|
|
application.createModule<MContraction::Gamma3pt>(modName, gamma3ptPar);
|
|
}
|
|
}
|
|
|
|
/*******************************************************************************
|
|
* Name: weakContraction[Eye,NonEye]
|
|
* Purpose: Create Weak Hamiltonian contraction module for Eye/NonEye topology
|
|
* and add to application module.
|
|
* Parameters: application - main application that stores modules.
|
|
* npt - specify n-point correlator (for labelling).
|
|
* q1 - quark propagator 1.
|
|
* q2 - quark propagator 2.
|
|
* q3 - quark propagator 3.
|
|
* q4 - quark propagator 4.
|
|
* label - unique label to construct module name.
|
|
* Returns: None.
|
|
******************************************************************************/
|
|
#define HW_CONTRACTION(top) \
|
|
inline void weakContraction##top(Application &application, unsigned int npt,\
|
|
std::string &q1, std::string &q2, \
|
|
std::string &q3, std::string &q4, \
|
|
std::string &label)\
|
|
{\
|
|
std::string modName = std::to_string(npt) + "pt_" + label;\
|
|
if (!(Environment::getInstance().hasModule(modName)))\
|
|
{\
|
|
MContraction::WeakHamiltonian##top::Par weakPar;\
|
|
weakPar.output = std::to_string(npt) + "pt/" + label;\
|
|
weakPar.q1 = q1;\
|
|
weakPar.q2 = q2;\
|
|
weakPar.q3 = q3;\
|
|
weakPar.q4 = q4;\
|
|
application.createModule<MContraction::WeakHamiltonian##top>(modName, weakPar);\
|
|
}\
|
|
}
|
|
HW_CONTRACTION(Eye) // weakContractionEye
|
|
HW_CONTRACTION(NonEye) // weakContractionNonEye
|
|
|
|
/*******************************************************************************
|
|
* Name: disc0Contraction
|
|
* Purpose: Create contraction module for 4pt Weak Hamiltonian + current
|
|
* disconnected topology for neutral mesons and add to application
|
|
* module.
|
|
* Parameters: application - main application that stores modules.
|
|
* q1 - quark propagator 1.
|
|
* q2 - quark propagator 2.
|
|
* q3 - quark propagator 3.
|
|
* q4 - quark propagator 4.
|
|
* label - unique label to construct module name.
|
|
* Returns: None.
|
|
******************************************************************************/
|
|
inline void disc0Contraction(Application &application,
|
|
std::string &q1, std::string &q2,
|
|
std::string &q3, std::string &q4,
|
|
std::string &label)
|
|
{
|
|
std::string modName = "4pt_" + label;
|
|
if (!(Environment::getInstance().hasModule(modName)))
|
|
{
|
|
MContraction::WeakNeutral4ptDisc::Par disc0Par;
|
|
disc0Par.output = "4pt/" + label;
|
|
disc0Par.q1 = q1;
|
|
disc0Par.q2 = q2;
|
|
disc0Par.q3 = q3;
|
|
disc0Par.q4 = q4;
|
|
application.createModule<MContraction::WeakNeutral4ptDisc>(modName, disc0Par);
|
|
}
|
|
}
|
|
|
|
/*******************************************************************************
|
|
* Name: discLoopContraction
|
|
* Purpose: Create contraction module for disconnected loop and add to
|
|
* application module.
|
|
* Parameters: application - main application that stores modules.
|
|
* q_loop - loop quark propagator.
|
|
* modName - unique module name.
|
|
* gamma - gamma matrix to use in contraction.
|
|
* Returns: None.
|
|
******************************************************************************/
|
|
inline void discLoopContraction(Application &application,
|
|
std::string &q_loop, std::string &modName,
|
|
Gamma::Algebra gamma = Gamma::Algebra::Identity)
|
|
{
|
|
if (!(Environment::getInstance().hasModule(modName)))
|
|
{
|
|
MContraction::DiscLoop::Par discPar;
|
|
discPar.output = "disc/" + modName;
|
|
discPar.q_loop = q_loop;
|
|
discPar.gamma = gamma;
|
|
application.createModule<MContraction::DiscLoop>(modName, discPar);
|
|
}
|
|
}
|