TestUpdate3.py

#!/usr/bin/env python

"""
This way uses a wxClientDC

"""


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):
        print("OnPaint Called")
        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)
            dc = wx.ClientDC(self)
            dc.SetPen(wx.Pen("Red", 3))
            x1, y1 = self.LineData[-1][-2]
            x2, y2 =self.LineData[-1][-1]
            dc.DrawLine(x1, y1, x2, y2)

    def Animate(self, event):
        self.Refresh()
        self.Update()
        self.LineData.append( [(0,0)] )
        dc = wx.ClientDC(self)
        dc.SetPen(wx.Pen("Red", 3))
        for i in xrange(10,500,5):
            self.LineData[-1].append((i,i))
            x1, y1 = self.LineData[-1][-2]
            x2, y2 =self.LineData[-1][-1]
            dc.DrawLine(x1, y1, x2, y2)
            #self.Update()
            wx.GetApp().Yield(1)
            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()