ensure RAW exports get the unscaled data

src/py4D_browser/menu_actions.py

@@ -96,7 +96,7 @@ def export_datacube(self, save_format: str):
         )
 
         if response == QMessageBox.Cancel:
-            print("Cancelling due to user guilt")
+            self.statusBar().showMessage("Cancelling due to user guilt", 5_000)
             return
 
     filename = self.get_savefile_name(save_format)
@@ -135,6 +135,11 @@ def export_virtual_image(self, im_format: str, im_type: str):
     elif im_format == "TIFF (raw)":
         from tifffile import TiffWriter
 
+        vimg = (
+            self.unscaled_realspace_image
+            if im_type == "image"
+            else self.unscaled_diffraction_image
+        )
         with TiffWriter(filename) as tw:
             tw.write(vimg)
 

src/py4D_browser/update_views.py

@@ -171,6 +171,9 @@ def update_real_space_view(self, reset=False):
         new_view = np.sqrt(np.maximum(vimg, 0))
     else:
         raise ValueError("Mode not recognized")
+
+    self.unscaled_realspace_image = vimg
+
     self.real_space_widget.setImage(
         new_view.T,
         autoLevels=reset or self.realspace_rescale_button.latched,
@@ -232,6 +235,8 @@ def update_diffraction_space_view(self, reset=False):
     else:
         raise ValueError("Detector shape not recognized")
 
+    self.unscaled_diffraction_image = DP
+
     if scaling_mode == "Linear":
         new_view = DP
     elif scaling_mode == "Log":