60 {
62
64
65 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<int>> measurementRegionHandle(
m_measurementRegionKey, ctx);
66 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<int>> measurementDetectorHandle(
m_measurementDetectorKey, ctx);
67 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<int>> measurementLayerHandle(
m_measurementLayerKey, ctx);
68 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<int>> measurementTypeHandle(
m_measurementTypeKey, ctx);
69 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> chi2HitPredictedHandle(
m_chi2HitPredictedKey, ctx);
70 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> chi2HitFilteredHandle(
m_chi2HitFilteredKey, ctx);
71
72 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<int>> measurementPhiWidthHandle(
m_measurementPhiWidthKey, ctx);
73 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<int>> measurementEtaWidthHandle(
m_measurementEtaWidthKey, ctx);
74
75 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> residualLocXhandle(
m_residualLocXkey, ctx);
76 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> pullLocXhandle(
m_pullLocXkey, ctx);
77 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> measurementLocXhandle(
m_measurementLocXkey, ctx);
78 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> trackParameterLocXhandle(
m_trackParameterLocXkey, ctx);
79 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> measurementLocCovXhandle(
m_measurementLocCovXkey, ctx);
80 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> trackParameterLocCovXhandle(
m_trackParameterLocCovXkey, ctx);
81
82 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> residualLocYhandle(
m_residualLocYkey, ctx);
83 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> pullLocYhandle(
m_pullLocYkey, ctx);
84 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> measurementLocYhandle(
m_measurementLocYkey, ctx);
85 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> trackParameterLocYhandle(
m_trackParameterLocYkey, ctx);
86 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> measurementLocCovYhandle(
m_measurementLocCovYkey, ctx);
87 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> trackParameterLocCovYhandle(
m_trackParameterLocCovYkey, ctx);
88
92
94
95 std::optional<ActsTrk::TrackContainer::ConstTrackProxy> optional_track =
getActsTrack(*track_particle);
96
97 if ( not optional_track.has_value() ) {
98 ATH_MSG_WARNING(
"Invalid track link for particle " << track_particle->index() <<
". Skipping track..");
99 continue;
100 }
101
102 ATH_MSG_DEBUG(
"Track link found for track particle with index " << track_particle->index());
103 ActsTrk::TrackContainer::ConstTrackProxy
track = optional_track.value();
104
105 std::vector<int>&
regions{measurementRegionHandle(*track_particle)};
107 std::vector<int>&
detectors{measurementDetectorHandle(*track_particle)};
109 std::vector<int>&
layers{measurementLayerHandle(*track_particle)};
111 std::vector<int>&
types{measurementTypeHandle(*track_particle)};
113 std::vector<float>& predchi2s{chi2HitPredictedHandle(*track_particle)};
114 predchi2s.reserve(
track.nMeasurements());
115 std::vector<float>& filtchi2s{chi2HitFilteredHandle(*track_particle)};
116 filtchi2s.reserve(
track.nMeasurements());
117
118 std::vector<int>& sizesPhi{measurementPhiWidthHandle(*track_particle)};
119 sizesPhi.reserve(
track.nMeasurements());
120 std::vector<int>& sizesEta{measurementEtaWidthHandle(*track_particle)};
121 sizesEta.reserve(
track.nMeasurements());
122 std::vector<float>& residualsLocX{residualLocXhandle(*track_particle)};
123 residualsLocX.reserve(
track.nMeasurements());
124 std::vector<float>& pullsLocX{pullLocXhandle(*track_particle)};
125 pullsLocX.reserve(
track.nMeasurements());
126 std::vector<float>& measurementsLocX{measurementLocXhandle(*track_particle)};
127 measurementsLocX.reserve(
track.nMeasurements());
128 std::vector<float>& trackParametersLocX{trackParameterLocXhandle(*track_particle)};
129 trackParametersLocX.reserve(
track.nMeasurements());
130 std::vector<float>& measurementsLocCovX{measurementLocCovXhandle(*track_particle)};
131 measurementsLocCovX.reserve(
track.nMeasurements());
132 std::vector<float>& trackParametersLocCovX{trackParameterLocCovXhandle(*track_particle)};
133 trackParametersLocCovX.reserve(
track.nMeasurements());
134 std::vector<float>& residualsLocY{residualLocYhandle(*track_particle)};
135 residualsLocY.reserve(
track.nMeasurements());
136 std::vector<float>& pullsLocY{pullLocYhandle(*track_particle)};
137 pullsLocY.reserve(
track.nMeasurements());
138 std::vector<float>& measurementsLocY{measurementLocYhandle(*track_particle)};
139 measurementsLocY.reserve(
track.nMeasurements());
140 std::vector<float>& trackParametersLocY{trackParameterLocYhandle(*track_particle)};
141 trackParametersLocY.reserve(
track.nMeasurements());
142 std::vector<float>& measurementsLocCovY{measurementLocCovYhandle(*track_particle)};
143 measurementsLocCovY.reserve(
track.nMeasurements());
144 std::vector<float>& trackParametersLocCovY{trackParameterLocCovYhandle(*track_particle)};
145 trackParametersLocCovY.reserve(
track.nMeasurements());
146
147 for (
const auto state :
track.trackStatesReversed()) {
148
149 auto flag = state.typeFlags();
150
151 bool anyHit =
flag.isHole() or
flag.hasMeasurement() or state.hasUncalibratedSourceLink();
152 if (not anyHit) {
153 ATH_MSG_DEBUG(
"--- This is not a hit measurement, skipping...");
154 continue;
155 }
156
157
158
160 int region = -999;
163 float chi2_hit_predicted = -999.;
164 float chi2_hit_filtered = -999.;
165 int sizePhi = -999;
166 int sizeEta = -999;
167 float residualLocX = -999.;
168 float pullLocX = -999.;
169 float measurementLocX = -999.;
170 float trackParameterLocX = -999.;
171 float measurementLocCovX = -999.;
172 float trackParameterLocCovX = -999;
173 float residualLocY = -999.;
174 float pullLocY = -999.;
175 float measurementLocY = -999.;
176 float trackParameterLocY = -999.;
177 float measurementLocCovY = -999.;
178 float trackParameterLocCovY = -999.;
179 bool isAnnulusBound = false;
180
181
185 }
else if (
flag.isOutlier()) {
188 }
else if (
flag.hasMeasurement()) {
191 }
192
193
194 if (state.hasReferenceSurface() and state.referenceSurface().isSensitive()) {
196 if (!detectorElement) {
197 ATH_MSG_WARNING(
"--- TrackState reference surface returned an invalid associated detector element");
198 continue;
199 }
200 const InDetDD::SiDetectorElement * siliconDetectorElement = dynamic_cast<const InDetDD::SiDetectorElement *>(detectorElement->upstreamDetectorElement());
201 if (!siliconDetectorElement) {
202 ATH_MSG_WARNING(
"--- TrackState associated detector element is not silicon");
203 continue;
204 }
205
206 const Acts::AnnulusBounds* annulusBounds = dynamic_cast<const Acts::AnnulusBounds*>(&(state.referenceSurface().bounds()));
207 isAnnulusBound = annulusBounds ? true : false;
208
209 if (siliconDetectorElement) {
210 Identifier detectorIdentifier = siliconDetectorElement->
identify();
211 if (siliconDetectorElement->
isPixel()) {
212 const PixelID* pixel_id =
static_cast<const PixelID *
>(siliconDetectorElement->
getIdHelper());
213 int layer_disk = pixel_id->
layer_disk(detectorIdentifier);
215 if (pixel_id->
barrel_ec(detectorIdentifier) == 0) {
216 if (layer_disk == 0) {
220 } else {
223 }
224 }
else if (siliconDetectorElement->
isSCT()) {
225 const SCT_ID* sct_id =
static_cast<const SCT_ID *
>(siliconDetectorElement->
getIdHelper());
227 region = sct_id->
barrel_ec(detectorIdentifier) == 0 ?
230 }
else ATH_MSG_WARNING(
"--- Unknown detector type - It is not pixel nor strip detecor element!");
232 }
else ATH_MSG_WARNING(
"--- Missing reference surface or associated detector element!");
233
234
235
236
237
239
240
241
243 chi2_hit_filtered = state.chi2();
244 }
245
246 if (state.hasUncalibratedSourceLink()) {
248 }
249
250
251 if (!state.hasSmoothed() || !state.hasProjector())
252 continue;
253
254
255 const auto &calibratedParameters = state.effectiveCalibrated();
256 const auto &calibratedCovariance = state.effectiveCalibratedCovariance();
257
258
259
260
261 bool evaluateUnbiased =
flag.isMeasurement();
262
263 if (evaluateUnbiased) {
264 ATH_MSG_DEBUG(
"--- Good for unbiased parameters evaluation!");
266
267 if (state.hasUncalibratedSourceLink()) {
277 } else {
278 ATH_MSG_DEBUG(
"xAOD::UncalibratedMeasurement is neither xAOD::PixelCluster nor xAOD::StripCluster");
279 }
280 }
281 }
282
283 const auto& [unbiasedParameters, unbiasedCovariance] =
284 evaluateUnbiased ? Acts::calculateUnbiasedParametersCovariance(state) : std::make_pair(state.
parameters(), state.covariance());
285
286 measurementLocX = calibratedParameters[Acts::eBoundLoc0];
287 measurementLocCovX = calibratedCovariance(Acts::eBoundLoc0, Acts::eBoundLoc0);
288
289 if (!isAnnulusBound) {
290
291 trackParameterLocX = unbiasedParameters[Acts::eBoundLoc0];
292 residualLocX = (measurementLocX - trackParameterLocX) / 1_um;
293 trackParameterLocCovX = unbiasedCovariance(Acts::eBoundLoc0, Acts::eBoundLoc0);
294 }
295 else {
296
297
298 float locR = unbiasedParameters[Acts::eBoundLoc0];
299 float covR = unbiasedCovariance(Acts::eBoundLoc0,Acts::eBoundLoc0);
300 float locphi = unbiasedParameters[Acts::eBoundLoc1];
301 float covphi = unbiasedCovariance(Acts::eBoundLoc1,Acts::eBoundLoc1);
302 float covRphi = unbiasedCovariance(Acts::eBoundLoc0,Acts::eBoundLoc1);
303
304 trackParameterLocX = locphi;
305 residualLocX =
locR * (measurementLocX - trackParameterLocX) / 1_um;
306
307 trackParameterLocCovX =
locR*
locR*covphi + locphi*locphi*covR + 2*locphi*
locR*covRphi;
308
309
310 measurementLocCovX =
locR*
locR * measurementLocCovX;
311 }
312
313 pullLocX =
evaluatePull(residualLocX, measurementLocCovX,
314 trackParameterLocCovX, evaluateUnbiased);
315
316 if (state.calibratedSize() == 2) {
317 measurementLocY = calibratedParameters[Acts::eBoundLoc1];
318 trackParameterLocY = unbiasedParameters[Acts::eBoundLoc1];
319 residualLocY = (measurementLocY - trackParameterLocY) / 1_um;
320 measurementLocCovY = calibratedCovariance(Acts::eBoundLoc1, Acts::eBoundLoc1);
321 trackParameterLocCovY = unbiasedCovariance(Acts::eBoundLoc1, Acts::eBoundLoc1);
322 pullLocY =
evaluatePull(residualLocY, measurementLocCovY,
323 trackParameterLocCovY, evaluateUnbiased);
324 }
325
326 }
327
329
330
331 auto pred = state.predicted();
332 trackParameterLocX = pred[Acts::eBoundLoc0];
333 trackParameterLocY = pred[Acts::eBoundLoc1];
334 }
335
336 else {
339 }
340
341
342
346 types.push_back(type);
347 predchi2s.push_back(chi2_hit_predicted);
348 filtchi2s.push_back(chi2_hit_filtered);
349 sizesPhi.push_back(sizePhi);
350 sizesEta.push_back(sizeEta);
351 residualsLocX.push_back(residualLocX);
352 pullsLocX.push_back(pullLocX);
353 measurementsLocX.push_back(measurementLocX);
354 trackParametersLocX.push_back(trackParameterLocX);
355 measurementsLocCovX.push_back(measurementLocCovX);
356 trackParametersLocCovX.push_back(trackParameterLocCovX);
357 residualsLocY.push_back(residualLocY);
358 pullsLocY.push_back(pullLocY);
359 measurementsLocY.push_back(measurementLocY);
360 trackParametersLocY.push_back(trackParameterLocY);
361 measurementsLocCovY.push_back(measurementLocCovY);
362 trackParametersLocCovY.push_back(trackParameterLocCovY);
363
364
365 }
366
367 }
368
369 return StatusCode::SUCCESS;
370 }
const ActsDetectorElement * getActsDetectorElement(const Acts::Surface &surf)
Attempts to retrieve the ActsDetectorElement associated to the passed ActsSurface.
#define ATH_CHECK
Evaluate an expression and check for errors.
#define ATH_MSG_WARNING(x)
static const std::vector< std::string > types
static const std::vector< std::string > regions
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_chi2HitPredictedKey
ContextUtility m_ctxProvider
Context provider for geometry, magnetic field and calibration contexts.
float evaluatePull(const float residual, const float measurementCovariance, const float trackParameterCovariance, const bool evaluateUnbiased) const
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_measurementRegionKey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_residualLocYkey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_trackParameterLocYkey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_measurementLayerKey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_measurementDetectorKey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_measurementLocYkey
float getChi2Contribution(const typename ActsTrk::TrackStateBackend::ConstTrackStateProxy &state) const
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_measurementEtaWidthKey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_pullLocYkey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_measurementLocXkey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_measurementTypeKey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_residualLocXkey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_pullLocXkey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_measurementLocCovXkey
SG::ReadHandleKey< xAOD::TrackParticleContainer > m_trackParticlesKey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_trackParameterLocXkey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_trackParameterLocCovYkey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_measurementLocCovYkey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_trackParameterLocCovXkey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_chi2HitFilteredKey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_measurementPhiWidthKey
static const xAOD::UncalibratedMeasurement * unpack(const Acts::SourceLink &sl)
Helper method to unpack an Acts source link to an uncalibrated measurement.
virtual Identifier identify() const override final
identifier of this detector element (inline)
const AtlasDetectorID * getIdHelper() const
Returns the id helper (inline).
int layer_disk(const Identifier &id) const
int barrel_ec(const Identifier &id) const
Values of different levels (failure returns 0).
int layer_disk(const Identifier &id) const
int barrel_ec(const Identifier &id) const
Values of different levels (failure returns 0).
virtual bool isValid() override final
Can the handle be successfully dereferenced?
const_pointer_type cptr()
Dereference the pointer.
virtual xAOD::UncalibMeasType type() const =0
Returns the type of the measurement type as a simple enumeration.
std::optional< ActsTrk::TrackContainer::ConstTrackProxy > getActsTrack(const xAOD::TrackParticle &trkPart)
Return the proxy to the Acts track from which the track particle was made frome.
layers(flags, cells_name, *args, **kw)
Here we define wrapper functions to set up all of the standard corrections.
SG::ReadCondHandle< T > makeHandle(const SG::ReadCondHandleKey< T > &key, const EventContext &ctx=Gaudi::Hive::currentContext())
StripCluster_v1 StripCluster
Define the version of the strip cluster class.
UncalibratedMeasurement_v1 UncalibratedMeasurement
Define the version of the uncalibrated measurement class.
TrackParticle_v1 TrackParticle
Reference the current persistent version:
PixelCluster_v1 PixelCluster
Define the version of the pixel cluster class.
UncalibMeasType
Define the type of the uncalibrated measurement.
TrackParticleContainer_v1 TrackParticleContainer
Definition of the current "TrackParticle container version".