ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitHzz4l/LeptonSelection/src/MuonSelection.cc
Revision: 1.29
Committed: Thu Jun 21 20:50:25 2012 UTC (12 years, 10 months ago) by dkralph
Content type: text/plain
Branch: MAIN
Changes since 1.28: +22 -1 lines
Log Message:
Denominator def'n was changed at some point to differ from the lepton preselection fcns, so add it here...

File Contents

# Content
1 #include <math.h>
2 #include <iostream>
3
4 #include "MuonSelection.h"
5 #include "ParseArgs.h"
6 #include "SelectionStatus.h"
7
8 #include "Track.h"
9 #include "MuonTools.h"
10 #include "MuonQuality.h"
11 #include "MuonIDMVA.h"
12
13 mithep::MuonIDMVA * muIDMVA;
14 mithep::MuonTools muTools;
15
16 extern vector<bool> PFnoPUflag;
17 extern Float_t computePFMuonIso(const mithep::Muon *mu,
18 const mithep::Vertex * vtx,
19 const mithep::Array<mithep::PFCandidate> * pfCandidates,
20 const Double_t dRMax);
21
22 bool muon2012CutBasedIDTight(const mithep::Muon *mu, const mithep::Vertex * vtx, const mithep::Array<mithep::PFCandidate> * pfCandidates, const mithep::Array<mithep::PileupEnergyDensity> * puEnergyDensity)
23 {
24
25 mithep::MuonTools muT;
26
27 Float_t charged_iso = 0;
28 Float_t neutral_iso = 0;
29 Float_t gamma_iso = 0;
30
31 for(int k=0; k<pfCandidates->GetEntries(); ++k) {
32
33 if( !(PFnoPUflag[k]) ) continue;
34 const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[k]);
35
36 Double_t dr = mithep::MathUtils::DeltaR(mu->Phi(),mu->Eta(), pf->Phi(), pf->Eta());
37 if (dr > 0.4) continue;
38
39 if (pf->Charge() != 0 && (pf->HasTrackerTrk()||pf->HasGsfTrk()) ) {
40
41 if (abs(pf->PFType()) == mithep::PFCandidate::eElectron || abs(pf->PFType()) == mithep::PFCandidate::eMuon) continue;
42
43 charged_iso += pf->Pt();
44 }
45
46 else if (abs(pf->PFType()) == mithep::PFCandidate::eGamma) {
47
48 gamma_iso += pf->Pt();
49 }
50
51 else {
52 neutral_iso += pf->Pt();
53 }
54 }
55
56 mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion = mithep::MuonTools::kMuEAData2012;
57 float rho = puEnergyDensity->At(0)->RhoKt6PFJetsCentralNeutral();
58
59 double iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
60 -rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
61 mu->Eta(),EffectiveAreaVersion)));
62
63 bool pass = kTRUE;
64 if(!mu->IsGlobalMuon()) pass = kFALSE;
65
66 //there is not necessarily a global track or a tracker track if is global muon is false
67 if(!pass) return pass;
68
69 if(!mu->IsPFMuon()) pass = kFALSE;
70 if(mu->GlobalTrk()->RChi2() > 10) pass= kFALSE;
71 if(mu->NValidHits() == 0 ) pass= kFALSE;
72 if(mu->NMatches() <= 1) pass = kFALSE;
73 if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2) pass = kFALSE;
74 if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5) pass = kFALSE;
75 if(mu->BestTrk()->NPixelHits() == 0) pass = kFALSE;
76 if(mu->NTrkLayersHit() <= 5) pass = kFALSE;
77 if(iso_sum/mu->Pt() > 0.12 ) pass = kFALSE;
78
79 return pass;
80
81
82 }
83
84 bool trackToPFMuonSelection( const mithep::Track *track, const mithep::MuonCol * muoncol )
85 {
86 bool pass = false;
87 for(UInt_t i=0; i<muoncol->GetEntries(); ++i) {
88 const mithep::Muon *mu = muoncol->At(i);
89 if(mu->TrackerTrk() == track && (mu->IsGlobalMuon() || mu->IsTrackerMuon()) && mu->IsPFMuon())
90 pass = true;
91 }
92
93 return pass;
94
95 }
96
97 //--------------------------------------------------------------------------------------------------
98 SelectionStatus muonDummyVeto(ControlFlags &ctrl,
99 const mithep::Muon *muon,
100 const mithep::Vertex * vtx)
101 //--------------------------------------------------------------------------------------------------
102 {
103 SelectionStatus status;
104 status.setStatus(SelectionStatus::PRESELECTION);
105 return status;
106 }
107
108 //--------------------------------------------------------------------------------------------------
109 SelectionStatus muonCutBasedVeto(ControlFlags &ctrl,
110 const mithep::Muon *muon,
111 const mithep::Vertex *vtx)
112 //--------------------------------------------------------------------------------------------------
113 {
114 //
115 // Loose cut-based ID for isolation veto
116 //
117 bool ret = true;
118
119 if(!(muon->IsGlobalMuon() || muon->IsTrackerMuon())) ret = false;
120 if( muon->NValidHits() < 11 ) ret = false;
121 if( fabs(muon->Ip3dPVSignificance()) >= 4 ) ret = false;
122
123 SelectionStatus status;
124 if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
125 return status;
126 }
127
128
129 //--------------------------------------------------------------------------------------------------
130 SelectionStatus noPreSelection( ControlFlags &ctrl, const mithep::Muon * mu )
131
132 //--------------------------------------------------------------------------------------------------
133 {
134 SelectionStatus status;
135 status.setStatus(SelectionStatus::PRESELECTION);
136 if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
137 return status;
138 }
139
140 //--------------------------------------------------------------------------------------------------
141 SelectionStatus muonPreSelection( ControlFlags &ctrl,
142 const mithep::Muon * mu,
143 const mithep::Vertex * vtx,
144 const mithep::Array<mithep::PFCandidate> * pfCandidates )
145 //--------------------------------------------------------------------------------------------------
146 {
147 bool ret = true;
148
149 ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
150 ret &= ( mu->Pt() >= 5 );
151 ret &= ( fabs(mu->Eta()) <= 2.4 );
152 ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
153 // if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
154 // ret &= ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
155 ret &= ( mu->IsoR03SumPt()/mu->Pt() < 0.7 );
156
157 SelectionStatus status;
158 if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
159 if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
160 return status;
161 }
162
163
164
165 //--------------------------------------------------------------------------------------------------
166 SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,
167 const mithep::Muon * mu,
168 const mithep::Vertex * vtx,
169 const mithep::Array<mithep::PFCandidate> * pfCandidates )
170 //--------------------------------------------------------------------------------------------------
171 {
172 bool ret = true;
173
174 if(ctrl.debug) cout << "inside muonpresel ... " << endl;
175 ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
176 if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;
177 ret &= ( mu->Pt() >= 5 );
178 if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
179 ret &= ( fabs(mu->Eta()) <= 2.4 );
180 if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
181 ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
182 if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
183 ret &= (mu->HasTrackerTrk() &&
184 fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
185 fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
186 if(ctrl.debug) cout << "d0 and dz? ... " << ret << endl;
187
188 // if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
189 // ret &= ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
190 // if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
191 // ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
192 // if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret << endl;
193
194
195 SelectionStatus status;
196 if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
197 if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
198 return status;
199 }
200 //--------------------------------------------------------------------------------------------------
201 SelectionStatus muonDenominatorSelection( ControlFlags &ctrl,
202 const mithep::Muon * mu,
203 const mithep::Vertex * vtx,
204 const mithep::Array<mithep::PFCandidate> * pfCandidates )
205 //--------------------------------------------------------------------------------------------------
206 {
207 bool ret = true;
208
209 ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
210 ret &= ( mu->Pt() >= 5 );
211 ret &= ( fabs(mu->Eta()) <= 2.4 );
212 ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
213 ret &= (mu->HasTrackerTrk() &&
214 fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
215 fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
216
217 SelectionStatus status;
218 if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
219 return status;
220 }
221
222 //--------------------------------------------------------------------------------------------------
223 SelectionStatus muonPreSelectionNoD0DzIP( ControlFlags &ctrl,
224 const mithep::Muon * mu,
225 const mithep::Vertex * vtx,
226 const mithep::Array<mithep::PFCandidate> * pfCandidates )
227 //--------------------------------------------------------------------------------------------------
228 {
229 bool ret = true;
230
231 if(ctrl.debug) cout << "inside muonpresel ... " << endl;
232 ret &= ( mu->Pt() >= 5 );
233 if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;
234 ret &= ( fabs(mu->Eta()) <= 2.4 );
235 if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;
236 ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
237 if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
238
239 // if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )
240 // ret &= ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
241 // if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;
242 // ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);
243 // if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret << endl;
244
245
246 SelectionStatus status;
247 if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
248 if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
249 return status;
250 }
251
252
253
254
255 //--------------------------------------------------------------------------------------------------
256 SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
257 const mithep::Muon *mu,
258 const mithep::Vertex * vtx )
259 //--------------------------------------------------------------------------------------------------
260 {
261 double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
262
263 //
264 // WARNING !!! KH hacked out to sync
265 //
266 /*
267 double mvaval = muIDMVA->MVAValue_ID(mu->Pt(),
268 mu->Eta(),
269 mu->IsGlobalMuon(),
270 mu->IsTrackerMuon(),
271 mu->TrackerTrk()->RChi2(),
272 global_rchi2,
273 (Double_t)(mu->NValidHits()),
274 (Double_t)(mu->TrackerTrk()->NHits()),
275 (Double_t)(mu->TrackerTrk()->NPixelHits()),
276 (Double_t)(mu->NMatches()),
277 mu->TrkKink(),
278 muTools.GetSegmentCompatability(mu),
279 muTools.GetCaloCompatability(mu,true,true),
280 mu->HadEnergy(),
281 mu->EmEnergy(),
282 mu->HadS9Energy(),
283 mu->EmS9Energy(),
284 (Bool_t)(ctrl.debug) );
285 */
286 double mvaval = 0.0;
287
288 SelectionStatus status;
289 bool pass = false;
290
291 if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
292 && fabs(mu->Eta()) <= 1.5 && mu->Pt() <= 10 && mvaval >= MUON_IDMVA_CUT_BIN0) pass = true;
293 else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
294 && fabs(mu->Eta()) <= 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN1) pass = true;
295 else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
296 && fabs(mu->Eta()) > 1.5 && mu->Pt() <= 10 && mvaval >= MUON_IDMVA_CUT_BIN2) pass = true;
297 else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
298 && fabs(mu->Eta()) > 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN3) pass = true;
299 else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon() && mvaval >= MUON_IDMVA_CUT_BIN4 ) pass = true;
300 else if( mu->IsGlobalMuon() && !(mu->IsTrackerMuon()) && mvaval >= MUON_IDMVA_CUT_BIN5 ) pass = true;
301
302
303
304 if( pass ) {
305 status.orStatus(SelectionStatus::LOOSEID);
306 status.orStatus(SelectionStatus::TIGHTID);
307 }
308
309 return status;
310 }
311
312 //--------------------------------------------------------------------------------------------------
313 void initMuonIDMVA()
314 //--------------------------------------------------------------------------------------------------
315 {
316 muIDMVA = new mithep::MuonIDMVA();
317 vector<string> weightFiles;
318 weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_barrel_lowpt_V2.weights.xml");
319 weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_barrel_highpt_V2.weights.xml");
320 weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_endcap_lowpt_V2.weights.xml");
321 weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_endcap_highpt_V2.weights.xml");
322 weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_tracker_V2.weights.xml");
323 weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_global_V2.weights.xml");
324 muIDMVA->Initialize( "MuonIDMVA",
325 mithep::MuonIDMVA::kIDV0,
326 kTRUE, weightFiles);
327 }
328
329 //--------------------------------------------------------------------------------------------------
330 SelectionStatus muonIDPFSelection(ControlFlags &ctrl,
331 const mithep::Muon *mu,
332 const mithep::Vertex * vtx,
333 const mithep::Array<mithep::PFCandidate> * pfCandidates )
334 //--------------------------------------------------------------------------------------------------
335 {
336 bool pass = false;
337 SelectionStatus status; // init'ed to FAIL
338
339 // check that it matches to a PF muon
340 for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
341 // tmp
342 if( !(PFnoPUflag[i]) ) continue; // my PF no PU hack
343
344 const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
345 if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
346 pass = true;
347 break;
348 }
349 }
350 //
351 // if(mu->IsPFMuon()) pass = true;
352 // else pass = false;
353
354 //
355 // NB : ipsig here for reference synch ...
356 //
357 #ifdef SYNC
358 if( pass && fabs(mu->Ip3dPVSignificance()) < 4 ) {
359 status.orStatus(SelectionStatus::LOOSEID);
360 status.orStatus(SelectionStatus::TIGHTID);
361 }
362 #else
363 if( pass ) {
364 status.orStatus(SelectionStatus::LOOSEID);
365 status.orStatus(SelectionStatus::TIGHTID);
366 }
367 #endif
368
369 return status;
370 }
371 //----------------------------------------------------------------------------------------
372 SelectionStatus PassWwMuonSel(const mithep::Muon *mu, const mithep::Vertex *vtx,
373 const mithep::Array<mithep::PFCandidate> *pfArr
374 )
375 {
376 bool passID=true;
377 bool passIso=true;
378
379 if (mu->Pt() < 5)
380 passID = false;
381 if (fabs(mu->Eta()) > 2.4)
382 passID = false;
383 if (mu->BestTrk()->PtErr()/mu->Pt() > 0.1)
384 passID = false;
385 if (fabs(mu->BestTrk()->DzCorrected(*vtx)) > 0.1)
386 passID = false;
387 Bool_t isGlobal = (mu->IsGlobalMuon()) && (mu->BestTrk()->RChi2() < 10) &&
388 (mu->NMatches() > 1) && (mu->NValidHits() > 0);
389 Bool_t isTracker = (mu->IsTrackerMuon()) &&
390 (mu->Quality().Quality(mithep::MuonQuality::TMLastStationTight));
391 if (!isGlobal && !isTracker)
392 passID = false;
393 int ntrkhits = (mu->HasTrackerTrk()) ? mu->TrackerTrk()->NHits() : 0;
394 if (ntrkhits < 10)
395 passID = false;
396 if (mu->BestTrk()->NPixelHits() < 1)
397 passID = false;
398 if (fabs(mu->BestTrk()->D0Corrected(*vtx)) > 0.02)
399 passID = false;
400
401 // note: this isn't really ww muon isolation
402 float pfiso = computePFMuonIso(mu,vtx,pfArr,0.3);
403
404 if (pfiso > 0.2*mu->Pt())
405 passIso = false;
406
407 SelectionStatus status;
408 if(passID) status |= SelectionStatus::TIGHTIDANDPRE;
409 if(passIso) status |= SelectionStatus::TIGHTISOANDPRE;
410
411 return status;
412
413 }