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

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

@ -52,7 +52,9 @@ class ProtoPaletteUI(wx.Panel):
validator=wx.DefaultValidator, name="treeCtrl") validator=wx.DefaultValidator, name="treeCtrl")
self.rootName = "Proto Objects" self.rootName = "Proto Objects"
self.root = self.tree.AddRoot(self.rootName) 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.editorTxt = "Proto Objects Editor"
@ -109,27 +111,21 @@ class ProtoPaletteUI(wx.Panel):
item, cookie = self.tree.GetNextChild(parent, cookie) item, cookie = self.tree.GetNextChild(parent, cookie)
return None return None
def addTreeNode(self, itemText, parentItem, items): def addTreeNodes(self, parentItem, parentItemName, items, itemKeys):
newItem = wx.TreeItemId roots = []
parentText = items[itemText] rootItems = []
if parentText == self.palette.rootName: #import pdb;set_trace()
newItem = self.tree.AppendItem(parentItem, itemText) for key in itemKeys:
self.tree.SetItemPyData(newItem, itemText) if parentItemName == items[key]:
else: roots.append(key)
item = self.traverse(parentItem, parentText) for root in roots:
if item is None: newItem = self.tree.AppendItem(parentItem, root)
item = self.addTreeNode(parentText, parentItem, items) 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): def onBeginDrag(self, event):
item = event.GetItem() item = event.GetItem()