#include <random>
#include <iostream>

#include <cholmod.h>

#include "randutils/randutils.hpp"

#include <graph.hpp>
#include <network.hpp>
#include <hooks.hpp>
#include "animate.hpp"

#include <csignal>
#include <cstring>
#include <atomic>

std::atomic<bool> quit(false);    // signal flag

void got_signal(int) {
    quit.store(true);
}

int main(int argc, char* argv[]) {
  struct sigaction sa;
  memset( &sa, 0, sizeof(sa) );
  sa.sa_handler = got_signal;
  sigfillset(&sa.sa_mask);
  sigaction(SIGINT, &sa, NULL);

  int opt;

  unsigned N = 1;
  double Lx = 16.0;
  double Ly = 16.0;
  double  beta = 0.5;

  while ((opt = getopt(argc, argv, "X:Y:N:b:")) != -1) {
    switch (opt) {
      case 'N':
        N = (unsigned)atof(optarg);
        break;
      case 'X':
        Lx = atof(optarg);
        break;
      case 'Y':
        Ly = atof(optarg);
        break;
      case 'b':
        beta = atof(optarg);
        break;
      default:
        exit(1);
    }
  }

  cholmod_common c;
  CHOL_F(start)(&c);

  animate meas(Lx, Ly, 700, argc, argv);

  randutils::auto_seed_128 seeds;
  std::mt19937 rng{seeds};

  for (unsigned trial = 0; trial < N; trial++) {
    graph G(Lx, Ly, rng);
    elastic_network elastic_network(G, &c);
    elastic_network.set_thresholds(beta, rng);
    elastic_network.fracture(meas);

    if (quit.load())
      break;
  }

  CHOL_F(finish)(&c);

  return 0;
}