""" TSintro.py <fine> <coarse> <quantized> <plot_file>
"""
import matplotlib, sys, os.path, scipy
matplotlib.use('PDF')
import pylab
name_fine, name_coarse, name_quantized, plot_file = sys.argv[1:5]

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

fine = read_data(name_fine)
coarse = read_data(name_coarse)
quantized = read_data(name_quantized)

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

pylab.subplot(2,1,1)
pylab.plot(fine[0],fine[1],'b-',coarse[0],coarse[1],'ro')
pylab.xlabel(r'$\tau$')
pylab.ylabel(r'$x_1(\tau)$')
pylab.axis(ymax=17)


pylab.subplot(2,1,2)
pylab.plot(quantized[0],quantized[1],'kd')
pylab.xlabel(r'$t$')
pylab.ylabel(r'$y(t)$')
pylab.axis(ymin=0.5,ymax=4.5)

pylab.savefig(plot_file)