]> git.saurik.com Git - wxWidgets.git/blame - wxPython/wx/tools/XRCed/tree.py
warning fix
[wxWidgets.git] / wxPython / wx / tools / XRCed / tree.py
CommitLineData
d14a1e28
RD
1# Name: tree.py
2# Purpose: XRC editor, XML_tree class
3# Author: Roman Rolinsky <rolinsky@mema.ucl.ac.be>
4# Created: 02.12.2002
5# RCS-ID: $Id$
1fded56b 6
d14a1e28 7from xxx import * # xxx imports globals and params
fdc1a8be 8import types
a4c013b2 9import traceback
d14a1e28
RD
10
11# Constant to define standart window name
12STD_NAME = '_XRCED_T_W'
13
14# Icons
15import images
16
17class MemoryFile:
18 def __init__(self, name):
19 self.name = name
20 self.buffer = ''
21 def write(self, data):
7353d818 22 self.buffer += data.encode(g.currentEncoding)
d14a1e28
RD
23 def close(self):
24 wxMemoryFSHandler_AddFile(self.name, self.buffer)
25
26################################################################################
27
28# Redefine writing to include encoding
29class MyDocument(minidom.Document):
30 def __init__(self):
31 minidom.Document.__init__(self)
32 self.encoding = ''
33 def writexml(self, writer, indent="", addindent="", newl="", encoding=""):
34 if encoding: encdstr = 'encoding="%s"' % encoding
35 else: encdstr = ''
36 writer.write('<?xml version="1.0" %s?>\n' % encdstr)
37 for node in self.childNodes:
38 node.writexml(writer, indent, addindent, newl)
39
40################################################################################
41
42# Ids for menu commands
43class ID_NEW:
44 PANEL = wxNewId()
45 DIALOG = wxNewId()
46 FRAME = wxNewId()
47 TOOL_BAR = wxNewId()
48 TOOL = wxNewId()
49 MENU_BAR = wxNewId()
50 MENU = wxNewId()
51
52 STATIC_TEXT = wxNewId()
53 TEXT_CTRL = wxNewId()
54
55 BUTTON = wxNewId()
56 BITMAP_BUTTON = wxNewId()
57 RADIO_BUTTON = wxNewId()
58 SPIN_BUTTON = wxNewId()
3d49f2fb 59 TOGGLE_BUTTON = wxNewId()
d14a1e28
RD
60
61 STATIC_BOX = wxNewId()
62 CHECK_BOX = wxNewId()
63 RADIO_BOX = wxNewId()
64 COMBO_BOX = wxNewId()
65 LIST_BOX = wxNewId()
66
67 STATIC_LINE = wxNewId()
68 STATIC_BITMAP = wxNewId()
69 CHOICE = wxNewId()
70 SLIDER = wxNewId()
71 GAUGE = wxNewId()
72 SCROLL_BAR = wxNewId()
73 TREE_CTRL = wxNewId()
74 LIST_CTRL = wxNewId()
75 CHECK_LIST = wxNewId()
76 NOTEBOOK = wxNewId()
68ae5821 77 SPLITTER_WINDOW = wxNewId()
d14a1e28
RD
78 SCROLLED_WINDOW = wxNewId()
79 HTML_WINDOW = wxNewId()
80 CALENDAR_CTRL = wxNewId()
81 GENERIC_DIR_CTRL = wxNewId()
82 SPIN_CTRL = wxNewId()
83 UNKNOWN = wxNewId()
84
85 BOX_SIZER = wxNewId()
86 STATIC_BOX_SIZER = wxNewId()
87 GRID_SIZER = wxNewId()
88 FLEX_GRID_SIZER = wxNewId()
a4c013b2 89 GRID_BAG_SIZER = wxNewId()
d14a1e28
RD
90 SPACER = wxNewId()
91 TOOL_BAR = wxNewId()
92 TOOL = wxNewId()
93 MENU = wxNewId()
94 MENU_ITEM = wxNewId()
95 SEPARATOR = wxNewId()
96 LAST = wxNewId()
97
98class PullDownMenu:
99 ID_EXPAND = wxNewId()
100 ID_COLLAPSE = wxNewId()
101 ID_PASTE_SIBLING = wxNewId()
2481bf3c 102 ID_SUBCLASS = wxNewId()
d14a1e28
RD
103
104 def __init__(self, parent):
105 self.ID_DELETE = parent.ID_DELETE
106 EVT_MENU_RANGE(parent, ID_NEW.PANEL, ID_NEW.LAST, parent.OnCreate)
107 EVT_MENU_RANGE(parent, 1000 + ID_NEW.PANEL, 1000 + ID_NEW.LAST, parent.OnReplace)
108 EVT_MENU(parent, self.ID_COLLAPSE, parent.OnCollapse)
109 EVT_MENU(parent, self.ID_EXPAND, parent.OnExpand)
110 EVT_MENU(parent, self.ID_PASTE_SIBLING, parent.OnPaste)
2481bf3c 111 EVT_MENU(parent, self.ID_SUBCLASS, parent.OnSubclass)
d14a1e28
RD
112 # We connect to tree, but process in frame
113 EVT_MENU_HIGHLIGHT_ALL(g.tree, parent.OnPullDownHighlight)
114
115 # Mapping from IDs to element names
116 self.createMap = {
117 ID_NEW.PANEL: 'wxPanel',
118 ID_NEW.DIALOG: 'wxDialog',
119 ID_NEW.FRAME: 'wxFrame',
120 ID_NEW.TOOL_BAR: 'wxToolBar',
121 ID_NEW.TOOL: 'tool',
122 ID_NEW.MENU_BAR: 'wxMenuBar',
123 ID_NEW.MENU: 'wxMenu',
124 ID_NEW.MENU_ITEM: 'wxMenuItem',
125 ID_NEW.SEPARATOR: 'separator',
126
127 ID_NEW.STATIC_TEXT: 'wxStaticText',
128 ID_NEW.TEXT_CTRL: 'wxTextCtrl',
129
130 ID_NEW.BUTTON: 'wxButton',
131 ID_NEW.BITMAP_BUTTON: 'wxBitmapButton',
132 ID_NEW.RADIO_BUTTON: 'wxRadioButton',
133 ID_NEW.SPIN_BUTTON: 'wxSpinButton',
3d49f2fb 134 ID_NEW.TOGGLE_BUTTON: 'wxToggleButton',
d14a1e28
RD
135
136 ID_NEW.STATIC_BOX: 'wxStaticBox',
137 ID_NEW.CHECK_BOX: 'wxCheckBox',
138 ID_NEW.RADIO_BOX: 'wxRadioBox',
139 ID_NEW.COMBO_BOX: 'wxComboBox',
140 ID_NEW.LIST_BOX: 'wxListBox',
141
142 ID_NEW.STATIC_LINE: 'wxStaticLine',
143 ID_NEW.STATIC_BITMAP: 'wxStaticBitmap',
144 ID_NEW.CHOICE: 'wxChoice',
145 ID_NEW.SLIDER: 'wxSlider',
146 ID_NEW.GAUGE: 'wxGauge',
147 ID_NEW.SCROLL_BAR: 'wxScrollBar',
148 ID_NEW.TREE_CTRL: 'wxTreeCtrl',
149 ID_NEW.LIST_CTRL: 'wxListCtrl',
edfeb1b8 150 ID_NEW.CHECK_LIST: 'wxCheckListBox',
d14a1e28 151 ID_NEW.NOTEBOOK: 'wxNotebook',
68ae5821 152 ID_NEW.SPLITTER_WINDOW: 'wxSplitterWindow',
d14a1e28
RD
153 ID_NEW.SCROLLED_WINDOW: 'wxScrolledWindow',
154 ID_NEW.HTML_WINDOW: 'wxHtmlWindow',
155 ID_NEW.CALENDAR_CTRL: 'wxCalendarCtrl',
156 ID_NEW.GENERIC_DIR_CTRL: 'wxGenericDirCtrl',
157 ID_NEW.SPIN_CTRL: 'wxSpinCtrl',
158
159 ID_NEW.BOX_SIZER: 'wxBoxSizer',
160 ID_NEW.STATIC_BOX_SIZER: 'wxStaticBoxSizer',
161 ID_NEW.GRID_SIZER: 'wxGridSizer',
162 ID_NEW.FLEX_GRID_SIZER: 'wxFlexGridSizer',
a4c013b2 163 ID_NEW.GRID_BAG_SIZER: 'wxGridBagSizer',
d14a1e28
RD
164 ID_NEW.SPACER: 'spacer',
165 ID_NEW.UNKNOWN: 'unknown',
166 }
167 self.topLevel = [
168 (ID_NEW.PANEL, 'Panel', 'Create panel'),
169 (ID_NEW.DIALOG, 'Dialog', 'Create dialog'),
170 (ID_NEW.FRAME, 'Frame', 'Create frame'),
171 None,
172 (ID_NEW.TOOL_BAR, 'ToolBar', 'Create toolbar'),
173 (ID_NEW.MENU_BAR, 'MenuBar', 'Create menubar'),
174 (ID_NEW.MENU, 'Menu', 'Create menu')
175 ]
176 self.containers = [
177 (ID_NEW.PANEL, 'Panel', 'Create panel'),
178 (ID_NEW.NOTEBOOK, 'Notebook', 'Create notebook control'),
68ae5821 179 (ID_NEW.SPLITTER_WINDOW, 'SplitterWindow', 'Create splitter window'),
d14a1e28
RD
180 (ID_NEW.TOOL_BAR, 'ToolBar', 'Create toolbar'),
181 ]
182 self.sizers = [
183 (ID_NEW.BOX_SIZER, 'BoxSizer', 'Create box sizer'),
184 (ID_NEW.STATIC_BOX_SIZER, 'StaticBoxSizer',
185 'Create static box sizer'),
186 (ID_NEW.GRID_SIZER, 'GridSizer', 'Create grid sizer'),
187 (ID_NEW.FLEX_GRID_SIZER, 'FlexGridSizer',
188 'Create flexgrid sizer'),
a4c013b2
RR
189 (ID_NEW.GRID_BAG_SIZER, 'GridBagSizer',
190 'Create gridbag sizer'),
d14a1e28
RD
191 (ID_NEW.SPACER, 'Spacer', 'Create spacer'),
192 ]
193 self.controls = [
194 ['control', 'Various controls',
195 (ID_NEW.STATIC_TEXT, 'Label', 'Create label'),
196 (ID_NEW.STATIC_BITMAP, 'Bitmap', 'Create bitmap'),
197 (ID_NEW.STATIC_LINE, 'Line', 'Create line'),
198 (ID_NEW.TEXT_CTRL, 'TextBox', 'Create text box'),
199 (ID_NEW.CHOICE, 'Choice', 'Create choice'),
200 (ID_NEW.SLIDER, 'Slider', 'Create slider'),
201 (ID_NEW.GAUGE, 'Gauge', 'Create gauge'),
202 (ID_NEW.SPIN_CTRL, 'SpinCtrl', 'Create spin'),
203 (ID_NEW.SCROLL_BAR, 'ScrollBar', 'Create scroll bar'),
204 (ID_NEW.TREE_CTRL, 'TreeCtrl', 'Create tree'),
205 (ID_NEW.LIST_CTRL, 'ListCtrl', 'Create list'),
206 (ID_NEW.CHECK_LIST, 'CheckList', 'Create check list'),
207 (ID_NEW.SCROLLED_WINDOW, 'ScrolledWindow', 'Create scrolled window'),
208 (ID_NEW.HTML_WINDOW, 'HtmlWindow', 'Create HTML window'),
209 (ID_NEW.CALENDAR_CTRL, 'CalendarCtrl', 'Create calendar control'),
210 (ID_NEW.GENERIC_DIR_CTRL, 'GenericDirCtrl', 'Create generic dir control'),
211 (ID_NEW.UNKNOWN, 'Unknown', 'Create custom control placeholder'),
212 ],
213 ['button', 'Buttons',
214 (ID_NEW.BUTTON, 'Button', 'Create button'),
215 (ID_NEW.BITMAP_BUTTON, 'BitmapButton', 'Create bitmap button'),
216 (ID_NEW.RADIO_BUTTON, 'RadioButton', 'Create radio button'),
217 (ID_NEW.SPIN_BUTTON, 'SpinButton', 'Create spin button'),
3d49f2fb 218 (ID_NEW.TOGGLE_BUTTON, 'ToggleButton', 'Create toggle button'),
d14a1e28
RD
219 ],
220 ['box', 'Boxes',
221 (ID_NEW.STATIC_BOX, 'StaticBox', 'Create static box'),
222 (ID_NEW.CHECK_BOX, 'CheckBox', 'Create check box'),
223 (ID_NEW.RADIO_BOX, 'RadioBox', 'Create radio box'),
224 (ID_NEW.COMBO_BOX, 'ComboBox', 'Create combo box'),
225 (ID_NEW.LIST_BOX, 'ListBox', 'Create list box'),
226 ],
227 ['container', 'Containers',
228 (ID_NEW.PANEL, 'Panel', 'Create panel'),
229 (ID_NEW.NOTEBOOK, 'Notebook', 'Create notebook control'),
68ae5821 230 (ID_NEW.SPLITTER_WINDOW, 'SplitterWindow', 'Create splitter window'),
d14a1e28
RD
231 (ID_NEW.TOOL_BAR, 'ToolBar', 'Create toolbar'),
232 ],
233 ['sizer', 'Sizers',
234 (ID_NEW.BOX_SIZER, 'BoxSizer', 'Create box sizer'),
235 (ID_NEW.STATIC_BOX_SIZER, 'StaticBoxSizer',
236 'Create static box sizer'),
237 (ID_NEW.GRID_SIZER, 'GridSizer', 'Create grid sizer'),
238 (ID_NEW.FLEX_GRID_SIZER, 'FlexGridSizer',
239 'Create flexgrid sizer'),
a4c013b2
RR
240 (ID_NEW.GRID_BAG_SIZER, 'GridBagSizer',
241 'Create gridbag sizer'),
d14a1e28
RD
242 (ID_NEW.SPACER, 'Spacer', 'Create spacer'),
243 ]
244 ]
245 self.menuControls = [
246 (ID_NEW.MENU, 'Menu', 'Create menu'),
247 (ID_NEW.MENU_ITEM, 'MenuItem', 'Create menu item'),
248 (ID_NEW.SEPARATOR, 'Separator', 'Create separator'),
249 ]
250 self.toolBarControls = [
251 (ID_NEW.TOOL, 'Tool', 'Create tool'),
252 (ID_NEW.SEPARATOR, 'Separator', 'Create separator'),
253 ['control', 'Various controls',
254 (ID_NEW.STATIC_TEXT, 'Label', 'Create label'),
255 (ID_NEW.STATIC_BITMAP, 'Bitmap', 'Create bitmap'),
256 (ID_NEW.STATIC_LINE, 'Line', 'Create line'),
257 (ID_NEW.TEXT_CTRL, 'TextBox', 'Create text box'),
258 (ID_NEW.CHOICE, 'Choice', 'Create choice'),
259 (ID_NEW.SLIDER, 'Slider', 'Create slider'),
260 (ID_NEW.GAUGE, 'Gauge', 'Create gauge'),
261 (ID_NEW.SCROLL_BAR, 'ScrollBar', 'Create scroll bar'),
262 (ID_NEW.LIST_CTRL, 'ListCtrl', 'Create list control'),
263 (ID_NEW.CHECK_LIST, 'CheckList', 'Create check list'),
264 ],
265 ['button', 'Buttons',
266 (ID_NEW.BUTTON, 'Button', 'Create button'),
267 (ID_NEW.BITMAP_BUTTON, 'BitmapButton', 'Create bitmap button'),
268 (ID_NEW.RADIO_BUTTON, 'RadioButton', 'Create radio button'),
269 (ID_NEW.SPIN_BUTTON, 'SpinButton', 'Create spin button'),
270 ],
271 ['box', 'Boxes',
272 (ID_NEW.STATIC_BOX, 'StaticBox', 'Create static box'),
273 (ID_NEW.CHECK_BOX, 'CheckBox', 'Create check box'),
274 (ID_NEW.RADIO_BOX, 'RadioBox', 'Create radio box'),
275 (ID_NEW.COMBO_BOX, 'ComboBox', 'Create combo box'),
276 (ID_NEW.LIST_BOX, 'ListBox', 'Create list box'),
277 ],
278 ]
279
280################################################################################
281
282# Set menu to list items.
283# Each menu command is a tuple (id, label, help)
284# submenus are lists [id, label, help, submenu]
285# and separators are any other type
286def SetMenu(m, list):
287 for l in list:
288 if type(l) == types.TupleType:
289 apply(m.Append, l)
290 elif type(l) == types.ListType:
291 subMenu = wxMenu()
292 SetMenu(subMenu, l[2:])
293 m.AppendMenu(wxNewId(), l[0], subMenu, l[1])
294 else: # separator
295 m.AppendSeparator()
296# Same, but adds 1000 to all IDs
297def SetMenu2(m, list):
298 for l in list:
299 if type(l) == types.TupleType:
300 # Shift ID
301 l = (1000 + l[0],) + l[1:]
302 apply(m.Append, l)
303 elif type(l) == types.ListType:
304 subMenu = wxMenu()
305 SetMenu2(subMenu, l[2:])
306 m.AppendMenu(wxNewId(), l[0], subMenu, l[1])
307 else: # separator
308 m.AppendSeparator()
309
310################################################################################
311
312class HighLightBox:
313 def __init__(self, pos, size):
314 if size.width == -1: size.width = 0
315 if size.height == -1: size.height = 0
316 w = g.testWin.panel
317 l1 = wxWindow(w, -1, pos, wxSize(size.width, 2))
318 l1.SetBackgroundColour(wxRED)
319 l2 = wxWindow(w, -1, pos, wxSize(2, size.height))
320 l2.SetBackgroundColour(wxRED)
321 l3 = wxWindow(w, -1, wxPoint(pos.x + size.width - 2, pos.y), wxSize(2, size.height))
322 l3.SetBackgroundColour(wxRED)
323 l4 = wxWindow(w, -1, wxPoint(pos.x, pos.y + size.height - 2), wxSize(size.width, 2))
324 l4.SetBackgroundColour(wxRED)
325 self.lines = [l1, l2, l3, l4]
326 # Move highlight to a new position
327 def Replace(self, pos, size):
328 if size.width == -1: size.width = 0
329 if size.height == -1: size.height = 0
330 self.lines[0].SetDimensions(pos.x, pos.y, size.width, 2)
331 self.lines[1].SetDimensions(pos.x, pos.y, 2, size.height)
332 self.lines[2].SetDimensions(pos.x + size.width - 2, pos.y, 2, size.height)
333 self.lines[3].SetDimensions(pos.x, pos.y + size.height - 2, size.width, 2)
334 # Remove it
335 def Remove(self):
336 map(wxWindow.Destroy, self.lines)
337 g.testWin.highLight = None
338
339################################################################################
340
341class XML_Tree(wxTreeCtrl):
342 def __init__(self, parent, id):
343 wxTreeCtrl.__init__(self, parent, id, style = wxTR_HAS_BUTTONS)
344 self.SetBackgroundColour(wxColour(224, 248, 224))
345 # Register events
346 EVT_TREE_SEL_CHANGED(self, self.GetId(), self.OnSelChanged)
347 # One works on Linux, another on Windows
348 if wxPlatform == '__WXGTK__':
349 EVT_TREE_ITEM_ACTIVATED(self, self.GetId(), self.OnItemActivated)
350 else:
351 EVT_LEFT_DCLICK(self, self.OnDClick)
352 EVT_RIGHT_DOWN(self, self.OnRightDown)
353 EVT_TREE_ITEM_EXPANDED(self, self.GetId(), self.OnItemExpandedCollapsed)
354 EVT_TREE_ITEM_COLLAPSED(self, self.GetId(), self.OnItemExpandedCollapsed)
355
356 self.selection = None
357 self.needUpdate = False
358 self.pendingHighLight = None
359 self.ctrl = self.shift = False
360 self.dom = None
361 # Create image list
362 il = wxImageList(16, 16, True)
363 self.rootImage = il.Add(images.getTreeRootImage().Scale(16,16).ConvertToBitmap())
364 xxxObject.image = il.Add(images.getTreeDefaultImage().Scale(16,16).ConvertToBitmap())
365 xxxPanel.image = il.Add(images.getTreePanelImage().Scale(16,16).ConvertToBitmap())
366 xxxDialog.image = il.Add(images.getTreeDialogImage().Scale(16,16).ConvertToBitmap())
367 xxxFrame.image = il.Add(images.getTreeFrameImage().Scale(16,16).ConvertToBitmap())
368 xxxMenuBar.image = il.Add(images.getTreeMenuBarImage().Scale(16,16).ConvertToBitmap())
369 xxxMenu.image = il.Add(images.getTreeMenuImage().Scale(16,16).ConvertToBitmap())
370 xxxMenuItem.image = il.Add(images.getTreeMenuItemImage().Scale(16,16).ConvertToBitmap())
371 xxxToolBar.image = il.Add(images.getTreeToolBarImage().Scale(16,16).ConvertToBitmap())
372 xxxTool.image = il.Add(images.getTreeToolImage().Scale(16,16).ConvertToBitmap())
373 xxxSeparator.image = il.Add(images.getTreeSeparatorImage().Scale(16,16).ConvertToBitmap())
374 xxxSizer.imageH = il.Add(images.getTreeSizerHImage().Scale(16,16).ConvertToBitmap())
375 xxxSizer.imageV = il.Add(images.getTreeSizerVImage().Scale(16,16).ConvertToBitmap())
376 xxxStaticBoxSizer.imageH = il.Add(images.getTreeStaticBoxSizerHImage().Scale(16,16).ConvertToBitmap())
377 xxxStaticBoxSizer.imageV = il.Add(images.getTreeStaticBoxSizerVImage().Scale(16,16).ConvertToBitmap())
378 xxxGridSizer.image = il.Add(images.getTreeSizerGridImage().Scale(16,16).ConvertToBitmap())
379 xxxFlexGridSizer.image = il.Add(images.getTreeSizerFlexGridImage().Scale(16,16).ConvertToBitmap())
380 self.il = il
381 self.SetImageList(il)
382
383 def RegisterKeyEvents(self):
384 EVT_KEY_DOWN(self, g.tools.OnKeyDown)
385 EVT_KEY_UP(self, g.tools.OnKeyUp)
386 EVT_ENTER_WINDOW(self, g.tools.OnMouse)
387 EVT_LEAVE_WINDOW(self, g.tools.OnMouse)
388
389 def Unselect(self):
390 self.selection = None
391 wxTreeCtrl.Unselect(self)
392 g.tools.UpdateUI()
393
394 def ExpandAll(self, item):
395 if self.ItemHasChildren(item):
396 self.Expand(item)
fdc1a8be 397 i, cookie = self.GetFirstChild(item)
d14a1e28
RD
398 children = []
399 while i.IsOk():
400 children.append(i)
401 i, cookie = self.GetNextChild(item, cookie)
402 for i in children:
403 self.ExpandAll(i)
404 def CollapseAll(self, item):
405 if self.ItemHasChildren(item):
fdc1a8be 406 i, cookie = self.GetFirstChild(item)
d14a1e28
RD
407 children = []
408 while i.IsOk():
409 children.append(i)
410 i, cookie = self.GetNextChild(item, cookie)
411 for i in children:
412 self.CollapseAll(i)
413 self.Collapse(item)
414
415 # Clear tree
416 def Clear(self):
417 self.DeleteAllItems()
418 # Add minimal structure
419 if self.dom: self.dom.unlink()
420 self.dom = MyDocument()
421 self.dummyNode = self.dom.createComment('dummy node')
422 # Create main node
423 self.mainNode = self.dom.createElement('resource')
424 self.dom.appendChild(self.mainNode)
425 self.rootObj = xxxMainNode(self.dom)
426 self.root = self.AddRoot('XML tree', self.rootImage,
427 data=wxTreeItemData(self.rootObj))
428 self.SetItemHasChildren(self.root)
d14a1e28 429 self.Unselect()
fdc1a8be 430 self.Expand(self.root)
d14a1e28
RD
431
432 # Clear old data and set new
433 def SetData(self, dom):
434 self.DeleteAllItems()
435 # Add minimal structure
436 if self.dom: self.dom.unlink()
437 self.dom = dom
438 self.dummyNode = self.dom.createComment('dummy node')
439 # Find 'resource' child, add it's children
440 self.mainNode = dom.documentElement
441 self.rootObj = xxxMainNode(self.dom)
442 self.root = self.AddRoot('XML tree', self.rootImage,
443 data=wxTreeItemData(self.rootObj))
444 self.SetItemHasChildren(self.root)
445 nodes = self.mainNode.childNodes[:]
446 for node in nodes:
447 if IsObject(node):
448 self.AddNode(self.root, None, node)
449 else:
450 self.mainNode.removeChild(node)
451 node.unlink()
452 self.Expand(self.root)
453 self.Unselect()
454
455 # Add tree item for given parent item if node is DOM element node with
456 # 'object' tag. xxxParent is parent xxx object
457 def AddNode(self, itemParent, xxxParent, node):
458 # Set item data to current node
459 try:
460 xxx = MakeXXXFromDOM(xxxParent, node)
461 except:
462 print 'ERROR: MakeXXXFromDom(%s, %s)' % (xxxParent, node)
463 raise
464 treeObj = xxx.treeObject()
465 # Append tree item
466 item = self.AppendItem(itemParent, treeObj.treeName(),
467 image=treeObj.treeImage(),
468 data=wxTreeItemData(xxx))
469 # Try to find children objects
470 if treeObj.hasChildren:
471 nodes = treeObj.element.childNodes[:]
472 for n in nodes:
473 if IsObject(n):
474 self.AddNode(item, treeObj, n)
475 elif n.nodeType != minidom.Node.ELEMENT_NODE:
476 treeObj.element.removeChild(n)
477 n.unlink()
478
479 # Insert new item at specific position
480 def InsertNode(self, itemParent, parent, elem, nextItem):
481 # Insert in XML tree and wxWin
482 xxx = MakeXXXFromDOM(parent, elem)
483 # If nextItem is None, we append to parent, otherwise insert before it
484 if nextItem.IsOk():
485 node = self.GetPyData(nextItem).element
486 parent.element.insertBefore(elem, node)
487 # Inserting before is difficult, se we insert after or first child
488 index = self.ItemIndex(nextItem)
489 newItem = self.InsertItemBefore(itemParent, index,
490 xxx.treeName(), image=xxx.treeImage())
491 self.SetPyData(newItem, xxx)
492 else:
493 parent.element.appendChild(elem)
494 newItem = self.AppendItem(itemParent, xxx.treeName(), image=xxx.treeImage(),
495 data=wxTreeItemData(xxx))
496 # Add children items
497 if xxx.hasChildren:
498 treeObj = xxx.treeObject()
499 for n in treeObj.element.childNodes:
500 if IsObject(n):
501 self.AddNode(newItem, treeObj, n)
502 return newItem
503
504 # Remove leaf of tree, return it's data object
505 def RemoveLeaf(self, leaf):
506 xxx = self.GetPyData(leaf)
507 node = xxx.element
508 parent = node.parentNode
509 parent.removeChild(node)
510 self.Delete(leaf)
511 # Reset selection object
512 self.selection = None
513 return node
514 # Find position relative to the top-level window
515 def FindNodePos(self, item):
516 # Root at (0,0)
517 if item == g.testWin.item: return wxPoint(0, 0)
518 itemParent = self.GetItemParent(item)
519 # Select NB page
520 obj = self.FindNodeObject(item)
521 if self.GetPyData(itemParent).treeObject().__class__ == xxxNotebook:
522 notebook = self.FindNodeObject(itemParent)
523 # Find position
524 for i in range(notebook.GetPageCount()):
525 if notebook.GetPage(i) == obj:
fd919451
RR
526 if notebook.GetSelection() != i:
527 notebook.SetSelection(i)
528 # Remove highlight - otherwise highlight window won't be visible
529 if g.testWin.highLight:
530 g.testWin.highLight.Remove()
d14a1e28
RD
531 break
532 # Find first ancestor which is a wxWindow (not a sizer)
533 winParent = itemParent
534 while self.GetPyData(winParent).isSizer:
535 winParent = self.GetItemParent(winParent)
fd919451
RR
536 # Notebook children are layed out in a little strange way
537 if self.GetPyData(itemParent).treeObject().__class__ == xxxNotebook:
538 parentPos = wxPoint(0,0)
539 else:
540 parentPos = self.FindNodePos(winParent)
d14a1e28
RD
541 # Position (-1,-1) is really (0,0)
542 pos = obj.GetPosition()
543 if pos == (-1,-1): pos = (0,0)
544 return parentPos + pos
545
546 # Find window (or sizer) corresponding to a tree item.
547 def FindNodeObject(self, item):
548 testWin = g.testWin
549 # If top-level, return testWin (or panel its panel)
550 if item == testWin.item: return testWin.panel
551 itemParent = self.GetItemParent(item)
552 xxx = self.GetPyData(item).treeObject()
553 parentWin = self.FindNodeObject(itemParent)
554 # Top-level sizer? return window's sizer
555 if xxx.isSizer and isinstance(parentWin, wxWindow):
556 return parentWin.GetSizer()
557 # Otherwise get parent's object and it's child
558 child = parentWin.GetChildren()[self.ItemIndex(item)]
559 # Return window or sizer for sizer items
560 if child.GetClassName() == 'wxSizerItem':
561 if child.IsWindow(): child = child.GetWindow()
562 elif child.IsSizer():
563 child = child.GetSizer()
564 # Test for notebook sizers
565 if isinstance(child, wxNotebookSizer):
566 child = child.GetNotebook()
567 return child
568
569 def OnSelChanged(self, evt):
fd919451
RR
570 self.ChangeSelection(evt.GetItem())
571
572 def ChangeSelection(self, item):
d14a1e28
RD
573 # Apply changes
574 # !!! problem with wxGTK - GetOldItem is Ok if nothing selected
575 #oldItem = evt.GetOldItem()
576 status = ''
577 oldItem = self.selection
578 if oldItem:
579 xxx = self.GetPyData(oldItem)
580 # If some data was modified, apply changes
581 if g.panel.IsModified():
582 self.Apply(xxx, oldItem)
583 #if conf.autoRefresh:
584 if g.testWin:
585 if g.testWin.highLight:
586 g.testWin.highLight.Remove()
587 self.needUpdate = True
588 status = 'Changes were applied'
589 g.frame.SetStatusText(status)
590 # Generate view
fd919451 591 self.selection = item
d14a1e28
RD
592 if not self.selection.IsOk():
593 self.selection = None
594 return
595 xxx = self.GetPyData(self.selection)
596 # Update panel
597 g.panel.SetData(xxx)
598 # Update tools
599 g.tools.UpdateUI()
600 # Hightlighting is done in OnIdle
601 self.pendingHighLight = self.selection
602
603 # Check if item is in testWin subtree
604 def IsHighlatable(self, item):
605 if item == g.testWin.item: return False
606 while item != self.root:
607 item = self.GetItemParent(item)
608 if item == g.testWin.item: return True
609 return False
610
611 # Highlight selected item
612 def HighLight(self, item):
613 self.pendingHighLight = None
614 # Can highlight only with some top-level windows
615 if not g.testWin or self.GetPyData(g.testWin.item).treeObject().__class__ \
616 not in [xxxDialog, xxxPanel, xxxFrame]:
617 return
618 # If a control from another window is selected, remove highlight
619 if not self.IsHighlatable(item):
620 if g.testWin.highLight: g.testWin.highLight.Remove()
621 return
622 # Get window/sizer object
623 obj, pos = self.FindNodeObject(item), self.FindNodePos(item)
624 size = obj.GetSize()
625 # Highlight
626 # Nagative positions are not working wuite well
627 if g.testWin.highLight:
628 g.testWin.highLight.Replace(pos, size)
629 else:
630 g.testWin.highLight = HighLightBox(pos, size)
631 g.testWin.highLight.item = item
632
633 def ShowTestWindow(self, item):
634 xxx = self.GetPyData(item)
635 if g.panel.IsModified():
636 self.Apply(xxx, item) # apply changes
637 treeObj = xxx.treeObject()
638 if treeObj.className not in ['wxFrame', 'wxPanel', 'wxDialog',
639 'wxMenuBar', 'wxToolBar']:
640 wxLogMessage('No view for this element (yet)')
641 return
642 # Show item in bold
643 if g.testWin: # Reset old
644 self.SetItemBold(g.testWin.item, False)
645 self.CreateTestWin(item)
646 # Maybe an error occured, so we need to test
647 if g.testWin: self.SetItemBold(g.testWin.item)
648
649 # Double-click on Linux
650 def OnItemActivated(self, evt):
651 if evt.GetItem() != self.root:
652 self.ShowTestWindow(evt.GetItem())
653
654 # Double-click on Windows
655 def OnDClick(self, evt):
656 item, flags = self.HitTest(evt.GetPosition())
657 if flags in [wxTREE_HITTEST_ONITEMBUTTON, wxTREE_HITTEST_ONITEMLABEL]:
658 if item != self.root: self.ShowTestWindow(item)
659 else:
660 evt.Skip()
661
662 def OnItemExpandedCollapsed(self, evt):
663 # Update tool palette
664 g.tools.UpdateUI()
665 evt.Skip()
666
667 # (re)create test window
668 def CreateTestWin(self, item):
669 testWin = g.testWin
670 # Create a window with this resource
671 xxx = self.GetPyData(item).treeObject()
672
673 # If frame
674# if xxx.__class__ == xxxFrame:
675 # Frame can't have many children,
676 # but it's first child possibly can...
fdc1a8be 677# child = self.GetFirstChild(item)[0]
d14a1e28
RD
678# if child.IsOk() and self.GetPyData(child).__class__ == xxxPanel:
679# # Clean-up before recursive call or error
680# wxMemoryFSHandler_RemoveFile('xxx.xrc')
681# wxEndBusyCursor()
682# self.CreateTestWin(child)
683# return
684
685 wxBeginBusyCursor()
686 wxYield()
687 # Close old window, remember where it was
688 highLight = None
689 if testWin:
690 pos = testWin.GetPosition()
691 if item == testWin.item:
692 # Remember highlight if same top-level window
693 if testWin.highLight:
694 highLight = testWin.highLight.item
695 if xxx.className == 'wxPanel':
696 if testWin.highLight:
697 testWin.pendingHighLight = highLight
698 testWin.highLight.Remove()
699 testWin.panel.Destroy()
700 testWin.panel = None
701 else:
702 testWin.Destroy()
703 testWin = g.testWin = None
704 else:
705 testWin.Destroy()
706 testWin = g.testWin = None
707 else:
708 pos = g.testWinPos
709 # Save in memory FS
710 memFile = MemoryFile('xxx.xrc')
711 # Create partial XML file - faster for big files
712
713 dom = MyDocument()
714 mainNode = dom.createElement('resource')
715 dom.appendChild(mainNode)
716
717 # Remove temporarily from old parent
718 elem = xxx.element
719 # Change window id to _XRCED_T_W. This gives some name for
720 # unnamed windows, and for named gives the possibility to
721 # write sawfish scripts.
722 if not xxx.name:
723 name = 'noname'
724 else:
725 name = xxx.name
726 elem.setAttribute('name', STD_NAME)
727 parent = elem.parentNode
728 next = elem.nextSibling
729 parent.replaceChild(self.dummyNode, elem)
730 # Append to new DOM, write it
731 mainNode.appendChild(elem)
732 dom.writexml(memFile, encoding=self.rootObj.params['encoding'].value())
733 # Put back in place
734 mainNode.removeChild(elem)
735 dom.unlink()
736 parent.replaceChild(elem, self.dummyNode)
737 # Remove temporary name or restore changed
738 if not xxx.name:
739 elem.removeAttribute('name')
740 else:
741 elem.setAttribute('name', xxx.name)
742 memFile.close() # write to wxMemoryFS
743 xmlFlags = wxXRC_NO_SUBCLASSING
744 # Use translations if encoding is not specified
745 if g.currentEncoding == 'ascii':
746 xmlFlags != wxXRC_USE_LOCALE
747 res = wxXmlResource('', xmlFlags)
748 res.Load('memory:xxx.xrc')
a4c013b2
RR
749 try:
750 if xxx.__class__ == xxxFrame:
751 # Frame can't have many children,
752 # but it's first child possibly can...
753 # child = self.GetFirstChild(item)[0]
754 # if child.IsOk() and self.GetPyData(child).__class__ == xxxPanel:
755 # # Clean-up before recursive call or error
756 # wxMemoryFSHandler_RemoveFile('xxx.xrc')
757 # wxEndBusyCursor()
758 # self.CreateTestWin(child)
759 # return
760 # This currently works under GTK, but not under MSW
761 testWin = g.testWin = wxPreFrame()
762 res.LoadOnFrame(testWin, g.frame, STD_NAME)
763 # Create status bar
764 testWin.panel = testWin
765 testWin.CreateStatusBar()
766 testWin.SetClientSize(testWin.GetBestSize())
a4c013b2
RR
767 testWin.SetPosition(pos)
768 testWin.Show(True)
769 elif xxx.__class__ == xxxPanel:
770 # Create new frame
771 if not testWin:
772 testWin = g.testWin = wxFrame(g.frame, -1, 'Panel: ' + name,
773 pos=pos, name=STD_NAME)
774 testWin.panel = res.LoadPanel(testWin, STD_NAME)
775 testWin.SetClientSize(testWin.GetBestSize())
776 testWin.Show(True)
777 elif xxx.__class__ == xxxDialog:
778 testWin = g.testWin = res.LoadDialog(None, STD_NAME)
779 testWin.panel = testWin
780 testWin.Layout()
781 testWin.SetPosition(pos)
782 testWin.Show(True)
783 # Dialog's default code does not produce EVT_CLOSE
784 EVT_BUTTON(testWin, wxID_OK, self.OnCloseTestWin)
785 EVT_BUTTON(testWin, wxID_CANCEL, self.OnCloseTestWin)
786 elif xxx.__class__ == xxxMenuBar:
787 testWin = g.testWin = wxFrame(g.frame, -1, 'MenuBar: ' + name,
788 pos=pos, name=STD_NAME)
789 testWin.panel = None
790 # Set status bar to display help
791 testWin.CreateStatusBar()
792 testWin.menuBar = res.LoadMenuBar(STD_NAME)
793 testWin.SetMenuBar(testWin.menuBar)
794 testWin.Show(True)
795 elif xxx.__class__ == xxxToolBar:
796 testWin = g.testWin = wxFrame(g.frame, -1, 'ToolBar: ' + name,
d14a1e28 797 pos=pos, name=STD_NAME)
a4c013b2
RR
798 testWin.panel = None
799 # Set status bar to display help
800 testWin.CreateStatusBar()
801 testWin.toolBar = res.LoadToolBar(testWin, STD_NAME)
802 testWin.SetToolBar(testWin.toolBar)
803 testWin.Show(True)
804 testWin.item = item
805 EVT_CLOSE(testWin, self.OnCloseTestWin)
806 testWin.highLight = None
807 if highLight and not self.pendingHighLight:
808 self.HighLight(highLight)
809 except:
810 if g.testWin:
811 self.SetItemBold(item, False)
812 g.testWinPos = g.testWin.GetPosition()
813 g.testWin.Destroy()
814 g.testWin = None
815 inf = sys.exc_info()
816 wxLogError(traceback.format_exception(inf[0], inf[1], None)[-1])
817 wxLogError('Error loading resource')
d14a1e28 818 wxMemoryFSHandler_RemoveFile('xxx.xrc')
d14a1e28
RD
819 wxEndBusyCursor()
820
821 def OnCloseTestWin(self, evt):
822 self.SetItemBold(g.testWin.item, False)
823 g.testWinPos = g.testWin.GetPosition()
824 g.testWin.Destroy()
825 g.testWin = None
826
827 # Return item index in parent
828 def ItemIndex(self, item):
829 n = 0 # index of sibling
830 prev = self.GetPrevSibling(item)
831 while prev.IsOk():
832 prev = self.GetPrevSibling(prev)
833 n += 1
834 return n
835
836 # Full tree index of an item - list of positions
837 def ItemFullIndex(self, item):
838 if not item.IsOk(): return None
839 l = []
840 while item != self.root:
841 l.insert(0, self.ItemIndex(item))
842 item = self.GetItemParent(item)
843 return l
844 # Get item position from full index
845 def ItemAtFullIndex(self, index):
846 if index is None: return wxTreeItemId()
847 item = self.root
848 for i in index:
fdc1a8be 849 item = self.GetFirstChild(item)[0]
d14a1e28
RD
850 for k in range(i): item = self.GetNextSibling(item)
851 return item
852
853 # True if next item should be inserted after current (vs. appended to it)
854 def NeedInsert(self, item):
855 xxx = self.GetPyData(item)
856 if item == self.root: return False # root item
857 if xxx.hasChildren and not self.GetChildrenCount(item, False):
858 return False
859 return not (self.IsExpanded(item) and self.GetChildrenCount(item, False))
860
861 # Pull-down
862 def OnRightDown(self, evt):
863 pullDownMenu = g.pullDownMenu
864 # select this item
865 pt = evt.GetPosition();
866 item, flags = self.HitTest(pt)
867 if item.Ok() and flags & wxTREE_HITTEST_ONITEM:
868 self.SelectItem(item)
869
870 # Setup menu
871 menu = wxMenu()
872
873 item = self.selection
874 if not item:
875 menu.Append(g.pullDownMenu.ID_EXPAND, 'Expand', 'Expand tree')
876 menu.Append(g.pullDownMenu.ID_COLLAPSE, 'Collapse', 'Collapse tree')
877 else:
878# self.ctrl = evt.ControlDown() # save Ctrl state
879# self.shift = evt.ShiftDown() # and Shift too
880 m = wxMenu() # create menu
881 if self.ctrl:
882 needInsert = True
883 else:
884 needInsert = self.NeedInsert(item)
885 if item == self.root or needInsert and self.GetItemParent(item) == self.root:
886 SetMenu(m, pullDownMenu.topLevel)
887 else:
888 xxx = self.GetPyData(item).treeObject()
889 # Check parent for possible child nodes if inserting sibling
890 if needInsert: xxx = xxx.parent
891 if xxx.__class__ == xxxMenuBar:
892 m.Append(ID_NEW.MENU, 'Menu', 'Create menu')
893 elif xxx.__class__ in [xxxToolBar, xxxTool] or \
894 xxx.__class__ == xxxSeparator and xxx.parent.__class__ == xxxToolBar:
895 SetMenu(m, pullDownMenu.toolBarControls)
896 elif xxx.__class__ in [xxxMenu, xxxMenuItem]:
897 SetMenu(m, pullDownMenu.menuControls)
898 else:
899 SetMenu(m, pullDownMenu.controls)
900 if xxx.__class__ == xxxNotebook:
901 m.Enable(m.FindItem('sizer'), False)
902 elif not (xxx.isSizer or xxx.parent and xxx.parent.isSizer):
903 m.Enable(ID_NEW.SPACER, False)
904 # Select correct label for create menu
905 if not needInsert:
906 if self.shift:
907 menu.AppendMenu(wxNewId(), 'Insert Child', m,
908 'Create child object as the first child')
909 else:
910 menu.AppendMenu(wxNewId(), 'Append Child', m,
911 'Create child object as the last child')
912 else:
913 if self.shift:
914 menu.AppendMenu(wxNewId(), 'Create Sibling', m,
915 'Create sibling before selected object')
916 else:
917 menu.AppendMenu(wxNewId(), 'Create Sibling', m,
918 'Create sibling after selected object')
919 # Build replace menu
920 if item != self.root:
921 xxx = self.GetPyData(item).treeObject()
922 m = wxMenu() # create replace menu
923 if xxx.__class__ == xxxMenuBar:
924 m.Append(1000 + ID_NEW.MENU, 'Menu', 'Create menu')
925 elif xxx.__class__ in [xxxMenu, xxxMenuItem]:
926 SetMenu2(m, pullDownMenu.menuControls)
927 elif xxx.__class__ == xxxToolBar and \
928 self.GetItemParent(item) == self.root:
929 SetMenu2(m, [])
930 elif xxx.__class__ in [xxxFrame, xxxDialog, xxxPanel]:
931 SetMenu2(m, [
932 (ID_NEW.PANEL, 'Panel', 'Create panel'),
933 (ID_NEW.DIALOG, 'Dialog', 'Create dialog'),
934 (ID_NEW.FRAME, 'Frame', 'Create frame')])
935 elif xxx.isSizer:
936 SetMenu2(m, pullDownMenu.sizers)
937 else:
938 SetMenu2(m, pullDownMenu.controls)
939 id = wxNewId()
940 menu.AppendMenu(id, 'Replace With', m)
941 if not m.GetMenuItemCount(): menu.Enable(id, False)
2481bf3c
RR
942 menu.Append(pullDownMenu.ID_SUBCLASS, 'Subclass...',
943 'Set subclass property')
d14a1e28
RD
944 menu.AppendSeparator()
945 # Not using standart IDs because we don't want to show shortcuts
946 menu.Append(wxID_CUT, 'Cut', 'Cut to the clipboard')
947 menu.Append(wxID_COPY, 'Copy', 'Copy to the clipboard')
948 if self.ctrl and item != self.root:
949 menu.Append(pullDownMenu.ID_PASTE_SIBLING, 'Paste Sibling',
950 'Paste from the clipboard as a sibling')
951 else:
952 menu.Append(wxID_PASTE, 'Paste', 'Paste from the clipboard')
953 menu.Append(pullDownMenu.ID_DELETE,
954 'Delete', 'Delete object')
955 if self.ItemHasChildren(item):
956 menu.AppendSeparator()
957 menu.Append(pullDownMenu.ID_EXPAND, 'Expand', 'Expand subtree')
958 menu.Append(pullDownMenu.ID_COLLAPSE, 'Collapse', 'Collapse subtree')
959 self.PopupMenu(menu, evt.GetPosition())
960 menu.Destroy()
961
962 # Apply changes
963 def Apply(self, xxx, item):
964 g.panel.Apply()
965 # Update tree view
966 xxx = xxx.treeObject()
967 if xxx.hasName and self.GetItemText(item) != xxx.name:
968 self.SetItemText(item, xxx.treeName())
969 # Item width may have changed
970 # !!! Tric to update tree width (wxGTK, ??)
971 self.SetIndent(self.GetIndent())
972 # Change tree icon for sizers
973 if isinstance(xxx, xxxBoxSizer):
974 self.SetItemImage(item, xxx.treeImage())
975 # Set global modified state
976 g.frame.modified = True
1fded56b 977