]> git.saurik.com Git - wxWidgets.git/blame_incremental - wxPython/samples/wxPIA_book/Chapter-06/example5.py
RemoveItem should return the same object it is passed, but adjusted
[wxWidgets.git] / wxPython / samples / wxPIA_book / Chapter-06 / example5.py
... / ...
CommitLineData
1import wx
2from example1 import SketchWindow
3
4
5class SketchFrame(wx.Frame):
6 def __init__(self, parent):
7 wx.Frame.__init__(self, parent, -1, "Sketch Frame",
8 size=(800,600))
9 self.sketch = SketchWindow(self, -1)
10 self.sketch.Bind(wx.EVT_MOTION, self.OnSketchMotion)
11 self.initStatusBar()
12 self.createMenuBar()
13 self.createToolBar()
14
15 def initStatusBar(self):
16 self.statusbar = self.CreateStatusBar()
17 self.statusbar.SetFieldsCount(3)
18 self.statusbar.SetStatusWidths([-1, -2, -3])
19
20 def OnSketchMotion(self, event):
21 self.statusbar.SetStatusText("Pos: %s" %
22 str(event.GetPositionTuple()), 0)
23 self.statusbar.SetStatusText("Current Pts: %s" %
24 len(self.sketch.curLine), 1)
25 self.statusbar.SetStatusText("Line Count: %s" %
26 len(self.sketch.lines), 2)
27 event.Skip()
28
29 def menuData(self):
30 return [("&File", (
31 ("&New", "New Sketch file", self.OnNew),
32 ("&Open", "Open sketch file", self.OnOpen),
33 ("&Save", "Save sketch file", self.OnSave),
34 ("", "", ""),
35 ("&Color", (
36 ("&Black", "", self.OnColor, wx.ITEM_RADIO),
37 ("&Red", "", self.OnColor, wx.ITEM_RADIO),
38 ("&Green", "", self.OnColor, wx.ITEM_RADIO),
39 ("&Blue", "", self.OnColor, wx.ITEM_RADIO))),
40 ("", "", ""),
41 ("&Quit", "Quit", self.OnCloseWindow)))]
42
43 def createMenuBar(self):
44 menuBar = wx.MenuBar()
45 for eachMenuData in self.menuData():
46 menuLabel = eachMenuData[0]
47 menuItems = eachMenuData[1]
48 menuBar.Append(self.createMenu(menuItems), menuLabel)
49 self.SetMenuBar(menuBar)
50
51 def createMenu(self, menuData):
52 menu = wx.Menu()
53 for eachItem in menuData:
54 if len(eachItem) == 2:
55 label = eachItem[0]
56 subMenu = self.createMenu(eachItem[1])
57 menu.AppendMenu(wx.NewId(), label, subMenu)
58 else:
59 self.createMenuItem(menu, *eachItem)
60 return menu
61
62 def createMenuItem(self, menu, label, status, handler, kind=wx.ITEM_NORMAL):
63 if not label:
64 menu.AppendSeparator()
65 return
66 menuItem = menu.Append(-1, label, status, kind)
67 self.Bind(wx.EVT_MENU, handler, menuItem)
68
69 def createToolBar(self):
70 toolbar = self.CreateToolBar()
71 for each in self.toolbarData():
72 self.createSimpleTool(toolbar, *each)
73 toolbar.AddSeparator()
74 for each in self.toolbarColorData():
75 self.createColorTool(toolbar, each)
76 toolbar.Realize()
77
78 def createSimpleTool(self, toolbar, label, filename, help, handler):
79 if not label:
80 toolbar.AddSeparator()
81 return
82 bmp = wx.Image(filename, wx.BITMAP_TYPE_BMP).ConvertToBitmap()
83 tool = toolbar.AddSimpleTool(-1, bmp, label, help)
84 self.Bind(wx.EVT_MENU, handler, tool)
85
86 def toolbarData(self):
87 return (("New", "new.bmp", "Create new sketch", self.OnNew),
88 ("", "", "", ""),
89 ("Open", "open.bmp", "Open existing sketch", self.OnOpen),
90 ("Save", "save.bmp", "Save existing sketch", self.OnSave))
91
92 def createColorTool(self, toolbar, color):
93 bmp = self.MakeBitmap(color)
94 tool = toolbar.AddRadioTool(-1, bmp, shortHelp=color)
95 self.Bind(wx.EVT_MENU, self.OnColor, tool)
96
97 def MakeBitmap(self, color):
98 bmp = wx.EmptyBitmap(16, 15)
99 dc = wx.MemoryDC()
100 dc.SelectObject(bmp)
101 dc.SetBackground(wx.Brush(color))
102 dc.Clear()
103 dc.SelectObject(wx.NullBitmap)
104 return bmp
105
106 def toolbarColorData(self):
107 return ("Black", "Red", "Green", "Blue")
108
109 def OnNew(self, event): pass
110 def OnOpen(self, event): pass
111 def OnSave(self, event): pass
112
113 def OnColor(self, event):
114 menubar = self.GetMenuBar()
115 itemId = event.GetId()
116 item = menubar.FindItemById(itemId)
117 if not item:
118 toolbar = self.GetToolBar()
119 item = toolbar.FindById(itemId)
120 color = item.GetShortHelp()
121 else:
122 color = item.GetLabel()
123 self.sketch.SetColor(color)
124
125 def OnCloseWindow(self, event):
126 self.Destroy()
127
128if __name__ == '__main__':
129 app = wx.PySimpleApp()
130 frame = SketchFrame(None)
131 frame.Show(True)
132 app.MainLoop()