AsgMetadataTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// System include(s):
#include <stdexcept>
 
// Local include(s):
#include "AsgTools/AsgMetadataTool.h"
 
#ifndef XAOD_STANDALONE
// Gaudi/Athena include(s):
#   include "GaudiKernel/Incident.h"
#   include "GaudiKernel/IIncidentSvc.h"
#endif // not XAOD_STANDALONE
 
#ifdef XAOD_STANDALONE
// xAOD include(s):
#   include "xAODRootAccessInterfaces/TActiveEvent.h"
#   include "xAODRootAccess/TEvent.h"
#endif // XAOD_STANDALONE
 
namespace asg {
 
#ifdef XAOD_STANDALONE
   // To be able to refer to xAOD::IncidentType as IncidentType...
   using namespace xAOD;
#endif // XAOD_STANDALONE
 
   AsgMetadataTool::AsgMetadataTool( const std::string& name )
      : AsgTool( name ),
#ifdef XAOD_STANDALONE
        m_inputMetaStore( SgTEventMeta::InputStore ),
        m_outputMetaStore( SgTEventMeta::OutputStore ),
#else // XAOD_STANDALONE
        m_inputMetaStore( "StoreGateSvc/InputMetaDataStore", name ),
        m_outputMetaStore( "StoreGateSvc/MetaDataStore", name ),
#endif // XAOD_STANDALONE
        m_beginInputFileCalled( false ),
        m_useIncidents (true)
   {
 
#ifdef XAOD_STANDALONE
      // Try to access the current active TEvent:
      xAOD::TVirtualEvent* vevent = xAOD::TActiveEvent::event();
      xAOD::TEvent* event = dynamic_cast< xAOD::TEvent* >( vevent );
      if( ! event ) {
         ATH_MSG_WARNING( "Couldn't find active xAOD::TEvent object" );
         ATH_MSG_WARNING( "Callbacks to the tool will not be available" );
         return;
      }
      // Register the tool for callbacks:
      if( event->addListener( this ).isFailure() ) {
         ATH_MSG_ERROR( "Couldn't register the tool for xAOD callbacks" );
      }
#else // XAOD_STANDALONE
      // Declare the interface implemented by this base class:
      declareInterface< IIncidentListener >( this );
#endif // XAOD_STANDALONE
   }
 
   AsgMetadataTool::~AsgMetadataTool() {
 
#ifdef XAOD_STANDALONE
      // Try to access the active TEvent:
      xAOD::TVirtualEvent* vevent = xAOD::TActiveEvent::event();
      xAOD::TEvent* event = dynamic_cast< xAOD::TEvent* >( vevent );
      // If we didn't succeed, fine. No need to complain about it...
      if( event ) {
         // But if there is a TEvent, then let's try to de-register from it
         // properly.
         if( event->removeListener( this ).isFailure() ) {
            ATH_MSG_WARNING( "Active xAOD::TEvent didn't have a callback for "
                             "this tool" );
         }
      }
#else
      //if initialized, then we should remove listeners
      if(FSMState() == Gaudi::StateMachine::INITIALIZED) {
    ServiceHandle< IIncidentSvc > incSvc( "IncidentSvc", name() );
        if( incSvc.retrieve().isSuccess() ) {
      incSvc->removeListener( this ); //removes entirely
    }
      }
#endif // XAOD_STANDALONE
   }
 
   AsgMetadataTool::MetaStorePtr_t AsgMetadataTool::inputMetaStore() const {
 
#ifdef XAOD_STANDALONE
      return &m_inputMetaStore;
#else // XAOD_STANDALONE
      return m_inputMetaStore;
#endif // XAOD_STANDALONE
   }
 
   AsgMetadataTool::MetaStorePtr_t AsgMetadataTool::outputMetaStore() const {
 
#ifdef XAOD_STANDALONE
      return &m_outputMetaStore;
#else // XAOD_STANDALONE
      return m_outputMetaStore;
#endif // XAOD_STANDALONE
   }
 
   StatusCode AsgMetadataTool::sysInitialize() {
 
#ifndef XAOD_STANDALONE
      if (m_useIncidents) {
         // Connect to the IncidentSvc:
         ServiceHandle< IIncidentSvc > incSvc( "IncidentSvc", name() );
         ATH_CHECK( incSvc.retrieve() );
 
         // Set up the right callbacks: don't rethrow exceptions, any failure and we should end
         incSvc->addListener( this, IncidentType::BeginEvent, 0, false );
      }
      // Let the base class do its thing:
      ATH_CHECK( AlgTool::sysInitialize() );
 
#endif // not XAOD_STANDALONE
 
      // Return gracefully:
      return StatusCode::SUCCESS;
   }
 
   void AsgMetadataTool::handle( const Incident& inc ) {
 
      // Tell the user what's happening:
      ATH_MSG_VERBOSE( "Callback received with incident: " << inc.type() );
 
      // Call the appropriate member function:
      if( inc.type() == IncidentType::BeginInputFile ) {
        m_beginInputFileCalled = true;
        if( beginInputFile().isFailure() ) {
          ATH_MSG_FATAL( "Failed to call beginInputFile()" );
          throw std::runtime_error( "Couldn't call beginInputFile()" );
         }
      } else if( inc.type() == IncidentType::EndInputFile ) {
        if( endInputFile().isFailure() ) {
          ATH_MSG_FATAL( "Failed to call endInputFile()" );
          throw std::runtime_error( "Couldn't call endInputFile()" );
         }
      } else if( inc.type() == IncidentType::BeginEvent ) {
         // If the tool didn't catch the begin input file incident for the
         // first input file of the job, then call the appropriate function
         // now.
         if( ! m_beginInputFileCalled ) {
            m_beginInputFileCalled = true;
            if( beginInputFile().isFailure() ) {
               ATH_MSG_FATAL( "Failed to call beginInputFile()" );
               throw std::runtime_error( "Couldn't call beginInputFile()" );
            }
         }
         if( beginEvent().isFailure() ) {
            ATH_MSG_FATAL( "Failed to call beginEvent()" );
            throw std::runtime_error( "Couldn't call beginEvent()" );
         }
 
     #ifdef XAOD_STANDALONE
      } else if( inc.type() == IncidentType::MetaDataStop ) {
         if( metaDataStop().isFailure() ) {
            ATH_MSG_FATAL( "Failed to call metaDataStop()" );
            throw std::runtime_error( "Couldn't call metaDataStop()" );
         }
         
     #endif // XAOD_STANDALONE
      } else {
         ATH_MSG_WARNING( "Unknown incident type received in AsgMetaDataTool: " << inc.type() );
      }
 
      return;
   }
 
   StatusCode AsgMetadataTool::beginInputFile() {
 
      // Return gracefully:
      return StatusCode::SUCCESS;
   }
 
   StatusCode AsgMetadataTool::endInputFile() {
 
      // Return gracefully:
      return StatusCode::SUCCESS;
   }
 
   StatusCode AsgMetadataTool::beginEvent() {
 
      // Return gracefully:
      return StatusCode::SUCCESS;
   }
 
   StatusCode AsgMetadataTool::metaDataStop() {
 
      // Return gracefully:
      return StatusCode::SUCCESS;
   }
 
} // namespace asg