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: treetest.h 43021 2006-11-04 11:26:51Z VZ $
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:
MyApp()26     MyApp() { m_showImages = true; m_showButtons = false; }
27 
28     bool OnInit();
29 
SetShowImages(bool show)30     void SetShowImages(bool show) { m_showImages = show; }
ShowImages()31     bool ShowImages() const { return m_showImages; }
32 
SetShowButtons(bool show)33     void SetShowButtons(bool show) { m_showButtons = show; }
ShowButtons()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:
MyTreeItemData(const wxString & desc)43     MyTreeItemData(const wxString& desc) : m_desc(desc) { }
44 
45     void ShowInfo(wxTreeCtrl *tree);
GetDesc()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 
MyTreeCtrl()64     MyTreeCtrl() { }
65     MyTreeCtrl(wxWindow *parent, const wxWindowID id,
66                const wxPoint& pos, const wxSize& size,
67                long style);
~MyTreeCtrl()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); }
DoEnsureVisible()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 
ImageSize(void)110     int ImageSize(void) const { return m_imageSize; }
111 
SetLastItem(wxTreeItemId id)112     void SetLastItem(wxTreeItemId id) { m_lastItem = id; }
113 
114 protected:
115     virtual int OnCompareItems(const wxTreeItemId& i1, const wxTreeItemId& i2);
116 
117     // is this the test item which we use in several event handlers?
IsTestItem(const wxTreeItemId & item)118     bool IsTestItem(const wxTreeItemId& item)
119     {
120         // the test item is the first child folder
121         return GetItemParent(item) == GetRootItem() && !GetPrevSibling(item);
122     }
123 
124 private:
125     void AddItemsRecursively(const wxTreeItemId& idParent,
126                              size_t nChildren,
127                              size_t depth,
128                              size_t folder);
129 
130     void LogEvent(const wxChar *name, const wxTreeEvent& event);
131 
132     int          m_imageSize;               // current size of images
133     bool         m_reverseSort;             // flag for OnCompareItems
134     wxTreeItemId m_lastItem,                // for OnEnsureVisible()
135                  m_draggedItem;             // item being dragged right now
136 
137     // NB: due to an ugly wxMSW hack you _must_ use DECLARE_DYNAMIC_CLASS()
138     //     if you want your overloaded OnCompareItems() to be called.
139     //     OTOH, if you don't want it you may omit the next line - this will
140     //     make default (alphabetical) sorting much faster under wxMSW.
141     DECLARE_DYNAMIC_CLASS(MyTreeCtrl)
142     DECLARE_EVENT_TABLE()
143 };
144 
145 // Define a new frame type
146 class MyFrame: public wxFrame
147 {
148 public:
149     // ctor and dtor
150     MyFrame(const wxString& title, int x, int y, int w, int h);
151     virtual ~MyFrame();
152 
153     // menu callbacks
154     void OnQuit(wxCommandEvent& event);
155     void OnAbout(wxCommandEvent& event);
156 
OnTogButtons(wxCommandEvent & event)157     void OnTogButtons(wxCommandEvent& event)
158         { TogStyle(event.GetId(), wxTR_HAS_BUTTONS); }
OnTogTwist(wxCommandEvent & event)159     void OnTogTwist(wxCommandEvent& event)
160         { TogStyle(event.GetId(), wxTR_TWIST_BUTTONS); }
OnTogLines(wxCommandEvent & event)161     void OnTogLines(wxCommandEvent& event)
162         { TogStyle(event.GetId(), wxTR_NO_LINES); }
OnTogEdit(wxCommandEvent & event)163     void OnTogEdit(wxCommandEvent& event)
164         { TogStyle(event.GetId(), wxTR_EDIT_LABELS); }
OnTogHideRoot(wxCommandEvent & event)165     void OnTogHideRoot(wxCommandEvent& event)
166         { TogStyle(event.GetId(), wxTR_HIDE_ROOT); }
OnTogRootLines(wxCommandEvent & event)167     void OnTogRootLines(wxCommandEvent& event)
168         { TogStyle(event.GetId(), wxTR_LINES_AT_ROOT); }
OnTogBorder(wxCommandEvent & event)169     void OnTogBorder(wxCommandEvent& event)
170         { TogStyle(event.GetId(), wxTR_ROW_LINES); }
OnTogFullHighlight(wxCommandEvent & event)171     void OnTogFullHighlight(wxCommandEvent& event)
172         { TogStyle(event.GetId(), wxTR_FULL_ROW_HIGHLIGHT); }
173 
OnResetStyle(wxCommandEvent & WXUNUSED (event))174     void OnResetStyle(wxCommandEvent& WXUNUSED(event))
175         { CreateTreeWithDefStyle(); }
176 
177     void OnSetFgColour(wxCommandEvent& event);
178     void OnSetBgColour(wxCommandEvent& event);
179 
180     void OnHighlight(wxCommandEvent& event);
181     void OnDump(wxCommandEvent& event);
182 #ifndef NO_MULTIPLE_SELECTION
183     void OnDumpSelected(wxCommandEvent& event);
184     void OnSelect(wxCommandEvent& event);
185     void OnUnselect(wxCommandEvent& event);
186     void OnToggleSel(wxCommandEvent& event);
187 #endif // NO_MULTIPLE_SELECTION
188     void OnSelectRoot(wxCommandEvent& event);
189     void OnDelete(wxCommandEvent& event);
190     void OnDeleteChildren(wxCommandEvent& event);
191     void OnDeleteAll(wxCommandEvent& event);
192 
193     void OnRecreate(wxCommandEvent& event);
194     void OnToggleButtons(wxCommandEvent& event);
195     void OnToggleImages(wxCommandEvent& event);
196     void OnSetImageSize(wxCommandEvent& event);
197     void OnCollapseAndReset(wxCommandEvent& event);
198 
OnSetBold(wxCommandEvent & WXUNUSED (event))199     void OnSetBold(wxCommandEvent& WXUNUSED(event)) { DoSetBold(true); }
OnClearBold(wxCommandEvent & WXUNUSED (event))200     void OnClearBold(wxCommandEvent& WXUNUSED(event)) { DoSetBold(false); }
201 
202     void OnEnsureVisible(wxCommandEvent& event);
203 
204     void OnCount(wxCommandEvent& event);
205     void OnCountRec(wxCommandEvent& event);
206 
207     void OnRename(wxCommandEvent& event);
OnSort(wxCommandEvent & WXUNUSED (event))208     void OnSort(wxCommandEvent& WXUNUSED(event)) { DoSort(); }
OnSortRev(wxCommandEvent & WXUNUSED (event))209     void OnSortRev(wxCommandEvent& WXUNUSED(event)) { DoSort(true); }
210 
211     void OnAddItem(wxCommandEvent& event);
212     void OnInsertItem(wxCommandEvent& event);
213 
214     void OnIncIndent(wxCommandEvent& event);
215     void OnDecIndent(wxCommandEvent& event);
216 
217     void OnIncSpacing(wxCommandEvent& event);
218     void OnDecSpacing(wxCommandEvent& event);
219 
220     void OnToggleIcon(wxCommandEvent& event);
221 
222     void OnIdle(wxIdleEvent& event);
223     void OnSize(wxSizeEvent& event);
224 
225 private:
226     void TogStyle(int id, long flag);
227 
228     void DoSort(bool reverse = false);
229 
230     void Resize();
231 
232     void CreateTreeWithDefStyle();
233     void CreateTree(long style);
234 
235     wxPanel *m_panel;
236     MyTreeCtrl *m_treeCtrl;
237 #if wxUSE_LOG
238     wxTextCtrl *m_textCtrl;
239 #endif // wxUSE_LOG
240 
241     void DoSetBold(bool bold = true);
242 
243     DECLARE_EVENT_TABLE()
244 };
245 
246 // menu and control ids
247 enum
248 {
249     TreeTest_Quit = wxID_EXIT,
250     TreeTest_About = wxID_ABOUT,
251     TreeTest_TogButtons = wxID_HIGHEST,
252     TreeTest_TogTwist,
253     TreeTest_TogLines,
254     TreeTest_TogEdit,
255     TreeTest_TogHideRoot,
256     TreeTest_TogRootLines,
257     TreeTest_TogBorder,
258     TreeTest_TogFullHighlight,
259     TreeTest_SetFgColour,
260     TreeTest_SetBgColour,
261     TreeTest_ResetStyle,
262     TreeTest_Highlight,
263     TreeTest_Dump,
264     TreeTest_DumpSelected,
265     TreeTest_Count,
266     TreeTest_CountRec,
267     TreeTest_Sort,
268     TreeTest_SortRev,
269     TreeTest_SetBold,
270     TreeTest_ClearBold,
271     TreeTest_Rename,
272     TreeTest_Delete,
273     TreeTest_DeleteChildren,
274     TreeTest_DeleteAll,
275     TreeTest_Recreate,
276     TreeTest_ToggleImages,
277     TreeTest_ToggleButtons,
278     TreeTest_SetImageSize,
279     TreeTest_ToggleSel,
280     TreeTest_CollapseAndReset,
281     TreeTest_EnsureVisible,
282     TreeTest_AddItem,
283     TreeTest_InsertItem,
284     TreeTest_IncIndent,
285     TreeTest_DecIndent,
286     TreeTest_IncSpacing,
287     TreeTest_DecSpacing,
288     TreeTest_ToggleIcon,
289     TreeTest_Select,
290     TreeTest_Unselect,
291     TreeTest_SelectRoot,
292     TreeTest_Ctrl = 1000
293 };
294