PIDPriors.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 <TObject.h>
 
#include <analysis/dbobjects/PIDPriorsTable.h>
#include <framework/gearbox/Const.h>
 
class TH2F;
 
namespace Belle2 {

  class PIDPriors : public TObject {

    ClassDef(PIDPriors, 1);
 
 
  public:

    PIDPriors() {};
 

    void setPriors(const Const::ChargedStable& particle, const PIDPriorsTable& table)
    {
      auto index = particle.getIndex();
      m_priors[index] = table;
      return;
    }
 

    void setPriors(const Const::ChargedStable& particle, const std::vector<float>& xAxisEdges, const std::vector<float>& yAxisEdges,
                   const std::vector<float>& priorsTable, const std::vector<float>& errorsTable)
    {
      auto index = particle.getIndex();
      m_priors[index].setBinEdges(xAxisEdges, yAxisEdges);
      m_priors[index].setBinEdges(xAxisEdges, yAxisEdges);
      m_priors[index].setPriorsTable(priorsTable);
      m_priors[index].setErrorsTable(errorsTable);
      return;
    }
 

    void setPriors(const Const::ChargedStable& particle, TH2F* priorHistogram);
 

    void setPriors(const Const::ChargedStable& particle, TH2F* counts, TH2F* normalization);
 

    void setPriorsAxes(const Const::ChargedStable& particle, const std::vector<float>& xAxisEdges, const std::vector<float>& yAxisEdges)
    {
      auto index = particle.getIndex();
      m_priors[index].setBinEdges(xAxisEdges, yAxisEdges);
      m_priors[index].setBinEdges(xAxisEdges, yAxisEdges);
      return;
    }
 

    void setPriorsTable(const Const::ChargedStable& particle, const std::vector<float>& priorsTable)
    {
      auto index = particle.getIndex();
      m_priors[index].setPriorsTable(priorsTable);
      return;
    }
 

    void setErrorsTable(const Const::ChargedStable& particle, const std::vector<float>& errorsTable)
    {
      auto index = particle.getIndex();
      m_priors[index].setErrorsTable(errorsTable);
      return;
    }
 

    void setAxisLabels(const Const::ChargedStable& particle, const std::string& xAxisLabel, const std::string& yAxisLabel)
    {
      auto index = particle.getIndex();
      m_priors[index].setAxisLabels(xAxisLabel, yAxisLabel);
      return;
    }
 

    PIDPriorsTable getPriorsTable(const Const::ChargedStable& particle) const
    {
      auto index = particle.getIndex();
      return m_priors[index];
    }
 

    float getPriorValue(const Const::ChargedStable& particle, float x, float y) const
    {
      auto index = particle.getIndex();
      return m_priors[index].getPriorValue(x, y);
    }
 

    float getPriorError(const Const::ChargedStable& particle, float x, float y) const
    {
      auto index = particle.getIndex();
      return m_priors[index].getErrorValue(x, y);
    }
 

    std::string getXAxisLabel(const Const::ChargedStable& particle) const
    {
      auto index = particle.getIndex();
      return m_priors[index].getXAxisLabel();
    }
 

    std::string getYAxisLabel(const Const::ChargedStable& particle) const
    {
      auto index = particle.getIndex();
      return m_priors[index].getYAxisLabel();
    }
 
  private:
    PIDPriorsTable m_priors[Const::ChargedStable::c_SetSize]; 
 
 
  };
 

} // Belle2 namespace