PartitionSwitchTool.cpp

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2024 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
// Framework include files
#include <GaudiKernel/AlgTool.h>
#include <GaudiKernel/IPartitionControl.h>
#include <GaudiKernel/MsgStream.h>
#include <GaudiKernel/SmartIF.h>

class PartitionSwitchTool : public extends<AlgTool, IPartitionControl> {
 
  using CSTR   = const std::string&;
  using STATUS = StatusCode;
 
private:
  Gaudi::Property<std::string> m_actorName{ this, "Actor", "EventDataService", "option to set the multi-service name" };
  SmartIF<IPartitionControl> m_actor;
 
public:
  using extends::extends;

  STATUS initialize() override {
    STATUS sc = AlgTool::initialize();
    if ( !sc.isSuccess() ) {
      error() << "Cannot initialize base class!" << endmsg;
      return sc;
    }
    m_actor = service<IPartitionControl>( m_actorName );
    if ( !m_actor ) {
      error() << "Cannot retrieve partition controller \"" << m_actorName.value() << "\"!" << endmsg;
      return StatusCode::FAILURE;
    }
    return sc;
  }

  STATUS finalize() override {
    m_actor = nullptr;
    return AlgTool::finalize();
  }
 
  void _check( STATUS sc, CSTR msg ) const { error() << msg << " Status=" << sc.getCode() << endmsg; }
#define CHECK( x, y )                                                                                                  \
  if ( !x.isSuccess() ) _check( x, y );                                                                                \
  return x;

  STATUS create( CSTR nam, CSTR typ ) override {
    STATUS sc = m_actor ? m_actor->create( nam, typ ) : IInterface::Status::NO_INTERFACE;
    CHECK( sc, "Cannot create partition: " + nam + " of type " + typ );
  }

  STATUS create( CSTR nam, CSTR typ, IInterface*& pPartition ) override {
    STATUS sc = m_actor ? m_actor->create( nam, typ, pPartition ) : IInterface::Status::NO_INTERFACE;
    CHECK( sc, "Cannot create partition: " + nam + " of type " + typ );
  }

  STATUS drop( CSTR nam ) override {
    STATUS sc = m_actor ? m_actor->drop( nam ) : IInterface::Status::NO_INTERFACE;
    CHECK( sc, "Cannot drop partition: " + nam );
  }

  STATUS drop( IInterface* pPartition ) override {
    STATUS sc = m_actor ? m_actor->drop( pPartition ) : IInterface::Status::NO_INTERFACE;
    CHECK( sc, "Cannot drop partition by Interface." );
  }

  STATUS activate( CSTR nam ) override {
    STATUS sc = m_actor ? m_actor->activate( nam ) : IInterface::Status::NO_INTERFACE;
    CHECK( sc, "Cannot activate partition: " + nam );
  }

  STATUS activate( IInterface* pPartition ) override {
    STATUS sc = m_actor ? m_actor->activate( pPartition ) : IInterface::Status::NO_INTERFACE;
    CHECK( sc, "Cannot activate partition by Interface." );
  }

  STATUS get( CSTR nam, IInterface*& pPartition ) const override {
    STATUS sc = m_actor ? m_actor->get( nam, pPartition ) : IInterface::Status::NO_INTERFACE;
    CHECK( sc, "Cannot get partition " + nam );
  }

  STATUS activePartition( std::string& nam, IInterface*& pPartition ) const override {
    STATUS sc = m_actor ? m_actor->activePartition( nam, pPartition ) : IInterface::Status::NO_INTERFACE;
    CHECK( sc, "Cannot determine active partition." );
  }
};
 
// Declaration of the Tool Factory
DECLARE_COMPONENT( PartitionSwitchTool )