From bfef71382e6a1180d750d2c800650942c5da7a21 Mon Sep 17 00:00:00 2001
From: wyzula-jan <133381102+wyzula-jan@users.noreply.github.com>
Date: Thu, 7 Sep 2023 15:27:33 +0200
Subject: [PATCH] feat: simulation and simple 2D plot for mca card stream

---
 bec_widgets/examples/mca_readout/mca_plot.py | 113 +++++++++++++++++++
 bec_widgets/examples/mca_readout/mca_sim.py  |  28 +++++
 2 files changed, 141 insertions(+)
 create mode 100644 bec_widgets/examples/mca_readout/mca_plot.py
 create mode 100644 bec_widgets/examples/mca_readout/mca_sim.py

diff --git a/bec_widgets/examples/mca_readout/mca_plot.py b/bec_widgets/examples/mca_readout/mca_plot.py
new file mode 100644
index 00000000..89ed1581
--- /dev/null
+++ b/bec_widgets/examples/mca_readout/mca_plot.py
@@ -0,0 +1,113 @@
+# import simulation_progress as SP
+import numpy as np
+import pyqtgraph as pg
+
+from PyQt5.QtCore import pyqtSlot, pyqtSignal
+from PyQt5.QtWidgets import (
+    QApplication,
+    QVBoxLayout,
+    QLabel,
+    QWidget,
+    QProgressBar,
+    QPushButton,
+)
+
+from bec_lib.core import MessageEndpoints, RedisConnector, BECMessage
+
+
+class StreamApp(QWidget):
+    update_plot = pyqtSignal()
+
+    def __init__(self):
+        super().__init__()
+
+        # Create layout and add widgets
+        self.layout = QVBoxLayout()
+        self.setLayout(self.layout)
+
+        # Create plot
+        self.plot_widget = pg.PlotWidget(title="2D plot for mcs data")
+        self.image_item = pg.ImageItem()
+        self.label_id = pg.LabelItem(justify="left")
+        self.plot_widget.addItem(self.label_id)
+        self.plot_widget.addItem(self.image_item)
+
+        # Add widgets to the layout
+        self.layout.addWidget(self.plot_widget)
+
+        self.stream_consumer = connector.stream_consumer(
+            topics=MessageEndpoints.device_async_readback(scanID="ScanID1", device="mca"),
+            cb=self._stremer_cb,
+            parent=self,
+        )
+
+        self.stream_consumer.start()
+
+        self.data = np.array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0])
+
+        self.scanID = None
+
+        self.update_plot.connect(self.plot_new)
+
+    def plot_new(self):
+        self.image_item.setImage(self.data)
+
+    @pyqtSlot(dict, dict)
+    def get_stream(self, msg, metadata):
+        print(msg)
+        print(metadata)
+
+    # def connect_stream_slot(self, slot, topic):
+
+    @staticmethod
+    def _stremer_cb(msg, *, parent, **_kwargs) -> None:
+        msgMCS = BECMessage.DeviceMessage.loads(msg.value)
+
+        row = msgMCS.content["signals"]["mca1"]
+        metadata = msgMCS.metadata
+
+        current_scanID = metadata.get("scanID", None)
+        if current_scanID is None:
+            return
+
+        if current_scanID != parent.scanID:
+            parent.scanID = current_scanID
+            parent.data = row
+            parent.image_item.clear()
+
+        # print(f"metadata: {metadata}")
+
+        # add row vertically to data
+        parent.data = np.vstack((parent.data, row))
+
+        parent.update_plot.emit()
+
+        # print(f"msg: {msgMCS}")
+        # print(f"row: {row}")
+
+
+if __name__ == "__main__":
+    from bec_widgets.bec_dispatcher import bec_dispatcher
+
+    # BECclient global variables
+    # client = bec_dispatcher.client
+    # client.start()
+    #
+    # dev = client.device_manager.devices
+    # scans = client.scans
+    # queue = client.queue
+
+    connector = RedisConnector("localhost:6379")
+
+    app = QApplication([])
+    progressApp = StreamApp()
+
+    # bec_dispatcher.connect_slot(
+    #     slot=progressApp.get_stream,
+    #     topic=MessageEndpoints.device_async_readback(scanID="ScanID1", device="mca"),
+    # )
+
+    # window = ProgressApp()
+    # window.show()
+    progressApp.show()
+    app.exec_()
diff --git a/bec_widgets/examples/mca_readout/mca_sim.py b/bec_widgets/examples/mca_readout/mca_sim.py
new file mode 100644
index 00000000..e059e7e6
--- /dev/null
+++ b/bec_widgets/examples/mca_readout/mca_sim.py
@@ -0,0 +1,28 @@
+from bec_lib.core import BECMessage, MessageEndpoints, RedisConnector
+import time
+
+connector = RedisConnector("localhost:6379")
+producer = connector.producer()
+metadata = {}
+
+scanID = "ScanID1"
+
+metadata.update(
+    {
+        "scanID": scanID,
+        "async_update": "append",
+    }
+)
+for ii in range(20):
+    data = {"mca1": [10, 2, 3, 4, 5, 6, 7, 8, 9, 10], "mca2": [10, 9, 8, 7, 6, 5, 4, 3, 2, 1]}
+    msg = BECMessage.DeviceMessage(
+        signals=data,
+        metadata=metadata,
+    ).dumps()
+    producer.xadd(
+        topic=MessageEndpoints.device_async_readback(scanID=scanID, device="mca"),
+        msg={"data": msg},
+        expire=1800,
+    )
+    print(f"Sent {ii}")
+    time.sleep(0.5)