wxPython/wx/tools/XRCed/tree.py

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