ThreadsController.h

Go to the documentation of this file.

#ifndef FWCore_Concurrency_ThreadsController_h
#define FWCore_Concurrency_ThreadsController_h
// -*- C++ -*-
//
// Package:     FWCore/Concurrency
// Class  :     ThreadsController
//
//
// Original Author:  FWCore
//         Created:  Fri, 18 Nov 2016 20:30:42 GMT
//
 
// system include files
#include "tbb/global_control.h"
#include "tbb/task_arena.h"
#include <memory>
 
// user include files
 
// forward declarations
 
namespace edm {
  class ThreadsController {
  public:
    ThreadsController() = delete;
    explicit ThreadsController(unsigned int iNThreads)
        : m_nThreads{tbb::global_control::max_allowed_parallelism, iNThreads}, m_stackSize{} {}
    ThreadsController(unsigned int iNThreads, size_t iStackSize)
        : m_nThreads{tbb::global_control::max_allowed_parallelism, iNThreads}, m_stackSize{makeStackSize(iStackSize)} {}
 
    // ---------- member functions ---------------------------
    void setStackSize(size_t iStackSize) { m_stackSize = makeStackSize(iStackSize); }
 
  private:
    static std::unique_ptr<tbb::global_control> makeStackSize(size_t iStackSize);
 
    // ---------- member data --------------------------------
    tbb::global_control m_nThreads;
    std::unique_ptr<tbb::global_control> m_stackSize;
  };
}  // namespace edm
 
#endif