Added the sample code from wxPython In Action to the samples dir

[wxWidgets.git] / wxPython / samples / wxPIA_book / Chapter-06 / example1.py

diff --git a/wxPython/samples/wxPIA_book/Chapter-06/example1.py b/wxPython/samples/wxPIA_book/Chapter-06/example1.py
new file mode 100644 (file)
index 0000000..9ff1165
--- /dev/null
+++ b/wxPython/samples/wxPIA_book/Chapter-06/example1.py
@@ -0,0 +1,106 @@
+import wx
+
+
+class SketchWindow(wx.Window):
+    def __init__(self, parent, ID):
+        wx.Window.__init__(self, parent, ID)
+        self.SetBackgroundColour("White")
+        self.color = "Black"
+        self.thickness = 1
+        self.pen = wx.Pen(self.color, self.thickness, wx.SOLID)
+        self.lines = []
+        self.curLine = []
+        self.pos = (0, 0)
+        self.InitBuffer()
+
+        self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
+        self.Bind(wx.EVT_LEFT_UP, self.OnLeftUp)
+        self.Bind(wx.EVT_MOTION, self.OnMotion)
+        self.Bind(wx.EVT_SIZE, self.OnSize)
+        self.Bind(wx.EVT_IDLE, self.OnIdle)
+        self.Bind(wx.EVT_PAINT, self.OnPaint)
+
+    def InitBuffer(self):
+        size = self.GetClientSize()
+        #self.buffer = wx.EmptyBitmap(max(1, size.width), max(1, size.height))
+        self.buffer = wx.EmptyBitmap(size.width, size.height)
+        if self.buffer.Ok():
+            dc = wx.BufferedDC(None, self.buffer)
+            dc.SetBackground(wx.Brush(self.GetBackgroundColour()))
+            dc.Clear()
+            self.DrawLines(dc)
+            self.reInitBuffer = False
+
+    def GetLinesData(self):
+        return self.lines[:]
+
+    def SetLinesData(self, lines):
+        self.lines = lines[:]
+        self.InitBuffer()
+        self.Refresh()
+
+    def OnLeftDown(self, event):
+        self.curLine = []
+        self.pos = event.GetPositionTuple()
+        self.CaptureMouse()
+
+    def OnLeftUp(self, event):
+        if self.HasCapture():
+            self.lines.append((self.color,
+                               self.thickness,
+                               self.curLine))
+            self.curLine = []
+            self.ReleaseMouse()
+
+    def OnMotion(self, event):
+        if event.Dragging() and event.LeftIsDown():
+            dc = wx.BufferedDC(wx.ClientDC(self), self.buffer)
+            self.drawMotion(dc, event)
+        event.Skip()
+
+    def drawMotion(self, dc, event):
+        dc.SetPen(self.pen)
+        newPos = event.GetPositionTuple()
+        coords = self.pos + newPos
+        self.curLine.append(coords)
+        dc.DrawLine(*coords)
+        self.pos = newPos
+
+    def OnSize(self, event):
+        self.reInitBuffer = True
+
+    def OnIdle(self, event):
+        if self.reInitBuffer:
+            self.InitBuffer()
+            self.Refresh(False)
+
+    def OnPaint(self, event):
+        dc = wx.BufferedPaintDC(self, self.buffer)
+
+    def DrawLines(self, dc):
+        for colour, thickness, line in self.lines:
+            pen = wx.Pen(colour, thickness, wx.SOLID)
+            dc.SetPen(pen)
+            for coords in line:
+                dc.DrawLine(*coords)
+
+    def SetColor(self, color):
+        self.color = color
+        self.pen = wx.Pen(self.color, self.thickness, wx.SOLID)
+
+    def SetThickness(self, num):
+        self.thickness = num
+        self.pen = wx.Pen(self.color, self.thickness, wx.SOLID)
+
+
+class SketchFrame(wx.Frame):
+    def __init__(self, parent):
+        wx.Frame.__init__(self, parent, -1, "Sketch Frame",
+                size=(800,600))
+        self.sketch = SketchWindow(self, -1)
+
+if __name__ == '__main__':
+    app = wx.PySimpleApp()
+    frame = SketchFrame(None)
+    frame.Show(True)
+    app.MainLoop()