From fb2ee87acc728c42b4c562607e7dac4873f72907 Mon Sep 17 00:00:00 2001
From: mariafernandatorrescabrera <maria.fernanda.torres.cabrera@cern.ch>
Date: Sun, 15 Nov 2026 15:11:44 -0600
Subject: [PATCH] Add DCAFitter normalized impact parameters and decay length
 to OmegaC0 ML inputs

---
 PWGHF/Core/HfMlResponseOmegacToOmegaPiQa.h            | 11 ++++++++++-
 PWGHF/DataModel/CandidateReconstructionTables.h       |  8 +++++++-
 PWGHF/TableProducer/treeCreatorOmegac0ToOmegaPiQa.cxx |  4 ++--
 3 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/PWGHF/Core/HfMlResponseOmegacToOmegaPiQa.h b/PWGHF/Core/HfMlResponseOmegacToOmegaPiQa.h
index 2c3fd60bcae..3b5f89f3650 100644
--- a/PWGHF/Core/HfMlResponseOmegacToOmegaPiQa.h
+++ b/PWGHF/Core/HfMlResponseOmegacToOmegaPiQa.h
@@ -71,6 +71,9 @@ enum class InputFeaturesOmegacToOmegaPi : uint8_t {
   cosPACharmBaryon,
   cosPACasc,
   normDecayLenCharmBar,
+  normImpParCascade,
+  normImpParPiFromCharmBar,
+  decLenCharmBaryon,
 
   // KFParticle variables
   cosPaOmegacToPv,
@@ -138,6 +141,9 @@ class HfMlResponseOmegacToOmegaPi : public HfMlResponse<TypeOutputScore>
           CHECK_AND_FILL_VEC_OMEGAC0(cosPACharmBaryon);
           CHECK_AND_FILL_VEC_OMEGAC0(cosPACasc);
           CHECK_AND_FILL_VEC_OMEGAC0(normDecayLenCharmBar);
+          CHECK_AND_FILL_VEC_OMEGAC0(normImpParCascade);
+          CHECK_AND_FILL_VEC_OMEGAC0(normImpParPiFromCharmBar);
+          CHECK_AND_FILL_VEC_OMEGAC0(decLenCharmBaryon);
         }
       }
 
@@ -195,7 +201,10 @@ class HfMlResponseOmegacToOmegaPi : public HfMlResponse<TypeOutputScore>
         FILL_MAP_OMEGAC0(impactParCascZ),
         FILL_MAP_OMEGAC0(cosPACharmBaryon),
         FILL_MAP_OMEGAC0(cosPACasc),
-        FILL_MAP_OMEGAC0(normDecayLenCharmBar)};
+        FILL_MAP_OMEGAC0(normDecayLenCharmBar),
+        FILL_MAP_OMEGAC0(normImpParCascade),
+        FILL_MAP_OMEGAC0(normImpParPiFromCharmBar),
+        FILL_MAP_OMEGAC0(decLenCharmBaryon)};
 
       MlResponse<TypeOutputScore>::mAvailableInputFeatures.insert(mapDcaFeatures.begin(), mapDcaFeatures.end());
     }
diff --git a/PWGHF/DataModel/CandidateReconstructionTables.h b/PWGHF/DataModel/CandidateReconstructionTables.h
index 6036b9238bc..7d60a63df6d 100644
--- a/PWGHF/DataModel/CandidateReconstructionTables.h
+++ b/PWGHF/DataModel/CandidateReconstructionTables.h
@@ -1224,6 +1224,10 @@ DECLARE_SOA_DYNAMIC_COLUMN(PtKaFromCasc, ptKaFromCasc,
 
 DECLARE_SOA_DYNAMIC_COLUMN(NormDecayLenCharmBar, normDecayLenCharmBar,
                            [](float decLen, float errDecLen) -> float { return decLen / errDecLen; });
+DECLARE_SOA_DYNAMIC_COLUMN(NormImpParCascade, normImpParCascade,
+                           [](float impactPar, float errImpactPar) -> float { return impactPar / errImpactPar; });
+DECLARE_SOA_DYNAMIC_COLUMN(NormImpParPiFromCharmBar, normImpParPiFromCharmBar,
+                           [](float impactPar, float errImpactPar) -> float { return impactPar / errImpactPar; });
 
 } // end of namespace hf_cand_xic0_omegac0
 
@@ -1292,7 +1296,9 @@ DECLARE_SOA_TABLE(HfCandToOmegaPi, "AOD", "HFCANDTOOMEGAPI",
                   hf_cand_xic0_omegac0::DcaZToPvV0Dau0, hf_cand_xic0_omegac0::DcaZToPvV0Dau1, hf_cand_xic0_omegac0::DcaZToPvCascDau,
                   hf_cand_xic0_omegac0::DcaCascDau, hf_cand_xic0_omegac0::DcaV0Dau, hf_cand_xic0_omegac0::DcaCharmBaryonDau,
                   hf_cand_xic0_omegac0::DecLenCharmBaryon, hf_cand_xic0_omegac0::DecLenCascade, hf_cand_xic0_omegac0::DecLenV0, hf_cand_xic0_omegac0::ErrorDecayLengthCharmBaryon, hf_cand_xic0_omegac0::ErrorDecayLengthXYCharmBaryon, hf_track_index::HFflag,
-                  hf_cand_xic0_omegac0::NormDecayLenCharmBar<hf_cand_xic0_omegac0::DecLenCharmBaryon, hf_cand_xic0_omegac0::ErrorDecayLengthCharmBaryon>);
+                  hf_cand_xic0_omegac0::NormDecayLenCharmBar<hf_cand_xic0_omegac0::DecLenCharmBaryon, hf_cand_xic0_omegac0::ErrorDecayLengthCharmBaryon>,
+                  hf_cand_xic0_omegac0::NormImpParCascade<hf_cand_xic0_omegac0::ImpactParCascXY, hf_cand_xic0_omegac0::ErrImpactParCascXY>,
+                  hf_cand_xic0_omegac0::NormImpParPiFromCharmBar<hf_cand_xic0_omegac0::ImpactParBachFromCharmBaryonXY, hf_cand_xic0_omegac0::ErrImpactParBachFromCharmBaryonXY>);
 
 DECLARE_SOA_TABLE(HfCandToOmegaK, "AOD", "HFCANDTOOMEGAK",
                   o2::soa::Index<>,
diff --git a/PWGHF/TableProducer/treeCreatorOmegac0ToOmegaPiQa.cxx b/PWGHF/TableProducer/treeCreatorOmegac0ToOmegaPiQa.cxx
index 55ba2da9efc..8170f589688 100644
--- a/PWGHF/TableProducer/treeCreatorOmegac0ToOmegaPiQa.cxx
+++ b/PWGHF/TableProducer/treeCreatorOmegac0ToOmegaPiQa.cxx
@@ -375,8 +375,8 @@ struct HfTreeCreatorToOmegaPiQa {
                    candidate.dcaCharmBaryonDau(),
                    candidate.decLenCharmBaryon(),
                    candidate.errorDecayLengthCharmBaryon(),
-                   candidate.impactParCascXY() / candidate.errImpactParCascXY(),
-                   candidate.impactParBachFromCharmBaryonXY() / candidate.errImpactParBachFromCharmBaryonXY(),
+                   candidate.normImpParCascade(),
+                   candidate.normImpParPiFromCharmBar(),
                    candidate.normDecayLenCharmBar(),
                    candidate.template bachelorFromCharmBaryon_as<MyTrackTable>().isGlobalTrackWoDCA(),
                    candidate.template bachelorFromCharmBaryon_as<MyTrackTable>().itsNCls(),