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.
5 #include "cc/resources/gpu_tile_task_worker_pool.h"
9 #include "base/debug/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"
23 class RasterBufferImpl
: public RasterBuffer
{
27 // Overridden from RasterBuffer:
28 void Playback(const RasterSource
* raster_source
,
29 const gfx::Rect
& rect
,
30 float scale
) override
{
35 DISALLOW_COPY_AND_ASSIGN(RasterBufferImpl
);
40 scoped_ptr
<TileTaskWorkerPool
> GpuTileTaskWorkerPool::Create(
41 base::SequencedTaskRunner
* task_runner
,
42 TaskGraphRunner
* task_graph_runner
) {
43 return make_scoped_ptr
<TileTaskWorkerPool
>(
44 new GpuTileTaskWorkerPool(task_runner
, task_graph_runner
));
47 // TODO(hendrikw): This class should be removed. See crbug.com/444938.
48 GpuTileTaskWorkerPool::GpuTileTaskWorkerPool(
49 base::SequencedTaskRunner
* task_runner
,
50 TaskGraphRunner
* task_graph_runner
)
51 : task_runner_(task_runner
),
52 task_graph_runner_(task_graph_runner
),
53 namespace_token_(task_graph_runner_
->GetNamespaceToken()),
54 run_tasks_on_origin_thread_pending_(false),
55 task_set_finished_weak_ptr_factory_(this),
56 weak_ptr_factory_(this) {
59 GpuTileTaskWorkerPool::~GpuTileTaskWorkerPool() {
60 DCHECK_EQ(0u, completed_tasks_
.size());
63 TileTaskRunner
* GpuTileTaskWorkerPool::AsTileTaskRunner() {
67 void GpuTileTaskWorkerPool::SetClient(TileTaskRunnerClient
* client
) {
71 void GpuTileTaskWorkerPool::Shutdown() {
72 TRACE_EVENT0("cc", "GpuTileTaskWorkerPool::Shutdown");
75 task_graph_runner_
->ScheduleTasks(namespace_token_
, &empty
);
76 task_graph_runner_
->WaitForTasksToFinishRunning(namespace_token_
);
79 void GpuTileTaskWorkerPool::ScheduleTasks(TileTaskQueue
* queue
) {
80 TRACE_EVENT0("cc", "GpuTileTaskWorkerPool::ScheduleTasks");
82 // Mark all task sets as pending.
85 unsigned priority
= kTileTaskPriorityBase
;
89 // Cancel existing OnTaskSetFinished callbacks.
90 task_set_finished_weak_ptr_factory_
.InvalidateWeakPtrs();
92 scoped_refptr
<TileTask
> new_task_set_finished_tasks
[kNumberOfTaskSets
];
94 size_t task_count
[kNumberOfTaskSets
] = {0};
96 for (TaskSet task_set
= 0; task_set
< kNumberOfTaskSets
; ++task_set
) {
97 new_task_set_finished_tasks
[task_set
] = CreateTaskSetFinishedTask(
99 base::Bind(&GpuTileTaskWorkerPool::OnTaskSetFinished
,
100 task_set_finished_weak_ptr_factory_
.GetWeakPtr(), task_set
));
103 for (TileTaskQueue::Item::Vector::const_iterator it
= queue
->items
.begin();
104 it
!= queue
->items
.end(); ++it
) {
105 const TileTaskQueue::Item
& item
= *it
;
106 RasterTask
* task
= item
.task
;
107 DCHECK(!task
->HasCompleted());
109 for (TaskSet task_set
= 0; task_set
< kNumberOfTaskSets
; ++task_set
) {
110 if (!item
.task_sets
[task_set
])
113 ++task_count
[task_set
];
115 graph_
.edges
.push_back(
116 TaskGraph::Edge(task
, new_task_set_finished_tasks
[task_set
].get()));
119 InsertNodesForRasterTask(&graph_
, task
, task
->dependencies(), priority
++);
122 for (TaskSet task_set
= 0; task_set
< kNumberOfTaskSets
; ++task_set
) {
123 InsertNodeForTask(&graph_
, new_task_set_finished_tasks
[task_set
].get(),
124 kTaskSetFinishedTaskPriority
, task_count
[task_set
]);
127 ScheduleTasksOnOriginThread(this, &graph_
);
128 task_graph_runner_
->ScheduleTasks(namespace_token_
, &graph_
);
130 std::copy(new_task_set_finished_tasks
,
131 new_task_set_finished_tasks
+ kNumberOfTaskSets
,
132 task_set_finished_tasks_
);
135 void GpuTileTaskWorkerPool::CheckForCompletedTasks() {
136 TRACE_EVENT0("cc", "GpuTileTaskWorkerPool::CheckForCompletedTasks");
138 task_graph_runner_
->CollectCompletedTasks(namespace_token_
,
140 CompleteTasks(completed_tasks_
);
141 completed_tasks_
.clear();
144 void GpuTileTaskWorkerPool::CompleteTasks(const Task::Vector
& tasks
) {
145 for (auto& task
: tasks
) {
146 RasterTask
* raster_task
= static_cast<RasterTask
*>(task
.get());
148 raster_task
->WillComplete();
149 raster_task
->CompleteOnOriginThread(this);
150 raster_task
->DidComplete();
152 raster_task
->RunReplyOnOriginThread();
154 completed_tasks_
.clear();
157 scoped_ptr
<RasterBuffer
> GpuTileTaskWorkerPool::AcquireBufferForRaster(
158 const Resource
* resource
) {
159 return make_scoped_ptr
<RasterBuffer
>(new RasterBufferImpl());
162 void GpuTileTaskWorkerPool::ReleaseBufferForRaster(
163 scoped_ptr
<RasterBuffer
> buffer
) {
164 // Nothing to do here. RasterBufferImpl destructor cleans up after itself.
167 void GpuTileTaskWorkerPool::OnTaskSetFinished(TaskSet task_set
) {
168 TRACE_EVENT1("cc", "GpuTileTaskWorkerPool::OnTaskSetFinished", "task_set",
171 DCHECK(tasks_pending_
[task_set
]);
172 tasks_pending_
[task_set
] = false;
173 client_
->DidFinishRunningTileTasks(task_set
);