嵌入Wx3

版权所有（C）2003-2004 Andrew Straw和Jeremy O’Donoghue等人

许可证：此作品根据PSF许可。该文档也应该在 https://docs.python.org/3/license.html 上提供

这是使用matplotlib和wx的另一个例子。 希望这是功能齐全的：

  • matplotlib工具栏和WX按钮
  • 完整的wxApp框架，包括小部件交互
  • XRC（XML wxWidgets资源）文件创建GUI（用XRCed制作）

这是从embedding_in_wx和dynamic_image_wxagg派生的。

感谢matplotlib和wx团队创建这样出色的软件！

  1. import matplotlib
  2. import matplotlib.cm as cm
  3. import matplotlib.cbook as cbook
  4. from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as FigureCanvas
  5. from matplotlib.backends.backend_wxagg import NavigationToolbar2WxAgg as NavigationToolbar
  6. from matplotlib.figure import Figure
  7. import numpy as np
  9. import wx
  10. import wx.xrc as xrc
  12. ERR_TOL = 1e-5  # floating point slop for peak-detection
  15. matplotlib.rc('image', origin='lower')
  18. class PlotPanel(wx.Panel):
  19.     def __init__(self, parent):
  20.         wx.Panel.__init__(self, parent, -1)
  22.         self.fig = Figure((5, 4), 75)
  23.         self.canvas = FigureCanvas(self, -1, self.fig)
  24.         self.toolbar = NavigationToolbar(self.canvas)  # matplotlib toolbar
  25.         self.toolbar.Realize()
  26.         # self.toolbar.set_active([0,1])
  28.         # Now put all into a sizer
  29.         sizer = wx.BoxSizer(wx.VERTICAL)
  30.         # This way of adding to sizer allows resizing
  31.         sizer.Add(self.canvas, 1, wx.LEFT | wx.TOP | wx.GROW)
  32.         # Best to allow the toolbar to resize!
  33.         sizer.Add(self.toolbar, 0, wx.GROW)
  34.         self.SetSizer(sizer)
  35.         self.Fit()
  37.     def init_plot_data(self):
  38.         a = self.fig.add_subplot(111)
  40.         x = np.arange(120.0) * 2 * np.pi / 60.0
  41.         y = np.arange(100.0) * 2 * np.pi / 50.0
  42.         self.x, self.y = np.meshgrid(x, y)
  43.         z = np.sin(self.x) + np.cos(self.y)
  44.         self.im = a.imshow(z, cmap=cm.RdBu)  # , interpolation='nearest')
  46.         zmax = np.max(z) - ERR_TOL
  47.         ymax_i, xmax_i = np.nonzero(z >= zmax)
  48.         if self.im.origin == 'upper':
  49.             ymax_i = z.shape[0] - ymax_i
  50.         self.lines = a.plot(xmax_i, ymax_i, 'ko')
  52.         self.toolbar.update()  # Not sure why this is needed - ADS
  54.     def GetToolBar(self):
  55.         # You will need to override GetToolBar if you are using an
  56.         # unmanaged toolbar in your frame
  57.         return self.toolbar
  59.     def OnWhiz(self, evt):
  60.         self.x += np.pi / 15
  61.         self.y += np.pi / 20
  62.         z = np.sin(self.x) + np.cos(self.y)
  63.         self.im.set_array(z)
  65.         zmax = np.max(z) - ERR_TOL
  66.         ymax_i, xmax_i = np.nonzero(z >= zmax)
  67.         if self.im.origin == 'upper':
  68.             ymax_i = z.shape[0] - ymax_i
  69.         self.lines[0].set_data(xmax_i, ymax_i)
  71.         self.canvas.draw()
  74. class MyApp(wx.App):
  75.     def OnInit(self):
  76.         xrcfile = cbook.get_sample_data('embedding_in_wx3.xrc',
  77.                                         asfileobj=False)
  78.         print('loading', xrcfile)
  80.         self.res = xrc.XmlResource(xrcfile)
  82.         # main frame and panel ---------
  84.         self.frame = self.res.LoadFrame(None, "MainFrame")
  85.         self.panel = xrc.XRCCTRL(self.frame, "MainPanel")
  87.         # matplotlib panel -------------
  89.         # container for matplotlib panel (I like to make a container
  90.         # panel for our panel so I know where it'll go when in XRCed.)
  91.         plot_container = xrc.XRCCTRL(self.frame, "plot_container_panel")
  92.         sizer = wx.BoxSizer(wx.VERTICAL)
  94.         # matplotlib panel itself
  95.         self.plotpanel = PlotPanel(plot_container)
  96.         self.plotpanel.init_plot_data()
  98.         # wx boilerplate
  99.         sizer.Add(self.plotpanel, 1, wx.EXPAND)
  100.         plot_container.SetSizer(sizer)
  102.         # whiz button ------------------
  103.         whiz_button = xrc.XRCCTRL(self.frame, "whiz_button")
  104.         whiz_button.Bind(wx.EVT_BUTTON, self.plotpanel.OnWhiz)
  106.         # bang button ------------------
  107.         bang_button = xrc.XRCCTRL(self.frame, "bang_button")
  108.         bang_button.Bind(wx.EVT_BUTTON, self.OnBang)
  110.         # final setup ------------------
  111.         sizer = self.panel.GetSizer()
  112.         self.frame.Show(1)
  114.         self.SetTopWindow(self.frame)
  116.         return True
  118.     def OnBang(self, event):
  119.         bang_count = xrc.XRCCTRL(self.frame, "bang_count")
  120.         bangs = bang_count.GetValue()
  121.         bangs = int(bangs) + 1
  122.         bang_count.SetValue(str(bangs))
  124. if __name__ == '__main__':
  125.     app = MyApp(0)
  126.     app.MainLoop()

下载这个示例