pxiii_bec/bec_widgets/__init__.py

@@ -0,0 +1 @@
+from .auto_updates import PlotUpdate

pxiii_bec/bec_widgets/auto_updates.py

@@ -0,0 +1,26 @@
+from bec_widgets.cli import AutoUpdates, ScanInfo
+
+
+class PlotUpdate(AutoUpdates):
+
+    # def simple_line_scan(self, info: ScanInfo) -> None:
+    #     """
+    #     Simple line scan.
+    #     """
+    #     dev_x = info.scan_report_devices[0]
+    #     dev_y = self.get_selected_device(info.monitored_devices, self.figure.selected_device)
+    #     if not dev_y:
+    #         return
+    #     self.figure.clear_all()
+    #     plt = self.figure.plot(dev_x, dev_y)
+    #     plt.set(title=f"PXIII: Scan {info.scan_number}", x_label=dev_x, y_label=dev_y)
+
+    def handler(self, info: ScanInfo) -> None:
+        # EXAMPLES:
+        # if info.scan_name == "line_scan" and info.scan_report_devices:
+        #     self.simple_line_scan(info)
+        #     return
+        # if info.scan_name == "grid_scan" and info.scan_report_devices:
+        #     self.run_grid_scan_update(info)
+        #     return
+        super().handler(info)

pyproject.toml

@@ -46,8 +46,8 @@ plugin_scans = "pxiii_bec.scans"
 plugin_ipython_client_pre = "pxiii_bec.bec_ipython_client.startup.pre_startup"
 plugin_ipython_client_post = "pxiii_bec.bec_ipython_client.startup"
 
-[project.entry-points."bec.widgets"]
-plugin_widgets = "pxiii_bec.bec_widgets"
+[project.entry-points."bec.widgets.auto_updates"]
+plugin_widgets_update = "pxiii_bec.bec_widgets.auto_updates:PlotUpdate"
 
 [tool.hatch.build.targets.wheel]
 include = ["*"]