MinimalEventLoopMgr.cpp

Go to the documentation of this file.

// $Id: MinimalEventLoopMgr.cpp,v 1.7 2008/07/15 12:48:18 marcocle Exp $
#define  GAUDIKERNEL_MINIMALEVENTLOOPMGR_CPP

#include "GaudiKernel/SmartIF.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/IAlgorithm.h"
#include "GaudiKernel/IAlgManager.h"
#include "GaudiKernel/ListItem.h"
#include "GaudiKernel/GaudiException.h"
#include "GaudiKernel/ThreadGaudi.h"
#include "GaudiKernel/Incident.h"
#include "GaudiKernel/IIncidentListener.h"
#include "GaudiKernel/IIncidentSvc.h"

#include "GaudiKernel/MinimalEventLoopMgr.h"

namespace {
  class AbortEventListener: public IIncidentListener {
  public:
    AbortEventListener(bool &flag, std::string &source):m_flag(flag),
                                                        m_source(source),
                                                        m_refCount(1) {}
    virtual ~AbortEventListener() {}
    virtual void handle(const Incident& i) {
      if (i.type() == IncidentType::AbortEvent) {
        m_flag = true;
        m_source = i.source();
      }
    }
    
    // --- These are needed because we do not inherit from any actual class.
    virtual StatusCode queryInterface(const InterfaceID& riid,
                                      void** ppvInterface){
      if ( 0 == ppvInterface ) {
        return StatusCode::FAILURE;
      }
      if ( IIncidentListener::interfaceID().versionMatch(riid) ) {
        *ppvInterface = static_cast<IIncidentListener*>( this );
      }
      else if ( IInterface::interfaceID().versionMatch(riid) ) {
        *ppvInterface = static_cast<IInterface*>( this );
      }
      else {
        *ppvInterface = 0;
        return NO_INTERFACE;
      } 
      // increment reference counter 
      addRef();
      return SUCCESS;
    }
    
    virtual unsigned long addRef() {return ++m_refCount;}

    virtual unsigned long release() {
      unsigned long count = --m_refCount;
      if( count == 0 ) {
        delete this;
      }
      return count;
    }

  private:
    bool &m_flag;
    std::string &m_source;
    unsigned long m_refCount;
  };
}

//--------------------------------------------------------------------------------------------
// Standard Constructor
//--------------------------------------------------------------------------------------------
MinimalEventLoopMgr::MinimalEventLoopMgr(const std::string& nam, ISvcLocator* svcLoc)
: Service(nam, svcLoc)
{
  declareProperty("TopAlg",         m_topAlgNames );
  declareProperty("OutStream",      m_outStreamNames );
  declareProperty("OutStreamType",  m_outStreamType = "OutputStream");
  m_topAlgNames.declareUpdateHandler   ( &MinimalEventLoopMgr::topAlgHandler, this );
  m_outStreamNames.declareUpdateHandler( &MinimalEventLoopMgr::outStreamHandler, this );
  svcLoc->queryInterface(IID_IAppMgrUI, pp_cast<void>(&m_appMgrUI)).ignore();
  m_incidentSvc = 0;
  m_state = OFFLINE;
  m_scheduledStop = false;
  m_abortEventListener = 0;
  m_abortEvent = false;
}

//--------------------------------------------------------------------------------------------
// Standard Destructor
//--------------------------------------------------------------------------------------------
MinimalEventLoopMgr::~MinimalEventLoopMgr()   {
  if( m_incidentSvc ) m_incidentSvc->release();
  if( m_appMgrUI ) m_appMgrUI->release();
  m_state = OFFLINE;
}

//--------------------------------------------------------------------------------------------
// implementation of IInterface: queryInterface
//--------------------------------------------------------------------------------------------
StatusCode MinimalEventLoopMgr::queryInterface(const InterfaceID& riid, void** ppvInterface)   {
  if ( riid == IID_IAppMgrUI )  {
    *ppvInterface = (IAppMgrUI*)this;
  }
  else if ( riid == IID_IEventProcessor )  {
    *ppvInterface = (IEventProcessor*)this;
  }
  else  {
    return Service::queryInterface(riid, ppvInterface);
  }
  addRef();
  return StatusCode::SUCCESS;
}


//--------------------------------------------------------------------------------------------
// implementation of IAppMgrUI::initalize
//--------------------------------------------------------------------------------------------
StatusCode MinimalEventLoopMgr::initialize()    {

  MsgStream log(msgSvc(), name());

  if ( 0 == m_appMgrUI ) {
    return StatusCode::FAILURE;
  }

  StatusCode sc = Service::initialize();
  if ( !sc.isSuccess() )   {
    log << MSG::ERROR << "Failed to initialize Service Base class." << endreq;
    return StatusCode::FAILURE;
  }
  
  SmartIF<IProperty> prpMgr(IID_IProperty, serviceLocator());
  if ( ! prpMgr.isValid() )   {
    log << MSG::ERROR << "Error retrieving AppMgr interface IProperty." << endreq;
    return StatusCode::FAILURE;
  }
  else {
    if ( m_topAlgNames.value().size() == 0 )    {
      setProperty(prpMgr->getProperty("TopAlg")).ignore();
    }
    if ( m_outStreamNames.value().size() == 0 )   {
      setProperty(prpMgr->getProperty("OutStream")).ignore();
    }
  }

  // Get the references to the services that are needed by the ApplicationMgr itself
  sc = serviceLocator()->service("IncidentSvc", m_incidentSvc, true);
  if( !sc.isSuccess() )  {
    log << MSG::FATAL << "Error retrieving IncidentSvc." << endreq;
    return sc;
  }
  
  m_abortEventListener = new AbortEventListener(m_abortEvent,m_abortEventSource);
  m_incidentSvc->addListener(m_abortEventListener,IncidentType::AbortEvent);
  
  // The state is changed at this moment to allow decodeXXXX() to do something
  m_state = INITIALIZED;

  //--------------------------------------------------------------------------------------------
  // Create output streams. Do not initialize them yet.
  // The state is updated temporarily in order to enable the handler, which
  // is also triggered by a change to the "OutputStream" Property.
  //--------------------------------------------------------------------------------------------
  sc = decodeOutStreams();
  if ( !sc.isSuccess() )    {
    log << MSG::ERROR << "Failed to initialize Output streams." << endreq;
    m_state = CONFIGURED;
    return sc;
  }
  //--------------------------------------------------------------------------------------------
  // Create all needed Top Algorithms. Do not initialize them yet.
  // The state is updated temporarily in order to enable the handler, which
  // is also triggered by a change to the "TopAlg" Property.
  //--------------------------------------------------------------------------------------------
  sc = decodeTopAlgs();
  if ( !sc.isSuccess() )    {
    log << MSG::ERROR << "Failed to initialize Top Algorithms streams." << endreq;
    m_state = CONFIGURED;
    return sc;
  }

  ListAlg::iterator ita;
  // Initialize all the new TopAlgs. In fact Algorithms are protected against getting
  // initialized twice.
  for (ita = m_topAlgList.begin(); ita != m_topAlgList.end(); ita++ ) {
    sc = (*ita)->sysInitialize();
    if( !sc.isSuccess() ) {
      log << MSG::ERROR << "Unable to initialize Algorithm: " << (*ita)->name() << endreq;
      return sc;
    }
  }
  for (ita = m_outStreamList.begin(); ita != m_outStreamList.end(); ita++ ) {
    sc = (*ita)->sysInitialize();
    if( !sc.isSuccess() ) {
      log << MSG::ERROR << "Unable to initialize Output Stream: " << (*ita)->name() << endreq;
      return sc;
    }
  }

  return StatusCode::SUCCESS;
}
//--------------------------------------------------------------------------------------------
// implementation of IAppMgrUI::start
//--------------------------------------------------------------------------------------------
StatusCode MinimalEventLoopMgr::start()    {

  StatusCode sc = Service::start();
  if ( ! sc.isSuccess() ) return sc;
  
  MsgStream log(msgSvc(), name());
  
  ListAlg::iterator ita;
  // Start all the new TopAlgs. In fact Algorithms are protected against getting
  // started twice.
  for (ita = m_topAlgList.begin(); ita != m_topAlgList.end(); ita++ ) {
    sc = (*ita)->sysStart();
    if( !sc.isSuccess() ) {
      log << MSG::ERROR << "Unable to initialize Algorithm: " << (*ita)->name() << endreq;
      return sc;
    }
  }
  for (ita = m_outStreamList.begin(); ita != m_outStreamList.end(); ita++ ) {
    sc = (*ita)->sysStart();
    if( !sc.isSuccess() ) {
      log << MSG::ERROR << "Unable to initialize Output Stream: " << (*ita)->name() << endreq;
      return sc;
    }
  }
  return StatusCode::SUCCESS;
}
//--------------------------------------------------------------------------------------------
// implementation of IAppMgrUI::stop
//--------------------------------------------------------------------------------------------
StatusCode MinimalEventLoopMgr::stop()    {

  StatusCode sc = StatusCode::SUCCESS;
  
  MsgStream log(msgSvc(), name());
  
  ListAlg::iterator ita;
  // Start all the new TopAlgs. In fact Algorithms are protected against getting
  // started twice.
  for (ita = m_topAlgList.begin(); ita != m_topAlgList.end(); ita++ ) {
    sc = (*ita)->sysStop();
    if( !sc.isSuccess() ) {
      log << MSG::ERROR << "Unable to initialize Algorithm: " << (*ita)->name() << endreq;
      return sc;
    }
  }
  for (ita = m_outStreamList.begin(); ita != m_outStreamList.end(); ita++ ) {
    sc = (*ita)->sysStop();
    if( !sc.isSuccess() ) {
      log << MSG::ERROR << "Unable to initialize Output Stream: " << (*ita)->name() << endreq;
      return sc;
    }
  }
  
  return Service::stop();
}

//--------------------------------------------------------------------------------------------
// implementation of IService::reinitialize
//--------------------------------------------------------------------------------------------
StatusCode MinimalEventLoopMgr::reinitialize() {
  MsgStream log( msgSvc(), name() );
  StatusCode sc = StatusCode::SUCCESS;
  ListAlg::iterator ita;

  // Renitialize all the TopAlgs.
  for (ita = m_topAlgList.begin(); ita != m_topAlgList.end(); ita++ ) {
    sc = (*ita)->sysReinitialize();
    if( !sc.isSuccess() ) {
      log << MSG::ERROR << "Unable to reinitialize Algorithm: " << (*ita)->name() << endreq;
      return sc;
    }
  }
  for (ita = m_outStreamList.begin(); ita != m_outStreamList.end(); ita++ ) {
    sc = (*ita)->sysReinitialize();
    if( !sc.isSuccess() ) {
      log << MSG::ERROR << "Unable to reinitialize Output Stream: " << (*ita)->name() << endreq;
      return sc;
    }
  }

  return sc;
}
//--------------------------------------------------------------------------------------------
// implementation of IService::restart
//--------------------------------------------------------------------------------------------
StatusCode MinimalEventLoopMgr::restart() {
  MsgStream log( msgSvc(), name() );
  StatusCode sc = StatusCode::SUCCESS;
  ListAlg::iterator ita;

  // Restart all the TopAlgs.
  for (ita = m_topAlgList.begin(); ita != m_topAlgList.end(); ita++ ) {
    sc = (*ita)->sysRestart();
    if( !sc.isSuccess() ) {
      log << MSG::ERROR << "Unable to restart Algorithm: " << (*ita)->name() << endreq;
      return sc;
    }
  }
  for (ita = m_outStreamList.begin(); ita != m_outStreamList.end(); ita++ ) {
    sc = (*ita)->sysRestart();
    if( !sc.isSuccess() ) {
      log << MSG::ERROR << "Unable to restart Output Stream: " << (*ita)->name() << endreq;
      return sc;
    }
  }

  return sc;
}

//--------------------------------------------------------------------------------------------
// implementation of IService::finalize
//--------------------------------------------------------------------------------------------
StatusCode MinimalEventLoopMgr::finalize()    {
  MsgStream log( msgSvc(), name() );
  StatusCode sc = StatusCode::SUCCESS;
  // Call the finalize() method of all top algorithms
  ListAlg::iterator ita;
  for ( ita = m_topAlgList.begin(); ita != m_topAlgList.end(); ita++ ) {
    sc = (*ita)->sysFinalize();
    if( !sc.isSuccess() ) {
      log << MSG::WARNING << "Finalization of algorithm " << (*ita)->name() << " failed" << endreq;
    }
  }
  // Call the finalize() method of all Output streams
  for ( ita = m_outStreamList.begin(); ita != m_outStreamList.end(); ita++ ) {
    sc = (*ita)->sysFinalize();
    if( !sc.isSuccess() ) {
      log << MSG::WARNING << "Finalization of algorithm " << (*ita)->name() << " failed" << endreq;
    }
  }
  // release all top algorithms
  SmartIF<IAlgManager> algMan(IID_IAlgManager, serviceLocator());
  for ( ita = m_topAlgList.begin(); ita != m_topAlgList.end(); ita++ ) {
    if (algMan->removeAlgorithm(*ita).isFailure()) {
      log << MSG::ERROR << "Problems removing Algorithm " << (*ita)->name()
          << endreq;
    }
    (*ita)->release();
  }
  m_topAlgList.erase(m_topAlgList.begin(), m_topAlgList.end() );

  // release all output streams
  for ( ita = m_outStreamList.begin(); ita != m_outStreamList.end(); ita++ ) {
    (*ita)->release();
  }
  m_outStreamList.erase(m_outStreamList.begin(), m_outStreamList.end() );
  if ( sc.isSuccess() ) m_state = FINALIZED;
  
  m_incidentSvc->removeListener(m_abortEventListener,IncidentType::AbortEvent);
  m_abortEventListener = releaseInterface(m_abortEventListener);
  
  m_incidentSvc = releaseInterface(m_incidentSvc);
  m_appMgrUI = releaseInterface(m_appMgrUI);
  
  return Service::finalize();
}

//--------------------------------------------------------------------------------------------
// implementation of IAppMgrUI::nextEvent
//--------------------------------------------------------------------------------------------
StatusCode MinimalEventLoopMgr::nextEvent(int /* maxevt */)   {
  MsgStream log(msgSvc(), name());
  log << MSG::ERROR << "This method cannot be called on an object of type " 
      << System::typeinfoName(typeid(*this)) << endreq;
  return StatusCode::FAILURE;
}

//--------------------------------------------------------------------------------------------
// IEventProcessing::executeRun
//--------------------------------------------------------------------------------------------
StatusCode MinimalEventLoopMgr::executeRun( int maxevt ) {
  StatusCode  sc;
  MsgStream log( msgSvc(), name() );
  ListAlg::iterator ita;
  bool eventfailed = false;

  // Call the beginRun() method of all top algorithms
  for (ita = m_topAlgList.begin(); ita != m_topAlgList.end(); ita++ ) {
    sc = (*ita)->sysBeginRun();
    if( !sc.isSuccess() ) {
      log << MSG::WARNING << "beginRun() of algorithm " << (*ita)->name() << " failed" << endreq;
      eventfailed = true;
    }
  }

  // Call now the nextEvent(...)
  sc = nextEvent(maxevt);
  if( !sc.isSuccess() ) {
    eventfailed = true;
  }

  // Call the endRun() method of all top algorithms
  for (ita = m_topAlgList.begin(); ita != m_topAlgList.end(); ita++ ) {
    sc = (*ita)->sysEndRun();
    if( !sc.isSuccess() ) {
      log << MSG::WARNING << "endRun() of algorithm " << (*ita)->name() << " failed" << endreq;
      eventfailed = true;
    }
  }

  if( eventfailed ){
    return StatusCode::FAILURE;
  }
  else {
    return StatusCode::SUCCESS;
  }
}

//--------------------------------------------------------------------------------------------
// implemenation of IEventProcessor::executeEvent(void* par)
//--------------------------------------------------------------------------------------------
StatusCode MinimalEventLoopMgr::executeEvent(void* /* par */)    {
  bool eventfailed = false;
  
  // Call the resetExecuted() method of ALL "known" algorithms
  // (before we were reseting only the topalgs)
  SmartIF<IAlgManager> algMan(IID_IAlgManager, serviceLocator());
  if ( algMan.isValid() ) {
    const ListAlg& allAlgs = algMan->getAlgorithms() ;
    for( ListAlg::const_iterator ialg = allAlgs.begin() ; allAlgs.end() != ialg ; ++ialg ) { 
      if ( 0 != *ialg ) (*ialg)->resetExecuted();
    }
  } 

  // Call the execute() method of all top algorithms
  for (ListAlg::const_iterator ita = m_topAlgList.begin(); ita != m_topAlgList.end(); ita++ ) {
    StatusCode sc(StatusCode::FAILURE);
    try {
      if (m_abortEvent){
        MsgStream log ( msgSvc() , name() );
        log << MSG::DEBUG << "AbortEvent incident fired by "
                          << m_abortEventSource << endreq;
        m_abortEvent = false;
        sc.ignore();
        break;
      }
      sc = (*ita)->sysExecute();
    } catch ( const GaudiException& Exception ) {
      MsgStream log ( msgSvc() , "MinimalEventLoopMgr.executeEvent()" );
      log << MSG::FATAL << " Exception with tag=" << Exception.tag()
          << " thrown by " << (*ita)->name() << endreq;
      log << MSG::ERROR << Exception << endreq;
    } catch ( const std::exception& Exception ) {
      MsgStream log ( msgSvc() , "MinimalEventLoopMgr.executeEvent()" );
      log << MSG::FATAL << " Standard std::exception thrown by " 
          << (*ita)->name() << endreq;
      log << MSG::ERROR << Exception.what()  << endreq;      
    } catch(...) {
      MsgStream log ( msgSvc() , "MinimalEventLoopMgr.executeEvent()" );
      log << MSG::FATAL << "UNKNOWN Exception thrown by " 
          << (*ita)->name() << endreq;
    }

    if( !sc.isSuccess() )  {
      MsgStream log( msgSvc(), name() );
      log << MSG::WARNING << "Execution of algorithm " << (*ita)->name() << " failed" << endreq;
      eventfailed = true;
    }
  }
  
  // Call the execute() method of all output streams
  for (ListAlg::const_iterator ito = m_outStreamList.begin(); ito != m_outStreamList.end(); ito++ ) {
    (*ito)->resetExecuted();
      StatusCode sc;
      sc = (*ito)->sysExecute();
    if( !sc.isSuccess() )  {
      MsgStream log( msgSvc(), name() );
      log << MSG::WARNING << "Execution of output stream " << (*ito)->name() << " failed" << endreq;
      eventfailed = true;
    }
  }

  // Check if there was an error processing current event
  if( eventfailed ){
    MsgStream log( msgSvc(), name() );
    log << MSG::ERROR << "Error processing event loop." << endreq;
    return StatusCode(StatusCode::FAILURE,true);
  }
  return StatusCode(StatusCode::SUCCESS,true);
}
//--------------------------------------------------------------------------------------------
// implemenation of IEventProcessor::stopRun()
//--------------------------------------------------------------------------------------------
StatusCode MinimalEventLoopMgr::stopRun() {
  m_scheduledStop = true;
  return StatusCode::SUCCESS;
}

//--------------------------------------------------------------------------------------------
// Top algorithm List handler
//--------------------------------------------------------------------------------------------
void MinimalEventLoopMgr::topAlgHandler( Property& /* theProp */ )  {
  if ( !(decodeTopAlgs( )).isSuccess() ) {
    throw GaudiException("Failed to initialize Top Algorithms streams.",
                         "MinimalEventLoopMgr::topAlgHandler", 
                         StatusCode::FAILURE);
  }
}

//--------------------------------------------------------------------------------------------
// decodeTopAlgNameList & topAlgNameListHandler
//--------------------------------------------------------------------------------------------
StatusCode MinimalEventLoopMgr::decodeTopAlgs()    {
  StatusCode sc = StatusCode::SUCCESS;
  if ( CONFIGURED == m_state || INITIALIZED == m_state ) {
    SmartIF<IAlgManager> algMan(IID_IAlgManager, serviceLocator());
    MsgStream log(msgSvc(), name());
    if ( algMan.isValid() )   {
      // Reset the existing Top Algorithm List
      m_topAlgList.clear( );
      const std::vector<std::string>& algNames = m_topAlgNames.value( );
      for (VectorName::const_iterator it = algNames.begin(); it != algNames.end(); it++) {
        ListItem item(*it);
        // Got the type and name. Now creating the algorithm, avoiding duplicate creation.
        IAlgorithm* ialg;
        std::string item_name = item.name() + getGaudiThreadIDfromName(name()) ;
        sc = algMan->getAlgorithm( item_name, ialg );
        if ( sc.isSuccess( ) ) {
          log << MSG::DEBUG << "Top Algorithm " << item_name << " already exists" << endreq;
          ialg->addRef();  // increment reference count
        } 
        else {
          log << MSG::DEBUG << "Creating Top Algorithm " << item.type() << " with name " << item_name << endreq;
          sc = algMan->createAlgorithm( item.type(), item_name, ialg );
          if( !sc.isSuccess() ) {
            log << MSG::ERROR << "Unable to create Top Algorithm " << item.type() << " with name " << item_name << endreq;
            return sc;
          }
        }
        m_topAlgList.push_back(ialg);
      }
      return sc;
    }
    sc = StatusCode::FAILURE;
  }
  return sc;
}

//--------------------------------------------------------------------------------------------
// Output stream List handler
//--------------------------------------------------------------------------------------------
void MinimalEventLoopMgr::outStreamHandler( Property& /* theProp */ )    {
  if ( !(decodeOutStreams( )).isSuccess() ) {
    throw GaudiException("Failed to initialize output streams.",
                         "MinimalEventLoopMgr::outStreamHandler", 
                         StatusCode::FAILURE);
  }
}

//--------------------------------------------------------------------------------------------
// decodeOutStreamNameList & outStreamNameListHandler
//--------------------------------------------------------------------------------------------
StatusCode MinimalEventLoopMgr::decodeOutStreams( )    {
  StatusCode sc = StatusCode::SUCCESS;
  if ( CONFIGURED == m_state || INITIALIZED == m_state ) {
    MsgStream log(msgSvc(), name());
    SmartIF<IAlgManager> algMan(IID_IAlgManager, serviceLocator());
    if ( algMan.isValid() )   {
      // Reset the existing Top Algorithm List
      m_outStreamList.clear();
      const std::vector<std::string>& algNames = m_outStreamNames.value( );
      for (VectorName::const_iterator it = algNames.begin(); it != algNames.end(); it++) {
        ListItem item(*it, m_outStreamType);
        IAlgorithm* ios;
        log << MSG::DEBUG << "Creating " << m_outStreamType <<  (*it) << endreq;
        sc = algMan->getAlgorithm( item.name(), ios );
        if ( sc.isSuccess( ) ) {
          log << MSG::DEBUG << "Output Stream " << item.name() << " already exists" << endreq;
        } 
        else {
          log << MSG::DEBUG << "Creating Output Stream " << (*it) << endreq;
          sc = algMan->createAlgorithm( item.type(), item.name(), ios );
          if( !sc.isSuccess() ) {
            log << MSG::ERROR << "Unable to create Output Stream " << (*it) << endreq;
            return sc;
          }
        }
        m_outStreamList.push_back( ios );
      }
     return sc;
    }
    sc = StatusCode::FAILURE;
  }
  return sc;
}