mirror of
https://github.com/rdkit/rdkit.git
synced 2026-06-05 22:04:27 +08:00
2e3f31990d
* backup * simple first pass, passes all tests * cleanup a bunch of existing uses * ensure that we can safely add atoms/bonds while in edit mode * add context manager on python side * handle exceptions properly in those * changes in response to review
104 lines
3.3 KiB
C++
104 lines
3.3 KiB
C++
//
|
|
// Copyright (C) 2020 Greg Landrum and T5 Informatics GmbH
|
|
//
|
|
// @@ 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 "MolEnumerator.h"
|
|
#include <RDGeneral/Exceptions.h>
|
|
#include <GraphMol/FileParsers/MolSGroupParsing.h>
|
|
|
|
namespace RDKit {
|
|
namespace MolEnumerator {
|
|
|
|
void PositionVariationOp::initFromMol(const ROMol &mol) {
|
|
dp_mol.reset(new ROMol(mol));
|
|
initFromMol();
|
|
}
|
|
void PositionVariationOp::initFromMol() {
|
|
d_variationPoints.clear();
|
|
if (!dp_mol) {
|
|
return;
|
|
}
|
|
if (!dp_mol->hasProp(detail::idxPropName)) {
|
|
detail::preserveOrigIndices(*dp_mol);
|
|
}
|
|
for (const auto bond : dp_mol->bonds()) {
|
|
std::string endpts;
|
|
std::string attach;
|
|
if (bond->getPropIfPresent(common_properties::_MolFileBondEndPts, endpts) &&
|
|
bond->getPropIfPresent(common_properties::_MolFileBondAttach, attach) &&
|
|
attach == "ANY") {
|
|
const Atom *atom = bond->getBeginAtom();
|
|
if (atom->getAtomicNum() == 0) {
|
|
atom = bond->getEndAtom();
|
|
if (atom->getAtomicNum() == 0) {
|
|
throw ValueErrorException(
|
|
"position variation bond does not have connection to a "
|
|
"non-dummy atom");
|
|
}
|
|
}
|
|
d_dummiesAtEachPoint.push_back(bond->getOtherAtomIdx(atom->getIdx()));
|
|
std::vector<unsigned int> oats =
|
|
RDKit::SGroupParsing::ParseV3000Array<unsigned int>(endpts);
|
|
// decrement the indices and do error checking:
|
|
for (auto &oat : oats) {
|
|
if (oat == 0 || oat > dp_mol->getNumAtoms()) {
|
|
throw ValueErrorException("Bad variation point index");
|
|
}
|
|
--oat;
|
|
}
|
|
d_variationPoints.push_back(std::make_pair(atom->getIdx(), oats));
|
|
}
|
|
}
|
|
}
|
|
|
|
std::vector<size_t> PositionVariationOp::getVariationCounts() const {
|
|
std::vector<size_t> res(d_variationPoints.size());
|
|
std::transform(
|
|
d_variationPoints.begin(), d_variationPoints.end(), res.begin(),
|
|
[](std::pair<unsigned int, std::vector<unsigned int>> pr) -> size_t {
|
|
return pr.second.size();
|
|
});
|
|
return res;
|
|
}
|
|
|
|
std::unique_ptr<ROMol> PositionVariationOp::operator()(
|
|
const std::vector<size_t> &which) const {
|
|
PRECONDITION(dp_mol, "no molecule");
|
|
if (which.size() != d_variationPoints.size()) {
|
|
throw ValueErrorException("bad element choice in enumeration");
|
|
}
|
|
// a bit of quick error checking before starting the real work
|
|
for (unsigned int i = 0; i < d_variationPoints.size(); ++i) {
|
|
if (which[i] >= d_variationPoints[i].second.size()) {
|
|
throw ValueErrorException("bad element value in enumeration");
|
|
}
|
|
}
|
|
RWMol *res = new RWMol(*dp_mol);
|
|
for (unsigned int i = 0; i < d_variationPoints.size(); ++i) {
|
|
const auto tpl = d_variationPoints[i];
|
|
auto begAtomIdx = tpl.first;
|
|
auto endAtomIdx = tpl.second[which[i]];
|
|
// do we already have a bond?
|
|
if (res->getBondBetweenAtoms(begAtomIdx, endAtomIdx)) {
|
|
continue;
|
|
}
|
|
res->addBond(begAtomIdx, endAtomIdx, Bond::BondType::SINGLE);
|
|
}
|
|
// now remove the dummies:
|
|
res->beginBatchEdit();
|
|
for (auto idx : d_dummiesAtEachPoint) {
|
|
res->removeAtom(idx);
|
|
}
|
|
res->commitBatchEdit();
|
|
return std::unique_ptr<ROMol>(static_cast<ROMol *>(res));
|
|
}
|
|
|
|
} // namespace MolEnumerator
|
|
|
|
} // namespace RDKit
|