RCUSvc.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#include "RCUSvc.h"
 
 
namespace Athena {
 
 
RCUSvc::RCUSvc (const std::string& name, ISvcLocator* svc)
  : extends<AthService, IRCUSvc, IIncidentListener> (name, svc),
  m_hiveWhiteBoardSvc ("EventDataSvc", name),
  m_incidentSvc ("IncidentSvc", name)
{
  declareProperty ("HiveWhiteBoardSvc", m_hiveWhiteBoardSvc);
  declareProperty ("IncidentSvc", m_incidentSvc);
}
 
 
StatusCode RCUSvc::initialize()
{
  ATH_CHECK( AthService::initialize() );
  ATH_CHECK( m_hiveWhiteBoardSvc.retrieve() );
  ATH_CHECK( m_incidentSvc.retrieve() );
 
  m_incidentSvc->addListener (this, IncidentType::EndEvent);
  return StatusCode::SUCCESS;
}
 
 
size_t RCUSvc::getNumSlots() const
{
  // In non-MT mode, `EventDataSvc' does not have an IHiveWhiteBoard interface.
  // Fall back to one slot in that case.
  const SmartIF<IHiveWhiteBoard> wb (m_hiveWhiteBoardSvc.get());
  if (wb) {
    return wb->getNumberOfStores();
  }
  return 1;
}
 
 
void RCUSvc::add (IRCUObject* obj)
{
  lock_t g (m_mutex);
  m_objs.insert (obj);
}
 
 
StatusCode RCUSvc::remove (IRCUObject* obj)
{
  lock_t g (m_mutex);
  set_t::iterator it = m_objs.find (obj);
  if (it == m_objs.end())
    return StatusCode::FAILURE;
  m_objs.erase (it);
  return StatusCode::SUCCESS;
}
 
 
void RCUSvc::handle (const Incident& inc)
{
  if (inc.type() == IncidentType::EndEvent) {
    lock_t g (m_mutex);
    // Be careful --- calling quiescent() below may lead to objects being
    // removed from the set.
    std::vector<IRCUObject*> objs (m_objs.begin(), m_objs.end());
    for (IRCUObject* p : objs) {
      if (m_objs.find(p) != m_objs.end()) {
        p->quiescent (inc.context());
      }
    }
  }
}
 
 
} // namespace Athena