Re-land: cc: Cancel and re-prioritize worker pool tasks.

cc/resources/raster_worker_pool.h

 // 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.
 
 #ifndef CC_RESOURCES_RASTER_WORKER_POOL_H_
 #define CC_RESOURCES_RASTER_WORKER_POOL_H_
 
-#include <string>
-
 #include "cc/base/worker_pool.h"
 
 namespace cc {
 class PicturePileImpl;
 
 // A worker thread pool that runs raster tasks.
 class CC_EXPORT RasterWorkerPool : public WorkerPool {
  public:
-  typedef base::Callback<void(PicturePileImpl* picture_pile)> RasterCallback;
+  class Task {
+   public:
+    typedef base::Callback<void(bool)> Reply;
+
+    class Schedule {
+     public:
+      Schedule();
+      ~Schedule();
+
+      void Append(const Task& task);
+      void AppendWithDependencies(const Task& task, Schedule* dependencies);
+
+     private:
+      friend class RasterWorkerPool;
+
+      internal::WorkerPoolTaskGraph task_graph_;
+    };
+
+    Task();
+    Task(const base::Closure& callback, const Reply& reply);
+    ~Task();
+
+    // Returns true if Task is null (doesn't refer to anything).
+    bool is_null() const { return !internal_; }
+
+    // Returns the Task into an uninitialized state.
+    void Reset();
+
+   protected:
+    explicit Task(scoped_refptr<internal::WorkerPoolTask> internal);
+
+    scoped_refptr<internal::WorkerPoolTask> internal_;
+  };
+
+  class PictureTask : public Task {
+   public:
+    typedef base::Callback<void(PicturePileImpl*)> Callback;
+
+    PictureTask(PicturePileImpl* picture_pile,
+                const Callback& callback,
+                const Reply& reply);
+  };
 
   virtual ~RasterWorkerPool();
 
   static scoped_ptr<RasterWorkerPool> Create(size_t num_threads) {
     return make_scoped_ptr(new RasterWorkerPool(num_threads));
   }
 
-  void PostRasterTaskAndReply(PicturePileImpl* picture_pile,
-                              const RasterCallback& task,
-                              const base::Closure& reply);
+  // Tells the worker pool to shutdown after canceling all previously
+  // scheduled tasks. Reply callbacks are still guaranteed to run.
+  virtual void Shutdown() OVERRIDE;
+
+  // Schedule running of tasks in |schedule|. All tasks previously
+  // scheduled but not present in |schedule| will be canceled. Once
+  // scheduled, reply callbacks are guaranteed to run for all tasks even
+  // if they later get canceled by another call to ScheduleTasks(). This
+  // consumed all tasks and queue will be empty when function returns.
+  void ScheduleTasks(Task::Schedule* schedule);
 
  private:
   explicit RasterWorkerPool(size_t num_threads);
 
   DISALLOW_COPY_AND_ASSIGN(RasterWorkerPool);
 };
 
 }  // namespace cc
 
 #endif  // CC_RESOURCES_RASTER_WORKER_POOL_H_