voxelcoreprojects/voxelcore
1
0
Fork 0
mirror of https://github.com/MihailRis/voxelcore.git synced 2026-10-12 22:41:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

add jobsSource to ThreadPool

Browse source
This commit is contained in:
MihailRis 2026-01-23 22:39:20 +03:00
parent d5f8fed633
commit 245e040a1c
1 changed files with 22 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/util/ThreadPool.hpp
View file

@ -2,6 +2,7 @@
#include <atomic> #include <atomic>
#include <chrono> #include <chrono>
#include <optional>
#include <condition_variable> #include <condition_variable>
#include <functional> #include <functional>
#include <iostream> #include <iostream>
@ -48,6 +49,7 @@ namespace util {
std::atomic<int> busyWorkers = 0; std::atomic<int> busyWorkers = 0;
std::atomic<uint> jobsDone = 0; std::atomic<uint> jobsDone = 0;
std::atomic<bool> working = true; std::atomic<bool> working = true;
supplier<std::optional<T>> jobsSource = nullptr;
bool failed = false; bool failed = false;
bool standaloneResults = true; bool standaloneResults = true;
bool stopOnFail = true; bool stopOnFail = true;
@ -151,6 +153,10 @@ namespace util {
return working; return working;
} }
void setJobsSource(supplier<std::optional<T>>&& source) {
jobsSource = std::move(source);
}
void terminate() override { void terminate() override {
if (!working) { if (!working) {
return; return;
@ -221,6 +227,22 @@ namespace util {
} }
} }
} }
if (jobsSource) {
bool jobsAdded = false;
std::lock_guard<std::mutex> jobsLock(jobsMutex);
while (true) {
auto job = jobsSource();
if (job.has_value()) {
jobs.push(std::move(job.value()));
jobsAdded = true;
} else {
break;
}
}
if (jobsAdded) {
jobsMutexCondition.notify_one();
}
}
if (failed) { if (failed) {
throw std::runtime_error("some job failed"); throw std::runtime_error("some job failed");
} }