186 lines
5.1 KiB
Python
186 lines
5.1 KiB
Python
import os
|
|
import traceback
|
|
import thread
|
|
|
|
###############################################################################
|
|
# 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)
|
|
|
|
###############################################################################
|
|
# Parameters
|
|
###############################################################################
|
|
|
|
|
|
E1 = 634
|
|
E2 = 660
|
|
TIME = 4.0 #min
|
|
DELAY = 1.0 #s
|
|
#PREFIX = 'Data'
|
|
RUNTYPE = "+/-"
|
|
ROUNDS = 1
|
|
OFFSETp = -1.9
|
|
OFFSETm = -1.9
|
|
PLOT_TYPE = 1
|
|
|
|
caput ("X11PHS-E:OPT","PGM+ID2")
|
|
|
|
print "\nStarting energy scan - Parameters: ",
|
|
print E1,E2,TIME,DELAY,RUNTYPE
|
|
|
|
file_prefix = time.strftime("%y%m%d")
|
|
#input_path = "/sls/X11MA/data/X11MA/fechner_beamtime/"+file_prefix+"/"
|
|
input_path = "/sls/X11MA/data/X11MA/pshell/home/data/2017_12/juraj/"
|
|
output_path = input_path #+file_prefix+"/" #"/sls/X/pshell/home/data/2017_12/juraj11MA/Data1/public/e10989/"+file_prefix+"/"
|
|
pol_str = None
|
|
mag_str = None
|
|
fid = get_next_fid(input_path, "o" + file_prefix)
|
|
number_of_scans = 1
|
|
|
|
|
|
wait_channel(OTF_DONE, 1, type = 'i')
|
|
|
|
|
|
###############################################################################
|
|
#Prepare scan for PGM+ID2
|
|
###############################################################################
|
|
|
|
caput ("X11PHS-E:OPT","PGM+ID2")
|
|
#caput(OTF_OFF1,-40) #detune ID2
|
|
#wait_channel(OTF_DONE, 1, type = 'i')/sls/X11MA/datmokuser/
|
|
|
|
if RUNTYPE in ["+/-", "+", "-"]:
|
|
if RUNTYPE == "+":
|
|
caput(OTF_MODE2,1) # circ+ in ID2
|
|
caput(OTF_OFF2,OFFSETp)
|
|
number_of_scans = ROUNDS
|
|
elif RUNTYPE == "-":
|
|
caput(OTF_MODE2,2) # circ- in ID2
|
|
caput(OTF_OFF2,OFFSETm)
|
|
number_of_scans = ROUNDS
|
|
elif RUNTYPE == "+/-":
|
|
caput(OTF_MODE2,1) # circ+ in ID2
|
|
caput(OTF_OFF2,OFFSETp)
|
|
number_of_scans = 2 * ROUNDS
|
|
wait_channel(OTF_DONE, 1, type = 'i')
|
|
|
|
elif RUNTYPE in ["LH/LV", "LH", "LV"]:
|
|
print "Wrong polarization selected!"
|
|
|
|
|
|
else:
|
|
raise Exception("Invalid run type: " + RUNTYPE)
|
|
|
|
#open_vg10()
|
|
#time.sleep(0.5)
|
|
#open_vg11()
|
|
#time.sleep(0.5)
|
|
#open_vg12()
|
|
#time.sleep(0.5)
|
|
open_vg13()
|
|
|
|
###############################################################################
|
|
#Main scan loop
|
|
###############################################################################
|
|
|
|
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)
|
|
otf_start.write(1) #Start the OTF scan
|
|
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"
|
|
|
|
|
|
|
|
if RUNTYPE in ["+/-"]:
|
|
pol_str = caget(OTF_MODE2)
|
|
if pol_str == "CIRC +":
|
|
caput(OTF_MODE2,2) # circ- in ID2
|
|
caput(OTF_OFF2,OFFSETm)
|
|
elif pol_str == "CIRC -":
|
|
caput(OTF_MODE2,1) # circ+ in ID2
|
|
caput(OTF_OFF2,OFFSETp)
|
|
else: print "Check pol. change"
|
|
wait_channel(OTF_DONE, 1, type = 'i')
|
|
else:
|
|
print "running in one polarization mode, no switching"
|
|
pol_str = caget(OTF_MODE2)
|
|
|
|
print("Converting data file: " + output_file);
|
|
mag_str = caget(MAG)
|
|
#convert_file(input_file, output_file, mag_str, pol_str)
|
|
convert_file(input_file, output_file, mag_str, pol_str) # tey and tfy saved - not normalized
|
|
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)
|
|
|
|
time.sleep(3.0)
|
|
fid = fid + 1
|
|
|
|
caput(OTF_ESET, E1)
|
|
#close_vg13()
|
|
|
|
print "Finished Energy scan"
|
|
|
|
print("Success")
|
|
|