Added XML simplification scripts for generating the wxPython metadata xml.

[wxWidgets.git] / wxPython / demo / wxTreeCtrl.py

diff --git a/wxPython/demo/wxTreeCtrl.py b/wxPython/demo/wxTreeCtrl.py
index eff938a684277276c21a8d8ab82307e9dba95de4..547d05c87cd7601fa5b79476dce79a52c669a78e 100644 (file)
--- a/wxPython/demo/wxTreeCtrl.py
+++ b/wxPython/demo/wxTreeCtrl.py
@@ -1,13 +1,17 @@
+# 11/21/2003 - Jeff Grimmett (grimmtooth@softhome.net)
+#
+# o Updated for wx namespace
+# 

 
 

-from wxPython.wx import *
-
-import string
+import  string
+import  wx
+import  images

 
 #---------------------------------------------------------------------------
 
 
 #---------------------------------------------------------------------------
 

-class MyTreeCtrl(wxTreeCtrl):
+class MyTreeCtrl(wx.TreeCtrl):

     def __init__(self, parent, id, pos, size, style, log):
     def __init__(self, parent, id, pos, size, style, log):

-        wxTreeCtrl.__init__(self, parent, id, pos, size, style)
+        wx.TreeCtrl.__init__(self, parent, id, pos, size, style)

         self.log = log
 
     def OnCompareItems(self, item1, item2):
         self.log = log
 
     def OnCompareItems(self, item1, item2):


@@ -20,75 +24,83 @@ class MyTreeCtrl(wxTreeCtrl):
 
 #---------------------------------------------------------------------------
 
 
 #---------------------------------------------------------------------------
 

-class TestTreeCtrlPanel(wxPanel):
+class TestTreeCtrlPanel(wx.Panel):

     def __init__(self, parent, log):
         # Use the WANTS_CHARS style so the panel doesn't eat the Return key.
     def __init__(self, parent, log):
         # Use the WANTS_CHARS style so the panel doesn't eat the Return key.

-        wxPanel.__init__(self, parent, -1, style=wxWANTS_CHARS)
-        EVT_SIZE(self, self.OnSize)
+        wx.Panel.__init__(self, parent, -1, style=wx.WANTS_CHARS)
+        self.Bind(wx.EVT_SIZE, self.OnSize)

 
         self.log = log
 
         self.log = log

-        tID = NewId()
+        tID = wx.NewId()

 
 

-        self.tree = MyTreeCtrl(self, tID, wxDefaultPosition, wxDefaultSize,
-                               wxTR_HAS_BUTTONS | wxTR_EDIT_LABELS# | wxTR_MULTIPLE
+        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)
 
                                , 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())

 
 

-        #import images
-        #il = wxImageList(16, 16)
-        #idx1 = il.Add(images.getSmilesBitmap())
-        #idx2 = il.Add(images.getOpenBitmap())
-        #idx3 = il.Add(images.getNewBitmap())
-        #idx4 = il.Add(images.getCopyBitmap())
-        #idx5 = il.Add(images.getPasteBitmap())
-
-        #self.tree.SetImageList(il)
-        #self.il = il
+        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
 
         # 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.
+        #        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.root = self.tree.AddRoot("The Root Item")
         self.tree.SetPyData(self.root, None)

-        #self.tree.SetItemImage(self.root, idx1)
+        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)
 
         for x in range(15):
             child = self.tree.AppendItem(self.root, "Item %d" % x)
             self.tree.SetPyData(child, None)

-            #self.tree.SetItemImage(child, idx2)
-            #self.tree.SetItemSelectedImage(child, idx3)
+            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)
             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, idx4)
-                #self.tree.SetItemSelectedImage(last, idx5)
+                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)
                 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.tree.Expand(self.root)

-        EVT_TREE_ITEM_EXPANDED  (self, tID, self.OnItemExpanded)
-        EVT_TREE_ITEM_COLLAPSED (self, tID, self.OnItemCollapsed)
-        EVT_TREE_SEL_CHANGED    (self, tID, self.OnSelChanged)
-        EVT_TREE_BEGIN_LABEL_EDIT(self, tID, self.OnBeginEdit)
-        EVT_TREE_END_LABEL_EDIT (self, tID, self.OnEndEdit)
-        EVT_TREE_ITEM_ACTIVATED (self, tID, self.OnActivate)
+        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)

 
 

-        EVT_LEFT_DCLICK(self.tree, self.OnLeftDClick)
-        EVT_RIGHT_DOWN(self.tree, self.OnRightClick)
-        EVT_RIGHT_UP(self.tree, self.OnRightUp)

 
 
     def OnRightClick(self, event):
         pt = event.GetPosition();
         item, flags = self.tree.HitTest(pt)
 
 
     def OnRightClick(self, event):
         pt = event.GetPosition();
         item, flags = self.tree.HitTest(pt)

-        self.log.WriteText("OnRightClick: %s, %s\n" % (self.tree.GetItemText(item),
-                                                       type(item)))
+        self.log.WriteText("OnRightClick: %s, %s, %s\n" %
+                           (self.tree.GetItemText(item), type(item), item.__class__))

         self.tree.SelectItem(item)
 
 
         self.tree.SelectItem(item)
 
 

-

     def OnRightUp(self, event):
         pt = event.GetPosition();
         item, flags = self.tree.HitTest(pt)
     def OnRightUp(self, event):
         pt = event.GetPosition();
         item, flags = self.tree.HitTest(pt)


@@ -102,10 +114,23 @@ class TestTreeCtrlPanel(wxPanel):
         self.log.WriteText("OnBeginEdit\n")
         # show how to prevent edit...
         if self.tree.GetItemText(event.GetItem()) == "The Root Item":
         self.log.WriteText("OnBeginEdit\n")
         # show how to prevent edit...
         if self.tree.GetItemText(event.GetItem()) == "The Root Item":

-            wxBell()
+            wx.Bell()

             self.log.WriteText("You can't edit this one...\n")
             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()
 
             event.Veto()
 

+

     def OnEndEdit(self, event):
         self.log.WriteText("OnEndEdit\n")
         # show how to reject edit, we'll not allow any digits
     def OnEndEdit(self, event):
         self.log.WriteText("OnEndEdit\n")
         # show how to reject edit, we'll not allow any digits


@@ -141,15 +166,15 @@ class TestTreeCtrlPanel(wxPanel):
     def OnSelChanged(self, event):
         self.item = event.GetItem()
         self.log.WriteText("OnSelChanged: %s\n" % self.tree.GetItemText(self.item))
     def OnSelChanged(self, event):
         self.item = event.GetItem()
         self.log.WriteText("OnSelChanged: %s\n" % self.tree.GetItemText(self.item))

-        if wxPlatform == '__WXMSW__':
+        if wx.Platform == '__WXMSW__':

             self.log.WriteText("BoundingRect: %s\n" %
             self.log.WriteText("BoundingRect: %s\n" %

-                               self.tree.GetBoundingRect(self.item, true))
+                               self.tree.GetBoundingRect(self.item, True))

         #items = self.tree.GetSelections()
         #print map(self.tree.GetItemText, items)
         event.Skip()
 
 
         #items = self.tree.GetSelections()
         #print map(self.tree.GetItemText, items)
         event.Skip()
 
 

-    def OnActivate(self, evt):
+    def OnActivate(self, event):

         self.log.WriteText("OnActivate: %s\n" % self.tree.GetItemText(self.item))
 
 
         self.log.WriteText("OnActivate: %s\n" % self.tree.GetItemText(self.item))
 
 


@@ -165,15 +190,17 @@ def runTest(frame, nb, log):
 
 
 
 
 
 

+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])])

 
 

-
-
-
-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.
-
-"""