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()