TestUpdate.py

#!/usr/bin/env python

import wx
import time

class TestFrame(wx.Frame):
    def __init__(self):
        wx.Frame.__init__(self, None, -1, "DrawLines Test",
                         wx.DefaultPosition,
                         size=(500,500),
                         style=wx.DEFAULT_FRAME_STYLE | wx.NO_FULL_REPAINT_ON_RESIZE)

        ## Set up the MenuBar
        MenuBar = wx.MenuBar()

        file_menu = wx.Menu()

        ID_CLEAR_MENU = wx.NewId()
        file_menu.Append(ID_CLEAR_MENU, "&Clear", "Clear the Screen")
        self.Bind(wx.EVT_MENU, self.Clear, id=ID_CLEAR_MENU)

        ID_ANIMATE_MENU = wx.NewId()
        file_menu.Append(ID_ANIMATE_MENU, "&Animate", "Animate the Screen")
        self.Bind(wx.EVT_MENU, self.Animate, id=ID_ANIMATE_MENU)

        file_menu.Append(wx.ID_EXIT, "E&xit", "Terminate the program")
        self.Bind(wx.EVT_MENU, self.OnQuit, id=wx.ID_EXIT)

        MenuBar.Append(file_menu, "&File")
        self.SetMenuBar(MenuBar)

        self.Bind(wx.EVT_PAINT, self.OnPaint)
        self.Bind(wx.EVT_MOTION, self.OnMouseMove)
        self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)

        self.LineData = []

    def OnPaint(self,event):
        dc = wx.PaintDC(self)
        dc.SetBackground(wx.Brush("Purple"))
        dc.Clear()
        dc.SetPen(wx.Pen("Red", 3))
        for Line in self.LineData:
            dc.DrawLines(Line)

    def Clear(self, event = None):
        self.LineData = []
        self.Refresh()

    def OnLeftDown(self,event):
        xy = event.GetPosition()
        self.LineData.append([xy])

    def OnMouseMove(self, event):
        if event.Dragging() and event.LeftIsDown():
            xy = event.GetPosition()
            self.LineData[-1].append(xy)
            # note: Update() doesn't work
            self.Refresh()

    def Animate(self, event):
        self.Refresh()
        self.LineData.append( [] )
        for i in xrange(0,500,5):
            self.LineData[-1].append((i,i))
            self.Refresh()
            self.Update()
            time.sleep(0.01)

    def OnQuit(self,event):
        self.Close(True)


class DemoApp(wx.App):
    def OnInit(self):
        frame = TestFrame()
        frame.Show(True)
        self.SetTopWindow(frame)

        return True


if __name__ == "__main__":
    app = DemoApp(0)
    app.MainLoop()