added a menu command to clear log
[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_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 SetShowButtons(bool show) { m_showButtons = show; }
34 bool ShowButtons() const { return m_showButtons; }
35
36 private:
37 bool m_showImages, m_showButtons;
38 };
39
40 class MyTreeItemData : public wxTreeItemData
41 {
42 public:
43 MyTreeItemData(const wxString& desc) : m_desc(desc) { }
44
45 void ShowInfo(wxTreeCtrl *tree);
46 const wxChar *GetDesc() const { return m_desc.c_str(); }
47
48 private:
49 wxString m_desc;
50 };
51
52 class MyTreeCtrl : public wxTreeCtrl
53 {
54 public:
55 enum
56 {
57 TreeCtrlIcon_File,
58 TreeCtrlIcon_FileSelected,
59 TreeCtrlIcon_Folder,
60 TreeCtrlIcon_FolderSelected,
61 TreeCtrlIcon_FolderOpened
62 };
63
64 MyTreeCtrl() { m_alternateImages = false; }
65 MyTreeCtrl(wxWindow *parent, const wxWindowID id,
66 const wxPoint& pos, const wxSize& size,
67 long style);
68 virtual ~MyTreeCtrl(){};
69
70 void OnBeginDrag(wxTreeEvent& event);
71 void OnBeginRDrag(wxTreeEvent& event);
72 void OnEndDrag(wxTreeEvent& event);
73 void OnBeginLabelEdit(wxTreeEvent& event);
74 void OnEndLabelEdit(wxTreeEvent& event);
75 void OnDeleteItem(wxTreeEvent& event);
76 void OnContextMenu(wxContextMenuEvent& event);
77 void OnItemMenu(wxTreeEvent& event);
78 void OnGetInfo(wxTreeEvent& event);
79 void OnSetInfo(wxTreeEvent& event);
80 void OnItemExpanded(wxTreeEvent& event);
81 void OnItemExpanding(wxTreeEvent& event);
82 void OnItemCollapsed(wxTreeEvent& event);
83 void OnItemCollapsing(wxTreeEvent& event);
84 void OnSelChanged(wxTreeEvent& event);
85 void OnSelChanging(wxTreeEvent& event);
86 void OnTreeKeyDown(wxTreeEvent& event);
87 void OnItemActivated(wxTreeEvent& event);
88 void OnItemRClick(wxTreeEvent& event);
89
90 void OnRMouseDown(wxMouseEvent& event);
91 void OnRMouseUp(wxMouseEvent& event);
92 void OnRMouseDClick(wxMouseEvent& event);
93
94 void GetItemsRecursively(const wxTreeItemId& idParent,
95 wxTreeItemIdValue cookie = 0);
96
97 void CreateImageList(int size = 16);
98 void CreateButtonsImageList(int size = 11);
99
100 void AddTestItemsToTree(size_t numChildren, size_t depth);
101
102 void DoSortChildren(const wxTreeItemId& item, bool reverse = false)
103 { m_reverseSort = reverse; wxTreeCtrl::SortChildren(item); }
104 void DoEnsureVisible() { if (m_lastItem.IsOk()) EnsureVisible(m_lastItem); }
105
106 void DoToggleIcon(const wxTreeItemId& item);
107
108 void ShowMenu(wxTreeItemId id, const wxPoint& pt);
109
110 int ImageSize(void) const { return m_imageSize; }
111
112 void SetLastItem(wxTreeItemId id) { m_lastItem = id; }
113
114 void SetAlternateImages(bool show) { m_alternateImages = show; }
115 bool AlternateImages() const { return m_alternateImages; }
116
117 protected:
118 virtual int OnCompareItems(const wxTreeItemId& i1, const wxTreeItemId& i2);
119
120 // is this the test item which we use in several event handlers?
121 bool IsTestItem(const wxTreeItemId& item)
122 {
123 // the test item is the first child folder
124 return GetItemParent(item) == GetRootItem() && !GetPrevSibling(item);
125 }
126
127 private:
128 void AddItemsRecursively(const wxTreeItemId& idParent,
129 size_t nChildren,
130 size_t depth,
131 size_t folder);
132
133 void LogEvent(const wxChar *name, const wxTreeEvent& event);
134
135 int m_imageSize; // current size of images
136 bool m_reverseSort; // flag for OnCompareItems
137 wxTreeItemId m_lastItem, // for OnEnsureVisible()
138 m_draggedItem; // item being dragged right now
139 bool m_alternateImages;
140
141 // NB: due to an ugly wxMSW hack you _must_ use DECLARE_DYNAMIC_CLASS()
142 // if you want your overloaded OnCompareItems() to be called.
143 // OTOH, if you don't want it you may omit the next line - this will
144 // make default (alphabetical) sorting much faster under wxMSW.
145 DECLARE_DYNAMIC_CLASS(MyTreeCtrl)
146 DECLARE_EVENT_TABLE()
147 };
148
149 // Define a new frame type
150 class MyFrame: public wxFrame
151 {
152 public:
153 // ctor and dtor
154 MyFrame(const wxString& title, int x, int y, int w, int h);
155 virtual ~MyFrame();
156
157 // menu callbacks
158 void OnQuit(wxCommandEvent& event);
159 void OnAbout(wxCommandEvent& event);
160 void OnClearLog(wxCommandEvent& event);
161
162 void OnTogButtons(wxCommandEvent& event)
163 { TogStyle(event.GetId(), wxTR_HAS_BUTTONS); }
164 void OnTogTwist(wxCommandEvent& event)
165 { TogStyle(event.GetId(), wxTR_TWIST_BUTTONS); }
166 void OnTogLines(wxCommandEvent& event)
167 { TogStyle(event.GetId(), wxTR_NO_LINES); }
168 void OnTogEdit(wxCommandEvent& event)
169 { TogStyle(event.GetId(), wxTR_EDIT_LABELS); }
170 void OnTogHideRoot(wxCommandEvent& event)
171 { TogStyle(event.GetId(), wxTR_HIDE_ROOT); }
172 void OnTogRootLines(wxCommandEvent& event)
173 { TogStyle(event.GetId(), wxTR_LINES_AT_ROOT); }
174 void OnTogBorder(wxCommandEvent& event)
175 { TogStyle(event.GetId(), wxTR_ROW_LINES); }
176 void OnTogFullHighlight(wxCommandEvent& event)
177 { TogStyle(event.GetId(), wxTR_FULL_ROW_HIGHLIGHT); }
178
179 void OnResetStyle(wxCommandEvent& WXUNUSED(event))
180 { CreateTreeWithDefStyle(); }
181
182 void OnSetFgColour(wxCommandEvent& event);
183 void OnSetBgColour(wxCommandEvent& event);
184
185 void OnHighlight(wxCommandEvent& event);
186 void OnDump(wxCommandEvent& event);
187 #ifndef NO_MULTIPLE_SELECTION
188 void OnDumpSelected(wxCommandEvent& event);
189 void OnSelect(wxCommandEvent& event);
190 void OnUnselect(wxCommandEvent& event);
191 void OnToggleSel(wxCommandEvent& event);
192 #endif // NO_MULTIPLE_SELECTION
193 void OnSelectRoot(wxCommandEvent& event);
194 void OnDelete(wxCommandEvent& event);
195 void OnDeleteChildren(wxCommandEvent& event);
196 void OnDeleteAll(wxCommandEvent& event);
197
198 void OnRecreate(wxCommandEvent& event);
199 void OnToggleButtons(wxCommandEvent& event);
200 void OnToggleImages(wxCommandEvent& event);
201 void OnToggleAlternateImages(wxCommandEvent& event);
202 void OnSetImageSize(wxCommandEvent& event);
203 void OnCollapseAndReset(wxCommandEvent& event);
204
205 void OnSetBold(wxCommandEvent& WXUNUSED(event)) { DoSetBold(true); }
206 void OnClearBold(wxCommandEvent& WXUNUSED(event)) { DoSetBold(false); }
207
208 void OnEnsureVisible(wxCommandEvent& event);
209
210 void OnCount(wxCommandEvent& event);
211 void OnCountRec(wxCommandEvent& event);
212
213 void OnRename(wxCommandEvent& event);
214 void OnSort(wxCommandEvent& WXUNUSED(event)) { DoSort(); }
215 void OnSortRev(wxCommandEvent& WXUNUSED(event)) { DoSort(true); }
216
217 void OnAddItem(wxCommandEvent& event);
218 void OnInsertItem(wxCommandEvent& event);
219
220 void OnIncIndent(wxCommandEvent& event);
221 void OnDecIndent(wxCommandEvent& event);
222
223 void OnIncSpacing(wxCommandEvent& event);
224 void OnDecSpacing(wxCommandEvent& event);
225
226 void OnToggleIcon(wxCommandEvent& event);
227
228 void OnIdle(wxIdleEvent& event);
229 void OnSize(wxSizeEvent& event);
230
231 private:
232 void TogStyle(int id, long flag);
233
234 void DoSort(bool reverse = false);
235
236 void Resize();
237
238 void CreateTreeWithDefStyle();
239 void CreateTree(long style);
240
241 wxPanel *m_panel;
242 MyTreeCtrl *m_treeCtrl;
243 #if wxUSE_LOG
244 wxTextCtrl *m_textCtrl;
245 #endif // wxUSE_LOG
246
247 void DoSetBold(bool bold = true);
248
249 DECLARE_EVENT_TABLE()
250 };
251
252 // menu and control ids
253 enum
254 {
255 TreeTest_Quit = wxID_EXIT,
256 TreeTest_About = wxID_ABOUT,
257 TreeTest_ClearLog = wxID_CLEAR,
258 TreeTest_TogButtons = wxID_HIGHEST,
259 TreeTest_TogTwist,
260 TreeTest_TogLines,
261 TreeTest_TogEdit,
262 TreeTest_TogHideRoot,
263 TreeTest_TogRootLines,
264 TreeTest_TogBorder,
265 TreeTest_TogFullHighlight,
266 TreeTest_SetFgColour,
267 TreeTest_SetBgColour,
268 TreeTest_ResetStyle,
269 TreeTest_Highlight,
270 TreeTest_Dump,
271 TreeTest_DumpSelected,
272 TreeTest_Count,
273 TreeTest_CountRec,
274 TreeTest_Sort,
275 TreeTest_SortRev,
276 TreeTest_SetBold,
277 TreeTest_ClearBold,
278 TreeTest_Rename,
279 TreeTest_Delete,
280 TreeTest_DeleteChildren,
281 TreeTest_DeleteAll,
282 TreeTest_Recreate,
283 TreeTest_ToggleImages,
284 TreeTest_ToggleAlternateImages,
285 TreeTest_ToggleButtons,
286 TreeTest_SetImageSize,
287 TreeTest_ToggleSel,
288 TreeTest_CollapseAndReset,
289 TreeTest_EnsureVisible,
290 TreeTest_AddItem,
291 TreeTest_InsertItem,
292 TreeTest_IncIndent,
293 TreeTest_DecIndent,
294 TreeTest_IncSpacing,
295 TreeTest_DecSpacing,
296 TreeTest_ToggleIcon,
297 TreeTest_Select,
298 TreeTest_Unselect,
299 TreeTest_SelectRoot,
300 TreeTest_Ctrl = 1000
301 };