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
|
//
// Copyright (C) 2020-2022 Greg Landrum and other RDKit contributors
//
// @@ 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 <RDBoost/python.h>
#include <GraphMol/RDKitBase.h>
#include <GraphMol/MolHash/MolHash.h>
#include <RDBoost/Wrap.h>
namespace python = boost::python;
using namespace RDKit;
namespace {
std::string MolHashHelper(const ROMol &mol, MolHash::HashFunction func,
bool useCXSmiles) {
RWMol cpy(mol);
return MolHash::MolHash(&cpy, func, useCXSmiles);
}
} // namespace
BOOST_PYTHON_MODULE(rdMolHash) {
python::scope().attr("__doc__") =
"Module containing functions to generate hashes for molecules";
python::enum_<MolHash::HashFunction>("HashFunction")
.value("AnonymousGraph", MolHash::HashFunction::AnonymousGraph)
.value("ElementGraph", MolHash::HashFunction::ElementGraph)
.value("CanonicalSmiles", MolHash::HashFunction::CanonicalSmiles)
.value("MurckoScaffold", MolHash::HashFunction::MurckoScaffold)
.value("ExtendedMurcko", MolHash::HashFunction::ExtendedMurcko)
.value("MolFormula", MolHash::HashFunction::MolFormula)
.value("AtomBondCounts", MolHash::HashFunction::AtomBondCounts)
.value("DegreeVector", MolHash::HashFunction::DegreeVector)
.value("Mesomer", MolHash::HashFunction::Mesomer)
.value("HetAtomTautomer", MolHash::HashFunction::HetAtomTautomer)
.value("HetAtomProtomer", MolHash::HashFunction::HetAtomProtomer)
.value("RedoxPair", MolHash::HashFunction::RedoxPair)
.value("Regioisomer", MolHash::HashFunction::Regioisomer)
.value("NetCharge", MolHash::HashFunction::NetCharge)
.value("SmallWorldIndexBR", MolHash::HashFunction::SmallWorldIndexBR)
.value("SmallWorldIndexBRL", MolHash::HashFunction::SmallWorldIndexBRL)
.value("ArthorSubstructureOrder",
MolHash::HashFunction::ArthorSubstructureOrder);
python::def("MolHash", MolHashHelper,
(python::arg("mol"), python::arg("func"),
python::arg("useCxSmiles") = false),
"Generate a hash for a molecule. The func argument determines "
"which hash is generated.");
}
|
