#include <random>
#include <iostream>

#include <cholmod.h>

#include "randutils/randutils.hpp"

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

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

int main(int argc, char* argv[]) {
  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 (use_aN) {
    sample meas(n, a, beta);

    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';
        }
      }
    }
  } else {
    sample meas(Lx, Ly, beta);

    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';
        }
      }
    }
  }

  CHOL_F(finish)(&c);

  return 0;
}