126 lines
3.8 KiB
Python
126 lines
3.8 KiB
Python
#If running from editor
|
|
if get_exec_pars().source == CommandSource.ui:
|
|
ROI = {"Region1": [10,5,20, 10], "Region2": [20,15,20, 10]}
|
|
SAVE_SPECTRUM = True
|
|
SAVE_IMAGES = True
|
|
WITH_I0 = True
|
|
SWITCH_POL = False
|
|
EXPOSURE = 1.0
|
|
AVERAGE = 2
|
|
NUMBER_SCANS =1
|
|
RANGES = [[500.0, 1000.0, 100.0]]
|
|
|
|
DRY_RUN = True
|
|
SAVE_DIAGS = False
|
|
|
|
log("ROIs: " + str(ROI))
|
|
|
|
if len(RANGES) == 0:
|
|
raise Exception("No scan range defined")
|
|
|
|
start = time.time()
|
|
|
|
#eiger.stop()
|
|
#eiger.grabMode=eiger.GrabMode.Single
|
|
time.sleep(0.3)
|
|
#set_exposure_time(EXPOSURE)
|
|
|
|
set_exec_pars(path="{data}/{year}_{month}/{date}/{seq}%03d_AbsortionSpectrum", format="csv", layout="table")
|
|
tag= "AbsortionSpectrum_{seq}%03d_{count}%02d"
|
|
|
|
rois = create_roi_devices(ROI)
|
|
|
|
if AVERAGE>1:
|
|
for i in range(len(rois)):
|
|
rois[i].monitored = True
|
|
rois[i] = create_averager(rois[i], AVERAGE, -1)
|
|
rois[i].monitored = True
|
|
|
|
|
|
frames=[]
|
|
|
|
|
|
class SyncEnergy(Writable):
|
|
def write(self,pos):
|
|
if not DRY_RUN:
|
|
put_energy(pos)
|
|
else:
|
|
print "Energy=" + str(pos)
|
|
sync_energy=SyncEnergy()
|
|
set_device_alias(sync_energy, "Energy") #Set display name
|
|
|
|
sensors = rois + [SR_current]
|
|
if WITH_I0:
|
|
sensors = rois + [keithley_1a]
|
|
|
|
|
|
|
|
av=None
|
|
def grab_image(position, scan):
|
|
global av
|
|
#frames = grab_frames(image, AVERAGE, roi=None, wait_next=True)
|
|
frames = []
|
|
for j in range( AVERAGE):
|
|
frames.append(Data(Arr.indexesDouble(600),30,20))
|
|
av = average_frames(frames)
|
|
if SAVE_IMAGES:
|
|
#Scan_Nr = (scan.recordIndex-1)/NUMBER_SCANS
|
|
#filename = get_exec_pars().path + "/" + ("%02d" % Scan_Nr) + "/s" + "{seq}%03d" + "_" + ("%03d.tif" % scan.recordIndex)
|
|
filename = get_exec_pars().path + "/" + str(scan.currentPass) + "/s" + "{seq}%03d" + "_" + ("%03d.tif" % scan.recordIndex)
|
|
filename = Setup.expandPath(filename)
|
|
print filename
|
|
save_as_tiff(av, filename, metadata=(get_diags() if SAVE_DIAGS else {}))
|
|
|
|
class Average(ReadableMatrix):
|
|
def read(self):
|
|
return av.matrix
|
|
def getWidth(self):
|
|
return 30 #eiger.dataMatrix.getWidth()
|
|
def getHeight(self):
|
|
return 20 #eiger.dataMatrix.getHeight()
|
|
averager=Average()
|
|
set_device_alias(averager, "Image") #Set display name
|
|
sensors.append(averager) #sensors.append(eiger.getDataMatrix())
|
|
|
|
#Initialize vartiables
|
|
if not DRY_RUN and str(get_setting("AUTO_SWITCH_VALVE")).lower() == "true":
|
|
open_vg10()
|
|
if not get_dry_run() and str(get_setting("AUTO_SWITCH_SHUTTER")).lower() == "true":
|
|
caput(FAST_SHTR, 'Open')
|
|
|
|
#put_id_offset(1, OFFSET_ID_1) #offset on ID1
|
|
#put_id_offset(2, OFFSET_ID_2) #offset on ID2
|
|
|
|
|
|
|
|
"""
|
|
if ID == "ID1":
|
|
current_pol=POL_IDS[POL_ID_1] #get_id_pol(1)
|
|
put_id_pol(1,current_pol)
|
|
caput('X11MA-ID2-GAP:SET',100) #open Gap ID2
|
|
elif ID == "ID2":
|
|
current_pol=POL_IDS[POL_ID_2]#get_id_pol(2)
|
|
put_id_pol(2,current_pol)
|
|
|
|
caput('X11MA-Ivg10D1-GAP:SET',100) #open Gap ID1
|
|
elif ID == "ID1_ID2":
|
|
polID1=POL_IDS[POL_ID_1]
|
|
polID2=POL_IDS[POL_ID_2]
|
|
put_id_pol(1,polID1)
|
|
put_id_pol(2,polID2)
|
|
wait_channel("X11PHS:alldone", 1)
|
|
"""
|
|
|
|
try:
|
|
#tscan(rois, 10, 0.1, format="csv", before_read=grab_image, passes = NUMBER_SCANS, tag=tag)
|
|
rscan(sync_energy, sensors, RANGES, latency = 0.0 , passes = NUMBER_SCANS, tag=tag, before_read=grab_image, save=SAVE_SPECTRUM, split=True, keep=False)
|
|
finally:
|
|
if AVERAGE>1:
|
|
for i in range(len(rois)):
|
|
rois[i].parent.monitored = False # Remove listeners on the image
|
|
if not DRY_RUN and str(get_setting("AUTO_SWITCH_VALVE")).lower() == "true":
|
|
close_vg10()
|
|
if not get_dry_run() and str(get_setting("AUTO_SWITCH_SHUTTER")).lower() == "true":
|
|
caput(FAST_SHTR, 'Close')
|
|
#restore_eiger()
|
|
print "Running time: " + str(time.time() - start) |