#ifndef COLLECTIVE_SCHEDULER_HPP
#define COLLECTIVE_SCHEDULER_HPP

#include <iostream>
#include <unistd.h>
#include <sys/wait.h>
#include <cmath>
#include "Task.hpp"

class CollectiveScheduler {
	private:
		int levels;
	public:
		CollectiveScheduler(int lvls) : levels(lvls) {}

		void execute_task(Task* task){
		pid_t root=getpid();

		for(int i=0; i<levels; i++) {
			fork();
		}

		if(getpid()!=root) {
			int worker_id = getpid()%root;
			int worker_count = std::pow(2,levels) - 1;

			if(task!=nullptr) {
				task->execute_task(worker_id,worker_count);
			}
		}

		int status;
		while(true){
			pid_t finished=waitpid(-1, &status , 0);
			if(finished == -1){
				break;
			}
		}

		if(getpid() !=root){
			_exit(0);
		}
	}
};

#endif