← Back to branch summary

~brian-sidebotham/wxwidgets-cmake/wxpython-2.9.4

« back to all changes in this revision

Viewing changes to wxPython/samples/wxPIA_book/Chapter-15/tree_icons.py

  • Committer: Brian Sidebotham
  • Date: 2013-08-03 14:30:08 UTC
  • Revision ID: brian.sidebotham@gmail.com-20130803143008-c7806tkych1tp6fc
Initial import into Bazaar

Show diffs side-by-side

added added

removed removed

Lines of Context:
wxPython/samples/wxPIA_book/Chapter-15/tree_icons.py
 
1
import wx
 
2
import data
 
3
 
 
4
class TestFrame(wx.Frame):
 
5
    def __init__(self):
 
6
        wx.Frame.__init__(self, None, 
 
7
                title="simple tree with icons", size=(400,500))
 
8
 
 
9
        # Create an image list
 
10
        il = wx.ImageList(16,16)
 
11
 
 
12
        # Get some standard images from the art provider and add them
 
13
        # to the image list
 
14
        self.fldridx = il.Add(
 
15
            wx.ArtProvider.GetBitmap(wx.ART_FOLDER, 
 
16
                    wx.ART_OTHER, (16,16)))
 
17
        self.fldropenidx = il.Add(
 
18
            wx.ArtProvider.GetBitmap(wx.ART_FILE_OPEN,   
 
19
                    wx.ART_OTHER, (16,16)))
 
20
        self.fileidx = il.Add(
 
21
            wx.ArtProvider.GetBitmap(wx.ART_NORMAL_FILE, 
 
22
                    wx.ART_OTHER, (16,16)))
 
23
        
 
24
 
 
25
        # Create the tree
 
26
        self.tree = wx.TreeCtrl(self)
 
27
        # Give it the image list
 
28
        self.tree.AssignImageList(il)
 
29
        root = self.tree.AddRoot("wx.Object")
 
30
        self.tree.SetItemImage(root, self.fldridx,
 
31
                               wx.TreeItemIcon_Normal)
 
32
        self.tree.SetItemImage(root, self.fldropenidx,
 
33
                               wx.TreeItemIcon_Expanded)
 
34
        
 
35
        self.AddTreeNodes(root, data.tree)
 
36
        self.tree.Expand(root)
 
37
        
 
38
 
 
39
    def AddTreeNodes(self, parentItem, items):
 
40
        for item in items:
 
41
            if type(item) == str:
 
42
                newItem = self.tree.AppendItem(parentItem, item)
 
43
                self.tree.SetItemImage(newItem, self.fileidx,
 
44
                                       wx.TreeItemIcon_Normal)
 
45
            else:
 
46
                newItem = self.tree.AppendItem(parentItem, item[0])
 
47
                self.tree.SetItemImage(newItem, self.fldridx,
 
48
                                       wx.TreeItemIcon_Normal)
 
49
                self.tree.SetItemImage(newItem, self.fldropenidx,
 
50
                                       wx.TreeItemIcon_Expanded)
 
51
   
 
52
                self.AddTreeNodes(newItem, item[1])
 
53
                
 
54
 
 
55
    def GetItemText(self, item):
 
56
        if item:
 
57
            return self.tree.GetItemText(item)
 
58
        else:
 
59
            return ""
 
60
      
 
61
app = wx.PySimpleApp(redirect=True)
 
62
frame = TestFrame()
 
63
frame.Show()
 
64
app.MainLoop()
 
65