research/analysis and stuff/tcs scripts/au_focus.py~

   1 #!/usr/bin/python
   2
   3 from process import *
   4 import odict
   5 import time
   6
   7 tcsdir = "../../research/TCS/"
   8 imagedir = "au_focus/"
   9
  10 os.system("mkdir -p " + imagedir.strip("/"))
  11
  12 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"]
  13
  14 focus = []
  15
  16 gnuplot.reset()
  17 gnuplot("set term postscript colour")
  18 gnuplot("set output \""+imagedir+"focus.eps")
  19 gnuplot("set key top left")
  20 for d in toplot:
  21         for f in FindDataFiles(tcsdir+d):
  22                 params = ReadParameters(f)
  23                 if not "Deflection Voltage" in params.keys():
  24                         continue
  25                 if not "Accelerating Voltage" in params.keys():
  26                         continue
  27                 if not "Focus Voltage" in params.keys():
  28                         continue
  29                 if not "Venault Voltage" in params.keys():
  30                         continue
  31                 if not "Sample" in params.keys():
  32                         continue
  33
  34                 params = [params["Deflection Voltage"], params["Accelerating Voltage"], params["Focus Voltage"], params["Venault Voltage"], params["Sample"]]
  35                 if params in focus:
  36                         continue
  37
  38                 print "Plotting " + str(f)
  39
  40                 focus.append(params)
  41                 title = "Vd="+str(params[0])+"\\tVa="+str(params[1])+"\\tVf="+str(params[2])+"\\tVw="+str(params[3])
  42                 title += "\\nSample: " + str(params[4])
  43
  44                 data = GetData(f)
  45                 ShowData(data, calibrate=True, normalise=True, plot=gnuplot.plot, title=d+BaseName(f),master_title=title+"\\nI(E) Normalised")
  46                 gnuplot("set title \""+title+"\\S(E)")
  47                 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)
  48
  49
  50
  51 #os.system("cd "+imagedir+"; for i in *.eps; do epstopdf --autorotate=All $i; done; rm *.eps")
  52