|
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
|
/* \class IsolatedPixelTrackCorrector
*
*
*/
#include <vector>
#include <memory>
#include <algorithm>
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/Common/interface/Ref.h"
#include "DataFormats/Common/interface/TriggerResults.h"
#include "DataFormats/HcalIsolatedTrack/interface/IsolatedPixelTrackCandidate.h"
#include "DataFormats/HcalIsolatedTrack/interface/IsolatedPixelTrackCandidateFwd.h"
#include "DataFormats/HLTReco/interface/TriggerFilterObjectWithRefs.h"
#include "DataFormats/Math/interface/deltaR.h"
#include "DataFormats/TrackReco/interface/Track.h"
// Framework
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/Utilities/interface/Exception.h"
//
///
// Math
#include "Math/GenVector/VectorUtil.h"
#include "Math/GenVector/PxPyPzE4D.h"
class IPTCorrector : public edm::global::EDProducer<> {
public:
IPTCorrector(const edm::ParameterSet& ps);
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
void produce(edm::StreamID, edm::Event&, edm::EventSetup const&) const override;
private:
const edm::EDGetTokenT<reco::TrackCollection> tok_cor_;
const edm::EDGetTokenT<trigger::TriggerFilterObjectWithRefs> tok_uncor_;
const double assocCone_;
};
IPTCorrector::IPTCorrector(const edm::ParameterSet& config)
: tok_cor_(consumes<reco::TrackCollection>(config.getParameter<edm::InputTag>("corTracksLabel"))),
tok_uncor_(consumes<trigger::TriggerFilterObjectWithRefs>(config.getParameter<edm::InputTag>("filterLabel"))),
assocCone_(config.getParameter<double>("associationCone")) {
// register the product
produces<reco::IsolatedPixelTrackCandidateCollection>();
}
void IPTCorrector::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.add<edm::InputTag>("corTracksLabel", edm::InputTag("hltIter0PFlowCtfWithMaterialTracks"));
desc.add<edm::InputTag>("filterLabel", edm::InputTag("hltIsolPixelTrackL2Filter"));
desc.add<double>("associationCone", 0.2);
descriptions.add("iptCorrector", desc);
}
void IPTCorrector::produce(edm::StreamID, edm::Event& theEvent, edm::EventSetup const&) const {
auto trackCollection = std::make_unique<reco::IsolatedPixelTrackCandidateCollection>();
const edm::Handle<reco::TrackCollection>& corTracks = theEvent.getHandle(tok_cor_);
const edm::Handle<trigger::TriggerFilterObjectWithRefs>& fiCand = theEvent.getHandle(tok_uncor_);
std::vector<edm::Ref<reco::IsolatedPixelTrackCandidateCollection> > isoPixTrackRefs;
fiCand->getObjects(trigger::TriggerTrack, isoPixTrackRefs);
int nCand = isoPixTrackRefs.size();
//loop over input ipt
for (int p = 0; p < nCand; p++) {
double iptEta = isoPixTrackRefs[p]->track()->eta();
double iptPhi = isoPixTrackRefs[p]->track()->phi();
int ntrk = 0;
double minDR = 100;
reco::TrackCollection::const_iterator citSel;
for (reco::TrackCollection::const_iterator cit = corTracks->begin(); cit != corTracks->end(); cit++) {
double dR = deltaR(cit->eta(), cit->phi(), iptEta, iptPhi);
if (dR < minDR && dR < assocCone_) {
minDR = dR;
ntrk++;
citSel = cit;
}
}
if (ntrk > 0) {
reco::IsolatedPixelTrackCandidate newCandidate(reco::TrackRef(corTracks, citSel - corTracks->begin()),
isoPixTrackRefs[p]->l1tau(),
isoPixTrackRefs[p]->maxPtPxl(),
isoPixTrackRefs[p]->sumPtPxl());
newCandidate.setEnergyIn(isoPixTrackRefs[p]->energyIn());
newCandidate.setEnergyOut(isoPixTrackRefs[p]->energyOut());
newCandidate.setNHitIn(isoPixTrackRefs[p]->nHitIn());
newCandidate.setNHitOut(isoPixTrackRefs[p]->nHitOut());
if (isoPixTrackRefs[p]->etaPhiEcalValid()) {
std::pair<double, double> etaphi = (isoPixTrackRefs[p]->etaPhiEcal());
newCandidate.setEtaPhiEcal(etaphi.first, etaphi.second);
}
trackCollection->push_back(newCandidate);
}
}
// put the product in the event
theEvent.put(std::move(trackCollection));
}
#include "FWCore/PluginManager/interface/ModuleDef.h"
#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(IPTCorrector);
|