cc/resources/gpu_tile_task_worker_pool.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "cc/resources/gpu_tile_task_worker_pool.h"
   6
   7 #include <algorithm>
   8
   9 #include "base/trace_event/trace_event.h"
  10 #include "cc/resources/raster_buffer.h"
  11 #include "cc/resources/raster_source.h"
  12 #include "cc/resources/resource.h"
  13 #include "cc/resources/scoped_gpu_raster.h"
  14 #include "gpu/command_buffer/client/gles2_interface.h"
  15 #include "third_party/skia/include/core/SkMultiPictureDraw.h"
  16 #include "third_party/skia/include/core/SkPictureRecorder.h"
  17 #include "third_party/skia/include/core/SkSurface.h"
  18 #include "third_party/skia/include/gpu/GrContext.h"
  19
  20 namespace cc {
  21 namespace {
  22
  23 class RasterBufferImpl : public RasterBuffer {
  24  public:
  25   RasterBufferImpl() {}
  26
  27   // Overridden from RasterBuffer:
  28   void Playback(const RasterSource* raster_source,
  29                 const gfx::Rect& rect,
  30                 float scale) override {
  31     // Don't do anything.
  32   }
  33
  34  private:
  35   DISALLOW_COPY_AND_ASSIGN(RasterBufferImpl);
  36 };
  37
  38 }  // namespace
  39 // static
  40 scoped_ptr<TileTaskWorkerPool> GpuTileTaskWorkerPool::Create(
  41     base::SequencedTaskRunner* task_runner,
  42     TaskGraphRunner* task_graph_runner,
  43     ResourceProvider* resource_provider) {
  44   return make_scoped_ptr<TileTaskWorkerPool>(
  45       new GpuTileTaskWorkerPool(
  46           task_runner, task_graph_runner, resource_provider));
  47 }
  48
  49 // TODO(hendrikw): This class should be removed.  See crbug.com/444938.
  50 GpuTileTaskWorkerPool::GpuTileTaskWorkerPool(
  51     base::SequencedTaskRunner* task_runner,
  52     TaskGraphRunner* task_graph_runner,
  53     ResourceProvider* resource_provider)
  54     : task_runner_(task_runner),
  55       task_graph_runner_(task_graph_runner),
  56       namespace_token_(task_graph_runner_->GetNamespaceToken()),
  57       resource_provider_(resource_provider),
  58       task_set_finished_weak_ptr_factory_(this),
  59       weak_ptr_factory_(this) {
  60 }
  61
  62 GpuTileTaskWorkerPool::~GpuTileTaskWorkerPool() {
  63   DCHECK_EQ(0u, completed_tasks_.size());
  64 }
  65
  66 TileTaskRunner* GpuTileTaskWorkerPool::AsTileTaskRunner() {
  67   return this;
  68 }
  69
  70 void GpuTileTaskWorkerPool::SetClient(TileTaskRunnerClient* client) {
  71   client_ = client;
  72 }
  73
  74 void GpuTileTaskWorkerPool::Shutdown() {
  75   TRACE_EVENT0("cc", "GpuTileTaskWorkerPool::Shutdown");
  76
  77   TaskGraph empty;
  78   task_graph_runner_->ScheduleTasks(namespace_token_, &empty);
  79   task_graph_runner_->WaitForTasksToFinishRunning(namespace_token_);
  80 }
  81
  82 void GpuTileTaskWorkerPool::ScheduleTasks(TileTaskQueue* queue) {
  83   TRACE_EVENT0("cc", "GpuTileTaskWorkerPool::ScheduleTasks");
  84
  85   // Mark all task sets as pending.
  86   tasks_pending_.set();
  87
  88   unsigned priority = kTileTaskPriorityBase;
  89
  90   graph_.Reset();
  91
  92   // Cancel existing OnTaskSetFinished callbacks.
  93   task_set_finished_weak_ptr_factory_.InvalidateWeakPtrs();
  94
  95   scoped_refptr<TileTask> new_task_set_finished_tasks[kNumberOfTaskSets];
  96
  97   size_t task_count[kNumberOfTaskSets] = {0};
  98
  99   for (TaskSet task_set = 0; task_set < kNumberOfTaskSets; ++task_set) {
 100     new_task_set_finished_tasks[task_set] = CreateTaskSetFinishedTask(
 101         task_runner_.get(),
 102         base::Bind(&GpuTileTaskWorkerPool::OnTaskSetFinished,
 103                    task_set_finished_weak_ptr_factory_.GetWeakPtr(), task_set));
 104   }
 105
 106   for (TileTaskQueue::Item::Vector::const_iterator it = queue->items.begin();
 107        it != queue->items.end(); ++it) {
 108     const TileTaskQueue::Item& item = *it;
 109     RasterTask* task = item.task;
 110     DCHECK(!task->HasCompleted());
 111
 112     for (TaskSet task_set = 0; task_set < kNumberOfTaskSets; ++task_set) {
 113       if (!item.task_sets[task_set])
 114         continue;
 115
 116       ++task_count[task_set];
 117
 118       graph_.edges.push_back(
 119           TaskGraph::Edge(task, new_task_set_finished_tasks[task_set].get()));
 120     }
 121
 122     InsertNodesForRasterTask(&graph_, task, task->dependencies(), priority++);
 123   }
 124
 125   for (TaskSet task_set = 0; task_set < kNumberOfTaskSets; ++task_set) {
 126     InsertNodeForTask(&graph_, new_task_set_finished_tasks[task_set].get(),
 127                       kTaskSetFinishedTaskPriorityBase + task_set,
 128                       task_count[task_set]);
 129   }
 130
 131   ScheduleTasksOnOriginThread(this, &graph_);
 132   task_graph_runner_->ScheduleTasks(namespace_token_, &graph_);
 133
 134   std::copy(new_task_set_finished_tasks,
 135             new_task_set_finished_tasks + kNumberOfTaskSets,
 136             task_set_finished_tasks_);
 137 }
 138
 139 void GpuTileTaskWorkerPool::CheckForCompletedTasks() {
 140   TRACE_EVENT0("cc", "GpuTileTaskWorkerPool::CheckForCompletedTasks");
 141
 142   task_graph_runner_->CollectCompletedTasks(namespace_token_,
 143                                             &completed_tasks_);
 144   CompleteTasks(completed_tasks_);
 145   completed_tasks_.clear();
 146 }
 147
 148 ResourceFormat GpuTileTaskWorkerPool::GetResourceFormat() {
 149   return resource_provider_->best_texture_format();
 150 }
 151
 152 void GpuTileTaskWorkerPool::CompleteTasks(const Task::Vector& tasks) {
 153   for (auto& task : tasks) {
 154     RasterTask* raster_task = static_cast<RasterTask*>(task.get());
 155
 156     raster_task->WillComplete();
 157     raster_task->CompleteOnOriginThread(this);
 158     raster_task->DidComplete();
 159
 160     raster_task->RunReplyOnOriginThread();
 161   }
 162   completed_tasks_.clear();
 163 }
 164
 165 scoped_ptr<RasterBuffer> GpuTileTaskWorkerPool::AcquireBufferForRaster(
 166     const Resource* resource) {
 167   return make_scoped_ptr<RasterBuffer>(new RasterBufferImpl());
 168 }
 169
 170 void GpuTileTaskWorkerPool::ReleaseBufferForRaster(
 171     scoped_ptr<RasterBuffer> buffer) {
 172   // Nothing to do here. RasterBufferImpl destructor cleans up after itself.
 173 }
 174
 175 void GpuTileTaskWorkerPool::OnTaskSetFinished(TaskSet task_set) {
 176   TRACE_EVENT1("cc", "GpuTileTaskWorkerPool::OnTaskSetFinished", "task_set",
 177                task_set);
 178
 179   DCHECK(tasks_pending_[task_set]);
 180   tasks_pending_[task_set] = false;
 181   client_->DidFinishRunningTileTasks(task_set);
 182 }
 183
 184 }  // namespace cc