ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitPhysics/Mods/src/MuonIDMod.cc
Revision: 1.65
Committed: Thu Jan 26 15:29:13 2012 UTC (13 years, 3 months ago) by sixie
Content type: text/plain
Branch: MAIN
Changes since 1.64: +18 -9 lines
Log Message: 
use old denominator for MVA muon ID instead of effective area corrected

File Contents

# User Rev Content
1 sixie 1.65 // $Id: MuonIDMod.cc,v 1.64 2012/01/23 20:27:14 sixie Exp $
2 loizides 1.1
3     #include "MitPhysics/Mods/interface/MuonIDMod.h"
4 loizides 1.6 #include "MitCommon/MathTools/interface/MathUtils.h"
5 ceballos 1.38 #include "MitAna/DataTree/interface/MuonFwd.h"
6     #include "MitAna/DataTree/interface/ElectronFwd.h"
7 loizides 1.23 #include "MitAna/DataTree/interface/VertexCol.h"
8 loizides 1.6 #include "MitPhysics/Init/interface/ModNames.h"
9 loizides 1.1
10     using namespace mithep;
11    
12     ClassImp(mithep::MuonIDMod)
13    
14     //--------------------------------------------------------------------------------------------------
15     MuonIDMod::MuonIDMod(const char *name, const char *title) :
16     BaseMod(name,title),
17 loizides 1.6 fMuonBranchName(Names::gkMuonBrn),
18     fCleanMuonsName(ModNames::gkCleanMuonsName),
19 ceballos 1.39 fNonIsolatedMuonsName("random"),
20     fNonIsolatedElectronsName("random"),
21 ceballos 1.35 fVertexName(ModNames::gkGoodVertexesName),
22 ceballos 1.42 fBeamSpotName(Names::gkBeamSpotBrn),
23 ceballos 1.38 fTrackName(Names::gkTrackBrn),
24     fPFCandidatesName(Names::gkPFCandidatesBrn),
25 ceballos 1.57 fMuonIDType("WWMuIdV3"),
26 ceballos 1.49 fMuonIsoType("PFIso"),
27 loizides 1.6 fMuonClassType("Global"),
28 ceballos 1.2 fTrackIsolationCut(3.0),
29     fCaloIsolationCut(3.0),
30 sixie 1.54 fCombIsolationCut(0.15),
31     fCombRelativeIsolationCut(0.15),
32     fPFIsolationCut(-1.0),
33 ceballos 1.3 fMuonPtMin(10),
34 loizides 1.21 fApplyD0Cut(kTRUE),
35 ceballos 1.42 fApplyDZCut(kTRUE),
36 ceballos 1.30 fD0Cut(0.020),
37 ceballos 1.50 fDZCut(0.10),
38 ceballos 1.42 fWhichVertex(-1),
39 ceballos 1.30 fEtaCut(2.4),
40 loizides 1.15 fMuIDType(kIdUndef),
41     fMuIsoType(kIsoUndef),
42     fMuClassType(kClassUndef),
43 ceballos 1.14 fMuons(0),
44 loizides 1.15 fVertices(0),
45 ceballos 1.42 fBeamSpot(0),
46 ceballos 1.38 fTracks(0),
47     fPFCandidates(0),
48 ceballos 1.56 fIntRadius(0.0),
49 ceballos 1.39 fNonIsolatedMuons(0),
50 ceballos 1.41 fNonIsolatedElectrons(0),
51     fPileupEnergyDensityName(Names::gkPileupEnergyDensityBrn),
52 sixie 1.59 fPileupEnergyDensity(0),
53     fMuonTools(0),
54     fMuonIDMVA(0),
55     fMuonMVAWeights_Subdet0Pt10To14p5(""),
56     fMuonMVAWeights_Subdet1Pt10To14p5(""),
57     fMuonMVAWeights_Subdet0Pt14p5To20(""),
58     fMuonMVAWeights_Subdet1Pt14p5To20(""),
59     fMuonMVAWeights_Subdet0Pt20ToInf(""),
60     fMuonMVAWeights_Subdet1Pt20ToInf("")
61 loizides 1.1 {
62     // Constructor.
63     }
64    
65     //--------------------------------------------------------------------------------------------------
66     void MuonIDMod::Process()
67     {
68     // Process entries of the tree.
69    
70 ceballos 1.38 if(fMuIsoType != kPFIsoNoL) {
71     LoadEventObject(fMuonBranchName, fMuons);
72     }
73     else {
74     fMuons = GetObjThisEvt<MuonOArr>(fMuonBranchName);
75     }
76 ceballos 1.42 LoadEventObject(fBeamSpotName, fBeamSpot);
77 ceballos 1.38 LoadEventObject(fTrackName, fTracks);
78     LoadEventObject(fPFCandidatesName, fPFCandidates);
79 sixie 1.59 if(fMuIsoType == kTrackCaloSliding ||
80     fMuIsoType == kCombinedRelativeConeAreaCorrected ||
81     fMuIsoType == kPFIsoEffectiveAreaCorrected ||
82     fMuIsoType == kMVAIso_BDTG_IDIso
83     ) {
84 ceballos 1.41 LoadEventObject(fPileupEnergyDensityName, fPileupEnergyDensity);
85     }
86 sixie 1.59
87 loizides 1.7 MuonOArr *CleanMuons = new MuonOArr;
88     CleanMuons->SetName(fCleanMuonsName);
89 loizides 1.1
90 ceballos 1.38 fVertices = GetObjThisEvt<VertexOArr>(fVertexName);
91    
92 loizides 1.1 for (UInt_t i=0; i<fMuons->GetEntries(); ++i) {
93 loizides 1.8 const Muon *mu = fMuons->At(i);
94 loizides 1.6
95     Bool_t pass = kFALSE;
96 ceballos 1.30 Double_t pt = 0; // make sure pt is taken from the correct track!
97     Double_t eta = 0; // make sure eta is taken from the correct track!
98 loizides 1.6 switch (fMuClassType) {
99     case kAll:
100     pass = kTRUE;
101 ceballos 1.30 if (mu->HasTrk()) {
102     pt = mu->Pt();
103     eta = TMath::Abs(mu->Eta());
104     }
105 loizides 1.6 break;
106     case kGlobal:
107 ceballos 1.31 pass = mu->HasGlobalTrk() && mu->IsTrackerMuon();
108 ceballos 1.32 if (pass && mu->TrackerTrk()) {
109 ceballos 1.30 pt = mu->TrackerTrk()->Pt();
110     eta = TMath::Abs(mu->TrackerTrk()->Eta());
111     }
112 ceballos 1.32 else {
113     pt = mu->Pt();
114     eta = TMath::Abs(mu->Eta());
115     }
116 ceballos 1.30 break;
117 ceballos 1.53 case kGlobalTracker:
118     pass = (mu->HasGlobalTrk() && mu->GlobalTrk()->Chi2()/mu->GlobalTrk()->Ndof() < 10 &&
119     (mu->NSegments() > 1 || mu->NMatches() > 1) && mu->NValidHits() > 0) ||
120     (mu->IsTrackerMuon() &&
121     mu->Quality().Quality(MuonQuality::TMLastStationTight));
122     if (pass) {
123     pt = mu->TrackerTrk()->Pt();
124     eta = TMath::Abs(mu->TrackerTrk()->Eta());
125     }
126     else {
127     pt = mu->Pt();
128     eta = TMath::Abs(mu->Eta());
129     }
130     break;
131 loizides 1.6 case kSta:
132 loizides 1.19 pass = mu->HasStandaloneTrk();
133 ceballos 1.30 if (pass) {
134     pt = mu->StandaloneTrk()->Pt();
135     eta = TMath::Abs(mu->StandaloneTrk()->Eta());
136     }
137 loizides 1.6 break;
138 loizides 1.24 case kTrackerMuon:
139 ceballos 1.28 pass = mu->HasTrackerTrk() && mu->IsTrackerMuon() &&
140     mu->Quality().Quality(MuonQuality::TrackerMuonArbitrated);
141 ceballos 1.30 if (pass) {
142     pt = mu->TrackerTrk()->Pt();
143     eta = TMath::Abs(mu->TrackerTrk()->Eta());
144     }
145 loizides 1.24 break;
146     case kCaloMuon:
147     pass = mu->HasTrackerTrk() && mu->IsCaloMuon();
148 ceballos 1.30 if (pass) {
149     pt = mu->TrackerTrk()->Pt();
150     eta = TMath::Abs(mu->TrackerTrk()->Eta());
151     }
152 loizides 1.24 break;
153     case kTrackerBased:
154 loizides 1.19 pass = mu->HasTrackerTrk();
155 ceballos 1.30 if (pass) {
156     pt = mu->TrackerTrk()->Pt();
157     eta = TMath::Abs(mu->TrackerTrk()->Eta());
158     }
159 loizides 1.6 break;
160     default:
161     break;
162 ceballos 1.5 }
163 loizides 1.6
164     if (!pass)
165     continue;
166    
167     if (pt <= fMuonPtMin)
168     continue;
169    
170 ceballos 1.30 if (eta >= fEtaCut)
171     continue;
172    
173 ceballos 1.34 Double_t RChi2 = 0.0;
174     if (mu->HasGlobalTrk()) {
175     RChi2 = mu->GlobalTrk()->Chi2()/mu->GlobalTrk()->Ndof();
176     }
177     else if(mu->BestTrk() != 0){
178     RChi2 = mu->BestTrk()->Chi2()/mu->BestTrk()->Ndof();
179     }
180 loizides 1.6 Bool_t idpass = kFALSE;
181     switch (fMuIDType) {
182 ceballos 1.34 case kWMuId:
183     idpass = mu->BestTrk() != 0 &&
184     mu->BestTrk()->NHits() > 10 &&
185     RChi2 < 10.0 &&
186 ceballos 1.36 (mu->NSegments() > 1 || mu->NMatches() > 1) &&
187 ceballos 1.34 mu->BestTrk()->NPixelHits() > 0 &&
188     mu->Quality().Quality(MuonQuality::GlobalMuonPromptTight);
189     break;
190     case kZMuId:
191     idpass = mu->BestTrk() != 0 &&
192     mu->BestTrk()->NHits() > 10 &&
193 ceballos 1.36 (mu->NSegments() > 1 || mu->NMatches() > 1) &&
194 ceballos 1.34 mu->BestTrk()->NPixelHits() > 0 &&
195     mu->Quality().Quality(MuonQuality::GlobalMuonPromptTight);
196     break;
197 loizides 1.6 case kLoose:
198 ceballos 1.32 idpass = mu->BestTrk() != 0 &&
199     mu->Quality().Quality(MuonQuality::TMOneStationLoose) &&
200 ceballos 1.28 mu->Quality().Quality(MuonQuality::TM2DCompatibilityLoose) &&
201 ceballos 1.29 mu->BestTrk()->NHits() > 10 &&
202 ceballos 1.34 RChi2 < 10.0 &&
203 ceballos 1.30 mu->Quality().Quality(MuonQuality::GlobalMuonPromptTight);
204 loizides 1.6 break;
205     case kTight:
206 ceballos 1.32 idpass = mu->BestTrk() != 0 &&
207     mu->Quality().Quality(MuonQuality::TMOneStationTight) &&
208 ceballos 1.28 mu->Quality().Quality(MuonQuality::TM2DCompatibilityTight) &&
209 ceballos 1.29 mu->BestTrk()->NHits() > 10 &&
210 ceballos 1.34 RChi2 < 10.0 &&
211 ceballos 1.30 mu->Quality().Quality(MuonQuality::GlobalMuonPromptTight);
212 ceballos 1.28 break;
213 ceballos 1.53 case kWWMuIdV1:
214 ceballos 1.32 idpass = mu->BestTrk() != 0 &&
215     mu->BestTrk()->NHits() > 10 &&
216 ceballos 1.53 mu->BestTrk()->NPixelHits() > 0 &&
217     mu->BestTrk()->PtErr()/mu->BestTrk()->Pt() < 0.1 &&
218 ceballos 1.34 RChi2 < 10.0 &&
219 ceballos 1.36 (mu->NSegments() > 1 || mu->NMatches() > 1) &&
220 ceballos 1.53 mu->Quality().Quality(MuonQuality::GlobalMuonPromptTight);
221     break;
222     case kWWMuIdV2:
223     idpass = mu->BestTrk() != 0 &&
224     mu->BestTrk()->NHits() > 10 &&
225 ceballos 1.36 mu->BestTrk()->NPixelHits() > 0 &&
226     mu->BestTrk()->PtErr()/mu->BestTrk()->Pt() < 0.1;
227 loizides 1.6 break;
228 ceballos 1.57 case kWWMuIdV3:
229     idpass = mu->BestTrk() != 0 &&
230     mu->BestTrk()->NHits() > 10 &&
231     mu->BestTrk()->NPixelHits() > 0 &&
232     mu->BestTrk()->PtErr()/mu->BestTrk()->Pt() < 0.1 &&
233     mu->TrkKink() < 20.0;
234     break;
235 sixie 1.59 case kMVAID_BDTG_IDIso:
236     {
237     Bool_t passDenominatorM2 = (mu->BestTrk() != 0 &&
238     mu->BestTrk()->NHits() > 10 &&
239     mu->BestTrk()->NPixelHits() > 0 &&
240     mu->BestTrk()->PtErr()/mu->BestTrk()->Pt() < 0.1 &&
241     MuonTools::PassD0Cut(mu, fVertices, 0.20, 0) &&
242     MuonTools::PassDZCut(mu, fVertices, 0.10, 0) &&
243 sixie 1.64 mu->TrkKink() < 20.0
244     );
245     idpass = passDenominatorM2;
246 sixie 1.62 //only evaluate MVA if muon passes M2 denominator to save time
247     if (idpass) idpass = PassMuonMVA_BDTG_IdIso(mu, fVertices->At(0), fPileupEnergyDensity);
248 sixie 1.59 }
249     break;
250 loizides 1.18 case kNoId:
251     idpass = kTRUE;
252     break;
253 loizides 1.6 default:
254     break;
255 ceballos 1.4 }
256 loizides 1.6
257     if (!idpass)
258     continue;
259    
260 ceballos 1.22 Bool_t isocut = kFALSE;
261 loizides 1.6 switch (fMuIsoType) {
262     case kTrackCalo:
263 ceballos 1.22 isocut = (mu->IsoR03SumPt() < fTrackIsolationCut) &&
264 loizides 1.6 (mu->IsoR03EmEt() + mu->IsoR03HadEt() < fCaloIsolationCut);
265     break;
266     case kTrackCaloCombined:
267 ceballos 1.38 isocut = (1.0 * mu->IsoR03SumPt() +
268     1.0 * mu->IsoR03EmEt() +
269     1.0 * mu->IsoR03HadEt() < fCombIsolationCut);
270 loizides 1.6 break;
271     case kTrackCaloSliding:
272     {
273 ceballos 1.41 const PileupEnergyDensity *rho = fPileupEnergyDensity->At(0);
274 sixie 1.54 Double_t totalIso = mu->IsoR03SumPt() + mu->IsoR03EmEt() + mu->IsoR03HadEt() - rho->Rho() * TMath::Pi() * 0.3 * 0.3 ;
275 ceballos 1.46 // trick to change the signal region cut
276     double theIsoCut = fCombIsolationCut;
277     if(theIsoCut < 0.20){
278     if(mu->Pt() > 20.0) theIsoCut = 0.15;
279     else theIsoCut = 0.10;
280     }
281     if (totalIso < (mu->Pt()*theIsoCut)) isocut = kTRUE;
282 ceballos 1.22 }
283 loizides 1.6 break;
284 ceballos 1.41 case kTrackCaloSlidingNoCorrection:
285 ceballos 1.40 {
286     Double_t totalIso = 1.0 * mu->IsoR03SumPt() +
287 ceballos 1.41 1.0 * mu->IsoR03EmEt() +
288     1.0 * mu->IsoR03HadEt();
289 ceballos 1.46 // trick to change the signal region cut
290     double theIsoCut = fCombIsolationCut;
291     if(theIsoCut < 0.20){
292     if(mu->Pt() > 20.0) theIsoCut = 0.15;
293     else theIsoCut = 0.10;
294     }
295     if (totalIso < (mu->Pt()*theIsoCut)) isocut = kTRUE;
296 ceballos 1.40 }
297     break;
298 sixie 1.54 case kCombinedRelativeConeAreaCorrected:
299     {
300     const PileupEnergyDensity *rho = fPileupEnergyDensity->At(0);
301     Double_t totalIso = mu->IsoR03SumPt() + mu->IsoR03EmEt() + mu->IsoR03HadEt() - rho->Rho() * TMath::Pi() * 0.3 * 0.3 ;
302     double theIsoCut = fCombRelativeIsolationCut;
303     if (totalIso < (mu->Pt()*theIsoCut)) isocut = kTRUE;
304     }
305     break;
306 sixie 1.64 case kCombinedRelativeEffectiveAreaCorrected:
307     {
308     Double_t tmpRho = 0;
309     if (!(TMath::IsNaN(fPileupEnergyDensity->At(0)->Rho()) || isinf(fPileupEnergyDensity->At(0)->Rho())))
310     tmpRho = fPileupEnergyDensity->At(0)->Rho();
311    
312     isocut = ( mu->IsoR03SumPt() + mu->IsoR03EmEt() + mu->IsoR03HadEt()
313     - tmpRho*MuonTools::MuonEffectiveArea(MuonTools::kMuEMIso03, mu->Eta())
314     - tmpRho*MuonTools::MuonEffectiveArea(MuonTools::kMuHadIso03, mu->Eta())
315     ) < (mu->Pt()* 0.40);
316     }
317     break;
318 ceballos 1.38 case kPFIso:
319 sixie 1.59 {
320 ceballos 1.49 Double_t pfIsoCutValue = 9999;
321 sixie 1.54 if(fPFIsolationCut > 0){
322     pfIsoCutValue = fPFIsolationCut;
323 ceballos 1.49 } else {
324 ceballos 1.51 if (mu->AbsEta() < 1.479) {
325     if (mu->Pt() > 20) {
326 ceballos 1.52 pfIsoCutValue = 0.13;
327 ceballos 1.51 } else {
328 ceballos 1.52 pfIsoCutValue = 0.06;
329 ceballos 1.51 }
330 ceballos 1.49 } else {
331 ceballos 1.51 if (mu->Pt() > 20) {
332 ceballos 1.52 pfIsoCutValue = 0.09;
333 ceballos 1.51 } else {
334 ceballos 1.52 pfIsoCutValue = 0.05;
335 ceballos 1.51 }
336     }
337 ceballos 1.49 }
338 ceballos 1.56 Double_t totalIso = IsolationTools::PFMuonIsolation(mu, fPFCandidates, fVertices->At(0), 0.1, 1.0, 0.3, 0.0, fIntRadius);
339 ceballos 1.49 if (totalIso < (mu->Pt()*pfIsoCutValue) )
340 ceballos 1.38 isocut = kTRUE;
341     }
342     break;
343 sixie 1.59 case kPFIsoEffectiveAreaCorrected:
344     {
345     Double_t pfIsoCutValue = 9999;
346     if(fPFIsolationCut > 0){
347     pfIsoCutValue = fPFIsolationCut;
348     } else {
349     pfIsoCutValue = fPFIsolationCut; //leave it like this for now
350     }
351     Double_t EffectiveAreaCorrectedPFIso = IsolationTools::PFMuonIsolation(mu, fPFCandidates, fVertices->At(0), 0.1, 1.0, 0.3, 0.0, fIntRadius)
352     - fPileupEnergyDensity->At(0)->Rho() * MuonTools::MuonEffectiveArea(MuonTools::kMuNeutralIso03, mu->Eta());
353     isocut = EffectiveAreaCorrectedPFIso < (mu->Pt() * pfIsoCutValue);
354     break;
355     }
356 ceballos 1.38 case kPFIsoNoL:
357     {
358 ceballos 1.39 fNonIsolatedMuons = GetObjThisEvt<MuonCol>(fNonIsolatedMuonsName);
359     fNonIsolatedElectrons = GetObjThisEvt<ElectronCol>(fNonIsolatedElectronsName);
360 ceballos 1.38
361 ceballos 1.58 Double_t pfIsoCutValue = 9999;
362     if(fPFIsolationCut > 0){
363     pfIsoCutValue = fPFIsolationCut;
364     } else {
365     if (mu->AbsEta() < 1.479) {
366     if (mu->Pt() > 20) {
367     pfIsoCutValue = 0.13;
368     } else {
369     pfIsoCutValue = 0.06;
370     }
371     } else {
372     if (mu->Pt() > 20) {
373     pfIsoCutValue = 0.09;
374     } else {
375     pfIsoCutValue = 0.05;
376     }
377     }
378     }
379     Double_t totalIso = IsolationTools::PFMuonIsolation(mu, fPFCandidates, fNonIsolatedMuons, fNonIsolatedElectrons, fVertices->At(0), 0.1, 1.0, 0.3, 0.0, fIntRadius);
380     if (totalIso < (mu->Pt()*pfIsoCutValue) )
381 ceballos 1.38 isocut = kTRUE;
382     }
383     break;
384 sixie 1.59 case kMVAIso_BDTG_IDIso:
385 sixie 1.64 {
386 sixie 1.65
387     // **************************************************************************
388     // Don't use effective area correction denominator. Instead use the old one.
389     // **************************************************************************
390    
391     // Double_t tmpRho = 0;
392     // if (!(TMath::IsNaN(fPileupEnergyDensity->At(0)->Rho()) || isinf(fPileupEnergyDensity->At(0)->Rho())))
393     // tmpRho = fPileupEnergyDensity->At(0)->Rho();
394    
395     // isocut = ( mu->IsoR03SumPt() + mu->IsoR03EmEt() + mu->IsoR03HadEt()
396     // - tmpRho*MuonTools::MuonEffectiveArea(MuonTools::kMuEMIso03, mu->Eta())
397     // - tmpRho*MuonTools::MuonEffectiveArea(MuonTools::kMuHadIso03, mu->Eta())
398     // ) < (mu->Pt()* 0.40);
399    
400     Double_t totalIso = IsolationTools::PFMuonIsolation(mu, fPFCandidates, fVertices->At(0), 0.1, 1.0, 0.3, 0.0, fIntRadius);
401     isocut = (totalIso < (mu->Pt()*0.4));
402    
403 sixie 1.64 }
404 sixie 1.59 break;
405 loizides 1.13 case kNoIso:
406 ceballos 1.22 isocut = kTRUE;
407 loizides 1.13 break;
408 loizides 1.6 case kCustomIso:
409     default:
410     break;
411 ceballos 1.4 }
412 ceballos 1.3
413 ceballos 1.22 if (isocut == kFALSE)
414 loizides 1.6 continue;
415    
416 ceballos 1.42 // apply d0 cut
417 loizides 1.21 if (fApplyD0Cut) {
418 ceballos 1.42 Bool_t passD0cut = kTRUE;
419 ceballos 1.46 if(fD0Cut < 0.05) { // trick to change the signal region cut
420     if (mu->Pt() > 20.0) fD0Cut = 0.02;
421     else if (mu->Pt() <= 20.0) fD0Cut = 0.01;
422     }
423 ceballos 1.42 if(fWhichVertex >= -1) passD0cut = MuonTools::PassD0Cut(mu, fVertices, fD0Cut, fWhichVertex);
424     else passD0cut = MuonTools::PassD0Cut(mu, fBeamSpot, fD0Cut);
425 ceballos 1.31 if (!passD0cut)
426 loizides 1.21 continue;
427 ceballos 1.14 }
428    
429 ceballos 1.42 // apply dz cut
430     if (fApplyDZCut) {
431 ceballos 1.45 Bool_t passDZcut = MuonTools::PassDZCut(mu, fVertices, fDZCut, fWhichVertex);
432 ceballos 1.42 if (!passDZcut)
433     continue;
434     }
435    
436 loizides 1.6 // add good muon
437     CleanMuons->Add(mu);
438 loizides 1.1 }
439    
440 loizides 1.10 // sort according to pt
441     CleanMuons->Sort();
442    
443 loizides 1.6 // add objects for other modules to use
444 loizides 1.7 AddObjThisEvt(CleanMuons);
445 loizides 1.1 }
446    
447     //--------------------------------------------------------------------------------------------------
448     void MuonIDMod::SlaveBegin()
449     {
450     // Run startup code on the computer (slave) doing the actual analysis. Here,
451 loizides 1.6 // we just request the muon collection branch.
452    
453 ceballos 1.38 // In this case we cannot have a branch
454     if (fMuonIsoType.CompareTo("PFIsoNoL") != 0) {
455     ReqEventObject(fMuonBranchName, fMuons, kTRUE);
456     }
457 ceballos 1.42 ReqEventObject(fBeamSpotName, fBeamSpot, kTRUE);
458 ceballos 1.38 ReqEventObject(fTrackName, fTracks, kTRUE);
459     ReqEventObject(fPFCandidatesName, fPFCandidates, kTRUE);
460 sixie 1.55 if (fMuonIsoType.CompareTo("TrackCaloSliding") == 0
461 sixie 1.59 || fMuonIsoType.CompareTo("CombinedRelativeConeAreaCorrected") == 0
462 sixie 1.64 || fMuonIsoType.CompareTo("CombinedRelativeEffectiveAreaCorrected") == 0
463     || fMuonIsoType.CompareTo("PFIsoEffectiveAreaCorrected") == 0
464 sixie 1.59 || fMuonIsoType.CompareTo("MVA_BDTG_IDIso") == 0
465     ) {
466 ceballos 1.41 ReqEventObject(fPileupEnergyDensityName, fPileupEnergyDensity, kTRUE);
467     }
468 loizides 1.1
469 sixie 1.59
470 ceballos 1.34 if (fMuonIDType.CompareTo("WMuId") == 0)
471     fMuIDType = kWMuId;
472     else if (fMuonIDType.CompareTo("ZMuId") == 0)
473     fMuIDType = kZMuId;
474     else if (fMuonIDType.CompareTo("Tight") == 0)
475 loizides 1.6 fMuIDType = kTight;
476     else if (fMuonIDType.CompareTo("Loose") == 0)
477     fMuIDType = kLoose;
478 ceballos 1.53 else if (fMuonIDType.CompareTo("WWMuIdV1") == 0)
479     fMuIDType = kWWMuIdV1;
480     else if (fMuonIDType.CompareTo("WWMuIdV2") == 0)
481     fMuIDType = kWWMuIdV2;
482 ceballos 1.57 else if (fMuonIDType.CompareTo("WWMuIdV3") == 0)
483     fMuIDType = kWWMuIdV3;
484 loizides 1.12 else if (fMuonIDType.CompareTo("NoId") == 0)
485     fMuIDType = kNoId;
486 loizides 1.6 else if (fMuonIDType.CompareTo("Custom") == 0) {
487     fMuIDType = kCustomId;
488     SendError(kWarning, "SlaveBegin",
489     "Custom muon identification is not yet implemented.");
490 sixie 1.59 } else if (fMuonIDType.CompareTo("MVA_BDTG_IDIso") == 0) {
491     fMuIDType = kMVAID_BDTG_IDIso;
492 loizides 1.6 } else {
493     SendError(kAbortAnalysis, "SlaveBegin",
494     "The specified muon identification %s is not defined.",
495     fMuonIDType.Data());
496     return;
497     }
498 loizides 1.1
499 loizides 1.6 if (fMuonIsoType.CompareTo("TrackCalo") == 0)
500     fMuIsoType = kTrackCalo;
501     else if (fMuonIsoType.CompareTo("TrackCaloCombined") == 0)
502     fMuIsoType = kTrackCaloCombined;
503     else if (fMuonIsoType.CompareTo("TrackCaloSliding") == 0)
504     fMuIsoType = kTrackCaloSliding;
505 ceballos 1.41 else if (fMuonIsoType.CompareTo("TrackCaloSlidingNoCorrection") == 0)
506     fMuIsoType = kTrackCaloSlidingNoCorrection;
507 sixie 1.54 else if (fMuonIsoType.CompareTo("CombinedRelativeConeAreaCorrected") == 0)
508     fMuIsoType = kCombinedRelativeConeAreaCorrected;
509 sixie 1.64 else if (fMuonIsoType.CompareTo("CombinedRelativeEffectiveAreaCorrected") == 0)
510     fMuIsoType = kCombinedRelativeEffectiveAreaCorrected;
511 ceballos 1.38 else if (fMuonIsoType.CompareTo("PFIso") == 0)
512     fMuIsoType = kPFIso;
513 sixie 1.59 else if (fMuonIsoType.CompareTo("PFIsoEffectiveAreaCorrected") == 0)
514     fMuIsoType = kPFIsoEffectiveAreaCorrected;
515 ceballos 1.38 else if (fMuonIsoType.CompareTo("PFIsoNoL") == 0)
516     fMuIsoType = kPFIsoNoL;
517 loizides 1.6 else if (fMuonIsoType.CompareTo("NoIso") == 0)
518     fMuIsoType = kNoIso;
519     else if (fMuonIsoType.CompareTo("Custom") == 0) {
520     fMuIsoType = kCustomIso;
521     SendError(kWarning, "SlaveBegin",
522     "Custom muon isolation is not yet implemented.");
523 sixie 1.59 } else if (fMuonIDType.CompareTo("MVA_BDTG_IDIso") == 0) {
524     fMuIsoType = kMVAIso_BDTG_IDIso;
525 loizides 1.6 } else {
526     SendError(kAbortAnalysis, "SlaveBegin",
527     "The specified muon isolation %s is not defined.",
528     fMuonIsoType.Data());
529     return;
530     }
531 loizides 1.1
532 loizides 1.6 if (fMuonClassType.CompareTo("All") == 0)
533     fMuClassType = kAll;
534     else if (fMuonClassType.CompareTo("Global") == 0)
535     fMuClassType = kGlobal;
536 ceballos 1.53 else if (fMuonClassType.CompareTo("GlobalTracker") == 0)
537     fMuClassType = kGlobalTracker;
538 loizides 1.6 else if (fMuonClassType.CompareTo("Standalone") == 0)
539     fMuClassType = kSta;
540 loizides 1.24 else if (fMuonClassType.CompareTo("TrackerMuon") == 0)
541     fMuClassType = kTrackerMuon;
542     else if (fMuonClassType.CompareTo("CaloMuon") == 0)
543     fMuClassType = kCaloMuon;
544     else if (fMuonClassType.CompareTo("TrackerBased") == 0)
545     fMuClassType = kTrackerBased;
546 loizides 1.6 else {
547     SendError(kAbortAnalysis, "SlaveBegin",
548     "The specified muon class %s is not defined.",
549     fMuonClassType.Data());
550     return;
551     }
552 sixie 1.59
553    
554     //If we use MVA ID, need to load MVA weights
555     if(fMuIsoType == kMVAIso_BDTG_IDIso || fMuIDType == kMVAID_BDTG_IDIso) {
556     fMuonTools = new MuonTools();
557     fMuonIDMVA = new MuonIDMVA();
558     fMuonIDMVA->Initialize("BDTG method",
559     fMuonMVAWeights_Subdet0Pt10To14p5,
560     fMuonMVAWeights_Subdet1Pt10To14p5,
561     fMuonMVAWeights_Subdet0Pt14p5To20,
562     fMuonMVAWeights_Subdet1Pt14p5To20,
563     fMuonMVAWeights_Subdet0Pt20ToInf,
564     fMuonMVAWeights_Subdet1Pt20ToInf,
565 sixie 1.64 MuonIDMVA::kIDIsoCombinedDetIso);
566 sixie 1.59 }
567    
568     }
569    
570    
571     //--------------------------------------------------------------------------------------------------
572 sixie 1.61 Bool_t MuonIDMod::PassMuonMVA_BDTG_IdIso(const Muon *mu, const Vertex *vertex,
573     const PileupEnergyDensityCol *PileupEnergyDensity) const
574 sixie 1.59 {
575    
576     const Track *muTrk=0;
577     if(mu->HasTrackerTrk()) { muTrk = mu->TrackerTrk(); }
578     else if(mu->HasStandaloneTrk()) { muTrk = mu->StandaloneTrk(); }
579 sixie 1.64
580 sixie 1.61 Double_t MVAValue = fMuonIDMVA->MVAValue(mu,vertex,fMuonTools,fPFCandidates,PileupEnergyDensity);
581 sixie 1.59
582     Int_t subdet = 0;
583     if (fabs(muTrk->Eta()) < 1.479) subdet = 0;
584     else subdet = 1;
585     Int_t ptBin = 0;
586     if (muTrk->Pt() > 14.5) ptBin = 1;
587     if (muTrk->Pt() > 20.0) ptBin = 2;
588    
589     Int_t MVABin = -1;
590     if (subdet == 0 && ptBin == 0) MVABin = 0;
591     if (subdet == 1 && ptBin == 0) MVABin = 1;
592     if (subdet == 0 && ptBin == 1) MVABin = 2;
593     if (subdet == 1 && ptBin == 1) MVABin = 3;
594     if (subdet == 0 && ptBin == 2) MVABin = 4;
595     if (subdet == 1 && ptBin == 2) MVABin = 5;
596    
597     Double_t MVACut = -999;
598 sixie 1.64 if (MVABin == 0) MVACut = -0.5618;
599     if (MVABin == 1) MVACut = -0.3002;
600     if (MVABin == 2) MVACut = -0.4642;
601     if (MVABin == 3) MVACut = -0.2478;
602     if (MVABin == 4) MVACut = 0.1706;
603     if (MVABin == 5) MVACut = 0.8146;
604 sixie 1.59
605     if (MVAValue > MVACut) return kTRUE;
606     return kFALSE;
607 loizides 1.1 }