threadPool.hpp

/*
  ANCH Framework: ANother C++ Hack is a C++ framework based on C++11 standard
  Copyright (C) 2012 Vincent Lachenal
 
  This file is part of ANCH Framework.
 
  ANCH Framework is free software: you can redistribute it and/or modify
  it under the terms of the GNU Lesser General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.
 
  ANCH Framework is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU Lesser General Public License for more details.
 
  You should have received a copy of the GNU Lesser General Public License
  along with ANCH Framework.  If not, see <http://www.gnu.org/licenses/>.
*/
#pragma once
 
#include <functional>
#include <thread>
#include <mutex>
#include <queue>
#include <chrono>
#include <atomic>
#include <condition_variable>
 
namespace anch {

  class ThreadPool {
    // Attributes +
  private:
    std::queue<std::function<void()> > _threads;

    std::mutex _mutex;

    uint32_t _maxThreads;

    std::atomic<uint32_t> _available;

    bool _running;

    std::atomic<bool> _terminating;

    std::condition_variable _termCV;
    // Attributes -
 
    // Constructors +
  public:
    ThreadPool(uint32_t maxThreads = 0);
    // Constructors -
 
    // Destructor +
  public:
    virtual ~ThreadPool();
    // Destructor -
 
    // Methods +
  public:
    void start();

    void stop();

    template<typename R, typename P = std::ratio<1>>
    void awaitTermination(std::chrono::duration<R,P> duration);

    template<typename Func, typename... Args>
    void add(Func function, Args... args);
 
  private:
    void launchTreatment(std::function<void()> func);

    void afterExecute();

    void execute(std::function<void()> func);
    // Methods -
 
  };
 
}
 
#include "impl/threadPool.hpp"