New config: use strict pid

Author: kegang02

PWGCF/Flow/Tasks/flowPidCme.cxx

@@ -9,7 +9,7 @@
 // granted to it by virtue of its status as an Intergovernmental Organization
 // or submit itself to any jurisdiction.
 
-/// \author ZhengqingWang(zhengqing.wang@cern.ch)
+/// \author ZhengqingWang(zhengqing.wang@cern.ch), KegangXiong(kxiong@cern.ch)
 /// \file   flowPidCme.cxx
 /// \brief  task to calculate the pikp cme signal and bacground.
 // C++/ROOT includes.
@@ -158,6 +158,7 @@ struct FillPIDcolums {
   Configurable<bool> cfgOpenITSCut{"cfgOpenITSCut", true, "open ITSnsigma cut"};
   Configurable<bool> cfgOpenITSCutQAPlots{"cfgOpenITSCutQAPlots", true, "open QA plots after ITS nsigma cut"};
   Configurable<bool> cfgOpenDetailPlotsTPCITSContaimination{"cfgOpenDetailPlotsTPCITSContaimination", false, "open detail TH3D plots for nSigmaTPC-ITS Pt-eta-Phi nSigmaITS-clustersize"};
+  Configurable<bool> cfgUseStrictPID{"cfgUseStrictPID", true, "More strict pid strategy"};
   Configurable<bool> cfgOpenAllowCrossTrack{"cfgOpenAllowCrossTrack", false, "Allow one track to be identified as different kind of PID particles"};
   Configurable<bool> cfgOpenCrossTrackQAPlots{"cfgOpenCrossTrackQAPlots", true, "open cross pid track QA plots"};
   Configurable<bool> cfgOpenTOFOnlyPID{"cfgOpenTOFOnlyPID", true, "only accept tracks who has TOF infomation and use TOFnsigma for PID(priority greater than TPConly and combined"};
@@ -311,9 +312,17 @@ struct FillPIDcolums {
       pidVectorUpper = pidVectorTPCPtUpper;
       pidVectorLower = pidVectorTPCPtLower;
     } else {
-      nSigmaToUse = (candidate.pt() > cfgPtMaxforTPCOnlyPID && candidate.hasTOF()) ? nSigmaCombined : nSigmaTPC;
-      pidVectorUpper = (candidate.pt() > cfgPtMaxforTPCOnlyPID && candidate.hasTOF()) ? cfgnSigmaCutRMSUpper.value : cfgnSigmaCutTPCUpper.value;
-      pidVectorLower = (candidate.pt() > cfgPtMaxforTPCOnlyPID && candidate.hasTOF()) ? cfgnSigmaCutRMSLower.value : cfgnSigmaCutTPCLower.value;
+      if(candidate.pt() > cfgPtMaxforTPCOnlyPID && candidate.hasTOF()){
+        nSigmaToUse = nSigmaCombined;
+        pidVectorUpper = cfgnSigmaCutRMSUpper.value;
+        pidVectorLower = cfgnSigmaCutRMSLower.value;
+      }else if(candidate.pt() > cfgPtMaxforTPCOnlyPID && !candidate.hasTOF() && cfgUseStrictPID){
+        return 0;
+      }else{
+        nSigmaToUse = nSigmaTPC;
+        pidVectorUpper = cfgnSigmaCutTPCUpper.value;
+        pidVectorLower = cfgnSigmaCutTPCLower.value;
+      }
     }
     float nsigma = 9999.99;
     const int nPOI = 3;
@@ -448,20 +457,27 @@ struct FillPIDcolums {
         }
       }
     }
-    if (cfgOpenAllowCrossTrack) {
-      // one track can be recognized as different PID particles
+    if(cfgUseStrictPID){
+      // Only use the track which was recognized as an unique PID particle
       int index = (kIsPr << 2) | (kIsKa << 1) | kIsPi;
-      const int map[] = {0, 1, 2, 7, 3, 8, 9, 10};
+      const int map[] = {0, 1, 2, 0, 3, 0, 0, 0};
       return map[index];
-    } else {
-      // Select particle with the lowest nsigma (If not allow cross track)
-      for (int i = 0; i < nPOI; ++i) {
-        if (std::abs(nSigmaToUse[i]) < nsigma && (nSigmaToUse[i] > pidVectorLower[i] && nSigmaToUse[i] < pidVectorUpper[i])) {
-          pid = i;
-          nsigma = std::abs(nSigmaToUse[i]);
+    }else{
+      if (cfgOpenAllowCrossTrack) {
+        // one track can be recognized as different PID particles
+        int index = (kIsPr << 2) | (kIsKa << 1) | kIsPi;
+        const int map[] = {0, 1, 2, 7, 3, 8, 9, 10};
+        return map[index];
+      } else {
+        // Select particle with the lowest nsigma (If not allow cross track)
+        for (int i = 0; i < nPOI; ++i) {
+          if (std::abs(nSigmaToUse[i]) < nsigma && (nSigmaToUse[i] > pidVectorLower[i] && nSigmaToUse[i] < pidVectorUpper[i])) {
+            pid = i;
+            nsigma = std::abs(nSigmaToUse[i]);
+          }
         }
+        return pid + 1; // shift the pid by 1, 1 = pion, 2 = kaon, 3 = proton
       }
-      return pid + 1; // shift the pid by 1, 1 = pion, 2 = kaon, 3 = proton
     }
     // Clear the vectors
     std::vector<float>().swap(pidVectorLower);