NtracksPlots.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/


#include "NtracksPlots.h"
#include "../TrackParametersHelper.h"
 

IDTPM::NtracksPlots::NtracksPlots(
    PlotMgr* pParent, const std::string& dirName, 
    const std::string& anaTag, const std::string& trackType,
    bool doTrigger, bool doTruthMuPlots ) :
        PlotMgr( dirName, anaTag, pParent ), 
        m_trackType( trackType ),
        m_doTrigger( doTrigger ),
        m_doTruthMuPlots( doTruthMuPlots ) { }
 

void IDTPM::NtracksPlots::initializePlots()
{
  StatusCode sc = bookPlots();
  if( sc.isFailure() ) {
    ATH_MSG_ERROR( "Failed to book track multiplicity plots" );
  }
}
 
 
StatusCode IDTPM::NtracksPlots::bookPlots()
{
  ATH_MSG_DEBUG( "Booking track multiplicity plots in " << getDirectory() ); 
 
  for( size_t i=0; i<NCOUNTERS; i++ ) {
    if( not m_doTrigger and i == INROI ) continue;

    ATH_CHECK( retrieveAndBook( m_nTracks[i], "num_"+m_trackType+"_"+m_counterName[i] ) );

    ATH_CHECK( retrieveAndBook(
        m_nTracks_vs_nVertices[i],
        "num_"+m_trackType+"_"+m_counterName[i]+"_vs_num_vtx_"+m_trackType+"_"+m_counterName[i] ) );

    ATH_CHECK( retrieveAndBook(
        m_nTracks_vs_actualMu[i], "num_"+m_trackType+"_"+m_counterName[i]+"_vs_actualMu" ) );
    if( m_doTruthMuPlots ) ATH_CHECK( retrieveAndBook(
        m_nTracks_vs_truthMu[i],  "num_"+m_trackType+"_"+m_counterName[i]+"_vs_truthMu" ) );

    ATH_CHECK( retrieveAndBook(
        m_avg_nTracks_vs_actualMu[i], "avgNum_"+m_trackType+"_"+m_counterName[i]+"_vs_actualMu" ) );
    if( m_doTruthMuPlots ) ATH_CHECK( retrieveAndBook(
        m_avg_nTracks_vs_truthMu[i],  "avgNum_"+m_trackType+"_"+m_counterName[i]+"_vs_truthMu" ) );
  }
 
  return StatusCode::SUCCESS;
}
 

StatusCode IDTPM::NtracksPlots::fillPlots(
    const std::vector< size_t >& trackCounts,
    const std::vector< size_t >& vertexCounts,
    float truthMu,
    float actualMu,
    float weight )
{
  if( trackCounts.size() != NCOUNTERS or vertexCounts.size() != NCOUNTERS ) {
    ATH_MSG_ERROR( "Counts vectors sizes are invalid" );
    return StatusCode::FAILURE;
  }

  for( size_t i=0; i<NCOUNTERS; i++ ) {
    if( not m_doTrigger and i == INROI ) continue;

    ATH_CHECK( fill( m_nTracks[i], trackCounts[i], weight ) );

    ATH_CHECK( fill( m_nTracks_vs_nVertices[i], vertexCounts[i], trackCounts[i], weight ) );

    ATH_CHECK( fill( m_nTracks_vs_actualMu[i], actualMu, trackCounts[i], weight ) );
    if( m_doTruthMuPlots ) ATH_CHECK( fill( m_nTracks_vs_truthMu[i], truthMu, trackCounts[i], weight ) );

    ATH_CHECK( fill( m_avg_nTracks_vs_actualMu[i], actualMu, trackCounts[i], weight ) );
    if( m_doTruthMuPlots ) ATH_CHECK( fill( m_avg_nTracks_vs_truthMu[i], truthMu, trackCounts[i], weight ) );
  }
 
  return StatusCode::SUCCESS;
}
 

void IDTPM::NtracksPlots::finalizePlots()
{
  ATH_MSG_DEBUG( "Finalising track multiplicity plots" );
}