mirror of
https://github.com/rdkit/rdkit.git
synced 2026-06-04 21:54:27 +08:00
e1cd9f68c5
* move distance constraint to forcefield * move position and torsion constraint to force field, also remove some unused includes * sorting includes * add missing files
68 lines
2.0 KiB
C++
68 lines
2.0 KiB
C++
//
|
|
// Copyright (C) 2013-2024 Paolo Tosco 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 "PositionConstraint.h"
|
|
#include "ForceField.h"
|
|
#include <RDGeneral/Invariant.h>
|
|
#include <algorithm>
|
|
#include <cmath>
|
|
|
|
namespace ForceFields {
|
|
PositionConstraintContrib::PositionConstraintContrib(ForceField *owner,
|
|
unsigned int idx,
|
|
double maxDispl,
|
|
double forceConst) {
|
|
PRECONDITION(owner, "bad owner");
|
|
const RDGeom::PointPtrVect &pos = owner->positions();
|
|
URANGE_CHECK(idx, pos.size());
|
|
|
|
dp_forceField = owner;
|
|
d_atIdx = idx;
|
|
d_maxDispl = maxDispl;
|
|
d_pos0 = *((RDGeom::Point3D *)pos[idx]);
|
|
d_forceConstant = forceConst;
|
|
}
|
|
|
|
double PositionConstraintContrib::getEnergy(double *pos) const {
|
|
PRECONDITION(dp_forceField, "no owner");
|
|
PRECONDITION(pos, "bad vector");
|
|
|
|
RDGeom::Point3D p(pos[3 * d_atIdx], pos[3 * d_atIdx + 1],
|
|
pos[3 * d_atIdx + 2]);
|
|
double dist = (p - d_pos0).length();
|
|
double distTerm = std::max(dist - d_maxDispl, 0.0);
|
|
double res = 0.5 * d_forceConstant * distTerm * distTerm;
|
|
|
|
return res;
|
|
}
|
|
|
|
void PositionConstraintContrib::getGrad(double *pos, double *grad) const {
|
|
PRECONDITION(dp_forceField, "no owner");
|
|
PRECONDITION(pos, "bad vector");
|
|
PRECONDITION(grad, "bad vector");
|
|
|
|
RDGeom::Point3D p(pos[3 * d_atIdx], pos[3 * d_atIdx + 1],
|
|
pos[3 * d_atIdx + 2]);
|
|
double dist = (p - d_pos0).length();
|
|
|
|
double preFactor = 0.0;
|
|
if (dist > d_maxDispl) {
|
|
preFactor = dist - d_maxDispl;
|
|
} else {
|
|
return;
|
|
}
|
|
preFactor *= d_forceConstant;
|
|
|
|
for (unsigned int i = 0; i < 3; ++i) {
|
|
double dGrad = preFactor * (p[i] - d_pos0[i]) / std::max(dist, 1.0e-8);
|
|
grad[3 * d_atIdx + i] += dGrad;
|
|
}
|
|
}
|
|
} // namespace ForceFields
|