"""
LDA.py mean AA AN C LDA1 LDA2

"""

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

mean, AA, AN, C, LDA1, LDA2, = sys.argv[1:7]

def read_data(data_file):
    # Read in "data_file" as an 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

mean = read_data(mean)
AA   = read_data(AA)
AN   = read_data(AN)
C    = read_data(C)

# Do the plotting
pylab.clf() # Clear figure window
pylab.figure(figsize=(5,5.5),dpi=80)
ax1 = pylab.subplot(2,1,1)
pylab.plot(mean[0],mean[1],'r-',mean[0],mean[2],'g-',mean[0],mean[3],'b-')
pylab.legend((r'$\mu_C$',r'$\mu_N$',r'$\mu_A$'))
pylab.xticks([], visible=False)
pylab.yticks([], visible=False)
pylab.ylabel(r'PSD')
pylab.xlim(0,250)

pylab.subplot(2,1,2)
pylab.plot(mean[0],mean[4],'r-',mean[0],mean[5],'b-')
pylab.legend((r'$v_1$',r'$v_2$'))
xrng = scipy.arange(0,250,80)
pylab.xticks(xrng,  [ '$%# 2i$' % l for l in xrng/8 ])
pylab.yticks([], visible=False)
pylab.xlabel(r'cpm')
pylab.ylabel(r'PSD')
pylab.xlim(0,250)

pylab.savefig(LDA1)

pylab.clf() # Clear figure window
pylab.figure(figsize=(4,9.5),dpi=200)
box = [-0.7,0.25,-.7,.5]
pylab.subplot(3,1,1)
pylab.plot(C[0],C[1],'r,')
pylab.axis(box)
pylab.yticks([], visible=False)
pylab.xticks([], visible=False)
pylab.legend(('C',))
pylab.subplot(3,1,2)
pylab.plot(AN[0],AN[1],'g,')
pylab.axis(box)
pylab.yticks([], visible=False)
pylab.xticks([], visible=False)
pylab.legend(('N',))
pylab.subplot(3,1,3)
pylab.plot(AA[0],AA[1],'b,')
pylab.axis(box)
pylab.yticks([], visible=False)
pylab.xticks([], visible=False)
pylab.legend(('A',))

pylab.savefig(LDA2)

# Local Variables:
# mode: python
# End: