Feat: add GPUErrorQA class

Author: ariedel-cern

Detectors/TPC/qc/include/TPCQC/GPUErrorQA.h

@@ -18,6 +18,7 @@
 #define AliceO2_TPC_QC_GPUERRORQA_H
 
 #include <memory>
+#include <unordered_map>
 #include <gsl/span>
 
 // root includes
@@ -55,11 +56,14 @@ class GPUErrorQA
   /// Reset all histograms
   void resetHistograms();
 
+  /// return histograms
+  std::unordered_map<std::string, std::unique_ptr<TH1>>& getMapHist() { return mMapHist; };
+
   /// Dump results to a file
   void dumpToFile(std::string filename);
 
  private:
-  std::unique_ptr<TH1F> mHist;
+  std::unordered_map<std::string, std::unique_ptr<TH1>> mMapHist;
   ClassDefNV(GPUErrorQA, 1)
 };
 } // namespace qc

Detectors/TPC/qc/src/GPUErrorQA.cxx

@@ -13,14 +13,15 @@
 
 #include <cmath>
 #include <memory>
+#include <unordered_map>
 
 // root includes
 #include "TFile.h"
 #include <TH1.h>
 
 // o2 includes
 #include "TPCQC/GPUErrorQA.h"
-#include "GPUErrors.h"
+#include "GPUDefMacros.h"
 
 ClassImp(o2::tpc::qc::GPUErrorQA);
 
@@ -30,26 +31,49 @@ using namespace o2::tpc::qc;
 void GPUErrorQA::initializeHistograms()
 {
   TH1::AddDirectory(false);
-  mHist = std::make_unique<TH1F>("ErrorCounter", "ErrorCounter", o2::gpu::GPUErrors::getMaxErrors(), 0, o2::gpu::GPUErrors::getMaxErrors());
+
+  // get gpu error names
+  // copied from GPUErrors.h
+  static std::unordered_map<uint32_t, const char*> errorNames = {
+#define GPUCA_ERROR_CODE(num, name, ...) {num, GPUCA_M_STR(name)},
+#include "GPUErrorCodes.h"
+#undef GPUCA_ERROR_CODE
+  };
+
+  // 1D histogram counting all reported errors
+  mMapHist["ErrorCounter"] = std::make_unique<TH1F>("ErrorCounter", "ErrorCounter", errorNames.size(), 0, errorNames.size());
+  mMapHist["ErrorCounter"]->GetXaxis()->SetTitle("Error Codes");
+  mMapHist["ErrorCounter"]->GetYaxis()->SetTitle("Entries");
+  // for convienence, label each bin with the error name
+  for (size_t bin = 1; bin < mMapHist["ErrorCounter"]->GetNbinsX(); bin++) {
+    auto const& it = errorNames.find(bin);
+    mMapHist["ErrorCounter"]->GetXaxis()->SetBinLabel(bin, it->second);
+  }
 }
 //______________________________________________________________________________
 void GPUErrorQA::resetHistograms()
 {
-  mHist->Reset();
+  for (const auto& pair : mMapHist) {
+    pair.second->Reset();
+  }
 }
 //______________________________________________________________________________
 void GPUErrorQA::processErrors(gsl::span<const std::array<uint32_t, 4>> errors)
 {
   for (const auto& error : errors) {
     uint32_t errorCode = error[0];
-    mHist->Fill(static_cast<float>(errorCode));
+    mMapHist["ErrorCounter"]->Fill(static_cast<float>(errorCode));
   }
 }
 
 //______________________________________________________________________________
 void GPUErrorQA::dumpToFile(const std::string filename)
 {
-  auto f = std::unique_ptr<TFile>(TFile::Open(filename.c_str(), "recreate"));
-  mHist->Write();
-  f->Close();
+  auto f = std::unique_ptr<TFile>(TFile::Open(filename.data(), "recreate"));
+  for (const auto& [name, hist] : mMapHist) {
+    TObjArray arr;
+    arr.SetName(name.data());
+    arr.Add(hist.get());
+    arr.Write(arr.GetName(), TObject::kSingleKey);
+  }
 }