From 87d546764318679cd80e56d17d590f0e31e51504 Mon Sep 17 00:00:00 2001
From: wyzula-jan <133381102+wyzula-jan@users.noreply.github.com>
Date: Fri, 8 Sep 2023 16:27:23 +0200
Subject: [PATCH] fix: moved mask as a last step of image processing

---
 bec_widgets/examples/eiger_plot/eiger_plot.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/bec_widgets/examples/eiger_plot/eiger_plot.py b/bec_widgets/examples/eiger_plot/eiger_plot.py
index b1efec96..e3847a1b 100644
--- a/bec_widgets/examples/eiger_plot/eiger_plot.py
+++ b/bec_widgets/examples/eiger_plot/eiger_plot.py
@@ -101,9 +101,6 @@ class EigerPlot(QWidget):
     @pyqtSlot()
     def on_image_update(self):
         # TODO first rotate then transpose
-        if self.mask is not None:
-            # self.image = np.ma.masked_array(self.image, mask=self.mask) #TODO test if np works
-            self.image = self.image * (1 - self.mask) + 1
 
         if self.comboBox_rotation.currentIndex() > 0:  # rotate
             self.image = np.rot90(self.image, k=self.comboBox_rotation.currentIndex(), axes=(0, 1))
@@ -114,6 +111,10 @@ class EigerPlot(QWidget):
         if self.checkBox_log.isChecked():
             self.image = np.log(self.image)
 
+        if self.mask is not None:
+            # self.image = np.ma.masked_array(self.image, mask=self.mask) #TODO test if np works
+            self.image = self.image * (1 - self.mask) + 1
+
         self.imageItem.setImage(self.image, autoLevels=False)
 
     ###############################