ThreadPoolSvc.h

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2025 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.                                             *
\***********************************************************************************/
#pragma once
 
#include <GaudiKernel/IThreadInitTool.h>
#include <GaudiKernel/IThreadPoolSvc.h>
#include <GaudiKernel/Service.h>
#include <GaudiKernel/ToolHandle.h>
 
#include <boost/thread/barrier.hpp>
 
#include <tbb/global_control.h>
#include <tbb/spin_mutex.h>
#include <tbb/task_arena.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, const int& maxParallelismExtra ) 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;
 
  tbb::task_arena* getArena() { return &m_arena; }
 
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<boost::barrier> m_barrier;

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

  tbb::task_arena m_arena;

  std::atomic<int> m_threadInitCount = 0;
};