#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 100

extern __code initSynchronizedQueueContext(struct Context* context);
extern void allocator(struct Context* context);

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

__code code1(struct Allocate* allocate) {
  allocate->size = sizeof(long);
  allocator(context);
  goto code2();
}

__code code2(long* count) {
    *count = 0;
    goto code3();
}

__code code3(long* count, struct Allocate* allocate) {
    long loop = *count;
    if(loop == NUM) {
        goto thread_exit();
    }
    allocate->size = sizeof(struct Element);
    allocator(context);
    goto code4();
}

__code code4(long* count, struct Allocate* allocate, struct Element* element) {
    allocate->after_put = Code3;
    element->value = (*count)++;
    goto sender();
}

__code code4_stub(struct Context* context) {
    goto code4(context, &context->data[Count]->count, &context->data[Allocate]->allocate, &context->data[context->dataNum]->element);
}

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

__code sender(struct Context* context, struct Queue* queue) {
    goto meta_sender(context, queue, Put);
}

__code meta_put(struct Context* context, struct Queue* queue, enum Code next) {
    // signal
    pthread_cond_signal(&queue->cond);
    // unlock
    pthread_mutex_unlock(&queue->mutex);
    goto (context->code[next])(context);
}

__code put_stub(struct Context* context) {
    goto put(context, &context->data[Allocate]->allocate, &context->data[Queue]->queue, &context->data[context->dataNum]->element);
}

__code put(struct Context* context, struct Allocate* allocate, struct Queue* queue, struct Element* element) {
    if(queue->first) {
        queue->last->next = element;
    } else {
        queue->first = element;
    }
    queue->last   = element;
    element->next = 0;
    queue->count++;
    printf("Put %d\n\n", element->value);
    goto meta_put(context, queue, allocate->after_put);
}

__code code5(struct Allocate* allocate) {
    allocate->size = sizeof(long);
    allocator(context);
    goto code6();
}

__code code5_stub(struct Context* context) {
    goto code5(context, &context->data[Allocate]->allocate);
}

__code code6(long* count) {
    *count = 0;
    goto code7();
}

__code code7(long* count, struct Allocate* allocate) {
    long loop = *count;
    if(loop == NUM) {
        goto meta(context, Thread_exit);
    }
    (*count)++;
    allocate->after_get = Code7;
    goto receiver();
}

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

__code receiver(struct Queue* queue) {
    goto meta_receiver(context, queue, Get);
}

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

__code get(struct Allocate* allocate, struct Queue* queue, struct Element* element) {
    // thread wait if queue is empty
    while (queue->count == 0) {
        pthread_cond_wait(&queue->cond, &queue->mutex);
    }
    printf("      Get %d\n\n", queue->first->value);
    queue->first = (queue->first->next) ? queue->first->next : 0;
    queue->count--;
    goto meta_get(context, allocate->after_get);
}

__code get_stub(struct Context* context) {
    goto get(context, &context->data[Allocate]->allocate, &context->data[Queue]->queue, &context->data[context->dataNum]->element);
}

__code thread_exit() {
    free(context->code);
    free(context->data);
    free(context->heapStart);
    pthread_exit(0);
}

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

void* thread_func2(void* context) {
    goto start_code((struct Context*)context, Code5);
    return 0;
}

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);
    struct Context* context4 = (struct Context*)malloc(sizeof(struct Context));
    initSynchronizedQueueContext(context4);
    context2->data[Queue] = context1->data[Queue];
    context3->data[Queue] = context1->data[Queue];
    context4->data[Queue] = context1->data[Queue];
    pthread_t thread1, thread2, thread3, thread4;
    pthread_create(&thread1, NULL, thread_func, (void *)context1);
    pthread_create(&thread2, NULL, thread_func, (void *)context2);
    pthread_create(&thread3, NULL, thread_func2, (void *)context3);
    pthread_create(&thread4, NULL, thread_func2, (void *)context4);
    pthread_join(thread1, NULL);
    pthread_join(thread2, NULL);
    pthread_join(thread3, NULL);
    pthread_join(thread4, NULL);
}