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.10 by khahn, Tue May 1 23:39:15 2012 UTC vs.
Revision 1.30 by dkralph, Thu Jul 5 13:43:13 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(ControlFlags &ctrl, const mithep::Muon *mu, const mithep::Vertex * vtx, const mithep::Array<mithep::PFCandidate> * pfCandidates,
23 +                             const mithep::Array<mithep::PileupEnergyDensity> * puEnergyDensity,   mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion
24 +                             )
25 + {
26 +
27 +  mithep::MuonTools   muT;
28 +
29 +  Float_t charged_iso = 0;
30 +  Float_t neutral_iso = 0;
31 +  Float_t gamma_iso = 0;
32 +  
33 +  for(int k=0; k<pfCandidates->GetEntries(); ++k) {
34 +    
35 +    if( !(PFnoPUflag[k]) ) continue;
36 +    const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[k]);
37 +    
38 +    Double_t dr = mithep::MathUtils::DeltaR(mu->Phi(),mu->Eta(), pf->Phi(), pf->Eta());
39 +    if (dr > 0.4) continue;
40 +    
41 +    if (pf->Charge() != 0 && (pf->HasTrackerTrk()||pf->HasGsfTrk()) ) {
42 +      
43 +      if (abs(pf->PFType()) == mithep::PFCandidate::eElectron || abs(pf->PFType()) == mithep::PFCandidate::eMuon) continue;
44 +      
45 +      charged_iso += pf->Pt();
46 +    }
47 +    
48 +    else if (abs(pf->PFType()) == mithep::PFCandidate::eGamma) {
49 +      
50 +      gamma_iso += pf->Pt();
51 +    }
52 +    
53 +    else {
54 +      neutral_iso += pf->Pt();
55 +    }
56 +  }
57 +
58 +  double iso_sum=999;
59 +  double rho;
60 +  if(ctrl.era == 2012) {
61 +    EffectiveAreaVersion  = mithep::MuonTools::kMuEAData2012;
62 +    rho = puEnergyDensity->At(0)->RhoKt6PFJetsCentralNeutral();
63 +    
64 +    iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
65 +                                      -rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
66 +                                                                 mu->Eta(),EffectiveAreaVersion)));
67 +  } else if(ctrl.era == 2011) {
68 +    setMuonRhoAndEaVersion(rho,EffectiveAreaVersion,puEnergyDensity);
69 +    iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
70 +                                      -rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
71 +                                                                 mu->Eta(),EffectiveAreaVersion)));
72 +  } else assert(0);
73 +
74 +  bool pass = kTRUE;
75 +  if(!mu->IsGlobalMuon()) pass = kFALSE;
76 +
77 +  //there is not necessarily a global track or a tracker track if is global muon is false
78 +  if(!pass) return pass;
79 +
80 +  if(!mu->IsPFMuon() && ctrl.era==2012)                 pass = kFALSE;
81 +  if(mu->GlobalTrk()->RChi2() > 10)                     pass = kFALSE;
82 +  if(mu->NValidHits() == 0 )                            pass = kFALSE;
83 +  if(mu->NMatches() <= 1)                               pass = kFALSE;
84 +  if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2)   pass = kFALSE;
85 +  if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5)   pass = kFALSE;
86 +  if(mu->BestTrk()->NPixelHits() == 0)                  pass = kFALSE;
87 +  if(mu->NTrkLayersHit() <= 5)                          pass = kFALSE;
88 +  if(iso_sum/mu->Pt() > 0.12 )                          pass = kFALSE;
89 +
90 +  return pass;
91 +
92 +
93 + }
94 +
95 + bool trackToPFMuonSelection( const mithep::Track *track, const mithep::MuonCol * muoncol )
96 + {
97 +        bool pass = false;
98 +  for(UInt_t i=0; i<muoncol->GetEntries(); ++i) {
99 +    const mithep::Muon *mu = muoncol->At(i);
100 +                if(mu->TrackerTrk() == track && (mu->IsGlobalMuon() || mu->IsTrackerMuon()) && mu->IsPFMuon())
101 +                        pass = true;
102 +        }              
103 +                
104 +        return pass;            
105 +                
106 + }
107 +
108   //--------------------------------------------------------------------------------------------------
109   SelectionStatus muonDummyVeto(ControlFlags &ctrl,
110                                const mithep::Muon *muon,
111 <                              const mithep::Vertex &vtx)
111 >                              const mithep::Vertex * vtx)
112   //--------------------------------------------------------------------------------------------------
113   {
114    SelectionStatus status;
# Line 32 | Line 119 | SelectionStatus muonDummyVeto(ControlFla
119   //--------------------------------------------------------------------------------------------------
120   SelectionStatus muonCutBasedVeto(ControlFlags &ctrl,
121                                   const mithep::Muon *muon,
122 <                                 const mithep::Vertex &vtx)
122 >                                 const mithep::Vertex *vtx)
123   //--------------------------------------------------------------------------------------------------
124   {
125    //
# Line 64 | Line 151 | SelectionStatus noPreSelection( ControlF
151   //--------------------------------------------------------------------------------------------------
152   SelectionStatus muonPreSelection( ControlFlags &ctrl,  
153                                    const mithep::Muon * mu,
154 <                                  const mithep::Vertex & vtx,
154 >                                  const mithep::Vertex * vtx,
155                                    const mithep::Array<mithep::PFCandidate> * pfCandidates )
156   //--------------------------------------------------------------------------------------------------
157   {
158    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->BestTrk()->DzCorrected(vtx) < 0.1);  
85  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
159  
160 +  ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
161 +  ret &= ( mu->Pt() >= 5 );
162 +  ret &= ( fabs(mu->Eta()) <= 2.4 );
163 +  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
164 + //   if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
165 + //     ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
166 +  ret &= (  mu->IsoR03SumPt()/mu->Pt() < 0.7 );
167  
168    SelectionStatus status;
169    if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
# Line 92 | Line 172 | SelectionStatus muonPreSelection( Contro
172   }
173  
174  
95 //--------------------------------------------------------------------------------------------------
96 bool isMuFO( const mithep::Muon * mu,
97             const mithep::Vertex & vtx,
98             const mithep::Array<mithep::PFCandidate> * pfCandidates )
99 //--------------------------------------------------------------------------------------------------
100 {
101  bool isgood=true;
102  float reliso = computePFMuonIso(mu,vtx,pfCandidates,0.3)/mu->Pt();
103
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 };
175  
176   //--------------------------------------------------------------------------------------------------
177 < SelectionStatus passSoftMuonSelection( ControlFlags &ctrl,
178 <                                       const mithep::Muon * mu ,
179 <                                       const mithep::Vertex & vtx)
177 > SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,  
178 >                                           const mithep::Muon * mu,
179 >                                           const mithep::Vertex * vtx,
180 >                                           const mithep::Array<mithep::PFCandidate> * pfCandidates )
181   //--------------------------------------------------------------------------------------------------
182   {
183 +  bool ret = true;
184  
185 <  int level=0;
186 <  unsigned failmask=0x0;
187 <  
188 <  // Use tracker track when available
189 <  const mithep::Track *muTrk=0;
190 <  if(mu->HasStandaloneTrk()) { muTrk = mu->StandaloneTrk(); }
191 <  else if(mu->HasTrackerTrk())         { muTrk = mu->TrackerTrk(); }
192 <  else if(mu->HasGlobalTrk())     { muTrk = mu->GlobalTrk(); }
193 <  assert(muTrk);                  
194 <
195 <  if(mu->TrackerTrk()->NHits() < 11 )  
196 <    failmask |= (1<<level);
197 <  level++;
198 <
199 <  if(fabs(muTrk->D0Corrected(vtx)) > 0.2)
200 <    failmask |= (1<<level);
201 <  level++;
202 <
203 <  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 <
185 >  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
186 >  ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
187 >  if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;  
188 >  ret &= ( mu->Pt() >= 5 );
189 >  if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;  
190 >  ret &= ( fabs(mu->Eta()) <= 2.4 );
191 >  if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;  
192 >  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
193 >  if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
194 >  ret &= (mu->HasTrackerTrk() &&
195 >          fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
196 >          fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
197 >  if(ctrl.debug) cout << "d0 and dz? ... " << ret << endl;  
198 >          
199 >  //  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
200 >  //    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
201 >  //  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;  
202 >  //  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);  
203 >  //  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
204  
218  level++;  
219  if( (mu->IsoR03SumPt()/mu->Pt()) > 0.7 ) {
220    failmask |= (1<<level);
221  }
205  
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  
206    SelectionStatus status;
207 <  if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
208 <  if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
207 >  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
208 >  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
209    return status;
210 <
235 <
236 <
237 < };
238 <
239 <
240 <
241 <
242 < //
243 < // Kevin's WW selection
244 < //
210 > }
211   //--------------------------------------------------------------------------------------------------
212 < SelectionStatus passMuonSelection( ControlFlags &ctrl,
213 <                                   const mithep::Muon * mu,
214 <                                   const mithep::Vertex & vtx   )
212 > SelectionStatus muonPreSelectionNoD0DzIP( ControlFlags &ctrl,  
213 >                                          const mithep::Muon * mu,
214 >                                          const mithep::Vertex * vtx,
215 >                                          const mithep::Array<mithep::PFCandidate> * pfCandidates )
216   //--------------------------------------------------------------------------------------------------
217   {
218 <  SelectionStatus status;
252 <  int level=0;
253 <  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 <
218 >  bool ret = true;
219  
220 <  level++;
221 <  if(fabs(muTrk->D0Corrected(vtx))>0.02)   {
222 <    failmask |= (1<<level);
223 <  }
220 >  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
221 >  ret &= ( mu->Pt() >= 5 );
222 >  if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;  
223 >  ret &= ( fabs(mu->Eta()) <= 2.4 );
224 >  if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;  
225 >  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
226 >  if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
227 >          
228 >  //  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
229 >  //    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
230 >  //  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;  
231 >  //  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);  
232 >  //  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
233  
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  */
234  
235 <  if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
236 <  if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
235 >  SelectionStatus status;
236 >  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
237 >  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
238    return status;
239 + }
240  
241  
341 };
242  
243  
244   //--------------------------------------------------------------------------------------------------
245   SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
246                                   const mithep::Muon *mu,
247 <                                 const mithep::Vertex & vtx   )
247 >                                 const mithep::Vertex * vtx   )
248   //--------------------------------------------------------------------------------------------------
249   {
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
250    double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
251  
252 +  //
253 +  // WARNING !!! KH hacked out to sync
254 +  //
255 +  /*
256    double mvaval = muIDMVA->MVAValue_ID(mu->Pt(),
257                                      mu->Eta(),
258                                      mu->IsGlobalMuon(),
# Line 370 | Line 260 | SelectionStatus muonIDMVASelection(Contr
260                                      mu->TrackerTrk()->RChi2(),
261                                      global_rchi2,
262                                      (Double_t)(mu->NValidHits()),
263 <                                    (Double_t)(mu->TrackerTrk()->NHits()),
263 >                                    (Double_t)(mu->TrackerTrk()->NHits()),
264                                      (Double_t)(mu->TrackerTrk()->NPixelHits()),
265                                      (Double_t)(mu->NMatches()),
266                                      mu->TrkKink(),
# Line 381 | Line 271 | SelectionStatus muonIDMVASelection(Contr
271                                      mu->HadS9Energy(),
272                                      mu->EmS9Energy(),
273                                      (Bool_t)(ctrl.debug) );
274 <
274 >  */
275 >  double mvaval = 0.0;
276  
277    SelectionStatus status;
278    bool pass = false;
# Line 394 | Line 285 | SelectionStatus muonIDMVASelection(Contr
285             && fabs(mu->Eta()) > 1.5 && mu->Pt() <= 10 && mvaval >= MUON_IDMVA_CUT_BIN2)  pass = true;
286    else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
287             && fabs(mu->Eta()) > 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN3)  pass = true;
288 <  else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon()
289 <           && (mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated) && mvaval >= MUON_IDMVA_CUT_BIN4)
290 <    pass = true;
288 >  else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon() && mvaval >= MUON_IDMVA_CUT_BIN4 )  pass = true;
289 >  else if( mu->IsGlobalMuon() && !(mu->IsTrackerMuon()) && mvaval >= MUON_IDMVA_CUT_BIN5 )  pass = true;
290 >  
291  
292  
293    if( pass ) {
# Line 427 | Line 318 | void initMuonIDMVA()
318   //--------------------------------------------------------------------------------------------------
319   SelectionStatus muonIDPFSelection(ControlFlags &ctrl,
320                                    const mithep::Muon *mu,
321 <                                  const mithep::Vertex & vtx,
321 >                                  const mithep::Vertex * vtx,
322                                    const mithep::Array<mithep::PFCandidate> * pfCandidates )
323   //--------------------------------------------------------------------------------------------------
324   {
325    bool pass = false;
326    SelectionStatus status; // init'ed to FAIL
436  
437  // check that muon is tracker
438  if( !(mu->IsTrackerMuon()) )   return status;
327  
328    // check that it matches to a PF muon
329    for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
330 +    // tmp
331 +    if( !(PFnoPUflag[i]) ) continue; // my PF no PU hack
332 +
333      const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
334 <    if( pf->HasTrackerTrk() ) {
335 <      if( pf->TrackerTrk() == mu->TrackerTrk() && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
336 <        pass = true;
446 <        break;
447 <      }
334 >    if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
335 >      pass = true;
336 >      break;
337      }
338    }
339 + //      
340 + //      if(mu->IsPFMuon()) pass = true;
341 + //      else pass = false;
342  
343    if( pass ) {
344      status.orStatus(SelectionStatus::LOOSEID);
# Line 455 | Line 347 | SelectionStatus muonIDPFSelection(Contro
347  
348    return status;
349   }
350 + //----------------------------------------------------------------------------------------
351 + SelectionStatus PassWwMuonSel(const mithep::Muon *mu, const mithep::Vertex *vtx,
352 +                              const mithep::Array<mithep::PFCandidate> *pfArr
353 + )
354 + {
355 +  bool passID=true;
356 +  bool passIso=true;
357 +  
358 +  if (mu->Pt() < 5)
359 +    passID = false;
360 +  if (fabs(mu->Eta()) > 2.4)
361 +    passID = false;
362 +  if (mu->BestTrk()->PtErr()/mu->Pt() > 0.1)
363 +    passID = false;
364 +  if (fabs(mu->BestTrk()->DzCorrected(*vtx)) > 0.1)
365 +    passID = false;
366 +  Bool_t isGlobal  = (mu->IsGlobalMuon()) && (mu->BestTrk()->RChi2() < 10) &&
367 +                      (mu->NMatches() > 1) && (mu->NValidHits() > 0);
368 +  Bool_t isTracker = (mu->IsTrackerMuon()) &&
369 +                      (mu->Quality().Quality(mithep::MuonQuality::TMLastStationTight));
370 +  if (!isGlobal && !isTracker)
371 +    passID = false;
372 +  int ntrkhits = (mu->HasTrackerTrk()) ? mu->TrackerTrk()->NHits() : 0;
373 +  if (ntrkhits < 10)
374 +    passID = false;
375 +  if (mu->BestTrk()->NPixelHits() < 1)
376 +    passID = false;
377 +  if (fabs(mu->BestTrk()->D0Corrected(*vtx)) > 0.02)
378 +    passID = false;
379 +  
380 +  // note: this isn't really ww muon isolation
381 +  float pfiso = computePFMuonIso(mu,vtx,pfArr,0.3);
382 +
383 +  if (pfiso > 0.2*mu->Pt())
384 +    passIso = false;
385 +
386 +  SelectionStatus status;
387 +  if(passID)  status |= SelectionStatus::TIGHTIDANDPRE;
388 +  if(passIso) status |= SelectionStatus::TIGHTISOANDPRE;
389 +  
390 +  return status;
391 +
392 + }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines