Made protoPalette working with new PaletteTreeCtrl

direct/src/leveleditor/MayaConverter.py

@@ -148,7 +148,7 @@ class MayaConverter(wx.Dialog):
 
         self.editor.protoPalette.add(itemData)
 
-        newItem = self.editor.ui.protoPaletteUI.tree.AppendItem(self.editor.ui.protoPaletteUI.root, name)
+        newItem = self.editor.ui.protoPaletteUI.tree.AppendItem(self.editor.ui.protoPaletteUI.tree.root, name)
         self.editor.ui.protoPaletteUI.tree.SetItemPyData(newItem, itemData)
         self.editor.ui.protoPaletteUI.tree.ScrollTo(newItem)
             

direct/src/leveleditor/PaletteTreeCtrl.py

@@ -10,7 +10,7 @@ class PaletteTreeCtrl(wx.TreeCtrl):
         wx.TreeCtrl.__init__(self, parent, style=treeStyle)
 
         self.rootName = rootName
-        self.AddRoot(self.rootName)
+        self.root = self.AddRoot(self.rootName)
 
         self.paletteUI = parent
 

direct/src/leveleditor/ProtoPaletteUI.py

@@ -149,9 +149,9 @@ class ProtoPaletteUI(wx.Panel):
         menuItem = self.popupmenu.FindItemById(event.GetId())
         text = menuItem.GetText()
         if text == self.opAdd:
-           self.AddGroup()
+           self.tree.AddGroup()
         elif text == self.opDelete:
-           self.DeleteSelected()
+           self.tree.DeleteSelected()
         elif text == self.opSortAlpha:
            self.opSort = self.opSortAlpha
            self.tree.SortTreeNodes(self.tree.GetRootItem())
@@ -174,7 +174,7 @@ class ProtoPaletteUI(wx.Panel):
         itemData = ObjectBase(name=name, model=modelname, actor=True)
         self.editor.protoPalette.add(itemData)
 
-        newItem = self.tree.AppendItem(self.editor.ui.protoPaletteUI.root, name)
+        newItem = self.tree.AppendItem(self.editor.ui.protoPaletteUI.tree.root, name)
         self.tree.SetItemPyData(newItem, itemData)
         self.tree.ScrollTo(newItem)