n2p2 - A neural network potential package
nnp::SymFncCompAngnWeighted Class Reference

Weighted narrow angular symmetry function with compact support (type 24) More...

#include <SymFncCompAngnWeighted.h>

Inheritance diagram for nnp::SymFncCompAngnWeighted:
Collaboration diagram for nnp::SymFncCompAngnWeighted:

Public Member Functions

 SymFncCompAngnWeighted (ElementMap const &elementMap)
 Constructor, sets type = 24. More...
 
virtual bool operator== (SymFnc const &rhs) const
 Overload == operator. More...
 
virtual bool operator< (SymFnc const &rhs) const
 Overload < operator. More...
 
virtual void calculate (Atom &atom, bool const derivatives) const
 Calculate symmetry function for one atom. More...
 
- Public Member Functions inherited from nnp::SymFncBaseCompAngWeighted
virtual void setParameters (std::string const &parameterString)
 Set symmetry function parameters. More...
 
virtual void changeLengthUnit (double convLength)
 Change length unit. More...
 
virtual std::string getSettingsLine () const
 Get settings file line from currently set parameters. More...
 
void getCompactAngle (double const x, double &fx, double &dfx) const
 
void getCompactRadial (double const x, double &fx, double &dfx) const
 
virtual std::string parameterLine () const
 Give symmetry function parameters in one line. More...
 
virtual std::vector< std::string > parameterInfo () const
 Get description with parameter names and values. More...
 
double getAngleLeft () const
 Get private angleLeft member variable. More...
 
double getAngleRight () const
 Get private angleRight member variable. More...
 
virtual double calculateRadialPart (double distance) const
 Calculate (partial) symmetry function value for one given distance. More...
 
virtual double calculateAngularPart (double angle) const
 Calculate (partial) symmetry function value for one given angle. More...
 
virtual bool checkRelevantElement (std::size_t index) const
 Check whether symmetry function is relevant for given element. More...
 
virtual std::vector< std::string > getCacheIdentifiers () const
 Get unique cache identifiers. More...
 
- Public Member Functions inherited from nnp::SymFncBaseComp
virtual std::vector< std::string > parameterInfo () const
 Get description with parameter names and values. More...
 
void setCompactFunction (std::string subtype)
 Set radial compact function. More...
 
std::string getSubtype () const
 Get private subtype member variable. More...
 
double getRl () const
 Get private rl member variable. More...
 
- Public Member Functions inherited from nnp::SymFnc
virtual ~SymFnc ()
 Virtual destructor. More...
 
virtual bool operator== (SymFnc const &rhs) const =0
 Overload == operator. More...
 
virtual bool operator< (SymFnc const &rhs) const =0
 Overload < operator. More...
 
bool operator!= (SymFnc const &rhs) const
 Overload != operator. More...
 
bool operator> (SymFnc const &rhs) const
 Overload > operator. More...
 
bool operator<= (SymFnc const &rhs) const
 Overload <= operator. More...
 
bool operator>= (SymFnc const &rhs) const
 Overload >= operator. More...
 
virtual void setParameters (std::string const &parameterString)=0
 Set parameters. More...
 
virtual void changeLengthUnit (double convLength)=0
 Change length unit. More...
 
virtual std::string getSettingsLine () const =0
 Get settings file line from currently set parameters. More...
 
virtual void calculate (Atom &atom, bool const derivatives) const =0
 Calculate symmetry function for one atom. More...
 
virtual std::string parameterLine () const =0
 Give symmetry function parameters in one line. More...
 
virtual std::vector< std::string > parameterInfo () const
 Get description with parameter names and values. More...
 
void setScalingType (ScalingType scalingType, std::string statisticsLine, double Smin, double Smax)
 Set symmetry function scaling type. More...
 
double scale (double value) const
 Apply symmetry function scaling and/or centering. More...
 
double unscale (double value) const
 Undo symmetry function scaling and/or centering. More...
 
std::size_t getType () const
 Get private type member variable. More...
 
std::size_t getIndex () const
 Get private index member variable. More...
 
std::size_t getLineNumber () const
 Get private lineNumber member variable. More...
 
std::size_t getEc () const
 Get private ec member variable. More...
 
std::size_t getMinNeighbors () const
 Get private minNeighbors member variable. More...
 
double getRc () const
 Get private rc member variable. More...
 
double getGmin () const
 Get private Gmin member variable. More...
 
double getGmax () const
 Get private Gmax member variable. More...
 
double getScalingFactor () const
 Get private scalingFactor member variable. More...
 
double getConvLength () const
 Get private convLength member variable. More...
 
std::set< std::string > getParameters () const
 Get private parameters member variable. More...
 
std::vector< std::size_t > getIndexPerElement () const
 Get private indexPerElement member variable. More...
 
void setIndex (std::size_t index)
 Set private index member variable. More...
 
void setIndexPerElement (std::size_t elementIndex, std::size_t index)
 Set private indexPerElement member variable. More...
 
void setLineNumber (std::size_t lineNumber)
 Set line number. More...
 
std::string scalingLine () const
 Get string with scaling information. More...
 
virtual double calculateRadialPart (double distance) const =0
 Calculate (partial) symmetry function value for one given distance. More...
 
virtual double calculateAngularPart (double angle) const =0
 Calculate (partial) symmetry function value for one given angle. More...
 
virtual bool checkRelevantElement (std::size_t index) const =0
 Check whether symmetry function is relevant for given element. More...
 
virtual std::vector< std::string > getCacheIdentifiers () const
 Get unique cache identifiers. More...
 
void addCacheIndex (std::size_t element, std::size_t cacheIndex, std::string cacheIdentifier)
 Add one cache index for given neighbor element and check identifier. More...
 
std::vector< std::vector< std::size_t > > getCacheIndices () const
 Getter for cacheIndices. More...
 

Additional Inherited Members

- Public Types inherited from nnp::SymFnc
enum  ScalingType {
  ST_NONE , ST_SCALE , ST_CENTER , ST_SCALECENTER ,
  ST_SCALESIGMA
}
 List of available scaling types. More...
 
- Protected Types inherited from nnp::SymFnc
typedef std::map< std::string, std::pair< std::string, std::string > > PrintFormat
 
typedef std::vector< std::string > PrintOrder
 
- Protected Member Functions inherited from nnp::SymFncBaseCompAngWeighted
 SymFncBaseCompAngWeighted (std::size_t type, ElementMap const &)
 Constructor, initializes type. More...
 
- Protected Member Functions inherited from nnp::SymFncBaseComp
 SymFncBaseComp (std::size_t type, ElementMap const &)
 Constructor, initializes type. More...
 
- Protected Member Functions inherited from nnp::SymFnc
 SymFnc (std::size_t type, ElementMap const &)
 Constructor, initializes type. More...
 
std::string getPrintFormat () const
 Generate format string for symmetry function parameter printing. More...
 
- Static Protected Member Functions inherited from nnp::SymFnc
static PrintFormat const initializePrintFormat ()
 Initialize static print format map for all possible parameters. More...
 
static PrintOrder const initializePrintOrder ()
 Initialize static print order vector for all possible parameters. More...
 
- Protected Attributes inherited from nnp::SymFncBaseCompAngWeighted
double angleLeft
 Left angle boundary. More...
 
double angleRight
 Right angle boundary. More...
 
double angleLeftRadians
 Left angle boundary in radians. More...
 
double angleRightRadians
 Right angle boundary in radians. More...
 
CompactFunction ca
 Compact function member for angular part. More...
 
- Protected Attributes inherited from nnp::SymFncBaseComp
bool asymmetric
 If asymmetric version of polynomials should be used. More...
 
double rl
 Lower bound of compact function, \(r_{l}\). More...
 
std::string subtype
 Subtype string (specifies e.g. polynom type). More...
 
CompactFunction cr
 Compact function for radial part. More...
 
- Protected Attributes inherited from nnp::SymFnc
std::size_t type
 Symmetry function type. More...
 
ElementMap elementMap
 Copy of element map. More...
 
std::size_t index
 Symmetry function index (per element). More...
 
std::size_t lineNumber
 Line number. More...
 
std::size_t ec
 Element index of center atom. More...
 
std::size_t minNeighbors
 Minimum number of neighbors required. More...
 
double Smin
 Minimum for scaling range. More...
 
double Smax
 Maximum for scaling range. More...
 
double Gmin
 Minimum unscaled symmetry function value. More...
 
double Gmax
 Maximum unscaled symmetry function value. More...
 
double Gmean
 Mean unscaled symmetry function value. More...
 
double Gsigma
 Sigma of unscaled symmetry function values. More...
 
double rc
 Cutoff radius \(r_c\). More...
 
double scalingFactor
 Scaling factor. More...
 
double convLength
 Data set normalization length conversion factor. More...
 
ScalingType scalingType
 Symmetry function scaling type used by this symmetry function. More...
 
std::set< std::string > parameters
 Set with symmetry function parameter IDs (lookup for printing). More...
 
std::vector< std::size_t > indexPerElement
 Per-element index for derivative memory in Atom::Neighbor::dGdr arrays. More...
 
std::vector< std::vector< std::size_t > > cacheIndices
 Cache indices for each element. More...
 
- Static Protected Attributes inherited from nnp::SymFnc
static std::size_t const sfinfoWidth = 12
 Width of the SFINFO parameter description field (see parameterInfo()). More...
 
static PrintFormat const printFormat = initializePrintFormat()
 Map of parameter format strings and empty strings. More...
 
static PrintOrder const printOrder = initializePrintOrder()
 Vector of parameters in order of printing. More...
 

Detailed Description

Weighted narrow angular symmetry function with compact support (type 24)

\[ G^{24}_i = \sum_{\substack{j,k\neq i \\ j < k}} Z_j Z_k C(r_{ij}, r_l, r_c) C(r_{ik}, r_l, r_c) C(r_{jk}, r_l, r_c) C(\theta_{ijk}, \theta_l, \theta_r), \]

where \(C(x, x_\text{low}, x_\text{high})\) is a function with compact support \(\left[x_\text{low}, x_\text{high}\right]\). \(Z_j\) is defined as the atomic number of the neighbor atom \(j\).

Parameter string:

<element-central> 24 <rlow> <rcutoff> <left> <right> <subtype>
std::string subtype
Subtype string (specifies e.g. polynom type).

where

  • <element-central> ..... element symbol of central atom
  • <rlow> ................ low radius boundary \(r_{l}\)
  • <rcutoff> ............. high radius boundary \(r_{c}\)
  • <left> ................ left angle boundary \(\theta_l\)
  • <right> ............... right angle boundary \(\theta_r\)
  • <subtype> ............. compact function specifier

See the description of SymFncBaseComp::setCompactFunction() for possible values of the <subtype> argument.

Note
If <subtype> specifies an asymmetric version of a polynomial core function the asymmetry only applies to the radial parts, i.e. "p2a" sets a "p2a" radial compact function and a "p2" angular compact function.

Definition at line 65 of file SymFncCompAngnWeighted.h.

Constructor & Destructor Documentation

◆ SymFncCompAngnWeighted()

SymFncCompAngnWeighted::SymFncCompAngnWeighted ( ElementMap const &  elementMap)

Constructor, sets type = 24.

Definition at line 31 of file SymFncCompAngnWeighted.cpp.

32 :
34{
35}
SymFncBaseCompAngWeighted(std::size_t type, ElementMap const &)
Constructor, initializes type.
ElementMap elementMap
Copy of element map.
Definition: SymFnc.h:270

Member Function Documentation

◆ operator==()

bool SymFncCompAngnWeighted::operator== ( SymFnc const &  rhs) const
virtual

Overload == operator.

Implements nnp::SymFnc.

Definition at line 37 of file SymFncCompAngnWeighted.cpp.

38{
39 if (ec != rhs.getEc() ) return false;
40 if (type != rhs.getType()) return false;
41 SymFncCompAngnWeighted const& c =
42 dynamic_cast<SymFncCompAngnWeighted const&>(rhs);
43 if (subtype != c.getSubtype()) return false;
44 if (rc != c.rc ) return false;
45 if (rl != c.rl ) return false;
46 if (angleLeft != c.angleLeft ) return false;
47 if (angleRight != c.angleRight ) return false;
48 return true;
49}
double angleLeft
Left angle boundary.
double angleRight
Right angle boundary.
std::string getSubtype() const
Get private subtype member variable.
double rl
Lower bound of compact function, .
Weighted narrow angular symmetry function with compact support (type 24)
std::size_t type
Symmetry function type.
Definition: SymFnc.h:268
double rc
Cutoff radius .
Definition: SymFnc.h:292
std::size_t ec
Element index of center atom.
Definition: SymFnc.h:276

References nnp::SymFncBaseCompAngWeighted::angleLeft, nnp::SymFncBaseCompAngWeighted::angleRight, nnp::SymFnc::ec, nnp::SymFnc::getEc(), nnp::SymFncBaseComp::getSubtype(), nnp::SymFnc::getType(), nnp::SymFnc::rc, nnp::SymFncBaseComp::rl, nnp::SymFncBaseComp::subtype, and nnp::SymFnc::type.

Here is the call graph for this function:

◆ operator<()

bool SymFncCompAngnWeighted::operator< ( SymFnc const &  rhs) const
virtual

Overload < operator.

Implements nnp::SymFnc.

Definition at line 51 of file SymFncCompAngnWeighted.cpp.

52{
53 if (ec < rhs.getEc() ) return true;
54 else if (ec > rhs.getEc() ) return false;
55 if (type < rhs.getType()) return true;
56 else if (type > rhs.getType()) return false;
57 SymFncCompAngnWeighted const& c =
58 dynamic_cast<SymFncCompAngnWeighted const&>(rhs);
59 if (subtype < c.getSubtype()) return true;
60 else if (subtype > c.getSubtype()) return false;
61 if (rc < c.rc ) return true;
62 else if (rc > c.rc ) return false;
63 if (rl < c.rl ) return true;
64 else if (rl > c.rl ) return false;
65 if (angleLeft < c.angleLeft ) return true;
66 else if (angleLeft > c.angleLeft ) return false;
67 if (angleRight < c.angleRight ) return true;
68 else if (angleRight > c.angleRight ) return false;
69 return false;
70}

References nnp::SymFncBaseCompAngWeighted::angleLeft, nnp::SymFncBaseCompAngWeighted::angleRight, nnp::SymFnc::ec, nnp::SymFnc::getEc(), nnp::SymFncBaseComp::getSubtype(), nnp::SymFnc::getType(), nnp::SymFnc::rc, nnp::SymFncBaseComp::rl, nnp::SymFncBaseComp::subtype, and nnp::SymFnc::type.

Here is the call graph for this function:

◆ calculate()

void SymFncCompAngnWeighted::calculate ( Atom atom,
bool const  derivatives 
) const
virtual

Calculate symmetry function for one atom.

Parameters
[in,out]atomAtom for which the symmetry function is caluclated.
[in]derivativesIf also symmetry function derivatives will be calculated and saved.

Implements nnp::SymFnc.

Definition at line 72 of file SymFncCompAngnWeighted.cpp.

73{
74 double r2l = 0.0;
75 if (rl > 0.0) r2l = rl * rl;
76 double r2c = rc * rc;
77 double result = 0.0;
78
79 size_t numNeighbors = atom.numNeighbors;
80 // Prevent problematic condition in loop test below (j < numNeighbors - 1).
81 if (numNeighbors == 0) numNeighbors = 1;
82
83 for (size_t j = 0; j < numNeighbors - 1; j++)
84 {
85 Atom::Neighbor& nj = atom.neighbors[j];
86 size_t const nej = nj.element;
87 double const rij = nj.d;
88 if (rij < rc && rij > rl)
89 {
90 double radij;
91 double dradij;
92#ifndef N2P2_NO_SF_CACHE
93 if (cacheIndices[nej].size() == 0) cr.fdf(rij, radij, dradij);
94 else
95 {
96 double& crad = nj.cache[cacheIndices[nej][0]];
97 double& cdrad = nj.cache[cacheIndices[nej][1]];
98 if (crad < 0) cr.fdf(rij, crad, cdrad);
99 radij = crad;
100 dradij = cdrad;
101 }
102#else
103 cr.fdf(rij, radij, dradij);
104#endif
105 for (size_t k = j + 1; k < numNeighbors; k++)
106 {
107 Atom::Neighbor& nk = atom.neighbors[k];
108 size_t const nek = nk.element;
109 double const rik = nk.d;
110 if (rik < rc && rik > rl)
111 {
112 // Energy calculation.
113 Vec3D drij = nj.dr;
114 Vec3D drik = nk.dr;
115 Vec3D drjk = nk.dr - nj.dr;
116 double rjk = drjk.norm2();
117 if (rjk >= r2c || rjk <= r2l) continue;
118 rjk = sqrt(rjk);
119
120 double radik;
121 double dradik;
122#ifndef N2P2_NO_SF_CACHE
123 if (cacheIndices[nek].size() == 0)
124 {
125 cr.fdf(rik, radik, dradik);
126 }
127 else
128 {
129 double& crad = nk.cache[cacheIndices[nek][0]];
130 double& cdrad = nk.cache[cacheIndices[nek][1]];
131 if (crad < 0) cr.fdf(rik, crad, cdrad);
132 radik = crad;
133 dradik = cdrad;
134 }
135#else
136 cr.fdf(rik, radik, dradik);
137#endif
138 double radjk;
139 double dradjk;
140 cr.fdf(rjk, radjk, dradjk);
141
142 double costijk = drij * drik;
143 double rinvijik = 1.0 / rij / rik;
144 costijk *= rinvijik;
145
146 // By definition, our polynomial is zero at 0 and 180 deg.
147 // Therefore, skip the whole rest which might yield some NaN
148 if (costijk <= -1.0 || costijk >= 1.0) continue;
149
150 // Regroup later: Get acos(cos)
151 double const acostijk = acos(costijk);
152 // Only go on if we are within our compact support
153 if (acostijk < angleLeftRadians ||
154 acostijk > angleRightRadians) continue;
155 double ang = 0.0;
156 double dang = 0.0;
157 ca.fdf(acostijk, ang, dang);
158
159 double const rad = radij * radik * radjk; // product of cutoff fcts
160 ang *= elementMap.atomicNumber(nej)
162 result += rad * ang;
163
164 // Force calculation.
165 if (!derivatives) continue;
166
167 double const dacostijk = -1.0
168 / sqrt(1.0 - costijk * costijk);
169 dang *= dacostijk
172 double const rinvij = rinvijik * rik;
173 double const rinvik = rinvijik * rij;
174 double const rinvjk = 1.0 / rjk;
175 double phiijik = rinvij * (rinvik - rinvij * costijk);
176 double phiikij = rinvik * (rinvij - rinvik * costijk);
177 double psiijik = rinvijik; // careful: sign flip w.r.t. notes due to nj.dGd...
178 phiijik *= dang;
179 phiikij *= dang;
180 psiijik *= dang;
181
182 // Cutoff function might be a divide by zero, but we screen that case before
183 double const chiij = rinvij * dradij * radik * radjk;
184 double const chiik = rinvik * radij * dradik * radjk;
185 double const chijk = -rinvjk * radij * radik * dradjk;
186
187 // rijs/rij due to the shifted radial part of the Gaussian
188 double const p1 = rad * phiijik + ang * chiij;
189 double const p2 = rad * phiikij + ang * chiik;
190 double const p3 = rad * psiijik + ang * chijk;
191 drij *= p1 * scalingFactor;
192 drik *= p2 * scalingFactor;
193 drjk *= p3 * scalingFactor;
194
195 // Save force contributions in Atom storage.
196 atom.dGdr[index] += drij + drik;
197#ifndef N2P2_FULL_SFD_MEMORY
198 nj.dGdr[indexPerElement[nej]] -= drij + drjk;
199 nk.dGdr[indexPerElement[nek]] -= drik - drjk;
200#else
201 nj.dGdr[index] -= drij + drjk;
202 nk.dGdr[index] -= drik - drjk;
203#endif
204 } // rik <= rc
205 } // k
206 } // rij <= rc
207 } // j
208
209 atom.G[index] = scale(result);
210
211 return;
212}
void fdf(double a, double &fa, double &dfa) const
Calculate compact function and derivative at once.
std::size_t atomicNumber(std::size_t index) const
Get atomic number from element index.
Definition: ElementMap.h:145
CompactFunction ca
Compact function member for angular part.
double angleLeftRadians
Left angle boundary in radians.
double angleRightRadians
Right angle boundary in radians.
CompactFunction cr
Compact function for radial part.
std::size_t index
Symmetry function index (per element).
Definition: SymFnc.h:272
double scalingFactor
Scaling factor.
Definition: SymFnc.h:294
std::vector< std::vector< std::size_t > > cacheIndices
Cache indices for each element.
Definition: SymFnc.h:306
double scale(double value) const
Apply symmetry function scaling and/or centering.
Definition: SymFnc.cpp:169
std::vector< std::size_t > indexPerElement
Per-element index for derivative memory in Atom::Neighbor::dGdr arrays.
Definition: SymFnc.h:302
Struct to store information on neighbor atoms.
Definition: Atom.h:35
std::vector< double > cache
Symmetry function cache (e.g. for cutoffs, compact functions).
Definition: Atom.h:48
std::size_t element
Element index of neighbor atom.
Definition: Atom.h:41
double d
Distance to neighbor atom.
Definition: Atom.h:43
Vec3D dr
Distance vector to neighbor atom.
Definition: Atom.h:45
std::vector< Vec3D > dGdr
Derivatives of symmetry functions with respect to neighbor coordinates.
Definition: Atom.h:59
std::vector< Neighbor > neighbors
Neighbor array (maximum number defined in macros.h.
Definition: Atom.h:148
std::vector< Vec3D > dGdr
Derivative of symmetry functions with respect to this atom's coordinates.
Definition: Atom.h:146
std::vector< double > G
Symmetry function values.
Definition: Atom.h:134
std::size_t numNeighbors
Total number of neighbors.
Definition: Atom.h:106
Vector in 3 dimensional real space.
Definition: Vec3D.h:29
double norm2() const
Calculate square of norm of vector.
Definition: Vec3D.h:261

References nnp::SymFncBaseCompAngWeighted::angleLeftRadians, nnp::SymFncBaseCompAngWeighted::angleRightRadians, nnp::ElementMap::atomicNumber(), nnp::SymFncBaseCompAngWeighted::ca, nnp::Atom::Neighbor::cache, nnp::SymFnc::cacheIndices, nnp::SymFncBaseComp::cr, nnp::Atom::Neighbor::d, nnp::Atom::Neighbor::dGdr, nnp::Atom::dGdr, nnp::Atom::Neighbor::dr, nnp::Atom::Neighbor::element, nnp::SymFnc::elementMap, nnp::CompactFunction::fdf(), nnp::Atom::G, nnp::SymFnc::index, nnp::SymFnc::indexPerElement, nnp::Atom::neighbors, nnp::Vec3D::norm2(), nnp::Atom::numNeighbors, nnp::SymFnc::rc, nnp::SymFncBaseComp::rl, nnp::SymFnc::scale(), and nnp::SymFnc::scalingFactor.

Here is the call graph for this function:

The documentation for this class was generated from the following files: