InDetPerfPlot_VertexTruthMatching.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETPHYSVALMONITORING_INDETPERFPLOT_VERTEXTRUTHMATCHING
#define INDETPHYSVALMONITORING_INDETPERFPLOT_VERTEXTRUTHMATCHING
 
// local includes
#include "InDetPlotBase.h"
#include "InDetPhysValMonitoringUtilities.h"
 
// Tracking includes:
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/VertexContainer.h"
#include "xAODTracking/VertexFwd.h"
 
// xAOD truth object includes:
#include "xAODTruth/TruthEventContainer.h"
#include "xAODTruth/TruthPileupEventContainer.h"
#include "xAODTruth/TruthVertex.h"
 
// Vertex validation:
#include "InDetTruthVertexValidation/InDetVertexTruthMatchTool.h"
#include "InDetTruthVertexValidation/InDetVertexTruthMatchUtils.h"
 
// std includes
#include <string>
 
// root includes
class TH1;
class TH2;
class TProfile;
class TEfficiency;
 
 
class InDetPerfPlot_VertexTruthMatching: public InDetPlotBase {
public:
    InDetPerfPlot_VertexTruthMatching(InDetPlotBase* pParent, const std::string& dirName, const int detailLevel = 10, bool isITk = false);
    void fill(const xAOD::Vertex& vertex, const xAOD::TruthVertex * tvrt =0, float weight=1.0);
    void fill(const xAOD::Vertex* recoHardScatter, const xAOD::VertexContainer& vertexContainer, const std::vector<const xAOD::TruthVertex*>& truthHSVertices, const std::vector<const xAOD::TruthVertex*>& truthPUVertices, float actualMu, float weight=1.0);
private:
    bool m_isITk{false};
    int m_detailLevel{10};
    float m_cutMinTruthRecoRadialDiff = 0.1;
    TH1* m_vx_type_truth{};
    TH1* m_vx_z_diff{};
    TH1* m_vx_z_diff_pull{};
 
    TH1* m_vx_time_diff{};
    TH1* m_vx_time_diff_pull{};
 
    TH1* m_vx_hs_classification{};
    TProfile* m_vx_nReco_vs_nTruth_inclusive;
    TProfile* m_vx_nReco_vs_nTruth_matched;
    TProfile* m_vx_nReco_vs_nTruth_merged;
    TProfile* m_vx_nReco_vs_nTruth_split;
    TProfile* m_vx_nReco_vs_nTruth_fake;
    TProfile* m_vx_nReco_vs_nTruth_dummy;
    TProfile* m_vx_nReco_vs_nTruth_clean;
    TProfile* m_vx_nReco_vs_nTruth_lowpu;
    TProfile* m_vx_nReco_vs_nTruth_highpu;
    TProfile* m_vx_nReco_vs_nTruth_hssplit;
    TProfile* m_vx_nReco_vs_nTruth_none;
    // HS vertex reconstruction efficiency vs PU
    TEfficiency* m_vx_hs_reco_eff;
    // HS vertex selection efficiency vs PU
    TEfficiency* m_vx_hs_sel_eff;
    // HS vertex selection efficiency vs nReco vertices
    TEfficiency* m_vx_hs_sel_eff_vs_nReco;
    // HS vertex reconstruction and selection efficiency vs PU
    TEfficiency* m_vx_hs_reco_sel_eff;
    // HS vertex selection efficiency vs PU (distance based)
    TEfficiency* m_vx_hs_sel_eff_dist;
    // HS vertex selection efficiency vs PU (actualInteractionsPerCrossing)
    TEfficiency* m_vx_hs_sel_eff_mu;  
    // HS vertex selection efficiency vs nReco vertices (distance based)
    TEfficiency* m_vx_hs_sel_eff_dist_vs_nReco;
    // HS vertex reconstruction efficiency vs n truth vertices
    TEfficiency* m_vx_hs_reco_eff_vs_ntruth;
    // HS vertex selection efficiency vs n truth vertices
    TEfficiency* m_vx_hs_sel_eff_vs_ntruth;
    // HS vertex reconstruction and selection efficiency vs n truth vertices
    TEfficiency* m_vx_hs_reco_sel_eff_vs_ntruth;
    // For reco (covariance) resolutions:
    TProfile* m_vx_hs_reco_long_reso;
    TProfile* m_vx_hs_reco_trans_reso;
    // For reco-truth resolutions:
    TH2* m_resHelper_PUdensity_hsVxTruthLong;
    TH1* m_resolution_vs_PUdensity_hsVxTruthLong;
    TH1* m_resmean_vs_PUdensity_hsVxTruthLong;
    TH2* m_resHelper_PUdensity_hsVxTruthTransv;
    TH1* m_resolution_vs_PUdensity_hsVxTruthTransv;
    TH1* m_resmean_vs_PUdensity_hsVxTruthTransv;
 
 
    TH1* m_vx_hs_z_pull;
    TH1* m_vx_hs_y_pull;
    TH1* m_vx_hs_x_pull;
    TH1* m_vx_all_z_pull;
    TH1* m_vx_all_y_pull;
    TH1* m_vx_all_x_pull;
    
    TH1* m_vx_hs_z_res;
    TH1* m_vx_hs_y_res;
    TH1* m_vx_hs_x_res;
    TH1* m_vx_all_z_res;
    TH1* m_vx_all_y_res;
    TH1* m_vx_all_x_res;
 
    TH2* m_vx_all_truth_z_res_vs_PU;
    TH2* m_vx_all_truth_x_res_vs_PU;
    TH2* m_vx_all_truth_y_res_vs_PU;
    TH2* m_vx_all_truth_z_pull_vs_PU;
    TH2* m_vx_all_truth_x_pull_vs_PU;
    TH2* m_vx_all_truth_y_pull_vs_PU;
 
    TH2* m_vx_all_truth_z_res_vs_nTrk;
    TH2* m_vx_all_truth_x_res_vs_nTrk;
    TH2* m_vx_all_truth_y_res_vs_nTrk;
    TH2* m_vx_all_truth_z_pull_vs_nTrk;
    TH2* m_vx_all_truth_x_pull_vs_nTrk;
    TH2* m_vx_all_truth_y_pull_vs_nTrk;
 
    TH2* m_vx_hs_truth_z_res_vs_PU;
    TH2* m_vx_hs_truth_x_res_vs_PU;
    TH2* m_vx_hs_truth_y_res_vs_PU;
    TH2* m_vx_hs_truth_z_pull_vs_PU;
    TH2* m_vx_hs_truth_x_pull_vs_PU;
    TH2* m_vx_hs_truth_y_pull_vs_PU;
 
    TH2* m_vx_hs_truth_z_res_vs_nTrk;
    TH2* m_vx_hs_truth_x_res_vs_nTrk;
    TH2* m_vx_hs_truth_y_res_vs_nTrk;
    TH2* m_vx_hs_truth_z_pull_vs_nTrk;
    TH2* m_vx_hs_truth_x_pull_vs_nTrk;
    TH2* m_vx_hs_truth_y_pull_vs_nTrk;
    // New Histograms for Expert Plots for observables for different vertex classifications
    TH1* m_vx_ntracks_matched;
    TH1* m_vx_ntracks_merged;
    TH1* m_vx_ntracks_split;
    TH1* m_vx_ntracks_HS_matched;
    TH1* m_vx_ntracks_HS_merged;
    TH1* m_vx_ntracks_HS_split;
    TH1* m_vx_ntracks_ALL_matched;
    TH1* m_vx_ntracks_ALL_merged;
    TH1* m_vx_ntracks_ALL_split;
    TH1* m_vx_sumpT_matched;
    TH1* m_vx_sumpT_merged;
    TH1* m_vx_sumpT_split;
    TH1* m_vx_sumpT_HS_matched;
    TH1* m_vx_sumpT_HS_merged;
    TH1* m_vx_sumpT_HS_split;
    
    TH1* m_vx_z_asym_matched;
    TH1* m_vx_z_asym_merged;
    TH1* m_vx_z_asym_split;
    TH1* m_vx_z_asym_HS_matched;
    TH1* m_vx_z_asym_HS_merged;
    TH1* m_vx_z_asym_HS_split;
    TH1* m_vx_z_asym_weighted_matched;
    TH1* m_vx_z_asym_weighted_merged;
    TH1* m_vx_z_asym_weighted_split;
    TH1* m_vx_z_asym_weighted_HS_matched;
    TH1* m_vx_z_asym_weighted_HS_merged;
    TH1* m_vx_z_asym_weighted_HS_split;
    
    TH1* m_vx_track_weight_matched;
    TH1* m_vx_track_weight_merged;
    TH1* m_vx_track_weight_split;
    TH1* m_vx_track_weight_HS_matched;
    TH1* m_vx_track_weight_HS_merged;
    TH1* m_vx_track_weight_HS_split;
 
    TH1* m_vx_normalised_track_weight_matched;
    TH1* m_vx_normalised_track_weight_merged;
    TH1* m_vx_normalised_track_weight_split;
    TH1* m_vx_normalised_track_weight_HS_matched;
    TH1* m_vx_normalised_track_weight_HS_merged;
    TH1* m_vx_normalised_track_weight_HS_split;
 
    TH1* m_vx_chi2Over_ndf_matched;
    TH1* m_vx_chi2Over_ndf_merged;
    TH1* m_vx_chi2Over_ndf_split;
    TH1* m_vx_chi2Over_ndf_HS_matched;
    TH1* m_vx_chi2Over_ndf_HS_merged;
    TH1* m_vx_chi2Over_ndf_HS_split;
 
    TH1* m_vx_z0_skewness_matched;
    TH1* m_vx_z0_skewness_merged;
    TH1* m_vx_z0_skewness_split;
    TH1* m_vx_z0_skewness_HS_matched;
    TH1* m_vx_z0_skewness_HS_merged;
    TH1* m_vx_z0_skewness_HS_split;
    TH1* m_vx_z0_kurtosis_matched;
    TH1* m_vx_z0_kurtosis_merged;
    TH1* m_vx_z0_kurtosis_split;
    TH1* m_vx_z0_kurtosis_HS_matched;
    TH1* m_vx_z0_kurtosis_HS_merged;
    TH1* m_vx_z0_kurtosis_HS_split;
 
    // ALL Vertices
    TH1* m_vx_sumpT_ALL_matched;
    TH1* m_vx_sumpT_ALL_merged;
    TH1* m_vx_sumpT_ALL_split;
    
    TH1* m_vx_z_asym_ALL_matched;
    TH1* m_vx_z_asym_ALL_merged;
    TH1* m_vx_z_asym_ALL_split;
    TH1* m_vx_z_asym_weighted_ALL_matched;
    TH1* m_vx_z_asym_weighted_ALL_merged;
    TH1* m_vx_z_asym_weighted_ALL_split;
    TH1* m_vx_track_weight_ALL_matched;
    TH1* m_vx_track_weight_ALL_merged;
    TH1* m_vx_track_weight_ALL_split;
    TH1* m_vx_normalised_track_weight_ALL_matched;
    TH1* m_vx_normalised_track_weight_ALL_merged;
    TH1* m_vx_normalised_track_weight_ALL_split;
 
    TH1* m_vx_chi2Over_ndf_ALL_matched;
    TH1* m_vx_chi2Over_ndf_ALL_merged;
    TH1* m_vx_chi2Over_ndf_ALL_split;
 
    TH1* m_vx_z0_skewness_ALL_matched;
    TH1* m_vx_z0_skewness_ALL_merged;
    TH1* m_vx_z0_skewness_ALL_split;
    TH1* m_vx_z0_kurtosis_ALL_matched;
    TH1* m_vx_z0_kurtosis_ALL_merged;
    TH1* m_vx_z0_kurtosis_ALL_split;
 
    TH1* m_vx_nVertices_ALL_matched;
    TH1* m_vx_nVertices_ALL_merged;
    TH1* m_vx_nVertices_ALL_split;
    TH1* m_vx_nVertices_ALL_fake;
    TH1* m_vx_nVertices_HS_matched;
    TH1* m_vx_nVertices_HS_merged;
    TH1* m_vx_nVertices_HS_split;
    TH1* m_vx_nVertices_HS_fake;
    TH1* m_vx_nVertices_matched;
    TH1* m_vx_nVertices_merged;
    TH1* m_vx_nVertices_split;
    TH1* m_vx_nVertices_fake;
 
    // delta z plot
    TH1* m_vx_all_dz;
    TH1* m_vx_hs_mindz;
    
    TH1* m_vx_PUdensity;
    TH1* m_vx_nTruth;
    TH2* m_vx_nTruth_vs_PUdensity;
 
private:
    // plot base has no default implementation of this; we use it to book the histos
    void initializePlots();
    static const xAOD::Vertex* getHSRecoVertexSumPt2(const xAOD::VertexContainer& recoVertices) ;
    template<typename U, typename V>
    float getRadialDiff2(const U* vtx1, const V* vtx2) const;
    float getLocalPUDensity(const xAOD::TruthVertex* vtxOfInterest, const std::vector<const xAOD::TruthVertex*>& truthHSVertices, const std::vector<const xAOD::TruthVertex*>& truthPUVertices, const float radialWindow = 2.0) const;
    static float getRecoLongitudinalReso(const xAOD::Vertex* recoVtx) ;
    static float getRecoTransverseReso(const xAOD::Vertex* recoVtx) ;
    const xAOD::TruthVertex* getTruthVertex(const xAOD::Vertex* recoVtx) const;
    static void fillResoHist(TH1* resoHist, const TH2* resoHist2D);
    void finalizePlots();
};
 
#endif