Loading...
Searching...
No Matches
SetQuery.h
Go to the documentation of this file.
//
// Copyright (c) 2003-2020 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.
//
#include <RDGeneral/export.h>
#ifndef RD_SETQUERY_H
#define RD_SETQUERY_H
#include <set>
#include "Query.h"
#include <sstream>
#include <algorithm>
#include <iterator>
namespace Queries {
//! \brief a Query implementing a set: arguments must
//! one of a set of values
//!
template <class MatchFuncArgType, class DataFuncArgType = MatchFuncArgType,
bool needsConversion = false>
class RDKIT_QUERY_EXPORT SetQuery
: public Query<MatchFuncArgType, DataFuncArgType, needsConversion> {
public:
typedef std::set<MatchFuncArgType> CONTAINER_TYPE;
SetQuery() : Query<MatchFuncArgType, DataFuncArgType, needsConversion>() {}
//! insert an entry into our \c set
void insert(const MatchFuncArgType what) {
if (d_set.find(what) == this->d_set.end()) {
this->d_set.insert(what);
}
}
//! clears our \c set
void clear() { this->d_set.clear(); }
bool Match(const DataFuncArgType what) const override {
MatchFuncArgType mfArg =
this->TypeConvert(what, Int2Type<needsConversion>());
return (this->d_set.find(mfArg) != this->d_set.end()) ^ this->getNegation();
}
Query<MatchFuncArgType, DataFuncArgType, needsConversion> *copy()
const override {
SetQuery<MatchFuncArgType, DataFuncArgType, needsConversion> *res =
new SetQuery<MatchFuncArgType, DataFuncArgType, needsConversion>();
res->setDataFunc(this->d_dataFunc);
typename std::set<MatchFuncArgType>::const_iterator i;
for (i = this->d_set.begin(); i != this->d_set.end(); ++i) {
res->insert(*i);
}
res->setNegation(this->getNegation());
res->d_description = this->d_description;
res->d_queryType = this->d_queryType;
return res;
}
typename CONTAINER_TYPE::const_iterator beginSet() const {
return d_set.begin();
}
typename CONTAINER_TYPE::const_iterator endSet() const { return d_set.end(); }
unsigned int size() const { return rdcast<unsigned int>(d_set.size()); }
std::string getFullDescription() const override {
std::ostringstream res;
res << this->getDescription() << " val";
if (this->getNegation()) {
res << " not in ";
} else {
res << " in (";
}
std::copy(d_set.begin(), d_set.end(),
std::ostream_iterator<MatchFuncArgType>(res, ", "));
res << ")";
return res.str();
}
protected:
CONTAINER_TYPE d_set;
};
} // namespace Queries
#endif
Query.h
Queries::Int2Type
class to allow integer values to pick templates
Definition Query.h:26
Queries::Query
Base class for all queries.
Definition Query.h:45
Queries::Query::d_queryType
std::string d_queryType
Definition Query.h:153
Queries::Query::setDataFunc
void setDataFunc(MatchFuncArgType(*what)(DataFuncArgType))
sets our data function
Definition Query.h:94
Queries::Query::setNegation
void setNegation(bool what)
sets whether or not we are negated
Definition Query.h:59
Queries::Query::d_description
std::string d_description
Definition Query.h:152
Queries::SetQuery
a Query implementing a set: arguments must one of a set of values
Definition SetQuery.h:26
Queries::SetQuery::beginSet
CONTAINER_TYPE::const_iterator beginSet() const
Definition SetQuery.h:63
Queries::SetQuery::copy
Query< MatchFuncArgType, DataFuncArgType, needsConversion > * copy() const override
returns a copy of this Query
Definition SetQuery.h:48
Queries::SetQuery::clear
void clear()
clears our set
Definition SetQuery.h:40
Queries::SetQuery::CONTAINER_TYPE
std::set< MatchFuncArgType > CONTAINER_TYPE
Definition SetQuery.h:28
Queries::SetQuery::insert
void insert(const MatchFuncArgType what)
insert an entry into our set
Definition SetQuery.h:33
Queries::SetQuery::size
unsigned int size() const
Definition SetQuery.h:67
Queries::SetQuery::d_set
CONTAINER_TYPE d_set
Definition SetQuery.h:84
Queries::SetQuery::Match
bool Match(const DataFuncArgType what) const override
returns whether or not we match the argument
Definition SetQuery.h:42
Queries::SetQuery::endSet
CONTAINER_TYPE::const_iterator endSet() const
Definition SetQuery.h:66
Queries::SetQuery::getFullDescription
std::string getFullDescription() const override
returns a fuller text description
Definition SetQuery.h:69
Queries::SetQuery::SetQuery
SetQuery()
Definition SetQuery.h:30
export.h
RDKIT_QUERY_EXPORT
#define RDKIT_QUERY_EXPORT
Definition export.h:589
Queries
Definition AndQuery.h:16