From 0a96dcc2930b3c5d883ce8a2a8edd51df6392afc Mon Sep 17 00:00:00 2001
From: Pritam Chakraborty <47203359+prchakra@users.noreply.github.com>
Date: Mon, 20 Oct 2025 18:36:09 +0200
Subject: [PATCH] FemtoUniverse: Add flag and condition to select only
 primaries

---
 .../TableProducer/femtoUniverseProducerTask.cxx           | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/PWGCF/FemtoUniverse/TableProducer/femtoUniverseProducerTask.cxx b/PWGCF/FemtoUniverse/TableProducer/femtoUniverseProducerTask.cxx
index 19fb3fc50d0..e45c4e0084d 100644
--- a/PWGCF/FemtoUniverse/TableProducer/femtoUniverseProducerTask.cxx
+++ b/PWGCF/FemtoUniverse/TableProducer/femtoUniverseProducerTask.cxx
@@ -241,6 +241,7 @@ struct FemtoUniverseProducerTask {
     Configurable<float> confDcaXYCustom11FilterCut{"confDcaXYCustom11FilterCut", 0.004, "Value for [1] custom DCAxy cut -> |DCAxy| < [1] + [2]/pT"};
     Configurable<float> confDcaXYCustom12FilterCut{"confDcaXYCustom12FilterCut", 0.013, "Value for [2] custom DCAxy cut -> |DCAxy| < [1] + [2]/pT"};
     Configurable<bool> confIsApplyTrkCutMCTruth{"confIsApplyTrkCutMCTruth", false, "Apply eta, pT selection cut on MCTruth tracks "};
+    Configurable<bool> confIsOnlyPrimary{"confIsOnlyPrimary", false, "Select only primaries"};
   } ConfFilterCuts;
 
   Filter globalCutFilter = requireGlobalTrackInFilter();
@@ -1367,6 +1368,13 @@ struct FemtoUniverseProducerTask {
           continue;
         }
       }
+
+      if (ConfFilterCuts.confIsOnlyPrimary) {
+        if (!mc.isPhysicalPrimary()) {
+          return;
+        }
+      }
+
       std::vector<int> childIDs = {0, 0};
       outputParts(outputCollision.lastIndex(),
                   mc.pt(),