[PWGEM] Apply calibration to cell energy (#8660)

Co-authored-by: peressounko <Dmitri.Peresunko@cern.ch>

Author: peressounko

PWGEM/Tasks/phosCellQA.cxx

@@ -14,9 +14,12 @@
 #include <map>
 #include <memory>
 #include <vector>
+#include <string>
 
+#include "CCDB/BasicCCDBManager.h"
 #include "Common/DataModel/EventSelection.h"
 #include "DataFormatsPHOS/Cell.h"
+#include "DataFormatsPHOS/CalibParams.h"
 #include "Framework/ConfigParamSpec.h"
 #include "Framework/runDataProcessing.h"
 #include "Framework/AnalysisTask.h"
@@ -43,13 +46,17 @@ using namespace o2::framework;
 using namespace o2::framework::expressions;
 
 struct phosCellQA {
-  ConfigurableAxis amplitudeAxisLarge{"amplitude", {1000, 0., 100.}, "Amplutude (GeV)"};
+
+  Service<o2::ccdb::BasicCCDBManager> ccdb;
+
+  ConfigurableAxis amplitudeAxisLarge{"amplitude", {1000, 0., 10.}, "Amplutude (GeV)"};
   ConfigurableAxis timeAxisLarge{"celltime", {1000, -1500.e-9, 3500.e-9}, "cell time (ns)"};
   Configurable<int> mEvSelTrig{"mEvSelTrig", kTVXinPHOS, "Select events with this trigger"};
-  Configurable<double> mMinCellAmplitude{"minCellAmplitude", 0., "Minimum cell amplitude for histograms."};
+  Configurable<double> mMinCellAmplitude{"minCellAmplitude", 0.5, "Minimum cell energy for histograms (GeV)"};
   Configurable<double> mMinCellTimeMain{"minCellTimeMain", -50, "Min. cell time of main bunch selection"};
   Configurable<double> mMaxCellTimeMain{"maxCellTimeMain", 100, "Max. cell time of main bunch selection"};
   Configurable<int> mVetoBCID{"vetoBCID", -1, "BC ID to be excluded"};
+  Configurable<std::string> mCalibPath{"calibPath", "PHS/Calib/CalibParams", "path to Calibration snapshot"};
 
   o2::framework::HistogramRegistry mHistManager{"phosCallQAHistograms"};
 
@@ -91,6 +98,15 @@ struct phosCellQA {
   void process(o2::aod::Calos const& cells, BCsWithBcSels const& bcs)
   {
     LOG(debug) << "Processing next event";
+
+    int64_t timestamp = 0;
+    if (bcs.begin() != bcs.end()) {
+      timestamp = bcs.begin().timestamp(); // timestamp for CCDB object retrieval
+    } else {
+      return;
+    }
+    const o2::phos::CalibParams* calibParams = ccdb->getForTimeStamp<o2::phos::CalibParams>(mCalibPath, timestamp);
+
     for (const auto& bc : bcs) {
       o2::InteractionRecord eventIR;
       eventIR.setFromLong(bc.globalBC());
@@ -110,45 +126,52 @@ struct phosCellQA {
       if (mVetoBCID >= 0 && cellIR.bc == mVetoBCID)
         continue;
       mHistManager.fill(HIST("cellBCSelected"), cellIR.bc);
-      //       mHistManager.fill(HIST("cellAmplitude"), cell.amplitude(), cell.cellNumber());
 
       if (!cell.bc_as<BCsWithBcSels>().alias_bit(mEvSelTrig))
         continue;
 
-      if (cell.amplitude() < mMinCellAmplitude)
+      bool isHighGain = cell.cellType();
+      double energy = calibParams->getGain(cell.cellNumber()) * cell.amplitude();
+      // if (isHighGain) {
+      //    energy = calibParams->getGain(cell.cellNumber()) * cell.amplitude();
+      // } else {
+      //    energy = calibParams->getGain(cell.cellNumber()) * cell.amplitude() * calibParams->getHGLGRatio(cell.cellNumber());
+      // }
+
+      if (energy < mMinCellAmplitude)
         continue;
       char relid[3];
       o2::phos::Geometry::absToRelNumbering(cell.cellNumber(), relid);
       if (relid[0] == 1) {
         mHistManager.fill(HIST("cellOccM1"), relid[1] - 0.5, relid[2] - 0.5);
-        mHistManager.fill(HIST("cellAmpM1"), relid[1] - 0.5, relid[2] - 0.5, cell.amplitude());
+        mHistManager.fill(HIST("cellAmpM1"), relid[1] - 0.5, relid[2] - 0.5, energy);
         mHistManager.fill(HIST("cellTimeM1"), relid[1] - 0.5, relid[2] - 0.5, cell.time());
         if (cell.time() > mMinCellTimeMain && cell.time() < mMaxCellTimeMain) {
-          mHistManager.fill(HIST("cellAmpTimeM1"), cell.time(), cell.amplitude());
+          mHistManager.fill(HIST("cellAmpTimeM1"), cell.time(), energy);
         }
       }
       if (relid[0] == 2) {
         mHistManager.fill(HIST("cellOccM2"), relid[1] - 0.5, relid[2] - 0.5);
-        mHistManager.fill(HIST("cellAmpM2"), relid[1] - 0.5, relid[2] - 0.5, cell.amplitude());
+        mHistManager.fill(HIST("cellAmpM2"), relid[1] - 0.5, relid[2] - 0.5, energy);
         mHistManager.fill(HIST("cellTimeM2"), relid[1] - 0.5, relid[2] - 0.5, cell.time());
         if (cell.time() > mMinCellTimeMain && cell.time() < mMaxCellTimeMain) {
-          mHistManager.fill(HIST("cellAmpTimeM2"), cell.time(), cell.amplitude());
+          mHistManager.fill(HIST("cellAmpTimeM2"), cell.time(), energy);
         }
       }
       if (relid[0] == 3) {
         mHistManager.fill(HIST("cellOccM3"), relid[1] - 0.5, relid[2] - 0.5);
-        mHistManager.fill(HIST("cellAmpM3"), relid[1] - 0.5, relid[2] - 0.5, cell.amplitude());
+        mHistManager.fill(HIST("cellAmpM3"), relid[1] - 0.5, relid[2] - 0.5, energy);
         mHistManager.fill(HIST("cellTimeM3"), relid[1] - 0.5, relid[2] - 0.5, cell.time());
         if (cell.time() > mMinCellTimeMain && cell.time() < mMaxCellTimeMain) {
-          mHistManager.fill(HIST("cellAmpTimeM3"), cell.time(), cell.amplitude());
+          mHistManager.fill(HIST("cellAmpTimeM3"), cell.time(), energy);
         }
       }
       if (relid[0] == 4) {
         mHistManager.fill(HIST("cellOccM4"), relid[1] - 0.5, relid[2] - 0.5);
-        mHistManager.fill(HIST("cellAmpM4"), relid[1] - 0.5, relid[2] - 0.5, cell.amplitude());
+        mHistManager.fill(HIST("cellAmpM4"), relid[1] - 0.5, relid[2] - 0.5, energy);
         mHistManager.fill(HIST("cellTimeM4"), relid[1] - 0.5, relid[2] - 0.5, cell.time());
         if (cell.time() > mMinCellTimeMain && cell.time() < mMaxCellTimeMain) {
-          mHistManager.fill(HIST("cellAmpTimeM4"), cell.time(), cell.amplitude());
+          mHistManager.fill(HIST("cellAmpTimeM4"), cell.time(), energy);
         }
       }
     }