blob: 0924c0a5118b24e3a97717b8e842d3045fe55286 [file] [log] [blame]
// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "content/renderer/raster_worker_pool.h"
#include <string>
#include <utility>
#include <vector>
#include "base/strings/stringprintf.h"
#include "base/threading/thread_restrictions.h"
#include "base/trace_event/trace_event.h"
#include "cc/base/math_util.h"
#include "cc/raster/task_category.h"
namespace content {
namespace {
// A thread which forwards to RasterWorkerPool::Run with the runnable
// categories.
class RasterWorkerPoolThread : public base::SimpleThread {
public:
explicit RasterWorkerPoolThread(const std::string& name_prefix,
const Options& options,
RasterWorkerPool* pool,
std::vector<cc::TaskCategory> categories)
: SimpleThread(name_prefix, options),
pool_(pool),
categories_(categories) {}
void Run() override { pool_->Run(categories_); }
private:
RasterWorkerPool* const pool_;
const std::vector<cc::TaskCategory> categories_;
};
} // namespace
// A sequenced task runner which posts tasks to a RasterWorkerPool.
class RasterWorkerPool::RasterWorkerPoolSequencedTaskRunner
: public base::SequencedTaskRunner {
public:
explicit RasterWorkerPoolSequencedTaskRunner(
cc::TaskGraphRunner* task_graph_runner)
: task_graph_runner_(task_graph_runner),
namespace_token_(task_graph_runner->GetNamespaceToken()) {}
// Overridden from base::TaskRunner:
bool PostDelayedTask(const tracked_objects::Location& from_here,
const base::Closure& task,
base::TimeDelta delay) override {
return PostNonNestableDelayedTask(from_here, task, delay);
}
bool RunsTasksOnCurrentThread() const override { return true; }
// Overridden from base::SequencedTaskRunner:
bool PostNonNestableDelayedTask(const tracked_objects::Location& from_here,
const base::Closure& task,
base::TimeDelta delay) override {
base::AutoLock lock(lock_);
// Remove completed tasks.
DCHECK(completed_tasks_.empty());
task_graph_runner_->CollectCompletedTasks(namespace_token_,
&completed_tasks_);
tasks_.erase(tasks_.begin(), tasks_.begin() + completed_tasks_.size());
tasks_.push_back(make_scoped_refptr(new ClosureTask(task)));
graph_.Reset();
for (const auto& graph_task : tasks_) {
int dependencies = 0;
if (!graph_.nodes.empty())
dependencies = 1;
// Treat any tasks that are enqueued through the SequencedTaskRunner as
// FOREGROUND priority. We don't have enough information to know the
// actual priority of such tasks, so we run them as soon as possible.
cc::TaskGraph::Node node(graph_task.get(), cc::TASK_CATEGORY_FOREGROUND,
0u /* priority */, dependencies);
if (dependencies) {
graph_.edges.push_back(
cc::TaskGraph::Edge(graph_.nodes.back().task, node.task));
}
graph_.nodes.push_back(node);
}
task_graph_runner_->ScheduleTasks(namespace_token_, &graph_);
completed_tasks_.clear();
return true;
}
private:
~RasterWorkerPoolSequencedTaskRunner() override {
task_graph_runner_->WaitForTasksToFinishRunning(namespace_token_);
task_graph_runner_->CollectCompletedTasks(namespace_token_,
&completed_tasks_);
};
// Lock to exclusively access all the following members that are used to
// implement the SequencedTaskRunner interfaces.
base::Lock lock_;
cc::TaskGraphRunner* task_graph_runner_;
// Namespace used to schedule tasks in the task graph runner.
cc::NamespaceToken namespace_token_;
// List of tasks currently queued up for execution.
cc::Task::Vector tasks_;
// Graph object used for scheduling tasks.
cc::TaskGraph graph_;
// Cached vector to avoid allocation when getting the list of complete
// tasks.
cc::Task::Vector completed_tasks_;
};
RasterWorkerPool::RasterWorkerPool()
: namespace_token_(GetNamespaceToken()),
has_ready_to_run_tasks_cv_(&lock_),
has_namespaces_with_finished_running_tasks_cv_(&lock_),
shutdown_(false) {}
void RasterWorkerPool::Start(
int num_threads,
const base::SimpleThread::Options& thread_options) {
DCHECK(threads_.empty());
while (threads_.size() < static_cast<size_t>(num_threads)) {
// Determine the categories that each thread can run.
std::vector<cc::TaskCategory> task_categories;
// The first thread can run nonconcurrent tasks.
if (threads_.size() == 0) {
task_categories.push_back(cc::TASK_CATEGORY_NONCONCURRENT_FOREGROUND);
}
// All threads can run foreground tasks.
task_categories.push_back(cc::TASK_CATEGORY_FOREGROUND);
// The last thread can run background tasks.
if (threads_.size() == (static_cast<size_t>(num_threads) - 1)) {
task_categories.push_back(cc::TASK_CATEGORY_BACKGROUND);
}
scoped_ptr<base::SimpleThread> thread(new RasterWorkerPoolThread(
base::StringPrintf("CompositorTileWorker%u",
static_cast<unsigned>(threads_.size() + 1))
.c_str(),
thread_options, this, task_categories));
thread->Start();
threads_.push_back(std::move(thread));
}
}
void RasterWorkerPool::Shutdown() {
WaitForTasksToFinishRunning(namespace_token_);
CollectCompletedTasks(namespace_token_, &completed_tasks_);
// Shutdown raster threads.
{
base::AutoLock lock(lock_);
DCHECK(!work_queue_.HasReadyToRunTasks());
DCHECK(!work_queue_.HasAnyNamespaces());
DCHECK(!shutdown_);
shutdown_ = true;
// Wake up all workers so they exit.
has_ready_to_run_tasks_cv_.Broadcast();
}
while (!threads_.empty()) {
threads_.back()->Join();
threads_.pop_back();
}
}
// Overridden from base::TaskRunner:
bool RasterWorkerPool::PostDelayedTask(
const tracked_objects::Location& from_here,
const base::Closure& task,
base::TimeDelta delay) {
base::AutoLock lock(lock_);
// Remove completed tasks.
DCHECK(completed_tasks_.empty());
CollectCompletedTasksWithLockAcquired(namespace_token_, &completed_tasks_);
cc::Task::Vector::iterator end = std::remove_if(
tasks_.begin(), tasks_.end(), [this](const scoped_refptr<cc::Task>& e) {
return std::find(this->completed_tasks_.begin(),
this->completed_tasks_.end(),
e) != this->completed_tasks_.end();
});
tasks_.erase(end, tasks_.end());
tasks_.push_back(make_scoped_refptr(new ClosureTask(task)));
graph_.Reset();
for (const auto& graph_task : tasks_) {
// Delayed tasks are assigned FOREGROUND category, ensuring that they run as
// soon as possible once their delay has expired.
graph_.nodes.push_back(
cc::TaskGraph::Node(graph_task.get(), cc::TASK_CATEGORY_FOREGROUND,
0u /* priority */, 0u /* dependencies */));
}
ScheduleTasksWithLockAcquired(namespace_token_, &graph_);
completed_tasks_.clear();
return true;
}
bool RasterWorkerPool::RunsTasksOnCurrentThread() const {
return true;
}
void RasterWorkerPool::Run(const std::vector<cc::TaskCategory>& categories) {
base::AutoLock lock(lock_);
while (true) {
if (!RunTaskWithLockAcquired(categories)) {
// Exit when shutdown is set and no more tasks are pending.
if (shutdown_)
break;
// Wait for more tasks.
has_ready_to_run_tasks_cv_.Wait();
continue;
}
}
}
void RasterWorkerPool::FlushForTesting() {
base::AutoLock lock(lock_);
while (!work_queue_.HasFinishedRunningTasksInAllNamespaces()) {
has_namespaces_with_finished_running_tasks_cv_.Wait();
}
}
scoped_refptr<base::SequencedTaskRunner>
RasterWorkerPool::CreateSequencedTaskRunner() {
return new RasterWorkerPoolSequencedTaskRunner(this);
}
RasterWorkerPool::~RasterWorkerPool() {}
cc::NamespaceToken RasterWorkerPool::GetNamespaceToken() {
base::AutoLock lock(lock_);
return work_queue_.GetNamespaceToken();
}
void RasterWorkerPool::ScheduleTasks(cc::NamespaceToken token,
cc::TaskGraph* graph) {
TRACE_EVENT2("disabled-by-default-cc.debug",
"RasterWorkerPool::ScheduleTasks", "num_nodes",
graph->nodes.size(), "num_edges", graph->edges.size());
{
base::AutoLock lock(lock_);
ScheduleTasksWithLockAcquired(token, graph);
}
}
void RasterWorkerPool::ScheduleTasksWithLockAcquired(cc::NamespaceToken token,
cc::TaskGraph* graph) {
DCHECK(token.IsValid());
DCHECK(!cc::TaskGraphWorkQueue::DependencyMismatch(graph));
DCHECK(!shutdown_);
work_queue_.ScheduleTasks(token, graph);
// If there is more work available, wake up the other worker threads.
if (work_queue_.HasReadyToRunTasks())
has_ready_to_run_tasks_cv_.Broadcast();
}
void RasterWorkerPool::WaitForTasksToFinishRunning(cc::NamespaceToken token) {
TRACE_EVENT0("disabled-by-default-cc.debug",
"RasterWorkerPool::WaitForTasksToFinishRunning");
DCHECK(token.IsValid());
{
base::AutoLock lock(lock_);
base::ThreadRestrictions::ScopedAllowWait allow_wait;
auto* task_namespace = work_queue_.GetNamespaceForToken(token);
if (!task_namespace)
return;
while (!work_queue_.HasFinishedRunningTasksInNamespace(task_namespace))
has_namespaces_with_finished_running_tasks_cv_.Wait();
}
}
void RasterWorkerPool::CollectCompletedTasks(
cc::NamespaceToken token,
cc::Task::Vector* completed_tasks) {
TRACE_EVENT0("disabled-by-default-cc.debug",
"RasterWorkerPool::CollectCompletedTasks");
{
base::AutoLock lock(lock_);
CollectCompletedTasksWithLockAcquired(token, completed_tasks);
}
}
void RasterWorkerPool::CollectCompletedTasksWithLockAcquired(
cc::NamespaceToken token,
cc::Task::Vector* completed_tasks) {
DCHECK(token.IsValid());
work_queue_.CollectCompletedTasks(token, completed_tasks);
}
bool RasterWorkerPool::RunTaskWithLockAcquired(
const std::vector<cc::TaskCategory>& categories) {
for (const auto& category : categories) {
if (work_queue_.HasReadyToRunTasksForCategory(category)) {
RunTaskInCategoryWithLockAcquired(category);
return true;
}
}
return false;
}
void RasterWorkerPool::RunTaskInCategoryWithLockAcquired(
cc::TaskCategory category) {
TRACE_EVENT0("toplevel", "TaskGraphRunner::RunTask");
lock_.AssertAcquired();
auto prioritized_task = work_queue_.GetNextTaskToRun(category);
cc::Task* task = prioritized_task.task;
// Call WillRun() before releasing |lock_| and running task.
task->WillRun();
{
base::AutoUnlock unlock(lock_);
task->RunOnWorkerThread();
}
// This will mark task as finished running.
task->DidRun();
work_queue_.CompleteTask(prioritized_task);
// We may have just dequeued more tasks, wake up the other worker threads.
if (work_queue_.HasReadyToRunTasks())
has_ready_to_run_tasks_cv_.Broadcast();
// If namespace has finished running all tasks, wake up origin threads.
if (work_queue_.HasFinishedRunningTasksInNamespace(
prioritized_task.task_namespace))
has_namespaces_with_finished_running_tasks_cv_.Broadcast();
}
RasterWorkerPool::ClosureTask::ClosureTask(const base::Closure& closure)
: closure_(closure) {}
// Overridden from cc::Task:
void RasterWorkerPool::ClosureTask::RunOnWorkerThread() {
closure_.Run();
closure_.Reset();
}
RasterWorkerPool::ClosureTask::~ClosureTask() {}
} // namespace content