Skip to content

[PWGCF] FemtoUniverse producer: Add option to fill only reconstructed MC truth collisions #14749

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
prchakra merged 1 commit into from
Feb 3, 2026
+12 −0
Merged

[PWGCF] FemtoUniverse producer: Add option to fill only reconstructed MC truth collisions #14749

Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
12 changes: 12 additions & 0 deletions PWGCF/FemtoUniverse/TableProducer/femtoUniverseProducerTask.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -135,6 +135,8 @@ struct FemtoUniverseProducerTask {
Configurable<bool> confStoreMCmothers{"confStoreMCmothers", false, "MC truth: Fill with not only primary particles and store mothers' PDG in tempFitVar."};
Configurable<bool> confFillCollExt{"confFillCollExt", false, "Option to fill collision extended table"};

Configurable<bool> confCollMCTruthOnlyReco{"confCollMCTruthOnlyReco", false, "Fill only MC truth collisions that were reconstructed and selected"};

/// Event filtering (used for v0-cascade analysis)
Configurable<std::string> zorroMask{"zorroMask", "", "zorro trigger class to select on (empty: none)"};

Expand Down Expand Up @@ -2601,6 +2603,7 @@ struct FemtoUniverseProducerTask {
{
// recos
std::set<int> recoMcIds;
std::set<int> mcCollisions;
for (const auto& col : collisions) {
auto groupedTracks = tracks.sliceBy(perCollisionTracks, col.globalIndex());
auto bc = col.bc_as<aod::BCsWithTimestamps>();
Expand All @@ -2616,6 +2619,7 @@ struct FemtoUniverseProducerTask {
}

if (colcheck) {
mcCollisions.insert(col.mcCollisionId());
fillCollisionsCentRun3ColExtra<true>(col, ir);
fillTracks<true>(groupedTracks);
}
Expand All @@ -2627,6 +2631,9 @@ struct FemtoUniverseProducerTask {

// truth
for (const auto& mccol : mccols) {
if (confCollMCTruthOnlyReco && !mcCollisions.contains(mccol.globalIndex())) {
continue;
}
auto groupedCollisions = collisions.sliceBy(recoCollsPerMCCollCentPbPb, mccol.globalIndex());
for (const auto& col : groupedCollisions) {
const auto colcheck = fillMCTruthCollisionsCentRun3(col); // fills the reco collisions for mc collision
Expand Down Expand Up @@ -2871,12 +2878,14 @@ struct FemtoUniverseProducerTask {

// MCReco
std::set<int> recoMcIds;
std::set<int> mcCollisions;
for (const auto& col : collisions) { // loop over collisions
auto groupedTracks = tracks.sliceBy(perCollisionTracks, col.globalIndex()); // slicing for tracks
auto groupedV0Parts = fullV0s.sliceBy(perCollisionV0s, col.globalIndex()); // slicing for V0
getMagneticFieldTesla(col.bc_as<aod::BCsWithTimestamps>());
const auto colcheck = fillCollisionsCentRun3<false>(col);
if (colcheck) {
mcCollisions.insert(col.mcCollisionId());
fillTracks<true>(groupedTracks);
fillV0<true>(col, groupedV0Parts, groupedTracks);
}
Expand All @@ -2888,6 +2897,9 @@ struct FemtoUniverseProducerTask {

// MCTruth
for (const auto& mccol : mccols) {
if (confCollMCTruthOnlyReco && !mcCollisions.contains(mccol.globalIndex())) {
continue;
}
auto groupedCollisions = collisions.sliceBy(recoCollsPerMCCollCentPbPb, mccol.globalIndex()); // slicing for MC collisions
auto groupedMCParticles = mcParticles.sliceBy(perMCCollision, mccol.globalIndex()); // slicing for MC particles
for (const auto& col : groupedCollisions) {
Expand Down
Loading