From 70d74c774d2b318d99c049f0f03743e77812df98 Mon Sep 17 00:00:00 2001
From: e20636 <e20636@x12sa-comp-1.psi.ch>
Date: Fri, 8 Sep 2023 11:42:19 +0200
Subject: [PATCH] feat: eigerplot added

---
 bec_widgets/eiger_plot_hist.py                | 104 ++++++++++++++++++
 .../motor_movement/csax_bec_config.yaml       |  10 ++
 2 files changed, 114 insertions(+)
 create mode 100644 bec_widgets/eiger_plot_hist.py
 create mode 100644 bec_widgets/examples/motor_movement/csax_bec_config.yaml

diff --git a/bec_widgets/eiger_plot_hist.py b/bec_widgets/eiger_plot_hist.py
new file mode 100644
index 00000000..18ddc362
--- /dev/null
+++ b/bec_widgets/eiger_plot_hist.py
@@ -0,0 +1,104 @@
+import threading
+import numpy as np
+import pyqtgraph as pg
+from PyQt5.QtCore import pyqtProperty, pyqtSlot, pyqtSignal
+from PyQt5.QtWidgets import QHBoxLayout, QWidget, QCheckBox
+
+import zmq
+import json
+
+
+
+
+
+
+class EigerPlot(QWidget):
+    update_signale = pyqtSignal()
+    def __init__(self, parent=None):
+        super().__init__(parent)
+
+        pg.setConfigOptions(background="w", foreground="k", antialias=True)
+
+        self.layout = QHBoxLayout()
+
+        self.setLayout(self.layout)
+
+
+        self.glw = pg.GraphicsLayoutWidget()
+
+        # self.glw.show()
+        # self.setCentralItem(self.glw)
+
+        self.checkBox_FFT = QCheckBox("FFT")
+
+        self.layout.addWidget(self.checkBox_FFT)
+
+        self.layout.addWidget(self.glw)
+        
+        
+        self.plot_item = pg.PlotItem()
+        self.plot_item.setAspectLocked(True)
+        self.imageItem = pg.ImageItem()
+        self.plot_item.addItem(self.imageItem)
+
+        self.glw.addItem(self.plot_item)
+
+        self.hist = pg.HistogramLUTItem()
+        self.hist.setImageItem(self.imageItem)
+        self.hist.setLevels(min=0,max=100)
+        self.hist.gradient.loadPreset('magma')
+
+        self.glw.addItem(self.hist)
+
+        # self.plot_item.addItem(self.hist)
+
+        # add plot and histogram to glw
+        # self.glw.addItem(self.plot_item)
+        # self.glw.addItem(self.hist)
+
+        # self.imageItem.setImage([[0,1,2],[4,5,6]])
+        self.update_signale.connect(self.on_image_update)
+        self.start_zmq_consumer()
+
+    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)
+
+
+
+if __name__ == "__main__":
+    import sys
+
+    from PyQt5.QtWidgets import QApplication
+
+    app = QApplication(sys.argv)
+
+    plot = EigerPlot()
+
+    plot.show()
+
+    sys.exit(app.exec_())
diff --git a/bec_widgets/examples/motor_movement/csax_bec_config.yaml b/bec_widgets/examples/motor_movement/csax_bec_config.yaml
new file mode 100644
index 00000000..d5f96a39
--- /dev/null
+++ b/bec_widgets/examples/motor_movement/csax_bec_config.yaml
@@ -0,0 +1,10 @@
+redis:
+  host: pc15543
+  port: 6379
+mongodb:
+  host: localhost
+  port: 27017
+scibec:
+  host: http://localhost
+  port: 3030
+  beamline: MyBeamline