TrackAnalysisDefinitionSvc.cxx

Go to the documentation of this file.

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


#include "TrackAnalysisDefinitionSvc.h"

#include <algorithm>
#include <memory>
#include <unordered_map>

#include "InDetPhysValMonitoring/ResolutionHelper.h"
 

StatusCode IDTPM::TrackAnalysisDefinitionSvc::initialize()
{
 
  ATH_MSG_DEBUG( "Initialising  using TEST = " << m_testTypeStr.value() <<
                 " and REFERENCE = " << m_refTypeStr.value() );

  m_isTestTrigger = m_testTypeStr.value().find("Trigger") != std::string::npos;
  m_isTestEFTrigger = m_testTypeStr.value().find("EFTrigger") != std::string::npos;
  m_isTestTruth   = m_testTypeStr.value().find("Truth") != std::string::npos;
  m_isTestOffline = m_testTypeStr.value().find("Offline") != std::string::npos;
 
  m_isRefTrigger = m_refTypeStr.value().find("Trigger") != std::string::npos;
  m_isRefEFTrigger = m_refTypeStr.value().find("EFTrigger") != std::string::npos;
  m_isRefTruth   = m_refTypeStr.value().find("Truth") != std::string::npos;
  m_isRefOffline = m_refTypeStr.value().find("Offline") != std::string::npos;
 
  m_useTrigger = m_isTestTrigger or m_isRefTrigger;
  m_useEFTrigger = m_isTestEFTrigger or m_isRefEFTrigger;
  m_useTruth   = m_isTestTruth or m_isRefTruth or m_matchingType.value().find("EFTruthMatch") != std::string::npos;;
  ATH_MSG_DEBUG( "USE TRUTH? " << m_useTruth );
  m_useOffline = m_isTestOffline or m_isRefOffline;

  if( m_doTrigNavigation ) {
    for( size_t ic=0 ; ic<m_chainNames.size() ; ic++ ) {
      ATH_MSG_DEBUG( "Input chain : " << m_chainNames[ic] );
      m_configuredChains.push_back( m_chainNames[ic] );
    }
  } else {
    m_configuredChains.push_back( "Offline" );
  }

  std::sort( m_configuredChains.begin(), m_configuredChains.end() );
  m_configuredChains.erase( std::unique( m_configuredChains.begin(), 
                                         m_configuredChains.end() ), 
                            m_configuredChains.end() );
 
  return StatusCode::SUCCESS;
}

StatusCode IDTPM::TrackAnalysisDefinitionSvc::finalize() {
  return StatusCode::SUCCESS;
}

std::string IDTPM::TrackAnalysisDefinitionSvc::plotsFullDir( std::string chain ) const
{
  std::string topDir( m_dirName );
  if( not topDir.empty() ) topDir += "/";

   if( not chain.empty() ) chain += "/";

  std::string subDir( m_subFolder );
  if( subDir.empty() ) {
    ATH_MSG_WARNING( "Empty plots sub-directory" );
  } else  {
    if( subDir[0] == '/' ) {
      subDir.erase( subDir.begin() );
    }
    if( subDir.back() != '/' ) subDir += "/";
  }
 
  return m_sortPlotsByChain.value() ?
         topDir + chain + subDir :
         topDir + subDir + chain;
}

unsigned int IDTPM::TrackAnalysisDefinitionSvc::resolutionMethod() const
{
  using methodMap_t = std::unordered_map<
      std::string, IDPVM::ResolutionHelper::methods >;
  methodMap_t methodMap = {
    { "iterRMS"         , IDPVM::ResolutionHelper::iterRMS_convergence },
    { "gaussFit"        , IDPVM::ResolutionHelper::Gauss_fit },
    { "iterRMSgaussFit" , IDPVM::ResolutionHelper::fusion_iterRMS_Gaussfit },
    { "iterGaussFit"    , IDPVM::ResolutionHelper::iterGaussFit_convergence }
  };
 
  methodMap_t::const_iterator mitr = methodMap.find( m_resolMethod.value() );
  if( mitr == methodMap.end() ) {
    ATH_MSG_DEBUG( "Method " << m_resolMethod.value() <<
                   " not found. Using iterRMS by default." );
    return IDPVM::ResolutionHelper::iterRMS_convergence;
  }
  return mitr->second;
}