import wx, wx.lib.customtreectrl, wx.gizmos try: import treemixin except ImportError: from wx.lib.mixins import treemixin overview = treemixin.__doc__ class TreeModel(object): ''' TreeModel holds the domain objects that are shown in the different tree controls. Each domain object is simply a two-tuple consisting of a label and a list of child tuples, i.e. (label, [list of child tuples]). ''' def __init__(self, *args, **kwargs): self.items = [] self.itemCounter = 0 super(TreeModel, self).__init__(*args, **kwargs) def GetItem(self, indices): text, children = 'Hidden root', self.items for index in indices: text, children = children[index] return text, children def GetText(self, indices): return self.GetItem(indices)[0] def GetChildren(self, indices): return self.GetItem(indices)[1] def GetChildrenCount(self, indices): return len(self.GetChildren(indices)) def SetChildrenCount(self, indices, count): children = self.GetChildren(indices) while len(children) > count: children.pop() while len(children) < count: children.append(('item %d'%self.itemCounter, [])) self.itemCounter += 1 def MoveItem(self, itemToMoveIndex, newParentIndex): itemToMove = self.GetItem(itemToMoveIndex) newParentChildren = self.GetChildren(newParentIndex) newParentChildren.append(itemToMove) oldParentChildren = self.GetChildren(itemToMoveIndex[:-1]) oldParentChildren.remove(itemToMove) class DemoTreeMixin(treemixin.VirtualTree, treemixin.DragAndDrop, treemixin.ExpansionState): def __init__(self, *args, **kwargs): self.model = kwargs.pop('treemodel') self.log = kwargs.pop('log') super(DemoTreeMixin, self).__init__(*args, **kwargs) self.CreateImageList() def CreateImageList(self): size = (16, 16) self.imageList = wx.ImageList(*size) for art in wx.ART_FOLDER, wx.ART_FILE_OPEN, wx.ART_NORMAL_FILE: self.imageList.Add(wx.ArtProvider.GetBitmap(art, wx.ART_OTHER, size)) self.AssignImageList(self.imageList) def OnGetItemText(self, indices): return self.model.GetText(indices) def OnGetChildrenCount(self, indices): return self.model.GetChildrenCount(indices) def OnGetItemFont(self, indices): # Show how to change the item font. Here we use a small font for # items that have children and the default font otherwise. if self.model.GetChildrenCount(indices) > 0: return wx.SMALL_FONT else: return super(DemoTreeMixin, self).OnGetItemFont(indices) def OnGetItemTextColour(self, indices): # Show how to change the item text colour. In this case second level # items are coloured red and third level items are blue. All other # items have the default text colour. if len(indices) % 2 == 0: return wx.RED elif len(indices) % 3 == 0: return wx.BLUE else: return super(DemoTreeMixin, self).OnGetItemTextColour(indices) def OnGetItemBackgroundColour(self, indices): # Show how to change the item background colour. In this case the # background colour of each third item is green. if indices[-1] == 2: return wx.GREEN else: return super(DemoTreeMixin, self).OnGetItemBackgroundColour(indices) def OnGetItemImage(self, indices, which): # Return the right icon depending on whether the item has children. if which in [wx.TreeItemIcon_Normal, wx.TreeItemIcon_Selected]: if self.model.GetChildrenCount(indices): return 0 else: return 2 else: return 1 def OnDrop(self, dropTarget, dragItem): dropIndex = self.GetIndexOfItem(dropTarget) dropText = self.model.GetText(dropIndex) dragIndex = self.GetIndexOfItem(dragItem) dragText = self.model.GetText(dragIndex) self.log.write('drop %s %s on %s %s'%(dragText, dragIndex, dropText, dropIndex)) self.model.MoveItem(dragIndex, dropIndex) self.GetParent().RefreshItems() class VirtualTreeCtrl(DemoTreeMixin, wx.TreeCtrl): pass class VirtualTreeListCtrl(DemoTreeMixin, wx.gizmos.TreeListCtrl): def __init__(self, *args, **kwargs): kwargs['style'] = wx.TR_DEFAULT_STYLE | wx.TR_FULL_ROW_HIGHLIGHT super(VirtualTreeListCtrl, self).__init__(*args, **kwargs) self.AddColumn('Column 0') self.AddColumn('Column 1') for art in wx.ART_TIP, wx.ART_WARNING: self.imageList.Add(wx.ArtProvider.GetBitmap(art, wx.ART_OTHER, (16, 16))) def OnGetItemText(self, indices, column=0): # Return a different label depending on column. return '%s, column %d'%\ (super(VirtualTreeListCtrl, self).OnGetItemText(indices), column) def OnGetItemImage(self, indices, which, column=0): # Also change the image of the other columns when the item has # children. if column == 0: return super(VirtualTreeListCtrl, self).OnGetItemImage(indices, which) elif self.OnGetChildrenCount(indices): return 4 else: return 3 class VirtualCustomTreeCtrl(DemoTreeMixin, wx.lib.customtreectrl.CustomTreeCtrl): def __init__(self, *args, **kwargs): self.checked = {} kwargs['ctstyle'] = wx.TR_DEFAULT_STYLE | wx.TR_HIDE_ROOT | \ wx.TR_HAS_BUTTONS | wx.TR_FULL_ROW_HIGHLIGHT super(VirtualCustomTreeCtrl, self).__init__(*args, **kwargs) self.Bind(wx.lib.customtreectrl.EVT_TREE_ITEM_CHECKED, self.OnItemChecked) def OnGetItemType(self, indices): if len(indices) == 1: return 1 elif len(indices) == 2: return 2 else: return 0 def OnGetItemChecked(self, indices): return self.checked.get(indices, False) def OnItemChecked(self, event): item = event.GetItem() itemIndex = self.GetIndexOfItem(item) if self.GetItemType(item) == 2: # It's a radio item; reset other items on the same level for nr in range(self.GetChildrenCount(self.GetItemParent(item))): self.checked[itemIndex[:-1]+(nr,)] = False self.checked[itemIndex] = True class TreeNotebook(wx.Notebook): def __init__(self, *args, **kwargs): treemodel = kwargs.pop('treemodel') log = kwargs.pop('log') super(TreeNotebook, self).__init__(*args, **kwargs) self.trees = [] for class_, title in [(VirtualTreeCtrl, 'TreeCtrl'), (VirtualTreeListCtrl, 'TreeListCtrl'), (VirtualCustomTreeCtrl, 'CustomTreeCtrl')]: tree = class_(self, treemodel=treemodel, log=log) self.trees.append(tree) self.AddPage(tree, title) self.Bind(wx.EVT_NOTEBOOK_PAGE_CHANGED, self.OnPageChanged) def OnPageChanged(self, event): oldTree = self.GetPage(event.OldSelection) newTree = self.GetPage(event.Selection) newTree.RefreshItems() newTree.SetExpansionState(oldTree.GetExpansionState()) event.Skip() def GetIndicesOfSelectedItems(self): tree = self.trees[self.GetSelection()] if tree.GetSelections(): return [tree.GetIndexOfItem(item) for item in tree.GetSelections()] else: return [()] def RefreshItems(self): tree = self.trees[self.GetSelection()] tree.RefreshItems() tree.UnselectAll() class TestPanel(wx.Panel): def __init__(self, parent, log): self.log = log super(TestPanel, self).__init__(parent) self.treemodel = TreeModel() self.CreateControls() self.LayoutControls() def CreateControls(self): self.notebook = TreeNotebook(self, treemodel=self.treemodel, log=self.log) self.label = wx.StaticText(self, label='Number of children: ') self.childrenCountCtrl = wx.SpinCtrl(self, value='0', max=10000) self.button = wx.Button(self, label='Update children') self.button.Bind(wx.EVT_BUTTON, self.OnEnter) def LayoutControls(self): hSizer = wx.BoxSizer(wx.HORIZONTAL) options = dict(flag=wx.ALIGN_CENTER_VERTICAL|wx.ALL, border=2) hSizer.Add(self.label, **options) hSizer.Add(self.childrenCountCtrl, 2, **options) hSizer.Add(self.button, **options) sizer = wx.BoxSizer(wx.VERTICAL) sizer.Add(self.notebook, 1, wx.EXPAND) sizer.Add(hSizer, 0, wx.EXPAND) self.SetSizer(sizer) def OnEnter(self, event): indicesList = self.notebook.GetIndicesOfSelectedItems() newChildrenCount = self.childrenCountCtrl.GetValue() for indices in indicesList: text = self.treemodel.GetText(indices) oldChildrenCount = self.treemodel.GetChildrenCount(indices) self.log.write('%s %s now has %d children (was %d)'%(text, indices, newChildrenCount, oldChildrenCount)) self.treemodel.SetChildrenCount(indices, newChildrenCount) self.notebook.RefreshItems() def runTest(frame, nb, log): win = TestPanel(nb, log) return win if __name__ == '__main__': import sys, os, run run.main(['', os.path.basename(sys.argv[0])] + sys.argv[1:])