| OLD | NEW |
| 1 // Copyright 2012 The Chromium Authors. All rights reserved. | 1 // Copyright 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "cc/resources/raster_worker_pool.h" | 5 #include "cc/resources/raster_worker_pool.h" |
| 6 | 6 |
| 7 #include "cc/resources/picture_pile_impl.h" | 7 #include "cc/resources/picture_pile_impl.h" |
| 8 | 8 |
| 9 namespace cc { | 9 namespace cc { |
| 10 | 10 |
| 11 namespace { | 11 namespace { |
| 12 | 12 |
| 13 class RasterWorkerPoolContainerTaskImpl : public internal::WorkerPoolTask { | 13 class RasterWorkerPoolTaskImpl : public internal::WorkerPoolTask { |
| 14 public: | 14 public: |
| 15 RasterWorkerPoolContainerTaskImpl( | 15 RasterWorkerPoolTaskImpl(PicturePileImpl* picture_pile, |
| 16 internal::WorkerPoolTask::TaskVector* dependencies) | 16 const RasterWorkerPool::RasterCallback& task, |
| 17 : internal::WorkerPoolTask(dependencies) { | 17 const base::Closure& reply) |
| 18 : internal::WorkerPoolTask(reply), |
| 19 picture_pile_(picture_pile), |
| 20 task_(task) { |
| 21 DCHECK(picture_pile_); |
| 18 } | 22 } |
| 19 | 23 |
| 20 // Overridden from internal::WorkerPoolTask: | |
| 21 virtual void RunOnThread(unsigned thread_index) OVERRIDE {} | |
| 22 virtual void DispatchCompletionCallback() OVERRIDE {} | |
| 23 | |
| 24 private: | |
| 25 virtual ~RasterWorkerPoolContainerTaskImpl() {} | |
| 26 }; | |
| 27 | |
| 28 class RasterWorkerPoolTaskImpl : public internal::WorkerPoolTask { | |
| 29 public: | |
| 30 RasterWorkerPoolTaskImpl(const base::Closure& callback, | |
| 31 const RasterWorkerPool::Task::Reply& reply) | |
| 32 : callback_(callback), | |
| 33 reply_(reply) { | |
| 34 } | |
| 35 | |
| 36 // Overridden from internal::WorkerPoolTask: | |
| 37 virtual void RunOnThread(unsigned thread_index) OVERRIDE { | 24 virtual void RunOnThread(unsigned thread_index) OVERRIDE { |
| 38 callback_.Run(); | 25 task_.Run(picture_pile_->GetCloneForDrawingOnThread(thread_index)); |
| 39 } | |
| 40 virtual void DispatchCompletionCallback() OVERRIDE { | |
| 41 reply_.Run(!HasFinishedRunning()); | |
| 42 } | 26 } |
| 43 | 27 |
| 44 private: | 28 private: |
| 45 virtual ~RasterWorkerPoolTaskImpl() {} | |
| 46 | |
| 47 const base::Closure callback_; | |
| 48 const RasterWorkerPool::Task::Reply reply_; | |
| 49 }; | |
| 50 | |
| 51 class RasterWorkerPoolPictureTaskImpl : public internal::WorkerPoolTask { | |
| 52 public: | |
| 53 RasterWorkerPoolPictureTaskImpl( | |
| 54 PicturePileImpl* picture_pile, | |
| 55 const RasterWorkerPool::PictureTask::Callback& callback, | |
| 56 const RasterWorkerPool::Task::Reply& reply, | |
| 57 internal::WorkerPoolTask::TaskVector* dependencies) | |
| 58 : internal::WorkerPoolTask(dependencies), | |
| 59 picture_pile_(picture_pile), | |
| 60 callback_(callback), | |
| 61 reply_(reply) { | |
| 62 DCHECK(picture_pile_); | |
| 63 } | |
| 64 | |
| 65 // Overridden from internal::WorkerPoolTask: | |
| 66 virtual void RunOnThread(unsigned thread_index) OVERRIDE { | |
| 67 callback_.Run(picture_pile_->GetCloneForDrawingOnThread(thread_index)); | |
| 68 } | |
| 69 virtual void DispatchCompletionCallback() OVERRIDE { | |
| 70 reply_.Run(!HasFinishedRunning()); | |
| 71 } | |
| 72 | |
| 73 private: | |
| 74 virtual ~RasterWorkerPoolPictureTaskImpl() {} | |
| 75 | |
| 76 scoped_refptr<PicturePileImpl> picture_pile_; | 29 scoped_refptr<PicturePileImpl> picture_pile_; |
| 77 const RasterWorkerPool::PictureTask::Callback callback_; | 30 RasterWorkerPool::RasterCallback task_; |
| 78 const RasterWorkerPool::Task::Reply reply_; | |
| 79 }; | 31 }; |
| 80 | 32 |
| 81 const char* kWorkerThreadNamePrefix = "CompositorRaster"; | 33 const char* kWorkerThreadNamePrefix = "CompositorRaster"; |
| 82 | 34 |
| 83 const int kCheckForCompletedTasksDelayMs = 6; | 35 const int kCheckForCompletedTasksDelayMs = 6; |
| 84 | 36 |
| 85 } // namespace | 37 } // namespace |
| 86 | 38 |
| 87 RasterWorkerPool::Task::Queue::Queue() { | 39 RasterWorkerPool::RasterWorkerPool(size_t num_threads) |
| 88 } | 40 : WorkerPool( |
| 89 | 41 num_threads, |
| 90 RasterWorkerPool::Task::Queue::~Queue() { | 42 base::TimeDelta::FromMilliseconds(kCheckForCompletedTasksDelayMs), |
| 91 } | 43 kWorkerThreadNamePrefix) { |
| 92 | |
| 93 void RasterWorkerPool::Task::Queue::Append(const Task& task) { | |
| 94 DCHECK(!task.is_null()); | |
| 95 tasks_.push_back(task.internal_); | |
| 96 } | |
| 97 | |
| 98 RasterWorkerPool::Task::Task() { | |
| 99 } | |
| 100 | |
| 101 RasterWorkerPool::Task::Task(const base::Closure& callback, | |
| 102 const Reply& reply) | |
| 103 : internal_(new RasterWorkerPoolTaskImpl(callback, reply)) { | |
| 104 } | |
| 105 | |
| 106 RasterWorkerPool::Task::Task(Queue* dependencies) | |
| 107 : internal_(new RasterWorkerPoolContainerTaskImpl(&dependencies->tasks_)) { | |
| 108 } | |
| 109 | |
| 110 RasterWorkerPool::Task::Task(scoped_refptr<internal::WorkerPoolTask> internal) | |
| 111 : internal_(internal) { | |
| 112 } | |
| 113 | |
| 114 RasterWorkerPool::Task::~Task() { | |
| 115 } | |
| 116 | |
| 117 void RasterWorkerPool::Task::Reset() { | |
| 118 internal_ = NULL; | |
| 119 } | |
| 120 | |
| 121 RasterWorkerPool::PictureTask::PictureTask(PicturePileImpl* picture_pile, | |
| 122 const Callback& callback, | |
| 123 const Reply& reply, | |
| 124 Task::Queue* dependencies) | |
| 125 : RasterWorkerPool::Task( | |
| 126 new RasterWorkerPoolPictureTaskImpl(picture_pile, | |
| 127 callback, | |
| 128 reply, | |
| 129 &dependencies->tasks_)) { | |
| 130 } | |
| 131 | |
| 132 RasterWorkerPool::RasterWorkerPool(size_t num_threads) : WorkerPool( | |
| 133 num_threads, | |
| 134 base::TimeDelta::FromMilliseconds(kCheckForCompletedTasksDelayMs), | |
| 135 kWorkerThreadNamePrefix) { | |
| 136 } | 44 } |
| 137 | 45 |
| 138 RasterWorkerPool::~RasterWorkerPool() { | 46 RasterWorkerPool::~RasterWorkerPool() { |
| 139 } | 47 } |
| 140 | 48 |
| 141 void RasterWorkerPool::Shutdown() { | 49 void RasterWorkerPool::PostRasterTaskAndReply(PicturePileImpl* picture_pile, |
| 142 // Cancel all previously scheduled tasks. | 50 const RasterCallback& task, |
| 143 WorkerPool::ScheduleTasks(NULL); | 51 const base::Closure& reply) { |
| 144 | 52 PostTask(make_scoped_ptr(new RasterWorkerPoolTaskImpl( |
| 145 WorkerPool::Shutdown(); | 53 picture_pile, |
| 146 } | 54 task, |
| 147 | 55 reply)).PassAs<internal::WorkerPoolTask>()); |
| 148 void RasterWorkerPool::ScheduleTasks(Task* task) { | |
| 149 WorkerPool::ScheduleTasks(task ? task->internal_ : NULL); | |
| 150 } | 56 } |
| 151 | 57 |
| 152 } // namespace cc | 58 } // namespace cc |
| OLD | NEW |