add boolean to store only true cascades in MC analysis

AliceO2Group · Oct 14, 2024 · 4cf4e1c · 4cf4e1c
1 parent 6cfbb3a
commit 4cf4e1c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/PWGLF/TableProducer/Strangeness/cascadeflow.cxx b/PWGLF/TableProducer/Strangeness/cascadeflow.cxx
@@ -128,6 +128,7 @@ struct cascadeFlow {
   ConfigurableAxis axisQVsNorm{"axisQVsNorm", {200, -1.f, 1.f}, "axisQVsNorm"};
 
   // Event selection criteria
+  Configurable<bool> isStoreTrueCascOnly{"isStoreTrueCascOnly", 1, ""};
   Configurable<float> cutzvertex{"cutzvertex", 10.0f, "Accepted z-vertex range (cm)"};
   Configurable<bool> sel8{"sel8", 1, "Apply sel8 event selection"};
   Configurable<bool> isNoSameBunchPileupCut{"isNoSameBunchPileupCut", 1, "Same found-by-T0 bunch crossing rejection"};
@@ -790,7 +791,7 @@ struct cascadeFlow {
       {
         pdgCode = 0;
       }
-
+      
       /// Add some minimal cuts for single track variables (min number of TPC clusters)
       auto negExtra = casc.negTrackExtra_as<DauTracks>();
       auto posExtra = casc.posTrackExtra_as<DauTracks>();
@@ -859,6 +860,9 @@ struct cascadeFlow {
         BDTresponse[0] = bdtScore[0][1];
         BDTresponse[1] = bdtScore[1][1];
       }
+      if (isStoreTrueCascOnly){
+	if (pdgCode==0) continue;
+      }
       if (isSelectedCasc[0] || isSelectedCasc[1])
         fillAnalysedTable(coll, casc, v2CSP, v2CEP, PsiT0C, BDTresponse[0], BDTresponse[1], pdgCode);
     }