#include <cstring>
#include <fstream>
#include <string>
#include <cinttypes>
#include <sstream>
#include <functional>
#include <iostream>
#include <array>

#include <hooks.hpp>

#include "analysis_tools.hpp"

class RealField {
  public:
    coordinate L;
    std::array<unsigned, 2> M;
    std::vector<long unsigned> N;
    std::vector<double> F;

    RealField(coordinate L, std::array<unsigned, 2> M) : L(L), M(M), N(M[0] * M[1], 0), F(M[0] * M[1], 0.0) {}

    void add(coordinate r, double x) {
      unsigned ind = (unsigned)floor(M[1] * (r.y / L.y)) + M[1] *
        (unsigned)floor(M[0] * (r.x / L.x));
      N[ind]++;
      F[ind] += x;
    }

};

class ma : public hooks {
  // need:
  //  - interface for turning on and off specific measurements
  //
  private:
    unsigned Mx;
    unsigned My;
    unsigned num;
    uint64_t Nc;
    uint64_t Nl;
    uint64_t Nm;
    uint64_t Ns;
    uint64_t Nn;
    uint64_t Nb;
    uint64_t Na;
    uint64_t NA;
    uint64_t Nq;
    
    // measurement storage
    std::vector<uint64_t> sc;
    std::vector<uint64_t> sn; // non-spanning cluster size distribution
    std::vector<uint64_t> ss; // minimal spanning cluster size distribution
    std::vector<uint64_t> sm; // spanning cluster size distribution
    std::vector<uint64_t> sl; // final avalanche size distribution
    std::vector<uint64_t> sb; // final avalanche size distribution
    std::vector<uint64_t> sd; // final avalanche size distribution
    std::vector<uint64_t> sa; // non-final avalanche size distribution
    std::vector<uint64_t> sA; // non-final avalanche size distribution

    std::vector<uint64_t> si;
    std::vector<uint64_t> sI;

    std::vector<uint64_t> cc;
    std::vector<uint64_t> cn;
    std::vector<uint64_t> cs;
    std::vector<uint64_t> cm;
    std::vector<uint64_t> cl;
    std::vector<uint64_t> cb;
    std::vector<uint64_t> ca;
    std::vector<uint64_t> cA;
    std::vector<uint64_t> cq;

    std::list<std::array<double, 2>> aG;
    RealField fS;
    std::vector<uint64_t> sp;

    ClusterTree last_clusters;
    current_info last_currents;
    std::vector<bool> last_backbone;

  public:
    long double lv;

    std::list<std::list<unsigned>> avalanches;
    std::list<unsigned> last_avalanche;
    std::string model_string;

    ma(unsigned Lx, unsigned Ly, double beta, double w);
    ma(unsigned n, double a, double beta, double w);
    ~ma();

    void pre_fracture(const network &) override;
    void bond_broken(const network& net, const current_info& cur, unsigned i) override;
    void post_fracture(network &n) override;
};