ATLAS Offline Software
Loading...
Searching...
No Matches
ActsTrk::IterativePriVtxFinderTool Class Reference

#include <IterativePriVtxFinderTool.h>

Inheritance diagram for ActsTrk::IterativePriVtxFinderTool:
Collaboration diagram for ActsTrk::IterativePriVtxFinderTool:

Classes

class  TrackWrapper

Public Member Functions

virtual StatusCode initialize () override
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::FullBilloirVertexFitter
using VertexSeedFinder = Acts::TrackDensityVertexFinder
using VertexFinder = Acts::IterativeVertexFinder

Private Member Functions

const Acts::Logger & logger () const
std::pair< xAOD::VertexContainer *, xAOD::VertexAuxContainer * > findVertex (const EventContext &ctx, const std::vector< std::unique_ptr< Trk::ITrackLink > > &trackVector) const
Trk::PerigeeactsBoundToTrkPerigee (const Acts::BoundTrackParameters &bound, const Acts::Vector3 &surfCenter) const

Private Attributes

ContextUtility m_ctxProvider {this}
 Auxiliary class to access the magnetic field, geometry and calibration context.
std::unique_ptr< const Acts::Logger > m_logger {nullptr}
 logging instance
std::shared_ptr< VertexFinderm_vertexFinder = nullptr
std::shared_ptr< ATLASMagneticFieldWrapperm_bField = nullptr
std::shared_ptr< Propagatorm_propagator = nullptr
std::optional< TrackLinearizerm_linearizer = std::nullopt
ServiceHandle< ActsTrk::ITrackingGeometrySvcm_trackingGeometrySvc {this, "TrackingGeometrySvc", "ActsTrackingGeometrySvc"}
ToolHandle< InDet::IInDetTrackSelectionToolm_trkFilter {this, "TrackSelector", "", "InDetTrackSelectionTool"}
SG::ReadCondHandleKey< InDet::BeamSpotDatam_beamSpotKey {this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot"}
UnsignedIntegerProperty m_fitterMaxIterations {this, "fitterMaxIterations", 5, "Vertex fitter max. iterations"}
BooleanProperty m_useBeamConstraint {this, "useBeamConstraint", true, "Use beam constraint"}
DoubleProperty m_significanceCutSeeding {this, "significanceCutSeeding", 10, "Seeding Cut Significance"}
DoubleProperty m_maximumChi2cutForSeeding {this, "maximumChi2cutForSeeding", 36., "Max. Chi2 cut for seeding"}
UnsignedIntegerProperty m_maxVertices {this, "maxVertices", 50, "Max number of vertices"}
BooleanProperty m_createSplitVertices {this, "createSplitVertices", false, "Create split vertices or not"}
UnsignedIntegerProperty m_splitVerticesTrkInvFraction {this, "splitVerticesTrkInvFraction", 2, "Track Inv. fraction for split vertices"}
BooleanProperty m_reassignTracksAfterFirstFit {this, "reassignTracksAfterFirstFit", false, "Whether re-assign tracks after first vertex fit or not"}
BooleanProperty m_doMaxTracksCut {this, "doMaxTracksCut", false, "Whether use max. tracks cut or not"}
UnsignedIntegerProperty m_maxTracks {this, "maxTracks", 5000, "Max. number of tracks to use for vertex finding"}
DoubleProperty m_cutOffTrackWeight {this, "cutOffTrackWeight", 0.01, "Min. track weight allowed"}
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_ipEstMaxIterations {this, "ipEstMaxIterations", 20, "IpEstimator max. iterations"}
DoubleProperty m_ipEstPrecision {this, "ipEstPrecision", 1e-10, "IpEstimator precision"}

Detailed Description

Definition at line 56 of file IterativePriVtxFinderTool.h.

Member Typedef Documentation

◆ Propagator

using ActsTrk::IterativePriVtxFinderTool::Propagator = Acts::Propagator<Acts::EigenStepper<>, Acts::Navigator>
private

Definition at line 107 of file IterativePriVtxFinderTool.h.

◆ TrackLinearizer

using ActsTrk::IterativePriVtxFinderTool::TrackLinearizer = Acts::HelicalTrackLinearizer
private

Definition at line 108 of file IterativePriVtxFinderTool.h.

◆ VertexFinder

using ActsTrk::IterativePriVtxFinderTool::VertexFinder = Acts::IterativeVertexFinder
private

Definition at line 111 of file IterativePriVtxFinderTool.h.

◆ VertexFitter

using ActsTrk::IterativePriVtxFinderTool::VertexFitter = Acts::FullBilloirVertexFitter
private

Definition at line 109 of file IterativePriVtxFinderTool.h.

◆ VertexSeedFinder

using ActsTrk::IterativePriVtxFinderTool::VertexSeedFinder = Acts::TrackDensityVertexFinder
private

Definition at line 110 of file IterativePriVtxFinderTool.h.

Member Function Documentation

◆ actsBoundToTrkPerigee()

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

Definition at line 391 of file IterativePriVtxFinderTool.cxx.

392 {
393 using namespace Acts::UnitLiterals;
394 AmgSymMatrix(5) cov = AmgSymMatrix(5)(bound.covariance()->block<5,5>(0,0));
395 cov.col(Trk::qOverP) *= 1_MeV;
396 cov.row(Trk::qOverP) *= 1_MeV;
397 Acts::Vector<5> params = bound.parameters().head<5>();
398 params[Trk::qOverP] *= 1_MeV;
399
400 return new Trk::Perigee(params, Trk::PerigeeSurface(surfCenter), std::move(cov));
401}
#define AmgSymMatrix(dim)
std::string head(std::string s, const std::string &pattern)
head of a string

◆ findVertex() [1/3]

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

Definition at line 195 of file IterativePriVtxFinderTool.cxx.

197{
198
199 using namespace Acts::UnitLiterals;
200
201 // Vertex finding algorithm begins
202 SG::ReadCondHandle<InDet::BeamSpotData> beamSpotHandle { m_beamSpotKey, ctx};
203
204 // The output vertex containers
205 xAOD::VertexContainer* theVertexContainer = new xAOD::VertexContainer;
206 xAOD::VertexAuxContainer* theVertexAuxContainer = new xAOD::VertexAuxContainer;
207 theVertexContainer->setStore(theVertexAuxContainer);
208
209 // bail out early with only Dummy vertex if multiplicity cut is applied and exceeded
210 if((m_doMaxTracksCut && (trackVector.size() > m_maxTracks)) || trackVector.empty()) {
211 ATH_MSG_WARNING(trackVector.size()
212 << " tracks - exceeds maximum (" << m_maxTracks
213 << "), skipping vertexing and returning only dummy...");
214 xAOD::Vertex* dummyxAODVertex = new xAOD::Vertex;
215 theVertexContainer->push_back(
216 dummyxAODVertex); // have to add vertex to container here first so it can use its aux store
217 dummyxAODVertex->setPosition(beamSpotHandle->beamVtx().position());
218 dummyxAODVertex->setCovariancePosition(
219 beamSpotHandle->beamVtx().covariancePosition());
220 dummyxAODVertex->vxTrackAtVertex() = std::vector<Trk::VxTrackAtVertex>();
221 dummyxAODVertex->setVertexType(xAOD::VxType::NoVtx);
222 return std::make_pair(theVertexContainer, theVertexAuxContainer);
223 }
224
225 const Acts::Vector3& beamSpotPos = beamSpotHandle->beamVtx().position();
226 Acts::Vertex beamSpotConstraintVtx(beamSpotPos);
227 beamSpotConstraintVtx.setCovariance(beamSpotHandle->beamVtx().covariancePosition());
228
229 std::shared_ptr<Acts::PerigeeSurface> perigeeSurface =
230 Acts::Surface::makeShared<Acts::PerigeeSurface>((trackVector[0])->parameters()->associatedSurface().transform());
231
232 // Get the magnetic field context
233 const Acts::MagneticFieldContext magFieldContext = m_ctxProvider.getMagneticFieldContext(ctx);
234 const Acts::GeometryContext geoContext = m_ctxProvider.getGeometryContext(ctx);
235
236 // Convert tracks to Acts::BoundParameters
237 std::vector<TrackWrapper> allTracks;
238
239 for (const auto& trk : trackVector) {
240
241 const auto& trkParams = trk->parameters();
242 const auto& params = trkParams->parameters();
243
244 Acts::BoundVector actsParams;
245 actsParams << params(0), params(1), params(2), params(3), params(4)*1./(1_MeV), 0.;
246
247 if(trkParams->covariance() == nullptr){
248 continue;
249 }
250 auto cov = *(trkParams->covariance());
251
252 // TODO: check if the following works as well:
253 // cov->col(4) *= 1./1_MeV;
254 // cov->row(4) *= 1./1_MeV;
255 Acts::BoundMatrix covMat;
256 covMat << cov(0,0) , cov(0,1) , cov(0,2) , cov(0,3) , cov(0,4) *1./(1_MeV), 0
257 , cov(1,0) , cov(1,1) , cov(1,2) , cov(1,3) , cov(1,4) *1./(1_MeV) , 0
258 , cov(2,0) , cov(2,1) , cov(2,2) , cov(2,3) , cov(2,4) *1./(1_MeV) , 0
259 , cov(3,0) , cov(3,1) , cov(3,2) , cov(3,3) , cov(3,4) *1./(1_MeV) , 0
260 , 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
261 , 0. , 0. , 0. , 0., 0., 1.;
262
263 allTracks.emplace_back(trk.get(),Acts::BoundTrackParameters(perigeeSurface, actsParams, covMat, Acts::ParticleHypothesis::pion()));
264 }
265
266 std::vector<Acts::InputTrack> allTrackPtrs;
267 allTrackPtrs.reserve(allTracks.size());
268
269for(const auto& trk : allTracks){
270 allTrackPtrs.emplace_back(&trk);
271 }
272
273 Acts::VertexingOptions vertexingOptions(geoContext,
274 magFieldContext);
275
277 beamSpotConstraintVtx.setPosition(Acts::Vector3::Zero());
278 beamSpotConstraintVtx.setCovariance(Acts::SquareMatrix<3>::Zero());
279 }
280 vertexingOptions.useConstraintInFit = m_useBeamConstraint;
281
282 //Adding 4th dimensional timing info to vertex constraint as needed by ACTS
283 Acts::Vector4 vtxConstraintPos;
284 Acts::SquareMatrix4 vtxConstraintCov;
285
286 auto beamSpotCov = beamSpotHandle->beamVtx().covariancePosition();
287
288 vtxConstraintPos << beamSpotPos(0), beamSpotPos(1), beamSpotPos(2), 0.;
289 vtxConstraintCov << beamSpotCov(0,0), beamSpotCov(0,1), beamSpotCov(0,2), 0.
290 , beamSpotCov(1,0), beamSpotCov(1,1), beamSpotCov(1,2), 0.
291 , beamSpotCov(2,0), beamSpotCov(2,1), beamSpotCov(2,2), 0.
292 , 0., 0., 0., 1.;
293
294 vertexingOptions.constraint.setFullPosition(vtxConstraintPos);
295 vertexingOptions.constraint.setFullCovariance(vtxConstraintCov);
296
297 auto finderState = m_vertexFinder->makeState(magFieldContext);
298
299 auto findResult = m_vertexFinder->find(allTrackPtrs, vertexingOptions, finderState);
300
301 if(!findResult.ok()){
302 xAOD::Vertex* dummyxAODVertex = new xAOD::Vertex;
303 theVertexContainer->push_back(dummyxAODVertex);
304 dummyxAODVertex->setPosition(beamSpotHandle->beamVtx().position());
305 dummyxAODVertex->setCovariancePosition(beamSpotHandle->beamVtx().covariancePosition());
306 dummyxAODVertex->vxTrackAtVertex() = std::vector<Trk::VxTrackAtVertex>();
307 dummyxAODVertex->setVertexType(xAOD::VxType::NoVtx);
308
309 return std::make_pair(theVertexContainer, theVertexAuxContainer);
310 }
311
312 std::vector<Acts::Vertex> allVertices = *findResult;
313
314 for(const auto& vtx : allVertices){
315 xAOD::Vertex* xAODVtx = new xAOD::Vertex;
316 xAODVtx->makePrivateStore();
317 xAODVtx->setPosition(vtx.position());
318 xAODVtx->setCovariancePosition(vtx.covariance());
319 xAODVtx->setFitQuality(vtx.fitQuality().first, vtx.fitQuality().second);
320
321 const auto& tracks = vtx.tracks();
322 std::vector<Trk::VxTrackAtVertex>* trkAtVtxVec = &(xAODVtx->vxTrackAtVertex());
323 for(const auto& trk : tracks){
324
325 Trk::Perigee* fittedPerigee = actsBoundToTrkPerigee(trk.fittedParams, beamSpotPos);
326 //Trk::Perigee* originalPerigee = actsBoundToTrkPerigee((trk.originalParams)->parameters(), beamSpotPos);
327 const TrackWrapper* originalParams = trk.originalParams.template as<TrackWrapper>();
328
329 //Trk::VxTrackAtVertex trkAtVtx(trk.chi2Track, fittedPerigee, originalPerigee);
330 Trk::VxTrackAtVertex trkAtVtx(originalParams->trackLink()->clone());
331 trkAtVtx.setPerigeeAtVertex(fittedPerigee);
332 trkAtVtx.setTrackQuality(Trk::FitQuality(trk.chi2Track, trk.ndf));
333 trkAtVtx.setVtxCompatibility(trk.vertexCompatibility);
334 trkAtVtx.setWeight(trk.trackWeight);
335 trkAtVtxVec->push_back(trkAtVtx);
336
337 const Trk::LinkToXAODTrackParticle* linkToXAODTP =
338 dynamic_cast<const Trk::LinkToXAODTrackParticle*>(originalParams->trackLink());
339 if (linkToXAODTP) {
340 xAODVtx->addTrackAtVertex(*linkToXAODTP, trk.trackWeight);
341 }
342 }
343
344 theVertexContainer->push_back(xAODVtx);
345 }
346
348 if (!theVertexContainer->empty()) {
349 xAOD::Vertex* primaryVtx = theVertexContainer->front();
350 if (!primaryVtx->vxTrackAtVertex().empty()) {
352 xAOD::Vertex* dummyxAODVertex = new xAOD::Vertex;
353 theVertexContainer->push_back(dummyxAODVertex);
354 dummyxAODVertex->setPosition(primaryVtx->position());
355 dummyxAODVertex->setCovariancePosition(primaryVtx->covariancePosition());
356 dummyxAODVertex->vxTrackAtVertex() = std::vector<Trk::VxTrackAtVertex>();
357 dummyxAODVertex->setVertexType(xAOD::VxType::NoVtx);
358 } else {
360 }
361 } else {
362 xAOD::Vertex* dummyxAODVertex = new xAOD::Vertex;
363 theVertexContainer->push_back(dummyxAODVertex);
364 dummyxAODVertex->setPosition(beamSpotHandle->beamVtx().position());
365 dummyxAODVertex->setCovariancePosition(beamSpotHandle->beamVtx().covariancePosition());
366 dummyxAODVertex->vxTrackAtVertex() = std::vector<Trk::VxTrackAtVertex>();
367 dummyxAODVertex->setVertexType(xAOD::VxType::NoVtx);
368 }
369
370 // loop over the pile up to set it as pile up (EXCLUDE first and last
371 // vertex, do not do that in split mode)
372 for (unsigned int i = 0; i < theVertexContainer->size() - 1; i++) {
373
375 " Vtx: " << i << " x= " << (*theVertexContainer)[i]->position().x()
376 << " y= " << (*theVertexContainer)[i]->position().y() << " z= "
377 << (*theVertexContainer)[i]->position().z() << " ntracks= "
378 << (*theVertexContainer)[i]->vxTrackAtVertex().size()
379 << " chi2= " << (*theVertexContainer)[i]->chiSquared()
380 << " ndf = " << (*theVertexContainer)[i]->numberDoF());
381 if (i > 0) {
382 (*theVertexContainer)[i]->setVertexType(xAOD::VxType::PileUp);
383 }
384 }
385 }
386
387 return std::make_pair(theVertexContainer, theVertexAuxContainer);
388}
#define ATH_MSG_WARNING(x)
#define ATH_MSG_DEBUG(x)
size_t size() const
Number of registered mappings.
#define y
#define x
#define z
ContextUtility m_ctxProvider
Auxiliary class to access the magnetic field, geometry and calibration context.
Trk::Perigee * actsBoundToTrkPerigee(const Acts::BoundTrackParameters &bound, const Acts::Vector3 &surfCenter) const
std::shared_ptr< VertexFinder > m_vertexFinder
SG::ReadCondHandleKey< InDet::BeamSpotData > m_beamSpotKey
value_type push_back(value_type pElem)
Add an element to the end of the collection.
const T * front() const
Access the first element in the collection as an rvalue.
size_type size() const noexcept
Returns the number of elements in the collection.
bool empty() const noexcept
Returns true if the collection is empty.
void setCovariancePosition(const AmgSymMatrix(3)&covariancePosition)
Sets the vertex covariance matrix.
void addTrackAtVertex(const ElementLink< TrackParticleContainer > &tr, float weight=1.0)
Add a new track to the vertex.
void setVertexType(VxType::VertexType vType)
Set the type of the vertex.
void setPosition(const Amg::Vector3D &position)
Sets the 3-position.
std::vector< Trk::VxTrackAtVertex > & vxTrackAtVertex()
Non-const access to the VxTrackAtVertex vector.
void setFitQuality(float chiSquared, float numberDoF)
Set the 'Fit Quality' information.
const Amg::Vector3D & position() const
Returns the 3-pos.
Amg::Vector3D transform(Amg::Vector3D &v, Amg::Transform3D &tr)
Transform a point from a Trasformation3D.
float chiSquared(const U &p)
@ PriVtx
Primary Vertex.
Definition VertexType.h:27
ParametersT< TrackParametersDim, Charged, PerigeeSurface > Perigee
const Amg::Vector3D & position() const
Method to retrieve the position of the Intersection.
VertexType
Vertex types.
@ PileUp
Pile-up vertex.
@ NoVtx
Dummy vertex. TrackParticle was not used in vertex fit.
VertexAuxContainer_v1 VertexAuxContainer
Definition of the current jet auxiliary container.
VertexContainer_v1 VertexContainer
Definition of the current "Vertex container version".
Vertex_v1 Vertex
Define the latest version of the vertex class.

◆ findVertex() [2/3]

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

Definition at line 119 of file IterativePriVtxFinderTool.cxx.

121{
122
123 SG::ReadCondHandle<InDet::BeamSpotData> beamSpotHandle { m_beamSpotKey, ctx};
124 const Trk::RecVertex& beamposition(beamSpotHandle->beamVtx());
125
126 std::vector<std::unique_ptr<Trk::ITrackLink>> selectedTracks;
127
128 typedef DataVector<Trk::Track>::const_iterator TrackDataVecIter;
129
130 bool selectionPassed;
131 for (TrackDataVecIter itr = (*trackTES).begin(); itr != (*trackTES).end(); ++itr) {
133 selectionPassed = static_cast<bool>(m_trkFilter->accept(**itr, &beamposition));
134 } else {
135 Trk::Vertex null(Amg::Vector3D(0, 0, 0));
136 selectionPassed = static_cast<bool>(m_trkFilter->accept(**itr, &null));
137 }
138 if (selectionPassed) {
139 ElementLink<TrackCollection> link;
140 link.setElement(*itr);
141 auto trkPtr = std::make_unique<Trk::LinkToTrack>(link);
142 trkPtr->setStorableObject(*trackTES);
143 selectedTracks.push_back(std::move(trkPtr));
144 }
145 }
146
147 std::pair<xAOD::VertexContainer*, xAOD::VertexAuxContainer*> returnContainers = findVertex(ctx, std::move(selectedTracks));
148
149 return returnContainers;
150}
virtual std::pair< xAOD::VertexContainer *, xAOD::VertexAuxContainer * > findVertex(const EventContext &ctx, const TrackCollection *trackTES) const override
ToolHandle< InDet::IInDetTrackSelectionTool > m_trkFilter
DataModel_detail::const_iterator< DataVector > const_iterator
Standard const_iterator.
Definition DataVector.h:838
Eigen::Matrix< double, 3, 1 > Vector3D
pointer & link(pointer p) const
Return a reference to the link for an element.

◆ findVertex() [3/3]

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

Definition at line 153 of file IterativePriVtxFinderTool.cxx.

155{
156
157 std::vector<std::unique_ptr<Trk::ITrackLink>> selectedTracks;
158 SG::ReadCondHandle<InDet::BeamSpotData> beamSpotHandle { m_beamSpotKey, ctx};
159 xAOD::Vertex beamposition;
160 beamposition.makePrivateStore();
161 beamposition.setPosition(beamSpotHandle->beamVtx().position());
162 beamposition.setCovariancePosition(beamSpotHandle->beamVtx().covariancePosition());
163
164 typedef DataVector<xAOD::TrackParticle>::const_iterator TrackParticleDataVecIter;
165
166 bool selectionPassed;
167 for (TrackParticleDataVecIter itr = (*trackParticles).begin(); itr != (*trackParticles).end(); ++itr) {
169 selectionPassed = static_cast<bool>(m_trkFilter->accept(**itr, &beamposition));
170 } else {
171 xAOD::Vertex null;
172 null.makePrivateStore();
173 null.setPosition(Amg::Vector3D(0, 0, 0));
174 AmgSymMatrix(3) vertexError;
175 vertexError.setZero();
176 null.setCovariancePosition(vertexError);
177 selectionPassed = static_cast<bool>(m_trkFilter->accept(**itr, &null));
178 }
179
180 if (selectionPassed) {
181 ElementLink<xAOD::TrackParticleContainer> link;
182 link.setElement(*itr);
183 auto trkPtr = std::make_unique<Trk::LinkToXAODTrackParticle>(link);
184 trkPtr->setStorableObject(*trackParticles);
185 selectedTracks.push_back(std::move(trkPtr));
186 }
187 }
188
189 std::pair<xAOD::VertexContainer*, xAOD::VertexAuxContainer*> returnContainers = findVertex(ctx, std::move(selectedTracks));
190
191 return returnContainers;
192}
if(pathvar)

◆ initialize()

StatusCode ActsTrk::IterativePriVtxFinderTool::initialize ( )
overridevirtual

Definition at line 41 of file IterativePriVtxFinderTool.cxx.

42{
43 using namespace std::literals::string_literals;
44
45 ATH_CHECK(m_beamSpotKey.initialize());
46 ATH_CHECK(m_trkFilter.retrieve());
47
48 m_logger = makeActsAthenaLogger(this, "Acts");
49
50 ATH_MSG_INFO("Initializing ACTS Iterative Vertex Finder tool");
51 ATH_CHECK(m_ctxProvider.initialize());
52 ATH_CHECK( m_trackingGeometrySvc.retrieve() );
53 std::shared_ptr<const Acts::TrackingGeometry> trackingGeometry
54 = m_trackingGeometrySvc->trackingGeometry();
55
56 Acts::Navigator navigator( Acts::Navigator::Config{ trackingGeometry },
57 logger().cloneWithSuffix("Navigator"));
58
59 m_bField = std::make_shared<ATLASMagneticFieldWrapper>();
60 auto stepper = Acts::EigenStepper<>(m_bField);
61 m_propagator = std::make_shared<Propagator>(std::move(stepper),
62 std::move(navigator),
63 logger().cloneWithSuffix("Prop"));
64 // IP Estimator
65 Acts::ImpactPointEstimator::Config ipEstCfg(m_bField, m_propagator);
66 ipEstCfg.maxIterations = m_ipEstMaxIterations;
67 ipEstCfg.precision = m_ipEstPrecision;
68 Acts::ImpactPointEstimator ipEst(ipEstCfg,
69 logger().cloneWithSuffix("ImpactPointEstimator"));
70
71 // Linearizer for Acts::BoundParameters type test
72 TrackLinearizer::Config ltConfig;
73 ltConfig.bField = m_bField;
74 ltConfig.propagator = m_propagator;
75 m_linearizer.emplace(ltConfig, logger().cloneWithSuffix("Linearizer"));
76
77 // Full Billoir Vertex fitter setup
78 VertexFitter::Config fitterCfg;
79 fitterCfg.maxIterations = m_fitterMaxIterations;
80 fitterCfg.extractParameters.connect<&TrackWrapper::extractParameters>();
81 fitterCfg.trackLinearizer.connect<&TrackLinearizer::linearizeTrack>(&*m_linearizer);
82 VertexFitter fitter(fitterCfg, logger().cloneWithSuffix("Fitter"));
83
84
85 // Seed finder setup
86 // Set up Gaussian track density
87 Acts::GaussianTrackDensity::Config trackDensityConfig;
88 trackDensityConfig.d0MaxSignificance = m_gaussianMaxD0Significance;
89 trackDensityConfig.z0MaxSignificance = m_gaussianMaxZ0Significance;
90 trackDensityConfig.extractParameters.connect<&TrackWrapper::extractParameters>();
91 Acts::GaussianTrackDensity trackDensity(trackDensityConfig);
92
93 // Vertex seed finder
94 VertexSeedFinder::Config seedFinderConfig{trackDensity};
95 auto seedFinder = std::make_shared<VertexSeedFinder>(seedFinderConfig);
96
97 // Iterative Vertex Finder setup
98 VertexFinder::Config finderConfig(std::move(fitter),
99 std::move(seedFinder),
100 ipEst);
101 finderConfig.significanceCutSeeding = m_significanceCutSeeding;
102 finderConfig.maximumChi2cutForSeeding = m_maximumChi2cutForSeeding;
103 finderConfig.maxVertices = m_maxVertices;
104 finderConfig.createSplitVertices = m_createSplitVertices;
105 finderConfig.splitVerticesTrkInvFraction = m_splitVerticesTrkInvFraction;
106 finderConfig.reassignTracksAfterFirstFit = m_reassignTracksAfterFirstFit;
107 finderConfig.doMaxTracksCut = m_doMaxTracksCut;
108 finderConfig.maxTracks = m_maxTracks;
109 finderConfig.cutOffTrackWeight = m_cutOffTrackWeight;
110 finderConfig.extractParameters.connect<&TrackWrapper::extractParameters>();
111 finderConfig.trackLinearizer.connect<&TrackLinearizer::linearizeTrack>(&*m_linearizer);
112 m_vertexFinder = std::make_shared<VertexFinder>(std::move(finderConfig), logger().cloneWithSuffix("Finder"));
113
114 ATH_MSG_INFO("ACTS Iterative Vertex Finder tool successfully initialized");
115 return StatusCode::SUCCESS;
116}
#define ATH_CHECK
Evaluate an expression and check for errors.
#define ATH_MSG_INFO(x)
std::unique_ptr< const Acts::Logger > makeActsAthenaLogger(IMessageSvc *svc, const std::string &name, int level, std::optional< std::string > parent_name)
static Acts::BoundTrackParameters extractParameters(const Acts::InputTrack &input)
std::shared_ptr< Propagator > m_propagator
std::optional< TrackLinearizer > m_linearizer
std::unique_ptr< const Acts::Logger > m_logger
logging instance
std::shared_ptr< ATLASMagneticFieldWrapper > m_bField
UnsignedIntegerProperty m_splitVerticesTrkInvFraction
ServiceHandle< ActsTrk::ITrackingGeometrySvc > m_trackingGeometrySvc
Acts::FullBilloirVertexFitter VertexFitter
const ShapeFitter * fitter

◆ logger()

const Acts::Logger & ActsTrk::IterativePriVtxFinderTool::logger ( ) const
inlineprivate

Definition at line 99 of file IterativePriVtxFinderTool.h.

99{ return *m_logger; }

Member Data Documentation

◆ m_beamSpotKey

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

Definition at line 124 of file IterativePriVtxFinderTool.h.

124{this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot"};

◆ m_bField

std::shared_ptr<ATLASMagneticFieldWrapper> ActsTrk::IterativePriVtxFinderTool::m_bField = nullptr
private

Definition at line 114 of file IterativePriVtxFinderTool.h.

◆ m_createSplitVertices

BooleanProperty ActsTrk::IterativePriVtxFinderTool::m_createSplitVertices {this, "createSplitVertices", false, "Create split vertices or not"}
private

Definition at line 137 of file IterativePriVtxFinderTool.h.

137{this, "createSplitVertices", false, "Create split vertices or not"};

◆ m_ctxProvider

ContextUtility ActsTrk::IterativePriVtxFinderTool::m_ctxProvider {this}
private

Auxiliary class to access the magnetic field, geometry and calibration context.

Definition at line 96 of file IterativePriVtxFinderTool.h.

96{this};

◆ m_cutOffTrackWeight

DoubleProperty ActsTrk::IterativePriVtxFinderTool::m_cutOffTrackWeight {this, "cutOffTrackWeight", 0.01, "Min. track weight allowed"}
private

Definition at line 142 of file IterativePriVtxFinderTool.h.

142{this, "cutOffTrackWeight", 0.01, "Min. track weight allowed"};

◆ m_doMaxTracksCut

BooleanProperty ActsTrk::IterativePriVtxFinderTool::m_doMaxTracksCut {this, "doMaxTracksCut", false, "Whether use max. tracks cut or not"}
private

Definition at line 140 of file IterativePriVtxFinderTool.h.

140{this, "doMaxTracksCut", false, "Whether use max. tracks cut or not"};

◆ m_fitterMaxIterations

UnsignedIntegerProperty ActsTrk::IterativePriVtxFinderTool::m_fitterMaxIterations {this, "fitterMaxIterations", 5, "Vertex fitter max. iterations"}
private

Definition at line 130 of file IterativePriVtxFinderTool.h.

130{this, "fitterMaxIterations", 5, "Vertex fitter max. iterations"};

◆ m_gaussianMaxD0Significance

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

Definition at line 145 of file IterativePriVtxFinderTool.h.

145{this, "gaussianMaxD0Significance", 3.5, "Gaussian seeder max d0 track significance"};

◆ m_gaussianMaxZ0Significance

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

Definition at line 146 of file IterativePriVtxFinderTool.h.

146{this, "gaussianMaxDZSignificance", 12.0, "Gaussian seeder max z0 track significance"};

◆ m_ipEstMaxIterations

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

Definition at line 149 of file IterativePriVtxFinderTool.h.

149{this, "ipEstMaxIterations", 20, "IpEstimator max. iterations"};

◆ m_ipEstPrecision

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

Definition at line 150 of file IterativePriVtxFinderTool.h.

150{this, "ipEstPrecision", 1e-10, "IpEstimator precision"};

◆ m_linearizer

std::optional<TrackLinearizer> ActsTrk::IterativePriVtxFinderTool::m_linearizer = std::nullopt
private

Definition at line 119 of file IterativePriVtxFinderTool.h.

◆ m_logger

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

logging instance

Definition at line 98 of file IterativePriVtxFinderTool.h.

98{nullptr};

◆ m_maximumChi2cutForSeeding

DoubleProperty ActsTrk::IterativePriVtxFinderTool::m_maximumChi2cutForSeeding {this, "maximumChi2cutForSeeding", 36., "Max. Chi2 cut for seeding"}
private

Definition at line 135 of file IterativePriVtxFinderTool.h.

135{this, "maximumChi2cutForSeeding", 36., "Max. Chi2 cut for seeding"};

◆ m_maxTracks

UnsignedIntegerProperty ActsTrk::IterativePriVtxFinderTool::m_maxTracks {this, "maxTracks", 5000, "Max. number of tracks to use for vertex finding"}
private

Definition at line 141 of file IterativePriVtxFinderTool.h.

141{this, "maxTracks", 5000, "Max. number of tracks to use for vertex finding"};

◆ m_maxVertices

UnsignedIntegerProperty ActsTrk::IterativePriVtxFinderTool::m_maxVertices {this, "maxVertices", 50, "Max number of vertices"}
private

Definition at line 136 of file IterativePriVtxFinderTool.h.

136{this, "maxVertices", 50, "Max number of vertices"};

◆ m_propagator

std::shared_ptr<Propagator> ActsTrk::IterativePriVtxFinderTool::m_propagator = nullptr
private

Definition at line 116 of file IterativePriVtxFinderTool.h.

◆ m_reassignTracksAfterFirstFit

BooleanProperty ActsTrk::IterativePriVtxFinderTool::m_reassignTracksAfterFirstFit {this, "reassignTracksAfterFirstFit", false, "Whether re-assign tracks after first vertex fit or not"}
private

Definition at line 139 of file IterativePriVtxFinderTool.h.

139{this, "reassignTracksAfterFirstFit", false, "Whether re-assign tracks after first vertex fit or not"};

◆ m_significanceCutSeeding

DoubleProperty ActsTrk::IterativePriVtxFinderTool::m_significanceCutSeeding {this, "significanceCutSeeding", 10, "Seeding Cut Significance"}
private

Definition at line 134 of file IterativePriVtxFinderTool.h.

134{this, "significanceCutSeeding", 10, "Seeding Cut Significance"};

◆ m_splitVerticesTrkInvFraction

UnsignedIntegerProperty ActsTrk::IterativePriVtxFinderTool::m_splitVerticesTrkInvFraction {this, "splitVerticesTrkInvFraction", 2, "Track Inv. fraction for split vertices"}
private

Definition at line 138 of file IterativePriVtxFinderTool.h.

138{this, "splitVerticesTrkInvFraction", 2, "Track Inv. fraction for split vertices"};

◆ m_trackingGeometrySvc

ServiceHandle<ActsTrk::ITrackingGeometrySvc> ActsTrk::IterativePriVtxFinderTool::m_trackingGeometrySvc {this, "TrackingGeometrySvc", "ActsTrackingGeometrySvc"}
private

Definition at line 122 of file IterativePriVtxFinderTool.h.

122{this, "TrackingGeometrySvc", "ActsTrackingGeometrySvc"};

◆ m_trkFilter

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

Definition at line 123 of file IterativePriVtxFinderTool.h.

123{this, "TrackSelector", "", "InDetTrackSelectionTool"};

◆ m_useBeamConstraint

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

Definition at line 133 of file IterativePriVtxFinderTool.h.

133{this, "useBeamConstraint", true, "Use beam constraint"};

◆ m_vertexFinder

std::shared_ptr<VertexFinder> ActsTrk::IterativePriVtxFinderTool::m_vertexFinder = nullptr
private

Definition at line 113 of file IterativePriVtxFinderTool.h.


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