#include <setjmp.h>
#include <signal.h>
#include <stdio.h>
#include <sys/time.h>
#include "umka.h"
#include "thread.h"

struct itimerval timeout = {.it_value = {.tv_sec = 0, .tv_usec = 10000}};

void scheduler(int signo, siginfo_t *info, void *context) {
    (void)signo;
    (void)info;
    (void)context;
//    printf("##### switching from task %u\n", kos_current_task);
    if (!sigsetjmp(*kos_slot_base[kos_current_task].fpu_state, 1)) {
//        printf("##### saved\n");
        kos_current_task += 1;
        if (kos_current_task == kos_task_count) {
            kos_current_task = 1;
        }
        kos_current_slot = kos_slot_base + kos_current_task;
        printf("##### kos_current_task: %u\n", kos_current_task);
        setitimer(ITIMER_PROF, &timeout, NULL);
        siglongjmp(*kos_slot_base[kos_current_task].fpu_state, 1);
    }
}

int main() {
    struct sigaction sa;

    sa.sa_sigaction = scheduler;
    sigemptyset(&sa.sa_mask);
    sa.sa_flags = SA_SIGINFO;

    if (sigaction(SIGPROF, &sa, NULL) == -1) {
        printf("Can't install signal handler!\n");
        return 1;
    }

    umka_os();

    return 0;
}