Matplotlib: как обновить figure.canvas

Я не могу понять, как обновить экземпляр FigureCanvasWxAgg. Вот пример:


import wx
import matplotlib
from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as FigureCanvas
from matplotlib.figure import Figure

class MainFrame(wx.Frame): 
    def __init__(self): 
        wx.Frame.__init__(self, None, wx.NewId(), "Main") 
        self.sizer = wx.BoxSizer(wx.VERTICAL)

        self.figure = Figure(figsize=(1,2))
        self.axe = self.figure.add_subplot(111)
        self.figurecanvas = FigureCanvas(self, -1, self.figure)

        self.buttonPlot = wx.Button(self, wx.NewId(), "Plot")
        self.buttonClear = wx.Button(self, wx.NewId(), "Clear")

        self.sizer.Add(self.figurecanvas, proportion=1, border=5, flag=wx.ALL | wx.EXPAND)
        self.sizer.Add(self.buttonPlot, proportion=0, border=2, flag=wx.ALL)
        self.sizer.Add(self.buttonClear, proportion=0, border=2, flag=wx.ALL)
        self.SetSizer(self.sizer)

        self.figurecanvas.Bind(wx.EVT_LEFT_DCLICK, self.on_dclick)
        self.buttonPlot.Bind(wx.EVT_BUTTON, self.on_button_plot)
        self.buttonClear.Bind(wx.EVT_BUTTON, self.on_button_clear)

        self.subframe_opened = False

    def on_dclick(self, evt):
        self.subframe = SubFrame(self, self.figure)
        self.subframe.Show(True)
        self.subframe_opened = True

    def on_button_plot(self, evt):
        self.axe.plot(range(10), color='green')
        self.figurecanvas.draw()

    def on_button_clear(self, evt):
        if self.subframe_opened:
            self.subframe.Close()
        self.figure.set_canvas(self.figurecanvas)
        self.axe.clear()
        self.figurecanvas.draw()


class SubFrame(wx.Frame): 
    def __init__(self, parent, figure): 
        wx.Frame.__init__(self, parent, wx.NewId(), "Sub")
        self.sizer = wx.BoxSizer(wx.VERTICAL)
        self.figurecanvas = FigureCanvas(self, -1, figure)
        self.sizer.Add(self.figurecanvas, proportion=1, border=5, flag=wx.ALL | wx.EXPAND)
        self.SetSizer(self.sizer)

        self.Bind(wx.EVT_CLOSE, self.on_close)

    def on_close(self, evt):
        self.GetParent().subframe_opened = False
        evt.Skip()



class MyApp(wx.App):
    def OnInit(self):
        frame = MainFrame()
        frame.Show(True)
        self.SetTopWindow(frame)
        return True

app = MyApp(0)
app.MainLoop()    

Меня интересует следующая последовательность операций:

  • запустите script
  • изменить размер основного фрейма
  • нажмите кнопку "Печать"
  • двойной щелчок по графику
  • нажмите кнопку очистки

Теперь я получаю беспорядок на графике основного кадра. Если я изменил размер кадра, он будет правильно перерисовываться. Мой вопрос заключается в том, что я должен добавить в свой код, чтобы сделать это без изменения размера?

Под "беспорядок" я имею в виду что-то вроде этого:

http://img227.imageshack.us/img227/5407/mess.png

Спасибо заранее.

Ответ 1

Как я уже сказал в комментариях, я не думаю, что эта проблема связана с обновлением canvas фигуры, на самом деле я думаю, что она делает именно то, что она должна (перерисовывая себя на основе последнего состояния [т.е. как это было в вашем подзаговор]). Я думаю, ваша проблема в том, что wxFrame не освежает.

Самый простой способ исправить это - это изменить его размер на событие "Очистить". Что-то вроде:

def on_button_clear(self, evt):
    if self.subframe_opened:
        self.subframe.Close()
    self.figure.set_canvas(self.figurecanvas)
    self.axe.clear()
    self.figurecanvas.draw()  
    self.SetSize((self.Size[0],self.figurecanvas.Size[1]))

Установленный размер заставит кадр и все элементы управления, которые он содержит, перерисовать.

Тем не менее, я думаю, что разделять свою фигуру между двумя фигурами - опасно. Я смог произвести некоторые серьезные ошибки, щелкнув/изменив размер в разных комбинациях. Иногда вам приходилось собирать мусор, когда субкадр был закрыт, что делает его недоступным для вашего основного кадра.