PIDPriors.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Umberto Tamponi (tamponi@to.infn.it)                     *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#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, std::vector<float> xAxisEdges, std::vector<float> yAxisEdges,
                   std::vector<float> priorsTable, 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, std::vector<float> xAxisEdges, 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, std::vector<float> priorsTable)
    {
      auto index = particle.getIndex();
      m_priors[index].setPriorsTable(priorsTable);
      return;
    }
 
 
    void setErrorsTable(const Const::ChargedStable& particle, std::vector<float> errorsTable)
    {
      auto index = particle.getIndex();
      m_priors[index].setErrorsTable(errorsTable);
      return;
    }
 
 
    void setAxisLabels(const Const::ChargedStable& particle, std::string xAxisLabel, 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