diff --git a/bec_widgets/widgets/figure/plots/image/image.py b/bec_widgets/widgets/figure/plots/image/image.py index 151b64bf..f0aee084 100644 --- a/bec_widgets/widgets/figure/plots/image/image.py +++ b/bec_widgets/widgets/figure/plots/image/image.py @@ -496,6 +496,7 @@ class BECImageShow(BECPlotBase): data = msg["data"] device = msg["device"] image = self._images["device_monitor"][device] + image.raw_data = data self.process_image(device, image, data) @pyqtSlot(str, np.ndarray) @@ -528,7 +529,7 @@ class BECImageShow(BECPlotBase): """ for source, images in self._images.items(): for image_id, image in images.items(): - data = image.get_data() + data = image.raw_data self.process_image(image_id, image, data) def _connect_device_monitor(self, monitor: str): diff --git a/bec_widgets/widgets/figure/plots/image/image_item.py b/bec_widgets/widgets/figure/plots/image/image_item.py index 71b5c0a8..16bfd9c5 100644 --- a/bec_widgets/widgets/figure/plots/image/image_item.py +++ b/bec_widgets/widgets/figure/plots/image/image_item.py @@ -71,6 +71,7 @@ class BECImageItem(BECConnector, pg.ImageItem): self.parent_image = parent_image self.colorbar_bar = None + self._raw_data = None self._add_color_bar( self.config.color_bar, self.config.vrange @@ -80,6 +81,14 @@ class BECImageItem(BECConnector, pg.ImageItem): self.set(**kwargs) self.connected = False + @property + def raw_data(self) -> np.ndarray: + return self._raw_data + + @raw_data.setter + def raw_data(self, data: np.ndarray): + self._raw_data = data + def apply_config(self): """ Apply current configuration.