DedxPDFs.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2017 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Jake Bennett                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/gearbox/Const.h>
 
#include <TFile.h>
#include <TObject.h>
#include <TH2F.h>
 
namespace Belle2 {
  class DedxPDFs: public TObject {
 
  public:
 
    DedxPDFs(): m_PXDpdfs() {};
 
    explicit DedxPDFs(TFile* pdffile)
    {
      for (unsigned int iPart = 0; iPart < Const::ChargedStable::c_SetSize; iPart++) {
        const int pdgCode = Const::chargedStableSet.at(iPart).getPDGCode();
 
        TH2F pxdhist;
        pdffile->Get(TString::Format("hist_d0_%d", pdgCode))->Copy(pxdhist);
        m_PXDpdfs.push_back(pxdhist);
 
        TH2F pxdhist_trunc;
        pdffile->Get(TString::Format("hist_d0_%d_trunc", pdgCode))->Copy(pxdhist_trunc);
        m_PXDpdfs_trunc.push_back(pxdhist_trunc);
 
        TH2F svdhist;
        pdffile->Get(TString::Format("hist_d1_%d", pdgCode))->Copy(svdhist);
        m_SVDpdfs.push_back(svdhist);
 
        TH2F svdhist_trunc;
        pdffile->Get(TString::Format("hist_d1_%d_trunc", pdgCode))->Copy(svdhist_trunc);
        m_SVDpdfs_trunc.push_back(svdhist_trunc);
 
        TH2F cdchist;
        pdffile->Get(TString::Format("hist_d2_%d", pdgCode))->Copy(cdchist);
        m_CDCpdfs.push_back(cdchist);
 
        TH2F cdchist_trunc;
        pdffile->Get(TString::Format("hist_d2_%d_trunc", pdgCode))->Copy(cdchist_trunc);
        m_CDCpdfs_trunc.push_back(cdchist_trunc);
      }
    };
 
    ~DedxPDFs() {};
 
    const TH2F* getPXDPDF(int part, bool truncated) const
    {
      return truncated ? &m_PXDpdfs_trunc[part] : &m_PXDpdfs[part];
    }
 
    const TH2F* getSVDPDF(int part, bool truncated) const
    {
      return truncated ? &m_SVDpdfs_trunc[part] : &m_SVDpdfs[part];
    }
 
    const TH2F* getCDCPDF(int part, bool truncated) const
    {
      return truncated ? &m_CDCpdfs_trunc[part] : &m_CDCpdfs[part];
    }
 
  private:
    std::vector<TH2F> m_PXDpdfs; 
    std::vector<TH2F> m_SVDpdfs; 
    std::vector<TH2F> m_CDCpdfs; 
    std::vector<TH2F> m_PXDpdfs_trunc; 
    std::vector<TH2F> m_SVDpdfs_trunc; 
    std::vector<TH2F> m_CDCpdfs_trunc; 
    ClassDef(DedxPDFs, 2); 
  };
} // end namespace Belle2