TheCloud/panda3d
1
0
Fork 0
mirror of https://github.com/panda3d/panda3d.git synced 2025-10-03 10:22:45 -04:00
Code Issues Packages Projects Releases Wiki Activity

improved tree building performance

Browse Source
This commit is contained in:
Arkady Trestman 2010-03-10 02:10:06 +00:00
parent bc9a4e1db1
commit 7a4844fa34
1 changed files with 18 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
direct/src/leveleditor/ProtoPaletteUI.py
View File

@ -52,8 +52,10 @@ class ProtoPaletteUI(wx.Panel):
validator=wx.DefaultValidator, name="treeCtrl")
self.rootName = "Proto Objects"
self.root = self.tree.AddRoot(self.rootName)
self.addTreeNodes(self.root, self.palette.dataStruct)
self.dataKeys = self.palette.dataStruct.keys()[:]
self.addTreeNodes(self.root, self.palette.rootName, self.palette.dataStruct, self.dataKeys)
self.editorTxt = "Proto Objects Editor"
self.opAdd = "Add Group"
@ -109,27 +111,21 @@ class ProtoPaletteUI(wx.Panel):
item, cookie = self.tree.GetNextChild(parent, cookie)
return None
def addTreeNode(self, itemText, parentItem, items):
newItem = wx.TreeItemId
parentText = items[itemText]
if parentText == self.palette.rootName:
newItem = self.tree.AppendItem(parentItem, itemText)
self.tree.SetItemPyData(newItem, itemText)
else:
item = self.traverse(parentItem, parentText)
if item is None:
item = self.addTreeNode(parentText, parentItem, items)
def addTreeNodes(self, parentItem, parentItemName, items, itemKeys):
roots = []
rootItems = []
#import pdb;set_trace()
for key in itemKeys:
if parentItemName == items[key]:
roots.append(key)
for root in roots:
newItem = self.tree.AppendItem(parentItem, root)
self.tree.SetItemPyData(newItem, root)
rootItems.append(newItem)
itemKeys.remove(root)
for rootItem in rootItems:
self.addTreeNodes(rootItem, self.tree.GetItemText(rootItem), items, itemKeys)
newItem = self.tree.AppendItem(item, itemText)
self.tree.SetItemPyData(newItem, itemText)
return newItem
def addTreeNodes(self, parentItem, items):
for key in items.keys():
item = self.traverse(parentItem, key)
if item is None:
self.addTreeNode(key, parentItem, items)
def onBeginDrag(self, event):
item = event.GetItem()