diff options
author | Hieu Hoang <hieuhoang@gmail.com> | 2012-11-12 23:56:18 +0400 |
---|---|---|
committer | Hieu Hoang <hieuhoang@gmail.com> | 2012-11-12 23:56:18 +0400 |
commit | 5e3ef23cef6101d2c098eb3445f562e8f595655b (patch) | |
tree | b8c332b6fa82bae84ea4910967a10ba1b08a7107 /moses/ThreadPool.h | |
parent | 8c785cff2b1be3cccd76ea9026f71b649762dfc3 (diff) |
move moses/src/* to moses/
Diffstat (limited to 'moses/ThreadPool.h')
-rw-r--r-- | moses/ThreadPool.h | 127 |
1 files changed, 127 insertions, 0 deletions
diff --git a/moses/ThreadPool.h b/moses/ThreadPool.h new file mode 100644 index 000000000..fad236a98 --- /dev/null +++ b/moses/ThreadPool.h @@ -0,0 +1,127 @@ +// $Id: ThreadPool.h 3045 2010-04-05 13:07:29Z hieuhoang1972 $ + +/*********************************************************************** +Moses - factored phrase-based language decoder +Copyright (C) 2009 University of Edinburgh + +This library 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 2.1 of the License, or (at your option) any later version. + +This library 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 this library; if not, write to the Free Software +Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA +***********************************************************************/ + +#ifndef moses_ThreadPool_h +#define moses_ThreadPool_h + +#include <iostream> +#include <queue> +#include <vector> + +#ifdef WITH_THREADS +#include <boost/bind.hpp> +#include <boost/thread.hpp> +#endif + +#ifdef BOOST_HAS_PTHREADS +#include <pthread.h> +#endif + + +//#include "Util.h" + + +/** + * Classes to implement a ThreadPool. +**/ +namespace Moses { + +/** A task to be executed by the ThreadPool + */ +class Task +{ +public: + virtual void Run() = 0; + virtual bool DeleteAfterExecution() { return true; } + virtual ~Task() {} +}; + +#ifdef WITH_THREADS + +class ThreadPool +{ + public: + /** + * Construct a thread pool of a fixed size. + **/ + explicit ThreadPool(size_t numThreads); + + ~ThreadPool() { + Stop(); + } + + /** + * Add a job to the threadpool. + **/ + void Submit(Task* task); + + /** + * Wait until all queued jobs have completed, and shut down + * the ThreadPool. + **/ + void Stop(bool processRemainingJobs = false); + + /** + * Set maximum number of queued threads (otherwise Submit blocks) + **/ + void SetQueueLimit( size_t limit ) { m_queueLimit = limit; } + +private: + /** + * The main loop executed by each thread. + **/ + void Execute(); + + std::queue<Task*> m_tasks; + boost::thread_group m_threads; + boost::mutex m_mutex; + boost::condition_variable m_threadNeeded; + boost::condition_variable m_threadAvailable; + bool m_stopped; + bool m_stopping; + size_t m_queueLimit; +}; + +class TestTask : public Task +{ +public: + TestTask(int id) : m_id(id) {} + + virtual void Run() { +#ifdef BOOST_HAS_PTHREADS + pthread_t tid = pthread_self(); +#else + typedef void * pthread_t; + pthread_t tid = 0; +#endif + std::cerr << "Executing " << m_id << " in thread id " << tid << std::endl; + } + + virtual ~TestTask() {} + +private: + int m_id; +}; + +#endif //WITH_THREADS + +} // namespace Moses +#endif // moses_ThreadPool_h |