1 /////////////////////////////////////////////////////////////////////////////
2 // Name:        treetest.cpp
3 // Purpose:     wxTreeCtrl sample
4 // Author:      Julian Smart
5 // Modified by:
6 // Created:     04/01/98
7 // RCS-ID:      $Id: treetest.cpp 52196 2008-02-29 19:51:05Z VZ $
8 // Copyright:   (c) Julian Smart
9 // Licence:     wxWindows license
10 /////////////////////////////////////////////////////////////////////////////
11 
12 // For compilers that support precompilation, includes "wx/wx.h".
13 #include "wx/wxprec.h"
14 
15 #ifdef __BORLANDC__
16     #pragma hdrstop
17 #endif
18 
19 #ifndef WX_PRECOMP
20     #include "wx/wx.h"
21     #include "wx/log.h"
22 #endif
23 
24 #include "wx/colordlg.h"
25 #include "wx/numdlg.h"
26 
27 #include "wx/image.h"
28 #include "wx/imaglist.h"
29 #include "wx/treectrl.h"
30 #include "wx/math.h"
31 
32 #ifdef __WIN32__
33     // this is not supported by native control
34     #define NO_VARIABLE_HEIGHT
35 #endif
36 
37 #include "treetest.h"
38 
39 #include "icon1.xpm"
40 #include "icon2.xpm"
41 #include "icon3.xpm"
42 #include "icon4.xpm"
43 #include "icon5.xpm"
44 
45 #ifndef __WXMSW__
46     #include "../sample.xpm"
47 #endif
48 
49 // verify that the item is ok and insult the user if it is not
50 #define CHECK_ITEM( item ) if ( !item.IsOk() ) {                                 \
51                              wxMessageBox(wxT("Please select some item first!"), \
52                                           wxT("Tree sample error"),              \
53                                           wxOK | wxICON_EXCLAMATION,             \
54                                           this);                                 \
55                              return;                                             \
56                            }
57 
58 #define MENU_LINK(name) EVT_MENU(TreeTest_##name, MyFrame::On##name)
59 
BEGIN_EVENT_TABLE(MyFrame,wxFrame)60 BEGIN_EVENT_TABLE(MyFrame, wxFrame)
61     EVT_IDLE(MyFrame::OnIdle)
62     EVT_SIZE(MyFrame::OnSize)
63 
64     MENU_LINK(Quit)
65     MENU_LINK(About)
66 
67     MENU_LINK(TogButtons)
68     MENU_LINK(TogTwist)
69     MENU_LINK(TogLines)
70     MENU_LINK(TogEdit)
71     MENU_LINK(TogHideRoot)
72     MENU_LINK(TogRootLines)
73     MENU_LINK(TogBorder)
74     MENU_LINK(TogFullHighlight)
75     MENU_LINK(SetFgColour)
76     MENU_LINK(SetBgColour)
77     MENU_LINK(ResetStyle)
78 
79     MENU_LINK(Highlight)
80     MENU_LINK(Dump)
81 #ifndef NO_MULTIPLE_SELECTION
82     MENU_LINK(DumpSelected)
83     MENU_LINK(Select)
84     MENU_LINK(Unselect)
85     MENU_LINK(ToggleSel)
86 #endif // NO_MULTIPLE_SELECTION
87     MENU_LINK(Rename)
88     MENU_LINK(Count)
89     MENU_LINK(CountRec)
90     MENU_LINK(Sort)
91     MENU_LINK(SortRev)
92     MENU_LINK(SetBold)
93     MENU_LINK(ClearBold)
94     MENU_LINK(Delete)
95     MENU_LINK(DeleteChildren)
96     MENU_LINK(DeleteAll)
97     MENU_LINK(Recreate)
98     MENU_LINK(ToggleImages)
99     MENU_LINK(ToggleButtons)
100     MENU_LINK(SetImageSize)
101     MENU_LINK(CollapseAndReset)
102     MENU_LINK(EnsureVisible)
103     MENU_LINK(AddItem)
104     MENU_LINK(InsertItem)
105     MENU_LINK(IncIndent)
106     MENU_LINK(DecIndent)
107     MENU_LINK(IncSpacing)
108     MENU_LINK(DecSpacing)
109     MENU_LINK(ToggleIcon)
110     MENU_LINK(SelectRoot)
111 #undef MENU_LINK
112 
113 END_EVENT_TABLE()
114 
115 #if USE_GENERIC_TREECTRL
116 BEGIN_EVENT_TABLE(MyTreeCtrl, wxGenericTreeCtrl)
117 #else
118 BEGIN_EVENT_TABLE(MyTreeCtrl, wxTreeCtrl)
119 #endif
120     EVT_TREE_BEGIN_DRAG(TreeTest_Ctrl, MyTreeCtrl::OnBeginDrag)
121     EVT_TREE_BEGIN_RDRAG(TreeTest_Ctrl, MyTreeCtrl::OnBeginRDrag)
122     EVT_TREE_END_DRAG(TreeTest_Ctrl, MyTreeCtrl::OnEndDrag)
123     EVT_TREE_BEGIN_LABEL_EDIT(TreeTest_Ctrl, MyTreeCtrl::OnBeginLabelEdit)
124     EVT_TREE_END_LABEL_EDIT(TreeTest_Ctrl, MyTreeCtrl::OnEndLabelEdit)
125     EVT_TREE_DELETE_ITEM(TreeTest_Ctrl, MyTreeCtrl::OnDeleteItem)
126 #if 0       // there are so many of those that logging them causes flicker
127     EVT_TREE_GET_INFO(TreeTest_Ctrl, MyTreeCtrl::OnGetInfo)
128 #endif
129     EVT_TREE_SET_INFO(TreeTest_Ctrl, MyTreeCtrl::OnSetInfo)
130     EVT_TREE_ITEM_EXPANDED(TreeTest_Ctrl, MyTreeCtrl::OnItemExpanded)
131     EVT_TREE_ITEM_EXPANDING(TreeTest_Ctrl, MyTreeCtrl::OnItemExpanding)
132     EVT_TREE_ITEM_COLLAPSED(TreeTest_Ctrl, MyTreeCtrl::OnItemCollapsed)
133     EVT_TREE_ITEM_COLLAPSING(TreeTest_Ctrl, MyTreeCtrl::OnItemCollapsing)
134 
135     EVT_TREE_SEL_CHANGED(TreeTest_Ctrl, MyTreeCtrl::OnSelChanged)
136     EVT_TREE_SEL_CHANGING(TreeTest_Ctrl, MyTreeCtrl::OnSelChanging)
137     EVT_TREE_KEY_DOWN(TreeTest_Ctrl, MyTreeCtrl::OnTreeKeyDown)
138     EVT_TREE_ITEM_ACTIVATED(TreeTest_Ctrl, MyTreeCtrl::OnItemActivated)
139 
140     // so many differents ways to handle right mouse button clicks...
141     EVT_CONTEXT_MENU(MyTreeCtrl::OnContextMenu)
142     // EVT_TREE_ITEM_MENU is the preferred event for creating context menus
143     // on a tree control, because it includes the point of the click or item,
144     // meaning that no additional placement calculations are required.
145     EVT_TREE_ITEM_MENU(TreeTest_Ctrl, MyTreeCtrl::OnItemMenu)
146     EVT_TREE_ITEM_RIGHT_CLICK(TreeTest_Ctrl, MyTreeCtrl::OnItemRClick)
147 
148     EVT_RIGHT_DOWN(MyTreeCtrl::OnRMouseDown)
149     EVT_RIGHT_UP(MyTreeCtrl::OnRMouseUp)
150     EVT_RIGHT_DCLICK(MyTreeCtrl::OnRMouseDClick)
151 END_EVENT_TABLE()
152 
153 IMPLEMENT_APP(MyApp)
154 
155 bool MyApp::OnInit()
156 {
157     // Create the main frame window
158     MyFrame *frame = new MyFrame(wxT("wxTreeCtrl Test"), 50, 50, 450, 600);
159 
160     // Show the frame
161     frame->Show(true);
162     SetTopWindow(frame);
163 
164     return true;
165 }
166 
167 
168 // My frame constructor
MyFrame(const wxString & title,int x,int y,int w,int h)169 MyFrame::MyFrame(const wxString& title, int x, int y, int w, int h)
170        : wxFrame((wxFrame *)NULL, wxID_ANY, title, wxPoint(x, y), wxSize(w, h)),
171          m_treeCtrl(NULL)
172 #if wxUSE_LOG
173          , m_textCtrl(NULL)
174 #endif // wxUSE_LOG
175 {
176     // This reduces flicker effects - even better would be to define
177     // OnEraseBackground to do nothing. When the tree control's scrollbars are
178     // show or hidden, the frame is sent a background erase event.
179     SetBackgroundColour(wxColour(255, 255, 255));
180 
181     // Give it an icon
182     SetIcon(wxICON(sample));
183 
184 #if wxUSE_MENUS
185     // Make a menubar
186     wxMenu *file_menu = new wxMenu,
187            *style_menu = new wxMenu,
188            *tree_menu = new wxMenu,
189            *item_menu = new wxMenu;
190 
191     file_menu->Append(TreeTest_About, wxT("&About..."));
192     file_menu->AppendSeparator();
193     file_menu->Append(TreeTest_Quit, wxT("E&xit\tAlt-X"));
194 
195     style_menu->AppendCheckItem(TreeTest_TogButtons, wxT("Toggle &normal buttons"));
196     style_menu->AppendCheckItem(TreeTest_TogTwist, wxT("Toggle &twister buttons"));
197     style_menu->AppendCheckItem(TreeTest_ToggleButtons, wxT("Toggle image &buttons"));
198     style_menu->AppendSeparator();
199     style_menu->AppendCheckItem(TreeTest_TogLines, wxT("Toggle &connecting lines"));
200     style_menu->AppendCheckItem(TreeTest_TogRootLines, wxT("Toggle &lines at root"));
201     style_menu->AppendCheckItem(TreeTest_TogHideRoot, wxT("Toggle &hidden root"));
202     style_menu->AppendCheckItem(TreeTest_TogBorder, wxT("Toggle &item border"));
203     style_menu->AppendCheckItem(TreeTest_TogFullHighlight, wxT("Toggle &full row highlight"));
204     style_menu->AppendCheckItem(TreeTest_TogEdit, wxT("Toggle &edit mode"));
205 #ifndef NO_MULTIPLE_SELECTION
206     style_menu->AppendCheckItem(TreeTest_ToggleSel, wxT("Toggle &selection mode"));
207 #endif // NO_MULTIPLE_SELECTION
208     style_menu->AppendCheckItem(TreeTest_ToggleImages, wxT("Toggle show ima&ges"));
209     style_menu->Append(TreeTest_SetImageSize, wxT("Set image si&ze..."));
210     style_menu->AppendSeparator();
211     style_menu->Append(TreeTest_SetFgColour, wxT("Set &foreground colour..."));
212     style_menu->Append(TreeTest_SetBgColour, wxT("Set &background colour..."));
213     style_menu->AppendSeparator();
214     style_menu->Append(TreeTest_ResetStyle, wxT("&Reset to default\tF10"));
215 
216     tree_menu->Append(TreeTest_Recreate, wxT("&Recreate the tree"));
217     tree_menu->Append(TreeTest_CollapseAndReset, wxT("C&ollapse and reset"));
218     tree_menu->AppendSeparator();
219     tree_menu->Append(TreeTest_AddItem, wxT("Append a &new item"));
220     tree_menu->Append(TreeTest_InsertItem, wxT("&Insert a new item"));
221     tree_menu->Append(TreeTest_Delete, wxT("&Delete this item"));
222     tree_menu->Append(TreeTest_DeleteChildren, wxT("Delete &children"));
223     tree_menu->Append(TreeTest_DeleteAll, wxT("Delete &all items"));
224     tree_menu->Append(TreeTest_SelectRoot, wxT("Select root item"));
225     tree_menu->AppendSeparator();
226     tree_menu->Append(TreeTest_Count, wxT("Count children of current item"));
227     tree_menu->Append(TreeTest_CountRec, wxT("Recursively count children of current item"));
228     tree_menu->AppendSeparator();
229     tree_menu->Append(TreeTest_Sort, wxT("Sort children of current item"));
230     tree_menu->Append(TreeTest_SortRev, wxT("Sort in reversed order"));
231     tree_menu->AppendSeparator();
232     tree_menu->Append(TreeTest_EnsureVisible, wxT("Make the last item &visible"));
233     tree_menu->AppendSeparator();
234     tree_menu->Append(TreeTest_IncIndent, wxT("Add 5 points to indentation\tAlt-I"));
235     tree_menu->Append(TreeTest_DecIndent, wxT("Reduce indentation by 5 points\tAlt-R"));
236     tree_menu->AppendSeparator();
237     tree_menu->Append(TreeTest_IncSpacing, wxT("Add 5 points to spacing\tCtrl-I"));
238     tree_menu->Append(TreeTest_DecSpacing, wxT("Reduce spacing by 5 points\tCtrl-R"));
239 
240     item_menu->Append(TreeTest_Dump, wxT("&Dump item children"));
241     item_menu->Append(TreeTest_Rename, wxT("&Rename item..."));
242 
243     item_menu->AppendSeparator();
244     item_menu->Append(TreeTest_SetBold, wxT("Make item &bold"));
245     item_menu->Append(TreeTest_ClearBold, wxT("Make item &not bold"));
246     item_menu->AppendSeparator();
247     item_menu->Append(TreeTest_ToggleIcon, wxT("Toggle the item's &icon"));
248 
249 #ifndef NO_MULTIPLE_SELECTION
250     item_menu->AppendSeparator();
251     item_menu->Append(TreeTest_DumpSelected, wxT("Dump selected items\tAlt-D"));
252     item_menu->Append(TreeTest_Select, wxT("Select current item\tAlt-S"));
253     item_menu->Append(TreeTest_Unselect, wxT("Unselect everything\tAlt-U"));
254 #endif // NO_MULTIPLE_SELECTION
255 
256     wxMenuBar *menu_bar = new wxMenuBar;
257     menu_bar->Append(file_menu, wxT("&File"));
258     menu_bar->Append(style_menu, wxT("&Style"));
259     menu_bar->Append(tree_menu, wxT("&Tree"));
260     menu_bar->Append(item_menu, wxT("&Item"));
261     SetMenuBar(menu_bar);
262 #endif // wxUSE_MENUS
263 
264     m_panel = new wxPanel(this);
265 
266 #if wxUSE_LOG
267     // create the controls
268     m_textCtrl = new wxTextCtrl(m_panel, wxID_ANY, wxT(""),
269                                 wxDefaultPosition, wxDefaultSize,
270                                 wxTE_MULTILINE | wxSUNKEN_BORDER);
271 #endif // wxUSE_LOG
272 
273     CreateTreeWithDefStyle();
274 
275     menu_bar->Check(TreeTest_ToggleImages, true);
276 
277 #if wxUSE_STATUSBAR
278     // create a status bar
279     CreateStatusBar(2);
280 #endif // wxUSE_STATUSBAR
281 
282 #if wxUSE_LOG
283 #ifdef __WXMOTIF__
284     // For some reason, we get a memcpy crash in wxLogStream::DoLogStream
285     // on gcc/wxMotif, if we use wxLogTextCtl. Maybe it's just gcc?
286     delete wxLog::SetActiveTarget(new wxLogStderr);
287 #else
288     // set our text control as the log target
289     wxLogTextCtrl *logWindow = new wxLogTextCtrl(m_textCtrl);
290     delete wxLog::SetActiveTarget(logWindow);
291 #endif
292 #endif // wxUSE_LOG
293 }
294 
~MyFrame()295 MyFrame::~MyFrame()
296 {
297 #if wxUSE_LOG
298     delete wxLog::SetActiveTarget(NULL);
299 #endif // wxUSE_LOG
300 }
301 
CreateTreeWithDefStyle()302 void MyFrame::CreateTreeWithDefStyle()
303 {
304     long style = wxTR_DEFAULT_STYLE |
305 #ifndef NO_VARIABLE_HEIGHT
306                  wxTR_HAS_VARIABLE_ROW_HEIGHT |
307 #endif
308                  wxTR_EDIT_LABELS;
309 
310     CreateTree(style | wxSUNKEN_BORDER);
311 
312     // as we don't know what wxTR_DEFAULT_STYLE could contain, test for
313     // everything
314     wxMenuBar *mbar = GetMenuBar();
315     mbar->Check(TreeTest_TogButtons, (style & wxTR_HAS_BUTTONS) != 0);
316     mbar->Check(TreeTest_TogButtons, (style & wxTR_TWIST_BUTTONS) != 0);
317     mbar->Check(TreeTest_TogLines, (style & wxTR_NO_LINES) == 0);
318     mbar->Check(TreeTest_TogRootLines, (style & wxTR_LINES_AT_ROOT) != 0);
319     mbar->Check(TreeTest_TogHideRoot, (style & wxTR_HIDE_ROOT) != 0);
320     mbar->Check(TreeTest_TogEdit, (style & wxTR_EDIT_LABELS) != 0);
321     mbar->Check(TreeTest_TogBorder, (style & wxTR_ROW_LINES) != 0);
322     mbar->Check(TreeTest_TogFullHighlight, (style & wxTR_FULL_ROW_HIGHLIGHT) != 0);
323 }
324 
CreateTree(long style)325 void MyFrame::CreateTree(long style)
326 {
327     m_treeCtrl = new MyTreeCtrl(m_panel, TreeTest_Ctrl,
328                                 wxDefaultPosition, wxDefaultSize,
329                                 style);
330 
331     GetMenuBar()->Enable(TreeTest_SelectRoot, !(style & wxTR_HIDE_ROOT));
332 
333     Resize();
334 }
335 
TogStyle(int id,long flag)336 void MyFrame::TogStyle(int id, long flag)
337 {
338     long style = m_treeCtrl->GetWindowStyle() ^ flag;
339 
340     // most treectrl styles can't be changed on the fly using the native
341     // control and the tree must be recreated
342 #ifndef __WXMSW__
343     m_treeCtrl->SetWindowStyle(style);
344 #else // MSW
345     delete m_treeCtrl;
346     CreateTree(style);
347 #endif // !MSW/MSW
348 
349     GetMenuBar()->Check(id, (style & flag) != 0);
350 }
351 
OnIdle(wxIdleEvent & event)352 void MyFrame::OnIdle(wxIdleEvent& event)
353 {
354 #if wxUSE_STATUSBAR
355     if ( m_treeCtrl )
356     {
357         wxTreeItemId idRoot = m_treeCtrl->GetRootItem();
358         wxString status;
359         if (idRoot.IsOk())
360         {
361             wxTreeItemId idLast = m_treeCtrl->GetLastChild(idRoot);
362             status = wxString::Format(
363                 _T("Root/last item is %svisible/%svisible"),
364                 m_treeCtrl->IsVisible(idRoot) ? _T("") : _T("not "),
365                 idLast.IsOk() && m_treeCtrl->IsVisible(idLast)
366                     ? _T("") : _T("not "));
367         }
368         else
369             status = _T("No root item");
370 
371         SetStatusText(status, 1);
372     }
373 #endif // wxUSE_STATUSBAR
374 
375     event.Skip();
376 }
377 
OnSize(wxSizeEvent & event)378 void MyFrame::OnSize(wxSizeEvent& event)
379 {
380     if ( m_treeCtrl
381 #if wxUSE_LOG
382                     && m_textCtrl
383 #endif // wxUSE_LOG
384                                   )
385     {
386         Resize();
387     }
388 
389     event.Skip();
390 }
391 
Resize()392 void MyFrame::Resize()
393 {
394     wxSize size = GetClientSize();
395     m_treeCtrl->SetSize(0, 0, size.x, size.y
396 #if !wxUSE_LOG
397                                             );
398 #else
399                                             *2/3);
400     m_textCtrl->SetSize(0, 2*size.y/3, size.x, size.y/3);
401 #endif
402 }
403 
OnQuit(wxCommandEvent & WXUNUSED (event))404 void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event))
405 {
406     Close(true);
407 }
408 
OnAbout(wxCommandEvent & WXUNUSED (event))409 void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event))
410 {
411     wxMessageBox(wxT("Tree test sample\n")
412                  wxT("(c) Julian Smart 1997, Vadim Zeitlin 1998"),
413                  wxT("About tree test"),
414                  wxOK | wxICON_INFORMATION, this);
415 }
416 
OnRename(wxCommandEvent & WXUNUSED (event))417 void MyFrame::OnRename(wxCommandEvent& WXUNUSED(event))
418 {
419     wxTreeItemId item = m_treeCtrl->GetSelection();
420 
421     CHECK_ITEM( item );
422 
423     // old code - now we edit in place
424 #if 0
425     static wxString s_text;
426     s_text = wxGetTextFromUser(wxT("New name: "), wxT("Tree sample question"),
427             s_text, this);
428     if ( !s_text.empty() )
429     {
430         m_treeCtrl->SetItemText(item, s_text);
431     }
432 #endif // 0
433 
434     // TODO demonstrate creating a custom edit control...
435     (void)m_treeCtrl->EditLabel(item);
436 }
437 
OnCount(wxCommandEvent & WXUNUSED (event))438 void MyFrame::OnCount(wxCommandEvent& WXUNUSED(event))
439 {
440     wxTreeItemId item = m_treeCtrl->GetSelection();
441 
442     CHECK_ITEM( item );
443 
444     int i = m_treeCtrl->GetChildrenCount( item, false );
445 
446     wxLogMessage(wxT("%d children"), i);
447 }
448 
OnCountRec(wxCommandEvent & WXUNUSED (event))449 void MyFrame::OnCountRec(wxCommandEvent& WXUNUSED(event))
450 {
451     wxTreeItemId item = m_treeCtrl->GetSelection();
452 
453     CHECK_ITEM( item );
454 
455     int i = m_treeCtrl->GetChildrenCount( item );
456 
457     wxLogMessage(wxT("%d children"), i);
458 }
459 
DoSort(bool reverse)460 void MyFrame::DoSort(bool reverse)
461 {
462     wxTreeItemId item = m_treeCtrl->GetSelection();
463 
464     CHECK_ITEM( item );
465 
466     m_treeCtrl->DoSortChildren(item, reverse);
467 }
468 
OnHighlight(wxCommandEvent & WXUNUSED (event))469 void MyFrame::OnHighlight(wxCommandEvent& WXUNUSED(event))
470 {
471     wxTreeItemId id = m_treeCtrl->GetSelection();
472 
473     CHECK_ITEM( id );
474 
475     wxRect r;
476     if ( !m_treeCtrl->GetBoundingRect(id, r, true /* text, not full row */) )
477     {
478         wxLogMessage(_T("Failed to get bounding item rect"));
479         return;
480     }
481 
482     wxClientDC dc(m_treeCtrl);
483     dc.SetBrush(*wxRED);
484     dc.SetPen(*wxTRANSPARENT_PEN);
485     dc.DrawRectangle(r);
486     m_treeCtrl->Update();
487 }
488 
OnDump(wxCommandEvent & WXUNUSED (event))489 void MyFrame::OnDump(wxCommandEvent& WXUNUSED(event))
490 {
491     wxTreeItemId root = m_treeCtrl->GetSelection();
492 
493     CHECK_ITEM( root );
494 
495     m_treeCtrl->GetItemsRecursively(root);
496 }
497 
498 #ifndef NO_MULTIPLE_SELECTION
499 
OnToggleSel(wxCommandEvent & event)500 void MyFrame::OnToggleSel(wxCommandEvent& event)
501 {
502     TogStyle(event.GetId(), wxTR_MULTIPLE);
503 }
504 
OnDumpSelected(wxCommandEvent & WXUNUSED (event))505 void MyFrame::OnDumpSelected(wxCommandEvent& WXUNUSED(event))
506 {
507     wxArrayTreeItemIds array;
508 
509     size_t count = m_treeCtrl->GetSelections(array);
510     wxLogMessage(wxT("%u items selected"), unsigned(count));
511 
512     for ( size_t n = 0; n < count; n++ )
513     {
514         wxLogMessage(wxT("\t%s"), m_treeCtrl->GetItemText(array.Item(n)).c_str());
515     }
516 }
517 
OnSelect(wxCommandEvent & WXUNUSED (event))518 void MyFrame::OnSelect(wxCommandEvent& WXUNUSED(event))
519 {
520     m_treeCtrl->SelectItem(m_treeCtrl->GetSelection());
521 }
522 
OnSelectRoot(wxCommandEvent & WXUNUSED (event))523 void MyFrame::OnSelectRoot(wxCommandEvent& WXUNUSED(event))
524 {
525     if ( !m_treeCtrl->HasFlag(wxTR_HIDE_ROOT) )
526         m_treeCtrl->SelectItem(m_treeCtrl->GetRootItem());
527 }
528 
OnUnselect(wxCommandEvent & WXUNUSED (event))529 void MyFrame::OnUnselect(wxCommandEvent& WXUNUSED(event))
530 {
531     m_treeCtrl->UnselectAll();
532 }
533 
534 #endif // NO_MULTIPLE_SELECTION
535 
DoSetBold(bool bold)536 void MyFrame::DoSetBold(bool bold)
537 {
538     wxTreeItemId item = m_treeCtrl->GetSelection();
539 
540     CHECK_ITEM( item );
541 
542     m_treeCtrl->SetItemBold(item, bold);
543 }
544 
OnDelete(wxCommandEvent & WXUNUSED (event))545 void MyFrame::OnDelete(wxCommandEvent& WXUNUSED(event))
546 {
547     wxTreeItemId item = m_treeCtrl->GetSelection();
548 
549     CHECK_ITEM( item );
550 
551     m_treeCtrl->Delete(item);
552 }
553 
OnDeleteChildren(wxCommandEvent & WXUNUSED (event))554 void MyFrame::OnDeleteChildren(wxCommandEvent& WXUNUSED(event))
555 {
556     wxTreeItemId item = m_treeCtrl->GetSelection();
557 
558     CHECK_ITEM( item );
559 
560     m_treeCtrl->DeleteChildren(item);
561 }
562 
OnDeleteAll(wxCommandEvent & WXUNUSED (event))563 void MyFrame::OnDeleteAll(wxCommandEvent& WXUNUSED(event))
564 {
565     m_treeCtrl->DeleteAllItems();
566 }
567 
OnRecreate(wxCommandEvent & event)568 void MyFrame::OnRecreate(wxCommandEvent& event)
569 {
570     OnDeleteAll(event);
571     m_treeCtrl->AddTestItemsToTree(5, 2);
572 }
573 
OnSetImageSize(wxCommandEvent & WXUNUSED (event))574 void MyFrame::OnSetImageSize(wxCommandEvent& WXUNUSED(event))
575 {
576     int size = wxGetNumberFromUser(wxT("Enter the size for the images to use"),
577                                     wxT("Size: "),
578                                     wxT("TreeCtrl sample"),
579                                     m_treeCtrl->ImageSize());
580     if ( size == -1 )
581         return;
582 
583     m_treeCtrl->CreateImageList(size);
584     wxGetApp().SetShowImages(true);
585 }
586 
OnToggleImages(wxCommandEvent & WXUNUSED (event))587 void MyFrame::OnToggleImages(wxCommandEvent& WXUNUSED(event))
588 {
589     if ( wxGetApp().ShowImages() )
590     {
591         m_treeCtrl->CreateImageList(-1);
592         wxGetApp().SetShowImages(false);
593     }
594     else
595     {
596         m_treeCtrl->CreateImageList(0);
597         wxGetApp().SetShowImages(true);
598     }
599 }
600 
OnToggleButtons(wxCommandEvent & WXUNUSED (event))601 void MyFrame::OnToggleButtons(wxCommandEvent& WXUNUSED(event))
602 {
603 #if USE_GENERIC_TREECTRL || !defined(__WXMSW__)
604     if ( wxGetApp().ShowButtons() )
605     {
606         m_treeCtrl->CreateButtonsImageList(-1);
607         wxGetApp().SetShowButtons(false);
608     }
609     else
610     {
611         m_treeCtrl->CreateButtonsImageList(15);
612         wxGetApp().SetShowButtons(true);
613     }
614 #endif
615 }
616 
OnCollapseAndReset(wxCommandEvent & WXUNUSED (event))617 void MyFrame::OnCollapseAndReset(wxCommandEvent& WXUNUSED(event))
618 {
619     m_treeCtrl->CollapseAndReset(m_treeCtrl->GetRootItem());
620 }
621 
OnEnsureVisible(wxCommandEvent & WXUNUSED (event))622 void MyFrame::OnEnsureVisible(wxCommandEvent& WXUNUSED(event))
623 {
624     m_treeCtrl->DoEnsureVisible();
625 }
626 
OnInsertItem(wxCommandEvent & WXUNUSED (event))627 void MyFrame::OnInsertItem(wxCommandEvent& WXUNUSED(event))
628 {
629     int image = wxGetApp().ShowImages() ? MyTreeCtrl::TreeCtrlIcon_File : -1;
630     m_treeCtrl->InsertItem(m_treeCtrl->GetRootItem(), image, wxT("2nd item"));
631 }
632 
OnAddItem(wxCommandEvent & WXUNUSED (event))633 void MyFrame::OnAddItem(wxCommandEvent& WXUNUSED(event))
634 {
635     static int s_num = 0;
636 
637     wxString text;
638     text.Printf(wxT("Item #%d"), ++s_num);
639 
640     m_treeCtrl->AppendItem(m_treeCtrl->GetRootItem(),
641                            text /*,
642                            MyTreeCtrl::TreeCtrlIcon_File */ );
643 }
644 
OnIncIndent(wxCommandEvent & WXUNUSED (event))645 void MyFrame::OnIncIndent(wxCommandEvent& WXUNUSED(event))
646 {
647     unsigned int indent = m_treeCtrl->GetIndent();
648     if (indent < 100)
649         m_treeCtrl->SetIndent( indent+5 );
650 }
651 
OnDecIndent(wxCommandEvent & WXUNUSED (event))652 void MyFrame::OnDecIndent(wxCommandEvent& WXUNUSED(event))
653 {
654     unsigned int indent = m_treeCtrl->GetIndent();
655     if (indent > 10)
656         m_treeCtrl->SetIndent( indent-5 );
657 }
658 
OnIncSpacing(wxCommandEvent & WXUNUSED (event))659 void MyFrame::OnIncSpacing(wxCommandEvent& WXUNUSED(event))
660 {
661     unsigned int indent = m_treeCtrl->GetSpacing();
662     if (indent < 100)
663         m_treeCtrl->SetSpacing( indent+5 );
664 }
665 
OnDecSpacing(wxCommandEvent & WXUNUSED (event))666 void MyFrame::OnDecSpacing(wxCommandEvent& WXUNUSED(event))
667 {
668     unsigned int indent = m_treeCtrl->GetSpacing();
669     if (indent > 10)
670         m_treeCtrl->SetSpacing( indent-5 );
671 }
672 
OnToggleIcon(wxCommandEvent & WXUNUSED (event))673 void MyFrame::OnToggleIcon(wxCommandEvent& WXUNUSED(event))
674 {
675     wxTreeItemId item = m_treeCtrl->GetSelection();
676 
677     CHECK_ITEM( item );
678 
679     m_treeCtrl->DoToggleIcon(item);
680 }
681 
OnSetFgColour(wxCommandEvent & WXUNUSED (event))682 void MyFrame::OnSetFgColour(wxCommandEvent& WXUNUSED(event))
683 {
684     wxColour col = wxGetColourFromUser(this, m_treeCtrl->GetForegroundColour());
685     if ( col.Ok() )
686         m_treeCtrl->SetForegroundColour(col);
687 }
688 
OnSetBgColour(wxCommandEvent & WXUNUSED (event))689 void MyFrame::OnSetBgColour(wxCommandEvent& WXUNUSED(event))
690 {
691     wxColour col = wxGetColourFromUser(this, m_treeCtrl->GetBackgroundColour());
692     if ( col.Ok() )
693         m_treeCtrl->SetBackgroundColour(col);
694 }
695 
696 // MyTreeCtrl implementation
697 #if USE_GENERIC_TREECTRL
IMPLEMENT_DYNAMIC_CLASS(MyTreeCtrl,wxGenericTreeCtrl)698 IMPLEMENT_DYNAMIC_CLASS(MyTreeCtrl, wxGenericTreeCtrl)
699 #else
700 IMPLEMENT_DYNAMIC_CLASS(MyTreeCtrl, wxTreeCtrl)
701 #endif
702 
703 MyTreeCtrl::MyTreeCtrl(wxWindow *parent, const wxWindowID id,
704                        const wxPoint& pos, const wxSize& size,
705                        long style)
706           : wxTreeCtrl(parent, id, pos, size, style)
707 {
708     m_reverseSort = false;
709 
710     CreateImageList();
711 
712     // Add some items to the tree
713     AddTestItemsToTree(5, 2);
714 }
715 
CreateImageList(int size)716 void MyTreeCtrl::CreateImageList(int size)
717 {
718     if ( size == -1 )
719     {
720         SetImageList(NULL);
721         return;
722     }
723     if ( size == 0 )
724         size = m_imageSize;
725     else
726         m_imageSize = size;
727 
728     // Make an image list containing small icons
729     wxImageList *images = new wxImageList(size, size, true);
730 
731     // should correspond to TreeCtrlIcon_xxx enum
732     wxBusyCursor wait;
733     wxIcon icons[5];
734     icons[0] = wxIcon(icon1_xpm);
735     icons[1] = wxIcon(icon2_xpm);
736     icons[2] = wxIcon(icon3_xpm);
737     icons[3] = wxIcon(icon4_xpm);
738     icons[4] = wxIcon(icon5_xpm);
739 
740     int sizeOrig = icons[0].GetWidth();
741     for ( size_t i = 0; i < WXSIZEOF(icons); i++ )
742     {
743         if ( size == sizeOrig )
744         {
745             images->Add(icons[i]);
746         }
747         else
748         {
749             images->Add(wxBitmap(wxBitmap(icons[i]).ConvertToImage().Rescale(size, size)));
750         }
751     }
752 
753     AssignImageList(images);
754 }
755 
756 #if USE_GENERIC_TREECTRL || !defined(__WXMSW__)
CreateButtonsImageList(int size)757 void MyTreeCtrl::CreateButtonsImageList(int size)
758 {
759     if ( size == -1 )
760     {
761         SetButtonsImageList(NULL);
762         return;
763     }
764 
765     // Make an image list containing small icons
766     wxImageList *images = new wxImageList(size, size, true);
767 
768     // should correspond to TreeCtrlIcon_xxx enum
769     wxBusyCursor wait;
770     wxIcon icons[4];
771     icons[0] = wxIcon(icon3_xpm);   // closed
772     icons[1] = wxIcon(icon3_xpm);   // closed, selected
773     icons[2] = wxIcon(icon5_xpm);   // open
774     icons[3] = wxIcon(icon5_xpm);   // open, selected
775 
776     for ( size_t i = 0; i < WXSIZEOF(icons); i++ )
777     {
778         int sizeOrig = icons[i].GetWidth();
779         if ( size == sizeOrig )
780         {
781             images->Add(icons[i]);
782         }
783         else
784         {
785             images->Add(wxBitmap(wxBitmap(icons[i]).ConvertToImage().Rescale(size, size)));
786         }
787     }
788 
789     AssignButtonsImageList(images);
790 #else
791 void MyTreeCtrl::CreateButtonsImageList(int WXUNUSED(size))
792 {
793 #endif
794 }
795 
796 int MyTreeCtrl::OnCompareItems(const wxTreeItemId& item1,
797                                const wxTreeItemId& item2)
798 {
799     if ( m_reverseSort )
800     {
801         // just exchange 1st and 2nd items
802         return wxTreeCtrl::OnCompareItems(item2, item1);
803     }
804     else
805     {
806         return wxTreeCtrl::OnCompareItems(item1, item2);
807     }
808 }
809 
810 void MyTreeCtrl::AddItemsRecursively(const wxTreeItemId& idParent,
811                                      size_t numChildren,
812                                      size_t depth,
813                                      size_t folder)
814 {
815     if ( depth > 0 )
816     {
817         bool hasChildren = depth > 1;
818 
819         wxString str;
820         for ( size_t n = 0; n < numChildren; n++ )
821         {
822             // at depth 1 elements won't have any more children
823             if ( hasChildren )
824                 str.Printf(wxT("%s child %u"), wxT("Folder"), unsigned(n + 1));
825             else
826                 str.Printf(wxT("%s child %u.%u"), wxT("File"), unsigned(folder), unsigned(n + 1));
827 
828             // here we pass to AppendItem() normal and selected item images (we
829             // suppose that selected image follows the normal one in the enum)
830             int image, imageSel;
831             if ( wxGetApp().ShowImages() )
832             {
833                 image = depth == 1 ? TreeCtrlIcon_File : TreeCtrlIcon_Folder;
834                 imageSel = image + 1;
835             }
836             else
837             {
838                 image = imageSel = -1;
839             }
840             wxTreeItemId id = AppendItem(idParent, str, image, imageSel,
841                                          new MyTreeItemData(str));
842 
843             // and now we also set the expanded one (only for the folders)
844             if ( hasChildren && wxGetApp().ShowImages() )
845             {
846                 SetItemImage(id, TreeCtrlIcon_FolderOpened,
847                              wxTreeItemIcon_Expanded);
848             }
849 
850             // remember the last child for OnEnsureVisible()
851             if ( !hasChildren && n == numChildren - 1 )
852             {
853                 m_lastItem = id;
854             }
855 
856             AddItemsRecursively(id, numChildren, depth - 1, n + 1);
857         }
858     }
859     //else: done!
860 }
861 
862 void MyTreeCtrl::AddTestItemsToTree(size_t numChildren,
863                                     size_t depth)
864 {
865     int image = wxGetApp().ShowImages() ? MyTreeCtrl::TreeCtrlIcon_Folder : -1;
866     wxTreeItemId rootId = AddRoot(wxT("Root"),
867                                   image, image,
868                                   new MyTreeItemData(wxT("Root item")));
869     if ( !HasFlag(wxTR_HIDE_ROOT) && image != -1 )
870     {
871         SetItemImage(rootId, TreeCtrlIcon_FolderOpened, wxTreeItemIcon_Expanded);
872     }
873 
874     AddItemsRecursively(rootId, numChildren, depth, 0);
875 
876     // set some colours/fonts for testing
877     if ( !HasFlag(wxTR_HIDE_ROOT) )
878         SetItemFont(rootId, *wxITALIC_FONT);
879 
880     wxTreeItemIdValue cookie;
881     wxTreeItemId id = GetFirstChild(rootId, cookie);
882     SetItemTextColour(id, *wxBLUE);
883 
884     id = GetNextChild(rootId, cookie);
885     id = GetNextChild(rootId, cookie);
886     SetItemTextColour(id, *wxRED);
887     SetItemBackgroundColour(id, *wxLIGHT_GREY);
888 }
889 
890 void MyTreeCtrl::GetItemsRecursively(const wxTreeItemId& idParent,
891                                      wxTreeItemIdValue cookie)
892 {
893     wxTreeItemId id;
894 
895     if ( !cookie )
896         id = GetFirstChild(idParent, cookie);
897     else
898         id = GetNextChild(idParent, cookie);
899 
900     if ( !id.IsOk() )
901         return;
902 
903     wxString text = GetItemText(id);
904     wxLogMessage(text);
905 
906     if (ItemHasChildren(id))
907         GetItemsRecursively(id);
908 
909     GetItemsRecursively(idParent, cookie);
910 }
911 
912 void MyTreeCtrl::DoToggleIcon(const wxTreeItemId& item)
913 {
914     int image = (GetItemImage(item) == TreeCtrlIcon_Folder)
915                     ? TreeCtrlIcon_File
916                     : TreeCtrlIcon_Folder;
917     SetItemImage(item, image, wxTreeItemIcon_Normal);
918 
919     image = (GetItemImage(item) == TreeCtrlIcon_FolderSelected)
920                     ? TreeCtrlIcon_FileSelected
921                     : TreeCtrlIcon_FolderSelected;
922     SetItemImage(item, image, wxTreeItemIcon_Selected);
923 }
924 
925 void MyTreeCtrl::LogEvent(const wxChar *name, const wxTreeEvent& event)
926 {
927     wxTreeItemId item = event.GetItem();
928     wxString text;
929     if ( item.IsOk() )
930         text << _T('"') << GetItemText(item).c_str() << _T('"');
931     else
932         text = _T("invalid item");
933     wxLogMessage(wxT("%s(%s)"), name, text.c_str());
934 }
935 
936 // avoid repetition
937 #define TREE_EVENT_HANDLER(name)                                 \
938 void MyTreeCtrl::name(wxTreeEvent& event)                        \
939 {                                                                \
940     LogEvent(_T(#name), event);                                  \
941     SetLastItem(wxTreeItemId());                                 \
942     event.Skip();                                                \
943 }
944 
945 TREE_EVENT_HANDLER(OnBeginRDrag)
946 TREE_EVENT_HANDLER(OnDeleteItem)
947 TREE_EVENT_HANDLER(OnGetInfo)
948 TREE_EVENT_HANDLER(OnSetInfo)
949 TREE_EVENT_HANDLER(OnItemExpanded)
950 TREE_EVENT_HANDLER(OnItemExpanding)
951 TREE_EVENT_HANDLER(OnItemCollapsed)
952 TREE_EVENT_HANDLER(OnSelChanged)
953 TREE_EVENT_HANDLER(OnSelChanging)
954 
955 #undef TREE_EVENT_HANDLER
956 
957 void LogKeyEvent(const wxChar *name, const wxKeyEvent& event)
958 {
959     wxString key;
960     long keycode = event.GetKeyCode();
961     {
962         switch ( keycode )
963         {
964             case WXK_BACK: key = wxT("BACK"); break;
965             case WXK_TAB: key = wxT("TAB"); break;
966             case WXK_RETURN: key = wxT("RETURN"); break;
967             case WXK_ESCAPE: key = wxT("ESCAPE"); break;
968             case WXK_SPACE: key = wxT("SPACE"); break;
969             case WXK_DELETE: key = wxT("DELETE"); break;
970             case WXK_START: key = wxT("START"); break;
971             case WXK_LBUTTON: key = wxT("LBUTTON"); break;
972             case WXK_RBUTTON: key = wxT("RBUTTON"); break;
973             case WXK_CANCEL: key = wxT("CANCEL"); break;
974             case WXK_MBUTTON: key = wxT("MBUTTON"); break;
975             case WXK_CLEAR: key = wxT("CLEAR"); break;
976             case WXK_SHIFT: key = wxT("SHIFT"); break;
977             case WXK_ALT: key = wxT("ALT"); break;
978             case WXK_CONTROL: key = wxT("CONTROL"); break;
979             case WXK_MENU: key = wxT("MENU"); break;
980             case WXK_PAUSE: key = wxT("PAUSE"); break;
981             case WXK_CAPITAL: key = wxT("CAPITAL"); break;
982             case WXK_END: key = wxT("END"); break;
983             case WXK_HOME: key = wxT("HOME"); break;
984             case WXK_LEFT: key = wxT("LEFT"); break;
985             case WXK_UP: key = wxT("UP"); break;
986             case WXK_RIGHT: key = wxT("RIGHT"); break;
987             case WXK_DOWN: key = wxT("DOWN"); break;
988             case WXK_SELECT: key = wxT("SELECT"); break;
989             case WXK_PRINT: key = wxT("PRINT"); break;
990             case WXK_EXECUTE: key = wxT("EXECUTE"); break;
991             case WXK_SNAPSHOT: key = wxT("SNAPSHOT"); break;
992             case WXK_INSERT: key = wxT("INSERT"); break;
993             case WXK_HELP: key = wxT("HELP"); break;
994             case WXK_NUMPAD0: key = wxT("NUMPAD0"); break;
995             case WXK_NUMPAD1: key = wxT("NUMPAD1"); break;
996             case WXK_NUMPAD2: key = wxT("NUMPAD2"); break;
997             case WXK_NUMPAD3: key = wxT("NUMPAD3"); break;
998             case WXK_NUMPAD4: key = wxT("NUMPAD4"); break;
999             case WXK_NUMPAD5: key = wxT("NUMPAD5"); break;
1000             case WXK_NUMPAD6: key = wxT("NUMPAD6"); break;
1001             case WXK_NUMPAD7: key = wxT("NUMPAD7"); break;
1002             case WXK_NUMPAD8: key = wxT("NUMPAD8"); break;
1003             case WXK_NUMPAD9: key = wxT("NUMPAD9"); break;
1004             case WXK_MULTIPLY: key = wxT("MULTIPLY"); break;
1005             case WXK_ADD: key = wxT("ADD"); break;
1006             case WXK_SEPARATOR: key = wxT("SEPARATOR"); break;
1007             case WXK_SUBTRACT: key = wxT("SUBTRACT"); break;
1008             case WXK_DECIMAL: key = wxT("DECIMAL"); break;
1009             case WXK_DIVIDE: key = wxT("DIVIDE"); break;
1010             case WXK_F1: key = wxT("F1"); break;
1011             case WXK_F2: key = wxT("F2"); break;
1012             case WXK_F3: key = wxT("F3"); break;
1013             case WXK_F4: key = wxT("F4"); break;
1014             case WXK_F5: key = wxT("F5"); break;
1015             case WXK_F6: key = wxT("F6"); break;
1016             case WXK_F7: key = wxT("F7"); break;
1017             case WXK_F8: key = wxT("F8"); break;
1018             case WXK_F9: key = wxT("F9"); break;
1019             case WXK_F10: key = wxT("F10"); break;
1020             case WXK_F11: key = wxT("F11"); break;
1021             case WXK_F12: key = wxT("F12"); break;
1022             case WXK_F13: key = wxT("F13"); break;
1023             case WXK_F14: key = wxT("F14"); break;
1024             case WXK_F15: key = wxT("F15"); break;
1025             case WXK_F16: key = wxT("F16"); break;
1026             case WXK_F17: key = wxT("F17"); break;
1027             case WXK_F18: key = wxT("F18"); break;
1028             case WXK_F19: key = wxT("F19"); break;
1029             case WXK_F20: key = wxT("F20"); break;
1030             case WXK_F21: key = wxT("F21"); break;
1031             case WXK_F22: key = wxT("F22"); break;
1032             case WXK_F23: key = wxT("F23"); break;
1033             case WXK_F24: key = wxT("F24"); break;
1034             case WXK_NUMLOCK: key = wxT("NUMLOCK"); break;
1035             case WXK_SCROLL: key = wxT("SCROLL"); break;
1036             case WXK_PAGEUP: key = wxT("PAGEUP"); break;
1037             case WXK_PAGEDOWN: key = wxT("PAGEDOWN"); break;
1038             case WXK_NUMPAD_SPACE: key = wxT("NUMPAD_SPACE"); break;
1039             case WXK_NUMPAD_TAB: key = wxT("NUMPAD_TAB"); break;
1040             case WXK_NUMPAD_ENTER: key = wxT("NUMPAD_ENTER"); break;
1041             case WXK_NUMPAD_F1: key = wxT("NUMPAD_F1"); break;
1042             case WXK_NUMPAD_F2: key = wxT("NUMPAD_F2"); break;
1043             case WXK_NUMPAD_F3: key = wxT("NUMPAD_F3"); break;
1044             case WXK_NUMPAD_F4: key = wxT("NUMPAD_F4"); break;
1045             case WXK_NUMPAD_HOME: key = wxT("NUMPAD_HOME"); break;
1046             case WXK_NUMPAD_LEFT: key = wxT("NUMPAD_LEFT"); break;
1047             case WXK_NUMPAD_UP: key = wxT("NUMPAD_UP"); break;
1048             case WXK_NUMPAD_RIGHT: key = wxT("NUMPAD_RIGHT"); break;
1049             case WXK_NUMPAD_DOWN: key = wxT("NUMPAD_DOWN"); break;
1050             case WXK_NUMPAD_PAGEUP: key = wxT("NUMPAD_PAGEUP"); break;
1051             case WXK_NUMPAD_PAGEDOWN: key = wxT("NUMPAD_PAGEDOWN"); break;
1052             case WXK_NUMPAD_END: key = wxT("NUMPAD_END"); break;
1053             case WXK_NUMPAD_BEGIN: key = wxT("NUMPAD_BEGIN"); break;
1054             case WXK_NUMPAD_INSERT: key = wxT("NUMPAD_INSERT"); break;
1055             case WXK_NUMPAD_DELETE: key = wxT("NUMPAD_DELETE"); break;
1056             case WXK_NUMPAD_EQUAL: key = wxT("NUMPAD_EQUAL"); break;
1057             case WXK_NUMPAD_MULTIPLY: key = wxT("NUMPAD_MULTIPLY"); break;
1058             case WXK_NUMPAD_ADD: key = wxT("NUMPAD_ADD"); break;
1059             case WXK_NUMPAD_SEPARATOR: key = wxT("NUMPAD_SEPARATOR"); break;
1060             case WXK_NUMPAD_SUBTRACT: key = wxT("NUMPAD_SUBTRACT"); break;
1061             case WXK_NUMPAD_DECIMAL: key = wxT("NUMPAD_DECIMAL"); break;
1062 
1063             default:
1064             {
1065                if ( keycode < 128 && wxIsprint((int)keycode) )
1066                    key.Printf(wxT("'%c'"), (char)keycode);
1067                else if ( keycode > 0 && keycode < 27 )
1068                    key.Printf(_("Ctrl-%c"), wxT('A') + keycode - 1);
1069                else
1070                    key.Printf(wxT("unknown (%ld)"), keycode);
1071             }
1072         }
1073     }
1074 
1075     wxLogMessage( wxT("%s event: %s (flags = %c%c%c%c)"),
1076                   name,
1077                   key.c_str(),
1078                   event.ControlDown() ? wxT('C') : wxT('-'),
1079                   event.AltDown() ? wxT('A') : wxT('-'),
1080                   event.ShiftDown() ? wxT('S') : wxT('-'),
1081                   event.MetaDown() ? wxT('M') : wxT('-'));
1082 }
1083 
1084 void MyTreeCtrl::OnTreeKeyDown(wxTreeEvent& event)
1085 {
1086     LogKeyEvent(wxT("Tree key down "), event.GetKeyEvent());
1087 
1088     event.Skip();
1089 }
1090 
1091 void MyTreeCtrl::OnBeginDrag(wxTreeEvent& event)
1092 {
1093     // need to explicitly allow drag
1094     if ( event.GetItem() != GetRootItem() )
1095     {
1096         m_draggedItem = event.GetItem();
1097 
1098         wxPoint clientpt = event.GetPoint();
1099         wxPoint screenpt = ClientToScreen(clientpt);
1100 
1101         wxLogMessage(wxT("OnBeginDrag: started dragging %s at screen coords (%i,%i)"),
1102                      GetItemText(m_draggedItem).c_str(),
1103                      screenpt.x, screenpt.y);
1104 
1105         event.Allow();
1106     }
1107     else
1108     {
1109         wxLogMessage(wxT("OnBeginDrag: this item can't be dragged."));
1110     }
1111 }
1112 
1113 void MyTreeCtrl::OnEndDrag(wxTreeEvent& event)
1114 {
1115     wxTreeItemId itemSrc = m_draggedItem,
1116                  itemDst = event.GetItem();
1117     m_draggedItem = (wxTreeItemId)0l;
1118 
1119     // where to copy the item?
1120     if ( itemDst.IsOk() && !ItemHasChildren(itemDst) )
1121     {
1122         // copy to the parent then
1123         itemDst = GetItemParent(itemDst);
1124     }
1125 
1126     if ( !itemDst.IsOk() )
1127     {
1128         wxLogMessage(wxT("OnEndDrag: can't drop here."));
1129 
1130         return;
1131     }
1132 
1133     wxString text = GetItemText(itemSrc);
1134     wxLogMessage(wxT("OnEndDrag: '%s' copied to '%s'."),
1135                  text.c_str(), GetItemText(itemDst).c_str());
1136 
1137     // just do append here - we could also insert it just before/after the item
1138     // on which it was dropped, but this requires slightly more work... we also
1139     // completely ignore the client data and icon of the old item but could
1140     // copy them as well.
1141     //
1142     // Finally, we only copy one item here but we might copy the entire tree if
1143     // we were dragging a folder.
1144     int image = wxGetApp().ShowImages() ? TreeCtrlIcon_File : -1;
1145     AppendItem(itemDst, text, image);
1146 }
1147 
1148 void MyTreeCtrl::OnBeginLabelEdit(wxTreeEvent& event)
1149 {
1150     wxLogMessage(wxT("OnBeginLabelEdit"));
1151 
1152     // for testing, prevent this item's label editing
1153     wxTreeItemId itemId = event.GetItem();
1154     if ( IsTestItem(itemId) )
1155     {
1156         wxMessageBox(wxT("You can't edit this item."));
1157 
1158         event.Veto();
1159     }
1160     else if ( itemId == GetRootItem() )
1161     {
1162         // test that it is possible to change the text of the item being edited
1163         SetItemText(itemId, _T("Editing root item"));
1164     }
1165 }
1166 
1167 void MyTreeCtrl::OnEndLabelEdit(wxTreeEvent& event)
1168 {
1169     wxLogMessage(wxT("OnEndLabelEdit"));
1170 
1171     // don't allow anything except letters in the labels
1172     if ( !event.GetLabel().IsWord() )
1173     {
1174         wxMessageBox(wxT("The new label should be a single word."));
1175 
1176         event.Veto();
1177     }
1178 }
1179 
1180 void MyTreeCtrl::OnItemCollapsing(wxTreeEvent& event)
1181 {
1182     wxLogMessage(wxT("OnItemCollapsing"));
1183 
1184     // for testing, prevent the user from collapsing the first child folder
1185     wxTreeItemId itemId = event.GetItem();
1186     if ( IsTestItem(itemId) )
1187     {
1188         wxMessageBox(wxT("You can't collapse this item."));
1189 
1190         event.Veto();
1191     }
1192 }
1193 
1194 void MyTreeCtrl::OnItemActivated(wxTreeEvent& event)
1195 {
1196     // show some info about this item
1197     wxTreeItemId itemId = event.GetItem();
1198     MyTreeItemData *item = (MyTreeItemData *)GetItemData(itemId);
1199 
1200     if ( item != NULL )
1201     {
1202         item->ShowInfo(this);
1203     }
1204 
1205     wxLogMessage(wxT("OnItemActivated"));
1206 }
1207 
1208 void MyTreeCtrl::OnItemMenu(wxTreeEvent& event)
1209 {
1210     wxTreeItemId itemId = event.GetItem();
1211     MyTreeItemData *item = itemId.IsOk() ? (MyTreeItemData *)GetItemData(itemId)
1212                                          : NULL;
1213     wxPoint clientpt = event.GetPoint();
1214     wxPoint screenpt = ClientToScreen(clientpt);
1215 
1216     wxLogMessage(wxT("OnItemMenu for item \"%s\" at screen coords (%i, %i)"),
1217                  item ? item->GetDesc() : _T(""), screenpt.x, screenpt.y);
1218 
1219     ShowMenu(itemId, clientpt);
1220     event.Skip();
1221 }
1222 
1223 void MyTreeCtrl::OnContextMenu(wxContextMenuEvent& event)
1224 {
1225     wxPoint pt = event.GetPosition();
1226 
1227     wxLogMessage(wxT("OnContextMenu at screen coords (%i, %i)"), pt.x, pt.y);
1228 }
1229 
1230 void MyTreeCtrl::ShowMenu(wxTreeItemId id, const wxPoint& pt)
1231 {
1232     wxString title;
1233     if ( id.IsOk() )
1234     {
1235         title << wxT("Menu for ") << GetItemText(id);
1236     }
1237     else
1238     {
1239         title = wxT("Menu for no particular item");
1240     }
1241 
1242 #if wxUSE_MENUS
1243     wxMenu menu(title);
1244     menu.Append(TreeTest_About, wxT("&About..."));
1245     menu.AppendSeparator();
1246     menu.Append(TreeTest_Highlight, wxT("&Highlight item"));
1247     menu.Append(TreeTest_Dump, wxT("&Dump"));
1248 
1249     PopupMenu(&menu, pt);
1250 #endif // wxUSE_MENUS
1251 }
1252 
1253 void MyTreeCtrl::OnItemRClick(wxTreeEvent& event)
1254 {
1255     wxTreeItemId itemId = event.GetItem();
1256     MyTreeItemData *item = itemId.IsOk() ? (MyTreeItemData *)GetItemData(itemId)
1257                                          : NULL;
1258 
1259     wxLogMessage(wxT("Item \"%s\" right clicked"), item ? item->GetDesc()
1260                                                         : _T(""));
1261 
1262     event.Skip();
1263 }
1264 
1265 void MyTreeCtrl::OnRMouseDown(wxMouseEvent& event)
1266 {
1267     wxLogMessage(wxT("Right mouse button down"));
1268 
1269     event.Skip();
1270 }
1271 
1272 void MyTreeCtrl::OnRMouseUp(wxMouseEvent& event)
1273 {
1274     wxLogMessage(wxT("Right mouse button up"));
1275 
1276     event.Skip();
1277 }
1278 
1279 void MyTreeCtrl::OnRMouseDClick(wxMouseEvent& event)
1280 {
1281     wxTreeItemId id = HitTest(event.GetPosition());
1282     if ( !id )
1283         wxLogMessage(wxT("No item under mouse"));
1284     else
1285     {
1286         MyTreeItemData *item = (MyTreeItemData *)GetItemData(id);
1287         if ( item )
1288             wxLogMessage(wxT("Item '%s' under mouse"), item->GetDesc());
1289     }
1290 
1291     event.Skip();
1292 }
1293 
1294 static inline const wxChar *Bool2String(bool b)
1295 {
1296     return b ? wxT("") : wxT("not ");
1297 }
1298 
1299 void MyTreeItemData::ShowInfo(wxTreeCtrl *tree)
1300 {
1301     wxLogMessage(wxT("Item '%s': %sselected, %sexpanded, %sbold,\n")
1302                  wxT("%u children (%u immediately under this item)."),
1303                  m_desc.c_str(),
1304                  Bool2String(tree->IsSelected(GetId())),
1305                  Bool2String(tree->IsExpanded(GetId())),
1306                  Bool2String(tree->IsBold(GetId())),
1307                  unsigned(tree->GetChildrenCount(GetId())),
1308                  unsigned(tree->GetChildrenCount(GetId(), false)));
1309 }
1310