19#include <boost/dynamic_bitset.hpp>
35 unsigned int bondStereo{
36 static_cast<unsigned int>(Bond::BondStereo::STEREONONE)};
37 unsigned int nbrSymClass{0};
38 unsigned int nbrIdx{0};
40 const canon_atom *controllingAtoms[4]{
nullptr,
nullptr,
nullptr,
nullptr};
41 const std::string *p_symbol{
43 unsigned int bondIdx{0};
47 unsigned int nsc,
unsigned int bidx)
49 bondStereo(static_cast<unsigned int>(bs)),
54 unsigned int nsc,
unsigned int bidx)
65 return compare(lhs, rhs) > 0;
69 unsigned int div = 1) {
103 unsigned int degree{0};
104 unsigned int totalNumHs{0};
105 bool hasRingNbr{
false};
106 bool isRingStereoAtom{
false};
107 unsigned int whichStereoGroup{0};
110 const std::string *p_symbol{
122 std::vector<std::pair<unsigned int, unsigned int>> &
result);
139 const boost::dynamic_bitset<> *dp_atomsInPlay{
nullptr},
140 *dp_bondsInPlay{
nullptr};
145 const boost::dynamic_bitset<> *atomsInPlay =
nullptr,
146 const boost::dynamic_bitset<> *bondsInPlay =
nullptr)
149 dp_atomsInPlay(atomsInPlay),
150 dp_bondsInPlay(bondsInPlay) {}
155 if (dp_atomsInPlay && !((*dp_atomsInPlay)[i] || (*dp_atomsInPlay)[j])) {
159 if (!dp_atomsInPlay || (*dp_atomsInPlay)[i]) {
162 if (!dp_atomsInPlay || (*dp_atomsInPlay)[j]) {
165 for (
unsigned int ii = 0;
166 ii < dp_atoms[i].bonds.size() && ii < dp_atoms[j].bonds.size(); ++ii) {
168 bondholder::compare(dp_atoms[i].bonds[ii], dp_atoms[j].bonds[ii]);
174 std::vector<std::pair<unsigned int, unsigned int>> swapsi;
175 std::vector<std::pair<unsigned int, unsigned int>> swapsj;
176 if (!dp_atomsInPlay || (*dp_atomsInPlay)[i]) {
179 if (!dp_atomsInPlay || (*dp_atomsInPlay)[j]) {
182 for (
unsigned int ii = 0; ii < swapsi.size() && ii < swapsj.size(); ++ii) {
183 int cmp = swapsi[ii].second - swapsj[ii].second;
196 const boost::dynamic_bitset<> *dp_atomsInPlay{
nullptr},
197 *dp_bondsInPlay{
nullptr};
202 const boost::dynamic_bitset<> *atomsInPlay =
nullptr,
203 const boost::dynamic_bitset<> *bondsInPlay =
nullptr)
206 dp_atomsInPlay(atomsInPlay),
207 dp_bondsInPlay(bondsInPlay) {}
212 if (dp_atomsInPlay && !((*dp_atomsInPlay)[i] || (*dp_atomsInPlay)[j])) {
216 if (dp_atoms[i].neighborNum < dp_atoms[j].neighborNum) {
218 }
else if (dp_atoms[i].neighborNum > dp_atoms[j].neighborNum) {
222 if (dp_atoms[i].revistedNeighbors < dp_atoms[j].revistedNeighbors) {
224 }
else if (dp_atoms[i].revistedNeighbors > dp_atoms[j].revistedNeighbors) {
228 if (!dp_atomsInPlay || (*dp_atomsInPlay)[i]) {
231 if (!dp_atomsInPlay || (*dp_atomsInPlay)[j]) {
234 for (
unsigned int ii = 0;
235 ii < dp_atoms[i].bonds.size() && ii < dp_atoms[j].bonds.size(); ++ii) {
237 bondholder::compare(dp_atoms[i].bonds[ii], dp_atoms[j].bonds[ii]);
243 if (dp_atoms[i].bonds.size() < dp_atoms[j].bonds.size()) {
245 }
else if (dp_atoms[i].bonds.size() > dp_atoms[j].bonds.size()) {
255 unsigned int res = 0;
256 std::vector<unsigned int>
perm;
257 perm.reserve(dp_atoms[i].atom->getDegree());
258 for (
const auto nbr : dp_mol->atomNeighbors(dp_atoms[i].atom)) {
259 auto rnk = dp_atoms[
nbr->getIdx()].index;
267 if (
perm.size() == dp_atoms[i].atom->getDegree()) {
268 auto ctag = dp_atoms[i].atom->getChiralTag();
284 unsigned int getAtomRingNbrCode(
unsigned int i)
const {
285 if (!dp_atoms[i].hasRingNbr) {
289 auto nbrs = dp_atoms[i].nbrIds.get();
290 unsigned int code = 0;
291 for (
unsigned j = 0; j < dp_atoms[i].degree; ++j) {
292 if (dp_atoms[nbrs[j]].isRingStereoAtom) {
293 code += dp_atoms[nbrs[j]].index * 10000 + 1;
299 int basecomp(
int i,
int j)
const {
300 unsigned int ivi, ivj;
303 ivi = dp_atoms[i].index;
304 ivj = dp_atoms[j].index;
307 }
else if (ivi > ivj) {
310 if (df_useAtomMaps) {
312 int molAtomMapNumber_i = 0;
313 int molAtomMapNumber_j = 0;
314 dp_atoms[i].atom->getPropIfPresent(common_properties::molAtomMapNumber,
316 dp_atoms[j].atom->getPropIfPresent(common_properties::molAtomMapNumber,
318 if (molAtomMapNumber_i < molAtomMapNumber_j) {
320 }
else if (molAtomMapNumber_i > molAtomMapNumber_j) {
325 ivi = dp_atoms[i].degree;
326 ivj = dp_atoms[j].degree;
329 }
else if (ivi > ivj) {
332 if (dp_atoms[i].p_symbol && dp_atoms[j].p_symbol) {
333 if (*(dp_atoms[i].p_symbol) < *(dp_atoms[j].p_symbol)) {
335 }
else if (*(dp_atoms[i].p_symbol) > *(dp_atoms[j].p_symbol)) {
343 ivi = dp_atoms[i].atom->getAtomicNum();
344 ivj = dp_atoms[j].atom->getAtomicNum();
347 }
else if (ivi > ivj) {
351 if (df_useIsotopes) {
352 ivi = dp_atoms[i].atom->getIsotope();
353 ivj = dp_atoms[j].atom->getIsotope();
356 }
else if (ivi > ivj) {
362 ivi = dp_atoms[i].totalNumHs;
363 ivj = dp_atoms[j].totalNumHs;
366 }
else if (ivi > ivj) {
370 ivi = dp_atoms[i].atom->getFormalCharge();
371 ivj = dp_atoms[j].atom->getFormalCharge();
374 }
else if (ivi > ivj) {
378 if (df_useChirality) {
380 ivi = dp_atoms[i].whichStereoGroup;
382 ivj = dp_atoms[j].whichStereoGroup;
386 }
else if (ivj && !ivi) {
388 }
else if (ivi && ivj) {
389 ivi =
static_cast<unsigned int>(dp_atoms[i].typeOfStereoGroup);
390 ivj =
static_cast<unsigned int>(dp_atoms[j].typeOfStereoGroup);
393 }
else if (ivi > ivj) {
396 ivi = dp_atoms[i].whichStereoGroup - 1;
397 ivj = dp_atoms[j].whichStereoGroup - 1;
399 std::set<unsigned int> sgi;
401 sgi.insert(dp_atoms[sgat->getIdx()].index);
403 std::set<unsigned int> sgj;
405 sgj.insert(dp_atoms[sgat->getIdx()].index);
409 }
else if (sgi > sgj) {
419 ivi = dp_atoms[i].atom->getChiralTag() != 0;
420 ivj = dp_atoms[j].atom->getChiralTag() != 0;
423 }
else if (ivi > ivj) {
429 ivi = getChiralRank(dp_mol, dp_atoms, i);
432 ivj = getChiralRank(dp_mol, dp_atoms, j);
436 }
else if (ivi > ivj) {
443 if (df_useChiralityRings) {
445 ivi = getAtomRingNbrCode(i);
446 ivj = getAtomRingNbrCode(j);
449 }
else if (ivi > ivj) {
459 const boost::dynamic_bitset<> *dp_atomsInPlay{
nullptr},
460 *dp_bondsInPlay{
nullptr};
461 bool df_useNbrs{
false};
462 bool df_useIsotopes{
true};
463 bool df_useChirality{
true};
464 bool df_useChiralityRings{
true};
465 bool df_useAtomMaps{
true};
469 const boost::dynamic_bitset<> *atomsInPlay =
nullptr,
470 const boost::dynamic_bitset<> *bondsInPlay =
nullptr)
473 dp_atomsInPlay(atomsInPlay),
474 dp_bondsInPlay(bondsInPlay),
476 df_useIsotopes(true),
477 df_useChirality(true),
478 df_useChiralityRings(true),
479 df_useAtomMaps(true) {}
481 if (dp_atomsInPlay && !((*dp_atomsInPlay)[i] || (*dp_atomsInPlay)[j])) {
484 int v = basecomp(i, j);
490 if (!dp_atomsInPlay || (*dp_atomsInPlay)[i]) {
493 if (!dp_atomsInPlay || (*dp_atomsInPlay)[j]) {
497 for (
unsigned int ii = 0;
498 ii < dp_atoms[i].bonds.size() && ii < dp_atoms[j].bonds.size();
501 bondholder::compare(dp_atoms[i].bonds[ii], dp_atoms[j].bonds[ii]);
507 if (dp_atoms[i].bonds.size() < dp_atoms[j].bonds.size()) {
509 }
else if (dp_atoms[i].bonds.size() > dp_atoms[j].bonds.size()) {
524 void getAtomNeighborhood(std::vector<bondholder> &nbrs)
const {
525 for (
unsigned j = 0; j < nbrs.size(); ++j) {
526 unsigned int nbrIdx = nbrs[j].nbrIdx;
531 const Atom *nbr = dp_atoms[nbrIdx].atom;
532 nbrs[j].nbrSymClass =
535 std::sort(nbrs.begin(), nbrs.end(), bondholder::greater);
539 int basecomp(
int i,
int j)
const {
541 unsigned int ivi, ivj;
544 ivi = dp_atoms[i].index;
545 ivj = dp_atoms[j].index;
548 }
else if (ivi > ivj) {
553 ivi = dp_atoms[i].atom->getAtomicNum();
554 ivj = dp_atoms[j].atom->getAtomicNum();
557 }
else if (ivi > ivj) {
562 ivi = dp_atoms[i].atom->getIsotope();
563 ivj = dp_atoms[j].atom->getIsotope();
566 }
else if (ivi > ivj) {
574 if (dp_atoms[i].atom->getPropIfPresent(common_properties::_CIPCode,
576 ivi = cipCode ==
"R" ? 2 : 1;
578 if (dp_atoms[j].atom->getPropIfPresent(common_properties::_CIPCode,
580 ivj = cipCode ==
"R" ? 2 : 1;
584 }
else if (ivi > ivj) {
595 bool df_useNbrs{
false};
598 : dp_atoms(atoms), dp_mol(&m), df_useNbrs(false) {}
603 int v = basecomp(i, j);
609 getAtomNeighborhood(dp_atoms[i].bonds);
610 getAtomNeighborhood(dp_atoms[j].bonds);
615 for (
unsigned int ii = 0;
616 ii < dp_atoms[i].bonds.size() && ii < dp_atoms[j].bonds.size();
618 int cmp = bondholder::compare(
624 for (
unsigned int ii = 0;
625 ii < dp_atoms[i].bonds.size() && ii < dp_atoms[j].bonds.size();
628 bondholder::compare(dp_atoms[i].bonds[ii], dp_atoms[j].bonds[ii]);
633 if (dp_atoms[i].bonds.size() < dp_atoms[j].bonds.size()) {
635 }
else if (dp_atoms[i].bonds.size() > dp_atoms[j].bonds.size()) {
648template <
typename CompareFunc>
681 start = order + offset;
699 for (
int k = 0;
k <
len; ++
k) {
706 for (i =
count[index]; i <
len; i++) {
716 for (
unsigned j = 0;
j < atoms[index].
degree; ++
j) {
724 for (i =
count[index]; i <
len; i++) {
726 for (
unsigned j = 0;
j < atoms[index].
degree; ++
j) {
745template <
typename CompareFunc>
756 for (
unsigned int i = 0; i <
nAtoms; i++) {
762 index = order[offset];
763 atoms[index].
index = offset;
768 if (atoms[index].degree < 1) {
771 for (
unsigned j = 0;
j < atoms[index].
degree; ++
j) {
798 int *order,
int *
count,
806 std::vector<unsigned int> &
res,
807 bool breakTies =
true,
813 const ROMol &mol, std::vector<unsigned int> &
res,
816 const std::vector<std::string> *atomSymbols,
817 const std::vector<std::string> *
bondSymbols,
bool breakTies,
821 const ROMol &mol, std::vector<unsigned int> &
res,
824 const std::vector<std::string> *atomSymbols =
nullptr,
833 std::vector<unsigned int> &
res);
836 std::vector<Canon::canon_atom> &atoms,
841 std::vector<Canon::canon_atom> &atoms,
843 const std::vector<std::string> *atomSymbols,
850 bool useSpecial =
false,
bool useChirality =
false,
851 const boost::dynamic_bitset<> *
atomsInPlay =
nullptr,
852 const boost::dynamic_bitset<> *
bondsInPlay =
nullptr);
#define PRECONDITION(expr, mess)
Defines the primary molecule class ROMol as well as associated typedefs.
Defines the class StereoGroup which stores relationships between the absolute configurations of atoms...
The class for representing atoms.
int getAtomicNum() const
returns our atomic number
@ CHI_TETRAHEDRAL_CW
tetrahedral: clockwise rotation (SMILES @@)
@ CHI_TETRAHEDRAL_CCW
tetrahedral: counter-clockwise rotation (SMILES
BondStereo
the nature of the bond's stereochem (for cis/trans)
AtomCompareFunctor(Canon::canon_atom *atoms, const ROMol &m, const boost::dynamic_bitset<> *atomsInPlay=nullptr, const boost::dynamic_bitset<> *bondsInPlay=nullptr)
int operator()(int i, int j) const
ChiralAtomCompareFunctor(Canon::canon_atom *atoms, const ROMol &m)
int operator()(int i, int j) const
ChiralAtomCompareFunctor()
int operator()(int i, int j) const
SpecialChiralityAtomCompareFunctor(Canon::canon_atom *atoms, const ROMol &m, const boost::dynamic_bitset<> *atomsInPlay=nullptr, const boost::dynamic_bitset<> *bondsInPlay=nullptr)
SpecialChiralityAtomCompareFunctor()
SpecialSymmetryAtomCompareFunctor()
SpecialSymmetryAtomCompareFunctor(Canon::canon_atom *atoms, const ROMol &m, const boost::dynamic_bitset<> *atomsInPlay=nullptr, const boost::dynamic_bitset<> *bondsInPlay=nullptr)
int operator()(int i, int j) const
const std::vector< StereoGroup > & getStereoGroups() const
Gets a reference to the groups of atoms with relative stereochemistry.
unsigned int getNumAtoms() const
returns our number of atoms
#define RDKIT_GRAPHMOL_EXPORT
void rankWithFunctor(T &ftor, bool breakTies, int *order, bool useSpecial=false, bool useChirality=false, const boost::dynamic_bitset<> *atomsInPlay=nullptr, const boost::dynamic_bitset<> *bondsInPlay=nullptr)
void initFragmentCanonAtoms(const ROMol &mol, std::vector< Canon::canon_atom > &atoms, bool includeChirality, const std::vector< std::string > *atomSymbols, const std::vector< std::string > *bondSymbols, const boost::dynamic_bitset<> &atomsInPlay, const boost::dynamic_bitset<> &bondsInPlay, bool needsInit)
RDKIT_GRAPHMOL_EXPORT void rankFragmentAtoms(const ROMol &mol, std::vector< unsigned int > &res, const boost::dynamic_bitset<> &atomsInPlay, const boost::dynamic_bitset<> &bondsInPlay, const std::vector< std::string > *atomSymbols, const std::vector< std::string > *bondSymbols, bool breakTies, bool includeChirality, bool includeIsotope, bool includeAtomMaps)
RDKIT_GRAPHMOL_EXPORT void CreateSinglePartition(unsigned int nAtoms, int *order, int *count, canon_atom *atoms)
RDKIT_GRAPHMOL_EXPORT void initCanonAtoms(const ROMol &mol, std::vector< Canon::canon_atom > &atoms, bool includeChirality=true)
RDKIT_GRAPHMOL_EXPORT void ActivatePartitions(unsigned int nAtoms, int *order, int *count, int &activeset, int *next, int *changed)
const unsigned int ATNUM_CLASS_OFFSET
RDKIT_GRAPHMOL_EXPORT void updateAtomNeighborNumSwaps(canon_atom *atoms, std::vector< bondholder > &nbrs, unsigned int atomIdx, std::vector< std::pair< unsigned int, unsigned int > > &result)
void BreakTies(const ROMol &mol, canon_atom *atoms, CompareFunc compar, int mode, int *order, int *count, int &activeset, int *next, int *changed, char *touchedPartitions)
void RefinePartitions(const ROMol &mol, canon_atom *atoms, CompareFunc compar, int mode, int *order, int *count, int &activeset, int *next, int *changed, char *touchedPartitions)
RDKIT_GRAPHMOL_EXPORT void rankMolAtoms(const ROMol &mol, std::vector< unsigned int > &res, bool breakTies=true, bool includeChirality=true, bool includeIsotopes=true, bool includeAtomMaps=true)
RDKIT_GRAPHMOL_EXPORT void chiralRankMolAtoms(const ROMol &mol, std::vector< unsigned int > &res)
RDKIT_GRAPHMOL_EXPORT void updateAtomNeighborIndex(canon_atom *atoms, std::vector< bondholder > &nbrs)
bool rdvalue_is(const RDValue_cast_t)
void hanoisort(int *base, int nel, int *count, int *changed, CompareFunc compar)
unsigned int countSwapsToInterconvert(const T &ref, T probe)
const std::string * p_symbol
static bool greater(const bondholder &lhs, const bondholder &rhs)
bool operator<(const bondholder &o) const
int compareStereo(const bondholder &o) const
bondholder(Bond::BondType bt, unsigned int bs, unsigned int ni, unsigned int nsc, unsigned int bidx)
bondholder(Bond::BondType bt, Bond::BondStereo bs, unsigned int ni, unsigned int nsc, unsigned int bidx)
static int compare(const bondholder &x, const bondholder &y, unsigned int div=1)
std::vector< bondholder > bonds
std::unique_ptr< int[]> nbrIds
std::vector< int > revistedNeighbors
std::vector< int > neighborNum