PWGHF: Implent the possibility to apply proton PID for baryons in HF finding

Author: fgrosa

PWGHF/Core/SelectorCuts.h

@@ -63,6 +63,16 @@ static const std::vector<std::string> labelsPtTrack{};
 static const std::vector<std::string> labelsCutVarTrack = {"min_dcaxytoprimary", "max_dcaxytoprimary"};
 } // namespace hf_cuts_single_track
 
+namespace hf_presel_proton_pid
+{
+// default values for the PID cuts for protons in the track-index-skim-creator
+constexpr float cutsProtonPid[3][6] = {{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> labelsCutsProtonPid = {"minPtTpc", "maxPtTpc", "nSigmaMaxTpc", "minPtTof", "maxPtTof", "nSigmaMaxTof"};
+static const std::vector<std::string> labelsRowsProtonPid = {"LcToPKPi", "XicToPKPi", "LcToPK0S"};
+} // namespace hf_presel_proton_pid
+
 namespace hf_cuts_bdt_multiclass
 {
 static constexpr int nBinsPt = 1;

PWGHF/DataModel/CandidateReconstructionTables.h

@@ -153,10 +153,12 @@ DECLARE_SOA_TABLE(HfSelCollision, "AOD", "HFSELCOLLISION", //!
 namespace hf_sel_track
 {
 DECLARE_SOA_COLUMN(IsSelProng, isSelProng, int); //!
+DECLARE_SOA_COLUMN(IsProton, isProton, int8_t);  //!
 } // namespace hf_sel_track
 
 DECLARE_SOA_TABLE(HfSelTrack, "AOD", "HFSELTRACK", //!
-                  hf_sel_track::IsSelProng);
+                  hf_sel_track::IsSelProng,
+                  hf_sel_track::IsProton);
 
 namespace hf_pv_refit_track
 {