wxPython实现的画图板

#coding:GBK
'''
Created on 2012-3-22

@author: cWX53049
'''
import wx
import time
from bsddb.test.test_pickle import cPickle
import os

_formatStr="%Y-%m-%d %H:%M:%S"

class SkethWindow(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)#创建一个pen
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_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(size.width,size.height)
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#处理一个resize事件
def OnIdle(self,event):
if self.reInitBuffer:
self.InitBuffer()
self.Refresh()
def OnPaint(self,event):
wx.BufferedPaintDC(self,self.buffer)
def DrawLines(self,dc):
for colur,thickness,line in self.lines:
pen = wx.Pen(colur,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 SkethFrame(wx.Frame):
def __init__(self,parent):
self.title="绘画窗体"
wx.Frame.__init__(self,parent,-1,self.title,size=(500,500))
self.sketh=SkethWindow(self,-1)
self.InitStatusBar()
self.sketh.Bind(wx.EVT_MOTION, self.showPosition)
self.initMenuBar()
self.wildcard="Sketch file(*.sketch)|All  files(*.*)|*.*"
self.fileName=""

def OnNew(self,event):
pass
def OnOpen(self,event):
pass
def OnSave(self,event):
if not self.fileName:
self.OnSaveAs(event)
else:
self.SaveFile()

def SaveFile(self):
if self.fileName:
data = self.sketh.GetLinesData()
f = open(self.fileName,'w')
cPickle.dump(data, f)
f.close()
def OnSaveAs(self,event):
dlf = wx.FileDialog(self,"Sketch file",os.getcwd(),style=wx.SAVE|wx.OVERWRITE_PROMPT)
if dlf.ShowModal()==wx.ID_OK:
fileName = dlf.GetPath()
if not os.path.splitext(fileName)[1]:
fileName=fileName+'.sketch'
self.fileName=fileName
self.SaveFile()
self.SetTitle(self.title+'--'+self.fileName)
dlf.Destroy()
def OnColor(self,event):
pass
def OnCloseWindow(self,event):
self.Destroy()
def menuData(self):
return [("&File",(
("&New","New Sketch file",self.OnNew),
("&Open","open sketch file",self.OnOpen),
("&Save","save sktech file",self.OnSave),
("","",""),
("&Color",(
("&Black","",self.OnColor,wx.ITEM_RADIO),
("&Red","",self.OnColor,wx.ITEM_RADIO),
("&Red","",self.OnColor,wx.ITEM_RADIO),
("&Blue","",self.OnColor,wx.ITEM_RADIO))),
("","",""),
("&Quit","quit sketch ",self.OnCloseWindow)))]

def initMenuBar(self):
menuBar=wx.MenuBar()
for eachMenu in self.menuData():
eachMenuLabel = eachMenu[0]
eachMenuItems = eachMenu[1]
menuBar.Append(self.createMenu(eachMenuItems),eachMenuLabel)
self.SetMenuBar(menuBar)

def createMenu(self,menuData):
menu = wx.Menu()
for eachMenu in menuData:
if len(eachMenu)==2:
label = eachMenu[0]
subBar =self.createMenu(eachMenu[1])
menu.AppendMenu(wx.NewId(),label,subBar)
else:
self.createMenuItem(menu,*eachMenu)
return menu

def createMenuItem(self,menu,label,status,handler,kind=wx.ITEM_NORMAL):
if not label:
menu.AppendSeparator()
return
menuItem=menu.Append(-1,label,status,kind)
self.Bind(wx.EVT_MENU,handler,menuItem)

def InitStatusBar(self):
self.statusBar=self.CreateStatusBar()
self.statusBar.SetFieldsCount(4)
self.statusBar.SetStatusWidths([-1,-2,-2,-2])
self.timer=wx.Timer(self)
self.Bind(wx.EVT_TIMER, self.updateTime, self.timer)
self.timer.Start(1000)
self.statusBar.SetStatusText("Time is %s"%time.strftime(_formatStr,time.localtime()),3)
self.statusBar.SetStatusText("Pos (0,0)",1)
self.statusBar.SetStatusText("Cur Line 0",2)
self.statusBar.SetStatusText("Lines 0" ,3)

def showPosition(self,event):
self.statusBar.SetStatusText("Pos %s"%str(event.GetPositionTuple()),0)
self.statusBar.SetStatusText("Cur Line %s"%len(self.sketh.curLine),1)
self.statusBar.SetStatusText("Lines %s"%len(self.sketh.lines),2)
event.Skip()

def updateTime(self,event):
self.statusBar.SetStatusText("Time is %s"%time.strftime(_formatStr,time.localtime()),3)

if __name__=="__main__":
app = wx.PySimpleApp(False)
frame=SkethFrame(None)
frame.Center()
frame.Show(True)
app.MainLoop()