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.14 by khahn, Sun May 6 23:51:01 2012 UTC

# Line 74 | Line 74 | SelectionStatus muonPreSelection( Contro
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 );
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;  
81 >  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
82 >  if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;  
83 >  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
84 >    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 >
89  
90    SelectionStatus status;
91    if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
# Line 92 | Line 94 | SelectionStatus muonPreSelection( Contro
94   }
95  
96  
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 };
97  
98   //--------------------------------------------------------------------------------------------------
99 < SelectionStatus passSoftMuonSelection( ControlFlags &ctrl,
100 <                                       const mithep::Muon * mu ,
101 <                                       const mithep::Vertex & vtx)
102 < //--------------------------------------------------------------------------------------------------
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 )
99 > SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,  
100 >                                           const mithep::Muon * mu,
101 >                                           const mithep::Vertex & vtx,
102 >                                           const mithep::Array<mithep::PFCandidate> * pfCandidates )
103   //--------------------------------------------------------------------------------------------------
104   {
105 <  int level=0;
106 <  unsigned failmask=0x0;
107 <
108 <  if(mu->Pt() < 5) {
109 <    failmask |= (1<<level);
110 <  }
111 <
112 <  level++;
113 <  if(fabs(mu->Eta()) > 2.4) {
114 <    failmask |= (1<<level);
115 <  }
116 <
117 <  level++;
118 <  if(!(mu->IsGlobalMuon())) {
119 <    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 <
105 >  bool ret = true;
106 >  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
107 >  ret &= ( mu->Ip3dPVSignificance() < 100 );
108 >  if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;  
109 >  ret &= ( mu->Pt() >= 5 );
110 >  if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;  
111 >  ret &= ( fabs(mu->Eta()) <= 2.4 );
112 >  if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;  
113 >  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
114 >  if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;  
115 >  //  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
116 >  //    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
117 >  //  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;  
118 >  //  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);  
119 >  //  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
120  
121  
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 {
122    SelectionStatus status;
123 <  int level=0;
124 <  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);
123 >  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
124 >  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
125    return status;
126 + }
127  
128  
341 };
129  
130  
131   //--------------------------------------------------------------------------------------------------
# Line 347 | Line 134 | SelectionStatus muonIDMVASelection(Contr
134                                   const mithep::Vertex & vtx   )
135   //--------------------------------------------------------------------------------------------------
136   {
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
137    double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
138  
139 +  //
140 +  // WARNING !!! KH hacked out to sync
141 +  //
142 +  /*
143    double mvaval = muIDMVA->MVAValue_ID(mu->Pt(),
144                                      mu->Eta(),
145                                      mu->IsGlobalMuon(),
# Line 370 | Line 147 | SelectionStatus muonIDMVASelection(Contr
147                                      mu->TrackerTrk()->RChi2(),
148                                      global_rchi2,
149                                      (Double_t)(mu->NValidHits()),
150 <                                    (Double_t)(mu->TrackerTrk()->NHits()),
150 >                                    (Double_t)(mu->TrackerTrk()->NHits()),
151                                      (Double_t)(mu->TrackerTrk()->NPixelHits()),
152                                      (Double_t)(mu->NMatches()),
153                                      mu->TrkKink(),
# Line 381 | Line 158 | SelectionStatus muonIDMVASelection(Contr
158                                      mu->HadS9Energy(),
159                                      mu->EmS9Energy(),
160                                      (Bool_t)(ctrl.debug) );
161 <
161 >  */
162 >  double mvaval = 0.0;
163  
164    SelectionStatus status;
165    bool pass = false;
# Line 394 | Line 172 | SelectionStatus muonIDMVASelection(Contr
172             && fabs(mu->Eta()) > 1.5 && mu->Pt() <= 10 && mvaval >= MUON_IDMVA_CUT_BIN2)  pass = true;
173    else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
174             && fabs(mu->Eta()) > 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN3)  pass = true;
175 <  else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon()
176 <           && (mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated) && mvaval >= MUON_IDMVA_CUT_BIN4)
177 <    pass = true;
175 >  else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon() && mvaval >= MUON_IDMVA_CUT_BIN4 )  pass = true;
176 >  else if( mu->IsGlobalMuon() && !(mu->IsTrackerMuon()) && mvaval >= MUON_IDMVA_CUT_BIN5 )  pass = true;
177 >  
178  
179  
180    if( pass ) {
# Line 433 | Line 211 | SelectionStatus muonIDPFSelection(Contro
211   {
212    bool pass = false;
213    SelectionStatus status; // init'ed to FAIL
436  
437  // check that muon is tracker
438  if( !(mu->IsTrackerMuon()) )   return status;
214  
215    // check that it matches to a PF muon
216    for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
217      const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
218      if( pf->HasTrackerTrk() ) {
219 <      if( pf->TrackerTrk() == mu->TrackerTrk() && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
219 >      if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
220          pass = true;
221          break;
222        }
223      }
224    }
225  
226 <  if( pass ) {
226 >  //
227 >  // NB : ipsig here for reference synch ...
228 >  //
229 >  if( pass && mu->Ip3dPVSignificance() < 4 ) {
230      status.orStatus(SelectionStatus::LOOSEID);
231      status.orStatus(SelectionStatus::TIGHTID);
232    }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines