nuclear@24: #include "threadpool.h"
nuclear@24: 
nuclear@24: ThreadPool::ThreadPool(int num_threads)
nuclear@24: {
nuclear@24: 	quit = false;
nuclear@24: 
nuclear@24: 	if(num_threads == -1) {
nuclear@24: 		num_threads = std::thread::hardware_concurrency();
nuclear@24: 	}
nuclear@24: 
nuclear@24: 	printf("creating thread pool with %d threads\n", num_threads);
nuclear@24: 
nuclear@24: 	thread = new std::thread[num_threads];
nuclear@24: 	for(int i=0; i<num_threads; i++) {
nuclear@24: 		thread[i] = std::thread(&ThreadPool::thread_func, this);
nuclear@24: 	}
nuclear@24: }
nuclear@24: 
nuclear@24: ThreadPool::~ThreadPool()
nuclear@24: {
nuclear@24: 	quit = true;
nuclear@24: 	condvar.notify_all();
nuclear@24: 
nuclear@24: 	printf("ThreadPool: waiting for %d worker threads to stop ", num_threads);
nuclear@24: 	fflush(stdout);
nuclear@24: 	for(int i=0; i<num_threads; i++) {
nuclear@24: 		thread[i].join();
nuclear@24: 		putchar('.');
nuclear@24: 		fflush(stdout);
nuclear@24: 	}
nuclear@24: 	putchar('\n');
nuclear@24: }
nuclear@24: 
nuclear@24: void ThreadPool::add_work(std::function<void ()> func)
nuclear@24: {
nuclear@24: 	std::unique_lock<std::mutex> lock(workq_mutex);
nuclear@24: 	workq.push_back(func);
nuclear@24: }
nuclear@24: 
nuclear@24: void ThreadPool::thread_func()
nuclear@24: {
nuclear@24: 	std::unique_lock<std::mutex> lock(workq_mutex);
nuclear@24: 	for(;;) {
nuclear@24: 		if(quit) break;
nuclear@24: 
nuclear@24: 		condvar.wait(lock);
nuclear@24: 
nuclear@24: 		if(!quit && !workq.empty()) {
nuclear@24: 			std::function<void ()> work = workq.front();
nuclear@24: 			workq.pop_front();
nuclear@24: 			lock.unlock();
nuclear@24: 
nuclear@24: 			work();
nuclear@24: 
nuclear@24: 			lock.lock();
nuclear@24: 		}
nuclear@24: 	}
nuclear@24: }