PulseStudio/VoxelEngine
PulseStudio/VoxelEngine
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

add ThreadPool maxWorkers constructor argument

Browse Source
This commit is contained in:
MihailRis 2024-10-01 23:24:01 +03:00
parent 572ae9fa39
commit a65cf6ed66
1 changed files with 29 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/util/ThreadPool.hpp
View File

@ -103,14 +103,40 @@ namespace util {
}
}
public:
static constexpr int UNLIMITED = 0;
static constexpr int HALF = -2;
static constexpr int QUARTER = -4;
/// @brief Main thread pool constructor
/// @param name thread pool name (used in logger)
/// @param workersSupplier workers factory function
/// @param resultConsumer workers results consumer function
/// @param maxWorkers max number of workers. Special values: 0 is
/// unlimited, -2 is half of auto count, -4 is quarter.
ThreadPool(
std::string name,
supplier<std::shared_ptr<Worker<T, R>>> workersSupplier,
consumer<R&> resultConsumer
consumer<R&> resultConsumer,
int maxWorkers=UNLIMITED
)
: logger(std::move(name)), resultConsumer(resultConsumer) {
const uint num_threads = std::thread::hardware_concurrency();
for (uint i = 0; i < num_threads; i++) {
uint numThreads = std::thread::hardware_concurrency();
switch (maxWorkers) {
case UNLIMITED:
break;
case HALF:
numThreads = std::max(1U, numThreads);
break;
case QUARTER:
numThreads = std::max(1U, numThreads / 4);
break;
default:
numThreads = std::max(
1U, std::min(numThreads, static_cast<uint>(maxWorkers))
);
break;
}
for (uint i = 0; i < numThreads; i++) {
threads.emplace_back(
&ThreadPool<T, R>::threadLoop, this, i, workersSupplier()
);