wxPython自製Control : 用Matplotlib畫圖表的FigurePanel

import wx
from wx import xrc
import matplotlib
matplotlib.use('WXAgg')
from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg
from matplotlib.backends.backend_wx import NavigationToolbar2Wx
from matplotlib.figure import Figure
 
__all__ = ["HAVE_TOOL_BAR", "FigurePanel", "FigurePanelXmlHandler"]
 
# this flag indicate that there is toolbar in figure panel
HAVE_TOOL_BAR = 1
 
class FigurePanel(wx.PyPanel):
    def __init__(self, figSize=None, figDpi=None, *args, **kwargs):
        wx.Panel.__init__(self, *args, **kwargs)
        self.figSize = figSize
        self.figDpi = figDpi
        self.SetBackgroundStyle(wx.BG_STYLE_CUSTOM)
        self.createFigure(figSize, figDpi)
 
    def createFigure(self, size, dpi):
        """Create figure canvas
 
        """
        self.fig = Figure(size, dpi)
        self.canvas = FigureCanvasWxAgg(self, wx.ID_ANY, self.fig)
        self.toolbar = None
        # 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)
 
        # add  tool bar
        if self.GetWindowStyle() & HAVE_TOOL_BAR:
            self.toolbar = self.createToolbar()
            # On Windows, default frame size behaviour is incorrect
            # you don't need this under Linux
            tw, th = self.toolbar.GetSizeTuple()
            fw, fh = self.canvas.GetSizeTuple()
            self.toolbar.SetSize(wx.Size(fw, th))
            # Best to allow the toolbar to resize!
            sizer.Add(self.toolbar, 0, wx.GROW)
 
        self.SetSizer(sizer)
        self.Fit()
 
    def createToolbar(self):
        """Create toolbar for controlling figure panel, override this method
        to provide custom toolbar
 
        """
        toolbar = NavigationToolbar2Wx(self.canvas)
        toolbar.Realize()
        return toolbar
 
    def GetToolBar(self):
        # You will need to override GetToolBar if you are using an
        # unmanaged toolbar in your frame
        return self.toolbar
 
    def onEraseBackground(self, evt):
        # this is supposed to prevent redraw flicker on some X servers...
        pass
 
    def getFigure(self):
        """Get figure of this panel
 
        """
        return self.fig
 
class FigurePanelXmlHandler(xrc.XmlResourceHandler):
    def __init__(self):
        xrc.XmlResourceHandler.__init__(self)
        # Specify the styles recognized by objects of this type
        self.AddStyle("wxHAVE_TOOL_BAR", HAVE_TOOL_BAR)
        self.AddWindowStyles()
 
    # This method and the next one are required for XmlResourceHandlers
    def CanHandle(self, node):
        return self.IsOfClass(node, "FigurePanel")
 
    def DoCreateResource(self):
        assert self.GetInstance() is None
 
        figSize = self.GetParamValue("figsize")
        if not len(figSize):
            figSize = None
        else:
            figSize = map(int, figSize.split(','))
 
        figDpi =  self.GetParamValue("figdpi")
        if not len(figDpi):
            figDpi = None
        else:
            figDpi = int(figDpi)
 
        # Now create the object
        panel = FigurePanel(
            figSize,
            figDpi,
            self.GetParentAsWindow(),
            self.GetID(),
            self.GetPosition(),
            self.GetSize(),
            self.GetStyle("style", 0),
            self.GetName()
        )
        return panel
 
if __name__ == '__main__':
    app = wx.PySimpleApp()
 
    class Frame(wx.Frame):
        def __init__(self, *args, **kwargs):
            wx.Frame.__init__(self, *args, **kwargs)
 
            self.figPanel = FigurePanel(parent=self, figSize=(5, 5), figDpi=75)
            self.draw()
 
            sizer = wx.BoxSizer()
            sizer.Add(self.figPanel, 1, wx.GROW)
            self.SetSizer(sizer)
            self.Fit()
 
        def draw(self):
            fig = self.figPanel.getFigure()
            plot = fig.add_subplot(111)
            plot.plot([1,2,3])
            plot.set_ylabel('some numbers')
 
    frame = Frame(parent=None, title='Figure panel demo')
    frame.Show()
    app.MainLoop()

from figure_panel import *
 
def getHandlers():
    """Get xrc handlers
 
    """
    from wx import xrc
    for key, value in globals().iteritems():
        try:
            if issubclass(value, xrc.XmlResourceHandler):
                yield value
        except TypeError:
            pass
 
def installHandlers(res):
    """Insert xml handlers into xml resource object
 
    @param res: resources object to insert
    """
    for handler in getHandlers():
        res.InsertHandler(handler())

<object class="wxPanel" name="m_panel2">
	<style>wxTAB_TRAVERSAL</style>
	<object class="wxBoxSizer">
		<orient>wxVERTICAL</orient>
		<object class="sizeritem">
			<option>1</option>
			<flag>wxALL|wxEXPAND</flag>
			<border>5</border>
			<object class="FigurePanel" name="figurePanel">
				<style>wxHAVE_TOOL_BAR</style>
			</object>
		</object>
	</object>
</object>