#include "switchchain.hpp"
#include "exports.hpp"
#include "graph.hpp"
#include "graph_ecm.hpp"
#include "graph_powerlaw.hpp"
#include <algorithm>
#include <fstream>
#include <iostream>
#include <numeric>
#include <random>
#include <vector>

int main(int argc, char *argv[]) {
    // Simulation parameters
    const int numVerticesMin = 1000;
    const int numVerticesMax = 10000;
    const int numVerticesStep = 1000;

    float tauValues[] = {2.05f, 2.1f, 2.2f, 2.3f, 2.4f, 2.5f,
                         2.6f,  2.7f, 2.8f, 2.9f, 2.95f};

    auto getMixingTime = [](int n, float tau) {
        return int(50.0f * (50.0f - 5.0f * (tau - 2.0f)) * n);
    };
    auto getMeasurements = [](int n, float tau) {
        (void)n;
        (void)tau;
        return 5000;
    };
    auto getMeasureSkip = [](int n, float tau) {
        (void)tau;
        return 30 * n; // Take a sample every ... steps
    };

    // Output file
    std::ofstream outfile;
    if (argc >= 2)
        outfile.open(argv[1]);
    else
        outfile.open("graphdata_ecm_initialtris.m");
    if (!outfile.is_open()) {
        std::cout << "ERROR: Could not open output file.\n";
        return 1;
    }

    // Output Mathematica-style comment to indicate file contents
    outfile << "(*\n";
    outfile << "n from " << numVerticesMin << " to " << numVerticesMax
            << " step " << numVerticesStep << std::endl;
    outfile << "tauValues: " << tauValues << std::endl;
    outfile << "Canonical degree sequence.\n";
    outfile << "mixingTime: 50 * (50 - 5 (tau - 2)) n\n";
    outfile << "measurements: 5000\n";
    outfile << "measureSkip: 30 n\n";
    outfile << "data:\n";
    outfile << "1: {n,tau}\n";
    outfile << "2: {uniform tri samples}\n";
    outfile << "3: {ECM initial tri samples} \n";
    outfile << "*)" << std::endl;

    // Mathematica does not accept normal scientific notation
    outfile << std::fixed;
    outfile << '{' << '\n';
    bool outputComma = false;

    std::mt19937 rng(std::random_device{}());
    Graph g;
    for (int numVertices = numVerticesMin; numVertices <= numVerticesMax;
         numVertices += numVerticesStep) {
        for (float tau : tauValues) {
            int mixingTime = getMixingTime(numVertices, tau);
            int measurements = getMeasurements(numVertices, tau);
            int measureSkip = getMeasureSkip(numVertices, tau);

            // For a single n,tau take samples over several instances of
            // the degree distribution.
            DegreeSequence ds;
            generateCanonicalPowerlawGraph(numVertices, tau, g, ds);

            std::cout << "Running (n,tau) = (" << numVertices << ',' << tau
                      << ")." << std::flush;

            //
            // ECM triangles
            //
            std::vector<int> ECMtris;

            for (int i = 0; i < measurements; ++i) {
                Graph gtemp;
                if (erasedConfigurationModel(ds, gtemp, rng)) {
                    ECMtris.push_back(gtemp.countTriangles());
                }
            }

            std::cout << " Finished ECM samples." << std::flush;

            // Uniform triangles
            std::vector<int> uniformTris;

            SwitchChain chain;
            if (!chain.initialize(g)) {
                std::cerr << "Could not initialize Markov chain.\n";
                return 1;
            }

            for (int i = 0; i < mixingTime; ++i) {
                chain.doMove();
            }
            chain.g.getTrackedTriangles() = chain.g.countTriangles();
            for (int i = 0; i < measurements; ++i) {
                for (int j = 0; j < measureSkip; ++j)
                    chain.doMove(true);
                uniformTris.push_back(chain.g.getTrackedTriangles());
            }

            std::cout << " Finished uniform samples." << std::flush;

            if (outputComma)
                outfile << ',' << '\n';
            outputComma = true;

            outfile << '{';
            outfile << '{' << numVertices << ',' << tau << '}';
            outfile << ',' << uniformTris;
            outfile << ',' << ECMtris;
            outfile << '}' << std::flush;

            std::cout << std::endl;
        }
    }
    outfile << '\n' << '}';
    return 0;
}