246 lines
10 KiB
Python
246 lines
10 KiB
Python
import traceback
|
|
|
|
is_panel = get_exec_pars().source != CommandSource.ui #Must be checked before callin "run"
|
|
|
|
run("Devices/Elements")
|
|
run("Devices/WireScanner")
|
|
run("Diagnostics/sig_process_wrapper")
|
|
|
|
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 3
|
|
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 10
|
|
plt = args[8] if is_panel else plot(None, title = "Wire Scan")[0]
|
|
save_raw = args[9] if is_panel else False
|
|
do_elog = True if is_panel else True
|
|
print "WireScan parameters: ", prefix, scan_type, scan_range, cycles, velocity, bpms, blms, bkgrd
|
|
|
|
|
|
#Plot setup
|
|
plt.clear()
|
|
plt.removeMarker(None)
|
|
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(2)
|
|
if save_raw:
|
|
channels.append (("blm" + str(i+1) + "_raw" , blms[i] + ":LOSS_SIGNAL_RAW"))
|
|
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)
|
|
set_attribute("/", "BPMs", bpms)
|
|
set_attribute("/", "BLMs", blms)
|
|
|
|
filename = get_exec_pars().path
|
|
|
|
|
|
|
|
#Stream creation
|
|
print "Starting stream..."
|
|
st = Stream("pulse_id", dispatcher)
|
|
for c in channels:
|
|
st.addScalar(c[0], c[1], int(100.0 / get_repetition_rate()), 0)
|
|
st.initialize()
|
|
st.start()
|
|
st.waitCacheChange(10000) #Wait stream be running before starting scan
|
|
|
|
|
|
class Timestamp(Readable):
|
|
def read(self):
|
|
return st.getTimestamp()
|
|
|
|
|
|
|
|
|
|
#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, cur_cycle, wire = None, None, 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(wait=True)
|
|
set_exec_pars(group = "background")
|
|
r = mscan (st, st.getReadables()[4:], bkgrd)
|
|
for i in range(len(r.getReadables())):
|
|
d = r.getReadable(i)
|
|
path = get_exec_pars().group + "/"+ r.getReadables()[i].name
|
|
set_attribute(path, "Mean", mean(d))
|
|
set_attribute(path, "Sigma", stdev(d) )
|
|
|
|
#Scan
|
|
def do_scan(index):
|
|
global scan_complete, cur_cycle, wire
|
|
wire = "y" if (index==1) or (scan_type in [WireScanner.WireY1, WireScanner.WireY2]) 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(wait=True)
|
|
scanner.curr_cycl.write(0)
|
|
scan_complete=False
|
|
cur_cycle = 1.0
|
|
for s in plt.getAllSeries():
|
|
s.clear()
|
|
plt.removeMarker(None)
|
|
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() + [Timestamp(),], -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:
|
|
#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)
|
|
calculate()
|
|
img_file = os.path.abspath(filename + "_" + 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)
|
|
|
|
msg = ""
|
|
def calculate():
|
|
global msg
|
|
stats = []
|
|
for i in range(len(blms)):
|
|
msg += "Wire " + wire + " - BLM " + str(i+1) + ":\n"
|
|
try:
|
|
bg = get_attributes("background/blm" + str(i+1))["Mean"] if bkgrd>0 else 0.0
|
|
samples = [[], [], [], []]
|
|
for cycle in range (cycles):
|
|
pos = load_data(wire+"_" + ("%04d" % (cycle+1)) + "/w_pos")
|
|
path = wire+"_" + ("%04d" % (cycle+1)) + "/blm" + str(i+1)
|
|
data = load_data(path)
|
|
sp = blm_remove_spikes(data)
|
|
sig = sp if bg is None else [v-bg for v in sp]
|
|
|
|
[rms_com, rms_sigma] = profile_rms_stats(pos, sig,noise_std=0, n_sigma=3.5)
|
|
set_attribute(path, "RMS COM", float("nan") if (rms_com is None) else rms_com)
|
|
set_attribute(path, "RMS Sigma", float("nan") if (rms_sigma is None) else rms_sigma)
|
|
|
|
#print [com, rms]
|
|
[off, amp, com, sigma] = profile_gauss_stats(pos, sig, off=None, amp=None, com=None, sigma=None)
|
|
set_attribute(path, "Gauss COM", float("nan") if (com is None) else com)
|
|
set_attribute(path, "Gauss Sigma", float("nan") if (sigma is None) else sigma)
|
|
|
|
samples[0].append(rms_com);samples[1].append(rms_sigma);samples[2].append(com);samples[3].append(sigma)
|
|
#print [off, amp, com, sigma]
|
|
|
|
#from mathutils import Gaussian
|
|
#g = Gaussian(amp, com, sigma)
|
|
#gauss = [g.value(v)+off for v in pos]
|
|
#plot([data, sp, sig, gauss], ["data", "sp", "signal", "gauss", ], xdata = pos, title="Fit blm" + str(i+1) + " - " + str(cycle+1))
|
|
|
|
stats.append([])
|
|
for sample in samples:
|
|
sample = [v for v in sample if v is not None]
|
|
stats[i].append( (mean(sample), stdev(sample)) if len(sample)>0 else (float("nan"), float("nan")) )
|
|
plt.addMarker(stats[i][2][0], None, "Gcom=" + "%.2f" % stats[i][2][0], plt.getSeries(i).color)
|
|
plt.addMarker(stats[i][0][0], None, "Rcom=" + "%.2f" % stats[i][0][0], plt.getSeries(i).color.brighter())
|
|
|
|
msg += " RMS COM: " + "%.4f" % stats[i][0][0] + " +- " +"%.4f" % stats[i][0][1] + "\n" #unichr(0x03C3) + "="
|
|
msg += " RMS Sigma: " + "%.4f" % stats[i][1][0] + " +- " + "%.4f" % stats[i][1][1] + "\n"
|
|
msg += " Gauss COM: " + "%.4f" % stats[i][2][0] + " +- " + "%.4f" % stats[i][2][1] + "\n"
|
|
msg += " Gauss Sigma: " + "%.4f" % stats[i][3][0] + " +- " + "%.4f" % stats[i][3][1] + "\n"
|
|
except Exception, e:
|
|
print >> sys.stderr, traceback.format_exc()
|
|
msg += str(e)+ "\n"
|
|
|
|
|
|
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()
|
|
|
|
print msg
|
|
|
|
# save the entry in the logbook
|
|
if do_elog:
|
|
if get_option("Generated data file:\n" + filename +"\n\n" + msg + "\n\n" + "Save to ELOG?", "YesNo") == "Yes":
|
|
log_msg = "Data file: " + filename
|
|
log_msg = log_msg + "\nWire Scanner: " + prefix
|
|
log_msg = log_msg + "\nScan Type: " + str(scan_type)
|
|
log_msg = log_msg + "\nRange: " + str(scan_range)
|
|
log_msg = log_msg + "\nCycles: " + str(cycles)
|
|
log_msg = log_msg + "\nWire Velocity: " + str(velocity)
|
|
log_msg = log_msg + "\nBackground Measures: " + str(bkgrd)
|
|
log_msg = log_msg + "\nBPMs: " + str(bpms)
|
|
log_msg = log_msg + "\nBLMs: " + str(blms)
|
|
|
|
log_msg = log_msg + "\n" + msg
|
|
elog("Wire Scan", log_msg, snapshots)
|
|
|