feat: eiger_plot.py in example folder with new GUI

bec_widgets/examples/eiger_plot/eiger_plot.py

@@ -0,0 +1,122 @@
+import json
+import os
+import threading
+import time
+
+import numpy as np
+import pyqtgraph as pg
+import zmq
+from PyQt5.QtCore import pyqtSlot, pyqtSignal
+from PyQt5.QtWidgets import QWidget
+from pyqtgraph.Qt import uic
+
+
+class EigerPlot(QWidget):
+    update_signale = pyqtSignal()
+
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        pg.setConfigOptions(background="w", foreground="k", antialias=True)
+
+        current_path = os.path.dirname(__file__)
+        uic.loadUi(os.path.join(current_path, "eiger_plot.ui"), self)
+
+        self.hist_lims = None
+
+        # UI
+        self.init_ui()
+        self.hook_signals()
+
+        # ZMQ Consumer
+        self.start_zmq_consumer()
+
+    def init_ui(self):
+        # Create Plot and add ImageItem
+        self.plot_item = pg.PlotItem()
+        self.plot_item.setAspectLocked(True)
+        self.imageItem = pg.ImageItem()
+        self.plot_item.addItem(self.imageItem)
+
+        # Setting up histogram
+        self.hist = pg.HistogramLUTItem()
+        self.hist.setImageItem(self.imageItem)
+        self.hist.gradient.loadPreset("magma")
+        self.update_hist()
+
+        # Adding Items to Graphical Layout
+        self.glw.addItem(self.plot_item)
+        self.glw.addItem(self.hist)
+
+    def hook_signals(self):
+        # Buttons
+        self.pushButton_test.clicked.connect(self.start_sim_stream)
+
+        # SpinBoxes
+        self.doubleSpinBox_hist_min.valueChanged.connect(self.update_hist)
+        self.doubleSpinBox_hist_max.valueChanged.connect(self.update_hist)
+
+        # Signal/Slots
+        self.update_signale.connect(self.on_image_update)
+
+    def update_hist(self):
+        self.hist_levels = [
+            self.doubleSpinBox_hist_min.value(),
+            self.doubleSpinBox_hist_max.value(),
+        ]
+        self.hist.setLevels(min=self.hist_levels[0], max=self.hist_levels[1])
+        self.hist.setHistogramRange(
+            self.hist_levels[0] - 0.1 * self.hist_levels[0],
+            self.hist_levels[1] + 0.1 * self.hist_levels[1],
+        )
+
+    def start_zmq_consumer(self):
+        consumer_thread = threading.Thread(target=self.zmq_consumer, daemon=True).start()
+
+    def zmq_consumer(self):
+        try:
+            print("starting consumer")
+            live_stream_url = "tcp://129.129.95.38:20000"
+            receiver = zmq.Context().socket(zmq.SUB)
+            receiver.connect(live_stream_url)
+            receiver.setsockopt_string(zmq.SUBSCRIBE, "")
+
+            while True:
+                raw_meta, raw_data = receiver.recv_multipart()
+                meta = json.loads(raw_meta.decode("utf-8"))
+                self.image = np.frombuffer(raw_data, dtype=meta["type"]).reshape(meta["shape"])
+                self.update_signale.emit()
+
+        finally:
+            receiver.disconnect(live_stream_url)
+            receiver.context.term()
+
+    @pyqtSlot()
+    def on_image_update(self):
+        self.imageItem.setImage(self.image, autoLevels=False)
+
+    ###############################
+    # just simulations from here
+    ###############################
+    def start_sim_stream(self):
+        sim_stream_thread = threading.Thread(target=self.sim_stream, daemon=True)
+        sim_stream_thread.start()
+
+    def sim_stream(self):
+        for i in range(100):
+            self.image = np.random.rand(100, 100) * 10
+            self.update_signale.emit()
+            time.sleep(0.1)
+
+
+if __name__ == "__main__":
+    import sys
+
+    from PyQt5.QtWidgets import QApplication
+
+    app = QApplication(sys.argv)
+
+    plot = EigerPlot()
+
+    plot.show()
+
+    sys.exit(app.exec_())