from pathlib import Path

import matplotlib.pyplot as plt
import numpy as np
from scipy.stats import linregress


def plot_timing(indir, outdir):
    files = [
        "lattice_sizes.txt",
        "sample_sizes.txt",
    ]
    labels = [
        "Lattice sizes",
        "Sample sizes",
    ]
    xlabels = [
        "Lattice size",
        "Sampling size"
    ]
    outfiles = [
        "lattice_size.pdf",
        "sample_sizes.pdf"
    ]

    for file, label, xlabel, outfile in zip(files, labels, xlabels, outfiles):
        figure1, ax1 = plt.subplots()
        x = []
        t = []

        with open(Path(indir, file)) as f:
            lines = f.readlines()
            for line in lines:
                l = line.strip().split(",")
                x.append(float(l[0]))
                t.append(float(l[1]))


        ax1.plot(x, t, label=label)
        ax1.set_xlabel(xlabel)
        ax1.set_ylabel("time (seconds)")

        figure1.legend()

        figure1.savefig(Path(outdir, outfile))

        plt.close(figure1)


if __name__ == "__main__":
    plot_timing(
        "output/timing/",
        "../latex/images/timing",
    )