worker_pool.h

Go to the documentation of this file.

/*
 * Copyright 2018 Universidad Carlos III de Madrid
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *     http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#ifndef GRPPI_NATIVE_WORKER_POOL_H
#define GRPPI_NATIVE_WORKER_POOL_H
 
#include <thread>
 
namespace grppi {
 
class worker_pool {
  public:
 
    worker_pool(int num_threads) noexcept : 
        num_threads_{num_threads},
        workers_{}
    {}
 
    ~worker_pool() noexcept { this->wait(); }
 
    worker_pool(worker_pool &&) noexcept = default;
    worker_pool & operator=(worker_pool &&) noexcept = default;
    
    template <typename E, typename F, typename ... Args>
    void launch(const E & ex, F f, Args && ... args) {
      workers_.emplace_back([=,&ex]() {
        auto manager = ex.thread_manager();
        f(args...);
      });
    }
 
    template <typename E, typename F, typename ... Args>
    void launch_tasks(const E & ex, F && f, Args && ... args) {
      for (int i=0; i<num_threads_; ++i) {
        workers_.emplace_back([=,&ex]() {
          auto manager = ex.thread_manager();
          f(args...);
        });
      }
    }
 
    void wait() noexcept {
      for (auto && w : workers_) { w.join(); }
      workers_.clear();
    }
 
  private:
    const int num_threads_;
    std::vector<std::thread> workers_;
};
 
}
 
#endif