mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-15 02:05:37 +00:00
298 lines
8.1 KiB
C++
298 lines
8.1 KiB
C++
/*************************************************************************************
|
|
|
|
Grid physics library, www.github.com/paboyle/Grid
|
|
|
|
Source file: Hadrons/Global.hpp
|
|
|
|
Copyright (C) 2015-2018
|
|
|
|
Author: Antonin Portelli <antonin.portelli@me.com>
|
|
Author: Lanny91 <andrew.lawson@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
|
|
*************************************************************************************/
|
|
/* END LEGAL */
|
|
|
|
#ifndef Hadrons_Global_hpp_
|
|
#define Hadrons_Global_hpp_
|
|
|
|
#include <set>
|
|
#include <stack>
|
|
#include <regex>
|
|
#include <Grid/Grid.h>
|
|
#include <cxxabi.h>
|
|
|
|
#ifndef SITE_SIZE_TYPE
|
|
#define SITE_SIZE_TYPE size_t
|
|
#endif
|
|
|
|
#ifndef DEFAULT_ASCII_PREC
|
|
#define DEFAULT_ASCII_PREC 16
|
|
#endif
|
|
|
|
/* the 'using Grid::operator<<;' statement prevents a very nasty compilation
|
|
* error with GCC 5 (clang & GCC 6 compile fine without it).
|
|
*/
|
|
|
|
#define BEGIN_HADRONS_NAMESPACE \
|
|
namespace Grid {\
|
|
using namespace QCD;\
|
|
namespace Hadrons {\
|
|
using Grid::operator<<;\
|
|
using Grid::operator>>;
|
|
#define END_HADRONS_NAMESPACE }}
|
|
|
|
#define BEGIN_MODULE_NAMESPACE(name)\
|
|
namespace name {\
|
|
using Grid::operator<<;\
|
|
using Grid::operator>>;
|
|
|
|
#define END_MODULE_NAMESPACE }
|
|
|
|
#define _HADRONS_IMPL(impl, sub) impl##sub
|
|
#define HADRONS_IMPL(impl, sub) _HADRONS_IMPL(impl, sub)
|
|
|
|
#ifndef FIMPLBASE
|
|
#define FIMPLBASE WilsonImpl
|
|
#endif
|
|
#define FIMPL HADRONS_IMPL(FIMPLBASE, R)
|
|
#define FIMPLF HADRONS_IMPL(FIMPLBASE, F)
|
|
#define FIMPLD HADRONS_IMPL(FIMPLBASE, D)
|
|
|
|
#ifndef ZFIMPLBASE
|
|
#define ZFIMPLBASE ZWilsonImpl
|
|
#endif
|
|
#define ZFIMPL HADRONS_IMPL(ZFIMPLBASE, R)
|
|
#define ZFIMPLF HADRONS_IMPL(ZFIMPLBASE, F)
|
|
#define ZFIMPLD HADRONS_IMPL(ZFIMPLBASE, D)
|
|
|
|
#ifndef SIMPLBASE
|
|
#define SIMPLBASE ScalarImplC
|
|
#endif
|
|
#define SIMPL HADRONS_IMPL(SIMPLBASE, R)
|
|
#define SIMPLF HADRONS_IMPL(SIMPLBASE, F)
|
|
#define SIMPLD HADRONS_IMPL(SIMPLBASE, D)
|
|
|
|
#ifndef GIMPLBASE
|
|
#define GIMPLBASE PeriodicGimpl
|
|
#endif
|
|
#define GIMPL HADRONS_IMPL(GIMPLBASE, R)
|
|
#define GIMPLF HADRONS_IMPL(GIMPLBASE, F)
|
|
#define GIMPLD HADRONS_IMPL(GIMPLBASE, D)
|
|
|
|
BEGIN_HADRONS_NAMESPACE
|
|
|
|
// type aliases
|
|
#define BASIC_TYPE_ALIASES(Impl, suffix)\
|
|
typedef typename Impl::Field ScalarField##suffix;\
|
|
typedef typename Impl::PropagatorField PropagatorField##suffix;\
|
|
typedef typename Impl::SitePropagator::scalar_object SitePropagator##suffix;\
|
|
typedef std::vector<SitePropagator##suffix> SlicedPropagator##suffix;
|
|
|
|
#define FERM_TYPE_ALIASES(FImpl, suffix)\
|
|
BASIC_TYPE_ALIASES(FImpl, suffix);\
|
|
typedef FermionOperator<FImpl> FMat##suffix;\
|
|
typedef typename FImpl::FermionField FermionField##suffix;\
|
|
typedef typename FImpl::GaugeField GaugeField##suffix;\
|
|
typedef typename FImpl::DoubledGaugeField DoubledGaugeField##suffix;\
|
|
typedef typename FImpl::ComplexField ComplexField##suffix;
|
|
|
|
#define GAUGE_TYPE_ALIASES(GImpl, suffix)\
|
|
typedef typename GImpl::GaugeField GaugeField##suffix;
|
|
|
|
#define SOLVER_TYPE_ALIASES(FImpl, suffix)\
|
|
typedef Solver<FImpl> Solver##suffix;
|
|
|
|
#define SINK_TYPE_ALIASES(suffix)\
|
|
typedef std::function<SlicedPropagator##suffix\
|
|
(const PropagatorField##suffix &)> SinkFn##suffix;
|
|
|
|
// logger
|
|
class HadronsLogger: public Logger
|
|
{
|
|
public:
|
|
HadronsLogger(int on, std::string nm): Logger("Hadrons", on, nm,
|
|
GridLogColours, "BLACK"){};
|
|
};
|
|
|
|
#define LOG(channel) std::cout << HadronsLog##channel
|
|
#define HADRONS_DEBUG_VAR(var) LOG(Debug) << #var << "= " << (var) << std::endl;
|
|
|
|
extern HadronsLogger HadronsLogError;
|
|
extern HadronsLogger HadronsLogWarning;
|
|
extern HadronsLogger HadronsLogMessage;
|
|
extern HadronsLogger HadronsLogIterative;
|
|
extern HadronsLogger HadronsLogDebug;
|
|
extern HadronsLogger HadronsLogIRL;
|
|
|
|
void initLogger(void);
|
|
|
|
// singleton pattern
|
|
#define SINGLETON(name)\
|
|
public:\
|
|
name(const name &e) = delete;\
|
|
void operator=(const name &e) = delete;\
|
|
static name & getInstance(void)\
|
|
{\
|
|
static name e;\
|
|
return e;\
|
|
}\
|
|
private:\
|
|
name(void);
|
|
|
|
#define SINGLETON_DEFCTOR(name)\
|
|
public:\
|
|
name(const name &e) = delete;\
|
|
void operator=(const name &e) = delete;\
|
|
static name & getInstance(void)\
|
|
{\
|
|
static name e;\
|
|
return e;\
|
|
}\
|
|
private:\
|
|
name(void) = default;
|
|
|
|
// type utilities
|
|
template <typename T>
|
|
const std::type_info * typeIdPt(const T &x)
|
|
{
|
|
return &typeid(x);
|
|
}
|
|
|
|
template <typename T>
|
|
const std::type_info * typeIdPt(void)
|
|
{
|
|
return &typeid(T);
|
|
}
|
|
|
|
size_t typeHash(const std::type_info *info);
|
|
|
|
template <typename T>
|
|
size_t typeHash(const T &x)
|
|
{
|
|
return typeHash(typeIdPt(x));
|
|
}
|
|
|
|
template <typename T>
|
|
size_t typeHash(void)
|
|
{
|
|
return typeHash(typeIdPt<T>());
|
|
}
|
|
|
|
std::string typeName(const std::type_info *info);
|
|
|
|
template <typename T>
|
|
std::string typeName(const T &x)
|
|
{
|
|
return typeName(typeIdPt(x));
|
|
}
|
|
|
|
template <typename T>
|
|
std::string typeName(void)
|
|
{
|
|
return typeName(typeIdPt<T>());
|
|
}
|
|
|
|
// default writers/readers
|
|
extern const std::string resultFileExt;
|
|
|
|
#ifdef HAVE_HDF5
|
|
typedef Hdf5Reader ResultReader;
|
|
typedef Hdf5Writer ResultWriter;
|
|
#else
|
|
typedef XmlReader ResultReader;
|
|
typedef XmlWriter ResultWriter;
|
|
#endif
|
|
|
|
#define RESULT_FILE_NAME(name, traj) \
|
|
name + "." + std::to_string(traj) + "." + resultFileExt
|
|
|
|
// recursive mkdir
|
|
#define MAX_PATH_LENGTH 512u
|
|
int mkdir(const std::string dirName);
|
|
std::string basename(const std::string &s);
|
|
std::string dirname(const std::string &s);
|
|
void makeFileDir(const std::string filename, GridBase *g = nullptr);
|
|
|
|
// default Schur convention
|
|
#ifndef HADRONS_DEFAULT_SCHUR
|
|
#define HADRONS_DEFAULT_SCHUR DiagTwo
|
|
#endif
|
|
#define _HADRONS_SCHUR_OP_(conv) Schur##conv##Operator
|
|
#define HADRONS_SCHUR_OP(conv) _HADRONS_SCHUR_OP_(conv)
|
|
#define HADRONS_DEFAULT_SCHUR_OP HADRONS_SCHUR_OP(HADRONS_DEFAULT_SCHUR)
|
|
#define _HADRONS_SCHUR_SOLVE_(conv) SchurRedBlack##conv##Solve
|
|
#define HADRONS_SCHUR_SOLVE(conv) _HADRONS_SCHUR_SOLVE_(conv)
|
|
#define HADRONS_DEFAULT_SCHUR_SOLVE HADRONS_SCHUR_SOLVE(HADRONS_DEFAULT_SCHUR)
|
|
#define _HADRONS_SCHUR_A2A_(conv) A2AVectorsSchur##conv
|
|
#define HADRONS_SCHUR_A2A(conv) _HADRONS_SCHUR_A2A_(conv)
|
|
#define HADRONS_DEFAULT_SCHUR_A2A HADRONS_SCHUR_A2A(HADRONS_DEFAULT_SCHUR)
|
|
|
|
// stringify macro
|
|
#define _HADRONS_STR(x) #x
|
|
#define HADRONS_STR(x) _HADRONS_STR(x)
|
|
|
|
// pretty print time profile
|
|
void printTimeProfile(const std::map<std::string, GridTime> &timing, GridTime total);
|
|
|
|
// token replacement utility
|
|
template <typename T>
|
|
void tokenReplace(std::string &str, const std::string token,
|
|
const T &x, const std::string mark = "@")
|
|
{
|
|
std::string fullToken = mark + token + mark;
|
|
|
|
auto pos = str.find(fullToken);
|
|
if (pos != std::string::npos)
|
|
{
|
|
str.replace(pos, fullToken.size(), std::to_string(x));
|
|
}
|
|
}
|
|
|
|
// trajectory range
|
|
class TrajRange: Serializable
|
|
{
|
|
public:
|
|
GRID_SERIALIZABLE_CLASS_MEMBERS(TrajRange,
|
|
unsigned int, start,
|
|
unsigned int, end,
|
|
unsigned int, step,
|
|
std::string, exclude);
|
|
|
|
inline std::vector<unsigned int> getTrajectoryList(void)
|
|
{
|
|
std::vector<unsigned int> excVec = strToVec<unsigned int>(exclude);
|
|
std::vector<unsigned int> list;
|
|
|
|
for (unsigned int t = start; t < end; t += step)
|
|
{
|
|
if (std::find(excVec.begin(), excVec.end(), t) == excVec.end())
|
|
{
|
|
list.push_back(t);
|
|
}
|
|
}
|
|
|
|
return list;
|
|
}
|
|
};
|
|
|
|
END_HADRONS_NAMESPACE
|
|
|
|
#include <Hadrons/Exceptions.hpp>
|
|
|
|
#endif // Hadrons_Global_hpp_
|