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.8 by khahn, Mon Apr 30 23:54:34 2012 UTC vs.
Revision 1.34 by dkralph, Mon Jul 16 10:21:12 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 64 | Line 204 | SelectionStatus noPreSelection( ControlF
204   //--------------------------------------------------------------------------------------------------
205   SelectionStatus muonPreSelection( ControlFlags &ctrl,  
206                                    const mithep::Muon * mu,
207 <                                  const mithep::Vertex & vtx,
207 >                                  const mithep::Vertex * vtx,
208                                    const mithep::Array<mithep::PFCandidate> * pfCandidates )
209   //--------------------------------------------------------------------------------------------------
210   {
211    bool ret = true;
212 <  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
213 <  //  bool ret = isMuFO(mu,vtx,pfCandidates);
214 <  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;  
212 >
213 >  ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
214 >  ret &= ( mu->Pt() >= 5 );
215    ret &= ( fabs(mu->Eta()) <= 2.4 );
216 <  if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;  
217 <  ret &=  (mu->IsTrackerMuon() && mu->HasTrackerTrk() &&
218 <           (mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated));
219 <  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;  
216 >  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
217 > //   if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
218 > //     ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
219 >  ret &= (  mu->IsoR03SumPt()/mu->Pt() < 0.7 );
220  
221    SelectionStatus status;
222    if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
# Line 92 | Line 225 | SelectionStatus muonPreSelection( Contro
225   }
226  
227  
228 +
229   //--------------------------------------------------------------------------------------------------
230 < bool isMuFO( const mithep::Muon * mu,
231 <             const mithep::Vertex & vtx,
232 <             const mithep::Array<mithep::PFCandidate> * pfCandidates )
230 > SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,  
231 >                                           const mithep::Muon * mu,
232 >                                           const mithep::Vertex * vtx,
233 >                                           const mithep::Array<mithep::PFCandidate> * pfCandidates )
234   //--------------------------------------------------------------------------------------------------
235   {
236 <  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 <
236 >  bool ret = true;
237  
238 <  // comment for more stats for mu fake shape
239 <  if( mu->Pt() < 20 ) {
240 <    if( reliso > 3 ) isgood=false;    
241 <  } else {
242 <    if( reliso > 5 ) isgood=false;    
243 <  }
238 >  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
239 >  ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
240 >  if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;  
241 >  ret &= ( mu->Pt() >= 5 );
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() );
246 >  if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
247 >  ret &= (mu->HasTrackerTrk() &&
248 >          fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
249 >          fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
250 >  if(ctrl.debug) cout << "d0 and dz? ... " << ret << endl;  
251 >          
252 >  //  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
253 >  //    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
254 >  //  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;  
255 >  //  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);  
256 >  //  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
257  
139  return isgood;
140 };
258  
259 +  SelectionStatus status;
260 +  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
261 +  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
262 +  return status;
263 + }
264   //--------------------------------------------------------------------------------------------------
265 < SelectionStatus passSoftMuonSelection( ControlFlags &ctrl,
266 <                                       const mithep::Muon * mu ,
267 <                                       const mithep::Vertex & vtx)
268 < //--------------------------------------------------------------------------------------------------
147 < {
148 <
149 <  int level=0;
150 <  unsigned failmask=0x0;
151 <  
152 <  // Use tracker track when available
153 <  const mithep::Track *muTrk=0;
154 <  if(mu->HasStandaloneTrk()) { muTrk = mu->StandaloneTrk(); }
155 <  else if(mu->HasTrackerTrk())         { muTrk = mu->TrackerTrk(); }
156 <  else if(mu->HasGlobalTrk())     { muTrk = mu->GlobalTrk(); }
157 <  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 )
265 > SelectionStatus muonPreSelectionNoD0DzIP( ControlFlags &ctrl,  
266 >                                          const mithep::Muon * mu,
267 >                                          const mithep::Vertex * vtx,
268 >                                          const mithep::Array<mithep::PFCandidate> * pfCandidates )
269   //--------------------------------------------------------------------------------------------------
270   {
271 <  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 <  }
271 >  bool ret = true;
272  
273 +  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
274 +  ret &= ( mu->Pt() >= 5 );
275 +  if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;  
276 +  ret &= ( fabs(mu->Eta()) <= 2.4 );
277 +  if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;  
278 +  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
279 +  if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
280 +          
281 +  //  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
282 +  //    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
283 +  //  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;  
284 +  //  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);  
285 +  //  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
286  
218  level++;  
219  if( (mu->IsoR03SumPt()/mu->Pt()) > 0.7 ) {
220    failmask |= (1<<level);
221  }
287  
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  
288    SelectionStatus status;
289 <  if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
290 <  if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
289 >  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
290 >  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
291    return status;
292 <
235 <
236 <
237 < };
238 <
239 <
240 <
241 <
242 < //
243 < // Kevin's WW selection
244 < //
292 > }
293   //--------------------------------------------------------------------------------------------------
294 < SelectionStatus passMuonSelection( ControlFlags &ctrl,
295 <                                   const mithep::Muon * mu,
296 <                                   const mithep::Vertex & vtx   )
294 > SelectionStatus muonPreSelectionNoIP( ControlFlags &ctrl,  
295 >                                          const mithep::Muon * mu,
296 >                                          const mithep::Vertex * vtx,
297 >                                          const mithep::Array<mithep::PFCandidate> * pfCandidates )
298   //--------------------------------------------------------------------------------------------------
299   {
300 <  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 <
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 <  */
300 >  bool ret = true;
301  
302 <  if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
303 <  if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
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 >          
309 >  SelectionStatus status;
310 >  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
311 >  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
312    return status;
313 <
340 <
341 < };
342 <
343 <
313 > }
314   //--------------------------------------------------------------------------------------------------
315   SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
316                                   const mithep::Muon *mu,
317 <                                 const mithep::Vertex & vtx   )
317 >                                 const mithep::Vertex * vtx   )
318   //--------------------------------------------------------------------------------------------------
319   {
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
320    double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
321  
322 +  //
323 +  // WARNING !!! KH hacked out to sync
324 +  //
325 +  /*
326    double mvaval = muIDMVA->MVAValue_ID(mu->Pt(),
327                                      mu->Eta(),
328                                      mu->IsGlobalMuon(),
# Line 370 | Line 330 | SelectionStatus muonIDMVASelection(Contr
330                                      mu->TrackerTrk()->RChi2(),
331                                      global_rchi2,
332                                      (Double_t)(mu->NValidHits()),
333 <                                    (Double_t)(mu->TrackerTrk()->NHits()),
333 >                                    (Double_t)(mu->TrackerTrk()->NHits()),
334                                      (Double_t)(mu->TrackerTrk()->NPixelHits()),
335                                      (Double_t)(mu->NMatches()),
336                                      mu->TrkKink(),
# Line 381 | Line 341 | SelectionStatus muonIDMVASelection(Contr
341                                      mu->HadS9Energy(),
342                                      mu->EmS9Energy(),
343                                      (Bool_t)(ctrl.debug) );
344 <
344 >  */
345 >  double mvaval = 0.0;
346  
347    SelectionStatus status;
348    bool pass = false;
# Line 394 | Line 355 | SelectionStatus muonIDMVASelection(Contr
355             && fabs(mu->Eta()) > 1.5 && mu->Pt() <= 10 && mvaval >= MUON_IDMVA_CUT_BIN2)  pass = true;
356    else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
357             && fabs(mu->Eta()) > 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN3)  pass = true;
358 <  else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon()
359 <           && (mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated) && mvaval >= MUON_IDMVA_CUT_BIN4)
360 <    pass = true;
358 >  else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon() && mvaval >= MUON_IDMVA_CUT_BIN4 )  pass = true;
359 >  else if( mu->IsGlobalMuon() && !(mu->IsTrackerMuon()) && mvaval >= MUON_IDMVA_CUT_BIN5 )  pass = true;
360 >  
361  
362  
363    if( pass ) {
# Line 423 | Line 384 | void initMuonIDMVA()
384                         mithep::MuonIDMVA::kIDV0,
385                         kTRUE, weightFiles);
386   }
387 +
388 + //--------------------------------------------------------------------------------------------------
389 + SelectionStatus muonIDPFSelection(ControlFlags &ctrl,
390 +                                  const mithep::Muon *mu,
391 +                                  const mithep::Vertex * vtx,
392 +                                  const mithep::Array<mithep::PFCandidate> * pfCandidates )
393 + //--------------------------------------------------------------------------------------------------
394 + {
395 +  bool pass = false;
396 +  SelectionStatus status; // init'ed to FAIL
397 +
398 +  // check that it matches to a PF muon
399 +  for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
400 +    // tmp
401 +    if( !(PFnoPUflag[i]) ) continue; // my PF no PU hack
402 +
403 +    const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
404 +    if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
405 +      pass = true;
406 +      break;
407 +    }
408 +  }
409 + //      
410 + //      if(mu->IsPFMuon()) pass = true;
411 + //      else pass = false;
412 +
413 +  if( pass ) {
414 +    status.orStatus(SelectionStatus::LOOSEID);
415 +    status.orStatus(SelectionStatus::TIGHTID);
416 +  }
417 +
418 +  return status;
419 + }
420 + //----------------------------------------------------------------------------------------
421 + SelectionStatus PassWwMuonSel(const mithep::Muon *mu, const mithep::Vertex *vtx,
422 +                              const mithep::Array<mithep::PFCandidate> *pfArr
423 + )
424 + {
425 +  bool passID=true;
426 +  bool passIso=true;
427 +  
428 +  if (mu->Pt() < 5)
429 +    passID = false;
430 +  if (fabs(mu->Eta()) > 2.4)
431 +    passID = false;
432 +  if (mu->BestTrk()->PtErr()/mu->Pt() > 0.1)
433 +    passID = false;
434 +  if (fabs(mu->BestTrk()->DzCorrected(*vtx)) > 0.1)
435 +    passID = false;
436 +  Bool_t isGlobal  = (mu->IsGlobalMuon()) && (mu->BestTrk()->RChi2() < 10) &&
437 +                      (mu->NMatches() > 1) && (mu->NValidHits() > 0);
438 +  Bool_t isTracker = (mu->IsTrackerMuon()) &&
439 +                      (mu->Quality().Quality(mithep::MuonQuality::TMLastStationTight));
440 +  if (!isGlobal && !isTracker)
441 +    passID = false;
442 +  int ntrkhits = (mu->HasTrackerTrk()) ? mu->TrackerTrk()->NHits() : 0;
443 +  if (ntrkhits < 10)
444 +    passID = false;
445 +  if (mu->BestTrk()->NPixelHits() < 1)
446 +    passID = false;
447 +  if (fabs(mu->BestTrk()->D0Corrected(*vtx)) > 0.02)
448 +    passID = false;
449 +  
450 +  // note: this isn't really ww muon isolation
451 +  float pfiso = computePFMuonIso(mu,vtx,pfArr,0.3);
452 +
453 +  if (pfiso > 0.2*mu->Pt())
454 +    passIso = false;
455 +
456 +  SelectionStatus status;
457 +  if(passID)  status |= SelectionStatus::TIGHTIDANDPRE;
458 +  if(passIso) status |= SelectionStatus::TIGHTISOANDPRE;
459 +  
460 +  return status;
461 +
462 + }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines