# Copyright (C) 2024 ISIS Rutherford Appleton Laboratory UKRI
# SPDX - License - Identifier: GPL-3.0-or-later
from __future__ import annotations
from typing import TYPE_CHECKING
from PyQt5.QtCore import pyqtSignal
from PyQt5.QtGui import QGuiApplication
from PyQt5.QtWidgets import QAction, QDockWidget, QInputDialog, QMenu, QMessageBox, QVBoxLayout, QWidget
from mantidimaging.core.data import ImageStack
from mantidimaging.core.utility.sensible_roi import SensibleROI
from mantidimaging.gui.widgets.mi_image_view.view import MIImageView
from mantidimaging.gui.utility.qt_helpers import INPUT_DIALOG_FLAGS
from ..stack_visualiser.presenter import StackVisualiserPresenter
from .metadata_dialog import MetadataDialog
from .presenter import SVNotification
from ...utility.qt_helpers import populate_menu
if TYPE_CHECKING:
from mantidimaging.gui.windows.main import MainWindowView # noqa:F401 # pragma: no cover
import numpy as np
[docs]
class StackVisualiserView(QDockWidget):
# Signal that signifies when the ROI is updated. Used to update previews in Filter views
roi_updated = pyqtSignal(SensibleROI)
image_view: MIImageView
presenter: StackVisualiserPresenter
layout: QVBoxLayout
def __init__(self, parent: MainWindowView, images: ImageStack):
# enforce not showing a single image
assert images.data.ndim == 3, \
f"Data does NOT have 3 dimensions! Dimensions found: {images.data.ndim}"
# We set the main window as the parent, the effect is the same as
# having no parent, the window will be inside the QDockWidget. If the
# dock is set as a parent the window will be an independent floating
# window
super().__init__(images.name, parent)
self.central_widget = QWidget(self)
self.layout = QVBoxLayout()
self.central_widget.setLayout(self.layout)
self.setWidget(self.central_widget)
self.parent_create_stack = self.parent().create_new_stack
self._main_window = parent
self.presenter = StackVisualiserPresenter(self, images)
self._actions = [("Show history and metadata", self.show_image_metadata),
("Duplicate whole data", lambda: self.presenter.notify(SVNotification.DUPE_STACK)),
("Duplicate current ROI of data",
lambda: self.presenter.notify(SVNotification.DUPE_STACK_ROI)),
("Mark as projections/sinograms", self.mark_as_sinograms), ("", None),
("Toggle averaged image", lambda: self.presenter.notify(SVNotification.TOGGLE_IMAGE_MODE)),
("Create sinograms from stack", lambda: self.presenter.notify(SVNotification.SWAP_AXES)),
("Set ROI", self.set_roi), ("Copy ROI to clipboard", self.copy_roi_to_clipboard), ("", None),
("Change window name", self.change_window_name_clicked),
("Goto projection", self.goto_projection), ("Goto angle", self.goto_angle)]
self._context_actions = self.build_context_menu()
self.image_view = MIImageView(self)
self.image_view.imageItem.menu = self._context_actions
self.actionCloseStack = QAction("Close window", self)
self.actionCloseStack.triggered.connect(self.close)
self.actionCloseStack.setShortcut("Ctrl+W")
nan_check_menu = [("Crop Coordinates", lambda: self._main_window.presenter.show_operation("Crop Coordinates")),
("NaN Removal", lambda: self._main_window.presenter.show_operation("NaN Removal"))]
self.image_view.enable_nan_check(actions=nan_check_menu)
self.addAction(self.actionCloseStack)
self.set_image(self.presenter.images)
self.image_view.roi_changed_callback = self.roi_changed_callback
self.layout.addWidget(self.image_view)
self.connection_stack_changed = self._main_window.stack_changed.connect(
lambda: self.presenter.notify(SVNotification.REFRESH_IMAGE))
@property
def name(self):
return self.windowTitle()
@name.setter
def name(self, name: str):
self.setWindowTitle(name)
@property
def current_roi(self) -> SensibleROI:
return SensibleROI.from_points(*self.image_view.get_roi())
@property
def image(self):
return self.image_view.imageItem
@image.setter
def image(self, to_display: np.ndarray):
self.image_view.setImage(to_display)
[docs]
def set_image(self, image_stack: ImageStack):
self.image = image_stack.data
self.image_view.angles = image_stack.real_projection_angles()
@property
def main_window(self) -> MainWindowView:
return self._main_window
@property
def context_actions(self):
return self._context_actions
@property
def actions(self):
return self._actions
@property
def id(self):
return self.presenter.images.id
[docs]
def closeEvent(self, event):
self.setFloating(False)
self.hide()
super().closeEvent(event)
[docs]
def roi_changed_callback(self, roi: SensibleROI):
self.roi_updated.emit(roi)
[docs]
def goto_projection(self):
projection_to_goto, accepted = QInputDialog.getInt(
self,
"Enter Projection",
"Projection",
0, # Default value
0, # Min projection value
self.presenter.get_num_images(), # Max possible value
flags=INPUT_DIALOG_FLAGS)
if accepted:
self.image_view.set_selected_image(projection_to_goto)
[docs]
def goto_angle(self):
projection_to_goto, accepted = QInputDialog.getDouble(
self,
"Enter Angle",
"Angle in Degrees",
0, # Default value
0, # Min projection value
2147483647, # Max possible value
4, # Digits/decimals
flags=INPUT_DIALOG_FLAGS)
if accepted:
self.image_view.set_selected_image(self.presenter.find_image_from_angle(projection_to_goto))
[docs]
def set_roi(self):
roi, accepted = QInputDialog.getText(
self,
"Manual ROI",
"Enter ROI in order left, top, right, bottom, with commas in-between each number",
text="0, 0, 50, 50",
flags=INPUT_DIALOG_FLAGS)
if accepted:
roi = [int(r.strip()) for r in roi.split(",")]
self.image_view.set_roi(roi)
self.image_view.roi.show()
[docs]
def copy_roi_to_clipboard(self):
pos, size = self.image_view.get_roi()
QGuiApplication.clipboard().setText(f"{pos.x}, {pos.y}, {pos.x + size.x}, {pos.y + size.y}")
[docs]
def change_window_name_clicked(self):
new_window_name, ok = QInputDialog().getText(self,
"Change window name",
"Name:",
text=self.name,
flags=INPUT_DIALOG_FLAGS)
if ok:
if new_window_name not in self.main_window.stack_names:
self.main_window.rename_stack(self.name, new_window_name)
else:
error = QMessageBox(self)
error.setWindowTitle("Stack name conflict")
error.setText(f"There is already a window named {new_window_name}")
error.exec()
[docs]
def mark_as_sinograms(self):
# 1 is position of sinograms, 0 is projections
current = 1 if self.presenter.images._is_sinograms else 0
item, accepted = QInputDialog.getItem(self,
"Select if projections or sinograms",
"Images are:", ["projections", "sinograms"],
current,
flags=INPUT_DIALOG_FLAGS)
if accepted:
self.presenter.images._is_sinograms = False if item == "projections" else True
self._main_window.stack_changed.emit()
[docs]
def ask_confirmation(self, msg: str):
response = QMessageBox.question(self, "Confirm action", msg, QMessageBox.Ok | QMessageBox.Cancel) # type:ignore
return response == QMessageBox.Ok
[docs]
def cleanup(self):
self._main_window.stack_changed.disconnect(self.connection_stack_changed)
self.presenter = None