|
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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
// system include files
#include <atomic>
#include <memory>
#include <cmath>
#include <iostream>
#include <sstream>
#include <fstream>
// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDFilter.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/Run.h"
#include "FWCore/Framework/interface/LuminosityBlock.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/Common/interface/TriggerNames.h"
#include "DataFormats/Common/interface/Handle.h"
//Triggers
#include "DataFormats/Common/interface/TriggerResults.h"
#include "HLTrigger/HLTcore/interface/HLTConfigProvider.h"
//#define EDM_ML_DEBUG
//
// class declaration
//
namespace alCaIsoTracksProducerFilter {
struct Counters {
Counters() : nAll_(0), nGood_(0) {}
mutable std::atomic<unsigned int> nAll_, nGood_;
};
} // namespace alCaIsoTracksProducerFilter
class AlCaIsoTracksProducerFilter
: public edm::stream::EDFilter<edm::GlobalCache<alCaIsoTracksProducerFilter::Counters> > {
public:
explicit AlCaIsoTracksProducerFilter(edm::ParameterSet const&, const alCaIsoTracksProducerFilter::Counters* count);
~AlCaIsoTracksProducerFilter() override = default;
static std::unique_ptr<alCaIsoTracksProducerFilter::Counters> initializeGlobalCache(edm::ParameterSet const& iConfig) {
return std::make_unique<alCaIsoTracksProducerFilter::Counters>();
}
bool filter(edm::Event&, edm::EventSetup const&) override;
void endStream() override;
static void globalEndJob(const alCaIsoTracksProducerFilter::Counters* counters);
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
private:
void beginRun(edm::Run const&, edm::EventSetup const&) override;
void endRun(edm::Run const&, edm::EventSetup const&) override;
// ----------member data ---------------------------
HLTConfigProvider hltConfig_;
unsigned int nRun_, nAll_, nGood_;
const std::vector<std::string> trigNames_;
const std::string processName_;
const edm::InputTag triggerResultsLabel_;
const edm::EDGetTokenT<edm::TriggerResults> tok_trigRes_;
};
AlCaIsoTracksProducerFilter::AlCaIsoTracksProducerFilter(const edm::ParameterSet& iConfig,
const alCaIsoTracksProducerFilter::Counters* count)
: nRun_(0),
nAll_(0),
nGood_(0),
trigNames_(iConfig.getParameter<std::vector<std::string> >("triggers")),
processName_(iConfig.getParameter<std::string>("processName")),
triggerResultsLabel_(iConfig.getParameter<edm::InputTag>("triggerResultLabel")),
tok_trigRes_(consumes<edm::TriggerResults>(triggerResultsLabel_)) {
edm::LogVerbatim("HcalIsoTrack") << "Use process name " << processName_ << " Labels " << triggerResultsLabel_
<< " selecting " << trigNames_.size() << " triggers\n";
for (unsigned int k = 0; k < trigNames_.size(); ++k) {
edm::LogVerbatim("HcalIsoTrack") << "Trigger[" << k << "] " << trigNames_[k] << std::endl;
}
}
bool AlCaIsoTracksProducerFilter::filter(edm::Event& iEvent, edm::EventSetup const& iSetup) {
++nAll_;
edm::LogVerbatim("HcalIsoTrack") << "Run " << iEvent.id().run() << " Event " << iEvent.id().event() << " Luminosity "
<< iEvent.luminosityBlock() << " Bunch " << iEvent.bunchCrossing() << std::endl;
//Find if the event passes one of the chosen triggers
bool triggerSatisfied(false);
if (trigNames_.empty()) {
triggerSatisfied = true;
} else {
auto const& triggerResults = iEvent.getHandle(tok_trigRes_);
if (triggerResults.isValid()) {
std::vector<std::string> modules;
const edm::TriggerNames& triggerNames = iEvent.triggerNames(*triggerResults);
const std::vector<std::string>& triggerNames_ = triggerNames.triggerNames();
for (unsigned int iHLT = 0; iHLT < triggerResults->size(); iHLT++) {
int hlt = triggerResults->accept(iHLT);
for (unsigned int i = 0; i < trigNames_.size(); ++i) {
if (triggerNames_[iHLT].find(trigNames_[i]) != std::string::npos) {
#ifdef EDM_ML_DEBUG
edm::LogVerbatim("HcalIsoTrack")
<< triggerNames_[iHLT] << " has got HLT flag " << hlt << ":" << triggerSatisfied;
#endif
if (hlt > 0) {
triggerSatisfied = true;
break;
}
}
}
if (triggerSatisfied)
break;
}
}
}
#ifdef EDM_ML_DEBUG
edm::LogVerbatim("HcalIsoTrack") << "AlCaIsoTracksProducerFilter:: triggerSatisfied: " << triggerSatisfied;
#endif
if (triggerSatisfied)
++nGood_;
return triggerSatisfied;
}
void AlCaIsoTracksProducerFilter::endStream() {
globalCache()->nAll_ += nAll_;
globalCache()->nGood_ += nGood_;
}
void AlCaIsoTracksProducerFilter::globalEndJob(const alCaIsoTracksProducerFilter::Counters* count) {
edm::LogVerbatim("HcalIsoTrack") << "Selects " << count->nGood_ << " in " << count->nAll_ << " events " << std::endl;
}
void AlCaIsoTracksProducerFilter::beginRun(edm::Run const& iRun, edm::EventSetup const& iSetup) {
bool changed(false);
bool flag = hltConfig_.init(iRun, iSetup, processName_, changed);
edm::LogVerbatim("HcalIsoTrack") << "Run[" << nRun_ << "] " << iRun.run() << " hltconfig.init " << flag << std::endl;
}
void AlCaIsoTracksProducerFilter::endRun(edm::Run const& iRun, edm::EventSetup const&) {
++nRun_;
edm::LogVerbatim("HcalIsoTrack") << "endRun[" << nRun_ << "] " << iRun.run() << std::endl;
}
void AlCaIsoTracksProducerFilter::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
//The following says we do not know what parameters are allowed so do no validation
// Please change this to state exactly what you do use, even if it is no parameters
edm::ParameterSetDescription desc;
desc.add<edm::InputTag>("triggerResultLabel", edm::InputTag("TriggerResults", "", "HLT"));
std::vector<std::string> trigger = {"HLT_IsoTrackHB", "HLT_IsoTrackHE"};
desc.add<std::vector<std::string> >("triggers", trigger);
desc.add<std::string>("processName", "HLT");
descriptions.add("alcaIsoTracksProducerFilter", desc);
}
#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(AlCaIsoTracksProducerFilter);
|