cc: Move GPU raster to DirectRasterWorkerPool.

cc/resources/raster_worker_pool_perftest.cc

 // Copyright 2013 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 "cc/resources/raster_worker_pool.h"
 
 #include "base/time/time.h"
+#include "cc/resources/direct_raster_worker_pool.h"
 #include "cc/resources/image_raster_worker_pool.h"
 #include "cc/resources/pixel_buffer_raster_worker_pool.h"
 #include "cc/resources/resource_provider.h"
 #include "cc/resources/scoped_resource.h"
 #include "cc/test/fake_output_surface.h"
 #include "cc/test/fake_output_surface_client.h"
 #include "cc/test/lap_timer.h"
 #include "cc/test/test_web_graphics_context_3d.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/perf/perf_test.h"
 #include "third_party/khronos/GLES2/gl2.h"
 
 namespace cc {
 namespace {
 
 enum RasterWorkerPoolType {
   RASTER_WORKER_POOL_TYPE_PIXEL_BUFFER,
-  RASTER_WORKER_POOL_TYPE_IMAGE
+  RASTER_WORKER_POOL_TYPE_IMAGE,
+  RASTER_WORKER_POOL_TYPE_DIRECT
 };
 
 static const int kTimeLimitMillis = 2000;
 static const int kWarmupRuns = 5;
 static const int kTimeCheckInterval = 10;
 
 class PerfWorkerPoolTaskImpl : public internal::WorkerPoolTask {
  public:
   PerfWorkerPoolTaskImpl() {}
 
   // Overridden from internal::Task:
   virtual void RunOnWorkerThread(unsigned thread_index) OVERRIDE {}
 
   // Overridden from internal::WorkerPoolTask:
   virtual void ScheduleOnOriginThread(internal::WorkerPoolTaskClient* client)
       OVERRIDE {}
+  virtual void RunOnOriginThread() OVERRIDE {}
   virtual void CompleteOnOriginThread(internal::WorkerPoolTaskClient* client)
       OVERRIDE {}
   virtual void RunReplyOnOriginThread() OVERRIDE { Reset(); }
 
   void Reset() {
     did_run_ = false;
     did_complete_ = false;
   }
 
  protected:
   virtual ~PerfWorkerPoolTaskImpl() {}
 
  private:
   DISALLOW_COPY_AND_ASSIGN(PerfWorkerPoolTaskImpl);
 };
 
 class PerfRasterWorkerPoolTaskImpl : public internal::RasterWorkerPoolTask {
  public:
   PerfRasterWorkerPoolTaskImpl(scoped_ptr<ScopedResource> resource,
-                               internal::Task::Vector* dependencies)
-      : internal::RasterWorkerPoolTask(resource.get(), dependencies, false),
+                               internal::WorkerPoolTask::Vector* dependencies)
+      : internal::RasterWorkerPoolTask(resource.get(), dependencies),
         resource_(resource.Pass()) {}
 
   // Overridden from internal::Task:
   virtual void RunOnWorkerThread(unsigned thread_index) OVERRIDE {}
 
   // Overridden from internal::WorkerPoolTask:
   virtual void ScheduleOnOriginThread(internal::WorkerPoolTaskClient* client)
       OVERRIDE {
     client->AcquireCanvasForRaster(this);
   }
+  virtual void RunOnOriginThread() OVERRIDE {}
   virtual void CompleteOnOriginThread(internal::WorkerPoolTaskClient* client)
       OVERRIDE {
     client->OnRasterCompleted(this, PicturePileImpl::Analysis());
   }
   virtual void RunReplyOnOriginThread() OVERRIDE { Reset(); }
 
-  // Overridden from internal::RasterWorkerPoolTask:
-  virtual void RunOnOriginThread(ResourceProvider* resource_provider,
-                                 ContextProvider* context_provider) OVERRIDE {}
-
   void Reset() {
     did_run_ = false;
     did_complete_ = false;
   }
 
  protected:
   virtual ~PerfRasterWorkerPoolTaskImpl() {}
 
  private:
   scoped_ptr<ScopedResource> resource_;
 
   DISALLOW_COPY_AND_ASSIGN(PerfRasterWorkerPoolTaskImpl);
 };
 
 class PerfTaskGraphRunnerImpl : public internal::TaskGraphRunner {
  public:
   PerfTaskGraphRunnerImpl() : internal::TaskGraphRunner(0, "Perf") {}
 };
 
 class PerfPixelBufferRasterWorkerPoolImpl : public PixelBufferRasterWorkerPool {
  public:
   PerfPixelBufferRasterWorkerPoolImpl(
       internal::TaskGraphRunner* task_graph_runner,
       ResourceProvider* resource_provider)
       : PixelBufferRasterWorkerPool(task_graph_runner,
                                     resource_provider,
-                                    NULL,
                                     std::numeric_limits<size_t>::max()) {}
 };
 
 class PerfImageRasterWorkerPoolImpl : public ImageRasterWorkerPool {
  public:
   PerfImageRasterWorkerPoolImpl(internal::TaskGraphRunner* task_graph_runner,
                                 ResourceProvider* resource_provider)
       : ImageRasterWorkerPool(task_graph_runner,
                               resource_provider,
-                              NULL,
                               GL_TEXTURE_2D) {}
 };
 
+class PerfDirectRasterWorkerPoolImpl : public DirectRasterWorkerPool {
+ public:
+  PerfDirectRasterWorkerPoolImpl(ResourceProvider* resource_provider,
+                                 ContextProvider* context_provider)
+      : DirectRasterWorkerPool(resource_provider, context_provider) {}
+};
+
 class RasterWorkerPoolPerfTest
     : public testing::TestWithParam<RasterWorkerPoolType>,
       public RasterWorkerPoolClient {
  public:
   class Task : public RasterWorkerPool::Task {
    public:
     typedef std::vector<Task> Vector;
 
     static Task Create() { return Task(new PerfWorkerPoolTaskImpl); }
 
-    void AppendTo(internal::Task::Vector* dependencies) const {
+    void AppendTo(internal::WorkerPoolTask::Vector* dependencies) const {
       dependencies->push_back(internal_);
     }
 
    private:
     explicit Task(internal::WorkerPoolTask* task)
         : RasterWorkerPool::Task(task) {}
   };
 
   class RasterTask : public RasterWorkerPool::RasterTask {
    public:
     typedef std::vector<RasterTask> Vector;
 
     static RasterTask Create(scoped_ptr<ScopedResource> resource,
                              const Task::Vector& image_decode_tasks) {
-      internal::Task::Vector dependencies;
+      internal::WorkerPoolTask::Vector dependencies;
       for (Task::Vector::const_iterator it = image_decode_tasks.begin();
            it != image_decode_tasks.end();
            ++it)
         it->AppendTo(&dependencies);
 
       return RasterTask(
           new PerfRasterWorkerPoolTaskImpl(resource.Pass(), &dependencies));
     }
 
    private:
@@ skipping 15 matching lines @@
 
     switch (GetParam()) {
       case RASTER_WORKER_POOL_TYPE_PIXEL_BUFFER:
         raster_worker_pool_.reset(new PerfPixelBufferRasterWorkerPoolImpl(
             task_graph_runner_.get(), resource_provider_.get()));
         break;
       case RASTER_WORKER_POOL_TYPE_IMAGE:
         raster_worker_pool_.reset(new PerfImageRasterWorkerPoolImpl(
             task_graph_runner_.get(), resource_provider_.get()));
         break;
+      case RASTER_WORKER_POOL_TYPE_DIRECT:
+        raster_worker_pool_.reset(new PerfDirectRasterWorkerPoolImpl(
+            resource_provider_.get(), context_provider_));
+        break;
     }
 
     DCHECK(raster_worker_pool_);
     raster_worker_pool_->SetClient(this);
   }
   virtual ~RasterWorkerPoolPerfTest() { resource_provider_.reset(); }
 
   // Overridden from testing::Test:
   virtual void TearDown() OVERRIDE {
     raster_worker_pool_->Shutdown();
     raster_worker_pool_->CheckForCompletedTasks();
   }
 
   // Overriden from RasterWorkerPoolClient:
   virtual bool ShouldForceTasksRequiredForActivationToComplete()
       const OVERRIDE {
     return false;
   }
-  virtual void DidFinishRunningTasks() OVERRIDE {}
+  virtual void DidFinishRunningTasks() OVERRIDE {
+    raster_worker_pool_->CheckForCompletedTasks();
+    base::MessageLoop::current()->Quit();
+  }
   virtual void DidFinishRunningTasksRequiredForActivation() OVERRIDE {}
 
+  void RunMessageLoopUntilAllTasksHaveCompleted() {
+    while (task_graph_runner_->RunTaskForTesting())
+      continue;
+    base::MessageLoop::current()->Run();
+  }
+
   void CreateImageDecodeTasks(unsigned num_image_decode_tasks,
                               Task::Vector* image_decode_tasks) {
     for (unsigned i = 0; i < num_image_decode_tasks; ++i)
       image_decode_tasks->push_back(Task::Create());
   }
 
   void CreateRasterTasks(unsigned num_raster_tasks,
                          const Task::Vector& image_decode_tasks,
                          RasterTask::Vector* raster_tasks) {
     const gfx::Size size(1, 1);
 
     for (unsigned i = 0; i < num_raster_tasks; ++i) {
       scoped_ptr<ScopedResource> resource(
           ScopedResource::Create(resource_provider_.get()));
       resource->Allocate(size, ResourceProvider::TextureUsageAny, RGBA_8888);
 
       raster_tasks->push_back(
           RasterTask::Create(resource.Pass(), image_decode_tasks));
     }
   }
 
-  void AppendRasterTasks(RasterWorkerPool::RasterTask::Queue* tasks,
-                         const RasterTask::Vector& raster_tasks) {
+  void BuildTaskQueue(RasterWorkerPool::RasterTask::Queue* tasks,
+                      const RasterTask::Vector& raster_tasks) {
     for (RasterTask::Vector::const_iterator it = raster_tasks.begin();
          it != raster_tasks.end();
          ++it)
       tasks->Append(*it, false);
   }
 
   void RunScheduleTasksTest(const std::string& test_name,
                             unsigned num_raster_tasks,
                             unsigned num_image_decode_tasks) {
     Task::Vector image_decode_tasks;
     RasterTask::Vector raster_tasks;
     CreateImageDecodeTasks(num_image_decode_tasks, &image_decode_tasks);
     CreateRasterTasks(num_raster_tasks, image_decode_tasks, &raster_tasks);
 
     // Avoid unnecessary heap allocations by reusing the same queue.
-    RasterWorkerPool::RasterTask::Queue tasks;
+    RasterWorkerPool::RasterTask::Queue queue;
 
     timer_.Reset();
     do {
-      tasks.Reset();
-      AppendRasterTasks(&tasks, raster_tasks);
-      raster_worker_pool_->ScheduleTasks(&tasks);
+      queue.Reset();
+      BuildTaskQueue(&queue, raster_tasks);
+      raster_worker_pool_->ScheduleTasks(&queue);
       raster_worker_pool_->CheckForCompletedTasks();
       timer_.NextLap();
     } while (!timer_.HasTimeLimitExpired());
 
     RasterWorkerPool::RasterTask::Queue empty;
     raster_worker_pool_->ScheduleTasks(&empty);
-    raster_worker_pool_->CheckForCompletedTasks();
+    RunMessageLoopUntilAllTasksHaveCompleted();
 
     perf_test::PrintResult("schedule_tasks",
                            "",
                            test_name,
                            timer_.LapsPerSecond(),
                            "runs/s",
                            true);
   }
 
   void RunScheduleAlternateTasksTest(const std::string& test_name,
                                      unsigned num_raster_tasks,
                                      unsigned num_image_decode_tasks) {
     const size_t kNumVersions = 2;
     Task::Vector image_decode_tasks[kNumVersions];
     RasterTask::Vector raster_tasks[kNumVersions];
     for (size_t i = 0; i < kNumVersions; ++i) {
       CreateImageDecodeTasks(num_image_decode_tasks, &image_decode_tasks[i]);
       CreateRasterTasks(
           num_raster_tasks, image_decode_tasks[i], &raster_tasks[i]);
     }
 
     // Avoid unnecessary heap allocations by reusing the same queue.
-    RasterWorkerPool::RasterTask::Queue tasks;
+    RasterWorkerPool::RasterTask::Queue queue;
 
     size_t count = 0;
     timer_.Reset();
     do {
-      tasks.Reset();
-      AppendRasterTasks(&tasks, raster_tasks[count % kNumVersions]);
-      raster_worker_pool_->ScheduleTasks(&tasks);
+      queue.Reset();
+      BuildTaskQueue(&queue, raster_tasks[count % kNumVersions]);
+      raster_worker_pool_->ScheduleTasks(&queue);
       raster_worker_pool_->CheckForCompletedTasks();
       ++count;
       timer_.NextLap();
     } while (!timer_.HasTimeLimitExpired());
 
     RasterWorkerPool::RasterTask::Queue empty;
     raster_worker_pool_->ScheduleTasks(&empty);
-    raster_worker_pool_->CheckForCompletedTasks();
+    RunMessageLoopUntilAllTasksHaveCompleted();
 
     perf_test::PrintResult("schedule_alternate_tasks",
                            "",
                            test_name,
                            timer_.LapsPerSecond(),
                            "runs/s",
                            true);
   }
 
   void RunScheduleAndExecuteTasksTest(const std::string& test_name,
                                       unsigned num_raster_tasks,
                                       unsigned num_image_decode_tasks) {
     Task::Vector image_decode_tasks;
     RasterTask::Vector raster_tasks;
     CreateImageDecodeTasks(num_image_decode_tasks, &image_decode_tasks);
     CreateRasterTasks(num_raster_tasks, image_decode_tasks, &raster_tasks);
 
     // Avoid unnecessary heap allocations by reusing the same queue.
-    RasterWorkerPool::RasterTask::Queue tasks;
+    RasterWorkerPool::RasterTask::Queue queue;
 
     timer_.Reset();
     do {
-      tasks.Reset();
-      AppendRasterTasks(&tasks, raster_tasks);
-      raster_worker_pool_->ScheduleTasks(&tasks);
-      while (task_graph_runner_->RunTaskForTesting())
-        continue;
-      raster_worker_pool_->CheckForCompletedTasks();
+      queue.Reset();
+      BuildTaskQueue(&queue, raster_tasks);
+      raster_worker_pool_->ScheduleTasks(&queue);
+      RunMessageLoopUntilAllTasksHaveCompleted();
       timer_.NextLap();
     } while (!timer_.HasTimeLimitExpired());
 
     RasterWorkerPool::RasterTask::Queue empty;
     raster_worker_pool_->ScheduleTasks(&empty);
-    raster_worker_pool_->CheckForCompletedTasks();
+    RunMessageLoopUntilAllTasksHaveCompleted();
 
     perf_test::PrintResult("schedule_and_execute_tasks",
                            "",
                            test_name,
                            timer_.LapsPerSecond(),
                            "runs/s",
                            true);
   }
 
  private:
   scoped_refptr<TestContextProvider> context_provider_;
   FakeOutputSurfaceClient output_surface_client_;
   scoped_ptr<FakeOutputSurface> output_surface_;
   scoped_ptr<ResourceProvider> resource_provider_;
   scoped_ptr<internal::TaskGraphRunner> task_graph_runner_;
   scoped_ptr<RasterWorkerPool> raster_worker_pool_;
-  std::vector<RasterWorkerPool::RasterTask> tasks_;
   LapTimer timer_;
 };
 
 TEST_P(RasterWorkerPoolPerfTest, ScheduleTasks) {
   RunScheduleTasksTest("1_0", 1, 0);
   RunScheduleTasksTest("32_0", 32, 0);
   RunScheduleTasksTest("1_1", 1, 1);
   RunScheduleTasksTest("32_1", 32, 1);
   RunScheduleTasksTest("1_4", 1, 4);
   RunScheduleTasksTest("32_4", 32, 4);
@@ skipping 13 matching lines @@
   RunScheduleAndExecuteTasksTest("32_0", 32, 0);
   RunScheduleAndExecuteTasksTest("1_1", 1, 1);
   RunScheduleAndExecuteTasksTest("32_1", 32, 1);
   RunScheduleAndExecuteTasksTest("1_4", 1, 4);
   RunScheduleAndExecuteTasksTest("32_4", 32, 4);
 }
 
 INSTANTIATE_TEST_CASE_P(RasterWorkerPoolPerfTests,
                         RasterWorkerPoolPerfTest,
                         ::testing::Values(RASTER_WORKER_POOL_TYPE_PIXEL_BUFFER,
-                                          RASTER_WORKER_POOL_TYPE_IMAGE));
+                                          RASTER_WORKER_POOL_TYPE_IMAGE,
+                                          RASTER_WORKER_POOL_TYPE_DIRECT));
 
 }  // namespace
 }  // namespace cc