PIDDetectorWeights.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 <set>
 
#include <TObject.h>
#include <ROOT/RDataFrame.hxx>
#include <ROOT/RCsvDS.hxx>
 
 
namespace Belle2 {

  class PIDDetectorWeights : public TObject {
 
  public:

    class WeightsTable {
    public:
 
      WeightsTable() :
        m_isEmpty(false),
        m_nPBins(0),
        m_nThetaBins(0)
      {}; 
 
      bool m_isEmpty; 
 
      std::set<double> m_pBinEdges; 
      std::set<double> m_thetaBinEdges; 
 
      int m_nPBins; 
      int m_nThetaBins; 

      std::unordered_map<double, unsigned int> m_linBinIdxsToRowIdxs;
      std::map<std::string, std::vector<double>> m_weightsPerDet;
 
    private:
 
      ClassDef(WeightsTable, 1); 
    };

    PIDDetectorWeights() {};

    ~PIDDetectorWeights() {};

    PIDDetectorWeights(const std::string& weightsCSVFileName)
    {
      m_weightsRDataFrame = ROOT::RDF::FromCSV(weightsCSVFileName);
      fillWeightsTablePerHypoFromRDF();
    };

    PIDDetectorWeights(const std::string& treeName, const std::string& weightsROOTFileName) :
      m_weightsRDataFrame(treeName, weightsROOTFileName)
    {
      fillWeightsTablePerHypoFromRDF();
    };

    ROOT::RDataFrame getWeightsRDF() const
    {
      return m_weightsRDataFrame;
    };

    double getWeight(Const::ChargedStable hypo, Const::EDetector det, double p, double theta) const;
 
  private:

    std::map<int, WeightsTable> m_weightsTablePerHypo;

    ROOT::RDataFrame m_weightsRDataFrame = ROOT::RDataFrame(1); 

    std::map<std::string, std::string> m_weightNames = {
      {Const::parseDetectors(Const::CDC), "ablat_s_CDC"},
      {Const::parseDetectors(Const::TOP), "ablat_s_TOP"},
      {Const::parseDetectors(Const::ARICH), "ablat_s_ARICH"},
      {Const::parseDetectors(Const::ECL), "ablat_s_ECL"},
      {Const::parseDetectors(Const::KLM), "ablat_s_KLM"}
    };

    void fillWeightsTablePerHypoFromRDF();
 
    ClassDef(PIDDetectorWeights, 4); 
 
  };

} // Belle2 namespace