1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
'''
Code for raster displays using wxPython.
'''
from __future__ import absolute_import, division, print_function, unicode_literals
import logging
import wx
from spectral.graphics.graphics import SpyWindow
logger = logging.getLogger('spectral')
class RasterWindow(wx.Frame, SpyWindow):
'''
RasterWindow is the primary wxWindows object for displaying SPy
images. The frames also handle left double-click events by
displaying an x-y plot of the spectrum for the associated pixel.
'''
def __init__(self, parent, index, rgb, **kwargs):
if 'title' in kwargs:
title = kwargs['title']
else:
title = 'SPy Image'
# wxFrame.__init__(self, parent, index, "SPy Frame")
# wxScrolledWindow.__init__(self, parent, index, style = wxSUNKEN_BORDER)
img = wx.EmptyImage(rgb.shape[0], rgb.shape[1])
img = wx.EmptyImage(rgb.shape[1], rgb.shape[0])
img.SetData(rgb.tostring())
self.bmp = img.ConvertToBitmap()
self.kwargs = kwargs
wx.Frame.__init__(self, parent, index, title,
wx.DefaultPosition)
self.SetClientSizeWH(self.bmp.GetWidth(), self.bmp.GetHeight())
wx.EVT_PAINT(self, self.on_paint)
wx.EVT_LEFT_DCLICK(self, self.left_double_click)
def on_paint(self, e):
dc = wx.PaintDC(self)
self.paint(dc)
def paint(self, dc):
dc.BeginDrawing()
dc.DrawBitmap(self.bmp, 0, 0)
# dc.Blit(0,0, bmp.GetWidth(), bmp.GetHeight(), mDC, 0, 0)
dc.EndDrawing()
def left_double_click(self, evt):
from spectral import settings
if "data source" in self.kwargs:
logger.info('{}'.format((evt.GetY(), evt.GetX()))),
settings.plotter.plot(self.kwargs["data source"],
[evt.GetY(), evt.GetX()],
source=self.kwargs["data source"])
|
