From 39c316d6eadfdfbd483661b67720a7e224a46712 Mon Sep 17 00:00:00 2001 From: wakonig_k Date: Thu, 10 Jul 2025 13:54:08 +0200 Subject: [PATCH] fix(image item): fix processor for nans in images --- bec_widgets/widgets/plots/image/image_item.py | 21 ++++++++++++------- 1 file changed, 13 insertions(+), 8 deletions(-) diff --git a/bec_widgets/widgets/plots/image/image_item.py b/bec_widgets/widgets/plots/image/image_item.py index edca5e87..df182538 100644 --- a/bec_widgets/widgets/plots/image/image_item.py +++ b/bec_widgets/widgets/plots/image/image_item.py @@ -213,14 +213,19 @@ class ImageItem(BECConnector, pg.ImageItem): """ Reprocess the current raw data and update the image display. """ - if self.raw_data is not None: - autorange = self.config.autorange - 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.image_transform is not None: - self.setTransform(self.image_transform) - self.autorange = autorange + if self.raw_data is None: + return + + if np.all(np.isnan(self.raw_data)): + return + + autorange = self.config.autorange + 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.image_transform is not None: + self.setTransform(self.image_transform) + self.autorange = autorange @property def fft(self) -> bool: