workerpool.h

Go to the documentation of this file.

 
#ifndef WORKERPOOL_H
#define WORKERPOOL_H
 
#include <vector>
#include <thread>
#include <functional>
#include <mutex>
#include <condition_variable>
 
/*****************************************************************************/
class WorkerPool
{
public:
    static constexpr unsigned WorkersMax = 8;
 
    WorkerPool();
    ~WorkerPool();

    void enqueue(std::function<void()> job);

    void waitAll();

    int workerCount() const {return activeWorkers;}
 
private:
    std::vector<std::thread> workers;
    std::vector<std::function<void()>> jobs;
 
    std::mutex queueMutex;
 
    std::condition_variable jobCV;
    std::condition_variable waitCV;
 
    bool stop;
    int jobsPrepared;
    int jobsUnfinished;
    int activeWorkers;
};
 
#endif // WORKERPOOL_H