import matplotlib.pyplot as plt

import numpy as np

from config import config

class PlotData():

def __init__(self, running_mean_len = config.graph_running_mean):

if config.graph_window:

plt.ion()

self.points = []

self.running_mean = []

self.running_mean_big = []

self.running_mean_len = running_mean_len

self.running_mean_big_len = running_mean_len * 5

def new_data(self, new_point):

self.points.append(new_point)

self.running_mean.append(np.mean(self.points[-self.running_mean_len:]))

self.running_mean_big.append(np.mean(self.points[-self.running_mean_big_len:]))

pass

def clear(self):

plt.ioff()

self.points = []

self.running_mean = []

def graph(self):

if True:

plt.scatter(list(range(len(self.points))), self.points, s=.05, c="y")

plt.plot(self.running_mean, c="b")

plt.plot(self.running_mean_big, c="r")

# plt.autoscale_view()

if config.graph_window:

plt.pause(0.05)

plt.show()

def save(self, name):

self.graph()

plt.savefig(name + ".png")