Index: cc/resources/raster_worker_pool_perftest.cc |
diff --git a/cc/resources/raster_worker_pool_perftest.cc b/cc/resources/raster_worker_pool_perftest.cc |
index da53d8ecfe4a9918435f52f7137d6a08a5140790..4008ef8d6010755879567e99d2c133b35202769b 100644 |
--- a/cc/resources/raster_worker_pool_perftest.cc |
+++ b/cc/resources/raster_worker_pool_perftest.cc |
@@ -147,6 +147,8 @@ class RasterWorkerPoolPerfTestBase { |
public: |
typedef std::vector<scoped_refptr<RasterTask> > RasterTaskVector; |
+ enum NamedTaskSet { REQUIRED_FOR_ACTIVATION = 0, ALL = 1 }; |
+ |
RasterWorkerPoolPerfTestBase() |
: context_provider_(make_scoped_refptr(new PerfContextProvider)), |
task_runner_(new base::TestSimpleTaskRunner), |
@@ -196,8 +198,11 @@ class RasterWorkerPoolPerfTestBase { |
const RasterTaskVector& raster_tasks) { |
for (size_t i = 0u; i < raster_tasks.size(); ++i) { |
bool required_for_activation = (i % 2) == 0; |
- queue->items.push_back(RasterTaskQueue::Item(raster_tasks[i].get(), |
- required_for_activation)); |
+ TaskSetCollection task_set_collection; |
+ task_set_collection[ALL] = true; |
+ task_set_collection[REQUIRED_FOR_ACTIVATION] = required_for_activation; |
+ queue->items.push_back( |
+ RasterTaskQueue::Item(raster_tasks[i].get(), task_set_collection)); |
} |
} |
@@ -261,14 +266,12 @@ class RasterWorkerPoolPerfTest |
} |
// Overriden from RasterizerClient: |
- virtual bool ShouldForceTasksRequiredForActivationToComplete() const |
- OVERRIDE { |
- return false; |
- } |
- virtual void DidFinishRunningTasks() OVERRIDE { |
+ virtual void DidFinishRunningTasks(TaskSet task_set) OVERRIDE { |
raster_worker_pool_->AsRasterizer()->CheckForCompletedTasks(); |
} |
- virtual void DidFinishRunningTasksRequiredForActivation() OVERRIDE {} |
+ virtual TaskSetCollection TasksThatShouldBeForcedToComplete() const OVERRIDE { |
+ return TaskSetCollection(); |
+ } |
void RunMessageLoopUntilAllTasksHaveCompleted() { |
task_graph_runner_->RunUntilIdle(); |