forked from cms-sw/cmssw
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Phase2L1TGMTFilter.cc
104 lines (87 loc) · 3.54 KB
/
Phase2L1TGMTFilter.cc
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
#include <memory>
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/StreamID.h"
#include "DataFormats/L1TMuonPhase2/interface/TrackerMuon.h"
#include "DataFormats/L1TMuonPhase2/interface/SAMuon.h"
#include "Node.h"
//
// class declaration
//
using namespace Phase2L1GMT;
using namespace l1t;
class Phase2L1TGMTFilter : public edm::stream::EDProducer<> {
public:
explicit Phase2L1TGMTFilter(const edm::ParameterSet&);
~Phase2L1TGMTFilter() override;
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
private:
void beginStream(edm::StreamID) override;
void produce(edm::Event&, const edm::EventSetup&) override;
void endStream() override;
edm::EDGetTokenT<std::vector<l1t::TrackerMuon> > srcMuons_;
bool applyLowPtFilter_;
int ptBarrelMin_;
int ptEndcapMin_;
double etaBE_;
};
Phase2L1TGMTFilter::Phase2L1TGMTFilter(const edm::ParameterSet& iConfig)
: srcMuons_(consumes<std::vector<l1t::TrackerMuon> >(iConfig.getParameter<edm::InputTag>("srcMuons"))),
applyLowPtFilter_(iConfig.getParameter<bool>("applyLowPtFilter")),
ptBarrelMin_(iConfig.getParameter<int>("ptBarrelMin")),
ptEndcapMin_(iConfig.getParameter<int>("ptEndcapMin")),
etaBE_(iConfig.getParameter<double>("etaBE")) {
produces<std::vector<l1t::TrackerMuon> >("l1tTkMuonsGmtLowPtFix").setBranchAlias("tkMuLowPtFix");
}
Phase2L1TGMTFilter::~Phase2L1TGMTFilter() {
// do anything here that needs to be done at destruction time
// (e.g. close files, deallocate resources etc.)
}
//
// member functions
//
// ------------ method called to produce the data ------------
void Phase2L1TGMTFilter::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
using namespace edm;
Handle<std::vector<l1t::TrackerMuon> > muonHandle;
iEvent.getByToken(srcMuons_, muonHandle);
std::vector<l1t::TrackerMuon> out;
for (uint i = 0; i < muonHandle->size(); ++i) {
auto mu = muonHandle->at(i);
bool noSAMatch = true;
if (applyLowPtFilter_) {
if ((fabs(mu.phEta()) < etaBE_ && mu.phPt() < ptBarrelMin_) ||
(fabs(mu.phEta()) > etaBE_ && mu.phPt() < ptEndcapMin_)) {
// if quality is already set to 0 don't continue the loop.
for (const auto& r : mu.muonRef()) {
if (r.isNonnull()) {
noSAMatch = false;
break;
}
}
if (noSAMatch)
mu.setHwQual(0);
}
}
out.push_back(mu); // store all muons otherwise
}
// store results
std::unique_ptr<std::vector<l1t::TrackerMuon> > out1 = std::make_unique<std::vector<l1t::TrackerMuon> >(out);
iEvent.put(std::move(out1));
}
// ------------ method called once each stream before processing any runs, lumis or events ------------
void Phase2L1TGMTFilter::beginStream(edm::StreamID) {}
// ------------ method called once each stream after processing all runs, lumis and events ------------
void Phase2L1TGMTFilter::endStream() {}
void Phase2L1TGMTFilter::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.setUnknown();
descriptions.addDefault(desc);
}
//define this as a plug-in
DEFINE_FWK_MODULE(Phase2L1TGMTFilter);