Export Window: Create/Add ImageView (Mapping) to Export Window (#2819)

Author: hussam-stfc

docs/release_notes/next/feature-2395-Export-Window-Add-ImageView-to-Tab

@@ -0,0 +1 @@
+#2395: Adds an Image preview to the Export tab and makes the Table/Image view switchable from within the Export tab.

mantidimaging/gui/widgets/spectrum_widgets/export_image_widget.py

@@ -0,0 +1,46 @@
+# Copyright (C) 2021 ISIS Rutherford Appleton Laboratory UKRI
+# SPDX - License - Identifier: GPL-3.0-or-later
+from __future__ import annotations
+
+import numpy as np
+from PyQt5.QtWidgets import QWidget, QVBoxLayout
+import pyqtgraph as pg
+
+
+class ExportImageViewWidget(QWidget):
+    """
+    Minimal image preview widget for the Export tab.
+    """
+
+    def __init__(self, parent: QWidget | None) -> None:
+        super().__init__(parent)
+
+        layout = QVBoxLayout(self)
+        layout.setContentsMargins(0, 0, 0, 0)
+
+        self.image_view = pg.ImageView()
+        self.image_view.ui.roiBtn.hide()
+        self.image_view.ui.menuBtn.hide()
+        self.image_view.ui.histogram.hide()
+
+        layout.addWidget(self.image_view)
+        self.clear()
+
+    def update_image(self, image: np.ndarray | None, autoLevels: bool = True) -> None:
+        if image is None:
+            self.clear()
+            return
+
+        arr = np.asarray(image)
+        if arr.ndim == 3 and arr.shape[0] > 0:
+            arr = arr.mean(axis=0)
+        self.image_view.setImage(arr, autoLevels=autoLevels)
+
+    def clear(self) -> None:
+        """Show a blank canvas."""
+        self.image_view.setImage(np.zeros((1, 1), dtype=np.float32), autoLevels=True)
+
+    @property
+    def image_data(self) -> np.ndarray | None:
+        """Return the currently displayed image array, or None."""
+        return getattr(self.image_view.imageItem, "image", None)

mantidimaging/gui/windows/spectrum_viewer/view.py

@@ -9,8 +9,8 @@
 from PyQt5 import QtWidgets
 from pyqtgraph import mkPen
 from PyQt5.QtGui import QPixmap
-from PyQt5.QtWidgets import (QCheckBox, QVBoxLayout, QFileDialog, QLabel, QGroupBox, QActionGroup, QAction, QSplitter,
-                             QTabWidget)
+from PyQt5.QtWidgets import (QSplitter, QTabWidget, QCheckBox, QVBoxLayout, QFileDialog, QLabel, QGroupBox,
+                             QActionGroup, QAction)
 from PyQt5.QtCore import QModelIndex
 from logging import getLogger
 
@@ -27,6 +27,7 @@
 from mantidimaging.gui.widgets.spectrum_widgets.fitting_param_form_widget import FittingParamFormWidget
 from mantidimaging.gui.widgets.spectrum_widgets.export_settings_widget import FitExportFormWidget
 from mantidimaging.gui.widgets.spectrum_widgets.export_data_table_widget import ExportDataTableWidget
+from mantidimaging.gui.widgets.spectrum_widgets.export_image_widget import ExportImageViewWidget
 
 import numpy as np
 
@@ -64,7 +65,6 @@ def __init__(self, main_window: MainWindowView):
         super().__init__(None, 'gui/ui/spectrum_viewer.ui')
 
         self.main_window = main_window
-
         splitter = self.findChild(QSplitter, "splitter")
         sidebar = self.findChild(QTabWidget, "formTabs")
 
@@ -99,6 +99,19 @@ def __init__(self, main_window: MainWindowView):
         self.scalable_roi_widget = FittingParamFormWidget(self.presenter)
         self.fittingFormLayout.layout().addWidget(self.scalable_roi_widget)
 
+        self.export_display_tabs = QTabWidget(self)
+        self.exportDataTableWidget = ExportDataTableWidget()
+        self.export_display_tabs.addTab(self.exportDataTableWidget, "Table")
+
+        self._export_image_widget = ExportImageViewWidget(self)
+        self.export_display_tabs.addTab(self._export_image_widget, "Image")
+        self.exportLayout.addWidget(self.export_display_tabs)
+        export_image = self.spectrum_widget.image.image_item.image
+        if export_image is not None:
+            self._export_image_widget.update_image(export_image)
+        self.spectrum_widget.image.image_item.sigImageChanged.connect(
+            lambda: self._export_image_widget.update_image(self.spectrum_widget.image.image_item.image))
+
         self.exportSettingsWidget = FitExportFormWidget()
         self.exportFormLayout.layout().addWidget(self.exportSettingsWidget)
         self.exportSettingsWidget.exportButton.clicked.connect(self.presenter.handle_export_table)
@@ -155,8 +168,6 @@ def __init__(self, main_window: MainWindowView):
 
         self.roi_form.exportButton.clicked.connect(self.presenter.handle_export_csv)
         self.roi_form.exportButtonRITS.clicked.connect(self.presenter.handle_rits_export)
-        self.exportDataTableWidget = ExportDataTableWidget()
-        self.exportLayout.addWidget(self.exportDataTableWidget)
 
         self.roi_form.table_view.clicked.connect(self.handle_table_click)