dEdxPDFs.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
 
/* basf2 headers. */
#include <framework/gearbox/Const.h>
 
/* C++ headers. */
#include <array>
#include <vector>
 
/* ROOT headers. */
#include <TH2F.h>
#include <TObject.h>
 
namespace Belle2 {
  class dEdxPDFs: public TObject {
 
  public:
 
    dEdxPDFs() = default;
 
    const TH2F* getPDF(const Const::ChargedStable& chargedStable, bool truncated) const
    {
      return getPDF(chargedStable.getIndex(), truncated);
    }
 
    const TH2F* getPDF(const unsigned int hypothesis, const bool truncated) const
    {
      return truncated ? &(m_dEdxPDFsTruncated.at(hypothesis)) : &(m_dEdxPDFs.at(hypothesis));
    }
 
    const std::vector<const TH2F*>& getPDFs(bool truncated) const;
 
    void setPDF(const TH2F& pdf, const Const::ChargedStable& chargedStable, bool truncated)
    {
      setPDF(pdf, chargedStable.getIndex(), truncated);
    }
 
    void setPDF(const TH2F& pdf, const unsigned int hypothesis, const bool truncated)
    {
      if (truncated)
        m_dEdxPDFsTruncated.at(hypothesis) = pdf;
      else
        m_dEdxPDFs.at(hypothesis) = pdf;
    }
 
    bool checkPDFs(bool truncated) const;
 
  private:
 
    std::array<TH2F, Const::ChargedStable::c_SetSize> m_dEdxPDFs;
 
    std::array<TH2F, Const::ChargedStable::c_SetSize> m_dEdxPDFsTruncated;
 
    mutable std::vector<const TH2F*> m_cachePDFs;  
 
    mutable std::vector<const TH2F*> m_cachePDFsTruncated;  
 
    ClassDef(dEdxPDFs, 1);
 
  };
 
}