Files
ALF/dioderata/online_abs.py
2023-10-23 19:03:37 +02:00

81 lines
1.8 KiB
Python
Executable File

#!/usr/bin/env python
from time import sleep
from collections import deque
import numpy as np
import scipy.signal
from zoetrope import aniplot as plt
from bstrd import BS, bsstream
plt.blit = False
plt.style.use('ggplot')
def unpumpedXrays(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 pumpedXrays(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
#chname_diode = "SLAAR11-LSCP1-FNS:CH0:VAL_GET"
chname_diode = "SARES11-SPEC125-M1.edge_amplitude"
#chname_diode = "SARES11-GES1:CH2_VAL_GET"
chname_events = "SAR-CVME-TIFALL4:EvtSet"
length = 500
# create channel
ch_diode = BS(chname_diode)
ch_events = BS(chname_events)
n = 10
sigs = np.empty(n)
evts = np.empty((n, 256))
# create a buffer for the plotting
pp_sigs = deque(maxlen=length)
# create the empty plot
pd = plt.plot([0])
# some plot settings
plt.suptitle(chname_diode)
plt.fig.set_figheight(5)
plt.fig.set_figwidth(15)
plt.tight_layout()
for counter, data in zip(plt.show(), bsstream):
print(counter)
for i in range(n):
evts[i] = ch_events.get()
sigs[i] = ch_diode.get()
next(bsstream) # this gets the next set of data
# sigs_p = pumpedXrays(evts, sigs)
# sigs_u = unpumpedXrays(evts, sigs)
sig = np.mean(sigs)
# sig_p = np.mean(sigs_p)
# sig_u = np.mean(sigs_u)
# sig = -np.log10(sig_p/sig_u)
pp_sigs.append(sig)
xs = np.arange(len(pp_sigs))
pd.set(xs, pp_sigs)
# this, I need to move into the library
pd.ax.relim()
pd.ax.autoscale_view()
bsstream.close()