|
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
#include "DataFormats/PatCandidates/interface/VIDCutFlowResult.h"
#include "FWCore/Utilities/interface/Exception.h"
namespace {
const std::string empty_str("");
}
namespace vid {
CutFlowResult::CutFlowResult(const std::string& name,
const std::string& hash,
const std::map<std::string, unsigned>& n2idx,
const std::vector<double>& values,
unsigned bitmap,
unsigned mask)
: name_(name), hash_(hash), bitmap_(bitmap), mask_(mask), values_(values) {
for (const auto& val : n2idx) {
names_.push_back(val.first);
indices_.push_back(val.second);
}
}
const std::string& CutFlowResult::getNameAtIndex(const unsigned idx) const {
unsigned internal_idx = 0;
for (const auto& value : indices_) {
if (value == idx)
return names_[internal_idx];
++internal_idx;
}
throw cms::Exception("IndexNotFound") << "index = " << idx << " has no corresponding cut name!";
return empty_str;
}
bool CutFlowResult::getCutResultByIndex(const unsigned idx) const {
if (idx >= indices_.size()) {
throw cms::Exception("OutOfBounds") << idx << " is out of bounds for this cut flow!";
}
return getCutBit(idx);
}
bool CutFlowResult::getCutResultByName(const std::string& name) const {
auto found_name = std::lower_bound(names_.begin(), names_.end(), name);
if (found_name == names_.end() || *found_name != name) {
throw cms::Exception("UnknownName") << "Cut name: " << name << " is not known for this cutflow!";
}
return getCutBit(indices_[std::distance(names_.begin(), found_name)]);
}
bool CutFlowResult::isCutMasked(const unsigned idx) const {
if (idx >= indices_.size()) {
throw cms::Exception("OutOfBounds") << idx << " is out of bounds for this cut flow!";
}
return getMaskBit(idx);
}
bool CutFlowResult::isCutMasked(const std::string& name) const {
auto found_name = std::lower_bound(names_.begin(), names_.end(), name);
if (found_name == names_.end() || *found_name != name) {
throw cms::Exception("UnknownName") << "Cut name: " << name << " is not known for this cutflow!";
}
return getMaskBit(indices_[std::distance(names_.begin(), found_name)]);
}
double CutFlowResult::getValueCutUpon(const unsigned idx) const {
if (idx >= indices_.size()) {
throw cms::Exception("OutOfBounds") << idx << " is out of bounds for this cut flow!";
}
return getCutValue(idx);
}
double CutFlowResult::getValueCutUpon(const std::string& name) const {
auto found_name = std::lower_bound(names_.begin(), names_.end(), name);
if (found_name == names_.end() || *found_name != name) {
throw cms::Exception("UnknownName") << "Cut name: " << name << " is not known for this cutflow!";
}
return getCutValue(indices_[std::distance(names_.begin(), found_name)]);
}
CutFlowResult CutFlowResult::getCutFlowResultMasking(const std::vector<unsigned>& idxs) const {
unsigned bitmap = bitmap_;
unsigned mask = mask_;
for (const unsigned idx : idxs) {
if (idx >= indices_.size()) {
throw cms::Exception("OutOfBounds") << idx << " is out of bounds for this cut flow!";
}
mask = mask | 1 << idx;
}
bitmap = bitmap | mask;
return CutFlowResult(name_, empty_str, names_, indices_, values_, bitmap, mask);
}
CutFlowResult CutFlowResult::getCutFlowResultMasking(const std::vector<std::string>& names) const {
unsigned bitmap = bitmap_;
unsigned mask = mask_;
for (const std::string& name : names) {
auto found_name = std::lower_bound(names_.begin(), names_.end(), name);
if (found_name == names_.end() || *found_name != name) {
throw cms::Exception("UnknownName") << "Cut name: " << name << " is not known for this cutflow!";
}
mask = mask | 1 << indices_[std::distance(names_.begin(), found_name)];
}
bitmap = bitmap | mask;
return CutFlowResult(name_, empty_str, names_, indices_, values_, bitmap, mask);
}
CutFlowResult CutFlowResult::getCutFlowResultMasking(const unsigned idx) const {
unsigned bitmap = bitmap_;
unsigned mask = mask_;
if (idx >= indices_.size()) {
throw cms::Exception("OutOfBounds") << idx << " is out of bounds for this cut flow!";
}
mask = mask | 1 << idx;
bitmap = bitmap | mask;
return CutFlowResult(name_, empty_str, names_, indices_, values_, bitmap, mask);
}
CutFlowResult CutFlowResult::getCutFlowResultMasking(const std::string& name) const {
unsigned bitmap = bitmap_;
unsigned mask = mask_;
auto found_name = std::lower_bound(names_.begin(), names_.end(), name);
if (found_name == names_.end() || *found_name != name) {
throw cms::Exception("UnknownName") << "Cut name: " << name << " is not known for this cutflow!";
}
mask = mask | 1 << indices_[std::distance(names_.begin(), found_name)];
bitmap = bitmap | mask;
return CutFlowResult(name_, empty_str, names_, indices_, values_, bitmap, mask);
}
} // namespace vid
|