diff --git a/script/EnergyScanID1onl.py b/script/EnergyScanID1onl.py new file mode 100644 index 0000000..c9cfb4c --- /dev/null +++ b/script/EnergyScanID1onl.py @@ -0,0 +1,222 @@ +import os +import traceback +############################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 + +def _startPlot(type): + global running + 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 (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" + + +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,OFFSET2) + 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 +############################################################################### +#caput ("X11PHS-E:OPT","PGM+ID1+ID2") + +number_of_scans = 1 +if RUNTYPE in ["+/-", "+", "-"]: + caput(OTF_MODE1,2) # 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) + + 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)) + + 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") diff --git a/script/EnergyScanID1only.py b/script/EnergyScanID1only.py new file mode 100644 index 0000000..c9cfb4c --- /dev/null +++ b/script/EnergyScanID1only.py @@ -0,0 +1,222 @@ +import os +import traceback +############################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 + +def _startPlot(type): + global running + 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 (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" + + +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,OFFSET2) + 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 +############################################################################### +#caput ("X11PHS-E:OPT","PGM+ID1+ID2") + +number_of_scans = 1 +if RUNTYPE in ["+/-", "+", "-"]: + caput(OTF_MODE1,2) # 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) + + 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)) + + 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")