HitsOnTracksPlots.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "HitsOnTracksPlots.h"
#include "../TrackParametersHelper.h"
 
 
IDTPM::HitsOnTracksPlots::HitsOnTracksPlots(
    PlotMgr* pParent,
    const std::string& dirName,
    const std::string& anaTag,
    const std::string& testType,
    const std::string& refType,
    bool isITk, bool doGlobalPlots, bool doTruthMuPlots, bool do1D ) :
        PlotMgr( dirName, anaTag, pParent ),
        m_testType( testType ), m_refType( refType ),
        m_isITk( isITk ), m_doGlobalPlots( doGlobalPlots ),
        m_doTruthMuPlots( doTruthMuPlots ), m_do1D( do1D )
{
  m_NHITPARAMS = NHITPARAMSBASE;
  m_NRUN3HITPARAMS = NRUN3HITPARAMSBASE;
}
 
 
IDTPM::HitsOnTracksPlots::HitsOnTracksPlots(
    PlotMgr* pParent,
    const std::string& dirName,
    const std::string& anaTag,
    const std::string& trackType,
    bool isITk, bool doGlobalPlots, bool doTruthMuPlots ) :
        PlotMgr( dirName, anaTag, pParent ),
        m_testType( trackType ), m_refType( trackType ),
        m_isITk( isITk ), m_doGlobalPlots( doGlobalPlots ),
        m_doTruthMuPlots( doTruthMuPlots ), m_do1D( true )
{
  m_NHITPARAMS = NHITPARAMSBASE;
  m_NRUN3HITPARAMS = NRUN3HITPARAMSBASE;
}
 
 
void IDTPM::HitsOnTracksPlots::initializePlots()
{
  StatusCode sc = bookPlots();
  if( sc.isFailure() ) {
    ATH_MSG_ERROR( "Failed to book hits on tracks plots" );
  }
}
 
 
StatusCode IDTPM::HitsOnTracksPlots::bookPlots()
{
  ATH_MSG_DEBUG( "Booking hits on tracks plots in " << getDirectory() ); 
 
  if( m_do1D ) {
    for( unsigned int ih=0; ih<m_NHITPARAMS; ih++ ) {
      ATH_CHECK( retrieveAndBook( m_hits[ih], m_testType+"_"+m_hitParamName[ih] ) );
    }
 
    if( not m_isITk ) {
      for( unsigned int ih=0; ih<m_NRUN3HITPARAMS; ih++ ) {
        ATH_CHECK( retrieveAndBook( m_hitsRun3[ih], m_testType+"_"+m_run3HitParamName[ih] ) );
      }
    }
  }
 
  for( unsigned int i=0; i<NPARAMS; i++ ) {
    for( unsigned int ih=0; ih<m_NHITPARAMS; ih++ ) {
      ATH_CHECK( retrieveAndBook( m_hits_vs[ih][i],
          m_testType+"_"+m_hitParamName[ih]+"_vs_"+m_refType+"_"+m_paramName[i] ) );
    }
 
    if( not m_isITk ) {
      for( unsigned int ih=0; ih<m_NRUN3HITPARAMS; ih++ ) {
        ATH_CHECK( retrieveAndBook( m_hitsRun3_vs[ih][i],
            m_testType+"_"+m_run3HitParamName[ih]+"_vs_"+m_refType+"_"+m_paramName[i] ) );
      }
    }
 
    for( unsigned int j=i+1; j<NPARAMS; j++ ) {
      for( unsigned int ih=0; ih<m_NHITPARAMS; ih++ ) {
        ATH_CHECK( retrieveAndBook( m_hits_vs2D[ih][i][j],
            m_testType+"_"+m_hitParamName[ih]+"_vs_"+m_refType+"_"+m_paramName[i]+"_vs_"+m_paramName[j] ) );
      }
 
      if( not m_isITk ) {
        for( unsigned int ih=0; ih<m_NRUN3HITPARAMS; ih++ ) {
          ATH_CHECK( retrieveAndBook( m_hitsRun3_vs2D[ih][i][j],
              m_testType+"_"+m_run3HitParamName[ih]+"_vs_"+m_refType+"_"+m_paramName[i]+"_vs_"+m_paramName[j] ) );
        }
      }
    } // close j loop over NPARAMS
  } // close i loop over NPARAMS
 
  if( m_doGlobalPlots ) {
    for( unsigned int i=0; i<NPARAMSMU; i++ ) {
      if( not m_doTruthMuPlots and i==TRUTHMU ) continue;
      if( m_do1D ) {
        for( unsigned int ih=0; ih<m_NHITPARAMS; ih++ ) {
          ATH_CHECK( retrieveAndBook( m_hits_vsMu[ih][i],
              m_testType+"_"+m_hitParamName[ih]+"_vs_"+m_paramMuName[i] ) );
        }
 
        if( not m_isITk ) {
          for( unsigned int ih=0; ih<m_NRUN3HITPARAMS; ih++ ) {
            ATH_CHECK( retrieveAndBook( m_hitsRun3_vsMu[ih][i],
                m_testType+"_"+m_run3HitParamName[ih]+"_vs_"+m_paramMuName[i] ) );
          }
        }
      } // close if m_do1D
 
      for( unsigned int j=0; j<NPARAMS; j++ ) {
        for( unsigned int ih=0; ih<m_NHITPARAMS; ih++ ) {
          ATH_CHECK( retrieveAndBook( m_hits_vsMu_vs[ih][i][j],
              m_testType+"_"+m_hitParamName[ih]+"_vs_"+m_paramMuName[i]+"_vs_"+m_refType+"_"+m_paramName[j] ) );
        }
 
        if( not m_isITk ) {
          for( unsigned int ih=0; ih<m_NRUN3HITPARAMS; ih++ ) {
            ATH_CHECK( retrieveAndBook( m_hitsRun3_vsMu_vs[ih][i][j],
                m_testType+"_"+m_run3HitParamName[ih]+"_vs_"+m_paramMuName[i]+"_vs_"+m_refType+"_"+m_paramName[j] ) );
          }
        }
      } // close j loop over NPARAMS
    } // close i loop over NPARAMSMU
  } // end if m_doGlobalPlots
 
  return StatusCode::SUCCESS;
}
 
 
template< typename PTEST, typename PREF >
StatusCode IDTPM::HitsOnTracksPlots::fillPlots(
    const PTEST& ptest, const PREF& pref, float truthMu, float actualMu, float weight )
{
  float hitP[ NHITPARAMSTOT ];
  hitP[ NInnerMostPixelHits ]        = nInnerMostPixelHits( ptest );
  hitP[ NInnerMostPixelEndcapHits ]  = nInnerMostPixelEndcapHits( ptest );
  hitP[ NNextToInnerMostPixelHits ]  = nNextToInnerMostPixelHits( ptest );
  hitP[ NNextToInnerMostPixelEndcapHits ]  = nNextToInnerMostPixelEndcapHits( ptest );
  hitP[ NInnerMostPixelSharedHits ]        = nInnerMostPixelSharedHits( ptest );
  hitP[ NInnerMostPixelSharedEndcapHits ]  = nInnerMostPixelSharedEndcapHits( ptest );
  hitP[ NPixelHits ]          = nPixelHits( ptest );
  hitP[ NPixelHoles ]         = nPixelHoles( ptest );
  hitP[ NPixelSharedHits ]    = nPixelSharedHits( ptest );
  hitP[ PixeldEdx ]           = pixeldEdx( ptest );
  hitP[ NSCTHits ]            = nSCTHits( ptest );
  hitP[ NSCTHoles ]           = nSCTHoles( ptest ); 
  hitP[ NSCTSharedHits ]      = nSCTSharedHits( ptest );
 
  float hitRun3P[ NRUN3HITPARAMSTOT ];
  hitRun3P[ NTRTHits ]                = ( not m_isITk ) ? nTRTHits( ptest ) : -9999.;
  hitRun3P[ NTRTHitsXe ]              = ( not m_isITk ) ? nTRTHitsXe( ptest ) : -9999.;
  hitRun3P[ NTRTHitsAr ]              = ( not m_isITk ) ? nTRTHitsAr( ptest ) : -9999.;
  hitRun3P[ NTRTHighThresholdHits ]   = ( not m_isITk ) ? nTRTHighThresholdHits( ptest ) : -9999.;
  hitRun3P[ NTRTHighThresholdHitsXe ] = ( not m_isITk ) ? nTRTHighThresholdHitsXe( ptest ) : -9999.;
  hitRun3P[ NTRTHighThresholdHitsAr ] = ( not m_isITk ) ? nTRTHighThresholdHitsAr( ptest ) : -9999.;
 
  float refP[ NPARAMS ];
  refP[ PT ]  = pT( pref ) / Gaudi::Units::GeV;
  refP[ ETA ] = eta( pref );
  refP[ PHI ] = phi( pref );
 
  float muP[ NPARAMSMU ];
  muP[ TRUTHMU ]  = truthMu;
  muP[ ACTUALMU ] = actualMu;
 
 
  if( m_do1D ) {
    for( unsigned int ih=0; ih<m_NHITPARAMS; ih++ ) {
      ATH_CHECK( fill( m_hits[ih], hitP[ih], weight ) );
    }
 
    if( not m_isITk ) {
      for( unsigned int ih=0; ih<m_NRUN3HITPARAMS; ih++ ) {
        ATH_CHECK( fill( m_hitsRun3[ih], hitRun3P[ih], weight ) );
      }
    }
  }
 
  for( unsigned int i=0; i<NPARAMS; i++ ) {
    for( unsigned int ih=0; ih<m_NHITPARAMS; ih++ ) {
      ATH_CHECK( fill( m_hits_vs[ih][i], refP[i], hitP[ih], weight ) );
    }
 
    if( not m_isITk ) {
      for( unsigned int ih=0; ih<m_NRUN3HITPARAMS; ih++ ) {
        ATH_CHECK( fill( m_hitsRun3_vs[ih][i], refP[i], hitRun3P[ih], weight ) );
      }
    }
 
    for( unsigned int j=i+1; j<NPARAMS; j++ ) {
      for( unsigned int ih=0; ih<m_NHITPARAMS; ih++ ) {
        ATH_CHECK( fill( m_hits_vs2D[ih][i][j], refP[i], refP[j], hitP[ih], weight ) );
      }
 
      if( not m_isITk ) {
        for( unsigned int ih=0; ih<m_NRUN3HITPARAMS; ih++ ) {
          ATH_CHECK( fill( m_hitsRun3_vs2D[ih][i][j], refP[i], refP[j], hitRun3P[ih], weight ) );
        }
      }
    } // close j loop over NPARAMS
  } // close i loop over NPARAMS
 
  if( m_doGlobalPlots ) {
    for( unsigned int i=0; i<NPARAMSMU; i++ ) {
      if( not m_doTruthMuPlots and i==TRUTHMU ) continue;
      if( m_do1D ) {
        for( unsigned int ih=0; ih<m_NHITPARAMS; ih++ ) {
          ATH_CHECK( fill( m_hits_vsMu[ih][i], muP[i], hitP[ih], weight ) );
        }
 
        if( not m_isITk ) {
          for( unsigned int ih=0; ih<m_NRUN3HITPARAMS; ih++ ) {
            ATH_CHECK( fill( m_hitsRun3_vsMu[ih][i], muP[i], hitRun3P[ih], weight ) );
          }
        }
      } // close if m_do1D
 
      for( unsigned int j=0; j<NPARAMS; j++ ) {
        for( unsigned int ih=0; ih<m_NHITPARAMS; ih++ ) {
          ATH_CHECK( fill( m_hits_vsMu_vs[ih][i][j], muP[i], refP[j], hitP[ih], weight ) );
        }
 
        if( not m_isITk ) {
          for( unsigned int ih=0; ih<m_NRUN3HITPARAMS; ih++ ) {
            ATH_CHECK( fill( m_hitsRun3_vsMu_vs[ih][i][j], muP[i], refP[j], hitRun3P[ih], weight ) );
          }
        }
      } // close j loop over NPARAMS
    } // close i loop over NPARAMSMU
  } // end if m_doGlobalPlots
 
  return StatusCode::SUCCESS;
}
 
template StatusCode IDTPM::HitsOnTracksPlots::fillPlots< xAOD::TrackParticle >(
    const xAOD::TrackParticle&, const xAOD::TrackParticle&, float, float, float );
 
template StatusCode IDTPM::HitsOnTracksPlots::fillPlots< xAOD::TruthParticle >(
    const xAOD::TruthParticle&, const xAOD::TruthParticle&, float, float, float );
 
template StatusCode IDTPM::HitsOnTracksPlots::fillPlots< xAOD::TrackParticle, xAOD::TruthParticle >(
    const xAOD::TrackParticle&, const xAOD::TruthParticle&, float, float, float );
 
template StatusCode IDTPM::HitsOnTracksPlots::fillPlots< xAOD::TruthParticle, xAOD::TrackParticle >(
    const xAOD::TruthParticle&, const xAOD::TrackParticle&, float, float, float );
 
 
template< typename PARTICLE >
StatusCode IDTPM::HitsOnTracksPlots::fillPlots(
    const PARTICLE& particle, float truthMu, float actualMu, float weight )
{
  ATH_CHECK( fillPlots< PARTICLE >( particle, particle, truthMu, actualMu, weight ) );
  return StatusCode::SUCCESS;
}
 
template StatusCode IDTPM::HitsOnTracksPlots::fillPlots< xAOD::TrackParticle >(
    const xAOD::TrackParticle&, float, float, float );
 
template StatusCode IDTPM::HitsOnTracksPlots::fillPlots< xAOD::TruthParticle >(
    const xAOD::TruthParticle&, float, float, float );
 
 
void IDTPM::HitsOnTracksPlots::finalizePlots()
{
  ATH_MSG_DEBUG( "Finalising hits on tracks plots" );
}