Skip to content

Commit f223077

Browse files
authored
changes adding data processing path
1 parent 1d2bbed commit f223077

File tree

1 file changed

+14
-7
lines changed

1 file changed

+14
-7
lines changed

PWGLF/Tasks/QC/straRecoStudy.cxx

Lines changed: 14 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ struct preProcessMCcollisions {
149149
histos.add("h2dPVResolutionXYWithXi", "h2dPVResolutionXYWithXi", kTH2D, {axisPVResolutionXY, axisTwenty});
150150
histos.add("h2dPVResolutionZWithXi", "h2dPVResolutionZWithXi", kTH2D, {axisPVResolutionZ, axisTwenty});
151151
histos.add("h2dPVResolutionTWithXi", "h2dPVResolutionTWithXi", kTH2D, {axisPVResolutionT, axisTwenty});
152-
152+
153153
// Helper to decipher this histogram
154154
histos.get<TH2>(HIST("h2dNContributors"))->GetYaxis()->SetBinLabel(1, "Recoed 1 time, 1st PV"); // size 1 = 0
155155
histos.get<TH2>(HIST("h2dNContributors"))->GetYaxis()->SetBinLabel(2, "Recoed 2 times, Biggest PV"); // size 2 = 1
@@ -190,7 +190,13 @@ struct preProcessMCcollisions {
190190
histos.add("h2dNContribSpecialCorr2b", "h2dNContribSpecialCorr2b", kTH2D, {axisContributorsTRD, axisContributorsTOF});
191191
}
192192

193-
void process(aod::McCollision const& mcCollision, soa::SmallGroups<soa::Join<aod::McCollisionLabels, aod::Collisions>> const& collisions, TracksCompleteIUMC const& tracks, aod::McParticles const& mcParticles, aod::BCs const&)
193+
void processData(TracksCompleteIU const& tracks)
194+
{
195+
// Dummy process
196+
}
197+
PROCESS_SWITCH(preProcessMCcollisions, processData, "Real data dummy", false);
198+
199+
void processMC(aod::McCollision const& mcCollision, soa::SmallGroups<soa::Join<aod::McCollisionLabels, aod::Collisions>> const& collisions, TracksCompleteIUMC const& tracks, aod::McParticles const& mcParticles, aod::BCs const&)
194200
{
195201
int lNumberOfXi = 0;
196202
for (auto& mcp : mcParticles) {
@@ -294,6 +300,7 @@ struct preProcessMCcollisions {
294300
}
295301
mcCollsExtra(collisions.size());
296302
}
303+
PROCESS_SWITCH(preProcessMCcollisions, processMC, "MC prepare", true);
297304
};
298305

299306
struct straRecoStudy {
@@ -380,7 +387,7 @@ struct straRecoStudy {
380387

381388
const AxisSpec axisITSClu{10, -0.5f, +9.5f, "ITS clusters"};
382389
const AxisSpec axisTPCCroRo{160, -0.5f, +159.5f, "TPC crossed rows"};
383-
390+
384391
// bit packed ITS cluster map
385392
const AxisSpec axisITSCluMap{(int)128, -0.5f, +127.5f, "Packed ITS map"};
386393
const AxisSpec axisRadius{(int)160, 0.0f, +80.0f, "Radius (cm)"};
@@ -432,7 +439,7 @@ struct straRecoStudy {
432439
histos.add("h2dOmegaMinusQADCABachToPV", "h2dOmegaMinusQADCABachToPV", kTH2F, {axisVsPtCoarse, axisDCA});
433440
histos.add("h2dOmegaMinusQADCACascToPV", "h2dOmegaMinusQADCACascToPV", kTH2F, {axisVsPtCoarse, axisDCAWD});
434441
histos.add("h2dOmegaMinusQAPointingAngle", "h2dOmegaMinusQAPointingAngle", kTH2F, {axisVsPtCoarse, axisPA});
435-
442+
436443
histos.add("h2dITSCluMap_V0Positive", "h2dITSCluMap_V0Positive", kTH2D, {axisITSCluMap, axisRadius});
437444
histos.add("h2dITSCluMap_V0Negative", "h2dITSCluMap_V0Negative", kTH2D, {axisITSCluMap, axisRadius});
438445
histos.add("h2dITSCluMap_CascPositive", "h2dITSCluMap_CascPositive", kTH2D, {axisITSCluMap, axisRadius});
@@ -543,7 +550,7 @@ struct straRecoStudy {
543550
resetCounters();
544551
}
545552
PROCESS_SWITCH(straRecoStudy, processV0, "Regular V0 analysis", true);
546-
553+
547554
void processV0RealData(soa::Join<aod::Collisions, aod::EvSels>::iterator const& collision, aod::V0Datas const& v0tables, aod::CascDataExt const& Cascades, TracksCompleteIU const& tracks, aod::V0sLinked const&)
548555
{
549556
evselstats[kEvSelAll]++;
@@ -702,7 +709,7 @@ struct straRecoStudy {
702709
auto v0 = v0index.v0Data(); // de-reference index to correct v0data in case it exists
703710
auto posPartTrack = v0.posTrack_as<TracksCompleteIU>();
704711
auto negPartTrack = v0.negTrack_as<TracksCompleteIU>();
705-
712+
706713
histos.fill(HIST("h2dITSCluMap_CascPositive"), (float)posPartTrack.itsClusterMap(), casc.v0radius());
707714
histos.fill(HIST("h2dITSCluMap_CascNegative"), (float)negPartTrack.itsClusterMap(), casc.v0radius());
708715
histos.fill(HIST("h2dITSCluMap_CascBachelor"), (float)bachPartTrack.itsClusterMap(), casc.cascradius());
@@ -747,7 +754,7 @@ struct straRecoStudy {
747754
}
748755
}
749756
PROCESS_SWITCH(straRecoStudy, processCascadeRealData, "Regular cascade analysis, real data", false);
750-
757+
751758
void processGeneratedReconstructible(soa::Filtered<RecoedMCCollisions>::iterator const& collision, aod::McParticles const& mcParticles)
752759
{
753760
// check if collision successfully reconstructed

0 commit comments

Comments
 (0)