forked from SwissFEL/c365
165 lines
5.3 KiB
Python
Executable File
165 lines
5.3 KiB
Python
Executable File
from collections import deque
|
|
import random
|
|
import wx
|
|
import numpy as np
|
|
import epics
|
|
|
|
from datetime import datetime
|
|
|
|
#from epics.wx import MotorPanel
|
|
|
|
from slic.gui.widgets import LabeledMathEntry
|
|
from slic.gui.widgets import make_filled_hbox, make_filled_vbox, EXPANDING
|
|
from slic.gui.widgets.plotting import PlotPanel
|
|
|
|
from bstrd import BS, bsstream
|
|
|
|
def unpumped(events, *arrays):
|
|
laser = events[:, 18]
|
|
darkShot = events[:, 21]
|
|
background_shots = np.logical_and.reduce((laser, darkShot))
|
|
return [a[background_shots] for a in arrays]
|
|
|
|
def pumped(events, *arrays):
|
|
fel = events[:, 13]
|
|
laser = events[:, 18]
|
|
darkShot = events[:, 21]
|
|
pumped_shots = np.logical_and.reduce((laser, np.logical_not(darkShot)))
|
|
return [a[pumped_shots] for a in arrays]
|
|
|
|
|
|
# config
|
|
nshots = 50
|
|
qlength = 100
|
|
|
|
# channels
|
|
|
|
#chname_diode = "SARES11-GES1:PR1_CH2_VAL_GET" #Prime Keysight CH2
|
|
chname_diode = "SARES11-GES1:PR1_CH1_VAL_GET" #Prime Keysight CH1
|
|
#chname_diode = "SARES12-GES1:PR1_CH1_VAL_GET" #Flex Keysight
|
|
#chname_diode = "SLAAR11-LSCP1-FNS:CH0:VAL_GET" #Laser IoXos
|
|
|
|
chname_i0 = "SAROP11-PBPS122:INTENSITY"
|
|
chname_events = "SAR-CVME-TIFALL4:EvtSet"
|
|
|
|
# create channels
|
|
ch_diode = BS(chname_diode)
|
|
ch_i0 = BS(chname_i0)
|
|
ch_events = BS(chname_events)
|
|
|
|
iso_format = "%Y-%m-%d %H:%M:%S"
|
|
|
|
class MainPanel(wx.Panel):
|
|
|
|
def __init__(self, parent):
|
|
super().__init__(parent)
|
|
|
|
self.evts = np.empty((nshots,256))
|
|
self.diode = np.empty(nshots)
|
|
self.i0 = np.empty(nshots)
|
|
self.diode_p_norm = np.empty(nshots)
|
|
self.diode_u_norm = np.empty(nshots)
|
|
|
|
self.diodes_p = deque(maxlen=qlength)
|
|
self.diodes_pstd = deque(maxlen=qlength)
|
|
self.diodes_u = deque(maxlen=qlength)
|
|
self.diodes_ustd = deque(maxlen=qlength)
|
|
self.pp = deque(maxlen=qlength)
|
|
self.ppstd = deque(maxlen=qlength)
|
|
|
|
self.plot_diodes = plot_diodes = PlotPanel(self, figsize=(6,3))
|
|
self.plot_pp = plot_pp = PlotPanel(self, figsize=(6,3))
|
|
|
|
plots1 = (plot_diodes,)
|
|
plots2 = (plot_pp,)
|
|
|
|
hb_plot1 = make_filled_hbox(plots1)
|
|
hb_plot2 = make_filled_hbox(plots2)
|
|
|
|
btn_clearQ = wx.Button(self, label="Clear plots")
|
|
btns = (btn_clearQ,)
|
|
hb_btns = make_filled_hbox(btns)
|
|
|
|
widgets = (hb_plot1, hb_plot2, hb_btns)
|
|
box = make_filled_vbox(widgets, border=1)
|
|
self.SetSizerAndFit(box)
|
|
|
|
btn_clearQ.Bind(wx.EVT_BUTTON, self.on_click_clearQ)
|
|
|
|
self.timer = wx.Timer(self)
|
|
self.Bind(wx.EVT_TIMER, self.on_update, self.timer)
|
|
self.timer.Start(100)
|
|
|
|
|
|
def on_update(self, _event):
|
|
|
|
self.time = datetime.now().strftime(iso_format)
|
|
|
|
self.evts = np.empty((nshots,256))
|
|
self.diode = np.empty(nshots)
|
|
self.i0 = np.empty(nshots)
|
|
|
|
for i in range(nshots):
|
|
tempa = ch_events.get()
|
|
self.evts[i] = tempa
|
|
tempb = ch_diode.get()
|
|
self.diode[i] = tempb
|
|
tempc = ch_i0.get()
|
|
self.i0[i] = tempc
|
|
next(bsstream)
|
|
wx.GetApp().Yield()
|
|
|
|
self.diode_p = pumped(self.evts, self.diode)
|
|
self.diode_u = unpumped(self.evts, self.diode)
|
|
self.i0_p = pumped(self.evts, self.i0)
|
|
self.i0_u = unpumped(self.evts, self.i0)
|
|
|
|
self.diode_p_norm = np.asarray(self.diode_p)/np.asarray(self.i0_p)
|
|
self.diode_u_norm = np.asarray(self.diode_u)/np.asarray(self.i0_u)
|
|
pump = np.nanmean(self.diode_p_norm)
|
|
unpump = np.nanmean(self.diode_u_norm)
|
|
pump_e = np.nanstd(self.diode_p_norm)
|
|
unpump_e = np.nanstd(self.diode_u_norm)
|
|
pp_e = np.sqrt((pump_e/pump)**2+(unpump_e/unpump)**2)
|
|
|
|
self.diodes_p.append(pump)
|
|
self.diodes_pstd.append(pump_e)
|
|
self.diodes_u.append(unpump)
|
|
self.diodes_ustd.append(unpump_e)
|
|
#self.pp.append(-np.log10(pump/unpump))
|
|
self.pp.append(pump - unpump)
|
|
|
|
#self.ppstd.append(pp_e/((pump/unpump)*np.log(10)))
|
|
self.ppstd.append(np.sqrt(pump**2+unpump**2))
|
|
|
|
self.draw_plot()
|
|
|
|
def draw_plot(self):
|
|
self.plot_diodes.clear()
|
|
self.plot_pp.clear()
|
|
|
|
self.plot_diodes.set_xlabel('time ago, a.u.')
|
|
self.plot_diodes.set_ylabel('intensity, a.u.')
|
|
self.plot_diodes.set_title(chname_diode)
|
|
|
|
self.plot_pp.set_xlabel('time ago, a.u.')
|
|
self.plot_pp.set_ylabel('Absorbance')
|
|
|
|
self.plot_diodes.plot(np.arange(0, len(self.diodes_p)), self.diodes_p, '-', color='royalblue', label='on')
|
|
self.plot_diodes.plot(np.arange(0, len(self.diodes_u)), self.diodes_u, '-', color='orange', label='off')
|
|
self.plot_diodes.legend(loc='lower left')
|
|
self.plot_pp.plot(np.arange(0, len(self.pp)), self.pp, '-', color='green')
|
|
#self.plot_pp.fill_between(np.arange(0, len(self.pp)), np.asarray(self.ppstd)+np.asarray(self.pp), -np.asarray(self.ppstd)+np.asarray(self.pp), color='lightgreen')
|
|
#self.plot_pp.set_ylim(-1)
|
|
self.plot_pp.grid()
|
|
|
|
self.plot_diodes.draw()
|
|
self.plot_pp.draw()
|
|
|
|
def on_click_clearQ(self, _event):
|
|
|
|
self.diodes_p.clear()
|
|
self.diodes_u.clear()
|
|
self.pp.clear()
|
|
self.ppstd.clear()
|