bec/bec_widgets
0
0
Fork 0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-14 11:41:49 +02:00
Code Activity

refactor: stream_plot.py changed client initialization

Browse Source
This commit is contained in:
wyzula-jan
2023-10-17 11:11:11 +02:00
parent 8f83115efc
commit 17ea7ab703
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
bec_widgets/examples/stream_plot/stream_plot.py
View File

@ -17,14 +17,14 @@ from pyqtgraph.Qt.QtCore import pyqtSignal
from bec_widgets.qt_utils import Crosshair, Colors from bec_widgets.qt_utils import Crosshair, Colors
from bec_widgets.bec_dispatcher import bec_dispatcher from bec_widgets.bec_dispatcher import bec_dispatcher
client = bec_dispatcher.client # client = bec_dispatcher.client
class StreamPlot(QtWidgets.QWidget): class StreamPlot(QtWidgets.QWidget):
update_signal = pyqtSignal() update_signal = pyqtSignal()
roi_signal = pyqtSignal(tuple) roi_signal = pyqtSignal(tuple)
def __init__(self, name="", y_value_list=["gauss_bpm"]) -> None: def __init__(self, name="", y_value_list=["gauss_bpm"], client=None, parent=None) -> None:
""" """
Basic plot widget for displaying scan data. Basic plot widget for displaying scan data.
@ -33,6 +33,9 @@ class StreamPlot(QtWidgets.QWidget):
y_value_list (list, optional): List of signals to be plotted. Defaults to ["gauss_bpm"]. y_value_list (list, optional): List of signals to be plotted. Defaults to ["gauss_bpm"].
""" """
# Client and device manager from BEC
self.client = bec_dispatcher.client if client is None else client
super(StreamPlot, self).__init__() super(StreamPlot, self).__init__()
# Set style for pyqtgraph plots # Set style for pyqtgraph plots
pg.setConfigOption("background", "w") pg.setConfigOption("background", "w")
@ -293,7 +296,7 @@ class StreamPlot(QtWidgets.QWidget):
time.sleep(0.1) time.sleep(0.1)
continue continue
endpoint = f"px_stream/projection_{self._current_proj}/data" endpoint = f"px_stream/projection_{self._current_proj}/data"
msgs = client.producer.lrange(topic=endpoint, start=-1, end=-1) msgs = self.client.producer.lrange(topic=endpoint, start=-1, end=-1)
data = [BECMessage.DeviceMessage.loads(msg) for msg in msgs] data = [BECMessage.DeviceMessage.loads(msg) for msg in msgs]
if not data: if not data:
continue continue
@ -318,7 +321,7 @@ class StreamPlot(QtWidgets.QWidget):
def new_proj(self, content: dict, _metadata: dict): def new_proj(self, content: dict, _metadata: dict):
proj_nr = content["signals"]["proj_nr"] proj_nr = content["signals"]["proj_nr"]
endpoint = f"px_stream/projection_{proj_nr}/metadata" endpoint = f"px_stream/projection_{proj_nr}/metadata"
msg_raw = client.producer.get(topic=endpoint) msg_raw = self.client.producer.get(topic=endpoint)
msg = BECMessage.DeviceMessage.loads(msg_raw) msg = BECMessage.DeviceMessage.loads(msg_raw)
self._current_q = msg.content["signals"]["q"] self._current_q = msg.content["signals"]["q"]
self._current_norm = msg.content["signals"]["norm_sum"] self._current_norm = msg.content["signals"]["norm_sum"]
@ -340,10 +343,13 @@ if __name__ == "__main__":
# default = ["gauss_bpm", "bpm4i", "bpm5i", "bpm6i", "xert"], # default = ["gauss_bpm", "bpm4i", "bpm5i", "bpm6i", "xert"],
value = parser.parse_args() value = parser.parse_args()
print(f"Plotting signals for: {', '.join(value.signals)}") print(f"Plotting signals for: {', '.join(value.signals)}")
# Client from dispatcher
client = bec_dispatcher.client client = bec_dispatcher.client
app = QtWidgets.QApplication([]) app = QtWidgets.QApplication([])
ctrl_c.setup(app) ctrl_c.setup(app)
plot = StreamPlot(y_value_list=value.signals) plot = StreamPlot(y_value_list=value.signals, client=client)
bec_dispatcher.connect_slot(plot.new_proj, "px_stream/proj_nr") bec_dispatcher.connect_slot(plot.new_proj, "px_stream/proj_nr")
bec_dispatcher.connect_slot( bec_dispatcher.connect_slot(