|
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
|
#ifndef CommonToolsUtilsExpressionEvaluatorTemplates_H
#define CommonToolsUtilsExpressionEvaluatorTemplates_H
#include <vector>
#include <algorithm>
#include <numeric>
#include <limits>
#include <memory>
#include <tuple>
namespace reco {
template <typename Ret, typename... Args>
struct genericExpression {
virtual Ret operator()(Args...) const = 0;
virtual ~genericExpression() {}
};
template <typename Object>
struct CutOnObject {
virtual bool eval(Object const&) const = 0;
virtual ~CutOnObject() {}
};
template <typename Object>
struct ValueOnObject {
virtual double eval(Object const&) const = 0;
virtual ~ValueOnObject() {}
};
template <typename Object>
struct MaskCollection {
using Collection = std::vector<Object const*>;
using Mask = std::vector<bool>;
template <typename F>
void mask(Collection const& cands, Mask& mask, F f) const {
mask.resize(cands.size());
std::transform(
cands.begin(), cands.end(), mask.begin(), [&](typename Collection::value_type const& c) { return f(*c); });
}
virtual void eval(Collection const&, Mask&) const = 0;
virtual ~MaskCollection() {}
};
template <typename Object>
struct SelectInCollection {
using Collection = std::vector<Object const*>;
template <typename F>
void select(Collection& cands, F f) const {
cands.erase(
std::remove_if(cands.begin(), cands.end(), [&](typename Collection::value_type const& c) { return !f(*c); }),
cands.end());
}
virtual void eval(Collection&) const = 0;
virtual ~SelectInCollection() {}
};
template <typename Object>
struct SelectIndecesInCollection {
using Collection = std::vector<Object const*>;
using Indices = std::vector<unsigned int>;
template <typename F>
void select(Collection const& cands, Indices& inds, F f) const {
unsigned int i = 0;
for (auto const& c : cands) {
if (f(*c))
inds.push_back(i);
++i;
}
}
virtual void eval(Collection const&, Indices&) const = 0;
virtual ~SelectIndecesInCollection() {}
};
} // namespace reco
#endif // CommonToolsUtilsExpressionEvaluatorTemplates_H
|