Skip to content

Commit fff55d1

Browse files
committed
Merge track propagation + strangeness builder
1 parent 849bede commit fff55d1

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

PWGLF/TableProducer/Strangeness/strangenessbuildertester.cxx

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@
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

Comments
 (0)