git.saurik.com Git - wxWidgets.git/blame_incremental

... / ...

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