+++ /dev/null
-import wx
-import wx.html
-import cPickle
-import os
-from wx.lib import buttons
-
-from example1 import SketchWindow
-
-class SketchFrame(wx.Frame):
- def __init__(self, parent):
- self.title = "Sketch Frame"
- wx.Frame.__init__(self, parent, -1, self.title,
- size=(800,600))
- self.filename = ""
- self.sketch = SketchWindow(self, -1)
- wx.EVT_MOTION(self.sketch, self.OnSketchMotion)
- self.initStatusBar()
- self.createMenuBar()
- self.createToolBar()
- self.createPanel()
-
- def createPanel(self):
- controlPanel = ControlPanel(self, -1, self.sketch)
- box = wx.BoxSizer(wx.HORIZONTAL)
- box.Add(controlPanel, 0, wx.EXPAND)
- box.Add(self.sketch, 1, wx.EXPAND)
- self.SetSizer(box)
-
- def initStatusBar(self):
- self.statusbar = self.CreateStatusBar()
- self.statusbar.SetFieldsCount(3)
- self.statusbar.SetStatusWidths([-1, -2, -3])
-
- def OnSketchMotion(self, event):
- self.statusbar.SetStatusText("Pos: %s" %
- str(event.GetPositionTuple()), 0)
- self.statusbar.SetStatusText("Current Pts: %s" %
- len(self.sketch.curLine), 1)
- self.statusbar.SetStatusText("Line Count: %s" %
- len(self.sketch.lines), 2)
- event.Skip()
-
- def menuData(self):
- return [("&File", (
- ("&New", "New Sketch file", self.OnNew),
- ("&Open", "Open sketch file", self.OnOpen),
- ("&Save", "Save sketch file", self.OnSave),
- ("", "", ""),
- ("&Color", (
- ("&Black", "", self.OnColor, wx.ITEM_RADIO),
- ("&Red", "", self.OnColor, wx.ITEM_RADIO),
- ("&Green", "", self.OnColor, wx.ITEM_RADIO),
- ("&Blue", "", self.OnColor, wx.ITEM_RADIO),
- ("&Other...", "", self.OnOtherColor, wx.ITEM_RADIO))),
- ("", "", ""),
- ("About...", "Show about window", self.OnAbout),
- ("&Quit", "Quit", self.OnCloseWindow)))]
-
- def createMenuBar(self):
- menuBar = wx.MenuBar()
- for eachMenuData in self.menuData():
- menuLabel = eachMenuData[0]
- menuItems = eachMenuData[1]
- menuBar.Append(self.createMenu(menuItems), menuLabel)
- self.SetMenuBar(menuBar)
-
- def createMenu(self, menuData):
- menu = wx.Menu()
- for eachItem in menuData:
- if len(eachItem) == 2:
- label = eachItem[0]
- subMenu = self.createMenu(eachItem[1])
- menu.AppendMenu(wx.NewId(), label, subMenu)
- else:
- self.createMenuItem(menu, *eachItem)
- 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 createToolBar(self):
- toolbar = self.CreateToolBar()
- for each in self.toolbarData():
- self.createSimpleTool(toolbar, *each)
- toolbar.AddSeparator()
- for each in self.toolbarColorData():
- self.createColorTool(toolbar, each)
- toolbar.Realize()
-
- def createSimpleTool(self, toolbar, label, filename, help, handler):
- if not label:
- toolbar.AddSeparator()
- return
- bmp = wx.Image(filename, wx.BITMAP_TYPE_BMP).ConvertToBitmap()
- tool = toolbar.AddSimpleTool(-1, bmp, label, help)
- self.Bind(wx.EVT_MENU, handler, tool)
-
- def toolbarData(self):
- return (("New", "new.bmp", "Create new sketch", self.OnNew),
- ("", "", "", ""),
- ("Open", "open.bmp", "Open existing sketch", self.OnOpen),
- ("Save", "save.bmp", "Save existing sketch", self.OnSave))
-
- def createColorTool(self, toolbar, color):
- bmp = self.MakeBitmap(color)
- tool = toolbar.AddRadioTool(-1, bmp, shortHelp=color)
- self.Bind(wx.EVT_MENU, self.OnColor, tool)
-
- def MakeBitmap(self, color):
- bmp = wx.EmptyBitmap(16, 15)
- dc = wx.MemoryDC()
- dc.SelectObject(bmp)
- dc.SetBackground(wx.Brush(color))
- dc.Clear()
- dc.SelectObject(wx.NullBitmap)
- return bmp
-
- def toolbarColorData(self):
- return ("Black", "Red", "Green", "Blue")
-
- def OnNew(self, event): pass
-
- def OnColor(self, event):
- menubar = self.GetMenuBar()
- itemId = event.GetId()
- item = menubar.FindItemById(itemId)
- if not item:
- toolbar = self.GetToolBar()
- item = toolbar.FindById(itemId)
- color = item.GetShortHelp()
- else:
- color = item.GetLabel()
- self.sketch.SetColor(color)
-
- def OnCloseWindow(self, event):
- self.Destroy()
-
- def SaveFile(self):
- if self.filename:
- data = self.sketch.GetLinesData()
- f = open(self.filename, 'w')
- cPickle.dump(data, f)
- f.close()
-
- def ReadFile(self):
- if self.filename:
- try:
- f = open(self.filename, 'r')
- data = cPickle.load(f)
- f.close()
- self.sketch.SetLinesData(data)
- except cPickle.UnpicklingError:
- wx.MessageBox("%s is not a sketch file." % self.filename,
- "oops!", style=wx.OK|wx.ICON_EXCLAMATION)
-
- wildcard = "Sketch files (*.sketch)|*.sketch|All files (*.*)|*.*"
-
- def OnOpen(self, event):
- 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 OnSave(self, event):
- if not self.filename:
- self.OnSaveAs(event)
- else:
- self.SaveFile()
-
- def OnSaveAs(self, event):
- dlg = wx.FileDialog(self, "Save sketch as...", os.getcwd(),
- style=wx.SAVE | wx.OVERWRITE_PROMPT,
- wildcard = self.wildcard)
- if dlg.ShowModal() == wx.ID_OK:
- filename = dlg.GetPath()
- if not os.path.splitext(filename)[1]:
- filename = filename + '.sketch'
- self.filename = filename
- self.SaveFile()
- self.SetTitle(self.title + ' -- ' + self.filename)
- dlg.Destroy()
-
- def OnOtherColor(self, event):
- dlg = wx.ColourDialog(self)
- dlg.GetColourData().SetChooseFull(True)
- if dlg.ShowModal() == wx.ID_OK:
- self.sketch.SetColor(dlg.GetColourData().GetColour())
- dlg.Destroy()
-
- def OnAbout(self, event):
- dlg = SketchAbout(self)
- dlg.ShowModal()
- dlg.Destroy()
-
-
-class SketchAbout(wx.Dialog):
- text = '''
-<html>
-<body bgcolor="#ACAA60">
-<center><table bgcolor="#455481" width="100%" cellspacing="0"
-cellpadding="0" border="1">
-<tr>
- <td align="center"><h1>Sketch!</h1></td>
-</tr>
-</table>
-</center>
-<p><b>Sketch</b> is a demonstration program for <b>wxPython In Action</b>
-Chapter 7. It is based on the SuperDoodle demo included with wxPython,
-available at http://www.wxpython.org/
-</p>
-
-<p><b>SuperDoodle</b> and <b>wxPython</b> are brought to you by
-<b>Robin Dunn</b> and <b>Total Control Software</b>, Copyright
-© 1997-2006.</p>
-</body>
-</html>
-'''
-
- def __init__(self, parent):
- wx.Dialog.__init__(self, parent, -1, 'About Sketch',
- size=(440, 400) )
-
- html = wx.html.HtmlWindow(self)
- html.SetPage(self.text)
- button = wx.Button(self, wx.ID_OK, "Okay")
-
- sizer = wx.BoxSizer(wx.VERTICAL)
- sizer.Add(html, 1, wx.EXPAND|wx.ALL, 5)
- sizer.Add(button, 0, wx.ALIGN_CENTER|wx.ALL, 5)
-
- self.SetSizer(sizer)
- self.Layout()
-
-
-
-class ControlPanel(wx.Panel):
-
- BMP_SIZE = 16
- BMP_BORDER = 3
- NUM_COLS = 4
- SPACING = 4
-
- colorList = ('Black', 'Yellow', 'Red', 'Green', 'Blue', 'Purple',
- 'Brown', 'Aquamarine', 'Forest Green', 'Light Blue',
- 'Goldenrod', 'Cyan', 'Orange', 'Navy', 'Dark Grey',
- 'Light Grey')
- 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)
- thicknessGrid = self.createThicknessGrid(buttonSize)
- self.layout(colorGrid, thicknessGrid)
-
- def createColorGrid(self, parent, buttonSize):
- self.colorMap = {}
- self.colorButtons = {}
- colorGrid = wx.GridSizer(cols=self.NUM_COLS, hgap=2, vgap=2)
- for eachColor in self.colorList:
- bmp = parent.MakeBitmap(eachColor)
- b = buttons.GenBitmapToggleButton(self, -1, bmp, size=buttonSize)
- b.SetBezelWidth(1)
- b.SetUseFocusIndicator(False)
- self.Bind(wx.EVT_BUTTON, self.OnSetColour, b)
- colorGrid.Add(b, 0)
- self.colorMap[b.GetId()] = eachColor
- self.colorButtons[eachColor] = b
- self.colorButtons[self.colorList[0]].SetToggle(True)
- return colorGrid
-
- def createThicknessGrid(self, buttonSize):
- self.thicknessIdMap = {}
- self.thicknessButtons = {}
- thicknessGrid = wx.GridSizer(cols=self.NUM_COLS, hgap=2, vgap=2)
- for x in range(1, self.maxThickness + 1):
- b = buttons.GenToggleButton(self, -1, str(x), size=buttonSize)
- b.SetBezelWidth(1)
- b.SetUseFocusIndicator(False)
- self.Bind(wx.EVT_BUTTON, self.OnSetThickness, b)
- thicknessGrid.Add(b, 0)
- self.thicknessIdMap[b.GetId()] = x
- self.thicknessButtons[x] = b
- self.thicknessButtons[1].SetToggle(True)
- return thicknessGrid
-
- def layout(self, colorGrid, thicknessGrid):
- box = wx.BoxSizer(wx.VERTICAL)
- box.Add(colorGrid, 0, wx.ALL, self.SPACING)
- box.Add(thicknessGrid, 0, wx.ALL, self.SPACING)
- self.SetSizer(box)
- box.Fit(self)
-
- def OnSetColour(self, event):
- color = self.colorMap[event.GetId()]
- if color != self.sketch.color and self.sketch.color in self.colorButtons:
- self.colorButtons[self.sketch.color].SetToggle(False)
- self.sketch.SetColor(color)
-
- def OnSetThickness(self, event):
- thickness = self.thicknessIdMap[event.GetId()]
- if thickness != self.sketch.thickness:
- self.thicknessButtons[self.sketch.thickness].SetToggle(False)
- self.sketch.SetThickness(thickness)
-
-
-class SketchApp(wx.App):
-
- def OnInit(self):
- bmp = wx.Image("splash.png").ConvertToBitmap()
- wx.SplashScreen(bmp, wx.SPLASH_CENTRE_ON_SCREEN | wx.SPLASH_TIMEOUT,
- 1000, None, -1)
- wx.Yield()
-
- frame = SketchFrame(None)
- frame.Show(True)
- self.SetTopWindow(frame)
- return True
-
-if __name__ == '__main__':
- app = SketchApp(False)
- app.MainLoop()
projects / wxWidgets.git / blobdiff