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

#include <hooks.hpp>

#include "analysis_tools.hpp"

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 NC;
    uint64_t Nl;
    uint64_t Nm;
    uint64_t Ns;
    uint64_t Nn;
    uint64_t Nb;
    uint64_t NB;
    uint64_t Na;
    uint64_t Ne;
    uint64_t NA;
    uint64_t NE;
    uint64_t Nq;
    
    // measurement storage
    std::vector<uint64_t> sc;
    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> sB; // final avalanche size distribution
    std::vector<uint64_t> sd; // final avalanche size distribution
    std::vector<uint64_t> sD; // final avalanche size distribution
    std::vector<uint64_t> sf; // 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> se; // non-final avalanche size distribution
    std::vector<uint64_t> sE; // non-final avalanche size distribution

    std::vector<uint64_t> si;
    std::vector<uint64_t> sI;
    std::vector<uint64_t> sy;
    std::vector<uint64_t> sY;
    std::vector<uint64_t> sz;
    std::vector<uint64_t> sZ;

    std::vector<uint64_t> cc;
    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> cB;
    std::vector<uint64_t> ca;
    std::vector<uint64_t> ce;
    std::vector<uint64_t> cA;
    std::vector<uint64_t> cE;
    std::vector<uint64_t> cq;

    ClusterTree last_clusters;
    ClusterTree last_clusters_v;

    std::vector<bool> last_backbone_c;
    std::vector<bool> last_backbone_v;

  public:
    long double lc;
    long double lv;
    long double lm;

    std::list<double> moduliA;
    std::list<double> moduliE;
    std::list<std::list<unsigned>> avalanches;
    std::list<std::list<unsigned>> evalanches;
    std::list<unsigned> last_avalanche;
    std::list<unsigned> last_evalanche;
    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;
};