sys.path.append(lib_path)
import process
-dirs = {}
+dirs = process.GetDataSets(".")
-for f in os.listdir("."):
- if os.path.isdir(f) == True:
- data_set = []
- for datafile in os.listdir("./"+str(f)):
- data_set.append(map(lambda e : [e[1], e[2]], process.GetData("./"+str(f)+"/"+str(datafile))))
- avg = process.Average(data_set)
- dirs.update({f : avg})
+#process.gnuplot("set title \"Vary Deflection Voltage\"")
+#process.gnuplot("set xlabel \"DAC (counts)\"")
+#process.gnuplot("set ylabel \"Normalised S(E)\"")
+#process.gnuplot("set key outside right")
-process.gnuplot("set title \"Vary Deflection Voltage\"")
-process.gnuplot("set xlabel \"DAC (counts)\"")
-process.gnuplot("set ylabel \"Normalised S(E)\"")
-
-for p in dirs.items():
+max_curve = []
+for p in sorted(dirs.items(), key = lambda e : float(e[0].split("=")[1])):
if (len(p[1]) > 0):
- process.gnuplot.replot(process.Gnuplot.Data(p[1], title=str(p[0])+"", with_="lp"))
+ maxval = max(p[1], key = lambda e : e[1])
+ max_curve.append([float(p[0].split("=")[1]), maxval[1], maxval[0]])
+# process.gnuplot.replot(process.Gnuplot.Data(p[1], title=str(p[0])+"", with_="lp"))
+
+process.gnuplot.replot(process.Gnuplot.Data(max_curve, with_="lp", using="1:2", title="Max"))
+process.gnuplot.replot(process.Gnuplot.Data(max_curve, with_="lp", using="1:3", title="Max position"))
sys.stdin.readline()