1 /////////////////////////////////////////////////////////////////////////////
3 // Purpose: wxTreeCtrl sample
4 // Author: Julian Smart
8 // Copyright: (c) Julian Smart
9 // Licence: wxWindows license
10 /////////////////////////////////////////////////////////////////////////////
12 #define USE_GENERIC_TREECTRL 0
14 #if USE_GENERIC_TREECTRL
15 #include "wx/generic/treectlg.h"
17 #define wxTreeCtrl wxGenericTreeCtrl
18 #define sm_classwxTreeCtrl sm_classwxGenericTreeCtrl
22 // Define a new application type
23 class MyApp
: public wxApp
26 MyApp() { m_showImages
= true; m_showStates
= true; m_showButtons
= false; }
30 void SetShowImages(bool show
) { m_showImages
= show
; }
31 bool ShowImages() const { return m_showImages
; }
33 void SetShowStates(bool show
) { m_showStates
= show
; }
34 bool ShowStates() const { return m_showStates
; }
36 void SetShowButtons(bool show
) { m_showButtons
= show
; }
37 bool ShowButtons() const { return m_showButtons
; }
40 bool m_showImages
, m_showStates
, m_showButtons
;
43 class MyTreeItemData
: public wxTreeItemData
46 MyTreeItemData(const wxString
& desc
) : m_desc(desc
) { }
48 void ShowInfo(wxTreeCtrl
*tree
);
49 const wxChar
*GetDesc() const { return m_desc
.c_str(); }
55 class MyTreeCtrl
: public wxTreeCtrl
61 TreeCtrlIcon_FileSelected
,
63 TreeCtrlIcon_FolderSelected
,
64 TreeCtrlIcon_FolderOpened
67 MyTreeCtrl() { m_alternateImages
= false; m_alternateStates
= false; }
68 MyTreeCtrl(wxWindow
*parent
, const wxWindowID id
,
69 const wxPoint
& pos
, const wxSize
& size
,
71 virtual ~MyTreeCtrl(){};
73 void OnBeginDrag(wxTreeEvent
& event
);
74 void OnBeginRDrag(wxTreeEvent
& event
);
75 void OnEndDrag(wxTreeEvent
& event
);
76 void OnBeginLabelEdit(wxTreeEvent
& event
);
77 void OnEndLabelEdit(wxTreeEvent
& event
);
78 void OnDeleteItem(wxTreeEvent
& event
);
79 void OnContextMenu(wxContextMenuEvent
& event
);
80 void OnItemMenu(wxTreeEvent
& event
);
81 void OnGetInfo(wxTreeEvent
& event
);
82 void OnSetInfo(wxTreeEvent
& event
);
83 void OnItemExpanded(wxTreeEvent
& event
);
84 void OnItemExpanding(wxTreeEvent
& event
);
85 void OnItemCollapsed(wxTreeEvent
& event
);
86 void OnItemCollapsing(wxTreeEvent
& event
);
87 void OnSelChanged(wxTreeEvent
& event
);
88 void OnSelChanging(wxTreeEvent
& event
);
89 void OnTreeKeyDown(wxTreeEvent
& event
);
90 void OnItemActivated(wxTreeEvent
& event
);
91 void OnItemStateClick(wxTreeEvent
& event
);
92 void OnItemRClick(wxTreeEvent
& event
);
94 void OnRMouseDown(wxMouseEvent
& event
);
95 void OnRMouseUp(wxMouseEvent
& event
);
96 void OnRMouseDClick(wxMouseEvent
& event
);
98 void GetItemsRecursively(const wxTreeItemId
& idParent
,
99 wxTreeItemIdValue cookie
= 0);
101 void CreateImageList(int size
= 16);
102 void CreateButtonsImageList(int size
= 11);
103 void CreateStateImageList(bool del
= false);
105 void AddTestItemsToTree(size_t numChildren
, size_t depth
);
107 void DoSortChildren(const wxTreeItemId
& item
, bool reverse
= false)
108 { m_reverseSort
= reverse
; wxTreeCtrl::SortChildren(item
); }
109 void DoEnsureVisible() { if (m_lastItem
.IsOk()) EnsureVisible(m_lastItem
); }
111 void DoToggleIcon(const wxTreeItemId
& item
);
112 void DoToggleState(const wxTreeItemId
& item
);
114 void ShowMenu(wxTreeItemId id
, const wxPoint
& pt
);
116 int ImageSize(void) const { return m_imageSize
; }
118 void SetLastItem(wxTreeItemId id
) { m_lastItem
= id
; }
120 void SetAlternateImages(bool show
) { m_alternateImages
= show
; }
121 bool AlternateImages() const { return m_alternateImages
; }
123 void SetAlternateStates(bool show
) { m_alternateStates
= show
; }
124 bool AlternateStates() const { return m_alternateStates
; }
127 virtual int OnCompareItems(const wxTreeItemId
& i1
, const wxTreeItemId
& i2
);
129 // is this the test item which we use in several event handlers?
130 bool IsTestItem(const wxTreeItemId
& item
)
132 // the test item is the first child folder
133 return GetItemParent(item
) == GetRootItem() && !GetPrevSibling(item
);
137 void AddItemsRecursively(const wxTreeItemId
& idParent
,
142 void LogEvent(const wxChar
*name
, const wxTreeEvent
& event
);
144 int m_imageSize
; // current size of images
145 bool m_reverseSort
; // flag for OnCompareItems
146 wxTreeItemId m_lastItem
, // for OnEnsureVisible()
147 m_draggedItem
; // item being dragged right now
148 bool m_alternateImages
;
149 bool m_alternateStates
;
151 // NB: due to an ugly wxMSW hack you _must_ use DECLARE_DYNAMIC_CLASS()
152 // if you want your overloaded OnCompareItems() to be called.
153 // OTOH, if you don't want it you may omit the next line - this will
154 // make default (alphabetical) sorting much faster under wxMSW.
155 DECLARE_DYNAMIC_CLASS(MyTreeCtrl
)
156 DECLARE_EVENT_TABLE()
159 // Define a new frame type
160 class MyFrame
: public wxFrame
164 MyFrame(const wxString
& title
, int x
, int y
, int w
, int h
);
168 void OnQuit(wxCommandEvent
& event
);
169 void OnAbout(wxCommandEvent
& event
);
170 void OnClearLog(wxCommandEvent
& event
);
172 void OnTogButtons(wxCommandEvent
& event
)
173 { TogStyle(event
.GetId(), wxTR_HAS_BUTTONS
); }
174 void OnTogTwist(wxCommandEvent
& event
)
175 { TogStyle(event
.GetId(), wxTR_TWIST_BUTTONS
); }
176 void OnTogLines(wxCommandEvent
& event
)
177 { TogStyle(event
.GetId(), wxTR_NO_LINES
); }
178 void OnTogEdit(wxCommandEvent
& event
)
179 { TogStyle(event
.GetId(), wxTR_EDIT_LABELS
); }
180 void OnTogHideRoot(wxCommandEvent
& event
)
181 { TogStyle(event
.GetId(), wxTR_HIDE_ROOT
); }
182 void OnTogRootLines(wxCommandEvent
& event
)
183 { TogStyle(event
.GetId(), wxTR_LINES_AT_ROOT
); }
184 void OnTogBorder(wxCommandEvent
& event
)
185 { TogStyle(event
.GetId(), wxTR_ROW_LINES
); }
186 void OnTogFullHighlight(wxCommandEvent
& event
)
187 { TogStyle(event
.GetId(), wxTR_FULL_ROW_HIGHLIGHT
); }
189 void OnResetStyle(wxCommandEvent
& WXUNUSED(event
))
190 { CreateTreeWithDefStyle(); }
192 void OnSetFgColour(wxCommandEvent
& event
);
193 void OnSetBgColour(wxCommandEvent
& event
);
195 void OnHighlight(wxCommandEvent
& event
);
196 void OnDump(wxCommandEvent
& event
);
197 #ifndef NO_MULTIPLE_SELECTION
198 void OnDumpSelected(wxCommandEvent
& event
);
199 void OnSelect(wxCommandEvent
& event
);
200 void OnUnselect(wxCommandEvent
& event
);
201 void OnToggleSel(wxCommandEvent
& event
);
202 #endif // NO_MULTIPLE_SELECTION
203 void OnSelectRoot(wxCommandEvent
& event
);
204 void OnDelete(wxCommandEvent
& event
);
205 void OnDeleteChildren(wxCommandEvent
& event
);
206 void OnDeleteAll(wxCommandEvent
& event
);
208 void OnRecreate(wxCommandEvent
& event
);
209 void OnToggleButtons(wxCommandEvent
& event
);
210 void OnToggleImages(wxCommandEvent
& event
);
211 void OnToggleStates(wxCommandEvent
& event
);
212 void OnToggleAlternateImages(wxCommandEvent
& event
);
213 void OnToggleAlternateStates(wxCommandEvent
& event
);
214 void OnSetImageSize(wxCommandEvent
& event
);
215 void OnCollapseAndReset(wxCommandEvent
& event
);
217 void OnSetBold(wxCommandEvent
& WXUNUSED(event
)) { DoSetBold(true); }
218 void OnClearBold(wxCommandEvent
& WXUNUSED(event
)) { DoSetBold(false); }
220 void OnEnsureVisible(wxCommandEvent
& event
);
222 void OnCount(wxCommandEvent
& event
);
223 void OnCountRec(wxCommandEvent
& event
);
225 void OnRename(wxCommandEvent
& event
);
226 void OnSort(wxCommandEvent
& WXUNUSED(event
)) { DoSort(); }
227 void OnSortRev(wxCommandEvent
& WXUNUSED(event
)) { DoSort(true); }
229 void OnAddItem(wxCommandEvent
& event
);
230 void OnInsertItem(wxCommandEvent
& event
);
232 void OnIncIndent(wxCommandEvent
& event
);
233 void OnDecIndent(wxCommandEvent
& event
);
235 void OnIncSpacing(wxCommandEvent
& event
);
236 void OnDecSpacing(wxCommandEvent
& event
);
238 void OnToggleIcon(wxCommandEvent
& event
);
239 void OnToggleState(wxCommandEvent
& event
);
241 void OnShowFirstVisible(wxCommandEvent
& WXUNUSED(event
))
242 { DoShowFirstOrLast(&wxTreeCtrl::GetFirstVisibleItem
, "first visible"); }
243 #ifdef wxHAS_LAST_VISIBLE // we don't have it currently but may add later
244 void OnShowLastVisible(wxCommandEvent
& WXUNUSED(event
))
245 { DoShowFirstOrLast(&wxTreeCtrl::GetLastVisibleItem
, "last visible"); }
246 #endif // wxHAS_LAST_VISIBLE
248 void OnShowNextVisible(wxCommandEvent
& WXUNUSED(event
))
249 { DoShowNextOrPrev(&wxTreeCtrl::GetNextVisible
, "next visible"); }
250 void OnShowPrevVisible(wxCommandEvent
& WXUNUSED(event
))
251 { DoShowNextOrPrev(&wxTreeCtrl::GetPrevVisible
, "previous visible"); }
253 void OnIdle(wxIdleEvent
& event
);
254 void OnSize(wxSizeEvent
& event
);
257 void TogStyle(int id
, long flag
);
259 void DoSort(bool reverse
= false);
263 void CreateTreeWithDefStyle();
264 void CreateTree(long style
);
266 // common parts of OnShowFirst/LastVisible() and OnShowNext/PrevVisible()
267 typedef wxTreeItemId (wxTreeCtrl::*TreeFunc0_t
)() const;
268 void DoShowFirstOrLast(TreeFunc0_t pfn
, const wxString
& label
);
270 typedef wxTreeItemId (wxTreeCtrl::*TreeFunc1_t
)(const wxTreeItemId
&) const;
271 void DoShowNextOrPrev(TreeFunc1_t pfn
, const wxString
& label
);
275 MyTreeCtrl
*m_treeCtrl
;
277 wxTextCtrl
*m_textCtrl
;
280 void DoSetBold(bool bold
= true);
282 DECLARE_EVENT_TABLE()
285 // menu and control ids
288 TreeTest_Quit
= wxID_EXIT
,
289 TreeTest_About
= wxID_ABOUT
,
290 TreeTest_ClearLog
= wxID_CLEAR
,
291 TreeTest_TogButtons
= wxID_HIGHEST
,
295 TreeTest_TogHideRoot
,
296 TreeTest_TogRootLines
,
298 TreeTest_TogFullHighlight
,
299 TreeTest_SetFgColour
,
300 TreeTest_SetBgColour
,
304 TreeTest_DumpSelected
,
313 TreeTest_DeleteChildren
,
316 TreeTest_ToggleImages
,
317 TreeTest_ToggleStates
,
318 TreeTest_ToggleAlternateImages
,
319 TreeTest_ToggleAlternateStates
,
320 TreeTest_ToggleButtons
,
321 TreeTest_SetImageSize
,
323 TreeTest_CollapseAndReset
,
324 TreeTest_EnsureVisible
,
332 TreeTest_ToggleState
,
336 TreeTest_ShowFirstVisible
,
337 TreeTest_ShowLastVisible
,
338 TreeTest_ShowNextVisible
,
339 TreeTest_ShowPrevVisible
,