""" SGO.py <simS_data> <simY_data> <decodeS_data> <simS_plot>
<simY_plot> <decodeS_plot>
"""
import matplotlib, sys, os.path, scipy
matplotlib.use('PDF')
import pylab
simS_data, simY_data, decodeS_data, simS_plot, simY_plot, decodeS_plot = \
    sys.argv[1:7]
   
def read_data(data_file):
    # Read in "data_file" as a 2-d array
    f = file(data_file, 'r')
    data = [[float(x) for x in line.split()] for line in f.xreadlines()]
    f.close()
    return scipy.array(data).T

params = {'axes.labelsize': 12,
               'text.fontsize': 10,
               'legend.fontsize': 10,
               'text.usetex': True,
               'xtick.labelsize': 11,
               'ytick.labelsize': 11}
pylab.rcParams.update(params)

for data_file,plot_file,y_label,yrange in (
        (simS_data, simS_plot, 'S(t)',(-.1,1.1)),
        (simY_data, simY_plot, 'y(t)',(-4,4)),
        (decodeS_data, decodeS_plot, 'S(t)',(-.1,1.1)) ):
    data = read_data(data_file)
    pylab.plot(data[0],data[1],'b-')
    pylab.xlabel(r'$t$')
    pylab.xticks(pylab.arange(0,101,25))
    pylab.ylabel(y_label)
    V = list(pylab.axis())
    V[2] = yrange[0]
    V[3] = yrange[1]
    pylab.axis(V)
    pylab.yticks(pylab.arange(int(yrange[0]),int(yrange[1])+.1,int(yrange[1])))
    pylab.savefig(plot_file)
    pylab.close()