Source code for mantidimaging.gui.windows.recon.image_view

# Copyright (C) 2022 ISIS Rutherford Appleton Laboratory UKRI
# SPDX - License - Identifier: GPL-3.0-or-later
from math import isnan
from typing import Optional

import numpy as np
from PyQt5 import QtCore
from pyqtgraph import GraphicsLayoutWidget, InfiniteLine

from mantidimaging.core.utility.close_enough_point import CloseEnoughPoint
from mantidimaging.core.utility.data_containers import Degrees
from mantidimaging.core.utility.histogram import set_histogram_log_scale
from mantidimaging.gui.widgets.mi_mini_image_view.view import MIMiniImageView



[docs]
class ReconImagesView(GraphicsLayoutWidget):
    sigSliceIndexChanged = QtCore.pyqtSignal(int)

    def __init__(self, parent):
        super().__init__(parent)
        self.parent = parent

        self.imageview_projection = MIMiniImageView(name="Projection")
        self.imageview_sinogram = MIMiniImageView(name="Sinogram")
        self.imageview_recon = MIMiniImageView(name="Recon")

        self.projection, self.projection_vb, self.projection_hist = self.imageview_projection.get_parts()
        self.sinogram, self.sinogram_vb, self.sinogram_hist = self.imageview_sinogram.get_parts()
        self.recon, self.recon_vb, self.recon_hist = self.imageview_recon.get_parts()

        self.slice_line = InfiniteLine(pos=1024,
                                       angle=0,
                                       bounds=[0, self.imageview_projection.image_item.width()],
                                       movable=True)
        self.projection_vb.addItem(self.slice_line)
        self.tilt_line = InfiniteLine(pos=1024, angle=90, pen=(255, 0, 0, 255), movable=True)

        self.addItem(self.imageview_projection, 0, 0)
        self.addItem(self.imageview_recon, 0, 1, rowspan=2)
        self.addItem(self.imageview_sinogram, 1, 0)

        self.imageview_projection.image_item.mouseClickEvent = lambda ev: self.mouse_click(ev, self.slice_line)
        self.slice_line.sigPositionChangeFinished.connect(self.slice_line_moved)

        # Work around for https://github.com/mantidproject/mantidimaging/issues/565
        self.scene().contextMenu = [item for item in self.scene().contextMenu if "export" not in item.text().lower()]

        self.imageview_projection.enable_nan_check()
        self.imageview_sinogram.enable_nan_check()
        self.imageview_recon.enable_nan_check()
        self.imageview_projection.enable_nonpositive_check()
        self.imageview_sinogram.enable_nonpositive_check()


[docs]
    def slice_line_moved(self):
        self.slice_changed(int(self.slice_line.value()))



[docs]
    def update_projection(self, image_data: np.ndarray, preview_slice_index: int, tilt_angle: Optional[Degrees]):
        self.imageview_projection.clear()
        self.imageview_projection.setImage(image_data)
        self.projection_hist.imageChanged(autoLevel=True, autoRange=True)
        self.slice_line.setPos(preview_slice_index)
        if tilt_angle:
            self.set_tilt(tilt_angle, image_data.shape[1] // 2)
        else:
            self.hide_tilt()
        set_histogram_log_scale(self.projection_hist)



[docs]
    def update_sinogram(self, image):
        self.imageview_sinogram.clear()
        self.imageview_sinogram.setImage(image)
        self.sinogram_hist.imageChanged(autoLevel=True, autoRange=True)
        set_histogram_log_scale(self.sinogram_hist)



[docs]
    def update_recon(self, image_data):
        self.imageview_recon.clear()
        self.imageview_recon.setImage(image_data, autoLevels=False)
        set_histogram_log_scale(self.recon_hist)



[docs]
    def update_recon_hist(self):
        self.recon_hist.imageChanged(autoLevel=True, autoRange=True)



[docs]
    def mouse_click(self, ev, line: InfiniteLine):
        line.setPos(ev.pos())
        self.slice_changed(CloseEnoughPoint(ev.pos()).y)



[docs]
    def slice_changed(self, slice_index):
        self.parent.presenter.do_preview_reconstruct_slice(slice_idx=slice_index)
        self.sigSliceIndexChanged.emit(slice_index)



[docs]
    def clear_recon(self):
        self.imageview_recon.clear()



[docs]
    def reset_slice_and_tilt(self, slice_index):
        self.slice_line.setPos(slice_index)
        self.hide_tilt()



[docs]
    def hide_tilt(self):
        """
        Hides the tilt line. This stops infinite zooming out loop that messes up the image view
        (the line likes to be unbound when the degree isn't a multiple o 90 - and the tilt never is)
        :return:
        """
        if self.tilt_line.scene() is not None:
            self.projection_vb.removeItem(self.tilt_line)



[docs]
    def set_tilt(self, tilt: Degrees, pos: Optional[int] = None):
        if not isnan(tilt.value):  # is isnan it means there is no tilt, i.e. the line is vertical
            if pos is not None:
                self.tilt_line.setAngle(90)
                self.tilt_line.setPos(pos)
            self.tilt_line.setAngle(90 + tilt.value)
        self.projection_vb.addItem(self.tilt_line)



[docs]
    def reset_recon_histogram(self):
        self.recon_hist.autoHistogramRange()