#pragma once

#include <functional>
#include <queue>
#include <vector>
#include <thread>
#include <mutex>

using namespace std;


class Scheduler {
	struct Job {
		function<void()> callback;

		Job(const function<void()> callback)
				: callback(callback) {}
	};

	queue<Job*> jobs;
	bool terminateFlag = false;
	mutex commMut;

	vector<thread> workers;

	void workerEntry();

public:
	const int nthreads;

	Scheduler(int nthreads);
	~Scheduler();

	// func is run in child thread
	// doneCallback is run in host thread
	void submit(const function<void()> &func);

	void finish();
};