convert more style_creators to use local static functions

This commit is contained in:
Axel Kohlmeyer
2022-02-11 18:29:52 -05:00
parent 5ab9b46b67
commit 6a0e93a18a
10 changed files with 110 additions and 152 deletions

View File

@ -54,6 +54,15 @@ using namespace MathConst;
#define DELTA_PERATOM 64 #define DELTA_PERATOM 64
#define EPSILON 1.0e-6 #define EPSILON 1.0e-6
/* ----------------------------------------------------------------------
one instance per AtomVec style in style_atom.h
------------------------------------------------------------------------- */
template <typename T> static AtomVec *avec_creator(LAMMPS *lmp)
{
return new T(lmp);
}
/* ---------------------------------------------------------------------- */ /* ---------------------------------------------------------------------- */
/** \class LAMMPS_NS::Atom /** \class LAMMPS_NS::Atom
@ -743,16 +752,6 @@ AtomVec *Atom::new_avec(const std::string &style, int trysuffix, int &sflag)
return nullptr; return nullptr;
} }
/* ----------------------------------------------------------------------
one instance per AtomVec style in style_atom.h
------------------------------------------------------------------------- */
template <typename T>
AtomVec *Atom::avec_creator(LAMMPS *lmp)
{
return new T(lmp);
}
/* ---------------------------------------------------------------------- */ /* ---------------------------------------------------------------------- */
void Atom::init() void Atom::init()

View File

@ -416,9 +416,6 @@ class Atom : protected Pointers {
void set_atomflag_defaults(); void set_atomflag_defaults();
void setup_sort_bins(); void setup_sort_bins();
int next_prime(int); int next_prime(int);
private:
template <typename T> static AtomVec *avec_creator(LAMMPS *);
}; };
} // namespace LAMMPS_NS } // namespace LAMMPS_NS

View File

@ -47,6 +47,15 @@ using namespace LAMMPS_NS;
#define DELTAREGION 4 #define DELTAREGION 4
#define BONDSTRETCH 1.1 #define BONDSTRETCH 1.1
/* ----------------------------------------------------------------------
one instance per region style in style_region.h
------------------------------------------------------------------------- */
template <typename T> static Region *region_creator(LAMMPS *lmp, int narg, char ** arg)
{
return new T(lmp, narg, arg);
}
/* ---------------------------------------------------------------------- /* ----------------------------------------------------------------------
default is periodic default is periodic
------------------------------------------------------------------------- */ ------------------------------------------------------------------------- */
@ -1797,16 +1806,6 @@ void Domain::add_region(int narg, char **arg)
nregion++; nregion++;
} }
/* ----------------------------------------------------------------------
one instance per region style in style_region.h
------------------------------------------------------------------------- */
template <typename T>
Region *Domain::region_creator(LAMMPS *lmp, int narg, char ** arg)
{
return new T(lmp, narg, arg);
}
/* ---------------------------------------------------------------------- /* ----------------------------------------------------------------------
delete a region delete a region
------------------------------------------------------------------------- */ ------------------------------------------------------------------------- */

View File

@ -175,9 +175,6 @@ class Domain : protected Pointers {
protected: protected:
double small[3]; // fractions of box lengths double small[3]; // fractions of box lengths
private:
template <typename T> static Region *region_creator(LAMMPS *, int, char **);
}; };
} // namespace LAMMPS_NS } // namespace LAMMPS_NS

View File

@ -100,6 +100,14 @@ static const char cite_neigh_multi[] =
" year = {2020}\n" " year = {2020}\n"
"}\n\n"; "}\n\n";
// template for factory functions:
// there will be one instance for each style keyword in the respective style_xxx.h files
template <typename S, typename T> static S *style_creator(LAMMPS *lmp)
{
return new T(lmp);
}
//#define NEIGH_LIST_DEBUG 1 //#define NEIGH_LIST_DEBUG 1
/* ---------------------------------------------------------------------- */ /* ---------------------------------------------------------------------- */
@ -693,7 +701,7 @@ void Neighbor::init_styles()
#define NBIN_CLASS #define NBIN_CLASS
#define NBinStyle(key,Class,bitmasks) \ #define NBinStyle(key,Class,bitmasks) \
binnames[nbclass] = (char *) #key; \ binnames[nbclass] = (char *) #key; \
binclass[nbclass] = &bin_creator<Class>; \ binclass[nbclass] = &style_creator<NBin, Class>; \
binmasks[nbclass++] = bitmasks; binmasks[nbclass++] = bitmasks;
#include "style_nbin.h" // IWYU pragma: keep #include "style_nbin.h" // IWYU pragma: keep
#undef NBinStyle #undef NBinStyle
@ -717,7 +725,7 @@ void Neighbor::init_styles()
#define NSTENCIL_CLASS #define NSTENCIL_CLASS
#define NStencilStyle(key,Class,bitmasks) \ #define NStencilStyle(key,Class,bitmasks) \
stencilnames[nsclass] = (char *) #key; \ stencilnames[nsclass] = (char *) #key; \
stencilclass[nsclass] = &stencil_creator<Class>; \ stencilclass[nsclass] = &style_creator<NStencil, Class>; \
stencilmasks[nsclass++] = bitmasks; stencilmasks[nsclass++] = bitmasks;
#include "style_nstencil.h" // IWYU pragma: keep #include "style_nstencil.h" // IWYU pragma: keep
#undef NStencilStyle #undef NStencilStyle
@ -741,7 +749,7 @@ void Neighbor::init_styles()
#define NPAIR_CLASS #define NPAIR_CLASS
#define NPairStyle(key,Class,bitmasks) \ #define NPairStyle(key,Class,bitmasks) \
pairnames[npclass] = (char *) #key; \ pairnames[npclass] = (char *) #key; \
pairclass[npclass] = &pair_creator<Class>; \ pairclass[npclass] = &style_creator<NPair, Class>; \
pairmasks[npclass++] = bitmasks; pairmasks[npclass++] = bitmasks;
#include "style_npair.h" // IWYU pragma: keep #include "style_npair.h" // IWYU pragma: keep
#undef NPairStyle #undef NPairStyle
@ -2028,36 +2036,6 @@ int Neighbor::request(void *requestor, int instance)
return nrequest-1; return nrequest-1;
} }
/* ----------------------------------------------------------------------
one instance per entry in style_neigh_bin.h
------------------------------------------------------------------------- */
template <typename T>
NBin *Neighbor::bin_creator(LAMMPS *lmp)
{
return new T(lmp);
}
/* ----------------------------------------------------------------------
one instance per entry in style_neigh_stencil.h
------------------------------------------------------------------------- */
template <typename T>
NStencil *Neighbor::stencil_creator(LAMMPS *lmp)
{
return new T(lmp);
}
/* ----------------------------------------------------------------------
one instance per entry in style_neigh_pair.h
------------------------------------------------------------------------- */
template <typename T>
NPair *Neighbor::pair_creator(LAMMPS *lmp)
{
return new T(lmp);
}
/* ---------------------------------------------------------------------- /* ----------------------------------------------------------------------
setup neighbor binning and neighbor stencils setup neighbor binning and neighbor stencils
called before run and every reneighbor if box size/shape changes called before run and every reneighbor if box size/shape changes

View File

@ -231,10 +231,6 @@ class Neighbor : protected Pointers {
int choose_stencil(class NeighRequest *); int choose_stencil(class NeighRequest *);
int choose_pair(class NeighRequest *); int choose_pair(class NeighRequest *);
template <typename T> static NBin *bin_creator(class LAMMPS *);
template <typename T> static NStencil *stencil_creator(class LAMMPS *);
template <typename T> static NPair *pair_creator(class LAMMPS *);
// dummy functions provided by NeighborKokkos, called in init() // dummy functions provided by NeighborKokkos, called in init()
// otherwise NeighborKokkos would have to overwrite init() // otherwise NeighborKokkos would have to overwrite init()

View File

@ -43,6 +43,15 @@ using namespace LAMMPS_NS;
enum {SETUP, WRITE, RESET_DT}; enum {SETUP, WRITE, RESET_DT};
/* ----------------------------------------------------------------------
one instance per dump style in style_dump.h
------------------------------------------------------------------------- */
template <typename T> static Dump *dump_creator(LAMMPS *lmp, int narg, char ** arg)
{
return new T(lmp, narg, arg);
}
/* ---------------------------------------------------------------------- /* ----------------------------------------------------------------------
initialize all output initialize all output
------------------------------------------------------------------------- */ ------------------------------------------------------------------------- */
@ -784,16 +793,6 @@ void Output::add_dump(int narg, char **arg)
ndump++; ndump++;
} }
/* ----------------------------------------------------------------------
one instance per dump style in style_dump.h
------------------------------------------------------------------------- */
template <typename T>
Dump *Output::dump_creator(LAMMPS *lmp, int narg, char ** arg)
{
return new T(lmp, narg, arg);
}
/* ---------------------------------------------------------------------- /* ----------------------------------------------------------------------
modify parameters of a Dump modify parameters of a Dump
------------------------------------------------------------------------- */ ------------------------------------------------------------------------- */

View File

@ -93,7 +93,6 @@ class Output : protected Pointers {
void memory_usage(); // print out memory usage void memory_usage(); // print out memory usage
private: private:
template <typename T> static Dump *dump_creator(LAMMPS *, int, char **);
void calculate_next_dump(int, int, bigint); void calculate_next_dump(int, int, bigint);
}; };

View File

@ -1,4 +1,3 @@
// clang-format off
/* ---------------------------------------------------------------------- /* ----------------------------------------------------------------------
LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator
https://www.lammps.org/, Sandia National Laboratories https://www.lammps.org/, Sandia National Laboratories
@ -19,10 +18,10 @@
#include "comm.h" #include "comm.h"
#include "compute.h" #include "compute.h"
#include "integrate.h"
#include "error.h" #include "error.h"
#include "fix.h" #include "fix.h"
#include "force.h" #include "force.h"
#include "integrate.h"
#include "min.h" #include "min.h"
#include "modify.h" #include "modify.h"
#include "neighbor.h" #include "neighbor.h"
@ -32,6 +31,19 @@
using namespace LAMMPS_NS; using namespace LAMMPS_NS;
// template for factory functions:
// there will be one instance for each style keyword in the respective style_xxx.h files
template <typename T> static Integrate *integrate_creator(LAMMPS *lmp, int narg, char **arg)
{
return new T(lmp, narg, arg);
}
template <typename T> static Min *minimize_creator(LAMMPS *lmp)
{
return new T(lmp);
}
/* ---------------------------------------------------------------------- */ /* ---------------------------------------------------------------------- */
Update::Update(LAMMPS *lmp) : Pointers(lmp) Update::Update(LAMMPS *lmp) : Pointers(lmp)
@ -67,8 +79,7 @@ Update::Update(LAMMPS *lmp) : Pointers(lmp)
integrate_map = new IntegrateCreatorMap(); integrate_map = new IntegrateCreatorMap();
#define INTEGRATE_CLASS #define INTEGRATE_CLASS
#define IntegrateStyle(key,Class) \ #define IntegrateStyle(key, Class) (*integrate_map)[#key] = &integrate_creator<Class>;
(*integrate_map)[#key] = &integrate_creator<Class>;
#include "style_integrate.h" // IWYU pragma: keep #include "style_integrate.h" // IWYU pragma: keep
#undef IntegrateStyle #undef IntegrateStyle
#undef INTEGRATE_CLASS #undef INTEGRATE_CLASS
@ -76,8 +87,7 @@ Update::Update(LAMMPS *lmp) : Pointers(lmp)
minimize_map = new MinimizeCreatorMap(); minimize_map = new MinimizeCreatorMap();
#define MINIMIZE_CLASS #define MINIMIZE_CLASS
#define MinimizeStyle(key,Class) \ #define MinimizeStyle(key, Class) (*minimize_map)[#key] = &minimize_creator<Class>;
(*minimize_map)[#key] = &minimize_creator<Class>;
#include "style_minimize.h" // IWYU pragma: keep #include "style_minimize.h" // IWYU pragma: keep
#undef MinimizeStyle #undef MinimizeStyle
#undef MINIMIZE_CLASS #undef MINIMIZE_CLASS
@ -113,8 +123,10 @@ void Update::init()
// if neither (e.g. from write_restart) then just return // if neither (e.g. from write_restart) then just return
if (whichflag == 0) return; if (whichflag == 0) return;
if (whichflag == 1) integrate->init(); if (whichflag == 1)
else if (whichflag == 2) minimize->init(); integrate->init();
else if (whichflag == 2)
minimize->init();
// only set first_update if a run or minimize is being performed // only set first_update if a run or minimize is being performed
@ -299,15 +311,16 @@ void Update::set_units(const char *style)
dt = 0.00045; dt = 0.00045;
neighbor->skin = 0.1; neighbor->skin = 0.1;
} else error->all(FLERR,"Illegal units command"); } else
error->all(FLERR, "Illegal units command");
delete[] unit_style; delete[] unit_style;
unit_style = utils::strdup(style); unit_style = utils::strdup(style);
// check if timestep was changed from default value // check if timestep was changed from default value
if (!dt_default && (comm->me == 0)) { if (!dt_default && (comm->me == 0)) {
error->warning(FLERR,"Changing timestep from {:.6} to {:.6} due to " error->warning(FLERR, "Changing timestep from {:.6} to {:.6} due to changing units to {}",
"changing units to {}", dt_old, dt, unit_style); dt_old, dt, unit_style);
} }
dt_default = 1; dt_default = 1;
} }
@ -332,8 +345,10 @@ void Update::create_integrate(int narg, char **arg, int trysuffix)
std::string estyle = arg[0]; std::string estyle = arg[0];
if (sflag) { if (sflag) {
estyle += "/"; estyle += "/";
if (sflag == 1) estyle += lmp->suffix; if (sflag == 1)
else estyle += lmp->suffix2; estyle += lmp->suffix;
else
estyle += lmp->suffix2;
} }
integrate_style = utils::strdup(estyle); integrate_style = utils::strdup(estyle);
} }
@ -342,8 +357,7 @@ void Update::create_integrate(int narg, char **arg, int trysuffix)
create the Integrate style, first with suffix appended create the Integrate style, first with suffix appended
------------------------------------------------------------------------- */ ------------------------------------------------------------------------- */
void Update::new_integrate(char *style, int narg, char **arg, void Update::new_integrate(char *style, int narg, char **arg, int trysuffix, int &sflag)
int trysuffix, int &sflag)
{ {
if (trysuffix && lmp->suffix_enable) { if (trysuffix && lmp->suffix_enable) {
if (lmp->suffix) { if (lmp->suffix) {
@ -377,16 +391,6 @@ void Update::new_integrate(char *style, int narg, char **arg,
error->all(FLERR, "Illegal integrate style"); error->all(FLERR, "Illegal integrate style");
} }
/* ----------------------------------------------------------------------
one instance per integrate style in style_integrate.h
------------------------------------------------------------------------- */
template <typename T>
Integrate *Update::integrate_creator(LAMMPS *lmp, int narg, char ** arg)
{
return new T(lmp, narg, arg);
}
/* ---------------------------------------------------------------------- */ /* ---------------------------------------------------------------------- */
void Update::create_minimize(int narg, char **arg, int trysuffix) void Update::create_minimize(int narg, char **arg, int trysuffix)
@ -402,8 +406,10 @@ void Update::create_minimize(int narg, char **arg, int trysuffix)
std::string estyle = arg[0]; std::string estyle = arg[0];
if (sflag) { if (sflag) {
estyle += "/"; estyle += "/";
if (sflag == 1) estyle += lmp->suffix; if (sflag == 1)
else estyle += lmp->suffix2; estyle += lmp->suffix;
else
estyle += lmp->suffix2;
} }
minimize_style = utils::strdup(estyle); minimize_style = utils::strdup(estyle);
} }
@ -412,8 +418,7 @@ void Update::create_minimize(int narg, char **arg, int trysuffix)
create the Minimize style, first with suffix appended create the Minimize style, first with suffix appended
------------------------------------------------------------------------- */ ------------------------------------------------------------------------- */
void Update::new_minimize(char *style, int /* narg */, char ** /* arg */, void Update::new_minimize(char *style, int /* narg */, char ** /* arg */, int trysuffix, int &sflag)
int trysuffix, int &sflag)
{ {
if (trysuffix && lmp->suffix_enable) { if (trysuffix && lmp->suffix_enable) {
if (lmp->suffix) { if (lmp->suffix) {
@ -447,16 +452,6 @@ void Update::new_minimize(char *style, int /* narg */, char ** /* arg */,
error->all(FLERR, "Illegal minimize style"); error->all(FLERR, "Illegal minimize style");
} }
/* ----------------------------------------------------------------------
one instance per minimize style in style_minimize.h
------------------------------------------------------------------------- */
template <typename T>
Min *Update::minimize_creator(LAMMPS *lmp)
{
return new T(lmp);
}
/* ---------------------------------------------------------------------- /* ----------------------------------------------------------------------
reset timestep called from input script reset timestep called from input script
------------------------------------------------------------------------- */ ------------------------------------------------------------------------- */
@ -540,7 +535,9 @@ void Update::update_time()
double Update::memory_usage() double Update::memory_usage()
{ {
double bytes = 0; double bytes = 0;
if (whichflag == 1) bytes += integrate->memory_usage(); if (whichflag == 1)
else if (whichflag == 2) bytes += minimize->memory_usage(); bytes += integrate->memory_usage();
else if (whichflag == 2)
bytes += minimize->memory_usage();
return bytes; return bytes;
} }

View File

@ -73,9 +73,6 @@ class Update : protected Pointers {
private: private:
void new_integrate(char *, int, char **, int, int &); void new_integrate(char *, int, char **, int, int &);
void new_minimize(char *, int, char **, int, int &); void new_minimize(char *, int, char **, int, int &);
template <typename T> static Integrate *integrate_creator(LAMMPS *, int, char **);
template <typename T> static Min *minimize_creator(LAMMPS *);
}; };
} // namespace LAMMPS_NS } // namespace LAMMPS_NS