#include <random>
#include <iostream>

#include <cholmod.h>

#include "randutils/randutils.hpp"

#include <graph.hpp>
#include <network.hpp>
#include <hooks.hpp>
#include "measurements.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;
  unsigned Lx = 16;
  unsigned Ly = 16;
  double  beta = 0.5;

  unsigned n = 128;
  double a = 1.0;
  bool use_aN = false;
  double w = 0.5;

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

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

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

  if (w == 0 || w == 1) {
  if (use_aN) {
    ma meas(n, a, beta, w);

    for (unsigned trial = 0; trial < N; trial++) {
      while (true) {
        try {
          graph G(n, a, rng);
          fuse_network n(G, &c);
          n.set_thresholds(beta, rng);
          n.fracture(meas);
          break;
        } catch (std::exception &e) {
          std::cout << e.what() << '\n';
        }
      }

      if (quit.load())
        break;
    }
  } else {
    ma meas(Lx, Ly, beta, w);

    const graph G(Lx, Ly);
    const fuse_network n(G, &c);

    for (unsigned trial = 0; trial < N; trial++) {
      while (true) {
        try {
          fuse_network net = n;
          net.set_thresholds(beta, rng);
          net.fracture(meas);
          break;
        } catch (std::exception &e) {
          std::cout << e.what() << '\n';
        }
      }

      if (quit.load())
        break;
    }
  }
  } else {
  if (use_aN) {
    ma meas(n, a, beta, w);

    for (unsigned trial = 0; trial < N; trial++) {
      while (true) {
        try {
          graph G(n, a, rng);
          elastic_network fuse_network(G, &c, w);
          fuse_network.set_thresholds(beta, rng);
          fuse_network.fracture(meas);
          break;
        } catch (std::exception &e) {
          std::cout << e.what() << '\n';
        }
      }

      if (quit.load())
        break;
    }
  } else {
    ma meas(Lx, Ly, beta, w);

    const graph G(Lx, Ly);
    const elastic_network fuse_network(G, &c, w);

    for (unsigned trial = 0; trial < N; trial++) {
      while (true) {
        try {
          elastic_network net = fuse_network;
          net.set_thresholds(beta, rng);
          net.fracture(meas);
          break;
        } catch (std::exception &e) {
          std::cout << e.what() << '\n';
        }
      }

      if (quit.load())
        break;
    }
  }
  }

  CHOL_F(finish)(&c);

  return 0;
}