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.13 by khahn, Sat May 5 21:43:54 2012 UTC vs.
Revision 1.27 by anlevin, Wed Jun 6 18:17:21 2012 UTC

# Line 13 | Line 13
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,
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)
100 >                              const mithep::Vertex * vtx)
101   //--------------------------------------------------------------------------------------------------
102   {
103    SelectionStatus status;
# Line 32 | Line 108 | SelectionStatus muonDummyVeto(ControlFla
108   //--------------------------------------------------------------------------------------------------
109   SelectionStatus muonCutBasedVeto(ControlFlags &ctrl,
110                                   const mithep::Muon *muon,
111 <                                 const mithep::Vertex &vtx)
111 >                                 const mithep::Vertex *vtx)
112   //--------------------------------------------------------------------------------------------------
113   {
114    //
# Line 64 | Line 140 | SelectionStatus noPreSelection( ControlF
140   //--------------------------------------------------------------------------------------------------
141   SelectionStatus muonPreSelection( ControlFlags &ctrl,  
142                                    const mithep::Muon * mu,
143 <                                  const mithep::Vertex & vtx,
143 >                                  const mithep::Vertex * vtx,
144                                    const mithep::Array<mithep::PFCandidate> * pfCandidates )
145   //--------------------------------------------------------------------------------------------------
146   {
147    bool ret = true;
148 <  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
149 <  //  bool ret = isMuFO(mu,vtx,pfCandidates);
150 <  if(ctrl.debug) cout << "iS fo? ... " << ret << endl;  
75 <  ret &= ( fabs(mu->Ip3dPVSignificance()) < 4 );
76 <  if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;  
77 <  ret &= ( mu->Pt() > 5 );
78 <  if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;  
148 >
149 >  ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
150 >  ret &= ( mu->Pt() >= 5 );
151    ret &= ( fabs(mu->Eta()) <= 2.4 );
80 –  if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;  
152    ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
153 <  if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;  
154 <  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
155 <    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
85 <  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;  
86 <  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);  
87 <  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
88 <
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);
# Line 98 | Line 165 | SelectionStatus muonPreSelection( Contro
165   //--------------------------------------------------------------------------------------------------
166   SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,  
167                                             const mithep::Muon * mu,
168 <                                           const mithep::Vertex & vtx,
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 &= ( mu->Ip3dPVSignificance() < 100 );
175 >  ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
176    if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;  
177 <  ret &= ( mu->Pt() > 5 );
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;  
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 > //--------------------------------------------------------------------------------------------------
202 > SelectionStatus muonPreSelectionNoD0DzIP( ControlFlags &ctrl,  
203 >                                          const mithep::Muon * mu,
204 >                                          const mithep::Vertex * vtx,
205 >                                          const mithep::Array<mithep::PFCandidate> * pfCandidates )
206 > //--------------------------------------------------------------------------------------------------
207 > {
208 >  bool ret = true;
209 >
210 >  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
211 >  ret &= ( mu->Pt() >= 5 );
212 >  if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;  
213 >  ret &= ( fabs(mu->Eta()) <= 2.4 );
214 >  if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;  
215 >  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
216 >  if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
217 >          
218    //  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
219    //    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
220    //  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;  
# Line 131 | Line 234 | SelectionStatus muonReferencePreSelectio
234   //--------------------------------------------------------------------------------------------------
235   SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
236                                   const mithep::Muon *mu,
237 <                                 const mithep::Vertex & vtx   )
237 >                                 const mithep::Vertex * vtx   )
238   //--------------------------------------------------------------------------------------------------
239   {
240    double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
# Line 205 | Line 308 | void initMuonIDMVA()
308   //--------------------------------------------------------------------------------------------------
309   SelectionStatus muonIDPFSelection(ControlFlags &ctrl,
310                                    const mithep::Muon *mu,
311 <                                  const mithep::Vertex & vtx,
311 >                                  const mithep::Vertex * vtx,
312                                    const mithep::Array<mithep::PFCandidate> * pfCandidates )
313   //--------------------------------------------------------------------------------------------------
314   {
# Line 214 | Line 317 | SelectionStatus muonIDPFSelection(Contro
317  
318    // check that it matches to a PF muon
319    for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
320 +    // tmp
321 +    if( !(PFnoPUflag[i]) ) continue; // my PF no PU hack
322 +
323      const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
324 <    if( pf->HasTrackerTrk() ) {
325 <      if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
326 <        pass = true;
221 <        break;
222 <      }
324 >    if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
325 >      pass = true;
326 >      break;
327      }
328    }
329 + //      
330 + //      if(mu->IsPFMuon()) pass = true;
331 + //      else pass = false;
332  
333    //
334    // NB : ipsig here for reference synch ...
335    //
336 <  if( pass && mu->Ip3dPVSignificance() < 4 ) {
336 > #ifdef SYNC
337 >  if( pass && fabs(mu->Ip3dPVSignificance()) < 4 ) {
338 >    status.orStatus(SelectionStatus::LOOSEID);
339 >    status.orStatus(SelectionStatus::TIGHTID);
340 >  }
341 > #else
342 >  if( pass ) {
343      status.orStatus(SelectionStatus::LOOSEID);
344      status.orStatus(SelectionStatus::TIGHTID);
345    }
346 + #endif
347  
348    return status;
349   }

Diff Legend

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