blob: bc5eef2db04c2fc394fb058f342e0c1df0848d97 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
/*
Copyright 2018 Google Inc. All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS-IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
#include "geometrical_acoustics/parallel_for.h"
#include <condition_variable>
#include <memory>
#include <mutex>
#include <vector>
#include "base/logging.h"
#include "utils/task_thread_pool.h"
namespace vraudio {
void ParallelFor(unsigned int num_threads, size_t num_iterations,
const std::function<void(const size_t)>& function) {
TaskThreadPool worker_thread_pool;
CHECK(worker_thread_pool.StartThreadPool(num_threads));
for (size_t i = 0; i < num_iterations; ++i) {
while (!worker_thread_pool.WaitUntilWorkerBecomesAvailable()) {
}
const TaskThreadPool::TaskClosure task = [i, &function]() { function(i); };
CHECK(worker_thread_pool.RunOnWorkerThread(task));
}
worker_thread_pool.StopThreadPool();
}
} // namespace vraudio
|