#include <TrackParamsEstimationTool.h>

Inheritance diagram for ActsTrk::TrackParamsEstimationTool:

Collaboration diagram for ActsTrk::TrackParamsEstimationTool:

Public Member Functions
	TrackParamsEstimationTool (const std::string &type, const std::string &name, const IInterface *parent)

virtual	~TrackParamsEstimationTool ()=default

virtual StatusCode	initialize () override

virtual std::optional< Acts::BoundTrackParameters >	estimateTrackParameters (const ActsTrk::Seed &seed, bool useTopSp, const Acts::GeometryContext &geoContext, const Acts::MagneticFieldContext &magFieldContext, std::function< const Acts::Surface &(const ActsTrk::Seed &seed, bool useTopSp)> retrieveSurface) const override

virtual std::optional< Acts::BoundTrackParameters >	estimateTrackParameters (const ActsTrk::Seed &seed, bool useTopSp, const Acts::GeometryContext &geoContext, const Acts::MagneticFieldContext &magFieldContext, const Acts::Surface &surface, const Acts::Vector3 &bField) const override

Private Types
using	Stepper = Acts::SympyStepper

using	Navigator = Acts::VoidNavigator

using	Extrapolator = Acts::Propagator< Stepper >

Private Member Functions
const Acts::Logger &	logger () const
	Private access to the logger. More...

Private Attributes
Gaudi::Property< double >	m_sigmaLoc0

Gaudi::Property< double >	m_sigmaLoc1

Gaudi::Property< double >	m_sigmaPhi

Gaudi::Property< double >	m_sigmaTheta

Gaudi::Property< double >	m_sigmaQOverP

Gaudi::Property< double >	m_sigmaT0

Gaudi::Property< double >	m_initialSigmaPtRel

Gaudi::Property< std::vector< double > >	m_initialVarInflation

Gaudi::Property< bool >	m_useLongSeeds

std::optional< Extrapolator >	m_extrapolator

std::unique_ptr< const Acts::Logger >	m_logger
	logging instance More...

Detailed Description

Definition at line 19 of file TrackParamsEstimationTool.h.

Member Typedef Documentation

◆ Extrapolator

using ActsTrk::TrackParamsEstimationTool::Extrapolator = Acts::Propagator<Stepper>

private

Definition at line 73 of file TrackParamsEstimationTool.h.

◆ Navigator

using ActsTrk::TrackParamsEstimationTool::Navigator = Acts::VoidNavigator

private

Definition at line 72 of file TrackParamsEstimationTool.h.

◆ Stepper

using ActsTrk::TrackParamsEstimationTool::Stepper = Acts::SympyStepper

private

Definition at line 71 of file TrackParamsEstimationTool.h.

Constructor & Destructor Documentation

◆ TrackParamsEstimationTool()

ActsTrk::TrackParamsEstimationTool::TrackParamsEstimationTool	(	const std::string &	type,
		const std::string &	name,
		const IInterface *	parent
	)

Definition at line 11 of file TrackParamsEstimationTool.cxx.

14 : base_class(type, name, parent)

15 {}

◆ ~TrackParamsEstimationTool()

virtual ActsTrk::TrackParamsEstimationTool::~TrackParamsEstimationTool ( )

virtualdefault

Member Function Documentation

◆ estimateTrackParameters() [1/2]

std::optional< Acts::BoundTrackParameters > ActsTrk::TrackParamsEstimationTool::estimateTrackParameters	(	const ActsTrk::Seed &	seed,
		bool	useTopSp,
		const Acts::GeometryContext &	geoContext,
		const Acts::MagneticFieldContext &	magFieldContext,
		const Acts::Surface &	surface,
		const Acts::Vector3 &	bField
	)		const

overridevirtual

Definition at line 68 of file TrackParamsEstimationTool.cxx.

   {
     // Get SPs
     const auto& sp_collection = seed.sp();
     if ( sp_collection.size() < 3 ) return std::nullopt;
  
     // Compute free parameters
     Acts::FreeVector freeParams = useTopSp ?
       Acts::estimateTrackParamsFromSeed(sp_collection | std::views::reverse | std::views::take(3),
                                         bField) :
       Acts::estimateTrackParamsFromSeed(sp_collection | std::views::take(3),
                                         bField);
  
     if (m_useLongSeeds && sp_collection.size() > 3ul) {
       ActsTrk::Seed::container_type sp_collection2;
       if (useTopSp)
         sp_collection2.assign({sp_collection.back(), sp_collection.at(sp_collection.size()/2ul), sp_collection.front()});
       else
         sp_collection2.assign({sp_collection.front(), sp_collection.at(sp_collection.size()/2ul), sp_collection.back()});
       Acts::FreeVector freeParams2 = Acts::estimateTrackParamsFromSeed(sp_collection2, bField);
       ATH_MSG_DEBUG("update seed p = " << 1.0 / freeParams[Acts::eFreeQOverP] << " to " << 1.0 / freeParams2[Acts::eFreeQOverP]);
       freeParams[Acts::eFreeQOverP] = freeParams2[Acts::eFreeQOverP];
     }
  
     if (useTopSp) {
       // reverse direction so momentum vector pointing outwards
       freeParams = Acts::reflectFreeParameters(freeParams);
     }
  
     // Convert free params to curvilinear params for extrapolation
     Acts::BoundTrackParameters curvilinearParams = Acts::BoundTrackParameters::createCurvilinear(
       freeParams.segment<4>(Acts::eFreePos0),
       freeParams.segment<3>(Acts::eFreeDir0),
       freeParams[Acts::eFreeQOverP],
       std::nullopt,
       Acts::ParticleHypothesis::pion());
  
     // Extrapolate to surface
     Acts::PropagatorPlainOptions propOptions(geoContext, magFieldContext);
     propOptions.direction = Acts::Direction::fromScalarZeroAsPositive(
         surface.intersect(
           geoContext,
           freeParams.segment<3>(Acts::eFreePos0),
           freeParams.segment<3>(Acts::eFreeDir0)
         ).closest().pathLength());
     auto boundParamsResult = m_extrapolator->propagateToSurface(curvilinearParams, surface, propOptions);
     if (!boundParamsResult.ok()) {
       ATH_MSG_DEBUG("Extrapolation failed");
       return std::nullopt;
     }
  
     // Get extrapolated parameters
     Acts::BoundTrackParameters boundParams = *boundParamsResult;
  
     // Estimate covariance
     Acts::EstimateTrackParamCovarianceConfig covarianceEstimationConfig = {
       .initialSigmas = {m_sigmaLoc0, m_sigmaLoc1, m_sigmaPhi, m_sigmaTheta, m_sigmaQOverP, m_sigmaT0},
       .initialSigmaPtRel = m_initialSigmaPtRel,
       .initialVarInflation = Eigen::Map<const Acts::BoundVector>(m_initialVarInflation.value().data()),
       .noTimeVarInflation = 1.0,
     };
     boundParams.covariance() = Acts::estimateTrackParamCovariance(
       covarianceEstimationConfig,
       boundParams.parameters(),
       false);
  
     return boundParams;
   }

◆ estimateTrackParameters() [2/2]

std::optional< Acts::BoundTrackParameters > ActsTrk::TrackParamsEstimationTool::estimateTrackParameters	(	const ActsTrk::Seed &	seed,
		bool	useTopSp,
		const Acts::GeometryContext &	geoContext,
		const Acts::MagneticFieldContext &	magFieldContext,
		std::function< const Acts::Surface &(const ActsTrk::Seed &seed, bool useTopSp)>	retrieveSurface
	)		const

overridevirtual

Definition at line 38 of file TrackParamsEstimationTool.cxx.

   {
     const auto& sp_collection = seed.sp();
     if ( sp_collection.size() < 3 ) return std::nullopt;
     const auto& bottom_sp = useTopSp ? sp_collection.back() : sp_collection.front();
  
     // Magnetic Field
     ATLASMagneticFieldWrapper magneticField;
     Acts::MagneticFieldProvider::Cache magFieldCache = magneticField.makeCache( magFieldContext );
     Acts::Vector3 bField = *magneticField.getField( Acts::Vector3(bottom_sp->x(), bottom_sp->y(), bottom_sp->z()),
                                                     magFieldCache );
  
     // Get the surface
     const Acts::Surface& surface = retrieveSurface(seed, useTopSp);
  
     return estimateTrackParameters(
                    seed,
                    useTopSp,
                    geoContext,
                    magFieldContext,
                    surface,
                    bField);
   }

◆ initialize()

StatusCode ActsTrk::TrackParamsEstimationTool::initialize ( )

overridevirtual

Definition at line 17 of file TrackParamsEstimationTool.cxx.

   {
     ATH_MSG_INFO( "Initializing " << name() << "..." );
  
     ATH_MSG_DEBUG( "Properties Summary:" );
     ATH_MSG_DEBUG( "   " << m_sigmaLoc0 );
     ATH_MSG_DEBUG( "   " << m_sigmaLoc1 );
     ATH_MSG_DEBUG( "   " << m_sigmaPhi );
     ATH_MSG_DEBUG( "   " << m_sigmaTheta );
     ATH_MSG_DEBUG( "   " << m_sigmaQOverP );
     ATH_MSG_DEBUG( "   " << m_sigmaT0 );
     ATH_MSG_DEBUG( "   " << m_initialVarInflation );
  
     m_logger = makeActsAthenaLogger(this, "Acts");
  
     m_extrapolator = Extrapolator(Stepper(std::make_shared<ATLASMagneticFieldWrapper>()), Navigator(), logger().cloneWithSuffix("Prop"));
  
     return StatusCode::SUCCESS;
   }

◆ logger()

const Acts::Logger& ActsTrk::TrackParamsEstimationTool::logger ( ) const

inlineprivate

Private access to the logger.

Definition at line 78 of file TrackParamsEstimationTool.h.

     {
       return *m_logger;
     }

Member Data Documentation

◆ m_extrapolator

std::optional<Extrapolator> ActsTrk::TrackParamsEstimationTool::m_extrapolator

private

Definition at line 75 of file TrackParamsEstimationTool.h.

◆ m_initialSigmaPtRel

Gaudi::Property< double > ActsTrk::TrackParamsEstimationTool::m_initialSigmaPtRel

private

Initial value:

{this, "initialSigmaPtRel", 0.1,

"Initial relative pT resolution"}

Definition at line 64 of file TrackParamsEstimationTool.h.

◆ m_initialVarInflation

Gaudi::Property< std::vector<double> > ActsTrk::TrackParamsEstimationTool::m_initialVarInflation

private

Initial value:

{this, "initialVarInflation", {1., 1., 1., 1., 1., 1.},

"Inflate tracks"}

Definition at line 66 of file TrackParamsEstimationTool.h.

◆ m_logger

std::unique_ptr<const Acts::Logger> ActsTrk::TrackParamsEstimationTool::m_logger

private

logging instance

Definition at line 84 of file TrackParamsEstimationTool.h.

◆ m_sigmaLoc0

Gaudi::Property< double > ActsTrk::TrackParamsEstimationTool::m_sigmaLoc0

private

Initial value:

{this, "sigmaLoc0", 1 * Acts::UnitConstants::mm,

"Constant term of the loc0 resolution"}

Definition at line 52 of file TrackParamsEstimationTool.h.

◆ m_sigmaLoc1

Gaudi::Property< double > ActsTrk::TrackParamsEstimationTool::m_sigmaLoc1

private

Initial value:

{this, "sigmaLoc1", 1 * Acts::UnitConstants::mm,

"Constant term of the loc1 resolution"}

Definition at line 54 of file TrackParamsEstimationTool.h.

◆ m_sigmaPhi

Gaudi::Property< double > ActsTrk::TrackParamsEstimationTool::m_sigmaPhi

private

Initial value:

{this, "sigmaPhi", 0.1 * Acts::UnitConstants::degree,

"Phi angular resolution"}

Definition at line 56 of file TrackParamsEstimationTool.h.

◆ m_sigmaQOverP

Gaudi::Property< double > ActsTrk::TrackParamsEstimationTool::m_sigmaQOverP

private

Initial value:

{this, "sigmaQOverP", 0.1 * Acts::UnitConstants::e / Acts::UnitConstants::GeV,

"q/p resolution"}

Definition at line 60 of file TrackParamsEstimationTool.h.

◆ m_sigmaT0

Gaudi::Property< double > ActsTrk::TrackParamsEstimationTool::m_sigmaT0

private

Initial value:

{this, "sigmaT0", 1 * Acts::UnitConstants::ns,

"Time resolution"}

Definition at line 62 of file TrackParamsEstimationTool.h.

◆ m_sigmaTheta

Gaudi::Property< double > ActsTrk::TrackParamsEstimationTool::m_sigmaTheta

private

Initial value:

{this, "sigmaTheta", 0.1 * Acts::UnitConstants::degree,

"Theta angular resolution"}

Definition at line 58 of file TrackParamsEstimationTool.h.

◆ m_useLongSeeds

Gaudi::Property< bool > ActsTrk::TrackParamsEstimationTool::m_useLongSeeds

private

Initial value:

{this, "useLongSeeds", true,

"use seeds with 4 or more SPs for better pT measurement"}

Definition at line 68 of file TrackParamsEstimationTool.h.

The documentation for this class was generated from the following files:

Public Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ Extrapolator

◆ Navigator

◆ Stepper

Constructor & Destructor Documentation

◆ TrackParamsEstimationTool()

◆ ~TrackParamsEstimationTool()

Member Function Documentation

◆ estimateTrackParameters() [1/2]

◆ estimateTrackParameters() [2/2]

◆ initialize()

◆ logger()

Member Data Documentation

◆ m_extrapolator

◆ m_initialSigmaPtRel

◆ m_initialVarInflation

◆ m_logger

◆ m_sigmaLoc0

◆ m_sigmaLoc1

◆ m_sigmaPhi

◆ m_sigmaQOverP

◆ m_sigmaT0

◆ m_sigmaTheta

◆ m_useLongSeeds