is_panel = get_exec_pars().source != CommandSource.ui #Must be checked before callin "run" run("Devices/Elements") run("Devices/WireScanner") BPM_SENSORS = [("x","X1"), ("y","Y1"), ("q","Q1")] #(logic name sufix, channel sufix) #Paramter parsing prefix = args[0] if is_panel else "S30CB09-DWSC440" #"SINDI01-DWSC090" scan_type = args[1] if is_panel else WireScanner.WireX1 scan_range = args[2] if is_panel else [-200, 200, -200, 200] cycles = args[3] if is_panel else 5 velocity = args[4] if is_panel else 200 bpms = args[5] if is_panel else get_wire_scanners_bpms(prefix) blms = args[6] if is_panel else get_wire_scanners_blms(prefix) bkgrd = args[7] if is_panel else 20 plt = args[8] if is_panel else plot(None, title = "Wire Scan")[0] do_elog = True if is_panel else False print "WireScan parameters: ", prefix, scan_type, scan_range, cycles, cycles, bpms #Plot setup plt.clear() plt.getAxis(plt.AxisId.X).setLabel("Position"); plt.getAxis(plt.AxisId.Y).setLabel(""); plt.getAxis(plt.AxisId.Y2).setLabel(""); plt.setLegendVisible(True); snapshots = [] #Creating WireScanner object print "Creating scanner..." if prefix not in get_wire_scanners(): raise Exception("Invalid wire scan: " + prefix) scanner = WireScanner(prefix, scan_range, cycles, velocity, True) #List of stream channels channels = [("m_pos", scanner.motor_bs_readback.get_name()), ("cur_cycle", scanner.curr_cycl.get_name()), ("scanning", scanner.status_channels[0].get_name())] for i in range (len(blms)): channels.append (("blm" + str(i+1), blms[i] + ":B1_LOSS")) series = LinePlotSeries(blms[i], None, min(i+1, 2)) plt.addSeries(series) series.setLinesVisible(False) series.setPointSize(3) for i in range (len(bpms)): for sensor in BPM_SENSORS: channels.append (("bpm" + str(i+1) + "_" + sensor[0], bpms[i] + ":" + sensor[1])) #Metadata set_attribute("/", "Wire Scanner", prefix) set_attribute("/", "Scan Type", scan_type) set_attribute("/", "Range", scan_range) set_attribute("/", "Cycles", cycles) set_attribute("/", "Motor Velocity", velocity*math.sqrt(2)) set_attribute("/", "Wire Velocity", velocity) set_attribute("/", "Background Measures", bkgrd) #Stream creation print "Starting stream..." st = Stream("pulse_id", dispatcher) for c in channels: st.addScalar(c[0], c[1], 10, 0) st.initialize() st.start() st.waitCacheChange(10000) #Wait stream be running before starting scan #Pseudo-device returning the wire position class w_pos(Readable): def read(self): return scanner.get_sel_wire_pos(st.getChildren()[0].take()) #End of scan checking scan_complete = None cur_cycle = None def check_end_scan(record, scan): global scan_complete,cur_cycle if record[4]<1: print "Data aquisition completed" scan_complete=True scan.abort() record.cancel() #So it won't be saved else: position = record[0] if record[3] != cur_cycle: cur_cycle = record[3] get_context().dataManager.splitScanData(scan) #for s in plt.getAllSeries(): # s.clear() for i in range (len(blms)): plt.getSeries(i).appendData(position, record[5 + i]) #Process background def do_background(): #Store Background if bkgrd>0: #scanner.park() #scanner.waitValue("At start", 60000) set_exec_pars(group = "background") r = mscan (st, st.getReadables()[4:], bkgrd) for i in range(len(r.getReadables())): d = r.getReadable(i) m,s = mean(d), stdev(d) path = get_exec_pars().group + "/"+ r.getReadables()[i].name set_attribute(path, "Mean", m) set_attribute(path, "Sigma", s) #Scan def do_scan(index): global scan_complete, cur_cycle wire = "y" if (index==1) or (scan_type in [WireScanner.WireY1, WireScanner.WireY1]) else "x" set_exec_pars(group=wire+"_{count}", reset=True) scanner.set_selection(get_scan_selection(scan_type, index)) if wire == "x": plt.getAxis(plt.AxisId.X).setRange(scan_range[0], scan_range[1]) else: plt.getAxis(plt.AxisId.X).setRange(scan_range[2], scan_range[3]) scanner.init() scanner.waitValue("At start", 60000) scanner.curr_cycl.write(0) scan_complete=False cur_cycle = 1.0 for s in plt.getAllSeries(): s.clear() try: scanner.scan() #scanner.waitState(State.Busy, 60000) Not needed as stream filter will make the wait st.getChild("scanning").waitValue(1.0, 10000) mscan (st, [w_pos()] + st.getReadables(), -1, -1, take_initial = True, after_read = check_end_scan) except: if not scanner.isReady(): print "Aborting scan" scanner.abort() if not scan_complete: raise finally: #TODO: Display average of cycles #Combining data of multiple series #s=plt.getSeries(0) #indexes = sorted(range(len(s.x)),key=lambda x:s.x[x]) #x,y = [s.x[x] for x in indexes], [s.y[x] for x in indexes] #plot(y, xdata = x) img_file = os.path.abspath(get_exec_pars().path + "_" + get_exec_pars().group[0:1] + ".png") time.sleep(0.1) #Give some time to plot finish (async) plt.saveSnapshot(img_file, "png") snapshots.append(img_file) print "Starting scan..." try: do_background() st.setFilter(scanner.curr_cycl.get_name() + ">0") #scanner.status_channels[0].get_name() + ">0" not used because we must the transition to know when the finished do_scan(0) if scan_type in [WireScanner.Set1, WireScanner.Set2]: do_scan(1) finally: print "Closing scanner" scanner.close() print "Closing stream" st.close() # save the entry in the logbook if do_elog: if get_option("Generated data file:\n" + get_exec_pars().path +"\n\nSave to ELOG?", "YesNo") == "Yes": gsa_log_msg = "Data file: " + get_exec_pars().path gsa_log_msg = gsa_log_msg + "\nWire Scanner: " + prefix gsa_log_msg = gsa_log_msg + "\nScan Type: " + str(scan_type) gsa_log_msg = gsa_log_msg + "\nRange: " + str(scan_range) gsa_log_msg = gsa_log_msg + "\nCycles: " + str(cycles) gsa_log_msg = gsa_log_msg + "\nWire Velocity: " + str(velocity) elog("Wire Scan", gsa_log_msg, snapshots)