PWGEM/PhotonMeson: update createPCM for really TPC only tracks

Author: dsekihat

PWGEM/PhotonMeson/TableProducer/createPCM.cxx

@@ -76,6 +76,7 @@ struct createPCM {
   Configurable<float> maxeta{"maxeta", 0.9, "eta acceptance for single track"};
   Configurable<int> mincrossedrows{"mincrossedrows", 10, "min crossed rows"};
   Configurable<float> maxchi2tpc{"maxchi2tpc", 4.0, "max chi2/NclsTPC"};
+  Configurable<bool> useTPConly{"useTPConly", false, "Use truly TPC only tracks for V0 finder"};
 
   int mRunNumber;
   float d_bz;
@@ -178,13 +179,6 @@ struct createPCM {
     array<float, 3> pvec0 = {0.};
     array<float, 3> pvec1 = {0.};
 
-    if (ele.tpcNClsCrossedRows() < mincrossedrows) {
-      return;
-    }
-    if (pos.tpcNClsCrossedRows() < mincrossedrows) {
-      return;
-    }
-
     auto pTrack = getTrackParCov(pos); // positive
     auto nTrack = getTrackParCov(ele); // negative
 
@@ -244,6 +238,22 @@ struct createPCM {
       auto posTracks_coll = posTracks->sliceByCached(o2::aod::track::collisionId, collision.globalIndex(), cache);
 
       for (auto& [ele, pos] : combinations(CombinationsFullIndexPolicy(negTracks_coll, posTracks_coll))) {
+        if (ele.tpcNClsCrossedRows() < mincrossedrows) {
+          continue;
+        }
+        if (pos.tpcNClsCrossedRows() < mincrossedrows) {
+          continue;
+        }
+
+        if (useTPConly) {
+          if ((ele.hasITS() || ele.hasTOF() || ele.hasTRD())) {
+            continue;
+          }
+          if ((pos.hasITS() || pos.hasTOF() || pos.hasTRD())) {
+            continue;
+          }
+        }
+
         fillV0Table(collision, ele, pos);
       }
     } // end of collision loop
@@ -285,6 +295,15 @@ struct createPCM {
           continue;
         }
 
+        if (useTPConly) {
+          if ((ele.hasITS() || ele.hasTOF() || ele.hasTRD())) {
+            continue;
+          }
+          if ((pos.hasITS() || pos.hasTOF() || pos.hasTRD())) {
+            continue;
+          }
+        }
+
         if (ele.sign() < 0) {
           fillV0Table(collision, ele, pos);
         } else {