wxpython动态实时绘图_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()