From b984f0f36e2178690eaaec091d4a7b9443f2378f Mon Sep 17 00:00:00 2001
From: wyzula-jan <133381102+wyzula-jan@users.noreply.github.com>
Date: Fri, 8 Sep 2023 16:34:39 +0200
Subject: [PATCH] feat: added FFT

---
 bec_widgets/examples/eiger_plot/eiger_plot.py | 10 ++++++----
 bec_widgets/examples/eiger_plot/eiger_plot.ui |  4 ++--
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/bec_widgets/examples/eiger_plot/eiger_plot.py b/bec_widgets/examples/eiger_plot/eiger_plot.py
index e3847a1b..74e72241 100644
--- a/bec_widgets/examples/eiger_plot/eiger_plot.py
+++ b/bec_widgets/examples/eiger_plot/eiger_plot.py
@@ -101,6 +101,12 @@ 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.checkBox_FFT.isChecked():
+            self.image = np.abs(np.fft.fftshift(np.fft.fft2(self.image)))
 
         if self.comboBox_rotation.currentIndex() > 0:  # rotate
             self.image = np.rot90(self.image, k=self.comboBox_rotation.currentIndex(), axes=(0, 1))
@@ -111,10 +117,6 @@ 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)
 
     ###############################
diff --git a/bec_widgets/examples/eiger_plot/eiger_plot.ui b/bec_widgets/examples/eiger_plot/eiger_plot.ui
index 994c94ea..eff6ab39 100644
--- a/bec_widgets/examples/eiger_plot/eiger_plot.ui
+++ b/bec_widgets/examples/eiger_plot/eiger_plot.ui
@@ -69,9 +69,9 @@
        </property>
        <layout class="QVBoxLayout" name="verticalLayout">
         <item>
-         <widget class="QCheckBox" name="checkBox">
+         <widget class="QCheckBox" name="checkBox_FFT">
           <property name="enabled">
-           <bool>false</bool>
+           <bool>true</bool>
           </property>
           <property name="text">
            <string>FFT</string>