X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/c6281ceb776325251aa119c7b496e6152303996a..299647acac7960652aadb008775429c1f8ea9b8d:/wxPython/demo/TreeCtrl.py diff --git a/wxPython/demo/TreeCtrl.py b/wxPython/demo/TreeCtrl.py new file mode 100644 index 0000000000..547d05c87c --- /dev/null +++ b/wxPython/demo/TreeCtrl.py @@ -0,0 +1,206 @@ +# 11/21/2003 - Jeff Grimmett (grimmtooth@softhome.net) +# +# o Updated for wx namespace +# + +import string +import wx +import images + +#--------------------------------------------------------------------------- + +class MyTreeCtrl(wx.TreeCtrl): + def __init__(self, parent, id, pos, size, style, log): + wx.TreeCtrl.__init__(self, parent, id, pos, size, style) + self.log = log + + def OnCompareItems(self, item1, item2): + t1 = self.GetItemText(item1) + t2 = self.GetItemText(item2) + self.log.WriteText('compare: ' + t1 + ' <> ' + t2 + '\n') + if t1 < t2: return -1 + if t1 == t2: return 0 + return 1 + +#--------------------------------------------------------------------------- + +class TestTreeCtrlPanel(wx.Panel): + def __init__(self, parent, log): + # Use the WANTS_CHARS style so the panel doesn't eat the Return key. + wx.Panel.__init__(self, parent, -1, style=wx.WANTS_CHARS) + self.Bind(wx.EVT_SIZE, self.OnSize) + + self.log = log + tID = wx.NewId() + + self.tree = MyTreeCtrl(self, tID, wx.DefaultPosition, wx.DefaultSize, + wx.TR_HAS_BUTTONS + | wx.TR_EDIT_LABELS + #| wx.TR_MULTIPLE + #| wx.TR_HIDE_ROOT + , self.log) + + isz = (16,16) + il = wx.ImageList(isz[0], isz[1]) + fldridx = il.Add(wx.ArtProvider_GetBitmap(wx.ART_FOLDER, wx.ART_OTHER, isz)) + fldropenidx = il.Add(wx.ArtProvider_GetBitmap(wx.ART_FILE_OPEN, wx.ART_OTHER, isz)) + fileidx = il.Add(wx.ArtProvider_GetBitmap(wx.ART_REPORT_VIEW, wx.ART_OTHER, isz)) + smileidx = il.Add(images.getSmilesBitmap()) + + self.tree.SetImageList(il) + self.il = il + + # NOTE: For some reason tree items have to have a data object in + # order to be sorted. Since our compare just uses the labels + # we don't need any real data, so we'll just use None below for + # the item data. + + self.root = self.tree.AddRoot("The Root Item") + self.tree.SetPyData(self.root, None) + self.tree.SetItemImage(self.root, fldridx, wx.TreeItemIcon_Normal) + self.tree.SetItemImage(self.root, fldropenidx, wx.TreeItemIcon_Expanded) + + + for x in range(15): + child = self.tree.AppendItem(self.root, "Item %d" % x) + self.tree.SetPyData(child, None) + self.tree.SetItemImage(child, fldridx, wx.TreeItemIcon_Normal) + self.tree.SetItemImage(child, fldropenidx, wx.TreeItemIcon_Expanded) + + for y in range(5): + last = self.tree.AppendItem(child, "item %d-%s" % (x, chr(ord("a")+y))) + self.tree.SetPyData(last, None) + self.tree.SetItemImage(last, fldridx, wx.TreeItemIcon_Normal) + self.tree.SetItemImage(last, fldropenidx, wx.TreeItemIcon_Expanded) + + for z in range(5): + item = self.tree.AppendItem(last, "item %d-%s-%d" % (x, chr(ord("a")+y), z)) + self.tree.SetPyData(item, None) + self.tree.SetItemImage(item, fileidx, wx.TreeItemIcon_Normal) + self.tree.SetItemImage(item, smileidx, wx.TreeItemIcon_Selected) + + self.tree.Expand(self.root) + self.Bind(wx.EVT_TREE_ITEM_EXPANDED, self.OnItemExpanded, self.tree) + self.Bind(wx.EVT_TREE_ITEM_COLLAPSED, self.OnItemCollapsed, self.tree) + self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnSelChanged, self.tree) + self.Bind(wx.EVT_TREE_BEGIN_LABEL_EDIT, self.OnBeginEdit, self.tree) + self.Bind(wx.EVT_TREE_END_LABEL_EDIT, self.OnEndEdit, self.tree) + self.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnActivate, self.tree) + + self.tree.Bind(wx.EVT_LEFT_DCLICK, self.OnLeftDClick) + self.tree.Bind(wx.EVT_RIGHT_DOWN, self.OnRightClick) + self.tree.Bind(wx.EVT_RIGHT_UP, self.OnRightUp) + + + + def OnRightClick(self, event): + pt = event.GetPosition(); + item, flags = self.tree.HitTest(pt) + self.log.WriteText("OnRightClick: %s, %s, %s\n" % + (self.tree.GetItemText(item), type(item), item.__class__)) + self.tree.SelectItem(item) + + + def OnRightUp(self, event): + pt = event.GetPosition(); + item, flags = self.tree.HitTest(pt) + self.log.WriteText("OnRightUp: %s (manually starting label edit)\n" + % self.tree.GetItemText(item)) + self.tree.EditLabel(item) + + + + def OnBeginEdit(self, event): + self.log.WriteText("OnBeginEdit\n") + # show how to prevent edit... + if self.tree.GetItemText(event.GetItem()) == "The Root Item": + wx.Bell() + self.log.WriteText("You can't edit this one...\n") + + # Lets just see what's visible of its children + cookie = 0 + root = event.GetItem() + (child, cookie) = self.tree.GetFirstChild(root, cookie) + + while child.IsOk(): + self.log.WriteText("Child [%s] visible = %d" % + (self.tree.GetItemText(child), + self.tree.IsVisible(child))) + (child, cookie) = self.tree.GetNextChild(root, cookie) + + event.Veto() + + + def OnEndEdit(self, event): + self.log.WriteText("OnEndEdit\n") + # show how to reject edit, we'll not allow any digits + for x in event.GetLabel(): + if x in string.digits: + self.log.WriteText("You can't enter digits...\n") + event.Veto() + return + + + def OnLeftDClick(self, event): + pt = event.GetPosition(); + item, flags = self.tree.HitTest(pt) + self.log.WriteText("OnLeftDClick: %s\n" % self.tree.GetItemText(item)) + parent = self.tree.GetItemParent(item) + self.tree.SortChildren(parent) + event.Skip() + + + def OnSize(self, event): + w,h = self.GetClientSizeTuple() + self.tree.SetDimensions(0, 0, w, h) + + + def OnItemExpanded(self, event): + item = event.GetItem() + self.log.WriteText("OnItemExpanded: %s\n" % self.tree.GetItemText(item)) + + def OnItemCollapsed(self, event): + item = event.GetItem() + self.log.WriteText("OnItemCollapsed: %s\n" % self.tree.GetItemText(item)) + + def OnSelChanged(self, event): + self.item = event.GetItem() + self.log.WriteText("OnSelChanged: %s\n" % self.tree.GetItemText(self.item)) + if wx.Platform == '__WXMSW__': + self.log.WriteText("BoundingRect: %s\n" % + self.tree.GetBoundingRect(self.item, True)) + #items = self.tree.GetSelections() + #print map(self.tree.GetItemText, items) + event.Skip() + + + def OnActivate(self, event): + self.log.WriteText("OnActivate: %s\n" % self.tree.GetItemText(self.item)) + + +#--------------------------------------------------------------------------- + +def runTest(frame, nb, log): + win = TestTreeCtrlPanel(nb, log) + return win + +#--------------------------------------------------------------------------- + + + + + +overview = """\ +A tree control presents information as a hierarchy, with items that may be +expanded to show further items. Items in a tree control are referenced by +wxTreeItemId handles. + +""" + + + +if __name__ == '__main__': + import sys,os + import run + run.main(['', os.path.basename(sys.argv[0])]) +