]> git.saurik.com Git - wxWidgets.git/blob - samples/treectrl/treetest.cpp
projects / wxWidgets.git / blob
? search:


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