65{
66
67 std::map<Identifier, const SCT_RDORawData*> idToRAWDataMap;
69 SG::ReadHandle<SCT_RDO_Container> rdoContainer{
m_rdoContainer, ctx};
71
72 for (const auto collection: *rdoContainer) {
73
74 for (const auto rdo : *collection) {
75 if (rdo==nullptr) {
77 continue;
78 }
79 Identifier rdoId{rdo->identify()};
80 idToRAWDataMap.insert(std::pair<Identifier, const SCT_RDORawData*>{rdoId, rdo});
81 }
82 }
85 }
86 }
87 ATH_MSG_DEBUG(
"Size of RDO map is " << idToRAWDataMap.size());
88
89 const PRD_MultiTruthCollection* prdmtColl{nullptr};
90 const xAODTruthParticleLinkVector *truth_particle_links{nullptr};
92 SG::ReadHandle<PRD_MultiTruthCollection> prdmtCollHandle{
m_multiTruth, ctx};
93 if (prdmtCollHandle.
isValid()) {
94 prdmtColl = &*prdmtCollHandle;
95 }
98 if (truthParticleLinksHandle.
isValid()) {
99 truth_particle_links = truthParticleLinksHandle.
cptr();
100 }
101 }
102 }
103
104 const InDetSimDataCollection* sdoCollection{nullptr};
106 SG::ReadHandle<InDetSimDataCollection> sdoCollectionHandle{
m_SDOcontainer, ctx};
107 if (sdoCollectionHandle.
isValid()) {
108 sdoCollection = &*sdoCollectionHandle;
109 }
110 }
111
112 std::vector<std::vector<const SiHit*>> siHits(
m_SCTHelper->wafer_hash_max());
115 if (sihitCollection.
isValid()) {
116 for (const SiHit& siHit: *sihitCollection) {
117
118 if (not siHit.isSCT()) continue;
119
120 Identifier wafer_id{
m_SCTHelper->wafer_id(siHit.getBarrelEndcap(),
121 siHit.getLayerDisk(),
122 siHit.getPhiModule(),
123 siHit.getEtaModule(),
124 siHit.getSide())};
125 IdentifierHash wafer_hash{
m_SCTHelper->wafer_hash(wafer_id)};
126 siHits[wafer_hash].push_back(&siHit);
127 }
128 }
129 }
130
131
132 SG::ReadHandle<InDet::SCT_ClusterContainer> sctClusterContainer{
m_clustercontainer, ctx};
133 if (not sctClusterContainer.
isValid()) {
135 return StatusCode::FAILURE;
136 }
137
138
139 SG::WriteHandle<xAOD::TrackMeasurementValidationContainer> xaod{
m_xAodContainer, ctx};
140 ATH_CHECK(xaod.
record(std::make_unique<xAOD::TrackMeasurementValidationContainer>(),
141 std::make_unique<xAOD::TrackMeasurementValidationAuxContainer>()));
142
143 SG::WriteHandle<std::vector<unsigned int>> offsets{
m_xAodOffset, ctx};
145
146 unsigned int have_truth_link=0
u;
147 unsigned int missing_truth_particle=0
u;
148 unsigned int missing_parent_particle=0
u;
149
151 for (const auto clusterCollection: *sctClusterContainer) {
152
153 (*offsets)[clusterCollection->identifyHash()] =
counter;
154
155
156 if (clusterCollection->empty()) continue;
157
158 xaod->resize(counter + clusterCollection->size());
159
160 for (const InDet::SCT_Cluster* prd: *clusterCollection) {
161 Identifier clusterId{prd->identify()};
164 }
165
166
168 xaod->at(counter) = xprd;
170
171
173
174
177
178
181 float locX{
static_cast<float>(locpos.x())};
182 if ((not std::isinf(locpos.y()) or std::isnan(locpos.y()))) {
183 if (locpos.y()>=1e-07)
locY = locpos.y();
184 } else {
186 }
187
188
190
192 if (localCov.size() == 1) {
194 } else if (localCov.size() == 4) {
196 } else {
198 }
199
200
201 std::vector<uint64_t> rdoIdentifierList;
202 rdoIdentifierList.reserve(prd->rdoList().size());
203 for (const auto& hitIdentifier: prd->rdoList()) {
204 rdoIdentifierList.push_back(hitIdentifier.get_compact());
205 }
207
208
209 const InDet::SiWidth cw{prd->width()};
210 AUXDATA(xprd,
int, SiWidth) =
static_cast<int>(cw.
colRow()[0]);
211 AUXDATA(xprd,
int, hitsInThirdTimeBin) =
static_cast<int>(prd->hitsInThirdTimeBin());
212
218
219
220 const InDetDD::SiDetectorElement* de{prd->detectorElement()};
222 if (de) {
226 }
227 }
228 AUXDATA(xprd, uint64_t, detectorElementID) = detElementId;
229
230
233 }
234
235
237 if (prdmtColl) {
238 auto range{prdmtColl->equal_range(clusterId)};
239 if (truth_particle_links) {
240 std::vector<unsigned int> tp_indices;
242 ElementLink<xAOD::TruthParticleContainer> a_truth_particle_link = truth_particle_links->
find(
i->second);
243 if (a_truth_particle_link) {
245 if (truth_particle) {
246 ++have_truth_link;
247 tp_indices.push_back(static_cast<int>(truth_particle->index()));
248 }
249 else {
250 ++missing_parent_particle;
251 }
252 }
253 else {
254 tp_indices.push_back(std::numeric_limits<unsigned int>::max());
255 ++missing_truth_particle;
256 }
257 }
258 AUXDATA(xprd, std::vector<unsigned int>, truth_index) = std::move(tp_indices);
259 }
260 std::vector<int> uniqueIDs;
263 }
264 AUXDATA(xprd, std::vector<int>, truth_barcode) = std::move(uniqueIDs);
265 }
266 }
267
268
269
271 if (sdoCollection) {
273 }
274 }
275
276
277
280 }
281 }
282 }
283 ATH_MSG_DEBUG(
" recorded SCT_PrepData objects: size " << xaod->size());
284
289
290 return StatusCode::SUCCESS;
291}
#define ATH_CHECK
Evaluate an expression and check for errors.
#define ATH_MSG_WARNING(x)
bool is_valid() const
Check if id is in a valid state.
value_type get_compact() const
Get the compact id.
virtual Identifier identify() const override final
identifier of this detector element (inline)
const Amg::Vector2D & colRow() const
void addSiHitInformation(xAOD::TrackMeasurementValidation *xprd, const InDet::SCT_Cluster *prd, const std::vector< const SiHit * > *siHits) const
std::atomic< unsigned int > m_missingParentParticle
void addSDOInformation(xAOD::TrackMeasurementValidation *xprd, const InDet::SCT_Cluster *prd, const InDetSimDataCollection *sdoCollection) const
SG::ReadHandleKey< SiHitCollection > m_sihitContainer
SG::ReadHandleKey< PRD_MultiTruthCollection > m_multiTruth
SG::ReadHandleKey< SCT_RDO_Container > m_rdoContainer
BooleanProperty m_writeSDOs
SG::ReadHandleKey< InDetSimDataCollection > m_SDOcontainer
std::atomic_bool m_firstEventWarnings
BooleanProperty m_writeSiHits
SG::WriteHandleKey< xAOD::TrackMeasurementValidationContainer > m_xAodContainer
SG::WriteHandleKey< std::vector< unsigned int > > m_xAodOffset
BooleanProperty m_writeRDOinformation
SG::ReadHandleKey< xAODTruthParticleLinkVector > m_truthParticleLinks
std::atomic< unsigned int > m_missingTruthParticle
SG::ReadHandleKey< InDet::SCT_ClusterContainer > m_clustercontainer
void addRDOInformation(xAOD::TrackMeasurementValidation *, const InDet::SCT_Cluster *, const std::map< Identifier, const SCT_RDORawData * > &idToRAWDataMap) const
std::atomic< unsigned int > m_haveTruthLink
BooleanProperty m_useTruthInfo
virtual bool isValid() override final
Can the handle be successfully dereferenced?
const_pointer_type cptr()
Dereference the pointer.
StatusCode record(std::unique_ptr< T > data)
Record a const object to the store.
ElementLink< xAOD::TruthParticleContainer > find(const HepMcParticleLink &hepMCLink) const
void setRdoIdentifierList(const std::vector< uint64_t > &rdoIdentifierList)
Sets the list of RDO identifiers.
void setLocalPositionError(float localXError, float localYError, float localXYCorrelation)
Sets the local position error.
void setLocalPosition(float localX, float localY)
Sets the local position.
void setIdentifier(uint64_t identifier)
Sets the identifier.
void setGlobalPosition(float globalX, float globalY, float globalZ)
Sets the global position.
Eigen::Matrix< double, Eigen::Dynamic, Eigen::Dynamic > MatrixX
Dynamic Matrix - dynamic allocation.
Eigen::Matrix< double, 3, 1 > Vector3D
@ u
Enums for curvilinear frames.
TrackMeasurementValidation_v1 TrackMeasurementValidation
Reference the current persistent version:
TruthParticle_v1 TruthParticle
Typedef to implementation.