mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-05 19:58:52 +08:00
Tony S Yu
b14514e018
nose and autodoc imports the viewer modules so all PyQt4 imports must be wrapped in a try-except block. In addition, any classes derived from PyQt4 must be proxied since the class definition are run on import. This is really hacky.
100 lines
3.1 KiB
Python
100 lines
3.1 KiB
Python
import os
|
|
from textwrap import dedent
|
|
|
|
try:
|
|
from PyQt4 import QtGui
|
|
except ImportError:
|
|
print("Could not import PyQt4 -- skimage.viewer not available.")
|
|
|
|
from skimage import io
|
|
from .core import BaseWidget
|
|
|
|
|
|
__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 = QtGui.QPushButton('OK')
|
|
self.ok.clicked.connect(self.update_original_image)
|
|
self.ok.setMaximumWidth(button_width)
|
|
self.cancel = QtGui.QPushButton('Cancel')
|
|
self.cancel.clicked.connect(self.close_plugin)
|
|
self.cancel.setMaximumWidth(button_width)
|
|
|
|
self.layout = QtGui.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 = QtGui.QLabel()
|
|
self.name_label.setText(name)
|
|
|
|
self.save_file = QtGui.QPushButton('File')
|
|
self.save_file.clicked.connect(self.save_to_file)
|
|
self.save_stack = QtGui.QPushButton('Stack')
|
|
self.save_stack.clicked.connect(self.save_to_stack)
|
|
|
|
self.layout = QtGui.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.image_viewer.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 = str(QtGui.QFileDialog.getSaveFileName())
|
|
if len(filename) == 0:
|
|
return
|
|
#TODO: io plugins should assign default image formats
|
|
basename, ext = os.path.splitext(filename)
|
|
if not ext:
|
|
filename = '%s.%s' % (filename, self.default_format)
|
|
io.imsave(filename, self.plugin.image_viewer.image)
|
|
|
|
|
|
def notify(msg):
|
|
msglabel = QtGui.QLabel(msg)
|
|
dialog = QtGui.QDialog()
|
|
ok = QtGui.QPushButton('OK', dialog)
|
|
ok.clicked.connect(dialog.accept)
|
|
ok.setDefault(True)
|
|
dialog.layout = QtGui.QGridLayout(dialog)
|
|
dialog.layout.addWidget(msglabel, 0, 0, 1, 3)
|
|
dialog.layout.addWidget(ok, 1, 1)
|
|
dialog.exec_()
|