#include <AdaptiveMultiPriVtxFinderTool.h>

Inheritance diagram for ActsTrk::AdaptiveMultiPriVtxFinderTool:

Collaboration diagram for ActsTrk::AdaptiveMultiPriVtxFinderTool:

Classes
class	TrackWrapper

Public Member Functions
virtual StatusCode	initialize () override

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

virtual std::pair< xAOD::VertexContainer , xAOD::VertexAuxContainer >	findVertex (const EventContext &ctx, const TrackCollection *trackTES) const override

virtual std::pair< xAOD::VertexContainer , xAOD::VertexAuxContainer >	findVertex (const EventContext &ctx, const xAOD::TrackParticleContainer *trackParticles) const override

Private Types
using	Propagator = Acts::Propagator< Acts::EigenStepper<>, Acts::Navigator >

using	TrackLinearizer = Acts::HelicalTrackLinearizer

using	VertexFitter = Acts::AdaptiveMultiVertexFitter

using	VertexSeedFinder = Acts::TrackDensityVertexFinder

using	VertexFinder = Acts::AdaptiveMultiVertexFinder

Private Member Functions
std::pair< xAOD::VertexContainer , xAOD::VertexAuxContainer >	findVertex (const EventContext &ctx, const std::vector< std::unique_ptr< Trk::ITrackLink >> &trackVector) const

Trk::Perigee *	actsBoundToTrkPerigee (const Acts::BoundTrackParameters &bound, const Acts::Vector3 &surfCenter) const

double	estimateSignalCompatibility (xAOD::Vertex *vtx) const

virtual const IActsTrackingGeometryTool *	trackingGeometryTool () const

const Acts::Logger &	logger () const

void	initializeVertexFinder (VertexFinder::Config &finderConfig)

Private Attributes
std::unique_ptr< const Acts::Logger >	m_logger {nullptr}
	logging instance More...

std::shared_ptr< VertexFinder >	m_vertexFinder = nullptr

std::shared_ptr< Propagator >	m_propagator = nullptr

std::optional< TrackLinearizer >	m_linearizer = std::nullopt

ToolHandle< IActsTrackingGeometryTool >	m_trackingGeometryTool {this, "TrackingGeometryTool", "", "ActsTrackingGeometryTool"}

ToolHandle< IActsExtrapolationTool >	m_extrapolationTool {this, "ExtrapolationTool", "", "ActsExtrapolationTool"}

ToolHandle< InDet::IInDetTrackSelectionTool >	m_trkFilter {this, "TrackSelector", "", "InDetTrackSelectionTool"}

SG::ReadCondHandleKey< InDet::BeamSpotData >	m_beamSpotKey {this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot"}

DoubleArrayProperty	m_annealingTemps {this, "annealingTemps", {1.0}, "Annealing temperatures"}

DoubleProperty	m_annealingCutOff {this, "annealingCutOff", 9., "Annealing cut-off value"}

UnsignedIntegerProperty	m_fitterMaxIterations {this, "fitterMaxIterations", 30, "Vertex fitter max. iterations"}

DoubleProperty	m_fitterMaxDistToLinPoint {this, "fitterMaxDistToLinPoint", 0.5, "Vertex fitter max. distance to LinPoint"}

DoubleProperty	m_fitterMaxRelativeShift {this, "fitterMaxRelativeShift", 0.01, "Vertex fitter max. relative shift"}

BooleanProperty	m_fitterDoSmoothing {this, "fitterDoSmoothing", true, "Vertex fitter doSmoothing"}

DoubleProperty	m_minWeightFitter {this, "minWeightFitter", 0.001, "Min track weight for fitter"}

BooleanProperty	m_useBeamConstraint {this, "useBeamConstraint", true, "Use beam constraint"}

DoubleProperty	m_tracksMaxZinterval {this, "tracksMaxZinterval", 1., "Tracks max. Z-interval"}

DoubleProperty	m_tracksMaxSignificance {this, "tracksMaxSignificance", 5., "Tracks max. significance"}

DoubleProperty	m_maxVertexChi2 {this, "maxVertexChi2", 18.42, "Max vertex chi2"}

BooleanProperty	m_doRealMultiVertex {this, "doRealMultiVertex", true, "Do real multivertex fit"}

BooleanProperty	m_useFastCompatibility {this, "useFastCompatibility", true, "Use fast compatibility estimation"}

DoubleProperty	m_maxMergeVertexSignificance {this, "maxMergeVertexSignificance", 3., "Max merge vertex significance"}

DoubleProperty	m_minWeight {this, "minWeight", 0.0001, "Min track weight for finder"}

UnsignedIntegerProperty	m_maxIterations {this, "maxIterations", 100, "Vertex finder max. iterations"}

BooleanProperty	m_addSingleTrackVertices {this, "addSingleTrackVertices", false, "Add single-track vertices"}

BooleanProperty	m_doFullSplitting {this, "doFullSplitting", false, "Do full-splitting"}

DoubleProperty	m_maximumVertexContamination {this, "maximumVertexContamination", 0.5, "Max. vertex contamination"}

DoubleProperty	m_looseConstrValue {this, "looseConstrValue", 1e+8, "Loose constraint value"}

BooleanProperty	m_useVertexCovForIPEstimation {this, "useVertexCovForIPEstimation", false, "Use seed vertex cov for IPEstimation"}

BooleanProperty	m_useSeedConstraint {this, "useSeedConstraint", false, "Use seed constraint in fit"}

DoubleProperty	m_finalCutMaxVertexChi2 {this, "finalCutMaxVertexChi2", 18.42, "Final cut max. vertex chi2"}

DoubleProperty	m_gaussianMaxD0Significance {this, "gaussianMaxD0Significance", 3.5, "Gaussian seeder max d0 track significance"}

DoubleProperty	m_gaussianMaxZ0Significance {this, "gaussianMaxDZSignificance", 12.0, "Gaussian seeder max z0 track significance"}

UnsignedIntegerProperty	m_gridMainGridSize {this, "GridMainGridSize", 20, "Main grid size for Acts::GaussianGridTrackDensity"}

UnsignedIntegerProperty	m_gridTrkGridSize {this, "GridTrkGridSize", 10, "Track grid size for Acts::GaussianGridTrackDensity"}

BooleanProperty	m_gridUseHighestSumZPosition {this, "GridUseHighestSumZPosition", false, "Use highest sum Z position in the grid seeder"}

DoubleProperty	m_gridMaxD0Significance {this, "gridMaxD0Significance", 3.5, "Grid seeder max d0 track significance"}

DoubleProperty	m_gridMaxZ0Significance {this, "gridMaxZ0Significance", 12.0, "Grid seeder max z0 track significance"}

UnsignedIntegerProperty	m_ipEstMaxIterations {this, "ipEstMaxIterations", 20, "IpEstimator max. iterations"}

DoubleProperty	m_ipEstPrecision {this, "ipEstPrecision", 1e-10, "IpEstimator precision"}

StringProperty	m_seederType {this, "seederType", "Gaussian", "type of seeder, \"Grid\" for Grid seeder, other for Gaussian seeder (default)"}

Detailed Description

Definition at line 54 of file AdaptiveMultiPriVtxFinderTool.h.

Member Typedef Documentation

◆ Propagator

using ActsTrk::AdaptiveMultiPriVtxFinderTool::Propagator = Acts::Propagator<Acts::EigenStepper<>, Acts::Navigator>

private

Definition at line 118 of file AdaptiveMultiPriVtxFinderTool.h.

◆ TrackLinearizer

using ActsTrk::AdaptiveMultiPriVtxFinderTool::TrackLinearizer = Acts::HelicalTrackLinearizer

private

Definition at line 119 of file AdaptiveMultiPriVtxFinderTool.h.

◆ VertexFinder

using ActsTrk::AdaptiveMultiPriVtxFinderTool::VertexFinder = Acts::AdaptiveMultiVertexFinder

private

Definition at line 122 of file AdaptiveMultiPriVtxFinderTool.h.

◆ VertexFitter

using ActsTrk::AdaptiveMultiPriVtxFinderTool::VertexFitter = Acts::AdaptiveMultiVertexFitter

private

Definition at line 120 of file AdaptiveMultiPriVtxFinderTool.h.

◆ VertexSeedFinder

using ActsTrk::AdaptiveMultiPriVtxFinderTool::VertexSeedFinder = Acts::TrackDensityVertexFinder

private

Definition at line 121 of file AdaptiveMultiPriVtxFinderTool.h.

Constructor & Destructor Documentation

◆ AdaptiveMultiPriVtxFinderTool()

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

Definition at line 48 of file AdaptiveMultiPriVtxFinderTool.cxx.

51 : base_class(type, name, parent)

52 {}

Member Function Documentation

◆ actsBoundToTrkPerigee()

Trk::Perigee * ActsTrk::AdaptiveMultiPriVtxFinderTool::actsBoundToTrkPerigee	(	const Acts::BoundTrackParameters &	bound,
		const Acts::Vector3 &	surfCenter
	)		const

private

Definition at line 449 of file AdaptiveMultiPriVtxFinderTool.cxx.

 {
   using namespace Acts::UnitLiterals;
   AmgSymMatrix(5) cov =  AmgSymMatrix(5)(bound.covariance()->block<5,5>(0,0));
   cov.col(Trk::qOverP) *= 1_MeV;
   cov.row(Trk::qOverP) *= 1_MeV;
   Acts::ActsVector<5> params = bound.parameters().head<5>();
   params[Trk::qOverP] *= 1_MeV;
  
   return new Trk::Perigee(params, Trk::PerigeeSurface(surfCenter), std::move(cov));
 }

◆ estimateSignalCompatibility()

double ActsTrk::AdaptiveMultiPriVtxFinderTool::estimateSignalCompatibility ( xAOD::Vertex * vtx ) const

private

Definition at line 463 of file AdaptiveMultiPriVtxFinderTool.cxx.

 {
   double totalPt2 = 0;
   unsigned int nTracks = 0;
  
   for(const auto& trk : vtx->vxTrackAtVertex()){
     if ((trk.vtxCompatibility() < m_finalCutMaxVertexChi2 && m_useFastCompatibility) ||
       (trk.weight() > m_minWeight
          && trk.trackQuality().chiSquared() < m_finalCutMaxVertexChi2
          && !m_useFastCompatibility)) {
       const Trk::TrackParameters* perigee = nullptr;
       if (trk.perigeeAtVertex() != nullptr) {
           perigee = trk.perigeeAtVertex();
       } else {
           ATH_MSG_VERBOSE("Only initialPerigee is available");
           perigee = trk.initialPerigee();
       }
       if (perigee == nullptr) {
           ATH_MSG_ERROR("Neutrals are not supported. Skipping track in pT calculation...");
           continue;
       }
       totalPt2 +=
       std::pow(std::fabs(1. / perigee->parameters()[Trk::qOverP]) * std::sin(perigee->parameters()[Trk::theta]), 2);
       nTracks += 1;
     }
   }
   return totalPt2 * std::sqrt((double) nTracks);
 }

◆ findVertex() [1/3]

std::pair< xAOD::VertexContainer , xAOD::VertexAuxContainer > ActsTrk::AdaptiveMultiPriVtxFinderTool::findVertex	(	const EventContext &	ctx,
		const std::vector< std::unique_ptr< Trk::ITrackLink >> &	trackVector
	)		const

private

Definition at line 267 of file AdaptiveMultiPriVtxFinderTool.cxx.

 {
     using namespace Acts::UnitLiterals; // !!!
     
     SG::ReadCondHandle<InDet::BeamSpotData> beamSpotHandle { m_beamSpotKey, ctx};
     const Acts::Vector3& beamSpotPos = beamSpotHandle->beamVtx().position();
     Acts::Vertex beamSpotConstraintVtx(beamSpotPos);
     beamSpotConstraintVtx.setCovariance(beamSpotHandle->beamVtx().covariancePosition());
  
     // Get the magnetic field context
     Acts::MagneticFieldContext magFieldContext = m_extrapolationTool->getMagneticFieldContext(ctx);
  
     const auto& geoContext = m_trackingGeometryTool->getGeometryContext(ctx).context();
  
     // The output vertex containers
     xAOD::VertexContainer* theVertexContainer = new xAOD::VertexContainer;
     xAOD::VertexAuxContainer* theVertexAuxContainer = new xAOD::VertexAuxContainer;
     theVertexContainer->setStore(theVertexAuxContainer);
  
     if(trackVector.empty()){
       xAOD::Vertex* dummyxAODVertex = new xAOD::Vertex;
       theVertexContainer->push_back(dummyxAODVertex);
       dummyxAODVertex->setPosition(beamSpotHandle->beamVtx().position());
       dummyxAODVertex->setCovariancePosition(beamSpotHandle->beamVtx().covariancePosition());
       dummyxAODVertex->setVertexType(xAOD::VxType::NoVtx);
  
       return std::make_pair(theVertexContainer, theVertexAuxContainer);
     }
  
     std::shared_ptr<Acts::PerigeeSurface> perigeeSurface =
       Acts::Surface::makeShared<Acts::PerigeeSurface>((trackVector[0])->parameters()->associatedSurface().transform());
  
     // Convert tracks to Acts::BoundParameters
     std::vector<TrackWrapper> allTracks;
  
     for (const auto& trk : trackVector) {
       const auto& trkParams = trk->parameters();
       const auto& params = trkParams->parameters();
  
       Acts::BoundVector actsParams;
       actsParams << params(0), params(1), params(2), params(3), params(4)*1./(1_MeV), 0.;
  
       if(trkParams->covariance() == nullptr){
           continue;
       }
  
       auto cov = *(trkParams->covariance());
       
       // TODO: check if the following works as well:
       // cov->col(4) *= 1./1_MeV;
       // cov->row(4) *= 1./1_MeV;
       Acts::BoundSquareMatrix covMat;
       covMat << cov(0,0) , cov(0,1) , cov(0,2) , cov(0,3) , cov(0,4) *1./(1_MeV), 0      
       , cov(1,0) , cov(1,1) , cov(1,2) , cov(1,3) , cov(1,4) *1./(1_MeV) , 0
       , cov(2,0) , cov(2,1) , cov(2,2) , cov(2,3) , cov(2,4) *1./(1_MeV) , 0
       , cov(3,0) , cov(3,1) , cov(3,2) , cov(3,3) , cov(3,4) *1./(1_MeV) , 0 
       , cov(4,0) *1./(1_MeV) , cov(4,1) *1./(1_MeV) , cov(4,2) *1./(1_MeV) , cov(4,3) *1./(1_MeV) , cov(4,4) *1./(1_MeV*1_MeV), 0
       , 0. , 0. , 0. , 0., 0., 1.;
       
       allTracks.emplace_back(trk.get(),Acts::BoundTrackParameters(perigeeSurface, actsParams, covMat, Acts::ParticleHypothesis::pion()));
     }
  
     std::vector<Acts::InputTrack> allTrackPtrs;
     allTrackPtrs.reserve(allTracks.size());
     
     for(const auto& trk : allTracks){
       allTrackPtrs.emplace_back(&trk);
     }
  
     Acts::VertexingOptions vertexingOptions( geoContext, magFieldContext );
  
     if(!m_useBeamConstraint){
       beamSpotConstraintVtx.setPosition(Acts::Vector3::Zero());
       Amg::MatrixX looseConstraintCovariance(3, 3);
       looseConstraintCovariance.setIdentity();
       looseConstraintCovariance = looseConstraintCovariance * 1e+8;
       beamSpotConstraintVtx.setCovariance(looseConstraintCovariance);
     }
  
     vertexingOptions.useConstraintInFit = m_useBeamConstraint;
     vertexingOptions.constraint = beamSpotConstraintVtx;
  
     auto finderState = m_vertexFinder->makeState(magFieldContext);
  
     auto findResult = m_vertexFinder->find(allTrackPtrs, vertexingOptions, finderState);
     
     if(!findResult.ok()){
       xAOD::Vertex* dummyxAODVertex = new xAOD::Vertex;
       theVertexContainer->push_back(dummyxAODVertex);
       dummyxAODVertex->setPosition(beamSpotHandle->beamVtx().position());
       dummyxAODVertex->setCovariancePosition(beamSpotHandle->beamVtx().covariancePosition());
       dummyxAODVertex->setVertexType(xAOD::VxType::NoVtx);
       return std::make_pair(theVertexContainer, theVertexAuxContainer);
     }
  
     std::vector<Acts::Vertex> allVertices = *findResult;
  
     std::vector<VertexAndSignalComp> vtxList;
     
     // Reserve memory for all vertices
     vtxList.reserve(allVertices.size());
  
     for(const auto& vtx : allVertices){
       //skip the vertex with negative covariance
       if(vtx.covariance()(0,0)<0||vtx.covariance()(1,1)<0||vtx.covariance()(2,2)<0)
     continue;
       xAOD::Vertex* xAODVtx = new xAOD::Vertex;
       xAODVtx->makePrivateStore();
       xAODVtx->setPosition(vtx.position());
       xAODVtx->setCovariancePosition(vtx.covariance());
       xAODVtx->setFitQuality(vtx.fitQuality().first, vtx.fitQuality().second);
  
       const auto& tracks = vtx.tracks();
       std::vector<Trk::VxTrackAtVertex>* trkAtVtxVec = &(xAODVtx->vxTrackAtVertex());
       for(const auto& trk : tracks){
  
         // Clean up incompatible tracks
         if ((trk.vertexCompatibility > m_maxVertexChi2 && m_useFastCompatibility) ||
           ((trk.trackWeight < m_minWeight
             || trk.chi2Track > m_maxVertexChi2)
           && !m_useFastCompatibility)) {
           continue;
         }
  
         const TrackWrapper* originalParams = trk.originalParams.template as<TrackWrapper>();
  
         Trk::Perigee* fittedPerigee = actsBoundToTrkPerigee(trk.fittedParams, beamSpotPos);
         Trk::VxTrackAtVertex trkAtVtx(originalParams->trackLink()->clone());
         trkAtVtx.setPerigeeAtVertex(fittedPerigee);
         trkAtVtx.setTrackQuality(Trk::FitQuality(trk.chi2Track, trk.ndf));
         trkAtVtx.setVtxCompatibility(trk.vertexCompatibility);
         trkAtVtx.setWeight(trk.trackWeight);
         trkAtVtxVec->push_back(trkAtVtx);
  
         const Trk::LinkToXAODTrackParticle* linkToXAODTP =
         dynamic_cast<const Trk::LinkToXAODTrackParticle*>(originalParams->trackLink());
         if (linkToXAODTP) {
           xAODVtx->addTrackAtVertex(*linkToXAODTP, trk.trackWeight);
         }
       }
       // Find signal compatibility
       double sigComp = estimateSignalCompatibility(xAODVtx);
             // Insert vertex at correct position according to sigComp value
       VertexAndSignalComp vertexAndSig(xAODVtx, sigComp);
       auto it = std::lower_bound( vtxList.begin(), vtxList.end(), vertexAndSig );
       vtxList.insert( it, vertexAndSig );
     }
   
     for(unsigned int i = 0; i < vtxList.size(); i++){
       auto vtx = vtxList[i].first;
       theVertexContainer->push_back(vtx);
       if(i == 0){
     vtx->setVertexType(xAOD::VxType::PriVtx);
       }
       else{
     vtx->setVertexType(xAOD::VxType::PileUp);
       }
     }
  
     // Add dummy vertex to collection
     xAOD::Vertex* dummyxAODVertex = new xAOD::Vertex;
     theVertexContainer->push_back(dummyxAODVertex);
  
     if(!vtxList.empty()){
       // If HS vertex exists, create dummy with same position/covariance
       xAOD::Vertex* primaryVtx = theVertexContainer->front();
       dummyxAODVertex->setPosition(primaryVtx->position());
       dummyxAODVertex->setCovariancePosition(primaryVtx->covariancePosition());
       dummyxAODVertex->setVertexType(xAOD::VxType::NoVtx);
     }
     else{
       dummyxAODVertex->setPosition(beamSpotHandle->beamVtx().position());
       dummyxAODVertex->setCovariancePosition(beamSpotHandle->beamVtx().covariancePosition());
       dummyxAODVertex->setVertexType(xAOD::VxType::NoVtx);
     }
  
     return std::make_pair(theVertexContainer, theVertexAuxContainer);
 }

◆ findVertex() [2/3]

std::pair< xAOD::VertexContainer , xAOD::VertexAuxContainer > ActsTrk::AdaptiveMultiPriVtxFinderTool::findVertex	(	const EventContext &	ctx,
		const TrackCollection *	trackTES
	)		const

overridevirtual

Definition at line 191 of file AdaptiveMultiPriVtxFinderTool.cxx.

 {
   SG::ReadCondHandle<InDet::BeamSpotData> beamSpotHandle { m_beamSpotKey, ctx};
   const Trk::RecVertex& beamposition(beamSpotHandle->beamVtx());
  
   std::vector<std::unique_ptr<Trk::ITrackLink>> selectedTracks;
  
   typedef DataVector<Trk::Track>::const_iterator TrackDataVecIter;
  
   bool selectionPassed;
   for (TrackDataVecIter itr = (*trackTES).begin(); itr != (*trackTES).end(); ++itr) {
     if (m_useBeamConstraint) {
       selectionPassed = static_cast<bool>(m_trkFilter->accept(**itr, &beamposition));
     } else {
         Trk::Vertex null(Amg::Vector3D(0, 0, 0));
         selectionPassed = static_cast<bool>(m_trkFilter->accept(**itr, &null));
     }
     if (selectionPassed) {
         ElementLink<TrackCollection> link;
         link.setElement(*itr);
         auto trkPtr = std::make_unique<Trk::LinkToTrack>(link);
         trkPtr->setStorableObject(*trackTES);
         selectedTracks.push_back(std::move(trkPtr));
     }
   }
  
   std::pair<xAOD::VertexContainer*, xAOD::VertexAuxContainer*> returnContainers = findVertex(ctx, std::move(selectedTracks));
  
   return returnContainers;
 }

◆ findVertex() [3/3]

std::pair< xAOD::VertexContainer , xAOD::VertexAuxContainer > ActsTrk::AdaptiveMultiPriVtxFinderTool::findVertex	(	const EventContext &	ctx,
		const xAOD::TrackParticleContainer *	trackParticles
	)		const

overridevirtual

Definition at line 224 of file AdaptiveMultiPriVtxFinderTool.cxx.

 {
  
   std::vector<std::unique_ptr<Trk::ITrackLink>> selectedTracks;
   SG::ReadCondHandle<InDet::BeamSpotData> beamSpotHandle { m_beamSpotKey, ctx};
   xAOD::Vertex beamposition;
   beamposition.makePrivateStore();
   beamposition.setPosition(beamSpotHandle->beamVtx().position());
   beamposition.setCovariancePosition(beamSpotHandle->beamVtx().covariancePosition());
  
   typedef DataVector<xAOD::TrackParticle>::const_iterator TrackParticleDataVecIter;
  
   bool selectionPassed;
   for (TrackParticleDataVecIter itr = (*trackParticles).begin(); itr != (*trackParticles).end(); ++itr) {
     if (m_useBeamConstraint) {
       selectionPassed = static_cast<bool>(m_trkFilter->accept(**itr, &beamposition));
     } else {
         xAOD::Vertex null;
         null.makePrivateStore();
         null.setPosition(Amg::Vector3D(0, 0, 0));
         AmgSymMatrix(3) vertexError;
         vertexError.setZero();
         null.setCovariancePosition(vertexError);
         selectionPassed = static_cast<bool>(m_trkFilter->accept(**itr, &null));
     }
  
     if (selectionPassed) {
         ElementLink<xAOD::TrackParticleContainer> link;
         link.setElement(*itr);
         auto trkPtr = std::make_unique<Trk::LinkToXAODTrackParticle>(link);
         trkPtr->setStorableObject(*trackParticles);
         selectedTracks.push_back(std::move(trkPtr));
     }
   }
  
   std::pair<xAOD::VertexContainer*, xAOD::VertexAuxContainer*> returnContainers = findVertex(ctx, std::move(selectedTracks));
  
   return returnContainers;
 }

◆ initialize()

StatusCode ActsTrk::AdaptiveMultiPriVtxFinderTool::initialize ( )

overridevirtual

Definition at line 55 of file AdaptiveMultiPriVtxFinderTool.cxx.

 {
     using namespace std::literals::string_literals;
  
     ATH_CHECK(m_beamSpotKey.initialize());
     ATH_CHECK(m_trkFilter.retrieve());
  
     ATH_MSG_INFO("Initializing ACTS AMVF tool");
     ATH_CHECK( m_trackingGeometryTool.retrieve() );
     std::shared_ptr<const Acts::TrackingGeometry> trackingGeometry
     = m_trackingGeometryTool->trackingGeometry();
  
     ATH_CHECK( m_extrapolationTool.retrieve() );
  
     // Logger
     m_logger = makeActsAthenaLogger(this, "Acts");
     
     Acts::Navigator navigator( Acts::Navigator::Config{ trackingGeometry },
                    logger().cloneWithSuffix("Navigator"));
  
     auto bField = std::make_shared<ATLASMagneticFieldWrapper>();
     auto stepper = Acts::EigenStepper<>(bField);
     m_propagator = std::make_shared<Propagator>(std::move(stepper), 
                         std::move(navigator),
                         logger().cloneWithSuffix("Prop"));
  
     // IP Estimator
     Acts::ImpactPointEstimator::Config ipEstCfg(bField, m_propagator);
     ipEstCfg.maxIterations = m_ipEstMaxIterations;
     ipEstCfg.precision = m_ipEstPrecision;
     Acts::ImpactPointEstimator ipEst(ipEstCfg,
                      logger().cloneWithSuffix("ImpactPointEstimator"));
  
     Acts::AnnealingUtility::Config annealingConfig;
     annealingConfig.setOfTemperatures = m_annealingTemps;
     annealingConfig.cutOff = m_annealingCutOff;
     Acts::AnnealingUtility annealingUtility(annealingConfig);
  
     // Linearizer for Acts::BoundParameters type test
     TrackLinearizer::Config ltConfig;
     ltConfig.bField = bField;
     ltConfig.propagator = m_propagator;
     m_linearizer.emplace(ltConfig, logger().cloneWithSuffix("Linearizer"));
  
     // Vertex fitter configuration
     VertexFitter::Config fitterCfg(ipEst);
     fitterCfg.annealingTool = annealingUtility;
     fitterCfg.maxIterations = m_fitterMaxIterations;
     fitterCfg.maxDistToLinPoint = m_fitterMaxDistToLinPoint;
     fitterCfg.minWeight = m_minWeightFitter;
     fitterCfg.maxRelativeShift = m_fitterMaxRelativeShift;
     fitterCfg.doSmoothing = m_fitterDoSmoothing;
     fitterCfg.extractParameters.connect<&TrackWrapper::extractParameters>();
     fitterCfg.trackLinearizer.connect<&TrackLinearizer::linearizeTrack>(&*m_linearizer);
     VertexFitter fitter(fitterCfg, logger().cloneWithSuffix("Fitter"));
  
   std::string seederType = m_seederType;
  
     if (seederType == "Grid") {
       Acts::GaussianGridTrackDensity::Config trackDensityConfig;
       trackDensityConfig.mainGridSize           = m_gridMainGridSize;
       trackDensityConfig.trkGridSize            = m_gridTrkGridSize;
       trackDensityConfig.useHighestSumZPosition = m_gridUseHighestSumZPosition;
       Acts::GaussianGridTrackDensity trackDensity(trackDensityConfig);
  
       Acts::GridDensityVertexFinder::Config gridSeedFinderConfig(trackDensity);
       gridSeedFinderConfig.extractParameters.connect<&TrackWrapper::extractParameters>();
       gridSeedFinderConfig.maxD0TrackSignificance = m_gridMaxD0Significance;
       gridSeedFinderConfig.maxZ0TrackSignificance = m_gridMaxZ0Significance;
  
       Acts::GridDensityVertexFinder gridSeedFinder(gridSeedFinderConfig);
  
       VertexFinder::Config finderConfig(
           std::move(fitter),
           std::make_shared<Acts::GridDensityVertexFinder>(gridSeedFinder),
           ipEst,
           bField
       );
  
       initializeVertexFinder(finderConfig);
     }
     else if (seederType == "Gaussian") {
       Acts::GaussianTrackDensity::Config trackDensityConfig;
       trackDensityConfig.d0MaxSignificance = m_gaussianMaxD0Significance;
       trackDensityConfig.z0MaxSignificance = m_gaussianMaxZ0Significance;
       trackDensityConfig.extractParameters.connect<&TrackWrapper::extractParameters>();
  
       Acts::GaussianTrackDensity trackDensity(trackDensityConfig);
       VertexSeedFinder::Config seedFinderConfig(trackDensity);
       auto seedFinder = std::make_shared<VertexSeedFinder>(seedFinderConfig);
  
       VertexFinder::Config finderConfig(
           std::move(fitter),
           seedFinder,
           ipEst,
           bField
       );
  
       initializeVertexFinder(finderConfig);
     }
     else {
       ATH_MSG_ERROR("Unknown seederType '" << seederType << "'.");
       return StatusCode::FAILURE;
     }
  
     ATH_MSG_INFO("ACTS AMVF tool successfully initialized");
     return StatusCode::SUCCESS;
   }

◆ initializeVertexFinder()

void ActsTrk::AdaptiveMultiPriVtxFinderTool::initializeVertexFinder ( VertexFinder::Config & finderConfig )

private

Definition at line 165 of file AdaptiveMultiPriVtxFinderTool.cxx.

   {
  
   finderConfig.tracksMaxZinterval          = m_tracksMaxZinterval;
   finderConfig.tracksMaxSignificance       = m_tracksMaxSignificance;
   finderConfig.maxVertexChi2               = m_maxVertexChi2;
   finderConfig.doRealMultiVertex           = m_doRealMultiVertex;
   finderConfig.useFastCompatibility        = m_useFastCompatibility;
   finderConfig.maxMergeVertexSignificance  = m_maxMergeVertexSignificance;
   finderConfig.minWeight                   = m_minWeight;
   finderConfig.maxIterations               = m_maxIterations;
   finderConfig.addSingleTrackVertices      = m_addSingleTrackVertices;
   finderConfig.doFullSplitting             = m_doFullSplitting;
   finderConfig.maximumVertexContamination  = m_maximumVertexContamination;
   finderConfig.initialVariances            = Acts::Vector4::Constant(m_looseConstrValue);
   finderConfig.useVertexCovForIPEstimation = m_useVertexCovForIPEstimation;
   finderConfig.useSeedConstraint           = m_useSeedConstraint;
  
   finderConfig.extractParameters.connect<&TrackWrapper::extractParameters>();
  
   m_vertexFinder = std::make_shared<VertexFinder>(
       std::move(finderConfig),
       logger().cloneWithSuffix("Finder"));
 }

◆ logger()

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

inlineprivate

Definition at line 116 of file AdaptiveMultiPriVtxFinderTool.h.

116 { return *m_logger; }

◆ trackingGeometryTool()

virtual const IActsTrackingGeometryTool* ActsTrk::AdaptiveMultiPriVtxFinderTool::trackingGeometryTool ( ) const

inlineprivatevirtual

Definition at line 109 of file AdaptiveMultiPriVtxFinderTool.h.

     {
       return m_trackingGeometryTool.get();
     }

Member Data Documentation

◆ m_addSingleTrackVertices

BooleanProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_addSingleTrackVertices {this, "addSingleTrackVertices", false, "Add single-track vertices"}

private

Definition at line 161 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_annealingCutOff

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_annealingCutOff {this, "annealingCutOff", 9., "Annealing cut-off value"}

private

Definition at line 143 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_annealingTemps

DoubleArrayProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_annealingTemps {this, "annealingTemps", {1.0}, "Annealing temperatures"}

private

Definition at line 142 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_beamSpotKey

SG::ReadCondHandleKey<InDet::BeamSpotData> ActsTrk::AdaptiveMultiPriVtxFinderTool::m_beamSpotKey {this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot"}

private

Definition at line 135 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_doFullSplitting

BooleanProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_doFullSplitting {this, "doFullSplitting", false, "Do full-splitting"}

private

Definition at line 162 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_doRealMultiVertex

BooleanProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_doRealMultiVertex {this, "doRealMultiVertex", true, "Do real multivertex fit"}

private

Definition at line 156 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_extrapolationTool

ToolHandle<IActsExtrapolationTool> ActsTrk::AdaptiveMultiPriVtxFinderTool::m_extrapolationTool {this, "ExtrapolationTool", "", "ActsExtrapolationTool"}

private

Definition at line 133 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_finalCutMaxVertexChi2

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_finalCutMaxVertexChi2 {this, "finalCutMaxVertexChi2", 18.42, "Final cut max. vertex chi2"}

private

Definition at line 168 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_fitterDoSmoothing

BooleanProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_fitterDoSmoothing {this, "fitterDoSmoothing", true, "Vertex fitter doSmoothing"}

private

Definition at line 148 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_fitterMaxDistToLinPoint

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_fitterMaxDistToLinPoint {this, "fitterMaxDistToLinPoint", 0.5, "Vertex fitter max. distance to LinPoint"}

private

Definition at line 146 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_fitterMaxIterations

UnsignedIntegerProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_fitterMaxIterations {this, "fitterMaxIterations", 30, "Vertex fitter max. iterations"}

private

Definition at line 145 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_fitterMaxRelativeShift

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_fitterMaxRelativeShift {this, "fitterMaxRelativeShift", 0.01, "Vertex fitter max. relative shift"}

private

Definition at line 147 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_gaussianMaxD0Significance

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_gaussianMaxD0Significance {this, "gaussianMaxD0Significance", 3.5, "Gaussian seeder max d0 track significance"}

private

Definition at line 170 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_gaussianMaxZ0Significance

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_gaussianMaxZ0Significance {this, "gaussianMaxDZSignificance", 12.0, "Gaussian seeder max z0 track significance"}

private

Definition at line 171 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_gridMainGridSize

UnsignedIntegerProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_gridMainGridSize {this, "GridMainGridSize", 20, "Main grid size for Acts::GaussianGridTrackDensity"}

private

Definition at line 173 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_gridMaxD0Significance

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_gridMaxD0Significance {this, "gridMaxD0Significance", 3.5, "Grid seeder max d0 track significance"}

private

Definition at line 177 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_gridMaxZ0Significance

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_gridMaxZ0Significance {this, "gridMaxZ0Significance", 12.0, "Grid seeder max z0 track significance"}

private

Definition at line 178 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_gridTrkGridSize

UnsignedIntegerProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_gridTrkGridSize {this, "GridTrkGridSize", 10, "Track grid size for Acts::GaussianGridTrackDensity"}

private

Definition at line 174 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_gridUseHighestSumZPosition

BooleanProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_gridUseHighestSumZPosition {this, "GridUseHighestSumZPosition", false, "Use highest sum Z position in the grid seeder"}

private

Definition at line 175 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_ipEstMaxIterations

UnsignedIntegerProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_ipEstMaxIterations {this, "ipEstMaxIterations", 20, "IpEstimator max. iterations"}

private

Definition at line 181 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_ipEstPrecision

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_ipEstPrecision {this, "ipEstPrecision", 1e-10, "IpEstimator precision"}

private

Definition at line 182 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_linearizer

std::optional<TrackLinearizer> ActsTrk::AdaptiveMultiPriVtxFinderTool::m_linearizer = std::nullopt

private

Definition at line 130 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_logger

std::unique_ptr<const Acts::Logger> ActsTrk::AdaptiveMultiPriVtxFinderTool::m_logger {nullptr}

private

logging instance

Definition at line 115 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_looseConstrValue

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_looseConstrValue {this, "looseConstrValue", 1e+8, "Loose constraint value"}

private

Definition at line 164 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_maximumVertexContamination

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_maximumVertexContamination {this, "maximumVertexContamination", 0.5, "Max. vertex contamination"}

private

Definition at line 163 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_maxIterations

UnsignedIntegerProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_maxIterations {this, "maxIterations", 100, "Vertex finder max. iterations"}

private

Definition at line 160 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_maxMergeVertexSignificance

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_maxMergeVertexSignificance {this, "maxMergeVertexSignificance", 3., "Max merge vertex significance"}

private

Definition at line 158 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_maxVertexChi2

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_maxVertexChi2 {this, "maxVertexChi2", 18.42, "Max vertex chi2"}

private

Definition at line 155 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_minWeight

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_minWeight {this, "minWeight", 0.0001, "Min track weight for finder"}

private

Definition at line 159 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_minWeightFitter

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_minWeightFitter {this, "minWeightFitter", 0.001, "Min track weight for fitter"}

private

Definition at line 149 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_propagator

std::shared_ptr<Propagator> ActsTrk::AdaptiveMultiPriVtxFinderTool::m_propagator = nullptr

private

Definition at line 127 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_seederType

StringProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_seederType {this, "seederType", "Gaussian", "type of seeder, \"Grid\" for Grid seeder, other for Gaussian seeder (default)"}

private

Definition at line 183 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_trackingGeometryTool

ToolHandle<IActsTrackingGeometryTool> ActsTrk::AdaptiveMultiPriVtxFinderTool::m_trackingGeometryTool {this, "TrackingGeometryTool", "", "ActsTrackingGeometryTool"}

private

Definition at line 132 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_tracksMaxSignificance

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_tracksMaxSignificance {this, "tracksMaxSignificance", 5., "Tracks max. significance"}

private

Definition at line 154 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_tracksMaxZinterval

DoubleProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_tracksMaxZinterval {this, "tracksMaxZinterval", 1., "Tracks max. Z-interval"}

private

Definition at line 153 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_trkFilter

ToolHandle<InDet::IInDetTrackSelectionTool> ActsTrk::AdaptiveMultiPriVtxFinderTool::m_trkFilter {this, "TrackSelector", "", "InDetTrackSelectionTool"}

private

Definition at line 134 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_useBeamConstraint

BooleanProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_useBeamConstraint {this, "useBeamConstraint", true, "Use beam constraint"}

private

Definition at line 152 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_useFastCompatibility

BooleanProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_useFastCompatibility {this, "useFastCompatibility", true, "Use fast compatibility estimation"}

private

Definition at line 157 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_useSeedConstraint

BooleanProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_useSeedConstraint {this, "useSeedConstraint", false, "Use seed constraint in fit"}

private

Definition at line 166 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_useVertexCovForIPEstimation

BooleanProperty ActsTrk::AdaptiveMultiPriVtxFinderTool::m_useVertexCovForIPEstimation {this, "useVertexCovForIPEstimation", false, "Use seed vertex cov for IPEstimation"}

private

Definition at line 165 of file AdaptiveMultiPriVtxFinderTool.h.

◆ m_vertexFinder

std::shared_ptr<VertexFinder> ActsTrk::AdaptiveMultiPriVtxFinderTool::m_vertexFinder = nullptr

private

Definition at line 126 of file AdaptiveMultiPriVtxFinderTool.h.

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

Classes

Public Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ Propagator

◆ TrackLinearizer

◆ VertexFinder

◆ VertexFitter

◆ VertexSeedFinder

Constructor & Destructor Documentation

◆ AdaptiveMultiPriVtxFinderTool()

Member Function Documentation

◆ actsBoundToTrkPerigee()

◆ estimateSignalCompatibility()

◆ findVertex() [1/3]

◆ findVertex() [2/3]

◆ findVertex() [3/3]

◆ initialize()

◆ initializeVertexFinder()

◆ logger()

◆ trackingGeometryTool()

Member Data Documentation

◆ m_addSingleTrackVertices

◆ m_annealingCutOff

◆ m_annealingTemps

◆ m_beamSpotKey

◆ m_doFullSplitting

◆ m_doRealMultiVertex

◆ m_extrapolationTool

◆ m_finalCutMaxVertexChi2

◆ m_fitterDoSmoothing

◆ m_fitterMaxDistToLinPoint

◆ m_fitterMaxIterations

◆ m_fitterMaxRelativeShift

◆ m_gaussianMaxD0Significance

◆ m_gaussianMaxZ0Significance

◆ m_gridMainGridSize

◆ m_gridMaxD0Significance

◆ m_gridMaxZ0Significance

◆ m_gridTrkGridSize

◆ m_gridUseHighestSumZPosition

◆ m_ipEstMaxIterations

◆ m_ipEstPrecision

◆ m_linearizer

◆ m_logger

◆ m_looseConstrValue

◆ m_maximumVertexContamination

◆ m_maxIterations

◆ m_maxMergeVertexSignificance

◆ m_maxVertexChi2

◆ m_minWeight

◆ m_minWeightFitter

◆ m_propagator

◆ m_seederType

◆ m_trackingGeometryTool

◆ m_tracksMaxSignificance

◆ m_tracksMaxZinterval

◆ m_trkFilter

◆ m_useBeamConstraint

◆ m_useFastCompatibility

◆ m_useSeedConstraint

◆ m_useVertexCovForIPEstimation

◆ m_vertexFinder