embedding matplotlib in wxpython

[Soren]

Hi,

I've been trying to embed matplotlib in wxpython. I want to be able to
put a figure (axes) in a wx.Panel and place it somewhere in my GUI.
The GUI should have other panels with buttons etc. that can control
the output on the figure. I've been looking at the examples from the
matplotlib website, but can't seem to get it to work..

Does anyone here have experience in embedding matplotlib in wxpython?

I have attached my code.. it makes two panels.. one with a matplotlib
plot, and one with a button.. but the plotpanel is just a small square
in the corner.... !! ..

Any help is appreciated!

Thanks,
Soren
Code:
-----------------------------------

import wx
import pylab
from matplotlib.nume rix import arange, sin, cos, pi
import matplotlib

matplotlib.use( 'WXAgg')
from matplotlib.back ends.backend_wx agg import FigureCanvasWxA gg
from matplotlib.back ends.backend_wx import NavigationToolb ar2Wx
from matplotlib.figu re import Figure
matplotlib.inte ractive(False)

class App(wx.App):

def OnInit(self):
self.frame = MainFrame("BioX tas - Autoplotter", (50,60),
(700,700))
self.frame.Show ()

return True

class MainFrame(wx.Fr ame):

def __init__(self, title, pos, size):
wx.Frame.__init __(self, None, -1, title, pos, size)

pPanel = PlotPanel(self, -1) # Plot panel

bPanel = ButtonPanel(sel f, 100,500, (200,100)) # button
panel

sizer = wx.BoxSizer(wx. VERTICAL)

sizer.Add(pPane l,0)
sizer.Add(bPane l,0)

self.SetSizer(s izer)

class ButtonPanel(wx. Panel):

def __init__(self, Parent, xPos, yPos, insize):

pos = (xPos, yPos)
wx.Panel.__init __(self, Parent, -1, pos, style =
wx.RAISED_BORDE R, size = insize)

button = wx.Button(self, -1, 'HELLO!!', (10,10), (150,50))

class NoRepaintCanvas (FigureCanvasWx Agg):
"""We subclass FigureCanvasWxA gg, overriding the _onPaint method,
so that
the draw method is only called for the first two paint events.
After that,
the canvas will only be redrawn when it is resized.
"""
def __init__(self, *args, **kwargs):
FigureCanvasWxA gg.__init__(sel f, *args, **kwargs)
self._drawn = 0

def _onPaint(self, evt):
"""
Called when wxPaintEvt is generated
"""
if not self._isRealize d:
self.realize()

if self._drawn < 2:
self.draw(repai nt = False)
self._drawn += 1

self.gui_repain t(drawDC=wx.Pai ntDC(self))
class PlotPanel(wx.Pa nel):

def __init__(self, parent, id = -1, color = None,\
dpi = None, style = wx.NO_FULL_REPA INT_ON_RESIZE,
**kwargs):

wx.Panel.__init __(self, parent, id = id, style = style,
**kwargs)

self.figure = Figure(None, dpi)
self.canvas = NoRepaintCanvas (self, -1, self.figure)
self._resizefla g = True

self.Bind(wx.EV T_IDLE, self._onIdle)
self.Bind(wx.EV T_SIZE, self._onSize)

self._SetSize()

def draw(self):
if not hasattr(self, 'subplot'):
self.subplot = self.figure.add _subplot(111)
theta = arange(0, 45*2*pi, 0.02)
rad = (0.8*theta/(2*pi)+1)
r = rad*(8 + sin(theta*7+rad/1.8))
x = r*cos(theta)
y = r*sin(theta)
#Now draw it
self.subplot.pl ot(x,y, '-r')

def _onSize(self, event):
self._resizefla g = True

def _onIdle(self, evt):
if self._resizefla g:
self._resizefla g = False
self._SetSize()
self.draw()

def _SetSize(self, pixels = None):
"""
This method can be called to force the Plot to be a desired
size, which defaults to
the ClientSize of the panel
"""
if not pixels:
pixels = self.GetClientS ize()
self.canvas.Set Size(pixels)
self.figure.set _figsize_inches (pixels[0]/
self.figure.get _dpi(),
pixels[1]/self.figure.get _dpi())
if __name__ == "__main__":

app = App(0)
app.MainLoop()