From 3a0f92f1dfd71a69c9ed17d406a4226efe6da815 Mon Sep 17 00:00:00 2001 From: X11MA Date: Tue, 9 Feb 2016 11:05:41 +0100 Subject: [PATCH] Startup --- script/DelayScan.py | 150 +++++++++++++ script/EnergyScan_backup_160208.py | 328 +++++++++++++++++++++++++++++ 2 files changed, 478 insertions(+) create mode 100644 script/DelayScan.py create mode 100644 script/EnergyScan_backup_160208.py diff --git a/script/DelayScan.py b/script/DelayScan.py new file mode 100644 index 0000000..c4fce1c --- /dev/null +++ b/script/DelayScan.py @@ -0,0 +1,150 @@ +import os +import traceback +import math + +print "\nStarting delay scan - Parameters: ", +print START,END,SIZE,ENERGIES + + +############################################################################### +# Plotting +############################################################################### +""" +task = None +running = False +MCP1 = [] +MCP2 = [] + +def _startPlot(type): + global running + sep = "\t" + line_sep = "\r\n" + print "Starting plot: type " + str(type) + running = True + p = plot(None,name="Energy")[0] + s = p.getSeries(0) + cur = 0 + time.sleep(3.0) + while running: + try: + if otf_start.read() == 0: + break + e = energy.read() + if (MCP_1 == 1 or MCP_2 == 1): + aK1=keithley_1a.read() + aP=FillingPattern.read() + aPr = [] + for z in aP: + aPr.append(z) + SumaP=0 + CamaP=0 + for z in xrange(0,480): + SumaP = SumaP + aPr[z] + for z in xrange(460,470): + CamaP = CamaP + aPr[z] + aK1norm = aK1 * CamaP / SumaP + if MCP_1 == 1: + a1=MCPArray1.read() + a1r = [] + for i in a1: + a1r.append(i) + Sum1a1 = 0 + Cam1a1 = 0 + Sum2a1 = 0 + Cam2a1 = 0 + MCP1.append(line_sep) + MCP1.append(e) + for i in xrange(0,480): + Sum1a1 = Sum1a1 + a1r[i] + for i in xrange(Cam_start,Cam_end+1): + Cam1a1 = Cam1a1 + a1r[i] + MCP1.append(Sum1a1) + MCP1.append(Cam1a1) + MCP1.append(aK1norm) + if NrCounters.read() > 460: + for i in xrange(480,960): + Sum2a1 = Sum2a1 + a1r[i] + for i in xrange(Cam_start+480,Cam_end+481): + Cam2a1 = Cam2a1 + a1r[i] + MCP1.append(Sum2a1) + MCP1.append(Cam2a1) + if Save_array_data == 1: + MCP1.append(a1r) + if MCP_2 == 1: + a2=MCPArray2.read() + a2r = [] + for j in a2: + a2r.append(j) + Sum1a2 = 0 + Cam1a2 = 0 + Sum2a2 = 0 + Cam2a2 = 0 + MCP2.append(line_sep) + MCP2.append(e) + for j in xrange(0,480): + Sum1a2 = Sum1a2 + a2r[j] + for j in xrange(Cam_start,Cam_end+1): + Cam1a2 = Cam1a2 + a2r[j] + MCP2.append(Sum1a2) + MCP2.append(Cam1a2) + MCP2.append(aK1norm) + if NrCounters.read() > 460: + for j in xrange(480,960): + Sum2a2 = Sum2a2 + a2r[j] + for j in xrange(Cam_start+480,Cam_end+481): + Cam2a2 = Cam2a2 + a2r[j] + MCP2.append(Sum2a2) + MCP2.append(Cam2a2) + if Save_array_data == 1: + MCP2.append(a2r) + time.sleep(SamplingTime.read()*0.001) + if (abs(e-cur)) > 0.1: + v = abs((keithley_2a.read() / ((keithley_1a if (type==1) else keithley_3a).read() ))) + s.appendData(e,v) + cur = e + # time.sleep(0.2) + except: + pass + print "Done Plotting" + if MCP_1 == 1: + output_file_MCP1 = output_path+"MCP_1_"+file_prefix+"_" + suffix + ".dat" + MCP1out = open(output_file_MCP1, "a+") + sMCP1 = sep.join(str(x) for x in MCP1) + line_sep # MCP1.write("%s" + sep % i) + sMCP1out = sep + "rbkenergy" + sep + "Sum1" + sep + "Cam1" + sep + "Keithley1_norm" + if NrCounters.read() > 460: + sMCP1out = sMCP1out + sep + "Sum2" + sep + "Cam2" + if Save_array_data == 1: + sMCP1out = sMCP1out + sep + "Array" + sMCP1out = sMCP1out + line_sep + sMCP1 + MCP1out.write(sMCP1out) + MCP1out.close() + print "Saved MCP signal data" + if MCP_2 == 1: + output_file_MCP2 = output_path+"MCP_2_"+file_prefix+"_" + suffix + ".dat" + MCP2out = open(output_file_MCP2, "a+") + sMCP2 = sep.join(str(y) for y in MCP2) + line_sep + sMCP2out = sep + "rbkenergy" + sep + "Sum1" + sep + "Cam1" + sep + "Keithley1_norm" + if NrCounters.read() > 460: + sMCP2out = sMCP2out + sep + "Sum2" + sep + "Cam2" + if Save_array_data == 1: + sMCP2out = sMCP2out + sep + "Array" + sMCP2out = sMCP2out + line_sep + sMCP2 + MCP2out.write(sMCP2out) + MCP2out.close() + print "Saved MCP I_0 data" +def startPlot(type = 1): + global task + task = fork((_startPlot,(type,)),) + +def stopPlot(): + global task, running + running = False + ret = join(task) +""" + + + +print "Finished Delay scan" + +print("Success") + diff --git a/script/EnergyScan_backup_160208.py b/script/EnergyScan_backup_160208.py new file mode 100644 index 0000000..55065fa --- /dev/null +++ b/script/EnergyScan_backup_160208.py @@ -0,0 +1,328 @@ +import os +import traceback +import thread + +############################PGM+ID1+ID2################################################### + +if str(SET_OFFSETS) == "1": + print "Setting offsets" + if RUNTYPE in ["+/-", "+" , "-"]: + caput(OTF_MODE1,1) # circ + in ID1 + caput(OTF_MODE2,2) # circ - in ID + elif RUNTYPE in ["LH/LV", "LH", "LV"]: + caput(OTF_MODE1,0) + caput(OTF_MODE2,0) + wait_channel(OTF_DONE, 1, type = 'i') + caput(OTF_ALPHA1, 0.0) # LH in ID1 + caput(OTF_ALPHA2, 90.0) # LV in ID2 + wait_channel(OTF_DONE, 1, type = 'i') + else: + raise Exception("Invalid run type: " + RUNTYPE) + + caput(OTF_OFF1,OFFSET1) + caput(OTF_OFF2,OFFSET2-40) #detune ID2 + wait_channel(OTF_DONE, 1, type = 'i') + print "Offsets are set" + import sys + sys.exit(0) +else: + print "Running full script" + +file_prefix = time.strftime("%y%m%d") +input_path = "/sls/X11MA/Data1/public/X11MA/temp/"+file_prefix+"/" +output_path = input_path #+file_prefix+"/" #"/sls/X11MA/Data1/public/e10989/"+file_prefix+"/" + + +#Parameters +""" +E1 = 680 +E2 = 750 +TIME = 2 #min +DELAY = 10.0 #s +OFFSET1 = 1.0 #eV +OFFSET2 = -1.0 #eV +PREFIX = 'Data' +RUNTYPE = "+/-" +ROUNDS = 1 +PLOT_TYPE = 1 +""" +print "\nStarting energy scan - Parameters: ", +print E1,E2,TIME,DELAY,OFFSET1,OFFSET2,RUNTYPE#,ALPHA1,ALPHA2 + + +############################################################################### +# Plotting +############################################################################### + +task = None +running = False +MCP1 = [] +MCP2 = [] + +def _startPlot(type): + global running + sep = "\t" + line_sep = "\r\n" + print "Starting plot: type " + str(type) + running = True + p = plot(None,name="Energy")[0] + s = p.getSeries(0) + cur = 0 + time.sleep(3.0) + while running: + try: + if otf_start.read() == 0: + break + e = energy.read() + if (MCP_1 == 1 or MCP_2 == 1): + aK1=keithley_1a.read() + aP=FillingPattern.read() + aPr = [] + for z in aP: + aPr.append(z) + SumaP=0 + CamaP=0 + for z in xrange(0,480): + SumaP = SumaP + aPr[z] + for z in xrange(460,470): + CamaP = CamaP + aPr[z] + aK1norm = aK1 * CamaP / SumaP + if MCP_1 == 1: + a1=MCPArray1.read() + a1r = [] + for i in a1: + a1r.append(i) + Sum1a1 = 0 + Cam1a1 = 0 + Sum2a1 = 0 + Cam2a1 = 0 + MCP1.append(line_sep) + MCP1.append(e) + for i in xrange(0,480): + Sum1a1 = Sum1a1 + a1r[i] + for i in xrange(Cam_start,Cam_end+1): + Cam1a1 = Cam1a1 + a1r[i] + MCP1.append(Sum1a1) + MCP1.append(Cam1a1) + MCP1.append(aK1norm) + if NrCounters.read() > 460: + for i in xrange(480,960): + Sum2a1 = Sum2a1 + a1r[i] + for i in xrange(Cam_start+480,Cam_end+481): + Cam2a1 = Cam2a1 + a1r[i] + MCP1.append(Sum2a1) + MCP1.append(Cam2a1) + if Save_array_data == 1: + MCP1.append(a1r) + if MCP_2 == 1: + a2=MCPArray2.read() + a2r = [] + for j in a2: + a2r.append(j) + Sum1a2 = 0 + Cam1a2 = 0 + Sum2a2 = 0 + Cam2a2 = 0 + MCP2.append(line_sep) + MCP2.append(e) + for j in xrange(0,480): + Sum1a2 = Sum1a2 + a2r[j] + for j in xrange(Cam_start,Cam_end+1): + Cam1a2 = Cam1a2 + a2r[j] + MCP2.append(Sum1a2) + MCP2.append(Cam1a2) + MCP2.append(aK1norm) + if NrCounters.read() > 460: + for j in xrange(480,960): + Sum2a2 = Sum2a2 + a2r[j] + for j in xrange(Cam_start+480,Cam_end+481): + Cam2a2 = Cam2a2 + a2r[j] + MCP2.append(Sum2a2) + MCP2.append(Cam2a2) + if Save_array_data == 1: + MCP2.append(a2r) + time.sleep(SamplingTime.read()*0.001) + if (abs(e-cur)) > 0.1: + v = abs((keithley_2a.read() / ((keithley_1a if (type==1) else keithley_3a).read() ))) + s.appendData(e,v) + cur = e + # time.sleep(0.2) + except: + pass + print "Done Plotting" + if MCP_1 == 1: + output_file_MCP1 = output_path+"MCP_1_"+file_prefix+"_" + suffix + ".dat" + MCP1out = open(output_file_MCP1, "a+") + sMCP1 = sep.join(str(x) for x in MCP1) + line_sep # MCP1.write("%s" + sep % i) + sMCP1out = sep + "rbkenergy" + sep + "Sum1" + sep + "Cam1" + sep + "Keithley1_norm" + if NrCounters.read() > 460: + sMCP1out = sMCP1out + sep + "Sum2" + sep + "Cam2" + if Save_array_data == 1: + sMCP1out = sMCP1out + sep + "Array" + sMCP1out = sMCP1out + line_sep + sMCP1 + MCP1out.write(sMCP1out) + MCP1out.close() + print "Saved MCP signal data" + if MCP_2 == 1: + output_file_MCP2 = output_path+"MCP_2_"+file_prefix+"_" + suffix + ".dat" + MCP2out = open(output_file_MCP2, "a+") + sMCP2 = sep.join(str(y) for y in MCP2) + line_sep + sMCP2out = sep + "rbkenergy" + sep + "Sum1" + sep + "Cam1" + sep + "Keithley1_norm" + if NrCounters.read() > 460: + sMCP2out = sMCP2out + sep + "Sum2" + sep + "Cam2" + if Save_array_data == 1: + sMCP2out = sMCP2out + sep + "Array" + sMCP2out = sMCP2out + line_sep + sMCP2 + MCP2out.write(sMCP2out) + MCP2out.close() + print "Saved MCP I_0 data" +def startPlot(type = 1): + global task + task = fork((_startPlot,(type,)),) + +def stopPlot(): + global task, running + running = False + ret = join(task) + + +############################PGM+ID1+ID2################################################### + + +def switchpol(activeID, runtype): + global pol_str + if activeID == 1: + caput(OTF_OFF1,OFFSET1) + caput(OTF_OFF2,OFFSET2-40) #detune ID2 + if runtype in ["+/-", "+"]: + pol_str = "circ +" + elif runtype in ["LH/LV", "LH"]: + pol_str = "Lin. Horizontal" + elif activeID == 2: + caput(OTF_OFF1,OFFSET1-40) #detune ID1 + caput(OTF_OFF2,OFFSET2) + if runtype in ["+/-", "-"]: + pol_str = "circ -" + elif runtype in ["LH/LV", "LV"]: + pol_str = "Lin. Vertical" + else: + raise Exception("Invalid parameter") + + +pol_str = None +polswitch = 1 + +fid = get_next_fid(input_path, "o" + file_prefix) + + + +############################################################################### +#Prepare scan +############################################################################### +if ID1 == 1: + caput ("X11PHS-E:OPT","PGM+ID1") +elif ID2 == 1: + caput ("X11PHS-E:OPT","PGM+ID2") +else: + caput ("X11PHS-E:OPT","PGM+ID1+ID2") + +number_of_scans = 1 +if RUNTYPE in ["+/-", "+", "-"]: + caput(OTF_MODE1,1) # circ + in ID1 + caput(OTF_MODE2,2) # circ - in ID2 + wait_channel(OTF_DONE, 1, type = 'i') + if RUNTYPE == "+/-": + number_of_scans = 2 * ROUNDS + else: + number_of_scans = ROUNDS +elif RUNTYPE in ["LH/LV", "LH", "LV"]: + caput(OTF_MODE1,0) + caput(OTF_MODE2,0) + wait_channel(OTF_DONE, 1, type = 'i') + caput(OTF_ALPHA1, 0.0) # LH in ID1 + caput(OTF_ALPHA2, 90.0) # LV in ID2 + wait_channel(OTF_DONE, 1, type = 'i') + if RUNTYPE == "LH/LV": + number_of_scans = 2 * ROUNDS + else: + number_of_scans = ROUNDS +else: + raise Exception("Invalid run type: " + RUNTYPE) + +if RUNTYPE in ["-", "LV"]: + switchpol(2, RUNTYPE) # tune ID2 --> polarization: C- or LV + polswitch = 0 +elif RUNTYPE in ["+/-", "+", "LH/LV", "LH"]: + switchpol(1, RUNTYPE) # tune ID1 --> polarization: C+ or LH +time.sleep(1.0) + +wait_channel(OTF_DONE, 1, type = 'i') + +open_vg10() +time.sleep(0.5) +open_vg11() +time.sleep(0.5) +open_vg12() +time.sleep(0.5) +open_vg13() + +for scan_no in range(number_of_scans): + suffix = ("%03d" % fid) + input_file = input_path + "o" + file_prefix + "_" + suffix + ".dat" + + caput(OTF_E1, E1) + caput(OTF_E2, E2) + caput(OTF_TIME, TIME) + caput(OTF_FTS,file_prefix) + caput(OTF_FID,fid) + time.sleep(2.0) + caput(OTF_ESET, E1) + wait_channel(OTF_DONE, 1, type = 'i') + time.sleep(DELAY) + time.sleep(2.0) + + startPlot(PLOT_TYPE) + #Start the OTF scan + #caput(OTF_START, 'GO') + otf_start.write(1) + time.sleep(3.0) + print "Running scan " + str(scan_no+1) + " out of " + str(number_of_scans) + try: + #wait_channel(OTF_START, 'STOP', timeout = int(TIME*60), type = 's') + otf_start.waitValue(0, (15 + int(TIME*60)) *1000) + except: + print "******** OTF STOP TIMEOUT **********" + otf_start.write(0) + finally: + stopPlot() + + time.sleep(5.0) + #TODO: wait for file instead of sleep + #Convert file + output_file = output_path+"os"+file_prefix+"_" + suffix + ".dat" + + print("Converting data file: " + output_file); + convert_file(input_file, output_file, pol_str) + plot_file(output_file, file_prefix+"_" + suffix) #"Scan " + str(scan_no+1)) + print "Finished scan " + str(scan_no+1) + " out of " + str(number_of_scans) + + if RUNTYPE in ["+/-", "LH/LV"]: + if polswitch == 1: + switchpol(2, RUNTYPE) # tune ID2 --> polarization: C- or LV + polswitch = 0 + else: + polswitch = 1 + switchpol(1, RUNTYPE) # tune ID1 --> polarization: C+ or LH + else: + print "running in one polarization mode, no switching" + + time.sleep(3.0) + fid = fid + 1 + +caput(OTF_ESET, E1) +close_vg13() + +print "Finished Energy scan" + +print("Success") +