#include "context.h"
#include "stack.h"
#include "queue.h"
#include "origin_cs.h"
#include <stdio.h>

union Data* createTaskManager(struct Context* context) {
    struct TaskManager* taskManager = &ALLOCATE(context, TaskManager)->taskManager;
    struct TaskManager* taskManagerImpl = &ALLOCATE(context, TaskManagerImpl)->taskManagerImpl;
    taskManagerImpl -> activeQueue = createSynchronizedQueue(context);
    taskManagerImpl -> waitQueue = createSynchronizedQueue(context);
    taskManager->taskManager = (union Data*)taskManagerImpl;
    taskManager->spawn = C_spawnTaskManager;
    taskManager->shutdown  = C_shutdownTaskManager;
    taskManager->deadLockDetected  = C_deadLockDetected;
    return (union Data*)(taskManager);
}

__code spawnTaskManager(struct Context* context, struct Task* task, struct Element* element, struct Queue* queue, struct Queue* activeQueue, struct Queue* waitQueue) {
    if (task->idsCount == 0) {
        // enqueue activeQueue
        queue->queue = (union Data*)activeQueue;
    } else {
        // enqueue waitQueue
        queue->queue = (union Data*)waitQueue;
    }
    queue->data = element->data;
    queue->next = context->next;
    goto meta(context, queue->queue->put);
}

__code spawnTaskManager_stub(struct Context* context) {
    goto spawnTask(context,
            context->data[D_Element]->element.data,
            &context->data[D_Element]->element,
            &context->data[D_ActiveQueue]->Queue,
            &context->data[D_WaitQueue]->Queue);
}

__code shutdownTaskManager(struct Context* context, struct LoopCounter* loopCounter, struct Worker* worker) {
    int i = loopCounter->i;

    if (i < worker->num) {
        pthread_join(worker->contexts[i].thread, NULL);
        loopCounter->i++;

        goto meta(context, TaskManager);
    }

    loopCounter->i = 0;

    Time *t = &context->data[D_Time]->Time;
    t->next = C_code2;
    goto meta(context, EndTime);
}

__code shutdownTaskManager_stub(struct Context* context) {
    goto shutdownTaskManager(context, &context->data[D_LoopCounter]->loopCounter, &context->data[D_Worker]->worker);
}