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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
//
// Copyright (C) 2017 Novartis Institutes for BioMedical Research
//
// @@ 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.
//
#ifndef RGROUP_MATCH_DATA
#define RGROUP_MATCH_DATA
#include "RGroupData.h"
namespace RDKit {
typedef boost::shared_ptr<RGroupData> RData;
typedef std::map<int, RData> R_DECOMP;
//! RGroupMatch is the decomposition for a single molecule
struct RGroupMatch {
public:
size_t core_idx; // index of the matching core
size_t numberMissingUserRGroups;
R_DECOMP rgroups; // rlabel->RGroupData mapping
RWMOL_SPTR matchedCore; // Core with dummy or query atoms and bonds matched
RGroupMatch(size_t core_index, size_t numberMissingUserRGroups,
R_DECOMP input_rgroups, RWMOL_SPTR matchedCore)
: core_idx(core_index),
numberMissingUserRGroups(numberMissingUserRGroups),
rgroups(std::move(input_rgroups)),
matchedCore(std::move(matchedCore)) {}
std::string toString() const {
auto rGroupsString = std::accumulate(
rgroups.cbegin(), rgroups.cend(), std::string(),
[](std::string s, const std::pair<int, RData> &rgroup) {
return std::move(s) + "\n\t(" + std::to_string(rgroup.first) + ':' +
rgroup.second->toString() + ')';
});
std::stringstream ss;
ss << "Match coreIdx " << core_idx << " missing count "
<< numberMissingUserRGroups << " " << rGroupsString;
return ss.str();
}
//! Set the target molecule to be used for highlighting R groups
//! \param targetMol the target molecule
void setTargetMoleculeForHighlights(const RWMOL_SPTR &targetMol) {
targetMolForHighlights = targetMol;
targetMolWasTrimmed = false;
}
//! Get the target molecule to be used for highlighting R groups
//! \param trimHs whether explicit hydrogens should be removed,
//! except for those corresponding to R groups (if any)
//! \return the target molecule (can be null if it was never set)
RWMOL_SPTR getTargetMoleculeForHighlights(bool trimHs) {
if (!targetMolForHighlights || !trimHs || targetMolWasTrimmed) {
return targetMolForHighlights;
}
// if trimHs is true and this has not been done before, we need
// to remove explicit Hs, except for those corresponding to R groups
// (if any). Removal of hydrogens will change atom and bond indices,
// therefore common_properties::_rgroupTargetAtoms and
// common_properties::_rgroupTargetBonds need to be updated
int numMolAtoms = targetMolForHighlights->getNumAtoms();
std::vector<int> storedAtomMapNums(numMolAtoms);
std::vector<std::pair<int, int>> oldBondEnds(
targetMolForHighlights->getNumBonds(), std::make_pair(-1, -1));
auto atoms = targetMolForHighlights->atoms();
// we use atom map numbers to track original atom indices
// ahead of removing Hs, so we store existing values to be able
// to restore them afterwards
std::transform(atoms.begin(), atoms.end(), storedAtomMapNums.begin(),
[](auto atom) {
auto res = atom->getAtomMapNum();
atom->setAtomMapNum(0);
return res;
});
for (const auto &pair : rgroups) {
auto &combinedMol = pair.second->combinedMol;
std::vector<int> bondIndices;
if (combinedMol->getPropIfPresent(common_properties::_rgroupTargetBonds,
bondIndices)) {
std::for_each(bondIndices.begin(), bondIndices.end(),
[this, &oldBondEnds](const auto &bondIdx) {
const auto bond =
targetMolForHighlights->getBondWithIdx(bondIdx);
const auto beginAtom = bond->getBeginAtom();
const auto endAtom = bond->getEndAtom();
oldBondEnds[bondIdx].first = beginAtom->getIdx();
oldBondEnds[bondIdx].second = endAtom->getIdx();
beginAtom->setAtomMapNum(beginAtom->getIdx() + 1);
endAtom->setAtomMapNum(endAtom->getIdx() + 1);
});
}
}
// remove Hs except those involved in R groups
std::vector<int> oldToNewAtomIndices(numMolAtoms, -1);
MolOps::RemoveHsParameters rhps;
rhps.removeMapped = false;
MolOps::removeHs(*targetMolForHighlights, rhps);
for (auto atom : targetMolForHighlights->atoms()) {
auto atomMapNum = atom->getAtomMapNum();
if (atomMapNum) {
--atomMapNum;
oldToNewAtomIndices[atomMapNum] = atom->getIdx();
atom->setAtomMapNum(storedAtomMapNums.at(atomMapNum));
}
}
// update atom and bond indices after removing Hs
for (const auto &pair : rgroups) {
auto &combinedMol = pair.second->combinedMol;
std::vector<int> atomIndices;
if (combinedMol->getPropIfPresent(common_properties::_rgroupTargetAtoms,
atomIndices)) {
std::transform(
atomIndices.begin(), atomIndices.end(), atomIndices.begin(),
[&oldToNewAtomIndices](auto &atomIdx) {
auto newAtomIdx = oldToNewAtomIndices.at(atomIdx);
CHECK_INVARIANT(newAtomIdx != -1, "newAtomIdx must be >=0");
return newAtomIdx;
});
}
combinedMol->setProp(common_properties::_rgroupTargetAtoms, atomIndices);
std::vector<int> bondIndices;
if (combinedMol->getPropIfPresent(common_properties::_rgroupTargetBonds,
bondIndices)) {
std::transform(
bondIndices.begin(), bondIndices.end(), bondIndices.begin(),
[this, &oldBondEnds, &oldToNewAtomIndices](auto &bondIdx) {
const auto &oldPair = oldBondEnds.at(bondIdx);
CHECK_INVARIANT(oldPair.first != -1 && oldPair.second != -1,
"oldPair members must be >=0");
const auto newBeginAtomIdx =
oldToNewAtomIndices.at(oldPair.first);
const auto newEndAtomIdx = oldToNewAtomIndices.at(oldPair.second);
CHECK_INVARIANT(newBeginAtomIdx != -1 && newEndAtomIdx != -1,
"newBeginAtomIdx and newEndAtomIdx must be >=0");
const auto bond = targetMolForHighlights->getBondBetweenAtoms(
newBeginAtomIdx, newEndAtomIdx);
CHECK_INVARIANT(bond, "bond must not be null");
return bond->getIdx();
});
}
combinedMol->setProp(common_properties::_rgroupTargetBonds, bondIndices);
}
targetMolWasTrimmed = true;
return targetMolForHighlights;
}
private:
bool targetMolWasTrimmed = false;
RWMOL_SPTR targetMolForHighlights;
};
} // namespace RDKit
#endif
|
