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
|
//
// 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
|
