mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-23 13:10:18 +08:00
207 lines
6.8 KiB
Python
207 lines
6.8 KiB
Python
from util import prepare_for_display, window_manager, GuiLockError, ColorMixer
|
|
|
|
import numpy as np
|
|
import sys
|
|
|
|
try:
|
|
# We try to aquire the gui lock first or else the gui import might
|
|
# trample another GUI's PyOS_InputHook.
|
|
window_manager.acquire('qt')
|
|
|
|
except GuiLockError, gle:
|
|
print gle
|
|
|
|
else:
|
|
try:
|
|
from PyQt4.QtGui import (QApplication, QMainWindow, QImage, QPixmap,
|
|
QLabel, QWidget, QVBoxLayout, QSlider)
|
|
from PyQt4 import QtCore, QtGui
|
|
|
|
except ImportError:
|
|
print 'PyQT4 libraries not installed. Plugin not loaded.'
|
|
window_manager._release('qt')
|
|
|
|
else:
|
|
|
|
app = None
|
|
|
|
class LabelImage(QLabel):
|
|
def __init__(self, parent, arr):
|
|
QLabel.__init__(self)
|
|
self.parent = parent
|
|
# we need to hold a reference to
|
|
# arr because QImage doesn't copy the data
|
|
# and the buffer must be alive as long
|
|
# as the image is alive.
|
|
self.arr = arr
|
|
|
|
# we also need to pass in the row-stride to
|
|
# the constructor, because we can't guarantee
|
|
# that every row of the numpy data is
|
|
# 4-byte aligned. Which Qt would require
|
|
# if we didnt pass the stride.
|
|
self.img = QImage(arr.data, arr.shape[1], arr.shape[0],
|
|
arr.strides[0], QImage.Format_RGB888)
|
|
|
|
self.pm = QPixmap.fromImage(self.img)
|
|
self.setPixmap(self.pm)
|
|
|
|
def mouseMoveEvent(self, evt):
|
|
self.parent.label_mouseMoveEvent(evt)
|
|
|
|
|
|
class ImageWindow(QMainWindow):
|
|
def __init__(self, arr, mgr):
|
|
QMainWindow.__init__(self)
|
|
self.mgr = mgr
|
|
self.main_widget = QWidget()
|
|
self.layout = QtGui.QHBoxLayout(self.main_widget)
|
|
self.setCentralWidget(self.main_widget)
|
|
|
|
self.label = LabelImage(self, arr)
|
|
self.layout.addWidget(self.label)
|
|
self.mgr.add_window(self)
|
|
self.main_widget.show()
|
|
|
|
def closeEvent(self, event):
|
|
# Allow window to be destroyed by removing any
|
|
# references to it
|
|
self.mgr.remove_window(self)
|
|
|
|
def label_mouseMoveEvent(self, evt):
|
|
pass
|
|
|
|
|
|
class SliderBlock(QWidget):
|
|
def __init__(self, srange, callback):
|
|
QWidget.__init__(self)
|
|
|
|
self.callback = callback
|
|
|
|
low = srange[0]
|
|
high = srange[1]
|
|
init = srange[2]
|
|
|
|
self.rslider = QSlider()
|
|
self.rslider.setRange(low, high)
|
|
self.rslider.setValue(init)
|
|
|
|
self.gslider = QSlider()
|
|
self.gslider.setRange(low, high)
|
|
self.gslider.setValue(init)
|
|
|
|
self.bslider = QSlider()
|
|
self.bslider.setRange(low, high)
|
|
self.bslider.setValue(init)
|
|
|
|
self.rslider.valueChanged.connect(self.rslider_changed)
|
|
self.gslider.valueChanged.connect(self.gslider_changed)
|
|
self.bslider.valueChanged.connect(self.bslider_changed)
|
|
|
|
self.layout = QtGui.QHBoxLayout(self)
|
|
self.layout.addWidget(self.rslider)
|
|
self.layout.addWidget(self.gslider)
|
|
self.layout.addWidget(self.bslider)
|
|
|
|
def rslider_changed(self, val):
|
|
self.callback('RED', val)
|
|
|
|
def gslider_changed(self, val):
|
|
self.callback('GREEN', val)
|
|
|
|
def bslider_changed(self, val):
|
|
self.callback('BLUE', val)
|
|
|
|
|
|
class FancyImageWindow(ImageWindow):
|
|
def __init__(self, arr, mgr):
|
|
ImageWindow.__init__(self, arr, mgr)
|
|
self.arr = arr
|
|
|
|
self.statusBar().showMessage('X: Y: ')
|
|
self.label.setScaledContents(True)
|
|
self.label.setMouseTracking(True)
|
|
|
|
self.mixer = ColorMixer(self.arr)
|
|
|
|
self.sliders = SliderBlock((-255, 255, 0), self.svalueChanged)
|
|
self.msliders = SliderBlock((0, 1000, 500), self.mvalueChanged)
|
|
|
|
self.layout.addWidget(self.sliders)
|
|
self.layout.addWidget(self.msliders)
|
|
|
|
self.sliders.show()
|
|
self.msliders.show()
|
|
|
|
def svalueChanged(self, who, val):
|
|
if who == 'RED':
|
|
self.mixer.add(self.mixer.RED, val)
|
|
elif who == 'GREEN':
|
|
self.mixer.add(self.mixer.GREEN, val)
|
|
elif who == 'BLUE':
|
|
self.mixer.add(self.mixer.BLUE, val)
|
|
else:
|
|
return
|
|
|
|
pm = QPixmap.fromImage(self.label.img)
|
|
self.label.setPixmap(pm)
|
|
|
|
def mvalueChanged(self, who, val):
|
|
val = val / 500.
|
|
if who == 'RED':
|
|
self.mixer.multiply(self.mixer.RED, val)
|
|
elif who == 'GREEN':
|
|
self.mixer.multiply(self.mixer.GREEN, val)
|
|
elif who == 'BLUE':
|
|
self.mixer.multiply(self.mixer.BLUE, val)
|
|
else:
|
|
return
|
|
|
|
pm = QPixmap.fromImage(self.label.img)
|
|
self.label.setPixmap(pm)
|
|
|
|
def scale_mouse_pos(self, x, y):
|
|
width = self.label.width()
|
|
height = self.label.height()
|
|
x_frac = 1. * x / width
|
|
y_frac = 1. * y / height
|
|
width = self.arr.shape[1]
|
|
height = self.arr.shape[0]
|
|
new_x = int(width * x_frac)
|
|
new_y = int(height * y_frac)
|
|
return(new_x, new_y)
|
|
|
|
def label_mouseMoveEvent(self, evt):
|
|
x = evt.x()
|
|
y = evt.y()
|
|
x, y = self.scale_mouse_pos(x, y)
|
|
msg = 'X: %d, Y: %d ' % (x, y)
|
|
R = self.arr[y,x,0]
|
|
G = self.arr[y,x,1]
|
|
B = self.arr[y,x,2]
|
|
msg += 'R: %s, G:, %s, B: %s' % (R, G, B)
|
|
self.statusBar().showMessage(msg)
|
|
|
|
|
|
def imshow(arr, fancy=False):
|
|
global app
|
|
|
|
if not app:
|
|
app = QApplication([])
|
|
|
|
arr = prepare_for_display(arr)
|
|
|
|
if not fancy:
|
|
iw = ImageWindow(arr, window_manager)
|
|
else:
|
|
iw = FancyImageWindow(arr, window_manager)
|
|
|
|
iw.show()
|
|
|
|
def _app_show():
|
|
global app
|
|
if app and window_manager.has_windows():
|
|
app.exec_()
|
|
else:
|
|
print 'No images to show. See `imshow`.'
|