Merge pull request #46920 from mmusich/mm_dev_DeDxCalibration_PI

add stub of `DeDxCalibration` Payload Inspector

CondCore/PhysicsToolsPlugins/plugins/BuildFile.xml

@@ -8,3 +8,8 @@
   <use name="CondCore/CondDB"/>
   <use name="CondFormats/PhysicsToolsObjects"/>
 </library>
+
+<library file="DeDxCalibration_PayloadInspector.cc" name="DeDxCalibration_PayloadInspector">
+  <use name="CondCore/Utilities"/>
+  <use name="CondCore/CondDB"/>
+</library>

CondCore/PhysicsToolsPlugins/plugins/DeDxCalibration_PayloadInspector.cc

@@ -0,0 +1,201 @@
+#include "CondCore/CondDB/interface/Time.h"
+#include "CondCore/Utilities/interface/PayloadInspector.h"
+#include "CondCore/Utilities/interface/PayloadInspectorModule.h"
+#include "FWCore/MessageLogger/interface/MessageLogger.h"
+
+// the data format of the condition to be inspected
+#include "CondFormats/PhysicsToolsObjects/interface/DeDxCalibration.h"
+
+#include <array>
+#include <fstream>
+#include <iomanip>
+#include <iostream>
+#include <map>
+#include <memory>
+#include <numeric>
+#include <sstream>
+
+// include ROOT
+#include "TCanvas.h"
+#include "TF1.h"
+#include "TH1F.h"
+#include "TH2F.h"
+#include "TLatex.h"
+#include "TLegend.h"
+#include "TLine.h"
+#include "TPad.h"
+#include "TPave.h"
+#include "TPaveStats.h"
+#include "TStyle.h"
+
+namespace {
+
+  using namespace cond::payloadInspector;
+
+  /************************************************
+     DeDxCalibration Payload Inspector of 1 IOV 
+  *************************************************/
+  class DeDxCalibrationTest : public Histogram1D<DeDxCalibration, SINGLE_IOV> {
+  public:
+    DeDxCalibrationTest()
+        : Histogram1D<DeDxCalibration, SINGLE_IOV>("Test DeDxCalibration", "Test DeDxCalibration", 1, 0.0, 1.0) {}
+
+    bool fill() override {
+      auto tag = PlotBase::getTag<0>();
+      for (auto const& iov : tag.iovs) {
+        std::shared_ptr<DeDxCalibration> payload = Base::fetchPayload(std::get<1>(iov));
+        if (payload.get()) {
+          const auto& thresholds = payload->thr();
+          const auto& alphas = payload->alpha();
+          const auto& sigmas = payload->sigma();
+
+          assert(thresholds.size() == alphas.size());
+          assert(alphas.size() == sigmas.size());
+
+          for (unsigned int i = 0; i < thresholds.size(); i++) {
+            std::cout << "threshold:" << thresholds[i] << " alpha: " << alphas[i] << " sigma: " << sigmas[i]
+                      << std::endl;
+          }
+        }
+      }
+      return true;
+    }
+  };
+
+  // Inspector to show the values of thresholds, alphas, sigmas, and gains in DeDxCalibration
+  class DeDxCalibrationInspector : public PlotImage<DeDxCalibration, SINGLE_IOV> {
+  public:
+    DeDxCalibrationInspector() : PlotImage<DeDxCalibration, SINGLE_IOV>("DeDxCalibration Inspector") {}
+
+    bool fill() override {
+      gStyle->SetPalette(1);
+      gStyle->SetOptStat(0);
+
+      auto tag = PlotBase::getTag<0>();
+      auto iov = tag.iovs.front();
+
+      std::shared_ptr<DeDxCalibration> payload = fetchPayload(std::get<1>(iov));
+
+      if (payload.get()) {
+        const std::vector<double>& thr = payload->thr();
+        const std::vector<double>& alpha = payload->alpha();
+        const std::vector<double>& sigma = payload->sigma();
+
+        // Ensure that thr, alpha, sigma have the same size
+        assert(thr.size() == alpha.size());
+        assert(alpha.size() == sigma.size());
+
+        // Create a 2D histogram
+        int nBinsX = 3;           // For thr, alpha, sigma, gain
+        int nBinsY = thr.size();  // Number of elements in the vectors
+
+        TH2D h2("h2", "DeDxCalibration Values;Variable Type;Value", nBinsX, 0, nBinsX, nBinsY, 0, nBinsY);
+
+        // Label the x-axis with the variable names
+        h2.GetXaxis()->SetBinLabel(1, "Threshold");
+        h2.GetXaxis()->SetBinLabel(2, "#alpha");
+        h2.GetXaxis()->SetBinLabel(3, "#sigma");
+
+        // Fill the histogram
+        for (size_t i = 0; i < thr.size(); ++i) {
+          h2.Fill(0.5, i, thr[i]);
+          h2.Fill(1.5, i, alpha[i]);
+          h2.Fill(2.5, i, sigma[i]);
+        }
+
+        // Draw the histogram on a canvas
+        TCanvas canvas("Canvas", "DeDxCalibration Values", 1200, 800);
+        canvas.cd();
+        h2.Draw("COLZ TEXT");
+
+        std::string fileName(m_imageFileName);
+        canvas.SaveAs(fileName.c_str());
+
+        return true;
+      } else {
+        return false;
+      }
+    }
+  };
+
+  class DeDxCalibrationPlot : public cond::payloadInspector::PlotImage<DeDxCalibration, SINGLE_IOV> {
+  public:
+    DeDxCalibrationPlot()
+        : cond::payloadInspector::PlotImage<DeDxCalibration, SINGLE_IOV>(
+              "DeDxCalibration Thresholds, Alphas, Sigmas, and Gains") {}
+
+    bool fill() override {
+      gStyle->SetPalette(1);
+      gStyle->SetOptStat(0);
+
+      auto tag = PlotBase::getTag<0>();
+      auto iov = tag.iovs.front();
+
+      std::shared_ptr<DeDxCalibration> payload = fetchPayload(std::get<1>(iov));
+
+      if (!payload) {
+        return false;
+      }
+
+      // Prepare canvas
+      TCanvas canvas("DeDxCalibration", "DeDxCalibration", 1200, 800);
+      canvas.Divide(2, 2);
+
+      // Extract data
+      const auto& thresholds = payload->thr();
+      const auto& alphas = payload->alpha();
+      const auto& sigmas = payload->sigma();
+      const auto& gains = payload->gain();
+
+      // 1. Plot thresholds
+      canvas.cd(1);
+      auto h_thr = new TH1F("Thresholds", "Thresholds;Index;Value", thresholds.size(), 0, thresholds.size());
+      for (size_t i = 0; i < thresholds.size(); ++i) {
+        h_thr->SetBinContent(i + 1, thresholds[i]);
+      }
+      h_thr->SetFillColor(kBlue);
+      h_thr->Draw();
+
+      // 2. Plot alphas
+      canvas.cd(2);
+      auto h_alpha = new TH1F("Alphas", "Alphas;Index;Value", alphas.size(), 0, alphas.size());
+      for (size_t i = 0; i < alphas.size(); ++i) {
+        h_alpha->SetBinContent(i + 1, alphas[i]);
+      }
+      h_alpha->SetFillColor(kGreen);
+      h_alpha->Draw();
+
+      // 3. Plot sigmas
+      canvas.cd(3);
+      auto h_sigma = new TH1F("Sigmas", "Sigmas;Index;Value", sigmas.size(), 0, sigmas.size());
+      for (size_t i = 0; i < sigmas.size(); ++i) {
+        h_sigma->SetBinContent(i + 1, sigmas[i]);
+      }
+      h_sigma->SetFillColor(kRed);
+      h_sigma->Draw();
+
+      // 4. Plot aggregated gain values
+      canvas.cd(4);
+      const int numBins = 100;  // Set number of bins for aggregated gain
+      auto h_gain = new TH1F("Gains", "Aggregated Gains;Gain Range;Count", numBins, 0, 1.0);  // Adjust range if needed
+      for (const auto& [chip, gain] : gains) {
+        h_gain->Fill(gain);
+      }
+      h_gain->SetFillColor(kYellow);
+      h_gain->Draw();
+
+      // Save the canvas to a file
+      std::string fileName(m_imageFileName);
+      canvas.SaveAs(fileName.c_str());
+
+      return true;
+    }
+  };
+}  // namespace
+
+// Register the classes as boost python plugin
+PAYLOAD_INSPECTOR_MODULE(DeDxCalibration) {
+  PAYLOAD_INSPECTOR_CLASS(DeDxCalibrationTest);
+  PAYLOAD_INSPECTOR_CLASS(DeDxCalibrationInspector);
+  PAYLOAD_INSPECTOR_CLASS(DeDxCalibrationPlot);
+}

CondCore/PhysicsToolsPlugins/test/BuildFile.xml

@@ -4,3 +4,6 @@
 </bin>
 <bin file="test_PfCalibration_PayloadInspector.cpp" name="test_PfCalibration_PayloadInspector">
 </bin>
+<bin file="test_DeDxCalibration_PayloadInspector.cpp" name="test_DeDxCalibration_PayloadInspector">
+</bin>
+

CondCore/PhysicsToolsPlugins/test/test_DeDxCalibration_PayloadInspector.cpp

@@ -0,0 +1,42 @@
+#include <iostream>
+#include <sstream>
+#include "CondCore/Utilities/interface/PayloadInspector.h"
+#include "CondCore/PhysicsToolsPlugins/plugins/DeDxCalibration_PayloadInspector.cc"
+
+#include "FWCore/PluginManager/interface/PluginManager.h"
+#include "FWCore/PluginManager/interface/standard.h"
+#include "FWCore/ServiceRegistry/interface/ServiceRegistry.h"
+#include "FWCore/MessageLogger/interface/MessageLogger.h"
+
+int main(int argc, char** argv) {
+  Py_Initialize();
+  edmplugin::PluginManager::Config config;
+  edmplugin::PluginManager::configure(edmplugin::standard::config());
+
+  std::vector<edm::ParameterSet> psets;
+  edm::ParameterSet pSet;
+  pSet.addParameter("@service_type", std::string("SiteLocalConfigService"));
+  psets.push_back(pSet);
+  edm::ServiceToken servToken(edm::ServiceRegistry::createSet(psets));
+  edm::ServiceRegistry::Operate operate(servToken);
+
+  std::string connectionString("frontier://FrontierProd/CMS_CONDITIONS");
+
+  std::string tag = "DeDxCalibration_HI_2024_v2";
+  cond::Time_t start = static_cast<unsigned long long>(1);
+  cond::Time_t end = static_cast<unsigned long long>(1);
+
+  edm::LogPrint("test_DeDxCalibration_PayloadInspector") << "## test Inspector" << std::endl;
+
+  DeDxCalibrationInspector histo1;
+  histo1.process(connectionString, PI::mk_input(tag, start, start));
+  edm::LogPrint("test_DeDxCalibration_PayloadInspector") << histo1.data() << std::endl;
+
+  edm::LogPrint("test_DeDxCalibration_PayloadInspector") << "## test Plot" << std::endl;
+
+  DeDxCalibrationPlot histo2;
+  histo2.process(connectionString, PI::mk_input(tag, start, end));
+  edm::LogPrint("test_DeDxCalibration_PayloadInspector") << histo2.data() << std::endl;
+
+  Py_Finalize();
+}

CondCore/PhysicsToolsPlugins/test/test_DeDxCalibration_PayloadInspector.sh

@@ -0,0 +1,41 @@
+#!/bin/bash
+# Save current working dir so img can be outputted there later
+W_DIR=$(pwd);
+# Set SCRAM architecture var
+SCRAM_ARCH=slc7_amd64_gcc900;
+export SCRAM_ARCH;
+source /afs/cern.ch/cms/cmsset_default.sh;
+eval `scram run -sh`;
+
+mkdir -p $W_DIR/plots
+
+getPayloadData.py \
+    --plugin pluginDeDxCalibration_PayloadInspector \
+    --plot plot_DeDxCalibrationTest \
+    --tag DeDxCalibration_HI_2024_v1 \
+    --time_type Run \
+    --iovs '{"start_iov": "1", "end_iov": "1"}' \
+    --db Prod \
+    --test
+
+getPayloadData.py \
+    --plugin pluginDeDxCalibration_PayloadInspector \
+    --plot plot_DeDxCalibrationInspector \
+    --tag DeDxCalibration_HI_2024_v2 \
+    --time_type Run \
+    --iovs '{"start_iov": "1", "end_iov": "1"}' \
+    --db Prod \
+    --test
+
+mv *.png $W_DIR/plots/Inspector.png
+
+getPayloadData.py \
+    --plugin pluginDeDxCalibration_PayloadInspector \
+    --plot plot_DeDxCalibrationPlot \
+    --tag DeDxCalibration_HI_2024_v2 \
+    --time_type Run \
+    --iovs '{"start_iov": "1", "end_iov": "1"}' \
+    --db Prod \
+    --test
+
+mv *.png $W_DIR/plots/Plot.png