32 : base_class(
t,
n,
p) {}
68 <<
", y: " << last_param->
position().y()
69 <<
", z: " << last_param->
position().z());
71 <<
", py: " << last_param->
momentum().y()
72 <<
", pz: " << last_param->
momentum().z());
84 bool is_pos_endcap = last_param->
eta() > 0;
88 is_pos_endcap ?
"HGTD::PositiveEndcap" :
"HGTD::NegativeEndcap");
90 if (not hgtd_trk_volume) {
96 hgtd_trk_volume->confinedLayers();
98 if (not confined_layers) {
105 std::span<Trk::Layer const * const>
layers = confined_layers->arrayObjects();
106 size_t layer_size =
layers.size();
108 short position = is_pos_endcap ? 0 : layer_size - 1;
109 short step = is_pos_endcap ? 1 : -1;
110 short hgtd_layer_i = -1;
111 for (
size_t i = 0;
i < layer_size - 1;
i++, position = position +
step) {
115 if (
layer->layerType() != 1) {
126 std::unique_ptr<const Trk::TrackParameters> extrap_result =
nullptr;
137 if (not extrap_result) {
139 result.m_hits.at(hgtd_layer_i) =
nullptr;
140 result.m_truth_primary_hits.at(hgtd_layer_i) =
nullptr;
146 if (hgtd_layer_i == 0) {
147 result.m_extrap_x = extrap_result->position().x();
148 result.m_extrap_y = extrap_result->position().y();
152 << extrap_result->position().x()
153 <<
", y: " << extrap_result->position().y()
154 <<
", z: " << extrap_result->position().z());
156 << extrap_result->momentum().x()
157 <<
", py: " << extrap_result->momentum().y()
158 <<
", pz: " << extrap_result->momentum().z());
163 if (compatible_surfaces.empty()) {
165 result.m_hits.at(hgtd_layer_i) =
nullptr;
166 result.m_truth_primary_hits.at(hgtd_layer_i) =
nullptr;
171 auto extrapolated_params =
174 std::unique_ptr<const Trk::TrackStateOnSurface> updated_state =
178 if (not updated_state) {
179 result.m_hits.at(hgtd_layer_i) =
nullptr;
180 result.m_truth_primary_hits.at(hgtd_layer_i) =
nullptr;
186 last_param = updated_state->trackParameters();
189 std::pair<const HGTD_Cluster*, HGTD::ClusterTruthInfo> truth_info =
193 result.m_hits.at(hgtd_layer_i) = std::move(updated_state);
194 result.m_truth_primary_hits.at(hgtd_layer_i) = truth_info.first;
195 result.m_truth_primary_info.at(hgtd_layer_i) = truth_info.second;
205 track.trackStateOnSurfaces();
212 for (
auto i = tsos->rbegin();
i != tsos->rend(); ++
i) {
213 const auto* curr_last_tsos = *
i;
214 if (not curr_last_tsos) {
218 curr_last_tsos->trackParameters() and
219 curr_last_tsos->measurementOnTrack()) {
220 return curr_last_tsos;
226 std::vector<const Trk::Surface*>
230 std::vector<const Trk::Surface*> surfaces;
235 const auto* surface_arr =
layer->surfaceArray();
239 const Trk::Surface* module_surface = surface_arr->object(
241 if (!module_surface) {
244 surfaces.push_back(module_surface);
256 if (!additional_surface) {
260 if (
std::find(surfaces.begin(), surfaces.end(), additional_surface) ==
262 surfaces.push_back(additional_surface);
269 std::vector<std::unique_ptr<const Trk::TrackParameters>>
272 const std::vector<const Trk::Surface*>& surfaces)
const {
274 std::vector<std::unique_ptr<const Trk::TrackParameters>>
params;
275 params.reserve(surfaces.size());
277 for (
const auto* surface : surfaces) {
278 std::unique_ptr<const Trk::TrackParameters> extrapolated_params =
nullptr;
283 if (not extrapolated_params) {
286 params.push_back(std::move(extrapolated_params));
292 std::unique_ptr<const Trk::TrackStateOnSurface>
295 const std::vector<std::unique_ptr<const Trk::TrackParameters>>&
params,
297 ATH_MSG_DEBUG(
"[updateStateWithBestFittingCluster] start updating");
299 std::unique_ptr<const Trk::TrackStateOnSurface> updated_state =
nullptr;
301 double lowest_chi2 = -1.;
303 for (
const auto& param :
params) {
304 std::unique_ptr<const Trk::TrackStateOnSurface> best_tsos =
307 ATH_MSG_DEBUG(
"[updateStateWithBestFittingCluster] tsos is null");
310 ATH_MSG_DEBUG(
"[updateStateWithBestFittingCluster] tsos found");
312 double chi2 = best_tsos->fitQualityOnSurface().chiSquared() /
313 best_tsos->fitQualityOnSurface().doubleNumberDoF();
315 "[updateStateWithBestFittingCluster] found state with chi2 of "
318 if (!updated_state or
chi2 < lowest_chi2) {
319 updated_state.swap(best_tsos);
324 return updated_state;
328 std::unique_ptr<const Trk::TrackStateOnSurface>
335 std::unique_ptr<const Trk::TrackStateOnSurface> tsos =
nullptr;
337 double lowest_chi2 = -1;
338 for (
const auto* collection : *container) {
340 if (collection->identify() !=
345 "[findBestCompatibleCluster] found collection of given surface");
347 for (
const auto* cluster : *collection) {
349 std::unique_ptr<const Trk::TrackStateOnSurface> candidate =
355 if (not candidate->measurementOnTrack() and
356 not candidate->fitQualityOnSurface()) {
359 double chi2 = candidate->fitQualityOnSurface().chiSquared() /
360 candidate->fitQualityOnSurface().doubleNumberDoF();
367 if (not tsos or
chi2 < lowest_chi2) {
368 tsos.swap(candidate);
380 std::unique_ptr<const Trk::TrackStateOnSurface>
391 std::pair<float, float> corr_time_and_res =
395 std::unique_ptr<HGTD_ClusterOnTrack> cot =
396 std::make_unique<HGTD_ClusterOnTrack>(
399 corr_time_and_res.second, det_el->identifyHash());
403 std::unique_ptr<Trk::TrackParameters>
pars =
m_updator->addToState(
407 auto uniqueQuality= std::unique_ptr<Trk::FitQualityOnSurface>(
quality);
409 return std::make_unique<const Trk::TrackStateOnSurface>(QoS, std::move(cot),
413 std::pair<const HGTD_Cluster*, HGTD::ClusterTruthInfo>
415 const std::vector<const Trk::Surface*>& surfaces,
420 if (not truth_ptkl or not sim_data) {
429 std::vector<Identifier>
ids;
430 std::for_each(surfaces.begin(), surfaces.end(),
432 ids.push_back(surf->associatedDetectorElementIdentifier());
435 for (
const auto *
const collection : *container) {
442 for (
const auto* cluster : *collection) {
445 cluster, truth_ptkl, sim_data, hs_event);
449 return {cluster, truth_info};