#include <time.h>
#include <stdarg.h>
#include <stdbool.h>
#include <inttypes.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "../src/nissy.h"
#include "nissy_extra.h"

static void log_stderr(const char *, ...);
static void log_stdout(const char *, ...);
static double timerun(void (*)(void));
static void getfilename(const char *, const char *, char *);
static void writetable(const char *, int64_t, const char *);
static int64_t generatetable(const char *, const char *, char **);
static int64_t derivetable(const char *, const char *, const char *, char **);
static int getdata(const char *, const char *, char **, const char *);
static void gendata_run(const char *, const char *, uint64_t[static 21]);
static void derivedata_run(
    const char *, const char *, const char *, const char *);

static void
log_stderr(const char *str, ...)
{
	va_list args;

	va_start(args, str);
	vfprintf(stderr, str, args);
	va_end(args);
}

static void
write_stdout(const char *str, ...)
{
	va_list args;

	va_start(args, str);
	vfprintf(stdout, str, args);
	va_end(args);
}

static double
timerun(void (*run)(void))
{
	struct timespec start, end;
	double tdiff, tdsec, tdnano;

	fflush(stdout);

	if (run == NULL) {
		printf("nothing to run!\n");
		fflush(stdout);
		return -1.0;
	}

	clock_gettime(CLOCK_MONOTONIC, &start);
	run();
	clock_gettime(CLOCK_MONOTONIC, &end);

	tdsec = end.tv_sec - start.tv_sec;
	tdnano = end.tv_nsec - start.tv_nsec;
	tdiff = tdsec + 1e-9 * tdnano;

	printf("---------\n");
	printf("\nTotal time: %.4fs\n", tdiff);
	fflush(stdout);

	return tdiff;
}

static void
getfilename(const char *solver, const char *options, char *filename)
{
	uint8_t h, k;

	/* Only h48 supported for now */
	parse_h48_options(options, &h, &k, NULL);

	sprintf(filename, "tables/%sh%dk%d", solver, h, k);
}

static void
writetable(const char *buf, int64_t size, const char *filename)
{
	FILE *f;

	if ((f = fopen(filename, "wb")) == NULL) {
		fprintf(stderr, "Could not write tables to file %s"
		    ", will be regenerated next time.\n", filename);
	} else {
		fwrite(buf, size, 1, f);
		fclose(f);
		fprintf(stderr, "Table written to %s.\n", filename);
	}
}

static int64_t
generatetable(const char *solver, const char *options, char **buf)
{
	int64_t size, gensize;

	size = nissy_datasize(solver, options);
	if (size == -1) {
		printf("Error getting table size.\n");
		return -1;
	}

	*buf = malloc(size);
	gensize = nissy_gendata(solver, options, *buf);

	if (gensize != size) {
		fprintf(stderr, "Error generating table");
		if (gensize != -1)
			fprintf(stderr, " (got %" PRId64 " bytes)", gensize);
		fprintf(stderr, "\n");
		return -2;
	}

	return gensize;
}

static int64_t
derivetable(
	const char *opts_large,
	const char *opts_small,
	const char *filename_large,
	char **buf
)
{
	uint8_t h;
	int64_t size, gensize;
	char *fulltable;

	if (getdata("h48", opts_large, &fulltable, filename_large) != 0) {
		printf("Error reading full table.\n");
		return -1;
	}

	size = nissy_datasize("h48", opts_small);
	if (size == -1) {
		printf("Error getting table size.\n");
		free(fulltable);
		return -1;
	}

	h = atoi(opts_small); /* TODO: use option parser */
	*buf = malloc(size);
	gensize = gendata_h48_derive(h, fulltable, *buf);

	if (gensize != size) {
		fprintf(stderr, "Error deriving table\n");
		free(fulltable);
		return -2;
	}

	free(fulltable);
	return gensize;
}

static int
getdata(
	const char *solver,
	const char *options,
	char **buf,
	const char *filename
) {
	int64_t size, sizeread;
	FILE *f;

	if ((f = fopen(filename, "rb")) == NULL) {
		fprintf(stderr, "Table file not found, generating it.\n");
		size = generatetable(solver, options, buf);
		switch (size) {
		case -1:
			goto getdata_error_nofree;
		case -2:
			goto getdata_error;
		default:
			writetable(*buf, size, filename);
			break;
		}
	} else {
		fprintf(stderr, "Reading tables from file %s\n", filename);
		size = nissy_datasize(solver, options);
		*buf = malloc(size);
		sizeread = fread(*buf, size, 1, f);
		fclose(f);
		if (sizeread != 1) {
			fprintf(stderr, "Error reading table, stopping\n");
			goto getdata_error;
		}
	}

	return 0;

getdata_error:
	free(*buf);
getdata_error_nofree:
	return 1;
}

static void
gendata_run(
	const char *solver,
	const char *options,
	uint64_t expected[static 21]
) {
	int64_t size;
	char *buf, filename[1024];

	getfilename(solver, options, filename);
	size = generatetable(solver, options, &buf);
	switch (size) {
	case -1:
		return;
	case -2:
		goto gendata_run_finish;
	default:
		nissy_datainfo(buf, write_stdout);
		printf("\n");
		printf("Succesfully generated %" PRId64 " bytes. "
		       "See above for details on the tables.\n", size);

		/* TODO: check that the table is correct */
		writetable(buf, size, filename);
		break;
	}

gendata_run_finish:
	free(buf);
}

static void
derivedata_run(
	const char *opts_large,
	const char *opts_small,
	const char *filename_large,
	const char *filename_small
)
{
	int64_t size;
	char *buf;

	size = derivetable(opts_large, opts_small, filename_large, &buf);
	switch (size) {
	case -1:
		return;
	case -2:
		goto derivedata_run_finish;
	default:
		nissy_datainfo(buf, write_stdout);
		printf("\n");
		printf("Succesfully generated %" PRId64 " bytes. "
		       "See above for details on the tables.\n", size);

		writetable(buf, size, filename_small);
		break;
	}

derivedata_run_finish:
	free(buf);
}