1 |
+ |
#ifndef ISOLATIONSELECTION |
2 |
+ |
#define ISOLATIONSELECTION |
3 |
+ |
|
4 |
|
#include "Array.h" |
5 |
|
#include "Electron.h" |
6 |
|
#include "Muon.h" |
18 |
|
#include "SelectionStatus.h" |
19 |
|
#include "SimpleLepton.h" |
20 |
|
|
21 |
+ |
using namespace std; |
22 |
+ |
using namespace mithep; |
23 |
+ |
|
24 |
|
#define MUON_ISOMVA_CUT_BIN0 -0.615 |
25 |
|
#define MUON_ISOMVA_CUT_BIN1 0.235 |
26 |
|
#define MUON_ISOMVA_CUT_BIN2 -0.825 |
170 |
|
|
171 |
|
SelectionStatus electronIsoSelection(ControlFlags &, const mithep::Electron *, const mithep::Vertex &, const mithep::Array<mithep::PFCandidate> * fPFCandidateCol ); |
172 |
|
SelectionStatus muonIsoSelection(ControlFlags &, const mithep::Muon *, const mithep::Vertex *, const mithep::Array<mithep::PFCandidate> * fPFCandidateCol ); |
167 |
– |
bool pairwiseIsoSelection( ControlFlags &, vector<SimpleLepton> &, float rho ); |
173 |
|
bool noIso(ControlFlags &, vector<SimpleLepton> &, float rho); |
174 |
|
|
175 |
|
void initMuonIsoMVA(); |
196 |
|
const mithep::Array<mithep::PFCandidate> * fPFCandidates, |
197 |
|
const mithep::Array<mithep::PileupEnergyDensity> * fPUEnergyDensity, |
198 |
|
mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion, |
199 |
< |
vector<const mithep::Muon*> muonsToVeto, |
200 |
< |
vector<const mithep::Electron*> electronsToVeto); |
196 |
< |
// hacked version |
197 |
< |
double muonPFIso04(ControlFlags &ctrl, |
198 |
< |
const mithep::Muon * mu, |
199 |
< |
const mithep::Vertex * vtx, |
200 |
< |
const mithep::Array<mithep::PFCandidate> * fPFCandidates, |
201 |
< |
float rho, // the hack |
202 |
< |
mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion, |
203 |
< |
vector<const mithep::Muon*> muonsToVeto, |
204 |
< |
vector<const mithep::Electron*> electronsToVeto); |
205 |
< |
// |
199 |
> |
vector<const mithep::PFCandidate*> photonsToVeto); |
200 |
> |
|
201 |
|
SelectionStatus muonReferenceIsoSelection(ControlFlags &ctrl, |
202 |
|
const mithep::Muon * mu, |
203 |
|
const mithep::Vertex * vtx, |
204 |
|
const mithep::Array<mithep::PFCandidate> * fPFCandidates, |
205 |
|
const mithep::Array<mithep::PileupEnergyDensity> * fPUEnergyDensity, |
206 |
|
mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion, |
207 |
< |
vector<const mithep::Muon*> muonsToVeto, |
213 |
< |
vector<const mithep::Electron*> electronsToVeto); |
214 |
< |
// hacked version |
215 |
< |
SelectionStatus muonReferenceIsoSelection(ControlFlags &ctrl, |
216 |
< |
const mithep::Muon * mu, |
217 |
< |
const mithep::Vertex * vtx, |
218 |
< |
const mithep::Array<mithep::PFCandidate> * fPFCandidates, |
219 |
< |
float, |
220 |
< |
mithep::MuonTools::EMuonEffectiveAreaTarget EffectiveAreaVersion, |
221 |
< |
vector<const mithep::Muon*> muonsToVeto, |
222 |
< |
vector<const mithep::Electron*> electronsToVeto); |
207 |
> |
vector<const mithep::PFCandidate*> photonsToVeto); |
208 |
|
|
209 |
|
void initElectronIsoMVA(); |
210 |
|
SelectionStatus electronIsoMVASelection(ControlFlags &ctrl, |
230 |
|
const mithep::Array<mithep::PFCandidate> * fPFCandidates, |
231 |
|
const mithep::Array<mithep::PileupEnergyDensity> * fPUEnergyDensity, |
232 |
|
mithep::ElectronTools::EElectronEffectiveAreaTarget EffectiveAreaVersion, |
233 |
< |
vector<const mithep::Muon*> muonsToVeto, |
234 |
< |
vector<const mithep::Electron*> electronsToVeto) ; |
250 |
< |
// hacked version |
251 |
< |
float electronPFIso04(ControlFlags &ctrl, |
252 |
< |
const mithep::Electron * ele, |
253 |
< |
const mithep::Vertex * vtx, |
254 |
< |
const mithep::Array<mithep::PFCandidate> * fPFCandidates, |
255 |
< |
float rho, // the hack |
256 |
< |
mithep::ElectronTools::EElectronEffectiveAreaTarget EffectiveAreaVersion, |
257 |
< |
vector<const mithep::Muon*> muonsToVeto, |
258 |
< |
vector<const mithep::Electron*> electronsToVeto) ; |
259 |
< |
// |
233 |
> |
vector<const mithep::PFCandidate*> photonsToVeto); |
234 |
> |
|
235 |
|
SelectionStatus electronReferenceIsoSelection(ControlFlags &ctrl, |
236 |
|
const mithep::Electron * ele, |
237 |
|
const mithep::Vertex * vtx, |
238 |
|
const mithep::Array<mithep::PFCandidate> * fPFCandidates, |
239 |
|
const mithep::Array<mithep::PileupEnergyDensity> * fPUEnergyDensity, |
240 |
|
mithep::ElectronTools::EElectronEffectiveAreaTarget EffectiveAreaVersion, |
241 |
< |
vector<const mithep::Muon*> muonsToVeto, |
267 |
< |
vector<const mithep::Electron*> electronsToVeto); |
268 |
< |
// hacked version |
269 |
< |
SelectionStatus electronReferenceIsoSelection(ControlFlags &ctrl, |
270 |
< |
const mithep::Electron * ele, |
271 |
< |
const mithep::Vertex * vtx, |
272 |
< |
const mithep::Array<mithep::PFCandidate> * fPFCandidates, |
273 |
< |
float, |
274 |
< |
mithep::ElectronTools::EElectronEffectiveAreaTarget EffectiveAreaVersion, |
275 |
< |
vector<const mithep::Muon*> muonsToVeto, |
276 |
< |
vector<const mithep::Electron*> electronsToVeto); |
241 |
> |
vector<const mithep::PFCandidate*> photonsToVeto); |
242 |
|
|
243 |
|
double dbetaCorrectedIsoDr03(ControlFlags & ctrl, |
244 |
|
const mithep::PFCandidate * photon, |
245 |
< |
const mithep::Muon * mu, // need to veto |
245 |
> |
const mithep::ChargedParticle * lep, // need to veto |
246 |
|
const mithep::Array<mithep::PFCandidate> * fPFCandidates); |
247 |
|
|
248 |
< |
double dbetaCorrectedIsoDr03(ControlFlags & ctrl, |
249 |
< |
const mithep::PFCandidate * photon, |
250 |
< |
const mithep::Electron * el, // need to veto |
251 |
< |
const mithep::Array<mithep::PFCandidate> * fPFCandidates); |
252 |
< |
|
253 |
< |
double betaCorrectedIsoDr03(ControlFlags & ctrl, |
254 |
< |
const mithep::PFCandidate * photon, |
255 |
< |
const mithep::Muon * mu, // need to veto |
256 |
< |
const mithep::Array<mithep::PFCandidate> * fPFCandidates); |
257 |
< |
|
258 |
< |
double betaCorrectedIsoDr03(ControlFlags & ctrl, |
294 |
< |
const mithep::PFCandidate * photon, |
295 |
< |
const mithep::Electron * el, // need to veto |
296 |
< |
const mithep::Array<mithep::PFCandidate> * fPFCandidates); |
248 |
> |
double nonCorrectedIsoDr03(ControlFlags & ctrl, |
249 |
> |
const mithep::PFCandidate * photon, |
250 |
> |
const mithep::ChargedParticle *lep, // need to veto |
251 |
> |
const mithep::Array<mithep::PFCandidate> * fPFCandidates); |
252 |
> |
|
253 |
> |
double isoDr03ForFsr(ControlFlags & ctrl, |
254 |
> |
const PFCandidate * photon, |
255 |
> |
const ChargedParticle * lep, |
256 |
> |
const Array<PFCandidate> * fPFCandidates, |
257 |
> |
bool doDBetaCorr=false); |
258 |
> |
#endif |