remove extra shared_ptr use from util::ThreadPool

This commit is contained in:
MihailRis
2024-10-24 11:49:06 +03:00
parent 7f0e12bb4e
commit 054b3926cd
6 changed files with 25 additions and 26 deletions
+8 -9
View File
@@ -272,15 +272,15 @@ public:
LoaderWorker(AssetsLoader* loader) : loader(loader) { LoaderWorker(AssetsLoader* loader) : loader(loader) {
} }
assetload::postfunc operator()(const std::shared_ptr<aloader_entry>& entry assetload::postfunc operator()(const aloader_entry& entry
) override { ) override {
aloader_func loadfunc = loader->getLoader(entry->tag); aloader_func loadfunc = loader->getLoader(entry.tag);
return loadfunc( return loadfunc(
loader, loader,
loader->getPaths(), loader->getPaths(),
entry->filename, entry.filename,
entry->alias, entry.alias,
entry->config entry.config
); );
} }
}; };
@@ -290,14 +290,13 @@ std::shared_ptr<Task> AssetsLoader::startTask(runnable onDone) {
std::make_shared<util::ThreadPool<aloader_entry, assetload::postfunc>>( std::make_shared<util::ThreadPool<aloader_entry, assetload::postfunc>>(
"assets-loader-pool", "assets-loader-pool",
[=]() { return std::make_shared<LoaderWorker>(this); }, [=]() { return std::make_shared<LoaderWorker>(this); },
[=](assetload::postfunc& func) { func(assets); } [=](const assetload::postfunc& func) { func(assets); }
); );
pool->setOnComplete(std::move(onDone)); pool->setOnComplete(std::move(onDone));
while (!entries.empty()) { while (!entries.empty()) {
const aloader_entry& entry = entries.front(); aloader_entry entry = std::move(entries.front());
auto ptr = std::make_shared<aloader_entry>(entry);
pool->enqueueJob(ptr);
entries.pop(); entries.pop();
pool->enqueueJob(std::move(entry));
} }
return pool; return pool;
} }
+2 -2
View File
@@ -43,8 +43,8 @@ using aloader_func = std::function<
struct aloader_entry { struct aloader_entry {
AssetType tag; AssetType tag;
const std::string filename; std::string filename;
const std::string alias; std::string alias;
std::shared_ptr<AssetCfg> config; std::shared_ptr<AssetCfg> config;
}; };
+5 -5
View File
@@ -27,8 +27,8 @@ public:
: converter(std::move(converter)) { : converter(std::move(converter)) {
} }
int operator()(const std::shared_ptr<ConvertTask>& task) override { int operator()(const ConvertTask& task) override {
converter->convert(*task); converter->convert(task);
return 0; return 0;
} }
}; };
@@ -169,10 +169,10 @@ std::shared_ptr<Task> WorldConverter::startTask(
); );
auto& converterTasks = converter->tasks; auto& converterTasks = converter->tasks;
while (!converterTasks.empty()) { while (!converterTasks.empty()) {
const ConvertTask& task = converterTasks.front(); ConvertTask task = std::move(converterTasks.front());
auto ptr = std::make_shared<ConvertTask>(task);
pool->enqueueJob(ptr);
converterTasks.pop(); converterTasks.pop();
pool->enqueueJob(std::move(task));
} }
pool->setOnComplete([=]() { pool->setOnComplete([=]() {
converter->write(); converter->write();
+1 -1
View File
@@ -12,7 +12,7 @@
static debug::Logger logger("chunks-render"); static debug::Logger logger("chunks-render");
class RendererWorker : public util::Worker<Chunk, RendererResult> { class RendererWorker : public util::Worker<std::shared_ptr<Chunk>, RendererResult> {
Level* level; Level* level;
BlocksRenderer renderer; BlocksRenderer renderer;
public: public:
+1 -1
View File
@@ -30,7 +30,7 @@ class ChunksRenderer {
std::unordered_map<glm::ivec2, std::shared_ptr<Mesh>> meshes; std::unordered_map<glm::ivec2, std::shared_ptr<Mesh>> meshes;
std::unordered_map<glm::ivec2, bool> inwork; std::unordered_map<glm::ivec2, bool> inwork;
util::ThreadPool<Chunk, RendererResult> threadPool; util::ThreadPool<std::shared_ptr<Chunk>, RendererResult> threadPool;
public: public:
ChunksRenderer( ChunksRenderer(
Level* level, Level* level,
+8 -8
View File
@@ -17,7 +17,7 @@ namespace util {
template <class J, class T> template <class J, class T>
struct ThreadPoolResult { struct ThreadPoolResult {
std::shared_ptr<J> job; J job;
std::condition_variable& variable; std::condition_variable& variable;
int workerIndex; int workerIndex;
bool& locked; bool& locked;
@@ -29,13 +29,13 @@ namespace util {
public: public:
Worker() = default; Worker() = default;
virtual ~Worker() = default; virtual ~Worker() = default;
virtual R operator()(const std::shared_ptr<T>&) = 0; virtual R operator()(const T&) = 0;
}; };
template <class T, class R> template <class T, class R>
class ThreadPool : public Task { class ThreadPool : public Task {
debug::Logger logger; debug::Logger logger;
std::queue<std::shared_ptr<T>> jobs; std::queue<T> jobs;
std::queue<ThreadPoolResult<T, R>> results; std::queue<ThreadPoolResult<T, R>> results;
std::mutex resultsMutex; std::mutex resultsMutex;
std::vector<std::thread> threads; std::vector<std::thread> threads;
@@ -43,7 +43,7 @@ namespace util {
std::mutex jobsMutex; std::mutex jobsMutex;
std::vector<std::unique_lock<std::mutex>> workersBlocked; std::vector<std::unique_lock<std::mutex>> workersBlocked;
consumer<R&> resultConsumer; consumer<R&> resultConsumer;
consumer<std::shared_ptr<T>&> onJobFailed = nullptr; consumer<T&> onJobFailed = nullptr;
runnable onComplete = nullptr; runnable onComplete = nullptr;
std::atomic<int> busyWorkers = 0; std::atomic<int> busyWorkers = 0;
std::atomic<uint> jobsDone = 0; std::atomic<uint> jobsDone = 0;
@@ -57,7 +57,7 @@ namespace util {
std::mutex mutex; std::mutex mutex;
bool locked = false; bool locked = false;
while (working) { while (working) {
std::shared_ptr<T> job; T job;
{ {
std::unique_lock<std::mutex> lock(jobsMutex); std::unique_lock<std::mutex> lock(jobsMutex);
jobsMutexCondition.wait(lock, [this] { jobsMutexCondition.wait(lock, [this] {
@@ -66,7 +66,7 @@ namespace util {
if (!working || failed) { if (!working || failed) {
break; break;
} }
job = jobs.front(); job = std::move(jobs.front());
jobs.pop(); jobs.pop();
busyWorkers++; busyWorkers++;
@@ -229,10 +229,10 @@ namespace util {
} }
} }
void enqueueJob(const std::shared_ptr<T>& job) { void enqueueJob(T job) {
{ {
std::lock_guard<std::mutex> lock(jobsMutex); std::lock_guard<std::mutex> lock(jobsMutex);
jobs.push(job); jobs.push(std::move(job));
} }
jobsMutexCondition.notify_one(); jobsMutexCondition.notify_one();
} }