|
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
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
|
#include "DataFormats/EgammaReco/interface/ElectronSeed.h"
#include <climits>
using namespace reco;
ElectronSeed::ElectronSeed()
: TrajectorySeed(),
ctfTrack_(),
caloCluster_(),
hitInfo_(),
nrLayersAlongTraj_(0),
isEcalDriven_(false),
isTrackerDriven_(false)
{}
ElectronSeed::ElectronSeed(const TrajectorySeed& seed)
: TrajectorySeed(seed),
ctfTrack_(),
caloCluster_(),
hitInfo_(),
nrLayersAlongTraj_(0),
isEcalDriven_(false),
isTrackerDriven_(false) {}
ElectronSeed::ElectronSeed(PTrajectoryStateOnDet& pts, RecHitContainer& rh, PropagationDirection& dir)
: TrajectorySeed(pts, rh, dir),
ctfTrack_(),
caloCluster_(),
hitInfo_(),
nrLayersAlongTraj_(0),
isEcalDriven_(false),
isTrackerDriven_(false) {}
ElectronSeed::~ElectronSeed() = default;
void ElectronSeed::setCtfTrack(const CtfTrackRef& ctfTrack) {
ctfTrack_ = ctfTrack;
isTrackerDriven_ = true;
}
//the hit mask tells us which hits were used in the seed
//typically all are used at the HLT but this could change in the future
//RECO only uses some of them
unsigned int ElectronSeed::hitsMask() const {
int mask = 0;
for (size_t hitNr = 0; hitNr < nHits(); hitNr++) {
int bitNr = 0x1 << hitNr;
int hitDetId = (recHits().begin() + hitNr)->geographicalId().rawId();
auto detIdMatcher = [hitDetId](const ElectronSeed::PMVars& var) { return hitDetId == var.detId; };
if (std::find_if(hitInfo_.begin(), hitInfo_.end(), detIdMatcher) != hitInfo_.end()) {
mask |= bitNr;
}
}
return mask;
}
void ElectronSeed::initTwoHitSeed(const unsigned char hitMask) {
hitInfo_.resize(2);
std::vector<unsigned int> hitNrs = hitNrsFromMask(hitMask);
if (hitNrs.size() != 2) {
throw cms::Exception("LogicError")
<< "in ElectronSeed::" << __FUNCTION__ << "," << __LINE__ << ": number of hits in hit mask is " << hitNrs.size()
<< "\n"
<< "pre-2017 pixel upgrade ecalDriven ElectronSeeds should have exactly 2 hits\n "
<< "mask " << static_cast<unsigned int>(hitMask) << std::endl;
}
if (hitNrs[0] >= nHits() || hitNrs[1] >= nHits()) {
throw cms::Exception("LogicError") << "in ElectronSeed::" << __FUNCTION__ << "," << __LINE__ << ": hits are "
<< hitNrs[0] << " and " << hitNrs[1] << " while number of hits are " << nHits()
<< "\n"
<< "this means there was a bug in storing or creating the electron seeds "
<< "mask " << static_cast<unsigned int>(hitMask) << std::endl;
}
for (size_t hitNr = 0; hitNr < hitInfo_.size(); hitNr++) {
auto& info = hitInfo_[hitNr];
info.setDPhi(std::numeric_limits<float>::max(), std::numeric_limits<float>::max());
info.setDRZ(std::numeric_limits<float>::max(), std::numeric_limits<float>::max());
info.setDet((recHits().begin() + hitNrs[hitNr])->geographicalId(), -1);
}
}
void ElectronSeed::setNegAttributes(float dRZ2, float dPhi2, float dRZ1, float dPhi1) {
if (hitInfo_.size() != 2) {
throw cms::Exception("LogicError")
<< "ElectronSeed::setNegAttributes should only operate on seeds with exactly two hits. This is because it is a "
"legacy function to preverse backwards compatiblity and should not be used on new code which matches "
"variable number of hits";
}
hitInfo_[0].dRZNeg = dRZ1;
hitInfo_[1].dRZNeg = dRZ2;
hitInfo_[0].dPhiNeg = dPhi1;
hitInfo_[1].dPhiNeg = dPhi2;
}
void ElectronSeed::setPosAttributes(float dRZ2, float dPhi2, float dRZ1, float dPhi1) {
if (hitInfo_.size() != 2) {
throw cms::Exception("LogicError")
<< "ElectronSeed::setPosAttributes should only operate on seeds with exactly two hits. This is because it is a "
"legacy function to preverse backwards compatiblity and should not be used on new code which matches "
"variable number of hits";
}
hitInfo_[0].dRZPos = dRZ1;
hitInfo_[1].dRZPos = dRZ2;
hitInfo_[0].dPhiPos = dPhi1;
hitInfo_[1].dPhiPos = dPhi2;
}
std::vector<unsigned int> ElectronSeed::hitNrsFromMask(unsigned int hitMask) {
std::vector<unsigned int> hitNrs;
for (size_t bitNr = 0; bitNr < sizeof(hitMask) * CHAR_BIT; bitNr++) {
char bit = 0x1 << bitNr;
if ((hitMask & bit) != 0)
hitNrs.push_back(bitNr);
}
return hitNrs;
}
std::vector<ElectronSeed::PMVars> ElectronSeed::createHitInfo(const float dPhi1Pos,
const float dPhi1Neg,
const float dRZ1Pos,
const float dRZ1Neg,
const float dPhi2Pos,
const float dPhi2Neg,
const float dRZ2Pos,
const float dRZ2Neg,
const char hitMask,
TrajectorySeed::RecHitRange const& recHits) {
if (hitMask == 0)
return std::vector<ElectronSeed::PMVars>(); //was trackerDriven so no matched hits
size_t nrRecHits = std::distance(recHits.begin(), recHits.end());
std::vector<unsigned int> hitNrs = hitNrsFromMask(hitMask);
if (hitNrs.size() != 2) {
throw cms::Exception("LogicError") << "in ElectronSeed::" << __FUNCTION__ << "," << __LINE__
<< ": number of hits in hit mask is " << nrRecHits << "\n"
<< "pre-2017 pixel upgrade ecalDriven ElectronSeeds should have exactly 2 hits\n"
<< "mask " << static_cast<unsigned int>(hitMask) << std::endl;
}
if (hitNrs[0] >= nrRecHits || hitNrs[1] >= nrRecHits) {
throw cms::Exception("LogicError") << "in ElectronSeed::" << __FUNCTION__ << "," << __LINE__ << ": hits are "
<< hitNrs[0] << " and " << hitNrs[1] << " while number of hits are " << nrRecHits
<< "\n"
<< "this means there was a bug in storing or creating the electron seeds "
<< "mask " << static_cast<unsigned int>(hitMask) << std::endl;
}
std::vector<PMVars> hitInfo(2);
hitInfo[0].setDPhi(dPhi1Pos, dPhi1Neg);
hitInfo[0].setDRZ(dRZ1Pos, dRZ1Neg);
hitInfo[0].setDet(
(recHits.begin() + hitNrs[0])->geographicalId(),
-1); //getting the layer information needs tracker topo, hence why its stored in the first as its a pain to access
hitInfo[1].setDPhi(dPhi2Pos, dPhi2Neg);
hitInfo[1].setDRZ(dRZ2Pos, dRZ2Neg);
hitInfo[1].setDet(
(recHits.begin() + hitNrs[1])->geographicalId(),
-1); //getting the layer information needs tracker topo, hence why its stored in the first as its a pain to access
return hitInfo;
}
ElectronSeed::PMVars::PMVars()
: dRZPos(std::numeric_limits<float>::max()),
dRZNeg(std::numeric_limits<float>::max()),
dPhiPos(std::numeric_limits<float>::max()),
dPhiNeg(std::numeric_limits<float>::max()),
detId(0),
layerOrDiskNr(-1) {}
void ElectronSeed::PMVars::setDPhi(float pos, float neg) {
dPhiPos = pos;
dPhiNeg = neg;
}
void ElectronSeed::PMVars::setDRZ(float pos, float neg) {
dRZPos = pos;
dRZNeg = neg;
}
void ElectronSeed::PMVars::setDet(int iDetId, int iLayerOrDiskNr) {
detId = iDetId;
layerOrDiskNr = iLayerOrDiskNr;
}
|