#!/usr/bin/python

from process import *
import odict
import time

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

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

allplots = [ ("set1", odict.odict([
	#("Si (initial)" , [tcsdir+"2012-09-27/Si/095131.dat", 1])
	("Si", [tcsdir+"2012-09-27/Si (focused)/140416.dat", 5]) # For some reason I altered the focus of the gun, not a good idea
	,("Au (30min 4.0A)", [tcsdir+"2012-10-01/average.dat", 1])
	,("Au (100min 4.0A)", [tcsdir+"2012-10-03/Au (100min) on Si/average.dat", 1])
	,("Au (20min 4.6A + 120min 4.0A)", [tcsdir+"2012-10-03/Au (140min) on Si/average.dat", 1])
	,("Au (80min 4.6A + 120min 4.0A)", [tcsdir+"2012-10-03/Au (200min) on Si/average.dat", 1])
	,("Au (30min 4.8A + 80min 4.6A + 120min 4.0A)", [tcsdir+"2012-10-04/Au (230min) on Si/average.dat", 1])
	#,("Au (30min 4.8A + 80min 4.6A + 120min 4.0A) - REFOCUS", [tcsdir+"2012-10-04/Au (230min) on Si - Refocused/average.dat", 1])
])), ("set2", odict.odict([
	("Au (30min 4.8A + 80min 4.6A + 120min 4.0A)", [tcsdir+"2012-10-09/Au (230min) on Si/average.dat", 1])
	,("Au (30min 5.2A + 30min 4.8A + 80min 4.6A + 120min 4.0A)", [tcsdir+"2012-10-09/Au (260min) on Si/average.dat", 1])]))

]

for p in allplots:

	toplot = p[1]
	# I(E) (normalised) data
	gnuplot.reset()

	for f in toplot.items():
		d = ShowData(f[1][0], title=f[0], plot=gnuplot.replot, calibrate=True, normalise=True)

	gnuplot("set term postscript colour")
	gnuplot("set output \""+imagedir+str(p[0])+"_"+"normalised_data.eps\"")
	title="Time Evolution of Au on Si I(E) curves"
	if (p[0] == "set1"):
		title += "\\nFocus not altered between curves; different focus to set2"
	else:
		title += "\\nFocus not altered between curves; different focus to set1"
	gnuplot("set title \""+str(title)+"\"")
	gnuplot("set key top left")
	gnuplot.replot()

	time.sleep(1)

	# TCS (normalised) data
	gnuplot.reset()
	gnuplot("set term wxt")

	for f in toplot.items():
		d = ShowTCS(f[1][0], title=f[0], plot=gnuplot.replot, calibrate=True, normalise=True, with_="lp", step=f[1][1])

	gnuplot("set term postscript colour")
	gnuplot("set output \""+imagedir+str(p[0])+"_"+"tcs_normalised.eps\"")
	title="Time Evolution of Au on Si I(E) curves"
	if (p[0] == "set1"):
		title += "\\nFocus not altered between curves; different focus to set2"
	else:
		title += "\\nFocus not altered between curves; different focus to set1"
	gnuplot("set title \""+str(title)+"\"")
	gnuplot("set key top right")
	gnuplot.replot()

	time.sleep(1)


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