import matplotlib.pyplot as plt

def main():
    fig_a, ax_a = plt.subplots()
    fig_b, ax_b = plt.subplots()
    # fig_c, ax_c = plt.subplots()
    for i in range(6):
        x = []
        abs_err = []
        rel_err = []
        with open(f"output/error/out_{10**(i+1)}.txt", "r") as f:
            lines = f.readlines()
            for line in lines:
                x_i, abs_err_i, rel_err_i = line.strip().split(",")
                x.append(float(x_i))
                abs_err.append(float(abs_err_i))
                rel_err.append(float(rel_err_i))

        ax_a.plot(x, abs_err, label=f"abs_err {10**(i+1)} steps")
        ax_b.plot(x, rel_err, label=f"rel_err {10**(i+1)} steps")

    ax_a.legend(bbox_to_anchor=(1.05, 1), loc=2, borderaxespad=0.)
    ax_b.legend(bbox_to_anchor=(1.05, 1), loc=2, borderaxespad=0.)
    fig_a.savefig("../latex/images/problem8_a.pdf", bbox_inches="tight")
    fig_b.savefig("../latex/images/problem8_b.pdf", bbox_inches="tight")

if __name__ == "__main__":
    main()