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
|
//
// Copyright (C) 2018 Boran Adas, Google Summer of Code
//
// @@ 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 <RDGeneral/export.h>
#ifndef RD_TOPOLOGICALTORSIONGEN_H_2018_07
#define RD_TOPOLOGICALTORSIONGEN_H_2018_07
#include <GraphMol/Fingerprints/FingerprintGenerator.h>
#include <GraphMol/Fingerprints/FingerprintUtil.h>
namespace RDKit {
namespace TopologicalTorsion {
template <typename OutputType>
class RDKIT_FINGERPRINTS_EXPORT TopologicalTorsionArguments
: public FingerprintArguments<OutputType> {
public:
const bool df_includeChirality;
const uint32_t d_torsionAtomCount;
OutputType getResultSize() const override;
std::string infoString() const override;
/**
\brief Construct a new Topological Torsion Arguments object
\param includeChirality if set, chirality will be used in sparse result
\param torsionAtomCount the number of atoms to include in the "torsions"
\param useCountSimulation if set, use count simulation while
generating the fingerprint
\param countBounds boundaries for count simulation, corresponding bit will
be set if the count is higher than the number provided for that spot
\param fpSize size of the generated fingerprint, does not affect the sparse
versions
*/
TopologicalTorsionArguments(const bool includeChirality,
const uint32_t torsionAtomCount,
const bool countSimulation,
const std::vector<std::uint32_t> countBounds,
const std::uint32_t fpSize);
};
template <typename OutputType>
class RDKIT_FINGERPRINTS_EXPORT TopologicalTorsionAtomEnv
: public AtomEnvironment<OutputType> {
const OutputType d_bitId;
const INT_VECT d_atomPath;
public:
OutputType getBitId(FingerprintArguments<OutputType> *arguments,
const std::vector<std::uint32_t> *atomInvariants,
const std::vector<std::uint32_t> *bondInvariants,
const AdditionalOutput *additionalOutput,
const bool hashResults = false,
const std::uint64_t fpSize = 0) const override;
/**
\brief Construct a new Topological Torsion Atom Env object
\param bitId bitId generated for this environment
*/
TopologicalTorsionAtomEnv(OutputType bitId, INT_VECT atomPath)
: d_bitId(bitId), d_atomPath(std::move(atomPath)) {}
};
template <typename OutputType>
class RDKIT_FINGERPRINTS_EXPORT TopologicalTorsionEnvGenerator
: public AtomEnvironmentGenerator<OutputType> {
public:
std::vector<AtomEnvironment<OutputType> *> getEnvironments(
const ROMol &mol, FingerprintArguments<OutputType> *arguments,
const std::vector<std::uint32_t> *fromAtoms,
const std::vector<std::uint32_t> *ignoreAtoms, const int confId,
const AdditionalOutput *additionalOutput,
const std::vector<std::uint32_t> *atomInvariants,
const std::vector<std::uint32_t> *bondInvariants,
const bool hashResults = false) const override;
std::string infoString() const override;
};
/**
\brief Get the Topological Torsion Generator object
\tparam OutputType determines the size of the bitIds and the result, can only
be 64 bit unsigned integer for this type
\param includeChirality includeChirality argument for both the default atom
invariants generator and the fingerprint arguments
\param torsionAtomCount the number of atoms to include in the "torsions"
\param atomInvariantsGenerator custom atom invariants generator to use
\param useCountSimulation if set, use count simulation while
generating the fingerprint
\param countBounds boundaries for count simulation, corresponding bit will
be set if the count is higher than the number provided for that spot
\param fpSize size of the generated fingerprint, does not affect the sparse
versions
\param ownsAtomInvGen if set atom invariants generator is destroyed with the
fingerprint generator
/return FingerprintGenerator<OutputType>* that generates topological-torsion
fingerprints
This generator supports the following \c AdditionalOutput types:
- \c atomToBits : which bits each atom is involved in
- \c atomCounts : how many bits each atom sets
- \c bitPaths : map from bitId to vectors of atom indices
*/
template <typename OutputType>
RDKIT_FINGERPRINTS_EXPORT FingerprintGenerator<OutputType> *
getTopologicalTorsionGenerator(
const bool includeChirality = false, const uint32_t torsionAtomCount = 4,
AtomInvariantsGenerator *atomInvariantsGenerator = nullptr,
const bool countSimulation = true,
const std::vector<std::uint32_t> countBounds = {1, 2, 4, 8},
const std::uint32_t fpSize = 2048, const bool ownsAtomInvGen = false);
} // namespace TopologicalTorsion
} // namespace RDKit
#endif
|
