| 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 |