#include <assert.h>
#include <signal.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>

#include <SDL2/SDL.h>

#include "../game.h"
#include "logging.h"

Game games[2] = {0};

// Tsoding is now a 3-Star Developer!
const char *tsoding_shift(int *argc, char ***argv) {
  assert(*argc > 0);
  const char* result = **argv;
  *argc -= 1;
  *argv += 1;
  return result;
}

void usage(FILE *stream) {
  fprintf(stream, "Usage: ./gp_trainer [OPTIONS]\n");
  fprintf(stream, "       --gen|g <NUMBER>\n");
  fprintf(stream, "         Amount of generations (default: 100)\n");
  fprintf(stream, "       --out|o <FILEPATH>\n");
  fprintf(stream, "         Output filepath (default: out.bin)\n");
  fprintf(stream, "       --plot|p\n");
  fprintf(stream, "         Enable Gnuplot live reporting\n");
}

FILE* gnulog;
int gnuplot;
int current;
const char* output_filepath;

void ctrlc(int signum) {
  if (gnuplot) {
    log_close_pipe();
    fclose(gnulog);
  }

  dump_game(output_filepath, &games[current]);
  exit(1);
}

int main(int argc, char *argv[]) {

  signal(SIGINT, ctrlc);

  tsoding_shift(&argc, &argv); //Skip Program Name

  srand(time(NULL));
  int gen_count = 100;
  output_filepath = "out.bin";
  gnuplot = 0;

  while (argc > 0) {
    const char *flag = tsoding_shift(&argc, &argv);

    if (strcmp(flag, "--gen") == 0 || strcmp(flag, "-g") == 0) {
      if (argc == 0) {
        usage(stderr);
        fprintf(stderr, "ERROR: no value provided for flag %s\n", flag);
        exit(1);
      }
      const char *value = tsoding_shift(&argc, &argv);
      gen_count = atoi(value);

    } else if (strcmp(flag, "--out") == 0 || strcmp(flag, "-o") == 0) {
      if (argc == 0) {
        usage(stderr);
        fprintf(stderr, "ERROR: no value provided for flag %s\n", flag);
        exit(1);
      }
      const char *value = tsoding_shift(&argc, &argv);
      output_filepath = value;

    } else if (strcmp(flag, "--plot") == 0 || strcmp(flag, "-p") == 0) {
      gnuplot = 1;

    } else if (strcmp(flag, "--help") == 0 || strcmp(flag, "-h") == 0) {
      usage(stdout);
      exit(0);
    } else {
      usage(stderr);
      fprintf(stderr, "ERROR: unknown flag: %s\n", flag);
      exit(1);
    }
  }

  if (gnuplot) {
    gnulog = fopen("log.dat", "w");
    log_header(gnulog);
  }

  current = 0;

  init_game(&games[current]);

  for (int i = 0; i < gen_count; ++i) {
    fprintf(stderr, "Generation %d... ", i);
    fflush(stderr);

    clock_t begin = clock();
    while (!is_everyone_dead(&games[current])) {
      step_game(&games[current]);
    }

    printf("%fs\n", (float)(clock() - begin) / (float) CLOCKS_PER_SEC);
    fflush(stdout);

    if (gnuplot) {
      log_generation(gnulog, i, &games[current]);
      fflush(gnulog);
      log_live_update();
    }

    int next = 1 -  current;
    make_next_generation(&games[current], &games[next]);
    current = next;
  }

  if (gnuplot) {
    log_close_pipe();
    fclose(gnulog);
  }

  dump_game(output_filepath, &games[current]);

  return 0;
}