#include "gtools.h"
#include "geng.h"
#include <ucontext.h>
#include <stdint.h>
#include <stdbool.h>

// TODO: only on macos
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wdeprecated-declarations"

void
printgraph(graph *g, int n)
{
    set *gi;
    for (int i = 0; i < n; ++i)
    {
        gi = GRAPHROW(g, 1, i);
        for (int j = 0; j < n; ++j)
        {
            int q = ISELEMENT(gi, j);
            printf("%i ", q);
        }
        printf("\n");
    }
    printf("\n");
}

// TODO: probably don't need to name this function with macro
void
OUTPROC(__attribute__((unused)) FILE *outfile,
        graph *g, int n, struct geng_iterator *iter)
{
    iter->cur = g;
    iter->graph_size = n;

    // TODO: add support for generating graphs in batches (for speed)
#if 0
    static int i = 1;
    i++;
    if (i % 100 == 0)
    {
        swapcontext(&geng_worker, &geng_user);
        i = 0;
    }
#else
    swapcontext(&iter->geng_worker, &iter->geng_user);
#endif
}

// TODO: probably don't need to name this function with macro
// TODO: move to geng.h maybe?
extern int
GENG_MAIN(int argc, char *argv[]);

// geng_iterator_init gives ownership of iterator memory to caller
void
geng_iterator_create(struct geng_iterator **iterator_ptr,
                     size_t graph_size,
                     size_t batch_capacity)
{
    // TODO: malloc geng_iterator batch
    *iterator_ptr = malloc(sizeof(struct geng_iterator));
    
    struct geng_iterator *iterator = *iterator_ptr;
    iterator->iteration_done = false;

    // TODO: add support for more arguments
    int geng_argc = 3;
    char **geng_argv;
    geng_argv = malloc((geng_argc + 1) * sizeof(char *));
    geng_argv[0] = "geng";
    geng_argv[1] = "-q";
    char n_str[20];
    snprintf(n_str, sizeof(n_str), "%zu", graph_size);
    geng_argv[2] = n_str;
    geng_argv[3] = NULL;

    uint32_t p_argv[2];
    p_argv[0] = (uint32_t) (((size_t) &geng_argv) & ((1llu << 32) - 1llu));
    p_argv[1] = ((size_t) &geng_argv) >> 32;

    uint32_t p_iter[2];
    p_iter[0] = (uint32_t) (((size_t) &iterator) & ((1llu << 32) - 1llu));
    p_iter[1] = ((size_t) &iterator) >> 32;

    getcontext(&iterator->geng_user);
    iterator->geng_worker = iterator->geng_user;
    iterator->geng_worker.uc_stack.ss_sp = iterator->geng_stack;
    iterator->geng_worker.uc_stack.ss_size = sizeof(iterator->geng_stack);
    iterator->geng_worker.uc_link = &iterator->geng_user;

    makecontext(
        &iterator->geng_worker, (void (*) (void)) GENG_MAIN,
        5, geng_argc, p_argv[0], p_argv[1], p_iter[0], p_iter[1]
    );
    swapcontext(&iterator->geng_user, &iterator->geng_worker);
    free(geng_argv);
}

bool
geng_iterator_next(struct geng_iterator *iter, graph *g)
{
    if (iter->iteration_done) return false;
    else
    {
        memcpy(g, iter->cur, sizeof(set) * iter->graph_size);
        swapcontext(&iter->geng_user, &iter->geng_worker);
        if (!iter->iteration_done && iter->generation_done)
            iter->iteration_done = true;
        return true;
    }
}

// TODO: only on macos
#pragma GCC diagnostic pop