ThreadPoolSvc.h

Go to the documentation of this file.

#ifndef GAUDIHIVE_THREADPOOLSVC_H
#define GAUDIHIVE_THREADPOOLSVC_H

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

#include "boost/thread.hpp"
#include "tbb/spin_mutex.h"
#include "tbb/task_scheduler_init.h"
#define TBB_PREVIEW_GLOBAL_CONTROL 1
#include "tbb/global_control.h"

#include <memory>
#include <vector>

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

  StatusCode initialize() override final;

  StatusCode finalize() override final;

  StatusCode initPool( const int& poolSize ) override final;

  StatusCode terminatePool() override final;

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

  virtual bool isInit() const { return m_init; }

  virtual void initThisThread() override;

private:
  StatusCode launchTasks( bool finalize = false );

  ToolHandleArray<IThreadInitTool> m_threadInitTools = {this};

  bool m_init = false;

  int m_threadPoolSize = 0;

  tbb::spin_mutex m_initMutex;

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

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

  std::unique_ptr<tbb::global_control> m_tbbgc;
};

#endif