+++ /dev/null
-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()