]>
Commit | Line | Data |
---|---|---|
1 | import wx | |
2 | from example1 import SketchWindow | |
3 | ||
4 | ||
5 | class 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 | ||
128 | if __name__ == '__main__': | |
129 | app = wx.PySimpleApp() | |
130 | frame = SketchFrame(None) | |
131 | frame.Show(True) | |
132 | app.MainLoop() |