ECLLeakageCorrections.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2020 - Belle II Collaboration                             *
 *                                                                        *
 * DB object to store leakage corrections, including dependence on nCrys     *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Christopher Hearty (hearty@physics.ubc.ca)               *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
#include <TObject.h>
#include <framework/logging/Logger.h>
#include <TH2F.h>
 
namespace Belle2 {
  class ECLLeakageCorrections: public TObject {
 
 
//---------------------------------------------------
  public:
 
    ECLLeakageCorrections() :
      m_logEnergiesFwd(),
      m_logEnergiesBrl(),
      m_logEnergiesBwd(),
      m_thetaBoundaries(),
      m_thetaCorrections(),
      m_phiCorrections(),
      m_nCrystalCorrections()
    { };
 
    //---------------------------------------------------
    //..Getters for the various calibration components
 
    const std::vector<float>& getlogEnergiesFwd() const {return m_logEnergiesFwd;}
 
    const std::vector<float>& getlogEnergiesBrl() const {return m_logEnergiesBrl;}
 
    const std::vector<float>& getlogEnergiesBwd() const {return m_logEnergiesBwd;}
 
    const std::vector<float>& getThetaBoundaries() const {return m_thetaBoundaries;}
 
    const TH2F getThetaCorrections() const {return m_thetaCorrections;}
 
    const TH2F getPhiCorrections() const {return m_phiCorrections;}
 
    const TH2F getnCrystalCorrections() const {return m_nCrystalCorrections;}
 
 
    //---------------------------------------------------
    //..Setters for the various calibration components
 
    void setlogEnergiesFwd(const std::vector<float>& logEnergiesFwd) {m_logEnergiesFwd = logEnergiesFwd;}
 
    void setlogEnergiesBrl(const std::vector<float>& logEnergiesBrl) {m_logEnergiesBrl = logEnergiesBrl;}
 
    void setlogEnergiesBwd(const std::vector<float>& logEnergiesBwd) {m_logEnergiesBwd = logEnergiesBwd;}
 
    void setThetaBoundaries(const std::vector<float>& thetaBoundaries)
    {
      if (thetaBoundaries.size() != 70) {B2FATAL("ECLLeakageCorrections: wrong size vector uploaded for thetaBoundaries, " << thetaBoundaries.size() << " instead of 70");}        m_thetaBoundaries
        = thetaBoundaries;
    }
 
    void setThetaCorrections(const TH2F& thetaCorrections) {m_thetaCorrections = thetaCorrections;}
 
    void setPhiCorrections(const TH2F& phiCorrections) {m_phiCorrections = phiCorrections;}
 
    void setnCrystalCorrections(const TH2F& nCrystalCorrections) {m_nCrystalCorrections = nCrystalCorrections;}
 
 
//---------------------------------------------------
  private:
    std::vector<float> m_logEnergiesFwd; 
    std::vector<float> m_logEnergiesBrl; 
    std::vector<float> m_logEnergiesBwd; 
    std::vector<float> m_thetaBoundaries; 
    TH2F m_thetaCorrections; 
    TH2F m_phiCorrections; 
    TH2F m_nCrystalCorrections; 
    ClassDef(ECLLeakageCorrections, 1); 
  };
}