Amebis/wxWidgets
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
03319b655c0a672e255a5b64e5caeed7c3e93fac
wxWidgets/wxPython/wx/tools/XRCed/tree.py
Roman Rolinsky 03319b655c 0.1.6-1 
Support for object_ref. CreateTestWin saves all DOM in temporary
memory file now, to allow reference resolution.

Tree changed to multiple selection (not really supported yet), so
after deleting Unselect can be used (otherwise parent item is
selected automatically which results in scrolling jumps - distracting).

!!! Not tested on wxMSW yet.


git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@35165 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
2005-08-09 19:27:27 +00:00

1091 lines
46 KiB
Python
Raw Blame History

# Name: tree.py
# Purpose: XRC editor, XML_tree class
# Author: Roman Rolinsky <rolinsky@mema.ucl.ac.be>
# Created: 02.12.2002
# RCS-ID: $Id$
from xxx import * # xxx imports globals and params
import types
import traceback
# Constant to define standart window name
STD_NAME = '_XRCED_T_W'
# Icons
import images
class MemoryFile:
def __init__(self, name):
self.name = name
self.buffer = ''
def write(self, data):
if g.currentEncoding:
encoding = g.currentEncoding
else:
encoding = wxGetDefaultPyEncoding()
try:
self.buffer += data.encode(encoding)
except UnicodeEncodeError:
self.buffer += data.encode(encoding, 'xmlcharrefreplace')
def close(self):
wxMemoryFSHandler_AddFile(self.name, self.buffer)
################################################################################
# Redefine writing to include encoding
class MyDocument(minidom.Document):
def __init__(self):
minidom.Document.__init__(self)
self.encoding = ''
def writexml(self, writer, indent="", addindent="", newl="", encoding=""):
if encoding: encdstr = 'encoding="%s"' % encoding
else: encdstr = ''
writer.write('<?xml version="1.0" %s?>\n' % encdstr)
for node in self.childNodes:
node.writexml(writer, indent, addindent, newl)
################################################################################
# Ids for menu commands
class ID_NEW:
PANEL = wxNewId()
DIALOG = wxNewId()
FRAME = wxNewId()
TOOL_BAR = wxNewId()
TOOL = wxNewId()
MENU_BAR = wxNewId()
MENU = wxNewId()
STATIC_TEXT = wxNewId()
TEXT_CTRL = wxNewId()
BUTTON = wxNewId()
BITMAP_BUTTON = wxNewId()
RADIO_BUTTON = wxNewId()
SPIN_BUTTON = wxNewId()
TOGGLE_BUTTON = wxNewId()
STATIC_BOX = wxNewId()
CHECK_BOX = wxNewId()
RADIO_BOX = wxNewId()
COMBO_BOX = wxNewId()
LIST_BOX = wxNewId()
STATIC_LINE = wxNewId()
STATIC_BITMAP = wxNewId()
CHOICE = wxNewId()
SLIDER = wxNewId()
GAUGE = wxNewId()
SCROLL_BAR = wxNewId()
TREE_CTRL = wxNewId()
LIST_CTRL = wxNewId()
CHECK_LIST = wxNewId()
NOTEBOOK = wxNewId()
SPLITTER_WINDOW = wxNewId()
SCROLLED_WINDOW = wxNewId()
HTML_WINDOW = wxNewId()
CALENDAR_CTRL = wxNewId()
GENERIC_DIR_CTRL = wxNewId()
SPIN_CTRL = wxNewId()
UNKNOWN = wxNewId()
WIZARD = wxNewId()
WIZARD_PAGE = wxNewId()
WIZARD_PAGE_SIMPLE = wxNewId()
STATUS_BAR = wxNewId()
BOX_SIZER = wxNewId()
STATIC_BOX_SIZER = wxNewId()
GRID_SIZER = wxNewId()
FLEX_GRID_SIZER = wxNewId()
GRID_BAG_SIZER = wxNewId()
STD_DIALOG_BUTTON_SIZER = wxNewId()
SPACER = wxNewId()
TOOL_BAR = wxNewId()
TOOL = wxNewId()
MENU = wxNewId()
MENU_ITEM = wxNewId()
SEPARATOR = wxNewId()
OK_BUTTON = wxNewId()
YES_BUTTON = wxNewId()
SAVE_BUTTON = wxNewId()
APPLY_BUTTON = wxNewId()
NO_BUTTON = wxNewId()
CANCEL_BUTTON = wxNewId()
HELP_BUTTON = wxNewId()
CONTEXT_HELP_BUTTON = wxNewId()
REF = wxNewId()
LAST = wxNewId()
class PullDownMenu:
ID_EXPAND = wxNewId()
ID_COLLAPSE = wxNewId()
ID_PASTE_SIBLING = wxNewId()
ID_TOOL_PASTE = wxNewId()
ID_SUBCLASS = wxNewId()
def __init__(self, parent):
self.ID_DELETE = parent.ID_DELETE
EVT_MENU_RANGE(parent, ID_NEW.PANEL, ID_NEW.LAST, parent.OnCreate)
EVT_MENU_RANGE(parent, 1000 + ID_NEW.PANEL, 1000 + ID_NEW.LAST, parent.OnReplace)
EVT_MENU(parent, self.ID_COLLAPSE, parent.OnCollapse)
EVT_MENU(parent, self.ID_EXPAND, parent.OnExpand)
EVT_MENU(parent, self.ID_PASTE_SIBLING, parent.OnPaste)
EVT_MENU(parent, self.ID_SUBCLASS, parent.OnSubclass)
# We connect to tree, but process in frame
EVT_MENU_HIGHLIGHT_ALL(g.tree, parent.OnPullDownHighlight)
# Mapping from IDs to element names
self.createMap = {
ID_NEW.PANEL: 'wxPanel',
ID_NEW.DIALOG: 'wxDialog',
ID_NEW.FRAME: 'wxFrame',
ID_NEW.WIZARD: 'wxWizard',
ID_NEW.WIZARD_PAGE: 'wxWizardPage',
ID_NEW.WIZARD_PAGE_SIMPLE: 'wxWizardPageSimple',
ID_NEW.TOOL_BAR: 'wxToolBar',
ID_NEW.TOOL: 'tool',
ID_NEW.MENU_BAR: 'wxMenuBar',
ID_NEW.MENU: 'wxMenu',
ID_NEW.MENU_ITEM: 'wxMenuItem',
ID_NEW.SEPARATOR: 'separator',
ID_NEW.STATIC_TEXT: 'wxStaticText',
ID_NEW.TEXT_CTRL: 'wxTextCtrl',
ID_NEW.BUTTON: 'wxButton',
ID_NEW.BITMAP_BUTTON: 'wxBitmapButton',
ID_NEW.RADIO_BUTTON: 'wxRadioButton',
ID_NEW.SPIN_BUTTON: 'wxSpinButton',
ID_NEW.TOGGLE_BUTTON: 'wxToggleButton',
ID_NEW.STATIC_BOX: 'wxStaticBox',
ID_NEW.CHECK_BOX: 'wxCheckBox',
ID_NEW.RADIO_BOX: 'wxRadioBox',
ID_NEW.COMBO_BOX: 'wxComboBox',
ID_NEW.LIST_BOX: 'wxListBox',
ID_NEW.STATIC_LINE: 'wxStaticLine',
ID_NEW.STATIC_BITMAP: 'wxStaticBitmap',
ID_NEW.CHOICE: 'wxChoice',
ID_NEW.SLIDER: 'wxSlider',
ID_NEW.GAUGE: 'wxGauge',
ID_NEW.SCROLL_BAR: 'wxScrollBar',
ID_NEW.TREE_CTRL: 'wxTreeCtrl',
ID_NEW.LIST_CTRL: 'wxListCtrl',
ID_NEW.CHECK_LIST: 'wxCheckListBox',
ID_NEW.NOTEBOOK: 'wxNotebook',
ID_NEW.SPLITTER_WINDOW: 'wxSplitterWindow',
ID_NEW.SCROLLED_WINDOW: 'wxScrolledWindow',
ID_NEW.HTML_WINDOW: 'wxHtmlWindow',
ID_NEW.CALENDAR_CTRL: 'wxCalendarCtrl',
ID_NEW.GENERIC_DIR_CTRL: 'wxGenericDirCtrl',
ID_NEW.SPIN_CTRL: 'wxSpinCtrl',
ID_NEW.BOX_SIZER: 'wxBoxSizer',
ID_NEW.STATIC_BOX_SIZER: 'wxStaticBoxSizer',
ID_NEW.GRID_SIZER: 'wxGridSizer',
ID_NEW.FLEX_GRID_SIZER: 'wxFlexGridSizer',
ID_NEW.GRID_BAG_SIZER: 'wxGridBagSizer',
ID_NEW.STD_DIALOG_BUTTON_SIZER: 'wxStdDialogButtonSizer',
ID_NEW.SPACER: 'spacer',
ID_NEW.UNKNOWN: 'unknown',
ID_NEW.OK_BUTTON: 'wxButton',
ID_NEW.YES_BUTTON: 'wxButton',
ID_NEW.SAVE_BUTTON: 'wxButton',
ID_NEW.APPLY_BUTTON: 'wxButton',
ID_NEW.NO_BUTTON: 'wxButton',
ID_NEW.CANCEL_BUTTON: 'wxButton',
ID_NEW.HELP_BUTTON: 'wxButton',
ID_NEW.CONTEXT_HELP_BUTTON: 'wxButton',
}
self.topLevel = [
(ID_NEW.PANEL, 'Panel', 'Create panel'),
(ID_NEW.DIALOG, 'Dialog', 'Create dialog'),
(ID_NEW.FRAME, 'Frame', 'Create frame'),
(ID_NEW.WIZARD, 'Wizard', 'Create wizard'),
None,
(ID_NEW.TOOL_BAR, 'ToolBar', 'Create toolbar'),
(ID_NEW.MENU_BAR, 'MenuBar', 'Create menubar'),
(ID_NEW.MENU, 'Menu', 'Create menu')
]
self.containers = [
(ID_NEW.PANEL, 'Panel', 'Create panel'),
(ID_NEW.NOTEBOOK, 'Notebook', 'Create notebook control'),
(ID_NEW.SPLITTER_WINDOW, 'SplitterWindow', 'Create splitter window'),
(ID_NEW.TOOL_BAR, 'ToolBar', 'Create toolbar'),
# (ID_NEW.WIZARD_PAGE, 'WizardPage', 'Create wizard page'),
(ID_NEW.WIZARD_PAGE_SIMPLE, 'WizardPageSimple', 'Create simple wizard page'),
]
self.sizers = [
(ID_NEW.BOX_SIZER, 'BoxSizer', 'Create box sizer'),
(ID_NEW.STATIC_BOX_SIZER, 'StaticBoxSizer',
'Create static box sizer'),
(ID_NEW.GRID_SIZER, 'GridSizer', 'Create grid sizer'),
(ID_NEW.FLEX_GRID_SIZER, 'FlexGridSizer',
'Create flexgrid sizer'),
(ID_NEW.GRID_BAG_SIZER, 'GridBagSizer',
'Create gridbag sizer'),
# (ID_NEW.STD_DIALOG_BUTTON_SIZER, 'StdDialogButtonSizer',
# 'Create standard button sizer'),
(ID_NEW.SPACER, 'Spacer', 'Create spacer'),
]
self.controls = [
['control', 'Various controls',
(ID_NEW.STATIC_TEXT, 'Label', 'Create label'),
(ID_NEW.STATIC_BITMAP, 'Bitmap', 'Create bitmap'),
(ID_NEW.STATIC_LINE, 'Line', 'Create line'),
(ID_NEW.TEXT_CTRL, 'TextBox', 'Create text box'),
(ID_NEW.CHOICE, 'Choice', 'Create choice'),
(ID_NEW.SLIDER, 'Slider', 'Create slider'),
(ID_NEW.GAUGE, 'Gauge', 'Create gauge'),
(ID_NEW.SPIN_CTRL, 'SpinCtrl', 'Create spin'),
(ID_NEW.SCROLL_BAR, 'ScrollBar', 'Create scroll bar'),
(ID_NEW.TREE_CTRL, 'TreeCtrl', 'Create tree'),
(ID_NEW.LIST_CTRL, 'ListCtrl', 'Create list'),
(ID_NEW.CHECK_LIST, 'CheckList', 'Create check list'),
(ID_NEW.SCROLLED_WINDOW, 'ScrolledWindow', 'Create scrolled window'),
(ID_NEW.HTML_WINDOW, 'HtmlWindow', 'Create HTML window'),
(ID_NEW.CALENDAR_CTRL, 'CalendarCtrl', 'Create calendar control'),
(ID_NEW.GENERIC_DIR_CTRL, 'GenericDirCtrl', 'Create generic dir control'),
(ID_NEW.UNKNOWN, 'Unknown', 'Create custom control placeholder'),
],
['button', 'Buttons',
(ID_NEW.BUTTON, 'Button', 'Create button'),
(ID_NEW.BITMAP_BUTTON, 'BitmapButton', 'Create bitmap button'),
(ID_NEW.RADIO_BUTTON, 'RadioButton', 'Create radio button'),
(ID_NEW.SPIN_BUTTON, 'SpinButton', 'Create spin button'),
(ID_NEW.TOGGLE_BUTTON, 'ToggleButton', 'Create toggle button'),
],
['box', 'Boxes',
(ID_NEW.STATIC_BOX, 'StaticBox', 'Create static box'),
(ID_NEW.CHECK_BOX, 'CheckBox', 'Create check box'),
(ID_NEW.RADIO_BOX, 'RadioBox', 'Create radio box'),
(ID_NEW.COMBO_BOX, 'ComboBox', 'Create combo box'),
(ID_NEW.LIST_BOX, 'ListBox', 'Create list box'),
],
['container', 'Containers',
(ID_NEW.PANEL, 'Panel', 'Create panel'),
(ID_NEW.NOTEBOOK, 'Notebook', 'Create notebook control'),
(ID_NEW.SPLITTER_WINDOW, 'SplitterWindow', 'Create splitter window'),
(ID_NEW.TOOL_BAR, 'ToolBar', 'Create toolbar'),
# (ID_NEW.WIZARD_PAGE, 'Wizard Page', 'Create wizard page'),
(ID_NEW.WIZARD_PAGE_SIMPLE, 'WizardPageSimple', 'Create simple wizard page'),
],
['sizer', 'Sizers',
(ID_NEW.BOX_SIZER, 'BoxSizer', 'Create box sizer'),
(ID_NEW.STATIC_BOX_SIZER, 'StaticBoxSizer',
'Create static box sizer'),
(ID_NEW.GRID_SIZER, 'GridSizer', 'Create grid sizer'),
(ID_NEW.FLEX_GRID_SIZER, 'FlexGridSizer',
'Create flexgrid sizer'),
(ID_NEW.GRID_BAG_SIZER, 'GridBagSizer',
'Create gridbag sizer'),
(ID_NEW.SPACER, 'Spacer', 'Create spacer'),
(ID_NEW.STD_DIALOG_BUTTON_SIZER, 'StdDialogButtonSizer',
'Create standard button sizer'),
]
]
self.menuControls = [
(ID_NEW.MENU, 'Menu', 'Create menu'),
(ID_NEW.MENU_ITEM, 'MenuItem', 'Create menu item'),
(ID_NEW.SEPARATOR, 'Separator', 'Create separator'),
]
self.toolBarControls = [
(ID_NEW.TOOL, 'Tool', 'Create tool'),
(ID_NEW.SEPARATOR, 'Separator', 'Create separator'),
['control', 'Various controls',
(ID_NEW.STATIC_TEXT, 'Label', 'Create label'),
(ID_NEW.STATIC_BITMAP, 'Bitmap', 'Create bitmap'),
(ID_NEW.STATIC_LINE, 'Line', 'Create line'),
(ID_NEW.TEXT_CTRL, 'TextBox', 'Create text box'),
(ID_NEW.CHOICE, 'Choice', 'Create choice'),
(ID_NEW.SLIDER, 'Slider', 'Create slider'),
(ID_NEW.GAUGE, 'Gauge', 'Create gauge'),
(ID_NEW.SCROLL_BAR, 'ScrollBar', 'Create scroll bar'),
(ID_NEW.LIST_CTRL, 'ListCtrl', 'Create list control'),
(ID_NEW.CHECK_LIST, 'CheckList', 'Create check list'),
],
['button', 'Buttons',
(ID_NEW.BUTTON, 'Button', 'Create button'),
(ID_NEW.BITMAP_BUTTON, 'BitmapButton', 'Create bitmap button'),
(ID_NEW.RADIO_BUTTON, 'RadioButton', 'Create radio button'),
(ID_NEW.SPIN_BUTTON, 'SpinButton', 'Create spin button'),
],
['box', 'Boxes',
(ID_NEW.STATIC_BOX, 'StaticBox', 'Create static box'),
(ID_NEW.CHECK_BOX, 'CheckBox', 'Create check box'),
(ID_NEW.RADIO_BOX, 'RadioBox', 'Create radio box'),
(ID_NEW.COMBO_BOX, 'ComboBox', 'Create combo box'),
(ID_NEW.LIST_BOX, 'ListBox', 'Create list box'),
],
]
self.stdButtons = [
(ID_NEW.OK_BUTTON, 'OK Button', 'Create standard button'),
(ID_NEW.YES_BUTTON, 'YES Button', 'Create standard button'),
(ID_NEW.SAVE_BUTTON, 'SAVE Button', 'Create standard button'),
(ID_NEW.APPLY_BUTTON, 'APPLY Button', 'Create standard button'),
(ID_NEW.NO_BUTTON, 'NO Button', 'Create standard button'),
(ID_NEW.CANCEL_BUTTON, 'CANCEL Button', 'Create standard button'),
(ID_NEW.HELP_BUTTON, 'HELP Button', 'Create standard button'),
(ID_NEW.CONTEXT_HELP_BUTTON, 'CONTEXT HELP Button', 'Create standard button'),
]
self.stdButtonIDs = {
ID_NEW.OK_BUTTON: ('wxID_OK', '&Ok'),
ID_NEW.YES_BUTTON: ('wxID_YES', '&Yes'),
ID_NEW.SAVE_BUTTON: ('wxID_SAVE', '&Save'),
ID_NEW.APPLY_BUTTON: ('wxID_APPLY', '&Apply'),
ID_NEW.NO_BUTTON: ('wxID_NO', '&No'),
ID_NEW.CANCEL_BUTTON: ('wxID_CANCEL', '&Cancel'),
ID_NEW.HELP_BUTTON: ('wxID_HELP', '&Help'),
ID_NEW.CONTEXT_HELP_BUTTON: ('wxID_CONTEXT_HELP', '&Help'),
}
################################################################################
# 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. Shift is for making
# alternative sets of IDs. (+1000).
def SetMenu(m, list, shift=False):
for l in list:
if type(l) == types.TupleType:
# Shift ID
if shift: l = (1000 + l[0],) + l[1:]
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()
################################################################################
class HighLightBox:
def __init__(self, pos, size):
if size.width == -1: size.width = 0
if size.height == -1: size.height = 0
w = g.testWin.panel
l1 = wxWindow(w, -1, pos, wxSize(size.width, 2))
l1.SetBackgroundColour(wxRED)
l2 = wxWindow(w, -1, pos, wxSize(2, size.height))
l2.SetBackgroundColour(wxRED)
l3 = wxWindow(w, -1, wxPoint(pos.x + size.width - 2, pos.y), wxSize(2, size.height))
l3.SetBackgroundColour(wxRED)
l4 = wxWindow(w, -1, wxPoint(pos.x, pos.y + size.height - 2), wxSize(size.width, 2))
l4.SetBackgroundColour(wxRED)
self.lines = [l1, l2, l3, l4]
# Move highlight to a new position
def Replace(self, pos, size):
if size.width == -1: size.width = 0
if size.height == -1: size.height = 0
self.lines[0].SetDimensions(pos.x, pos.y, size.width, 2)
self.lines[1].SetDimensions(pos.x, pos.y, 2, size.height)
self.lines[2].SetDimensions(pos.x + size.width - 2, pos.y, 2, size.height)
self.lines[3].SetDimensions(pos.x, pos.y + size.height - 2, size.width, 2)
# Remove it
def Remove(self):
map(wxWindow.Destroy, self.lines)
g.testWin.highLight = None
def Refresh(self):
map(wxWindow.Refresh, self.lines)
################################################################################
class XML_Tree(wxTreeCtrl):
def __init__(self, parent, id):
wxTreeCtrl.__init__(self, parent, id, style = wxTR_HAS_BUTTONS | wxTR_MULTIPLE)
self.SetBackgroundColour(wxColour(224, 248, 224))
# Register events
EVT_TREE_SEL_CHANGED(self, self.GetId(), self.OnSelChanged)
# One works on Linux, another on Windows
if wxPlatform == '__WXGTK__':
EVT_TREE_ITEM_ACTIVATED(self, self.GetId(), self.OnItemActivated)
else:
EVT_LEFT_DCLICK(self, self.OnDClick)
EVT_RIGHT_DOWN(self, self.OnRightDown)
EVT_TREE_ITEM_EXPANDED(self, self.GetId(), self.OnItemExpandedCollapsed)
EVT_TREE_ITEM_COLLAPSED(self, self.GetId(), self.OnItemExpandedCollapsed)
self.selection = None
self.needUpdate = False
self.pendingHighLight = None
self.ctrl = self.shift = False
self.dom = None
# Create image list
il = wxImageList(16, 16, True)
self.rootImage = il.Add(images.getTreeRootImage().Scale(16,16).ConvertToBitmap())
xxxObject.image = il.Add(images.getTreeDefaultImage().Scale(16,16).ConvertToBitmap())
xxxPanel.image = il.Add(images.getTreePanelImage().Scale(16,16).ConvertToBitmap())
xxxDialog.image = il.Add(images.getTreeDialogImage().Scale(16,16).ConvertToBitmap())
xxxFrame.image = il.Add(images.getTreeFrameImage().Scale(16,16).ConvertToBitmap())
xxxMenuBar.image = il.Add(images.getTreeMenuBarImage().Scale(16,16).ConvertToBitmap())
xxxMenu.image = il.Add(images.getTreeMenuImage().Scale(16,16).ConvertToBitmap())
xxxMenuItem.image = il.Add(images.getTreeMenuItemImage().Scale(16,16).ConvertToBitmap())
xxxToolBar.image = il.Add(images.getTreeToolBarImage().Scale(16,16).ConvertToBitmap())
xxxTool.image = il.Add(images.getTreeToolImage().Scale(16,16).ConvertToBitmap())
xxxSeparator.image = il.Add(images.getTreeSeparatorImage().Scale(16,16).ConvertToBitmap())
xxxSizer.imageH = il.Add(images.getTreeSizerHImage().Scale(16,16).ConvertToBitmap())
xxxSizer.imageV = il.Add(images.getTreeSizerVImage().Scale(16,16).ConvertToBitmap())
xxxStaticBoxSizer.imageH = il.Add(images.getTreeStaticBoxSizerHImage().Scale(16,16).ConvertToBitmap())
xxxStaticBoxSizer.imageV = il.Add(images.getTreeStaticBoxSizerVImage().Scale(16,16).ConvertToBitmap())
xxxGridSizer.image = il.Add(images.getTreeSizerGridImage().Scale(16,16).ConvertToBitmap())
xxxFlexGridSizer.image = il.Add(images.getTreeSizerFlexGridImage().Scale(16,16).ConvertToBitmap())
self.il = il
self.SetImageList(il)
def RegisterKeyEvents(self):
EVT_KEY_DOWN(self, g.tools.OnKeyDown)
EVT_KEY_UP(self, g.tools.OnKeyUp)
EVT_ENTER_WINDOW(self, g.tools.OnMouse)
EVT_LEAVE_WINDOW(self, g.tools.OnMouse)
def ExpandAll(self, item):
if self.ItemHasChildren(item):
self.Expand(item)
i, cookie = self.GetFirstChild(item)
children = []
while i.IsOk():
children.append(i)
i, cookie = self.GetNextChild(item, cookie)
for i in children:
self.ExpandAll(i)
def CollapseAll(self, item):
if self.ItemHasChildren(item):
i, cookie = self.GetFirstChild(item)
children = []
while i.IsOk():
children.append(i)
i, cookie = self.GetNextChild(item, cookie)
for i in children:
self.CollapseAll(i)
self.Collapse(item)
# Clear tree
def Clear(self):
self.DeleteAllItems()
# Add minimal structure
if self.dom: self.dom.unlink()
self.dom = MyDocument()
self.dummyNode = self.dom.createComment('dummy node')
# Create main node
self.mainNode = self.dom.createElement('resource')
self.dom.appendChild(self.mainNode)
self.rootObj = xxxMainNode(self.dom)
self.root = self.AddRoot('XML tree', self.rootImage,
data=wxTreeItemData(self.rootObj))
self.SetItemHasChildren(self.root)
self.Unselect()
self.Expand(self.root)
# Clear old data and set new
def SetData(self, dom):
self.DeleteAllItems()
# Add minimal structure
if self.dom: self.dom.unlink()
self.dom = dom
self.dummyNode = self.dom.createComment('dummy node')
# Find 'resource' child, add it's children
self.mainNode = dom.documentElement
self.rootObj = xxxMainNode(self.dom)
self.root = self.AddRoot('XML tree', self.rootImage,
data=wxTreeItemData(self.rootObj))
self.SetItemHasChildren(self.root)
nodes = self.mainNode.childNodes[:]
for node in nodes:
if IsObject(node):
self.AddNode(self.root, None, node)
else:
self.mainNode.removeChild(node)
node.unlink()
self.Expand(self.root)
self.Unselect()
# Add tree item for given parent item if node is DOM element node with
# object/object_ref tag. xxxParent is parent xxx object
def AddNode(self, itemParent, xxxParent, node):
# Set item data to current node
try:
xxx = MakeXXXFromDOM(xxxParent, node)
except:
print 'ERROR: MakeXXXFromDom(%s, %s)' % (xxxParent, node)
raise
treeObj = xxx.treeObject()
# Append tree item
item = self.AppendItem(itemParent, treeObj.treeName(),
image=treeObj.treeImage(),
data=wxTreeItemData(xxx))
# Different color for references
if treeObj.ref:
self.SetItemTextColour(item, 'DarkGreen')
# 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()
# Insert new item at specific position
def InsertNode(self, itemParent, parent, elem, nextItem):
# Insert in XML tree and wxWin
xxx = MakeXXXFromDOM(parent, elem)
# If nextItem is None, we append to parent, otherwise insert before it
if nextItem.IsOk():
node = self.GetPyData(nextItem).element
parent.element.insertBefore(elem, node)
# Inserting before is difficult, se we insert after or first child
index = self.ItemIndex(nextItem)
newItem = self.InsertItemBefore(itemParent, index,
xxx.treeName(), image=xxx.treeImage())
self.SetPyData(newItem, xxx)
else:
parent.element.appendChild(elem)
newItem = self.AppendItem(itemParent, xxx.treeName(), image=xxx.treeImage(),
data=wxTreeItemData(xxx))
# Different color for references
if xxx.treeObject().ref: self.SetItemTextColour(newItem, 'DarkGreen')
# Add children items
if xxx.hasChildren:
treeObj = xxx.treeObject()
for n in treeObj.element.childNodes:
if IsObject(n):
self.AddNode(newItem, treeObj, n)
return newItem
# 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)
# Reset selection object
self.selection = None
return node
# Find position relative to the top-level window
def FindNodePos(self, item, obj=None):
# Root at (0,0)
if item == g.testWin.item: return wxPoint(0, 0)
itemParent = self.GetItemParent(item)
# Select NB page
if not obj: obj = self.FindNodeObject(item)
if self.GetPyData(itemParent).treeObject().__class__ == xxxNotebook:
notebook = self.FindNodeObject(itemParent)
# Find position
for i in range(notebook.GetPageCount()):
if notebook.GetPage(i) == obj:
if notebook.GetSelection() != i:
notebook.SetSelection(i)
# Remove highlight - otherwise highlight window won't be visible
if g.testWin.highLight:
g.testWin.highLight.Remove()
break
# Find first ancestor which is a wxWindow (not a sizer)
winParent = itemParent
while self.GetPyData(winParent).isSizer:
winParent = self.GetItemParent(winParent)
# Notebook children are layed out in a little strange way
if self.GetPyData(itemParent).treeObject().__class__ == xxxNotebook:
parentPos = wxPoint(0,0)
else:
parentPos = self.FindNodePos(winParent)
# Position (-1,-1) is really (0,0)
pos = obj.GetPosition()
if pos == (-1,-1): pos = (0,0)
return parentPos + pos
# Find window (or sizer) corresponding to a tree item.
def FindNodeObject(self, item):
testWin = g.testWin
# If top-level, return testWin (or panel its panel)
if item == testWin.item: return testWin.panel
itemParent = self.GetItemParent(item)
xxx = self.GetPyData(item).treeObject()
parentWin = self.FindNodeObject(itemParent)
# Top-level sizer? return window's sizer
if xxx.isSizer and isinstance(parentWin, wxWindow):
return parentWin.GetSizer()
elif isinstance(xxx.parent, xxxToolBar):
# How to get tool from toolbar?
return parentWin
elif isinstance(xxx.parent, xxxStdDialogButtonSizer):
# This sizer returns non-existing children
for ch in parentWin.GetChildren():
if ch.GetWindow() and ch.GetWindow().GetName() == xxx.name:
return ch.GetWindow()
return None
# Otherwise get parent's object and it's child
child = parentWin.GetChildren()[self.ItemIndex(item)]
# 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, wxNotebookSizer):
child = child.GetNotebook()
return child
def OnSelChanged(self, evt):
self.ChangeSelection(evt.GetItem())
def ChangeSelection(self, item):
# Apply changes
# !!! problem with wxGTK - GetOldItem is Ok if nothing selected
#oldItem = evt.GetOldItem()
status = ''
oldItem = self.selection
if oldItem:
xxx = self.GetPyData(oldItem)
# If some data was modified, apply changes
if g.panel.IsModified():
self.Apply(xxx, oldItem)
#if conf.autoRefresh:
if g.testWin:
if g.testWin.highLight:
g.testWin.highLight.Remove()
self.needUpdate = True
status = 'Changes were applied'
g.frame.SetStatusText(status)
# Generate view
self.selection = item
if not self.selection.IsOk():
self.selection = None
return
xxx = self.GetPyData(self.selection)
# Update panel
g.panel.SetData(xxx)
# Update tools
g.tools.UpdateUI()
# Highlighting is done in OnIdle
self.pendingHighLight = self.selection
# Check if item is in testWin subtree
def IsHighlatable(self, item):
if item == g.testWin.item: return False
while item != self.root:
item = self.GetItemParent(item)
if item == g.testWin.item: return True
return False
# Highlight selected item
def HighLight(self, item):
self.pendingHighLight = None
# Can highlight only with some top-level windows
if not g.testWin or self.GetPyData(g.testWin.item).treeObject().__class__ \
not in [xxxDialog, xxxPanel, xxxFrame]:
return
# If a control from another window is selected, remove highlight
if not self.IsHighlatable(item):
if g.testWin.highLight: g.testWin.highLight.Remove()
return
# Get window/sizer object
obj = self.FindNodeObject(item)
if not obj: return
pos = self.FindNodePos(item, obj)
size = obj.GetSize()
# Highlight
# Negative positions are not working quite well
if g.testWin.highLight:
g.testWin.highLight.Replace(pos, size)
else:
g.testWin.highLight = HighLightBox(pos, size)
g.testWin.highLight.Refresh()
g.testWin.highLight.item = item
def ShowTestWindow(self, item):
xxx = self.GetPyData(item)
if g.panel.IsModified():
self.Apply(xxx, item) # apply changes
treeObj = xxx.treeObject()
if treeObj.className not in ['wxFrame', 'wxPanel', 'wxDialog',
'wxMenuBar', 'wxToolBar', 'wxWizard',
'wxWizardPageSimple']:
wxLogMessage('No view for this element (yet)')
return
# Show item in bold
if g.testWin: # Reset old
self.SetItemBold(g.testWin.item, False)
self.CreateTestWin(item)
# Maybe an error occurred, so we need to test
if g.testWin: self.SetItemBold(g.testWin.item)
# Double-click on Linux
def OnItemActivated(self, evt):
if evt.GetItem() != self.root:
self.ShowTestWindow(evt.GetItem())
# Double-click on Windows
def OnDClick(self, evt):
item, flags = self.HitTest(evt.GetPosition())
if flags in [wxTREE_HITTEST_ONITEMBUTTON, wxTREE_HITTEST_ONITEMLABEL]:
if item != self.root: self.ShowTestWindow(item)
else:
evt.Skip()
def OnItemExpandedCollapsed(self, evt):
# Update tool palette
g.tools.UpdateUI()
evt.Skip()
# (re)create test window
def CreateTestWin(self, item):
testWin = g.testWin
# Create a window with this resource
xxx = self.GetPyData(item).treeObject()
# If frame
# if xxx.__class__ == xxxFrame:
# Frame can't have many children,
# but it's first child possibly can...
# child = self.GetFirstChild(item)[0]
# if child.IsOk() and self.GetPyData(child).__class__ == xxxPanel:
# # Clean-up before recursive call or error
# wxMemoryFSHandler_RemoveFile('xxx.xrc')
# wxEndBusyCursor()
# self.CreateTestWin(child)
# return
wxBeginBusyCursor()
wxYield()
# Close old window, remember where it was
highLight = None
if testWin:
pos = testWin.GetPosition()
if item == testWin.item:
# Remember highlight if same top-level window
if testWin.highLight:
highLight = testWin.highLight.item
if xxx.className == 'wxPanel':
if testWin.highLight:
testWin.pendingHighLight = highLight
testWin.highLight.Remove()
testWin.panel.Destroy()
testWin.panel = None
else:
testWin.Destroy()
testWin = g.testWin = None
else:
testWin.Destroy()
testWin = g.testWin = None
else:
pos = g.testWinPos
# Save in memory FS
memFile = MemoryFile('xxx.xrc')
# Create memory XML file
elem = xxx.element
# Change window id to _XRCED_T_W. This gives some name for
# unnamed windows, and for named gives the possibility to
# write sawfish scripts.
if not xxx.name:
name = 'noname'
else:
name = xxx.name
elem.setAttribute('name', STD_NAME)
# Replace wizard page class temporarily
if xxx.__class__ in [xxxWizardPage, xxxWizardPageSimple]:
oldCl = elem.getAttribute('class')
elem.setAttribute('class', 'wxPanel')
parent = elem.parentNode
next = elem.nextSibling
encd = self.rootObj.params['encoding'].value()
if not encd: encd = None
self.dom.writexml(open('ttt.xrc','w'), encoding=encd)
self.dom.writexml(memFile, encoding=encd)
# Put back in place
# Remove temporary name or restore changed
if not xxx.name:
elem.removeAttribute('name')
else:
elem.setAttribute('name', xxx.name)
if xxx.__class__ in [xxxWizardPage, xxxWizardPageSimple]:
elem.setAttribute('class', oldCl)
memFile.close() # write to wxMemoryFS
xmlFlags = wxXRC_NO_SUBCLASSING
# Use translations if encoding is not specified
if not g.currentEncoding:
xmlFlags != wxXRC_USE_LOCALE
res = wxXmlResource('', xmlFlags)
res.Load('memory:xxx.xrc')
try:
if xxx.__class__ == xxxFrame:
# Frame can't have many children,
# but it's first child possibly can...
# child = self.GetFirstChild(item)[0]
# if child.IsOk() and self.GetPyData(child).__class__ == xxxPanel:
# # Clean-up before recursive call or error
# wxMemoryFSHandler_RemoveFile('xxx.xrc')
# wxEndBusyCursor()
# self.CreateTestWin(child)
# return
# This currently works under GTK, but not under MSW
testWin = g.testWin = wxPreFrame()
res.LoadOnFrame(testWin, g.frame, STD_NAME)
# Create status bar
testWin.panel = testWin
testWin.CreateStatusBar()
testWin.SetClientSize(testWin.GetBestSize())
testWin.SetPosition(pos)
testWin.Show(True)
elif xxx.__class__ == xxxPanel:
# Create new frame
if not testWin:
testWin = g.testWin = wxFrame(g.frame, -1, 'Panel: ' + name,
pos=pos, name=STD_NAME)
testWin.panel = res.LoadPanel(testWin, STD_NAME)
testWin.SetClientSize(testWin.GetBestSize())
testWin.Show(True)
elif xxx.__class__ == xxxDialog:
testWin = g.testWin = res.LoadDialog(None, STD_NAME)
testWin.panel = testWin
testWin.Layout()
testWin.SetPosition(pos)
testWin.Show(True)
# Dialog's default code does not produce EVT_CLOSE
EVT_BUTTON(testWin, wxID_OK, self.OnCloseTestWin)
EVT_BUTTON(testWin, wxID_CANCEL, self.OnCloseTestWin)
elif xxx.__class__ == xxxWizard:
wiz = wxPreWizard()
res.LoadOnObject(wiz, None, STD_NAME, 'wxWizard')
# Find first page (don't know better way)
firstPage = None
for w in wiz.GetChildren():
if isinstance(w, wxWizardPage):
firstPage = w
break
if not firstPage:
wxLogError('Wizard is empty')
else:
# Wizard should be modal
self.SetItemBold(item)
wiz.RunWizard(w)
self.SetItemBold(item, False)
wiz.Destroy()
elif xxx.__class__ in [xxxWizardPage, xxxWizardPageSimple]:
# Create new frame
if not testWin:
testWin = g.testWin = wxFrame(g.frame, -1, 'Wizard page: ' + name,
pos=pos, name=STD_NAME)
testWin.panel = wxPrePanel()
res.LoadOnObject(testWin.panel, testWin, STD_NAME, 'wxPanel')
testWin.SetClientSize(testWin.GetBestSize())
testWin.Show(True)
elif xxx.__class__ == xxxMenuBar:
testWin = g.testWin = wxFrame(g.frame, -1, 'MenuBar: ' + name,
pos=pos, name=STD_NAME)
testWin.panel = None
# Set status bar to display help
testWin.CreateStatusBar()
testWin.menuBar = res.LoadMenuBar(STD_NAME)
testWin.SetMenuBar(testWin.menuBar)
testWin.Show(True)
elif xxx.__class__ == xxxToolBar:
testWin = g.testWin = wxFrame(g.frame, -1, 'ToolBar: ' + name,
pos=pos, name=STD_NAME)
testWin.panel = None
# Set status bar to display help
testWin.CreateStatusBar()
testWin.toolBar = res.LoadToolBar(testWin, STD_NAME)
testWin.SetToolBar(testWin.toolBar)
testWin.Show(True)
if testWin:
testWin.item = item
EVT_CLOSE(testWin, self.OnCloseTestWin)
testWin.highLight = None
if highLight and not self.pendingHighLight:
self.HighLight(highLight)
except:
if g.testWin:
self.SetItemBold(item, False)
g.testWinPos = g.testWin.GetPosition()
g.testWin.Destroy()
g.testWin = None
inf = sys.exc_info()
wxLogError(traceback.format_exception(inf[0], inf[1], None)[-1])
wxLogError('Error loading resource')
wxMemoryFSHandler_RemoveFile('xxx.xrc')
wxEndBusyCursor()
def CloseTestWindow(self):
if not g.testWin: return
self.SetItemBold(g.testWin.item, False)
g.frame.tb.ToggleTool(g.frame.ID_TOOL_LOCATE, False)
g.testWinPos = g.testWin.GetPosition()
g.testWin.Destroy()
g.testWin = None
def OnCloseTestWin(self, evt):
self.CloseTestWindow()
# Return item index in parent
def ItemIndex(self, item):
n = 0 # index of sibling
prev = self.GetPrevSibling(item)
while prev.IsOk():
prev = self.GetPrevSibling(prev)
n += 1
return n
# Full tree index of an item - list of positions
def ItemFullIndex(self, item):
if not item.IsOk(): return None
l = []
while item != self.root:
l.insert(0, self.ItemIndex(item))
item = self.GetItemParent(item)
return l
# Get item position from full index
def ItemAtFullIndex(self, index):
if index is None: return wxTreeItemId()
item = self.root
for i in index:
item = self.GetFirstChild(item)[0]
for k in range(i): item = self.GetNextSibling(item)
return item
# True if next item should be inserted after current (vs. appended to it)
def NeedInsert(self, item):
xxx = self.GetPyData(item)
if item == self.root: return False # root item
if xxx.hasChildren and not self.GetChildrenCount(item, False):
return False
return not (self.IsExpanded(item) and self.GetChildrenCount(item, False))
# Pull-down
def OnRightDown(self, evt):
pullDownMenu = g.pullDownMenu
# select this item
pt = evt.GetPosition();
item, flags = self.HitTest(pt)
if item.Ok() and flags & wxTREE_HITTEST_ONITEM:
self.SelectItem(item)
# Setup menu
menu = wxMenu()
item = self.selection
if not item:
menu.Append(g.pullDownMenu.ID_EXPAND, 'Expand', 'Expand tree')
menu.Append(g.pullDownMenu.ID_COLLAPSE, 'Collapse', 'Collapse tree')
else:
# self.ctrl = evt.ControlDown() # save Ctrl state
# self.shift = evt.ShiftDown() # and Shift too
m = wxMenu() # create menu
if self.ctrl:
needInsert = True
else:
needInsert = self.NeedInsert(item)
if item == self.root or needInsert and self.GetItemParent(item) == self.root:
SetMenu(m, pullDownMenu.topLevel)
m.AppendSeparator()
m.Append(ID_NEW.REF, 'reference...', 'Create object_ref node')
else:
xxx = self.GetPyData(item).treeObject()
# Check parent for possible child nodes if inserting sibling
if needInsert: xxx = xxx.parent
if xxx.__class__ == xxxMenuBar:
m.Append(ID_NEW.MENU, 'Menu', 'Create menu')
elif xxx.__class__ in [xxxToolBar, xxxTool] or \
xxx.__class__ == xxxSeparator and xxx.parent.__class__ == xxxToolBar:
SetMenu(m, pullDownMenu.toolBarControls)
elif xxx.__class__ in [xxxMenu, xxxMenuItem]:
SetMenu(m, pullDownMenu.menuControls)
elif xxx.__class__ == xxxStdDialogButtonSizer:
SetMenu(m, pullDownMenu.stdButtons)
else:
SetMenu(m, pullDownMenu.controls)
if xxx.__class__ == xxxNotebook:
m.Enable(m.FindItem('sizer'), False)
elif not (xxx.isSizer or xxx.parent and xxx.parent.isSizer):
m.Enable(ID_NEW.SPACER, False)
m.AppendSeparator()
m.Append(ID_NEW.REF, 'reference...', 'Create object_ref node')
# Select correct label for create menu
if not needInsert:
if self.shift:
menu.AppendMenu(wxNewId(), 'Insert Child', m,
'Create child object as the first child')
else:
menu.AppendMenu(wxNewId(), 'Append Child', m,
'Create child object as the last child')
else:
if self.shift:
menu.AppendMenu(wxNewId(), 'Create Sibling', m,
'Create sibling before selected object')
else:
menu.AppendMenu(wxNewId(), 'Create Sibling', m,
'Create sibling after selected object')
# Build replace menu
if item != self.root:
xxx = self.GetPyData(item).treeObject()
m = wxMenu() # create replace menu
if xxx.__class__ == xxxMenuBar:
m.Append(1000 + ID_NEW.MENU, 'Menu', 'Create menu')
elif xxx.__class__ in [xxxMenu, xxxMenuItem]:
SetMenu(m, pullDownMenu.menuControls, shift=True)
elif xxx.__class__ == xxxToolBar and \
self.GetItemParent(item) == self.root:
SetMenu(m, [], shift=True)
elif xxx.__class__ in [xxxFrame, xxxDialog, xxxPanel]:
SetMenu(m, [
(ID_NEW.PANEL, 'Panel', 'Create panel'),
(ID_NEW.DIALOG, 'Dialog', 'Create dialog'),
(ID_NEW.FRAME, 'Frame', 'Create frame')], shift=True)
elif xxx.isSizer:
SetMenu(m, pullDownMenu.sizers, shift=True)
else:
SetMenu(m, pullDownMenu.controls, shift=True)
id = wxNewId()
menu.AppendMenu(id, 'Replace With', m)
if not m.GetMenuItemCount(): menu.Enable(id, False)
menu.Append(pullDownMenu.ID_SUBCLASS, 'Subclass...',
'Set "subclass" property')
menu.AppendSeparator()
# Not using standart IDs because we don't want to show shortcuts
menu.Append(wxID_CUT, 'Cut', 'Cut to the clipboard')
menu.Append(wxID_COPY, 'Copy', 'Copy to the clipboard')
if self.ctrl and item != self.root:
menu.Append(pullDownMenu.ID_PASTE_SIBLING, 'Paste Sibling',
'Paste from the clipboard as a sibling')
else:
menu.Append(wxID_PASTE, 'Paste', 'Paste from the clipboard')
menu.Append(pullDownMenu.ID_DELETE,
'Delete', 'Delete object')
if self.ItemHasChildren(item):
menu.AppendSeparator()
menu.Append(pullDownMenu.ID_EXPAND, 'Expand', 'Expand subtree')
menu.Append(pullDownMenu.ID_COLLAPSE, 'Collapse', 'Collapse subtree')
self.PopupMenu(menu, evt.GetPosition())
menu.Destroy()
# Apply changes
def Apply(self, xxx, item):
g.panel.Apply()
# Update tree view
xxx = xxx.treeObject()
if xxx.hasName and self.GetItemText(item) != xxx.name:
self.SetItemText(item, xxx.treeName())
# Item width may have changed
# !!! Tric to update tree width (wxGTK, ??)
self.SetIndent(self.GetIndent())
# Change tree icon for sizers
if isinstance(xxx, xxxBoxSizer):
self.SetItemImage(item, xxx.treeImage())
# Set global modified state
g.frame.SetModified()
Reference in New Issue View Git Blame Copy Permalink