PWGCF: Update trigger (#4007)

* PWGCF: Update trigger

Add options to reject tracks whose assumed PID during tracking does not
correspond to the selected PID selected in the analysis. This rejection
can be turned on/off with a switch and can be applied up to a
configurabel p_TPC threshold. It can also be configured independently
for protons and deuterons.

* Fix: fix linter complaint

Author: ariedel-cern

EventFiltering/PWGCF/CFFilterAll.cxx

@@ -96,6 +96,7 @@ static const std::vector<std::string> TPCTOFAvgName{"TPC Avg", "TOF Avg"};
 static const std::vector<std::string> PidCutsName{"TPC min", "TPC max", "TOF min", "TOF max", "TPCTOF max"};
 static const std::vector<std::string> PtCutsName{"Pt min (particle)", "Pt max (particle)", "Pt min (antiparticle)", "Pt max (antiparticle)", "P thres"};
 static const std::vector<std::string> MomCorCutsName{"Momemtum Correlation min", "Momemtum Correlation max"};
+static const std::vector<std::string> PIDForTrackingName{"Switch", "Momemtum Threshold"};
 static const std::vector<std::string> ThreeBodyFilterNames{"PPP", "PPL", "PLL", "LLL"};
 static const std::vector<std::string> TwoBodyFilterNames{"PD", "LD"};
 static const std::vector<std::string> ParticleNames{"PPP", "aPaPaP", "PPL", "aPaPaL", "PLL", "aPaLaL", "LLL", "aLaLaL", "PD", "aPaD", "LD", "aLaD"};
@@ -135,6 +136,9 @@ static const float NClustersMin[1][nTracks]{
 static const float MomCorLimits[2][nMomCorCuts] =
   {{-99, 99},
    {-99, 99}};
+static const float PIDForTrackingTable[2][nTracks]{
+  {-1, 0.75},
+  {-1, 1.2}};
 
 static const float triggerSwitches[1][nAllTriggers]{
   {1, 1, 1, 1, 1, 1}};
@@ -383,6 +387,11 @@ struct CFFilter {
     "Cut on momentum correlation ratio (antipartilce)"};
   Configurable<bool> ConfMomCorRatioCutFlag{"ConfMomCorRatioFlag", false, "Flag for cut on momentum correlation ratio"};
 
+  Configurable<LabeledArray<float>> ConfPIDForTracking{
+    "ConfPIDForTracking",
+    {CFTrigger::PIDForTrackingTable[0], CFTrigger::nTracks, 2, CFTrigger::SpeciesName, CFTrigger::PIDForTrackingName},
+    "Use PID used in tracking up to momentum threshold"};
+
   Configurable<LabeledArray<float>> ConfPtCuts{
     "ConfPtCuts",
     {CFTrigger::ptcutsTable[0], CFTrigger::kNParticleSpecies, CFTrigger::nPtCuts, CFTrigger::SpeciesNameAll, CFTrigger::PtCutsName},
@@ -971,6 +980,22 @@ struct CFFilter {
     bool pThres = true;
     float nSigma = -999.;
 
+    // check tracking PID
+    uint8_t SpeciesForTracking = 0;
+    if (partSpecies == CFTrigger::kProton) {
+      SpeciesForTracking = o2::track::PID::Proton;
+    } else if (partSpecies == CFTrigger::kDeuteron) {
+      SpeciesForTracking = o2::track::PID::Deuteron;
+    } else {
+      LOG(warn) << "Unknown PID for tracking encountered";
+    }
+
+    if (ConfPIDForTracking->get(partSpecies, "Switch") > 0 && track.tpcInnerParam() < ConfPIDForTracking->get(partSpecies, "Momemtum Threshold")) {
+      if (track.pidForTracking() != SpeciesForTracking) {
+        return false;
+      }
+    }
+
     // check momentum threshold
     if (track.tpcInnerParam() <= ConfPtCuts->get(partSpecies, "P thres")) {
       pThres = true;