12 #include "Acts/Definitions/Common.hpp"
13 #include "Acts/Definitions/Algebra.hpp"
14 #include "Acts/Utilities/VectorHelpers.hpp"
15 #include "Acts/Utilities/Delegate.hpp"
16 #include "Acts/Utilities/Result.hpp"
17 #include "Acts/Geometry/GeometryContext.hpp"
18 #include "Acts/Utilities/CalibrationContext.hpp"
19 #include "Acts/EventData/TrackParameters.hpp"
20 #include "Acts/Surfaces/Surface.hpp"
21 #include "Acts/Utilities/Logger.hpp"
22 #include "Acts/TrackFinding/CombinatorialKalmanFilter.hpp"
23 #include "Acts/TrackFinding/TrackStateCreator.hpp"
36 #include "boost/container/small_vector.hpp"
39 #include <type_traits>
45 template <std::
size_t NMeasMax,
typename traj_t,
typename measurement_container_variant_t>
48 static constexpr
bool s_fullPreCalibration=
true;
53 template <std::
size_t DIM>
72 template <std::
size_t DIM>
94 template <std::
size_t NMeasMax,
typename traj_t,
typename measurement_container_variant_t>
98 template <std::
size_t N>
102 template <std::
size_t N>
107 template <std::
size_t N>
114 template <std::
size_t N>
119 template <std::
size_t N>
123 template <std::
size_t N>
127 template <
typename T_MeasurementRangeIterator>
128 struct MeasurementContainerTraits {
129 using value_type =
typename T_MeasurementRangeIterator::value_type;
144 using BoundState = std::tuple<BoundTrackParameters, BoundMatrix, double>;
154 template <std::
size_t NMeasMax,
typename traj_t,
typename measurement_container_variant_t>
157 AtlasMeasurementSelector<NMeasMax, traj_t, measurement_container_variant_t> ,
158 measurement_container_variant_t >
163 measurement_container_variant_t >;
167 template <std::
size_t DIM>
173 template <std::
size_t DIM,
typename measurement_t>
175 std::pair < typename traits::template CalibratedMeasurement<DIM>,
177 (
const Acts::GeometryContext&,
178 const Acts::CalibrationContext&,
179 const measurement_t &,
180 const typename traits::BoundTrackParameters &)>;
186 template <std::
size_t DIM>
188 template <std::
size_t DIM>
192 template <
typename T_Container>
209 std::conditional<s_fullPreCalibration,
221 template <std::
size_t DIM,
typename T_ValueType>
223 m_calibrators.template setCalibrator<DIM, T_ValueType>(calibrator);
226 template <std::
size_t DIM>
227 static constexpr
bool s_CanPreCalibrate = std::is_same< typename traits::template PreSelectionMeasurement<DIM>,
229 && std::is_same< typename traits::template PreSelectionMeasurementCovariance<DIM>,
232 template < std::
size_t DIM,
typename T_ValueType >
238 template <
typename T_Value>
241 static_assert( !std::is_same<std::remove_pointer_t<T_Value>, T_Value>::
value );
248 template <std::
size_t DIM>
251 const Acts::CalibrationContext& calibrationContext,
252 const Acts::Surface& surface)
const {
257 template <std::
size_t DIM,
typename measurement_t>
260 return m_calibrators.template calibrator<DIM,measurement_t>();
264 template <std::
size_t DIM,
typename measurement_t>
267 if constexpr(s_fullPreCalibration) {
277 return []( [[maybe_unused]]
const Acts::GeometryContext&,
278 [[maybe_unused]]
const Acts::CalibrationContext&,
279 const measurement_t &measurement,
280 [[maybe_unused]]
const typename traits::BoundTrackParameters &) {
281 return std::make_pair( measurement.template localPosition<DIM>(), measurement.template localCovariance<DIM>() );
286 std::tuple<const measurement_container_variant_t *, abstract_measurement_range_t, bool >
290 if (std::get<0>(ret))
return ret;
296 template <
typename MeasurementRangeList_t>
297 static std::tuple<const measurement_container_variant_t *, abstract_measurement_range_t, bool >
299 typename MeasurementRangeList_t::const_iterator range_iter = measurementRanges.find(surface.geometryId().value());
300 if (range_iter == measurementRanges.end())
306 range_iter->second.elementEndIndex()};
307 assert( !range_iter->second.isMeasurementExpected() ||
range.begin() <=
range.end());
309 return { forced || range_iter->second.isMeasurementExpected() ? &(measurementRanges.container(range_iter->second.containerIndex())) :
nullptr,
310 std::move(
range), forced};
315 [[maybe_unused]]
const measurement_container_variant_t *container_variant_ptr,
317 return (abstract_range.begin()<=abstract_range.end());
320 template <
typename measurement_container_t>
324 unsigned int begin_idx = abstract_range.front();
325 auto begin_iter = concrete_container.container().begin() + begin_idx;
326 auto end_iter = begin_iter +
static_cast<unsigned int>(abstract_range.size());
327 return std::ranges::subrange(begin_iter, end_iter);
334 Acts::VectorMultiTrajectory>;
336 static constexpr std::size_t gAbsoluteMaxBranchesPerSurface = 3;
342 template <
typename track_container_t>
345 using TheAtlasMeasurementSelector
347 gAbsoluteMaxBranchesPerSurface,
348 typename track_container_t::TrackStateContainerBackend,
354 using BoundState = std::tuple<Acts::BoundTrackParameters, Acts::BoundMatrix, double>;
359 TheAtlasMeasurementSelector::Config &&
config)
360 : m_calibrator( std::move(calibrator)),
361 m_measurementSelector(std::move(
config),
367 if constexpr( s_fullPreCalibration) {
368 m_measurementSelector.template setPreCalibrator<2,xAOD::PixelCluster>(m_calibrator.pixelPreCalibrator());
369 m_measurementSelector.template setPreCalibrator<1,xAOD::StripCluster>(m_calibrator.stripPreCalibrator());
370 m_measurementSelector.template setPreCalibrator<3,xAOD::HGTDCluster>(m_calibrator.hgtdPreCalibrator());
371 m_measurementSelector.template setCalibrator<2,xAOD::PixelCluster>(m_calibrator.pixelPostCalibrator());
372 m_measurementSelector.template setCalibrator<1,xAOD::StripCluster>(m_calibrator.stripPostCalibrator());
373 m_measurementSelector.template setCalibrator<3,xAOD::HGTDCluster>(m_calibrator.hgtdPostCalibrator());
376 m_measurementSelector.template setCalibrator<2,xAOD::PixelCluster>(m_calibrator.pixelPostCalibrator());
377 m_measurementSelector.template setCalibrator<1,xAOD::StripCluster>(m_calibrator.stripPostCalibrator());
378 m_measurementSelector.template setCalibrator<3,xAOD::HGTDCluster>(m_calibrator.hgtdPostCalibrator());
383 void connect(std::any delegate_ptr)
const override {
384 using TrackStateCreator = Acts::CombinatorialKalmanFilterExtensions<RecoTrackContainer>::TrackStateCreator;
386 auto delegate = std::any_cast< TrackStateCreator *>(delegate_ptr);
387 delegate->template connect< & TheAtlasMeasurementSelector::createTrackStates >(&m_measurementSelector);
391 m_measurementSelector.setMeasurementRangesForced(measurementRangesForced);
398 TheAtlasMeasurementSelector m_measurementSelector;
406 const std::vector<float> &etaBinsf,
407 const std::vector<std::pair<float, float> > &chi2CutOffOutlier,
408 const std::vector<size_t> &numMeasurementsCutOff) {
414 using AtlMeasurementSelector = AtlasActsMeasurmentSelector<RecoTrackContainer>;
415 using AtlMeasurementSelectorConfig = AtlMeasurementSelector::TheAtlasMeasurementSelector::Config;
417 std::unique_ptr<ActsTrk::IMeasurementSelector>
418 selector(
new AtlMeasurementSelector(
419 std::move(atl_measurement_calibrator),
421 AtlMeasurementSelectorConfig{ {Acts::GeometryIdentifier(),
422 AtlMeasurementSelectorCuts{ etaBinsf,
424 numMeasurementsCutOff} }}));