# checkConvergence.py

import matplotlib.pyplot as plt

number = input("1 for constrain score, 2 for len score, 3 for orientation score, 4 for total score")

if number == 1 :
    fp = open("Episode-constrain-score.txt", "r")
    print("now showing Episode-constrain-score")
elif number == 2 :
    fp = open("Episode-len-score.txt", "r")
    print("now showing Episode-len-score")
elif number == 3 :
    fp = open("Episode-orientation-score.txt", "r")
    print("now showing Episode-orientation-score")
else :
    fp = open("Episode-score.txt", "r")
    print("now showing Episode-score")

lines = fp.read().splitlines()
scores = list(map(float, lines))
episode = list(range(1, 1+len(scores)))
print(scores)
print(episode)
plt.figure()
plt.plot(episode, scores)
plt.xlabel("Episode",fontsize=13,fontweight='bold')
plt.ylabel("Score",fontsize=13,fontweight='bold')
#plt.show()
plt.savefig('trend.png')