EMCALTimeCalibData.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#ifndef EMCAL_TIME_DATA_H_
#define EMCAL_TIME_DATA_H_
 
#include "DetectorsCalibration/TimeSlotCalibration.h"
#include "DetectorsCalibration/TimeSlot.h"
#include "DataFormatsEMCAL/Cell.h"
#include "EMCALBase/Geometry.h"
#include "CCDB/CcdbObjectInfo.h"
#include "EMCALCalib/TimeCalibrationParams.h"
#include "EMCALCalib/GainCalibrationFactors.h"
#include "EMCALCalibration/EMCALCalibParams.h"
 
#include "Framework/Logger.h"
#include "CommonUtils/MemFileHelper.h"
#include "CCDB/CcdbApi.h"
#include "DetectorsCalibration/Utils.h"
#include <boost/histogram.hpp>
#include <boost/histogram/ostream.hpp>
#include <boost/format.hpp>
 
#include <array>
#include <boost/histogram.hpp>
namespace o2
{
namespace emcal
{
 
class EMCALTimeCalibData
{
 public:
  using Cells = o2::emcal::Cell;
  using boostHisto = boost::histogram::histogram<std::tuple<boost::histogram::axis::regular<double, boost::use_default, boost::use_default, boost::use_default>, boost::histogram::axis::regular<double, boost::use_default, boost::use_default, boost::use_default>>>;
 
  o2::emcal::Geometry* mGeometry = o2::emcal::Geometry::GetInstanceFromRunNumber(300000);
  int NCELLS = mGeometry->GetNCells();
 
  EMCALTimeCalibData()
  {
 
    mNThreads = EMCALCalibParams::Instance().nThreads_tc;
    mTimeHisto.resize(mNThreads);
    mVecNEntriesInHisto.resize(mNThreads);
    for (size_t i = 0; i < mNThreads; ++i) {
      mTimeHisto[i] = boost::histogram::make_histogram(boost::histogram::axis::regular<>(EMCALCalibParams::Instance().nBinsTimeAxis_tc, EMCALCalibParams::Instance().minValueTimeAxis_tc, EMCALCalibParams::Instance().maxValueTimeAxis_tc), boost::histogram::axis::regular<>(NCELLS, -0.5, NCELLS - 0.5));
      mVecNEntriesInHisto[i] = 0;
    }
    LOG(debug) << "initialize time histogram with " << NCELLS << " cells";
  }
 
  ~EMCALTimeCalibData() = default;
 
  void fill(const gsl::span<const o2::emcal::Cell> data);
 
  void merge(EMCALTimeCalibData* prev);
 
  bool hasEnoughData() const;
 
  void print();
 
  void setNEvents(int nevt) { mEvents = nevt; }
  void AddEvents(int nevt) { mEvents += nevt; }
  int getNEvents() const { return mEvents; }
 
  long unsigned int getNEntriesInHisto() const { return mNEntriesInHisto; }
  void setNEntriesInHisto(long unsigned int n) { mNEntriesInHisto = n; }
  void addNEntriesInHisto(long unsigned int n) { mNEntriesInHisto += n; }
 
  const boostHisto& getHisto()
  {
    // set the summed histogram to one of the existing histograms
    mHistoSummed = mTimeHisto[0];
    // reset the histogram
    mHistoSummed.reset();
    // Sum up all entries
    for (const auto& h : mTimeHisto) {
      mHistoSummed += h;
    }
    return mHistoSummed;
  }
 
  void setGainCalibFactors(o2::emcal::GainCalibrationFactors* calibFactors)
  {
    mGainCalibFactors = calibFactors;
    mApplyGainCalib = true;
  }
 
  void setHisto(boostHisto hist) { mTimeHisto[0] = hist; }
 
  void PrintStream(std::ostream& stream) const;
 
  o2::emcal::TimeCalibrationParams process();
 
 private:
  unsigned int mNThreads = 1;
  std::vector<boostHisto> mTimeHisto;                   
  boostHisto mHistoSummed;                              
  int mEvents = 0;                                      
  long unsigned int mNEntriesInHisto = 0;               
  std::vector<long unsigned int> mVecNEntriesInHisto;   
  bool mApplyGainCalib = false;                         
  o2::emcal::GainCalibrationFactors* mGainCalibFactors; 
 
  ClassDefNV(EMCALTimeCalibData, 1);
};
 
} // end namespace emcal
} // end namespace o2
 
#endif /*EMCAL_TIME_DATA_H_ */