Generalize Cluster task to also accept KrClusters (#794)

Depends on AliceO2Group/AliceO2#6798

Author: wiechula

Modules/TPC/include/TPC/Clusters.h

@@ -62,6 +62,9 @@ class Clusters /*final*/ : public TaskInterface // todo add back the "final" whe
   std::vector<std::unique_ptr<TCanvas>> mSigmaTimeCanvasVec{}; ///< summary canvases of the SigmaTime object
   std::vector<std::unique_ptr<TCanvas>> mSigmaPadCanvasVec{};  ///< summary canvases of the SigmaPad object
   std::vector<std::unique_ptr<TCanvas>> mTimeBinCanvasVec{};   ///< summary canvases of the TimeBin object
+
+  void processClusterNative(o2::framework::InputRecord& inputs);
+  void processKrClusters(o2::framework::InputRecord& inputs);
 };
 
 } // namespace o2::quality_control_modules::tpc

Modules/TPC/run/tpcQCKrClusters_direct.json

@@ -0,0 +1,52 @@
+{
+  "qc": {
+    "config": {
+      "database": {
+        "implementation": "CCDB",
+        "host": "ccdb-test.cern.ch:8080",
+        "username": "not_applicable",
+        "password": "not_applicable",
+        "name": "not_applicable"
+      },
+      "Activity": {
+        "number": "42",
+        "type": "2"
+      },
+      "monitoring": {
+        "url": "infologger:///debug?qc"
+      },
+      "consul": {
+        "url": "http://consul-test.cern.ch:8500"
+      },
+      "conditionDB": {
+        "url": "ccdb-test.cern.ch:8080"
+      }
+    },
+    "tasks": {
+      "KrClusters": {
+        "active": "true",
+        "className": "o2::quality_control_modules::tpc::Clusters",
+        "moduleName": "QcTPC",
+        "detectorName": "TPC",
+        "cycleDurationSeconds": "10",
+        "maxNumberCycles": "-1",
+        "resetAfterCycles": "5",
+        "dataSource": {
+          "type": "direct",
+          "query" : "krClusters:TPC/KRCLUSTERS"
+        },
+        "taskParameters": {
+          "NClustersNBins": "100",  "NClustersXMin": "0", "NClustersXMax": "100",
+          "QmaxNBins":      "200",  "QmaxXMin":      "0", "QmaxXMax":      "1024",
+          "QtotNBins":      "600",  "QtotXMin":      "0", "QtotXMax":      "6000",
+          "SigmaPadNBins":  "200",  "SigmaPadXMin":  "0", "SigmaPadXMax":  "3.8",
+          "SigmaTimeNBins": "200",  "SigmaTimeXMin": "0", "SigmaTimeXMax": "2.5",
+          "TimeBinNBins":   "1000", "TimeBinXMin":   "0", "TimeBinXMax":   "100000"
+        },
+        "location": "remote"
+      }
+    }
+  },
+  "dataSamplingPolicies": [
+  ]
+}

Modules/TPC/src/Clusters.cxx

@@ -16,14 +16,19 @@
 
 // O2 includes
 #include "Framework/ProcessingContext.h"
+#include "DataFormatsTPC/KrCluster.h"
 #include "DataFormatsTPC/ClusterNative.h"
 #include "TPCBase/Painter.h"
+#include "Framework/InputRecordWalker.h"
 
 // QC includes
 #include "QualityControl/QcInfoLogger.h"
 #include "TPC/Clusters.h"
 #include "TPC/Utility.h"
 
+using namespace o2::framework;
+using namespace o2::tpc;
+
 namespace o2::quality_control_modules::tpc
 {
 
@@ -63,9 +68,12 @@ void Clusters::startOfCycle()
   QcInfoLogger::GetInstance() << "startOfCycle" << AliceO2::InfoLogger::InfoLogger::endm;
 }
 
-void Clusters::monitorData(o2::framework::ProcessingContext& ctx)
+void Clusters::processClusterNative(o2::framework::InputRecord& inputs)
 {
-  o2::tpc::ClusterNativeAccess clusterIndex = clusterHandler(ctx.inputs());
+  o2::tpc::ClusterNativeAccess clusterIndex = clusterHandler(inputs);
+  if (!clusterIndex.nClustersTotal) {
+    return;
+  }
 
   for (int isector = 0; isector < o2::tpc::constants::MAXSECTOR; ++isector) {
     for (int irow = 0; irow < o2::tpc::constants::MAXGLOBALPADROW; ++irow) {
@@ -76,6 +84,27 @@ void Clusters::monitorData(o2::framework::ProcessingContext& ctx)
       }
     }
   }
+}
+
+void Clusters::processKrClusters(o2::framework::InputRecord& inputs)
+{
+  std::vector<InputSpec> filterKr = {
+    { "krClusters", ConcreteDataTypeMatcher{ "TPC", "KRCLUSTERS" }, Lifetime::Timeframe },
+    { "sampled-krClusters", ConcreteDataTypeMatcher{ "DS", "KRCLUSTERS" }, Lifetime::Timeframe },
+  };
+
+  for (auto const& inputRef : InputRecordWalker(inputs, filterKr)) {
+    auto krClusters = inputs.get<gsl::span<o2::tpc::KrCluster>>(inputRef);
+    for (const auto& cl : krClusters) {
+      mQCClusters.processCluster(cl, o2::tpc::Sector(cl.sector), int(cl.meanRow));
+    }
+  }
+}
+
+void Clusters::monitorData(o2::framework::ProcessingContext& ctx)
+{
+  processClusterNative(ctx.inputs());
+  processKrClusters(ctx.inputs());
 
   mQCClusters.analyse();