PIDPriorsTable.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 <vector>
#include <string>
 
namespace Belle2 {
  class PIDPriorsTable {
 
  public:
 
    void setBinEdges(const std::vector<float>& binEdgesX, const std::vector<float>& binEdgesY);
 
 
    void setPriorsTable(const std::vector<float>& priors)
    {
      m_priors = priors;
      return;
    };
 
 
    void setErrorsTable(const std::vector<float>& errors)
    {
      m_errors = errors;
      return;
    };
 
 
    void setPriorValue(float x, float y, float value);
 
 
    void setErrorValue(float x, float y, float value);
 
 
    void setAxisLabels(const std::string& labelX, const std::string& labelY)
    {
      m_xAxisLabel = labelX;
      m_yAxisLabel = labelY;
      return ;
    }
 
 
    float getPriorInBin(int iX, int iY) const
    {
      return m_priors[iX + (m_binEdgesX.size() - 1) * iY];
    };
 
 
    float getErrorInBin(int iX, int iY) const
    {
      return m_errors[iX + (m_binEdgesX.size() - 1) * iY];
    };
 
 
    float getPriorValue(float x, float y) const;
 
 
    float getErrorValue(float x, float y) const;
 
 
    std::string getXAxisLabel() const
    {
      return m_xAxisLabel;
    };
 
    std::string getYAxisLabel() const
    {
      return m_yAxisLabel;
    };
 
 
    void printPrior() const;
 
    void printError() const;
 
 
  private:
 
    std::vector<float> m_binEdgesX = {}; 
    std::vector<float> m_binEdgesY = {}; 
    std::vector<float> m_priors; 
    std::vector<float> m_errors; 
    std::string m_xAxisLabel = "" ; 
    std::string m_yAxisLabel = "" ; 
    bool checkRange(const std::string& text, float val, const std::vector<float>& edges) const;
 
    short findBin(float val, std::vector<float> array) const;
 
    // cppcheck-suppress unusedPrivateFunction
    short findBinFast(float val, std::vector<float> array) const;
 
    // cppcheck-suppress unusedPrivateFunction
    short findBinWithFixedWidth(float val, std::vector<float> array) const;
 
 
  };
} // Belle2 namespace