Seele/src/Engine/ThreadPool.h

#pragma once
#include <thread>
#include <functional>
#include "Containers/Array.h"
#include "Containers/List.h"

namespace Seele
{
class ThreadPool
{
public:
    ThreadPool(uint32 numWorkers = std::thread::hardware_concurrency());
    ~ThreadPool();
    void runAndWait(List<std::function<void()>> functions);
private:
    struct Task
    {
        uint64 numRemaining = 0;
        List<std::function<void()>> functions;
    };
    void work();
    Array<std::thread> workers;
    std::mutex taskLock;
    std::condition_variable taskCV;
    std::condition_variable completedCV;
    Task currentTask;
    bool running = true;
};
}