|
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
|
#ifndef DQMOffline_Trigger_VarRangeCut_h
#define DQMOffline_Trigger_VarRangeCut_h
//********************************************************************************
//
// Description:
// A object containing a minimal set of selection cuts.
// These selection cuts are intended to be simple selections on kinematic variables.
// Currently these are implimented as simple allowed ranges X<var<Y which are ORed together
// So for example we may want to have an eta cut of 0<|eta|<1.4442 || 1.556<|eta|<2.5
//
// Implimentation:
// std::function holds the function which generates the variable from the object
// the name of the variable is also stored so we can determine if we should not apply
// a given selection cut
//
// Author : Sam Harper , RAL, May 2017
//
//***********************************************************************************
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "DQMOffline/Trigger/interface/FunctionDefs.h"
#include <boost/algorithm/string.hpp>
template <typename ObjType>
class VarRangeCut {
public:
explicit VarRangeCut(const edm::ParameterSet& config);
static edm::ParameterSetDescription makePSetDescription();
bool operator()(const ObjType& obj) const;
const std::string& varName() const { return varName_; }
private:
std::string varName_;
std::function<float(const ObjType&)> varFunc_;
std::vector<std::pair<float, float> > allowedRanges_;
};
template <typename ObjType>
VarRangeCut<ObjType>::VarRangeCut(const edm::ParameterSet& config) {
varName_ = config.getParameter<std::string>("rangeVar");
varFunc_ = hltdqm::getUnaryFuncFloat<ObjType>(varName_);
auto ranges = config.getParameter<std::vector<std::string> >("allowedRanges");
for (auto range : ranges) {
std::vector<std::string> splitRange;
boost::split(splitRange, range, boost::is_any_of(":"));
if (splitRange.size() != 2)
throw cms::Exception("ConfigError")
<< "in VarRangeCut::VarRangeCut range " << range << " is not of format X:Y" << std::endl;
allowedRanges_.push_back({std::stof(splitRange[0]), std::stof(splitRange[1])});
}
}
template <typename ObjType>
edm::ParameterSetDescription VarRangeCut<ObjType>::makePSetDescription() {
edm::ParameterSetDescription desc;
desc.add<std::string>("rangeVar", "");
desc.add<std::vector<std::string> >("allowedRanges", std::vector<std::string>());
return desc;
}
template <typename ObjType>
bool VarRangeCut<ObjType>::operator()(const ObjType& obj) const {
if (!varFunc_)
return true; //auto pass if we dont specify a variable function
else {
float varVal = varFunc_(obj);
for (auto& range : allowedRanges_) {
if (varVal >= range.first && varVal < range.second)
return true;
}
return false;
}
}
#endif
|