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> |
11 | 11 |
12 #include "base/cancelable_callback.h" | 12 #include "base/cancelable_callback.h" |
| 13 #include "base/containers/scoped_ptr_hash_map.h" |
13 #include "base/memory/ref_counted.h" | 14 #include "base/memory/ref_counted.h" |
14 #include "base/memory/scoped_ptr.h" | 15 #include "base/memory/scoped_ptr.h" |
15 #include "base/memory/weak_ptr.h" | 16 #include "base/memory/weak_ptr.h" |
16 #include "base/message_loop/message_loop.h" | 17 #include "base/message_loop/message_loop.h" |
17 #include "cc/base/cc_export.h" | 18 #include "cc/base/cc_export.h" |
18 #include "cc/base/scoped_ptr_hash_map.h" | |
19 | 19 |
20 namespace cc { | 20 namespace cc { |
21 namespace internal { | 21 namespace internal { |
22 | 22 |
23 class CC_EXPORT WorkerPoolTask | 23 class CC_EXPORT WorkerPoolTask |
24 : public base::RefCountedThreadSafe<WorkerPoolTask> { | 24 : public base::RefCountedThreadSafe<WorkerPoolTask> { |
25 public: | 25 public: |
26 virtual void RunOnWorkerThread(unsigned thread_index) = 0; | 26 virtual void RunOnWorkerThread(unsigned thread_index) = 0; |
27 virtual void CompleteOnOriginThread() = 0; | 27 virtual void CompleteOnOriginThread() = 0; |
28 | 28 |
(...skipping 77 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
106 virtual void Shutdown(); | 106 virtual void Shutdown(); |
107 | 107 |
108 // Force a check for completed tasks. | 108 // Force a check for completed tasks. |
109 virtual void CheckForCompletedTasks(); | 109 virtual void CheckForCompletedTasks(); |
110 | 110 |
111 protected: | 111 protected: |
112 // A task graph contains a unique set of tasks with edges between | 112 // A task graph contains a unique set of tasks with edges between |
113 // dependencies pointing in the direction of the dependents. Each task | 113 // dependencies pointing in the direction of the dependents. Each task |
114 // need to be assigned a unique priority and a run count that matches | 114 // need to be assigned a unique priority and a run count that matches |
115 // the number of dependencies. | 115 // the number of dependencies. |
116 typedef ScopedPtrHashMap<internal::WorkerPoolTask*, internal::GraphNode> | 116 typedef base::ScopedPtrHashMap<internal::WorkerPoolTask*, internal::GraphNode> |
117 GraphNodeMap; | 117 GraphNodeMap; |
118 typedef GraphNodeMap TaskGraph; | 118 typedef GraphNodeMap TaskGraph; |
119 | 119 |
120 WorkerPool(size_t num_threads, const std::string& thread_name_prefix); | 120 WorkerPool(size_t num_threads, const std::string& thread_name_prefix); |
121 | 121 |
122 // Schedule running of tasks in |graph|. Any previously scheduled tasks | 122 // Schedule running of tasks in |graph|. Any previously scheduled tasks |
123 // that are not already running will be canceled. Canceled tasks don't run | 123 // that are not already running will be canceled. Canceled tasks don't run |
124 // but completion of them is still processed. | 124 // but completion of them is still processed. |
125 void SetTaskGraph(TaskGraph* graph); | 125 void SetTaskGraph(TaskGraph* graph); |
126 | 126 |
127 private: | 127 private: |
128 class Inner; | 128 class Inner; |
129 friend class Inner; | 129 friend class Inner; |
130 | 130 |
131 typedef std::vector<scoped_refptr<internal::WorkerPoolTask> > TaskVector; | 131 typedef std::vector<scoped_refptr<internal::WorkerPoolTask> > TaskVector; |
132 | 132 |
133 void ProcessCompletedTasks(const TaskVector& completed_tasks); | 133 void ProcessCompletedTasks(const TaskVector& completed_tasks); |
134 | 134 |
135 bool in_dispatch_completion_callbacks_; | 135 bool in_dispatch_completion_callbacks_; |
136 | 136 |
137 // Hide the gory details of the worker pool in |inner_|. | 137 // Hide the gory details of the worker pool in |inner_|. |
138 const scoped_ptr<Inner> inner_; | 138 const scoped_ptr<Inner> inner_; |
139 }; | 139 }; |
140 | 140 |
141 } // namespace cc | 141 } // namespace cc |
142 | 142 |
143 #endif // CC_RESOURCES_WORKER_POOL_H_ | 143 #endif // CC_RESOURCES_WORKER_POOL_H_ |
OLD | NEW |