TrackingPerformanceEvaluationModule.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 <framework/core/Module.h>
#include <tracking/modules/trackingPerformanceEvaluation/PerformanceEvaluationBaseClass.h>
#include <mdst/dataobjects/Track.h>
#include <mdst/dataobjects/MCParticle.h>
#include <tracking/dataobjects/MCParticleInfo.h>
#include <tracking/dataobjects/RecoTrack.h>
 
#include <TList.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TH3F.h>
 
 
// forward declarations
namespace Belle2 {
  class Track;
  class TrackFitResult;
 
  class TrackingPerformanceEvaluationModule : public Module, PerformanceEvaluationBaseClass {
 
  public:
 
    TrackingPerformanceEvaluationModule();
 
    ~TrackingPerformanceEvaluationModule();
 
    void initialize() override;
 
    void beginRun() override;
 
    void event() override;
 
    void endRun() override;
 
    void terminate() override;
 
  private:
 
    //list of functions to fill histograms
    void fillTrackParams1DHistograms(const TrackFitResult* fitResult,
                                     MCParticleInfo mcParticleInfo); 
    void fillTrackErrParams2DHistograms(const TrackFitResult* fitResult); 
    void fillHitsUsedInTrackFitHistograms(const Track& track); 
    bool isTraceable(const MCParticle& the_mcParticle); 
    void addMoreEfficiencyPlots(TList* histoList); 
    void addMoreInefficiencyPlots(TList* histoList); 
    /* user-defined parameters */
    std::string m_MCParticlesName; 
    std::string m_MCRecoTracksName; 
    std::string m_RecoTracksName; 
    std::string m_TracksName; 
    int m_ParticleHypothesis;  
    StoreArray<MCParticle> m_MCParticles; 
    StoreArray<RecoTrack> m_PRRecoTracks; 
    StoreArray<RecoTrack> m_MCRecoTracks; 
    StoreArray<Track> m_Tracks;           
    /* list of histograms filled per MCParticle found in the event */
 
    TH1F* m_multiplicityTracks = nullptr; 
    TH1F* m_multiplicityRecoTracks = nullptr; 
    TH1F* m_multiplicityMCRecoTracks = nullptr; 
    TH1F* m_multiplicityFittedTracks = nullptr; 
    TH1F* m_multiplicityFittedTracksPerMCRT = nullptr; 
    TH1F* m_multiplicityMCParticlesPerTrack = nullptr;  
    TH1F* m_multiplicityRecoTracksPerMCRT = nullptr;  
    TH1F* m_multiplicityMCRecoTracksPerRT = nullptr;  
    //error on track parameters
    TH1F* m_h1_d0_err = nullptr; 
    TH1F* m_h1_phi_err = nullptr; 
    TH1F* m_h1_omega_err = nullptr; 
    TH1F* m_h1_z0_err = nullptr; 
    TH1F* m_h1_cotTheta_err = nullptr; 
    //residuals on track parameters
    TH1F* m_h1_d0_res = nullptr; 
    TH1F* m_h1_phi_res = nullptr; 
    TH1F* m_h1_omega_res = nullptr; 
    TH1F* m_h1_z0_res = nullptr; 
    TH1F* m_h1_cotTheta_res = nullptr; 
    //residuals on momentum parameters
    TH1F* m_h1_px_res = nullptr; 
    TH1F* m_h1_py_res = nullptr; 
    TH1F* m_h1_pz_res = nullptr; 
    TH1F* m_h1_p_res = nullptr; 
    TH1F* m_h1_pt_res = nullptr; 
    //residuals on position parameters
    TH1F* m_h1_x_res = nullptr; 
    TH1F* m_h1_y_res = nullptr; 
    TH1F* m_h1_z_res = nullptr; 
    TH1F* m_h1_r_res = nullptr; 
    TH1F* m_h1_rtot_res = nullptr; 
    //pulls on track parameters
    TH1F* m_h1_d0_pll = nullptr; 
    TH1F* m_h1_phi_pll = nullptr; 
    TH1F* m_h1_omega_pll = nullptr; 
    TH1F* m_h1_z0_pll = nullptr; 
    TH1F* m_h1_cotTheta_pll = nullptr; 
    TH2F* m_h2_d0errphi0err_xy = nullptr; 
    TH2F* m_h2_d0errphi0err_rz = nullptr; 
    TH2F* m_h2_z0errcotThetaerr_xy = nullptr; 
    TH2F* m_h2_VXDhitsPR_xy = nullptr; 
    TH2F* m_h2_VXDhitsPR_rz = nullptr; 
    TH1F* m_h1_nVXDhitsPR = nullptr; 
    TH1F* m_h1_nVXDhitsWeighted = nullptr; 
    TH1F* m_h1_nVXDhitsUsed = nullptr; 
    TH1F* m_h1_nCDChitsPR = nullptr; 
    TH1F* m_h1_nCDChitsWeighted = nullptr; 
    TH1F* m_h1_nCDChitsUsed = nullptr;
    TH1F* m_h1_nHitDetID = nullptr; 
    TH2F* m_h2_TrackPointFitWeightVXD = nullptr; 
    TH2F* m_h2_TrackPointFitWeightCDC = nullptr; 
    TH1F* m_h1_pValue = nullptr; 
    TH2F* m_h2_OmegaerrOmegaVSpt = nullptr;  
    TH2F* m_h2_z0errVSpt_wtpxd = nullptr;  
    TH2F* m_h2_z0errVSpt_wfpxd = nullptr;  
    TH2F* m_h2_z0errVSpt_wpxd = nullptr;  
    TH2F* m_h2_z0errVSpt_wopxd = nullptr; 
    TH2F* m_h2_z0errVSpt = nullptr; 
    TH2F* m_h2_d0errVSpt_wtpxd = nullptr; 
    TH2F* m_h2_d0errVSpt_wfpxd = nullptr; 
    TH2F* m_h2_d0errVSpt_wpxd = nullptr; 
    TH2F* m_h2_d0errVSpt_wopxd = nullptr; 
    TH2F* m_h2_d0errVSpt = nullptr; 
    TH2F* m_h2_d0errMSVSpt = nullptr; 
    TH2F* m_h2_chargeVSchargeMC = nullptr; 
    //histograms used for efficiency plots
    TH1F* m_h1_HitsRecoTrackPerMCRecoTrack = nullptr;  
    TH1F* m_h1_HitsMCRecoTrack = nullptr;  
    TH3F* m_h3_MCParticle = nullptr;  
    TH3F* m_h3_MCParticleswPXDHits = nullptr;  
    TH3F* m_h3_TracksPerMCParticle = nullptr;  
    TH3F* m_h3_TrackswPXDHitsPerMCParticle = nullptr;  
    TH3F* m_h3_RecoTrackswPXDHitsPerMCParticle = nullptr;  
    TH3F* m_h3_RecoTrackswPXDHitsPerMCParticlewPXDHits = nullptr;  
    TH3F* m_h3_MCRecoTrack = nullptr;  
    TH3F* m_h3_TracksPerMCRecoTrack = nullptr;  
    TH3F* m_h3_MCParticle_plus = nullptr;  
    TH3F* m_h3_TracksPerMCParticle_plus = nullptr;  
    TH3F* m_h3_MCRecoTrack_plus = nullptr;  
    TH3F* m_h3_TracksPerMCRecoTrack_plus = nullptr;  
    TH3F* m_h3_MCParticle_minus = nullptr;  
    TH3F* m_h3_TracksPerMCParticle_minus = nullptr;  
    TH3F* m_h3_MCRecoTrack_minus = nullptr;  
    TH3F* m_h3_TracksPerMCRecoTrack_minus = nullptr;  
    //histograms used for purity plots
    TH3F* m_h3_MCParticlesPerTrack = nullptr;  
    TH3F* m_h3_Tracks = nullptr; 
  };
} // end of namespace