ThreadPool.h

Go to the documentation of this file.

#ifndef THREADPOOL_H_
#define THREADPOOL_H_

// #include <boost/function/function0.hpp>
#include <boost/thread/thread.hpp>
#include <boost/thread/mutex.hpp>
#include <boost/thread/condition.hpp>
#include <vector>

using boost::condition;
using boost::try_mutex;
using boost::mutex;
using boost::thread;


class WorkerMainLoop ;

class ThreadPoolJob
{
public :
    virtual void start() = 0;
    virtual ~ThreadPoolJob() {};
};

class WorkerState
{

public :

    WorkerState() ;
    virtual ~WorkerState() { };

    void assignJob(ThreadPoolJob &j);

    virtual void finish();

private :

    bool toBeFinished ;

    ThreadPoolJob *job;

    friend class WorkerMainLoop ;
};

class WorkerMainLoop
{
public:
    WorkerMainLoop(WorkerState *st, boost::condition &c1, boost::mutex &m1,
                   boost::condition &c2, boost::mutex &m2);

    virtual ~WorkerMainLoop();

    void operator() ();

protected:

    condition &cond1;
    mutex &mtx1;


    condition &cond2;
    mutex &mtx2;

    WorkerState *state ;
};

class ThreadPool
{
public:
    ThreadPool(int maxNumThreads);

    virtual ~ThreadPool();

    int dispatch(int id, ThreadPoolJob & f) ;

    void waitAll();

     int getNumThreads() {
         return _maxNumThreads;
     }

protected :
    int _maxNumThreads ;

    std::vector<thread*> threads;
    std::vector<WorkerState*> workerstates;
    std::vector<condition*> cond1;
    std::vector<condition*> cond2;
    std::vector<mutex*> mtx1;
    std::vector<mutex*> mtx2;

};

#endif /*THREADPOOL_H_*/

---