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.3 by anlevin, Wed Feb 29 10:10:18 2012 UTC vs.
Revision 1.28 by dkralph, Wed Jun 13 14:11:22 2012 UTC

# Line 1 | Line 1
1   #include <math.h>
2   #include <iostream>
3  
4 #include "HiggsAnaDefs.hh"
5
4   #include "MuonSelection.h"
5   #include "ParseArgs.h"
6   #include "SelectionStatus.h"
7  
8 < SelectionStatus noPreSelection( ControlFlags &ctrl,  const mithep::TMuon * mu ) {
9 <        SelectionStatus status;
10 <        status.setStatus(SelectionStatus::PRESELECTION);
11 <        if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
12 <        return status;
13 < }
14 <
15 < SelectionStatus muonPreSelection( ControlFlags &ctrl,  const mithep::TMuon * mu ) {
16 <  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
17 <  bool ret = isMuFO(mu);
18 <  if(ctrl.debug) cout << "iS fo? ... " << ret << endl;  
19 <  ret &= ( fabs(mu->ip3dSig) < 4 );
20 <  if(ctrl.debug) cout << "and pass IP (" << mu->ip3dSig << ") ? ... " << ret << endl;  
21 <  ret &= ( mu->pt > 5 );
22 <  if(ctrl.debug) cout << "and >5 GeV ? ... " << ret << endl;  
23 <  ret &= ( fabs(mu->eta) < 2.4 );
24 <  if(ctrl.debug) cout << "and < 2.4 eta ? ... " << ret << endl;  
25 <
26 <  SelectionStatus status;
27 <  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
28 <  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
29 <  return status;
30 < }
31 <
32 < bool isMuFO( const mithep::TMuon * mu ) {
33 <  bool isgood=true;
34 <  float reliso = mu->pfIso03/mu->pt;
35 <
36 <  if( mu->pt < 5  )         isgood=false;                          
37 <  if ( fabs(mu->dz) > 0.2 ) isgood=false;   // needed to remove cosmics in HF sample
38 <
39 <  //
40 <
41 <  // HF seems to not want tkquality, SS does
42 <  if( mu->pt>20 ) {
43 <    if ( !((mu->typeBits & kGlobal) && mu->nSeg>0  ) ) isgood=false; //&& mu->nValidHits>0
44 <  } else {
45 <    if ( !((mu->typeBits & kGlobal) && mu->nSeg>0 ) //&& //&& mu->nValidHits>0
46 <         && !( (mu->typeBits & kTracker) &&  (mu->qualityBits & kTMOneStationLoose)
47 <               )
48 <         )   isgood=false;
49 <  }
50 <
51 <
52 <  //
53 <  // for HF MC to agree w/ data, cannot put any Tk quality bits
56 <  // but to get to 5% closure in the SS test, we need at least Tk oneStationLoose
57 <
58 <  /*
59 < && ( (mu->qualityBits & kTMOneStationLoose) ||
60 <                                               (mu->qualityBits & kTMLastStationOptimizedBarrelLowPtLoose) ||
61 <                                               (mu->qualityBits & kTMLastStationOptimizedLowPtLoose)
62 <                                               )
63 <  */
64 <
65 <
66 <  // comment for more stats for mu fake shape
67 <  if( mu->pt < 20 ) {
68 <    if( reliso > 3 ) isgood=false;    
69 <  } else {
70 <    if( reliso > 5 ) isgood=false;    
8 > #include "Track.h"
9 > #include "MuonTools.h"
10 > #include "MuonQuality.h"
11 > #include "MuonIDMVA.h"
12 >
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,
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    }
72
73  return isgood;
74 };
75
76
77 SelectionStatus passSoftMuonSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) {
78
79  int level=0;
80  unsigned failmask=0x0;
55    
56 <  if(mu->nTkHits < 11 )  
57 <    failmask |= (1<<level);
84 <  level++;
85 <
86 <  if(fabs(mu->d0) > 0.2)
87 <    failmask |= (1<<level);
88 <  level++;
89 <
90 <  if(fabs(mu->dz) > 0.1)
91 <    failmask |= (1<<level);
92 <  level++;
93 <
94 <  if(!(mu->typeBits & kTracker))
95 <    failmask |= (1<<level);
96 <  level++;
97 <
98 <  if(!(mu->qualityBits & kTMLastStationAngTight))
99 <    failmask |= (1<<level);      
100 <  level++;
101 <
102 <  Double_t iso = (mu->trkIso03 + mu->emIso03 + mu->hadIso03)/mu->pt;
103 <  if(mu->pt>20 && iso<0.1)
104 <    failmask |= (1<<level);
105 <
106 <  bool passtight  = !(failmask);
107 <  bool passloose  = !(failmask);
108 <  return SelectionStatus( passtight | passloose );
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 < //unsigned passMuonSelectionZZ( const mithep::TMuon * mu ) {
99 < SelectionStatus passMuonSelectionZZ( ControlFlags &ctrl, const mithep::TMuon * mu ) {
100 <  int level=0;
101 <  unsigned failmask=0x0;
102 <
103 <  if(mu->pt < 5) {
104 <    failmask |= (1<<level);
105 <  }
106 <
122 <  level++;
123 <  if(fabs(mu->eta) > 2.4) {
124 <    failmask |= (1<<level);
125 <  }
126 <
127 <  level++;
128 <  if(!(mu->typeBits & kGlobal)) {
129 <    failmask |= (1<<level);
130 <  }
131 <
132 <  level++;
133 <  if(mu->nTkHits          < 10) {
134 <    failmask |= (1<<level);
135 <  }
136 <
137 <
138 <  level++;  
139 <  if( (mu->trkIso03/mu->pt) > 0.7 ) {
140 <    failmask |= (1<<level);
141 <  }
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 <  //  if(muon->nPixHits   < 1)     return false;
109 <  //  if(muon->muNchi2    > 10)    return false;
110 <  //  if(muon->nMatch     < 2)     return false;
111 <  //  if(muon->nValidHits         < 1)     return false;
112 <  //  if(muon->pterr/muon->pt > 0.1)   return false;
113 <  //  if(fabs(muon->dz)       > 0.1)   return false;  
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( !failmask ) status.setStatus(SelectionStatus::LOOSEID);
152 <  if( !failmask ) status.setStatus(SelectionStatus::TIGHTID);
124 >  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
125    return status;
126 + }
127  
128  
129 + //--------------------------------------------------------------------------------------------------
130 + SelectionStatus noPreSelection( ControlFlags &ctrl,  const mithep::Muon * mu )
131  
132 < };
132 > //--------------------------------------------------------------------------------------------------
133 > {
134 >        SelectionStatus status;
135 >        status.setStatus(SelectionStatus::PRESELECTION);
136 >        if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
137 >        return status;
138 > }
139  
140 + //--------------------------------------------------------------------------------------------------
141 + SelectionStatus muonPreSelection( ControlFlags &ctrl,  
142 +                                  const mithep::Muon * mu,
143 +                                  const mithep::Vertex * vtx,
144 +                                  const mithep::Array<mithep::PFCandidate> * pfCandidates )
145 + //--------------------------------------------------------------------------------------------------
146 + {
147 +  bool ret = true;
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);
159 +  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
160 +  return status;
161 + }
162  
163  
162 //
163 // Kevin's WW selection
164 //
165 SelectionStatus passMuonSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) {
166  int level=0;
167  unsigned failmask=0x0;
164  
165 + //--------------------------------------------------------------------------------------------------
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 <  // 0x1
175 <  if(mu->pt < 5) {
176 <    failmask |= (1<<level);
177 <  }
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  
175  // 0x2
176  level++;
177  if(fabs(mu->eta) > 2.4) {
178    failmask |= (1<<level);
179  }
194  
195 <  // 0x4
196 <  level++;
197 <  if(mu->ptErr/mu->pt > 0.1)   {
198 <    failmask |= (1<<level);
199 <  }
195 >  SelectionStatus status;
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 <  // 0x8
202 <  level++;
203 <  if( fabs(mu->dz) > 0.1 )   {
204 <    failmask |= (1<<level);
205 <  }
201 > //--------------------------------------------------------------------------------------------------
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 >  bool ret = true;
209  
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  
194  Bool_t isGlobal  = (mu->typeBits & kGlobal) && (mu->muNchi2 < 10) && (mu->nMatch > 1) && (mu->nValidHits > 0);
195  Bool_t isTracker = (mu->typeBits & kTracker) && (mu->qualityBits & kTMLastStationTight);
224  
225 <  // 0x10
226 <  level++;
227 <  if(!isGlobal && !isTracker) {
228 <    failmask |= (1<<level);
229 <  }
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  
203  // 0x20
204  level++;
205  if( mu->nTkHits < 10 ) {
206    failmask |= (1<<level);
207  }
231  
209  level++;
210  if(mu->nPixHits         < 1) {
211    failmask |= (1<<level);
212  }
232  
233  
234 <  level++;
235 <  if(fabs(mu->d0)>0.02)   {
236 <    failmask |= (1<<level);
237 <  }
234 > //--------------------------------------------------------------------------------------------------
235 > SelectionStatus muonIDMVASelection(ControlFlags &ctrl,
236 >                                 const mithep::Muon *mu,
237 >                                 const mithep::Vertex * vtx   )
238 > //--------------------------------------------------------------------------------------------------
239 > {
240 >  double global_rchi2 = (mu->IsGlobalMuon()) ? mu->GlobalTrk()->RChi2() : 0.;
241  
242 +  //
243 +  // WARNING !!! KH hacked out to sync
244 +  //
245    /*
246 <  if(mu->pt>20) {
247 <    if(fabs(mu->d0)>0.02)   {
248 <      failmask |= (1<<level);
249 <    }
250 <  } else {
251 <    if(fabs(mu->d0)>0.01)  {
252 <    failmask |= (1<<level);
253 <    }
254 <  }
246 >  double mvaval = muIDMVA->MVAValue_ID(mu->Pt(),
247 >                                    mu->Eta(),
248 >                                    mu->IsGlobalMuon(),
249 >                                    mu->IsTrackerMuon(),
250 >                                    mu->TrackerTrk()->RChi2(),
251 >                                    global_rchi2,
252 >                                    (Double_t)(mu->NValidHits()),
253 >                                    (Double_t)(mu->TrackerTrk()->NHits()),
254 >                                    (Double_t)(mu->TrackerTrk()->NPixelHits()),
255 >                                    (Double_t)(mu->NMatches()),
256 >                                    mu->TrkKink(),
257 >                                    muTools.GetSegmentCompatability(mu),
258 >                                    muTools.GetCaloCompatability(mu,true,true),
259 >                                    mu->HadEnergy(),
260 >                                    mu->EmEnergy(),
261 >                                    mu->HadS9Energy(),
262 >                                    mu->EmS9Energy(),
263 >                                    (Bool_t)(ctrl.debug) );
264    */
265 +  double mvaval = 0.0;
266  
267    SelectionStatus status;
268 <  if( !failmask ) status.setStatus(SelectionStatus::LOOSEID);
269 <  if( !failmask ) status.setStatus(SelectionStatus::TIGHTID);
270 <  return status;
271 <
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;
272 >  else if( mu->IsGlobalMuon() && mu->IsTrackerMuon()
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;
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() && 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 ) {
284 >    status.orStatus(SelectionStatus::LOOSEID);
285 >    status.orStatus(SelectionStatus::TIGHTID);
286 >  }
287  
288 < };
288 >  return status;
289 > }
290  
291 < SelectionStatus passMuonSelectionBackDoor( ControlFlags &ctrl, const mithep::TMuon * mu ) {
292 <  int level=0;
293 <  unsigned failmask=0x0;
291 > //--------------------------------------------------------------------------------------------------
292 > void initMuonIDMVA()
293 > //--------------------------------------------------------------------------------------------------
294 > {
295 >  muIDMVA = new mithep::MuonIDMVA();
296 >  vector<string> weightFiles;
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 <  if(mu->pt < 5) {
309 <    failmask |= (1<<level);
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 <  level++;
334 <  if(fabs(mu->eta) > 2.5) {
335 <    failmask |= (1<<level);
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 <  level++;
349 <  if(!(mu->typeBits & kGlobal) || !(mu->typeBits & kTracker)) {
350 <    failmask |= (1<<level);
351 <  }
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 <  level++;
384 <  if(mu->nTkHits          <= 10) {
260 <    failmask |= (1<<level);
261 <  }
383 >  if (pfiso > 0.2)
384 >    passIso = false;
385  
263        level++;
264        if(mu->ip3dSig > 4) {
265                failmask |= (1<<level);
266        }
267        
268        level++;
269  if(mu->muNchi2          >= 10)  {
270                failmask |= (1<<level);
271        }
272                
273        level++;
274  if(mu->nValidHits       < 1)   {
275                failmask |= (1<<level);
276        }
277        
278        level++;
279  if(mu->nMatch   < 2)   {
280                failmask |= (1<<level);
281        }
282  
386    SelectionStatus status;
387 <        
388 <  if( !failmask ) status.setStatus(SelectionStatus::LOOSEID);
389 <  if( !failmask ) status.setStatus(SelectionStatus::TIGHTID);
390 <        
288 <        return status;
387 >  if(passID)  status |= SelectionStatus::TIGHTIDANDPRE;
388 >  if(passIso) status |= SelectionStatus::TIGHTISOANDPRE;
389 >  
390 >  return status;
391  
392 < };
392 > }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines