| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CC_RESOURCES_WORKER_POOL_H_ | 5 #ifndef CC_RESOURCES_WORKER_POOL_H_ |
| 6 #define CC_RESOURCES_WORKER_POOL_H_ | 6 #define CC_RESOURCES_WORKER_POOL_H_ |
| 7 | 7 |
| 8 #include <deque> | 8 #include <deque> |
| 9 #include <string> | 9 #include <string> |
| 10 #include <vector> | 10 #include <vector> |
| (...skipping 87 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 98 // A worker thread pool that runs tasks provided by task graph and | 98 // A worker thread pool that runs tasks provided by task graph and |
| 99 // guarantees completion of all pending tasks at shutdown. | 99 // guarantees completion of all pending tasks at shutdown. |
| 100 class CC_EXPORT WorkerPool { | 100 class CC_EXPORT WorkerPool { |
| 101 public: | 101 public: |
| 102 virtual ~WorkerPool(); | 102 virtual ~WorkerPool(); |
| 103 | 103 |
| 104 // Tells the worker pool to shutdown and returns once all pending tasks have | 104 // Tells the worker pool to shutdown and returns once all pending tasks have |
| 105 // completed. | 105 // completed. |
| 106 virtual void Shutdown(); | 106 virtual void Shutdown(); |
| 107 | 107 |
| 108 protected: | 108 // Sets the number of threads to use for running raster tasks. |
| 109 // Can only be called once prior to GetNumRasterThreads(). |
| 110 // Caller is responsible for correct ordering. |
| 111 // TODO(reveman): Move this to RasterWorkerPool: crbug.com/331844 |
| 112 static void SetNumRasterThreads(int num_threads); |
| 113 |
| 114 // Gets the number of threads to use for running raster tasks. |
| 115 // TODO(reveman): Move this to RasterWorkerPool: crbug.com/331844 |
| 116 static int GetNumRasterThreads(); |
| 117 |
| 109 // A task graph contains a unique set of tasks with edges between | 118 // A task graph contains a unique set of tasks with edges between |
| 110 // dependencies pointing in the direction of the dependents. Each task | 119 // dependencies pointing in the direction of the dependents. Each task |
| 111 // need to be assigned a unique priority and a run count that matches | 120 // need to be assigned a unique priority and a run count that matches |
| 112 // the number of dependencies. | 121 // the number of dependencies. |
| 113 typedef base::ScopedPtrHashMap<internal::WorkerPoolTask*, internal::GraphNode> | 122 typedef base::ScopedPtrHashMap<internal::WorkerPoolTask*, internal::GraphNode> |
| 114 GraphNodeMap; | 123 GraphNodeMap; |
| 115 typedef GraphNodeMap TaskGraph; | 124 typedef GraphNodeMap TaskGraph; |
| 125 typedef std::vector<scoped_refptr<internal::WorkerPoolTask> > TaskVector; |
| 116 | 126 |
| 117 WorkerPool(size_t num_threads, const std::string& thread_name_prefix); | 127 protected: |
| 128 WorkerPool(); |
| 118 | 129 |
| 119 // Schedule running of tasks in |graph|. Any previously scheduled tasks | 130 // Schedule running of tasks in |graph|. Any previously scheduled tasks |
| 120 // that are not already running will be canceled. Canceled tasks don't run | 131 // that are not already running will be canceled. Canceled tasks don't run |
| 121 // but completion of them is still processed. | 132 // but completion of them is still processed. |
| 122 void SetTaskGraph(TaskGraph* graph); | 133 void SetTaskGraph(TaskGraph* graph); |
| 123 | 134 |
| 124 // Force a check for completed tasks. | 135 // Force a check for completed tasks. |
| 125 void CheckForCompletedWorkerTasks(); | 136 void CheckForCompletedWorkerTasks(); |
| 126 | 137 |
| 127 private: | 138 private: |
| 128 class Inner; | |
| 129 friend class Inner; | |
| 130 | |
| 131 typedef std::vector<scoped_refptr<internal::WorkerPoolTask> > TaskVector; | |
| 132 | |
| 133 void ProcessCompletedTasks(const TaskVector& completed_tasks); | 139 void ProcessCompletedTasks(const TaskVector& completed_tasks); |
| 134 | 140 |
| 135 bool in_dispatch_completion_callbacks_; | 141 bool in_dispatch_completion_callbacks_; |
| 136 | |
| 137 // Hide the gory details of the worker pool in |inner_|. | |
| 138 const scoped_ptr<Inner> inner_; | |
| 139 }; | 142 }; |
| 140 | 143 |
| 141 } // namespace cc | 144 } // namespace cc |
| 142 | 145 |
| 143 #endif // CC_RESOURCES_WORKER_POOL_H_ | 146 #endif // CC_RESOURCES_WORKER_POOL_H_ |
| OLD | NEW |