Add AthosScreens
This commit is contained in:
Executable
+45
@@ -0,0 +1,45 @@
|
||||
import numpy as np
|
||||
from scipy.optimize import curve_fit
|
||||
import sys
|
||||
|
||||
|
||||
def gaus(x, a, x0, sigma, offset):
|
||||
return offset + a * np.exp(-(x - x0) ** 2 / (2 * sigma ** 2))
|
||||
|
||||
#Return [amp, mean_val, sigma, offset]
|
||||
def fit_energy(e_from, e_to, steps, num_shots, data):
|
||||
energy_range = np.linspace(e_from, e_to, steps)
|
||||
energy_range_fit = np.linspace(energy_range[0], energy_range[-1], len(energy_range)*10)
|
||||
centre_line_out = data[:,:,int(data.shape[2]/2)].mean(axis=1)
|
||||
try:
|
||||
popt,pcov = curve_fit(gaus,energy_range,centre_line_out,p0=[1,energy_range[np.argmax(centre_line_out)],energy_range.mean()*1e-3,1e3*num_shots])
|
||||
except:
|
||||
raise Exception('Fit failed: spectrum might not be near scan range center \n' + str(sys.exc_info()[1]))
|
||||
#print('Fit failed: spectrum might not be near scan range center')
|
||||
#return None
|
||||
max_ind = np.argmax(centre_line_out)
|
||||
max_photon_energy=energy_range[max_ind]
|
||||
print(max_photon_energy)
|
||||
return popt, centre_line_out
|
||||
|
||||
|
||||
#Return [amp, mean_val, sigma, offset]
|
||||
def fit_crystal_height(xstal_from, xstal_to, steps, data):
|
||||
xstal_range = np.linspace(xstal_from, xstal_to, steps)
|
||||
projection = data.mean(axis=1).mean(axis=1)
|
||||
offset = np.mean(projection[0:100])
|
||||
signal_centre = xstal_range[np.argmax(projection)]
|
||||
xstal_range_fit = np.linspace(xstal_range[0], xstal_range[-1], len(xstal_range)*10)
|
||||
try:
|
||||
popt,pcov = curve_fit(gaus,xstal_range,projection,p0=[100,signal_centre,-0.2,offset])
|
||||
except:
|
||||
raise Exception('Fit failed: spectrum might not be near scan range center \n' + str(sys.exc_info()[1]))
|
||||
#print('Fit failed: spectrum might not be near scan range center')
|
||||
#return None
|
||||
return popt, projection
|
||||
|
||||
|
||||
def get_signal_centre(data, data_range):
|
||||
projection = data.mean(axis=1).mean(axis=1)
|
||||
signal_centre = data_range[np.argmax(projection)]
|
||||
return signal_centre, projection
|
||||
+32
@@ -0,0 +1,32 @@
|
||||
import numpy as np
|
||||
import matplotlib.pyplot as plt
|
||||
|
||||
|
||||
def gaus(x, a, x0, sigma, offset):
|
||||
return offset + a * np.exp(-(x - x0) ** 2 / (2 * sigma ** 2))
|
||||
|
||||
def plot_energy(E_from, E_to, steps, Scan_spec, popt, measured_offset):
|
||||
Energy_range = np.linspace(E_from, E_to, steps)
|
||||
centre_line_out = Scan_spec[:,:,int(Scan_spec.shape[2]/2)].mean(axis=1)
|
||||
Energy_range_fit = np.linspace(Energy_range[0], Energy_range[-1], len(Energy_range)*10)
|
||||
|
||||
|
||||
plt.figure(figsize=[10,5])
|
||||
plt.subplot(121)
|
||||
plt.title('PSSS scan of set photon energy')
|
||||
plt.pcolormesh(np.arange(0,Scan_spec.shape[2]), Energy_range, Scan_spec.mean(axis=1),cmap='CMRmap')
|
||||
plt.vlines(int(Scan_spec.shape[2]/2), Energy_range[0], Energy_range[-1],linestyles='--', colors='orange')
|
||||
plt.xlim([0,Scan_spec.shape[2]])
|
||||
plt.xlabel('Camera pixel')
|
||||
plt.ylabel('Set PSSS energy [eV] \n SARFE10-PSSS059:ENERGY')
|
||||
|
||||
plt.subplot(122)
|
||||
plt.title('At camera centre pixel %1i \nCalibrated energy = %.1f [eV]\n Offset from machine = %.1f [eV]'%(int(Scan_spec.shape[2]/2),popt[1],measured_offset))
|
||||
plt.plot(centre_line_out,Energy_range,linewidth = 2, color = 'orange',label ='measured')
|
||||
try:
|
||||
plt.plot(gaus(Energy_range_fit,*popt),Energy_range_fit,'r:',label='fit')
|
||||
except:
|
||||
pass
|
||||
plt.xticks([])
|
||||
plt.legend()
|
||||
plt.grid(True)
|
||||
+36
@@ -0,0 +1,36 @@
|
||||
from jeputils import *
|
||||
|
||||
RELOAD_CPYTHON = not App.isDetached()
|
||||
|
||||
def fit_energy(e_from, e_to, steps, num_shots, data):
|
||||
data = to_array(data, 'd')
|
||||
dims = [len(data), len(data[0]), len(data[0][0])]
|
||||
data = Convert.flatten(data)
|
||||
arr = to_npa(data, dims)
|
||||
popt, centre_line_out = call_jep("cpython/psss", "fit_energy", [e_from, e_to, steps, num_shots, arr], reload=RELOAD_CPYTHON)
|
||||
return popt.getData(), centre_line_out.getData()
|
||||
|
||||
def fit_crystal_height(xstal_from, xstal_to, steps, data):
|
||||
data = to_array(data, 'd')
|
||||
dims = [len(data), len(data[0]), len(data[0][0])]
|
||||
data = Convert.flatten(data)
|
||||
arr = to_npa(data, dims)
|
||||
popt, projection = call_jep("cpython/psss", "fit_crystal_height", [xstal_from, xstal_to, steps, arr], reload=RELOAD_CPYTHON)
|
||||
return popt.getData(), projection.getData()
|
||||
|
||||
def get_signal_centre(data, data_range):
|
||||
data = to_array(data, 'd')
|
||||
dims = [len(data), len(data[0]), len(data[0][0])]
|
||||
data = Convert.flatten(data)
|
||||
arr = to_npa(data, dims)
|
||||
data_range = to_npa(to_array(data_range, 'd'))
|
||||
signal_centre, projection = call_jep("cpython/psss", "get_signal_centre", [arr, data_range], reload=RELOAD_CPYTHON)
|
||||
return signal_centre, projection.getData()
|
||||
|
||||
def plot_energy(e_from, e_to, steps, data, popt, measured_offset):
|
||||
data = to_array(data, 'd')
|
||||
dims = [len(data), len(data[0]), len(data[0][0])]
|
||||
data = Convert.flatten(data)
|
||||
arr = to_npa(data, dims)
|
||||
ret = call_jep("cpython/psss_plot", "plot_energy", [e_from, e_to, steps, arr, popt, measured_offset], reload=RELOAD_CPYTHON)
|
||||
return ret
|
||||
Executable
+184
@@ -0,0 +1,184 @@
|
||||
###################################################################################################
|
||||
# Deployment specific global definitions - executed after startup.py
|
||||
###################################################################################################
|
||||
|
||||
from mathutils import estimate_peak_indexes, fit_gaussians, create_fit_point_list
|
||||
from mathutils import fit_polynomial,fit_gaussian, fit_harmonic, calculate_peaks, fit_gaussian_offset
|
||||
from mathutils import PolynomialFunction, Gaussian, HarmonicOscillator, GaussianOffset
|
||||
from plotutils import plot_function, plot_data
|
||||
|
||||
import java.awt.Color as Color
|
||||
run("psss/psss")
|
||||
|
||||
###################################################################################################
|
||||
# DRY RUN
|
||||
###################################################################################################
|
||||
|
||||
def set_dry_run(value):
|
||||
global dry_run
|
||||
dry_run = value
|
||||
|
||||
def is_dry_run():
|
||||
if "dry_run" in globals():
|
||||
return True if dry_run else False
|
||||
return False
|
||||
|
||||
|
||||
###################################################################################################
|
||||
# Machine utilities
|
||||
###################################################################################################
|
||||
|
||||
def is_laser_on():
|
||||
return (caget ("SIN-TIMAST-TMA:Beam-Las-Delay-Sel",'d') == 0 )
|
||||
|
||||
def is_timing_ok():
|
||||
return caget("SIN-TIMAST-TMA:SOS-COUNT-CHECK") == 0
|
||||
|
||||
def get_repetition_rate():
|
||||
return caget("SIN-TIMAST-TMA:Evt-15-Freq-I")
|
||||
|
||||
|
||||
###################################################################################################
|
||||
# Shortcut to maths utilities
|
||||
###################################################################################################
|
||||
|
||||
def gfit(ydata, xdata = None):
|
||||
"""
|
||||
Gaussian fit
|
||||
"""
|
||||
if xdata is None:
|
||||
xdata = frange(0, len(ydata), 1)
|
||||
#ydata = to_list(ydata)
|
||||
#xdata = to_list(xdata)
|
||||
max_y= max(ydata)
|
||||
index_max = ydata.index(max_y)
|
||||
max_x= xdata[index_max]
|
||||
print "Max index:" + str(index_max),
|
||||
print " x:" + str(max_x),
|
||||
print " y:" + str(max_y)
|
||||
gaussians = fit_gaussians(ydata, xdata, [index_max,])
|
||||
(norm, mean, sigma) = gaussians[0]
|
||||
p = plot([ydata],["data"],[xdata], title="Fit" )[0]
|
||||
fitted_gaussian_function = Gaussian(norm, mean, sigma)
|
||||
scale_x = [float(min(xdata)), float(max(xdata)) ]
|
||||
points = max((len(xdata)+1), 100)
|
||||
resolution = (scale_x[1]-scale_x[0]) / points
|
||||
fit_y = []
|
||||
fit_x = frange(scale_x[0],scale_x[1],resolution, True)
|
||||
for x in fit_x:
|
||||
fit_y.append(fitted_gaussian_function.value(x))
|
||||
p.addSeries(LinePlotSeries("fit"))
|
||||
p.getSeries(1).setData(fit_x, fit_y)
|
||||
|
||||
if abs(mean - xdata[index_max]) < ((scale_x[0] + scale_x[1])/2):
|
||||
print "Mean -> " + str(mean)
|
||||
p.addMarker(mean, None, "Mean="+str(round(norm,2)), Color.MAGENTA.darker())
|
||||
return (norm, mean, sigma)
|
||||
else:
|
||||
p.addMarker(max_x, None, "Max="+str(round(max_x,2)), Color.GRAY)
|
||||
print "Invalid gaussian fit: " + str(mean)
|
||||
return (None, None, None)
|
||||
|
||||
|
||||
def hfit(ydata, xdata = None):
|
||||
"""
|
||||
Harmonic fit
|
||||
"""
|
||||
if xdata is None:
|
||||
xdata = frange(0, len(ydata), 1)
|
||||
|
||||
max_y= max(ydata)
|
||||
index_max = ydata.index(max_y)
|
||||
max_x= xdata[index_max]
|
||||
|
||||
start,end = min(xdata), max(xdata)
|
||||
(amplitude, angular_frequency, phase) = fit_harmonic(ydata, xdata)
|
||||
fitted_harmonic_function = HarmonicOscillator(amplitude, angular_frequency, phase)
|
||||
|
||||
print "amplitude = ", amplitude
|
||||
print "angular frequency = ", angular_frequency
|
||||
print "phase = ", phase
|
||||
|
||||
f = angular_frequency/ (2* math.pi)
|
||||
print "frequency = ", f
|
||||
|
||||
resolution = 4.00 # 1.00
|
||||
fit_y = []
|
||||
for x in frange(start,end,resolution, True):
|
||||
fit_y.append(fitted_harmonic_function.value(x))
|
||||
fit_x = frange(start, end+resolution, resolution)
|
||||
|
||||
p = plot(ydata,"data", xdata, title="HFit")[0]
|
||||
p.addSeries(LinePlotSeries("fit"))
|
||||
p.getSeries(1).setData(fit_x, fit_y)
|
||||
|
||||
#m = (phase + math.pi)/ angular_frequency
|
||||
m = -phase / angular_frequency
|
||||
if (m<start):
|
||||
m+=(1.0/f)
|
||||
|
||||
if start <=m <=end:
|
||||
print "fit = ", m
|
||||
p.addMarker(m, None, "Fit="+str(round(m ,2)), Color.MAGENTA.darker())
|
||||
return (amplitude, angular_frequency, phase, True, m, fit_x, fit_y)
|
||||
else:
|
||||
print "max = ",max_x
|
||||
p.addMarker(max_x, None, "Max="+str(round(max_x ,2)), Color.MAGENTA.darker())
|
||||
return (amplitude, angular_frequency, phase, False, max_x, fit_x, fit_y)
|
||||
|
||||
|
||||
|
||||
def plot_gauss_fit(xdata, ydata, gauss_pars=None, p=None, title = "Data"):
|
||||
if gauss_pars is None:
|
||||
gauss_pars= fit_gaussian_offset(ydata, xdata, None)
|
||||
(offset, amp, mean_value, sigma) = gauss_pars
|
||||
print "Gauss plot: ", (offset, amp, mean_value, sigma)
|
||||
fitted_gaussian_function = GaussianOffset(offset, amp, mean_value, abs(sigma))
|
||||
|
||||
if p is None:
|
||||
p = plot(None, title=title)[0]
|
||||
p.clear()
|
||||
plot_data(p, ydata, title, xdata=xdata, show_points = True, color=Color.BLUE)
|
||||
fit_range = frange(xdata[0],xdata[-1],float(xdata[1]-xdata[0])/100, True)
|
||||
plot_function(p, fitted_gaussian_function, "Gauss", fit_range, show_points=False, color=Color.RED)
|
||||
p.setLegendVisible(True)
|
||||
p.addMarker(mean_value, None, "Mean=" + str(round(mean_value,2)), Color.LIGHT_GRAY)
|
||||
return p,(amp, mean_value, sigma)
|
||||
|
||||
###################################################################################################
|
||||
# Tools
|
||||
###################################################################################################
|
||||
|
||||
def elog(title, message, attachments = [], author = None, category = "Info", domain = "", logbook = "Bernina", encoding=1):
|
||||
"""
|
||||
Add entry to ELOG.
|
||||
"""
|
||||
if author is None:
|
||||
author = "pshell" #get_context().user.name
|
||||
typ = "pshell"
|
||||
entry = ""
|
||||
|
||||
cmd = 'G_CS_ELOG_add -l "' + logbook + '" '
|
||||
cmd = cmd + '-a "Author=' + author + '" '
|
||||
cmd = cmd + '-a "Type=' + typ + '" '
|
||||
cmd = cmd + '-a "Entry=' + entry + '" '
|
||||
cmd = cmd + '-a "Title=' + title + '" '
|
||||
cmd = cmd + '-a "Category=' + category + '" '
|
||||
cmd = cmd + '-a "Domain=' + domain + '" '
|
||||
for attachment in attachments:
|
||||
cmd = cmd + '-f "' + attachment + '" '
|
||||
cmd = cmd + '-n ' + str(encoding)
|
||||
cmd = cmd + ' "' + message + '"'
|
||||
#print cmd
|
||||
#os.system (cmd)
|
||||
#print os.popen(cmd).read()
|
||||
import subprocess
|
||||
proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
|
||||
(out, err) = proc.communicate()
|
||||
if (err is not None) and err!="":
|
||||
raise Exception(err)
|
||||
print out
|
||||
try:
|
||||
return int(out[out.find("ID=") +3 : ])
|
||||
except:
|
||||
print out
|
||||
+73
@@ -0,0 +1,73 @@
|
||||
#Scan the PSSS camera position
|
||||
#Purpose:
|
||||
#To set or confirm the camera is positioned with the measured spectrum in the centre of the spectral integration window
|
||||
|
||||
#If running from editor
|
||||
if get_exec_pars().source == CommandSource.ui:
|
||||
#User inputs - define travel range of camera
|
||||
RANGE_FROM = -17
|
||||
RANGE_TO = -11
|
||||
STEPS = 20
|
||||
NUM_SHOTS= 10 #100
|
||||
PLOT=None
|
||||
p = plot(None, title="Data")[0] if (PLOT is None) else PLOT
|
||||
p.clear()
|
||||
p.removeMarker(None)
|
||||
p.setLegendVisible(True)
|
||||
p.addSeries(LinePlotSeries("PSSS Spectrum Average"))
|
||||
|
||||
|
||||
run("cpython/wrapper")
|
||||
|
||||
if not is_dry_run():
|
||||
cam_x=Channel("SARFE10-PSSS059:MOTOR_X5.VAL", name="cam_x")
|
||||
else:
|
||||
cam_x=DummyRegister("cam_x")
|
||||
|
||||
av = create_averager(psss_spectrum_y, NUM_SHOTS, interval=-1, name="spectrum_average")
|
||||
av_samples = av.samples
|
||||
av_samples.alias = "spectrum_samples"
|
||||
|
||||
|
||||
#Scan and take data
|
||||
def after_read(record, scan):
|
||||
p.getSeries(0).setData(psss_spectrum_x.take(), record[av])
|
||||
p.setTitle("Cam X = %1.3f" %(record[cam_x]))
|
||||
|
||||
r = lscan(cam_x, (av, av_samples), RANGE_FROM, RANGE_TO, STEPS, latency=0.0, after_read = after_read, save=False)
|
||||
average, samples, cam_range = r.getReadable(0), r.getReadable(1), r.getPositions(0)
|
||||
|
||||
signal_centre, projection = get_signal_centre(samples, cam_range)
|
||||
|
||||
#Set max position
|
||||
cam_x.write(signal_centre)
|
||||
cam_x.close()
|
||||
|
||||
|
||||
|
||||
"""
|
||||
plt.figure(figsize=[10,5])
|
||||
plt.subplot(121)
|
||||
plt.title('PSSS scan of camera position')
|
||||
plt.pcolormesh(np.arange(0,Scan_spec.shape[2]), Cam_range, Scan_spec.mean(axis=1),cmap='CMRmap')
|
||||
plt.xlim([0,Scan_spec.shape[2]])
|
||||
plt.xlabel('Camera pixel dispersive direction')
|
||||
plt.ylabel('Set PSSS cam_x _pos [mm] \n'+PSSS_cam_x_PV_name[0:-4])
|
||||
plt.subplot(122)
|
||||
plt.plot(projection,Cam_range,linewidth = 2, color = 'orange',label ='projected signal')
|
||||
plt.title('Spectrum centred at %.1f [mm] (from signal max) \n trace should have hard edges'%signal_centre)
|
||||
plt.xticks([])
|
||||
plt.legend()
|
||||
plt.grid(True)
|
||||
"""
|
||||
#PLOT.clear()
|
||||
#plot_data(PLOT, projection, "Data", xdata=cam_range, show_points = True, color=Color.BLUE)
|
||||
#p,pars = plot_gauss_fit(cam_range, projection, gauss_pars=None, p=PLOT, title = "Data")
|
||||
|
||||
p.clear()
|
||||
p.setTitle("")
|
||||
plot_data(p, projection, "Projection", xdata=cam_range, show_points = True, color=Color.BLUE)
|
||||
p.addMarker(signal_centre, None, "Signal Centre=" + str(round(signal_centre,2)), Color.LIGHT_GRAY)
|
||||
|
||||
|
||||
set_return(signal_centre)
|
||||
+73
@@ -0,0 +1,73 @@
|
||||
###############################################################################
|
||||
#Scan the PSSS crystal height
|
||||
#Purpose:
|
||||
#The PSSS signal level is very sensitive to the crystal height. This script will scan the height and set the position to the maximum signal
|
||||
|
||||
if get_exec_pars().source == CommandSource.ui:
|
||||
#User inputs - define travel range of camera
|
||||
RANGE_FROM = -0.8
|
||||
RANGE_TO = -1.7
|
||||
STEPS = 10 #20
|
||||
NUM_SHOTS= 10 # 100
|
||||
PLOT=None
|
||||
# get current camera ROIs and then set to max for scan
|
||||
roi_min = psss_roi_min.read()
|
||||
roi_max = psss_roi_max.read()
|
||||
psss_roi_min.write(1)
|
||||
psss_roi_max.write(2000)
|
||||
|
||||
p = plot(None, title="Data")[0] if (PLOT is None) else PLOT
|
||||
p.clear()
|
||||
p.removeMarker(None)
|
||||
p.setLegendVisible(True)
|
||||
p.addSeries(LinePlotSeries("PSSS Spectrum Average"))
|
||||
|
||||
run("cpython/wrapper")
|
||||
|
||||
|
||||
#Setup and functions setup¶
|
||||
#if not is_dry_run(): # C.arrell commented out 20.01.21
|
||||
xstal_height=Channel("SARFE10-PSSS059:MOTOR_Y3.VAL", name="xstal_height")
|
||||
#else:
|
||||
# xstal_height=DummyRegister("xstal_height")
|
||||
|
||||
av = create_averager(psss_spectrum_y, NUM_SHOTS, interval=-1, name="spectrum_average")
|
||||
av_samples = av.samples
|
||||
av_samples.alias = "spectrum_samples"
|
||||
|
||||
#Scan and take data
|
||||
def after_read(record, scan):
|
||||
p.getSeries(0).setData(psss_spectrum_x.take(), record[av])
|
||||
p.setTitle("Xtal Height = %1.3f" %(record[xstal_height]))
|
||||
|
||||
r = lscan(xstal_height, (av, av_samples), RANGE_FROM, RANGE_TO, STEPS, latency=2.0, after_read = after_read, save=False)
|
||||
|
||||
#User inputs - define travel range of crystal
|
||||
#It is unlikely these values need to be changed
|
||||
average, samples, xstal_range = r.getReadable(0), r.getReadable(1), r.getPositions(0)
|
||||
|
||||
#return maxium position
|
||||
[amp, mean_val, sigma, offset], projection = fit_crystal_height(RANGE_FROM, RANGE_TO, STEPS+1, samples)
|
||||
print(mean_val)
|
||||
|
||||
if not (RANGE_FROM < mean_val < RANGE_TO or RANGE_TO < mean_val < RANGE_FROM):
|
||||
raise Exception ("Invalid fit mean: " + str(mean_val))
|
||||
|
||||
|
||||
#Set max position
|
||||
#Cell below will push the maximum position to the xstal height
|
||||
xstal_height.write(mean_val)
|
||||
xstal_height.close()
|
||||
|
||||
# return ROI to inital value
|
||||
psss_roi_min.write(roi_min)
|
||||
psss_roi_max.write(roi_max)
|
||||
|
||||
#Plots
|
||||
|
||||
p.clear()
|
||||
p.setTitle("")
|
||||
plot_gauss_fit(xstal_range, projection, gauss_pars=(offset, amp, mean_val, sigma), p=p, title = "Data")
|
||||
|
||||
|
||||
set_return(mean_val)
|
||||
+89
@@ -0,0 +1,89 @@
|
||||
###############################################################################
|
||||
#Scan the PSSS photon energy
|
||||
#Purpose: To find and centre the PSSS photon energy so the measured spectrum is centred on the camera chip
|
||||
|
||||
#PARAMETERS
|
||||
#User inputs - define energy range to scan below by running the appropiate cell
|
||||
#Below is for a large scan range assuming offset from machine upto $\pm$ 300 eV
|
||||
|
||||
#If running from editor
|
||||
if get_exec_pars().source == CommandSource.ui:
|
||||
RANGE_OFF = None
|
||||
RANGE_FROM = 11100
|
||||
RANGE_TO = 11300
|
||||
STEPS = 5 #60
|
||||
NUM_SHOTS= 10 #100
|
||||
PLOT=None
|
||||
p = plot(None, title="Data")[0] if (PLOT is None) else PLOT
|
||||
p.clear()
|
||||
p.removeMarker(None)
|
||||
p.setLegendVisible(True)
|
||||
p.addSeries(LinePlotSeries("PSSS Spectrum Average"))
|
||||
|
||||
if RANGE_OFF is not None:
|
||||
RANGE_FROM = energy_machine.read()-RANGE_OFF
|
||||
RANGE_TO = energy_machine.read()+RANGE_OFF
|
||||
|
||||
run("cpython/wrapper")
|
||||
|
||||
# get current camera ROIs and then set to max for scan
|
||||
roi_min = psss_roi_min.read()
|
||||
roi_max = psss_roi_max.read()
|
||||
psss_roi_min.write(1)
|
||||
psss_roi_max.write(2000)
|
||||
|
||||
|
||||
#Scan and take data
|
||||
class PSSS_energy(Writable):
|
||||
def write(self, value):
|
||||
#if not is_dry_run():
|
||||
psss_energy.write(value)
|
||||
exec_cpython("/ioc/modules/qt/PSSS_motion.py", args = ["-m1", "SARFE10-PSSS059"])
|
||||
# python / ioc / modules / qt / PSSS_motion.py - m1 SARFE10 - PSSS059
|
||||
time.sleep(1)
|
||||
print(value)
|
||||
|
||||
en = PSSS_energy()
|
||||
en.alias = "energy"
|
||||
|
||||
av = create_averager(psss_spectrum_y, NUM_SHOTS, interval=-1, name="spectrum_average")
|
||||
av_samples = av.samples
|
||||
av_samples.alias = "spectrum_samples"
|
||||
|
||||
|
||||
def after_read(record, scan):
|
||||
p.getSeries(0).setData(psss_spectrum_x.take(), record[av])
|
||||
p.setTitle("Energy = %1.3f" %(record[en]))
|
||||
|
||||
r = lscan(en, (av, av_samples), RANGE_FROM, RANGE_TO, STEPS, latency=0.0, after_read = after_read, save=False )
|
||||
average, samples, energy_range = r.getReadable(0), r.getReadable(1), r.getPositions(0)
|
||||
# return ROI to inital value
|
||||
psss_roi_min.write(roi_min)
|
||||
psss_roi_max.write(roi_max)
|
||||
|
||||
[amp, mean_val, sigma, offset],centre_line_out = fit_energy(RANGE_FROM, RANGE_TO, STEPS+1, NUM_SHOTS, samples)
|
||||
|
||||
if not (RANGE_FROM < mean_val < RANGE_TO or RANGE_TO < mean_val < RANGE_FROM):
|
||||
|
||||
raise Exception ("Invalid fit mean: " + str(mean_val))
|
||||
|
||||
|
||||
measured_offset = energy_machine.read() - mean_val
|
||||
#Set fitted energy
|
||||
print "measured offset", measured_offset
|
||||
|
||||
en.write(mean_val)
|
||||
|
||||
|
||||
p.clear()
|
||||
p.setTitle("")
|
||||
plot_gauss_fit(energy_range, centre_line_out, gauss_pars=(offset, amp, mean_val, sigma), p=PLOT, title = "Data")
|
||||
|
||||
|
||||
set_return(mean_val)
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Executable
+184
@@ -0,0 +1,184 @@
|
||||
from collections import deque
|
||||
|
||||
PSSS_CAMERA_NAME = "SARFE10-PSSS059";
|
||||
|
||||
def integrate_arrays(arrays):
|
||||
if arrays is None or (len(arrays)==0):
|
||||
return None
|
||||
ret = arrays[0]
|
||||
for a in arrays[1:]:
|
||||
ret=arradd(ret, a)
|
||||
return ret
|
||||
|
||||
def average_arrays(arrays):
|
||||
ret = integrate_arrays(arrays)
|
||||
if ret is not None:
|
||||
s=len(arrays)
|
||||
ret = [x/s for x in ret]
|
||||
return ret
|
||||
|
||||
|
||||
def get_psss_data(average=1):
|
||||
ax,ay,ac,af=[],[],[],[]
|
||||
x = psss_spectrum_x.take()
|
||||
for i in range(average):
|
||||
y = psss_spectrum_y.take()
|
||||
center,fwhm = psss_center.take(), psss_fwhm.take()
|
||||
if average==1:
|
||||
return x,y,center,fwhm
|
||||
#ax.append(x)
|
||||
ay.append(y)
|
||||
ac.append(center)
|
||||
af.append(fwhm)
|
||||
if i < (average-1):
|
||||
psss_spectrum_y.waitCacheChange(2000)
|
||||
#psss_center.waitCacheChange(1)
|
||||
#psss_fwhm.waitCacheChange(1)
|
||||
#x=average_arrays(ax)
|
||||
y=average_arrays(ay)
|
||||
center=mean(ac)
|
||||
fwhm=mean(af)
|
||||
return x,y,center,fwhm
|
||||
|
||||
def plot_psss(p, h=None, average = None):
|
||||
"""
|
||||
if len(p.getMarkers())==0:
|
||||
m1=p.addMarker(0,None,"",Color.WHITE)
|
||||
m2=p.addMarker(0,None,"",Color.WHITE)
|
||||
m2.setLabelAnchor(RectangleAnchor.TOP)
|
||||
else:
|
||||
m1,m2 = p.getMarkers()
|
||||
"""
|
||||
|
||||
#Manipulate axis (use PSSS_PLOT for the global object):
|
||||
#p.getAxis(LinePlot.AxisId.X).
|
||||
|
||||
# Setup queues
|
||||
if p.getNumberOfSeries()==0:
|
||||
center_queue = deque(maxlen=100)
|
||||
fwhm_queue = deque(maxlen=100)
|
||||
|
||||
# Setup figures
|
||||
|
||||
if p.getNumberOfSeries()==0:
|
||||
p.addSeries(LinePlotSeries("spectrum"))
|
||||
p.addSeries(LinePlotSeries("average"))
|
||||
p.setLegendVisible(True)
|
||||
p.getAxis(LinePlot.AxisId.X)
|
||||
p.getAxis(LinePlot.AxisId.X).setLabel("Energy [eV]")
|
||||
p.getAxis(LinePlot.AxisId.Y).setLabel("Sum counts")
|
||||
if len(p.getMarkers())==0:
|
||||
paint = RangeSelectionPlot().getSelectionColor() #p.chart.getBackgroundPaint()
|
||||
m=p.addIntervalMarker(0,0, None,"", paint)
|
||||
m.setLabelAnchor(RectangleAnchor.BOTTOM)
|
||||
m.alpha=0.2
|
||||
m.setLabelPaint(Color.WHITE)
|
||||
else:
|
||||
m = p.getMarkers()[0]
|
||||
|
||||
|
||||
x,y, = psss_spectrum_x.take(), psss_spectrum_y.take()
|
||||
# update spectral plot
|
||||
if (x is None) or (y is None):
|
||||
p.getSeries(0).clear()
|
||||
else:
|
||||
p.getSeries(0).setData(x,y)
|
||||
if (x is None) or (y is None):
|
||||
p.getSeries(0).clear()
|
||||
else:
|
||||
p.getSeries(0).setData(x,y)
|
||||
|
||||
if average is not None:
|
||||
print "Average: ", average
|
||||
x,y, center,fwhm = get_psss_data(average)
|
||||
else:
|
||||
y = psss_spectrum_y_average.take()
|
||||
center = psss_center_average.take()
|
||||
fwhm = psss_fwhm_average.take()
|
||||
|
||||
if (x is None) or (y is None):
|
||||
p.getSeries(1).clear()
|
||||
else:
|
||||
p.getSeries(1).setData(x,y)
|
||||
|
||||
|
||||
if (center!= None) and (fwhm!=None):
|
||||
center=center.doubleValue()
|
||||
fwhm=fwhm.doubleValue()
|
||||
m.startValue, m.endValue = center - fwhm/2, center + fwhm/2
|
||||
m.label = str(center)
|
||||
|
||||
if h:
|
||||
if h.getNumberOfSeries()==0:
|
||||
h.addSeries(TimePlotSeries("centre"))
|
||||
h.addSeries(TimePlotSeries("Energy spread SS",2))
|
||||
h.addSeries(TimePlotSeries("Energy spread cum avg",2))
|
||||
h.setLegendVisible(True)
|
||||
h.setTimeAxisLabel("")
|
||||
h.getAxis(Timeplot.AxisId.Y1).setLabel("Central energy [eV]")
|
||||
per_mil = (fwhm/center)*1e3
|
||||
per_mil_avg = psss_fwhm_avg.take()
|
||||
h.getSeries(0).appendData(center)
|
||||
h.getSeries(1).appendData(per_mil)
|
||||
h.getSeries(2).appendData(per_mil_avg)
|
||||
return center,fwhm
|
||||
|
||||
ovmin, ovmax, ovavg = None, None, None
|
||||
|
||||
def update_psss_image(renderer):
|
||||
global ovmin, ovmax
|
||||
#if ovmin: ovmin.update(Point(0,psss_roi_min.take()))
|
||||
#if ovmax: ovmax.update(Point(0,psss_roi_max.take()))
|
||||
|
||||
|
||||
width=psss_spectrum_x.size
|
||||
if ovmin: ovmin.update(Point(0,psss_roi_min.take()), Point(width, psss_roi_min.take()))
|
||||
if ovmax: ovmax.update(Point(0,psss_roi_max.take()), Point(width, psss_roi_max.take()))
|
||||
try:
|
||||
data = renderer.data
|
||||
av = "%1.2f" %(data.integrate(False)/data.width/data.height)
|
||||
except:
|
||||
av = ""
|
||||
if ovavg: ovavg.update(av)
|
||||
|
||||
def enable_psss_image(enabled, renderer):
|
||||
global ovmin, ovmax, ovavg
|
||||
try:
|
||||
if (enabled):
|
||||
#Start or connect to ScreenPanel pipeline
|
||||
renderer.setDevice(cam_server)
|
||||
renderer.setProfile(renderer.Profile.Both)
|
||||
renderer.setShowProfileLimits(False)
|
||||
|
||||
#Changing colormap
|
||||
#print Colormap.values() #Check values
|
||||
cam_server.config.colormap=Colormap.Temperature
|
||||
|
||||
|
||||
cam_server.start(PSSS_CAMERA_NAME + "_sp", True)
|
||||
#ovmin, ovmax= Overlays.Crosshairs(renderer.getPenMarker(), Dimension(-1,1)), \
|
||||
# Overlays.Crosshairs(renderer.getPenMarker(), Dimension(-1,1))
|
||||
ovmin, ovmax= Overlays.Line(renderer.getPenMarker()), Overlays.Line(renderer.getPenMarker())
|
||||
|
||||
ovavg = Overlays.Text(Pen(java.awt.Color.GREEN.darker()), "", \
|
||||
java.awt.Font("Verdana", java.awt.Font.PLAIN, 12), java.awt.Point(-50,20))
|
||||
ovavg.fixed=True
|
||||
ovavg.anchor=Overlay.ANCHOR_IMAGE_TOP_RIGHT
|
||||
renderer.addOverlays([ovmin, ovmax, ovavg])
|
||||
update_psss_image(renderer)
|
||||
else:
|
||||
ovmin, ovmax, ovavg = None, None, None
|
||||
renderer.setDevice(None)
|
||||
renderer.clearOverlays()
|
||||
cam_server.stop()
|
||||
except:
|
||||
log(sys.exc_info()[1])
|
||||
|
||||
|
||||
def get_psss_averaging():
|
||||
return psss_spectrum_y_average.config.measures
|
||||
|
||||
def set_psss_averaging(measures):
|
||||
psss_spectrum_y_average.config.measures=measures
|
||||
psss_center_average.config.measures=measures
|
||||
psss_fwhm_average.config.measures=measures
|
||||
Reference in New Issue
Block a user