Files
rdkit/Code/GraphMol/QueryOps.h
Greg Landrum 270f7b76e5 Partial support for reading CXSMILES (#1237)
* add a SmilesParserParams object to prepare for this

* add a SmilesParserParams object to prepare for this

* add tests for the SmilesParseParmas

* support name parsing, should it be the default?

* rename CXNSmiles to CXSmiles;
add a spirit parser for CXSmiles coordinate that is at least syntax correct

* abandon boost::spirit for now; crude atom token parser

* support params in smiles parser (not tested, may not build)

* can read coords and atom labels along with mol names; crude, but works

* read coordinate bonds

* remove some compiler warnings with VS2015

* remove a bunch of compiler warnings on windows

* remove more warnings on windows

* remove more warnings on windows

* backup commit: first pass at parsing query features

* radical spec parsing

* handle attachment points using atom mapping

* switch to a named property for atom labels

* fix handling of the "A" atom query

* add functions to construct A and Q queries (needs more work)

* fix a problem created while cleaning up warnings earlier

* add some additional convenience functions for making generic atoms.
Still need M and to recognize these while writing CXSMILES

* add M queries; update some tests

* fix a linux compile problem

* get the cxsmiles stuff working in python; basic testing

* support "M" in CXSMILES
2017-01-31 13:50:36 -05:00

770 lines
25 KiB
C++

//
// Copyright (C) 2003-2017 Greg Landrum and 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.
//
//! \file QueryOps.h
/*!
\brief Includes a bunch of functionality for handling Atom and Bond queries.
*/
#ifndef _RD_QUERY_OPS_H
#define _RD_QUERY_OPS_H
#include <GraphMol/RDKitBase.h>
#include <Query/QueryObjects.h>
#include <Query/Query.h>
#ifdef RDK_THREADSAFE_SSS
#include <RDGeneral/BoostStartInclude.h>
#include <boost/thread/mutex.hpp>
#include <RDGeneral/BoostEndInclude.h>
#endif
namespace RDKit {
typedef Queries::Query<bool, Atom const *, true> ATOM_BOOL_QUERY;
typedef Queries::Query<bool, Bond const *, true> BOND_BOOL_QUERY;
typedef Queries::AndQuery<int, Atom const *, true> ATOM_AND_QUERY;
typedef Queries::AndQuery<int, Bond const *, true> BOND_AND_QUERY;
typedef Queries::OrQuery<int, Atom const *, true> ATOM_OR_QUERY;
typedef Queries::OrQuery<int, Bond const *, true> BOND_OR_QUERY;
typedef Queries::XOrQuery<int, Atom const *, true> ATOM_XOR_QUERY;
typedef Queries::XOrQuery<int, Bond const *, true> BOND_XOR_QUERY;
typedef Queries::EqualityQuery<int, Atom const *, true> ATOM_EQUALS_QUERY;
typedef Queries::EqualityQuery<int, Bond const *, true> BOND_EQUALS_QUERY;
typedef Queries::GreaterQuery<int, Atom const *, true> ATOM_GREATER_QUERY;
typedef Queries::GreaterQuery<int, Bond const *, true> BOND_GREATER_QUERY;
typedef Queries::GreaterEqualQuery<int, Atom const *, true>
ATOM_GREATEREQUAL_QUERY;
typedef Queries::GreaterEqualQuery<int, Bond const *, true>
BOND_GREATEREQUAL_QUERY;
typedef Queries::LessQuery<int, Atom const *, true> ATOM_LESS_QUERY;
typedef Queries::LessQuery<int, Bond const *, true> BOND_LESS_QUERY;
typedef Queries::LessEqualQuery<int, Atom const *, true> ATOM_LESSEQUAL_QUERY;
typedef Queries::LessEqualQuery<int, Bond const *, true> BOND_LESSEQUAL_QUERY;
typedef Queries::RangeQuery<int, Atom const *, true> ATOM_RANGE_QUERY;
typedef Queries::RangeQuery<int, Bond const *, true> BOND_RANGE_QUERY;
typedef Queries::SetQuery<int, Atom const *, true> ATOM_SET_QUERY;
typedef Queries::SetQuery<int, Bond const *, true> BOND_SET_QUERY;
typedef Queries::Query<int, Bond const *, true> BOND_NULL_QUERY;
typedef Queries::Query<int, Atom const *, true> ATOM_NULL_QUERY;
// -------------------------------------------------
// common atom queries
static inline int queryAtomAromatic(Atom const *at) {
return at->getIsAromatic();
};
static inline int queryAtomAliphatic(Atom const *at) {
return !(at->getIsAromatic());
};
static inline int queryAtomExplicitDegree(Atom const *at) {
return at->getDegree();
};
static inline int queryAtomTotalDegree(Atom const *at) {
return at->getTotalDegree();
};
static inline int queryAtomHeavyAtomDegree(Atom const *at) {
return at->getTotalDegree() - at->getTotalNumHs(true);
};
static inline int queryAtomHCount(Atom const *at) {
return at->getTotalNumHs(true);
};
static inline int queryAtomImplicitHCount(Atom const *at) {
return at->getTotalNumHs(false);
};
static inline int queryAtomHasImplicitH(Atom const *at) {
return int(at->getTotalNumHs(false) > 0);
};
static inline int queryAtomImplicitValence(Atom const *at) {
return at->getImplicitValence();
};
static inline int queryAtomExplicitValence(Atom const *at) {
return at->getExplicitValence() - at->getNumExplicitHs();
};
static inline int queryAtomTotalValence(Atom const *at) {
return at->getExplicitValence() + at->getImplicitValence();
};
static inline int queryAtomUnsaturated(Atom const *at) {
return static_cast<int>(at->getDegree()) < at->getExplicitValence();
};
static inline int queryAtomNum(Atom const *at) { return at->getAtomicNum(); };
const int massIntegerConversionFactor = 1000;
static inline int queryAtomMass(Atom const *at) {
return static_cast<int>(round(massIntegerConversionFactor * at->getMass()));
};
static inline int queryAtomIsotope(Atom const *at) {
return static_cast<int>(at->getIsotope());
};
static inline int queryAtomFormalCharge(Atom const *at) {
return static_cast<int>(at->getFormalCharge());
};
static inline int queryAtomHybridization(Atom const *at) {
return at->getHybridization();
};
static inline int queryAtomNumRadicalElectrons(Atom const *at) {
return at->getNumRadicalElectrons();
};
static inline int queryAtomHasChiralTag(Atom const *at) {
return at->getChiralTag() != Atom::CHI_UNSPECIFIED;
};
static inline int queryAtomMissingChiralTag(Atom const *at) {
return at->getChiralTag() == Atom::CHI_UNSPECIFIED &&
at->hasProp(common_properties::_ChiralityPossible);
};
unsigned int queryAtomBondProduct(Atom const *at);
unsigned int queryAtomAllBondProduct(Atom const *at);
// -------------------------------------------------
// common bond queries
static inline int queryBondOrder(Bond const *bond) {
return static_cast<int>(bond->getBondType());
};
static inline int queryBondDir(Bond const *bond) {
return static_cast<int>(bond->getBondDir());
};
static inline int queryIsBondInNRings(Bond const *at) {
return at->getOwningMol().getRingInfo()->numBondRings(at->getIdx());
};
static inline int queryBondHasStereo(Bond const *bnd) {
return bnd->getStereo() > Bond::STEREONONE;
};
// -------------------------------------------------
// ring queries
static inline int queryIsAtomInNRings(Atom const *at) {
return at->getOwningMol().getRingInfo()->numAtomRings(at->getIdx());
};
static inline int queryIsAtomInRing(Atom const *at) {
return at->getOwningMol().getRingInfo()->numAtomRings(at->getIdx()) != 0;
};
static inline int queryAtomHasRingBond(Atom const *at) {
ROMol::OBOND_ITER_PAIR atomBonds = at->getOwningMol().getAtomBonds(at);
while (atomBonds.first != atomBonds.second) {
unsigned int bondIdx =
at->getOwningMol().getTopology()[*atomBonds.first]->getIdx();
if (at->getOwningMol().getRingInfo()->numBondRings(bondIdx)) {
return 1;
}
++atomBonds.first;
}
return 0;
};
static inline int queryIsBondInRing(Bond const *bond) {
return bond->getOwningMol().getRingInfo()->numBondRings(bond->getIdx()) != 0;
};
static inline int queryAtomMinRingSize(Atom const *at) {
return at->getOwningMol().getRingInfo()->minAtomRingSize(at->getIdx());
};
static inline int queryBondMinRingSize(Bond const *bond) {
return bond->getOwningMol().getRingInfo()->minBondRingSize(bond->getIdx());
};
static inline int queryAtomRingBondCount(Atom const *at) {
// EFF: cache this result
int res = 0;
ROMol::OBOND_ITER_PAIR atomBonds = at->getOwningMol().getAtomBonds(at);
while (atomBonds.first != atomBonds.second) {
unsigned int bondIdx =
at->getOwningMol().getTopology()[*atomBonds.first]->getIdx();
if (at->getOwningMol().getRingInfo()->numBondRings(bondIdx)) {
res++;
}
++atomBonds.first;
}
return res;
}
template <int tgt>
int queryAtomIsInRingOfSize(Atom const *at) {
if (at->getOwningMol().getRingInfo()->isAtomInRingOfSize(at->getIdx(), tgt)) {
return tgt;
} else {
return 0;
}
};
template <int tgt>
int queryBondIsInRingOfSize(Bond const *bond) {
if (bond->getOwningMol().getRingInfo()->isBondInRingOfSize(bond->getIdx(),
tgt)) {
return tgt;
} else {
return 0;
}
};
template <class T>
T *makeAtomSimpleQuery(int what, int func(Atom const *),
const std::string &description = "Atom Simple") {
T *res = new T;
res->setVal(what);
res->setDataFunc(func);
res->setDescription(description);
return res;
}
//! returns a Query for matching atomic number
template <class T>
T *makeAtomNumQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryAtomNum, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomNumQuery(int what);
//! returns a Query for matching implicit valence
template <class T>
T *makeAtomImplicitValenceQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryAtomImplicitValence, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomImplicitValenceQuery(int what);
//! returns a Query for matching explicit valence
template <class T>
T *makeAtomExplicitValenceQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryAtomExplicitValence, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomExplicitValenceQuery(int what);
//! returns a Query for matching total valence
template <class T>
T *makeAtomTotalValenceQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryAtomTotalValence, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomTotalValenceQuery(int what);
//! returns a Query for matching explicit degree
template <class T>
T *makeAtomExplicitDegreeQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryAtomExplicitDegree, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomExplicitDegreeQuery(int what);
//! returns a Query for matching atomic degree
template <class T>
T *makeAtomTotalDegreeQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryAtomTotalDegree, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomTotalDegreeQuery(int what);
//! returns a Query for matching hydrogen count
template <class T>
T *makeAtomHCountQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryAtomHCount, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomHCountQuery(int what);
//! returns a Query for matching ring atoms
template <class T>
T *makeAtomHasImplicitHQuery(const std::string &descr) {
return makeAtomSimpleQuery<T>(true, queryAtomHasImplicitH, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomHasImplicitHQuery();
//! returns a Query for matching implicit hydrogen count
template <class T>
T *makeAtomImplicitHCountQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryAtomImplicitHCount, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomImplicitHCountQuery(int what);
//! returns a Query for matching the \c isAromatic flag
template <class T>
T *makeAtomAromaticQuery(const std::string &descr) {
return makeAtomSimpleQuery<T>(true, queryAtomAromatic, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomAromaticQuery();
//! returns a Query for matching aliphatic atoms
template <class T>
T *makeAtomAliphaticQuery(const std::string &descr) {
return makeAtomSimpleQuery<T>(true, queryAtomAliphatic, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomAliphaticQuery();
//! returns a Query for matching atoms with a particular mass
template <class T>
T *makeAtomMassQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(massIntegerConversionFactor * what,
queryAtomMass, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomMassQuery(int what);
//! returns a Query for matching atoms with a particular isotope
template <class T>
T *makeAtomIsotopeQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryAtomIsotope, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomIsotopeQuery(int what);
//! returns a Query for matching formal charge
template <class T>
T *makeAtomFormalChargeQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryAtomFormalCharge, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomFormalChargeQuery(int what);
//! returns a Query for matching hybridization
template <class T>
T *makeAtomHybridizationQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryAtomHybridization, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomHybridizationQuery(int what);
//! returns a Query for matching the number of radical electrons
template <class T>
T *makeAtomNumRadicalElectronsQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryAtomNumRadicalElectrons, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomNumRadicalElectronsQuery(int what);
//! returns a Query for matching whether or not chirality has been set on the
//! atom
template <class T>
T *makeAtomHasChiralTagQuery(const std::string &descr) {
return makeAtomSimpleQuery<T>(true, queryAtomHasChiralTag, descr);
}
//! \overloadquery
ATOM_EQUALS_QUERY *makeAtomHasChiralTagQuery();
//! returns a Query for matching whether or not a potentially chiral atom is
//! missing a chiral tag
template <class T>
T *makeAtomMissingChiralTagQuery(const std::string &descr) {
return makeAtomSimpleQuery<T>(true, queryAtomMissingChiralTag, descr);
}
//! \overloadquery
ATOM_EQUALS_QUERY *makeAtomMissingChiralTagQuery();
//! returns a Query for matching atoms with unsaturation:
template <class T>
T *makeAtomUnsaturatedQuery(const std::string &descr) {
return makeAtomSimpleQuery<T>(true, queryAtomUnsaturated, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomUnsaturatedQuery();
//! returns a Query for matching ring atoms
template <class T>
T *makeAtomInRingQuery(const std::string &descr) {
return makeAtomSimpleQuery<T>(true, queryIsAtomInRing, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomInRingQuery();
//! returns a Query for matching atoms in a particular number of rings
template <class T>
T *makeAtomInNRingsQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryIsAtomInNRings, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomInNRingsQuery(int what);
//! returns a Query for matching atoms in rings of a particular size
ATOM_EQUALS_QUERY *makeAtomInRingOfSizeQuery(int tgt);
//! returns a Query for matching an atom's minimum ring size
template <class T>
T *makeAtomMinRingSizeQuery(int tgt, const std::string &descr) {
return makeAtomSimpleQuery<T>(tgt, queryAtomMinRingSize, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomMinRingSizeQuery(int tgt);
//! returns a Query for matching atoms with a particular number of ring bonds
template <class T>
T *makeAtomRingBondCountQuery(int what, const std::string &descr) {
return makeAtomSimpleQuery<T>(what, queryAtomRingBondCount, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomRingBondCountQuery(int what);
//! returns a Query for matching generic A atoms (heavy atoms)
ATOM_EQUALS_QUERY *makeAAtomQuery();
//! returns a Query for matching generic AH atoms (any atom)
ATOM_EQUALS_QUERY *makeAHAtomQuery();
//! returns a Query for matching generic Q atoms (heteroatoms)
ATOM_OR_QUERY *makeQAtomQuery();
//! returns a Query for matching generic QH atoms (heteroatom or H)
ATOM_EQUALS_QUERY *makeQHAtomQuery();
//! returns a Query for matching generic X atoms (halogens)
ATOM_OR_QUERY *makeXAtomQuery();
//! returns a Query for matching generic XH atoms (halogen or H)
ATOM_OR_QUERY *makeXHAtomQuery();
//! returns a Query for matching generic M atoms (metals)
ATOM_OR_QUERY *makeMAtomQuery();
//! returns a Query for matching generic MH atoms (metals or H)
ATOM_OR_QUERY *makeMHAtomQuery();
//! returns a Query for matching atoms that have ring bonds
template <class T>
T *makeAtomHasRingBondQuery(const std::string &descr) {
return makeAtomSimpleQuery<T>(1, queryAtomHasRingBond, descr);
}
//! \overload
ATOM_EQUALS_QUERY *makeAtomHasRingBondQuery();
//! returns a Query for matching bond orders
BOND_EQUALS_QUERY *makeBondOrderEqualsQuery(Bond::BondType what);
//! returns a Query for matching bond directions
BOND_EQUALS_QUERY *makeBondDirEqualsQuery(Bond::BondDir what);
//! returns a Query for matching bonds with stereo set
BOND_EQUALS_QUERY *makeBondHasStereoQuery();
//! returns a Query for matching ring bonds
BOND_EQUALS_QUERY *makeBondIsInRingQuery();
//! returns a Query for matching bonds in rings of a particular size
BOND_EQUALS_QUERY *makeBondInRingOfSizeQuery(int what);
//! returns a Query for matching a bond's minimum ring size
BOND_EQUALS_QUERY *makeBondMinRingSizeQuery(int what);
//! returns a Query for matching bonds in a particular number of rings
BOND_EQUALS_QUERY *makeBondInNRingsQuery(int tgt);
//! returns a Query for matching any bond
BOND_NULL_QUERY *makeBondNullQuery();
//! returns a Query for matching any atom
ATOM_NULL_QUERY *makeAtomNullQuery();
static inline int queryAtomRingMembership(Atom const *at) {
return static_cast<int>(
at->getOwningMol().getRingInfo()->numAtomRings(at->getIdx()));
}
// I'm pretty sure that this typedef shouldn't be necessary,
// but VC++ generates a warning about const Atom const * in
// the definition of Match, then complains about an override
// that differs only by const/volatile (c4301), then generates
// incorrect code if we don't do this... so let's do it.
typedef Atom const *ConstAtomPtr;
class AtomRingQuery : public Queries::EqualityQuery<int, ConstAtomPtr, true> {
public:
AtomRingQuery() : Queries::EqualityQuery<int, ConstAtomPtr, true>(-1) {
// default is to just do a number of rings query:
this->setDescription("AtomInNRings");
this->setDataFunc(queryAtomRingMembership);
};
explicit AtomRingQuery(int v)
: Queries::EqualityQuery<int, ConstAtomPtr, true>(v) {
// default is to just do a number of rings query:
this->setDescription("AtomInNRings");
this->setDataFunc(queryAtomRingMembership);
};
virtual bool Match(const ConstAtomPtr what) const {
int v = this->TypeConvert(what, Queries::Int2Type<true>());
bool res;
if (this->d_val < 0) {
res = v != 0;
} else {
res = !Queries::queryCmp(v, this->d_val, this->d_tol);
}
if (this->getNegation()) {
res = !res;
}
return res;
}
//! returns a copy of this query
Queries::Query<int, ConstAtomPtr, true> *copy() const {
AtomRingQuery *res = new AtomRingQuery(this->d_val);
res->setNegation(getNegation());
res->setTol(this->getTol());
res->d_description = this->d_description;
res->d_dataFunc = this->d_dataFunc;
return res;
}
};
//! allows use of recursive structure queries (e.g. recursive SMARTS)
class RecursiveStructureQuery
: public Queries::SetQuery<int, Atom const *, true> {
public:
RecursiveStructureQuery()
: Queries::SetQuery<int, Atom const *, true>(), d_serialNumber(0) {
setDataFunc(getAtIdx);
setDescription("RecursiveStructure");
};
//! initialize from an ROMol pointer
/*!
<b>Notes</b>
- this takes over ownership of the pointer
*/
RecursiveStructureQuery(ROMol const *query, unsigned int serialNumber = 0)
: Queries::SetQuery<int, Atom const *, true>(),
d_serialNumber(serialNumber) {
setQueryMol(query);
setDataFunc(getAtIdx);
setDescription("RecursiveStructure");
};
//! returns the index of an atom
static inline int getAtIdx(Atom const *at) {
PRECONDITION(at, "bad atom argument");
return at->getIdx();
};
//! sets the molecule we'll use recursively
/*!
<b>Notes</b>
- this takes over ownership of the pointer
*/
void setQueryMol(ROMol const *query) { dp_queryMol.reset(query); }
//! returns a pointer to our query molecule
ROMol const *getQueryMol() const { return dp_queryMol.get(); };
//! returns a copy of this query
Queries::Query<int, Atom const *, true> *copy() const {
RecursiveStructureQuery *res = new RecursiveStructureQuery();
res->dp_queryMol.reset(new ROMol(*dp_queryMol, true));
std::set<int>::const_iterator i;
for (i = d_set.begin(); i != d_set.end(); i++) {
res->insert(*i);
}
res->setNegation(getNegation());
res->d_description = d_description;
res->d_serialNumber = d_serialNumber;
return res;
}
unsigned int getSerialNumber() const { return d_serialNumber; };
#ifdef RDK_THREADSAFE_SSS
boost::mutex d_mutex;
#endif
private:
boost::shared_ptr<const ROMol> dp_queryMol;
unsigned int d_serialNumber;
};
template <typename T>
int nullDataFun(T) {
return 1;
}
template <typename T>
bool nullQueryFun(T) {
return true;
}
typedef Bond const *ConstBondPtr;
// ! Query whether an atom has a property
template <class TargetPtr>
class HasPropQuery : public Queries::EqualityQuery<int, TargetPtr, true> {
std::string propname;
public:
HasPropQuery() : Queries::EqualityQuery<int, TargetPtr, true>(), propname() {
// default is to just do a number of rings query:
this->setDescription("AtomHasProp");
this->setDataFunc(0);
};
explicit HasPropQuery(const std::string &v)
: Queries::EqualityQuery<int, TargetPtr, true>(), propname(v) {
// default is to just do a number of rings query:
this->setDescription("AtomHasProp");
this->setDataFunc(0);
};
virtual bool Match(const TargetPtr what) const {
bool res = what->hasProp(propname);
if (this->getNegation()) {
res = !res;
}
return res;
}
//! returns a copy of this query
Queries::Query<int, TargetPtr, true> *copy() const {
HasPropQuery *res = new HasPropQuery(this->propname);
res->setNegation(this->getNegation());
res->d_description = this->d_description;
return res;
}
};
typedef Queries::EqualityQuery<int, Atom const *, true> ATOM_PROP_QUERY;
typedef Queries::EqualityQuery<int, Bond const *, true> BOND_PROP_QUERY;
//! returns a Query for matching atoms that have a particular property
template <class Target>
Queries::EqualityQuery<int, const Target *, true> *makeHasPropQuery(
const std::string &property) {
return new HasPropQuery<const Target *>(property);
}
// ! Query whether an atom has a property with a value
template <class TargetPtr, class T>
class HasPropWithValueQuery
: public Queries::EqualityQuery<int, TargetPtr, true> {
std::string propname;
T val;
T tolerance;
public:
HasPropWithValueQuery()
: Queries::EqualityQuery<int, TargetPtr, true>(), propname(), val() {
// default is to just do a number of rings query:
this->setDescription("HasPropWithValue");
this->setDataFunc(0);
};
explicit HasPropWithValueQuery(const std::string &prop, const T &v,
const T &tol = 0.0)
: Queries::EqualityQuery<int, TargetPtr, true>(),
propname(prop),
val(v),
tolerance(tol) {
// default is to just do a number of rings query:
this->setDescription("HasPropWithValue");
this->setDataFunc(0);
};
virtual bool Match(const TargetPtr what) const {
bool res = what->hasProp(propname);
if (res) {
try {
T atom_val = what->template getProp<T>(propname);
res = Queries::queryCmp(atom_val, this->val, this->tolerance) == 0;
} catch (KeyErrorException e) {
res = false;
} catch (boost::bad_any_cast) {
res = false;
}
#ifdef __GNUC__
#if (__GNUC__ < 4 || (__GNUC__ == 4 && __GNUC_MINOR__ < 2))
catch (...) {
// catch all -- this is currently necessary to
// trap some bugs in boost+gcc configurations
// Normally, this is not the correct thing to
// do, but the only exception above is due
// to the boost any_cast which is trapped
// by the Boost python wrapper when it shouldn't
// be.
res = false;
}
#endif
#endif
}
if (this->getNegation()) {
res = !res;
}
return res;
}
//! returns a copy of this query
Queries::Query<int, TargetPtr, true> *copy() const {
HasPropWithValueQuery *res =
new HasPropWithValueQuery(this->propname, this->val, this->tolerance);
res->setNegation(this->getNegation());
res->d_description = this->d_description;
return res;
}
};
template <class TargetPtr>
class HasPropWithValueQuery<TargetPtr, std::string>
: public Queries::EqualityQuery<int, TargetPtr, true> {
std::string propname;
std::string val;
public:
HasPropWithValueQuery()
: Queries::EqualityQuery<int, TargetPtr, true>(), propname(), val() {
// default is to just do a number of rings query:
this->setDescription("HasPropWithValue");
this->setDataFunc(0);
};
explicit HasPropWithValueQuery(const std::string &prop, const std::string &v,
const std::string &tol = "")
: Queries::EqualityQuery<int, TargetPtr, true>(), propname(prop), val(v) {
RDUNUSED_PARAM(tol);
// default is to just do a number of rings query:
this->setDescription("HasPropWithValue");
this->setDataFunc(0);
};
virtual bool Match(const TargetPtr what) const {
bool res = what->hasProp(propname);
if (res) {
try {
std::string atom_val = what->template getProp<std::string>(propname);
res = atom_val == this->val;
} catch (KeyErrorException) {
res = false;
} catch (boost::bad_any_cast) {
res = false;
}
#ifdef __GNUC__
#if (__GNUC__ < 4 || (__GNUC__ == 4 && __GNUC_MINOR__ < 2))
catch (...) {
// catch all -- this is currently necessary to
// trap some bugs in boost+gcc configurations
// Normally, this is not the correct thing to
// do, but the only exception above is due
// to the boost any_cast which is trapped
// by the Boost python wrapper when it shouldn't
// be.
res = false;
}
#endif
#endif
}
if (this->getNegation()) {
res = !res;
}
return res;
}
//! returns a copy of this query
Queries::Query<int, TargetPtr, true> *copy() const {
HasPropWithValueQuery<TargetPtr, std::string> *res =
new HasPropWithValueQuery<TargetPtr, std::string>(this->propname,
this->val);
res->setNegation(this->getNegation());
res->d_description = this->d_description;
return res;
}
};
template <class Target, class T>
Queries::EqualityQuery<int, const Target *, true> *makePropQuery(
const std::string &propname, const T &val, const T &tolerance = T()) {
return new HasPropWithValueQuery<const Target *, T>(propname, val, tolerance);
}
bool isComplexQuery(const Bond *b);
bool isComplexQuery(const Atom *a);
bool isAtomAromatic(const Atom *a);
};
#endif