From 2ce2ace3e228a9d9535111a9c505afee0d7f50d7 Mon Sep 17 00:00:00 2001
From: fuchuncui <162277233+fuchuncui@users.noreply.github.com>
Date: Wed, 24 Jun 2026 16:25:56 +0800
Subject: [PATCH] Add files via upload

---
 .../Tasks/cascDiHadronCorr.cxx                   | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/PWGCF/TwoParticleCorrelations/Tasks/cascDiHadronCorr.cxx b/PWGCF/TwoParticleCorrelations/Tasks/cascDiHadronCorr.cxx
index 1f1f31a09d0..76bb9e239d6 100644
--- a/PWGCF/TwoParticleCorrelations/Tasks/cascDiHadronCorr.cxx
+++ b/PWGCF/TwoParticleCorrelations/Tasks/cascDiHadronCorr.cxx
@@ -266,6 +266,9 @@ struct CascDiHadronCorr {
   using ValidCollisions = std::vector<std::vector<ValidCollision>>;
   ValidCollisions validCollisions;
 
+  double masslow = 0;
+  double massup = 0;
+
   // persistent caches
   std::vector<float> efficiencyAssociatedCache;
 
@@ -278,6 +281,10 @@ struct CascDiHadronCorr {
     }
     const AxisSpec axisPhi{72, 0.0, constants::math::TwoPI, "#varphi"};
     const AxisSpec axisEta{40, -1., 1., "#eta"};
+    o2::framework::AxisSpec axismass = axisInvMass;
+    int nMasssBinEdges = axismass.binEdges.size();
+    masslow = axismass.binEdges[0];
+    massup = axismass.binEdges[nMasssBinEdges - 1];
     cfgNSigma = cfgNSigmapid;
 
     ccdb->setURL("http://alice-ccdb.cern.ch");
@@ -530,6 +537,8 @@ struct CascDiHadronCorr {
       return false;
 
     if (cfgOutputXi) {
+      if (casc.mXi() > massup || casc.mXi() < masslow)
+        return false;
       if (casc.sign() < 0) {
         if (std::fabs(bachelor.tpcNSigmaPi()) > cfgNSigma[0])
           return false;
@@ -571,6 +580,8 @@ struct CascDiHadronCorr {
         return false;
     }
     if (cfgOutputOmega) {
+      if (casc.mOmega() > massup || casc.mOmega() < masslow)
+        return false;
       if (casc.sign() < 0) {
         if (std::fabs(bachelor.tpcNSigmaKa()) > cfgNSigma[2])
           return false;
@@ -817,7 +828,10 @@ struct CascDiHadronCorr {
           float deltaPhi = RecoDecay::constrainAngle(track1.phi() - track2.phi, -PIHalf);
           float deltaEta = track1.eta() - track2.eta;
 
-          mixed->getPairHist()->Fill(step, fSampleIndex, posZ, track1.pt(), track2.pt, deltaPhi, deltaEta, eventWeight * triggerWeight * associatedWeight);
+          if (cfgOutputXi)
+            same->getPairHist()->Fill(step, fSampleIndex, posZ, track1.pt(), track2.pt, deltaPhi, deltaEta, track1.mXi(), eventWeight * triggerWeight * associatedWeight);
+          if (cfgOutputOmega)
+            same->getPairHist()->Fill(step, fSampleIndex, posZ, track1.pt(), track2.pt, deltaPhi, deltaEta, track1.mOmega(), eventWeight * triggerWeight * associatedWeight);
           registry.fill(HIST("deltaEta_deltaPhi_mixed"), deltaPhi, deltaEta, eventWeight * triggerWeight * associatedWeight);
         }
       }