[PWGHF] Implement TPC pid for light nuclei based on Bethe-Bloch parametrization in HF track skimming task (#14729)

Co-authored-by: ALICE Action Bot <alibuild@cern.ch>

Author: zhangbiao-phy

PWGHF/Core/SelectorCuts.h

@@ -67,29 +67,38 @@ static const std::vector<std::string> labelsCutVarTrack = {"min_dcaxytoprimary",
 
 namespace hf_presel_pid
 {
-static constexpr int NPidRows = 7; // number of PID channels / rows
+static constexpr int NPidRows = 4; // number of PID channels / rows
 static constexpr int NPidCuts = 6; // number of cuts per PID (TPC+TOF)
 // default values for the PID cuts for protons in the track-index-skim-creator
 constexpr float CutsPid[NPidRows][NPidCuts] = {{0.f, 1000.f, 5.f, 0.f, 1000.f, 5.f},
-                                               {0.f, 1000.f, 5.f, 0.f, 1000.f, 5.f},
-                                               {0.f, 1000.f, 5.f, 0.f, 1000.f, 5.f},
-                                               {0.f, 1000.f, 5.f, 0.f, 1000.f, 5.f},
                                                {0.f, 1000.f, 5.f, 0.f, 1000.f, 5.f},
                                                {0.f, 1000.f, 5.f, 0.f, 1000.f, 5.f},
                                                {0.f, 1000.f, 5.f, 0.f, 1000.f, 5.f}};
 static const std::vector<std::string> labelsCutsPid = {"minPtTpc", "maxPtTpc", "nSigmaMaxTpc", "minPtTof", "maxPtTof", "nSigmaMaxTof"};
-static const std::vector<std::string> labelsRowsPid = {"ProtonInLcToPKPi", "ProtonInXicToPKPi", "ProtonInLcToPK0S", "KaonIn3Prongs", "DeuteronInCdToDeKPi", "TritonInCtToTrKPi", "HeliumInChToHeKPi"};
+static const std::vector<std::string> labelsRowsPid = {"ProtonInLcToPKPi", "ProtonInXicToPKPi", "ProtonInLcToPK0S", "KaonIn3Prongs"};
 } // namespace hf_presel_pid
 
 namespace hf_presel_lightnuclei
 {
 
+static constexpr int NParticleRows = 4;     // number of particles / rows
+static constexpr int NVarCuts = 10;         // number of cuts for each particles
+static constexpr int NBetheBlochParams = 6; // number of parameters for Bethe-Bloch
+
 // default values for the track cuts for lightnuclei in the track-index-skim-creator
-constexpr float CutsTrackQuality[3][9] = {{-4, 3, 5., 0., 100, 100, 0.83, 160., 1.},
-                                          {-4, 3, 5., 0., 100, 100, 0.83, 160., 1.},
-                                          {-4, 3, 5., 0., 100, 100, 0.83, 160., 1.}};
-static const std::vector<std::string> labelsCutsTrack = {"nSigmaMinIts", "minItsClusterSizes", "minItsCluster", "minItsIbCluster", "minTpcCluster", "minTpcRow", "minTpcCrossedOverFound", "maxTpcShared", "maxTpcFracShared"};
-static const std::vector<std::string> labelsRowsNucleiType = {"Deutron", "Triton", "Helium3"};
+constexpr float CutsTrackQuality[NParticleRows][NVarCuts] = {{-4.f, 3.f, 5.f, 0.f, 100.f, 100.f, 0.83, 160.f, 1.f, 5.f},
+                                                             {-4.f, 3.f, 5.f, 0.f, 100.f, 100.f, 0.83, 160.f, 1.f, 5.f},
+                                                             {-4.f, 3.f, 5.f, 0.f, 100.f, 100.f, 0.83, 160.f, 1.f, 5.f},
+                                                             {-4.f, 3.f, 5.f, 0.f, 100.f, 100.f, 0.83, 160.f, 1.f, 5.f}};
+static const std::vector<std::string> labelsCutsTrack = {"nSigmaMinIts", "minItsClusterSizes", "minItsCluster", "minItsIbCluster", "minTpcCluster", "minTpcRow", "minTpcCrossedOverFound", "maxTpcShared", "maxTpcFracShared", "maxTPCnSigmaBB"};
+static const std::vector<std::string> labelsRowsNucleiType = {"Deutron", "Triton", "Helium3", "Alpha"};
+
+constexpr float BetheBlochParams[NParticleRows][NBetheBlochParams] = {{5.39302, 7.859534, 0.004048, 2.323197, 1.609307, 0.09},
+                                                                      {5.39302, 7.859534, 0.004048, 2.323197, 1.609307, 0.09},
+                                                                      {-126.55736, -0.858569, 1.11164, 1.21032, 2.656374, 0.09},
+                                                                      {-126.55736, -0.858569, 1.11164, 1.21032, 2.656374, 0.09}};
+
+static const std::vector<std::string> labelsBetheBlochParams = {"p0", "p1", "p2", "p3", "p4", "resolution"};
 
 } // namespace hf_presel_lightnuclei
 

PWGHF/DataModel/TrackIndexSkimmingTables.h

@@ -301,6 +301,7 @@ enum DecayType {
   CdToDeKPi,
   CtToTrKPi,
   ChToHeKPi,
+  CaToAlKPi,
   N3ProngDecays
 };
 } // namespace hf_cand_3prong