def __init__(self, parent=None, id=wx.ID_ANY, title="Simple wx viewer"):
# Initialize wx window
wx.Frame.__init__(self, parent, id, title, size=(WIDTH, HEIGHT))
# Window dimensions are used to size the splitter proportions.
window_width, window_height = self.GetSizeTuple()
self.splitter = wx.SplitterWindow(self, -1, style=wx.SP_LIVE_UPDATE)
# Figure Panel
self.fig_panel = wx.Panel(self.splitter, -1)
# Initialize image data
self.img = ImageData()
# Initialize matplotlib figure
self.figure = Figure()
self.img_plot = SingleImage(self.figure, self.img.data)
# Initialize Canvas and figure sizer
self.figure.canvas = FigureCanvas(self.fig_panel, -1, self.figure)
self.fig_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.fig_sizer.Add(self.figure.canvas, 1, wx.LEFT | wx.TOP | wx.GROW)
self.figure.canvas.SetMinSize((100, 100))
self.fig_panel.SetSizer(self.fig_sizer)
# Initialize Control Panel for control widgets
self.ctrl_panel = wx.Panel(self.splitter, -1)
self.ctrl_sizer = wx.BoxSizer(wx.VERTICAL)
# Affine TextCtrl
# StaticBox for label
box = wx.StaticBox(self.ctrl_panel, -1, 'Affine Transform:')
bsizer = wx.StaticBoxSizer(box, wx.VERTICAL)
# GridSizer to handle layout of 16 TextCtrl widgets
self.gridsizer = wx.GridSizer(4, 4, 2, 2)
self.affine_txt = []
for elem in range(16):
self.affine_txt.append(wx.TextCtrl(self.ctrl_panel, -1, '0.0',
size=(60, -1),
style=wx.TE_READONLY))
self.gridsizer.AddMany(self.affine_txt)
bsizer.Add(self.gridsizer, 0)
self.ctrl_sizer.Add(bsizer, 0)
# Radio box for selecting orthogonal slice
self.slice_plane = wx.RadioBox(
self.ctrl_panel, -1, "View Slice:", wx.DefaultPosition,
wx.DefaultSize, _slice_planes, 1, wx.RA_SPECIFY_COLS
)
self.Bind(wx.EVT_RADIOBOX, self.EvtSelectSlice, self.slice_plane)
#rb.SetToolTip(wx.ToolTip("This is a ToolTip!"))
#rb.SetLabel("wx.RadioBox")
self.ctrl_sizer.Add(self.slice_plane, 0, wx.ALL, 10)
# Range slider for selecing slice to view
# wx.Slider(parent, id, value, min, max, position, size, style)
self.slider = wx.Slider(self.ctrl_panel, -1, 0, 0, 100, (-1, -1),
(-1, -1), wx.SL_HORIZONTAL |
wx.SL_AUTOTICKS | wx.SL_LABELS)
self.ctrl_sizer.Add(self.slider, 0, wx.ALL, 10)
self.ctrl_panel.SetSizer(self.ctrl_sizer)
self.Bind(wx.EVT_SCROLL, self.EvtSlider, self.slider)
# Add widgets to splitter
self.splitter.SetMinimumPaneSize(10)
self.splitter.SplitVertically(self.fig_panel, self.ctrl_panel,
int(window_width * 0.60))
# Setting up the menu.
filemenu = wx.Menu()
filemenu.Append(ID_OPEN, "&Open", "Open a document")
filemenu.AppendSeparator()
filemenu.Append(ID_EXIT,"E&xit"," Terminate the program")
# Creating the menubar.
menuBar = wx.MenuBar()
menuBar.Append(filemenu,"&File") # Adding the "filemenu" to the MenuBar
self.SetMenuBar(menuBar) # Adding the MenuBar to the Frame content.
# Event handlers for menus
wx.EVT_MENU(self, ID_EXIT, self.OnExit)
wx.EVT_MENU(self, ID_OPEN, self.OnOpen)
self.Centre()
self.Show(True)
class CanvasFrame(wx.Frame):
def __init__(self, parent=None, id=wx.ID_ANY, title="Simple wx viewer"):
# Initialize wx window
wx.Frame.__init__(self, parent, id, title, size=(WIDTH, HEIGHT))
# Window dimensions are used to size the splitter proportions.
window_width, window_height = self.GetSizeTuple()
self.splitter = wx.SplitterWindow(self, -1, style=wx.SP_LIVE_UPDATE)
# Figure Panel
self.fig_panel = wx.Panel(self.splitter, -1)
# Initialize image data
self.img = ImageData()
# Initialize matplotlib figure
self.figure = Figure()
self.img_plot = SingleImage(self.figure, self.img.data)
# Initialize Canvas and figure sizer
self.figure.canvas = FigureCanvas(self.fig_panel, -1, self.figure)
self.fig_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.fig_sizer.Add(self.figure.canvas, 1, wx.LEFT | wx.TOP | wx.GROW)
self.figure.canvas.SetMinSize((100, 100))
self.fig_panel.SetSizer(self.fig_sizer)
# Initialize Control Panel for control widgets
self.ctrl_panel = wx.Panel(self.splitter, -1)
self.ctrl_sizer = wx.BoxSizer(wx.VERTICAL)
# Affine TextCtrl
# StaticBox for label
box = wx.StaticBox(self.ctrl_panel, -1, 'Affine Transform:')
bsizer = wx.StaticBoxSizer(box, wx.VERTICAL)
# GridSizer to handle layout of 16 TextCtrl widgets
self.gridsizer = wx.GridSizer(4, 4, 2, 2)
self.affine_txt = []
for elem in range(16):
self.affine_txt.append(wx.TextCtrl(self.ctrl_panel, -1, '0.0',
size=(60, -1),
style=wx.TE_READONLY))
self.gridsizer.AddMany(self.affine_txt)
bsizer.Add(self.gridsizer, 0)
self.ctrl_sizer.Add(bsizer, 0)
# Radio box for selecting orthogonal slice
self.slice_plane = wx.RadioBox(
self.ctrl_panel, -1, "View Slice:", wx.DefaultPosition,
wx.DefaultSize, _slice_planes, 1, wx.RA_SPECIFY_COLS
)
self.Bind(wx.EVT_RADIOBOX, self.EvtSelectSlice, self.slice_plane)
#rb.SetToolTip(wx.ToolTip("This is a ToolTip!"))
#rb.SetLabel("wx.RadioBox")
self.ctrl_sizer.Add(self.slice_plane, 0, wx.ALL, 10)
# Range slider for selecing slice to view
# wx.Slider(parent, id, value, min, max, position, size, style)
self.slider = wx.Slider(self.ctrl_panel, -1, 0, 0, 100, (-1, -1),
(-1, -1), wx.SL_HORIZONTAL |
wx.SL_AUTOTICKS | wx.SL_LABELS)
self.ctrl_sizer.Add(self.slider, 0, wx.ALL, 10)
self.ctrl_panel.SetSizer(self.ctrl_sizer)
self.Bind(wx.EVT_SCROLL, self.EvtSlider, self.slider)
# Add widgets to splitter
self.splitter.SetMinimumPaneSize(10)
self.splitter.SplitVertically(self.fig_panel, self.ctrl_panel,
int(window_width * 0.60))
# Setting up the menu.
filemenu = wx.Menu()
filemenu.Append(ID_OPEN, "&Open", "Open a document")
filemenu.AppendSeparator()
filemenu.Append(ID_EXIT,"E&xit"," Terminate the program")
# Creating the menubar.
menuBar = wx.MenuBar()
menuBar.Append(filemenu,"&File") # Adding the "filemenu" to the MenuBar
self.SetMenuBar(menuBar) # Adding the MenuBar to the Frame content.
# Event handlers for menus
wx.EVT_MENU(self, ID_EXIT, self.OnExit)
wx.EVT_MENU(self, ID_OPEN, self.OnOpen)
self.Centre()
self.Show(True)
def OnPaint(self, event):
self.draw()
def draw(self):
self.img_plot.draw()
def update_slider(self):
rng = self.img.get_range()
self.slider.SetRange(rng[0], rng[1])
def update_image(self):
"""Update data in the matplotlib figure."""
self.img.update_data()
self.img_plot.set_data(self.img.data)
self.draw()
def set_affine(self, affine):
for ind, val in enumerate(affine.flat):
self.affine_txt[ind].ChangeValue(str(val))
#
# Event handlers
#
def OnExit(self, event):
self.Close(True)
def OnOpen(self, event):
self.dirname = ''
dlg = wx.FileDialog(self, "Choose a file", self.dirname, "", "*.*",
wx.OPEN)
if dlg.ShowModal() == wx.ID_OK:
filename = dlg.GetFilename()
dirname = dlg.GetDirectory()
self.img.load_image(os.path.join(dirname, filename))
self.set_affine(self.img.get_affine())
self.update_slider()
dlg.Destroy()
self.update_image()
def EvtSelectSlice(self, evt):
self.img.set_slice_plane(_slice_planes[evt.GetInt()])
self.update_slider()
self.update_image()
def EvtSlider(self, evt):
self.img.set_slice_index(self.slider.GetValue())
self.update_image()
def __init__(self):
super(MainWindow, self).__init__()
# Initialize our nipy image object
self.img = ImageData()
# Initialize our matplotlib figure
self.img_plot = SingleImage(self.figure, self.img.data)
class MainWindow(HasTraits):
"""Main window for the viewer built using Traits."""
# mpl figure
figure = Instance(Figure)
# Range slider for selecing slice to view
slice_index_low = Int(0) # These have to be trait ints or they don't work
slice_index_high = Int(91) # with the dynamic updating of the Range slider.
slice_index = Range(low='slice_index_low',
high='slice_index_high')
# Radio box for selecting orthogonal slice
slice_plane = Enum(_slice_planes)
# Affine TextCtrl
affine = Array(Float, (4,4))
def __init__(self):
super(MainWindow, self).__init__()
# Initialize our nipy image object
self.img = ImageData()
# Initialize our matplotlib figure
self.img_plot = SingleImage(self.figure, self.img.data)
#
# Initializers for Traited attrs
#
def _figure_default(self):
"""Initialize matplotlib figure."""
figure = Figure()
return figure
def _slice_index_default(self):
"""Initialize slice_index attr without triggering the
on_trait_change method.
"""
return 0
#
# Event handlers
#
@on_trait_change('slice_index, slice_plane')
def update_slice_index(self):
self.img.set_slice_index(self.slice_index)
self.update_image_slicing()
self.image_show()
#
# Data Model methods
#
def update_affine(self):
self.affine = self.img.get_affine()
def update_image_slicing(self):
# XXX: BUG: self.slice_index is set by the slider of the
# current slice. When we switch the slice plane, this index
# may be outside the range of the new slice. Need to handle
# this.
if self.slice_plane == 'Axial':
self.img.set_slice_plane(_slice_planes[0])
elif self.slice_plane == 'Sagittal':
self.img.set_slice_plane(_slice_planes[1])
elif self.slice_plane == 'Coronal':
self.img.set_slice_plane(_slice_planes[2])
else:
raise AttributeError('Unknown slice plane')
# update image array
self.img.update_data()
# update figure data
self.img_plot.set_data(self.img.data)
# get range information for slider
low, high = self.img.get_range()
# update range slider
self.slice_index_low = low
self.slice_index_high = high
def image_show(self):
self.img_plot.draw()
#
# View code
#
# Menus
def open_menu(self):
dlg = FileDialog()
dlg.open()
if dlg.return_code == OK:
self.img.load_image(dlg.path)
self.update_affine()
self.update_slice_index()
menu_open_action = Action(name='Open Nifti', action='open_menu')
file_menubar = MenuBar(Menu(menu_open_action,
name='File'))
# Items
fig_item = Item('figure', editor=MPLFigureEditor())
# radio button to pick slice
_slice_opts = {'Axial' : '1:Axial',
'Sagittal' : '2:Sagittal',
'Coronal' : '3:Coronal'}
slice_opt_item = Item(name='slice_plane',
editor=EnumEditor(values=_slice_opts),
style='custom')
affine_item = Item('affine', label='Affine', style='readonly')
# BUG: The rendering with the 'readonly' style creates an ugly wx
# "multi-line" control.
traits_view = View(HSplit(Group(fig_item),
Group(affine_item,
slice_opt_item,
Item('slice_index'))
),
menubar=file_menubar,
width=0.80, height=0.80,
resizable=True)
def __init__(self):
super(MainWindow, self).__init__()
# Initialize our nipy image object
self.img = ImageData()
# Initialize our matplotlib figure
self.img_plot = SingleImage(self.figure, self.img.data)
class MainWindow(HasTraits):
"""Main window for the viewer built using Traits."""
# mpl figure
figure = Instance(Figure)
# Range slider for selecing slice to view
slice_index_low = Int(0) # These have to be trait ints or they don't work
slice_index_high = Int(
91) # with the dynamic updating of the Range slider.
slice_index = Range(low='slice_index_low', high='slice_index_high')
# Radio box for selecting orthogonal slice
slice_plane = Enum(_slice_planes)
# Affine TextCtrl
affine = Array(Float, (4, 4))
def __init__(self):
super(MainWindow, self).__init__()
# Initialize our nipy image object
self.img = ImageData()
# Initialize our matplotlib figure
self.img_plot = SingleImage(self.figure, self.img.data)
#
# Initializers for Traited attrs
#
def _figure_default(self):
"""Initialize matplotlib figure."""
figure = Figure()
return figure
def _slice_index_default(self):
"""Initialize slice_index attr without triggering the
on_trait_change method.
"""
return 0
#
# Event handlers
#
@on_trait_change('slice_index, slice_plane')
def update_slice_index(self):
self.img.set_slice_index(self.slice_index)
self.update_image_slicing()
self.image_show()
#
# Data Model methods
#
def update_affine(self):
self.affine = self.img.get_affine()
def update_image_slicing(self):
# XXX: BUG: self.slice_index is set by the slider of the
# current slice. When we switch the slice plane, this index
# may be outside the range of the new slice. Need to handle
# this.
if self.slice_plane == 'Axial':
self.img.set_slice_plane(_slice_planes[0])
elif self.slice_plane == 'Sagittal':
self.img.set_slice_plane(_slice_planes[1])
elif self.slice_plane == 'Coronal':
self.img.set_slice_plane(_slice_planes[2])
else:
raise AttributeError('Unknown slice plane')
# update image array
self.img.update_data()
# update figure data
self.img_plot.set_data(self.img.data)
# get range information for slider
low, high = self.img.get_range()
# update range slider
self.slice_index_low = low
self.slice_index_high = high
def image_show(self):
self.img_plot.draw()
#
# View code
#
# Menus
def open_menu(self):
dlg = FileDialog()
dlg.open()
if dlg.return_code == OK:
self.img.load_image(dlg.path)
self.update_affine()
self.update_slice_index()
menu_open_action = Action(name='Open Nifti', action='open_menu')
file_menubar = MenuBar(Menu(menu_open_action, name='File'))
# Items
fig_item = Item('figure', editor=MPLFigureEditor())
# radio button to pick slice
_slice_opts = {
'Axial': '1:Axial',
'Sagittal': '2:Sagittal',
'Coronal': '3:Coronal'
}
slice_opt_item = Item(name='slice_plane',
editor=EnumEditor(values=_slice_opts),
style='custom')
affine_item = Item('affine', label='Affine', style='readonly')
# BUG: The rendering with the 'readonly' style creates an ugly wx
# "multi-line" control.
traits_view = View(HSplit(
Group(fig_item), Group(affine_item, slice_opt_item,
Item('slice_index'))),
menubar=file_menubar,
width=0.80,
height=0.80,
resizable=True)
