Added high-res pass

negstation/widgets/export_widget.py

@@ -0,0 +1,33 @@
+import dearpygui.dearpygui as dpg
+import numpy as np
+
+from .pipeline_stage_widget import PipelineStageWidget
+
+
+class ExportStage(PipelineStageWidget):
+    name = "Export Image"
+    register = True
+    has_pipeline_in = True
+    has_pipeline_out = False
+
+    def __init__(self, manager, logger):
+        super().__init__(manager, logger, default_stage_out="opened_image")
+        self.manager.bus.subscribe(
+            "process_full_res", self._on_process_full_res, True)
+
+    def create_pipeline_stage_content(self):
+        dpg.add_text("Some export fields")
+
+    def _on_process_full_res(self, data):
+        self.logger.info("Starting full res pipeline export")
+
+    def on_pipeline_data(self, img):
+        if img is None:
+            return
+        self.logger.info("low res image received, ignore")
+
+    def on_full_res_pipeline_data(self, img):
+        if img is None:
+            return
+        h, w, _ = img.shape
+        self.logger.info(f"Full res image received: {w}x{h}")