PIDCalibrationWeight.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 <framework/gearbox/Const.h>
#include <framework/logging/Logger.h>
 
#include <boost/algorithm/string.hpp>
 
#include <TObject.h>
 
namespace Belle2 {

  typedef std::vector<std::vector<double>> WeightMatrix;

  class PIDCalibrationWeight : public TObject {
 
  public:
    PIDCalibrationWeight() {};

    explicit PIDCalibrationWeight(const WeightMatrix& weightMatrix)
      : m_weightMatrix(weightMatrix)
    {};

    void setWeightMatrix(const WeightMatrix& weightMatrix)
    {
      m_weightMatrix = weightMatrix;
    };

    const WeightMatrix& getWeightMatrix() const
    {
      return m_weightMatrix;
    };

    double getWeight(int pdg, std::string detector) const
    {
      std::vector<double> weightVector = getWeights(pdg);
 
      Const::EDetector det = Const::invalidDetector; // default value, will be overwritten
      boost::to_lower(detector);
      if (detector == "svd") det = Const::SVD;
      else if (detector == "cdc") det = Const::CDC;
      else if (detector == "top") det = Const::TOP;
      else if (detector == "arich") det = Const::ARICH;
      else if (detector == "ecl") det = Const::ECL;
      else if (detector == "klm") det = Const::KLM;
      else B2FATAL("Unknown detector component: " << detector);
      int det_index = Const::PIDDetectors::c_set.getIndex(det);
 
      return weightVector[det_index];
    };

    double getWeight(int pdg, Const::EDetector det) const
    {
      std::vector<double> weightVector = getWeights(pdg);
 
      int det_index = Const::PIDDetectors::c_set.getIndex(det);
      return weightVector[det_index];
    };

    std::vector<double> getWeights(int pdg) const
    {
      int p_index = -1;
      for (const auto& pdgIter : Const::chargedStableSet) {
        if (pdgIter.getPDGCode() == pdg) {
          p_index = pdgIter.getIndex();
          break;
        }
      }
      if (p_index == -1)
        B2FATAL("Invalid particle: " << pdg);
 
      return m_weightMatrix[p_index];
    };
 
  private:

    WeightMatrix m_weightMatrix;
 
    ClassDef(PIDCalibrationWeight, 1); 
 
  };

} // Belle2 namespace