#coding=utf-8
'''
Created on 2013年12月13日
@author: hxp
'''
import wx
import SketchWindow
import cPickle
import os
class MenuSketchFrame(wx.Frame):
'''
classdocs
'''
def __init__(self):
'''
Constructor
'''
wx.Frame.__init__(self,None,-1,"menu sketch frame",size=(1000,600))
self.sketch=SketchWindow.SketchWindow(self)
self.sketch.Bind(wx.EVT_MOTION, self.onMove)
self.fileName="a.sketch"
self.wildcard="Sketch files (*.sketch)|*.sketch|All files (*.*)|*.*"
self.initStatuBar()
self.initMenuBar()
self.createPanel()
def initStatuBar(self):
self.statuBar=self.CreateStatusBar()
self.statuBar.SetFieldsCount(3)
self.statuBar.SetStatusWidths([-1,-2,-3])
def onMove(self,evt):
self.statuBar.SetStatusText(str(evt.GetPositionTuple()),0)
self.statuBar.SetStatusText("current lines %s"%str(len(self.sketch.curLine)),1)
self.statuBar.SetStatusText("total lines %s"%str(len(self.sketch.lines)),2)
evt.Skip()
def menuData(self):
return [("&File",(("&New","New file",self.onNewFile),("&Open","open file",self.onOpenFile),("&Save","save file",self.onSaveFile),("","","")
,("&Color",(("&Black","black",self.onColor,wx.ITEM_RADIO),("&Red","red",self.onColor,wx.ITEM_RADIO),("&Green","green",self.onColor,wx.ITEM_RADIO),("&Blue","blue",self.onColor,wx.ITEM_RADIO),("&Other","other",self.onOther))),
("","",""),
("&Quit","quit",self.onCloseWindow)))]
def initMenuBar(self):
menuBar=wx.MenuBar()
for menuData in self.menuData():
label=menuData[0]
menuItem=menuData[1]
menuBar.Append(self.createMenu(menuItem),label)
self.SetMenuBar(menuBar)
def createMenu(self,menuData):
menu=wx.Menu()
for item in menuData:
if len(item)==2:
label=item[0]
submenu=self.createMenu(item[1])
menu.AppendMenu(-1,label,submenu)
else:
self.createMenuItem(menu,*item)
return menu
def createMenuItem(self,menu,label,status,handler,kind=wx.ITEM_NORMAL):
if not label:
menu.AppendSeparator()
else:
menuItem=menu.Append(-1,label,status,kind)
self.Bind(wx.EVT_MENU,handler,menuItem)
def onNewFile(self,evt):
pass
def onOpenFile(self,evt):
dlg=wx.FileDialog(self,"open sketch file...",os.getcwd(),style=wx.OPEN,wildcard=self.wildcard)
if dlg.ShowModal()==wx.ID_OK:
self.fileName=dlg.GetPath()
self.readFile()
self.SetTitle(self.Title+"---"+self.fileName)
dlg.Destroy()
def readFile(self):
if self.fileName:
try:
f=open(self.fileName,"r")
data=cPickle.load(f)
f.close()
self.sketch.setLineData(data)
except:
wx.MessageBox("error","read file error")
def onSaveFile(self,evt):
if self.fileName:
data=self.sketch.getLineData()
f=open(self.fileName,"w")
print data
cPickle.dump(data, f)
f.close()
evt.Skip()
def onOther(self,evt):
dlg=wx.ColourDialog(self)
dlg.GetColourData().SetChooseFull(True)
if dlg.ShowModal()==wx.ID_OK:
self.sketch.setColor(dlg.GetColourData().GetColour())
dlg.Destroy()
evt.Skip()
def onColor(self,evt):
menuBar=self.GetMenuBar()
id=evt.GetId()
item=menuBar.FindItemById(id)
color=item.GetLabel()
self.sketch.setColor(color)
def onCloseWindow(self,evt):
pass
def createPanel(self):
controlPanel=ControlPanel(self,-1,self.sketch)
box=wx.BoxSizer(wx.HORIZONTAL)
box.Add(controlPanel,0,wx.EXPAND)
box.Add(self.sketch,0,wx.EXPAND)
self.SetSizer(box)
class ControlPanel(wx.Panel):
BMP_SIZE=16
BMP_BORDER=16
NUM_COLOR=4
SPACE=4
colorList=["black","yellow","red","green","blue","purple"]
maxThickness=16
def __init__(self,parent,id,sketch):
wx.Panel.__init__(self,parent,id,style=wx.RAISED_BORDER)
self.sketch=sketch
buttonSize=(self.BMP_SIZE+2*self.BMP_BORDER,self.BMP_SIZE+2*self.BMP_BORDER)
colorGrid=self.createColorGrid(parent,buttonSize)
thickGrid=self.createThickGrid(buttonSize)
self.layout(colorGrid,thickGrid)
def createColorGrid(self,parent,buttonSize):
self.colorMap={}
self.colorButton={}
colorGrid=wx.GridSizer(cols=self.NUM_COLOR,hgap=2,vgap=2)
for color in self.colorList:
b=wx.ToggleButton(self,-1,color,size=buttonSize)
self.Bind(wx.EVT_TOGGLEBUTTON, self.onColor, b)
colorGrid.Add(b,0)
self.colorMap[b.GetId()]=color
self.colorButton[color]=b
return colorGrid
def createThickGrid(self,buttonSize):
self.thicknessMap={}
self.thicknessButton={}
thicknessGrid=wx.GridSizer(cols=self.NUM_COLOR,hgap=2,vgap=2)
for thickness in range(1,self.maxThickness+1):
b=wx.ToggleButton(self,-1,str(thickness),size=buttonSize)
self.Bind(wx.EVT_TOGGLEBUTTON, self.onThickness, b)
thicknessGrid.Add(b,0)
self.thicknessMap[b.GetId()]=thickness
self.thicknessMap[thickness]=b
return thicknessGrid
def layout(self,colorGrid,thickGrid):
box=wx.BoxSizer(wx.VERTICAL)
box.Add(colorGrid,0,wx.ALL,self.SPACE)
box.Add(thickGrid,0,wx.ALL,self.SPACE)
self.SetSizer(box)
box.Fit(self)
def onColor(self,evt):
color=self.colorMap[evt.GetId()]
self.sketch.setColor(color)
evt.Skip()
def onThickness(self,evt):
thick=self.thicknessMap[evt.GetId()]
self.sketch.setThick(thick)
evt.Skip()
if __name__=="__main__":
app=wx.PySimpleApp()
frame=MenuSketchFrame()
frame.Show()
app.MainLoop()
