| 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_WORKER_POOL_H_ | 5 #ifndef CC_WORKER_POOL_H_ |
| 6 #define CC_WORKER_POOL_H_ | 6 #define CC_WORKER_POOL_H_ |
| 7 | 7 |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
| 11 #include "base/callback.h" | 11 #include "base/callback.h" |
| 12 #include "base/cancelable_callback.h" | 12 #include "base/cancelable_callback.h" |
| 13 #include "base/memory/scoped_ptr.h" | 13 #include "base/memory/scoped_ptr.h" |
| 14 #include "base/memory/weak_ptr.h" | 14 #include "base/memory/weak_ptr.h" |
| 15 #include "base/threading/thread.h" | 15 #include "base/threading/thread.h" |
| 16 #include "cc/rendering_stats.h" | 16 #include "cc/rendering_stats.h" |
| 17 #include "cc/scoped_ptr_deque.h" | 17 #include "cc/scoped_ptr_deque.h" |
| 18 | 18 |
| 19 namespace cc { | 19 namespace cc { |
| 20 namespace internal { | 20 namespace internal { |
| 21 | 21 |
| 22 class WorkerPoolTask { | 22 class WorkerPoolTask { |
| 23 public: | 23 public: |
| 24 virtual ~WorkerPoolTask(); | 24 virtual ~WorkerPoolTask(); |
| 25 | 25 |
| 26 virtual void Run(RenderingStats* rendering_stats) = 0; | 26 virtual void Run(RenderingStats* rendering_stats) = 0; |
| 27 virtual void DeferToThread(base::Thread* thread) = 0; |
| 27 | 28 |
| 28 bool HasCompleted(); | 29 bool HasCompleted(); |
| 29 void DidComplete(); | 30 void DidComplete(); |
| 30 | 31 |
| 31 protected: | 32 protected: |
| 32 WorkerPoolTask(const base::Closure& reply); | 33 WorkerPoolTask(const base::Closure& reply); |
| 33 | 34 |
| 34 const base::Closure reply_; | 35 const base::Closure reply_; |
| 35 | 36 |
| 36 // Accessed from multiple threads. Set to 1 when task has completed. | 37 // Accessed from multiple threads. Set to 1 when task has completed. |
| (...skipping 77 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 114 | 115 |
| 115 WorkerPool* worker_pool_; | 116 WorkerPool* worker_pool_; |
| 116 ScopedPtrDeque<internal::WorkerPoolTask> pending_tasks_; | 117 ScopedPtrDeque<internal::WorkerPoolTask> pending_tasks_; |
| 117 scoped_ptr<RenderingStats> rendering_stats_; | 118 scoped_ptr<RenderingStats> rendering_stats_; |
| 118 bool record_rendering_stats_; | 119 bool record_rendering_stats_; |
| 119 }; | 120 }; |
| 120 | 121 |
| 121 WorkerPool(WorkerPoolClient* client, size_t num_threads); | 122 WorkerPool(WorkerPoolClient* client, size_t num_threads); |
| 122 | 123 |
| 123 WorkerPool::Worker* GetWorkerForNextTask(); | 124 WorkerPool::Worker* GetWorkerForNextTask(); |
| 125 bool CanPostCheapTask() const; |
| 126 void PostCheapTask(scoped_ptr<internal::WorkerPoolTask> task); |
| 124 | 127 |
| 125 private: | 128 private: |
| 126 class NumPendingTasksComparator { | 129 class NumPendingTasksComparator { |
| 127 public: | 130 public: |
| 128 bool operator() (const Worker* a, const Worker* b) const { | 131 bool operator() (const Worker* a, const Worker* b) const { |
| 129 return a->num_pending_tasks() < b->num_pending_tasks(); | 132 return a->num_pending_tasks() < b->num_pending_tasks(); |
| 130 } | 133 } |
| 131 }; | 134 }; |
| 132 | 135 |
| 133 // Schedule a completed tasks check if not already pending. | 136 // Schedule a completed tasks check if not already pending. |
| (...skipping 10 matching lines...) Expand all Loading... |
| 144 | 147 |
| 145 // Check for completed tasks and run reply callbacks. | 148 // Check for completed tasks and run reply callbacks. |
| 146 void CheckForCompletedTasks(); | 149 void CheckForCompletedTasks(); |
| 147 | 150 |
| 148 // Called when processing task completion. | 151 // Called when processing task completion. |
| 149 void OnTaskCompleted(); | 152 void OnTaskCompleted(); |
| 150 | 153 |
| 151 // Ensure workers are sorted by number of pending tasks. | 154 // Ensure workers are sorted by number of pending tasks. |
| 152 void SortWorkersIfNeeded(); | 155 void SortWorkersIfNeeded(); |
| 153 | 156 |
| 157 // Run pending cheap tasks on origin thread. |
| 158 void RunCheapTasks(); |
| 159 |
| 154 typedef std::vector<Worker*> WorkerVector; | 160 typedef std::vector<Worker*> WorkerVector; |
| 155 WorkerVector workers_; | 161 WorkerVector workers_; |
| 156 WorkerPoolClient* client_; | 162 WorkerPoolClient* client_; |
| 157 scoped_refptr<base::MessageLoopProxy> origin_loop_; | 163 scoped_refptr<base::MessageLoopProxy> origin_loop_; |
| 158 base::WeakPtrFactory<WorkerPool> weak_ptr_factory_; | 164 base::WeakPtrFactory<WorkerPool> weak_ptr_factory_; |
| 159 bool workers_need_sorting_; | 165 bool workers_need_sorting_; |
| 160 bool shutdown_; | 166 bool shutdown_; |
| 161 base::CancelableClosure check_for_completed_tasks_callback_; | 167 base::CancelableClosure check_for_completed_tasks_callback_; |
| 162 bool check_for_completed_tasks_pending_; | 168 bool check_for_completed_tasks_pending_; |
| 163 base::Closure idle_callback_; | 169 base::Closure idle_callback_; |
| 170 base::Closure cheap_task_callback_; |
| 164 // Accessed from multiple threads. 0 when worker pool is idle. | 171 // Accessed from multiple threads. 0 when worker pool is idle. |
| 165 base::subtle::Atomic32 pending_task_count_; | 172 base::subtle::Atomic32 pending_task_count_; |
| 166 | 173 |
| 174 bool cheap_tasks_allowed_; |
| 175 bool run_cheap_tasks_pending_; |
| 176 ScopedPtrDeque<internal::WorkerPoolTask> pending_cheap_tasks_; |
| 177 scoped_ptr<RenderingStats> cheap_rendering_stats_; |
| 178 |
| 167 DISALLOW_COPY_AND_ASSIGN(WorkerPool); | 179 DISALLOW_COPY_AND_ASSIGN(WorkerPool); |
| 168 }; | 180 }; |
| 169 | 181 |
| 170 } // namespace cc | 182 } // namespace cc |
| 171 | 183 |
| 172 #endif // CC_WORKER_POOL_H_ | 184 #endif // CC_WORKER_POOL_H_ |
| OLD | NEW |