SGHiveMgrSvc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef STOREGATE_HIVEMGRSVC_H
#define STOREGATE_HIVEMGRSVC_H
 
#include <vector>
#include <string>
#include <mutex>
 
#include "GaudiKernel/Service.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/StatusCode.h"
#include "GaudiKernel/IHiveWhiteBoard.h"
 
#include "StoreGate/StoreGateSvc.h"
#include "StoreGate/SGHiveEventSlot.h"
 
class StoreGateSvc;
class ISvcLocator;
class HltEventLoopMgr;       // friend
 
namespace SG {
class HiveMgrSvc : public extends<Service, IHiveWhiteBoard> {
  friend class TestSGHiveMgrSvc;
  friend class ::HltEventLoopMgr;
public:
 
  virtual StatusCode selectStore(size_t slotIndex) override;
 
  virtual StatusCode clearStore(size_t slotIndex) override;
  
  virtual StatusCode setNumberOfStores(size_t slots) override;
 
  virtual size_t getNumberOfStores() const override;
 
    virtual bool exists( const DataObjID& ) override;
  
  virtual size_t allocateStore( int evtnumber ) override;
  
  virtual StatusCode freeStore( size_t slotIndex ) override;
  
  
  virtual size_t getPartitionNumber(int eventnumber) const override;
  
  virtual size_t freeSlots() override;
    
  virtual StatusCode initialize() override;
  virtual StatusCode finalize() override;
  virtual StatusCode start() override;
 
  HiveMgrSvc(const std::string& name, ISvcLocator* svc);
 
  virtual ~HiveMgrSvc() {}
 
private:
  static void setNumProcs(size_t numProcs);
 
  ServiceHandle<StoreGateSvc> m_hiveStore;
  size_t m_nSlots; //property settable also by setNumberOfStores
  std::vector<SG::HiveEventSlot> m_slots;
  std::mutex m_mutex; //< protects m_slots access
  std::atomic<size_t> m_freeSlots {0};
 
};
} //namespace SG
#endif // STOREGATE_HIVEMGRSVC_H