63 {
65
67
68 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<int>> measurementRegionHandle(
m_measurementRegionKey, ctx);
69 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<int>> measurementDetectorHandle(
m_measurementDetectorKey, ctx);
70 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<int>> measurementLayerHandle(
m_measurementLayerKey, ctx);
71 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<int>> measurementTypeHandle(
m_measurementTypeKey, ctx);
72 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> chi2HitPredictedHandle(
m_chi2HitPredictedKey, ctx);
73 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> chi2HitFilteredHandle(
m_chi2HitFilteredKey, ctx);
74
75 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<int>> measurementPhiWidthHandle(
m_measurementPhiWidthKey, ctx);
76 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<int>> measurementEtaWidthHandle(
m_measurementEtaWidthKey, ctx);
77
78 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> residualLocXhandle(
m_residualLocXkey, ctx);
79 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> pullLocXhandle(
m_pullLocXkey, ctx);
80 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> measurementLocXhandle(
m_measurementLocXkey, ctx);
81 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> trackParameterLocXhandle(
m_trackParameterLocXkey, ctx);
82 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> measurementLocCovXhandle(
m_measurementLocCovXkey, ctx);
83 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> trackParameterLocCovXhandle(
m_trackParameterLocCovXkey, ctx);
84
85 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> residualLocYhandle(
m_residualLocYkey, ctx);
86 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> pullLocYhandle(
m_pullLocYkey, ctx);
87 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> measurementLocYhandle(
m_measurementLocYkey, ctx);
88 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> trackParameterLocYhandle(
m_trackParameterLocYkey, ctx);
89 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> measurementLocCovYhandle(
m_measurementLocCovYkey, ctx);
90 SG::WriteDecorHandle<xAOD::TrackParticleContainer, std::vector<float>> trackParameterLocCovYhandle(
m_trackParameterLocCovYkey, ctx);
91
95
97
99 ElementLink<ActsTrk::TrackContainer> link_to_track = actsTrackLink(*track_particle);
101
102
103 static_assert( std::is_same<ElementLink<ActsTrk::TrackContainer>::ElementConstReference,
104 std::optional<ActsTrk::TrackContainer::ConstTrackProxy> >::value);
105 std::optional<ActsTrk::TrackContainer::ConstTrackProxy> optional_track = *link_to_track;
106
107 if ( not optional_track.has_value() ) {
108 ATH_MSG_WARNING(
"Invalid track link for particle " << track_particle->index() <<
". Skipping track..");
109 continue;
110 }
111
112 ATH_MSG_DEBUG(
"Track link found for track particle with index " << track_particle->index());
113 ActsTrk::TrackContainer::ConstTrackProxy
track = optional_track.value();
114
115 std::vector<int>&
regions{measurementRegionHandle(*track_particle)};
117 std::vector<int>&
detectors{measurementDetectorHandle(*track_particle)};
119 std::vector<int>&
layers{measurementLayerHandle(*track_particle)};
121 std::vector<int>&
types{measurementTypeHandle(*track_particle)};
123 std::vector<float>& predchi2s{chi2HitPredictedHandle(*track_particle)};
124 predchi2s.reserve(
track.nMeasurements());
125 std::vector<float>& filtchi2s{chi2HitFilteredHandle(*track_particle)};
126 filtchi2s.reserve(
track.nMeasurements());
127
128 std::vector<int>& sizesPhi{measurementPhiWidthHandle(*track_particle)};
129 sizesPhi.reserve(
track.nMeasurements());
130 std::vector<int>& sizesEta{measurementEtaWidthHandle(*track_particle)};
131 sizesEta.reserve(
track.nMeasurements());
132 std::vector<float>& residualsLocX{residualLocXhandle(*track_particle)};
133 residualsLocX.reserve(
track.nMeasurements());
134 std::vector<float>& pullsLocX{pullLocXhandle(*track_particle)};
135 pullsLocX.reserve(
track.nMeasurements());
136 std::vector<float>& measurementsLocX{measurementLocXhandle(*track_particle)};
137 measurementsLocX.reserve(
track.nMeasurements());
138 std::vector<float>& trackParametersLocX{trackParameterLocXhandle(*track_particle)};
139 trackParametersLocX.reserve(
track.nMeasurements());
140 std::vector<float>& measurementsLocCovX{measurementLocCovXhandle(*track_particle)};
141 measurementsLocCovX.reserve(
track.nMeasurements());
142 std::vector<float>& trackParametersLocCovX{trackParameterLocCovXhandle(*track_particle)};
143 trackParametersLocCovX.reserve(
track.nMeasurements());
144 std::vector<float>& residualsLocY{residualLocYhandle(*track_particle)};
145 residualsLocY.reserve(
track.nMeasurements());
146 std::vector<float>& pullsLocY{pullLocYhandle(*track_particle)};
147 pullsLocY.reserve(
track.nMeasurements());
148 std::vector<float>& measurementsLocY{measurementLocYhandle(*track_particle)};
149 measurementsLocY.reserve(
track.nMeasurements());
150 std::vector<float>& trackParametersLocY{trackParameterLocYhandle(*track_particle)};
151 trackParametersLocY.reserve(
track.nMeasurements());
152 std::vector<float>& measurementsLocCovY{measurementLocCovYhandle(*track_particle)};
153 measurementsLocCovY.reserve(
track.nMeasurements());
154 std::vector<float>& trackParametersLocCovY{trackParameterLocCovYhandle(*track_particle)};
155 trackParametersLocCovY.reserve(
track.nMeasurements());
156
157 for (
const auto state :
track.trackStatesReversed()) {
158
159 auto flag = state.typeFlags();
160
161 bool anyHit =
flag.test(Acts::TrackStateFlag::HoleFlag) or
flag.test(Acts::TrackStateFlag::MeasurementFlag);
162 if (not anyHit) {
163 ATH_MSG_DEBUG(
"--- This is not a hit measurement, skipping...");
164 continue;
165 }
166
167
168
170 int region = -999;
173 float chi2_hit_predicted = -999.;
174 float chi2_hit_filtered = -999.;
175 int sizePhi = -999;
176 int sizeEta = -999;
177 float residualLocX = -999.;
178 float pullLocX = -999.;
179 float measurementLocX = -999.;
180 float trackParameterLocX = -999.;
181 float measurementLocCovX = -999.;
182 float trackParameterLocCovX = -999;
183 float residualLocY = -999.;
184 float pullLocY = -999.;
185 float measurementLocY = -999.;
186 float trackParameterLocY = -999.;
187 float measurementLocCovY = -999.;
188 float trackParameterLocCovY = -999.;
189 bool isAnnulusBound = false;
190
191
192 if (
flag.test(Acts::TrackStateFlag::HoleFlag)) {
195 }
else if (
flag.test(Acts::TrackStateFlag::OutlierFlag)) {
198 } else {
201 }
202
203
204 if (state.hasReferenceSurface() and state.referenceSurface().associatedDetectorElement()) {
205 const ActsDetectorElement * detectorElement = dynamic_cast<const ActsDetectorElement *>(state.referenceSurface().associatedDetectorElement());
206 if (!detectorElement) {
207 ATH_MSG_WARNING(
"--- TrackState reference surface returned an invalid associated detector element");
208 continue;
209 }
210 const InDetDD::SiDetectorElement * siliconDetectorElement =
dynamic_cast<const InDetDD::SiDetectorElement *
>(detectorElement->
upstreamDetectorElement());
211 if (!siliconDetectorElement) {
212 ATH_MSG_WARNING(
"--- TrackState associated detector element is not silicon");
213 continue;
214 }
215
216 const Acts::AnnulusBounds* annulusBounds = dynamic_cast<const Acts::AnnulusBounds*>(&(state.referenceSurface().bounds()));
217 isAnnulusBound = annulusBounds ? true : false;
218
219 if (siliconDetectorElement) {
220 Identifier detectorIdentifier = siliconDetectorElement->
identify();
221 if (siliconDetectorElement->
isPixel()) {
222 const PixelID* pixel_id =
static_cast<const PixelID *
>(siliconDetectorElement->
getIdHelper());
223 int layer_disk = pixel_id->
layer_disk(detectorIdentifier);
225 if (pixel_id->
barrel_ec(detectorIdentifier) == 0) {
226 if (layer_disk == 0) {
230 } else {
233 }
234 }
else if (siliconDetectorElement->
isSCT()) {
235 const SCT_ID* sct_id =
static_cast<const SCT_ID *
>(siliconDetectorElement->
getIdHelper());
237 region = sct_id->
barrel_ec(detectorIdentifier) == 0 ?
240 }
else ATH_MSG_WARNING(
"--- Unknown detector type - It is not pixel nor strip detecor element!");
242 }
else ATH_MSG_WARNING(
"--- Missing reference surface or associated detector element!");
243
244
245
246
247
249
250
251
253 chi2_hit_filtered = state.chi2();
254 }
255
256 if (state.hasUncalibratedSourceLink()) {
258 }
259
260
261 if (!state.hasSmoothed() || !state.hasProjector())
262 continue;
263
264
265 const auto &calibratedParameters = state.effectiveCalibrated();
266 const auto &calibratedCovariance = state.effectiveCalibratedCovariance();
267
268
269
270
271 bool evaluateUnbiased = (!
flag.test(Acts::TrackStateFlag::OutlierFlag));
272
273 if (evaluateUnbiased) {
274 ATH_MSG_DEBUG(
"--- Good for unbiased parameters evaluation!");
276
277 if (state.hasUncalibratedSourceLink()) {
288 } else {
289 ATH_MSG_DEBUG(
"xAOD::UncalibratedMeasurement is neither xAOD::PixelCluster nor xAOD::StripCluster");
290 }
291 }
292 }
293
294 const auto& [unbiasedParameters, unbiasedCovariance] =
295 evaluateUnbiased ? Acts::calculateUnbiasedParametersCovariance(state) : std::make_pair(state.
parameters(), state.covariance());
296
297 measurementLocX = calibratedParameters[Acts::eBoundLoc0];
298 measurementLocCovX = calibratedCovariance(Acts::eBoundLoc0, Acts::eBoundLoc0);
299
300 if (!isAnnulusBound) {
301
302 trackParameterLocX = unbiasedParameters[Acts::eBoundLoc0];
303 residualLocX = (measurementLocX - trackParameterLocX) / 1_um;
304 trackParameterLocCovX = unbiasedCovariance(Acts::eBoundLoc0, Acts::eBoundLoc0);
305 }
306 else {
307
308
309 float locR = unbiasedParameters[Acts::eBoundLoc0];
310 float covR = unbiasedCovariance(Acts::eBoundLoc0,Acts::eBoundLoc0);
311 float locphi = unbiasedParameters[Acts::eBoundLoc1];
312 float covphi = unbiasedCovariance(Acts::eBoundLoc1,Acts::eBoundLoc1);
313 float covRphi = unbiasedCovariance(Acts::eBoundLoc0,Acts::eBoundLoc1);
314
315 trackParameterLocX = locphi;
316 residualLocX =
locR * (measurementLocX - trackParameterLocX) / 1_um;
317
318 trackParameterLocCovX =
locR*
locR*covphi + locphi*locphi*covR + 2*locphi*
locR*covRphi;
319
320
321 measurementLocCovX =
locR*
locR * measurementLocCovX;
322 }
323
324 pullLocX =
evaluatePull(residualLocX, measurementLocCovX,
325 trackParameterLocCovX, evaluateUnbiased);
326
327 if (state.calibratedSize() == 2) {
328 measurementLocY = calibratedParameters[Acts::eBoundLoc1];
329 trackParameterLocY = unbiasedParameters[Acts::eBoundLoc1];
330 residualLocY = (measurementLocY - trackParameterLocY) / 1_um;
331 measurementLocCovY = calibratedCovariance(Acts::eBoundLoc1, Acts::eBoundLoc1);
332 trackParameterLocCovY = unbiasedCovariance(Acts::eBoundLoc1, Acts::eBoundLoc1);
333 pullLocY =
evaluatePull(residualLocY, measurementLocCovY,
334 trackParameterLocCovY, evaluateUnbiased);
335 }
336
337 }
338
340
341
342 auto pred = state.predicted();
343 trackParameterLocX = pred[Acts::eBoundLoc0];
344 trackParameterLocY = pred[Acts::eBoundLoc1];
345 }
346
347
348
352 types.push_back(type);
353 predchi2s.push_back(chi2_hit_predicted);
354 filtchi2s.push_back(chi2_hit_filtered);
355 sizesPhi.push_back(sizePhi);
356 sizesEta.push_back(sizeEta);
357 residualsLocX.push_back(residualLocX);
358 pullsLocX.push_back(pullLocX);
359 measurementsLocX.push_back(measurementLocX);
360 trackParametersLocX.push_back(trackParameterLocX);
361 measurementsLocCovX.push_back(measurementLocCovX);
362 trackParametersLocCovX.push_back(trackParameterLocCovX);
363 residualsLocY.push_back(residualLocY);
364 pullsLocY.push_back(pullLocY);
365 measurementsLocY.push_back(measurementLocY);
366 trackParametersLocY.push_back(trackParameterLocY);
367 measurementsLocCovY.push_back(measurementLocCovY);
368 trackParametersLocCovY.push_back(trackParameterLocCovY);
369
370
371 }
372
373 }
374
375 return StatusCode::SUCCESS;
376 }
#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
const GeoVDetectorElement * upstreamDetectorElement() const
Returns the underllying GeoModel detectorelement that this one is based on.
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer > m_chi2HitPredictedKey
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
PublicToolHandle< ActsTrk::ITrackingGeometryTool > m_trackingGeometryTool
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
bool isValid() const
Test to see if the link can be dereferenced.
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)
SG::ConstAccessor< T, ALLOC > ConstAccessor
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.
T * get(TKey *tobj)
get a TObject* from a TKey* (why can't a TObject be a TKey?)
const xAOD::UncalibratedMeasurement & getUncalibratedMeasurement(const ATLASUncalibSourceLink &source_link)
const xAOD::UncalibratedMeasurement * ATLASUncalibSourceLink
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.
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".
UncalibratedMeasurement_v1 UncalibratedMeasurement
Define the version of the uncalibrated measurement class.