Including Leg momentum 2 at inner TPC update and fill it also for electrons

Author: JonathanWitte

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;

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();