Add TF Boarder cut

BongHwi · Feb 29, 2024 · d2c6784 · d2c6784
1 parent 7d7fe75
commit d2c6784
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/PWGLF/TableProducer/LFResonanceInitializer.cxx b/PWGLF/TableProducer/LFResonanceInitializer.cxx
@@ -103,6 +103,7 @@ struct reso2initializer {
   Configurable<bool> ConfEvtTriggerCheck{"ConfEvtTriggerCheck", false, "Evt sel: check for trigger"};
   Configurable<int> ConfEvtTriggerSel{"ConfEvtTriggerSel", 8, "Evt sel: trigger"};
   Configurable<bool> ConfEvtOfflineCheck{"ConfEvtOfflineCheck", true, "Evt sel: check for offline selection"};
+  Configurable<bool> ConfEvtTFBorderCut{"ConfEvtTFBorderCut", false, "Evt sel: apply TF border cut"};
 
   Configurable<std::string> cfgMultName{"cfgMultName", "FT0M", "The name of multiplicity estimator"};
 
@@ -847,6 +848,7 @@ struct reso2initializer {
       colCuts.setCuts(ConfEvtZvtx, ConfEvtTriggerCheck, ConfEvtTriggerSel, ConfEvtOfflineCheck, true);
     }
     colCuts.init(&qaRegistry);
+    colCuts.setApplyTFBorderCut(ConfEvtTFBorderCut);
     if (!ConfBypassCCDB) {
       ccdb->setURL(ccdburl.value);
       ccdb->setCaching(true);

diff --git a/PWGLF/Utils/collisionCuts.h b/PWGLF/Utils/collisionCuts.h
@@ -22,6 +22,7 @@
 
 #include "Framework/HistogramRegistry.h"
 #include "Framework/Logger.h"
+#include "Common/DataModel/EventSelection.h"
 
 using namespace o2::framework;
 
@@ -46,6 +47,7 @@ class CollisonCuts
     mTrigger = trig;
     mCheckOffline = checkOffline;
     mCheckIsRun3 = checkRun3;
+    mApplyTFBorderCut = false;
   }
 
   /// Initializes histograms for the task
@@ -76,6 +78,9 @@ class CollisonCuts
   /// Scan the trigger alias of the event
   void setInitialTriggerScan(bool checkTrigger) { mInitialTriggerScan = checkTrigger; }
 
+  /// Set the time frame border cut
+  void setApplyTFBorderCut(bool applyTFBorderCut) { mApplyTFBorderCut = applyTFBorderCut; }
+
   /// Check whether the collisions fulfills the specified selections
   /// \tparam T type of the collision
   /// \param col Collision
@@ -92,6 +97,11 @@ class CollisonCuts
         LOGF(debug, "Offline selection failed (Run3)");
         return false;
       }
+      if (!col.selection_bit(aod::evsel::kNoTimeFrameBorder) && mApplyTFBorderCut)
+      {
+        LOGF(debug, "Time frame border cut failed");
+        return false;
+      }
     } else { // Run2 case
       if (mCheckOffline && !col.sel7()) {
         LOGF(debug, "Offline selection failed (sel7)");
@@ -148,6 +158,7 @@ class CollisonCuts
   bool mCheckOffline = false;                      ///< Check for offline criteria (might change)
   bool mCheckIsRun3 = false;                       ///< Check if running on Pilot Beam
   bool mInitialTriggerScan = false;                ///< Check trigger when the event is first selected
+  bool mApplyTFBorderCut = false;                  ///< Apply time frame border cut
   int mTrigger = kINT7;                            ///< Trigger to check for
   float mZvtxMax = 999.f;                          ///< Maximal deviation from nominal z-vertex (cm)
 };