#!/usr/bin/python

from process import *
import odict
import time

tcsdir = "../../research/TCS/"
imagedir = "au_focus/"

os.system("mkdir -p " + imagedir.strip("/"))

toplot = ["2012-10-10/focusA", "2012-10-11/before_refocus", "2012-10-11/attempt1", "2012-10-11/attempt2", "2012-10-11/attempt3", "2012-10-11/attempt4", "2012-10-11", "2012-10-12"]

focus = []

gnuplot.reset()
gnuplot("set term postscript colour")
gnuplot("set output \""+imagedir+"focus.eps")
gnuplot("set key top left")
for d in toplot:
	for f in FindDataFiles(tcsdir+d):
		params = ReadParameters(f)
		if not "Deflection Voltage" in params.keys():
			continue
		if not "Accelerating Voltage" in params.keys():
			continue
		if not "Focus Voltage" in params.keys():
			continue
		if not "Venault Voltage" in params.keys():
			continue
		if not "Sample" in params.keys():
			continue

		params = [params["Deflection Voltage"], params["Accelerating Voltage"], params["Focus Voltage"], params["Venault Voltage"], params["Sample"]]
		if params in focus:
			continue

		print "Plotting " + str(f)
		
		focus.append(params)
		title = "Vd="+str(params[0])+"\\tVa="+str(params[1])+"\\tVf="+str(params[2])+"\\tVw="+str(params[3])
		title += "\\nSample: " + str(params[4])
		
		data = GetData(f)
		ShowData(data, calibrate=True, normalise=True, plot=gnuplot.plot, title=d+BaseName(f),master_title=title+"\\nI(E) Normalised")
		gnuplot("set title \""+title+"\\S(E)")
		ShowTCS(data, calibrate=True, normalise=True, plot=gnuplot.plot, title=d+BaseName(f), master_title=title+"\\nTotal Current Spectrum S(E)", step=len(data)/4000)

		

#os.system("cd "+imagedir+"; for i in *.eps; do epstopdf --autorotate=All $i; done; rm *.eps")