Embedding in wx #3#

Copyright (C) 2003-2004 Andrew Straw, Jeremy O'Donoghue and others

This is yet another example of using matplotlib with wx. Hopefully this is pretty full-featured:

• both matplotlib toolbar and WX buttons manipulate plot

• full wxApp framework, including widget interaction

• XRC (XML wxWidgets resource) file to create GUI (made with XRCed)

This was derived from embedding_in_wx and dynamic_image_wxagg.

Thanks to matplotlib and wx teams for creating such great software!

import matplotlib.cm as cm
import matplotlib.cbook as cbook
from matplotlib.backends.backend_wxagg import (
FigureCanvasWxAgg as FigureCanvas,
from matplotlib.figure import Figure
import numpy as np

import wx
import wx.xrc as xrc

ERR_TOL = 1e-5  # floating point slop for peak-detection

class PlotPanel(wx.Panel):
def __init__(self, parent):
super().__init__(parent, -1)

self.fig = Figure((5, 4), 75)
self.canvas = FigureCanvas(self, -1, self.fig)
self.toolbar = NavigationToolbar(self.canvas)  # matplotlib toolbar
self.toolbar.Realize()

# Now put all into a sizer
sizer = wx.BoxSizer(wx.VERTICAL)
# This way of adding to sizer allows resizing
sizer.Add(self.canvas, 1, wx.LEFT | wx.TOP | wx.GROW)
# Best to allow the toolbar to resize!
self.SetSizer(sizer)
self.Fit()

def init_plot_data(self):

x = np.arange(120.0) * 2 * np.pi / 60.0
y = np.arange(100.0) * 2 * np.pi / 50.0
self.x, self.y = np.meshgrid(x, y)
z = np.sin(self.x) + np.cos(self.y)
self.im = ax.imshow(z, cmap=cm.RdBu, origin='lower')

zmax = np.max(z) - ERR_TOL
ymax_i, xmax_i = np.nonzero(z >= zmax)
if self.im.origin == 'upper':
ymax_i = z.shape[0] - ymax_i
self.lines = ax.plot(xmax_i, ymax_i, 'ko')

self.toolbar.update()  # Not sure why this is needed - ADS

def GetToolBar(self):
# You will need to override GetToolBar if you are using an
# unmanaged toolbar in your frame
return self.toolbar

def OnWhiz(self, event):
self.x += np.pi / 15
self.y += np.pi / 20
z = np.sin(self.x) + np.cos(self.y)
self.im.set_array(z)

zmax = np.max(z) - ERR_TOL
ymax_i, xmax_i = np.nonzero(z >= zmax)
if self.im.origin == 'upper':
ymax_i = z.shape[0] - ymax_i
self.lines[0].set_data(xmax_i, ymax_i)

self.canvas.draw()

class MyApp(wx.App):
def OnInit(self):
xrcfile = cbook.get_sample_data('embedding_in_wx3.xrc',
asfileobj=False)

self.res = xrc.XmlResource(xrcfile)

# main frame and panel ---------

self.panel = xrc.XRCCTRL(self.frame, "MainPanel")

# matplotlib panel -------------

# container for matplotlib panel (I like to make a container
# panel for our panel so I know where it'll go when in XRCed.)
plot_container = xrc.XRCCTRL(self.frame, "plot_container_panel")
sizer = wx.BoxSizer(wx.VERTICAL)

# matplotlib panel itself
self.plotpanel = PlotPanel(plot_container)
self.plotpanel.init_plot_data()

# wx boilerplate
plot_container.SetSizer(sizer)

# whiz button ------------------
whiz_button = xrc.XRCCTRL(self.frame, "whiz_button")
whiz_button.Bind(wx.EVT_BUTTON, self.plotpanel.OnWhiz)

# bang button ------------------
bang_button = xrc.XRCCTRL(self.frame, "bang_button")
bang_button.Bind(wx.EVT_BUTTON, self.OnBang)

# final setup ------------------
self.frame.Show()

self.SetTopWindow(self.frame)

return True

def OnBang(self, event):
bang_count = xrc.XRCCTRL(self.frame, "bang_count")
bangs = bang_count.GetValue()
bangs = int(bangs) + 1
bang_count.SetValue(str(bangs))

if __name__ == '__main__':
app = MyApp()
app.MainLoop()


Gallery generated by Sphinx-Gallery