ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitPhysics/Utils/src/MetTools.cc
Revision: 1.7
Committed: Fri May 13 08:27:16 2011 UTC (13 years, 11 months ago) by ceballos
Content type: text/plain
Branch: MAIN
CVS Tags: Mit_023, Mit_022a, Mit_022, Mit_021, Mit_021pre2
Changes since 1.6: +2 -2 lines
Log Message:
fixed in isolation

File Contents

# User Rev Content
1 ceballos 1.7 // $Id: MetTools.cc,v 1.6 2011/04/25 21:24:53 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    
17     // muons Pt
18     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
19     trackNumeratorX -= fMuons->At(m)->Px();
20     trackNumeratorY -= fMuons->At(m)->Py();
21     }
22    
23     // PF candidates pT
24     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
25    
26     // charged
27 mzanetti 1.3 if (fPFCandidates->At(i)->HasTrackerTrk()){
28 mzanetti 1.1
29     bool isMuonTrack = false;
30     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
31     if (fMuons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) {
32     isMuonTrack = true;
33     break;
34     }
35     }
36     if (isMuonTrack) continue;
37 ceballos 1.7
38 mzanetti 1.1 if (fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) {
39     trackNumeratorX -= fPFCandidates->At(i)->Px();
40     trackNumeratorY -= fPFCandidates->At(i)->Py();
41     }
42     }
43    
44     // neutral
45     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
46     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
47     neutralNumeratorX -= fPFCandidates->At(i)->Px();
48     neutralNumeratorY -= fPFCandidates->At(i)->Py();
49     }
50     }
51     }
52    
53 mzanetti 1.3 fCorrectedMet = mithep::Met(trackNumeratorX+neutralNumeratorX, trackNumeratorY+neutralNumeratorY);
54     fCorrectedTrackMet = mithep::Met(trackNumeratorX, trackNumeratorY);
55 mzanetti 1.1 }
56    
57     MetTools::MetTools(const ElectronCol *fElectrons, const PFCandidateCol *fPFCandidates, const Vertex *fVertex,
58     float deltaZCut, float ptCut, float etaCut) {
59    
60     float trackNumeratorX =0, trackNumeratorY =0;
61     float neutralNumeratorX=0, neutralNumeratorY=0;
62    
63     // electrons Pt
64     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
65     trackNumeratorX -= fElectrons->At(m)->Px();
66     trackNumeratorY -= fElectrons->At(m)->Py();
67     }
68    
69     // PF candidates pT
70     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
71    
72     // charged
73 ceballos 1.6 if (fPFCandidates->At(i)->HasTrackerTrk() || fPFCandidates->At(i)->HasGsfTrk()){
74 mzanetti 1.1 bool isElectronTrack = false;
75     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
76 mzanetti 1.3 if ( (fElectrons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) or
77     (fElectrons->At(m)->HasGsfTrk() and fElectrons->At(m)->GsfTrk() == fPFCandidates->At(i)->GsfTrk()) ) {
78 mzanetti 1.1 isElectronTrack = true;
79     break;
80     }
81     }
82     if (isElectronTrack) continue;
83    
84 ceballos 1.6 if ((fPFCandidates->At(i)->HasTrackerTrk() && fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) ||
85     (fPFCandidates->At(i)->HasGsfTrk() && fabs(fPFCandidates->At(i)->GsfTrk()->DzCorrected(*fVertex) ) < deltaZCut)) {
86 mzanetti 1.1 trackNumeratorX -= fPFCandidates->At(i)->Px();
87     trackNumeratorY -= fPFCandidates->At(i)->Py();
88     }
89     }
90    
91     // neutral
92     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
93     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
94     neutralNumeratorX -= fPFCandidates->At(i)->Px();
95     neutralNumeratorY -= fPFCandidates->At(i)->Py();
96     }
97     }
98     }
99 mzanetti 1.3
100     fCorrectedMet = mithep::Met(trackNumeratorX+neutralNumeratorX, trackNumeratorY+neutralNumeratorY);
101     fCorrectedTrackMet = mithep::Met(trackNumeratorX, trackNumeratorY);
102 ceballos 1.2 }
103    
104 mzanetti 1.4
105     //--------------------------------------------------------------------------------------------------
106     MetTools::MetTools(const MuonCol *fMuons, const PFCandidateCol *fPFCandidates, const PFJetCol *pfJets, const Vertex *fVertex,
107     float deltaZCut, float ptCut, float etaCut) {
108    
109     float trackNumeratorX =0, trackNumeratorY =0;
110     float neutralNumeratorX=0, neutralNumeratorY=0;
111    
112     // muons Pt
113     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
114     trackNumeratorX -= fMuons->At(m)->Px();
115     trackNumeratorY -= fMuons->At(m)->Py();
116     }
117    
118     // jets Pt
119     for (UInt_t j = 0; j < pfJets->GetEntries(); ++j) {
120     trackNumeratorX -= pfJets->At(j)->Px();
121     trackNumeratorY -= pfJets->At(j)->Py();
122     }
123    
124     // PF candidates pT
125     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
126    
127     // jets
128     bool inTheJet = false;
129     for (UInt_t j = 0; j < pfJets->GetEntries(); ++j) {
130 mzanetti 1.5 for (UInt_t c=0;c<pfJets->At(j)->NPFCands();++c){
131     if (pfJets->At(j)->PFCand(c) == fPFCandidates->At(i)) {
132 mzanetti 1.4 inTheJet=true;
133     break;
134     }
135     }
136     if (inTheJet) break;
137     }
138     if (inTheJet) continue;
139    
140     // charged
141     if (fPFCandidates->At(i)->HasTrackerTrk()){
142    
143     bool isMuonTrack = false;
144     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
145     if (fMuons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) {
146     isMuonTrack = true;
147     break;
148     }
149     }
150     if (isMuonTrack) continue;
151    
152     if (fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) {
153     trackNumeratorX -= fPFCandidates->At(i)->Px();
154     trackNumeratorY -= fPFCandidates->At(i)->Py();
155     }
156     }
157    
158     // neutral
159     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
160     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
161     neutralNumeratorX -= fPFCandidates->At(i)->Px();
162     neutralNumeratorY -= fPFCandidates->At(i)->Py();
163     }
164     }
165     }
166    
167     fCorrectedMet = mithep::Met(trackNumeratorX+neutralNumeratorX, trackNumeratorY+neutralNumeratorY);
168     fCorrectedTrackMet = mithep::Met(trackNumeratorX, trackNumeratorY);
169     }
170    
171    
172     MetTools::MetTools(const ElectronCol *fElectrons, const PFCandidateCol *fPFCandidates, const PFJetCol *pfJets, const Vertex *fVertex,
173     float deltaZCut, float ptCut, float etaCut) {
174    
175     float trackNumeratorX =0, trackNumeratorY =0;
176     float neutralNumeratorX=0, neutralNumeratorY=0;
177    
178     // electrons Pt
179     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
180     trackNumeratorX -= fElectrons->At(m)->Px();
181     trackNumeratorY -= fElectrons->At(m)->Py();
182     }
183    
184     // jets Pt
185     for (UInt_t j = 0; j < pfJets->GetEntries(); ++j) {
186     trackNumeratorX -= pfJets->At(j)->Px();
187     trackNumeratorY -= pfJets->At(j)->Py();
188     }
189    
190     // PF candidates pT
191     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
192    
193     // jets
194     bool inTheJet = false;
195     for (UInt_t j = 0; j < pfJets->GetEntries(); ++j) {
196 mzanetti 1.5 for (UInt_t c=0;c<pfJets->At(j)->NPFCands();++c){
197     if (pfJets->At(j)->PFCand(c) == fPFCandidates->At(i)) {
198 mzanetti 1.4 inTheJet=true;
199     break;
200     }
201     }
202     if (inTheJet) break;
203     }
204     if (inTheJet) continue;
205    
206     // charged
207 ceballos 1.6 if (fPFCandidates->At(i)->HasTrackerTrk() || fPFCandidates->At(i)->HasGsfTrk()){
208 mzanetti 1.4 bool isElectronTrack = false;
209     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
210     if ( (fElectrons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) or
211     (fElectrons->At(m)->HasGsfTrk() and fElectrons->At(m)->GsfTrk() == fPFCandidates->At(i)->GsfTrk()) ) {
212     isElectronTrack = true;
213     break;
214     }
215     }
216     if (isElectronTrack) continue;
217    
218 ceballos 1.6 if ((fPFCandidates->At(i)->HasTrackerTrk() && fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) ||
219     (fPFCandidates->At(i)->HasGsfTrk() && fabs(fPFCandidates->At(i)->GsfTrk()->DzCorrected(*fVertex) ) < deltaZCut)) {
220 mzanetti 1.4 trackNumeratorX -= fPFCandidates->At(i)->Px();
221     trackNumeratorY -= fPFCandidates->At(i)->Py();
222     }
223     }
224    
225     // neutral
226     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
227     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
228     neutralNumeratorX -= fPFCandidates->At(i)->Px();
229     neutralNumeratorY -= fPFCandidates->At(i)->Py();
230     }
231     }
232     }
233    
234     fCorrectedMet = mithep::Met(trackNumeratorX+neutralNumeratorX, trackNumeratorY+neutralNumeratorY);
235     fCorrectedTrackMet = mithep::Met(trackNumeratorX, trackNumeratorY);
236     }
237    
238    
239 ceballos 1.2 MetTools::MetTools(const MuonCol *fMuons, const ElectronCol *fElectrons, const PFCandidateCol *fPFCandidates,
240     const Vertex *fVertex, float deltaZCut, float ptCut, float etaCut) {
241 mzanetti 1.1
242 ceballos 1.2 float trackNumeratorX =0, trackNumeratorY =0;
243     float neutralNumeratorX=0, neutralNumeratorY=0;
244    
245     // muons Pt
246     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
247     trackNumeratorX -= fMuons->At(m)->Px();
248     trackNumeratorY -= fMuons->At(m)->Py();
249     }
250    
251     // electrons Pt
252     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
253     trackNumeratorX -= fElectrons->At(m)->Px();
254     trackNumeratorY -= fElectrons->At(m)->Py();
255     }
256    
257     // PF candidates pT
258     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
259    
260     // charged
261 ceballos 1.6 if (fPFCandidates->At(i)->HasTrackerTrk() || fPFCandidates->At(i)->HasGsfTrk()){
262 ceballos 1.2 bool isMuonTrack = false;
263     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
264     if (fMuons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) {
265     isMuonTrack = true;
266     break;
267     }
268     }
269     if (isMuonTrack) continue;
270    
271     bool isElectronTrack = false;
272     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
273 mzanetti 1.3 if ( (fElectrons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) or
274     (fElectrons->At(m)->HasGsfTrk() and fElectrons->At(m)->GsfTrk() == fPFCandidates->At(i)->GsfTrk()) ) {
275 ceballos 1.2 isElectronTrack = true;
276     break;
277     }
278     }
279     if (isElectronTrack) continue;
280 mzanetti 1.1
281 ceballos 1.6 if ((fPFCandidates->At(i)->HasTrackerTrk() && fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) ||
282     (fPFCandidates->At(i)->HasGsfTrk() && fabs(fPFCandidates->At(i)->GsfTrk()->DzCorrected(*fVertex) ) < deltaZCut)) {
283 ceballos 1.2 trackNumeratorX -= fPFCandidates->At(i)->Px();
284     trackNumeratorY -= fPFCandidates->At(i)->Py();
285     }
286     }
287 mzanetti 1.1
288 ceballos 1.2 // neutral
289     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
290     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
291     neutralNumeratorX -= fPFCandidates->At(i)->Px();
292     neutralNumeratorY -= fPFCandidates->At(i)->Py();
293     }
294     }
295     }
296 mzanetti 1.3 fCorrectedMet = mithep::Met(trackNumeratorX+neutralNumeratorX, trackNumeratorY+neutralNumeratorY);
297     fCorrectedTrackMet = mithep::Met(trackNumeratorX, trackNumeratorY);
298 ceballos 1.2 }
299 mzanetti 1.1
300 mzanetti 1.4
301     MetTools::MetTools(const MuonCol *fMuons, const ElectronCol *fElectrons, const PFCandidateCol *fPFCandidates,
302     const PFJetCol *pfJets,
303     const Vertex *fVertex, float deltaZCut, float ptCut, float etaCut) {
304    
305     float trackNumeratorX =0, trackNumeratorY =0;
306     float neutralNumeratorX=0, neutralNumeratorY=0;
307    
308     // muons Pt
309     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
310     trackNumeratorX -= fMuons->At(m)->Px();
311     trackNumeratorY -= fMuons->At(m)->Py();
312     }
313    
314     // electrons Pt
315     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
316     trackNumeratorX -= fElectrons->At(m)->Px();
317     trackNumeratorY -= fElectrons->At(m)->Py();
318     }
319    
320     // jets Pt
321     for (UInt_t j = 0; j < pfJets->GetEntries(); ++j) {
322     trackNumeratorX -= pfJets->At(j)->Px();
323     trackNumeratorY -= pfJets->At(j)->Py();
324     }
325    
326     // PF candidates pT
327     for (UInt_t i=0; i<fPFCandidates->GetEntries(); ++i) {
328    
329     // jets
330     bool inTheJet = false;
331     for (UInt_t j = 0; j < pfJets->GetEntries(); ++j) {
332 mzanetti 1.5 for (UInt_t c=0;c<pfJets->At(j)->NPFCands();++c){
333     if (pfJets->At(j)->PFCand(c) == fPFCandidates->At(i)) {
334 mzanetti 1.4 inTheJet=true;
335     break;
336     }
337     }
338     if (inTheJet) break;
339     }
340     if (inTheJet) continue;
341    
342    
343     // charged
344 ceballos 1.6 if (fPFCandidates->At(i)->HasTrackerTrk() || fPFCandidates->At(i)->HasGsfTrk()){
345 mzanetti 1.4 bool isMuonTrack = false;
346     for (UInt_t m = 0; m < fMuons->GetEntries(); ++m) {
347     if (fMuons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) {
348     isMuonTrack = true;
349     break;
350     }
351     }
352     if (isMuonTrack) continue;
353    
354     bool isElectronTrack = false;
355     for (UInt_t m = 0; m < fElectrons->GetEntries(); ++m) {
356     if ( (fElectrons->At(m)->TrackerTrk() == fPFCandidates->At(i)->TrackerTrk()) or
357     (fElectrons->At(m)->HasGsfTrk() and fElectrons->At(m)->GsfTrk() == fPFCandidates->At(i)->GsfTrk()) ) {
358     isElectronTrack = true;
359     break;
360     }
361     }
362     if (isElectronTrack) continue;
363    
364 ceballos 1.6 if ((fPFCandidates->At(i)->HasTrackerTrk() && fabs(fPFCandidates->At(i)->TrackerTrk()->DzCorrected(*fVertex)) < deltaZCut) ||
365     (fPFCandidates->At(i)->HasGsfTrk() && fabs(fPFCandidates->At(i)->GsfTrk()->DzCorrected(*fVertex) ) < deltaZCut)) {
366 mzanetti 1.4 trackNumeratorX -= fPFCandidates->At(i)->Px();
367     trackNumeratorY -= fPFCandidates->At(i)->Py();
368     }
369     }
370    
371     // neutral
372     if (fPFCandidates->At(i)->PFType()== PFCandidate::eNeutralHadron || fPFCandidates->At(i)->PFType()== PFCandidate::eGamma) {
373     if (fPFCandidates->At(i)->Pt() > ptCut and fabs(fPFCandidates->At(i)->Eta()) < etaCut ) {
374     neutralNumeratorX -= fPFCandidates->At(i)->Px();
375     neutralNumeratorY -= fPFCandidates->At(i)->Py();
376     }
377     }
378     }
379     fCorrectedMet = mithep::Met(trackNumeratorX+neutralNumeratorX, trackNumeratorY+neutralNumeratorY);
380     fCorrectedTrackMet = mithep::Met(trackNumeratorX, trackNumeratorY);
381     }
382    
383    
384 mzanetti 1.3 Met MetTools::GetMinimumMet(const Met *UncorrectedMet) {
385 mzanetti 1.1
386     return (fCorrectedMet.Pt() < UncorrectedMet->Pt()) ? fCorrectedMet : *UncorrectedMet;
387     }
388    
389 mzanetti 1.3 Met MetTools::GetMinimumTrackMet(const Met *UncorrectedMet) {
390 ceballos 1.2
391     return (fCorrectedTrackMet.Pt() < UncorrectedMet->Pt()) ? fCorrectedTrackMet : *UncorrectedMet;
392     }
393 mzanetti 1.1