[PWGHF] Use the Hf event selection for both tracks and charm hadrons (#8706)

Author: zhangbiao-phy

PWGHF/HFC/TableProducer/CMakeLists.txt

@@ -61,5 +61,5 @@ o2physics_add_dpl_workflow(correlator-lc-hadrons
 
 o2physics_add_dpl_workflow(femto-dream-producer
                     SOURCES femtoDreamProducer.cxx
-                    PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
+                    PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore O2Physics::EventFilteringUtils
                     COMPONENT_NAME Analysis)

PWGHF/HFC/TableProducer/femtoDreamProducer.cxx

@@ -12,6 +12,7 @@
 /// \file femtoDreamProducer.cxx
 /// \brief Tasks that produces the track tables used for the pairing
 /// \author Ravindra Singh, GSI, ravindra.singh@cern.ch
+/// \author Biao Zhang, Heidelberg University, biao.zhang@cern.ch
 
 #include <string>
 #include <vector>
@@ -39,11 +40,25 @@
 #include "PWGHF/DataModel/CandidateReconstructionTables.h"
 #include "PWGHF/DataModel/CandidateSelectionTables.h"
 #include "PWGHF/Utils/utilsBfieldCCDB.h"
+#include "PWGHF/Utils/utilsEvSelHf.h"
+#include "PWGHF/Core/CentralityEstimation.h"
 
 using namespace o2;
 using namespace o2::framework;
 using namespace o2::framework::expressions;
 using namespace o2::analysis::femtoDream;
+using namespace o2::hf_evsel;
+using namespace o2::hf_centrality;
+
+// event types
+enum Event : uint8_t {
+  kAll = 0,
+  kRejEveSel,
+  kRejNoTracksAndCharm,
+  kTrackSelected,
+  kCharmSelected,
+  kPairSelected
+};
 
 struct HfFemtoDreamProducer {
 
@@ -69,19 +84,6 @@ struct HfFemtoDreamProducer {
 
   // Configurable<bool> isForceGRP{"isForceGRP", false, "Set true if the magnetic field configuration is not available in the usual CCDB directory (e.g. for Run 2 converted data or unanchorad Monte Carlo)"};
 
-  /// Event selection
-  // Configurable<float> evtZvtx{"evtZvtx", 10.f, "Evt sel: Max. z-Vertex (cm)"};
-  // Configurable<bool> evtTriggerCheck{"evtTriggerCheck", true, "Evt sel: check for trigger"};
-  // Configurable<int> evtTriggerSel{"evtTriggerSel", kINT7, "Evt sel: trigger"};
-  // Configurable<bool> evtOfflineCheck{"evtOfflineCheck", false, "Evt sel: check for offline selection"};
-  // Configurable<bool> evtAddOfflineCheck{"evtAddOfflineCheck", false, "Evt sel: additional checks for offline selection (not part of sel8 yet)"};
-
-  Configurable<bool> evtAddOfflineCheck{"evtAddOfflineCheck", false, "Evt sel: additional checks for offline selection (not part of sel8 yet)"};
-  Configurable<bool> evtOfflineCheck{"evtOfflineCheck", false, "Evt sel: check for offline selection"};
-  Configurable<bool> evtTriggerCheck{"evtTriggerCheck", true, "Evt sel: check for trigger"};
-  Configurable<int> evtTriggerSel{"evtTriggerSel", kINT7, "Evt sel: trigger"};
-  Configurable<float> evtZvtx{"evtZvtx", 10.f, "Evt sel: Max. z-Vertex (cm)"};
-
   Configurable<bool> isDebug{"isDebug", true, "Enable Debug tables"};
   Configurable<bool> isRun3{"isRun3", true, "Running on Run3 or pilot"};
 
@@ -121,7 +123,6 @@ struct HfFemtoDreamProducer {
 
   using GeneratedMc = soa::Filtered<soa::Join<aod::McParticles, aod::HfCand3ProngMcGen>>;
 
-  FemtoDreamCollisionSelection colCuts;
   FemtoDreamTrackSelection trackCuts;
 
   Filter filterSelectCandidateLc = (aod::hf_sel_candidate_lc::isSelLcToPKPi >= selectionFlagLc || aod::hf_sel_candidate_lc::isSelLcToPiKP >= selectionFlagLc);
@@ -130,6 +131,7 @@ struct HfFemtoDreamProducer {
   HistogramRegistry TrackRegistry{"Tracks", {}, OutputObjHandlingPolicy::AnalysisObject};
 
   HfHelper hfHelper;
+  o2::hf_evsel::HfEventSelection hfEvSel;
 
   float magField;
   int runNumber;
@@ -148,8 +150,21 @@ struct HfFemtoDreamProducer {
     int CutBits = 8 * sizeof(o2::aod::femtodreamparticle::cutContainerType);
     TrackRegistry.add("AnalysisQA/CutCounter", "; Bit; Counter", kTH1F, {{CutBits + 1, -0.5, CutBits + 0.5}});
 
-    colCuts.setCuts(evtZvtx.value, evtTriggerCheck.value, evtTriggerSel.value, evtOfflineCheck.value, evtAddOfflineCheck.value, isRun3.value);
-    colCuts.init(&qaRegistry);
+    // event QA histograms
+    constexpr int kEventTypes = kPairSelected + 1;
+    std::string labels[kEventTypes];
+    labels[Event::kAll] = "All events";
+    labels[Event::kRejEveSel] = "rejected by event selection";
+    labels[Event::kRejNoTracksAndCharm] = "rejected by no tracks and charm";
+    labels[Event::kTrackSelected] = "with tracks ";
+    labels[Event::kCharmSelected] = "with charm hadrons ";
+    labels[Event::kPairSelected] = "with pairs";
+
+    static const AxisSpec axisEvents = {kEventTypes, 0.5, kEventTypes + 0.5, ""};
+    qaRegistry.add("hEventQA", "Events;;entries", HistType::kTH1F, {axisEvents});
+    for (int iBin = 0; iBin < kEventTypes; iBin++) {
+      qaRegistry.get<TH1>(HIST("hEventQA"))->GetXaxis()->SetBinLabel(iBin + 1, labels[iBin].data());
+    }
 
     trackCuts.setSelection(trkCharge, femtoDreamTrackSelection::kSign, femtoDreamSelection::kEqual);
     trackCuts.setSelection(trkPtmin, femtoDreamTrackSelection::kpTMin, femtoDreamSelection::kLowerLimit);
@@ -175,6 +190,8 @@ struct HfFemtoDreamProducer {
     ccdb->setCaching(true);
     ccdb->setLocalObjectValidityChecking();
 
+    hfEvSel.addHistograms(qaRegistry); // collision monitoring
+
     int64_t now = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now().time_since_epoch()).count();
     ccdb->setCreatedNotAfter(now);
   }
@@ -335,8 +352,7 @@ struct HfFemtoDreamProducer {
   {
     const auto vtxZ = col.posZ();
     const auto sizeCand = candidates.size();
-
-    const auto spher = colCuts.computeSphericity(col, tracks);
+    const auto spher = 2.; // dummy value for the moment
     float mult = 0;
     int multNtr = 0;
     if (isRun3) {
@@ -351,15 +367,20 @@ struct HfFemtoDreamProducer {
       multNtr = col.multTracklets();
     }
 
-    colCuts.fillQA(col, mult);
+    const auto rejectionMask = hfEvSel.getHfCollisionRejectionMask<true, CentralityEstimator::None, aod::BCsWithTimestamps>(col, mult, ccdb, qaRegistry);
+
+    qaRegistry.fill(HIST("hEventQA"), 1 + Event::kAll);
 
-    // check whether the basic event selection criteria are fulfilled
-    // that included checking if there is at least on usable track or V0
-    if (!colCuts.isSelectedCollision(col)) {
+    /// monitor the satisfied event selections
+    hfEvSel.fillHistograms(col, rejectionMask, mult);
+    if (rejectionMask != 0) {
+      /// at least one event selection not satisfied --> reject the candidate
+      qaRegistry.fill(HIST("hEventQA"), 1 + Event::kRejEveSel);
       return;
     }
 
-    if (colCuts.isEmptyCollision(col, tracks, trackCuts)) {
+    if (isNoSelectedTracks(col, tracks, trackCuts) && sizeCand <= 0) {
+      qaRegistry.fill(HIST("hEventQA"), 1 + Event::kRejNoTracksAndCharm);
       return;
     }
 
@@ -437,18 +458,39 @@ struct HfFemtoDreamProducer {
     aod::femtodreamcollision::BitMaskType bitTrack = 0;
     if (isTrackFilled) {
       bitTrack |= 1 << 0;
+      qaRegistry.fill(HIST("hEventQA"), 1 + Event::kTrackSelected);
     }
 
     aod::femtodreamcollision::BitMaskType bitCand = 0;
     if (sizeCand > 0) {
       bitCand |= 1 << 0;
+      qaRegistry.fill(HIST("hEventQA"), 1 + Event::kCharmSelected);
     }
 
+    if (isTrackFilled && (sizeCand > 0))
+      qaRegistry.fill(HIST("hEventQA"), 1 + Event::kPairSelected);
+
     rowMasks(static_cast<aod::femtodreamcollision::BitMaskType>(bitTrack),
              static_cast<aod::femtodreamcollision::BitMaskType>(bitCand),
              0);
   }
 
+  // check if there is no selected track
+  /// \param C type of the collision
+  /// \param T type of the tracks
+  /// \param TC type of the femto track cuts
+  /// \return whether or not the tracks fulfills the all selections
+  template <typename C, typename T, typename TC>
+  bool isNoSelectedTracks(C const& /*col*/, T const& tracks, TC& trackCuts)
+  {
+    for (auto const& track : tracks) {
+      if (trackCuts.isSelectedMinimal(track)) {
+        return false;
+      }
+    }
+    return true;
+  }
+
   template <typename ParticleType>
   void fillCharmHadMcGen(ParticleType particles)
   {

PWGHF/Utils/utilsEvSelHf.h

@@ -106,7 +106,7 @@ struct HfEventSelection : o2::framework::ConfigurableGroup {
   o2::framework::Configurable<bool> useNoCollInRofStandard{"useNoCollInRofStandard", false, "Reject collisions in ROF standard"};
   o2::framework::Configurable<std::string> softwareTrigger{"softwareTrigger", "", "Label of software trigger. Multiple triggers can be selected dividing them by a comma. Set None if you want bcs that are not selected by any trigger"};
   o2::framework::Configurable<uint64_t> bcMarginForSoftwareTrigger{"bcMarginForSoftwareTrigger", 100, "Number of BCs of margin for software triggers"};
-  o2::framework::Configurable<std::string> ccdbPathSoftwareTrigger{"ccdbPathSoftwareTrigger", "Users/m/mpuccio/EventFiltering/OTS/", "ccdb path for ZORRO objects"};
+  o2::framework::Configurable<std::string> ccdbPathSoftwareTrigger{"ccdbPathSoftwareTrigger", "Users/m/mpuccio/EventFiltering/OTS/Chunked/", "ccdb path for ZORRO objects"};
   o2::framework::ConfigurableAxis th2ConfigAxisCent{"th2ConfigAxisCent", {100, 0., 100.}, ""};
   o2::framework::ConfigurableAxis th2ConfigAxisOccupancy{"th2ConfigAxisOccupancy", {14, 0, 140000}, ""};