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 doGlobalPlots ) :
        PlotMgr( dirName, anaTag, pParent ), 
        m_trackType( trackType ),
        m_doTrigger( doTrigger ),
        m_doGlobalPlots( doGlobalPlots ) { }
 
 
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( unsigned int 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] ) );
  }
 
  if( m_doGlobalPlots ) {
    ATH_CHECK( retrieveAndBook( m_nTracks_vs_truthMu,  "num_"+m_trackType+"_"+m_counterName[ FS ]+"_vs_truthMu" ) );
    ATH_CHECK( retrieveAndBook( m_nTracks_vs_actualMu, "num_"+m_trackType+"_"+m_counterName[ FS ]+"_vs_actualMu" ) );
  }
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode IDTPM::NtracksPlots::fillPlots(
    const std::vector< unsigned int >& counts,
    float truthMu,
    float actualMu,
    float weight )
{
  if( counts.size() != NCOUNTERS ) {
    ATH_MSG_ERROR( "Counts vecotor size is invalid" );
    return StatusCode::FAILURE;
  }
 
  for( unsigned int i=0; i<NCOUNTERS; i++ ) {
    if( not m_doTrigger and i == INROI ) continue;
    ATH_CHECK( fill( m_nTracks[i], counts[i], weight ) );
  }
 
  if( m_doGlobalPlots ) {
    ATH_CHECK( fill( m_nTracks_vs_truthMu,   truthMu,  counts[ INROI ], weight ) );
    ATH_CHECK( fill( m_nTracks_vs_actualMu,  actualMu, counts[ INROI ], weight ) );
  }
 
  return StatusCode::SUCCESS;
}
 
 
void IDTPM::NtracksPlots::finalizePlots()
{
  ATH_MSG_DEBUG( "Finalising track multiplicity plots" );
}