#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <math.h>

#include "../../../context.h"
#interface "TaskManager.h"

int cpu_num = 1;
int length = 1024;
int split  = 8;
int gpu_num = 0;
int CPU_ANY = -1;
int CPU_CUDA = -1;

__code initDataGears(struct LoopCounter* loopCounter, struct TaskManager* taskManager) {
    // loopCounter->tree = createRedBlackTree(context);
    loopCounter->i = 0;
    taskManager->taskManager = (union Data*)createTaskManagerImpl(context, cpu_num, gpu_num, 0);
    goto code1();
}

__code code1(struct LoopCounter* loopCounter) {
    printf("cpus:\t\t%d\n", cpu_num);
    printf("gpus:\t\t%d\n", gpu_num);
    printf("length:\t\t%d\n", length);
    printf("length/task:\t%d\n", length/split);
    /* puts("queue"); */
    /* print_queue(context->data[ActiveQueue]->queue.first); */
    /* puts("tree"); */
    /* print_tree(context->data[Tree]->tree.root); */
    /* puts("result"); */

    goto createTask1();
}

__code createTask1(struct LoopCounter* loopCounter, struct TaskManager* taskManager) {
    struct SortArray* outputSortArray = new SortArray();
    struct SortArray* inputSortArray = outputSortArray;
    struct Timer* timer = createTimerImpl(context);

    par goto makeArray(outputSortArray, timer, __exit);

    for (int i=2; i <= length; i=2*i) {
        int first = 1;
        for (int j=i>>1; j > 0; j=j>>1) {
            outputSortArray = new SortArray();
            inputSortArray->prefix = length/2/split;
            inputSortArray->block = j;
            inputSortArray->first = first;
            par goto bitonicSwap(inputSortArray, outputSortArray, iterate(split), __exit);
            first = 0;
            inputSortArray = outputSortArray;
        }
    }

    par goto printArray(inputSortArray, timer, __exit);

    goto code2();
}

__code code2(struct TaskManager* taskManager) {
    goto taskManager->shutdown(exit_code);
}

__code code2_stub(struct Context* context) {
    goto code2(context, &Gearef(context, TaskManager)->taskManager->TaskManager);
}

void init(int argc, char** argv) {
    if (gpu_num == 1) {
        split = length/2;
    }
    for (int i = 1; argv[i]; ++i) {
        if (strcmp(argv[i], "-cpu") == 0)
            cpu_num = (int)atoi(argv[i+1]);
        else if (strcmp(argv[i], "-l") == 0)
            length = (int)atoi(argv[i+1]);
        else if (strcmp(argv[i], "-s") == 0)
            split = (int)atoi(argv[i+1]);
        else if (strcmp(argv[i], "-cuda") == 0) {
            gpu_num = 1;
            CPU_CUDA = 0;
        }
    }
}

int main(int argc, char** argv) {
    init(argc, argv);
    goto initDataGears();
}