wait-group.hpp

struct WaitGroup {
	std::atomic_int counter = 0;

	void add(int i)
	{
		counter += i;
		// printf("COUNTER[UP]: %d\n", counter.load());
	}

	void done()
	{
		counter--;
		// printf("COUNTER[DN]: %d\n", counter.load());
		assert(counter.load() >= 0);
	}

	void wait(std::function<void()> fn)
	{
		// TODO add time out to WaitGroup functionality?
		while (counter > 0)
		{
			fn();
		}
	}
};