InterCalib.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.
 
#include <mutex>
#include <memory>
#include <TH1.h>
#include <TH2.h>
#include <THnBase.h>
#include <THnSparse.h>
#include <TMinuit.h>
#include "ZDCBase/Constants.h"
#include "CommonDataFormat/FlatHisto1D.h"
#include "CommonDataFormat/FlatHisto2D.h"
#include "DataFormatsZDC/RecEvent.h"
#include "ZDCReconstruction/ZDCEnergyParam.h"
#include "ZDCReconstruction/ZDCTowerParam.h"
#include "ZDCCalib/InterCalibData.h"
#include "ZDCCalib/InterCalibConfig.h"
#include "CCDB/CcdbObjectInfo.h"
#ifndef ALICEO2_ZDC_INTERCALIB_H
#define ALICEO2_ZDC_INTERCALIB_H
namespace o2
{
namespace zdc
{
class InterCalib
{
  using CcdbObjectInfo = o2::ccdb::CcdbObjectInfo;
 
 public:
  InterCalib() = default;
  int init();
  static constexpr int HidZNA = 0;
  static constexpr int HidZPA = 1;
  static constexpr int HidZNC = 2;
  static constexpr int HidZPC = 3;
  static constexpr int HidZEM = 4;
  static constexpr int HidZNI = 5;
  static constexpr int HidZPI = 6;
  static constexpr int HidZPAX = 7;
  static constexpr int HidZPCX = 8;
  static constexpr int NH = InterCalibData::NH;
  static constexpr int NPAR = InterCalibData::NPAR;
  void clear(int ih = -1);
  int process(const gsl::span<const o2::zdc::BCRecData>& bcrec,
              const gsl::span<const o2::zdc::ZDCEnergy>& energy,
              const gsl::span<const o2::zdc::ZDCTDCData>& tdc,
              const gsl::span<const uint16_t>& info); // Calibration of RUN3 data - direct
  int process(const InterCalibData& data);            // Calibration of RUN3 data - aggregator node
  int endOfRun();                                     // Perform minimization
  int process(const char* hname, int ic);             // Calibration of RUN2 data
  void replay(int ih, THnSparse* hs, int ic);         // Test of calibration using RUN2 data
  int mini(int ih);
  void add(int ih, o2::dataformats::FlatHisto1D<float>& h1);
  void add(int ih, o2::dataformats::FlatHisto2D<float>& h2);
  int saveDebugHistos(const std::string fn = "ZDCInterCalib.root");
  void setSaveDebugHistos() { mSaveDebugHistos = true; }
  void setDontSaveDebugHistos() { mSaveDebugHistos = false; }
 
  static double mAdd[NPAR][NPAR]; 
  static void fcn(int& npar, double* gin, double& chi, double* par, int iflag);
  void cumulate(int ih, double tc, double t1, double t2, double t3, double t4, double w);
 
  const ZDCTowerParam& getTowerParamUpd() const { return mTowerParamUpd; };
 
  CcdbObjectInfo& getCcdbObjectInfo() { return mInfo; }
 
  void setEnergyParam(const ZDCEnergyParam* param) { mEnergyParam = param; };
  const ZDCEnergyParam* getEnergyParam() const { return mEnergyParam; };
  void setTowerParam(const ZDCTowerParam* param) { mTowerParam = param; };
  const ZDCTowerParam* getTowerParam() const { return mTowerParam; };
  void setInterCalibConfig(const InterCalibConfig* param) { mInterCalibConfig = param; };
  const InterCalibConfig* getInterCalibConfig() const { return mInterCalibConfig; };
 
  InterCalibData& getData() { return mData; };
 
  void setVerbosity(int v) { mVerbosity = v; }
  int getVerbosity() const { return mVerbosity; }
 
  // Histograms for intercalibration of side C vs A are duplicate -> No title
  static constexpr const char* mHUncN[2 * NH] = {"hZNAS", "hZPAS", "hZNCS", "hZPCS", "hZEM2", "", "", "hZPASX", "hZPCSX",
                                                 "hZNAC", "hZPAC", "hZNCC", "hZPCC", "hZEM1", "", "", "hZPACX", "hZPCCX"};
  static constexpr const char* mHUncT[2 * NH] = {"ZNA sum", "ZPA sum", "ZNC sum", "ZPC sum", "ZEM2", "", "", "ZPA sum", "ZPC sum",
                                                 "ZNA TC", "ZPA TC", "ZNC TC", "ZPC TC", "ZEM1", "", "", "ZPA TC", "ZPC TC"};
  static constexpr const char* mCUncN[NH] = {"cZNA", "cZPA", "cZNC", "cZPC", "cZEM", "cZNI", "cZPI", "cZPAX", "cZPCX"};
  static constexpr const char* mCUncT[NH] = {"ZNA;TC;SUM", "ZPA;TC;SUM", "ZNC;TC;SUM", "ZPC;TC;SUM", "ZEM;ZEM1;ZEM2", "ZN;ZNAC;ZNCC", "ZP;ZPAC;ZPCC", "ZPA (x cut);TC;SUM", "ZPC (x cut);TC;SUM"};
 
 private:
  std::array<o2::dataformats::FlatHisto1D<float>*, 2 * NH> mHUnc{};
  std::array<o2::dataformats::FlatHisto2D<float>*, NH> mCUnc{};
  std::array<std::unique_ptr<TH1>, NH> mHCorr{};
  std::array<std::unique_ptr<TH2>, NH> mCCorr{};
  std::array<std::unique_ptr<TMinuit>, NH> mMn{};
  InterCalibData mData;
  bool mInitDone = false;
  bool mSaveDebugHistos = false;
  int32_t mVerbosity = DbgMinimal;
  static std::mutex mMtx; 
  double mPar[NH][NPAR] = {0};
  double mErr[NH][NPAR] = {0};
  const InterCalibConfig* mInterCalibConfig = nullptr; 
  const ZDCEnergyParam* mEnergyParam = nullptr;        
  const ZDCTowerParam* mTowerParam = nullptr;          
  ZDCTowerParam mTowerParamUpd;                        
  CcdbObjectInfo mInfo;                                
  void assign(int ih, bool ismod);                     
};
} // namespace zdc
} // namespace o2
 
#endif