[PWGEM/Dilepton] update taggingHFE.cxx for ITSib requirement

Author: dsekihat

PWGEM/Dilepton/Tasks/taggingHFE.cxx

@@ -199,8 +199,8 @@ struct taggingHFE {
     Configurable<float> cfg_max_chi2its{"cfg_max_chi2its", 36.0, "max chi2/NclsITS"};
     Configurable<int> cfg_min_ncluster_its{"cfg_min_ncluster_its", 2, "min ncluster its"};
     Configurable<int> cfg_min_ncluster_itsib{"cfg_min_ncluster_itsib", 0, "min ncluster itsib"};
+    Configurable<int> cfg_itsib_type{"cfg_itsib_type", 0, "0:free, 1:OR, 2:AND between 2 legs, else:free"};
     Configurable<float> cfg_min_dcaxy{"cfg_min_dcaxy", 0.1, "min dca XY for v0 legs in cm"};
-
     Configurable<float> cfg_max_alpha_veto{"cfg_max_alpha_veto", 0.95, "max alpha for photon conversion rejection"};
     Configurable<float> cfg_max_qt_veto{"cfg_max_qt_veto", 0.01, "max qT for photon conversion rejection"};
 
@@ -241,6 +241,12 @@ struct taggingHFE {
     Configurable<float> cfg_min_dcaxy_v0leg{"cfg_min_dcaxy_v0leg", 0.1, "min dca XY for v0 legs in cm"};
     Configurable<float> cfg_min_dcaxy_bachelor{"cfg_min_dcaxy_bachelor", 0.05, "min dca XY for bachelor in cm"};
     Configurable<float> cfg_min_dcaxy_v0{"cfg_min_dcaxy_v0", 0.0, "min dca XY for V0 in cm"};
+
+    Configurable<int> cfg_itsib_type{"cfg_itsib_type", 0, "0:free, 1:OR, 2:AND between 2 legs, else:free"};
+    Configurable<int> cfg_min_ncluster_its_v0leg{"cfg_min_ncluster_its_v0leg", 2, "min ncluster its"};
+    Configurable<int> cfg_min_ncluster_itsib_v0leg{"cfg_min_ncluster_itsib_v0leg", 0, "min ncluster itsib"};
+    Configurable<int> cfg_min_ncluster_its_bachelor{"cfg_min_ncluster_its_bachelor", 2, "min ncluster its"};
+    Configurable<int> cfg_min_ncluster_itsib_bachelor{"cfg_min_ncluster_itsib_bachelor", 0, "min ncluster itsib"};
   } cascadeCut;
 
   struct : ConfigurableGroup {
@@ -768,9 +774,10 @@ struct taggingHFE {
     return (cascadeCut.cfg_min_mass_Omega < cascade.mOmega() && cascade.mOmega() < cascadeCut.cfg_max_mass_Omega) && (cascade.mXi() < cascadeCut.cfg_min_mass_Xi_veto || cascadeCut.cfg_max_mass_Xi_veto < cascade.mXi());
   }
 
-  template <bool isMC = true, typename TTrack>
+  template <int trackType = 0, bool isMC = true, typename TTrack>
   bool isSelectedV0Leg(TTrack const& track)
   {
+    // trackType = 0:v0leg, 1:v0leg in cascade, 2:bachelor of cascade only for ITS requirements
     if constexpr (isMC) {
       if (!track.has_mcParticle()) {
         return false;
@@ -781,15 +788,21 @@ struct taggingHFE {
       return false;
     }
 
-    if (track.itsChi2NCl() > v0Cut.cfg_max_chi2its) {
-      return false;
-    }
-
-    if (track.itsNCls() < v0Cut.cfg_min_ncluster_its) {
-      return false;
+    if constexpr (trackType == 0) {
+      if (track.itsNCls() < v0Cut.cfg_min_ncluster_its) { // must be 2
+        return false;
+      }
+    } else if constexpr (trackType == 1) {
+      if (track.itsNCls() < cascadeCut.cfg_min_ncluster_its_v0leg) { // must be 2
+        return false;
+      }
+    } else if constexpr (trackType == 2) {
+      if (track.itsNCls() < cascadeCut.cfg_min_ncluster_its_bachelor) { // must be 2
+        return false;
+      }
     }
 
-    if (track.itsNClsInnerBarrel() < v0Cut.cfg_min_ncluster_itsib) {
+    if (track.itsChi2NCl() > v0Cut.cfg_max_chi2its) {
       return false;
     }
 
@@ -816,6 +829,35 @@ struct taggingHFE {
     return true;
   }
 
+  template <int typeSV, typename TTrack>
+  bool checkITSibForV0Legs(TTrack const& t1, TTrack const& t2)
+  {
+    // typeSV = 0:v0, 1:cascade
+    if constexpr (typeSV == 0) {       // V0 legs
+      if (v0Cut.cfg_itsib_type == 0) { // free
+        return true;
+      } else if (v0Cut.cfg_itsib_type == 1) { // OR
+        return t1.itsNClsInnerBarrel() >= v0Cut.cfg_min_ncluster_itsib || t2.itsNClsInnerBarrel() >= v0Cut.cfg_min_ncluster_itsib;
+      } else if (v0Cut.cfg_itsib_type == 2) { // AND
+        return t1.itsNClsInnerBarrel() >= v0Cut.cfg_min_ncluster_itsib && t2.itsNClsInnerBarrel() >= v0Cut.cfg_min_ncluster_itsib;
+      } else {
+        return true;
+      }
+    } else if constexpr (typeSV == 1) {     // V0 legs in cascade
+      if (cascadeCut.cfg_itsib_type == 0) { // free
+        return true;
+      } else if (cascadeCut.cfg_itsib_type == 1) { // OR
+        return t1.itsNClsInnerBarrel() >= cascadeCut.cfg_min_ncluster_itsib_v0leg || t2.itsNClsInnerBarrel() >= cascadeCut.cfg_min_ncluster_itsib_v0leg;
+      } else if (cascadeCut.cfg_itsib_type == 2) { // AND
+        return t1.itsNClsInnerBarrel() >= cascadeCut.cfg_min_ncluster_itsib_v0leg && t2.itsNClsInnerBarrel() >= cascadeCut.cfg_min_ncluster_itsib_v0leg;
+      } else {
+        return true;
+      }
+    } else {
+      return true;
+    }
+  }
+
   template <typename TCollision>
   void fillEventHistograms(TCollision const& collision)
   {
@@ -1180,7 +1222,7 @@ struct taggingHFE {
         if (pos.sign() * neg.sign() > 0) {
           continue;
         }
-        if (!isSelectedV0Leg(pos) || !isSelectedV0Leg(neg)) {
+        if (!isSelectedV0Leg<0, true>(pos) || !isSelectedV0Leg<0, true>(neg)) {
           continue;
         }
 
@@ -1257,7 +1299,7 @@ struct taggingHFE {
           continue;
         }
 
-        if (!isSelectedV0Leg(pos) || !isSelectedV0Leg(neg) || !isSelectedV0Leg(bachelor)) {
+        if (!isSelectedV0Leg<1, true>(pos) || !isSelectedV0Leg<1, true>(neg) || !isSelectedV0Leg<2, true>(bachelor)) {
           continue;
         }
 
@@ -1517,7 +1559,11 @@ struct taggingHFE {
       for (const auto& v0 : v0s_per_coll) {
         auto pos = v0.template posTrack_as<TTracks>();
         auto neg = v0.template negTrack_as<TTracks>();
-        if (!isSelectedV0Leg<isMC>(pos) || !isSelectedV0Leg<isMC>(neg)) {
+        if (!isSelectedV0Leg<0, isMC>(pos) || !isSelectedV0Leg<0, isMC>(neg)) {
+          continue;
+        }
+
+        if (!checkITSibForV0Legs<0>(pos, neg)) {
           continue;
         }
 
@@ -1570,7 +1616,10 @@ struct taggingHFE {
           continue;
         }
 
-        if (!isSelectedV0Leg<isMC>(pos) || !isSelectedV0Leg<isMC>(neg) || !isSelectedV0Leg<isMC>(bachelor)) {
+        if (!isSelectedV0Leg<1, isMC>(pos) || !isSelectedV0Leg<1, isMC>(neg) || !isSelectedV0Leg<2, isMC>(bachelor)) {
+          continue;
+        }
+        if (!checkITSibForV0Legs<1>(pos, neg)) {
           continue;
         }