2 from wxPython
.wx
import *
6 #---------------------------------------------------------------------------
8 class MyTreeCtrl(wxTreeCtrl
):
9 def __init__(self
, parent
, id, pos
, size
, style
):
10 wxTreeCtrl
.__init
__(self
, parent
, id, pos
, size
, style
)
13 def OnCompareItems(self
, item1
, item2
):
14 t1
= self
.GetItemText(item1
)
15 t2
= self
.GetItemText(item2
)
20 #---------------------------------------------------------------------------
22 class TestTreeCtrlPanel(wxPanel
):
23 def __init__(self
, parent
, log
):
24 # Use the WANTS_CHARS style so the panel doesn't eat the Return key.
25 wxPanel
.__init
__(self
, parent
, -1, style
=wxWANTS_CHARS
)
26 EVT_SIZE(self
, self
.OnSize
)
31 self
.tree
= MyTreeCtrl(self
, tID
, wxDefaultPosition
, wxDefaultSize
,
32 wxTR_HAS_BUTTONS | wxTR_EDIT_LABELS
)# | wxTR_MULTIPLE)
34 #il = wxImageList(16, 16)
35 #idx1 = il.Add(wxBitmap('bitmaps/smiles.bmp', wxBITMAP_TYPE_BMP))
36 #idx2 = il.Add(wxBitmap('bitmaps/open.bmp', wxBITMAP_TYPE_BMP))
37 #idx3 = il.Add(wxBitmap('bitmaps/new.bmp', wxBITMAP_TYPE_BMP))
38 #idx4 = il.Add(wxBitmap('bitmaps/copy.bmp', wxBITMAP_TYPE_BMP))
39 #idx5 = il.Add(wxBitmap('bitmaps/paste.bmp', wxBITMAP_TYPE_BMP))
41 #self.tree.SetImageList(il)
44 # NOTE: For some reason tree items have to have a data object in
45 # order to be sorted. Since our compare just uses the labels
46 # we don't need any real data, so we'll just use None.
48 self
.root
= self
.tree
.AddRoot("The Root Item")
49 self
.tree
.SetPyData(self
.root
, None)
50 #self.tree.SetItemImage(self.root, idx1)
53 child
= self
.tree
.AppendItem(self
.root
, "Item %d" % x
)
54 self
.tree
.SetPyData(child
, None)
55 #self.tree.SetItemImage(child, idx2)
56 #self.tree.SetItemSelectedImage(child, idx3)
58 last
= self
.tree
.AppendItem(child
, "item %d-%s" % (x
, chr(ord("a")+y
)))
59 self
.tree
.SetPyData(last
, None)
60 #self.tree.SetItemImage(last, idx4)
61 #self.tree.SetItemSelectedImage(last, idx5)
63 item
= self
.tree
.AppendItem(last
, "item %d-%s-%d" % (x
, chr(ord("a")+y
), z
))
64 self
.tree
.SetPyData(item
, None)
66 self
.tree
.Expand(self
.root
)
67 EVT_TREE_ITEM_EXPANDED (self
, tID
, self
.OnItemExpanded
)
68 EVT_TREE_ITEM_COLLAPSED (self
, tID
, self
.OnItemCollapsed
)
69 EVT_TREE_SEL_CHANGED (self
, tID
, self
.OnSelChanged
)
70 EVT_TREE_BEGIN_LABEL_EDIT(self
, tID
, self
.OnBeginEdit
)
71 EVT_TREE_END_LABEL_EDIT (self
, tID
, self
.OnEndEdit
)
72 EVT_TREE_ITEM_ACTIVATED (self
, tID
, self
.OnActivate
)
74 EVT_LEFT_DCLICK(self
.tree
, self
.OnLeftDClick
)
75 EVT_RIGHT_DOWN(self
.tree
, self
.OnRightClick
)
76 EVT_RIGHT_UP(self
.tree
, self
.OnRightUp
)
79 def OnRightClick(self
, event
):
80 pt
= event
.GetPosition();
81 item
, flags
= self
.tree
.HitTest(pt
)
82 self
.log
.WriteText("OnRightClick: %s, %s\n" % (self
.tree
.GetItemText(item
),
84 self
.tree
.SelectItem(item
)
88 def OnRightUp(self
, event
):
89 pt
= event
.GetPosition();
90 item
, flags
= self
.tree
.HitTest(pt
)
91 self
.log
.WriteText("OnRightUp: %s (manually starting label edit)\n"
92 % self
.tree
.GetItemText(item
))
93 self
.tree
.EditLabel(item
)
97 def OnBeginEdit(self
, event
):
98 self
.log
.WriteText("OnBeginEdit\n")
99 # show how to prevent edit...
100 if self
.tree
.GetItemText(event
.GetItem()) == "The Root Item":
102 self
.log
.WriteText("You can't edit this one...\n")
105 def OnEndEdit(self
, event
):
106 self
.log
.WriteText("OnEndEdit\n")
107 # show how to reject edit, we'll not allow any digits
108 for x
in event
.GetLabel():
109 if x
in string
.digits
:
110 self
.log
.WriteText("You can't enter digits...\n")
115 def OnLeftDClick(self
, event
):
116 pt
= event
.GetPosition();
117 item
, flags
= self
.tree
.HitTest(pt
)
118 self
.log
.WriteText("OnLeftDClick: %s\n" % self
.tree
.GetItemText(item
))
119 parent
= self
.tree
.GetItemParent(item
)
120 self
.tree
.SortChildren(parent
)
124 def OnSize(self
, event
):
125 w
,h
= self
.GetClientSizeTuple()
126 self
.tree
.SetDimensions(0, 0, w
, h
)
129 def OnItemExpanded(self
, event
):
130 item
= event
.GetItem()
131 self
.log
.WriteText("OnItemExpanded: %s\n" % self
.tree
.GetItemText(item
))
133 def OnItemCollapsed(self
, event
):
134 item
= event
.GetItem()
135 self
.log
.WriteText("OnItemCollapsed: %s\n" % self
.tree
.GetItemText(item
))
137 def OnSelChanged(self
, event
):
138 self
.item
= event
.GetItem()
139 self
.log
.WriteText("OnSelChanged: %s\n" % self
.tree
.GetItemText(self
.item
))
140 if wxPlatform
== '__WXMSW__':
141 self
.log
.WriteText("BoundingRect: %s\n" %
142 self
.tree
.GetBoundingRect(self
.item
, true
))
143 #items = self.tree.GetSelections()
144 #print map(self.tree.GetItemText, items)
148 def OnActivate(self
, evt
):
149 self
.log
.WriteText("OnActivate: %s\n" % self
.tree
.GetItemText(self
.item
))
152 #---------------------------------------------------------------------------
154 def runTest(frame
, nb
, log
):
155 win
= TestTreeCtrlPanel(nb
, log
)
158 #---------------------------------------------------------------------------
173 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.
176 -------------------------
180 wxTreeCtrl(wxWindow* parent, wxWindowID id, const wxPoint& pos = wxDefaultPosition, const wxSize& size = wxDefaultSize, long style = wxTR_HAS_BUTTONS, const wxValidator& validator = wxDefaultValidator, const wxString& name = "listCtrl")
182 Constructor, creating and showing a tree control.
187 parent = Parent window. Must not be NULL.
189 id = Window identifier. A value of -1 indicates a default value.
191 pos = Window position.
193 size = Window size. If the default size (-1, -1) is specified then the window is sized appropriately.
195 style = Window style. See wxTreeCtrl.
197 validator = Window validator.