[PWGLF] debugging angular correlation pp histo filling

Author: chiarapinto

PWGLF/Tasks/Nuspex/angularCorrelationsInJets.cxx

@@ -539,6 +539,56 @@ struct AngularCorrelationsInJets {
     return true;
   }
 
+  bool isSmartProtonPID(const auto& track)
+  {
+    if (track.sign() < 0)
+      return false;
+
+    float pt = track.pt();
+    float maxSigmaTPC = 4.0;
+    float maxSigmaTOF = 3.0;
+  
+    // Loosen PID at high momentum
+    if (pt > 1.5) {
+      maxSigmaTPC = 3.0;
+      maxSigmaTOF = 2.0;
+    }
+    if (pt > 3.0) {
+      maxSigmaTPC = 2.0;
+      maxSigmaTOF = 1.0;
+    }
+  
+    bool tpcOK = std::abs(track.tpcNSigmaProton()) < maxSigmaTPC;
+    bool tofOK = track.hasTOF() ? std::abs(track.tofNSigmaProton()) < maxSigmaTOF : false;
+  
+    return tpcOK || tofOK;
+  }
+
+  bool isSmartAntiProtonPID(const auto& track)
+  {
+    if (track.sign() > 0)
+      return false;
+
+    float pt = track.pt();
+    float maxSigmaTPC = 4.0;
+    float maxSigmaTOF = 3.0;
+  
+    // Loosen PID at high momentum
+    if (pt > 1.5) {
+      maxSigmaTPC = 3.0;
+      maxSigmaTOF = 2.0;
+    }
+    if (pt > 3.0) {
+      maxSigmaTPC = 2.0;
+      maxSigmaTOF = 1.0;
+    }
+  
+    bool tpcOK = std::abs(track.tpcNSigmaProton()) < maxSigmaTPC;
+    bool tofOK = track.hasTOF() ? std::abs(track.tofNSigmaProton()) < maxSigmaTOF : false;
+  
+    return tpcOK || tofOK;
+  }
+
   template <typename T>
   bool isNucleus(const T& track)
   {
@@ -1085,12 +1135,14 @@ struct AngularCorrelationsInJets {
         }
       }
       if (measureCorrelations) {
-        if (isProton(jetParticle, true)) {
+        if (isSmartProtonPID(jetParticle)) {
+        //if (isProton(jetParticle, true)) { // while debugging issue for proton correlations
           registryData.fill(HIST("trackProtocol"), 5); // # high purity protons
           jetProtons.emplace_back(jetParticle);
           registryData.fill(HIST("dcaZJetProton"), jetParticle.pt(), jetParticle.dcaZ());
         }
-        if (isAntiproton(jetParticle, true)) {
+        if (isSmartAntiProtonPID(jetParticle)) {
+        //if (isAntiproton(jetParticle, true)) { // while debugging issue for antiproton correlations
           registryData.fill(HIST("trackProtocol"), 7); // # high purity antiprotons
           jetAntiprotons.emplace_back(jetParticle);
           registryData.fill(HIST("dcaZJetAntiproton"), jetParticle.pt(), jetParticle.dcaZ());