PIDCalibrationWeightUtil.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <analysis/dbobjects/PIDCalibrationWeight.h>
#include <analysis/dbobjects/PIDDetectorWeights.h>
#include <framework/database/DBObjPtr.h>
#include <framework/logging/Logger.h>
 
#include <framework/gearbox/Const.h>
 
namespace Belle2 {
  class PIDCalibrationWeightUtil {
 
  public:
 
    PIDCalibrationWeightUtil()
    {
      m_pidCalibWeightDB = std::make_unique<DBObjPtr<PIDCalibrationWeight>>();
 
      if (!(*m_pidCalibWeightDB))
        B2FATAL("The dbobject PIDCalibrationWeight could not be found! It is necessary for the weighted PID variables.");
    };
 
    PIDCalibrationWeightUtil(std::string matrixName)
    {
      m_matrixName = matrixName;
 
      if (m_matrixName.find("PIDDetectorWeights") != std::string::npos) {
        m_pidDetWeightDB = std::make_unique<DBObjPtr<PIDDetectorWeights>>(m_matrixName);
        if (!(*m_pidDetWeightDB))
          B2FATAL("The dbobject PIDDetectorWeights, " << m_matrixName <<
                  " could not be found! It is necessary for the weighted PID variables.");
      } else {
        m_pidCalibWeightDB = std::make_unique<DBObjPtr<PIDCalibrationWeight>>(m_matrixName);
        if (!(*m_pidCalibWeightDB))
          B2FATAL("The dbobject PIDCalibrationWeight, " << m_matrixName <<
                  " could not be found! It is necessary for the weighted PID variables.");
      }
    };
 
    double getWeight(int pdg, Const::EDetector det, double p, double theta) const
    {
      if (m_matrixName.find("PIDDetectorWeights") != std::string::npos) {
        return (*m_pidDetWeightDB)->getWeight(Const::ParticleType(pdg), det, p, theta);
      } else {
        return (*m_pidCalibWeightDB)->getWeight(pdg, det);
      }
    }
 
  private:
    std::string m_matrixName = "PIDCalibrationWeight"; 
    std::unique_ptr<DBObjPtr<PIDCalibrationWeight>> m_pidCalibWeightDB; 
    std::unique_ptr<DBObjPtr<PIDDetectorWeights>>
                                               m_pidDetWeightDB; 
  };
 
} // Belle2 namespace