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 |