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 |
< |
|
9 |
< |
SelectionStatus muonPreSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) { |
10 |
< |
if(ctrl.debug) cout << "inside muonpresel ... " << endl; |
11 |
< |
bool ret = isMuFO(mu); |
12 |
< |
if(ctrl.debug) cout << "iS fo? ... " << ret << endl; |
13 |
< |
ret &= ( fabs(mu->ip3dSig) < 4 ); |
14 |
< |
if(ctrl.debug) cout << "and pass IP (" << mu->ip3dSig << ") ? ... " << ret << endl; |
15 |
< |
ret &= ( mu->pt > 5 ); |
16 |
< |
if(ctrl.debug) cout << "and >5 GeV ? ... " << ret << endl; |
17 |
< |
ret &= ( fabs(mu->eta) < 2.4 ); |
18 |
< |
if(ctrl.debug) cout << "and < 2.4 eta ? ... " << ret << endl; |
19 |
< |
|
20 |
< |
SelectionStatus status; |
21 |
< |
if( ret ) status.setStatus(SelectionStatus::PRESELECTION); |
22 |
< |
if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl; |
23 |
< |
return status; |
24 |
< |
} |
25 |
< |
|
26 |
< |
bool isMuFO( const mithep::TMuon * mu ) { |
27 |
< |
bool isgood=true; |
28 |
< |
float reliso = mu->pfIso03/mu->pt; |
29 |
< |
|
30 |
< |
if( mu->pt < 5 ) isgood=false; |
31 |
< |
if ( fabs(mu->dz) > 0.2 ) isgood=false; // needed to remove cosmics in HF sample |
32 |
< |
|
33 |
< |
// |
34 |
< |
|
35 |
< |
// HF seems to not want tkquality, SS does |
36 |
< |
if( mu->pt>20 ) { |
37 |
< |
if ( !((mu->typeBits & kGlobal) && mu->nSeg>0 ) ) isgood=false; //&& mu->nValidHits>0 |
38 |
< |
} else { |
39 |
< |
if ( !((mu->typeBits & kGlobal) && mu->nSeg>0 ) //&& //&& mu->nValidHits>0 |
40 |
< |
&& !( (mu->typeBits & kTracker) && (mu->qualityBits & kTMOneStationLoose) |
41 |
< |
) |
42 |
< |
) 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 |
|
} |
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 |
< |
// for HF MC to agree w/ data, cannot put any Tk quality bits |
116 |
< |
// but to get to 5% closure in the SS test, we need at least Tk oneStationLoose |
117 |
< |
|
52 |
< |
/* |
53 |
< |
&& ( (mu->qualityBits & kTMOneStationLoose) || |
54 |
< |
(mu->qualityBits & kTMLastStationOptimizedBarrelLowPtLoose) || |
55 |
< |
(mu->qualityBits & kTMLastStationOptimizedLowPtLoose) |
56 |
< |
) |
57 |
< |
*/ |
58 |
< |
|
59 |
< |
|
60 |
< |
// comment for more stats for mu fake shape |
61 |
< |
if( mu->pt < 20 ) { |
62 |
< |
if( reliso > 3 ) isgood=false; |
63 |
< |
} else { |
64 |
< |
if( reliso > 5 ) isgood=false; |
65 |
< |
} |
66 |
< |
|
67 |
< |
return isgood; |
68 |
< |
}; |
69 |
< |
|
70 |
< |
|
71 |
< |
SelectionStatus passSoftMuonSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) { |
72 |
< |
|
73 |
< |
int level=0; |
74 |
< |
unsigned failmask=0x0; |
115 |
> |
// Loose cut-based ID for isolation veto |
116 |
> |
// |
117 |
> |
bool ret = true; |
118 |
|
|
119 |
< |
if(mu->nTkHits < 11 ) |
120 |
< |
failmask |= (1<<level); |
121 |
< |
level++; |
79 |
< |
|
80 |
< |
if(fabs(mu->d0) > 0.2) |
81 |
< |
failmask |= (1<<level); |
82 |
< |
level++; |
83 |
< |
|
84 |
< |
if(fabs(mu->dz) > 0.1) |
85 |
< |
failmask |= (1<<level); |
86 |
< |
level++; |
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 |
< |
if(!(mu->typeBits & kTracker)) |
124 |
< |
failmask |= (1<<level); |
125 |
< |
level++; |
123 |
> |
SelectionStatus status; |
124 |
> |
if( ret ) status.setStatus(SelectionStatus::PRESELECTION); |
125 |
> |
return status; |
126 |
> |
} |
127 |
|
|
92 |
– |
if(!(mu->qualityBits & kTMLastStationAngTight)) |
93 |
– |
failmask |= (1<<level); |
94 |
– |
level++; |
128 |
|
|
129 |
< |
Double_t iso = (mu->trkIso03 + mu->emIso03 + mu->hadIso03)/mu->pt; |
130 |
< |
if(mu->pt>20 && iso<0.1) |
98 |
< |
failmask |= (1<<level); |
129 |
> |
//-------------------------------------------------------------------------------------------------- |
130 |
> |
SelectionStatus noPreSelection( ControlFlags &ctrl, const mithep::Muon * mu ) |
131 |
|
|
132 |
< |
bool passtight = !(failmask); |
133 |
< |
bool passloose = !(failmask); |
134 |
< |
return SelectionStatus( passtight | passloose ); |
135 |
< |
|
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 |
< |
//unsigned passMuonSelectionZZ( const mithep::TMuon * mu ) { |
142 |
< |
SelectionStatus passMuonSelectionZZ( ControlFlags &ctrl, const mithep::TMuon * mu ) { |
143 |
< |
int level=0; |
144 |
< |
unsigned failmask=0x0; |
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 |
< |
if(mu->pt < 5) { |
158 |
< |
failmask |= (1<<level); |
159 |
< |
} |
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 |
|
|
116 |
– |
level++; |
117 |
– |
if(fabs(mu->eta) > 2.4) { |
118 |
– |
failmask |= (1<<level); |
119 |
– |
} |
163 |
|
|
121 |
– |
level++; |
122 |
– |
if(!(mu->typeBits & kGlobal)) { |
123 |
– |
failmask |= (1<<level); |
124 |
– |
} |
164 |
|
|
165 |
< |
level++; |
166 |
< |
if(mu->nTkHits < 10) { |
167 |
< |
failmask |= (1<<level); |
168 |
< |
} |
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 |
+ |
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 |
|
|
132 |
– |
level++; |
133 |
– |
if( (mu->trkIso03/mu->pt) > 0.7 ) { |
134 |
– |
failmask |= (1<<level); |
135 |
– |
} |
194 |
|
|
137 |
– |
// if(muon->nPixHits < 1) return false; |
138 |
– |
// if(muon->muNchi2 > 10) return false; |
139 |
– |
// if(muon->nMatch < 2) return false; |
140 |
– |
// if(muon->nValidHits < 1) return false; |
141 |
– |
// if(muon->pterr/muon->pt > 0.1) return false; |
142 |
– |
// if(fabs(muon->dz) > 0.1) return false; |
143 |
– |
|
195 |
|
SelectionStatus status; |
196 |
< |
if( !failmask ) status.setStatus(SelectionStatus::LOOSEID); |
197 |
< |
if( !failmask ) status.setStatus(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 |
+ |
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 |
< |
|
211 |
< |
}; |
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 |
|
|
224 |
|
|
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 |
|
|
156 |
– |
// |
157 |
– |
// Kevin's WW selection |
158 |
– |
// |
159 |
– |
SelectionStatus passMuonSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) { |
160 |
– |
int level=0; |
161 |
– |
unsigned failmask=0x0; |
232 |
|
|
163 |
– |
// 0x1 |
164 |
– |
if(mu->pt < 5) { |
165 |
– |
failmask |= (1<<level); |
166 |
– |
} |
233 |
|
|
234 |
< |
// 0x2 |
235 |
< |
level++; |
236 |
< |
if(fabs(mu->eta) > 2.4) { |
237 |
< |
failmask |= (1<<level); |
238 |
< |
} |
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 |
< |
// 0x4 |
243 |
< |
level++; |
244 |
< |
if(mu->ptErr/mu->pt > 0.1) { |
245 |
< |
failmask |= (1<<level); |
246 |
< |
} |
242 |
> |
// |
243 |
> |
// WARNING !!! KH hacked out to sync |
244 |
> |
// |
245 |
> |
/* |
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 |
< |
// 0x8 |
268 |
< |
level++; |
269 |
< |
if( fabs(mu->dz) > 0.1 ) { |
270 |
< |
failmask |= (1<<level); |
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; |
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 |
+ |
return status; |
289 |
+ |
} |
290 |
|
|
291 |
< |
Bool_t isGlobal = (mu->typeBits & kGlobal) && (mu->muNchi2 < 10) && (mu->nMatch > 1) && (mu->nValidHits > 0); |
292 |
< |
Bool_t isTracker = (mu->typeBits & kTracker) && (mu->qualityBits & kTMLastStationTight); |
293 |
< |
|
294 |
< |
// 0x10 |
295 |
< |
level++; |
296 |
< |
if(!isGlobal && !isTracker) { |
297 |
< |
failmask |= (1<<level); |
298 |
< |
} |
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 |
< |
// 0x20 |
309 |
< |
level++; |
310 |
< |
if( mu->nTkHits < 10 ) { |
311 |
< |
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(mu->nPixHits < 1) { |
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 |
+ |
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(fabs(mu->d0)>0.02) { |
210 |
< |
failmask |= (1<<level); |
211 |
< |
} |
212 |
< |
|
213 |
< |
/* |
214 |
< |
if(mu->pt>20) { |
215 |
< |
if(fabs(mu->d0)>0.02) { |
216 |
< |
failmask |= (1<<level); |
217 |
< |
} |
218 |
< |
} else { |
219 |
< |
if(fabs(mu->d0)>0.01) { |
220 |
< |
failmask |= (1<<level); |
221 |
< |
} |
222 |
< |
} |
223 |
< |
*/ |
383 |
> |
if (pfiso > 0.2) |
384 |
> |
passIso = false; |
385 |
|
|
386 |
|
SelectionStatus status; |
387 |
< |
if( !failmask ) status.setStatus(SelectionStatus::LOOSEID); |
388 |
< |
if( !failmask ) status.setStatus(SelectionStatus::TIGHTID); |
387 |
> |
if(passID) status |= SelectionStatus::TIGHTIDANDPRE; |
388 |
> |
if(passIso) status |= SelectionStatus::TIGHTISOANDPRE; |
389 |
> |
|
390 |
|
return status; |
391 |
|
|
392 |
< |
|
231 |
< |
}; |
232 |
< |
|
392 |
> |
} |