4040#include " DataFormatsCalibration/MeanVertexObject.h"
4141#include " CommonConstants/GeomConstants.h"
4242#include " PWGLF/Utils/strangenessBuilderModule.h"
43+ #include " Common/Tools/TrackPropagationModule.h"
4344#include " Common/Tools/StandardCCDBLoader.h"
4445
4546// The Run 3 AO2D stores the tracks at the point of innermost update. For a track with ITS this is the innermost (or second innermost)
@@ -85,6 +86,11 @@ struct StrangenessBuilderTester {
8586 o2::pwglf::strangenessbuilder::preSelectOpts preSelectOpts;
8687 o2::pwglf::strangenessbuilder::BuilderModule strangenessBuilderModule;
8788
89+ // track propagation
90+ o2::common::TrackPropagationProducts trackPropagationProducts;
91+ o2::common::TrackPropagationConfigurables trackPropagationConfigurables;
92+ o2::common::TrackPropagationModule trackPropagation;
93+
8894 // registry
8995 HistogramRegistry histos{" histos" };
9096
@@ -96,30 +102,35 @@ struct StrangenessBuilderTester {
96102 ccdb->setURL (ccdburl.value );
97103
98104 // task-specific
105+ trackPropagation.init (trackPropagationConfigurables, histos, initContext);
99106 strangenessBuilderModule.init (baseOpts, v0BuilderOpts, cascadeBuilderOpts, preSelectOpts, histos, initContext);
100107 }
101108
102109 void processRealData (soa::Join<aod::Collisions, aod::EvSels> const & collisions, aod::V0s const & v0s, aod::Cascades const & cascades, aod::TrackedCascades const & trackedCascades, FullTracksExtIU const & tracks, aod::BCsWithTimestamps const & bcs)
103110 {
104111 ccdbLoader.initCCDBfromBCs (standardCCDBLoaderConfigurables, ccdb, bcs);
112+ trackPropagation.fillTrackTables <false >(trackPropagationConfigurables, ccdbLoader, tracks, trackPropagationProducts, histos);
105113 strangenessBuilderModule.dataProcess (ccdb, histos, collisions, static_cast <TObject*>(nullptr ), v0s, cascades, trackedCascades, tracks, bcs, static_cast <TObject*>(nullptr ), products);
106114 }
107115
108116 void processMonteCarlo (soa::Join<aod::Collisions, aod::EvSels, aod::McCollisionLabels> const & collisions, aod::McCollisions const & mccollisions, aod::V0s const & v0s, aod::Cascades const & cascades, aod::TrackedCascades const & trackedCascades, FullTracksExtLabeledIU const & tracks, aod::BCsWithTimestamps const & bcs, aod::McParticles const & mcParticles)
109117 {
110118 ccdbLoader.initCCDBfromBCs (standardCCDBLoaderConfigurables, ccdb, bcs);
119+ trackPropagation.fillTrackTables <true >(trackPropagationConfigurables, ccdbLoader, tracks, trackPropagationProducts, histos);
111120 strangenessBuilderModule.dataProcess (ccdb, histos, collisions, mccollisions, v0s, cascades, trackedCascades, tracks, bcs, mcParticles, products);
112121 }
113122
114123 void processRealDataWithPID (soa::Join<aod::Collisions, aod::EvSels> const & collisions, aod::V0s const & v0s, aod::Cascades const & cascades, aod::TrackedCascades const & trackedCascades, FullTracksExtIUWithPID const & tracks, aod::BCsWithTimestamps const & bcs)
115124 {
116125 ccdbLoader.initCCDBfromBCs (standardCCDBLoaderConfigurables, ccdb, bcs);
126+ trackPropagation.fillTrackTables <false >(trackPropagationConfigurables, ccdbLoader, tracks, trackPropagationProducts, histos);
117127 strangenessBuilderModule.dataProcess (ccdb, histos, collisions, static_cast <TObject*>(nullptr ), v0s, cascades, trackedCascades, tracks, bcs, static_cast <TObject*>(nullptr ), products);
118128 }
119129
120130 void processMonteCarloWithPID (soa::Join<aod::Collisions, aod::EvSels, aod::McCollisionLabels> const & collisions, aod::McCollisions const & mccollisions, aod::V0s const & v0s, aod::Cascades const & cascades, aod::TrackedCascades const & trackedCascades, FullTracksExtLabeledIUWithPID const & tracks, aod::BCsWithTimestamps const & bcs, aod::McParticles const & mcParticles)
121131 {
122132 ccdbLoader.initCCDBfromBCs (standardCCDBLoaderConfigurables, ccdb, bcs);
133+ trackPropagation.fillTrackTables <true >(trackPropagationConfigurables, ccdbLoader, tracks, trackPropagationProducts, histos);
123134 strangenessBuilderModule.dataProcess (ccdb, histos, collisions, mccollisions, v0s, cascades, trackedCascades, tracks, bcs, mcParticles, products);
124135 }
125136
0 commit comments