# Name: xrced.py # Purpose: XRC editor, main module # Author: Roman Rolinsky # Created: 20.08.2001 from wxPython.wx import * from wxPython.xrc import * from wxPython.html import * import wxPython.lib.wxpTag from xml.dom import minidom import os import tempfile import images # String constants htmlHeader = '\n' htmlFooter = '\n' progname = 'XRCed' version = '0.0.3' # Local modules from xxx import * # Globals testWin = None testWinPos = wxDefaultPosition # 1 adds CMD command to Help menu debug = 1 if debug: import traceback import time # Set menu to list items. # Each menu command is a tuple (id, label, help) # submenus are lists [id, label, help, submenu] # and separators are any other type def SetMenu(m, list): for l in list: if type(l) == types.TupleType: apply(m.Append, l) elif type(l) == types.ListType: subMenu = wxMenu() SetMenu(subMenu, l[2:]) m.AppendMenu(wxNewId(), l[0], subMenu, l[1]) else: # separator m.AppendSeparator() # Properties panel class Panel(wxHtmlWindow): def __init__(self, parent, id): wxHtmlWindow.__init__(self, parent, id) self.SetBorders(5) self.SetFonts('', '', [8, 10, 12, 14, 16, 19, 24]) EVT_CHECKBOX(self, xxxObject.ID_CHECK_PARAMS, self.OnCheckParams) EVT_CHECKBOX(self, xxxChildContainer.ID_CHECK_PARAMS, self.OnCheckParams) self.modified = false def Clear(self): self.SetPage(htmlHeader + 'select a tree item on the left' + htmlFooter) def OnCheckParams(self, evt): selected = tree.GetSelection() xxx = tree.GetPyData(selected) if xxx.hasChild and evt.GetId() != xxxChildContainer.ID_CHECK_PARAMS: xxx = xxx.child # Set current object param = evt.GetEventObject().GetName()[6:] if xxx.hasChild: w = GetRegistered('_'+param) else: w = GetRegistered(param) elem = xxx.element if evt.IsChecked(): # Ad new text node in order of allParams w.SetValue('') textElem = tree.dom.createElement(param) textNode = tree.dom.createTextNode('') textElem.appendChild(textNode) # Find place to put new element: first present element after param found = false for p in xxx.allParams[xxx.allParams.index(param) + 1:]: # Content params don't have same type if xxx.params.has_key(p) and p != 'content': found = true break if found: nextTextElem = xxx.params[p].parentNode elem.insertBefore(textElem, nextTextElem) else: elem.appendChild(textElem) xxx.params[param] = textNode else: # Remove parameter element and following text node textElem = xxx.params[param].parentNode newline = textElem.nextSibling if newline and newline.nodeType == minidom.Node.TEXT_NODE: elem.removeChild(newline) elem.removeChild(textElem) del xxx.params[param] w.SetValue('') w.Enable(evt.IsChecked()) # Set modified flas self.SetModified(true) # If some parameter was changed def IsModified(self): return self.modified def SetModified(self, value): self.modified = value class HightLightBox: def __init__(self, pos, size): w = testWin.panel l1 = wxWindow(w, -1, pos, wxSize(size.x, 2)) l1.SetBackgroundColour(wxRED) l2 = wxWindow(w, -1, pos, wxSize(2, size.y)) l2.SetBackgroundColour(wxRED) l3 = wxWindow(w, -1, wxPoint(pos.x + size.x - 2, pos.y), wxSize(2, size.y)) l3.SetBackgroundColour(wxRED) l4 = wxWindow(w, -1, wxPoint(pos.x, pos.y + size.y - 2), wxSize(size.x, 2)) l4.SetBackgroundColour(wxRED) self.lines = [l1, l2, l3, l4] # Move highlight to a new position def Replace(self, pos, size): self.lines[0].SetDimensions(pos.x, pos.y, size.x, 2, wxSIZE_ALLOW_MINUS_ONE) self.lines[1].SetDimensions(pos.x, pos.y, 2, size.y, wxSIZE_ALLOW_MINUS_ONE) self.lines[2].SetDimensions(pos.x + size.x - 2, pos.y, 2, size.y, wxSIZE_ALLOW_MINUS_ONE) self.lines[3].SetDimensions(pos.x, pos.y + size.y - 2, size.x, 2, wxSIZE_ALLOW_MINUS_ONE) # Remove it def Remove(self): map(wxWindow.Destroy, self.lines) testWin.highLight = None class MemoryFile: def __init__(self, name): self.name = name self.buffer = '' def write(self, data): self.buffer = self.buffer + data.encode() def close(self): f = open(self.name, 'w') f.write(self.buffer) f.close() # !!! memory FS will work someday #self.file = wxMemoryFSHandler_AddFile(self.name, self.buffer) class XML_Tree(wxTreeCtrl): def __init__(self, parent, id): wxTreeCtrl.__init__(self, parent, id) self.SetBackgroundColour(wxColour(224, 248, 224)) EVT_TREE_SEL_CHANGED(self, self.GetId(), self.OnSelChanged) EVT_TREE_ITEM_ACTIVATED(self, self.GetId(), self.OnItemActivated) EVT_RIGHT_DOWN(self, self.OnRightDown) self.needUpdate = false self.pendingHighLight = None self.ctrl = false self.dom = None # Create image list il = wxImageList(16, 16, true) xxxPanel.image = il.AddIcon( wxIconFromXPMData(images.getTreePanelData()) ) xxxDialog.image = il.AddIcon( wxIconFromXPMData(images.getTreeDialogData()) ) xxxFrame.image = il.AddIcon( wxIconFromXPMData(images.getTreeFrameData()) ) xxxMenuBar.image = il.AddIcon( wxIconFromXPMData(images.getTreeMenuBarData()) ) xxxMenu.image = il.AddIcon( wxIconFromXPMData(images.getTreeMenuData()) ) xxxSizer.imageH = il.AddIcon( wxIconFromXPMData(images.getTreeSizerHData()) ) xxxSizer.imageV = il.AddIcon( wxIconFromXPMData(images.getTreeSizerVData()) ) xxxStaticBoxSizer.imageH = il.AddIcon( wxIconFromXPMData(images.getTreeStaticBoxSizerHData()) ) xxxStaticBoxSizer.imageV = il.AddIcon( wxIconFromXPMData(images.getTreeStaticBoxSizerVData()) ) xxxGridSizer.image = il.AddIcon( wxIconFromXPMData(images.getTreeSizerGridData()) ) xxxFlexGridSizer.image = il.AddIcon( wxIconFromXPMData(images.getTreeSizerFlexGridData()) ) self.il = il self.SetImageList(il) # !!! temporary solution for GetOldItem problem def Unselect(self): self.selection = wxTreeItemId() wxTreeCtrl.Unselect(self) def GetSelection(self): return self.selection def ExpandAll(self, item): if self.ItemHasChildren(item): self.Expand(item) i, cookie = self.GetFirstChild(item, 0) children = [] while i.IsOk(): children.append(i) i, cookie = self.GetNextChild(item, cookie) for i in children: self.ExpandAll(i) def SetData(self, dom): self.dom = dom # Find 'resource' child, add it's children self.mainNode = dom.getElementsByTagName('resource')[0] nodes = self.mainNode.childNodes[:] for node in nodes: if IsObject(node): self.AddNode(self.GetRootItem(), None, node) else: self.mainNode.removeChild(node) node.unlink() self.Unselect() # Add tree item for given parent item if node is DOM element node with # 'object' tag. xxxParent is parent xxx object def AddNode(self, itemParent, xxxParent, node): # Set item data to current node xxx = MakeXXXFromDOM(xxxParent, node) treeObj = xxx.treeObject() # Append tree item item = self.AppendItem(itemParent, treeObj.treeName(), image=treeObj.treeImage(), data=wxTreeItemData(xxx)) # Try to find children objects if treeObj.hasChildren: nodes = treeObj.element.childNodes[:] for n in nodes: if IsObject(n): self.AddNode(item, treeObj, n) elif n.nodeType != minidom.Node.ELEMENT_NODE: treeObj.element.removeChild(n) n.unlink() # Remove leaf of tree, return it's data object def RemoveLeaf(self, leaf): xxx = self.GetPyData(leaf) node = xxx.element parent = node.parentNode parent.removeChild(node) self.Delete(leaf) # Update view? #if testWin and self.GetItemAncestor(leaf) == testWin.item: # if testWin.highLight: # testWin.highLight.Remove() # self.needUpdate = true return node # Find position relative to the top-level window def FindNodePos(self, item): itemParent = self.GetItemParent(item) if itemParent == self.GetRootItem(): return wxPoint(0, 0) obj = self.FindNodeObject(item) # Find first ancestor which is a wxWindow (not a sizer) winParent = itemParent while self.GetPyData(winParent).isSizer: winParent = self.GetItemParent(winParent) parentPos = self.FindNodePos(winParent) return parentPos + obj.GetPosition() # Find window (or sizer) corresponding to a tree item. def FindNodeObject(self, item): itemParent = self.GetItemParent(item) # If top-level, return testWin (or panel if wxFrame) if itemParent == self.GetRootItem(): return testWin.panel xxx = self.GetPyData(item).treeObject() parentWin = self.FindNodeObject(itemParent) # Top-level sizer? return window's sizer if xxx.isSizer and isinstance(parentWin, wxWindowPtr): return parentWin.GetSizer() # Otherwise get parent's object and it's child n = 0 # index of sibling prev = self.GetPrevSibling(item) while prev.IsOk(): prev = self.GetPrevSibling(prev) n += 1 child = parentWin.GetChildren()[n] # Return window or sizer for sizer items if child.GetClassName() == 'wxSizerItem': if child.IsWindow(): child = child.GetWindow() elif child.IsSizer(): child = child.GetSizer() # Test for notebook sizers if isinstance(child, wxNotebookSizerPtr): child = child.GetNotebook() return child def OnSelChanged(self, evt): # Apply changes # !!! problem with wxGTK #oldItem = evt.GetOldItem() oldItem = self.selection if oldItem.IsOk(): xxx = self.GetPyData(oldItem) # If some data was modified, apply changes if xxx: if panel.IsModified(): self.Apply(xxx, oldItem) #if conf.autoRefresh: if testWin and testWin.highLight: testWin.highLight.Remove() self.needUpdate = true # Generate HTML view item = evt.GetItem() self.selection = item # !!! fix xxx = self.GetPyData(item) html = htmlHeader # List of parameters tuples (parameter, isDefined) if not xxx: # root item html += 'this item has no properties' + htmlFooter panel.SetPage(html) if testWin and testWin.highLight: testWin.highLight.Remove() return # Normal nodes ClearRegister() # empty register html += xxx.generateHtml() html += htmlFooter panel.SetPage(html) # Set values, checkboxes to false, disable defaults if xxx.hasChild: prefix = '_' else: prefix = '' for param in xxx.allParams: if xxx.params.has_key(param): if param == 'content': value = [] for text in xxx.params[param]: value.append(str(text.data)) # convert from unicode else: value = xxx.params[param].data GetRegistered(prefix + param).SetValue(value) if not param in xxx.required: panel.FindWindowByName('check_' + param).SetValue(true) else: GetRegistered(prefix + param).Enable(false) # Same for the child of sizeritem if xxx.hasChild: xxx = xxx.child for param in xxx.allParams: if xxx.params.has_key(param): if param == 'content': value = [] for text in xxx.params[param]: value.append(str(text.data)) # convert from unicode else: value = xxx.params[param].data GetRegistered(param).SetValue(value) if not param in xxx.required: panel.FindWindowByName('check_' + param).SetValue(true) else: GetRegistered(param).Enable(false) # Clear flag panel.SetModified(false) # Hightlighting is done in OnIdle tree.pendingHighLight = item # Find top-level parent def GetItemAncestor(self, item): while self.GetItemParent(item) != self.GetRootItem(): item = self.GetItemParent(item) return item # Highlight selected item def HighLight(self, item): self.pendingHighLight = None if not testWin or self.GetPyData(testWin.item).className \ not in ['wxDialog', 'wxPanel', 'wxFrame']: return # Top-level does not have highlight if item == testWin.item: if testWin.highLight: testWin.highLight.Remove() return # If a control from another window is selected, remove highlight if self.GetItemAncestor(item) != testWin.item and testWin.highLight: testWin.highLight.Remove() return # Get window/sizer object obj, pos = self.FindNodeObject(item), self.FindNodePos(item) size = obj.GetSize() # For notebook, select item's page. # For children of page, nothing happens (too much work) if isinstance(self.GetPyData(item).parent, xxxNotebook): notebook = self.FindNodeObject(self.GetItemParent(item)) # Find position n = 0 prev = self.GetPrevSibling(item) while prev.IsOk(): n += 1 prev = self.GetPrevSibling(prev) notebook.SetSelection(n) # Highlight try: # finally I use exceptions testWin.highLight.Replace(pos, size) except AttributeError: testWin.highLight = HightLightBox(pos, size) testWin.highLight.item = item # Double-click def OnItemActivated(self, evt): item = evt.GetItem() xxx = self.GetPyData(item) if not xxx: return # if root selected, do nothing if panel.IsModified(): self.Apply(xxx, item) # apply changes self.CreateTestWin(item) # (re)create test window def CreateTestWin(self, node): global testWin # Create a window with this resource xxx = self.GetPyData(node).treeObject() if not xxx: return # if root selected, do nothing # If noname element, display error if not xxx.hasName or not xxx.name: wxLogError("Can't display a noname element") return # Close old window, remember where it was highLight = None if testWin: pos = testWin.GetPosition() if node == testWin.item: # Remember highlight if same top-level window if testWin.highLight: highLight = testWin.highLight.item # !!! if 0 is removed, refresh is broken (notebook not deleted?) if 0 and xxx.className == 'wxPanel': if testWin.highLight: testWin.pendingHighLight = highLight testWin.highLight.Remove() testWin.panel.Destroy() testWin.panel = None else: testWin.Destroy() testWin = None else: testWin.Destroy() testWin = None else: pos = testWinPos # Save in temporary file before activating memFile = MemoryFile(tempfile.mktemp('xrc')) #memFile = MemoryFile('core.xrc') # to write debug file # Create partial XML file - faster for big files dom = minidom.Document() mainNode = dom.createElement('resource') dom.appendChild(mainNode) # Remove temporarily from old parent elem = xxx.element parent = elem.parentNode next = elem.nextSibling parent.replaceChild(self.dummyNode, elem) # Append to new DOM, write it mainNode.appendChild(elem) dom.writexml(memFile) # Put back in place mainNode.removeChild(elem) dom.unlink() parent.replaceChild(elem, self.dummyNode) memFile.close() # write to wxMemoryFS res = wxXmlResource('') res.Load(memFile.name) if xxx.className == 'wxFrame': # Create new frame testWin = wxPreFrame() res.LoadFrame(testWin, frame, xxx.name) testWin.panel = testWin testWin.SetPosition(pos) testWin.Show(true) elif xxx.className == 'wxPanel': # Create new frame if not testWin: testWin = wxFrame(frame, -1, 'Panel: ' + xxx.name, pos=pos) testWin.panel = res.LoadPanel(testWin, xxx.name) testWin.SetSize(testWin.panel.GetSize()) testWin.Show(true) elif xxx.className == 'wxDialog': # Create new frame testWin = res.LoadDialog(None, xxx.name) testWin.panel = testWin testWin.SetPosition(pos) testWin.Show(true) elif xxx.className == 'wxMenuBar': testWin = wxFrame(frame, -1, 'MenuBar: ' + xxx.name, pos=pos) # Set status bar to display help testWin.CreateStatusBar() testWin.menuBar = res.LoadMenuBar(xxx.name) testWin.SetMenuBar(testWin.menuBar) testWin.Show(true) else: wxLogMessage('No view for this element yet') return os.unlink(memFile.name) # remove tmp file testWin.item = node testWin.Connect(testWin.GetId(), -1, wxEVT_CLOSE_WINDOW, self.OnCloseTestWin) testWin.highLight = None if highLight and not tree.pendingHighLight: self.HighLight(highLight) def OnCloseTestWin(self, evt): global testWin, testWinPos testWinPos = testWin.GetPosition() testWin.Destroy() testWin = None evt.Skip() # True if next item should be inserted after current (vs. appended to it) def NeedInsert(self, item): xxx = self.GetPyData(item) if not xxx: return false # root item if self.ctrl: return true # if Ctrl pressed, always insert if xxx.hasChildren and not self.ItemHasChildren(item): return false return not (self.IsExpanded(item) and self.ItemHasChildren(item)) # Pull-down def OnRightDown(self, evt): # Setup menu pullDownMenu.menu = wxMenu() item = self.GetSelection() if not item.IsOk(): pullDownMenu.menu.Append(pullDownMenu.ID_EXPAND, 'Expand', 'Expand tree') else: self.ctrl = evt.ControlDown() # save Ctrl state m = wxMenu() # create menu if item != self.GetRootItem(): needInsert = self.NeedInsert(item) if item == self.GetRootItem() or \ self.GetItemParent(item) == self.GetRootItem() and needInsert: m.Append(pullDownMenu.ID_NEW_PANEL, 'Panel', 'Create panel') m.Append(pullDownMenu.ID_NEW_DIALOG, 'Dialog', 'Create dialog') m.Append(pullDownMenu.ID_NEW_FRAME, 'Frame', 'Create frame') m.AppendSeparator() m.Append(pullDownMenu.ID_NEW_MENU_BAR, 'MenuBar', 'Create menu bar') m.Append(pullDownMenu.ID_NEW_MENU, 'Menu', 'Create menu') else: xxx = self.GetPyData(item) if xxx.__class__ == xxxMenuBar: m.Append(pullDownMenu.ID_NEW_MENU, 'Menu', 'Create menu') elif xxx.__class__ in [xxxMenu, xxxMenuItem]: SetMenu(m, pullDownMenu.menuControls) else: SetMenu(m, pullDownMenu.controls) if not (xxx.isSizer or \ xxx.parent and xxx.parent.isSizer): m.Enable(pullDownMenu.ID_NEW_SPACER, false) # Select correct label for create menu if item == self.GetRootItem(): pullDownMenu.menu.AppendMenu(wxNewId(), 'Create', m, 'Create top-level object') else: if not needInsert: pullDownMenu.menu.AppendMenu(wxNewId(), 'Create child', m, 'Create child object') else: pullDownMenu.menu.AppendMenu(wxNewId(), 'Create Sibling', m, 'Create sibling of selected object') pullDownMenu.menu.AppendSeparator() pullDownMenu.menu.Append(wxID_CUT, 'Cut', 'Cut to the clipboard') pullDownMenu.menu.Append(wxID_COPY, 'Copy', 'Copy to the clipboard') pullDownMenu.menu.Append(wxID_PASTE, 'Paste', 'Paste from the clipboard') pullDownMenu.menu.Append(pullDownMenu.ID_DELETE, 'Delete', 'Delete object') if item.IsOk() and self.ItemHasChildren(item): pullDownMenu.menu.AppendSeparator() pullDownMenu.menu.Append(pullDownMenu.ID_EXPAND, 'Expand', 'Expand subtree') self.PopupMenu(pullDownMenu.menu, evt.GetPosition()) pullDownMenu.menu.Destroy() pullDownMenu.menu = None # Clear tree def Clear(self): self.DeleteAllItems() # Add minimal structure root = self.AddRoot('XML tree') self.Unselect() if self.dom: self.dom.unlink() self.dom = minidom.Document() self.dummyNode = self.dom.createComment('dummy node') # Create main node self.mainNode = self.dom.createElement('resource') self.dom.appendChild(self.mainNode) # Apply changes def Apply(self, xxx, item): if not xxx: return # !!! Save undo info if xxx.undo: xxx.undo.unlink() xxx.undo = xxx.element.cloneNode(false) if xxx.hasName: name = GetRegistered('name').GetValue() if xxx.name != name: xxx.name = name xxx.element.setAttribute('name', name) self.SetItemText(item, xxx.treeName()) if xxx.hasChild: prefix = '_' else: prefix = '' for param, data in xxx.params.items(): value = GetRegistered(prefix + param).GetValue() if param == 'content': # If number if items is not the same, recreate children if len(value) != len(data): elem = xxx.element.getElementsByTagName('content')[0] for n in elem.childNodes: elem.removeChild(n) data = [] for str in value: itemElem = tree.dom.createElement('item') itemText = tree.dom.createTextNode(str) itemElem.appendChild(itemText) elem.appendChild(itemElem) data.append(itemText) xxx.params[param] = data else: for i in range(len(value)): data[i].data = value[i] else: data.data = value if xxx.hasChild: self.Apply(xxx.child, item) else: # Change tree icon for sizers if isinstance(xxx, xxxBoxSizer): self.SetItemImage(item, xxx.treeImage()) # Set global modified state frame.modified = true class Frame(wxFrame): def __init__(self, size): wxFrame.__init__(self, None, -1, '', size=size) self.CreateStatusBar() # Make menus menuBar = wxMenuBar() menu = wxMenu() menu.Append(wxID_NEW, '&New\tCtrl-N', 'New file') menu.Append(wxID_OPEN, '&Open...\tCtrl-O', 'Open XRC file') menu.Append(wxID_SAVE, '&Save\tCtrl-S', 'Save XRC file') menu.Append(wxID_SAVEAS, 'Save &As...', 'Save XRC file under different name') menu.AppendSeparator() menu.Append(wxID_EXIT, '&Quit\tCtrl-Q', 'Exit application') menuBar.Append(menu, '&File') menu = wxMenu() menu.Append(wxID_UNDO, '&Undo\tCtrl-Z', 'Undo') menu.Append(wxID_REDO, '&Redo\tCtrl-R', 'Redo') menu.AppendSeparator() menu.Append(wxID_CUT, 'Cut\tCtrl-X', 'Cut to the clipboard') menu.Append(wxID_COPY, '&Copy\tCtrl-C', 'Copy to the clipboard') menu.Append(wxID_PASTE, '&Paste\tCtrl-V', 'Paste from the clipboard') self.ID_DELETE = wxNewId() menu.Append(self.ID_DELETE, '&Delete\tCtrl-D', 'Delete object') menuBar.Append(menu, '&Edit') menu = wxMenu() self.ID_REFRESH = wxNewId() menu.Append(self.ID_REFRESH, '&Refresh\tCtrl-R', 'Refresh view') self.ID_AUTO_REFRESH = wxNewId() menu.Append(self.ID_AUTO_REFRESH, '&Auto-refresh\tCtrl-A', 'Toggle auto-refresh mode', true) menu.Check(self.ID_AUTO_REFRESH, conf.autoRefresh) menuBar.Append(menu, '&View') menu = wxMenu() menu.Append(wxID_ABOUT, 'About...', 'About XCRed') if debug: self.ID_DEBUG_CMD = wxNewId() menu.Append(self.ID_DEBUG_CMD, 'CMD', 'Python command line') EVT_MENU(self, self.ID_DEBUG_CMD, self.OnDebugCMD) menuBar.Append(menu, '&Help') self.menuBar = menuBar self.SetMenuBar(menuBar) # Create toolbar tb = self.CreateToolBar()#wxTB_DOCKABLE | wxTB_FLAT) tb.SetToolBitmapSize((24,23)) tb.AddSimpleTool(wxID_NEW, images.getNewBitmap(), 'New', 'New file') tb.AddSimpleTool(wxID_OPEN, images.getOpenBitmap(), 'Open', 'Open file') tb.AddSimpleTool(wxID_SAVE, images.getSaveBitmap(), 'Save', 'Save file') tb.AddSeparator() tb.AddSimpleTool(wxID_CUT, images.getCutBitmap(), 'Cut', 'Cut') tb.AddSimpleTool(wxID_COPY, images.getCopyBitmap(), 'Copy', 'Copy') tb.AddSimpleTool(wxID_PASTE, images.getPasteBitmap(), 'Paste', 'Paste') tb.AddSeparator() tb.AddSimpleTool(self.ID_REFRESH, images.getRefreshBitmap(), 'Refresh', 'Refresh view') tb.AddSimpleTool(self.ID_AUTO_REFRESH, images.getAutoRefreshBitmap(), 'Auto-refresh', 'Toggle auto-refresh mode', true) tb.ToggleTool(self.ID_AUTO_REFRESH, conf.autoRefresh) self.tb = tb tb.Realize() # File EVT_MENU(self, wxID_NEW, self.OnNew) EVT_MENU(self, wxID_OPEN, self.OnOpen) EVT_MENU(self, wxID_SAVE, self.OnSaveOrSaveAs) EVT_MENU(self, wxID_SAVEAS, self.OnSaveOrSaveAs) EVT_MENU(self, wxID_EXIT, self.OnExit) # Edit EVT_MENU(self, wxID_UNDO, self.OnUndo) EVT_MENU(self, wxID_REDO, self.OnRedo) EVT_MENU(self, wxID_CUT, self.OnCut) EVT_MENU(self, wxID_COPY, self.OnCopy) EVT_MENU(self, wxID_PASTE, self.OnPaste) EVT_MENU(self, self.ID_DELETE, self.OnDelete) # View EVT_MENU(self, self.ID_REFRESH, self.OnRefresh) EVT_MENU(self, self.ID_AUTO_REFRESH, self.OnAutoRefresh) # Help EVT_MENU(self, wxID_ABOUT, self.OnAbout) # Update events EVT_UPDATE_UI(self, wxID_CUT, self.OnUpdateUI) EVT_UPDATE_UI(self, wxID_COPY, self.OnUpdateUI) EVT_UPDATE_UI(self, wxID_PASTE, self.OnUpdateUI) EVT_UPDATE_UI(self, self.ID_DELETE, self.OnUpdateUI) # Build interface splitter = wxSplitterWindow(self, -1) # Create tree global tree tree = XML_Tree(splitter, -1) sys.modules['xxx'].tree = tree # Create panel for parameters global panel #panel = wxPanel(self, -1) # Sizer for static box #sizer = wxBoxSizer() panel = Panel(splitter, -1) sys.modules['params'].panel = panel #sizer.Add(panel, 1, wxEXPAND) #box = wxStaticBox(panel, -1, 'Parameters') #boxSizer = wxStaticBoxSizer(box) #boxSizer.Add(wxButton(panel, -1, 'BUTT ON')) #sizer.Add(boxSizer, 1, wxEXPAND | wxALL, 10) #panel.SetAutoLayout(true) #panel.SetSizer(sizer) # Set plitter windows splitter.SplitVertically(tree, panel, 200) #topSizer = wxBoxSizer() #topSizer.Add(splitter, 1, wxEXPAND) #self.SetAutoLayout(true) #self.SetSizer(topSizer) # Init pull-down menu data class MenuData: pass global pullDownMenu pullDownMenu = MenuData() pullDownMenu.menu = None pullDownMenu.ID_NEW_PANEL = wxNewId() pullDownMenu.ID_NEW_DIALOG = wxNewId() pullDownMenu.ID_NEW_FRAME = wxNewId() pullDownMenu.ID_NEW_MENU_BAR = wxNewId() pullDownMenu.ID_NEW_MENU = wxNewId() pullDownMenu.ID_NEW_STATIC_TEXT = wxNewId() pullDownMenu.ID_NEW_TEXT_CTRL = wxNewId() pullDownMenu.ID_NEW_BUTTON = wxNewId() pullDownMenu.ID_NEW_BITMAP_BUTTON = wxNewId() pullDownMenu.ID_NEW_RADIO_BUTTON = wxNewId() pullDownMenu.ID_NEW_SPIN_BUTTON = wxNewId() pullDownMenu.ID_NEW_STATIC_BOX = wxNewId() pullDownMenu.ID_NEW_CHECK_BOX = wxNewId() pullDownMenu.ID_NEW_RADIO_BOX = wxNewId() pullDownMenu.ID_NEW_COMBO_BOX = wxNewId() pullDownMenu.ID_NEW_LIST_BOX = wxNewId() pullDownMenu.ID_NEW_STATIC_LINE = wxNewId() pullDownMenu.ID_NEW_CHOICE = wxNewId() pullDownMenu.ID_NEW_SLIDER = wxNewId() pullDownMenu.ID_NEW_GAUGE = wxNewId() pullDownMenu.ID_NEW_SCROLL_BAR = wxNewId() pullDownMenu.ID_NEW_TREE_CTRL = wxNewId() pullDownMenu.ID_NEW_LIST_CTRL = wxNewId() pullDownMenu.ID_NEW_CHECK_LIST = wxNewId() pullDownMenu.ID_NEW_NOTEBOOK = wxNewId() pullDownMenu.ID_NEW_HTML_WINDOW = wxNewId() pullDownMenu.ID_NEW_CALENDAR = wxNewId() pullDownMenu.ID_NEW_BOX_SIZER = wxNewId() pullDownMenu.ID_NEW_STATIC_BOX_SIZER = wxNewId() pullDownMenu.ID_NEW_GRID_SIZER = wxNewId() pullDownMenu.ID_NEW_FLEX_GRID_SIZER = wxNewId() pullDownMenu.ID_NEW_SPACER = wxNewId() pullDownMenu.ID_NEW_MENU = wxNewId() pullDownMenu.ID_NEW_MENU_ITEM = wxNewId() pullDownMenu.ID_NEW_SEPARATOR = wxNewId() pullDownMenu.ID_NEW_LAST = wxNewId() pullDownMenu.ID_DELETE = self.ID_DELETE pullDownMenu.ID_EXPAND = wxNewId() EVT_MENU_RANGE(self, pullDownMenu.ID_NEW_PANEL, pullDownMenu.ID_NEW_LAST, self.OnCreate) EVT_MENU(self, pullDownMenu.ID_EXPAND, self.OnExpand) # We connect to tree, but process in frame EVT_MENU_HIGHLIGHT_ALL(tree, self.OnPullDownHighlight) # Mapping from IDs to element names self.createMap = { pullDownMenu.ID_NEW_PANEL: 'wxPanel', pullDownMenu.ID_NEW_DIALOG: 'wxDialog', pullDownMenu.ID_NEW_FRAME: 'wxFrame', pullDownMenu.ID_NEW_MENU_BAR: 'wxMenuBar', pullDownMenu.ID_NEW_MENU: 'wxMenu', pullDownMenu.ID_NEW_STATIC_TEXT: 'wxStaticText', pullDownMenu.ID_NEW_TEXT_CTRL: 'wxTextCtrl', pullDownMenu.ID_NEW_BUTTON: 'wxButton', pullDownMenu.ID_NEW_BITMAP_BUTTON: 'wxBitmapButton', pullDownMenu.ID_NEW_RADIO_BUTTON: 'wxRadioButton', pullDownMenu.ID_NEW_SPIN_BUTTON: 'wxSpinButton', pullDownMenu.ID_NEW_STATIC_BOX: 'wxStaticBox', pullDownMenu.ID_NEW_CHECK_BOX: 'wxCheckBox', pullDownMenu.ID_NEW_RADIO_BOX: 'wxRadioBox', pullDownMenu.ID_NEW_COMBO_BOX: 'wxComboBox', pullDownMenu.ID_NEW_LIST_BOX: 'wxListBox', pullDownMenu.ID_NEW_STATIC_LINE: 'wxStaticLine', pullDownMenu.ID_NEW_CHOICE: 'wxChoice', pullDownMenu.ID_NEW_SLIDER: 'wxSlider', pullDownMenu.ID_NEW_GAUGE: 'wxGauge', pullDownMenu.ID_NEW_SCROLL_BAR: 'wxScrollBar', pullDownMenu.ID_NEW_TREE_CTRL: 'wxTreeCtrl', pullDownMenu.ID_NEW_LIST_CTRL: 'wxListCtrl', pullDownMenu.ID_NEW_CHECK_LIST: 'wxCheckList', pullDownMenu.ID_NEW_NOTEBOOK: 'wxNotebook', pullDownMenu.ID_NEW_HTML_WINDOW: 'wxHtmlWindow', pullDownMenu.ID_NEW_CALENDAR: 'wxCalendar', pullDownMenu.ID_NEW_BOX_SIZER: 'wxBoxSizer', pullDownMenu.ID_NEW_STATIC_BOX_SIZER: 'wxStaticBoxSizer', pullDownMenu.ID_NEW_GRID_SIZER: 'wxGridSizer', pullDownMenu.ID_NEW_FLEX_GRID_SIZER: 'wxFlexGridSizer', pullDownMenu.ID_NEW_SPACER: 'spacer', pullDownMenu.ID_NEW_MENU: 'wxMenu', pullDownMenu.ID_NEW_MENU_ITEM: 'wxMenuItem', pullDownMenu.ID_NEW_SEPARATOR: 'separator', } pullDownMenu.controls = [ ['control', 'Various controls', (pullDownMenu.ID_NEW_STATIC_TEXT, 'Label', 'Create static label'), (pullDownMenu.ID_NEW_STATIC_LINE, 'Line', 'Create static line'), (pullDownMenu.ID_NEW_TEXT_CTRL, 'TextBox', 'Create text box control'), (pullDownMenu.ID_NEW_CHOICE, 'Choice', 'Create choice control'), (pullDownMenu.ID_NEW_SLIDER, 'Slider', 'Create slider control'), (pullDownMenu.ID_NEW_GAUGE, 'Gauge', 'Create gauge control'), (pullDownMenu.ID_NEW_SCROLL_BAR, 'ScrollBar', 'Create scroll bar'), (pullDownMenu.ID_NEW_TREE_CTRL, 'TreeCtrl', 'Create tree control'), (pullDownMenu.ID_NEW_LIST_CTRL, 'ListCtrl', 'Create list control'), (pullDownMenu.ID_NEW_CHECK_LIST, 'CheckList', 'Create check list control'), (pullDownMenu.ID_NEW_HTML_WINDOW, 'HtmlWindow', 'Create HTML window'), (pullDownMenu.ID_NEW_CALENDAR, 'Calendar', 'Create calendar control'), (pullDownMenu.ID_NEW_PANEL, 'Panel', 'Create panel'), (pullDownMenu.ID_NEW_NOTEBOOK, 'Notebook', 'Create notebook control'), ], ['button', 'Buttons', (pullDownMenu.ID_NEW_BUTTON, 'Button', 'Create button'), (pullDownMenu.ID_NEW_BITMAP_BUTTON, 'BitmapButton', 'Create bitmap button'), (pullDownMenu.ID_NEW_RADIO_BUTTON, 'RadioButton', 'Create radio button'), (pullDownMenu.ID_NEW_SPIN_BUTTON, 'SpinButton', 'Create spin button'), ], ['box', 'Boxes', (pullDownMenu.ID_NEW_STATIC_BOX, 'StaticBox', 'Create static box'), (pullDownMenu.ID_NEW_CHECK_BOX, 'CheckBox', 'Create check box'), (pullDownMenu.ID_NEW_RADIO_BOX, 'RadioBox', 'Create radio box'), (pullDownMenu.ID_NEW_COMBO_BOX, 'ComboBox', 'Create combo box'), (pullDownMenu.ID_NEW_LIST_BOX, 'ListBox', 'Create list box'), ], ['sizer', 'Sizers', (pullDownMenu.ID_NEW_BOX_SIZER, 'BoxSizer', 'Create box sizer'), (pullDownMenu.ID_NEW_STATIC_BOX_SIZER, 'StaticBoxSizer', 'Create static box sizer'), (pullDownMenu.ID_NEW_GRID_SIZER, 'GridSizer', 'Create grid sizer'), (pullDownMenu.ID_NEW_FLEX_GRID_SIZER, 'FlexGridSizer', 'Create flexgrid sizer'), (pullDownMenu.ID_NEW_SPACER, 'Spacer', 'Create spacer'), ] ] pullDownMenu.menuControls = [ (pullDownMenu.ID_NEW_MENU, 'Menu', 'Create menu'), (pullDownMenu.ID_NEW_MENU_ITEM, 'MenuItem', 'Create menu item'), (pullDownMenu.ID_NEW_SEPARATOR, 'Separator', 'Create separator'), ] # Initialize self.Clear() # Other events EVT_IDLE(self, self.OnIdle) EVT_CLOSE(self, self.OnCloseWindow) def OnNew(self, evt): self.Clear() def OnOpen(self, evt): if not self.AskSave(): return dlg = wxFileDialog(self, 'Open', os.path.dirname(self.dataFile), '', '*.xrc', wxOPEN | wxCHANGE_DIR) if dlg.ShowModal() == wxID_OK: path = dlg.GetPath() self.SetStatusText('Loading...') wxYield() wxBeginBusyCursor() self.Open(path) wxEndBusyCursor() self.SetStatusText('Ready') dlg.Destroy() def OnSaveOrSaveAs(self, evt): if evt.GetId() == wxID_SAVEAS or not self.dataFile: if self.dataFile: defaultName = '' else: defaultName = 'UNTITLED.xrc' dlg = wxFileDialog(self, 'Save As', os.path.dirname(self.dataFile), defaultName, '*.xrc', wxSAVE | wxOVERWRITE_PROMPT | wxCHANGE_DIR) if dlg.ShowModal() == wxID_CANCEL: return path = dlg.GetPath() dlg.Destroy() else: path = self.dataFile self.SetStatusText('Saving...') wxYield() wxBeginBusyCursor() self.Save(path) self.dataFile = path wxEndBusyCursor() self.SetStatusText('Ready') def OnExit(self, evt): self.Close() def OnUndo(self, evt): print '*** being implemented' print self.lastOp, self.undo if self.lastOp == 'DELETE': parent, prev, elem = self.undo if prev.IsOk(): xxx = MakeXXXFromDOM(tree.GetPyData(parent).treeObject(), elem) item = tree.InsertItem( parent, prev, xxx.treeObject().className, data=wxTreeItemData(xxx) ) def OnRedo(self, evt): print '*** being implemented' def OnCut(self, evt): selected = tree.GetSelection() # Undo info self.lastOp = 'CUT' self.undo = [tree.GetItemParent(selected), tree.GetPrevSibling(selected)] # Delete testWin? global testWin if testWin: # If deleting top-level item, delete testWin if selected == testWin.item: testWin.Destroy() testWin = None else: # Remove highlight, update testWin if tree.GetItemAncestor(selected) == testWin.item: if testWin.highLight: testWin.highLight.Remove() tree.needUpdate = true self.clipboard = tree.RemoveLeaf(selected) tree.pendingHighLight = None tree.Unselect() panel.Clear() self.modified = true def OnCopy(self, evt): selected = tree.GetSelection() xxx = tree.GetPyData(selected) self.clipboard = xxx.element.cloneNode(true) def OnPaste(self, evt): selected = tree.GetSelection() appendChild = not tree.NeedInsert(selected) xxx = tree.GetPyData(selected) if not appendChild: # If has next item, insert, else append to parent nextItem = tree.GetNextSibling(selected) if nextItem.IsOk(): # Insert before nextItem parentLeaf = tree.GetItemParent(selected) else: # last child: change selected to parent appendChild = true selected = tree.GetItemParent(selected) # Expanded container (must have children) elif tree.IsExpanded(selected) and tree.ItemHasChildren(selected): appendChild = false nextItem = tree.GetFirstChild(selected, 0)[0] parentLeaf = selected # Parent should be tree element or None if appendChild: parent = tree.GetPyData(selected) else: parent = tree.GetPyData(parentLeaf) if parent and parent.hasChild: parent = parent.child # Create a copy of clipboard element elem = self.clipboard.cloneNode(true) # Tempopary xxx object to test things xxx = MakeXXXFromDOM(parent, elem) className = xxx.treeObject().className # Check parent and child relationships # Parent is sizer or notebook, child is of wrong class or # parent is normal window, child is child container: detach child isChildContainer = isinstance(xxx, xxxChildContainer) if not parent and isChildContainer or \ (parent.isSizer and not isinstance(xxx, xxxSizerItem)) or \ (isinstance(parent, xxxNotebook) and not isinstance(xxx, xxxNotebookPage)) or \ (not parent.isSizer and not isinstance(parent, xxxNotebook) and \ isChildContainer): if isChildContainer: elem.removeChild(xxx.child.element) # detach child elem.unlink() # delete child container elem = xxx.child.element # replace # This should help garbage collection (!!! maybe not needed?) xxx.child.parent = None xxx.child = None if parent: # Parent is sizer or notebook, child is not child container if parent.isSizer and not isChildContainer and \ not isinstance(xxx, xxxSpacer): # Create sizer item element sizerItemElem = MakeEmptyDOM('sizeritem') sizerItemElem.appendChild(elem) elem = sizerItemElem elif isinstance(parent, xxxNotebook) and not isChildContainer: pageElem = MakeEmptyDOM('notebookpage') pageElem.appendChild(elem) elem = pageElem xxx = MakeXXXFromDOM(parent, elem) # Figure out if we must append a new child or sibling if appendChild: if parent: node = parent.element else: node = tree.mainNode node.appendChild(elem) newItem = tree.AppendItem(selected, xxx.treeName(), image=xxx.treeImage(), data=wxTreeItemData(xxx)) else: node = tree.GetPyData(nextItem).element if parent: elemParent = parent.element else: elemParent = tree.mainNode elemParent.insertBefore(elem, node) # Inserting before is difficult, se we insert after or first child newItem = tree.InsertItem(parentLeaf, selected, xxx.treeName(), image=xxx.treeImage(), data=wxTreeItemData(xxx)) # Add children items if xxx.hasChildren: treeObj = xxx.treeObject() for n in treeObj.element.childNodes: if IsObject(n): tree.AddNode(newItem, treeObj, n) # Scroll to show new item tree.EnsureVisible(newItem) tree.SelectItem(newItem) if not tree.IsVisible(newItem): tree.ScrollTo(newItem) tree.Refresh() # Update view? if testWin and tree.GetItemAncestor(newItem) == testWin.item: if conf.autoRefresh: tree.needUpdate = true tree.pendingHighLight = newItem else: tree.pendingHighLight = None self.modified = true def OnDelete(self, evt): selected = tree.GetSelection() # Undo info self.lastOp = 'DELETE' self.undo = [tree.GetItemParent(selected), tree.GetPrevSibling(selected)] # Delete testWin? global testWin if testWin: # If deleting top-level item, delete testWin if selected == testWin.item: testWin.Destroy() testWin = None else: # Remove highlight, update testWin if tree.GetItemAncestor(selected) == testWin.item: if testWin.highLight: testWin.highLight.Remove() tree.needUpdate = true xnode = tree.RemoveLeaf(selected) self.undo.append(xnode.cloneNode(true)) xnode.unlink() tree.pendingHighLight = None tree.Unselect() panel.Clear() self.modified = true def OnRefresh(self, evt): # If modified, apply first selection = tree.GetSelection() if selection.IsOk(): xxx = tree.GetPyData(selection) if xxx and panel.IsModified(): tree.Apply(xxx, selection) if testWin: # (re)create tree.CreateTestWin(testWin.item) tree.needUpdate = false def OnAutoRefresh(self, evt): conf.autoRefresh = evt.IsChecked() self.menuBar.Check(self.ID_AUTO_REFRESH, conf.autoRefresh) self.tb.ToggleTool(self.ID_AUTO_REFRESH, conf.autoRefresh) def OnAbout(self, evt): dlg = wxMessageDialog(self, '%s %s\n\nRoman Rolinsky ' % \ (progname, version), 'About %s' % progname, wxOK | wxCENTRE) dlg.ShowModal() dlg.Destroy() # Simple emulation of python command line def OnDebugCMD(self, evt): while 1: try: exec raw_input('C:\> ') except EOFError: print '^D' break except: (etype, value, tb) =sys.exc_info() tblist =traceback.extract_tb(tb)[1:] msg =string.join(traceback.format_exception_only(etype, value) +traceback.format_list(tblist)) print msg def OnCreate(self, evt): selected = tree.GetSelection() appendChild = not tree.NeedInsert(selected) xxx = tree.GetPyData(selected) if not appendChild: # If has next item, insert, else append to parent nextItem = tree.GetNextSibling(selected) if nextItem.IsOk(): # Insert before nextItem parentLeaf = tree.GetItemParent(selected) else: # last child: change selected to parent appendChild = true selected = tree.GetItemParent(selected) # Expanded container (must have children) elif tree.IsExpanded(selected) and tree.ItemHasChildren(selected): appendChild = false nextItem = tree.GetFirstChild(selected, 0)[0] parentLeaf = selected # Parent should be tree element or None if appendChild: parent = tree.GetPyData(selected) else: parent = tree.GetPyData(parentLeaf) if parent and parent.hasChild: parent = parent.child # Create element className = self.createMap[evt.GetId()] xxx = MakeEmptyXXX(parent, className) # Figure out if we must append a new child or sibling elem = xxx.element if appendChild: if parent: node = parent.element else: node = tree.mainNode # Insert newline for debug purposes node.appendChild(tree.dom.createTextNode('\n')) node.appendChild(elem) newItem = tree.AppendItem(selected, xxx.treeName(), image=xxx.treeImage(), data=wxTreeItemData(xxx)) else: node = tree.GetPyData(nextItem).element if parent: elemParent = parent.element else: elemParent = tree.mainNode elemParent.insertBefore(tree.dom.createTextNode('\n'), node) elemParent.insertBefore(elem, node) # Inserting before is difficult, se we insert after or first child newItem = tree.InsertItem(parentLeaf, selected, xxx.treeName(), image=xxx.treeImage(), data=wxTreeItemData(xxx)) tree.EnsureVisible(newItem) tree.SelectItem(newItem) if not tree.IsVisible(newItem): tree.ScrollTo(newItem) tree.Refresh() # Update view? if testWin and tree.GetItemAncestor(newItem) == testWin.item: if conf.autoRefresh: tree.needUpdate = true tree.pendingHighLight = newItem else: tree.pendingHighLight = None def OnExpand(self, evt): if tree.GetSelection().IsOk(): tree.ExpandAll(tree.GetSelection()) else: tree.ExpandAll(tree.GetRootItem()) def OnPullDownHighlight(self, evt): menuId = evt.GetMenuId() help = '' if menuId != -1: help = pullDownMenu.GetHelpString(menuId) self.SetStatusText(help) def OnUpdateUI(self, evt): if evt.GetId() in [wxID_CUT, wxID_COPY, self.ID_DELETE]: enable = tree.GetSelection().IsOk() and \ tree.GetSelection() != tree.GetRootItem() evt.Enable(enable) elif evt.GetId() == wxID_PASTE: enable = self.clipboard != None evt.Enable(enable) def OnIdle(self, evt): if tree.needUpdate: if conf.autoRefresh: if testWin: # (re)create tree.CreateTestWin(testWin.item) tree.needUpdate = false elif tree.pendingHighLight: tree.HighLight(tree.pendingHighLight) evt.Skip() def OnCloseWindow(self, evt): if not self.AskSave(): return if testWin: testWin.Destroy() conf.width, conf.height = self.GetSize() evt.Skip() def Clear(self): self.dataFile = '' self.clipboard = None self.modified = false panel.SetModified(false) panel.Clear() tree.Clear() global testWin if testWin: testWin.Destroy() testWin = None self.SetTitle(progname) def Open(self, path): # Try to read the file try: open(path) self.Clear() # Build wx tree dom = minidom.parse(path) tree.SetData(dom) self.dataFile = path self.SetTitle(progname + ': ' + os.path.basename(path)) except: wxLogError('Error reading file: ' + path) def Save(self, path): try: memFile = MemoryFile(path) tree.dom.writexml(memFile) memFile.close() self.modified = false panel.SetModified(false) except: wxLogError('Error writing file: ' + path) def AskSave(self): if not (self.modified or panel.IsModified()): return true flags = wxICON_EXCLAMATION | wxYES_NO | wxCANCEL | wxCENTRE dlg = wxMessageDialog( self, 'File is modified. Save before exit?', 'Save before too late?', flags ) say = dlg.ShowModal() dlg.Destroy() if say == wxID_YES: self.OnSaveOrSaveAs(wxCommandEvent(wxID_SAVE)) # If save was successful, modified flag is unset if not self.modified: return true elif say == wxID_NO: self.modified = false panel.SetModified(false) return true return false class App(wxApp): def OnInit(self): self.SetAppName("xrced") # Settings global conf # !!! wxConfigBase_Get doesn't seem to work conf = wxConfig(style=wxCONFIG_USE_LOCAL_FILE) conf.autoRefresh = conf.ReadInt('autorefresh', true) size = conf.ReadInt('width', 800), conf.ReadInt('height', 600) # Add handlers wxFileSystem_AddHandler(wxMemoryFSHandler()) wxInitAllImageHandlers() # Create main frame global frame frame = self.frame = Frame(size) self.frame.Show(true) # Load resources from XRC file (!!! should be transformed to .py later) sys.modules['params'].frame = frame frame.res = wxXmlResource('') frame.res.Load(os.path.join(sys.path[0], 'xrced.xrc')) return true def OnExit(self): # Write config wc = wxConfigBase_Get() wc.WriteInt('autorefresh', conf.autoRefresh) wc.WriteInt('width', conf.width) wc.WriteInt('height', conf.height) wc.Flush() def main(): app = App() app.MainLoop() app.OnExit() if __name__ == '__main__': main()