FIT calibration: new feature for debugging - extraInfo option, applied only for FT0 time spectra calibration

Author: afurs

Detectors/FIT/FT0/calibration/include/FT0Calibration/FT0TimeOffsetSlotContainer.h

@@ -18,6 +18,8 @@
 #include "CommonDataFormat/FlatHisto2D.h"
 #include "DataFormatsFT0/SpectraInfoObject.h"
 
+#include "TList.h"
+
 #include "Rtypes.h"
 namespace o2::ft0
 {
@@ -34,10 +36,10 @@ class FT0TimeOffsetSlotContainer final
   FT0TimeOffsetSlotContainer& operator=(FT0TimeOffsetSlotContainer&&) = default;
   bool hasEnoughEntries() const;
   void fill(const gsl::span<const float>& data);
-  SpectraInfoObject getSpectraInfoObject(std::size_t channelID) const;
+  SpectraInfoObject getSpectraInfoObject(std::size_t channelID, TList* listHists) const;
   void merge(FT0TimeOffsetSlotContainer* prev);
   void print() const;
-  TimeSpectraInfoObject generateCalibrationObject() const;
+  TimeSpectraInfoObject generateCalibrationObject(long tsStartMS, long tsEndMS, const std::string& pathToHists) const;
   typedef float FlatHistoValue_t;
   typedef o2::dataformats::FlatHisto2D<FlatHistoValue_t> FlatHisto2D_t;
 

Detectors/FIT/FT0/calibration/include/FT0Calibration/GlobalOffsetsContainer.h

@@ -38,7 +38,7 @@ class GlobalOffsetsContainer final
   int getMeanGaussianFitValue() const;
   void merge(GlobalOffsetsContainer* prev);
   void print() const;
-  GlobalOffsetsCalibrationObject generateCalibrationObject() const;
+  GlobalOffsetsCalibrationObject generateCalibrationObject(long, long, const std::string&) const;
   void updateFirstCreation(std::uint64_t creation)
   {
     if (creation < mFirstCreation) {

Detectors/FIT/FT0/calibration/src/FT0TimeOffsetSlotContainer.cxx

@@ -16,6 +16,7 @@
 #include <Framework/Logger.h>
 
 #include "TH1.h"
+#include "TFile.h"
 #include "TFitResult.h"
 
 using namespace o2::ft0;
@@ -107,7 +108,7 @@ void FT0TimeOffsetSlotContainer::merge(FT0TimeOffsetSlotContainer* prev)
   mCurrentSlot++;
 }
 
-SpectraInfoObject FT0TimeOffsetSlotContainer::getSpectraInfoObject(std::size_t channelID) const
+SpectraInfoObject FT0TimeOffsetSlotContainer::getSpectraInfoObject(std::size_t channelID, TList* listHists) const
 {
   uint32_t statusBits{};
   double minFitRange{0};
@@ -145,19 +146,40 @@ SpectraInfoObject FT0TimeOffsetSlotContainer::getSpectraInfoObject(std::size_t c
       LOG(debug) << "Bad gaus fit: meanGaus " << meanGaus << " sigmaGaus " << sigmaGaus << " meanHist " << meanHist << " rmsHist " << rmsHist << "resultFit " << ((int)resultFit);
     }
   }
+  if (listHists != nullptr) {
+    auto histPtr = hist.release();
+    const std::string histName = "histCh" + std::to_string(channelID);
+    histPtr->SetName(histName.c_str());
+    listHists->Add(histPtr);
+  }
   return SpectraInfoObject{meanGaus, sigmaGaus, constantGaus, fitChi2, meanHist, rmsHist, stat, statusBits};
 }
 
-TimeSpectraInfoObject FT0TimeOffsetSlotContainer::generateCalibrationObject() const
+TimeSpectraInfoObject FT0TimeOffsetSlotContainer::generateCalibrationObject(long tsStartMS, long tsEndMS, const std::string& extraInfo) const
 {
+  TList* listHists = nullptr;
+  bool storeHists{false};
+  if (extraInfo.size() > 0) {
+    storeHists = true;
+    listHists = new TList();
+    listHists->SetOwner(true);
+    listHists->SetName("output");
+  }
   TimeSpectraInfoObject calibrationObject;
   for (unsigned int iCh = 0; iCh < sNCHANNELS; ++iCh) {
-    calibrationObject.mTime[iCh] = getSpectraInfoObject(iCh);
+    calibrationObject.mTime[iCh] = getSpectraInfoObject(iCh, listHists);
+  }
+  calibrationObject.mTimeA = getSpectraInfoObject(sNCHANNELS, listHists);
+  calibrationObject.mTimeC = getSpectraInfoObject(sNCHANNELS + 1, listHists);
+  calibrationObject.mSumTimeAC = getSpectraInfoObject(sNCHANNELS + 2, listHists);
+  calibrationObject.mDiffTimeCA = getSpectraInfoObject(sNCHANNELS + 3, listHists);
+  if (storeHists) {
+    const std::string filename = extraInfo + "/histsTimeSpectra" + std::to_string(tsStartMS) + "_" + std::to_string(tsEndMS) + ".root";
+    TFile fileHists(filename.c_str(), "RECREATE");
+    fileHists.WriteObject(listHists, listHists->GetName(), "SingleKey");
+    fileHists.Close();
+    delete listHists;
   }
-  calibrationObject.mTimeA = getSpectraInfoObject(sNCHANNELS);
-  calibrationObject.mTimeC = getSpectraInfoObject(sNCHANNELS + 1);
-  calibrationObject.mSumTimeAC = getSpectraInfoObject(sNCHANNELS + 2);
-  calibrationObject.mDiffTimeCA = getSpectraInfoObject(sNCHANNELS + 3);
   return calibrationObject;
 }
 

Detectors/FIT/FT0/calibration/src/GlobalOffsetsContainer.cxx

@@ -75,7 +75,7 @@ int GlobalOffsetsContainer::getMeanGaussianFitValue() const
     return 0;
   }
 }
-GlobalOffsetsCalibrationObject GlobalOffsetsContainer::generateCalibrationObject() const
+GlobalOffsetsCalibrationObject GlobalOffsetsContainer::generateCalibrationObject(long, long, const std::string&) const
 {
   GlobalOffsetsCalibrationObject calibrationObject;
   calibrationObject.mCollisionTimeOffsets = getMeanGaussianFitValue();

Detectors/FIT/FT0/calibration/testWorkflow/FT0TimeOffsetCalibration-Workflow.cxx

@@ -56,7 +56,8 @@ WorkflowSpec defineDataProcessing(ConfigContext const& config)
     o2::framework::AlgorithmSpec{o2::framework::adaptFromTask<CalibrationDeviceType>(ccdbRequest, outputDataDescriptor)},
     o2::framework::Options{
       {"tf-per-slot", o2::framework::VariantType::UInt32, 56000u, {""}},
-      {"max-delay", o2::framework::VariantType::UInt32, 3u, {""}}}};
+      {"max-delay", o2::framework::VariantType::UInt32, 3u, {""}},
+      {"extra-info-per-slot", o2::framework::VariantType::String, "", {"Extra info for time slot(usually for debugging)"}}}};
 
   WorkflowSpec workflow;
   workflow.emplace_back(dataProcessorSpec);

Detectors/FIT/FV0/calibration/include/FV0Calibration/FV0ChannelTimeOffsetSlotContainer.h

@@ -56,7 +56,7 @@ class FV0ChannelTimeOffsetSlotContainer final
   {
     return mFirstCreation;
   }
-  FV0ChannelTimeCalibrationObject generateCalibrationObject() const;
+  FV0ChannelTimeCalibrationObject generateCalibrationObject(long, long, const std::string&) const;
 
  private:
   std::size_t mMinEntries;

Detectors/FIT/FV0/calibration/src/FV0ChannelTimeOffsetSlotContainer.cxx

@@ -103,7 +103,7 @@ int16_t FV0ChannelTimeOffsetSlotContainer::getMeanGaussianFitValue(std::size_t c
 
   return static_cast<int16_t>(std::round(outputGaussianFitValues[MEAN_VALUE_INDEX_IN_OUTPUT_VECTOR]));
 }
-FV0ChannelTimeCalibrationObject FV0ChannelTimeOffsetSlotContainer::generateCalibrationObject() const
+FV0ChannelTimeCalibrationObject FV0ChannelTimeOffsetSlotContainer::generateCalibrationObject(long, long, const std::string&) const
 {
   FV0ChannelTimeCalibrationObject calibrationObject;
 

Detectors/FIT/common/calibration/include/FITCalibration/FITCalibrationDevice.h

@@ -46,12 +46,12 @@ class FITCalibrationDevice : public o2::framework::Task
     o2::base::GRPGeomHelper::instance().setRequest(mCCDBRequest);
     auto slotL = context.options().get<uint32_t>("tf-per-slot");
     auto delay = context.options().get<uint32_t>("max-delay");
-
+    const std::string extraInfo = context.options().get<std::string>("extra-info-per-slot");
     mCalibrator = std::make_unique<CalibratorType>();
 
     mCalibrator->setSlotLength(slotL);
     mCalibrator->setMaxSlotsDelay(delay);
-
+    mCalibrator->setExtraInfo(extraInfo);
     //    o2::ccdb::BasicCCDBManager::instance().setURL(sDEFAULT_CCDB_URL);
   }
 

Detectors/FIT/common/calibration/include/FITCalibration/FITCalibrator.h

@@ -67,7 +67,7 @@ class FITCalibrator final : public o2::calibration::TimeSlotCalibration<TimeSlot
     auto starting = slot.getStartTimeMS();
     auto stopping = slot.getEndTimeMS();
     LOGP(info, "!!!! {}({})<=TF<={}({}), starting: {} stopping {}", slot.getTFStart(), slot.getStartTimeMS(), slot.getTFEnd(), slot.getEndTimeMS(), starting, stopping);
-    auto calibrationObject = container->generateCalibrationObject();
+    auto calibrationObject = container->generateCalibrationObject(slot.getStartTimeMS(), slot.getEndTimeMS(), mExtraInfo);
     std::vector<CalibObjWithInfoType> preparedCalibObjects;
     preparedCalibObjects.emplace_back(doSerializationAndPrepareObjectInfo(calibrationObject, starting, stopping));
     mStoredCalibrationObjects.insert(mStoredCalibrationObjects.end(),
@@ -99,10 +99,15 @@ class FITCalibrator final : public o2::calibration::TimeSlotCalibration<TimeSlot
               << " start " << starting << " end " << stopping;
     return result;
   }
+  void setExtraInfo(const std::string& extraInfo)
+  {
+    mExtraInfo = extraInfo;
+  }
 
  private:
   std::vector<CalibObjWithInfoType> mStoredCalibrationObjects{};
   const unsigned int mMinEntries;
+  std::string mExtraInfo;
 };
 
 } // namespace o2::fit