From b22ecbb29e6c727c006e16a1bd5c9bc2682e6b23 Mon Sep 17 00:00:00 2001
From: tiantian cheng <chengtt0406@163.com>
Date: Mon, 4 Aug 2025 11:05:53 +0200
Subject: [PATCH 1/2] Add ToiMask info in the tree creator

---
 PWGHF/TableProducer/treeCreatorOmegacSt.cxx | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/PWGHF/TableProducer/treeCreatorOmegacSt.cxx b/PWGHF/TableProducer/treeCreatorOmegacSt.cxx
index c291852c448..87d44151567 100644
--- a/PWGHF/TableProducer/treeCreatorOmegacSt.cxx
+++ b/PWGHF/TableProducer/treeCreatorOmegacSt.cxx
@@ -165,6 +165,7 @@ DECLARE_SOA_COLUMN(DecayLengthXYCasc, decayLengthXYCasc, float);
 DECLARE_SOA_INDEX_COLUMN_FULL(MotherCasc, motherCasc, int, HfStChBarGens, "_Casc");
 DECLARE_SOA_INDEX_COLUMN_FULL(MotherPionOrKaon, motherPionOrKaon, int, HfStChBarGens, "_PionOrKaon");
 DECLARE_SOA_COLUMN(OriginMcRec, originMcRec, int);
+DECLARE_SOA_COLUMN(ToiMask, toiMask, uint32_t);
 } // namespace hf_st_charmed_baryon
 
 DECLARE_SOA_TABLE(HfStChBars, "AOD", "HFSTCHBAR",
@@ -220,7 +221,8 @@ DECLARE_SOA_TABLE(HfStChBars, "AOD", "HFSTCHBAR",
                   hf_st_charmed_baryon::DecayLengthXYCasc,
                   hf_st_charmed_baryon::MotherCascId,
                   hf_st_charmed_baryon::MotherPionOrKaonId,
-                  hf_st_charmed_baryon::OriginMcRec);
+                  hf_st_charmed_baryon::OriginMcRec,
+                  hf_st_charmed_baryon::ToiMask);
 } // namespace o2::aod
 
 struct HfTreeCreatorOmegacSt {
@@ -256,6 +258,7 @@ struct HfTreeCreatorOmegacSt {
   Configurable<float> maxNSigmaPion{"maxNSigmaPion", 5., "Max Nsigma for pion to be paired with Omega"};
   Configurable<float> maxNSigmaKaon{"maxNSigmaKaon", 5., "Max Nsigma for kaon to be paired with Omega"};
   Configurable<bool> bzOnly{"bzOnly", true, "Use B_z instead of full field map"};
+  Configurable<std::string> cfgTriggersOfInterest{"cfgTriggersOfInterest", "fTrackedOmega,fHfCharmBarToXiBach", "Triggers of interest, comma separated for Zorro"};
 
   const int itsNClsMin = 4;
   const float tpcNclsFindableFraction = 0.8;
@@ -446,7 +449,7 @@ struct HfTreeCreatorOmegacSt {
           if (runNumber == 0) {
             zorroSummary.setObject(zorro.getZorroSummary());
           }
-          zorro.initCCDB(ccdb.service, bc.runNumber(), bc.timestamp(), "fTrackedOmega");
+          zorro.initCCDB(ccdb.service, bc.runNumber(), bc.timestamp(), cfgTriggersOfInterest.value);
           zorro.populateHistRegistry(registry, bc.runNumber());
         }
         runNumber = bc.runNumber();
@@ -463,8 +466,15 @@ struct HfTreeCreatorOmegacSt {
         }
         df2.setBz(bz);
       }
+      uint32_t toiMask = 0;
       if (skimmedProcessing) {
-        zorro.isSelected(collision.bc().globalBC());
+          bool sel = zorro.isSelected(bc.globalBC());
+          if(sel){
+              std::vector<bool> toivect = zorro.getTriggerOfInterestResults();
+              for(size_t i{0}; i < toivect.size(); i++){
+                  toiMask |= static_cast<uint32_t>(toivect[i]) << i;
+              }
+          }
       }
 
       const auto primaryVertex = getPrimaryVertex(collision);
@@ -770,7 +780,8 @@ struct HfTreeCreatorOmegacSt {
                                   decayLengthCascXY,
                                   trackCascMotherId,
                                   trackMotherId,
-                                  origin);
+                                  origin,
+                                  toiMask);
                     }
                   } else {
                     continue;

From 90db8edc26d525cb63f106ab3d90cd0c235626f2 Mon Sep 17 00:00:00 2001
From: tiantian cheng <chengtt0406@163.com>
Date: Thu, 7 Aug 2025 14:21:59 +0200
Subject: [PATCH 2/2] fix clang issue

---
 PWGHF/TableProducer/treeCreatorOmegacSt.cxx | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/PWGHF/TableProducer/treeCreatorOmegacSt.cxx b/PWGHF/TableProducer/treeCreatorOmegacSt.cxx
index 87d44151567..1bd4466893e 100644
--- a/PWGHF/TableProducer/treeCreatorOmegacSt.cxx
+++ b/PWGHF/TableProducer/treeCreatorOmegacSt.cxx
@@ -468,13 +468,13 @@ struct HfTreeCreatorOmegacSt {
       }
       uint32_t toiMask = 0;
       if (skimmedProcessing) {
-          bool sel = zorro.isSelected(bc.globalBC());
-          if(sel){
-              std::vector<bool> toivect = zorro.getTriggerOfInterestResults();
-              for(size_t i{0}; i < toivect.size(); i++){
-                  toiMask |= static_cast<uint32_t>(toivect[i]) << i;
-              }
+        bool sel = zorro.isSelected(bc.globalBC());
+        if (sel) {
+          std::vector<bool> toivect = zorro.getTriggerOfInterestResults();
+          for (size_t i{0}; i < toivect.size(); i++) {
+            toiMask |= static_cast<uint32_t>(toivect[i]) << i;
           }
+        }
       }
 
       const auto primaryVertex = getPrimaryVertex(collision);