[PWGDQ] add fraction of shared cluster and local phi in var manager (#18716)

Co-authored-by: Alexander Tiekötter <alexander.tiekoetter@cern.ch>

Author: glegras

PWGDQ/Core/VarManager.cxx

@@ -139,6 +139,10 @@ void VarManager::SetVariableDependencies()
   }
   if (fgUsedVars[kTrackIsInsideTPCModule]) {
     fgUsedVars[kPhiTPCOuter] = true;
+    fgUsedVars[kLocalSectorPhi] = true;
+  }
+  if (fgUsedVars[kLocalSectorPhi]) {
+    fgUsedVars[kPhiTPCOuter] = true;
   }
 }
 
@@ -1027,6 +1031,8 @@ void VarManager::SetDefaultVarNames()
   fgVariableUnits[kEta] = "";
   fgVariableNames[kPhi] = "#varphi";
   fgVariableUnits[kPhi] = "rad.";
+  fgVariableNames[kLocalSectorPhi] = "local sector #varphi";
+  fgVariableUnits[kLocalSectorPhi] = "rad.";
   fgVariableNames[kRap] = "y";
   fgVariableUnits[kRap] = "";
   fgVariableNames[kMass] = "mass";
@@ -1086,6 +1092,8 @@ void VarManager::SetDefaultVarNames()
   fgVariableUnits[kTPCnCRoverFindCls] = "";
   fgVariableNames[kTPCchi2] = "TPC chi2";
   fgVariableUnits[kTPCchi2] = "";
+  fgVariableNames[kTPCFractionSharedCls] = "TPC fraction shared clusters";
+  fgVariableUnits[kTPCFractionSharedCls] = "";
   fgVariableNames[kTPCsignal] = "TPC dE/dx";
   fgVariableUnits[kTPCsignal] = "";
   fgVariableNames[kPhiTPCOuter] = "#varphi_{TPCout}";
@@ -2415,6 +2423,7 @@ void VarManager::SetDefaultVarNames()
   fgVarNamesMap["kUsedKF"] = kUsedKF;
   fgVarNamesMap["kKFMass"] = kKFMass;
   fgVarNamesMap["kKFMassGeoTop"] = kKFMassGeoTop;
+  fgVarNamesMap["kLocalSectorPhi"] = kLocalSectorPhi;
   fgVarNamesMap["kPt1"] = kPt1;
   fgVarNamesMap["kEta1"] = kEta1;
   fgVarNamesMap["kPhi1"] = kPhi1;
@@ -2470,6 +2479,7 @@ void VarManager::SetDefaultVarNames()
   fgVarNamesMap["kTPCnclsCR"] = kTPCnclsCR;
   fgVarNamesMap["kTPCnCRoverFindCls"] = kTPCnCRoverFindCls;
   fgVarNamesMap["kTPCchi2"] = kTPCchi2;
+  fgVarNamesMap["kTPCFractionSharedCls"] = kTPCFractionSharedCls;
   fgVarNamesMap["kTPCsignal"] = kTPCsignal;
   fgVarNamesMap["kPhiTPCOuter"] = kPhiTPCOuter;
   fgVarNamesMap["kTrackIsInsideTPCModule"] = kTrackIsInsideTPCModule;

PWGDQ/Core/VarManager.h

@@ -587,6 +587,7 @@ class VarManager : public TObject
     // Barrel track variables
     kPin,
     kSignedPin,
+    kLocalSectorPhi,
     kTOFExpMom,
     kTrackTime,
     kTrackTimeRes,
@@ -618,6 +619,7 @@ class VarManager : public TObject
     kTPCnclsCR,
     kTPCnCRoverFindCls,
     kTPCchi2,
+    kTPCFractionSharedCls,
     kTPCsignal,
     kPhiTPCOuter,
     kTrackIsInsideTPCModule,
@@ -3086,16 +3088,18 @@ void VarManager::FillTrack(T const& track, float* values)
     if (fgUsedVars[kPhiTPCOuter]) {
       values[kPhiTPCOuter] = RecoDecay::constrainAngle(track.phi() - track.sign() * (o2::constants::math::PIHalf - std::acos(0.22 * fgMagField / track.pt())));
     }
+    if (fgUsedVars[kLocalSectorPhi]) {
+      values[kLocalSectorPhi] = values[kPhiTPCOuter] - std::floor(18.0 * values[kPhiTPCOuter] / o2::constants::math::TwoPI) * (o2::constants::math::TwoPI / 18.0);
+    }
     if (fgUsedVars[kTrackIsInsideTPCModule]) {
-      float localSectorPhi = values[kPhiTPCOuter] - std::floor(18.0 * values[kPhiTPCOuter] / o2::constants::math::TwoPI) * (o2::constants::math::TwoPI / 18.0);
       float edge = fgTPCInterSectorBoundary / 2.0 / 246.6; // minimal inter-sector boundary as angle
       float curvature = 3.0 * 3.33 * track.pt() / fgMagField * (1.0 - std::sin(std::acos(0.22 * fgMagField / track.pt())));
       if (curvature / o2::dqconstants::defaults::CurvatureNormalizationFactor > edge) {
         edge = curvature / o2::dqconstants::defaults::CurvatureNormalizationFactor;
       }
       double min = edge;
       double max = o2::constants::math::TwoPI / 18.0 - edge;
-      values[kTrackIsInsideTPCModule] = (localSectorPhi > min && localSectorPhi < max ? 1.0 : 0.0);
+      values[kTrackIsInsideTPCModule] = (values[kLocalSectorPhi] > min && values[kLocalSectorPhi] < max ? 1.0 : 0.0);
     }
 
     if constexpr ((fillMap & ReducedTrack) > 0 && !((fillMap & Pair) > 0)) {
@@ -3193,6 +3197,7 @@ void VarManager::FillTrack(T const& track, float* values)
     values[kITSchi2] = track.itsChi2NCl();
     values[kTPCncls] = track.tpcNClsFound();
     values[kTPCchi2] = track.tpcChi2NCl();
+    values[kTPCFractionSharedCls] = track.tpcFractionSharedCls();
     values[kTrackLength] = track.length();
     values[kTPCnclsCR] = track.tpcNClsCrossedRows();
     values[kTRDPattern] = track.trdPattern();

PWGDQ/DataModel/ReducedInfoTables.h

@@ -482,6 +482,7 @@ DECLARE_SOA_TABLE(ReducedTracksBarrel, "AOD", "RTBARREL", //!
                   track::IsPVContributor<track::Flags>,
                   track::TPCNClsFound<track::TPCNClsFindable, track::TPCNClsFindableMinusFound>,
                   track::TPCNClsCrossedRows<track::TPCNClsFindable, track::TPCNClsFindableMinusCrossedRows>,
+                  track::TPCFractionSharedCls<track::TPCNClsShared, track::TPCNClsFindable, track::TPCNClsFindableMinusFound>,
                   reducedtrack::HasITS<reducedtrack::DetectorMap>, reducedtrack::HasTRD<reducedtrack::DetectorMap>,
                   reducedtrack::HasTOF<reducedtrack::DetectorMap>, reducedtrack::HasTPC<reducedtrack::DetectorMap>);