68ae5821b6
git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@27015 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
965 lines
40 KiB
Python
965 lines
40 KiB
Python
# 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):
|
|
self.buffer += data.encode()
|
|
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()
|
|
|
|
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()
|
|
|
|
BOX_SIZER = wxNewId()
|
|
STATIC_BOX_SIZER = wxNewId()
|
|
GRID_SIZER = wxNewId()
|
|
FLEX_GRID_SIZER = wxNewId()
|
|
GRID_BAG_SIZER = wxNewId()
|
|
SPACER = wxNewId()
|
|
TOOL_BAR = wxNewId()
|
|
TOOL = wxNewId()
|
|
MENU = wxNewId()
|
|
MENU_ITEM = wxNewId()
|
|
SEPARATOR = wxNewId()
|
|
LAST = wxNewId()
|
|
|
|
class PullDownMenu:
|
|
ID_EXPAND = wxNewId()
|
|
ID_COLLAPSE = wxNewId()
|
|
ID_PASTE_SIBLING = 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.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.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.SPACER: 'spacer',
|
|
ID_NEW.UNKNOWN: 'unknown',
|
|
}
|
|
self.topLevel = [
|
|
(ID_NEW.PANEL, 'Panel', 'Create panel'),
|
|
(ID_NEW.DIALOG, 'Dialog', 'Create dialog'),
|
|
(ID_NEW.FRAME, 'Frame', 'Create frame'),
|
|
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'),
|
|
]
|
|
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.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'),
|
|
],
|
|
['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'),
|
|
],
|
|
['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'),
|
|
]
|
|
]
|
|
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'),
|
|
],
|
|
]
|
|
|
|
################################################################################
|
|
|
|
# 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()
|
|
# Same, but adds 1000 to all IDs
|
|
def SetMenu2(m, list):
|
|
for l in list:
|
|
if type(l) == types.TupleType:
|
|
# Shift ID
|
|
l = (1000 + l[0],) + l[1:]
|
|
apply(m.Append, l)
|
|
elif type(l) == types.ListType:
|
|
subMenu = wxMenu()
|
|
SetMenu2(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
|
|
|
|
################################################################################
|
|
|
|
class XML_Tree(wxTreeCtrl):
|
|
def __init__(self, parent, id):
|
|
wxTreeCtrl.__init__(self, parent, id, style = wxTR_HAS_BUTTONS)
|
|
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 Unselect(self):
|
|
self.selection = None
|
|
wxTreeCtrl.Unselect(self)
|
|
g.tools.UpdateUI()
|
|
|
|
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' 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))
|
|
# 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))
|
|
# 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):
|
|
# Root at (0,0)
|
|
if item == g.testWin.item: return wxPoint(0, 0)
|
|
itemParent = self.GetItemParent(item)
|
|
# Select NB page
|
|
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)
|
|
break
|
|
# 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)
|
|
# 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()
|
|
# 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):
|
|
# 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 = evt.GetItem()
|
|
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()
|
|
# Hightlighting 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, pos = self.FindNodeObject(item), self.FindNodePos(item)
|
|
size = obj.GetSize()
|
|
# Highlight
|
|
# Nagative positions are not working wuite well
|
|
if g.testWin.highLight:
|
|
g.testWin.highLight.Replace(pos, size)
|
|
else:
|
|
g.testWin.highLight = HighLightBox(pos, size)
|
|
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']:
|
|
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 occured, 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 partial XML file - faster for big files
|
|
|
|
dom = MyDocument()
|
|
mainNode = dom.createElement('resource')
|
|
dom.appendChild(mainNode)
|
|
|
|
# Remove temporarily from old parent
|
|
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)
|
|
parent = elem.parentNode
|
|
next = elem.nextSibling
|
|
parent.replaceChild(self.dummyNode, elem)
|
|
# Append to new DOM, write it
|
|
mainNode.appendChild(elem)
|
|
dom.writexml(memFile, encoding=self.rootObj.params['encoding'].value())
|
|
# Put back in place
|
|
mainNode.removeChild(elem)
|
|
dom.unlink()
|
|
parent.replaceChild(elem, self.dummyNode)
|
|
# Remove temporary name or restore changed
|
|
if not xxx.name:
|
|
elem.removeAttribute('name')
|
|
else:
|
|
elem.setAttribute('name', xxx.name)
|
|
memFile.close() # write to wxMemoryFS
|
|
xmlFlags = wxXRC_NO_SUBCLASSING
|
|
# Use translations if encoding is not specified
|
|
if g.currentEncoding == 'ascii':
|
|
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.panel = testWin
|
|
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__ == 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)
|
|
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 OnCloseTestWin(self, evt):
|
|
self.SetItemBold(g.testWin.item, False)
|
|
g.testWinPos = g.testWin.GetPosition()
|
|
g.testWin.Destroy()
|
|
g.testWin = None
|
|
|
|
# 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)
|
|
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)
|
|
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)
|
|
# 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]:
|
|
SetMenu2(m, pullDownMenu.menuControls)
|
|
elif xxx.__class__ == xxxToolBar and \
|
|
self.GetItemParent(item) == self.root:
|
|
SetMenu2(m, [])
|
|
elif xxx.__class__ in [xxxFrame, xxxDialog, xxxPanel]:
|
|
SetMenu2(m, [
|
|
(ID_NEW.PANEL, 'Panel', 'Create panel'),
|
|
(ID_NEW.DIALOG, 'Dialog', 'Create dialog'),
|
|
(ID_NEW.FRAME, 'Frame', 'Create frame')])
|
|
elif xxx.isSizer:
|
|
SetMenu2(m, pullDownMenu.sizers)
|
|
else:
|
|
SetMenu2(m, pullDownMenu.controls)
|
|
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.modified = True
|
|
|