Expose a SequencedTaskRunner that runs tasks on raster threads.

content/renderer/render_thread_impl.cc

 // Copyright (c) 2012 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/render_thread_impl.h"
 
 #include <algorithm>
 #include <limits>
 #include <map>
 #include <vector>
@@ skipping 248 matching lines @@
   }
 
  private:
   const std::string scheme_;
   const std::string host_;
   const double zoom_level_;
 
   DISALLOW_COPY_AND_ASSIGN(RenderViewZoomer);
 };
 
-class CompositorRasterThread : public base::SimpleThread {
- public:
-  CompositorRasterThread(cc::TaskGraphRunner* task_graph_runner,
-                         const std::string& name_prefix,
-                         base::SimpleThread::Options options)
-      : base::SimpleThread(name_prefix, options),
-        task_graph_runner_(task_graph_runner) {}
-
-  // Overridden from base::SimpleThread:
-  void Run() override { task_graph_runner_->Run(); }
-
- private:
-  cc::TaskGraphRunner* task_graph_runner_;
-
-  DISALLOW_COPY_AND_ASSIGN(CompositorRasterThread);
-};
-
 std::string HostToCustomHistogramSuffix(const std::string& host) {
   if (host == "mail.google.com")
     return ".gmail";
   if (host == "docs.google.com" || host == "drive.google.com")
     return ".docs";
   if (host == "plus.google.com")
     return ".plus";
   if (host == "inbox.google.com")
     return ".inbox";
   return std::string();
@@ skipping 91 matching lines @@
 blink::WebGraphicsContext3D::Attributes GetOffscreenAttribs() {
   blink::WebGraphicsContext3D::Attributes attributes;
   attributes.shareResources = true;
   attributes.depth = false;
   attributes.stencil = false;
   attributes.antialias = false;
   attributes.noAutomaticFlushes = true;
   return attributes;
 }
 
+class RasterWorkerPool : public base::SequencedTaskRunner,
+                         public base::DelegateSimpleThread::Delegate {
+ public:
+  RasterWorkerPool()
+      : namespace_token_(task_graph_runner_.GetNamespaceToken()) {}
+
+  void Start(int num_raster_threads,
+             const base::SimpleThread::Options& thread_options) {
+    DCHECK(raster_threads_.empty());
+    while (raster_threads_.size() < static_cast<size_t>(num_raster_threads)) {
+      scoped_ptr<base::DelegateSimpleThread> raster_thread(
+          new base::DelegateSimpleThread(
+              this, base::StringPrintf("CompositorTileWorker%u",
+                                       static_cast<unsigned>(
+                                           raster_threads_.size() + 1)).c_str(),
+              thread_options));
+      raster_thread->Start();
+      raster_threads_.push_back(raster_thread.Pass());
+    }
+  }
+
+  void Shutdown() {
+    // Shutdown raster threads.
+    task_graph_runner_.Shutdown();
+    while (!raster_threads_.empty()) {
+      raster_threads_.back()->Join();
+      raster_threads_.pop_back();
+    }
+  }
+
+  // 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);
+  }
+
+  // Overridden from base::TaskRunner:

[reveman, 2015/06/30 18:03:52]

nit: remove line 417 and 416 so these functions become a group of functions
affected by the comment on line 410

[Daniele Castagna, 2015/06/30 18:55:25]

Done.

+  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_);
+    DCHECK(!raster_threads_.empty());
+
+    // Remove completed tasks.
+    DCHECK(completed_tasks_.empty());
+    task_graph_runner_.CollectCompletedTasks(namespace_token_,
+                                             &completed_tasks_);
+    DCHECK_LE(completed_tasks_.size(), tasks_.size());
+    DCHECK(std::equal(completed_tasks_.begin(), completed_tasks_.end(),
+                      tasks_.begin()));
+    tasks_.erase(tasks_.begin(), tasks_.begin() + completed_tasks_.size());
+    completed_tasks_.clear();
+
+    tasks_.push_back(make_scoped_refptr(new ClosureTask(task)));
+
+    graph_.Reset();
+    for (const auto& task : tasks_) {
+      cc::TaskGraph::Node node(task.get(), 0, graph_.nodes.size());
+      if (graph_.nodes.size()) {
+        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_);
+    return true;
+  }
+
+  // Overridden from base::DelegateSimpleThread::Delegate:
+  void Run() override { task_graph_runner_.Run(); }
+
+  cc::TaskGraphRunner* GetTaskGraphRunner() { return &task_graph_runner_; }
+
+ protected:
+  ~RasterWorkerPool() override {}
+
+ private:
+  // Simple Task for the TaskGraphRunner that wraps a closure.
+  class ClosureTask : public cc::Task {
+   public:
+    ClosureTask(const base::Closure& closure) : closure_(closure) {}
+    void RunOnWorkerThread() override {

[reveman, 2015/06/30 18:03:52]

nit: blank line after ctor and add: // Overridden from cc:Task:

[Daniele Castagna, 2015/06/30 18:55:24]

Done.

+      closure_.Run();
+      closure_.Reset();
+    };
+
+   protected:
+    ~ClosureTask() override {}
+
+   private:
+    base::Closure closure_;
+
+    DISALLOW_COPY_AND_ASSIGN(ClosureTask);
+  };
+
+  // The actual threads where work is done.
+  ScopedVector<base::DelegateSimpleThread> raster_threads_;

[reveman, 2015/06/30 18:03:52]

nit: just "threads_" as "raster" is implicit from the class name and it would
make a future rasterworkerpool->workerpool transition cleaner.

[Daniele Castagna, 2015/06/30 18:55:24]

Done.

+  cc::TaskGraphRunner task_graph_runner_;
+
+  // Lock to exclusively access all the following members that are used to
+  // implement the SequencedTaskRunner interface.
+  base::Lock lock_;
+  // List of tasks currently queued up for execution.
+  ClosureTask::Vector tasks_;
+  // Cached vector to avoid allocation when getting the list of complete tasks.
+  ClosureTask::Vector completed_tasks_;
+  // Graph object used for scheduling tasks.
+  cc::TaskGraph graph_;
+  // Namespace where the SequencedTaskRunner tasks run.
+  cc::NamespaceToken namespace_token_;

[reveman, 2015/06/30 18:03:52]

nit: can you move this above the lock_ and make it const?

[Daniele Castagna, 2015/06/30 18:55:24]

Done.

+};
+
 }  // namespace
 
 // For measuring memory usage after each task. Behind a command line flag.
 class MemoryObserver : public base::MessageLoop::TaskObserver {
  public:
   MemoryObserver() {}
   ~MemoryObserver() override {}
 
   void WillProcessTask(const base::PendingTask& pending_task) override {}
 
@@ skipping 49 matching lines @@
 }
 
 RenderThreadImpl* RenderThreadImpl::current() {
   return lazy_tls.Pointer()->Get();
 }
 
 RenderThreadImpl::RenderThreadImpl(const InProcessChildThreadParams& params)
     : ChildThreadImpl(Options::Builder()
                           .InBrowserProcess(params)
                           .UseMojoChannel(ShouldUseMojoChannel())
-                          .Build()) {
+                          .Build()),
+      raster_worker_pool_(new RasterWorkerPool()) {
   Init();
 }
 
 // When we run plugins in process, we actually run them on the render thread,
 // which means that we need to make the render thread pump UI events.
 RenderThreadImpl::RenderThreadImpl(
     scoped_ptr<base::MessageLoop> main_message_loop)
     : ChildThreadImpl(Options::Builder()
                           .UseMojoChannel(ShouldUseMojoChannel())
                           .Build()),
-      main_message_loop_(main_message_loop.Pass()) {
+      main_message_loop_(main_message_loop.Pass()),
+      raster_worker_pool_(new RasterWorkerPool()) {
   Init();
 }
 
 void RenderThreadImpl::Init() {
   TRACE_EVENT_BEGIN_ETW("RenderThreadImpl::Init", 0, "");
 
   base::trace_event::TraceLog::GetInstance()->SetThreadSortIndex(
       base::PlatformThread::CurrentId(),
       kTraceEventRendererMainThreadSortIndex);
 
@@ skipping 152 matching lines @@
     is_distance_field_text_enabled_ = false;
   }
 
   // Note that under Linux, the media library will normally already have
   // been initialized by the Zygote before this instance became a Renderer.
   media::InitializeMediaLibrary();
 
   memory_pressure_listener_.reset(new base::MemoryPressureListener(
       base::Bind(&RenderThreadImpl::OnMemoryPressure, base::Unretained(this))));
 
-  compositor_task_graph_runner_.reset(new cc::TaskGraphRunner);
-
   is_gather_pixel_refs_enabled_ = false;
 
   int num_raster_threads = 0;
   std::string string_value =
       command_line.GetSwitchValueASCII(switches::kNumRasterThreads);
   bool parsed_num_raster_threads =
       base::StringToInt(string_value, &num_raster_threads);
   DCHECK(parsed_num_raster_threads) << string_value;
   DCHECK_GT(num_raster_threads, 0);
 
   // Note: Currently, gathering of pixel refs when using a single
   // raster thread doesn't provide any benefit. This might change
   // in the future but we avoid it for now to reduce the cost of
   // Picture::Create.
   is_gather_pixel_refs_enabled_ = num_raster_threads > 1;
 
   base::SimpleThread::Options thread_options;
 #if defined(OS_ANDROID) || defined(OS_LINUX)
   if (!command_line.HasSwitch(
           switches::kUseNormalPriorityForTileTaskWorkerThreads)) {
     thread_options.set_priority(base::ThreadPriority::BACKGROUND);
   }
 #endif
-  while (compositor_raster_threads_.size() <
-         static_cast<size_t>(num_raster_threads)) {
-    scoped_ptr<CompositorRasterThread> raster_thread(new CompositorRasterThread(
-        compositor_task_graph_runner_.get(),
-        base::StringPrintf("CompositorTileWorker%u",
-                           static_cast<unsigned>(
-                               compositor_raster_threads_.size() + 1)).c_str(),
-        thread_options));
-    raster_thread->Start();
-    compositor_raster_threads_.push_back(raster_thread.Pass());
-  }
+
+  raster_worker_pool_->Start(num_raster_threads, thread_options);
 
   // TODO(boliu): In single process, browser main loop should set up the
   // discardable memory manager, and should skip this if kSingleProcess.
   // See crbug.com/503724.
   base::DiscardableMemoryAllocator::SetInstance(
       ChildThreadImpl::discardable_shared_memory_manager());
 
   service_registry()->AddService<RenderFrameSetup>(
       base::Bind(CreateRenderFrameSetup));
 
@@ skipping 58 matching lines @@
   }
 
   media_thread_.reset();
 
   // AudioMessageFilter may be accessed on |media_thread_|, so shutdown after.
   RemoveFilter(audio_message_filter_.get());
   audio_message_filter_ = NULL;
 
   compositor_thread_.reset();
 
-  // Shutdown raster threads.
-  compositor_task_graph_runner_->Shutdown();
-  while (!compositor_raster_threads_.empty()) {
-    compositor_raster_threads_.back()->Join();
-    compositor_raster_threads_.pop_back();
-  }
-  compositor_task_graph_runner_.reset();
+  raster_worker_pool_->Shutdown();
 
   main_input_callback_.Cancel();
   input_handler_manager_.reset();
   if (input_event_filter_.get()) {
     RemoveFilter(input_event_filter_.get());
     input_event_filter_ = NULL;
   }
 
   // RemoveEmbeddedWorkerRoute may be called while deleting
   // EmbeddedWorkerDispatcher. So it must be deleted before deleting
@@ skipping 672 matching lines @@
   if (SynchronousCompositorFactory* factory =
           SynchronousCompositorFactory::GetInstance()) {
     return factory->CreateExternalBeginFrameSource(routing_id);
   }
 #endif
   return make_scoped_ptr(new CompositorExternalBeginFrameSource(
       compositor_message_filter_.get(), sync_message_filter(), routing_id));
 }
 
 cc::TaskGraphRunner* RenderThreadImpl::GetTaskGraphRunner() {
-  return compositor_task_graph_runner_.get();
+  return raster_worker_pool_->GetTaskGraphRunner();
 }
 
 bool RenderThreadImpl::IsGatherPixelRefsEnabled() {
   return is_gather_pixel_refs_enabled_;
 }
 
 bool RenderThreadImpl::IsMainThread() {
   return !!current();
 }
 
@@ skipping 316 matching lines @@
     media_thread_->Start();
 
 #if defined(OS_ANDROID)
     renderer_demuxer_ = new RendererDemuxerAndroid();
     AddFilter(renderer_demuxer_.get());
 #endif
   }
   return media_thread_->task_runner();
 }
 
+base::SequencedTaskRunner* RenderThreadImpl::GetWorkerSequencedTaskRunner() {
+  return raster_worker_pool_.get();
+}
+
 void RenderThreadImpl::SampleGamepads(blink::WebGamepads* data) {
   blink_platform_impl_->sampleGamepads(*data);
 }
 
 bool RenderThreadImpl::RendererIsHidden() const {
   return widget_count_ > 0 && hidden_widget_count_ == widget_count_;
 }
 
 void RenderThreadImpl::WidgetCreated() {
   bool renderer_was_hidden = RendererIsHidden();
@@ skipping 70 matching lines @@
 }
 
 void RenderThreadImpl::PendingRenderFrameConnect::OnConnectionError() {
   size_t erased =
       RenderThreadImpl::current()->pending_render_frame_connects_.erase(
           routing_id_);
   DCHECK_EQ(1u, erased);
 }
 
 }  // namespace content