[PWGLF] Add more granularity to the number of selected collisions in StraSelections (#17159)

romainschotter · alibuild · web-flow · commit 94f0396e8cbb · 2026-07-22T14:06:25.000+02:00
Co-authored-by: ALICE Builder &lt;alibuild@users.noreply.github.com&gt;
diff --git a/PWGLF/DataModel/LFStrangenessTables.h b/PWGLF/DataModel/LFStrangenessTables.h
@@ -47,12 +47,32 @@ DECLARE_SOA_TABLE(StraOrigins, "AOD", "STRAORIGIN", //! Table which contains the
 // for keeping track of the number of selected collisions
 namespace straselections
 {
+// Event selection criteria
+enum EventSelectionFlags {
+  kIsTriggerTVX = 0,     // FT0 vertex (acceptable FT0C-FT0A time difference) at trigger level
+  kNoITSROFrameBorder,   // bunch crossing is far from ITS RO Frame border
+  kNoTimeFrameBorder,    // bunch crossing is far from Time Frame borders
+  kIsGoodZvtxAcceptance, // Zvtx within detector acceptance
+  kNoSameBunchPileup,    // reject collisions in case of pileup with another collision in the same foundBC
+  kIsGoodRCT,            // Good RCT
+  kNsel                  // counter
+};
+
 DECLARE_SOA_COLUMN(TotalNbrOfCollisions, totalNbrOfCollisions, int);       //! total number of analysed collisions
 DECLARE_SOA_COLUMN(TotalNbrOfSelCollisions, totalNbrOfSelCollisions, int); //! total number of selected collisions
+DECLARE_SOA_COLUMN(TotalIsTriggerTVXCollisions, totalIsTriggerTVXCollisions, int);     //! N selected collisions after applying IsTriggerTVX
+DECLARE_SOA_COLUMN(TotalNoITSROFBorderCollisions, totalNoITSROFBorderCollisions, int); //! N selected collisions after applying IsTriggerTVX, NoITSROF
+DECLARE_SOA_COLUMN(TotalNoTFBorderCollisions, totalNoTFBorderCollisions, int);         //! N selected collisions after applying IsTriggerTVX, NoITSROF, NoTF
+DECLARE_SOA_COLUMN(TotalIsGoodZvtxCollisions, totalIsGoodZvtxCollisions, int);         //! N selected collisions after applying IsTriggerTVX, NoITSROF, NoTF, |Zvtx| < X cm
+DECLARE_SOA_COLUMN(TotalNoSBPileupCollisions, totalNoSBPileupCollisions, int);         //! N selected collisions after applying IsTriggerTVX, NoITSROF, NoTF, |Zvtx| < X cm, NoSameBunchPileup
+DECLARE_SOA_COLUMN(TotalIsGoodRCTCollisions, totalIsGoodRCTCollisions, int);           //! N selected collisions after applying IsTriggerTVX, NoITSROF, NoTF, |Zvtx| < X cm, NoSameBunchPileup, Good RCT
 } // namespace straselections
 
 DECLARE_SOA_TABLE(StraSelections, "AOD", "STRASELECTIONS", //! keep track of the number of analysed collisions in this DF
-                  o2::soa::Index<>, straselections::TotalNbrOfCollisions, straselections::TotalNbrOfSelCollisions);
+                  o2::soa::Index<>, straselections::TotalNbrOfCollisions, straselections::TotalNbrOfSelCollisions,
+                  straselections::TotalIsTriggerTVXCollisions, straselections::TotalNoITSROFBorderCollisions,
+                  straselections::TotalNoTFBorderCollisions, straselections::TotalIsGoodZvtxCollisions,
+                  straselections::TotalNoSBPileupCollisions, straselections::TotalIsGoodRCTCollisions);
 
 namespace stracollision
 {
diff --git a/PWGLF/TableProducer/Strangeness/strangederivedbuilder.cxx b/PWGLF/TableProducer/Strangeness/strangederivedbuilder.cxx
@@ -287,32 +287,38 @@ struct strangederivedbuilder {
   }
 
   template <typename TCollision>
-  bool isCollisionAccepted(TCollision collision)
+  bool isCollisionAccepted(TCollision collision, std::array<int, o2::aod::straselections::kNsel>& nSelected)
   // check whether the collision passes our collision selections
   {
     if (requireTriggerTVX && !collision.selection_bit(aod::evsel::kIsTriggerTVX)) {
       return false;
     }
+    nSelected[o2::aod::straselections::kIsTriggerTVX]++;
 
     if (rejectITSROFBorder && !collision.selection_bit(o2::aod::evsel::kNoITSROFrameBorder)) {
       return false;
     }
+    nSelected[o2::aod::straselections::kNoITSROFrameBorder]++;
 
     if (rejectTFBorder && !collision.selection_bit(o2::aod::evsel::kNoTimeFrameBorder)) {
       return false;
     }
+    nSelected[o2::aod::straselections::kNoTimeFrameBorder]++;
 
     if (std::abs(collision.posZ()) > maxZVtxPosition) {
       return false;
     }
+    nSelected[o2::aod::straselections::kIsGoodZvtxAcceptance]++;
 
     if (rejectSameBunchPileup && !collision.selection_bit(o2::aod::evsel::kNoSameBunchPileup)) {
       return false;
     }
+    nSelected[o2::aod::straselections::kNoSameBunchPileup]++;
 
     if (!cfgApplyRCTrequirement && !rctFlagsChecker(collision)) {
       return false;
     }
+    nSelected[o2::aod::straselections::kIsGoodRCT]++;
 
     return true;
   }
@@ -559,6 +565,7 @@ struct strangederivedbuilder {
 
     int totalNbrCollisions = collisions.size();
     int totalNbrSelectedCollisions = 0;
+    std::array<int, o2::aod::straselections::kNsel> totalNbrCollisionsPerSelection = {};
 
     // +-<*>-+-<*>-+-<*>-+-<*>-+-<*>-+-<*>-+-<*>-+-<*>-+-<*>-+-<*>-+-<*>-+
     for (const auto& collision : collisions) {
@@ -639,7 +646,7 @@ struct strangederivedbuilder {
       for (const auto& track : TrackTable_thisColl)
         TrackCollIndices[track.globalIndex()] = bc.globalBC();
 
-      if (fillOnlySelectedCollisions && !isCollisionAccepted(collision)) {
+      if (fillOnlySelectedCollisions && !isCollisionAccepted(collision, totalNbrCollisionsPerSelection)) {
         continue;
       }
 
@@ -771,7 +778,13 @@ struct strangederivedbuilder {
     for (const auto& casc : TraCascades) {
       products.tracasccollref(TraCascadeCollIndices[casc.globalIndex()]);
     }
-    products.straSelection(totalNbrCollisions, totalNbrSelectedCollisions);
+    products.straSelection(totalNbrCollisions, totalNbrSelectedCollisions,
+                           totalNbrCollisionsPerSelection[o2::aod::straselections::kIsTriggerTVX],
+                           totalNbrCollisionsPerSelection[o2::aod::straselections::kNoITSROFrameBorder],
+                           totalNbrCollisionsPerSelection[o2::aod::straselections::kNoTimeFrameBorder],
+                           totalNbrCollisionsPerSelection[o2::aod::straselections::kIsGoodZvtxAcceptance],
+                           totalNbrCollisionsPerSelection[o2::aod::straselections::kNoSameBunchPileup],
+                           totalNbrCollisionsPerSelection[o2::aod::straselections::kIsGoodRCT]);
   }
 
   // helper function to estimate collision time
@@ -781,6 +794,8 @@ struct strangederivedbuilder {
     std::vector<double> collisionEventTime(collisions.size(), 0.0);
     std::vector<double> collisionEventTimeErr(collisions.size(), 0.0);
     std::vector<int> collisionNtracks(collisions.size(), 0);
+
+    std::array<int, o2::aod::straselections::kNsel> totalNbrCollisionsPerSelection = {};
     for (const auto& track : tracks) {
       if (track.hasTOF() && track.collisionId() >= 0) {
         collisionEventTime[track.collisionId()] += track.tofEvTime();
@@ -797,7 +812,7 @@ struct strangederivedbuilder {
         collisionEventTime[collision.globalIndex()] = -1e+6; // undefined
         collisionEventTimeErr[collision.globalIndex()] = -1e-6; // undefined
       }
-      if (fillOnlySelectedCollisions && !isCollisionAccepted(collision)) {
+      if (fillOnlySelectedCollisions && !isCollisionAccepted(collision, totalNbrCollisionsPerSelection)) {
         continue;
       }
       histos.fill(HIST("h2dCollisionTimesVsNTracks"), collisionNtracks[collision.globalIndex()], collisionEventTime[collision.globalIndex()]);
diff --git a/PWGLF/Tasks/Strangeness/derivedlambdakzeroanalysis.cxx b/PWGLF/Tasks/Strangeness/derivedlambdakzeroanalysis.cxx
@@ -673,9 +673,15 @@ struct derivedlambdakzeroanalysis {
 
     //
     if (doprocessAnalysedCollisions) {
-      histos.add("hEventPreSelection", "hEventPreSelection", kTH1D, {{2, -0.5f, +1.5f}});
+      histos.add("hEventPreSelection", "hEventPreSelection", kTH1D, {{8, -0.5f, +7.5f}});
       histos.get<TH1>(HIST("hEventPreSelection"))->GetXaxis()->SetBinLabel(1, "All collisions");
-      histos.get<TH1>(HIST("hEventPreSelection"))->GetXaxis()->SetBinLabel(2, "Preselected collisions");
+      histos.get<TH1>(HIST("hEventPreSelection"))->GetXaxis()->SetBinLabel(2, "kIsTriggerTVX");
+      histos.get<TH1>(HIST("hEventPreSelection"))->GetXaxis()->SetBinLabel(3, "kNoITSROFrameBorder");
+      histos.get<TH1>(HIST("hEventPreSelection"))->GetXaxis()->SetBinLabel(4, "kNoTimeFrameBorder");
+      histos.get<TH1>(HIST("hEventPreSelection"))->GetXaxis()->SetBinLabel(5, "posZ cut");
+      histos.get<TH1>(HIST("hEventPreSelection"))->GetXaxis()->SetBinLabel(6, "kNoSameBunchPileup");
+      histos.get<TH1>(HIST("hEventPreSelection"))->GetXaxis()->SetBinLabel(7, "RCT flags");
+      histos.get<TH1>(HIST("hEventPreSelection"))->GetXaxis()->SetBinLabel(8, "Preselected collisions");
     }
 
     // Event Counters
@@ -3166,8 +3172,15 @@ struct derivedlambdakzeroanalysis {
   void processAnalysedCollisions(aod::StraSelections const& straSelections)
   {
     for (auto const& straSelection : straSelections) {
+      // Event selection criteria
       histos.get<TH1>(HIST("hEventPreSelection"))->AddBinContent(1, straSelection.totalNbrOfCollisions() /* all collisions */);
-      histos.get<TH1>(HIST("hEventPreSelection"))->AddBinContent(2, straSelection.totalNbrOfSelCollisions() /* preselected collisions */);
+      histos.get<TH1>(HIST("hEventPreSelection"))->AddBinContent(2, straSelection.totalIsTriggerTVXCollisions() /* preselected IsTriggerTVX collisions */);
+      histos.get<TH1>(HIST("hEventPreSelection"))->AddBinContent(3, straSelection.totalNoITSROFBorderCollisions() /* + preselected NoITSROF collisions */);
+      histos.get<TH1>(HIST("hEventPreSelection"))->AddBinContent(4, straSelection.totalNoTFBorderCollisions() /* + preselected NoTF collisions */);
+      histos.get<TH1>(HIST("hEventPreSelection"))->AddBinContent(5, straSelection.totalIsGoodZvtxCollisions() /* + preselected |Zvtx| < X cm collisions */);
+      histos.get<TH1>(HIST("hEventPreSelection"))->AddBinContent(6, straSelection.totalNoSBPileupCollisions() /* + preselected NoSameBunchPileup collisions */);
+      histos.get<TH1>(HIST("hEventPreSelection"))->AddBinContent(7, straSelection.totalIsGoodRCTCollisions() /* + preselected Good RCT collisions */);
+      histos.get<TH1>(HIST("hEventPreSelection"))->AddBinContent(8, straSelection.totalNbrOfSelCollisions() /* total number of preselected collisions */);
     }
   }
 

Original file line number	Diff line number	Diff line change
`@@ -287,32 +287,38 @@ struct strangederivedbuilder {`
`287`	`287`	`}`
`288`	`288`
`289`	`289`	`template <typename TCollision>`
`290`		`- bool isCollisionAccepted(TCollision collision)`
	`290`	`+ bool isCollisionAccepted(TCollision collision, std::array<int, o2::aod::straselections::kNsel>& nSelected)`
`291`	`291`	`// check whether the collision passes our collision selections`
`292`	`292`	`{`
`293`	`293`	`if (requireTriggerTVX && !collision.selection_bit(aod::evsel::kIsTriggerTVX)) {`
`294`	`294`	`return false;`
`295`	`295`	`}`
	`296`	`+ nSelected[o2::aod::straselections::kIsTriggerTVX]++;`
`296`	`297`
`297`	`298`	`if (rejectITSROFBorder && !collision.selection_bit(o2::aod::evsel::kNoITSROFrameBorder)) {`
`298`	`299`	`return false;`
`299`	`300`	`}`
	`301`	`+ nSelected[o2::aod::straselections::kNoITSROFrameBorder]++;`
`300`	`302`
`301`	`303`	`if (rejectTFBorder && !collision.selection_bit(o2::aod::evsel::kNoTimeFrameBorder)) {`
`302`	`304`	`return false;`
`303`	`305`	`}`
	`306`	`+ nSelected[o2::aod::straselections::kNoTimeFrameBorder]++;`
`304`	`307`
`305`	`308`	`if (std::abs(collision.posZ()) > maxZVtxPosition) {`
`306`	`309`	`return false;`
`307`	`310`	`}`
	`311`	`+ nSelected[o2::aod::straselections::kIsGoodZvtxAcceptance]++;`
`308`	`312`
`309`	`313`	`if (rejectSameBunchPileup && !collision.selection_bit(o2::aod::evsel::kNoSameBunchPileup)) {`
`310`	`314`	`return false;`
`311`	`315`	`}`
	`316`	`+ nSelected[o2::aod::straselections::kNoSameBunchPileup]++;`
`312`	`317`
`313`	`318`	`if (!cfgApplyRCTrequirement && !rctFlagsChecker(collision)) {`
`314`	`319`	`return false;`
`315`	`320`	`}`
	`321`	`+ nSelected[o2::aod::straselections::kIsGoodRCT]++;`
`316`	`322`
`317`	`323`	`return true;`
`318`	`324`	`}`
`@@ -559,6 +565,7 @@ struct strangederivedbuilder {`
`559`	`565`
`560`	`566`	`int totalNbrCollisions = collisions.size();`
`561`	`567`	`int totalNbrSelectedCollisions = 0;`
	`568`	`+ std::array<int, o2::aod::straselections::kNsel> totalNbrCollisionsPerSelection = {};`
`562`	`569`
`563`	`570`	`// +-<>-+-<>-+-<>-+-<>-+-<>-+-<>-+-<>-+-<>-+-<>-+-<>-+-<*>-+`
`564`	`571`	`for (const auto& collision : collisions) {`
`@@ -639,7 +646,7 @@ struct strangederivedbuilder {`
`639`	`646`	`for (const auto& track : TrackTable_thisColl)`
`640`	`647`	`TrackCollIndices[track.globalIndex()] = bc.globalBC();`
`641`	`648`
`642`		`- if (fillOnlySelectedCollisions && !isCollisionAccepted(collision)) {`
	`649`	`+ if (fillOnlySelectedCollisions && !isCollisionAccepted(collision, totalNbrCollisionsPerSelection)) {`
`643`	`650`	`continue;`
`644`	`651`	`}`
`645`	`652`
`@@ -771,7 +778,13 @@ struct strangederivedbuilder {`
`771`	`778`	`for (const auto& casc : TraCascades) {`
`772`	`779`	`products.tracasccollref(TraCascadeCollIndices[casc.globalIndex()]);`
`773`	`780`	`}`
`774`		`- products.straSelection(totalNbrCollisions, totalNbrSelectedCollisions);`
	`781`	`+ products.straSelection(totalNbrCollisions, totalNbrSelectedCollisions,`
	`782`	`+ totalNbrCollisionsPerSelection[o2::aod::straselections::kIsTriggerTVX],`
	`783`	`+ totalNbrCollisionsPerSelection[o2::aod::straselections::kNoITSROFrameBorder],`
	`784`	`+ totalNbrCollisionsPerSelection[o2::aod::straselections::kNoTimeFrameBorder],`
	`785`	`+ totalNbrCollisionsPerSelection[o2::aod::straselections::kIsGoodZvtxAcceptance],`
	`786`	`+ totalNbrCollisionsPerSelection[o2::aod::straselections::kNoSameBunchPileup],`
	`787`	`+ totalNbrCollisionsPerSelection[o2::aod::straselections::kIsGoodRCT]);`
`775`	`788`	`}`
`776`	`789`
`777`	`790`	`// helper function to estimate collision time`
`@@ -781,6 +794,8 @@ struct strangederivedbuilder {`
`781`	`794`	`std::vector<double> collisionEventTime(collisions.size(), 0.0);`
`782`	`795`	`std::vector<double> collisionEventTimeErr(collisions.size(), 0.0);`
`783`	`796`	`std::vector<int> collisionNtracks(collisions.size(), 0);`
	`797`	`+`
	`798`	`+ std::array<int, o2::aod::straselections::kNsel> totalNbrCollisionsPerSelection = {};`
`784`	`799`	`for (const auto& track : tracks) {`
`785`	`800`	`if (track.hasTOF() && track.collisionId() >= 0) {`
`786`	`801`	`collisionEventTime[track.collisionId()] += track.tofEvTime();`
`@@ -797,7 +812,7 @@ struct strangederivedbuilder {`
`797`	`812`	`collisionEventTime[collision.globalIndex()] = -1e+6; // undefined`
`798`	`813`	`collisionEventTimeErr[collision.globalIndex()] = -1e-6; // undefined`
`799`	`814`	`}`
`800`		`- if (fillOnlySelectedCollisions && !isCollisionAccepted(collision)) {`
	`815`	`+ if (fillOnlySelectedCollisions && !isCollisionAccepted(collision, totalNbrCollisionsPerSelection)) {`
`801`	`816`	`continue;`
`802`	`817`	`}`
`803`	`818`	`histos.fill(HIST("h2dCollisionTimesVsNTracks"), collisionNtracks[collision.globalIndex()], collisionEventTime[collision.globalIndex()]);`