105 lines
3.3 KiB
Python
105 lines
3.3 KiB
Python
|
from textwrap import dedent
|
||
|
|
|
||
|
|
from ..qt import QtGui, QtCore, QtWidgets
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
from ... import io
|
||
|
|
from ...util import img_as_ubyte
|
||
|
|
from .core import BaseWidget
|
||
|
|
from ..utils import dialogs
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = ['OKCancelButtons', 'SaveButtons']
|
||
|
|
|
||
|
|
|
||
|
|
class OKCancelButtons(BaseWidget):
|
||
|
|
"""Buttons that close the parent plugin.
|
||
|
|
|
||
|
|
OK will replace the original image with the current (filtered) image.
|
||
|
|
Cancel will just close the plugin.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, button_width=80):
|
||
|
|
name = 'OK/Cancel'
|
||
|
|
super(OKCancelButtons, self).__init__(name)
|
||
|
|
|
||
|
|
self.ok = QtWidgets.QPushButton('OK')
|
||
|
|
self.ok.clicked.connect(self.update_original_image)
|
||
|
|
self.ok.setMaximumWidth(button_width)
|
||
|
|
self.ok.setFocusPolicy(QtCore.Qt.NoFocus)
|
||
|
|
self.cancel = QtWidgets.QPushButton('Cancel')
|
||
|
|
self.cancel.clicked.connect(self.close_plugin)
|
||
|
|
self.cancel.setMaximumWidth(button_width)
|
||
|
|
self.cancel.setFocusPolicy(QtCore.Qt.NoFocus)
|
||
|
|
|
||
|
|
self.layout = QtWidgets.QHBoxLayout(self)
|
||
|
|
self.layout.addStretch()
|
||
|
|
self.layout.addWidget(self.cancel)
|
||
|
|
self.layout.addWidget(self.ok)
|
||
|
|
|
||
|
|
def update_original_image(self):
|
||
|
|
image = self.plugin.image_viewer.image
|
||
|
|
self.plugin.image_viewer.original_image = image
|
||
|
|
self.plugin.close()
|
||
|
|
|
||
|
|
def close_plugin(self):
|
||
|
|
# Image viewer will restore original image on close.
|
||
|
|
self.plugin.close()
|
||
|
|
|
||
|
|
|
||
|
|
class SaveButtons(BaseWidget):
|
||
|
|
"""Buttons to save image to io.stack or to a file."""
|
||
|
|
|
||
|
|
def __init__(self, name='Save to:', default_format='png'):
|
||
|
|
super(SaveButtons, self).__init__(name)
|
||
|
|
|
||
|
|
self.default_format = default_format
|
||
|
|
|
||
|
|
self.name_label = QtWidgets.QLabel()
|
||
|
|
self.name_label.setText(name)
|
||
|
|
|
||
|
|
self.save_file = QtWidgets.QPushButton('File')
|
||
|
|
self.save_file.clicked.connect(self.save_to_file)
|
||
|
|
self.save_file.setFocusPolicy(QtCore.Qt.NoFocus)
|
||
|
|
self.save_stack = QtWidgets.QPushButton('Stack')
|
||
|
|
self.save_stack.clicked.connect(self.save_to_stack)
|
||
|
|
self.save_stack.setFocusPolicy(QtCore.Qt.NoFocus)
|
||
|
|
|
||
|
|
self.layout = QtWidgets.QHBoxLayout(self)
|
||
|
|
self.layout.addWidget(self.name_label)
|
||
|
|
self.layout.addWidget(self.save_stack)
|
||
|
|
self.layout.addWidget(self.save_file)
|
||
|
|
|
||
|
|
def save_to_stack(self):
|
||
|
|
image = self.plugin.filtered_image.copy()
|
||
|
|
io.push(image)
|
||
|
|
|
||
|
|
msg = dedent('''\
|
||
|
|
The image has been pushed to the io stack.
|
||
|
|
Use io.pop() to retrieve the most recently pushed image.
|
||
|
|
NOTE: The io stack only works in interactive sessions.''')
|
||
|
|
notify(msg)
|
||
|
|
|
||
|
|
def save_to_file(self, filename=None):
|
||
|
|
if not filename:
|
||
|
|
filename = dialogs.save_file_dialog()
|
||
|
|
if not filename:
|
||
|
|
return
|
||
|
|
image = self.plugin.filtered_image
|
||
|
|
if image.dtype == np.bool:
|
||
|
|
# TODO: This check/conversion should probably be in `imsave`.
|
||
|
|
image = img_as_ubyte(image)
|
||
|
|
io.imsave(filename, image)
|
||
|
|
|
||
|
|
|
||
|
|
def notify(msg):
|
||
|
|
msglabel = QtWidgets.QLabel(msg)
|
||
|
|
dialog = QtWidgets.QDialog()
|
||
|
|
ok = QtWidgets.QPushButton('OK', dialog)
|
||
|
|
ok.clicked.connect(dialog.accept)
|
||
|
|
ok.setDefault(True)
|
||
|
|
dialog.layout = QtWidgets.QGridLayout(dialog)
|
||
|
|
dialog.layout.addWidget(msglabel, 0, 0, 1, 3)
|
||
|
|
dialog.layout.addWidget(ok, 1, 1)
|
||
|
|
dialog.exec_()
|