1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
//
// Copyright (C) 2024 Niels Maeder and other RDKit contributors
//
// @@ All Rights Reserved @@
// This file is part of the RDKit.
// The contents are covered by the terms of the BSD license
// which is included in the file license.txt, found at the root
// of the RDKit source tree.
//
#include <RDGeneral/export.h>
#ifndef RD_DISTANCECONSTRAINTS_H
#define RD_DISTANCECONSTRAINTS_H
#include <vector>
#include <iostream>
#include "Contrib.h"
namespace ForceFields {
struct DistanceConstraintContribsParams {
unsigned int idx1{0}; //!< index of atom1 of the distance constraint
unsigned int idx2{0}; //!< index of atom2 of the distance constraint
double minLen{0.0}; //!< lower bound of the flat bottom potential
double maxLen{0.0}; //!< upper bound of the flat bottom potential
double forceConstant{1.0}; //!< force constant for distance constraint
DistanceConstraintContribsParams(unsigned int idx1, unsigned int idx2,
double minLen, double maxLen,
double forceConstant = 1.0)
: idx1(idx1),
idx2(idx2),
minLen(minLen),
maxLen(maxLen),
forceConstant(forceConstant) {};
};
//! A term to capture all flat bottom distance constraint potentials.
class RDKIT_FORCEFIELD_EXPORT DistanceConstraintContribs
: public ForceFieldContrib {
public:
DistanceConstraintContribs() = default;
//! Constructor
/*!
\param owner pointer to the owning ForceField
*/
DistanceConstraintContribs(ForceField *owner);
~DistanceConstraintContribs() override = default;
//! Add contribution to this contrib.
/*!
\param idx1 index of atom1 in the ForceField's positions
\param idx2 index of atom2 in the ForceField's positions
\param minLen minimum distance
\param maxLen maximum distance
\param forceConst force Constant
*/
void addContrib(unsigned int idx1, unsigned int idx2, double minLen,
double maxLen, double forceConstant);
//! Add contribution to this contrib.
/*!
\param idx1 index of atom1 in the ForceField's positions
\param idx2 index of atom2 in the ForceField's positions
\param relative whether to add the provided distance to the
current distance
\param minLen minimum distance
\param maxLen maximum distance
\param forceConst force Constant
*/
void addContrib(unsigned int idx1, unsigned int idx2, bool relative,
double minLen, double maxLen, double forceConstant);
//! return the contribution of this contrib to the energy of a given state
/*!
\param pos positions of the atoms in the current state
*/
double getEnergy(double *pos) const override;
//! calculate the contribution of this contrib to the gradient at a given
/// state
/*!
\param pos positions of the atoms in the current state
\param grad gradients to be adapted
*/
void getGrad(double *pos, double *grad) const override;
//! Copy constructor
DistanceConstraintContribs *copy() const override {
return new DistanceConstraintContribs(*this);
}
//! Return true if there are no contributions in this contrib
bool empty() const { return d_contribs.empty(); }
//! Get number of contributions in this contrib
unsigned int size() const { return d_contribs.size(); }
private:
std::vector<DistanceConstraintContribsParams> d_contribs;
};
} // namespace ForceFields
#endif
|
