From 28f77069c3142cbe61bfa9126705d7e8ecba92c1 Mon Sep 17 00:00:00 2001
From: Jonathan Witte <Jonathan.Witte@gmx.net>
Date: Fri, 13 Nov 2026 17:37:01 +0100
Subject: [PATCH] Including Leg momentum 2 at inner TPC update and fill it also
 for electrons

---
 PWGDQ/Core/VarManager.cxx | 3 +++
 PWGDQ/Core/VarManager.h   | 5 +++++
 2 files changed, 8 insertions(+)

diff --git a/PWGDQ/Core/VarManager.cxx b/PWGDQ/Core/VarManager.cxx
index b85e284dd40..92baef2d6ac 100644
--- a/PWGDQ/Core/VarManager.cxx
+++ b/PWGDQ/Core/VarManager.cxx
@@ -1032,6 +1032,8 @@ void VarManager::SetDefaultVarNames()
   fgVariableUnits[kPin] = "GeV/c";
   fgVariableNames[kPin_leg1] = "p_{IN}";
   fgVariableUnits[kPin_leg1] = "GeV/c";
+  fgVariableNames[kPin_leg2] = "p_{IN}";
+  fgVariableUnits[kPin_leg2] = "GeV/c";
   fgVariableNames[kSignedPin] = "p_{IN} x charge";
   fgVariableUnits[kSignedPin] = "GeV/c";
   fgVariableNames[kTOFExpMom] = "TOF expected momentum";
@@ -2405,6 +2407,7 @@ void VarManager::SetDefaultVarNames()
   fgVarNamesMap["kTPCnSigmaEl1"] = kTPCnSigmaEl1;
   fgVarNamesMap["kPin_leg1"] = kPin_leg1;
   fgVarNamesMap["kTPCnSigmaKa_leg1"] = kTPCnSigmaKa_leg1;
+  fgVarNamesMap["kPin_leg2"] = kPin_leg2;
   fgVarNamesMap["kPt2"] = kPt2;
   fgVarNamesMap["kEta2"] = kEta2;
   fgVarNamesMap["kPhi2"] = kPhi2;
diff --git a/PWGDQ/Core/VarManager.h b/PWGDQ/Core/VarManager.h
index a3b9d9d367b..07736cfe210 100644
--- a/PWGDQ/Core/VarManager.h
+++ b/PWGDQ/Core/VarManager.h
@@ -566,6 +566,7 @@ class VarManager : public TObject
     kTPCnSigmaEl1,
     kPin_leg1,
     kTPCnSigmaKa_leg1,
+    kPin_leg2,
     kPt2,
     kEta2,
     kPhi2,
@@ -4094,6 +4095,10 @@ void VarManager::FillPair(T1 const& t1, T2 const& t2, float* values)
       values[kITSclusterMap1] = t1.itsClusterMap();
       values[kITSclusterMap2] = t2.itsClusterMap();
     }
+    if constexpr ((fillMap & TrackExtra) > 0 || (fillMap & ReducedTrackBarrel) > 0) {
+      values[kPin_leg1] = t1.tpcInnerParam();
+      values[kPin_leg2] = t2.tpcInnerParam();
+    }
 
     if constexpr ((fillMap & ReducedTrackBarrelPID) > 0) {
       values[kTPCnSigmaEl1] = t1.tpcNSigmaEl();