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
|
// $Id$
//
// Copyright (C) 2004-2006 Rational Discovery LLC
//
// @@ 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 <GraphMol/RDKitBase.h>
#include <GraphMol/SmilesParse/SmilesParse.h>
#include "MolChemicalFeature.h"
#include "MolChemicalFeatureDef.h"
namespace RDKit {
const std::string &MolChemicalFeature::getFamily() const {
return dp_def->getFamily(); // return d_family;
};
const std::string &MolChemicalFeature::getType() const {
return dp_def->getType(); // return d_type;
};
void MolChemicalFeature::setActiveConformer(int confId) {
PRECONDITION(dp_mol, "bad molecule");
d_activeConf = confId;
}
RDGeom::Point3D MolChemicalFeature::getPos() const {
return this->getPos(d_activeConf);
}
RDGeom::Point3D MolChemicalFeature::getPos(int confId) const {
PRECONDITION(dp_mol, "bad molecule");
PRECONDITION(dp_mol->getNumConformers(), "molecule has no conformers");
if (confId == -1) confId = (*dp_mol->beginConformers())->getId();
// -------------
// Check to see if we've got the value cached:
PointCacheType::const_iterator cacheIt = d_locs.find(confId);
if (cacheIt != d_locs.end()) {
return cacheIt->second;
}
// --------------
// Nope, we have to figure it out on our own:
RDGeom::Point3D res(0, 0, 0);
bool setNeg1 = false;
RDUNUSED_PARAM(setNeg1);
if (confId == -1) {
setNeg1 = true;
}
if (d_atoms.size() == 1) {
res = dp_mol->getConformer(confId).getAtomPos((*d_atoms.begin())->getIdx());
} else {
PRECONDITION(dp_def, "bad definition");
PRECONDITION(dp_def->getNumWeights() == this->getNumAtoms(),
"weight/atom mismatch");
auto weightIt = dp_def->beginWeights();
const Conformer &conf = dp_mol->getConformer(confId);
for (auto atomIt = d_atoms.begin(); atomIt != d_atoms.end();
atomIt++, weightIt++) {
const Atom *atom = *atomIt;
RDGeom::Point3D p = conf.getAtomPos(atom->getIdx());
p *= *weightIt;
res += p;
}
}
d_locs[confId] = res;
return res;
}
}
|