wxPython/wx/tools/XRCed/xrced.py

   1 # Name:         xrced.py
   2 # Purpose:      XRC editor, main module
   3 # Author:       Roman Rolinsky <rolinsky@mema.ucl.ac.be>
   4 # Created:      20.08.2001
   5 # RCS-ID:       $Id$
   6
   7 """
   8
   9 xrced -- Simple resource editor for XRC format used by wxWindows/wxPython
  10          GUI toolkit.
  11
  12 Usage:
  13
  14     xrced [ -h ] [ -v ] [ XRC-file ]
  15
  16 Options:
  17
  18     -h          output short usage info and exit
  19
  20     -v          output version info and exit
  21 """
  22
  23 from globals import *
  24 import os, sys, getopt, re, traceback, tempfile, shutil, cPickle
  25
  26 # Local modules
  27 from tree import *                      # imports xxx which imports params
  28 from panel import *
  29 from tools import *
  30 # Cleanup recursive import sideeffects, otherwise we can't create undoMan
  31 import undo
  32 undo.ParamPage = ParamPage
  33 undoMan = g.undoMan = UndoManager()
  34
  35 # Set application path for loading resources
  36 if __name__ == '__main__':
  37     basePath = os.path.dirname(sys.argv[0])
  38 else:
  39     basePath = os.path.dirname(__file__)
  40
  41 # 1 adds CMD command to Help menu
  42 debug = 0
  43
  44 g.helpText = """\
  45 <HTML><H2>Welcome to XRC<font color="blue">ed</font></H2><H3><font color="green">DON'T PANIC :)</font></H3>
  46 Read this note before clicking on anything!<P>
  47 To start select tree root, then popup menu with your right mouse button,
  48 select "Append Child", and then any command.<P>
  49 Or just press one of the buttons on the tools palette.<P>
  50 Enter XML ID, change properties, create children.<P>
  51 To test your interface select Test command (View menu).<P>
  52 Consult README file for the details.</HTML>
  53 """
  54
  55 defaultIDs = {xxxPanel:'PANEL', xxxDialog:'DIALOG', xxxFrame:'FRAME',
  56               xxxMenuBar:'MENUBAR', xxxMenu:'MENU', xxxToolBar:'TOOLBAR',
  57               xxxWizard:'WIZARD', xxxBitmap:'BITMAP', xxxIcon:'ICON'}
  58
  59 defaultName = 'UNTITLED.xrc'
  60
  61 ################################################################################
  62
  63 # ScrolledMessageDialog - modified from wxPython lib to set fixed-width font
  64 class ScrolledMessageDialog(wxDialog):
  65     def __init__(self, parent, msg, caption, pos = wxDefaultPosition, size = (500,300)):
  66         from wxPython.lib.layoutf import Layoutf
  67         wxDialog.__init__(self, parent, -1, caption, pos, size)
  68         text = wxTextCtrl(self, -1, msg, wxDefaultPosition,
  69                              wxDefaultSize, wxTE_MULTILINE | wxTE_READONLY)
  70         text.SetFont(g.modernFont())
  71         dc = wxWindowDC(text)
  72         # !!! possible bug - GetTextExtent without font returns sysfont dims
  73         w, h = dc.GetFullTextExtent(' ', g.modernFont())[:2]
  74         ok = wxButton(self, wxID_OK, "OK")
  75         text.SetConstraints(Layoutf('t=t5#1;b=t5#2;l=l5#1;r=r5#1', (self,ok)))
  76         text.SetSize((w * 80 + 30, h * 40))
  77         text.ShowPosition(1)
  78         ok.SetConstraints(Layoutf('b=b5#1;x%w50#1;w!80;h!25', (self,)))
  79         self.SetAutoLayout(True)
  80         self.Fit()
  81         self.CenterOnScreen(wxBOTH)
  82
  83 ################################################################################
  84
  85 # Event handler for using during location
  86 class Locator(wxEvtHandler):
  87     def ProcessEvent(self, evt):
  88         print evt
  89
  90 class Frame(wxFrame):
  91     def __init__(self, pos, size):
  92         wxFrame.__init__(self, None, -1, '', pos, size)
  93         global frame
  94         frame = g.frame = self
  95         bar = self.CreateStatusBar(2)
  96         bar.SetStatusWidths([-1, 40])
  97         self.SetIcon(images.getIconIcon())
  98
  99         # Idle flag
 100         self.inIdle = False
 101
 102         # Load our own resources
 103         self.res = wxXmlResource('')
 104         # !!! Blocking of assert failure occurring in older unicode builds
 105         try:
 106             self.res.Load(os.path.join(basePath, 'xrced.xrc'))
 107         except wx._core.PyAssertionError:
 108             print 'PyAssertionError was ignored'
 109
 110         # Make menus
 111         menuBar = wxMenuBar()
 112
 113         menu = wxMenu()
 114         menu.Append(wxID_NEW, '&New\tCtrl-N', 'New file')
 115         menu.AppendSeparator()
 116         menu.Append(wxID_OPEN, '&Open...\tCtrl-O', 'Open XRC file')
 117         self.recentMenu = wxMenu()
 118         self.AppendRecent(self.recentMenu)
 119         menu.AppendMenu(-1, 'Open Recent', self.recentMenu, 'Open a recent file')
 120         menu.AppendSeparator()
 121         menu.Append(wxID_SAVE, '&Save\tCtrl-S', 'Save XRC file')
 122         menu.Append(wxID_SAVEAS, 'Save &As...', 'Save XRC file under different name')
 123         menu.AppendSeparator()
 124         menu.Append(wxID_EXIT, '&Quit\tCtrl-Q', 'Exit application')
 125
 126         menuBar.Append(menu, '&File')
 127
 128         menu = wxMenu()
 129         menu.Append(wxID_UNDO, '&Undo\tCtrl-Z', 'Undo')
 130         menu.Append(wxID_REDO, '&Redo\tCtrl-Y', 'Redo')
 131         menu.AppendSeparator()
 132         menu.Append(wxID_CUT, 'Cut\tCtrl-X', 'Cut to the clipboard')
 133         menu.Append(wxID_COPY, '&Copy\tCtrl-C', 'Copy to the clipboard')
 134         menu.Append(wxID_PASTE, '&Paste\tCtrl-V', 'Paste from the clipboard')
 135         self.ID_DELETE = wxNewId()
 136         menu.Append(self.ID_DELETE, '&Delete\tCtrl-D', 'Delete object')
 137         menu.AppendSeparator()
 138         self.ID_LOCATE = wxNewId()
 139         self.ID_TOOL_LOCATE = wxNewId()
 140         self.ID_TOOL_PASTE = wxNewId()
 141         menu.Append(self.ID_LOCATE, '&Locate\tCtrl-L', 'Locate control in test window and select it')
 142         menuBar.Append(menu, '&Edit')
 143
 144         menu = wxMenu()
 145         self.ID_EMBED_PANEL = wxNewId()
 146         menu.Append(self.ID_EMBED_PANEL, '&Embed Panel',
 147                     'Toggle embedding properties panel in the main window', True)
 148         menu.Check(self.ID_EMBED_PANEL, conf.embedPanel)
 149         self.ID_SHOW_TOOLS = wxNewId()
 150         menu.Append(self.ID_SHOW_TOOLS, 'Show &Tools', 'Toggle tools', True)
 151         menu.Check(self.ID_SHOW_TOOLS, conf.showTools)
 152         menu.AppendSeparator()
 153         self.ID_TEST = wxNewId()
 154         menu.Append(self.ID_TEST, '&Test\tF5', 'Show test window')
 155         self.ID_REFRESH = wxNewId()
 156         menu.Append(self.ID_REFRESH, '&Refresh\tCtrl-R', 'Refresh test window')
 157         self.ID_AUTO_REFRESH = wxNewId()
 158         menu.Append(self.ID_AUTO_REFRESH, '&Auto-refresh\tCtrl-A',
 159                     'Toggle auto-refresh mode', True)
 160         menu.Check(self.ID_AUTO_REFRESH, conf.autoRefresh)
 161         self.ID_TEST_HIDE = wxNewId()
 162         menu.Append(self.ID_TEST_HIDE, '&Hide\tCtrl-H', 'Close test window')
 163         menuBar.Append(menu, '&View')
 164
 165         menu = wxMenu()
 166         menu.Append(wxID_ABOUT, '&About...', 'About XCRed')
 167         self.ID_README = wxNewId()
 168         menu.Append(self.ID_README, '&Readme...', 'View the README file')
 169         if debug:
 170             self.ID_DEBUG_CMD = wxNewId()
 171             menu.Append(self.ID_DEBUG_CMD, 'CMD', 'Python command line')
 172             EVT_MENU(self, self.ID_DEBUG_CMD, self.OnDebugCMD)
 173         menuBar.Append(menu, '&Help')
 174
 175         self.menuBar = menuBar
 176         self.SetMenuBar(menuBar)
 177
 178         # Create toolbar
 179         tb = self.CreateToolBar(wxTB_HORIZONTAL | wxNO_BORDER | wxTB_FLAT)
 180         tb.SetToolBitmapSize((24,24))
 181         new_bmp  = wx.ArtProvider.GetBitmap(wx.ART_NORMAL_FILE, wx.ART_TOOLBAR)
 182         open_bmp = wx.ArtProvider.GetBitmap(wx.ART_FILE_OPEN, wx.ART_TOOLBAR)
 183         save_bmp = wx.ArtProvider.GetBitmap(wx.ART_FILE_SAVE, wx.ART_TOOLBAR)
 184         undo_bmp = wx.ArtProvider.GetBitmap(wx.ART_UNDO, wx.ART_TOOLBAR)
 185         redo_bmp = wx.ArtProvider.GetBitmap(wx.ART_REDO, wx.ART_TOOLBAR)
 186         cut_bmp  = wx.ArtProvider.GetBitmap(wx.ART_CUT, wx.ART_TOOLBAR)
 187         copy_bmp = wx.ArtProvider.GetBitmap(wx.ART_COPY, wx.ART_TOOLBAR)
 188         paste_bmp= wx.ArtProvider.GetBitmap(wx.ART_PASTE, wx.ART_TOOLBAR)
 189
 190         tb.AddSimpleTool(wxID_NEW, new_bmp, 'New', 'New file')
 191         tb.AddSimpleTool(wxID_OPEN, open_bmp, 'Open', 'Open file')
 192         tb.AddSimpleTool(wxID_SAVE, save_bmp, 'Save', 'Save file')
 193         tb.AddControl(wxStaticLine(tb, -1, size=(-1,23), style=wxLI_VERTICAL))
 194         tb.AddSimpleTool(wxID_UNDO, undo_bmp, 'Undo', 'Undo')
 195         tb.AddSimpleTool(wxID_REDO, redo_bmp, 'Redo', 'Redo')
 196         tb.AddControl(wxStaticLine(tb, -1, size=(-1,23), style=wxLI_VERTICAL))
 197         tb.AddSimpleTool(wxID_CUT, cut_bmp, 'Cut', 'Cut')
 198         tb.AddSimpleTool(wxID_COPY, copy_bmp, 'Copy', 'Copy')
 199         tb.AddSimpleTool(self.ID_TOOL_PASTE, paste_bmp, 'Paste', 'Paste')
 200         tb.AddControl(wxStaticLine(tb, -1, size=(-1,23), style=wxLI_VERTICAL))
 201         tb.AddSimpleTool(self.ID_TOOL_LOCATE,
 202                         images.getLocateBitmap(), #images.getLocateArmedBitmap(),
 203                         'Locate', 'Locate control in test window and select it', True)
 204         tb.AddControl(wxStaticLine(tb, -1, size=(-1,23), style=wxLI_VERTICAL))
 205         tb.AddSimpleTool(self.ID_TEST, images.getTestBitmap(), 'Test', 'Test window')
 206         tb.AddSimpleTool(self.ID_REFRESH, images.getRefreshBitmap(),
 207                          'Refresh', 'Refresh view')
 208         tb.AddSimpleTool(self.ID_AUTO_REFRESH, images.getAutoRefreshBitmap(),
 209                          'Auto-refresh', 'Toggle auto-refresh mode', True)
 210 #        if wxPlatform == '__WXGTK__':
 211 #            tb.AddSeparator()   # otherwise auto-refresh sticks in status line
 212         tb.ToggleTool(self.ID_AUTO_REFRESH, conf.autoRefresh)
 213         tb.Realize()
 214
 215         self.tb = tb
 216         self.minWidth = tb.GetSize()[0] # minimal width is the size of toolbar
 217
 218         # File
 219         EVT_MENU(self, wxID_NEW, self.OnNew)
 220         EVT_MENU(self, wxID_OPEN, self.OnOpen)
 221         EVT_MENU(self, wxID_SAVE, self.OnSaveOrSaveAs)
 222         EVT_MENU(self, wxID_SAVEAS, self.OnSaveOrSaveAs)
 223         EVT_MENU(self, wxID_EXIT, self.OnExit)
 224         # Edit
 225         EVT_MENU(self, wxID_UNDO, self.OnUndo)
 226         EVT_MENU(self, wxID_REDO, self.OnRedo)
 227         EVT_MENU(self, wxID_CUT, self.OnCutDelete)
 228         EVT_MENU(self, wxID_COPY, self.OnCopy)
 229         EVT_MENU(self, wxID_PASTE, self.OnPaste)
 230         EVT_MENU(self, self.ID_TOOL_PASTE, self.OnPaste)
 231         EVT_MENU(self, self.ID_DELETE, self.OnCutDelete)
 232         EVT_MENU(self, self.ID_LOCATE, self.OnLocate)
 233         EVT_MENU(self, self.ID_TOOL_LOCATE, self.OnLocate)
 234         # View
 235         EVT_MENU(self, self.ID_EMBED_PANEL, self.OnEmbedPanel)
 236         EVT_MENU(self, self.ID_SHOW_TOOLS, self.OnShowTools)
 237         EVT_MENU(self, self.ID_TEST, self.OnTest)
 238         EVT_MENU(self, self.ID_REFRESH, self.OnRefresh)
 239         EVT_MENU(self, self.ID_AUTO_REFRESH, self.OnAutoRefresh)
 240         EVT_MENU(self, self.ID_TEST_HIDE, self.OnTestHide)
 241         # Help
 242         EVT_MENU(self, wxID_ABOUT, self.OnAbout)
 243         EVT_MENU(self, self.ID_README, self.OnReadme)
 244
 245         # Update events
 246         EVT_UPDATE_UI(self, wxID_SAVE, self.OnUpdateUI)
 247         EVT_UPDATE_UI(self, wxID_CUT, self.OnUpdateUI)
 248         EVT_UPDATE_UI(self, wxID_COPY, self.OnUpdateUI)
 249         EVT_UPDATE_UI(self, wxID_PASTE, self.OnUpdateUI)
 250         EVT_UPDATE_UI(self, self.ID_LOCATE, self.OnUpdateUI)
 251         EVT_UPDATE_UI(self, self.ID_TOOL_LOCATE, self.OnUpdateUI)
 252         EVT_UPDATE_UI(self, self.ID_TOOL_PASTE, self.OnUpdateUI)
 253         EVT_UPDATE_UI(self, wxID_UNDO, self.OnUpdateUI)
 254         EVT_UPDATE_UI(self, wxID_REDO, self.OnUpdateUI)
 255         EVT_UPDATE_UI(self, self.ID_DELETE, self.OnUpdateUI)
 256         EVT_UPDATE_UI(self, self.ID_TEST, self.OnUpdateUI)
 257         EVT_UPDATE_UI(self, self.ID_REFRESH, self.OnUpdateUI)
 258
 259         # Build interface
 260         sizer = wxBoxSizer(wxVERTICAL)
 261         sizer.Add(wxStaticLine(self, -1), 0, wxEXPAND)
 262         # Horizontal sizer for toolbar and splitter
 263         self.toolsSizer = sizer1 = wxBoxSizer()
 264         splitter = wxSplitterWindow(self, -1, style=wxSP_3DSASH)
 265         self.splitter = splitter
 266         splitter.SetMinimumPaneSize(100)
 267         # Create tree
 268         global tree
 269         g.tree = tree = XML_Tree(splitter, -1)
 270
 271         # Init pull-down menu data
 272         global pullDownMenu
 273         g.pullDownMenu = pullDownMenu = PullDownMenu(self)
 274
 275         # Vertical toolbar for GUI buttons
 276         g.tools = tools = Tools(self)
 277         tools.Show(conf.showTools)
 278         if conf.showTools: sizer1.Add(tools, 0, wxEXPAND)
 279
 280         tree.RegisterKeyEvents()
 281
 282         # !!! frame styles are broken
 283         # Miniframe for not embedded mode
 284         miniFrame = wxFrame(self, -1, 'Properties & Style',
 285                             (conf.panelX, conf.panelY),
 286                             (conf.panelWidth, conf.panelHeight))
 287         self.miniFrame = miniFrame
 288         sizer2 = wxBoxSizer()
 289         miniFrame.SetAutoLayout(True)
 290         miniFrame.SetSizer(sizer2)
 291         EVT_CLOSE(self.miniFrame, self.OnCloseMiniFrame)
 292         # Create panel for parameters
 293         global panel
 294         if conf.embedPanel:
 295             panel = Panel(splitter)
 296             # Set plitter windows
 297             splitter.SplitVertically(tree, panel, conf.sashPos)
 298         else:
 299             panel = Panel(miniFrame)
 300             sizer2.Add(panel, 1, wxEXPAND)
 301             miniFrame.Show(True)
 302             splitter.Initialize(tree)
 303         sizer1.Add(splitter, 1, wxEXPAND)
 304         sizer.Add(sizer1, 1, wxEXPAND)
 305         self.SetAutoLayout(True)
 306         self.SetSizer(sizer)
 307
 308         # Initialize
 309         self.Clear()
 310
 311         # Other events
 312         EVT_IDLE(self, self.OnIdle)
 313         EVT_CLOSE(self, self.OnCloseWindow)
 314         EVT_KEY_DOWN(self, tools.OnKeyDown)
 315         EVT_KEY_UP(self, tools.OnKeyUp)
 316         EVT_ICONIZE(self, self.OnIconize)
 317
 318     def AppendRecent(self, menu):
 319         # add recently used files to the menu
 320         for id,name in conf.recentfiles.iteritems():
 321             menu.Append(id,name)
 322             EVT_MENU(self,id,self.OnRecentFile)
 323         return
 324
 325     def OnRecentFile(self,evt):
 326         # open recently used file
 327         if not self.AskSave(): return
 328         wxBeginBusyCursor()
 329         try:
 330             path=conf.recentfiles[evt.GetId()]
 331             if self.Open(path):
 332                 self.SetStatusText('Data loaded')
 333             else:
 334                 self.SetStatusText('Failed')
 335         except KeyError:
 336             self.SetStatusText('No such file')
 337         wxEndBusyCursor()
 338
 339     def OnNew(self, evt):
 340         if not self.AskSave(): return
 341         self.Clear()
 342
 343     def OnOpen(self, evt):
 344         if not self.AskSave(): return
 345         dlg = wxFileDialog(self, 'Open', os.path.dirname(self.dataFile),
 346                            '', '*.xrc', wxOPEN | wxCHANGE_DIR)
 347         if dlg.ShowModal() == wxID_OK:
 348             path = dlg.GetPath()
 349             self.SetStatusText('Loading...')
 350             wxBeginBusyCursor()
 351             try:
 352                 if self.Open(path):
 353                     self.SetStatusText('Data loaded')
 354                 else:
 355                     self.SetStatusText('Failed')
 356                 self.SaveRecent(path)
 357             finally:
 358                 wxEndBusyCursor()
 359         dlg.Destroy()
 360
 361     def OnSaveOrSaveAs(self, evt):
 362         if evt.GetId() == wxID_SAVEAS or not self.dataFile:
 363             if self.dataFile: name = ''
 364             else: name = defaultName
 365             dirname = os.path.abspath(os.path.dirname(self.dataFile))
 366             dlg = wxFileDialog(self, 'Save As', dirname, name, '*.xrc',
 367                                wxSAVE | wxOVERWRITE_PROMPT | wxCHANGE_DIR)
 368             if dlg.ShowModal() == wxID_OK:
 369                 path = dlg.GetPath()
 370                 dlg.Destroy()
 371             else:
 372                 dlg.Destroy()
 373                 return
 374         else:
 375             path = self.dataFile
 376         self.SetStatusText('Saving...')
 377         wxBeginBusyCursor()
 378         try:
 379             try:
 380                 tmpFile,tmpName = tempfile.mkstemp(prefix='xrced-')
 381                 os.close(tmpFile)
 382                 self.Save(tmpName) # save temporary file first
 383                 shutil.move(tmpName, path)
 384                 self.dataFile = path
 385                 self.SetStatusText('Data saved')
 386                 self.SaveRecent(path)
 387             except IOError:
 388                 self.SetStatusText('Failed')
 389         finally:
 390             wxEndBusyCursor()
 391
 392     def SaveRecent(self,path):
 393         # append to recently used files
 394         if path not in conf.recentfiles.values():
 395             newid = wxNewId()
 396             self.recentMenu.Append(newid, path)
 397             EVT_MENU(self, newid, self.OnRecentFile)
 398             conf.recentfiles[newid] = path
 399
 400     def OnExit(self, evt):
 401         self.Close()
 402
 403     def OnUndo(self, evt):
 404         # Extra check to not mess with idle updating
 405         if undoMan.CanUndo():
 406             undoMan.Undo()
 407
 408     def OnRedo(self, evt):
 409         if undoMan.CanRedo():
 410             undoMan.Redo()
 411
 412     def OnCopy(self, evt):
 413         selected = tree.selection
 414         if not selected: return         # key pressed event
 415         xxx = tree.GetPyData(selected)
 416         wx.TheClipboard.Open()
 417         data = wx.CustomDataObject('XRCED')
 418         data.SetData(cPickle.dumps(xxx.element))
 419         wx.TheClipboard.SetData(data)
 420         wx.TheClipboard.Close()
 421         self.SetStatusText('Copied')
 422
 423     def OnPaste(self, evt):
 424         selected = tree.selection
 425         if not selected: return         # key pressed event
 426         # For pasting with Ctrl pressed
 427         appendChild = True
 428         if evt.GetId() == pullDownMenu.ID_PASTE_SIBLING: appendChild = False
 429         elif evt.GetId() == self.ID_TOOL_PASTE:
 430             if g.tree.ctrl: appendChild = False
 431             else: appendChild = not tree.NeedInsert(selected)
 432         else: appendChild = not tree.NeedInsert(selected)
 433         xxx = tree.GetPyData(selected)
 434         if not appendChild:
 435             # If has next item, insert, else append to parent
 436             nextItem = tree.GetNextSibling(selected)
 437             parentLeaf = tree.GetItemParent(selected)
 438         # Expanded container (must have children)
 439         elif tree.IsExpanded(selected) and tree.GetChildrenCount(selected, False):
 440             # Insert as first child
 441             nextItem = tree.GetFirstChild(selected)[0]
 442             parentLeaf = selected
 443         else:
 444             # No children or unexpanded item - appendChild stays True
 445             nextItem = wxTreeItemId()   # no next item
 446             parentLeaf = selected
 447         parent = tree.GetPyData(parentLeaf).treeObject()
 448
 449         # Create a copy of clipboard pickled element
 450         wx.TheClipboard.Open()
 451         data = wx.CustomDataObject('XRCED')
 452         if not wx.TheClipboard.IsSupported(data.GetFormat()):
 453             wx.TheClipboard.Close()
 454             wx.LogError('unsupported clipboard format')
 455             return
 456         wx.TheClipboard.GetData(data)
 457         wx.TheClipboard.Close()
 458         elem = cPickle.loads(data.GetData())
 459         # Tempopary xxx object to test things
 460         xxx = MakeXXXFromDOM(parent, elem)
 461
 462         # Check compatibility
 463         error = False
 464         # Top-level
 465         x = xxx.treeObject()
 466         if x.__class__ in [xxxDialog, xxxFrame, xxxMenuBar, xxxWizard]:
 467             # Top-level classes
 468             if parent.__class__ != xxxMainNode: error = True
 469         elif x.__class__ == xxxToolBar:
 470             # Toolbar can be top-level of child of panel or frame
 471             if parent.__class__ not in [xxxMainNode, xxxPanel, xxxFrame] and \
 472                not parent.isSizer: error = True
 473         elif x.__class__ == xxxPanel and parent.__class__ == xxxMainNode:
 474             pass
 475         elif x.__class__ == xxxSpacer:
 476             if not parent.isSizer: error = True
 477         elif x.__class__ == xxxSeparator:
 478             if not parent.__class__ in [xxxMenu, xxxToolBar]: error = True
 479         elif x.__class__ == xxxTool:
 480             if parent.__class__ != xxxToolBar: error = True
 481         elif x.__class__ == xxxMenu:
 482             if not parent.__class__ in [xxxMainNode, xxxMenuBar, xxxMenu]: error = True
 483         elif x.__class__ == xxxMenuItem:
 484             if not parent.__class__ in [xxxMenuBar, xxxMenu]: error = True
 485         elif x.isSizer and parent.__class__ in [xxxNotebook, xxxChoicebook, xxxListbook]:
 486             error = True
 487         else:                           # normal controls can be almost anywhere
 488             if parent.__class__ == xxxMainNode or \
 489                parent.__class__ in [xxxMenuBar, xxxMenu]: error = True
 490         if error:
 491             if parent.__class__ == xxxMainNode: parentClass = 'root'
 492             else: parentClass = parent.className
 493             wxLogError('Incompatible parent/child: parent is %s, child is %s!' %
 494                        (parentClass, x.className))
 495             return
 496
 497         # Check parent and child relationships.
 498         # If parent is sizer or notebook, child is of wrong class or
 499         # parent is normal window, child is child container then detach child.
 500         isChildContainer = isinstance(xxx, xxxChildContainer)
 501         parentIsBook = parent.__class__ in [xxxNotebook, xxxChoicebook, xxxListbook]
 502         if isChildContainer and \
 503            ((parent.isSizer and not isinstance(xxx, xxxSizerItem)) or \
 504             (parentIsBook and not isinstance(xxx, xxxPage)) or \
 505            not (parent.isSizer or parentIsBook)):
 506             elem.removeChild(xxx.child.element) # detach child
 507             elem.unlink()           # delete child container
 508             elem = xxx.child.element # replace
 509             # This may help garbage collection
 510             xxx.child.parent = None
 511             isChildContainer = False
 512         # Parent is sizer or notebook, child is not child container
 513         if parent.isSizer and not isChildContainer and not isinstance(xxx, xxxSpacer):
 514             # Create sizer item element
 515             sizerItemElem = MakeEmptyDOM(parent.itemTag)
 516             sizerItemElem.appendChild(elem)
 517             elem = sizerItemElem
 518         elif isinstance(parent, xxxNotebook) and not isChildContainer:
 519             pageElem = MakeEmptyDOM('notebookpage')
 520             pageElem.appendChild(elem)
 521             elem = pageElem
 522         elif isinstance(parent, xxxChoicebook) and not isChildContainer:
 523             pageElem = MakeEmptyDOM('choicebookpage')
 524             pageElem.appendChild(elem)
 525             elem = pageElem
 526         elif isinstance(parent, xxxListbook) and not isChildContainer:
 527             pageElem = MakeEmptyDOM('listbookpage')
 528             pageElem.appendChild(elem)
 529             elem = pageElem
 530         # Insert new node, register undo
 531         newItem = tree.InsertNode(parentLeaf, parent, elem, nextItem)
 532         undoMan.RegisterUndo(UndoPasteCreate(parentLeaf, parent, newItem, selected))
 533         # Scroll to show new item (!!! redundant?)
 534         tree.EnsureVisible(newItem)
 535         tree.SelectItem(newItem)
 536         if not tree.IsVisible(newItem):
 537             tree.ScrollTo(newItem)
 538             tree.Refresh()
 539         # Update view?
 540         if g.testWin and tree.IsHighlatable(newItem):
 541             if conf.autoRefresh:
 542                 tree.needUpdate = True
 543                 tree.pendingHighLight = newItem
 544             else:
 545                 tree.pendingHighLight = None
 546         self.SetModified()
 547         self.SetStatusText('Pasted')
 548
 549     def OnCutDelete(self, evt):
 550         selected = tree.selection
 551         if not selected: return         # key pressed event
 552         # Undo info
 553         if evt.GetId() == wxID_CUT:
 554             self.lastOp = 'CUT'
 555             status = 'Removed to clipboard'
 556         else:
 557             self.lastOp = 'DELETE'
 558             status = 'Deleted'
 559         # Delete testWin?
 560         if g.testWin:
 561             # If deleting top-level item, delete testWin
 562             if selected == g.testWin.item:
 563                 g.testWin.Destroy()
 564                 g.testWin = None
 565             else:
 566                 # Remove highlight, update testWin
 567                 if g.testWin.highLight:
 568                     g.testWin.highLight.Remove()
 569                 tree.needUpdate = True
 570         # Prepare undo data
 571         panel.Apply()
 572         index = tree.ItemFullIndex(selected)
 573         parent = tree.GetPyData(tree.GetItemParent(selected)).treeObject()
 574         elem = tree.RemoveLeaf(selected)
 575         undoMan.RegisterUndo(UndoCutDelete(index, parent, elem))
 576         if evt.GetId() == wxID_CUT:
 577             wx.TheClipboard.Open()
 578             data = wx.CustomDataObject('XRCED')
 579             data.SetData(cPickle.dumps(elem))
 580             wx.TheClipboard.SetData(data)
 581             wx.TheClipboard.Close()
 582         tree.pendingHighLight = None
 583         tree.UnselectAll()
 584         tree.selection = None
 585         # Update tools
 586         g.tools.UpdateUI()
 587         panel.Clear()
 588         self.SetModified()
 589         self.SetStatusText(status)
 590
 591     def OnSubclass(self, evt):
 592         selected = tree.selection
 593         xxx = tree.GetPyData(selected).treeObject()
 594         elem = xxx.element
 595         subclass = xxx.subclass
 596         dlg = wxTextEntryDialog(self, 'Subclass:', defaultValue=subclass)
 597         if dlg.ShowModal() == wxID_OK:
 598             subclass = dlg.GetValue()
 599             if subclass:
 600                 elem.setAttribute('subclass', subclass)
 601             elif elem.hasAttribute('subclass'):
 602                 elem.removeAttribute('subclass')
 603             self.SetModified()
 604             xxx.subclass = elem.getAttribute('subclass')
 605             tree.SetItemText(selected, xxx.treeName())
 606             panel.pages[0].box.SetLabel(xxx.panelName())
 607         dlg.Destroy()
 608
 609     def OnEmbedPanel(self, evt):
 610         conf.embedPanel = evt.IsChecked()
 611         if conf.embedPanel:
 612             # Remember last dimentions
 613             conf.panelX, conf.panelY = self.miniFrame.GetPosition()
 614             conf.panelWidth, conf.panelHeight = self.miniFrame.GetSize()
 615             size = self.GetSize()
 616             pos = self.GetPosition()
 617             sizePanel = panel.GetSize()
 618             panel.Reparent(self.splitter)
 619             self.miniFrame.GetSizer().Remove(panel)
 620             # Widen
 621             self.SetDimensions(pos.x, pos.y, size.width + sizePanel.width, size.height)
 622             self.splitter.SplitVertically(tree, panel, conf.sashPos)
 623             self.miniFrame.Show(False)
 624         else:
 625             conf.sashPos = self.splitter.GetSashPosition()
 626             pos = self.GetPosition()
 627             size = self.GetSize()
 628             sizePanel = panel.GetSize()
 629             self.splitter.Unsplit(panel)
 630             sizer = self.miniFrame.GetSizer()
 631             panel.Reparent(self.miniFrame)
 632             panel.Show(True)
 633             sizer.Add(panel, 1, wxEXPAND)
 634             self.miniFrame.Show(True)
 635             self.miniFrame.SetDimensions(conf.panelX, conf.panelY,
 636                                          conf.panelWidth, conf.panelHeight)
 637             # Reduce width
 638             self.SetDimensions(pos.x, pos.y,
 639                                max(size.width - sizePanel.width, self.minWidth), size.height)
 640
 641     def OnShowTools(self, evt):
 642         conf.showTools = evt.IsChecked()
 643         g.tools.Show(conf.showTools)
 644         if conf.showTools:
 645             self.toolsSizer.Prepend(g.tools, 0, wxEXPAND)
 646         else:
 647             self.toolsSizer.Remove(g.tools)
 648         self.toolsSizer.Layout()
 649
 650     def OnTest(self, evt):
 651         if not tree.selection: return   # key pressed event
 652         tree.ShowTestWindow(tree.selection)
 653
 654     def OnTestHide(self, evt):
 655         tree.CloseTestWindow()
 656
 657     # Find object by relative position
 658     def FindObject(self, item, obj):
 659         # We simply perform depth-first traversal, sinse it's too much
 660         # hassle to deal with all sizer/window combinations
 661         w = tree.FindNodeObject(item)
 662         if w == obj:
 663             return item
 664         if tree.ItemHasChildren(item):
 665             child = tree.GetFirstChild(item)[0]
 666             while child:
 667                 found = self.FindObject(child, obj)
 668                 if found: return found
 669                 child = tree.GetNextSibling(child)
 670         return None
 671
 672     def OnTestWinLeftDown(self, evt):
 673         pos = evt.GetPosition()
 674         self.SetHandler(g.testWin)
 675         g.testWin.Disconnect(wxID_ANY, wxID_ANY, wxEVT_LEFT_DOWN)
 676         item = self.FindObject(g.testWin.item, evt.GetEventObject())
 677         if item:
 678             tree.SelectItem(item)
 679         self.tb.ToggleTool(self.ID_TOOL_LOCATE, False)
 680         if item:
 681             self.SetStatusText('Selected %s' % tree.GetItemText(item))
 682         else:
 683             self.SetStatusText('Locate failed!')
 684
 685     def SetHandler(self, w, h=None):
 686         if h:
 687             w.SetEventHandler(h)
 688             w.SetCursor(wxCROSS_CURSOR)
 689         else:
 690             w.SetEventHandler(w)
 691             w.SetCursor(wxNullCursor)
 692         for ch in w.GetChildren():
 693             self.SetHandler(ch, h)
 694
 695     def OnLocate(self, evt):
 696         if g.testWin:
 697             if evt.GetId() == self.ID_LOCATE or \
 698                evt.GetId() == self.ID_TOOL_LOCATE and evt.IsChecked():
 699                 self.SetHandler(g.testWin, g.testWin)
 700                 g.testWin.Connect(wxID_ANY, wxID_ANY, wxEVT_LEFT_DOWN, self.OnTestWinLeftDown)
 701                 if evt.GetId() == self.ID_LOCATE:
 702                     self.tb.ToggleTool(self.ID_TOOL_LOCATE, True)
 703             elif evt.GetId() == self.ID_TOOL_LOCATE and not evt.IsChecked():
 704                 self.SetHandler(g.testWin, None)
 705                 g.testWin.Disconnect(wxID_ANY, wxID_ANY, wxEVT_LEFT_DOWN)
 706             self.SetStatusText('Click somewhere in your test window now')
 707
 708     def OnRefresh(self, evt):
 709         # If modified, apply first
 710         selection = tree.selection
 711         if selection:
 712             xxx = tree.GetPyData(selection)
 713             if xxx and panel.IsModified():
 714                 tree.Apply(xxx, selection)
 715         if g.testWin:
 716             # (re)create
 717             tree.CreateTestWin(g.testWin.item)
 718         panel.modified = False
 719         tree.needUpdate = False
 720
 721     def OnAutoRefresh(self, evt):
 722         conf.autoRefresh = evt.IsChecked()
 723         self.menuBar.Check(self.ID_AUTO_REFRESH, conf.autoRefresh)
 724         self.tb.ToggleTool(self.ID_AUTO_REFRESH, conf.autoRefresh)
 725
 726     def OnAbout(self, evt):
 727         str = '''\
 728 XRCed version %s
 729
 730 (c) Roman Rolinsky <rollrom@users.sourceforge.net>
 731 Homepage: http://xrced.sourceforge.net\
 732 ''' % version
 733         dlg = wxMessageDialog(self, str, 'About XRCed', wxOK | wxCENTRE)
 734         dlg.ShowModal()
 735         dlg.Destroy()
 736
 737     def OnReadme(self, evt):
 738         text = open(os.path.join(basePath, 'README.txt'), 'r').read()
 739         dlg = ScrolledMessageDialog(self, text, "XRCed README")
 740         dlg.ShowModal()
 741         dlg.Destroy()
 742
 743     # Simple emulation of python command line
 744     def OnDebugCMD(self, evt):
 745         while 1:
 746             try:
 747                 exec raw_input('C:\> ')
 748             except EOFError:
 749                 print '^D'
 750                 break
 751             except:
 752                 (etype, value, tb) =sys.exc_info()
 753                 tblist =traceback.extract_tb(tb)[1:]
 754                 msg =' '.join(traceback.format_exception_only(etype, value)
 755                         +traceback.format_list(tblist))
 756                 print msg
 757
 758     def OnCreate(self, evt):
 759         selected = tree.selection
 760         if tree.ctrl: appendChild = False
 761         else: appendChild = not tree.NeedInsert(selected)
 762         xxx = tree.GetPyData(selected)
 763         if not appendChild:
 764             # If insert before
 765             if tree.shift:
 766                 # If has previous item, insert after it, else append to parent
 767                 nextItem = selected
 768                 parentLeaf = tree.GetItemParent(selected)
 769             else:
 770                 # If has next item, insert, else append to parent
 771                 nextItem = tree.GetNextSibling(selected)
 772                 parentLeaf = tree.GetItemParent(selected)
 773         # Expanded container (must have children)
 774         elif tree.shift and tree.IsExpanded(selected) \
 775            and tree.GetChildrenCount(selected, False):
 776             nextItem = tree.GetFirstChild(selected)[0]
 777             parentLeaf = selected
 778         else:
 779             nextItem = wxTreeItemId()
 780             parentLeaf = selected
 781         parent = tree.GetPyData(parentLeaf)
 782         if parent.hasChild: parent = parent.child
 783
 784         # Create object_ref?
 785         if evt.GetId() == ID_NEW.REF:
 786             ref = wxGetTextFromUser('Create reference to:', 'Create reference')
 787             if not ref: return
 788             xxx = MakeEmptyRefXXX(parent, ref)
 789         else:
 790             # Create empty element
 791             className = pullDownMenu.createMap[evt.GetId()]
 792             xxx = MakeEmptyXXX(parent, className)
 793
 794         # Set default name for top-level windows
 795         if parent.__class__ == xxxMainNode:
 796             cl = xxx.treeObject().__class__
 797             frame.maxIDs[cl] += 1
 798             xxx.setTreeName('%s%d' % (defaultIDs[cl], frame.maxIDs[cl]))
 799         # And for some other standard controls
 800         elif parent.__class__ == xxxStdDialogButtonSizer:
 801             xxx.setTreeName(pullDownMenu.stdButtonIDs[evt.GetId()][0])
 802             # We can even set label
 803             obj = xxx.treeObject()
 804             elem = g.tree.dom.createElement('label')
 805             elem.appendChild(g.tree.dom.createTextNode(pullDownMenu.stdButtonIDs[evt.GetId()][1]))
 806             obj.params['label'] = xxxParam(elem)
 807             xxx.treeObject().element.appendChild(elem)
 808
 809         # Insert new node, register undo
 810         elem = xxx.element
 811         newItem = tree.InsertNode(parentLeaf, parent, elem, nextItem)
 812         undoMan.RegisterUndo(UndoPasteCreate(parentLeaf, parent, newItem, selected))
 813         tree.EnsureVisible(newItem)
 814         tree.SelectItem(newItem)
 815         if not tree.IsVisible(newItem):
 816             tree.ScrollTo(newItem)
 817             tree.Refresh()
 818         # Update view?
 819         if g.testWin and tree.IsHighlatable(newItem):
 820             if conf.autoRefresh:
 821                 tree.needUpdate = True
 822                 tree.pendingHighLight = newItem
 823             else:
 824                 tree.pendingHighLight = None
 825         tree.SetFocus()
 826         self.SetModified()
 827
 828     # Replace one object with another
 829     def OnReplace(self, evt):
 830         selected = tree.selection
 831         xxx = tree.GetPyData(selected).treeObject()
 832         elem = xxx.element
 833         parent = elem.parentNode
 834         undoMan.RegisterUndo(UndoReplace(selected))
 835         # New class
 836         className = pullDownMenu.createMap[evt.GetId() - 1000]
 837         # Create temporary empty node (with default values)
 838         dummy = MakeEmptyDOM(className)
 839         if className == 'spacer' and xxx.className != 'spacer':
 840             klass = xxxSpacer
 841         elif xxx.className == 'spacer' and className != 'spacer':
 842             klass = xxxSizerItem
 843         else:
 844             klass = xxxDict[className]
 845         # Remove non-compatible children
 846         if tree.ItemHasChildren(selected) and not klass.hasChildren:
 847             tree.DeleteChildren(selected)
 848         nodes = elem.childNodes[:]
 849         tags = []
 850         for node in nodes:
 851             if node.nodeType != minidom.Node.ELEMENT_NODE: continue
 852             remove = False
 853             tag = node.tagName
 854             if tag == 'object':
 855                 if not klass.hasChildren:  remove = True
 856             elif tag not in klass.allParams and \
 857                      (not klass.hasStyle or tag not in klass.styles):
 858                 remove = True
 859             else:
 860                 tags.append(tag)
 861             if remove:
 862                 elem.removeChild(node)
 863                 node.unlink()
 864
 865         # Remove sizeritem child if spacer
 866         if className == 'spacer' and xxx.className != 'spacer':
 867             sizeritem = elem.parentNode
 868             assert sizeritem.getAttribute('class') == 'sizeritem'
 869             sizeritem.removeChild(elem)
 870             elem.unlink()
 871             elem = sizeritem
 872             tree.GetPyData(selected).hasChild = false
 873         elif xxx.className == 'spacer' and className != 'spacer':
 874             # Create sizeritem element
 875             assert xxx.parent.isSizer
 876             elem.setAttribute('class', 'sizeritem')
 877             node = MakeEmptyDOM(className)
 878             elem.appendChild(node)
 879             # Replace to point to new object
 880             xxx = xxxSizerItem(xxx.parent, elem)
 881             elem = node
 882             tree.SetPyData(selected, xxx)
 883             xxx = xxx.child
 884         else:
 885             # Copy parameters present in dummy but not in elem
 886             for node in dummy.childNodes:
 887                 if node.tagName not in tags:  elem.appendChild(node.cloneNode(True))
 888         dummy.unlink()
 889
 890         # Change class name
 891         elem.setAttribute('class', className)
 892         if elem.hasAttribute('subclass'):
 893             elem.removeAttribute('subclass') # clear subclassing
 894         # Re-create xxx element
 895         xxx = MakeXXXFromDOM(xxx.parent, elem)
 896         # Update parent in child objects
 897         if tree.ItemHasChildren(selected):
 898             i, cookie = tree.GetFirstChild(selected)
 899             while i.IsOk():
 900                 x = tree.GetPyData(i)
 901                 x.parent = xxx
 902                 if x.hasChild: x.child.parent = xxx
 903                 i, cookie = tree.GetNextChild(selected, cookie)
 904
 905         # Update tree
 906         if tree.GetPyData(selected).hasChild: # child container
 907             container = tree.GetPyData(selected)
 908             container.child = xxx
 909             container.hasChildren = xxx.hasChildren
 910             container.isSizer = xxx.isSizer
 911             xxx = container
 912         else:
 913             tree.SetPyData(selected, xxx)
 914         tree.SetItemText(selected, xxx.treeName())
 915         tree.SetItemImage(selected, xxx.treeImage())
 916
 917         # Set default name for top-level windows
 918         if parent.__class__ == xxxMainNode:
 919             cl = xxx.treeObject().__class__
 920             frame.maxIDs[cl] += 1
 921             xxx.setTreeName('%s%d' % (defaultIDs[cl], frame.maxIDs[cl]))
 922
 923         # Update panel
 924         g.panel.SetData(xxx)
 925         # Update tools
 926         g.tools.UpdateUI()
 927
 928         #undoMan.RegisterUndo(UndoPasteCreate(parentLeaf, parent, newItem, selected))
 929         # Update view?
 930         if g.testWin and tree.IsHighlatable(selected):
 931             if conf.autoRefresh:
 932                 tree.needUpdate = True
 933                 tree.pendingHighLight = selected
 934             else:
 935                 tree.pendingHighLight = None
 936         tree.SetFocus()
 937         self.SetModified()
 938
 939     # Expand/collapse subtree
 940     def OnExpand(self, evt):
 941         if tree.selection: tree.ExpandAll(tree.selection)
 942         else: tree.ExpandAll(tree.root)
 943     def OnCollapse(self, evt):
 944         if tree.selection: tree.CollapseAll(tree.selection)
 945         else: tree.CollapseAll(tree.root)
 946
 947     def OnPullDownHighlight(self, evt):
 948         menuId = evt.GetMenuId()
 949         if menuId != -1:
 950             menu = evt.GetEventObject()
 951             help = menu.GetHelpString(menuId)
 952             self.SetStatusText(help)
 953         else:
 954             self.SetStatusText('')
 955
 956     def OnUpdateUI(self, evt):
 957         if evt.GetId() in [wxID_CUT, wxID_COPY, self.ID_DELETE]:
 958             evt.Enable(tree.selection is not None and tree.selection != tree.root)
 959         elif evt.GetId() == wxID_SAVE:
 960             evt.Enable(self.modified)
 961         elif evt.GetId() in [wxID_PASTE, self.ID_TOOL_PASTE]:
 962             evt.Enable(tree.selection is not None)
 963         elif evt.GetId() == self.ID_TEST:
 964             evt.Enable(tree.selection is not None and tree.selection != tree.root)
 965         elif evt.GetId() in [self.ID_LOCATE, self.ID_TOOL_LOCATE]:
 966             evt.Enable(g.testWin is not None)
 967         elif evt.GetId() == wxID_UNDO:  evt.Enable(undoMan.CanUndo())
 968         elif evt.GetId() == wxID_REDO:  evt.Enable(undoMan.CanRedo())
 969
 970     def OnIdle(self, evt):
 971         if self.inIdle: return          # Recursive call protection
 972         self.inIdle = True
 973         try:
 974             if tree.needUpdate:
 975                 if conf.autoRefresh:
 976                     if g.testWin:
 977                         self.SetStatusText('Refreshing test window...')
 978                         # (re)create
 979                         tree.CreateTestWin(g.testWin.item)
 980                         self.SetStatusText('')
 981                     tree.needUpdate = False
 982             elif tree.pendingHighLight:
 983                 try:
 984                     tree.HighLight(tree.pendingHighLight)
 985                 except:
 986                     # Remove highlight if any problem
 987                     if g.testWin.highLight:
 988                         g.testWin.highLight.Remove()
 989                     tree.pendingHighLight = None
 990                     raise
 991             else:
 992                 evt.Skip()
 993         finally:
 994             self.inIdle = False
 995
 996     # We don't let close panel window
 997     def OnCloseMiniFrame(self, evt):
 998         return
 999
1000     def OnIconize(self, evt):
1001         conf.x, conf.y = self.GetPosition()
1002         conf.width, conf.height = self.GetSize()
1003         if conf.embedPanel:
1004             conf.sashPos = self.splitter.GetSashPosition()
1005         else:
1006             conf.panelX, conf.panelY = self.miniFrame.GetPosition()
1007             conf.panelWidth, conf.panelHeight = self.miniFrame.GetSize()
1008             self.miniFrame.Iconize()
1009         evt.Skip()
1010
1011     def OnCloseWindow(self, evt):
1012         if not self.AskSave(): return
1013         if g.testWin: g.testWin.Destroy()
1014         if not panel.GetPageCount() == 2:
1015             panel.page2.Destroy()
1016         else:
1017             # If we don't do this, page does not get destroyed (a bug?)
1018             panel.RemovePage(1)
1019         if not self.IsIconized():
1020             conf.x, conf.y = self.GetPosition()
1021             conf.width, conf.height = self.GetSize()
1022             if conf.embedPanel:
1023                 conf.sashPos = self.splitter.GetSashPosition()
1024             else:
1025                 conf.panelX, conf.panelY = self.miniFrame.GetPosition()
1026                 conf.panelWidth, conf.panelHeight = self.miniFrame.GetSize()
1027         evt.Skip()
1028
1029     def Clear(self):
1030         self.dataFile = ''
1031         undoMan.Clear()
1032         self.SetModified(False)
1033         tree.Clear()
1034         panel.Clear()
1035         if g.testWin:
1036             g.testWin.Destroy()
1037             g.testWin = None
1038         # Numbers for new controls
1039         self.maxIDs = {}
1040         for cl in [xxxPanel, xxxDialog, xxxFrame,
1041                    xxxMenuBar, xxxMenu, xxxToolBar,
1042                    xxxWizard, xxxBitmap, xxxIcon]:
1043             self.maxIDs[cl] = 0
1044
1045     def SetModified(self, state=True):
1046         self.modified = state
1047         name = os.path.basename(self.dataFile)
1048         if not name: name = defaultName
1049         if state:
1050             self.SetTitle(progname + ': ' + name + ' *')
1051         else:
1052             self.SetTitle(progname + ': ' + name)
1053
1054     def Open(self, path):
1055         if not os.path.exists(path):
1056             wxLogError('File does not exists: %s' % path)
1057             return False
1058         # Try to read the file
1059         try:
1060             f = open(path)
1061             self.Clear()
1062             dom = minidom.parse(f)
1063             f.close()
1064             # Set encoding global variable and default encoding
1065             if dom.encoding:
1066                 g.currentEncoding = dom.encoding
1067                 wx.SetDefaultPyEncoding(g.currentEncoding.encode())
1068             else:
1069                 g.currentEncoding = ''
1070             # Change dir
1071             self.dataFile = path = os.path.abspath(path)
1072             dir = os.path.dirname(path)
1073             if dir: os.chdir(dir)
1074             tree.SetData(dom)
1075             self.SetTitle(progname + ': ' + os.path.basename(path))
1076         except:
1077             # Nice exception printing
1078             inf = sys.exc_info()
1079             wxLogError(traceback.format_exception(inf[0], inf[1], None)[-1])
1080             wxLogError('Error reading file: %s' % path)
1081             if debug: raise
1082             return False
1083         return True
1084
1085     def Indent(self, node, indent = 0):
1086         # Copy child list because it will change soon
1087         children = node.childNodes[:]
1088         # Main node doesn't need to be indented
1089         if indent:
1090             text = self.domCopy.createTextNode('\n' + ' ' * indent)
1091             node.parentNode.insertBefore(text, node)
1092         if children:
1093             # Append newline after last child, except for text nodes
1094             if children[-1].nodeType == minidom.Node.ELEMENT_NODE:
1095                 text = self.domCopy.createTextNode('\n' + ' ' * indent)
1096                 node.appendChild(text)
1097             # Indent children which are elements
1098             for n in children:
1099                 if n.nodeType == minidom.Node.ELEMENT_NODE:
1100                     self.Indent(n, indent + 2)
1101
1102     def Save(self, path):
1103         try:
1104             import codecs
1105             # Apply changes
1106             if tree.selection and panel.IsModified():
1107                 self.OnRefresh(wxCommandEvent())
1108             if g.currentEncoding:
1109                 f = codecs.open(path, 'wt', g.currentEncoding)
1110             else:
1111                 f = codecs.open(path, 'wt')
1112             # Make temporary copy for formatting it
1113             # !!! We can't clone dom node, it works only once
1114             #self.domCopy = tree.dom.cloneNode(True)
1115             self.domCopy = MyDocument()
1116             mainNode = self.domCopy.appendChild(tree.mainNode.cloneNode(True))
1117             # Remove first child (test element)
1118             testElem = mainNode.firstChild
1119             mainNode.removeChild(testElem)
1120             testElem.unlink()
1121             self.Indent(mainNode)
1122             self.domCopy.writexml(f, encoding = g.currentEncoding)
1123             f.close()
1124             self.domCopy.unlink()
1125             self.domCopy = None
1126             self.SetModified(False)
1127             panel.SetModified(False)
1128         except:
1129             inf = sys.exc_info()
1130             wxLogError(traceback.format_exception(inf[0], inf[1], None)[-1])
1131             wxLogError('Error writing file: %s' % path)
1132             raise
1133
1134     def AskSave(self):
1135         if not (self.modified or panel.IsModified()): return True
1136         flags = wxICON_EXCLAMATION | wxYES_NO | wxCANCEL | wxCENTRE
1137         dlg = wxMessageDialog( self, 'File is modified. Save before exit?',
1138                                'Save before too late?', flags )
1139         say = dlg.ShowModal()
1140         dlg.Destroy()
1141         if say == wxID_YES:
1142             self.OnSaveOrSaveAs(wxCommandEvent(wxID_SAVE))
1143             # If save was successful, modified flag is unset
1144             if not self.modified: return True
1145         elif say == wxID_NO:
1146             self.SetModified(False)
1147             panel.SetModified(False)
1148             return True
1149         return False
1150
1151     def SaveUndo(self):
1152         pass                            # !!!
1153
1154 ################################################################################
1155
1156 def usage():
1157     print >> sys.stderr, 'usage: xrced [-dhiv] [file]'
1158
1159 class App(wxApp):
1160     def OnInit(self):
1161         # Check version
1162         if wxVERSION[:3] < MinWxVersion:
1163             wxLogWarning('''\
1164 This version of XRCed may not work correctly on your version of wxWindows. \
1165 Please upgrade wxWindows to %d.%d.%d or higher.''' % MinWxVersion)
1166         global debug
1167         # Process comand-line
1168         opts = args = None
1169         try:
1170             opts, args = getopt.getopt(sys.argv[1:], 'dhiv')
1171             for o,a in opts:
1172                 if o == '-h':
1173                     usage()
1174                     sys.exit(0)
1175                 elif o == '-d':
1176                     debug = True
1177                 elif o == '-v':
1178                     print 'XRCed version', version
1179                     sys.exit(0)
1180
1181         except getopt.GetoptError:
1182             if wxPlatform != '__WXMAC__': # macs have some extra parameters
1183                 print >> sys.stderr, 'Unknown option'
1184                 usage()
1185                 sys.exit(1)
1186
1187         self.SetAppName('xrced')
1188         # Settings
1189         global conf
1190         conf = g.conf = wxConfig(style = wxCONFIG_USE_LOCAL_FILE)
1191         conf.autoRefresh = conf.ReadInt('autorefresh', True)
1192         pos = conf.ReadInt('x', -1), conf.ReadInt('y', -1)
1193         size = conf.ReadInt('width', 800), conf.ReadInt('height', 600)
1194         conf.embedPanel = conf.ReadInt('embedPanel', True)
1195         conf.showTools = conf.ReadInt('showTools', True)
1196         conf.sashPos = conf.ReadInt('sashPos', 200)
1197         # read recently used files
1198         recentfiles=conf.Read('recentFiles','')
1199         conf.recentfiles={}
1200         if recentfiles:
1201             for fil in recentfiles.split('|'):
1202                 conf.recentfiles[wxNewId()]=fil
1203         if not conf.embedPanel:
1204             conf.panelX = conf.ReadInt('panelX', -1)
1205             conf.panelY = conf.ReadInt('panelY', -1)
1206         else:
1207             conf.panelX = conf.panelY = -1
1208         conf.panelWidth = conf.ReadInt('panelWidth', 200)
1209         conf.panelHeight = conf.ReadInt('panelHeight', 200)
1210         conf.panic = not conf.HasEntry('nopanic')
1211         # Add handlers
1212         wxFileSystem_AddHandler(wxMemoryFSHandler())
1213         wxInitAllImageHandlers()
1214         # Create main frame
1215         frame = Frame(pos, size)
1216         frame.Show(True)
1217
1218         # Load file after showing
1219         if args:
1220             conf.panic = False
1221             frame.open = frame.Open(args[0])
1222
1223         return True
1224
1225     def OnExit(self):
1226         # Write config
1227         global conf
1228         wc = wxConfigBase_Get()
1229         wc.WriteInt('autorefresh', conf.autoRefresh)
1230         wc.WriteInt('x', conf.x)
1231         wc.WriteInt('y', conf.y)
1232         wc.WriteInt('width', conf.width)
1233         wc.WriteInt('height', conf.height)
1234         wc.WriteInt('embedPanel', conf.embedPanel)
1235         wc.WriteInt('showTools', conf.showTools)
1236         if not conf.embedPanel:
1237             wc.WriteInt('panelX', conf.panelX)
1238             wc.WriteInt('panelY', conf.panelY)
1239         wc.WriteInt('sashPos', conf.sashPos)
1240         wc.WriteInt('panelWidth', conf.panelWidth)
1241         wc.WriteInt('panelHeight', conf.panelHeight)
1242         wc.WriteInt('nopanic', True)
1243         wc.Write('recentFiles', '|'.join(conf.recentfiles.values()[-5:]))
1244         wc.Flush()
1245
1246 def main():
1247     app = App(0, useBestVisual=False)
1248     #app.SetAssertMode(wxPYAPP_ASSERT_LOG)
1249     app.MainLoop()
1250     app.OnExit()
1251     global conf
1252     del conf
1253
1254 if __name__ == '__main__':
1255     main()