|
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
|
#ifndef PreId_H
#define PreId_H
// Author F. Beaudette. March 2010
#include "DataFormats/Math/interface/Point3D.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecTrackFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFClusterFwd.h"
namespace reco {
class PreId {
public:
enum MatchingType { NONE = 0, ECALMATCH = 1, ESMATCH = 2, TRACKFILTERING = 3, MVA = 4, FINAL = 10 };
public:
PreId(unsigned nselection = 1)
: trackRef_(reco::TrackRef()),
clusterRef_(reco::PFClusterRef()),
matchingEop_(-999.),
EcalPos_(math::XYZPoint()),
meanShower_(math::XYZPoint()),
gsfChi2_(-999.),
dpt_(0.),
chi2Ratio_(0.) {
matching_.resize(nselection, false);
mva_.resize(nselection, -999.);
geomMatching_.resize(5, -999.);
}
void setTrack(reco::TrackRef trackref) { trackRef_ = trackref; }
void setECALMatchingProperties(PFClusterRef clusterRef,
const math::XYZPoint &ecalpos,
const math::XYZPoint &meanShower,
float deta,
float dphi,
float chieta,
float chiphi,
float chi2,
float eop) {
clusterRef_ = clusterRef;
EcalPos_ = ecalpos;
meanShower_ = meanShower;
geomMatching_[0] = deta;
geomMatching_[1] = dphi;
geomMatching_[2] = chieta;
geomMatching_[3] = chiphi;
geomMatching_[4] = chi2;
matchingEop_ = eop;
}
void setTrackProperties(float newchi2, float chi2ratio, float dpt) {
gsfChi2_ = newchi2;
chi2Ratio_ = chi2ratio;
dpt_ = dpt;
}
void setFinalDecision(bool accepted, unsigned n = 0) { setMatching(FINAL, accepted, n); }
void setECALMatching(bool accepted, unsigned n = 0) { setMatching(ECALMATCH, accepted, n); }
void setESMatching(bool accepted, unsigned n = 0) { setMatching(ESMATCH, accepted, n); }
void setTrackFiltering(bool accepted, unsigned n = 0) { setMatching(TRACKFILTERING, accepted, n); }
void setMVA(bool accepted, float mva, unsigned n = 0) {
setMatching(MVA, accepted, n);
if (n < mva_.size())
mva_[n] = mva;
}
void setMatching(MatchingType type, bool result, unsigned n = 0);
bool matching(MatchingType type, unsigned n = 0) const {
if (n < matching_.size()) {
return matching_[n] & (1 << type);
}
return false;
}
/// Access methods
inline const std::vector<float> &geomMatching() const { return geomMatching_; }
inline float eopMatch() const { return matchingEop_; }
inline float pt() const { return trackRef_->pt(); }
inline float eta() const { return trackRef_->eta(); }
inline float kfChi2() const { return trackRef_->normalizedChi2(); }
inline float kfNHits() const { return trackRef_->found(); }
const math::XYZPoint &ecalPos() const { return EcalPos_; }
const math::XYZPoint &meanShower() const { return meanShower_; }
inline float chi2Ratio() const { return chi2Ratio_; }
inline float gsfChi2() const { return gsfChi2_; }
inline bool ecalMatching(unsigned n = 0) const { return matching(ECALMATCH, n); }
inline bool esMatching(unsigned n = 0) const { return matching(ESMATCH, n); }
inline bool trackFiltered(unsigned n = 0) const { return matching(TRACKFILTERING, n); }
inline bool mvaSelected(unsigned n = 0) const { return matching(MVA, n); }
inline bool preIded(unsigned n = 0) const { return matching(FINAL, n); }
float mva(unsigned n = 0) const;
inline float dpt() const { return dpt_; }
reco::TrackRef trackRef() const { return trackRef_; }
PFClusterRef clusterRef() const { return clusterRef_; }
private:
reco::TrackRef trackRef_;
PFClusterRef clusterRef_;
std::vector<float> geomMatching_;
float matchingEop_;
math::XYZPoint EcalPos_;
math::XYZPoint meanShower_;
float gsfChi2_;
float dpt_;
float chi2Ratio_;
std::vector<float> mva_;
// bool goodpreid_;
// bool TkId_;
// bool EcalMatching_;
// bool PSMatching_;
std::vector<int> matching_;
};
} // namespace reco
#endif
|