git.saurik.com Git - wxWidgets.git/blame_incremental - wxPython/samples/wxPIA

... / ...

Commit	Line	Data
	1	import wx
	2
	3
	4	class SketchWindow(wx.Window):
	5	def __init__(self, parent, ID):
	6	wx.Window.__init__(self, parent, ID)
	7	self.SetBackgroundColour("White")
	8	self.color = "Black"
	9	self.thickness = 1
	10	self.pen = wx.Pen(self.color, self.thickness, wx.SOLID)
	11	self.lines = []
	12	self.curLine = []
	13	self.pos = (0, 0)
	14	self.InitBuffer()
	15
	16	self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
	17	self.Bind(wx.EVT_LEFT_UP, self.OnLeftUp)
	18	self.Bind(wx.EVT_MOTION, self.OnMotion)
	19	self.Bind(wx.EVT_SIZE, self.OnSize)
	20	self.Bind(wx.EVT_IDLE, self.OnIdle)
	21	self.Bind(wx.EVT_PAINT, self.OnPaint)
	22
	23	def InitBuffer(self):
	24	size = self.GetClientSize()
	25	#self.buffer = wx.EmptyBitmap(max(1, size.width), max(1, size.height))
	26	self.buffer = wx.EmptyBitmap(size.width, size.height)
	27	if self.buffer.Ok():
	28	dc = wx.BufferedDC(None, self.buffer)
	29	dc.SetBackground(wx.Brush(self.GetBackgroundColour()))
	30	dc.Clear()
	31	self.DrawLines(dc)
	32	self.reInitBuffer = False
	33
	34	def GetLinesData(self):
	35	return self.lines[:]
	36
	37	def SetLinesData(self, lines):
	38	self.lines = lines[:]
	39	self.InitBuffer()
	40	self.Refresh()
	41
	42	def OnLeftDown(self, event):
	43	self.curLine = []
	44	self.pos = event.GetPositionTuple()
	45	self.CaptureMouse()
	46
	47	def OnLeftUp(self, event):
	48	if self.HasCapture():
	49	self.lines.append((self.color,
	50	self.thickness,
	51	self.curLine))
	52	self.curLine = []
	53	self.ReleaseMouse()
	54
	55	def OnMotion(self, event):
	56	if event.Dragging() and event.LeftIsDown():
	57	dc = wx.BufferedDC(wx.ClientDC(self), self.buffer)
	58	self.drawMotion(dc, event)
	59	event.Skip()
	60
	61	def drawMotion(self, dc, event):
	62	dc.SetPen(self.pen)
	63	newPos = event.GetPositionTuple()
	64	coords = self.pos + newPos
	65	self.curLine.append(coords)
	66	dc.DrawLine(*coords)
	67	self.pos = newPos
	68
	69	def OnSize(self, event):
	70	self.reInitBuffer = True
	71
	72	def OnIdle(self, event):
	73	if self.reInitBuffer:
	74	self.InitBuffer()
	75	self.Refresh(False)
	76
	77	def OnPaint(self, event):
	78	dc = wx.BufferedPaintDC(self, self.buffer)
	79
	80	def DrawLines(self, dc):
	81	for colour, thickness, line in self.lines:
	82	pen = wx.Pen(colour, thickness, wx.SOLID)
	83	dc.SetPen(pen)
	84	for coords in line:
	85	dc.DrawLine(*coords)
	86
	87	def SetColor(self, color):
	88	self.color = color
	89	self.pen = wx.Pen(self.color, self.thickness, wx.SOLID)
	90
	91	def SetThickness(self, num):
	92	self.thickness = num
	93	self.pen = wx.Pen(self.color, self.thickness, wx.SOLID)
	94
	95
	96	class SketchFrame(wx.Frame):
	97	def __init__(self, parent):
	98	wx.Frame.__init__(self, parent, -1, "Sketch Frame",
	99	size=(800,600))
	100	self.sketch = SketchWindow(self, -1)
	101
	102	if __name__ == '__main__':
	103	app = wx.PySimpleApp()
	104	frame = SketchFrame(None)
	105	frame.Show(True)
	106	app.MainLoop()