|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
#ifndef ParticleFlowCandidate_PFCandidateElectronExtra_h
#define ParticleFlowCandidate_PFCandidateElectronExtra_h
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrackFwd.h"
#include "DataFormats/Math/interface/LorentzVector.h"
#include <iosfwd>
namespace reco {
/** \class reco::PFCandidateElectronExtra
*
* extra information on the electron particle candidate from particle flow
*
*/
class PFCandidateElectronExtra {
public:
enum StatusFlag {
X = 0, // undefined
Selected, // selected
ECALDrivenPreselected, // ECAL-driven electron pre-selected
MVASelected, // Passed the internal particle-flow selection (mva selection)
Rejected // Rejected
};
// if you had a variable update NMvaVariables
enum MvaVariable {
MVA_FIRST = 0,
MVA_LnPtGsf = MVA_FIRST,
MVA_EtaGsf,
MVA_SigmaPtOverPt,
MVA_Fbrem,
MVA_Chi2Gsf,
MVA_NhitsKf,
MVA_Chi2Kf,
MVA_EtotOverPin,
MVA_EseedOverPout,
MVA_EbremOverDeltaP,
MVA_DeltaEtaTrackCluster,
MVA_LogSigmaEtaEta,
MVA_HOverHE,
MVA_LateBrem,
MVA_FirstBrem,
MVA_MVA,
MVA_LAST
};
public:
/// constructor
PFCandidateElectronExtra();
/// constructor
PFCandidateElectronExtra(const GsfTrackRef&);
/// destructor
~PFCandidateElectronExtra() { ; }
/// set gsftrack reference
void setGsfTrackRef(const reco::GsfTrackRef& ref);
/// set kf track reference
void setKfTrackRef(const reco::TrackRef& ref);
/// return a reference to the corresponding GSF track
reco::GsfTrackRef gsfTrackRef() const { return gsfTrackRef_; }
/// return a reference to the corresponding KF track
reco::TrackRef kfTrackRef() const { return kfTrackRef_; }
/// set LateBrem
void setLateBrem(float val);
/// set EarlyBrem
void setEarlyBrem(float val);
/// set the pout (not trivial to get from the GSF track)
void setGsfTrackPout(const math::XYZTLorentzVector& pout);
/// set the cluster energies. the Pout should be saved first
void setClusterEnergies(const std::vector<float>& energies);
/// set the sigmaetaeta
void setSigmaEtaEta(float val);
/// set the delta eta
void setDeltaEta(float val);
/// set the had energy. The cluster energies should be entered before
void setHadEnergy(float val);
/// set the result (mostly for debugging)
void setMVA(float val);
/// set status
void setStatus(StatusFlag type, bool status = true);
/// access to the status
bool electronStatus(StatusFlag) const;
/// access to the status
int electronStatus() const { return status_; }
/// access to mva variable status
bool mvaStatus(MvaVariable flag) const;
/// access to the mva variables
const std::vector<float>& mvaVariables() const { return mvaVariables_; }
/// access to any variable
float mvaVariable(MvaVariable var) const;
/// access to specific variables
float hadEnergy() const { return hadEnergy_; }
float sigmaEtaEta() const { return sigmaEtaEta_; }
private:
void setVariable(MvaVariable type, float var);
private:
/// Ref to the GSF track
reco::GsfTrackRef gsfTrackRef_;
/// Ref to the KF track
reco::TrackRef kfTrackRef_;
/// energy of individual clusters (corrected). The first cluster is the seed
std::vector<float> clusterEnergies_;
/// mva variables - transient !
std::vector<float> mvaVariables_;
/// status of mva variables
int mvaStatus_;
/// Status of the electron
int status_;
/// Variables entering the MVA that should be saved
math::XYZTLorentzVector pout_;
float earlyBrem_;
float lateBrem_;
float sigmaEtaEta_;
float hadEnergy_;
float deltaEta_;
};
/// print the variables
std::ostream& operator<<(std::ostream& out, const PFCandidateElectronExtra& c);
} // namespace reco
#endif
|