MPIHiveEventLoopMgr.h

Go to the documentation of this file.

// -*- C++ -*-
 
/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ATHENASERVICES_MPIHIVEEVENTLOOPMGR_H
#define ATHENASERVICES_MPIHIVEEVENTLOOPMGR_H
 
// Base class headers
#include "AthenaHiveEventLoopMgr.h"
 
// Athena headers
#include "AthenaKernel/IMPIClusterSvc.h"
#include "EventInfo/EventID.h" /* number_type */
 
// Gaudi headers
#include "Gaudi/Property.h"
 
// Standard includes
#include <string>

class MPIHiveEventLoopMgr : public AthenaHiveEventLoopMgr {
  // for Hive
 protected:
  ServiceHandle<IMPIClusterSvc> m_clusterSvc{this, "MPIClusterSvc", "",
                                             "MPIClusterSvc"};

  StatusCode drainLocalScheduler();
  StatusCode insertEvent(int eventIdx, bool& endOfStream,
                         std::int64_t requestTime_ns);
  StatusCode workerEventLoop();
  StatusCode masterEventLoop(int maxEvt);
 
  // Keep track of how many failed events we've had
  int m_contiguousFailedEvts{0};
  int m_totalFailedEvts{0};
  // Keeps track of events already processed
  int m_nLocalCreatedEvts{0};
  int m_nLocalSkippedEvts{0};
  int m_nLocalFinishedEvts{0};
 
 public:
  MPIHiveEventLoopMgr(const std::string& name, ISvcLocator* svcLoc);
  virtual ~MPIHiveEventLoopMgr();
  virtual StatusCode initialize() override;
  virtual StatusCode finalize() override;
  virtual StatusCode nextEvent(int maxevt) override;
 
 private:
  UnsignedIntegerProperty m_firstEventIndex{
      this, "FirstEventIndex", 0, "First event index (Exec.SkipEvents)"};
  int m_evtSelectorCurrentPos = 0;
 
  StoreGateSvc* eventStore() const;
};
 
#endif  // ATHENASERVICES_MPIHIVEEVENTLOOPMGR_H