From 7fedd277f923f683a321d2b0a3493f26955d93bd Mon Sep 17 00:00:00 2001
From: fchinu <fabrizio.chinu@cern.ch>
Date: Thu, 17 Sep 2026 19:30:46 +0200
Subject: [PATCH] Add TPC crossed rows to HfMlResponseHfTracks

---
 PWGHF/Core/HfMlResponseHfTracks.h                | 4 ++++
 PWGHF/D2H/TableProducer/mlBasedTrackSelector.cxx | 1 +
 2 files changed, 5 insertions(+)

diff --git a/PWGHF/Core/HfMlResponseHfTracks.h b/PWGHF/Core/HfMlResponseHfTracks.h
index 60fb744a197..4b6a9df281f 100644
--- a/PWGHF/Core/HfMlResponseHfTracks.h
+++ b/PWGHF/Core/HfMlResponseHfTracks.h
@@ -74,6 +74,7 @@ enum class InputFeaturesTracks : uint8_t {
   itsChi2NCl,
   // TPC quality
   tpcNClsFound,
+  tpcNClsCrossedRows,
   tpcCrossedRowsOverFindableCls,
   tpcChi2NCl,
   tpcFractionSharedCls,
@@ -110,6 +111,7 @@ struct HfTrackMlFeatures {
   float itsChi2NCl{0.f};
   // TPC quality
   float tpcNClsFound{0.f};
+  float tpcNClsCrossedRows{0.f};
   float tpcCrossedRowsOverFindableCls{0.f};
   float tpcChi2NCl{0.f};
   float tpcFractionSharedCls{0.f};
@@ -171,6 +173,7 @@ class HfMlResponseHfTracks : public HfMlResponse<TypeOutputScore>
         CHECK_AND_FILL_VEC_HF_TRACK(itsNClsInnerBarrel);
         CHECK_AND_FILL_VEC_HF_TRACK(itsChi2NCl);
         CHECK_AND_FILL_VEC_HF_TRACK(tpcNClsFound);
+        CHECK_AND_FILL_VEC_HF_TRACK(tpcNClsCrossedRows);
         CHECK_AND_FILL_VEC_HF_TRACK(tpcCrossedRowsOverFindableCls);
         CHECK_AND_FILL_VEC_HF_TRACK(tpcChi2NCl);
         CHECK_AND_FILL_VEC_HF_TRACK(tpcFractionSharedCls);
@@ -204,6 +207,7 @@ class HfMlResponseHfTracks : public HfMlResponse<TypeOutputScore>
       FILL_MAP_HF_TRACK(itsNClsInnerBarrel),
       FILL_MAP_HF_TRACK(itsChi2NCl),
       FILL_MAP_HF_TRACK(tpcNClsFound),
+      FILL_MAP_HF_TRACK(tpcNClsCrossedRows),
       FILL_MAP_HF_TRACK(tpcCrossedRowsOverFindableCls),
       FILL_MAP_HF_TRACK(tpcChi2NCl),
       FILL_MAP_HF_TRACK(tpcFractionSharedCls),
diff --git a/PWGHF/D2H/TableProducer/mlBasedTrackSelector.cxx b/PWGHF/D2H/TableProducer/mlBasedTrackSelector.cxx
index 55a854ae9ca..fd69e2bc816 100644
--- a/PWGHF/D2H/TableProducer/mlBasedTrackSelector.cxx
+++ b/PWGHF/D2H/TableProducer/mlBasedTrackSelector.cxx
@@ -525,6 +525,7 @@ struct HfTrackSelectorTagSelTracks {
     features.itsNClsInnerBarrel = static_cast<float>(track.itsNClsInnerBarrel());
     features.itsChi2NCl = track.itsChi2NCl();
     features.tpcNClsFound = static_cast<float>(track.tpcNClsFound());
+    features.tpcNClsCrossedRows = static_cast<float>(track.tpcNClsCrossedRows());
     features.tpcCrossedRowsOverFindableCls = track.tpcCrossedRowsOverFindableCls();
     features.tpcChi2NCl = track.tpcChi2NCl();
     features.tpcFractionSharedCls = track.tpcFractionSharedCls();