8 |
|
// Authors: S.Xie |
9 |
|
//-------------------------------------------------------------------------------------------------- |
10 |
|
|
11 |
< |
#ifndef MITANA_UTILS_ISOLATIONTOOLS_H |
12 |
< |
#define MITANA_UTILS_ISOLATIONTOOLS_H |
11 |
> |
#ifndef MITPHYSICS_UTILS_ISOLATIONTOOLS_H |
12 |
> |
#define MITPHYSICS_UTILS_ISOLATIONTOOLS_H |
13 |
|
|
14 |
|
#include <TMath.h> |
15 |
|
#include "MitAna/DataTree/interface/Track.h" |
16 |
|
#include "MitAna/DataTree/interface/BasicCluster.h" |
17 |
|
#include "MitAna/DataTree/interface/SuperCluster.h" |
18 |
|
#include "MitAna/DataTree/interface/CaloTower.h" |
19 |
+ |
#include "MitAna/DataTree/interface/VertexCol.h" |
20 |
+ |
#include "MitAna/DataTree/interface/MuonCol.h" |
21 |
+ |
#include "MitAna/DataTree/interface/ElectronCol.h" |
22 |
+ |
#include "MitAna/DataTree/interface/PFCandidateCol.h" |
23 |
+ |
#include "MitAna/DataTree/interface/TrackCol.h" |
24 |
+ |
#include "MitAna/DataTree/interface/DecayParticleCol.h" |
25 |
|
|
26 |
|
namespace mithep |
27 |
|
{ |
28 |
|
class IsolationTools { |
29 |
|
public: |
30 |
|
static Double_t TrackIsolation(const mithep::Track *p, Double_t extRadius, |
31 |
< |
Double_t intRadius, Double_t ptLow, Double_t maxVtxZDist, |
32 |
< |
mithep::Collection<mithep::Track> *tracks); |
31 |
> |
Double_t intRadius, Double_t ptLow, Double_t maxVtxZDist, |
32 |
> |
const mithep::Collection<mithep::Track> *tracks); |
33 |
|
static Double_t EcalIsolation(const SuperCluster *sc, Double_t coneSize, Double_t etLow, |
34 |
< |
mithep::Collection<mithep::BasicCluster> *basicClusters); |
34 |
> |
const mithep::Collection<mithep::BasicCluster> *basicClusters); |
35 |
|
static Double_t CaloTowerHadIsolation(const ThreeVector *p, Double_t extRadius, |
36 |
|
Double_t intRadius, Double_t etLow, |
37 |
< |
mithep::Collection<mithep::CaloTower> *caloTowers); |
37 |
> |
const mithep::Collection<mithep::CaloTower> |
38 |
> |
*caloTowers); |
39 |
|
static Double_t CaloTowerEmIsolation(const ThreeVector *p, Double_t extRadius, |
40 |
|
Double_t intRadius, Double_t etLow, |
41 |
< |
mithep::Collection<mithep::CaloTower> *caloTowers); |
41 |
> |
const mithep::Collection<mithep::CaloTower> *caloTowers); |
42 |
> |
static Double_t PFMuonIsolation(const Muon *p, const PFCandidateCol *PFCands, |
43 |
> |
const Vertex *vertex, Double_t delta_z, Double_t ptMin, |
44 |
> |
Double_t extRadius, Double_t intRadius); |
45 |
> |
static Double_t PFMuonIsolation(const Muon *p, const PFCandidateCol *PFCands, const Vertex *vertex, |
46 |
> |
const MuonCol *goodMuons, const ElectronCol *goodElectrons, |
47 |
> |
Double_t delta_z = 0.2, Double_t ptMin = 1.0, Double_t extRadius = 0.4, |
48 |
> |
Double_t intRadius = 0.0, int isoType = 0, Double_t beta = 1.0); |
49 |
> |
static Double_t PFElectronIsolation(const Electron *p, const PFCandidateCol *PFCands, |
50 |
> |
const Vertex *vertex, Double_t delta_z, Double_t ptMin, |
51 |
> |
Double_t extRadius, Double_t intRadius); |
52 |
> |
static Double_t PFElectronIsolation(const Electron *p, const PFCandidateCol *PFCands, const Vertex *vertex, |
53 |
> |
const MuonCol *goodMuons, const ElectronCol *goodElectrons, |
54 |
> |
Double_t delta_z = 0.2 , Double_t ptMin = 1.0,Double_t extRadius = 0.4, |
55 |
> |
Double_t intRadius = 0.0, int isoType = 0, Double_t beta = 1.0); |
56 |
> |
static Double_t BetaM(const TrackCol *tracks, const Muon *p, const Vertex *vertex, |
57 |
> |
Double_t ptMin, Double_t delta_z, Double_t extRadius, |
58 |
> |
Double_t intRadius); |
59 |
> |
static Double_t BetaE(const TrackCol *tracks, const Electron *p, const Vertex *vertex, |
60 |
> |
Double_t ptMin, Double_t delta_z, Double_t extRadius, |
61 |
> |
Double_t intRadius); |
62 |
> |
|
63 |
> |
// method added by F.Stoeckli: computes the track isolation with NO constrint on the OV-track compatibility |
64 |
> |
static Double_t TrackIsolationNoPV(const mithep::Particle*, const BaseVertex*, |
65 |
> |
Double_t extRadius, |
66 |
> |
Double_t intRadius, |
67 |
> |
Double_t ptLow, |
68 |
> |
Double_t etaStrip, |
69 |
> |
Double_t maxD0, |
70 |
> |
mithep::TrackQuality::EQuality, |
71 |
> |
const mithep::Collection<mithep::Track> *tracks, |
72 |
> |
UInt_t maxNExpectedHitsInner = 999, |
73 |
> |
const mithep::DecayParticleCol *conversions = 0); |
74 |
> |
|
75 |
> |
ClassDef(IsolationTools, 0) // Isolation tools |
76 |
|
}; |
77 |
|
} |
78 |
|
#endif |