Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(16)

Side by Side Diff: content/renderer/raster_worker_pool.h

Issue 2021323002: content: Rename RasterWorkerPool -> CategorizedWorkerPool. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@task_graph_runner_test_2
Patch Set: rebase Created 4 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 // Copyright 2015 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 CONTENT_RENDERER_RASTER_WORKER_POOL_H_
6 #define CONTENT_RENDERER_RASTER_WORKER_POOL_H_
7
8 #include <memory>
9
10 #include "base/callback.h"
11 #include "base/containers/hash_tables.h"
12 #include "base/macros.h"
13 #include "base/sequenced_task_runner.h"
14 #include "base/synchronization/condition_variable.h"
15 #include "base/task_runner.h"
16 #include "base/threading/simple_thread.h"
17 #include "cc/raster/task_category.h"
18 #include "cc/raster/task_graph_runner.h"
19 #include "cc/raster/task_graph_work_queue.h"
20 #include "content/common/content_export.h"
21
22 namespace content {
23
24 // A pool of threads used to run raster work.
25 // Work can be scheduled on the threads using different interfaces.
26 // The pool itself implements TaskRunner interface and tasks posted via that
27 // interface might run in parallel.
28 // CreateSequencedTaskRunner creates a sequenced task runner that might run in
29 // parallel with other instances of sequenced task runners.
30 // It's also possible to get the underlying TaskGraphRunner to schedule a graph
31 // of tasks with their dependencies.
32 class CONTENT_EXPORT RasterWorkerPool : public base::TaskRunner,
33 public cc::TaskGraphRunner {
34 public:
35 RasterWorkerPool();
36
37 // Overridden from base::TaskRunner:
38 bool PostDelayedTask(const tracked_objects::Location& from_here,
39 const base::Closure& task,
40 base::TimeDelta delay) override;
41 bool RunsTasksOnCurrentThread() const override;
42
43 // Overridden from cc::TaskGraphRunner:
44 cc::NamespaceToken GetNamespaceToken() override;
45 void ScheduleTasks(cc::NamespaceToken token, cc::TaskGraph* graph) override;
46 void WaitForTasksToFinishRunning(cc::NamespaceToken token) override;
47 void CollectCompletedTasks(cc::NamespaceToken token,
48 cc::Task::Vector* completed_tasks) override;
49
50 // Runs a task from one of the provided categories. Categories listed first
51 // have higher priority.
52 void Run(const std::vector<cc::TaskCategory>& categories,
53 base::ConditionVariable* has_ready_to_run_tasks_cv);
54
55 void FlushForTesting();
56
57 // Spawn |num_threads| number of threads and start running work on the
58 // worker threads.
59 void Start(int num_threads);
60
61 // Finish running all the posted tasks (and nested task posted by those tasks)
62 // of all the associated task runners.
63 // Once all the tasks are executed the method blocks until the threads are
64 // terminated.
65 void Shutdown();
66
67 cc::TaskGraphRunner* GetTaskGraphRunner() { return this; }
68
69 // Create a new sequenced task graph runner.
70 scoped_refptr<base::SequencedTaskRunner> CreateSequencedTaskRunner();
71
72 protected:
73 ~RasterWorkerPool() override;
74
75 private:
76 class RasterWorkerPoolSequencedTaskRunner;
77 friend class RasterWorkerPoolSequencedTaskRunner;
78
79 // Simple Task for the TaskGraphRunner that wraps a closure.
80 // This class is used to schedule TaskRunner tasks on the
81 // |task_graph_runner_|.
82 class ClosureTask : public cc::Task {
83 public:
84 explicit ClosureTask(const base::Closure& closure);
85
86 // Overridden from cc::Task:
87 void RunOnWorkerThread() override;
88
89 protected:
90 ~ClosureTask() override;
91
92 private:
93 base::Closure closure_;
94
95 DISALLOW_COPY_AND_ASSIGN(ClosureTask);
96 };
97
98 void ScheduleTasksWithLockAcquired(cc::NamespaceToken token,
99 cc::TaskGraph* graph);
100 void CollectCompletedTasksWithLockAcquired(cc::NamespaceToken token,
101 cc::Task::Vector* completed_tasks);
102
103 // Runs a task from one of the provided categories. Categories listed first
104 // have higher priority. Returns false if there were no tasks to run.
105 bool RunTaskWithLockAcquired(const std::vector<cc::TaskCategory>& categories);
106
107 // Run next task for the given category. Caller must acquire |lock_| prior to
108 // calling this function and make sure at least one task is ready to run.
109 void RunTaskInCategoryWithLockAcquired(cc::TaskCategory category);
110
111 // Helper function which signals worker threads if tasks are ready to run.
112 void SignalHasReadyToRunTasksWithLockAcquired();
113
114 // Determines if we should run a new task for the given category. This factors
115 // in whether a task is available and whether the count of running tasks is
116 // low enough to start a new one.
117 bool ShouldRunTaskForCategoryWithLockAcquired(cc::TaskCategory category);
118
119 // The actual threads where work is done.
120 std::vector<std::unique_ptr<base::SimpleThread>> threads_;
121
122 // Lock to exclusively access all the following members that are used to
123 // implement the TaskRunner and TaskGraphRunner interfaces.
124 base::Lock lock_;
125 // Stores the tasks to be run, sorted by priority.
126 cc::TaskGraphWorkQueue work_queue_;
127 // Namespace used to schedule tasks in the task graph runner.
128 cc::NamespaceToken namespace_token_;
129 // List of tasks currently queued up for execution.
130 cc::Task::Vector tasks_;
131 // Graph object used for scheduling tasks.
132 cc::TaskGraph graph_;
133 // Cached vector to avoid allocation when getting the list of complete
134 // tasks.
135 cc::Task::Vector completed_tasks_;
136 // Condition variables for foreground and background tasks.
137 base::ConditionVariable has_ready_to_run_foreground_tasks_cv_;
138 base::ConditionVariable has_ready_to_run_background_tasks_cv_;
139 // Condition variable that is waited on by origin threads until a namespace
140 // has finished running all associated tasks.
141 base::ConditionVariable has_namespaces_with_finished_running_tasks_cv_;
142 // Set during shutdown. Tells Run() to return when no more tasks are pending.
143 bool shutdown_;
144 };
145
146 } // namespace content
147
148 #endif // CONTENT_RENDERER_RASTER_WORKER_POOL_H_
OLDNEW
« no previous file with comments | « content/renderer/categorized_worker_pool_unittest.cc ('k') | content/renderer/raster_worker_pool.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698