[PWGUD] Adding RCTflags selections to personal tasks (#11907)

Co-authored-by: ALICE Action Bot <alibuild@cern.ch>

Author: rolavick

PWGUD/TableProducer/tauEventTableProducer.cxx

@@ -68,6 +68,7 @@ struct TauEventTableProducer {
     Configurable<bool> useNumContribs{"useNumContribs", false, {"Use coll.numContribs as event cut"}};
     Configurable<int> cutRecoFlag{"cutRecoFlag", 1, {"0 = std mode, 1 = upc mode"}};
     Configurable<bool> useRecoFlag{"useRecoFlag", false, {"Use coll.flags as event cut"}};
+    Configurable<int> cutRCTflag{"cutRCTflag", 0, {"0 = off, 1 = CBT, 2 = CBT+ZDC, 3 = CBThadron, 4 = CBThadron+ZDC"}};
     Configurable<float> cutTrueGapSideFV0{"cutTrueGapSideFV0", 180000, "FV0A threshold for SG selector"};
     Configurable<float> cutTrueGapSideFT0A{"cutTrueGapSideFT0A", 150., "FT0A threshold for SG selector"};
     Configurable<float> cutTrueGapSideFT0C{"cutTrueGapSideFT0C", 50., "FT0C threshold for SG selector"};
@@ -153,6 +154,23 @@ struct TauEventTableProducer {
     return true;
   }
 
+  template <typename C>
+  bool isGoodRCTflag(C const& coll)
+  {
+    switch (cutSample.cutRCTflag) {
+      case 1:
+        return sgSelector.isCBTOk(coll);
+      case 2:
+        return sgSelector.isCBTZdcOk(coll);
+      case 3:
+        return sgSelector.isCBTHadronOk(coll);
+      case 4:
+        return sgSelector.isCBTHadronZdcOk(coll);
+      default:
+        return true;
+    }
+  }
+
   template <typename C>
   bool isGoodROFtime(C const& coll)
   {
@@ -324,15 +342,16 @@ struct TauEventTableProducer {
                      FullUDTracks const& tracks)
   {
 
-    int gapSide = collision.gapSide();
-    int trueGapSide = sgSelector.trueGap(collision, cutSample.cutTrueGapSideFV0, cutSample.cutTrueGapSideFT0A, cutSample.cutTrueGapSideFT0C, cutSample.cutTrueGapSideZDC);
-
-    if (cutSample.useTrueGap)
-      gapSide = trueGapSide;
+    if (!isGoodRCTflag(collision))
+      return;
 
     if (!isGoodROFtime(collision))
       return;
 
+    int gapSide = collision.gapSide();
+    int trueGapSide = sgSelector.trueGap(collision, cutSample.cutTrueGapSideFV0, cutSample.cutTrueGapSideFT0A, cutSample.cutTrueGapSideFT0C, cutSample.cutTrueGapSideZDC);
+    if (cutSample.useTrueGap)
+      gapSide = trueGapSide;
     if (gapSide != cutSample.whichGapSide)
       return;
 

PWGUD/TableProducer/twoTracksEventTableProducer.cxx

@@ -73,6 +73,7 @@ struct TwoTracksEventTableProducer {
     Configurable<bool> useNumContribs{"useNumContribs", true, {"Use coll.numContribs as event cut"}};
     Configurable<int> cutRecoFlag{"cutRecoFlag", 1, {"0 = std mode, 1 = upc mode"}};
     Configurable<bool> useRecoFlag{"useRecoFlag", false, {"Use coll.flags as event cut"}};
+    Configurable<int> cutRCTflag{"cutRCTflag", 0, {"0 = off, 1 = CBT, 2 = CBT+ZDC, 3 = CBThadron, 4 = CBThadron+ZDC"}};
     Configurable<float> cutTrueGapSideFV0{"cutTrueGapSideFV0", 180000, "FV0A threshold for SG selector"};
     Configurable<float> cutTrueGapSideFT0A{"cutTrueGapSideFT0A", 150., "FT0A threshold for SG selector"};
     Configurable<float> cutTrueGapSideFT0C{"cutTrueGapSideFT0C", 50., "FT0C threshold for SG selector"};
@@ -161,6 +162,23 @@ struct TwoTracksEventTableProducer {
     return true;
   }
 
+  template <typename C>
+  bool isGoodRCTflag(C const& coll)
+  {
+    switch (cutSample.cutRCTflag) {
+      case 1:
+        return sgSelector.isCBTOk(coll);
+      case 2:
+        return sgSelector.isCBTZdcOk(coll);
+      case 3:
+        return sgSelector.isCBTHadronOk(coll);
+      case 4:
+        return sgSelector.isCBTHadronZdcOk(coll);
+      default:
+        return true;
+    }
+  }
+
   template <typename C>
   bool isGoodROFtime(C const& coll)
   {
@@ -346,15 +364,18 @@ struct TwoTracksEventTableProducer {
     histos.get<TH1>(HIST("Reco/hSelections"))->Fill(nSelection);
     nSelection++;
 
+    if (!isGoodRCTflag(collision))
+      return;
+    histos.get<TH1>(HIST("Reco/hSelections"))->Fill(nSelection);
+    nSelection++;
+
     if (!isGoodROFtime(collision))
       return;
 
     int gapSide = collision.gapSide();
     int trueGapSide = sgSelector.trueGap(collision, cutSample.cutTrueGapSideFV0, cutSample.cutTrueGapSideFT0A, cutSample.cutTrueGapSideFT0C, cutSample.cutTrueGapSideZDC);
-
     if (cutSample.useTrueGap)
       gapSide = trueGapSide;
-
     if (gapSide != cutSample.whichGapSide)
       return;
     histos.get<TH1>(HIST("Reco/hSelections"))->Fill(nSelection);

PWGUD/Tasks/upcTauRl.cxx

@@ -151,6 +151,7 @@ struct UpcTauRl {
     Configurable<bool> useNumContribs{"useNumContribs", true, {"Use coll.numContribs as event cut"}};
     Configurable<int> cutRecoFlag{"cutRecoFlag", 1, {"0 = std mode, 1 = upc mode"}};
     Configurable<bool> useRecoFlag{"useRecoFlag", false, {"Use coll.flags as event cut"}};
+    Configurable<int> cutRCTflag{"cutRCTflag", 0, {"0 = off, 1 = CBT, 2 = CBT+ZDC, 3 = CBThadron, 4 = CBThadron+ZDC"}};
     Configurable<float> cutTrueGapSideFV0{"cutTrueGapSideFV0", -1, "FV0A threshold for SG selector"};
     Configurable<float> cutTrueGapSideFT0A{"cutTrueGapSideFT0A", 150., "FT0A threshold for SG selector"};
     Configurable<float> cutTrueGapSideFT0C{"cutTrueGapSideFT0C", 50., "FT0C threshold for SG selector"};
@@ -863,6 +864,23 @@ struct UpcTauRl {
     return true;
   }
 
+  template <typename C>
+  bool isGoodRCTflag(C const& coll)
+  {
+    switch (cutSample.cutRCTflag) {
+      case 1:
+        return sgSelector.isCBTOk(coll);
+      case 2:
+        return sgSelector.isCBTZdcOk(coll);
+      case 3:
+        return sgSelector.isCBTHadronOk(coll);
+      case 4:
+        return sgSelector.isCBTHadronZdcOk(coll);
+      default:
+        return true;
+    }
+  }
+
   template <typename C>
   bool isGoodROFtime(C const& coll)
   {
@@ -2280,6 +2298,9 @@ struct UpcTauRl {
     fillRejectionReasonDG(reconstructedCollision);
     outputGlobalRejectionHistogram();
 
+    if (!isGoodRCTflag(reconstructedCollision))
+      return;
+
     if (!isGoodROFtime(reconstructedCollision))
       return;
 
@@ -2313,6 +2334,9 @@ struct UpcTauRl {
     fillRejectionReasonSG(reconstructedCollision);
     outputGlobalRejectionHistogram();
 
+    if (!isGoodRCTflag(reconstructedCollision))
+      return;
+
     int gapSide = reconstructedCollision.gapSide();
     int trueGapSide = sgSelector.trueGap(reconstructedCollision, cutSample.cutTrueGapSideFV0, cutSample.cutTrueGapSideFT0A, cutSample.cutTrueGapSideFT0C, cutSample.cutTrueGapSideZDC);