import matplotlib.pyplot as plt
import numpy as np


def plot_learning_result(values, soften=20, legend=None):
    # TODO popraw srednia kroczaco - pierwsze n liczb nie jest usrednianych
    # https://stackoverflow.com/questions/14313510/how-to-calculate-moving-average-using-numpy
    if legend is not None:
        title_str = "Alpha: {}, Epsilon: {},\n Gamma: {}, Epsilon decay: {}, Alpha decay: {}".format(legend["alpha"],
                                                                                                    legend["epsilon"],
                                                                                                    legend['gamma'],
                                                                                                    legend['epsilon decay'],
                                                                                                     legend['alpha decay'])
    else:
        title_str = "Reward, moving average"
    values_avg = [np.mean(values[i-soften:i]) for i in range(soften, len(values))]
    plt.plot(np.arange(0, len(values_avg)), values_avg)
    plt.title(title_str)
    plt.show()