from sys import argv
from glob import glob
import matplotlib.pyplot as plt
import numpy as np


search_term = argv[1]


# for 250 ms, 10 spike
# Kaue...pt3/steady... 
# ycol=9 numvals=16 offset=10

# for 100 ms 1 spike

 
xcol=1
ycol=0
#errcol=8
numvals=1 # number of 2nd parameter shifts
offset=0 #
# errcol 0,1,2 are x,y,freq
# 25 ms steps after that

errcol=ycol+1
#print len(argv)
if len(argv) > 2:
	for i in range(len(argv)-2):
		things =argv[i+2]
		try: 
			exec(things)
		except:
			exec('%s =\'%s\'' % (things.split('=')[0],things.split('=')[1]))

xcol = int(xcol)
name = (ycol+1)*25
ycol = 2*int(ycol)+3
numvals = int(numvals)
errcol = int(ycol)+1
offset = int(offset)%numvals

files = glob('*%s*' % search_term)
print files
fig, ax = plt.subplots()

files.sort() # will always put Ex3d first
color = ['red','blue']
i=0
for fix in files:
	print fix

	#datafile = 'stats'+fix
	errorfile = fix
	#= np.loadtxt(datafile,unpack=True,usecols=[xcol,ycol])
	xdat,ydat,errdat, zdata = np.loadtxt(errorfile,unpack=True, usecols=[xcol,ycol,errcol,1-xcol])
	#print len(xdat)/51,len(errdat)
	plt.title('%d ms, %e' % (name,zdata[offset]))
	xdat = xdat[offset::numvals]
	ydat = ydat[offset::numvals]
	errdat= errdat[offset::numvals]
	#print xdat[0],ydat[0],errdat[0]
	plt.plot(xdat,ydat,color=color[i%2])
	plt.fill_between(xdat,np.subtract(ydat,errdat),np.add(ydat,errdat),alpha=0.5,color=color[i%2])
	i+=1


ax.set(xlim=(0,0.1),ylim=(0,1.2))

plt.show()