ThreadPoolSvc.h

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2019 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
#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/tbb_stddef.h"
#if TBB_INTERFACE_VERSION_MAJOR < 12
#  include "tbb/task_scheduler_init.h"
#  define TBB_PREVIEW_GLOBAL_CONTROL 1
#endif // TBB_INTERFACE_VERSION_MAJOR < 12
#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;

#if TBB_INTERFACE_VERSION_MAJOR < 12
  std::unique_ptr<tbb::task_scheduler_init> m_tbbSchedInit;
#endif // TBB_INTERFACE_VERSION_MAJOR < 12

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

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

#endif