StateHandler.h

Go to the documentation of this file.

//
// Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
//
// Dear emacs, this is -*- c++ -*-
//
 
#ifndef ATHXRTINTERFACES_STATEHANDLER_H
#define ATHXRTINTERFACES_STATEHANDLER_H
 
#include "GaudiKernel/IIncidentListener.h"
#include "GaudiKernel/IIncidentSvc.h"
#include "GaudiKernel/ConcurrencyFlags.h"
#include "GaudiKernel/ServiceHandle.h"
 
#include "AthenaBaseComps/AthCheckMacros.h"
#include "AthenaInterprocess/Incidents.h"
 
 
namespace AthXRT {
class StateHandler : virtual public ::IIncidentListener
{
 protected: 
 
  virtual StatusCode initialize_global()
  {
    return StatusCode::SUCCESS;
  }
 
  virtual StatusCode initialize_worker()
  {
    return StatusCode::SUCCESS;
  }
 
  virtual StatusCode stop_worker()
  {
    return StatusCode::SUCCESS;
  }
 
  virtual StatusCode initialize()
  {
    ATH_CHECK(this->initialize_global());
 
    const bool is_multiprocess = (Gaudi::Concurrency::ConcurrencyFlags::numProcs() > 0);
 
    if (is_multiprocess)
      {
        ServiceHandle<IIncidentSvc> incidentSvc("IncidentSvc","StateHandler");
        incidentSvc->addListener(this, AthenaInterprocess::UpdateAfterFork::type());
      }
    else
      {
        ATH_CHECK(this->initialize_worker());
      }
 
    return StatusCode::SUCCESS;
  }
 
  virtual StatusCode stop() {
    const bool is_multiprocess = (Gaudi::Concurrency::ConcurrencyFlags::numProcs() > 0);
 
    if (!is_multiprocess) {
      ATH_CHECK(this->stop_worker());
    } else {
      if (m_is_child){
              ATH_CHECK(this->stop_worker());
      }
    }
 
  return StatusCode::SUCCESS;
  }
 
  bool m_is_child = false;
 
 public:
 
  void handle(const Incident & incident) override
  {
    const bool is_multiprocess = (Gaudi::Concurrency::ConcurrencyFlags::numProcs() > 0);
    if (is_multiprocess && incident.type() == AthenaInterprocess::UpdateAfterFork::type())
      {
    m_is_child = true;
        if (!this->initialize_worker().isSuccess())
        {
          throw GaudiException("Failed to initialize the XRT setup!",
                               "StateHandler::handle",
                               StatusCode::FAILURE);
        }
      }
  }
  
  virtual ~StateHandler() = default;
 
};
 
}  // namespace AthXRT
 
#endif //ATHXRTCORE_STATEHANDLER_H