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()
def slice_line_moved(self):
self.slice_changed(int(self.slice_line.value()))
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)
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)
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)
def update_recon_hist(self):
self.recon_hist.imageChanged(autoLevel=True, autoRange=True)
def mouse_click(self, ev, line: InfiniteLine):
line.setPos(ev.pos())
self.slice_changed(CloseEnoughPoint(ev.pos()).y)
def slice_changed(self, slice_index):
self.parent.presenter.do_preview_reconstruct_slice(
slice_idx=slice_index)
self.sigSliceIndexChanged.emit(slice_index)
def clear_recon(self):
self.imageview_recon.clear()
def reset_slice_and_tilt(self, slice_index):
self.slice_line.setPos(slice_index)
self.hide_tilt()
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)
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)
def reset_recon_histogram(self):
self.recon_hist.autoHistogramRange()
