Skip to content

[PWGLF,Tutorial] [PWGMM] Modified the tutorial code for learning #14424

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
dibakardhar07 wants to merge 2 commits into from
+142 −8
Closed

[PWGLF,Tutorial] [PWGMM] Modified the tutorial code for learning #14424

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
6a31cbe
added some function in the example task code for learning o2
Jan 8, 2026
d2ef35d
dummy commit
Jan 8, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
98 changes: 93 additions & 5 deletions Tutorials/PWGMM/myExampleTask.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,33 +15,121 @@

#include "Framework/runDataProcessing.h"
#include "Framework/AnalysisTask.h"
#include "Common/DataModel/TrackSelectionTables.h"
#include "Framework/ASoAHelpers.h"

#include <TPDGCode.h> // for PDG codes

using namespace o2;
using namespace o2::framework;
using namespace o2::framework::expressions;

struct myExampleTask {
// Histogram registry: an object to hold your histograms
HistogramRegistry histos{"histos", {}, OutputObjHandlingPolicy::AnalysisObject};

Configurable<int> nBinsPt{"nBinsPt", 100, "N bins in pT histo"};
Configurable<float> maxDCAxy{"maxDCAxy", 0.2, "max DCAxy (in cm)"};
Configurable<float> minTPCCrossedRows{"minTPCCrossedRows", 70, "min crossed rows in the TPC"};

Filter trackDCA = nabs(aod::track::dcaXY) < maxDCAxy;

//This is an example of a convenient declaration of "using"
using myCompleteTracksMC = soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::McTrackLabels>;
using myCompleteTracks = soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA>;
using myFilteredTracksMC = soa::Filtered<myCompleteTracksMC>;
using myFilteredTracks = soa::Filtered<myCompleteTracks>;

Preslice<aod::Tracks> perCollision = aod::track::collisionId;

void init(InitContext const&)
{
// define axes you want to use
const AxisSpec axisCounter{1, 0, +1, ""};
const AxisSpec axisEta{30, -1.5, +1.5, "#eta"};

const AxisSpec axisPt{nBinsPt, 0, 10, "p_{T}"};
const AxisSpec axisDeltaPt{100, -1.0, +1.0, "#Delta(p_{T})"};
// create histograms
histos.add("eventCounter", "eventCounter", kTH1F, {axisCounter});
histos.add("etaHistogram", "etaHistogram", kTH1F, {axisEta});
histos.add("ptHistogram", "ptHistogram", kTH1F, {axisPt});
histos.add("ptResolution", "ptResolution", kTH2F, {axisPt, axisDeltaPt});

histos.add("ptHistogramPion", "ptHistogramPion", kTH1F, {axisPt});
histos.add("ptHistogramKaon", "ptHistogramKaon", kTH1F, {axisPt});
histos.add("ptHistogramProton", "ptHistogramProton", kTH1F, {axisPt});
histos.add("ptGeneratedPion", "ptGeneratedPion", kTH1F, {axisPt});
histos.add("ptGeneratedKaon", "ptGeneratedKaon", kTH1F, {axisPt});
histos.add("ptGeneratedProton", "ptGeneratedProton", kTH1F, {axisPt});

histos.add("numberOfRecoCollisions", "numberOfRecoCollisions", kTH1F, {{10,-0.5f, 9.5f}});
histos.add("multiplicityCorrelation", "multiplicityCorrelations", kTH2F, {{100, -0.5f, 99.5f}, {100,-0.5f, 99.5f}});
}

void process(aod::TracksIU const& tracks)
{
for (auto& track : tracks) {
template <bool fillResolution = true, typename TCollision, typename TTracks>
void processStuff(TCollision const& collision, TTracks const& tracks) {
histos.fill(HIST("eventCounter"), 0.5f);
for (const auto& track : tracks) {
if( track.tpcNClsCrossedRows() < minTPCCrossedRows ) continue; //badly tracked
histos.fill(HIST("etaHistogram"), track.eta());
histos.fill(HIST("ptHistogram"), track.pt());

// this part only done if dealing with MC
if constexpr (requires { track.has_mcParticle(); }) { // does the getter exist?

Check failure on line 78 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / PR formatting / whitespace

Trailing spaces 

Remove the trailing spaces at the end of the line.
if(track.has_mcParticle()){ // is the return 'true'? -> N.B. different question!

Check failure on line 79 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / PR formatting / whitespace

Trailing spaces 

Remove the trailing spaces at the end of the line.
auto mcParticle = track.mcParticle();

Check failure on line 80 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / PR formatting / whitespace

Trailing spaces 

Remove the trailing spaces at the end of the line.
if constexpr (fillResolution){ // compile-time check
histos.fill(HIST("ptResolution"), track.pt(), track.pt() - mcParticle.pt());
}
if(mcParticle.isPhysicalPrimary() && fabs(mcParticle.y())<0.5){ // do this in the context of the track ! (context matters!!!)

Check failure on line 84 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
if(abs(mcParticle.pdgCode())==kPiPlus) histos.fill(HIST("ptHistogramPion"), mcParticle.pt());

Check failure on line 85 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
if(abs(mcParticle.pdgCode())==kKPlus) histos.fill(HIST("ptHistogramKaon"), mcParticle.pt());

Check failure on line 86 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
if(abs(mcParticle.pdgCode())==kProton) histos.fill(HIST("ptHistogramProton"), mcParticle.pt());

Check failure on line 87 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
}
}
}
}
}

void processRecoInData(aod::Collision const& collision, myFilteredTracks const& tracks)
{
processStuff(collision, tracks);

Check failure on line 96 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / PR formatting / whitespace

Trailing spaces 

Remove the trailing spaces at the end of the line.
}
PROCESS_SWITCH(myExampleTask, processRecoInData, "process reconstructed information", true);

void processRecoInSim(aod::Collision const& collision, myFilteredTracksMC const& tracks, aod::McParticles const&)
{
processStuff(collision, tracks);

Check failure on line 102 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / PR formatting / whitespace

Trailing spaces 

Remove the trailing spaces at the end of the line.
}
PROCESS_SWITCH(myExampleTask, processRecoInSim, "process reconstructed information", false);

void processSim(aod::McCollision const& mcCollision, soa::SmallGroups<soa::Join<aod::McCollisionLabels, aod::Collisions>> const& collisions, aod::McParticles const& mcParticles, myFilteredTracksMC const& tracks)
{
histos.fill(HIST("numberOfRecoCollisions"), collisions.size()); // number of times coll was reco-ed

//Now loop over each time this collision has been reconstructed and aggregate tracks
std::vector<int> numberOfTracks;
for (auto& collision : collisions) {

Check failure on line 112 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[const-ref-in-for-loop] 

Use constant references for non-modified iterators in range-based for loops.
auto groupedTracks = tracks.sliceBy(perCollision, collision.globalIndex());
// size of grouped tracks may help in understanding why event was split!
numberOfTracks.emplace_back(groupedTracks.size());
}
if( collisions.size() == 2 ) histos.fill(HIST("multiplicityCorrelation"), numberOfTracks[0], numberOfTracks[1]);

Check failure on line 117 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[magic-number] 

Avoid magic numbers in expressions. Assign the value to a clearly named variable or constant.

//Loop over particles in this mcCollision (first argument of process: iterator)
for (const auto& mcParticle : mcParticles) {
if(mcParticle.isPhysicalPrimary() && fabs(mcParticle.y())<0.5){ // do this in the context of the MC loop ! (context matters!!!)

Check failure on line 121 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
if(abs(mcParticle.pdgCode())==kPiPlus) histos.fill(HIST("ptGeneratedPion"), mcParticle.pt());

Check failure on line 122 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
if(abs(mcParticle.pdgCode())==kKPlus) histos.fill(HIST("ptGeneratedKaon"), mcParticle.pt());

Check failure on line 123 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
if(abs(mcParticle.pdgCode())==kProton) histos.fill(HIST("ptGeneratedProton"), mcParticle.pt());

Check failure on line 124 in Tutorials/PWGMM/myExampleTask.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
}
}
}
PROCESS_SWITCH(myExampleTask, processSim, "process pure simulation information", false);
};

WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
{
return WorkflowSpec{
adaptAnalysisTask<myExampleTask>(cfgc)};
}
}
52 changes: 49 additions & 3 deletions Tutorials/Skimming/derivedBasicConsumer.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,24 +40,70 @@ struct DerivedBasicConsumer {
return deltaPhi;
}

Configurable<float> associatedMinPt{"associatedMinPt", 4.0f, "NSassociatedMinPt"};
Configurable<float> associatedMaxPt{"associatedMaxPt", 6.0f, "associatedMaxPt"};
Configurable<float> triggerMinPt{"triggerMinPt", 6.0f, "triggerMinPt"};
ConfigurableAxis axisPt{"axisPt", {200,0.0f,20.0f}, "pt axis"};

SliceCache cache;

// define partitions
Partition<aod::DrTracks> associatedTracks = aod::exampleTrackSpace::pt < associatedMaxPt && aod::exampleTrackSpace::pt > associatedMinPt;
Partition<aod::DrTracks> triggerTracks = aod::exampleTrackSpace::pt > triggerMinPt;

Filter collZfilter = nabs(aod::collision::posZ) < 10.0f;

// Histogram registry: an object to hold your histograms
HistogramRegistry histos{"histos", {}, OutputObjHandlingPolicy::AnalysisObject};

void init(InitContext const&)
{
// define axes you want to use
const AxisSpec axisCounter{1, 0, +1, ""};
histos.add("eventCounter", "eventCounter", kTH1F, {axisCounter});
const AxisSpec axisPVz{300, -15.0f, +15.0f, ""};
const AxisSpec axisDeltaPhi{100, -0.5*o2::constants::math::PI, +1.5*o2::constants::math::PI, "#Delta#phi"};
const AxisSpec axisDeltaEta{100, -1.0, +1.0, "#Delta#eta"};

histos.add("eventCounter", "eventCounter", kTH1D, {axisCounter});
histos.add("hEventPVz", "hEventPVz", kTH1D, {axisPVz});

histos.add("ptAssoHistogram", "ptAssoHistogram", kTH1D, {axisPt});
histos.add("ptTrigHistogram", "ptTrigHistogram", kTH1D, {axisPt});

histos.add("correlationFunction", "correlationFunction", kTH1D, {axisDeltaPhi});
histos.add("correlationFunctionO2", "correlationFunctionO2", kTH1D, {axisDeltaPhi});

histos.add("correlationFunction2d", "correlationFunction2d", kTH2F, {axisDeltaPhi, axisDeltaEta});
}

void process(aod::DrCollision const& /*collision*/)
void process(soa::Filtered<aod::DrCollisions>::iterator const& collision, aod::DrTracks const&)
{
histos.fill(HIST("eventCounter"), 0.5);
histos.fill(HIST("hEventPVz"), collision.posZ());

auto assoTracksThisCollision = associatedTracks->sliceByCached(aod::exampleTrackSpace::drCollisionId, collision.globalIndex(), cache);
auto trigTracksThisCollision = triggerTracks->sliceByCached(aod::exampleTrackSpace::drCollisionId, collision.globalIndex(), cache);

for (auto& track : assoTracksThisCollision)
histos.fill(HIST("ptAssoHistogram"), track.pt());
for (auto& track : trigTracksThisCollision)
histos.fill(HIST("ptTrigHistogram"), track.pt());

for (auto& trigger : trigTracksThisCollision){
for (auto& associated : assoTracksThisCollision){
histos.fill(HIST("correlationFunction"), ComputeDeltaPhi(trigger.phi(),associated.phi()));
}
}

for (auto& [trigger, associated] : combinations(o2::soa::CombinationsFullIndexPolicy(trigTracksThisCollision, assoTracksThisCollision))) {
histos.fill(HIST("correlationFunctionO2"), ComputeDeltaPhi(trigger.phi(),associated.phi()));
histos.fill(HIST("correlationFunction2d"), ComputeDeltaPhi(trigger.phi(),associated.phi()), trigger.eta() - associated.eta());
}
}
};

WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
{
WorkflowSpec workflow{adaptAnalysisTask<DerivedBasicConsumer>(cfgc, TaskName{"derived-basic-consumer"})};
return workflow;
}
}
Loading