ThreadPoolSvc.h

Go to the documentation of this file.

#ifndef GAUDIHIVE_THREADPOOLSVC_H
#define GAUDIHIVE_THREADPOOLSVC_H

#include "GaudiKernel/Service.h"
#include "GaudiKernel/IThreadPoolSvc.h"
#include "GaudiKernel/IThreadInitTool.h"
#include "GaudiKernel/ToolHandle.h"

#include "tbb/spin_mutex.h"
#include "tbb/task_scheduler_init.h"
#include "boost/thread.hpp"

#include <vector>
#include <memory>

class ThreadPoolSvc: public extends<Service,
                                    IThreadPoolSvc> {
public:
  ThreadPoolSvc( const std::string& name, ISvcLocator* svc );

  virtual StatusCode initialize() override final;

  virtual StatusCode finalize() override final;

  virtual StatusCode initPool(const int& poolSize) override final;

  virtual StatusCode terminatePool() override final;

  virtual int poolSize() const override final {
    return m_threadPoolSize;
  }

  virtual bool isInit() const { return m_init; }

private:

  StatusCode launchTasks(bool finalize=false);

  ToolHandleArray<IThreadInitTool> m_threadInitTools;

  bool m_init;

  int m_threadPoolSize;

  tbb::spin_mutex m_initMutex;

  std::unique_ptr<tbb::task_scheduler_init> m_tbbSchedInit;

  std::unique_ptr<boost::barrier> m_barrier;

};

#endif