CombinedPIDPerformanceModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Jake Bennett
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef COMBINEDPIDPERFORMANCEMODULE_H_
#define COMBINEDPIDPERFORMANCEMODULE_H_
 
#include <framework/core/Module.h>
#include <TFile.h>
#include <TList.h>
#include <TEfficiency.h>
#include <TH3F.h>
 
#include <mdst/dataobjects/MCParticle.h>
#include <mdst/dataobjects/PIDLikelihood.h>
#include <mdst/dataobjects/Track.h>
#include <mdst/dataobjects/TrackFitResult.h>
 
#include <framework/datastore/StoreArray.h>
 
namespace Belle2 {
  class CombinedPIDPerformanceModule : public Module {
 
  public:
 
    CombinedPIDPerformanceModule();
 
    virtual ~CombinedPIDPerformanceModule();
 
    virtual void initialize() override;
 
    virtual void event() override;
 
    virtual void terminate() override;
 
  private:
 
    // required input
    StoreArray<Track> m_tracks; 
    StoreArray<TrackFitResult> m_trackFitResults; 
    StoreArray<PIDLikelihood> m_pidLikelihoods; 
    StoreArray<MCParticle> m_mcParticles; 
    TList* m_histoList{nullptr};
 
    TEfficiency* createEfficiency(const char* name, const char* title,
                                  Int_t nbins, Double_t min, Double_t max, TList* histoList = NULL);
 
    void fillEfficiencyHistos(const TrackFitResult* fitResult, const PIDLikelihood* pid, int pdg);
 
    /* user-defined parameters */
    std::string m_rootFileName;   
    std::string m_mdstType;       
    int m_nbins;                  
    double m_pLow;                
    double m_pHigh;               
    /* ROOT file related parameters */
    TFile* m_rootFilePtr; 
    /* list of histograms filled per MCParticle found in the event */
    // TEfficiencies for each particle type
    std::vector< TEfficiency* > m_piK_Efficiencies; 
    std::vector< TEfficiency* > m_Kpi_Efficiencies; 
    std::vector< TEfficiency* > m_ppi_Efficiencies; 
    std::vector< TEfficiency* > m_pK_Efficiencies;  
    std::vector< TEfficiency* > m_dpi_Efficiencies; 
    std::vector< TEfficiency* > m_epi_Efficiencies; 
    std::vector< TEfficiency* > m_mpi_Efficiencies; 
    // TEfficiencies for fake rates
    std::vector< TEfficiency* > m_piK_FakeRates;    
    std::vector< TEfficiency* > m_Kpi_FakeRates;    
    std::vector< TEfficiency* > m_ppi_FakeRates;    
    std::vector< TEfficiency* > m_pK_FakeRates;     
    std::vector< TEfficiency* > m_dpi_FakeRates;    
    std::vector< TEfficiency* > m_epi_FakeRates;    
    std::vector< TEfficiency* > m_mpi_FakeRates;    
    TH3F* h_ROC[5][10] = {}; 
    double pidvalue(float pida, float pidb);
 
    bool pidavail(const PIDLikelihood* pid, Const::DetectorSet dets);
 
    // determine which detectors to use for each particle
    Const::DetectorSet chargedSet;  
    Const::DetectorSet electronSet; 
    Const::DetectorSet muonSet;     
    enum dettype {svd, cdc, top, arich, klm, ecl, dedx, dedxtop, dedxecl, all};
    std::vector<int> detset;  
    std::vector<int> edetset; 
  };
} // end of namespace
 
#endif