src/cc/resources/pixel_buffer_raster_worker_pool.h

   1 // Copyright 2013 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.
   4
   5 #ifndef CC_RESOURCES_PIXEL_BUFFER_RASTER_WORKER_POOL_H_
   6 #define CC_RESOURCES_PIXEL_BUFFER_RASTER_WORKER_POOL_H_
   7
   8 #include <deque>
   9
  10 #include "base/containers/hash_tables.h"
  11 #include "cc/resources/raster_worker_pool.h"
  12
  13 #if defined(COMPILER_GCC)
  14 namespace BASE_HASH_NAMESPACE {
  15 template <>
  16 struct hash<cc::internal::RasterWorkerPoolTask*> {
  17   size_t operator()(cc::internal::RasterWorkerPoolTask* ptr) const {
  18     return hash<size_t>()(reinterpret_cast<size_t>(ptr));
  19   }
  20 };
  21 }  // namespace BASE_HASH_NAMESPACE
  22 #endif  // COMPILER
  23
  24 namespace cc {
  25
  26 class CC_EXPORT PixelBufferRasterWorkerPool : public RasterWorkerPool {
  27  public:
  28   virtual ~PixelBufferRasterWorkerPool();
  29
  30   static scoped_ptr<RasterWorkerPool> Create(
  31       ResourceProvider* resource_provider,
  32       size_t max_transfer_buffer_usage_bytes);
  33
  34   // Overridden from RasterWorkerPool:
  35   virtual void Shutdown() OVERRIDE;
  36   virtual void ScheduleTasks(RasterTask::Queue* queue) OVERRIDE;
  37   virtual unsigned GetResourceTarget() const OVERRIDE;
  38   virtual ResourceFormat GetResourceFormat() const OVERRIDE;
  39   virtual void CheckForCompletedTasks() OVERRIDE;
  40
  41   // Overridden from internal::WorkerPoolTaskClient:
  42   virtual SkCanvas* AcquireCanvasForRaster(internal::RasterWorkerPoolTask* task)
  43       OVERRIDE;
  44   virtual void OnRasterCompleted(internal::RasterWorkerPoolTask* task,
  45                                  const PicturePileImpl::Analysis& analysis)
  46       OVERRIDE;
  47   virtual void OnImageDecodeCompleted(internal::WorkerPoolTask* task) OVERRIDE;
  48
  49  protected:
  50   PixelBufferRasterWorkerPool(internal::TaskGraphRunner* task_graph_runner,
  51                               ResourceProvider* resource_provider,
  52                               size_t max_transfer_buffer_usage_bytes);
  53
  54  private:
  55   enum RasterTaskState { UNSCHEDULED, SCHEDULED, UPLOADING, COMPLETED };
  56   typedef std::deque<scoped_refptr<internal::RasterWorkerPoolTask> >
  57       RasterTaskDeque;
  58   typedef internal::RasterWorkerPoolTask* RasterTaskMapKey;
  59   typedef base::hash_set<RasterTaskMapKey> RasterTaskSet;
  60   typedef base::hash_map<RasterTaskMapKey, RasterTaskState> RasterTaskStateMap;
  61
  62   // Overridden from RasterWorkerPool:
  63   virtual void OnRasterTasksFinished() OVERRIDE;
  64   virtual void OnRasterTasksRequiredForActivationFinished() OVERRIDE;
  65
  66   void FlushUploads();
  67   void CheckForCompletedUploads();
  68   void ScheduleCheckForCompletedRasterTasks();
  69   void OnCheckForCompletedRasterTasks();
  70   void CheckForCompletedRasterTasks();
  71   void ScheduleMoreTasks();
  72   unsigned PendingRasterTaskCount() const;
  73   bool HasPendingTasks() const;
  74   bool HasPendingTasksRequiredForActivation() const;
  75   void CheckForCompletedWorkerPoolTasks();
  76   bool IsRasterTaskRequiredForActivation(internal::RasterWorkerPoolTask* task)
  77       const;
  78
  79   const char* StateName() const;
  80   scoped_ptr<base::Value> StateAsValue() const;
  81   scoped_ptr<base::Value> ThrottleStateAsValue() const;
  82
  83   bool shutdown_;
  84
  85   RasterTask::Queue raster_tasks_;
  86   RasterTaskSet raster_tasks_required_for_activation_;
  87   RasterTaskStateMap raster_task_states_;
  88   RasterTaskDeque raster_tasks_with_pending_upload_;
  89   RasterTaskDeque completed_raster_tasks_;
  90   TaskDeque completed_image_decode_tasks_;
  91
  92   size_t scheduled_raster_task_count_;
  93   size_t bytes_pending_upload_;
  94   size_t max_bytes_pending_upload_;
  95   bool has_performed_uploads_since_last_flush_;
  96   base::TimeTicks check_for_completed_raster_tasks_time_;
  97   bool check_for_completed_raster_tasks_pending_;
  98
  99   bool should_notify_client_if_no_tasks_are_pending_;
 100   bool should_notify_client_if_no_tasks_required_for_activation_are_pending_;
 101   bool raster_finished_task_pending_;
 102   bool raster_required_for_activation_finished_task_pending_;
 103
 104   // Task graph used when scheduling tasks and vector used to gather
 105   // completed tasks.
 106   internal::TaskGraph graph_;
 107   internal::Task::Vector completed_tasks_;
 108
 109   base::WeakPtrFactory<PixelBufferRasterWorkerPool> weak_factory_;
 110
 111   DISALLOW_COPY_AND_ASSIGN(PixelBufferRasterWorkerPool);
 112 };
 113
 114 }  // namespace cc
 115
 116 #endif  // CC_RESOURCES_PIXEL_BUFFER_RASTER_WORKER_POOL_H_