RoundRobinSchedulerSvc.h

Go to the documentation of this file.

#ifndef GAUDIHIVE_ROUNDROBINSCHEDULERSVC_H
#define GAUDIHIVE_ROUNDROBINSCHEDULERSVC_H

// Framework include files
#include "GaudiKernel/IScheduler.h"
#include "GaudiKernel/IRunable.h"
#include "GaudiKernel/Service.h"
#include "GaudiKernel/IAlgResourcePool.h"

#include "AlgResourcePool.h"
#include "ExecutionFlowManager.h"
#include "DataFlowManager.h"


// C++ include files
#include <vector>
#include <string>
#include <unordered_map>
#include <functional>
#include <thread>

// External libs
#include "tbb/concurrent_queue.h"


//---------------------------------------------------------------------------

class RoundRobinSchedulerSvc: public extends1<Service, IScheduler> {
public:
  RoundRobinSchedulerSvc( const std::string& name, ISvcLocator* svc );

  ~RoundRobinSchedulerSvc();

  virtual StatusCode initialize();

  virtual StatusCode finalize();

  virtual StatusCode pushNewEvent(EventContext* eventContext);

  // Make multiple events available to the scheduler
  virtual StatusCode pushNewEvents(std::vector<EventContext*>& eventContexts);

  virtual StatusCode popFinishedEvent(EventContext*& eventContext);

  virtual StatusCode tryPopFinishedEvent(EventContext*& eventContext);

  virtual unsigned int freeSlots();


private:

  StatusCode processEvents();

  bool m_useTopAlgList;
  SmartIF<IAlgResourcePool> m_algResourcePool;

  //control flow manager
  concurrency::ExecutionFlowManager m_controlFlow;

  std::vector<std::string> m_algname_vect;

  std::unordered_map<std::string,unsigned int> m_algname_index_map;

  std::vector<std::vector<std::string>> m_algosDependencies;

  std::list<IAlgorithm*> m_algList;

  tbb::concurrent_bounded_queue<EventContext*> m_finishedEvents;

  unsigned int m_freeSlots;
  std::vector<EventContext*> m_evtCtx_buffer;

};

#endif // GAUDIHIVE_ROUNDROBINSCHEDULERSVC_H