[PWGLF] Add zorro to non-prompt cascade (#9882)

Author: mpuccio

PWGLF/Tasks/Strangeness/CMakeLists.txt

@@ -51,7 +51,7 @@ o2physics_add_dpl_workflow(cascadecorrelations
 
 o2physics_add_dpl_workflow(non-prompt-cascade
     SOURCES nonPromptCascade.cxx
-    PUBLIC_LINK_LIBRARIES O2::Framework O2::ReconstructionDataFormats O2Physics::AnalysisCore O2::DetectorsBase
+    PUBLIC_LINK_LIBRARIES O2::Framework O2::ReconstructionDataFormats O2Physics::AnalysisCore O2::DetectorsBase O2Physics::EventFilteringUtils
     COMPONENT_NAME Analysis)
 
 o2physics_add_dpl_workflow(k0mixedevents

PWGLF/Tasks/Strangeness/nonPromptCascade.cxx

@@ -26,6 +26,8 @@
 #include "DataFormatsTPC/BetheBlochAleph.h"
 #include "DCAFitter/DCAFitterN.h"
 #include "DetectorsBase/Propagator.h"
+#include "EventFiltering/Zorro.h"
+#include "EventFiltering/ZorroSummary.h"
 #include "Framework/AnalysisTask.h"
 #include "Framework/AnalysisDataModel.h"
 #include "Framework/ASoA.h"
@@ -191,6 +193,10 @@ struct NonPromptCascadeTask {
 
   Configurable<float> cfgCutNclusTPC{"cfgCutNclusTPC", 70, "Minimum number of TPC clusters"};
   Configurable<LabeledArray<float>> cfgCutsPID{"particlesCutsPID", {cutsPID[0], nParticles, nCutsPID, particlesNames, cutsNames}, "Nuclei PID selections"};
+  Configurable<bool> cfgSkimmedProcessing{"cfgSkimmedProcessing", true, "Skimmed dataset processing"};
+
+  Zorro zorro;
+  OutputObj<ZorroSummary> zorroSummary{"zorroSummary"};
 
   Service<o2::ccdb::BasicCCDBManager> ccdb;
   int mRunNumber = 0;
@@ -263,6 +269,7 @@ struct NonPromptCascadeTask {
 
   void init(InitContext const&)
   {
+    zorroSummary.setObject(zorro.getZorroSummary());
     ccdb->setURL(ccdbUrl);
     ccdb->setFatalWhenNull(true);
     ccdb->setCaching(true);
@@ -652,6 +659,18 @@ struct NonPromptCascadeTask {
                                   aod::V0s const& /*v0s*/, TracksExtData const& /*tracks*/,
                                   aod::BCsWithTimestamps const&)
   {
+    if (cfgSkimmedProcessing) {
+      int runNumber{-1};
+      for (const auto& coll : collisions) {
+        auto bc = coll.bc_as<aod::BCsWithTimestamps>();
+        if (runNumber != bc.runNumber()) {
+          zorro.initCCDB(ccdb.service, bc.runNumber(), bc.timestamp(), "fTrackedOmega");
+          zorro.populateHistRegistry(registry, bc.runNumber());
+          runNumber = bc.runNumber();
+        }
+        zorro.isSelected(bc.globalBC()); /// Just let Zorro do the accounting
+      }
+    }
     fillCandidatesVector<TracksExtData>(collisions, trackedCascades);
     for (const auto& c : candidates) {
       NPCTable(c.matchingChi2, c.deltaPt, c.itsClusSize, c.hasReassociatedCluster,