Commit before breaking everything

[matches/honours.git] / research / analysis and stuff / tcs scripts / au_focus.py

diff --git a/research/analysis and stuff/tcs scripts/au_focus.py b/research/analysis and stuff/tcs scripts/au_focus.py
new file mode 100755 (executable)
index 0000000..fd1e85f
--- /dev/null
+++ b/research/analysis and stuff/tcs scripts/au_focus.py
@@ -0,0 +1,52 @@
+#!/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/50))
+
+               
+
+#os.system("cd "+imagedir+"; for i in *.eps; do epstopdf --autorotate=All $i; done; rm *.eps")
+