pub_soft/rdkit
3
0
Fork 0
mirror of https://github.com/rdkit/rdkit.git synced 2026-06-04 21:54:27 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
d304f9f4165d360df8d923caa94fe4f13f03f737
rdkit/Code/GraphMol/Wrap/Bond.cpp
Brian Kelley d304f9f416 Trim spaces from RDProp strings to simulate reading from SDFiles (#8760) 
* Trim spaces from RDProp strings to simulate reading from SDFiles

* Update documentation

* Use the correct doc strings

---------

Co-authored-by: Brian Kelley <bkelley@glysade.com>
2025-09-26 06:14:23 +02:00

364 lines
16 KiB
C++
Raw Blame History

//
// Copyright (C) 2003-2017 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.
//
#define NO_IMPORT_ARRAY
#include <RDBoost/python.h>
#include <string>
#include "props.hpp"
#include <GraphMol/RDKitBase.h>
#include <GraphMol/QueryBond.h>
#include <GraphMol/SmilesParse/SmilesWrite.h>
#include <GraphMol/SmilesParse/SmartsWrite.h>
#include <RDGeneral/types.h>
namespace python = boost::python;
namespace RDKit {
void expandQuery(QueryBond *self, const QueryBond *other,
Queries::CompositeQueryType how, bool maintainOrder) {
if (other->hasQuery()) {
const QueryBond::QUERYBOND_QUERY *qry = other->getQuery();
self->expandQuery(qry->copy(), how, maintainOrder);
}
}
void setQuery(QueryBond *self, const QueryBond *other) {
if (other->hasQuery()) {
self->setQuery(other->getQuery()->copy());
}
}
int BondHasProp(const Bond *bond, const char *key) {
int res = bond->hasProp(key);
return res;
}
template <class T>
void BondSetProp(const Bond *bond, const char *key, const T &val) {
bond->setProp<T>(key, val);
}
void BondClearProp(const Bond *bond, const char *key) {
if (!bond->hasProp(key)) {
return;
}
bond->clearProp(key);
}
bool BondIsInRing(const Bond *bond) {
if (!bond->getOwningMol().getRingInfo()->isSssrOrBetter()) {
MolOps::findSSSR(bond->getOwningMol());
}
return bond->getOwningMol().getRingInfo()->numBondRings(bond->getIdx()) != 0;
}
bool BondIsInRingSize(const Bond *bond, int size) {
if (!bond->getOwningMol().getRingInfo()->isSssrOrBetter()) {
MolOps::findSSSR(bond->getOwningMol());
}
return bond->getOwningMol().getRingInfo()->isBondInRingOfSize(bond->getIdx(),
size);
return false;
}
INT_VECT getBondStereoAtoms(const Bond *bond) { return bond->getStereoAtoms(); }
std::string BondGetSmarts(const Bond *bond, bool allBondsExplicit) {
std::string res;
if (bond->hasQuery()) {
res = SmartsWrite::GetBondSmarts(static_cast<const QueryBond *>(bond));
} else {
res = SmilesWrite::GetBondSmiles(bond, -1, false, allBondsExplicit);
}
return res;
}
std::string bondClassDoc =
"The class to store Bonds.\n\
Note: unlike Atoms, is it currently impossible to construct Bonds from\n\
Python.\n";
struct bond_wrapper {
static void wrap() {
python::class_<Bond>("Bond", bondClassDoc.c_str(), python::no_init)
.def("HasOwningMol", &Bond::hasOwningMol, python::args("self"),
"Returns whether or not this instance belongs to a molecule.\n")
.def("GetOwningMol", &Bond::getOwningMol,
"Returns the Mol that owns this bond.\n",
python::return_value_policy<python::reference_existing_object>(),
python::args("self"))
.def("GetBondType", &Bond::getBondType, python::args("self"),
"Returns the type of the bond as a BondType\n")
.def("SetBondType", &Bond::setBondType, python::args("self", "bT"),
"Set the type of the bond as a BondType\n")
.def("GetBondTypeAsDouble", &Bond::getBondTypeAsDouble,
python::args("self"),
"Returns the type of the bond as a double (i.e. 1.0 for SINGLE, "
"1.5 for AROMATIC, 2.0 for DOUBLE)\n")
.def("GetBondDir", &Bond::getBondDir, python::args("self"),
"Returns the type of the bond as a BondDir\n")
.def("SetBondDir", &Bond::setBondDir, python::args("self", "what"),
"Set the type of the bond as a BondDir\n")
.def("GetStereo", &Bond::getStereo, python::args("self"),
"Returns the stereo configuration of the bond as a BondStereo\n")
.def("SetStereo", &Bond::setStereo, python::args("self", "what"),
"Set the stereo configuration of the bond as a BondStereo\n")
.def("GetStereoAtoms", getBondStereoAtoms, python::args("self"),
"Returns the indices of the atoms setting this bond's "
"stereochemistry.\n")
.def("SetStereoAtoms", &Bond::setStereoAtoms,
python::args("self", "bgnIdx", "endIdx"),
"Set the indices of the atoms setting this bond's "
"stereochemistry.\n")
.def("GetValenceContrib",
(double(Bond::*)(const Atom *) const) & Bond::getValenceContrib,
python::args("self", "at"),
"Returns the contribution of the bond to the valence of an "
"Atom.\n\n"
" ARGUMENTS:\n\n"
" - atom: the Atom to consider.\n")
.def("GetIsAromatic", &Bond::getIsAromatic, python::args("self"))
.def("SetIsAromatic", &Bond::setIsAromatic,
python::args("self", "what"))
.def("GetIsConjugated", &Bond::getIsConjugated, python::args("self"),
"Returns whether or not the bond is considered to be conjugated.")
.def("SetIsConjugated", &Bond::setIsConjugated,
python::args("self", "what"))
.def("GetIdx", &Bond::getIdx, python::args("self"),
"Returns the bond's index (ordering in the molecule)\n")
.def("GetBeginAtomIdx", &Bond::getBeginAtomIdx, python::args("self"),
"Returns the index of the bond's first atom.\n")
.def("GetEndAtomIdx", &Bond::getEndAtomIdx, python::args("self"),
"Returns the index of the bond's first atom.\n")
.def("GetOtherAtomIdx", &Bond::getOtherAtomIdx,
python::args("self", "thisIdx"),
"Given the index of one of the bond's atoms, returns the\n"
"index of the other.\n")
.def("GetBeginAtom", &Bond::getBeginAtom,
python::return_value_policy<python::reference_existing_object>(),
python::args("self"), "Returns the bond's first atom.\n")
.def("GetEndAtom", &Bond::getEndAtom,
python::return_value_policy<python::reference_existing_object>(),
python::args("self"), "Returns the bond's second atom.\n")
.def("GetOtherAtom", &Bond::getOtherAtom,
python::return_value_policy<python::reference_existing_object>(),
python::args("self", "what"),
"Given one of the bond's atoms, returns the other one.\n")
// FIX: query stuff
.def("Match", (bool(Bond::*)(const Bond *) const) & Bond::Match,
python::args("self", "what"),
"Returns whether or not this bond matches another Bond.\n\n"
" Each Bond (or query Bond) has a query function which is\n"
" used for this type of matching.\n\n"
" ARGUMENTS:\n"
" - other: the other Bond to which to compare\n")
.def("IsInRingSize", BondIsInRingSize, python::args("self", "size"),
"Returns whether or not the bond is in a ring of a particular "
"size.\n\n"
" ARGUMENTS:\n"
" - size: the ring size to look for\n")
.def("IsInRing", BondIsInRing, python::args("self"),
"Returns whether or not the bond is in a ring of any size.\n\n")
.def("HasQuery", &Bond::hasQuery, python::args("self"),
"Returns whether or not the bond has an associated query\n\n")
.def("DescribeQuery", describeQuery, python::args("self"),
"returns a text description of the query. Primarily intended for "
"debugging purposes.\n\n")
.def("GetSmarts", BondGetSmarts,
(python::arg("bond"), python::arg("allBondsExplicit") = false),
"returns the SMARTS (or SMILES) string for a Bond")
// properties
.def("SetProp", BondSetProp<std::string>,
(python::arg("self"), python::arg("key"), python::arg("val")),
"Sets a bond property\n\n"
" ARGUMENTS:\n"
" - key: the name of the property to be set (a string).\n"
" - value: the property value (a string).\n\n")
.def(
"GetProp", GetPyProp<Bond>,
(python::arg("self"), python::arg("key"),
python::arg("autoConvert") = false),
"Returns the value of the property.\n\n"
" ARGUMENTS:\n"
" - key: the name of the property to return (a string).\n\n"
" - autoConvert: if True attempt to convert the property into a python object\n\n"
" RETURNS: a string\n\n"
" NOTE:\n"
" - If the property has not been set, a KeyError exception "
"will be raised.\n",
boost::python::return_value_policy<return_pyobject_passthrough>())
.def("SetIntProp", BondSetProp<int>,
(python::arg("self"), python::arg("key"), python::arg("val")),
"Sets a bond property\n\n"
" ARGUMENTS:\n"
" - key: the name of the property to be set (a string).\n"
" - value: the property value (an int).\n\n")
.def("SetUnsignedProp", BondSetProp<unsigned int>,
(python::arg("self"), python::arg("key"), python::arg("val")),
"Sets a bond property\n\n"
" ARGUMENTS:\n"
" - key: the name of the property to be set (a string).\n"
" - value: the property value (an int >= 0).\n\n")
.def("GetIntProp", GetProp<Bond, int>, python::args("self", "key"),
"Returns the value of the property.\n\n"
" ARGUMENTS:\n"
" - key: the name of the property to return (an int).\n\n"
" RETURNS: an int\n\n"
" NOTE:\n"
" - If the property has not been set, a KeyError exception "
"will be raised.\n",
boost::python::return_value_policy<return_pyobject_passthrough>())
.def("GetUnsignedProp", GetProp<Bond, unsigned int>,
python::args("self", "key"),
"Returns the value of the property.\n\n"
" ARGUMENTS:\n"
" - key: the name of the property to return (an unsigned "
"integer).\n\n"
" RETURNS: an int (Python has no unsigned type)\n\n"
" NOTE:\n"
" - If the property has not been set, a KeyError exception "
"will be raised.\n",
boost::python::return_value_policy<return_pyobject_passthrough>())
.def("SetDoubleProp", BondSetProp<double>,
(python::arg("self"), python::arg("key"), python::arg("val")),
"Sets a bond property\n\n"
" ARGUMENTS:\n"
" - key: the name of the property to be set (a string).\n"
" - value: the property value (a double).\n\n")
.def("GetDoubleProp", GetProp<Bond, double>,
python::args("self", "key"),
"Returns the value of the property.\n\n"
" ARGUMENTS:\n"
" - key: the name of the property to return (a double).\n\n"
" RETURNS: a double\n\n"
" NOTE:\n"
" - If the property has not been set, a KeyError exception "
"will be raised.\n",
boost::python::return_value_policy<return_pyobject_passthrough>())
.def("SetBoolProp", BondSetProp<bool>,
(python::arg("self"), python::arg("key"), python::arg("val")),
"Sets a bond property\n\n"
" ARGUMENTS:\n"
" - key: the name of the property to be set (a string).\n"
" - value: the property value (a boolean).\n\n")
.def("GetBoolProp", GetProp<Bond, bool>, python::args("self", "key"),
"Returns the value of the property.\n\n"
" ARGUMENTS:\n"
" - key: the name of the property to return (a boolean).\n\n"
" RETURNS: a boolean\n\n"
" NOTE:\n"
" - If the property has not been set, a KeyError exception "
"will be raised.\n",
boost::python::return_value_policy<return_pyobject_passthrough>())
.def("HasProp", BondHasProp, python::args("self", "key"),
"Queries a Bond to see if a particular property has been "
"assigned.\n\n"
" ARGUMENTS:\n"
" - key: the name of the property to check for (a string).\n")
.def("ClearProp", BondClearProp, python::args("self", "key"),
"Removes a particular property from an Bond (does nothing if not "
"already set).\n\n"
" ARGUMENTS:\n"
" - key: the name of the property to be removed.\n")
.def("GetPropNames", &Bond::getPropList,
(python::arg("self"), python::arg("includePrivate") = false,
python::arg("includeComputed") = false),
"Returns a list of the properties set on the Bond.\n\n")
.def("GetPropsAsDict", GetPropsAsDict<Bond>,
(python::arg("self"), python::arg("includePrivate") = true,
python::arg("includeComputed") = true,
python::arg("autoConvertStrings") = true),
getPropsAsDictDocString.c_str())
;
python::enum_<Bond::BondType>("BondType")
.value("UNSPECIFIED", Bond::UNSPECIFIED)
.value("SINGLE", Bond::SINGLE)
.value("DOUBLE", Bond::DOUBLE)
.value("TRIPLE", Bond::TRIPLE)
.value("QUADRUPLE", Bond::QUADRUPLE)
.value("QUINTUPLE", Bond::QUINTUPLE)
.value("HEXTUPLE", Bond::HEXTUPLE)
.value("ONEANDAHALF", Bond::ONEANDAHALF)
.value("TWOANDAHALF", Bond::TWOANDAHALF)
.value("THREEANDAHALF", Bond::THREEANDAHALF)
.value("FOURANDAHALF", Bond::FOURANDAHALF)
.value("FIVEANDAHALF", Bond::FIVEANDAHALF)
.value("AROMATIC", Bond::AROMATIC)
.value("IONIC", Bond::IONIC)
.value("HYDROGEN", Bond::HYDROGEN)
.value("THREECENTER", Bond::THREECENTER)
.value("DATIVEONE", Bond::DATIVEONE)
.value("DATIVE", Bond::DATIVE)
.value("DATIVEL", Bond::DATIVEL)
.value("DATIVER", Bond::DATIVER)
.value("OTHER", Bond::OTHER)
.value("ZERO", Bond::ZERO);
python::enum_<Bond::BondDir>("BondDir")
.value("NONE", Bond::NONE)
.value("BEGINWEDGE", Bond::BEGINWEDGE)
.value("BEGINDASH", Bond::BEGINDASH)
.value("ENDDOWNRIGHT", Bond::ENDDOWNRIGHT)
.value("ENDUPRIGHT", Bond::ENDUPRIGHT)
.value("EITHERDOUBLE", Bond::EITHERDOUBLE)
.value("UNKNOWN", Bond::UNKNOWN);
python::enum_<Bond::BondStereo>("BondStereo")
.value("STEREONONE", Bond::STEREONONE)
.value("STEREOANY", Bond::STEREOANY)
.value("STEREOZ", Bond::STEREOZ)
.value("STEREOE", Bond::STEREOE)
.value("STEREOCIS", Bond::STEREOCIS)
.value("STEREOTRANS", Bond::STEREOTRANS)
.value("STEREOATROPCW", Bond::STEREOATROPCW)
.value("STEREOATROPCCW", Bond::STEREOATROPCCW);
bondClassDoc =
"The class to store QueryBonds.\n\
These cannot currently be constructed directly from Python\n";
python::class_<QueryBond, python::bases<Bond>>(
"QueryBond", bondClassDoc.c_str(), python::no_init)
.def("ExpandQuery", expandQuery,
(python::arg("self"), python::arg("other"),
python::arg("how") = Queries::COMPOSITE_AND,
python::arg("maintainOrder") = true),
"combines the query from other with ours")
.def("SetQuery", setQuery, (python::arg("self"), python::arg("other")),
"Replace our query with a copy of the other query");
};
};
} // namespace RDKit
void wrap_bond() { RDKit::bond_wrapper::wrap(); }
Reference in New Issue View Git Blame Copy Permalink