autorange after fft update

py4dstem · Feb 6, 2024 · 54215bd · 54215bd
1 parent 9e94fe9
commit 54215bd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/py4D_browser/update_views.py b/src/py4D_browser/update_views.py
@@ -189,7 +189,10 @@ def update_real_space_view(self, reset=False):
         self.fft_widget.setImage(
             fft.T, autoLevels=False, levels=levels, autoRange=mode_switch
         )
-        self.fft_widget.getImageItem().setRect(0,0,fft.shape[1],fft.shape[1])
+        self.fft_widget.getImageItem().setRect(0, 0, fft.shape[1], fft.shape[1])
+        if mode_switch:
+            # Need to autorange after setRect
+            self.fft_widget.autoRange()
 
 
 def update_diffraction_space_view(self, reset=False):