34 lines
986 B
Python
34 lines
986 B
Python
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}")
|