| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 #include "cc/raster/single_thread_task_graph_runner.h" | 5 #include "cc/raster/single_thread_task_graph_runner.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 | 8 |
| 9 #include "base/threading/simple_thread.h" | 9 #include "base/threading/simple_thread.h" |
| 10 #include "base/threading/thread_restrictions.h" | 10 #include "base/threading/thread_restrictions.h" |
| (...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 105 { | 105 { |
| 106 base::AutoLock lock(lock_); | 106 base::AutoLock lock(lock_); |
| 107 work_queue_.CollectCompletedTasks(token, completed_tasks); | 107 work_queue_.CollectCompletedTasks(token, completed_tasks); |
| 108 } | 108 } |
| 109 } | 109 } |
| 110 | 110 |
| 111 void SingleThreadTaskGraphRunner::Run() { | 111 void SingleThreadTaskGraphRunner::Run() { |
| 112 base::AutoLock lock(lock_); | 112 base::AutoLock lock(lock_); |
| 113 | 113 |
| 114 while (true) { | 114 while (true) { |
| 115 if (!RunTaskWithLockAcquired()) { | 115 if (!work_queue_.HasReadyToRunTasks()) { |
| 116 // Exit when shutdown is set and no more tasks are pending. | 116 // Exit when shutdown is set and no more tasks are pending. |
| 117 if (shutdown_) | 117 if (shutdown_) |
| 118 break; | 118 break; |
| 119 | 119 |
| 120 // Wait for more tasks. | 120 // Wait for more tasks. |
| 121 has_ready_to_run_tasks_cv_.Wait(); | 121 has_ready_to_run_tasks_cv_.Wait(); |
| 122 continue; | 122 continue; |
| 123 } | 123 } |
| 124 |
| 125 RunTaskWithLockAcquired(); |
| 124 } | 126 } |
| 125 } | 127 } |
| 126 | 128 |
| 127 bool SingleThreadTaskGraphRunner::RunTaskWithLockAcquired() { | 129 void SingleThreadTaskGraphRunner::RunTaskWithLockAcquired() { |
| 128 TRACE_EVENT0("toplevel", | 130 TRACE_EVENT0("toplevel", |
| 129 "SingleThreadTaskGraphRunner::RunTaskWithLockAcquired"); | 131 "SingleThreadTaskGraphRunner::RunTaskWithLockAcquired"); |
| 130 | 132 |
| 131 lock_.AssertAcquired(); | 133 lock_.AssertAcquired(); |
| 132 | 134 |
| 133 // Find the first category with any tasks to run. This task graph runner | 135 auto prioritized_task = work_queue_.GetNextTaskToRun(); |
| 134 // treats categories as an additional priority. | |
| 135 const auto& ready_to_run_namespaces = work_queue_.ready_to_run_namespaces(); | |
| 136 auto found = std::find_if( | |
| 137 ready_to_run_namespaces.cbegin(), ready_to_run_namespaces.cend(), | |
| 138 [](const std::pair<uint16_t, TaskGraphWorkQueue::TaskNamespace::Vector>& | |
| 139 pair) { return !pair.second.empty(); }); | |
| 140 | |
| 141 if (found == ready_to_run_namespaces.cend()) { | |
| 142 return false; | |
| 143 } | |
| 144 | |
| 145 const uint16_t category = found->first; | |
| 146 auto prioritized_task = work_queue_.GetNextTaskToRun(category); | |
| 147 Task* task = prioritized_task.task; | 136 Task* task = prioritized_task.task; |
| 148 | 137 |
| 149 // Call WillRun() before releasing |lock_| and running task. | 138 // Call WillRun() before releasing |lock_| and running task. |
| 150 task->WillRun(); | 139 task->WillRun(); |
| 151 | 140 |
| 152 { | 141 { |
| 153 base::AutoUnlock unlock(lock_); | 142 base::AutoUnlock unlock(lock_); |
| 154 task->RunOnWorkerThread(); | 143 task->RunOnWorkerThread(); |
| 155 } | 144 } |
| 156 | 145 |
| 157 // This will mark task as finished running. | 146 // This will mark task as finished running. |
| 158 task->DidRun(); | 147 task->DidRun(); |
| 159 | 148 |
| 160 work_queue_.CompleteTask(prioritized_task); | 149 work_queue_.CompleteTask(prioritized_task); |
| 161 | 150 |
| 162 // If namespace has finished running all tasks, wake up origin thread. | 151 // If namespace has finished running all tasks, wake up origin thread. |
| 163 if (work_queue_.HasFinishedRunningTasksInNamespace( | 152 if (work_queue_.HasFinishedRunningTasksInNamespace( |
| 164 prioritized_task.task_namespace)) | 153 prioritized_task.task_namespace)) |
| 165 has_namespaces_with_finished_running_tasks_cv_.Signal(); | 154 has_namespaces_with_finished_running_tasks_cv_.Signal(); |
| 166 | |
| 167 return true; | |
| 168 } | 155 } |
| 169 | 156 |
| 170 } // namespace cc | 157 } // namespace cc |
| OLD | NEW |