ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitPhysics/Utils/src/MetTools.cc
Revision: 1.13
Committed: Sat Jul 7 11:33:59 2012 UTC (12 years, 9 months ago) by ceballos
Content type: text/plain
Branch: MAIN
CVS Tags: Mit_029c, Mit_029b, Mit_029a, HEAD
Changes since 1.12: +23 -8 lines
Log Message: 
new pmet

File Contents

# User Rev Content
1 ceballos 1.13 // $Id: MetTools.cc,v 1.12 2012/04/28 19:10:01 ceballos Exp $
2 mzanetti 1.1
3     #include "MitPhysics/Utils/interface/MetTools.h"
4     #include <TFile.h>
5    
6     ClassImp(mithep::MetTools)
7    
8     using namespace mithep;
9    
10     //--------------------------------------------------------------------------------------------------
11     MetTools::MetTools(const MuonCol *fMuons, const PFCandidateCol *fPFCandidates, const Vertex *fVertex,
12     float deltaZCut, float ptCut, float etaCut) {
13    
14     float trackNumeratorX =0, trackNumeratorY =0;
15     float neutralNumeratorX=0, neutralNumeratorY=0;
16 ceballos 1.12 float CHSNumeratorX =0, CHSNumeratorY =0, NHSNumeratorX =0, NHSNumeratorY =0;
17 mzanetti 1.1
18     // muons Pt
19     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
20     trackNumeratorX -= fMuons->At(m)->Px();
21     trackNumeratorY -= fMuons->At(m)->Py();
22     }
23    
24     // PF candidates pT
25     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
26    
27 ceballos 1.12 // CHS computation
28     if (fPFCandidates->At(i)->HasTrk()){
29     CHSNumeratorX -= fPFCandidates->At(i)->Px();
30     CHSNumeratorY -= fPFCandidates->At(i)->Py();
31     }
32     // NHS computation
33     if (fPFCandidates->At(i)->HasTrk() ||
34     (fPFCandidates->At(i)->AbsEta() < 3.0 && fPFCandidates->At(i)->Pt() > 4.0)){
35     NHSNumeratorX -= fPFCandidates->At(i)->Px();
36     NHSNumeratorY -= fPFCandidates->At(i)->Py();
37     }
38    
39 mzanetti 1.1 // charged
40 mzanetti 1.3 if (fPFCandidates->At(i)->HasTrackerTrk()){
41 mzanetti 1.1
42     bool isMuonTrack = false;
43     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
44     if (fMuons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) {
45     isMuonTrack = true;
46     break;
47     }
48     }
49     if (isMuonTrack) continue;
50 ceballos 1.7
51 mzanetti 1.1 if (fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) {
52     trackNumeratorX -= fPFCandidates->At(i)->Px();
53     trackNumeratorY -= fPFCandidates->At(i)->Py();
54 sixie 1.10 fRecoil += fPFCandidates->At(i)->Mom();
55     fChargedRecoil += fPFCandidates->At(i)->Mom();
56 mzanetti 1.1 }
57     }
58    
59     // neutral
60     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
61     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
62     neutralNumeratorX -= fPFCandidates->At(i)->Px();
63     neutralNumeratorY -= fPFCandidates->At(i)->Py();
64 sixie 1.10 fRecoil += fPFCandidates->At(i)->Mom();
65 mzanetti 1.1 }
66     }
67     }
68    
69 mzanetti 1.3 fCorrectedMet = mithep::Met(trackNumeratorX+neutralNumeratorX, trackNumeratorY+neutralNumeratorY);
70     fCorrectedTrackMet = mithep::Met(trackNumeratorX, trackNumeratorY);
71 ceballos 1.12 fCHSMet = mithep::Met(CHSNumeratorX, CHSNumeratorY);
72     fNHSMet = mithep::Met(NHSNumeratorX, NHSNumeratorY);
73 mzanetti 1.1 }
74    
75     MetTools::MetTools(const ElectronCol *fElectrons, const PFCandidateCol *fPFCandidates, const Vertex *fVertex,
76     float deltaZCut, float ptCut, float etaCut) {
77    
78     float trackNumeratorX =0, trackNumeratorY =0;
79     float neutralNumeratorX=0, neutralNumeratorY=0;
80 ceballos 1.12 float CHSNumeratorX =0, CHSNumeratorY =0, NHSNumeratorX =0, NHSNumeratorY =0;
81 mzanetti 1.1
82     // electrons Pt
83     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
84     trackNumeratorX -= fElectrons->At(m)->Px();
85     trackNumeratorY -= fElectrons->At(m)->Py();
86     }
87    
88     // PF candidates pT
89     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
90    
91 ceballos 1.12 // CHS computation
92     if (fPFCandidates->At(i)->HasTrk()){
93     CHSNumeratorX -= fPFCandidates->At(i)->Px();
94     CHSNumeratorY -= fPFCandidates->At(i)->Py();
95     }
96     // NHS computation
97     if (fPFCandidates->At(i)->HasTrk() ||
98     (fPFCandidates->At(i)->AbsEta() < 3.0 && fPFCandidates->At(i)->Pt() > 4.0)){
99     NHSNumeratorX -= fPFCandidates->At(i)->Px();
100     NHSNumeratorY -= fPFCandidates->At(i)->Py();
101     }
102    
103 mzanetti 1.1 // charged
104 ceballos 1.6 if (fPFCandidates->At(i)->HasTrackerTrk() || fPFCandidates->At(i)->HasGsfTrk()){
105 mzanetti 1.1 bool isElectronTrack = false;
106     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
107 mzanetti 1.3 if ( (fElectrons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) or
108     (fElectrons->At(m)->HasGsfTrk() and fElectrons->At(m)->GsfTrk() == fPFCandidates->At(i)->GsfTrk()) ) {
109 mzanetti 1.1 isElectronTrack = true;
110     break;
111     }
112     }
113     if (isElectronTrack) continue;
114    
115 ceballos 1.6 if ((fPFCandidates->At(i)->HasTrackerTrk() && fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) ||
116     (fPFCandidates->At(i)->HasGsfTrk() && fabs(fPFCandidates->At(i)->GsfTrk()->DzCorrected(*fVertex) ) < deltaZCut)) {
117 mzanetti 1.1 trackNumeratorX -= fPFCandidates->At(i)->Px();
118     trackNumeratorY -= fPFCandidates->At(i)->Py();
119 sixie 1.10 fRecoil += fPFCandidates->At(i)->Mom();
120     fChargedRecoil += fPFCandidates->At(i)->Mom();
121 mzanetti 1.1 }
122     }
123    
124     // neutral
125     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
126     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
127     neutralNumeratorX -= fPFCandidates->At(i)->Px();
128     neutralNumeratorY -= fPFCandidates->At(i)->Py();
129 sixie 1.10 fRecoil += fPFCandidates->At(i)->Mom();
130 mzanetti 1.1 }
131     }
132     }
133 mzanetti 1.3
134     fCorrectedMet = mithep::Met(trackNumeratorX+neutralNumeratorX, trackNumeratorY+neutralNumeratorY);
135     fCorrectedTrackMet = mithep::Met(trackNumeratorX, trackNumeratorY);
136 ceballos 1.12 fCHSMet = mithep::Met(CHSNumeratorX, CHSNumeratorY);
137     fNHSMet = mithep::Met(NHSNumeratorX, NHSNumeratorY);
138 ceballos 1.2 }
139    
140 mzanetti 1.4
141     //--------------------------------------------------------------------------------------------------
142     MetTools::MetTools(const MuonCol *fMuons, const PFCandidateCol *fPFCandidates, const PFJetCol *pfJets, const Vertex *fVertex,
143     float deltaZCut, float ptCut, float etaCut) {
144    
145     float trackNumeratorX =0, trackNumeratorY =0;
146     float neutralNumeratorX=0, neutralNumeratorY=0;
147 ceballos 1.12 float CHSNumeratorX =0, CHSNumeratorY =0, NHSNumeratorX =0, NHSNumeratorY =0;
148 mzanetti 1.4
149     // muons Pt
150     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
151     trackNumeratorX -= fMuons->At(m)->Px();
152     trackNumeratorY -= fMuons->At(m)->Py();
153     }
154    
155     // jets Pt
156     for (UInt_t j = 0; j < pfJets->GetEntries(); ++j) {
157     trackNumeratorX -= pfJets->At(j)->Px();
158     trackNumeratorY -= pfJets->At(j)->Py();
159     }
160    
161     // PF candidates pT
162     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
163    
164 ceballos 1.12 // CHS computation
165     if (fPFCandidates->At(i)->HasTrk()){
166     CHSNumeratorX -= fPFCandidates->At(i)->Px();
167     CHSNumeratorY -= fPFCandidates->At(i)->Py();
168     }
169     // NHS computation
170     if (fPFCandidates->At(i)->HasTrk() ||
171     (fPFCandidates->At(i)->AbsEta() < 3.0 && fPFCandidates->At(i)->Pt() > 4.0)){
172     NHSNumeratorX -= fPFCandidates->At(i)->Px();
173     NHSNumeratorY -= fPFCandidates->At(i)->Py();
174     }
175    
176 mzanetti 1.4 // jets
177     bool inTheJet = false;
178     for (UInt_t j = 0; j < pfJets->GetEntries(); ++j) {
179 mzanetti 1.5 for (UInt_t c=0;c<pfJets->At(j)->NPFCands();++c){
180     if (pfJets->At(j)->PFCand(c) == fPFCandidates->At(i)) {
181 mzanetti 1.4 inTheJet=true;
182     break;
183     }
184     }
185     if (inTheJet) break;
186     }
187     if (inTheJet) continue;
188    
189     // charged
190     if (fPFCandidates->At(i)->HasTrackerTrk()){
191    
192     bool isMuonTrack = false;
193     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
194     if (fMuons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) {
195     isMuonTrack = true;
196     break;
197     }
198     }
199     if (isMuonTrack) continue;
200    
201     if (fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) {
202     trackNumeratorX -= fPFCandidates->At(i)->Px();
203     trackNumeratorY -= fPFCandidates->At(i)->Py();
204 sixie 1.10 fRecoil += fPFCandidates->At(i)->Mom();
205     fChargedRecoil += fPFCandidates->At(i)->Mom();
206 mzanetti 1.4 }
207     }
208    
209     // neutral
210     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
211     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
212     neutralNumeratorX -= fPFCandidates->At(i)->Px();
213     neutralNumeratorY -= fPFCandidates->At(i)->Py();
214 sixie 1.10 fRecoil += fPFCandidates->At(i)->Mom();
215 mzanetti 1.4 }
216     }
217     }
218    
219     fCorrectedMet = mithep::Met(trackNumeratorX+neutralNumeratorX, trackNumeratorY+neutralNumeratorY);
220     fCorrectedTrackMet = mithep::Met(trackNumeratorX, trackNumeratorY);
221 ceballos 1.12 fCHSMet = mithep::Met(CHSNumeratorX, CHSNumeratorY);
222     fNHSMet = mithep::Met(NHSNumeratorX, NHSNumeratorY);
223 mzanetti 1.4 }
224    
225    
226     MetTools::MetTools(const ElectronCol *fElectrons, const PFCandidateCol *fPFCandidates, const PFJetCol *pfJets, const Vertex *fVertex,
227     float deltaZCut, float ptCut, float etaCut) {
228    
229     float trackNumeratorX =0, trackNumeratorY =0;
230     float neutralNumeratorX=0, neutralNumeratorY=0;
231 ceballos 1.12 float CHSNumeratorX =0, CHSNumeratorY =0, NHSNumeratorX =0, NHSNumeratorY =0;
232 mzanetti 1.4
233     // electrons Pt
234     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
235     trackNumeratorX -= fElectrons->At(m)->Px();
236     trackNumeratorY -= fElectrons->At(m)->Py();
237     }
238    
239     // jets Pt
240     for (UInt_t j = 0; j < pfJets->GetEntries(); ++j) {
241     trackNumeratorX -= pfJets->At(j)->Px();
242     trackNumeratorY -= pfJets->At(j)->Py();
243     }
244    
245     // PF candidates pT
246     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
247    
248 ceballos 1.12 // CHS computation
249     if (fPFCandidates->At(i)->HasTrk()){
250     CHSNumeratorX -= fPFCandidates->At(i)->Px();
251     CHSNumeratorY -= fPFCandidates->At(i)->Py();
252     }
253     // NHS computation
254     if (fPFCandidates->At(i)->HasTrk() ||
255     (fPFCandidates->At(i)->AbsEta() < 3.0 && fPFCandidates->At(i)->Pt() > 4.0)){
256     NHSNumeratorX -= fPFCandidates->At(i)->Px();
257     NHSNumeratorY -= fPFCandidates->At(i)->Py();
258     }
259    
260 mzanetti 1.4 // jets
261     bool inTheJet = false;
262     for (UInt_t j = 0; j < pfJets->GetEntries(); ++j) {
263 mzanetti 1.5 for (UInt_t c=0;c<pfJets->At(j)->NPFCands();++c){
264     if (pfJets->At(j)->PFCand(c) == fPFCandidates->At(i)) {
265 mzanetti 1.4 inTheJet=true;
266     break;
267     }
268     }
269     if (inTheJet) break;
270     }
271     if (inTheJet) continue;
272    
273     // charged
274 ceballos 1.6 if (fPFCandidates->At(i)->HasTrackerTrk() || fPFCandidates->At(i)->HasGsfTrk()){
275 mzanetti 1.4 bool isElectronTrack = false;
276     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
277 sixie 1.8 if ( (fElectrons->At(m)->HasTrackerTrk() and fElectrons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) or
278 mzanetti 1.4 (fElectrons->At(m)->HasGsfTrk() and fElectrons->At(m)->GsfTrk() == fPFCandidates->At(i)->GsfTrk()) ) {
279     isElectronTrack = true;
280     break;
281     }
282     }
283     if (isElectronTrack) continue;
284    
285 ceballos 1.6 if ((fPFCandidates->At(i)->HasTrackerTrk() && fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) ||
286     (fPFCandidates->At(i)->HasGsfTrk() && fabs(fPFCandidates->At(i)->GsfTrk()->DzCorrected(*fVertex) ) < deltaZCut)) {
287 mzanetti 1.4 trackNumeratorX -= fPFCandidates->At(i)->Px();
288     trackNumeratorY -= fPFCandidates->At(i)->Py();
289 sixie 1.10 fRecoil += fPFCandidates->At(i)->Mom();
290     fChargedRecoil += fPFCandidates->At(i)->Mom();
291 mzanetti 1.4 }
292     }
293    
294     // neutral
295     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
296     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
297     neutralNumeratorX -= fPFCandidates->At(i)->Px();
298     neutralNumeratorY -= fPFCandidates->At(i)->Py();
299 sixie 1.10 fRecoil += fPFCandidates->At(i)->Mom();
300 mzanetti 1.4 }
301     }
302     }
303    
304     fCorrectedMet = mithep::Met(trackNumeratorX+neutralNumeratorX, trackNumeratorY+neutralNumeratorY);
305     fCorrectedTrackMet = mithep::Met(trackNumeratorX, trackNumeratorY);
306 ceballos 1.12 fCHSMet = mithep::Met(CHSNumeratorX, CHSNumeratorY);
307     fNHSMet = mithep::Met(NHSNumeratorX, NHSNumeratorY);
308 mzanetti 1.4 }
309    
310    
311 sixie 1.10 void MetTools::AddToCorrectedTrackMet( const Particle *p, bool debug ) {
312 phedex 1.9 float MetX = fCorrectedTrackMet.Mex();
313     float MetY = fCorrectedTrackMet.Mey();
314 sixie 1.10
315     if (debug) std::cout << "AddToCorrectedTrackMet:\n";
316     if (debug) std::cout << "Before: " << MetX << " " << MetY << std::endl;
317    
318 phedex 1.9 MetX -= p->Px();
319     MetY -= p->Py();
320    
321 sixie 1.10 if (debug) std::cout << "Add : " << p->Px() << " " << p->Py() << std::endl;
322     if (debug) std::cout << "After : " << MetX << " " << MetY << std::endl;
323    
324 phedex 1.9 fCorrectedTrackMet.SetMex(MetX);
325     fCorrectedTrackMet.SetMey(MetY);
326    
327     return;
328     }
329    
330     void MetTools::AddToCorrectedMet( const Particle *p) {
331     float MetX=fCorrectedMet.Mex();
332     float MetY=fCorrectedMet.Mey();
333    
334     MetX -= p->Px();
335     MetY -= p->Py();
336    
337     fCorrectedMet.SetMex(MetX);
338     fCorrectedMet.SetMey(MetY);
339    
340     return;
341     }
342    
343 sixie 1.10 void MetTools::AddToRecoil( const Particle *p) {
344    
345     if (p->Charge() != 0) {
346     fChargedRecoil += p->Mom();
347     }
348     fRecoil += p->Mom();
349    
350     return;
351     }
352    
353    
354    
355     void MetTools::RemoveParticleInIsoConeFromTrackMet( const Particle *p, const PFCandidateCol *fPFCandidates, const Vertex *fVertex, float deltaZCut , float deltaR, bool debug ) {
356     float MetX = fCorrectedTrackMet.Mex();
357     float MetY = fCorrectedTrackMet.Mey();
358    
359     if (debug) std::cout << "RemoveParticleInIsoConeFromTrackMet:\n";
360     if (debug) std::cout << "Before: " << MetX << " " << MetY << std::endl;
361    
362     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
363     //charged
364     if (fPFCandidates->At(i)->HasTrackerTrk() || fPFCandidates->At(i)->HasGsfTrk()) {
365     //passes dZ cut
366     if ((fPFCandidates->At(i)->HasTrackerTrk() && fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) ||
367     (fPFCandidates->At(i)->HasGsfTrk() && fabs(fPFCandidates->At(i)->GsfTrk()->DzCorrected(*fVertex) ) < deltaZCut)) {
368     //inside cone
369     if (MathUtils::DeltaR(fPFCandidates->At(i)->Mom(), p->Mom()) < deltaR ) {
370     MetX += fPFCandidates->At(i)->Px();
371     MetY += fPFCandidates->At(i)->Py();
372     if (debug) std::cout << "Subtract : " << fPFCandidates->At(i)->Px() << " " << fPFCandidates->At(i)->Py() << " : " << std::endl;
373     }
374     }
375     }
376     }
377    
378     if (debug) std::cout << "After : " << MetX << " " << MetY << std::endl;
379    
380     fCorrectedTrackMet.SetMex(MetX);
381     fCorrectedTrackMet.SetMey(MetY);
382    
383     return;
384     }
385    
386    
387     void MetTools::RemoveParticleInIsoConeFromCorrectedMet( const Particle *p, const PFCandidateCol *fPFCandidates, const Vertex *fVertex, float deltaZCut, float ptCut, float etaCut , float deltaR) {
388     float MetX = fCorrectedMet.Mex();
389     float MetY = fCorrectedMet.Mey();
390    
391     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
392    
393     //inside cone
394     if (MathUtils::DeltaR(fPFCandidates->At(i)->Mom(), p->Mom()) < deltaR ) {
395    
396     //charged
397     if (fPFCandidates->At(i)->HasTrackerTrk() || fPFCandidates->At(i)->HasGsfTrk()) {
398     //passes dZ cut
399     if ((fPFCandidates->At(i)->HasTrackerTrk() && fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) ||
400     (fPFCandidates->At(i)->HasGsfTrk() && fabs(fPFCandidates->At(i)->GsfTrk()->DzCorrected(*fVertex) ) < deltaZCut)) {
401     MetX += p->Px();
402     MetY += p->Py();
403     }
404     }
405    
406     //neutrals
407     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
408     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
409     MetX += fPFCandidates->At(i)->Px();
410     MetY += fPFCandidates->At(i)->Py();
411     }
412     }
413     }
414    
415     }
416    
417     fCorrectedMet.SetMex(MetX);
418     fCorrectedMet.SetMey(MetY);
419    
420     return;
421     }
422    
423    
424     void MetTools::RemoveParticleInIsoConeFromRecoil( const Particle *p, const PFCandidateCol *fPFCandidates, const Vertex *fVertex, float deltaZCut, float ptCut, float etaCut , float deltaR) {
425    
426     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
427    
428     //inside cone
429     if (MathUtils::DeltaR(fPFCandidates->At(i)->Mom(), p->Mom()) < deltaR ) {
430    
431     //charged
432     if (fPFCandidates->At(i)->HasTrackerTrk() || fPFCandidates->At(i)->HasGsfTrk()) {
433     //passes dZ cut
434     if ((fPFCandidates->At(i)->HasTrackerTrk() && fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) ||
435     (fPFCandidates->At(i)->HasGsfTrk() && fabs(fPFCandidates->At(i)->GsfTrk()->DzCorrected(*fVertex) ) < deltaZCut)) {
436     fChargedRecoil -= fPFCandidates->At(i)->Mom();
437     fRecoil -= fPFCandidates->At(i)->Mom();
438     }
439     }
440    
441     //neutrals
442     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
443     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
444     fRecoil -= fPFCandidates->At(i)->Mom();
445     }
446     }
447     }
448     }
449    
450     return;
451     }
452    
453 phedex 1.9
454 ceballos 1.2 MetTools::MetTools(const MuonCol *fMuons, const ElectronCol *fElectrons, const PFCandidateCol *fPFCandidates,
455 ceballos 1.13 const Vertex *fVertex, float deltaZCut, float ptCut, float etaCut, float intRadius,
456     const GenericParticle *genP) {
457 mzanetti 1.1
458 ceballos 1.2 float trackNumeratorX =0, trackNumeratorY =0;
459     float neutralNumeratorX=0, neutralNumeratorY=0;
460 ceballos 1.12 float CHSNumeratorX =0, CHSNumeratorY =0, NHSNumeratorX =0, NHSNumeratorY =0;
461 ceballos 1.2
462 ceballos 1.13 float trackNumeratorGenPX =0, trackNumeratorGenPY =0;
463     // added to the track quantities
464     if (genP) {
465     trackNumeratorGenPX -= genP->Px();
466     trackNumeratorGenPY -= genP->Py();
467     }
468    
469 ceballos 1.2 // muons Pt
470     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
471     trackNumeratorX -= fMuons->At(m)->Px();
472     trackNumeratorY -= fMuons->At(m)->Py();
473     }
474    
475     // electrons Pt
476     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
477     trackNumeratorX -= fElectrons->At(m)->Px();
478     trackNumeratorY -= fElectrons->At(m)->Py();
479     }
480    
481     // PF candidates pT
482     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
483    
484 ceballos 1.12 // CHS computation
485     if (fPFCandidates->At(i)->HasTrk()){
486     CHSNumeratorX -= fPFCandidates->At(i)->Px();
487     CHSNumeratorY -= fPFCandidates->At(i)->Py();
488     }
489     // NHS computation
490     if (fPFCandidates->At(i)->HasTrk() ||
491     (fPFCandidates->At(i)->AbsEta() < 3.0 && fPFCandidates->At(i)->Pt() > 4.0)){
492     NHSNumeratorX -= fPFCandidates->At(i)->Px();
493     NHSNumeratorY -= fPFCandidates->At(i)->Py();
494     }
495    
496 ceballos 1.2 // charged
497 ceballos 1.6 if (fPFCandidates->At(i)->HasTrackerTrk() || fPFCandidates->At(i)->HasGsfTrk()){
498 ceballos 1.2 bool isMuonTrack = false;
499     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
500     if (fMuons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) {
501     isMuonTrack = true;
502     break;
503     }
504 ceballos 1.11 if (intRadius > 0.0 && MathUtils::DeltaR(fPFCandidates->At(i)->Mom(), fMuons->At(m)->Mom()) < intRadius) {
505     isMuonTrack = true;
506     break;
507     }
508     }
509 ceballos 1.2 if (isMuonTrack) continue;
510    
511     bool isElectronTrack = false;
512     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
513 sixie 1.8 if ( (fElectrons->At(m)->HasTrackerTrk() and fElectrons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) or
514 mzanetti 1.3 (fElectrons->At(m)->HasGsfTrk() and fElectrons->At(m)->GsfTrk() == fPFCandidates->At(i)->GsfTrk()) ) {
515 ceballos 1.2 isElectronTrack = true;
516     break;
517     }
518 ceballos 1.11 if (intRadius > 0.0 && MathUtils::DeltaR(fPFCandidates->At(i)->Mom(), fElectrons->At(m)->Mom()) < intRadius) {
519     isElectronTrack = true;
520     break;
521     }
522     }
523 ceballos 1.2 if (isElectronTrack) continue;
524 mzanetti 1.1
525 ceballos 1.6 if ((fPFCandidates->At(i)->HasTrackerTrk() && fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) ||
526     (fPFCandidates->At(i)->HasGsfTrk() && fabs(fPFCandidates->At(i)->GsfTrk()->DzCorrected(*fVertex) ) < deltaZCut)) {
527 ceballos 1.2 trackNumeratorX -= fPFCandidates->At(i)->Px();
528     trackNumeratorY -= fPFCandidates->At(i)->Py();
529 sixie 1.10 fRecoil += fPFCandidates->At(i)->Mom();
530     fChargedRecoil += fPFCandidates->At(i)->Mom();
531 ceballos 1.2 }
532     }
533 mzanetti 1.1
534 ceballos 1.2 // neutral
535     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
536     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
537     neutralNumeratorX -= fPFCandidates->At(i)->Px();
538     neutralNumeratorY -= fPFCandidates->At(i)->Py();
539 sixie 1.10 fRecoil += fPFCandidates->At(i)->Mom();
540 ceballos 1.2 }
541     }
542     }
543 mzanetti 1.3 fCorrectedMet = mithep::Met(trackNumeratorX+neutralNumeratorX, trackNumeratorY+neutralNumeratorY);
544 ceballos 1.13 fCorrectedTrackMet = mithep::Met(trackNumeratorX+trackNumeratorGenPX, trackNumeratorY+trackNumeratorGenPY);
545     fCHSMet = mithep::Met(CHSNumeratorX+trackNumeratorGenPX, CHSNumeratorY+trackNumeratorGenPY);
546 ceballos 1.12 fNHSMet = mithep::Met(NHSNumeratorX, NHSNumeratorY);
547 ceballos 1.2 }
548 mzanetti 1.1
549 mzanetti 1.4
550     MetTools::MetTools(const MuonCol *fMuons, const ElectronCol *fElectrons, const PFCandidateCol *fPFCandidates,
551     const PFJetCol *pfJets,
552 ceballos 1.13 const Vertex *fVertex, float deltaZCut, float ptCut, float etaCut, float intRadius,
553     const GenericParticle *genP) {
554 mzanetti 1.4
555     float trackNumeratorX =0, trackNumeratorY =0;
556     float neutralNumeratorX=0, neutralNumeratorY=0;
557 ceballos 1.12 float CHSNumeratorX =0, CHSNumeratorY =0, NHSNumeratorX =0, NHSNumeratorY =0;
558 mzanetti 1.4
559 ceballos 1.13 float trackNumeratorGenPX =0, trackNumeratorGenPY =0;
560     // added to the track quantities
561     if (genP) {
562     trackNumeratorGenPX -= genP->Px();
563     trackNumeratorGenPY -= genP->Py();
564     }
565    
566 mzanetti 1.4 // muons Pt
567     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
568     trackNumeratorX -= fMuons->At(m)->Px();
569     trackNumeratorY -= fMuons->At(m)->Py();
570     }
571    
572     // electrons Pt
573     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
574     trackNumeratorX -= fElectrons->At(m)->Px();
575     trackNumeratorY -= fElectrons->At(m)->Py();
576     }
577    
578     // jets Pt
579     for (UInt_t j = 0; j < pfJets->GetEntries(); ++j) {
580     trackNumeratorX -= pfJets->At(j)->Px();
581     trackNumeratorY -= pfJets->At(j)->Py();
582     }
583    
584     // PF candidates pT
585     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
586    
587 ceballos 1.12 // CHS computation
588     if (fPFCandidates->At(i)->HasTrk()){
589     CHSNumeratorX -= fPFCandidates->At(i)->Px();
590     CHSNumeratorY -= fPFCandidates->At(i)->Py();
591     }
592     // NHS computation
593     if (fPFCandidates->At(i)->HasTrk() ||
594     (fPFCandidates->At(i)->AbsEta() < 3.0 && fPFCandidates->At(i)->Pt() > 4.0)){
595     NHSNumeratorX -= fPFCandidates->At(i)->Px();
596     NHSNumeratorY -= fPFCandidates->At(i)->Py();
597     }
598    
599 mzanetti 1.4 // jets
600     bool inTheJet = false;
601     for (UInt_t j = 0; j < pfJets->GetEntries(); ++j) {
602 mzanetti 1.5 for (UInt_t c=0;c<pfJets->At(j)->NPFCands();++c){
603     if (pfJets->At(j)->PFCand(c) == fPFCandidates->At(i)) {
604 mzanetti 1.4 inTheJet=true;
605     break;
606     }
607     }
608     if (inTheJet) break;
609     }
610     if (inTheJet) continue;
611    
612     // charged
613 ceballos 1.6 if (fPFCandidates->At(i)->HasTrackerTrk() || fPFCandidates->At(i)->HasGsfTrk()){
614 mzanetti 1.4 bool isMuonTrack = false;
615     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
616     if (fMuons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) {
617     isMuonTrack = true;
618     break;
619     }
620 ceballos 1.11 if (intRadius > 0.0 && MathUtils::DeltaR(fPFCandidates->At(i)->Mom(), fMuons->At(m)->Mom()) < intRadius) {
621     isMuonTrack = true;
622     break;
623     }
624 mzanetti 1.4 }
625     if (isMuonTrack) continue;
626    
627     bool isElectronTrack = false;
628     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
629 sixie 1.8 if ( (fElectrons->At(m)->HasTrackerTrk() and fElectrons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) or
630 mzanetti 1.4 (fElectrons->At(m)->HasGsfTrk() and fElectrons->At(m)->GsfTrk() == fPFCandidates->At(i)->GsfTrk()) ) {
631     isElectronTrack = true;
632     break;
633     }
634 ceballos 1.11 if (intRadius > 0.0 && MathUtils::DeltaR(fPFCandidates->At(i)->Mom(), fElectrons->At(m)->Mom()) < intRadius) {
635     isElectronTrack = true;
636     break;
637     }
638 mzanetti 1.4 }
639     if (isElectronTrack) continue;
640    
641 ceballos 1.6 if ((fPFCandidates->At(i)->HasTrackerTrk() && fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) ||
642     (fPFCandidates->At(i)->HasGsfTrk() && fabs(fPFCandidates->At(i)->GsfTrk()->DzCorrected(*fVertex) ) < deltaZCut)) {
643 mzanetti 1.4 trackNumeratorX -= fPFCandidates->At(i)->Px();
644     trackNumeratorY -= fPFCandidates->At(i)->Py();
645 sixie 1.10 fRecoil += fPFCandidates->At(i)->Mom();
646     fChargedRecoil += fPFCandidates->At(i)->Mom();
647 mzanetti 1.4 }
648     }
649    
650     // neutral
651     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
652     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
653     neutralNumeratorX -= fPFCandidates->At(i)->Px();
654     neutralNumeratorY -= fPFCandidates->At(i)->Py();
655 sixie 1.10 fRecoil += fPFCandidates->At(i)->Mom();
656 mzanetti 1.4 }
657     }
658     }
659     fCorrectedMet = mithep::Met(trackNumeratorX+neutralNumeratorX, trackNumeratorY+neutralNumeratorY);
660 ceballos 1.13 fCorrectedTrackMet = mithep::Met(trackNumeratorX+trackNumeratorGenPX, trackNumeratorY+trackNumeratorGenPY);
661     fCHSMet = mithep::Met(CHSNumeratorX+trackNumeratorGenPX, CHSNumeratorY+trackNumeratorGenPY);
662 ceballos 1.12 fNHSMet = mithep::Met(NHSNumeratorX, NHSNumeratorY);
663 mzanetti 1.4 }
664    
665    
666 mzanetti 1.3 Met MetTools::GetMinimumMet(const Met *UncorrectedMet) {
667 mzanetti 1.1
668     return (fCorrectedMet.Pt() < UncorrectedMet->Pt()) ? fCorrectedMet : *UncorrectedMet;
669     }
670    
671 mzanetti 1.3 Met MetTools::GetMinimumTrackMet(const Met *UncorrectedMet) {
672 ceballos 1.2
673     return (fCorrectedTrackMet.Pt() < UncorrectedMet->Pt()) ? fCorrectedTrackMet : *UncorrectedMet;
674     }
675 mzanetti 1.1