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.5 by khahn, Thu Apr 26 07:02:28 2012 UTC vs.
Revision 1.28 by dkralph, Wed Jun 13 14:11:22 2012 UTC

# Line 13 | Line 13
13   mithep::MuonIDMVA * muIDMVA;
14   mithep::MuonTools muTools;
15  
16 + extern vector<bool> PFnoPUflag;
17   extern Float_t computePFMuonIso(const mithep::Muon *mu,
18 <                        const mithep::Vertex & vtx,
18 >                        const mithep::Vertex * vtx,
19                          const mithep::Array<mithep::PFCandidate> * pfCandidates,
20                          const Double_t dRMax);
21  
22 + bool muon2012CutBasedIDTight(const mithep::Muon *mu, const mithep::Vertex * vtx, const mithep::Array<mithep::PFCandidate> * pfCandidates, const mithep::Array<mithep::PileupEnergyDensity> * puEnergyDensity)
23 + {
24 +
25 +  mithep::MuonTools   muT;
26 +
27 +  Float_t charged_iso = 0;
28 +  Float_t neutral_iso = 0;
29 +  Float_t gamma_iso = 0;
30 +  
31 +  for(int k=0; k<pfCandidates->GetEntries(); ++k) {
32 +    
33 +    if( !(PFnoPUflag[k]) ) continue;
34 +    const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[k]);
35 +    
36 +    Double_t dr = mithep::MathUtils::DeltaR(mu->Phi(),mu->Eta(), pf->Phi(), pf->Eta());
37 +    if (dr > 0.4) continue;
38 +    
39 +    if (pf->Charge() != 0 && (pf->HasTrackerTrk()||pf->HasGsfTrk()) ) {
40 +      
41 +      if (abs(pf->PFType()) == mithep::PFCandidate::eElectron || abs(pf->PFType()) == mithep::PFCandidate::eMuon) continue;
42 +      
43 +      charged_iso += pf->Pt();
44 +    }
45 +    
46 +    else if (abs(pf->PFType()) == mithep::PFCandidate::eGamma) {
47 +      
48 +      gamma_iso += pf->Pt();
49 +    }
50 +    
51 +    else {
52 +      neutral_iso += pf->Pt();
53 +    }
54 +  }
55 +  
56 +  mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion  = mithep::MuonTools::kMuEAData2012;
57 +  float rho = puEnergyDensity->At(0)->RhoKt6PFJetsCentralNeutral();
58 +  
59 +  double iso_sum = charged_iso + fmax(0.0,(gamma_iso + neutral_iso
60 +                                        -rho*muT.MuonEffectiveArea(muT.kMuGammaAndNeutralHadronIso04,
61 +                                                                   mu->Eta(),EffectiveAreaVersion)));
62 +
63 +  bool pass = kTRUE;
64 +  if(!mu->IsGlobalMuon()) pass = kFALSE;
65 +
66 +  //there is not necessarily a global track or a tracker track if is global muon is false
67 +  if(!pass) return pass;
68 +
69 +  if(!mu->IsPFMuon()) pass = kFALSE;
70 +  if(mu->GlobalTrk()->RChi2() > 10) pass= kFALSE;
71 +  if(mu->NValidHits() == 0 ) pass= kFALSE;
72 +  if(mu->NMatches() <= 1) pass = kFALSE;
73 +  if(fabs(mu->TrackerTrk()->D0Corrected(*vtx)) > 0.2) pass = kFALSE;
74 +  if(fabs(mu->TrackerTrk()->DzCorrected(*vtx)) > 0.5) pass = kFALSE;
75 +  if(mu->BestTrk()->NPixelHits() == 0) pass = kFALSE;
76 +  if(mu->NTrkLayersHit() <= 5) pass = kFALSE;
77 +  if(iso_sum/mu->Pt() > 0.12 ) pass = kFALSE;
78 +
79 +  return pass;
80 +
81 +
82 + }
83 +
84 + bool trackToPFMuonSelection( const mithep::Track *track, const mithep::MuonCol * muoncol )
85 + {
86 +        bool pass = false;
87 +  for(UInt_t i=0; i<muoncol->GetEntries(); ++i) {
88 +    const mithep::Muon *mu = muoncol->At(i);
89 +                if(mu->TrackerTrk() == track && (mu->IsGlobalMuon() || mu->IsTrackerMuon()) && mu->IsPFMuon())
90 +                        pass = true;
91 +        }              
92 +                
93 +        return pass;            
94 +                
95 + }
96 +
97 + //--------------------------------------------------------------------------------------------------
98 + SelectionStatus muonDummyVeto(ControlFlags &ctrl,
99 +                              const mithep::Muon *muon,
100 +                              const mithep::Vertex * vtx)
101 + //--------------------------------------------------------------------------------------------------
102 + {
103 +  SelectionStatus status;
104 +  status.setStatus(SelectionStatus::PRESELECTION);
105 +  return status;  
106 + }
107 +
108 + //--------------------------------------------------------------------------------------------------
109 + SelectionStatus muonCutBasedVeto(ControlFlags &ctrl,
110 +                                 const mithep::Muon *muon,
111 +                                 const mithep::Vertex *vtx)
112 + //--------------------------------------------------------------------------------------------------
113 + {
114 +  //
115 +  // Loose cut-based ID for isolation veto
116 +  //
117 +  bool ret = true;
118 +  
119 +  if(!(muon->IsGlobalMuon() || muon->IsTrackerMuon())) ret = false;
120 +  if( muon->NValidHits() < 11 )                        ret = false;
121 +  if( fabs(muon->Ip3dPVSignificance()) >= 4 )          ret = false;
122 +
123 +  SelectionStatus status;
124 +  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
125 +  return status;
126 + }
127 +
128 +
129   //--------------------------------------------------------------------------------------------------
130   SelectionStatus noPreSelection( ControlFlags &ctrl,  const mithep::Muon * mu )
131  
# Line 32 | Line 140 | SelectionStatus noPreSelection( ControlF
140   //--------------------------------------------------------------------------------------------------
141   SelectionStatus muonPreSelection( ControlFlags &ctrl,  
142                                    const mithep::Muon * mu,
143 <                                  const mithep::Vertex & vtx,
143 >                                  const mithep::Vertex * vtx,
144                                    const mithep::Array<mithep::PFCandidate> * pfCandidates )
145   //--------------------------------------------------------------------------------------------------
146   {
147    bool ret = true;
148 <  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
149 <  //  bool ret = isMuFO(mu,vtx,pfCandidates);
150 <  if(ctrl.debug) cout << "iS fo? ... " << ret << endl;  
151 <  ret &= ( fabs(mu->Ip3dPVSignificance()) < 4 );
152 <  if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;  
153 <  ret &= ( mu->Pt() > 5 );
154 <  if(ctrl.debug) cout << "and >5 GeV ? ... " << ret << endl;  
155 <  ret &= ( fabs(mu->Eta()) < 2.4 );
48 <  if(ctrl.debug) cout << "and < 2.4 eta ? ... " << ret << endl;  
49 <  ret &=  (mu->IsTrackerMuon() && mu->HasTrackerTrk());
50 <  if(ctrl.debug) cout << "and isTrackerMuon ? ... " << ret << endl;  
51 <  ret &=  (mu->IsoR03SumPt()/mu->Pt() < 0.7 );
52 <  if(ctrl.debug) cout << "and loose trk iso ? ... " << ret << endl;  
148 >
149 >  ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
150 >  ret &= ( mu->Pt() >= 5 );
151 >  ret &= ( fabs(mu->Eta()) <= 2.4 );
152 >  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
153 > //   if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
154 > //     ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
155 >  ret &= (  mu->IsoR03SumPt()/mu->Pt() < 0.7 );
156  
157    SelectionStatus status;
158    if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
# Line 58 | Line 161 | SelectionStatus muonPreSelection( Contro
161   }
162  
163  
61 //--------------------------------------------------------------------------------------------------
62 bool isMuFO( const mithep::Muon * mu,
63             const mithep::Vertex & vtx,
64             const mithep::Array<mithep::PFCandidate> * pfCandidates )
65 //--------------------------------------------------------------------------------------------------
66 {
67  bool isgood=true;
68  float reliso = computePFMuonIso(mu,vtx,pfCandidates,0.3)/mu->Pt();
69
70  if( mu->Pt() < 5  )         isgood=false;                          
71  if ( fabs(mu->BestTrk()->DzCorrected(vtx)) > 0.2 ) isgood=false;   // needed to remove cosmics in HF sample
72
73  //
74
75  // HF seems to not want tkquality, SS does
76  if( mu->Pt()>20 ) {
77    if ( !((mu->IsGlobalMuon()) && mu->NSegments()>0  ) ) isgood=false; //&& mu->nValidHits>0
78  } else {
79    if ( !((mu->IsGlobalMuon()) && mu->NSegments()>0 ) //&& //&& mu->nValidHits>0
80         && !( (mu->IsTrackerMuon()) &&  (mu->Quality().QualityMask().Mask() & muTools.kTMOneStationLoose)
81               )
82         )   isgood=false;
83  }
84
85
86  //
87  // for HF MC to agree w/ data, cannot put any Tk quality bits
88  // but to get to 5% closure in the SS test, we need at least Tk oneStationLoose
89
90  /*
91 && ( (mu->qualityBits & kTMOneStationLoose) ||
92                                               (mu->qualityBits & kTMLastStationOptimizedBarrelLowPtLoose) ||
93                                               (mu->qualityBits & kTMLastStationOptimizedLowPtLoose)
94                                               )
95  */
96
97
98  // comment for more stats for mu fake shape
99  if( mu->Pt() < 20 ) {
100    if( reliso > 3 ) isgood=false;    
101  } else {
102    if( reliso > 5 ) isgood=false;    
103  }
104
105  return isgood;
106 };
164  
165   //--------------------------------------------------------------------------------------------------
166 < SelectionStatus passSoftMuonSelection( ControlFlags &ctrl,
167 <                                       const mithep::Muon * mu ,
168 <                                       const mithep::Vertex & vtx)
166 > SelectionStatus muonReferencePreSelection( ControlFlags &ctrl,  
167 >                                           const mithep::Muon * mu,
168 >                                           const mithep::Vertex * vtx,
169 >                                           const mithep::Array<mithep::PFCandidate> * pfCandidates )
170   //--------------------------------------------------------------------------------------------------
171   {
172 +  bool ret = true;
173  
174 <  int level=0;
175 <  unsigned failmask=0x0;
176 <  
177 <  // Use tracker track when available
178 <  const mithep::Track *muTrk=0;
179 <  if(mu->HasStandaloneTrk()) { muTrk = mu->StandaloneTrk(); }
180 <  else if(mu->HasTrackerTrk())         { muTrk = mu->TrackerTrk(); }
181 <  else if(mu->HasGlobalTrk())     { muTrk = mu->GlobalTrk(); }
182 <  assert(muTrk);                  
183 <
184 <  if(mu->TrackerTrk()->NHits() < 11 )  
185 <    failmask |= (1<<level);
186 <  level++;
187 <
188 <  if(fabs(muTrk->D0Corrected(vtx)) > 0.2)
189 <    failmask |= (1<<level);
190 <  level++;
191 <
192 <  if(fabs(muTrk->DzCorrected(vtx)) > 0.1)
134 <    failmask |= (1<<level);
135 <  level++;
136 <
137 <  if(!(mu->IsTrackerMuon()))
138 <    failmask |= (1<<level);
139 <  level++;
140 <
141 <  //  if(!(mu->Quality().QualityMask().Mask() & muTools.kTMLastStationAngTight))
142 <  if(!(mu->Quality().QualityMask().Mask() & muTools.kTMLastStationTight))
143 <    failmask |= (1<<level);      
144 <  level++;
145 <
146 <  Double_t iso = (mu->IsoR03SumPt() + mu->IsoR03EmEt() + mu->IsoR03HadEt())/mu->Pt();
147 <  if(mu->Pt()>20 && iso<0.1)
148 <    failmask |= (1<<level);
149 <
150 <  bool passtight  = !(failmask);
151 <  bool passloose  = !(failmask);
152 <  return SelectionStatus( passtight | passloose );
153 <  
154 < }
155 <
156 < //--------------------------------------------------------------------------------------------------
157 < //unsigned passMuonSelectionZZ( const mithep::TMuon * mu ) {
158 < SelectionStatus passMuonSelectionZZ( ControlFlags &ctrl, const mithep::Muon * mu )
159 < //--------------------------------------------------------------------------------------------------
160 < {
161 <  int level=0;
162 <  unsigned failmask=0x0;
163 <
164 <  if(mu->Pt() < 5) {
165 <    failmask |= (1<<level);
166 <  }
167 <
168 <  level++;
169 <  if(fabs(mu->Eta()) > 2.4) {
170 <    failmask |= (1<<level);
171 <  }
172 <
173 <  level++;
174 <  if(!(mu->IsGlobalMuon())) {
175 <    failmask |= (1<<level);
176 <  }
177 <
178 <  level++;
179 <  if(mu->TrackerTrk()->NHits()  < 10) {
180 <    failmask |= (1<<level);
181 <  }
182 <
174 >  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
175 >  ret &= ( fabs(mu->Ip3dPVSignificance()) < 100 );
176 >  if(ctrl.debug) cout << "and pass IP (" << mu->Ip3dPVSignificance() << ") ? ... " << ret << endl;  
177 >  ret &= ( mu->Pt() >= 5 );
178 >  if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;  
179 >  ret &= ( fabs(mu->Eta()) <= 2.4 );
180 >  if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;  
181 >  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
182 >  if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
183 >  ret &= (mu->HasTrackerTrk() &&
184 >          fabs(mu->TrackerTrk()->D0Corrected(*vtx)) < 0.5 &&
185 >          fabs(mu->TrackerTrk()->DzCorrected(*vtx)) < 1.0);
186 >  if(ctrl.debug) cout << "d0 and dz? ... " << ret << endl;  
187 >          
188 >  //  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
189 >  //    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
190 >  //  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;  
191 >  //  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);  
192 >  //  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
193  
184  level++;  
185  if( (mu->IsoR03SumPt()/mu->Pt()) > 0.7 ) {
186    failmask |= (1<<level);
187  }
194  
189  //  if(muon->nPixHits   < 1)     return false;
190  //  if(muon->muNchi2    > 10)    return false;
191  //  if(muon->nMatch     < 2)     return false;
192  //  if(muon->nValidHits         < 1)     return false;
193  //  if(muon->pterr/muon->pt > 0.1)   return false;
194  //  if(fabs(muon->dz)       > 0.1)   return false;  
195  
195    SelectionStatus status;
196 <  if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
197 <  if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
196 >  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
197 >  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
198    return status;
199 + }
200  
201
202
203 };
204
205
206
207
208 //
209 // Kevin's WW selection
210 //
201   //--------------------------------------------------------------------------------------------------
202 < SelectionStatus passMuonSelection( ControlFlags &ctrl,
203 <                                   const mithep::Muon * mu,
204 <                                   const mithep::Vertex & vtx   )
202 > SelectionStatus muonPreSelectionNoD0DzIP( ControlFlags &ctrl,  
203 >                                          const mithep::Muon * mu,
204 >                                          const mithep::Vertex * vtx,
205 >                                          const mithep::Array<mithep::PFCandidate> * pfCandidates )
206   //--------------------------------------------------------------------------------------------------
207   {
208 <  SelectionStatus status;
218 <  int level=0;
219 <  unsigned failmask=0x0;
220 <
221 <  // Use tracker track when available
222 <  const mithep::Track *muTrk=0;
223 <  if(mu->HasTrackerTrk())         { muTrk = mu->TrackerTrk(); }
224 <  else if(mu->HasGlobalTrk())     { muTrk = mu->GlobalTrk(); }
225 <  else if(mu->HasStandaloneTrk()) { muTrk = mu->StandaloneTrk(); }
226 <  assert(muTrk);                  
227 <  
228 <
229 <
230 <  // 0x1
231 <  if(muTrk->Pt() < 5) {
232 <    failmask |= (1<<level);
233 <  }
234 <
235 <  // 0x2
236 <  level++;
237 <  if(fabs(muTrk->Eta()) > 2.4) {
238 <    failmask |= (1<<level);
239 <  }
240 <
241 <  // 0x4
242 <  level++;
243 <  if(muTrk->PtErr()/muTrk->Pt() > 0.1)   {
244 <    failmask |= (1<<level);
245 <  }
246 <
247 <  // 0x8
248 <  level++;
249 <  if( fabs(muTrk->DzCorrected(vtx)) > 0.1 )   {
250 <    failmask |= (1<<level);
251 <  }
252 <
253 <  double muNchi2;
254 <  if(mu->HasStandaloneTrk())      { muNchi2 = mu->StandaloneTrk()->RChi2(); }
255 <  else if(mu->HasGlobalTrk())     { muNchi2 = mu->GlobalTrk()->RChi2();     }
256 <  else if(mu->HasTrackerTrk())    { muNchi2 = mu->TrackerTrk()->RChi2();    }
257 <
258 <  unsigned qualityBits = mu->Quality().QualityMask().Mask();
259 <  
260 <  Bool_t isGlobal  = (mu->IsGlobalMuon()) && (muNchi2 < 10) && (mu->NMatches() > 1) && (mu->NValidHits() > 0);
261 <  Bool_t isTracker = (mu->IsTrackerMuon() ) && (qualityBits & muTools.kTMLastStationTight);
262 <
263 <  // 0x10
264 <  level++;
265 <  if((!isGlobal && !isTracker) || !(mu->HasTrackerTrk())) {
266 <    failmask |= (1<<level);
267 <    status.setStatus(SelectionStatus::FAIL);
268 <    return status;    
269 <  }
270 <
271 <  // 0x20
272 <  level++;
273 <  assert(mu->HasTrackerTrk());
274 <  assert(mu->TrackerTrk());
275 <  if( mu->TrackerTrk()->NHits() < 10 ) {
276 <    failmask |= (1<<level);
277 <  }
278 <
279 <  level++;
280 <  if(muTrk->NPixelHits()          < 1) {
281 <    failmask |= (1<<level);
282 <  }
283 <
208 >  bool ret = true;
209  
210 <  level++;
211 <  if(fabs(muTrk->D0Corrected(vtx))>0.02)   {
212 <    failmask |= (1<<level);
213 <  }
210 >  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
211 >  ret &= ( mu->Pt() >= 5 );
212 >  if(ctrl.debug) cout << "and >5 GeV (" << mu->Pt() << ") ? ... " << ret << endl;  
213 >  ret &= ( fabs(mu->Eta()) <= 2.4 );
214 >  if(ctrl.debug) cout << "and < 2.4 eta (" << mu->Eta() << ")? ... " << ret << endl;  
215 >  ret &= ( mu->IsTrackerMuon() || mu->IsGlobalMuon() );
216 >  if(ctrl.debug) cout << "is Tracker or Global? ... " << ret << endl;
217 >          
218 >  //  if( mu->IsTrackerMuon() && mu->HasTrackerTrk() && !mu->IsGlobalMuon() )  
219 >  //    ret &=  ( mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated);
220 >  //  if(ctrl.debug) cout << "if isTrackerMuon, arbitrated ? ... " << ret << endl;  
221 >  //  ret &= ( mu->BestTrk()->DzCorrected(vtx) < 0.1);  
222 >  //  if( ctrl.debug ) cout << "elepresel : ret after dZcorr : " << ret <<  endl;
223  
290  /*
291  if(mu->pt>20) {
292    if(fabs(mu->d0)>0.02)   {
293      failmask |= (1<<level);
294    }
295  } else {
296    if(fabs(mu->d0)>0.01)  {
297    failmask |= (1<<level);
298    }
299  }
300  */
224  
225 <  if( !failmask ) status.orStatus(SelectionStatus::LOOSEID);
226 <  if( !failmask ) status.orStatus(SelectionStatus::TIGHTID);
225 >  SelectionStatus status;
226 >  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
227 >  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
228    return status;
229 + }
230  
231  
307 };
232  
233  
234   //--------------------------------------------------------------------------------------------------
235   SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
236                                   const mithep::Muon *mu,
237 <                                 const mithep::Vertex & vtx   )
237 >                                 const mithep::Vertex * vtx   )
238   //--------------------------------------------------------------------------------------------------
239   {
316  assert(mu->IsTrackerMuon()); // must be
317  assert(mu->HasTrackerTrk());
318  assert(mu->TrackerTrk());
319
320  /*
321  cerr << "TrackerTrk: " << mu->TrackerTrk() << endl;
322  flush(cerr);
323  cerr << "Rchi2: " << mu->TrackerTrk()->RChi2() << endl;
324  flush(cerr);
325
326  cerr << "calling MVa ... "<< endl;
327  flush(cerr);
328  */
329
240    double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
241  
242 +  //
243 +  // WARNING !!! KH hacked out to sync
244 +  //
245 +  /*
246    double mvaval = muIDMVA->MVAValue_ID(mu->Pt(),
247                                      mu->Eta(),
248                                      mu->IsGlobalMuon(),
# Line 336 | Line 250 | SelectionStatus muonIDMVASelection(Contr
250                                      mu->TrackerTrk()->RChi2(),
251                                      global_rchi2,
252                                      (Double_t)(mu->NValidHits()),
253 <                                    (Double_t)(mu->TrackerTrk()->NHits()),
253 >                                    (Double_t)(mu->TrackerTrk()->NHits()),
254                                      (Double_t)(mu->TrackerTrk()->NPixelHits()),
255                                      (Double_t)(mu->NMatches()),
256                                      mu->TrkKink(),
# Line 347 | Line 261 | SelectionStatus muonIDMVASelection(Contr
261                                      mu->HadS9Energy(),
262                                      mu->EmS9Energy(),
263                                      (Bool_t)(ctrl.debug) );
264 <
264 >  */
265 >  double mvaval = 0.0;
266  
267    SelectionStatus status;
268    bool pass = false;
269    
270    if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
271 <      && fabs(mu->Eta()) < 1.5 && mu->Pt() < 10 && mvaval >= MUON_IDMVA_CUT_BIN0)  pass = true;
271 >      && fabs(mu->Eta()) <= 1.5 && mu->Pt() <= 10 && mvaval >= MUON_IDMVA_CUT_BIN0)  pass = true;
272    else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
273 <           && fabs(mu->Eta()) < 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN1)  pass = true;
273 >           && fabs(mu->Eta()) <= 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN1)  pass = true;
274    else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
275 <           && fabs(mu->Eta()) > 1.5 && mu->Pt() < 10 && mvaval >= MUON_IDMVA_CUT_BIN2)  pass = true;
275 >           && fabs(mu->Eta()) > 1.5 && mu->Pt() <= 10 && mvaval >= MUON_IDMVA_CUT_BIN2)  pass = true;
276    else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
277             && fabs(mu->Eta()) > 1.5 && mu->Pt() > 10 && mvaval >= MUON_IDMVA_CUT_BIN3)  pass = true;
278 <  else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon()
279 <           && (mu->Quality().QualityMask().Mask() & mithep::MuonQuality::AllArbitrated) && mvaval >= MUON_IDMVA_CUT_BIN4)
280 <    pass = true;
278 >  else if( !(mu->IsGlobalMuon()) && mu->IsTrackerMuon() && mvaval >= MUON_IDMVA_CUT_BIN4 )  pass = true;
279 >  else if( mu->IsGlobalMuon() && !(mu->IsTrackerMuon()) && mvaval >= MUON_IDMVA_CUT_BIN5 )  pass = true;
280 >  
281  
282  
283    if( pass ) {
# Line 379 | Line 294 | void initMuonIDMVA()
294   {
295    muIDMVA = new mithep::MuonIDMVA();
296    vector<string> weightFiles;
297 <  weightFiles.push_back("../MitPhysics/data/MuonIDMVAWeights/MuonIDMVA_BDTG_barrel_lowpt_V2.weights.xml");
298 <  weightFiles.push_back("../MitPhysics/data/MuonIDMVAWeights/MuonIDMVA_BDTG_barrel_highpt_V2.weights.xml");
299 <  weightFiles.push_back("../MitPhysics/data/MuonIDMVAWeights/MuonIDMVA_BDTG_endcap_lowpt_V2.weights.xml");
300 <  weightFiles.push_back("../MitPhysics/data/MuonIDMVAWeights/MuonIDMVA_BDTG_endcap_highpt_V2.weights.xml");
301 <  weightFiles.push_back("../MitPhysics/data/MuonIDMVAWeights/MuonIDMVA_BDTG_tracker_V2.weights.xml");
302 <  weightFiles.push_back("../MitPhysics/data/MuonIDMVAWeights/MuonIDMVA_BDTG_global_V2.weights.xml");
297 >  weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_barrel_lowpt_V2.weights.xml");
298 >  weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_barrel_highpt_V2.weights.xml");
299 >  weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_endcap_lowpt_V2.weights.xml");
300 >  weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_endcap_highpt_V2.weights.xml");
301 >  weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_tracker_V2.weights.xml");
302 >  weightFiles.push_back("./data/MuonIDMVAWeights/MuonIDMVA_BDTG_global_V2.weights.xml");
303    muIDMVA->Initialize( "MuonIDMVA",
304                         mithep::MuonIDMVA::kIDV0,
305                         kTRUE, weightFiles);
306   }
307 +
308 + //--------------------------------------------------------------------------------------------------
309 + SelectionStatus muonIDPFSelection(ControlFlags &ctrl,
310 +                                  const mithep::Muon *mu,
311 +                                  const mithep::Vertex * vtx,
312 +                                  const mithep::Array<mithep::PFCandidate> * pfCandidates )
313 + //--------------------------------------------------------------------------------------------------
314 + {
315 +  bool pass = false;
316 +  SelectionStatus status; // init'ed to FAIL
317 +
318 +  // check that it matches to a PF muon
319 +  for( int i=0; i<pfCandidates->GetEntries(); i++ ) {
320 +    // tmp
321 +    if( !(PFnoPUflag[i]) ) continue; // my PF no PU hack
322 +
323 +    const mithep::PFCandidate *pf = (mithep::PFCandidate*)((*pfCandidates)[i]);
324 +    if( (pf->TrackerTrk() == mu->TrackerTrk()) && abs(pf->PFType()) == mithep::PFCandidate::eMuon ) {
325 +      pass = true;
326 +      break;
327 +    }
328 +  }
329 + //      
330 + //      if(mu->IsPFMuon()) pass = true;
331 + //      else pass = false;
332 +
333 +  //
334 +  // NB : ipsig here for reference synch ...
335 +  //
336 + #ifdef SYNC
337 +  if( pass && fabs(mu->Ip3dPVSignificance()) < 4 ) {
338 +    status.orStatus(SelectionStatus::LOOSEID);
339 +    status.orStatus(SelectionStatus::TIGHTID);
340 +  }
341 + #else
342 +  if( pass ) {
343 +    status.orStatus(SelectionStatus::LOOSEID);
344 +    status.orStatus(SelectionStatus::TIGHTID);
345 +  }
346 + #endif
347 +
348 +  return status;
349 + }
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)
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