diff --git a/Examples/Scripts/Python/full_chain_itk.py b/Examples/Scripts/Python/full_chain_itk.py index f92ef67923c..85a74cc78ab 100755 --- a/Examples/Scripts/Python/full_chain_itk.py +++ b/Examples/Scripts/Python/full_chain_itk.py @@ -10,6 +10,7 @@ ParticleSelectorConfig, addDigitization, ) +from acts.examples import TrackSelector from acts.examples.reconstruction import ( addSeeding, TruthSeedRanges, @@ -93,13 +94,24 @@ outputDirRoot=outputDir, ) +ckfTrackParameters = "fittedTrackParameters" +vtxTrackParameters = "vtxTrackParameters" +trackSelector = TrackSelector( + level=acts.logging.INFO, + inputTrackParameters=ckfTrackParameters, + outputTrackParameters=vtxTrackParameters, + removeNeutral=True, + ptMin=1.0 * u.GeV, + absEtaMax=4.0, + ) +s.addAlgorithm(trackSelector) addVertexFitting( s, field, - TrackSelectorRanges(pt=(1.0 * u.GeV, None), absEta=(None, 4.0), removeNeutral=True), vertexFinder=VertexFinder.Iterative, outputDirRoot=outputDir, - trajectories="trajectories", + trajectories=None, + trackParameters=vtxTrackParameters, ) s.run()