/home/brandes/workspace/LAMA/src/lama/task/LAMAThreadPool.hpp

Go to the documentation of this file.

#ifndef LAMA_THREAD_POOL_HPP_
#define LAMA_THREAD_POOL_HPP_

// for dll_import
#include <lama/config.hpp>

// logging
#include <logging/logging.hpp>

// boost
#include <boost/shared_ptr.hpp>
#include <boost/function.hpp>
#include <boost/version.hpp>

#include <boost/thread/thread.hpp>
#include <boost/thread/mutex.hpp>
#include <boost/thread/condition.hpp>

#include <climits>
#include <queue>

namespace lama {

typedef unsigned int TaskId;

struct LAMA_DLL_IMPORTEXPORT LAMAThreadTask
{
    enum TaskState { DEFINED, QUEUED, RUNNING, FINISHED };

    boost::function<void()> mWork;  

    volatile TaskState mState;      

    bool mException;                

    unsigned int mTaskId;  

    int ompThreads;        

    static boost::shared_ptr<LAMAThreadTask> create( boost::function<void()> work, unsigned int taskId, int numOmpThreads = 0 );
};

class LAMA_DLL_IMPORTEXPORT LAMAThreadPool 
{

public:

    LAMAThreadPool( int size );

    boost::shared_ptr<LAMAThreadTask> schedule( boost::function<void()> work, int numOmpThreads = 0 );

    void wait( boost::shared_ptr<LAMAThreadTask> task );

    ~LAMAThreadPool();

private:

    void worker( int id );

    LAMA_LOG_DECL_STATIC_LOGGER( logger );

    unsigned int mTaskId;           // last given task id

    int mMaxSize;                   // number of worker threads

    std::vector<boost::thread*> mThreads;  // worker threads of this pool

    std::queue<boost::shared_ptr<LAMAThreadTask> > mTaskQueue;

    boost::condition mNotifyFinished;  // notify about finished tasks
    boost::condition mNotifyTask;      // notify about new task

    boost::mutex mTaskQueueMutex;      // make access to taskqueue thread-safe
    boost::mutex mNotifyFinishMutex;   // used for wait on mNotifyFinished

    enum WorkerState { WORKING, WAITING };

    WorkerState mWorkerState; //used to indicate the state of the worker
};

} // namespace lama

#endif // LAMA_THREAD_POOL_HPP_