#include <stdlib.h>
#include <stdio.h>

#include "synchronizedQueueContext.h"

#include "allocate.h"
#include "origin_cs.h"

#ifdef CLANG
#define _CbC_retrun __return
#define _CbC_environment __environment
#endif

#define NUM 50

extern __code initSynchronizedQueueContext(struct Context* context);

//__code code1(struct Context* context) {
//    context->data[Allocate]->allocate.size = sizeof(struct Element);
//    context->data[Allocate]->allocate.next = Code2;
//    goto meta(context, Allocator);
//}

__code meta(struct Context* context, enum Code next) {
    goto (context->code[next])(context);
}

//__code code2(struct Context* context) {
//    context->data[Allocate]->allocate.after_put = Code3;
//    context->data[context->dataNum] -> element.value = 1024;
//    goto meta(context, Sender);
//}

__code meta_sender(struct Context* context, enum Code next) {
    // lock
    pthread_mutex_lock(&context->data[Queue]->queue.mutex);
    goto (context->code[next])(context);
}

__code sender(struct Context* context) {
    goto meta(context, Put);
}

__code code1(struct Context* context) {
    context->data[Allocate]->allocate.size = sizeof(long);
    context->data[Allocate]->allocate.next = Code2;
    goto meta(context, Allocator);
}

__code code2(struct Context* context) {
    context->data[Counter] -> count = 0;
    goto meta(context, Code3);
}

__code code3(struct Context* context) {
    long loop = context->data[Counter]->count;
    if(loop == NUM) {
        goto meta(context, Exit);
    }
    context->data[Allocate]->allocate.size = sizeof(struct Element);
    context->data[Allocate]->allocate.next = Code4;
    goto meta(context, Allocator);
}

__code code4(struct Context* context) {
    context->data[Allocate]->allocate.after_put = Code3;
    context->data[context->dataNum] -> element.value = context->data[Counter]->count++;
    goto meta(context, Sender);
}

__code meta_put(struct Context* context, enum Code next) {
    if(context->data[Queue]->queue.first) {
        context->data[Queue]->queue.last->element.next = context->data[context->dataNum];
        context->data[Queue]->queue.last               = context->data[Queue]->queue.last->element.next;
    } else {
        context->data[Queue]->queue.first = context->data[context->dataNum];
        context->data[Queue]->queue.last  = context->data[Queue]->queue.first;
    }
    context->data[Queue]->queue.last->element.next = 0;
    context->data[Queue]->queue.count++;
    printf("Put %d\n\n", context->data[Queue]->queue.last->element.value);

    // unlock
    pthread_mutex_unlock(&context->data[Queue]->queue.mutex);
    goto (context->code[next])(context);
}

__code put(struct Context* context) {
    goto meta_put(context, context->data[Allocate]->allocate.after_put);
}

__code code5(struct Context* context) {
    context->data[Allocate]->allocate.size = sizeof(long);
    context->data[Allocate]->allocate.next = Code6;
    goto meta(context, Allocator);
}

__code code6(struct Context* context) {
    context->data[Counter] -> count = 0;
    goto meta(context, Code7);
}

__code code7(struct Context* context) {
    long loop = context->data[Counter]->count;
    if(loop == NUM) {
        goto meta(context, Exit);
    }
    context->data[Counter]->count++;
    context->data[Allocate]->allocate.after_get = Code7;
    goto meta(context, Receiver);
}

__code meta_receiver(struct Context* context, enum Code next) {
    // lock
    pthread_mutex_lock(&context->data[Queue]->queue.mutex);
    goto (context->code[next])(context);
}

__code receiver(struct Context* context) {
    goto meta(context, Get);
}

__code meta_get(struct Context* context, enum Code next) {
    if (context->data[Queue]->queue.first == context->data[Queue]->queue.last) {
        printf("queue is empty\n");
        goto meta(context, Receiver);
    } else {
        printf("Get %d\n\n", context->data[Queue]->queue.first->element.value);
        context->data[Queue]->queue.first = (context->data[Queue]->queue.first->element.next) ? context->data[Queue]->queue.first->element.next : 0;
        context->data[Queue]->queue.count--;
    }
    pthread_mutex_unlock(&context->data[Queue]->queue.mutex);
    goto (context->code[next])(context);
}

__code get(struct Context* context) {
    goto meta_get(context, context->data[Allocate]->allocate.after_get);
}

void* thread_func(void* context) {
    goto start_code((struct Context*)context, Code1);
    return NULL;
}

int main() {
    struct Context* context1 = (struct Context*)malloc(sizeof(struct Context));
    initSynchronizedQueueContext(context1);
    struct Context* context2 = (struct Context*)malloc(sizeof(struct Context));
    initSynchronizedQueueContext(context2);
    struct Context* context3 = (struct Context*)malloc(sizeof(struct Context));
    initSynchronizedQueueContext(context3);
    context2->data[Queue] = context1->data[Queue];
    context3->data[Queue] = context1->data[Queue];
    pthread_t thread1, thread2;
    pthread_create(&thread1, NULL, thread_func, (void *)context2);
    pthread_create(&thread2, NULL, thread_func, (void *)context3);
    goto start_code(context1, Code5);
}