InnerDetector/InDetValidation/InDetTrackPerfMon/src/plots/ResolutionPlots.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "ResolutionPlots.h"
#include "../TrackParametersHelper.h"
 
#include "InDetPhysValMonitoring/ResolutionHelper.h"
 
#include "FourMomUtils/xAODP4Helpers.h"
 
IDTPM::ResolutionPlots::ResolutionPlots(
    PlotMgr* pParent, const std::string& dirName, 
    const std::string& anaTag, const std::string& trackType ) :
        PlotMgr( dirName, anaTag, pParent ), 
        m_trackType( trackType ) { }
 
 
void IDTPM::ResolutionPlots::initializePlots()
{
  StatusCode sc = bookPlots();
  if( sc.isFailure() ) {
    ATH_MSG_ERROR( "Failed to book resolution plots" );
  }
}
 
 
StatusCode IDTPM::ResolutionPlots::bookPlots()
{
  ATH_MSG_DEBUG( "Booking resolution plots in " << getDirectory() );
 
  for(unsigned int i=0; i<NPARAMS; i++){
    ATH_CHECK( retrieveAndBook( m_resHelperEta[i],
                "resHelper_"+m_trackType+"_eta_" + m_paramProp[i] ) );
    ATH_CHECK( retrieveAndBook( m_reswidth_vs_eta[i],
                "resolution_vs_"+m_trackType+"_eta_" + m_paramProp[i] ) );
    ATH_CHECK( retrieveAndBook( m_resmean_vs_eta[i],
                "resmean_vs_"+m_trackType+"_eta_" + m_paramProp[i] ) );
 
    ATH_CHECK( retrieveAndBook( m_pullHelperEta[i],
                "pullHelper_"+m_trackType+"_eta_" + m_paramProp[i] ) );
    ATH_CHECK( retrieveAndBook( m_pullwidth_vs_eta[i],
                "pullwidth_vs_"+m_trackType+"_eta_" + m_paramProp[i] ) );
    ATH_CHECK( retrieveAndBook( m_pullmean_vs_eta[i],
                "pullmean_vs_"+m_trackType+"_eta_" + m_paramProp[i] ) );
 
    ATH_CHECK( retrieveAndBook( m_resHelperPt[i],
                "resHelper_"+m_trackType+"_pt_" + m_paramProp[i] ) );
    ATH_CHECK( retrieveAndBook( m_reswidth_vs_pt[i],
                "resolution_vs_"+m_trackType+"_pt_" + m_paramProp[i] ) );
    ATH_CHECK( retrieveAndBook( m_resmean_vs_pt[i],
                "resmean_vs_"+m_trackType+"_pt_" + m_paramProp[i] ) );
    ATH_CHECK( retrieveAndBook(m_corrHelper[i], 
        "corr_"+m_trackType+"_"+m_paramProp[i]));
    ATH_CHECK( retrieveAndBook( m_pullHelperPt[i],
                "pullHelper_"+m_trackType+"_pt_" + m_paramProp[i] ) );
    ATH_CHECK( retrieveAndBook( m_pullwidth_vs_pt[i],
                "pullwidth_vs_"+m_trackType+"_pt_" + m_paramProp[i] ) );
    ATH_CHECK( retrieveAndBook( m_pullmean_vs_pt[i],
                "pullmean_vs_"+m_trackType+"_pt_" + m_paramProp[i] ) );
  }
 
  return StatusCode::SUCCESS;
}
 
 
template< typename REF, typename TEST >
StatusCode IDTPM::ResolutionPlots::fillPlots(
    const REF& particle_ref, const TEST& track_test, float weight )
{
  float refP[NPARAMS];
  float testP[NPARAMS];
  float testErrorP[NPARAMS];
 
  refP[D0] = d0(particle_ref);
  refP[Z0] = z0(particle_ref);
  refP[QOVERP] = qOverP(particle_ref) * Gaudi::Units::GeV;
  refP[QOVERPT] = qOverPT(particle_ref) * Gaudi::Units::GeV;
  refP[THETA] = theta(particle_ref);
  refP[PHI] = phi(particle_ref);
  refP[PT] = pT(particle_ref) / Gaudi::Units::GeV;
  refP[Z0SIN] = z0SinTheta(particle_ref);
 
  testP[D0] = d0(track_test);
  testP[Z0] = z0(track_test);
  testP[QOVERP] = qOverP(track_test) * Gaudi::Units::GeV;
  testP[QOVERPT] = qOverPT(track_test) * Gaudi::Units::GeV;
  testP[THETA] = theta(track_test);
  testP[PHI] = phi(track_test);
  testP[PT] = pT(track_test) / Gaudi::Units::GeV;
  testP[Z0SIN] = z0SinTheta(track_test);
 
  testErrorP[D0] = trackParameterError(track_test, Trk::d0);
  testErrorP[Z0] = trackParameterError(track_test, Trk::z0);
  testErrorP[QOVERP] = trackParameterError(track_test, Trk::qOverP) * Gaudi::Units::GeV;
  testErrorP[QOVERPT] = qOverPtError(track_test) * Gaudi::Units::GeV;
  testErrorP[THETA] = trackParameterError(track_test, Trk::theta);
  testErrorP[PHI] = trackParameterError(track_test, Trk::phi);
  testErrorP[PT] = pTError(track_test) / Gaudi::Units::GeV;
  testErrorP[Z0SIN] = z0SinThetaError(track_test);
 
  float eta_ref = eta(particle_ref);
  float pt_ref = pT(particle_ref) / Gaudi::Units::GeV;
 
  for(unsigned int i=0; i<NPARAMS; i++){
    float residual = testP[i]-refP[i];
    if(i==PHI) residual = xAOD::P4Helpers::deltaPhi(testP[i], refP[i]);
    float pull = residual / testErrorP[i];
 
    // Relative q/pt resolution
    if(i==QOVERPT) residual = testP[i]/refP[i] - 1.;
 
    ATH_CHECK( fill( m_resHelperEta[i], eta_ref, residual, weight ) );
    ATH_CHECK( fill( m_resHelperPt[i], pt_ref, residual, weight ) );
    ATH_CHECK( fill( m_pullHelperEta[i], eta_ref, pull, weight ) );
    ATH_CHECK( fill( m_pullHelperPt[i], pt_ref, pull, weight ) );
    ATH_CHECK( fill( m_corrHelper[i], refP[i], testP[i], weight) );
  }
  
  return StatusCode::SUCCESS;
}
 
 
template StatusCode IDTPM::ResolutionPlots::fillPlots< xAOD::TrackParticle, xAOD::TrackParticle >(
    const xAOD::TrackParticle&, const xAOD::TrackParticle&, float weight );
 
template StatusCode IDTPM::ResolutionPlots::fillPlots< xAOD::TruthParticle, xAOD::TrackParticle >(
    const xAOD::TruthParticle&, const xAOD::TrackParticle&, float weight );
 
template StatusCode IDTPM::ResolutionPlots::fillPlots< xAOD::TrackParticle, xAOD::TruthParticle >(
    const xAOD::TrackParticle&, const xAOD::TruthParticle&, float weight );
 
template StatusCode IDTPM::ResolutionPlots::fillPlots< xAOD::TruthParticle, xAOD::TruthParticle >(
    const xAOD::TruthParticle&, const xAOD::TruthParticle&, float weight );
 
 
void IDTPM::ResolutionPlots::finalizePlots()
{
  ATH_MSG_DEBUG( "Finalising resolution plots" );
 
  IDPVM::ResolutionHelper resolutionHelper;
 
  for (unsigned int i = 0; i < NPARAMS; i++) {
    resolutionHelper.makeResolutions(m_resHelperEta[i],
                     m_reswidth_vs_eta[i], m_resmean_vs_eta[i],
                     nullptr, false,
                     IDPVM::ResolutionHelper::iterRMS_convergence);
    resolutionHelper.makeResolutions(m_resHelperPt[i],
                     m_reswidth_vs_pt[i], m_resmean_vs_pt[i],
                     nullptr, false,
                     IDPVM::ResolutionHelper::iterRMS_convergence);
  }
 
}