bec/bec_widgets
1
0
Fork 0
mirror of https://github.com/bec-project/bec_widgets.git synced 2026-05-02 21:04:18 +02:00
Code Activity

feat: added FFT

Browse Source
This commit is contained in:
wyzula-jan
2023-09-08 16:34:39 +02:00
parent 87d5467643
commit b984f0f36e
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bec_widgets/examples/eiger_plot/eiger_plot.py
+6 -4
View File
@@ -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)
###############################