mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-30 20:36:44 +08:00
Stefan van der Walt
54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
import numpy as np
|
|
import plugin
|
|
from util import prepare_for_display
|
|
|
|
try:
|
|
import wx
|
|
except ImportError:
|
|
pass
|
|
else:
|
|
|
|
# idea shamelessly taken from here:
|
|
# http://wiki.wxpython.org/WorkingWithImages
|
|
|
|
class ImagePanel(wx.Panel):
|
|
def __init__(self, parent, id):
|
|
wx.Panel.__init__(self, parent, id)
|
|
self.bitmap = None
|
|
self.Bind(wx.EVT_PAINT, self.OnPaint)
|
|
|
|
def display(self, npy_img):
|
|
self.bitmap = self.get_bitmap(npy_img)
|
|
self.Refresh(True)
|
|
|
|
def OnPaint(self, evt):
|
|
dc = wx.PaintDC(self)
|
|
if self.bitmap:
|
|
dc.DrawBitmap(self.bitmap, 0, 0)
|
|
|
|
def get_bitmap(self, npy_img):
|
|
width = npy_img.shape[1]
|
|
height = npy_img.shape[0]
|
|
wx_img = wx.EmptyImage(width, height)
|
|
wx_img.SetData(npy_img.data)
|
|
return wx.BitmapFromImage(wx_img)
|
|
|
|
|
|
class ImageFrame(wx.Frame):
|
|
def __init__(self, img):
|
|
self.img = img
|
|
width = img.shape[1]
|
|
height = img.shape[0]
|
|
wx.Frame.__init__(self, None, -1, 'wx', wx.DefaultPosition,
|
|
wx.Size(width, height))
|
|
|
|
self.iPanel = ImagePanel(self, -1)
|
|
self.iPanel.display(self.img)
|
|
|
|
def wx_imshow(img):
|
|
f = ImageFrame(prepare_for_display(img))
|
|
f.CenterOnScreen()
|
|
f.Show()
|
|
|
|
plugin.register('wx', show=wx_imshow)
|