PerformanceEvaluationBaseClass.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 <TFile.h>
#include <TList.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TH3F.h>
 
// forward declarations
class TFile;
 
namespace Belle2 {
  class PerformanceEvaluationBaseClass  {
 
  public:
 
    PerformanceEvaluationBaseClass();
 
    virtual ~PerformanceEvaluationBaseClass();
 
    TList* m_histoList = nullptr; 
    TList* m_histoList_multiplicity = nullptr; 
    TList* m_histoList_evtCharacterization = nullptr; 
    TList* m_histoList_trkQuality = nullptr; 
    TList* m_histoList_firstHit = nullptr; 
    TList* m_histoList_pr = nullptr; 
    TList* m_histoList_fit = nullptr; 
    TList* m_histoList_efficiency = nullptr; 
    TList* m_histoList_purity = nullptr; 
    TList* m_histoList_others = nullptr; 
    //list of functions to create histograms:
    TH1F* createHistogram1D(const char* name, const char* title,
                            Int_t nbins, Double_t min, Double_t max,
                            const char* xtitle, TList* histoList = nullptr);
 
    TH1F* createHistogram1D(const char* name, const char* title,
                            Int_t nbins, Double_t* bins,
                            const char* xtitle, TList* histoList = nullptr);
 
    TH2F* createHistogram2D(const char* name, const char* title,
                            Int_t nbinsX, Double_t minX, Double_t maxX, const char* titleX,
                            Int_t nbinsY, Double_t minY, Double_t maxY, const char* titleY,
                            TList* histoList = nullptr); 
    TH2F* createHistogram2D(const char* name, const char* title,
                            Int_t nbinsX, Double_t* binsX, const char* titleX,
                            Int_t nbinsY, Double_t* binsY, const char* titleY,
                            TList* histoList = nullptr);
 
    TH3F* createHistogram3D(const char* name, const char* title,
                            Int_t nbinsX, Double_t minX, Double_t maxX, const char* titleX,
                            Int_t nbinsY, Double_t minY, Double_t maxY, const char* titleY,
                            Int_t nbinsZ, Double_t minZ, Double_t maxZ, const char* titleZ,
                            TList* histoList = nullptr);
 
    TH3F* createHistogram3D(const char* name, const char* title,
                            Int_t nbinsX, Double_t* binsX, const char* titleX,
                            Int_t nbinsY, Double_t* binsY, const char* titleY,
                            Int_t nbinsZ, Double_t* binsZ, const char* titleZ,
                            TList* histoList = nullptr);
 
    TH1* duplicateHistogram(const char* newname, const char* newtitle,
                            TH1* h, TList* histoList = nullptr);
 
 
    TH1F* createHistogramsRatio(const char* name, const char* title,
                                TH1* hNum, TH1* hDen, bool isEffPlot,
                                int axisRef);
 
 
 
    void addEfficiencyPlots(TList* graphList = nullptr, TH3F* h3_xPerMCParticle = nullptr, TH3F* h3_MCParticle = nullptr);
    void addInefficiencyPlots(TList* graphList = nullptr, TH3F* h3_xPerMCParticle = nullptr,
                              TH3F* h3_MCParticle = nullptr);
    void addPurityPlots(TList* graphList = nullptr, TH3F* h3_xPerMCParticle = nullptr, TH3F* h3_MCParticle = nullptr);
 
    TH1F* effPlot1D(TH1F* h1_den, TH1F* h1_num,
                    const char* name, const char* title, bool geo_accettance, TList* histoList = nullptr);
 
    TH1F* effPlot1D(TH1F* h1_MC, TH1F* h1_RecoTrack, TH1F* h1_Track,
                    const char* name, const char* title, TList* histoList = nullptr);
 
    TH2F* effPlot2D(TH2F* h2_den, TH2F* h2_num,
                    const char* name, const char* title, bool geo_accettance, TList* histoList = nullptr);
 
    TH2F* effPlot2D(TH2F* h2_MC, TH2F* h2_RecoTrack, TH2F* h2_Track,
                    const char* name, const char* title, TList* histoList);
 
    TH1F* geoAcc1D(TH1F* h1_den, TH1F* h1_num,
                   const char* name, const char* title, TList* histoList = nullptr);
 
    TH2F* geoAcc2D(TH2F* h2_den, TH2F* h2_num,
                   const char* name, const char* title, TList* histoList = nullptr);
 
    TH1F* V0FinderEff(TH1F* h1_dau0, TH1F* h1_dau1, TH1F* h1_Mother,
                      const char* name, const char* title, TList* histoList = nullptr);
 
 
    /* user-defined parameters */
    std::string m_rootFileName;   
    /* ROOT file related parameters */
    TFile* m_rootFilePtr = nullptr; 
  };
} // end of namespace