import pylab, numpy, os, sys
from phi_driven_data import *

DATA = rundata["0"] + rundata["1"]

DATA = [elem for elem in DATA if elem[1] != None]

DATA.sort(lambda x, y: cmp(x[0], y[0]))

DATA2 = []

for i in range(len(DATA)):
	sum = 0
	if i < 100 or len(DATA)-i < 100:
		continue
	for z in range(1, 100):
		sum += DATA[i-z][1]
		sum += DATA[i+z][1]
		sum += DATA[i][1]
	DATA2.append([DATA[i][0], float(sum) / 200])


x = [el[0] for el in DATA2 if el[1] != None]
y = [el[1] for el in DATA2 if el[1] != None]

pylab.xlabel("Time of death (steps)")
pylab.ylabel("Integrated information")

pylab.title("Integrated Information of Polyworldians at Death over Evolutionary Time")

pylab.plot(x,y, 'g.')
pylab.show()