TmpMenu/src/thread_pool.hpp

#pragma once

namespace big
{
	struct thread_pool_job
	{
		std::function<void()> m_func;
		std::source_location m_source_location;
	};

	class thread_pool
	{
		std::atomic<bool> m_accept_jobs;
		std::condition_variable m_data_condition;

		std::stack<thread_pool_job> m_job_stack;
		std::mutex m_lock;
		std::vector<std::thread> m_thread_pool;

		std::thread m_managing_thread;

		std::atomic<size_t> m_available_thread_count;

	public:
		thread_pool();
		~thread_pool();

		void destroy();
		void push(std::function<void()> func, std::source_location location = std::source_location::current());

	private:
		void create();
		void done();
		void run();
	};

	inline thread_pool* g_thread_pool{};
}