ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitHzz4l/LeptonSelection/src/MuonSelection.cc
(Generate patch)

Comparing UserCode/MitHzz4l/LeptonSelection/src/MuonSelection.cc (file contents):
Revision 1.27 by anlevin, Wed Jun 6 18:17:21 2012 UTC vs.
Revision 1.36 by anlevin, Wed Oct 17 01:31:22 2012 UTC

# Line 19 | Line 19 | extern Float_t computePFMuonIso(const mi
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)
22 > bool muonPOG2012CutBasedIDTight(const mithep::Muon *mu, const mithep::Vertex * vtx, const mithep::Array<mithep::PFCandidate> * pfCandidates, const mithep::Array<mithep::PileupEnergyDensity> * puEnergyDensity)
23 > {
24 >
25 >  Float_t charged_iso = 0;
26 >  
27 >  Float_t sumChargedHadronPt = 0;
28 >  Float_t sumNeutralHadronPt = 0;
29 >  Float_t sumPhotonPt = 0;
30 >  Float_t sumPUPt = 0;
31 >
32 >  for(unsigned k=0; k<pfCandidates->GetEntries(); ++k) {
33 >    const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[k]);
34 >    Double_t dr = mithep::MathUtils::DeltaR(mu->Phi(),mu->Eta(), pf->Phi(), pf->Eta());
35 >    if(dr > 0.4) continue;
36 >    if(pf->TrackerTrk() && mu->TrackerTrk() && pf->TrackerTrk() == mu->TrackerTrk()) continue;
37 >
38 >    if(PFnoPUflag[k]){ //pf no pileup
39 >      if(abs(pf->PFType()) == mithep::PFCandidate::eHadron)
40 >        sumChargedHadronPt += pf->Pt();
41 >      else if (pf->Pt() > 0.5 && abs(pf->PFType()) == mithep::PFCandidate::eNeutralHadron)
42 >        sumNeutralHadronPt += pf->Pt();
43 >      else if (pf->Pt() > 0.5 && abs(pf->PFType()) == mithep::PFCandidate::eGamma)
44 >        sumPhotonPt += pf->Pt();
45 >    }
46 >    else {//pf pile up
47 >      if(pf->Charge() != 0)
48 >        sumPUPt += pf->Pt();
49 >    }
50 >  
51 >  }
52 >  double iso_sum =  sumChargedHadronPt+ max(0.,sumNeutralHadronPt+sumPhotonPt-0.5*sumPUPt);
53 >
54 >  bool pass = kTRUE;
55 >  if(!mu->IsGlobalMuon()) pass = kFALSE;
56 >
57 >  //there is not necessarily a global track or a tracker track if is global muon is false
58 >  if(!pass) return pass;
59 >
60 >  if(!mu->IsPFMuon()) pass = kFALSE;
61 >  if(mu->GlobalTrk()->RChi2() > 10) pass= kFALSE;
62 >  if(mu->NValidHits() == 0 ) pass= kFALSE;
63 >  if(mu->NMatches() <= 1) pass = kFALSE;
64 >  if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2) pass = kFALSE;
65 >  if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5) pass = kFALSE;
66 >  if(mu->BestTrk()->NPixelHits() == 0) pass = kFALSE;
67 >  if(mu->NTrkLayersHit() <= 5) pass = kFALSE;
68 >  if(iso_sum/mu->Pt() > 0.12 ) pass = kFALSE;
69 >
70 >  return pass;
71 >
72 >
73 > }
74 > //----------------------------------------------------------------------------------------
75 > bool muon2012CutBasedIDTightVersionWithOldIsoThatWorksOn2011(ControlFlags &ctrl, const mithep::Muon *mu, const mithep::Vertex * vtx, const mithep::Array<mithep::PFCandidate> * pfCandidates,
76 >                                                             const mithep::Array<mithep::PileupEnergyDensity> * puEnergyDensity,   mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion
77 >                                                             )
78   {
79  
80    mithep::MuonTools   muT;
# Line 52 | Line 107 | bool muon2012CutBasedIDTight(const mithe
107        neutral_iso += pf->Pt();
108      }
109    }
110 <  
111 <  mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion  = mithep::MuonTools::kMuEAData2012;
112 <  float rho = puEnergyDensity->At(0)->RhoKt6PFJetsCentralNeutral();
113 <  
114 <  double iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
115 <                                        -rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
116 <                                                                   mu->Eta(),EffectiveAreaVersion)));
110 >
111 >  double iso_sum=999;
112 >  double rho;
113 >  if(ctrl.era == 2012) {
114 >    EffectiveAreaVersion  = mithep::MuonTools::kMuEAData2012;
115 >    rho = puEnergyDensity->At(0)->RhoKt6PFJetsCentralNeutral();
116 >    
117 >    iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
118 >                                      -rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
119 >                                                                 mu->Eta(),EffectiveAreaVersion)));
120 >  } else if(ctrl.era == 2011) {
121 >    setMuonRhoAndEaVersion(rho,EffectiveAreaVersion,puEnergyDensity);
122 >    iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
123 >                                      -rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
124 >                                                                 mu->Eta(),EffectiveAreaVersion)));
125 >  } else assert(0);
126  
127    bool pass = kTRUE;
128    if(!mu->IsGlobalMuon()) pass = kFALSE;
# Line 66 | Line 130 | bool muon2012CutBasedIDTight(const mithe
130    //there is not necessarily a global track or a tracker track if is global muon is false
131    if(!pass) return pass;
132  
133 <  if(!mu->IsPFMuon()) pass = kFALSE;
134 <  if(mu->GlobalTrk()->RChi2() > 10) pass= kFALSE;
135 <  if(mu->NValidHits() == 0 ) pass= kFALSE;
136 <  if(mu->NMatches() <= 1) pass = kFALSE;
137 <  if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2) pass = kFALSE;
138 <  if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5) pass = kFALSE;
139 <  if(mu->BestTrk()->NPixelHits() == 0) pass = kFALSE;
140 <  if(mu->NTrkLayersHit() <= 5) pass = kFALSE;
141 <  if(iso_sum/mu->Pt() > 0.12 ) pass = kFALSE;
133 >  if(!mu->IsPFMuon() && ctrl.era==2012)                 pass = kFALSE;
134 >  if(mu->GlobalTrk()->RChi2() > 10)                     pass = kFALSE;
135 >  if(mu->NValidHits() == 0 )                            pass = kFALSE;
136 >  if(mu->NMatches() <= 1)                               pass = kFALSE;
137 >  if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2)   pass = kFALSE;
138 >  if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5)   pass = kFALSE;
139 >  if(mu->BestTrk()->NPixelHits() == 0)                  pass = kFALSE;
140 >  if(mu->NTrkLayersHit() <= 5)                          pass = kFALSE;
141 >  if(iso_sum/mu->Pt() > 0.12 )                          pass = kFALSE;
142  
143    return pass;
144  
# Line 178 | Line 242 | SelectionStatus muonReferencePreSelectio
242    if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;  
243    ret &= ( fabs(mu->Eta()) <= 2.4 );
244    if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;  
245 <  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
245 >  ret &= ( (mu->IsTrackerMuon() && ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated)) || mu->IsGlobalMuon() );
246    if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
247    ret &= (mu->HasTrackerTrk() &&
248            fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
# Line 197 | Line 261 | SelectionStatus muonReferencePreSelectio
261    if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
262    return status;
263   }
200 –
264   //--------------------------------------------------------------------------------------------------
265   SelectionStatus muonPreSelectionNoD0DzIP( ControlFlags &ctrl,  
266                                            const mithep::Muon * mu,
# Line 227 | Line 290 | SelectionStatus muonPreSelectionNoD0DzIP
290    if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
291    return status;
292   }
293 + //--------------------------------------------------------------------------------------------------
294 + SelectionStatus muonPreSelectionNoD0IP( ControlFlags &ctrl,  
295 +                                          const mithep::Muon * mu,
296 +                                          const mithep::Vertex * vtx,
297 +                                          const mithep::Array<mithep::PFCandidate> * pfCandidates )
298 + //--------------------------------------------------------------------------------------------------
299 + {
300 +  bool ret = true;
301  
302 <
303 <
304 <
302 >  ret &= ( mu->Pt() >= 5 );
303 >  ret &= ( fabs(mu->Eta()) <= 2.4 );
304 >  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
305 >  // ret &= (mu->HasTrackerTrk() &&
306 >  //      fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
307 >  //      fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
308 >  ret &= (mu->HasTrackerTrk() &&
309 >          fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
310 >          
311 >  SelectionStatus status;
312 >  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
313 >  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
314 >  return status;
315 > }
316   //--------------------------------------------------------------------------------------------------
317   SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
318                                   const mithep::Muon *mu,
# Line 330 | Line 412 | SelectionStatus muonIDPFSelection(Contro
412   //      if(mu->IsPFMuon()) pass = true;
413   //      else pass = false;
414  
333 –  //
334 –  // NB : ipsig here for reference synch ...
335 –  //
336 – #ifdef SYNC
337 –  if( pass && fabs(mu->Ip3dPVSignificance()) < 4 ) {
338 –    status.orStatus(SelectionStatus::LOOSEID);
339 –    status.orStatus(SelectionStatus::TIGHTID);
340 –  }
341 – #else
415    if( pass ) {
416      status.orStatus(SelectionStatus::LOOSEID);
417      status.orStatus(SelectionStatus::TIGHTID);
418    }
346 – #endif
419  
420    return status;
421   }
422 + //----------------------------------------------------------------------------------------
423 + SelectionStatus PassWwMuonSel(const mithep::Muon *mu, const mithep::Vertex *vtx,
424 +                              const mithep::Array<mithep::PFCandidate> *pfArr
425 + )
426 + {
427 +  bool passID=true;
428 +  bool passIso=true;
429 +  
430 +  if (mu->Pt() < 5)
431 +    passID = false;
432 +  if (fabs(mu->Eta()) > 2.4)
433 +    passID = false;
434 +  if (mu->BestTrk()->PtErr()/mu->Pt() > 0.1)
435 +    passID = false;
436 +  if (fabs(mu->BestTrk()->DzCorrected(*vtx)) > 0.1)
437 +    passID = false;
438 +  Bool_t isGlobal  = (mu->IsGlobalMuon()) && (mu->BestTrk()->RChi2() < 10) &&
439 +                      (mu->NMatches() > 1) && (mu->NValidHits() > 0);
440 +  Bool_t isTracker = (mu->IsTrackerMuon()) &&
441 +                      (mu->Quality().Quality(mithep::MuonQuality::TMLastStationTight));
442 +  if (!isGlobal && !isTracker)
443 +    passID = false;
444 +  int ntrkhits = (mu->HasTrackerTrk()) ? mu->TrackerTrk()->NHits() : 0;
445 +  if (ntrkhits < 10)
446 +    passID = false;
447 +  if (mu->BestTrk()->NPixelHits() < 1)
448 +    passID = false;
449 +  if (fabs(mu->BestTrk()->D0Corrected(*vtx)) > 0.02)
450 +    passID = false;
451 +  
452 +  // note: this isn't really ww muon isolation
453 +  float pfiso = computePFMuonIso(mu,vtx,pfArr,0.3);
454 +
455 +  if (pfiso > 0.2*mu->Pt())
456 +    passIso = false;
457 +
458 +  SelectionStatus status;
459 +  if(passID)  status |= SelectionStatus::TIGHTIDANDPRE;
460 +  if(passIso) status |= SelectionStatus::TIGHTISOANDPRE;
461 +  
462 +  return status;
463 +
464 + }

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines