| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 // Simple thread pool class | |
| 6 | |
| 7 #ifndef EXAMPLES_DEMO_VORONOI_THREADPOOL_H_ | |
| 8 #define EXAMPLES_DEMO_VORONOI_THREADPOOL_H_ | |
| 9 | |
| 10 #include <pthread.h> | |
| 11 #include <semaphore.h> | |
| 12 | |
| 13 // typdef helper for work function | |
| 14 typedef void (*WorkFunction)(int task_index, void* data); | |
| 15 | |
| 16 // ThreadPool is a class to manage num_threads and assign | |
| 17 // them num_tasks of work at a time. Each call | |
| 18 // to Dispatch(..) will block until all tasks complete. | |
| 19 // If 0 is passed in for num_threads, all tasks will be | |
| 20 // issued on the dispatch thread. | |
| 21 | |
| 22 class ThreadPool { | |
| 23 public: | |
| 24 void Dispatch(int num_tasks, WorkFunction work, void* data); | |
| 25 explicit ThreadPool(int num_threads); | |
| 26 ~ThreadPool(); | |
| 27 private: | |
| 28 int DecCounter(); | |
| 29 void Setup(int counter, WorkFunction work, void* data); | |
| 30 void DispatchMany(int num_tasks, WorkFunction work, void* data); | |
| 31 void DispatchHere(int num_tasks, WorkFunction work, void* data); | |
| 32 void WorkLoop(); | |
| 33 static void* WorkerThreadEntry(void* data); | |
| 34 void PostExitAndJoinAll(); | |
| 35 pthread_t* threads_; | |
| 36 int counter_; | |
| 37 const int num_threads_; | |
| 38 bool exiting_; | |
| 39 void* user_data_; | |
| 40 WorkFunction user_work_function_; | |
| 41 sem_t work_sem_; | |
| 42 sem_t done_sem_; | |
| 43 }; | |
| 44 #endif // EXAMPLES_DEMO_VORONOI_THREADPOOL_H_ | |
| 45 | |
| OLD | NEW |