From 21b2b02f1cf9c24317307759188ee2c90c46fd34 Mon Sep 17 00:00:00 2001
From: GiorgioAlbertoLucia <giogioalberto@gmail.com>
Date: Fri, 29 May 2026 14:45:08 +0200
Subject: [PATCH] added feature so exclude Li4 for the kstar resolution

---
 PWGLF/TableProducer/Nuspex/he3HadronFemto.cxx | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/PWGLF/TableProducer/Nuspex/he3HadronFemto.cxx b/PWGLF/TableProducer/Nuspex/he3HadronFemto.cxx
index fa2bb7a33a1..b836a7293c8 100644
--- a/PWGLF/TableProducer/Nuspex/he3HadronFemto.cxx
+++ b/PWGLF/TableProducer/Nuspex/he3HadronFemto.cxx
@@ -143,7 +143,7 @@ enum ParticleFlags {
   kFromLi4 = BIT(1),         // from Li4 decay
   kFromHypertriton = BIT(2), // from hypertriton decay
   kFromMaterial = BIT(3),    // from material
-  kFromOtherDecays = BIT(4), // from other decays
+  kFromOtherDecays = BIT(4), // from other (weak) decays
 };
 
 constexpr double kItsParamsDefault[static_cast<int>(Species::kAllSpecies)][6] = {
@@ -291,7 +291,9 @@ struct he3HadronFemto {
 
   Configurable<bool> settingFillMultiplicity{"settingFillMultiplicity", false, "Fill multiplicity table"};
   Configurable<bool> settingFillQa{"settingFillQa", false, "Fill QA table"};
-  Configurable<bool> settingFillPrimariesAndMixedMc{"settingFillPrimariesAndMixedMc", false, "Fill primary MC tracks and mixed tracks (e.g. a primary track and one from Li4)"};
+  Configurable<bool> settingFillPrimariesMc{"settingFillPrimariesMc", false, "Fill primary MC tracks"};
+  Configurable<bool> settingFillLi4DaughtersMc{"settingFillLi4DaughtersMc", true, "Fill tracks from Li4 decays in MC"};
+  Configurable<bool> settingFillMixedMc{"settingFillMixedMc", false, "Fill mixed MC tracks (e.g. a primary track and one from Li4)"};
 
   // Zorro
   Configurable<bool> settingSkimmedProcessing{"settingSkimmedProcessing", false, "Skimmed dataset processing"};
@@ -1059,7 +1061,7 @@ struct he3HadronFemto {
         } else if (std::abs(mother.pdgCode()) == H3LPDG) {
           flag |= ParticleFlags::kFromHypertriton;
         } else {
-          flag |= ParticleFlags::kFromOtherDecays;
+          return; // other decays is thought for weak decays
         }
       }
 
@@ -1260,7 +1262,16 @@ struct he3HadronFemto {
           he3Hadcand.flags |= Flags::kMixedPair;
         }
 
-        if (!settingFillPrimariesAndMixedMc && ((he3Hadcand.flags == Flags::kMixedPair) || he3Hadcand.flags == Flags::kBothPrimaries)) {
+        LOG(info) << "He3 flags: " << int(he3Hadcand.flagsHe3) << ", Hadron flags: " << int(he3Hadcand.flagsHad) << ", Pair flags: " << int(he3Hadcand.flags);
+        if (!settingFillPrimariesMc && (he3Hadcand.flags == Flags::kBothPrimaries)) {
+          continue;
+        }
+
+        if (!settingFillLi4DaughtersMc && (he3Hadcand.flags == Flags::kBothFromLi4)) {
+          continue;
+        }
+
+        if (!settingFillMixedMc && (he3Hadcand.flags == Flags::kMixedPair)) {
           continue;
         }