104 {
105
106 const std::size_t nSP = spacepoints.size();
107
109
110
111 std::vector<std::size_t> sortIdx(nSP);
112 std::iota(sortIdx.begin(), sortIdx.end(), 0);
113 std::ranges::sort(sortIdx, std::less{}, [&](std::size_t
i) {
114 return spacepoints[
i]->clusterList().first->detectorElement()->identify().get_compact();
115 });
116
117
119 std::vector<std::uint64_t> moduleIds(nSP);
120 std::vector<int>
ids(nSP);
121
122 for (std::size_t k = 0;
k < nSP; ++
k) {
123 const std::size_t origIdx = sortIdx[
k];
125
126 moduleIds[
k] = spacepoints[origIdx]->clusterList().first->detectorElement()->identify().get_compact();
127 ids[
k] =
static_cast<int>(
k);
130 }
131 }
132
133
135 std::unique_lock<std::mutex>
lock;
136 if (m_runMutex)
lock = std::unique_lock<std::mutex>(*m_runMutex);
137
138 if (edgeMap != nullptr) {
139 ScoredGraphHook hook;
140 auto result =
m_gnnPipeline->run(features, moduleIds, ids, ActsPlugins::Device::Cuda(0), hook);
141
142
143 const std::vector<float>& edgeScores = hook.getEdgeScores();
144 const std::vector<std::int64_t>& edgeIndex = hook.getEdgeIndex();
145 const std::size_t nEdges = edgeScores.size();
146
147
148 for (std::size_t i = 0;
i < nEdges; ++
i) {
149 std::int64_t
src = edgeIndex[
i];
150 std::int64_t dst = edgeIndex[nEdges +
i];
151 (*edgeMap)[sortIdx[
src]][sortIdx[dst]] = edgeScores[
i];
152 }
154 }
155
156 return m_gnnPipeline->run(features, moduleIds, ids, ActsPlugins::Device::Cuda(0));;
157 }();
158
160
161
162 tracks.clear();
164
165 for (const auto& candidate : candidates) {
167 continue;
168 }
169
170
171 std::vector<uint32_t>
track;
172 track.reserve(candidate.size());
173 for (int sortedIdx : candidate) {
175 }
176 tracks.push_back(std::move(track));
177 }
178
179 ATH_MSG_DEBUG(
"Returning " << tracks.size() <<
" track candidates after filtering (>= "
181
182 return StatusCode::SUCCESS;
183}
virtual void lock()=0
Interface to allow an object to lock itself when made const in SG.
float j(const xAOD::IParticle &, const xAOD::TrackMeasurementValidation &hit, const Eigen::Matrix3d &jab_inv)
1.16.1