import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd
import os
from utility import visualization, simulation


def plot_results():

    sim_names = [
        "96.3965",
        "204.3.486",
        "204.55.187",
        "408.33.844",
        "816.1A4.845",
        "999.111.3.5543",
        "999.111.3.5565",
        "PEGReg252x504"
    ]

    deserializer = simulation.SimulationDeSerializer(save_dir="sim_saves", results_dir="sim_results")

    data = []
    for sim_name in sim_names:
        df, metadata = deserializer.read_results(sim_name)
        df = df.loc[:, ~df.columns.str.contains('^Unnamed')]

        graph_title = sim_name
        line_labels = metadata["labels"]

        graph_tuple = (graph_title, df, line_labels)
        data.append(graph_tuple)

    sns.set_theme()
    fig = visualization.plot_BERs(title="Bit-Error-Rates of proximal decoder for different codes",
                                  data=data, num_cols=4)

    plt.show()


if __name__ == "__main__":
    plot_results()