From 324a5bd3d9ed278495c6ba62453b02061900ae32 Mon Sep 17 00:00:00 2001 From: wakonig_k Date: Fri, 4 Jul 2025 13:31:04 +0200 Subject: [PATCH] feat(image_item): add support for qtransform --- bec_widgets/widgets/plots/image/image_item.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/bec_widgets/widgets/plots/image/image_item.py b/bec_widgets/widgets/plots/image/image_item.py index 072d1bc1..4b0765fb 100644 --- a/bec_widgets/widgets/plots/image/image_item.py +++ b/bec_widgets/widgets/plots/image/image_item.py @@ -7,6 +7,7 @@ import pyqtgraph as pg from bec_lib.logger import bec_logger from pydantic import Field, ValidationError, field_validator from qtpy.QtCore import Signal +from qtpy.QtGui import QTransform from bec_widgets.utils import BECConnector, Colors, ConnectionConfig from bec_widgets.widgets.plots.image.image_processor import ( @@ -88,6 +89,7 @@ class ImageItem(BECConnector, pg.ImageItem): super().__init__(config=config, gui_id=gui_id, **kwargs) self.raw_data = None + self.transform = None self.buffer = [] self.max_len = 0 @@ -100,8 +102,9 @@ class ImageItem(BECConnector, pg.ImageItem): def parent(self): return self.parent_image - def set_data(self, data: np.ndarray): + def set_data(self, data: np.ndarray, transform: QTransform | None = None): self.raw_data = data + self.transform = transform self._process_image() ################################################################################ @@ -215,6 +218,8 @@ class ImageItem(BECConnector, pg.ImageItem): self._image_processor.set_config(self.config.processing) processed_data = self._image_processor.process_image(self.raw_data) self.setImage(processed_data, autoLevels=False) + if self.transform is not None: + self.setTransform(self.transform) self.autorange = autorange @property