Skip to content

[PWGLF] Improvements in the track criteria selection #15702

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

Open
lucasjsilva wants to merge 2 commits into
base: master
Choose a base branch
from
+65 −54
Open

[PWGLF] Improvements in the track criteria selection #15702

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
5118cf0
Improvements regarding track criteria selection
lucasjsilva Apr 9, 2026
1d99fa5
Improvements in the track criteria selection and exclusion of unused …
lucasjsilva Apr 9, 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
119 changes: 65 additions & 54 deletions PWGLF/Tasks/GlobalEventProperties/studyPnch.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -76,6 +76,7 @@ AxisSpec axisEvent{10, 0.5, 10.5, "#Event", "EventAxis"};
AxisSpec axisVtxZ{40, -20, 20, "Vertex Z", "VzAxis"};
AxisSpec axisEta{40, -2, 2, "#eta", "EtaAxis"};
AxisSpec axisPhi{629, 0, o2::constants::math::TwoPI, "#phi"};
AxisSpec axisCollSel{5, 0.5, 5.5, "#Event", "CollSelAxis"};
auto static constexpr kMinCharge = 3.f;

struct StudyPnch {
Expand All @@ -97,6 +98,7 @@ struct StudyPnch {
ConfigurableAxis ft0aMultHistBin{"ft0aMultHistBin", {501, -0.5, 500.5}, ""};
ConfigurableAxis ft0cMultHistBin{"ft0cMultHistBin", {501, -0.5, 500.5}, ""};
ConfigurableAxis ptHistBin{"ptHistBin", {200, 0., 20.}, ""};
ConfigurableAxis binsDCA{"binsDCA", {500, -10.0f, 10.0f}, ""};
ConfigurableAxis countNumberTracks{"countNumberTracks", {10, -0.5, 9.5}, ""};

Configurable<bool> isApplyTFcut{"isApplyTFcut", true, "Enable TimeFrameBorder cut"};
Expand All @@ -105,6 +107,8 @@ struct StudyPnch {
Configurable<bool> isApplyInelgt0{"isApplyInelgt0", false, "Enable INEL > 0 condition"};
Configurable<bool> isApplyExtraPhiCut{"isApplyExtraPhiCut", false, "Enable extra phi cut"};
Configurable<bool> isApplyTVX{"isApplyTVX", false, "Enable TVX trigger sel"};
Configurable<bool> isApplyCheckID{"isApplyCheckID", true, "Select Tracks evaluating Collision ID"};
Configurable<bool> isApplyDuplicatedTrack{"isApplyDuplicatedTrack", true, "Select tracks that are not duplicated"};

void init(InitContext const&)
{
Expand All @@ -115,6 +119,7 @@ struct StudyPnch {
AxisSpec axisFt0cMult = {ft0cMultHistBin, "ft0c", "FT0CMultAxis"};
AxisSpec axisPt = {ptHistBin, "pT", "pTAxis"};
AxisSpec axisCountNumberTracks = {countNumberTracks, "Count", "CountAxis"};
AxisSpec dcaAxis = {binsDCA, "DCA vs PV"};

histos.add("EventHist", "EventHist", kTH1D, {axisEvent}, false);
histos.add("VtxZHist", "VtxZHist", kTH1D, {axisVtxZ}, false);
Expand All @@ -129,8 +134,19 @@ struct StudyPnch {
x->SetBinLabel(6, "INEL > 0");
x->SetBinLabel(7, "|vz| < 10");

if (doprocessData || doprocessCorrelation || doprocessMonteCarlo || doprocessModifiedMonteCarlo) {
histos.add("SelCollsHist", "SelCollsHist", kTH1D, {axisCollSel}, false);
auto hstat_colls = histos.get<TH1>(HIST("SelCollsHist"));
auto* xColls = hstat_colls->GetXaxis();
xColls->SetBinLabel(1, "All collisions");
xColls->SetBinLabel(2, "Best Collision Selection");
xColls->SetBinLabel(3, "Has MC Collision Selection");

if (doprocessData || doprocessCorrelation || doprocessMonteCarlo) {
histos.add("PhiVsEtaHist", "PhiVsEtaHist", kTH2F, {axisPhi, axisEta}, false);
histos.add("EtaHist", "EtaHist", kTH1D, {axisEta}, false);
histos.add("PhiHist", "PhiHist", kTH1D, {axisPhi}, false);
histos.add("hdcaxy", "dca to pv in the xy plane", kTH1D, {dcaAxis}, false);
histos.add("hdcaz", "dca to pv in the z axis", kTH1D, {dcaAxis}, false);
}
if (doprocessData) {
histos.add("hMultiplicityData", "hMultiplicityData", kTH1F, {axisMult}, true);
Expand All @@ -143,17 +159,14 @@ struct StudyPnch {
histos.add("NPVtracks_vs_GlobalMult", "NPVtracks_vs_GlobalMult", kTH2F, {axisPV, axisMult}, true);
}
if (doprocessMonteCarlo) {
histos.add("PhiVsEtaGenHist", "PhiVsEtaGenHist", kTH2F, {axisPhi, axisEta}, false);
histos.add("EtaGenHist", "EtaGenHist", kTH1D, {axisEta}, false);
histos.add("PhiGenHist", "PhiGenHist", kTH1D, {axisPhi}, false);
histos.add("hMultiplicityMCrec", "hMultiplicityMCrec", kTH1F, {axisMult}, true);
histos.add("hMultiplicityMCgen", "hMultiplicityMCgen", kTH1F, {axisMult}, true);
histos.add("hResponseMatrix", "hResponseMatrix", kTH2F, {axisMult, axisMult}, true);
histos.add("hCountNTracks", "hCountNTracks", kTH1F, {axisCountNumberTracks}, true);
}
if (doprocessModifiedMonteCarlo) {
histos.add("hMultiplicityMCrecMod", "hMultiplicityMCrecMod", kTH1F, {axisMult}, true);
histos.add("hMultiplicityMCgenMod", "hMultiplicityMCgenMod", kTH1F, {axisMult}, true);
histos.add("hResponseMatrixMod", "hResponseMatrixMod", kTH2F, {axisMult, axisMult}, true);
histos.add("hCountNTracksMod", "hCountNTracksMod", kTH1F, {axisCountNumberTracks}, true);
}
if (doprocessEvtLossSigLossMC) {
histos.add("MCEventHist", "MCEventHist", kTH1F, {axisEvent}, false);
auto hstat = histos.get<TH1>(HIST("MCEventHist"));
Expand Down Expand Up @@ -242,6 +255,10 @@ struct StudyPnch {
if (!isTrackSelected(track)) {
continue;
}
histos.fill(HIST("hdcaxy"), track.dcaXY());
histos.fill(HIST("hdcaz"), track.dcaZ());
histos.fill(HIST("EtaHist"), track.eta());
histos.fill(HIST("PhiHist"), track.phi());
histos.fill(HIST("PhiVsEtaHist"), track.phi(), track.eta());
nTrk++;
}
Expand All @@ -256,11 +273,12 @@ struct StudyPnch {
if (!isGenTrackSelected(track)) {
continue;
}
if (track.mcCollisionId() != McCol.globalIndex()) {
if (track.mcCollisionId() != McCol.mcCollisionId()) {
continue;
}

histos.fill(HIST("PhiVsEtaHist"), track.phi(), track.eta());
histos.fill(HIST("EtaGenHist"), track.eta());
histos.fill(HIST("PhiGenHist"), track.phi());
histos.fill(HIST("PhiVsEtaGenHist"), track.phi(), track.eta());
nTrk++;
}
return nTrk;
Expand All @@ -270,30 +288,28 @@ struct StudyPnch {
int countNTracksMcCol(countTrk const& tracks, McColType const& McCol)
{
auto nTrk = 0;
std::unordered_map<int, int> recoFrequencies; // Map that stores globalIndex and the times it appears
std::vector<int> mcRecIDs;
for (const auto& track : tracks) {
if (!isTrackSelected(track)) {
continue;
}
if (track.has_mcParticle()) {
auto particle = track.mcParticle();
if (particle.mcCollisionId() != McCol.mcCollisionId()) {
if (isApplyCheckID && particle.mcCollisionId() != McCol.mcCollisionId()) {
continue;
}
auto globalIndex = particle.globalIndex();
recoFrequencies[globalIndex]++; // Increment the count for this globalIndex
}
histos.fill(HIST("PhiVsEtaHist"), track.phi(), track.eta());
}
// Loop to fill the histogram without cloned tracks
for (const auto& [globalIndex, frequency] : recoFrequencies) {
histos.fill(HIST("hCountNTracks"), frequency);
// Fill histogram with not cloned tracks
if (frequency == 1) {
if (isApplyDuplicatedTrack && find(mcRecIDs.begin(), mcRecIDs.end(), particle.globalIndex()) != mcRecIDs.end()) {
continue;
}
mcRecIDs.push_back(particle.globalIndex());
nTrk++;
}
histos.fill(HIST("hdcaxy"), track.dcaXY());
histos.fill(HIST("hdcaz"), track.dcaZ());
histos.fill(HIST("EtaHist"), track.eta());
histos.fill(HIST("PhiHist"), track.phi());
histos.fill(HIST("PhiVsEtaHist"), track.phi(), track.eta());
}
// return recoFrequencies;
return nTrk;
}

Expand Down Expand Up @@ -328,47 +344,31 @@ struct StudyPnch {
histos.fill(HIST("NPVtracks_vs_GlobalMult"), cols.multNTracksPV(), mult);
}

void processMonteCarlo(ColMCTrueTable::iterator const& mcCollision, ColMCRecTable const& RecCols, TrackMCTrueTable const& GenParticles, FilTrackMCRecTable const& RecTracks)
{
for (const auto& RecCol : RecCols) {
if (!isEventSelected(RecCol)) {
continue;
}
auto recTracksPart = RecTracks.sliceBy(perCollision, RecCol.globalIndex());
auto multrec = countNTracksMcCol(recTracksPart, RecCol);
if (multrec > 0) {
histos.fill(HIST("hMultiplicityMCrec"), multrec);
}
auto multgen = countGenTracks(GenParticles, mcCollision);
if (multgen > 0 && multrec > 0) {
histos.fill(HIST("hMultiplicityMCgen"), multgen);
histos.fill(HIST("hResponseMatrix"), multrec, multgen);
}
}
}

void processModifiedMonteCarlo(soa::Join<aod::McCollisions, aod::McCollsExtra>::iterator const& mcCollision, ColMCRecTable const& RecCols, TrackMCTrueTable const& GenParticles, FilTrackMCRecTable const& RecTracks)
void processMonteCarlo(soa::Join<aod::McCollisions, aod::McCollsExtra>::iterator const& mcCollision, ColMCRecTable const& RecCols, TrackMCTrueTable const& GenParticles, FilTrackMCRecTable const& RecTracks)
{
for (const auto& RecCol : RecCols) {
if (!isEventSelected(RecCol)) {
continue;
}
histos.fill(HIST("SelCollsHist"), 1);
// Evaluation of reconstructed collisions with more than 1 contributor
if (RecCol.globalIndex() != mcCollision.bestCollisionIndex()) {
continue;
}
histos.fill(HIST("SelCollsHist"), 2);
if (!RecCol.has_mcCollision()) {
continue;
}
histos.fill(HIST("SelCollsHist"), 3);
auto recTracksPart = RecTracks.sliceBy(perCollision, RecCol.globalIndex());
auto multrec = countNTracksMcCol(recTracksPart, RecCol);
if (multrec > 0) {
histos.fill(HIST("hMultiplicityMCrecMod"), multrec);
histos.fill(HIST("hMultiplicityMCrec"), multrec);
}
auto multgen = countGenTracks(GenParticles, mcCollision);
auto multgen = countGenTracks(GenParticles, RecCol);
if (multgen > 0 && multrec > 0) {
histos.fill(HIST("hMultiplicityMCgenMod"), multgen);
histos.fill(HIST("hResponseMatrixMod"), multrec, multgen);
histos.fill(HIST("hMultiplicityMCgen"), multgen);
histos.fill(HIST("hResponseMatrix"), multrec, multgen);
}
}
}
Expand All @@ -386,9 +386,15 @@ struct StudyPnch {
}
// All generated events
histos.fill(HIST("MCEventHist"), 1);
auto multAll = countGenTracks(GenParticles, mcCollision);
if (multAll > 0) {
histos.fill(HIST("hMultiplicityMCgenAll"), multAll);
auto nTrk_multAll = 0;
for (const auto& GenParticle : GenParticles) {
if (!isGenTrackSelected(GenParticle)) {
continue;
}
nTrk_multAll++;
}
if (nTrk_multAll > 0) {
histos.fill(HIST("hMultiplicityMCgenAll"), nTrk_multAll);
}

bool atLeastOne = false;
Expand All @@ -407,17 +413,22 @@ struct StudyPnch {

if (atLeastOne) {
histos.fill(HIST("MCEventHist"), 2);
auto multSel = countGenTracks(GenParticles, mcCollision);
if (multSel > 0) {
histos.fill(HIST("hMultiplicityMCgenSel"), multSel);
auto nTrk_multSel = 0;
for (const auto& GenParticle : GenParticles) {
if (!isGenTrackSelected(GenParticle)) {
continue;
}
nTrk_multSel++;
}
if (nTrk_multSel > 0) {
histos.fill(HIST("hMultiplicityMCgenSel"), nTrk_multSel);
}
}
}

PROCESS_SWITCH(StudyPnch, processData, "process data CentFT0C", false);
PROCESS_SWITCH(StudyPnch, processCorrelation, "do correlation study in data", false);
PROCESS_SWITCH(StudyPnch, processMonteCarlo, "process MC CentFT0C", false);
PROCESS_SWITCH(StudyPnch, processModifiedMonteCarlo, "process MC CentFT0C", false);
PROCESS_SWITCH(StudyPnch, processEvtLossSigLossMC, "process Signal Loss, Event Loss", false);
};

Expand Down
Loading