Skip to content

[PWGCF] Add pt spectra hist for QA #15215

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

Merged
victor-gonzalez merged 3 commits into from
Mar 2, 2026
+94 −2
Merged

[PWGCF] Add pt spectra hist for QA #15215

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
4cc6a45
[PWGCF] Add pt spectra hist for QA
huinaibing Mar 2, 2026
8a2e5bb
Please consider the following formatting changes
alibuild Mar 2, 2026
4c66cde
Merge pull request #6 from alibuild/alibot-cleanup-15215
huinaibing Mar 2, 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
96 changes: 94 additions & 2 deletions PWGCF/Flow/Tasks/pidFlowPtCorr.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -137,6 +137,8 @@ struct PidFlowPtCorr {
O2_DEFINE_CONFIGURABLE(cfgUsePtCentNUECorr, bool, true, "do NUA NUE, Using Eff(pt, cent) to do NUE")
O2_DEFINE_CONFIGURABLE(cfgDebugMyCode, bool, false, "output some graph for code debug")

O2_DEFINE_CONFIGURABLE(cfgOutPutPtSpectra, bool, false, "output pt spectra for data, MC and RECO")

/**
* @brief cfg for PID pt range
* @details default datas are from run2, note that separate pi-k and k-p needs to use difference pt range
Expand Down Expand Up @@ -197,7 +199,7 @@ struct PidFlowPtCorr {
Filter particleFilter = (nabs(aod::mcparticle::eta) < trkQualityOpts.cfgCutEta.value);
using FilteredMcParticles = soa::Filtered<aod::McParticles>;

using FilteredTracksWithMCLabel = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TrackSelection, aod::TracksDCA, aod::McTrackLabels>>;
using FilteredTracksWithMCLabel = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TrackSelection, aod::TracksDCA, aod::McTrackLabels, TracksPID>>;
using FilteredCollisionsWithMCLabel = soa::Filtered<soa::Join<AodCollisions, aod::EvSels, aod::McCollisionLabels>>;
// end using and filter

Expand Down Expand Up @@ -382,7 +384,7 @@ struct PidFlowPtCorr {
TH1* hPhiRunByRunBefore = registry.add<TH1>(Form("RunByRunQA%d/hPhiBefore", oneRun), "", {HistType::kTH1D, {cfgaxisPhi}}).get();
TH1* hPhiRunByRunAfter = registry.add<TH1>(Form("RunByRunQA%d/hPhiAfter", oneRun), "", {HistType::kTH1D, {cfgaxisPhi}}).get();

TH2* hITSnclsVsPhi = registry.add<TH2>(Form("RunByRunQA%d/hITSnclsVsPhi", oneRun), "", {HistType::kTH2D, {cfgaxisPhi, {7, 0, 7}}}).get();
TH2* hITSnclsVsPhi = registry.add<TH2>(Form("RunByRunQA%d/hITSnclsVsPhi", oneRun), "", {HistType::kTH2D, {cfgaxisPhi, {8, 0, 8}}}).get();
TH2* hITSChi2VsPhi = registry.add<TH2>(Form("RunByRunQA%d/hITSChi2VsPhi", oneRun), "", {HistType::kTH2D, {cfgaxisPhi, {100, 0, 10}}}).get();

qaHistVector.emplace_back(qaHist(hPhiRunByRunBefore, hPhiRunByRunAfter, hITSnclsVsPhi, hITSChi2VsPhi));
Expand All @@ -394,6 +396,16 @@ struct PidFlowPtCorr {
// hist for NUE correction
registry.add("correction/hPtCentMcRec", "", {HistType::kTH2D, {cfgaxisPt, axisMultiplicity}});
registry.add("correction/hPtCentMcGen", "", {HistType::kTH2D, {cfgaxisPt, axisMultiplicity}});

// hist for Pi eff
registry.add("correction/hPtCentMcRecPi", "", {HistType::kTH2D, {cfgaxisPt, axisMultiplicity}});
registry.add("correction/hPtCentMcGenPi", "", {HistType::kTH2D, {cfgaxisPt, axisMultiplicity}});
// hist for Ka eff
registry.add("correction/hPtCentMcRecKa", "", {HistType::kTH2D, {cfgaxisPt, axisMultiplicity}});
registry.add("correction/hPtCentMcGenKa", "", {HistType::kTH2D, {cfgaxisPt, axisMultiplicity}});
// hist for Pr eff
registry.add("correction/hPtCentMcRecPr", "", {HistType::kTH2D, {cfgaxisPt, axisMultiplicity}});
registry.add("correction/hPtCentMcGenPr", "", {HistType::kTH2D, {cfgaxisPt, axisMultiplicity}});
} // cfgoutputMC

// debug hists
Expand All @@ -406,6 +418,26 @@ struct PidFlowPtCorr {
}
} // cfgdebugmycode

if (cfgOutPutPtSpectra) {
registry.add("ptSpectra/hPtCentData", "", {HistType::kTH2D, {cfgaxisPt, axisMultiplicity}});
registry.add("ptSpectra/hCentEventCountData", "", {HistType::kTH1D, {axisMultiplicity}});
registry.add("ptSpectra/hCentEventCountMcGen", "", {HistType::kTH1D, {axisMultiplicity}});
registry.add("ptSpectra/hCentEventCountMcRec", "", {HistType::kTH1D, {axisMultiplicity}});

// (PosEta)(PosCh)
registry.add("ptSpectra/hPtCentDataPosEtaPosCh", "", {HistType::kTH2D, {cfgaxisPt, axisMultiplicity}});

// (PosEta)(NegCh)
registry.add("ptSpectra/hPtCentDataPosEtaNegCh", "", {HistType::kTH2D, {cfgaxisPt, axisMultiplicity}});

// (NegEta(PosCh)
registry.add("ptSpectra/hPtCentDataNegEtaPosCh", "", {HistType::kTH2D, {cfgaxisPt, axisMultiplicity}});

// (NegEta)(NegCh)
registry.add("ptSpectra/hPtCentDataNegEtaNegCh", "", {HistType::kTH2D, {cfgaxisPt, axisMultiplicity}});

} // cfgoutputptspectra

if (cfgOutputrunbyrun) {
// hist for NUA
registry.add("correction/hRunNumberPhiEtaVertex", "", {HistType::kTHnSparseF, {cfgaxisRun, cfgaxisPhi, cfgaxisEta, cfgaxisVertex}});
Expand Down Expand Up @@ -1262,6 +1294,11 @@ struct PidFlowPtCorr {
return;
// end collision cut

// pt spectra
if (cfgOutPutPtSpectra) {
registry.fill(HIST("ptSpectra/hCentEventCountData"), cent);
} // cfgOutPutPtSpectra

// correction
loadCorrections(bc.timestamp());
float vtxz = collision.posZ();
Expand Down Expand Up @@ -1388,6 +1425,27 @@ struct PidFlowPtCorr {
registry.fill(HIST("hPt"), track.pt());
// end fill QA hist

// pt spectra
if (cfgOutPutPtSpectra) {
registry.fill(HIST("ptSpectra/hPtCentData"), track.pt(), cent);

// region 1 +eta +ch
if (track.eta() > 0.05 && track.sign() > 0)
registry.fill(HIST("ptSpectra/hPtCentDataPosEtaPosCh"), track.pt(), cent);

// region 2 +eta -ch
if (track.eta() > 0.05 && track.sign() < 0)
registry.fill(HIST("ptSpectra/hPtCentDataPosEtaNegCh"), track.pt(), cent);

// region 3 -eta +ch
if (track.eta() < -0.05 && track.sign() > 0)
registry.fill(HIST("ptSpectra/hPtCentDataNegEtaPosCh"), track.pt(), cent);

// region 4 -eta -ch
if (track.eta() < -0.05 && track.sign() < 0)
registry.fill(HIST("ptSpectra/hPtCentDataNegEtaNegCh"), track.pt(), cent);
}

// track pt cut
if (!((track.pt() > trkQualityOpts.cfgCutPtMin.value) && (track.pt() < trkQualityOpts.cfgCutPtMax.value)))
continue;
Expand Down Expand Up @@ -1694,6 +1752,10 @@ struct PidFlowPtCorr {
return;
// end init && cut

if (cfgOutPutPtSpectra) {
registry.fill(HIST("ptSpectra/hCentEventCountMcRec"), cent);
} // cfgoutputptspectra

// loop tracks
for (const auto& track : tracks) {
// track cut
Expand All @@ -1706,7 +1768,20 @@ struct PidFlowPtCorr {
auto mcParticle = track.mcParticle();
// fill graph
if (particleSelected(mcParticle)) {
// graph for all particles
registry.fill(HIST("correction/hPtCentMcRec"), track.pt(), cent);

// identify particle and fill graph
if (isPion(track)) {
registry.fill(HIST("correction/hPtCentMcRecPi"), track.pt(), cent);
}
if (isKaon(track)) {
registry.fill(HIST("correction/hPtCentMcRecKa"), track.pt(), cent);
}
if (isProton(track)) {
registry.fill(HIST("correction/hPtCentMcRecPr"), track.pt(), cent);
}
// end identify particle and fill graph
}
// end fill graph
}
Expand Down Expand Up @@ -1755,11 +1830,28 @@ struct PidFlowPtCorr {
continue;
// end collision cut

if (cfgOutPutPtSpectra) {
registry.fill(HIST("ptSpectra/hCentEventCountMcGen"), cent);
} // cfgoutputptspectra

// loop mc particles
for (const auto& mcParticle : mcParticles) {
// fill graph
if (particleSelected(mcParticle)) {
// graph for all particles
registry.fill(HIST("correction/hPtCentMcGen"), mcParticle.pt(), cent);

// identify particle and fill graph
if (std::abs(mcParticle.pdgCode()) == PDG_t::kPiPlus) {
registry.fill(HIST("correction/hPtCentMcGenPi"), mcParticle.pt(), cent);
}
if (std::abs(mcParticle.pdgCode()) == PDG_t::kKPlus) {
registry.fill(HIST("correction/hPtCentMcGenKa"), mcParticle.pt(), cent);
}
if (std::abs(mcParticle.pdgCode()) == PDG_t::kProton) {
registry.fill(HIST("correction/hPtCentMcGenPr"), mcParticle.pt(), cent);
}
// end identify particle and fill graph
}
// end fill graph
}
Expand Down
Loading