From 7fa2768419617e26f7d77ba166104b6185ef1da0 Mon Sep 17 00:00:00 2001
From: Fabrizio Grosa <fabrizio.grosa@cern.ch>
Date: Wed, 19 Aug 2026 16:15:35 +0200
Subject: [PATCH 1/2] Add pT-dependent invariant mass cut for Lc->pKpi

---
 PWGHF/TableProducer/candidateSelectorLc.cxx | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/PWGHF/TableProducer/candidateSelectorLc.cxx b/PWGHF/TableProducer/candidateSelectorLc.cxx
index 34b183aa1ce..4cb81d160b9 100644
--- a/PWGHF/TableProducer/candidateSelectorLc.cxx
+++ b/PWGHF/TableProducer/candidateSelectorLc.cxx
@@ -108,6 +108,8 @@ struct HfCandidateSelectorLc {
   Configurable<std::vector<std::string>> onnxFileNames{"onnxFileNames", std::vector<std::string>{"ModelHandler_onnx_LcToPKPi.onnx"}, "ONNX file names for each pT bin (if not from CCDB full path)"};
   Configurable<int64_t> timestampCCDB{"timestampCCDB", -1, "timestamp of the ONNX file for ML model used to query in CCDB"};
   Configurable<bool> loadModelsFromCCDB{"loadModelsFromCCDB", false, "Flag to enable or disable the loading of models from CCDB"};
+  // Mass Cut for trigger analysis
+  Configurable<bool> useTriggerMassCut{"useTriggerMassCut", false, "Flag to enable parametrize pT differential mass cut for triggered data"};
 
   o2::analysis::HfMlResponseLcToPKPi<float, aod::hf_cand::VertexerType::DCAFitter> hfMlResponseDCA;
   o2::analysis::HfMlResponseLcToPKPi<float, aod::hf_cand::VertexerType::KfParticle> hfMlResponseKF;
@@ -117,6 +119,7 @@ struct HfCandidateSelectorLc {
   TrackSelectorPi selectorPion;
   TrackSelectorKa selectorKaon;
   TrackSelectorPr selectorProton;
+  HfTrigger3ProngCuts hfTriggerCuts;
 
   using TracksSel = soa::Join<aod::TracksWExtra,
                               aod::TracksPidPi, aod::PidTpcTofFullPi, aod::TracksPidKa, aod::PidTpcTofFullKa, aod::TracksPidPr, aod::PidTpcTofFullPr>;
@@ -327,6 +330,10 @@ struct HfCandidateSelectorLc {
         return false;
       }
 
+      if (useTriggerMassCut && !isCandidateInMassRange(massLc, o2::constants::physics::MassLambdaCPlus, ptCand, hfTriggerCuts)) {
+        return false;
+      }
+
       /// cut on the Kpi pair invariant mass, to study Lc->pK*(->Kpi)
       const double cutMassKPi = cuts->get(pTBin, "mass (Kpi)");
       if (cutMassKPi > 0 && std::abs(massKPi - massK0Star892) > cutMassKPi) {

From f615ba60c4623d390c903aa9a904d75f8f4599dc Mon Sep 17 00:00:00 2001
From: Fabrizio Grosa <fabrizio.grosa@cern.ch>
Date: Wed, 19 Aug 2026 16:17:47 +0200
Subject: [PATCH 2/2] Add missing header

---
 PWGHF/TableProducer/candidateSelectorLc.cxx | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/PWGHF/TableProducer/candidateSelectorLc.cxx b/PWGHF/TableProducer/candidateSelectorLc.cxx
index 4cb81d160b9..a3155b30e3e 100644
--- a/PWGHF/TableProducer/candidateSelectorLc.cxx
+++ b/PWGHF/TableProducer/candidateSelectorLc.cxx
@@ -24,6 +24,7 @@
 #include "PWGHF/DataModel/CandidateReconstructionTables.h"
 #include "PWGHF/DataModel/CandidateSelectionTables.h"
 #include "PWGHF/DataModel/TrackIndexSkimmingTables.h"
+#include "PWGHF/Utils/utilsAnalysis.h"
 
 #include "Common/Core/TrackSelectorPID.h"
 #include "Common/DataModel/PIDResponseCombined.h"
@@ -330,7 +331,7 @@ struct HfCandidateSelectorLc {
         return false;
       }
 
-      if (useTriggerMassCut && !isCandidateInMassRange(massLc, o2::constants::physics::MassLambdaCPlus, ptCand, hfTriggerCuts)) {
+      if (useTriggerMassCut && !isCandidateInMassRange(massLc, o2::constants::physics::MassLambdaCPlus, candpT, hfTriggerCuts)) {
         return false;
       }