MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t > Struct Template Reference

#include <MeasurementSelector.h>

Collaboration diagram for MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >:

Classes
struct	ProjectorBitSetMaker
struct	Flags
struct	PairWithFlags
struct	MatchingMeasurement

Public Types
using	Config = Acts::GeometryHierarchyMap<AtlasMeasurementSelectorCuts>
using	traits = MeasurementSelectorTraits<derived_t>

Public Member Functions
std::tuple< std::size_t, std::pair< float, float > >	getCuts (const Acts::Surface &surface, const T_BoundState &boundState, const Acts::Logger &logger) const

Public Attributes
Config	m_config

Protected Types
using	T_BoundState = typename MeasurementSelectorTraits<derived_t>::BoundState
using	trajectory_t = typename MeasurementSelectorTraits<derived_t>::trajectory_t
using	TrackStateProxy = typename MeasurementSelectorTraits<derived_t>::TrackStateProxy

Protected Member Functions
const derived_t &	derived () const
unsigned int	makeOutlierFlag (bool value)
template<std::size_t DIM, typename T_MeasurementRange>
Acts::Result< boost::container::small_vector< typename TrackStateProxy::IndexType, s_maxBranchesPerSurface > >	selectMeasurementsCreateTrackStates (const Acts::GeometryContext &geometryContext, const Acts::CalibrationContext &calibrationContext, const Acts::Surface &surface, const T_BoundState &boundState, T_MeasurementRange &&measurement_range, std::size_t prevTip, trajectory_t &trajectory, const Acts::Logger &logger, const std::size_t numMeasurementsCut, const std::pair< float, float > &maxChi2Cut, bool forced) const

Static Protected Member Functions
static void	createStates (std::size_t n_new_states, const T_BoundState &boundState, std::size_t prevTip, trajectory_t &trajectory, const Acts::BoundSubspaceIndices &subspaceIndices, boost::container::small_vector< typename TrackStateProxy::IndexType, s_maxBranchesPerSurface > &track_states, const Acts::Logger &logger, bool outlier_states)
template<typename parameters_t>
static std::size_t	getEtaBin (const parameters_t &boundParameters, const std::vector< float > &etaBins)

Static Protected Attributes
static constexpr std::size_t	s_maxBranchesPerSurface = MeasurementSelectorTraits<derived_t>::s_maxBranchesPerSurface

Detailed Description

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>
struct MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >

Definition at line 321 of file MeasurementSelector.h.

Member Typedef Documentation

Config

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>

using MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >::Config = Acts::GeometryHierarchyMap<AtlasMeasurementSelectorCuts>

Definition at line 323 of file MeasurementSelector.h.

T_BoundState

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>

using MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >::T_BoundState = typename MeasurementSelectorTraits<derived_t>::BoundState

protected

Definition at line 329 of file MeasurementSelector.h.

TrackStateProxy

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>

using MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >::TrackStateProxy = typename MeasurementSelectorTraits<derived_t>::TrackStateProxy

protected

Definition at line 331 of file MeasurementSelector.h.

traits

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>

using MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >::traits = MeasurementSelectorTraits<derived_t>

Definition at line 324 of file MeasurementSelector.h.

trajectory_t

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>

using MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >::trajectory_t = typename MeasurementSelectorTraits<derived_t>::trajectory_t

protected

Definition at line 330 of file MeasurementSelector.h.

Member Function Documentation

createStates()

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>

void MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >::createStates ( std::size_t n_new_states, const T_BoundState & boundState, std::size_t prevTip, trajectory_t & trajectory, const Acts::BoundSubspaceIndices & subspaceIndices, boost::container::small_vector< typename TrackStateProxy::IndexType, s_maxBranchesPerSurface > & track_states, const Acts::Logger & logger, bool outlier_states )

inlinestaticprotected

Definition at line 360 of file MeasurementSelector.h.

                                                 {
      // create track states for all compatible measurement candidates
      const auto &boundParams = derived_t::boundParams(boundState);
      const auto &pathLength = derived_t::pathLength(boundState);
 
      using PM = Acts::TrackStatePropMask;
 
      track_states.reserve( n_new_states );
      std::optional<TrackStateProxy> firstTrackState{
         std::nullopt};
      for (unsigned int state_i=0; state_i<n_new_states; ++state_i) {
         PM mask = PM::Predicted | PM::Filtered | PM::Jacobian | PM::Calibrated;
 
         if (firstTrackState.has_value()) {
            // subsequent track states don't need storage for these as they will
            // be shared
            mask &= ~PM::Predicted & ~PM::Jacobian;
         }
 
         if (outlier_states) {
            // outlier won't have separate filtered parameters
            mask &= ~PM::Filtered;
         }
 
         TrackStateProxy trackState = trajectory.makeTrackState(mask, prevTip);
         ACTS_VERBOSE("Create SourceLink output track state #"
                      << trackState.index() << " with mask: " << mask);
 
         if (firstTrackState.has_value()) {
            trackState.shareFrom(*firstTrackState, PM::Predicted);
            trackState.shareFrom(*firstTrackState, PM::Jacobian);
         }
         else {
            // only set these for first
            trackState.predicted() = boundParams.parameters();
            if (boundParams.covariance()) {
               trackState.predictedCovariance() = *boundParams.covariance();
            }
            trackState.jacobian() = derived_t::boundJacobiMatrix(boundState);
            firstTrackState = trackState;
         }
         trackState.pathLength() = pathLength;
 
         trackState.setReferenceSurface(boundParams.referenceSurface().getSharedPtr());
 
         trackState.setProjectorSubspaceIndices(subspaceIndices);
 
         auto typeFlags = trackState.typeFlags();
         if (trackState.referenceSurface().surfaceMaterial() != nullptr) {
            typeFlags.setHasMaterial();
         }
         typeFlags.setHasParameters();
 
         // @TODO these track states still need some additional processing. Should there be a special
         //       flag for this ?
         track_states.push_back( trackState.index());
      }
   }

derived()

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>

const derived_t & MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >::derived ( ) const

inlineprotected

Definition at line 334 of file MeasurementSelector.h.

{ return *static_cast<const derived_t *>(this); }

getCuts()

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>

std::tuple< std::size_t, std::pair< float, float > > MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >::getCuts ( const Acts::Surface & surface, const T_BoundState & boundState, const Acts::Logger & logger ) const

inline

Definition at line 677 of file MeasurementSelector.h.

                                                                                             {
      std::tuple< std::size_t, std::pair<float,float> > result;
      std::size_t &numMeasurementsCut = std::get<0>(result);
      std::pair<float,float> &maxChi2Cut = std::get<1>(result);
      // Get geoID of this surface
      auto geoID = surface.geometryId();
      // Find the appropriate cuts
      auto cuts = m_config.find(geoID);
      if (cuts == m_config.end()) {
         // indicates failure
         numMeasurementsCut = 0;
      }
      else {
         // num measurement Cut
         // getchi2 cut
         std::size_t eta_bin = getEtaBin(derived().boundParams(boundState), cuts->etaBins);
         numMeasurementsCut = (!cuts->numMeasurementsCutOff.empty()
                               ? cuts->numMeasurementsCutOff[ std::min(cuts->numMeasurementsCutOff.size()-1, eta_bin) ]
                               : NMeasMax);
         maxChi2Cut = ! cuts->chi2CutOff.empty()
            ? cuts->chi2CutOff[ std::min(cuts->chi2CutOff.size()-1, eta_bin) ]
            : std::make_pair<float,float>(std::numeric_limits<float>::max(),
                                          std::numeric_limits<float>::max());
         ACTS_VERBOSE("Get cut for eta-bin="
                      << (eta_bin < cuts->etaBins.size() ? cuts->etaBins[eta_bin] : std::numeric_limits<float>::max())
                      << ": chi2 (max,max-outlier) " << maxChi2Cut.first << ", " << maxChi2Cut.second
                      << " max.meas." << numMeasurementsCut);
      }
      return result;
   }

getEtaBin()

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>

template<typename parameters_t>

std::size_t MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >::getEtaBin ( const parameters_t & boundParameters, const std::vector< float > & etaBins )

inlinestaticprotected

Definition at line 658 of file MeasurementSelector.h.

                                                                 {
      if (etaBins.empty()) {
         return 0u;  // shortcut if no etaBins
      }
      const float eta = std::abs(std::atanh(std::cos(boundParameters.parameters()[Acts::eBoundTheta])));
      std::size_t bin = 0;
      for (auto etaBin : etaBins) {
         if (etaBin >= eta) {
            break;
         }
         bin++;
      }
      return bin;
   }

makeOutlierFlag()

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>

unsigned int MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >::makeOutlierFlag ( bool value )

inlineprotected

Definition at line 446 of file MeasurementSelector.h.

{ return static_cast<unsigned int>(value)<<2u; }

selectMeasurementsCreateTrackStates()

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>

template<std::size_t DIM, typename T_MeasurementRange>

Acts::Result< boost::container::small_vector< typename TrackStateProxy::IndexType, s_maxBranchesPerSurface > > MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >::selectMeasurementsCreateTrackStates ( const Acts::GeometryContext & geometryContext, const Acts::CalibrationContext & calibrationContext, const Acts::Surface & surface, const T_BoundState & boundState, T_MeasurementRange && measurement_range, std::size_t prevTip, trajectory_t & trajectory, const Acts::Logger & logger, const std::size_t numMeasurementsCut, const std::pair< float, float > & maxChi2Cut, bool forced ) const

inlineprotected

Definition at line 464 of file MeasurementSelector.h.

                                                          {
      Acts::Result<boost::container::small_vector< typename TrackStateProxy::IndexType, s_maxBranchesPerSurface> >
         result = boost::container::small_vector< typename TrackStateProxy::IndexType, s_maxBranchesPerSurface>{};
 
      using iterator_t = decltype(measurement_range.begin());
      using container_value_t  = typename MeasurementSelectorTraits<derived_t>::template MeasurementContainerTraits<iterator_t>::value_type;
      using BaseElementType = std::remove_cv_t<std::remove_pointer_t< container_value_t > >;
      // get calibrator
      using TheMatchingMeasurement = MatchingMeasurement<DIM, container_value_t >;
      using MeasCovPair = TheMatchingMeasurement::MeasCovPair;
 
      using Predicted = typename MeasurementSelectorTraits<derived_t>::template Predicted<DIM>;
      using PredictedCovariance = typename MeasurementSelectorTraits<derived_t>::template PredictedCovariance<DIM>;
 
 
 
      // get prediction in the measurement domain
      Acts::SubspaceIndices<DIM> parameter_map = derived().template parameterMap<DIM>(geometryContext,
                                                                                       calibrationContext,
                                                                                       surface);
      auto predicted
         = std::make_pair( project<DIM, Predicted>(parameter_map,
                                                   derived().boundParams(boundState).parameters()),
                           project<DIM, PredictedCovariance>(parameter_map,
                                                             derived().boundParams(boundState).covariance().value()));
 
      // select n measurents with the smallest chi2.
      auto preCalibrator = derived().template preCalibrator<DIM, BaseElementType>();
      auto postCalibrator = derived().template postCalibrator<DIM, BaseElementType>();
      TopCollection<NMeasMax, TheMatchingMeasurement > selected_measurements(numMeasurementsCut);
      for ( const auto &measurement : measurement_range ) {
         TheMatchingMeasurement &matching_measurement=selected_measurements.slot();
         if (forced && postCalibrator) {
            // skip preCalibrator and computeChi2, by doing everything else. We start with what's done if no preCalibrator.
            const auto &m = derived().forwardToCalibrator(measurement);
            matching_measurement.m_measurement = std::make_tuple( m.template localPosition<DIM>(), m.template localCovariance<DIM>(), 0u );
            matching_measurement.m_chi2 = 0.0f;
            matching_measurement.m_sourceLink=measurement;
            if (!selected_measurements.acceptNoSort()) break;
         } else {
            matching_measurement.m_measurement = preCalibrator(geometryContext,
                                                               calibrationContext,
                                                               surface,
                                                               derived().forwardToCalibrator(measurement),
                                                               derived().boundParams(boundState));
            matching_measurement.m_chi2 = computeChi2(matching_measurement.m_measurement.position(),
                                                      matching_measurement.m_measurement.covariance(),
                                                      predicted.first,
                                                      predicted.second);
            // only consider measurements which pass the outlier chi2 cut
            if (matching_measurement.m_chi2<maxChi2Cut.second) {
                matching_measurement.m_sourceLink=measurement;
                selected_measurements.acceptAndSort([](const TheMatchingMeasurement &a,
                                                       const TheMatchingMeasurement &b) {
                   return a.m_chi2 < b.m_chi2;
                });
            }
         }
      }
 
      if (selected_measurements.empty()) {
         auto config_for_surface = m_config.find(surface.geometryId());
         if (config_for_surface != m_config.end()) {
            if (config_for_surface->edgeTolerance.has_value()) {
               // if the  prediction failes the bound test, where the edgeTolerance is expected to be negative
               // then the "hole" is considered to be an edge hole which is not treated as a hole.
               auto local_coords  = derived().boundParams(boundState).parameters().template block<2,1>(0,0);
               if (!surface.insideBounds(local_coords,config_for_surface->edgeTolerance.value())) {
                  result = result.failure(Acts::CombinatorialKalmanFilterError::NoMeasurementExpected);
               }
            }
         }
      }
      else {
 
         // apply final calibration to n-best measurements
         using post_calib_meas_cov_pair_t
            = PairWithFlags<typename MeasurementSelectorTraits<derived_t>::template CalibratedMeasurement<DIM>,
                            typename MeasurementSelectorTraits<derived_t>::template CalibratedMeasurementCovariance<DIM> >;
 
         // First Create states without setting information about the calibrated measurement for the selected measurements
         // @TODO first create state then copy measurements, or crete state by state and set measurements ?
         //       the lastter has the "advantage" that the outlier flag can be set individually
         //       the former has the advantage that part of the state creation code is independent of the
         //       the measurement.
 
         Acts::BoundSubspaceIndices boundSubspaceIndices;
         std::copy(parameter_map.begin(), parameter_map.end(), boundSubspaceIndices.begin());
         createStates( selected_measurements.size(),
                       boundState,
                       prevTip,
                       trajectory,
                       boundSubspaceIndices,
                       *result,
                       logger,
                       (!selected_measurements.empty()
                        ? selected_measurements.getSlot( *(selected_measurements.begin())).m_measurement.flags().isOutlier() || postCalibrator
                        : false) );
         assert( result->size() == selected_measurements.size() );
 
         if (postCalibrator) {
            // either run post selection calibrator which will set the calibrated position and covariance
            // directly to the state.
            auto predicted_pos_dbl = predicted.first.template cast<double>();
            auto predicted_cov_dbl = predicted.second.template cast<double>();
            unsigned int state_i=0u;
            for (typename TopCollection<NMeasMax, MeasCovPair >::IndexType
                    idx: selected_measurements) {
               assert( state_i < result->size());
               TrackStateProxy trackState( trajectory.getTrackState( (*result)[state_i] ) );
               TheMatchingMeasurement &a_selected_measurement = selected_measurements.getSlot(idx);
               trackState.setUncalibratedSourceLink(ActsTrk::detail::MeasurementCalibratorBase::pack(std::move(a_selected_measurement.m_sourceLink.value())));
 
               // apply the calibration
               postCalibrator(geometryContext,
                              calibrationContext,
                              derived().forwardToCalibrator(a_selected_measurement.m_sourceLink.value()),
                              trackState
                              );
               // update chi2 using calibrated measurement
               trackState.chi2() = computeChi2(trackState.template calibrated<DIM>(),
                                               trackState.template calibratedCovariance<DIM>(),
                                               predicted_pos_dbl,
                                               predicted_cov_dbl);
               // ... and set outlier flag
               // and share or create storage for filtered.
               bool is_outlier = !forced && trackState.chi2() >= maxChi2Cut.first;
               if (is_outlier) {
                  trackState.typeFlags().setIsOutlier();
                  trackState.shareFrom(trackState, Acts::TrackStatePropMask::Predicted,
                                       Acts::TrackStatePropMask::Filtered);
               } else {
                  trackState.typeFlags().setIsMeasurement();
                  trackState.addComponents(Acts::TrackStatePropMask::Filtered);
               }
               ++state_i;
            }
         }
         else {
            // or copy the pre-selection calibrated position and covariance to the
            // track states and adjust the flags.
            unsigned int state_i=0u;
            for (typename TopCollection<NMeasMax, MeasCovPair >::IndexType
                    idx: selected_measurements) {
               assert( state_i < result->size());
               TrackStateProxy trackState( trajectory.getTrackState( (*result)[state_i] ) );
               TheMatchingMeasurement &a_selected_measurement = selected_measurements.getSlot(idx);
 
               bool is_outlier = !forced && a_selected_measurement.m_chi2 >= maxChi2Cut.first;
               auto type_flags = trackState.typeFlags();
               if (is_outlier) {
                  // the filtered parameters will not be computed for outliers, so has to be shared from the prediction.
                  type_flags.setIsOutlier();
                  trackState.shareFrom(trackState, Acts::TrackStatePropMask::Predicted,
                                       Acts::TrackStatePropMask::Filtered);
               } else {
                  // for measurements the filtered parameters will be computed, so space needs to be allocated.
                  type_flags.setIsMeasurement();
                  trackState.addComponents(Acts::TrackStatePropMask::Filtered);
               }
               if (a_selected_measurement.m_measurement.flags().isSplitHit()) {
                  type_flags.setIsSplitHit();
               }
 
               trackState.setUncalibratedSourceLink(ActsTrk::detail::MeasurementCalibratorBase
                                                        ::pack(std::move(a_selected_measurement.m_sourceLink.value())));
 
               trackState.allocateCalibrated(DIM);
               trackState.template calibrated<DIM>()
                  = MeasurementSelectorMatrixTraits::matrixTypeCast<typename MeasurementSelectorTraits<derived_t>
                                                                    ::MatrixFloatType>(a_selected_measurement.m_measurement.position());
               trackState.template calibratedCovariance<DIM>()
                  = MeasurementSelectorMatrixTraits::matrixTypeCast<typename MeasurementSelectorTraits<derived_t>
                                                                    ::MatrixFloatType>(a_selected_measurement.m_measurement.covariance());
               trackState.chi2() = a_selected_measurement.m_chi2;
               ++state_i;
            }
         }
      }
 
      return result;
   }

Member Data Documentation

m_config

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>

Config MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >::m_config

Definition at line 326 of file MeasurementSelector.h.

s_maxBranchesPerSurface

template<std::size_t NMeasMax, std::size_t DIMMAX, typename derived_t>

std::size_t MeasurementSelectorBase< NMeasMax, DIMMAX, derived_t >::s_maxBranchesPerSurface = MeasurementSelectorTraits<derived_t>::s_maxBranchesPerSurface

staticconstexprprotected

Definition at line 332 of file MeasurementSelector.h.

---

The documentation for this struct was generated from the following file:

• MeasurementSelector.h