Files
rdkit/Code/GraphMol/PeriodicTable.cpp
Yakov Pechersky c6cabf4153 Speed-up tautomer canonicalization, no API changes (#9134)
* Speed up tautomer canonicalization by deferring on SSSR calc

* Lazy kekulization for tautomer enumeration

Defer kekulization of tautomers until they are actually needed for
transform matching. This avoids creating kekulized copies for:
1. The initial tautomer (until first iteration)
2. New tautomers that may never be processed (if enumeration ends early)

The Tautomer class now supports lazy initialization of the kekulized
form via getKekulized() method.

Performance improvement: ~7% additional speedup (total ~22-24% from baseline)

* Use count-only substructure matching in tautomer scoring

* Add SubstructMatchCount regression test

* MolStandardize: reduce enumerate overhead

* MolStandardize: avoid per-tautomer ring recomputation

* Atom: cache PeriodicTable pointer in valence calcs

* Atom: reuse PeriodicTable in getEffectiveAtomicNum

* PeriodicTable: add atomic fast path for getTable

* GraphMol: reduce ROMol copy reallocations

* MolStandardize: use quickCopy for per-match product copies

Use RWMol(*kmol, true) in tautomer enumeration to avoid copying properties/bookmarks/conformers for each candidate. This reduces deep-copy overhead without changing chemistry.

* MolStandardize: pre-filter scoring patterns by element/connectivity

For tautomer scoring, pre-compute which SubstructTerms are relevant for
a given input molecule. Since tautomerization only moves H atoms and
changes bond orders (never creates/destroys heavy-atom bonds), patterns
requiring missing elements or connectivity can be skipped for all
tautomers of that molecule.

Two-stage filtering:
1. Element check: skip patterns requiring atoms not in the molecule
2. Connectivity check: skip patterns whose bond-order-agnostic structure
   doesn't match the input molecule's connectivity

This reduces the number of VF2 substructure calls per tautomer from 12
to typically 3-5, depending on the molecule's composition.

* MolStandardize: preserve molecule properties for canonical tautomer

Copy molecule properties from the original input to the canonical tautomer
result. Since quickCopy during enumeration skips d_props to avoid overhead,
extended SMILES data like link nodes (LN) was lost. This restores them
on the final result.

* TautomerQuery: preserve molecule properties (e.g. link nodes) in tautomers

TautomerQuery::fromMol() uses TautomerEnumerator::enumerate() which uses
quickCopy for performance. This doesn't copy molecule properties like
_molLinkNodes. Without this fix, XQMol output would lose link node
extensions in the SMILES.

Copy properties from the original query molecule to all enumerated
tautomers before constructing the TautomerQuery. This preserves extended
SMILES data without impacting enumeration performance.

* MolStandardize: use parallel iteration and cache bond lookups

Replace O(n) getAtomWithIdx/getBondWithIdx calls with parallel iteration
over atom/bond ranges in canonicalizeInPlace and enumerate. Cache bond
lookups in setTautomerStereoAndIsoHs to avoid repeated O(n) searches.

* perf: add specialized matchers for simple tautomer scoring patterns

Replace VF2 graph matching with O(n) loops for 6 simple patterns:
- countDoubleOrAromaticBonds: C=O, N=O, P=O patterns
- countMethyls: [CX4H3] methyl groups
- countCarbonDoubleHetero: [C]=[/home/dcvuser/rdkit;Code/GraphMol/MolStandardize/Tautomer.h] aliphatic C=hetero
- countAromaticCarbonExocyclicN: [c]=aromatic C=exocyclic N
Complex patterns (benzoquinone, oxim, guanidine, aci-nitro) still use VF2.
Combined with the pre-filtering optimization, this achieves ~3.7x speedup
(~2500ms vs ~9300ms original) for tautomer canonicalization.

* Fix tautomer canonicalize dropping conformers from quickCopy

quickCopy (RWMol(*mol, true)) skips conformers, so tautomer
enumeration products lose 2D/3D coordinates. This causes InChI
generation to omit the /b (double bond E/Z stereo) layer, since
E/Z is derived from atomic coordinates.

Fix: copy conformers from the original molecule onto the canonical
tautomer after pickCanonical in TautomerEnumerator::canonicalize().

Tests: SMILES-based E/Z check in testTautomer.cpp, molblock-based
conformer preservation check in catch_tests.cpp.

* add test on canonicalize losing stereo

* add regression test for exocyclic C=C tautomer canonicalization

The getTautomerStateKey() pre-filter (commit 2595ef748) can falsely
deduplicate distinct tautomers when their atom-index-ordered state
patterns happen to match, leading canonicalize() to pick the wrong
canonical form for molecules with STEREOTRANS-pinned exocyclic C=C
bonds after RemoveHs.

Test verifies that O=C(CC1=CC2=CC=COC2)NC1=O canonicalizes to the
exocyclic form O=C1CC(=CC2=CC=COC2)C(=O)N1, not the endocyclic form
O=C1C=C(C=C2CC=COC2)C(=O)N1.

Currently expected to FAIL until the state key dedup bug is fixed.

* MolStandardize: expand tautomer connectivity SMARTS

* MolStandardize: scope tautomer pattern enum

* MolStandardize: trim tautomer pattern enum

* MolStandardize: use symmetric ring scoring
2026-03-31 06:42:40 +02:00

129 lines
3.6 KiB
C++

// $Id$
//
// Copyright (C) 2001-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 "PeriodicTable.h"
#include <string>
#include <atomic>
#include <boost/tokenizer.hpp>
typedef boost::tokenizer<boost::char_separator<char>> tokenizer;
#include <sstream>
#include <locale>
#ifdef RDK_BUILD_THREADSAFE_SSS
#include <mutex>
#endif
namespace RDKit {
namespace {
std::atomic<PeriodicTable *> ds_rawInstance{nullptr};
} // namespace
class std::unique_ptr<PeriodicTable> PeriodicTable::ds_instance = nullptr;
PeriodicTable::PeriodicTable() {
// it is assumed that the atomic atomData string constains atoms
// in sequence and no atoms are missing in between
byanum.clear();
byname.clear();
boost::char_separator<char> eolSep("\n");
tokenizer tokens(periodicTableAtomData, eolSep);
for (tokenizer::iterator token = tokens.begin(); token != tokens.end();
++token) {
if (*token != " ") {
atomicData adata(*token);
std::string enam = adata.Symbol();
byname[enam] = adata.AtomicNum();
// there are, for backwards compatibility reasons, some duplicate rows for
// atomic numbers in the atomic_data data structure. It's ok to have
// multiple symbols map to the same atomic number (above), but we need to
// be sure that we only store one entry per atomic number.
// Note that this only works because the first atom in the adata list is
// the dummy atom (atomic number 0). This was #2784
if (rdcast<size_t>(adata.AtomicNum()) == byanum.size()) {
byanum.push_back(adata);
}
}
}
unsigned int lidx = 0;
std::istringstream istr;
istr.imbue(std::locale("C"));
while (isotopesAtomData[lidx] != "" && isotopesAtomData[lidx] != "EOS") {
tokenizer lines(isotopesAtomData[lidx++], eolSep);
boost::char_separator<char> spaceSep(" \t");
for (tokenizer::iterator line = lines.begin(); line != lines.end();
++line) {
if (*line != " ") {
tokenizer tokens(*line, spaceSep);
tokenizer::iterator token = tokens.begin();
int anum;
istr.clear();
istr.str(*token);
istr >> anum;
atomicData &adata = byanum[anum];
++token;
if (token == tokens.end()) {
continue;
}
++token;
if (token == tokens.end()) {
continue;
}
unsigned int isotope;
istr.clear();
istr.str(*token);
istr >> isotope;
++token;
if (token == tokens.end()) {
continue;
}
double mass;
istr.clear();
istr.str(*token);
istr >> mass;
++token;
if (token == tokens.end()) {
continue;
}
double abundance;
istr.clear();
istr.str(*token);
istr >> abundance;
adata.d_isotopeInfoMap[isotope] = std::make_pair(mass, abundance);
}
}
}
}
void PeriodicTable::initInstance() {
ds_instance = std::unique_ptr<PeriodicTable>(new PeriodicTable());
ds_rawInstance.store(ds_instance.get(), std::memory_order_release);
}
PeriodicTable *PeriodicTable::getTable() {
#ifdef RDK_BUILD_THREADSAFE_SSS
auto *raw = ds_rawInstance.load(std::memory_order_acquire);
if (raw) {
return raw;
}
static std::once_flag pt_init_once;
std::call_once(pt_init_once, initInstance);
#else
if (!ds_instance) {
initInstance();
}
#endif
return ds_rawInstance.load(std::memory_order_acquire);
}
} // namespace RDKit