update bakefile to 0.2.6
[wxWidgets.git] / samples / treectrl / treetest.h
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: treectrl.h
3 // Purpose: wxTreeCtrl sample
4 // Author: Julian Smart
5 // Modified by:
6 // Created: 04/01/98
7 // RCS-ID: $Id$
8 // Copyright: (c) Julian Smart
9 // Licence: wxWindows license
10 /////////////////////////////////////////////////////////////////////////////
11
12 #define USE_GENERIC_TREECTRL 0
13
14 #if USE_GENERIC_TREECTRL
15 #include "wx/generic/treectlg.h"
16 #ifndef wxTreeCtrl
17 #define wxTreeCtrl wxGenericTreeCtrl
18 #define sm_classwxTreeCtrl sm_classwxGenericTreeCtrl
19 #endif
20 #endif
21
22 // Define a new application type
23 class MyApp : public wxApp
24 {
25 public:
26 MyApp() { m_showImages = true; m_showStates = true; m_showButtons = false; }
27
28 bool OnInit();
29
30 void SetShowImages(bool show) { m_showImages = show; }
31 bool ShowImages() const { return m_showImages; }
32
33 void SetShowStates(bool show) { m_showStates = show; }
34 bool ShowStates() const { return m_showStates; }
35
36 void SetShowButtons(bool show) { m_showButtons = show; }
37 bool ShowButtons() const { return m_showButtons; }
38
39 private:
40 bool m_showImages, m_showStates, m_showButtons;
41 };
42
43 class MyTreeItemData : public wxTreeItemData
44 {
45 public:
46 MyTreeItemData(const wxString& desc) : m_desc(desc) { }
47
48 void ShowInfo(wxTreeCtrl *tree);
49 const wxChar *GetDesc() const { return m_desc.c_str(); }
50
51 private:
52 wxString m_desc;
53 };
54
55 class MyTreeCtrl : public wxTreeCtrl
56 {
57 public:
58 enum
59 {
60 TreeCtrlIcon_File,
61 TreeCtrlIcon_FileSelected,
62 TreeCtrlIcon_Folder,
63 TreeCtrlIcon_FolderSelected,
64 TreeCtrlIcon_FolderOpened
65 };
66
67 MyTreeCtrl() { m_alternateImages = false; m_alternateStates = false; }
68 MyTreeCtrl(wxWindow *parent, const wxWindowID id,
69 const wxPoint& pos, const wxSize& size,
70 long style);
71 virtual ~MyTreeCtrl(){};
72
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);
93
94 void OnRMouseDown(wxMouseEvent& event);
95 void OnRMouseUp(wxMouseEvent& event);
96 void OnRMouseDClick(wxMouseEvent& event);
97
98 void GetItemsRecursively(const wxTreeItemId& idParent,
99 wxTreeItemIdValue cookie = 0);
100
101 void CreateImageList(int size = 16);
102 void CreateButtonsImageList(int size = 11);
103 void CreateStateImageList(bool del = false);
104
105 void AddTestItemsToTree(size_t numChildren, size_t depth);
106
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); }
110
111 void DoToggleIcon(const wxTreeItemId& item);
112 void DoToggleState(const wxTreeItemId& item);
113
114 void ShowMenu(wxTreeItemId id, const wxPoint& pt);
115
116 int ImageSize(void) const { return m_imageSize; }
117
118 void SetLastItem(wxTreeItemId id) { m_lastItem = id; }
119
120 void SetAlternateImages(bool show) { m_alternateImages = show; }
121 bool AlternateImages() const { return m_alternateImages; }
122
123 void SetAlternateStates(bool show) { m_alternateStates = show; }
124 bool AlternateStates() const { return m_alternateStates; }
125
126 void ResetBrokenStateImages()
127 {
128 const size_t count = GetStateImageList()->GetImageCount();
129 int state = count > 0 ? count - 1 : wxTREE_ITEMSTATE_NONE;
130 DoResetBrokenStateImages(GetRootItem(), 0, state);
131 }
132
133 protected:
134 virtual int OnCompareItems(const wxTreeItemId& i1, const wxTreeItemId& i2);
135
136 // is this the test item which we use in several event handlers?
137 bool IsTestItem(const wxTreeItemId& item)
138 {
139 // the test item is the first child folder
140 return GetItemParent(item) == GetRootItem() && !GetPrevSibling(item);
141 }
142
143 private:
144 void AddItemsRecursively(const wxTreeItemId& idParent,
145 size_t nChildren,
146 size_t depth,
147 size_t folder);
148
149 void DoResetBrokenStateImages(const wxTreeItemId& idParent,
150 wxTreeItemIdValue cookie, int state);
151
152 void LogEvent(const wxChar *name, const wxTreeEvent& event);
153
154 int m_imageSize; // current size of images
155 bool m_reverseSort; // flag for OnCompareItems
156 wxTreeItemId m_lastItem, // for OnEnsureVisible()
157 m_draggedItem; // item being dragged right now
158 bool m_alternateImages;
159 bool m_alternateStates;
160
161 // NB: due to an ugly wxMSW hack you _must_ use DECLARE_DYNAMIC_CLASS()
162 // if you want your overloaded OnCompareItems() to be called.
163 // OTOH, if you don't want it you may omit the next line - this will
164 // make default (alphabetical) sorting much faster under wxMSW.
165 DECLARE_DYNAMIC_CLASS(MyTreeCtrl)
166 DECLARE_EVENT_TABLE()
167 };
168
169 // Define a new frame type
170 class MyFrame: public wxFrame
171 {
172 public:
173 // ctor and dtor
174 MyFrame(const wxString& title, int x, int y, int w, int h);
175 virtual ~MyFrame();
176
177 // menu callbacks
178 void OnQuit(wxCommandEvent& event);
179 void OnAbout(wxCommandEvent& event);
180 void OnClearLog(wxCommandEvent& event);
181
182 void OnTogButtons(wxCommandEvent& event)
183 { TogStyle(event.GetId(), wxTR_HAS_BUTTONS); }
184 void OnTogTwist(wxCommandEvent& event)
185 { TogStyle(event.GetId(), wxTR_TWIST_BUTTONS); }
186 void OnTogLines(wxCommandEvent& event)
187 { TogStyle(event.GetId(), wxTR_NO_LINES); }
188 void OnTogEdit(wxCommandEvent& event)
189 { TogStyle(event.GetId(), wxTR_EDIT_LABELS); }
190 void OnTogHideRoot(wxCommandEvent& event)
191 { TogStyle(event.GetId(), wxTR_HIDE_ROOT); }
192 void OnTogRootLines(wxCommandEvent& event)
193 { TogStyle(event.GetId(), wxTR_LINES_AT_ROOT); }
194 void OnTogBorder(wxCommandEvent& event)
195 { TogStyle(event.GetId(), wxTR_ROW_LINES); }
196 void OnTogFullHighlight(wxCommandEvent& event)
197 { TogStyle(event.GetId(), wxTR_FULL_ROW_HIGHLIGHT); }
198
199 void OnResetStyle(wxCommandEvent& WXUNUSED(event))
200 { CreateTreeWithDefStyle(); }
201
202 void OnSetFgColour(wxCommandEvent& event);
203 void OnSetBgColour(wxCommandEvent& event);
204
205 void OnHighlight(wxCommandEvent& event);
206 void OnDump(wxCommandEvent& event);
207 #ifndef NO_MULTIPLE_SELECTION
208 void OnDumpSelected(wxCommandEvent& event);
209 void OnSelect(wxCommandEvent& event);
210 void OnUnselect(wxCommandEvent& event);
211 void OnToggleSel(wxCommandEvent& event);
212 #endif // NO_MULTIPLE_SELECTION
213 void OnSelectRoot(wxCommandEvent& event);
214 void OnDelete(wxCommandEvent& event);
215 void OnDeleteChildren(wxCommandEvent& event);
216 void OnDeleteAll(wxCommandEvent& event);
217
218 void OnRecreate(wxCommandEvent& event);
219 void OnToggleButtons(wxCommandEvent& event);
220 void OnToggleImages(wxCommandEvent& event);
221 void OnToggleStates(wxCommandEvent& event);
222 void OnToggleAlternateImages(wxCommandEvent& event);
223 void OnToggleAlternateStates(wxCommandEvent& event);
224 void OnSetImageSize(wxCommandEvent& event);
225 void OnCollapseAndReset(wxCommandEvent& event);
226
227 void OnSetBold(wxCommandEvent& WXUNUSED(event)) { DoSetBold(true); }
228 void OnClearBold(wxCommandEvent& WXUNUSED(event)) { DoSetBold(false); }
229
230 void OnEnsureVisible(wxCommandEvent& event);
231 void OnSetFocus(wxCommandEvent& event);
232
233 void OnCount(wxCommandEvent& event);
234 void OnCountRec(wxCommandEvent& event);
235
236 void OnRename(wxCommandEvent& event);
237 void OnSort(wxCommandEvent& WXUNUSED(event)) { DoSort(); }
238 void OnSortRev(wxCommandEvent& WXUNUSED(event)) { DoSort(true); }
239
240 void OnAddItem(wxCommandEvent& event);
241 void OnInsertItem(wxCommandEvent& event);
242
243 void OnIncIndent(wxCommandEvent& event);
244 void OnDecIndent(wxCommandEvent& event);
245
246 void OnIncSpacing(wxCommandEvent& event);
247 void OnDecSpacing(wxCommandEvent& event);
248
249 void OnToggleIcon(wxCommandEvent& event);
250 void OnToggleState(wxCommandEvent& event);
251
252 void OnShowFirstVisible(wxCommandEvent& WXUNUSED(event))
253 { DoShowFirstOrLast(&wxTreeCtrl::GetFirstVisibleItem, "first visible"); }
254 #ifdef wxHAS_LAST_VISIBLE // we don't have it currently but may add later
255 void OnShowLastVisible(wxCommandEvent& WXUNUSED(event))
256 { DoShowFirstOrLast(&wxTreeCtrl::GetLastVisibleItem, "last visible"); }
257 #endif // wxHAS_LAST_VISIBLE
258
259 void OnShowNextVisible(wxCommandEvent& WXUNUSED(event))
260 { DoShowRelativeItem(&wxTreeCtrl::GetNextVisible, "next visible"); }
261 void OnShowPrevVisible(wxCommandEvent& WXUNUSED(event))
262 { DoShowRelativeItem(&wxTreeCtrl::GetPrevVisible, "previous visible"); }
263
264 void OnShowParent(wxCommandEvent& WXUNUSED(event))
265 { DoShowRelativeItem(&wxTreeCtrl::GetItemParent, "parent"); }
266 void OnShowPrevSibling(wxCommandEvent& WXUNUSED(event))
267 { DoShowRelativeItem(&wxTreeCtrl::GetPrevSibling, "previous sibling"); }
268 void OnShowNextSibling(wxCommandEvent& WXUNUSED(event))
269 { DoShowRelativeItem(&wxTreeCtrl::GetNextSibling, "next sibling"); }
270
271 void OnIdle(wxIdleEvent& event);
272 void OnSize(wxSizeEvent& event);
273
274 private:
275 void TogStyle(int id, long flag);
276
277 void DoSort(bool reverse = false);
278
279 void Resize();
280
281 void CreateTreeWithDefStyle();
282 void CreateTree(long style);
283
284 // common parts of OnShowFirst/LastVisible() and OnShowNext/PrevVisible()
285 typedef wxTreeItemId (wxTreeCtrl::*TreeFunc0_t)() const;
286 void DoShowFirstOrLast(TreeFunc0_t pfn, const wxString& label);
287
288 typedef wxTreeItemId (wxTreeCtrl::*TreeFunc1_t)(const wxTreeItemId&) const;
289 void DoShowRelativeItem(TreeFunc1_t pfn, const wxString& label);
290
291
292 wxPanel *m_panel;
293 MyTreeCtrl *m_treeCtrl;
294 #if wxUSE_LOG
295 wxTextCtrl *m_textCtrl;
296 #endif // wxUSE_LOG
297
298 void DoSetBold(bool bold = true);
299
300 DECLARE_EVENT_TABLE()
301 };
302
303 // menu and control ids
304 enum
305 {
306 TreeTest_Quit = wxID_EXIT,
307 TreeTest_About = wxID_ABOUT,
308 TreeTest_ClearLog = wxID_CLEAR,
309 TreeTest_TogButtons = wxID_HIGHEST,
310 TreeTest_TogTwist,
311 TreeTest_TogLines,
312 TreeTest_TogEdit,
313 TreeTest_TogHideRoot,
314 TreeTest_TogRootLines,
315 TreeTest_TogBorder,
316 TreeTest_TogFullHighlight,
317 TreeTest_SetFgColour,
318 TreeTest_SetBgColour,
319 TreeTest_ResetStyle,
320 TreeTest_Highlight,
321 TreeTest_Dump,
322 TreeTest_DumpSelected,
323 TreeTest_Count,
324 TreeTest_CountRec,
325 TreeTest_Sort,
326 TreeTest_SortRev,
327 TreeTest_SetBold,
328 TreeTest_ClearBold,
329 TreeTest_Rename,
330 TreeTest_Delete,
331 TreeTest_DeleteChildren,
332 TreeTest_DeleteAll,
333 TreeTest_Recreate,
334 TreeTest_ToggleImages,
335 TreeTest_ToggleStates,
336 TreeTest_ToggleAlternateImages,
337 TreeTest_ToggleAlternateStates,
338 TreeTest_ToggleButtons,
339 TreeTest_SetImageSize,
340 TreeTest_ToggleSel,
341 TreeTest_CollapseAndReset,
342 TreeTest_EnsureVisible,
343 TreeTest_SetFocus,
344 TreeTest_AddItem,
345 TreeTest_InsertItem,
346 TreeTest_IncIndent,
347 TreeTest_DecIndent,
348 TreeTest_IncSpacing,
349 TreeTest_DecSpacing,
350 TreeTest_ToggleIcon,
351 TreeTest_ToggleState,
352 TreeTest_Select,
353 TreeTest_Unselect,
354 TreeTest_SelectRoot,
355 TreeTest_ShowFirstVisible,
356 TreeTest_ShowLastVisible,
357 TreeTest_ShowNextVisible,
358 TreeTest_ShowPrevVisible,
359 TreeTest_ShowParent,
360 TreeTest_ShowPrevSibling,
361 TreeTest_ShowNextSibling,
362 TreeTest_Ctrl = 1000
363 };