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.9 by khahn, Tue May 1 16:38:35 2012 UTC vs.
Revision 1.38 by dkralph, Mon Dec 17 17:12:27 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 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;
81 +
82 +  Float_t charged_iso = 0;
83 +  Float_t neutral_iso = 0;
84 +  Float_t gamma_iso = 0;
85 +  
86 +  for(int k=0; k<pfCandidates->GetEntries(); ++k) {
87 +    
88 +    if( !(PFnoPUflag[k]) ) continue;
89 +    const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[k]);
90 +    
91 +    Double_t dr = mithep::MathUtils::DeltaR(mu->Phi(),mu->Eta(), pf->Phi(), pf->Eta());
92 +    if (dr > 0.4) continue;
93 +    
94 +    if (pf->Charge() != 0 && (pf->HasTrackerTrk()||pf->HasGsfTrk()) ) {
95 +      
96 +      if (abs(pf->PFType()) == mithep::PFCandidate::eElectron || abs(pf->PFType()) == mithep::PFCandidate::eMuon) continue;
97 +      
98 +      charged_iso += pf->Pt();
99 +    }
100 +    
101 +    else if (abs(pf->PFType()) == mithep::PFCandidate::eGamma) {
102 +      
103 +      gamma_iso += pf->Pt();
104 +    }
105 +    
106 +    else {
107 +      neutral_iso += pf->Pt();
108 +    }
109 +  }
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;
129 +
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() && 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 +
145 +
146 + }
147 +
148 + bool trackToPFMuonSelection( const mithep::Track *track, const mithep::MuonCol * muoncol )
149 + {
150 +        bool pass = false;
151 +  for(UInt_t i=0; i<muoncol->GetEntries(); ++i) {
152 +    const mithep::Muon *mu = muoncol->At(i);
153 +                if(mu->TrackerTrk() == track && (mu->IsGlobalMuon() || mu->IsTrackerMuon()) && mu->IsPFMuon())
154 +                        pass = true;
155 +        }              
156 +                
157 +        return pass;            
158 +                
159 + }
160 +
161   //--------------------------------------------------------------------------------------------------
162   SelectionStatus muonDummyVeto(ControlFlags &ctrl,
163                                const mithep::Muon *muon,
164 <                              const mithep::Vertex &vtx)
164 >                              const mithep::Vertex * vtx)
165   //--------------------------------------------------------------------------------------------------
166   {
167    SelectionStatus status;
# Line 32 | Line 172 | SelectionStatus muonDummyVeto(ControlFla
172   //--------------------------------------------------------------------------------------------------
173   SelectionStatus muonCutBasedVeto(ControlFlags &ctrl,
174                                   const mithep::Muon *muon,
175 <                                 const mithep::Vertex &vtx)
175 >                                 const mithep::Vertex *vtx)
176   //--------------------------------------------------------------------------------------------------
177   {
178    //
# Line 60 | Line 200 | SelectionStatus noPreSelection( ControlF
200          if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
201          return status;
202   }
63
203   //--------------------------------------------------------------------------------------------------
204 < SelectionStatus muonPreSelection( ControlFlags &ctrl,  
205 <                                  const mithep::Muon * mu,
206 <                                  const mithep::Vertex & vtx,
207 <                                  const mithep::Array<mithep::PFCandidate> * pfCandidates )
204 > SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,  
205 >                                           const mithep::Muon * mu,
206 >                                           const mithep::Vertex * vtx,
207 >                                           const mithep::Array<mithep::PFCandidate> * pfCandidates )
208   //--------------------------------------------------------------------------------------------------
209   {
210    bool ret = true;
72  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
73  //  bool ret = isMuFO(mu,vtx,pfCandidates);
74  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;  
79  ret &= ( fabs(mu->Eta()) <= 2.4 );
80  if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;  
81  ret &=  (mu->IsTrackerMuon() && mu->HasTrackerTrk() &&
82           (mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated));
83  if(ctrl.debug) cout << "and isTrackerMuon ? ... " << ret << endl;  
84  //  ret &=  (mu->IsoR03SumPt()/mu->Pt() < 0.7 );
85  ret &=  (mu->IsoR03SumPt()/mu->Pt() < 0.7 );
86  if(ctrl.debug) cout << "and loose trk iso ? ... " << ret << endl;  
211  
212 +  // if(ctrl.debug) cout << "inside muonpresel ... " << endl;
213 +  ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
214 +  // if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;  
215 +  ret &= ( mu->Pt() >= 5 );
216 +  // if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;  
217 +  ret &= ( fabs(mu->Eta()) <= 2.4 );
218 +  // if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;  
219 +  ret &= ( (mu->IsTrackerMuon() && ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated)) || mu->IsGlobalMuon() );
220 +  // if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
221 +  ret &= (mu->HasTrackerTrk() &&
222 +          fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
223 +          fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
224 +  // if(ctrl.debug) cout << "d0 and dz? ... " << ret << endl;  
225 +    
226    SelectionStatus status;
227    if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
228 <  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
228 >  // if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
229    return status;
230   }
93
94
95 //--------------------------------------------------------------------------------------------------
96 bool isMuFO( const mithep::Muon * mu,
97             const mithep::Vertex & vtx,
98             const mithep::Array<mithep::PFCandidate> * pfCandidates )
231   //--------------------------------------------------------------------------------------------------
232 < {
233 <  bool isgood=true;
234 <  float reliso = computePFMuonIso(mu,vtx,pfCandidates,0.3)/mu->Pt();
235 <
104 <  if( mu->Pt() < 5  )         isgood=false;                          
105 <  if ( fabs(mu->BestTrk()->DzCorrected(vtx)) > 0.2 ) isgood=false;   // needed to remove cosmics in HF sample
106 <
107 <  //
108 <
109 <  // HF seems to not want tkquality, SS does
110 <  if( mu->Pt()>20 ) {
111 <    if ( !((mu->IsGlobalMuon()) && mu->NSegments()>0  ) ) isgood=false; //&& mu->nValidHits>0
112 <  } else {
113 <    if ( !((mu->IsGlobalMuon()) && mu->NSegments()>0 ) //&& //&& mu->nValidHits>0
114 <         && !( (mu->IsTrackerMuon()) &&  (mu->Quality().QualityMask().Mask() & muTools.kTMOneStationLoose)
115 <               )
116 <         )   isgood=false;
117 <  }
118 <
119 <
120 <  //
121 <  // for HF MC to agree w/ data, cannot put any Tk quality bits
122 <  // but to get to 5% closure in the SS test, we need at least Tk oneStationLoose
123 <
124 <  /*
125 < && ( (mu->qualityBits & kTMOneStationLoose) ||
126 <                                               (mu->qualityBits & kTMLastStationOptimizedBarrelLowPtLoose) ||
127 <                                               (mu->qualityBits & kTMLastStationOptimizedLowPtLoose)
128 <                                               )
129 <  */
130 <
131 <
132 <  // comment for more stats for mu fake shape
133 <  if( mu->Pt() < 20 ) {
134 <    if( reliso > 3 ) isgood=false;    
135 <  } else {
136 <    if( reliso > 5 ) isgood=false;    
137 <  }
138 <
139 <  return isgood;
140 < };
141 <
142 < //--------------------------------------------------------------------------------------------------
143 < SelectionStatus passSoftMuonSelection( ControlFlags &ctrl,
144 <                                       const mithep::Muon * mu ,
145 <                                       const mithep::Vertex & vtx)
232 > SelectionStatus muonPreSelectionNoD0IP( ControlFlags &ctrl,  
233 >                                          const mithep::Muon * mu,
234 >                                          const mithep::Vertex * vtx,
235 >                                          const mithep::Array<mithep::PFCandidate> * pfCandidates )
236   //--------------------------------------------------------------------------------------------------
237   {
238 +  bool ret = true;
239  
240 <  int level=0;
241 <  unsigned failmask=0x0;
242 <  
243 <  // Use tracker track when available
244 <  const mithep::Track *muTrk=0;
245 <  if(mu->HasStandaloneTrk()) { muTrk = mu->StandaloneTrk(); }
246 <  else if(mu->HasTrackerTrk())         { muTrk = mu->TrackerTrk(); }
247 <  else if(mu->HasGlobalTrk())     { muTrk = mu->GlobalTrk(); }
248 <  assert(muTrk);                  
158 <
159 <  if(mu->TrackerTrk()->NHits() < 11 )  
160 <    failmask |= (1<<level);
161 <  level++;
162 <
163 <  if(fabs(muTrk->D0Corrected(vtx)) > 0.2)
164 <    failmask |= (1<<level);
165 <  level++;
166 <
167 <  if(fabs(muTrk->DzCorrected(vtx)) > 0.1)
168 <    failmask |= (1<<level);
169 <  level++;
170 <
171 <  if(!(mu->IsTrackerMuon()))
172 <    failmask |= (1<<level);
173 <  level++;
174 <
175 <  //  if(!(mu->Quality().QualityMask().Mask() & muTools.kTMLastStationAngTight))
176 <  if(!(mu->Quality().QualityMask().Mask() & muTools.kTMLastStationTight))
177 <    failmask |= (1<<level);      
178 <  level++;
179 <
180 <  Double_t iso = (mu->IsoR03SumPt() + mu->IsoR03EmEt() + mu->IsoR03HadEt())/mu->Pt();
181 <  if(mu->Pt()>20 && iso<0.1)
182 <    failmask |= (1<<level);
183 <
184 <  bool passtight  = !(failmask);
185 <  bool passloose  = !(failmask);
186 <  return SelectionStatus( passtight | passloose );
187 <  
188 < }
189 <
190 < //--------------------------------------------------------------------------------------------------
191 < //unsigned passMuonSelectionZZ( const mithep::TMuon * mu ) {
192 < SelectionStatus passMuonSelectionZZ( ControlFlags &ctrl, const mithep::Muon * mu )
193 < //--------------------------------------------------------------------------------------------------
194 < {
195 <  int level=0;
196 <  unsigned failmask=0x0;
197 <
198 <  if(mu->Pt() < 5) {
199 <    failmask |= (1<<level);
200 <  }
201 <
202 <  level++;
203 <  if(fabs(mu->Eta()) > 2.4) {
204 <    failmask |= (1<<level);
205 <  }
206 <
207 <  level++;
208 <  if(!(mu->IsGlobalMuon())) {
209 <    failmask |= (1<<level);
210 <  }
211 <
212 <  level++;
213 <  if(mu->TrackerTrk()->NHits()  < 10) {
214 <    failmask |= (1<<level);
215 <  }
216 <
217 <
218 <  level++;  
219 <  if( (mu->IsoR03SumPt()/mu->Pt()) > 0.7 ) {
220 <    failmask |= (1<<level);
221 <  }
222 <
223 <  //  if(muon->nPixHits   < 1)     return false;
224 <  //  if(muon->muNchi2    > 10)    return false;
225 <  //  if(muon->nMatch     < 2)     return false;
226 <  //  if(muon->nValidHits         < 1)     return false;
227 <  //  if(muon->pterr/muon->pt > 0.1)   return false;
228 <  //  if(fabs(muon->dz)       > 0.1)   return false;  
229 <  
230 <  SelectionStatus status;
231 <  if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
232 <  if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
233 <  return status;
234 <
235 <
236 <
237 < };
238 <
239 <
240 <
241 <
242 < //
243 < // Kevin's WW selection
244 < //
245 < //--------------------------------------------------------------------------------------------------
246 < SelectionStatus passMuonSelection( ControlFlags &ctrl,
247 <                                   const mithep::Muon * mu,
248 <                                   const mithep::Vertex & vtx   )
249 < //--------------------------------------------------------------------------------------------------
250 < {
240 >  ret &= ( mu->Pt() >= 5 );
241 >  ret &= ( fabs(mu->Eta()) <= 2.4 );
242 >  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
243 >  // they put this back in for the h.c.p. sync...
244 >  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
245 >    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
246 >  ret &= (mu->HasTrackerTrk() &&
247 >          fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
248 >          
249    SelectionStatus status;
250 <  int level=0;
251 <  unsigned failmask=0x0;
254 <
255 <  // Use tracker track when available
256 <  const mithep::Track *muTrk=0;
257 <  if(mu->HasTrackerTrk())         { muTrk = mu->TrackerTrk(); }
258 <  else if(mu->HasGlobalTrk())     { muTrk = mu->GlobalTrk(); }
259 <  else if(mu->HasStandaloneTrk()) { muTrk = mu->StandaloneTrk(); }
260 <  assert(muTrk);                  
261 <  
262 <
263 <
264 <  // 0x1
265 <  if(muTrk->Pt() < 5) {
266 <    failmask |= (1<<level);
267 <  }
268 <
269 <  // 0x2
270 <  level++;
271 <  if(fabs(muTrk->Eta()) > 2.4) {
272 <    failmask |= (1<<level);
273 <  }
274 <
275 <  // 0x4
276 <  level++;
277 <  if(muTrk->PtErr()/muTrk->Pt() > 0.1)   {
278 <    failmask |= (1<<level);
279 <  }
280 <
281 <  // 0x8
282 <  level++;
283 <  if( fabs(muTrk->DzCorrected(vtx)) > 0.1 )   {
284 <    failmask |= (1<<level);
285 <  }
286 <
287 <  double muNchi2;
288 <  if(mu->HasStandaloneTrk())      { muNchi2 = mu->StandaloneTrk()->RChi2(); }
289 <  else if(mu->HasGlobalTrk())     { muNchi2 = mu->GlobalTrk()->RChi2();     }
290 <  else if(mu->HasTrackerTrk())    { muNchi2 = mu->TrackerTrk()->RChi2();    }
291 <
292 <  unsigned qualityBits = mu->Quality().QualityMask().Mask();
293 <  
294 <  Bool_t isGlobal  = (mu->IsGlobalMuon()) && (muNchi2 < 10) && (mu->NMatches() > 1) && (mu->NValidHits() > 0);
295 <  Bool_t isTracker = (mu->IsTrackerMuon() ) && (qualityBits & muTools.kTMLastStationTight);
296 <
297 <  // 0x10
298 <  level++;
299 <  if((!isGlobal && !isTracker) || !(mu->HasTrackerTrk())) {
300 <    failmask |= (1<<level);
301 <    status.setStatus(SelectionStatus::FAIL);
302 <    return status;    
303 <  }
304 <
305 <  // 0x20
306 <  level++;
307 <  assert(mu->HasTrackerTrk());
308 <  assert(mu->TrackerTrk());
309 <  if( mu->TrackerTrk()->NHits() < 10 ) {
310 <    failmask |= (1<<level);
311 <  }
312 <
313 <  level++;
314 <  if(muTrk->NPixelHits()          < 1) {
315 <    failmask |= (1<<level);
316 <  }
317 <
318 <
319 <  level++;
320 <  if(fabs(muTrk->D0Corrected(vtx))>0.02)   {
321 <    failmask |= (1<<level);
322 <  }
323 <
324 <  /*
325 <  if(mu->pt>20) {
326 <    if(fabs(mu->d0)>0.02)   {
327 <      failmask |= (1<<level);
328 <    }
329 <  } else {
330 <    if(fabs(mu->d0)>0.01)  {
331 <    failmask |= (1<<level);
332 <    }
333 <  }
334 <  */
335 <
336 <  if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
337 <  if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
250 >  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
251 >  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
252    return status;
253 <
340 <
341 < };
342 <
343 <
253 > }
254   //--------------------------------------------------------------------------------------------------
255   SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
256                                   const mithep::Muon *mu,
257 <                                 const mithep::Vertex & vtx   )
257 >                                 const mithep::Vertex * vtx   )
258   //--------------------------------------------------------------------------------------------------
259   {
350  assert(mu->IsTrackerMuon()); // must be
351  assert(mu->HasTrackerTrk());
352  assert(mu->TrackerTrk());
353
354  /*
355  cerr << "TrackerTrk: " << mu->TrackerTrk() << endl;
356  flush(cerr);
357  cerr << "Rchi2: " << mu->TrackerTrk()->RChi2() << endl;
358  flush(cerr);
359
360  cerr << "calling MVa ... "<< endl;
361  flush(cerr);
362  */
363
260    double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
261  
262 +  //
263 +  // WARNING !!! KH hacked out to sync
264 +  //
265 +  /*
266    double mvaval = muIDMVA->MVAValue_ID(mu->Pt(),
267                                      mu->Eta(),
268                                      mu->IsGlobalMuon(),
# Line 370 | Line 270 | SelectionStatus muonIDMVASelection(Contr
270                                      mu->TrackerTrk()->RChi2(),
271                                      global_rchi2,
272                                      (Double_t)(mu->NValidHits()),
273 <                                    (Double_t)(mu->TrackerTrk()->NHits()),
273 >                                    (Double_t)(mu->TrackerTrk()->NHits()),
274                                      (Double_t)(mu->TrackerTrk()->NPixelHits()),
275                                      (Double_t)(mu->NMatches()),
276                                      mu->TrkKink(),
# Line 381 | Line 281 | SelectionStatus muonIDMVASelection(Contr
281                                      mu->HadS9Energy(),
282                                      mu->EmS9Energy(),
283                                      (Bool_t)(ctrl.debug) );
284 <
284 >  */
285 >  double mvaval = 0.0;
286  
287    SelectionStatus status;
288    bool pass = false;
# Line 394 | Line 295 | SelectionStatus muonIDMVASelection(Contr
295             && fabs(mu->Eta()) > 1.5 && mu->Pt() <= 10 && mvaval >= MUON_IDMVA_CUT_BIN2)  pass = true;
296    else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
297             && fabs(mu->Eta()) > 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN3)  pass = true;
298 <  else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon()
299 <           && (mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated) && mvaval >= MUON_IDMVA_CUT_BIN4)
300 <    pass = true;
298 >  else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon() && mvaval >= MUON_IDMVA_CUT_BIN4 )  pass = true;
299 >  else if( mu->IsGlobalMuon() && !(mu->IsTrackerMuon()) && mvaval >= MUON_IDMVA_CUT_BIN5 )  pass = true;
300 >  
301  
302  
303    if( pass ) {
# Line 427 | Line 328 | void initMuonIDMVA()
328   //--------------------------------------------------------------------------------------------------
329   SelectionStatus muonIDPFSelection(ControlFlags &ctrl,
330                                    const mithep::Muon *mu,
331 <                                  const mithep::Vertex & vtx,
331 >                                  const mithep::Vertex * vtx,
332                                    const mithep::Array<mithep::PFCandidate> * pfCandidates )
333   //--------------------------------------------------------------------------------------------------
334   {
335    bool pass = false;
336 <  SelectionStatus status; // init'ed to FAIL
436 <  
437 <  // check that muon is tracker
438 <  if( !(mu->IsTrackerMuon()) )   return status;
336 >  SelectionStatus status;
337  
338    // check that it matches to a PF muon
339    for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
340 +    if( !(PFnoPUflag[i]) ) continue;
341 +
342      const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
343 <    if( pf->HasTrackerTrk() ) {
344 <      if( pf->TrackerTrk() == mu->TrackerTrk() && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
345 <        pass = true;
446 <        break;
447 <      }
343 >    if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
344 >      pass = true;
345 >      break;
346      }
347    }
348  
# Line 455 | Line 353 | SelectionStatus muonIDPFSelection(Contro
353  
354    return status;
355   }
356 + //----------------------------------------------------------------------------------------
357 + SelectionStatus PassWwMuonSel(const mithep::Muon *mu, const mithep::Vertex *vtx,
358 +                              const mithep::Array<mithep::PFCandidate> *pfArr
359 + )
360 + {
361 +  bool passID=true;
362 +  bool passIso=true;
363 +  
364 +  if (mu->Pt() < 5)
365 +    passID = false;
366 +  if (fabs(mu->Eta()) > 2.4)
367 +    passID = false;
368 +  if (mu->BestTrk()->PtErr()/mu->Pt() > 0.1)
369 +    passID = false;
370 +  if (fabs(mu->BestTrk()->DzCorrected(*vtx)) > 0.1)
371 +    passID = false;
372 +  Bool_t isGlobal  = (mu->IsGlobalMuon()) && (mu->BestTrk()->RChi2() < 10) &&
373 +                      (mu->NMatches() > 1) && (mu->NValidHits() > 0);
374 +  Bool_t isTracker = (mu->IsTrackerMuon()) &&
375 +                      (mu->Quality().Quality(mithep::MuonQuality::TMLastStationTight));
376 +  if (!isGlobal && !isTracker)
377 +    passID = false;
378 +  int ntrkhits = (mu->HasTrackerTrk()) ? mu->TrackerTrk()->NHits() : 0;
379 +  if (ntrkhits < 10)
380 +    passID = false;
381 +  if (mu->BestTrk()->NPixelHits() < 1)
382 +    passID = false;
383 +  if (fabs(mu->BestTrk()->D0Corrected(*vtx)) > 0.02)
384 +    passID = false;
385 +  
386 +  // note: this isn't really ww muon isolation
387 +  float pfiso = computePFMuonIso(mu,vtx,pfArr,0.3);
388 +
389 +  if (pfiso > 0.2*mu->Pt())
390 +    passIso = false;
391 +
392 +  SelectionStatus status;
393 +  if(passID)  status |= SelectionStatus::TIGHTIDANDPRE;
394 +  if(passIso) status |= SelectionStatus::TIGHTISOANDPRE;
395 +  
396 +  return status;
397 +
398 + }
399 + // //--------------------------------------------------------------------------------------------------
400 + // SelectionStatus muonPreSelectionNoD0DzIP( ControlFlags &ctrl,  
401 + //                                        const mithep::Muon * mu,
402 + //                                        const mithep::Vertex * vtx,
403 + //                                        const mithep::Array<mithep::PFCandidate> * pfCandidates )
404 + // //--------------------------------------------------------------------------------------------------
405 + // {
406 + //   bool ret = true;
407 +
408 + //   if(ctrl.debug) cout << "inside muonpresel ... " << endl;
409 + //   ret &= ( mu->Pt() >= 5 );
410 + //   if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;  
411 + //   ret &= ( fabs(mu->Eta()) <= 2.4 );
412 + //   if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;  
413 + //   ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
414 + //   if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
415 +          
416 + //   //  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
417 + //   //    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
418 + //   //  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;  
419 + //   //  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);  
420 + //   //  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
421 +
422 +
423 + //   SelectionStatus status;
424 + //   if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
425 + //   if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
426 + //   return status;
427 + // }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines