class FigurePanel(wx.Panel):
"""
Display panel for a matplotlib figure.
"""
def __init__(self, parent, view, **kwargs):
"""
Initialize panel and connect the view.
Extends wx.App.__init__.
"""
self.hook = HookCollection(
init='madgui.widget.figure.init',
capture_mouse=None)
super(FigurePanel, self).__init__(parent, **kwargs)
self.capturing = False
self.view = view
# couple figure to canvas
self.canvas = Canvas(self, -1, view.figure.figure)
view.canvas = self.canvas
# create a toolbar
self.toolbar = Toolbar(self.canvas)
self.hook.init(self)
self.toolbar.Realize()
# put elements into sizer
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(self.canvas, 1, wx.EXPAND)
sizer.Add(self.toolbar, 0 , wx.LEFT | wx.EXPAND)
self.SetSizer(sizer)
# setup mouse capturing
self.hook.capture_mouse.connect(self.on_capture_mouse)
self.toolbar.Bind(wx.EVT_TOOL,
self.on_zoom_or_pan,
id=self.get_zoom_id())
self.toolbar.Bind(wx.EVT_TOOL,
self.on_zoom_or_pan,
id=self.get_pan_id())
# get notified when frame is destroyed
self.Bind(wx.EVT_WINDOW_DESTROY, self.on_destroy)
def on_destroy(self, event):
"""Invoked when C++ window is destroyed."""
self.view.destroy()
def on_zoom_or_pan(self, event):
"""Capture mouse, after Zoom/Pan tools were clicked."""
if event.IsChecked():
self.capturing = True
self.hook.capture_mouse()
self.capturing = False
event.Skip()
def on_capture_mouse(self):
"""Disable Zoom/Pan tools when someone captures the mouse."""
if self.capturing:
return
zoom_id = self.get_zoom_id()
if self.toolbar.GetToolState(zoom_id):
self.toolbar.zoom()
self.toolbar.ToggleTool(zoom_id, False)
pan_id = self.get_pan_id()
if self.toolbar.GetToolState(pan_id):
self.toolbar.pan()
self.toolbar.ToggleTool(pan_id, False)
def get_pan_id(self):
try:
return self.toolbar.wx_ids['Pan']
except AttributeError:
return self.toolbar._NTB2_PAN
def get_zoom_id(self):
try:
return self.toolbar.wx_ids['Zoom']
except AttributeError:
return self.toolbar._NTB2_ZOOM
