]> git.saurik.com Git - wxWidgets.git/blame_incremental - samples/treectrl/treetest.cpp
Small changes
[wxWidgets.git] / samples / treectrl / treetest.cpp
... / ...
CommitLineData
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 and Markus Holzem
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#endif
27
28#include "wx/log.h"
29
30#include "wx/image.h"
31#include "wx/imaglist.h"
32#include "wx/treectrl.h"
33
34#include "math.h"
35
36#ifdef __WXMSW__
37 // comment out this line to test multiple selection even under MSW (where
38 // it looks ugly - but works)
39 #define NO_MULTIPLE_SELECTION
40#endif
41
42#define NO_VARIABLE_HEIGHT
43
44#include "treetest.h"
45
46// under Windows the icons are in the .rc file
47#ifndef __WXMSW__
48 #include "icon1.xpm"
49 #include "icon2.xpm"
50 #include "icon3.xpm"
51 #include "icon4.xpm"
52 #include "icon5.xpm"
53 #include "mondrian.xpm"
54#endif
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("Please select some item first!", \
59 "Tree sample error", \
60 wxOK | wxICON_EXCLAMATION, \
61 this); \
62 return; \
63 }
64
65BEGIN_EVENT_TABLE(MyFrame, wxFrame)
66 EVT_MENU(TreeTest_Quit, MyFrame::OnQuit)
67 EVT_MENU(TreeTest_About, MyFrame::OnAbout)
68 EVT_MENU(TreeTest_Dump, MyFrame::OnDump)
69#ifndef NO_MULTIPLE_SELECTION
70 EVT_MENU(TreeTest_DumpSelected, MyFrame::OnDumpSelected)
71 EVT_MENU(TreeTest_Select, MyFrame::OnSelect)
72 EVT_MENU(TreeTest_Unselect, MyFrame::OnUnselect)
73#endif // NO_MULTIPLE_SELECTION
74 EVT_MENU(TreeTest_Rename, MyFrame::OnRename)
75 EVT_MENU(TreeTest_Count, MyFrame::OnCount)
76 EVT_MENU(TreeTest_CountRec, MyFrame::OnCountRec)
77 EVT_MENU(TreeTest_Sort, MyFrame::OnSort)
78 EVT_MENU(TreeTest_SortRev, MyFrame::OnSortRev)
79 EVT_MENU(TreeTest_Bold, MyFrame::OnSetBold)
80 EVT_MENU(TreeTest_UnBold, MyFrame::OnClearBold)
81 EVT_MENU(TreeTest_Delete, MyFrame::OnDelete)
82 EVT_MENU(TreeTest_DeleteChildren, MyFrame::OnDeleteChildren)
83 EVT_MENU(TreeTest_DeleteAll, MyFrame::OnDeleteAll)
84 EVT_MENU(TreeTest_Recreate, MyFrame::OnRecreate)
85 EVT_MENU(TreeTest_CollapseAndReset, MyFrame::OnCollapseAndReset)
86 EVT_MENU(TreeTest_EnsureVisible, MyFrame::OnEnsureVisible)
87 EVT_MENU(TreeTest_AddItem, MyFrame::OnAddItem)
88 EVT_MENU(TreeTest_IncIndent, MyFrame::OnIncIndent)
89 EVT_MENU(TreeTest_DecIndent, MyFrame::OnDecIndent)
90 EVT_MENU(TreeTest_IncSpacing, MyFrame::OnIncSpacing)
91 EVT_MENU(TreeTest_DecSpacing, MyFrame::OnDecSpacing)
92 EVT_MENU(TreeTest_ToggleIcon, MyFrame::OnToggleIcon)
93END_EVENT_TABLE()
94
95BEGIN_EVENT_TABLE(MyTreeCtrl, wxTreeCtrl)
96 EVT_TREE_BEGIN_DRAG(TreeTest_Ctrl, MyTreeCtrl::OnBeginDrag)
97 EVT_TREE_BEGIN_RDRAG(TreeTest_Ctrl, MyTreeCtrl::OnBeginRDrag)
98 EVT_TREE_BEGIN_LABEL_EDIT(TreeTest_Ctrl, MyTreeCtrl::OnBeginLabelEdit)
99 EVT_TREE_END_LABEL_EDIT(TreeTest_Ctrl, MyTreeCtrl::OnEndLabelEdit)
100 EVT_TREE_DELETE_ITEM(TreeTest_Ctrl, MyTreeCtrl::OnDeleteItem)
101#if 0 // there are so many of those that logging them causes flicker
102 EVT_TREE_GET_INFO(TreeTest_Ctrl, MyTreeCtrl::OnGetInfo)
103#endif
104 EVT_TREE_SET_INFO(TreeTest_Ctrl, MyTreeCtrl::OnSetInfo)
105 EVT_TREE_ITEM_EXPANDED(TreeTest_Ctrl, MyTreeCtrl::OnItemExpanded)
106 EVT_TREE_ITEM_EXPANDING(TreeTest_Ctrl, MyTreeCtrl::OnItemExpanding)
107 EVT_TREE_ITEM_COLLAPSED(TreeTest_Ctrl, MyTreeCtrl::OnItemCollapsed)
108 EVT_TREE_ITEM_COLLAPSING(TreeTest_Ctrl, MyTreeCtrl::OnItemCollapsing)
109 EVT_TREE_SEL_CHANGED(TreeTest_Ctrl, MyTreeCtrl::OnSelChanged)
110 EVT_TREE_SEL_CHANGING(TreeTest_Ctrl, MyTreeCtrl::OnSelChanging)
111 EVT_TREE_KEY_DOWN(TreeTest_Ctrl, MyTreeCtrl::OnTreeKeyDown)
112 EVT_TREE_ITEM_ACTIVATED(TreeTest_Ctrl, MyTreeCtrl::OnItemActivated)
113 EVT_RIGHT_DCLICK(MyTreeCtrl::OnRMouseDClick)
114END_EVENT_TABLE()
115
116IMPLEMENT_APP(MyApp)
117
118bool MyApp::OnInit()
119{
120 // Create the main frame window
121 MyFrame *frame = new MyFrame("wxTreeCtrl Test", 50, 50, 450, 340);
122
123 // Show the frame
124 frame->Show(TRUE);
125 SetTopWindow(frame);
126
127 return TRUE;
128}
129
130
131// My frame constructor
132MyFrame::MyFrame(const wxString& title, int x, int y, int w, int h)
133 : wxFrame((wxFrame *)NULL, -1, title, wxPoint(x, y), wxSize(w, h))
134{
135 // This reduces flicker effects - even better would be to define
136 // OnEraseBackground to do nothing. When the tree control's scrollbars are
137 // show or hidden, the frame is sent a background erase event.
138 SetBackgroundColour(wxColour(255, 255, 255));
139
140 // Give it an icon
141 SetIcon(wxICON(mondrian));
142
143 // Make a menubar
144 wxMenu *file_menu = new wxMenu,
145 *tree_menu = new wxMenu,
146 *item_menu = new wxMenu;
147
148 file_menu->Append(TreeTest_About, "&About...");
149 file_menu->AppendSeparator();
150 file_menu->Append(TreeTest_Quit, "E&xit");
151
152 tree_menu->Append(TreeTest_Recreate, "&Recreate the tree");
153 tree_menu->Append(TreeTest_CollapseAndReset, "C&ollapse and reset");
154 tree_menu->AppendSeparator();
155 tree_menu->Append(TreeTest_AddItem, "Append a &new item");
156 tree_menu->Append(TreeTest_Delete, "&Delete this item");
157 tree_menu->Append(TreeTest_DeleteChildren, "Delete &children");
158 tree_menu->Append(TreeTest_DeleteAll, "Delete &all items");
159 tree_menu->AppendSeparator();
160 tree_menu->Append(TreeTest_Count, "Count children of current item");
161 tree_menu->Append(TreeTest_CountRec, "Recursively count children of current item");
162 tree_menu->AppendSeparator();
163 tree_menu->Append(TreeTest_Sort, "Sort children of current item");
164 tree_menu->Append(TreeTest_SortRev, "Sort in reversed order");
165 tree_menu->AppendSeparator();
166 tree_menu->Append(TreeTest_EnsureVisible, "Make the last item &visible");
167 tree_menu->AppendSeparator();
168 tree_menu->Append(TreeTest_IncIndent, "Add 5 points to indentation\tAlt-I");
169 tree_menu->Append(TreeTest_DecIndent, "Reduce indentation by 5 points\tAlt-R");
170 tree_menu->AppendSeparator();
171 tree_menu->Append(TreeTest_IncSpacing, "Add 5 points to spacing\tCtrl-I");
172 tree_menu->Append(TreeTest_DecSpacing, "Reduce spacing by 5 points\tCtrl-R");
173
174 item_menu->Append(TreeTest_Dump, "&Dump item children");
175 item_menu->Append(TreeTest_Rename, "&Rename item...");
176
177 item_menu->AppendSeparator();
178 item_menu->Append(TreeTest_Bold, "Make item &bold");
179 item_menu->Append(TreeTest_UnBold, "Make item &not bold");
180 item_menu->AppendSeparator();
181 item_menu->Append(TreeTest_ToggleIcon, "Toggle the items &icon");
182
183#ifndef NO_MULTIPLE_SELECTION
184 item_menu->AppendSeparator();
185 item_menu->Append(TreeTest_DumpSelected, "Dump selected items\tAlt-D");
186 item_menu->Append(TreeTest_Select, "Select current item\tAlt-S");
187 item_menu->Append(TreeTest_Unselect, "Unselect everything\tAlt-U");
188#endif
189
190 wxMenuBar *menu_bar = new wxMenuBar;
191 menu_bar->Append(file_menu, "&File");
192 menu_bar->Append(tree_menu, "&Tree");
193 menu_bar->Append(item_menu, "&Item");
194 SetMenuBar(menu_bar);
195
196 m_treeCtrl = new MyTreeCtrl(this, TreeTest_Ctrl,
197 wxDefaultPosition, wxDefaultSize,
198 wxTR_HAS_BUTTONS |
199 wxTR_EDIT_LABELS |
200#ifndef NO_MULTIPLE_SELECTION
201 wxTR_MULTIPLE |
202#endif
203#ifndef NO_VARIABLE_HEIGHT
204 wxTR_HAS_VARIABLE_ROW_HEIGHT |
205#endif
206 wxSUNKEN_BORDER);
207 wxTextCtrl *textCtrl = new wxTextCtrl(this, -1, "",
208 wxDefaultPosition, wxDefaultSize,
209 wxTE_MULTILINE | wxSUNKEN_BORDER);
210
211 wxLayoutConstraints *c = new wxLayoutConstraints;
212 c->top.SameAs(this, wxTop);
213 c->left.SameAs(this, wxLeft);
214 c->right.SameAs(this, wxRight);
215 c->height.PercentOf(this, wxHeight, 66);
216 m_treeCtrl->SetConstraints(c);
217
218 c = new wxLayoutConstraints;
219 c->top.Below(m_treeCtrl);
220 c->left.SameAs(this, wxLeft);
221 c->right.SameAs(this, wxRight);
222 c->bottom.SameAs(this, wxBottom);
223 textCtrl->SetConstraints(c);
224 SetAutoLayout(TRUE);
225
226 // create a status bar with 3 panes
227 CreateStatusBar(3);
228 SetStatusText("", 0);
229
230#ifdef __WXMOTIF__
231 // For some reason, we get a memcpy crash in wxLogStream::DoLogStream
232 // on gcc/wxMotif, if we use wxLogTextCtl. Maybe it's just gcc?
233 delete wxLog::SetActiveTarget(new wxLogStderr);
234#else
235 // set our text control as the log target
236 wxLogTextCtrl *logWindow = new wxLogTextCtrl(textCtrl);
237 delete wxLog::SetActiveTarget(logWindow);
238#endif
239}
240
241MyFrame::~MyFrame()
242{
243 delete wxLog::SetActiveTarget(NULL);
244}
245
246void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event))
247{
248 Close(TRUE);
249}
250
251void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event))
252{
253 wxMessageBox("Tree test sample\n"
254 "(c) Julian Smart 1997, Vadim Zeitlin 1998",
255 "About tree test",
256 wxOK | wxICON_INFORMATION, this);
257}
258
259void MyFrame::OnRename(wxCommandEvent& WXUNUSED(event))
260{
261 wxTreeItemId item = m_treeCtrl->GetSelection();
262
263 CHECK_ITEM( item );
264
265 // old code - now we edit in place
266#if 0
267 static wxString s_text;
268 s_text = wxGetTextFromUser("New name: ", "Tree sample question",
269 s_text, this);
270 if ( !s_text.IsEmpty() )
271 {
272 m_treeCtrl->SetItemText(item, s_text);
273 }
274#endif // 0
275
276 // TODO demonstrate creating a custom edit control...
277 (void)m_treeCtrl->EditLabel(item);
278}
279
280void MyFrame::OnCount(wxCommandEvent& WXUNUSED(event))
281{
282 wxTreeItemId item = m_treeCtrl->GetSelection();
283
284 CHECK_ITEM( item );
285
286 int i = m_treeCtrl->GetChildrenCount( item, FALSE );
287
288 wxLogMessage(wxT("%d children"), i);
289}
290
291void MyFrame::OnCountRec(wxCommandEvent& WXUNUSED(event))
292{
293 wxTreeItemId item = m_treeCtrl->GetSelection();
294
295 CHECK_ITEM( item );
296
297 int i = m_treeCtrl->GetChildrenCount( item );
298
299 wxLogMessage(wxT("%d children"), i);
300}
301
302void MyFrame::DoSort(bool reverse)
303{
304 wxTreeItemId item = m_treeCtrl->GetSelection();
305
306 CHECK_ITEM( item );
307
308 m_treeCtrl->DoSortChildren(item, reverse);
309}
310
311void MyFrame::OnDump(wxCommandEvent& WXUNUSED(event))
312{
313 wxTreeItemId root = m_treeCtrl->GetSelection();
314
315 CHECK_ITEM( root );
316
317 m_treeCtrl->GetItemsRecursively(root, -1);
318}
319
320#ifndef NO_MULTIPLE_SELECTION
321
322void MyFrame::OnDumpSelected(wxCommandEvent& WXUNUSED(event))
323{
324 wxArrayTreeItemIds array;
325
326 size_t count = m_treeCtrl->GetSelections(array);
327 wxLogMessage(wxT("%u items selected"), count);
328
329 for ( size_t n = 0; n < count; n++ )
330 {
331 wxLogMessage("\t%s", m_treeCtrl->GetItemText(array.Item(n)).c_str());
332 }
333}
334
335void MyFrame::OnSelect(wxCommandEvent& event)
336{
337 m_treeCtrl->SelectItem(m_treeCtrl->GetSelection());
338}
339
340void MyFrame::OnUnselect(wxCommandEvent& event)
341{
342 m_treeCtrl->UnselectAll();
343}
344
345#endif // NO_MULTIPLE_SELECTION
346
347void MyFrame::DoSetBold(bool bold)
348{
349 wxTreeItemId item = m_treeCtrl->GetSelection();
350
351 CHECK_ITEM( item );
352
353 m_treeCtrl->SetItemBold(item, bold);
354}
355
356void MyFrame::OnDelete(wxCommandEvent& WXUNUSED(event))
357{
358 wxTreeItemId item = m_treeCtrl->GetSelection();
359
360 CHECK_ITEM( item );
361
362 m_treeCtrl->Delete(item);
363}
364
365void MyFrame::OnDeleteChildren(wxCommandEvent& WXUNUSED(event))
366{
367 wxTreeItemId item = m_treeCtrl->GetSelection();
368
369 CHECK_ITEM( item );
370
371 m_treeCtrl->DeleteChildren(item);
372}
373
374void MyFrame::OnDeleteAll(wxCommandEvent& WXUNUSED(event))
375{
376 m_treeCtrl->DeleteAllItems();
377}
378
379void MyFrame::OnRecreate(wxCommandEvent& event)
380{
381 OnDeleteAll(event);
382 m_treeCtrl->AddTestItemsToTree(3, 2);
383}
384
385void MyFrame::OnCollapseAndReset(wxCommandEvent& event)
386{
387 m_treeCtrl->CollapseAndReset(m_treeCtrl->GetRootItem());
388}
389
390void MyFrame::OnEnsureVisible(wxCommandEvent& event)
391{
392 m_treeCtrl->DoEnsureVisible();
393}
394
395void MyFrame::OnAddItem(wxCommandEvent& WXUNUSED(event))
396{
397 static int s_num = 0;
398
399 wxString text;
400 text.Printf("Item #%d", ++s_num);
401
402 m_treeCtrl->AppendItem(m_treeCtrl->GetRootItem(),
403 text /*,
404 MyTreeCtrl::TreeCtrlIcon_File */ );
405}
406
407void MyFrame::OnIncIndent(wxCommandEvent& WXUNUSED(event))
408{
409 unsigned int indent = m_treeCtrl->GetIndent();
410 if (indent < 100)
411 m_treeCtrl->SetIndent( indent+5 );
412}
413
414void MyFrame::OnDecIndent(wxCommandEvent& WXUNUSED(event))
415{
416 unsigned int indent = m_treeCtrl->GetIndent();
417 if (indent > 10)
418 m_treeCtrl->SetIndent( indent-5 );
419}
420
421void MyFrame::OnIncSpacing(wxCommandEvent& WXUNUSED(event))
422{
423 unsigned int indent = m_treeCtrl->GetSpacing();
424 if (indent < 100)
425 m_treeCtrl->SetSpacing( indent+5 );
426}
427
428void MyFrame::OnDecSpacing(wxCommandEvent& WXUNUSED(event))
429{
430 unsigned int indent = m_treeCtrl->GetSpacing();
431 if (indent > 10)
432 m_treeCtrl->SetSpacing( indent-5 );
433}
434
435void MyFrame::OnToggleIcon(wxCommandEvent& WXUNUSED(event))
436{
437 wxTreeItemId item = m_treeCtrl->GetSelection();
438
439 CHECK_ITEM( item );
440
441 m_treeCtrl->DoToggleIcon(item);
442}
443
444// MyTreeCtrl implementation
445IMPLEMENT_DYNAMIC_CLASS(MyTreeCtrl, wxTreeCtrl)
446
447MyTreeCtrl::MyTreeCtrl(wxWindow *parent, const wxWindowID id,
448 const wxPoint& pos, const wxSize& size,
449 long style)
450 : wxTreeCtrl(parent, id, pos, size, style)
451{
452#ifndef NO_VARIABLE_HEIGHT
453#if wxUSE_LIBJPEG
454 wxImage::AddHandler(new wxJPEGHandler);
455 wxImage image;
456
457 image.LoadFile(wxString("horse.jpg"), wxBITMAP_TYPE_JPEG );
458#endif
459#endif
460
461 m_reverseSort = FALSE;
462
463 // Make an image list containing small icons
464 m_imageListNormal = new wxImageList(16, 16, TRUE);
465
466 // should correspond to TreeCtrlIcon_xxx enum
467#if defined(__WXMSW__) && defined(__WIN16__)
468 // This is required in 16-bit Windows mode only because we can't load a specific (16x16)
469 // icon image, so it comes out stretched
470# ifndef NO_VARIABLE_HEIGHT
471 m_imageListNormal->Add(image.ConvertToBitmap());
472# else
473 m_imageListNormal->Add(wxBitmap("bitmap1", wxBITMAP_TYPE_BMP_RESOURCE));
474# endif
475 m_imageListNormal->Add(wxBitmap("bitmap2", wxBITMAP_TYPE_BMP_RESOURCE));
476 m_imageListNormal->Add(wxBitmap("bitmap3", wxBITMAP_TYPE_BMP_RESOURCE));
477 m_imageListNormal->Add(wxBitmap("bitmap4", wxBITMAP_TYPE_BMP_RESOURCE));
478 m_imageListNormal->Add(wxBitmap("bitmap5", wxBITMAP_TYPE_BMP_RESOURCE));
479#else
480# ifndef NO_VARIABLE_HEIGHT
481 m_imageListNormal->Add(image.ConvertToBitmap());
482# else
483 m_imageListNormal->Add(wxICON(icon1));
484# endif
485 m_imageListNormal->Add(wxICON(icon2));
486 m_imageListNormal->Add(wxICON(icon3));
487 m_imageListNormal->Add(wxICON(icon4));
488 m_imageListNormal->Add(wxICON(icon5));
489#endif
490
491 SetImageList(m_imageListNormal);
492
493 // Add some items to the tree
494 AddTestItemsToTree(3, 2);
495}
496
497MyTreeCtrl::~MyTreeCtrl()
498{
499 delete m_imageListNormal;
500}
501
502int MyTreeCtrl::OnCompareItems(const wxTreeItemId& item1,
503 const wxTreeItemId& item2)
504{
505 if ( m_reverseSort )
506 {
507 // just exchange 1st and 2nd items
508 return wxTreeCtrl::OnCompareItems(item2, item1);
509 }
510 else
511 {
512 return wxTreeCtrl::OnCompareItems(item1, item2);
513 }
514}
515
516void MyTreeCtrl::AddItemsRecursively(const wxTreeItemId& idParent,
517 size_t numChildren,
518 size_t depth,
519 size_t folder)
520{
521 if ( depth > 0 )
522 {
523 bool hasChildren = depth > 1;
524
525 wxString str;
526 for ( size_t n = 0; n < numChildren; n++ )
527 {
528 // at depth 1 elements won't have any more children
529 if ( hasChildren )
530 str.Printf("%s child %d", "Folder", n + 1);
531 else
532 str.Printf("%s child %d.%d", "File", folder, n + 1);
533
534 // here we pass to AppendItem() normal and selected item images (we
535 // suppose that selected image follows the normal one in the enum)
536 int image = depth == 1 ? TreeCtrlIcon_File : TreeCtrlIcon_Folder;
537 wxTreeItemId id = AppendItem(idParent, str, image, image + 1,
538 new MyTreeItemData(str));
539
540 // and now we also set the expanded one (only for the folders)
541 if ( hasChildren )
542 {
543 SetItemImage(id, TreeCtrlIcon_FolderOpened,
544 wxTreeItemIcon_Expanded);
545 }
546
547 // remember the last child for OnEnsureVisible()
548 if ( !hasChildren && n == numChildren - 1 )
549 {
550 m_lastItem = id;
551 }
552
553 AddItemsRecursively(id, numChildren, depth - 1, n + 1);
554 }
555 }
556 //else: done!
557}
558
559void MyTreeCtrl::AddTestItemsToTree(size_t numChildren,
560 size_t depth)
561{
562 wxTreeItemId rootId = AddRoot("Root",
563 TreeCtrlIcon_Folder, TreeCtrlIcon_Folder,
564 new MyTreeItemData("Root item"));
565 SetItemImage(rootId, TreeCtrlIcon_FolderOpened, wxTreeItemIcon_Expanded);
566
567 AddItemsRecursively(rootId, numChildren, depth, 0);
568
569 // set some colours/fonts for testing
570 SetItemFont(rootId, *wxITALIC_FONT);
571
572 long cookie;
573 wxTreeItemId id = GetFirstChild(rootId, cookie);
574 SetItemTextColour(id, *wxBLUE);
575
576 id = GetNextChild(rootId, cookie);
577 id = GetNextChild(rootId, cookie);
578 SetItemTextColour(id, *wxRED);
579 SetItemBackgroundColour(id, *wxLIGHT_GREY);
580}
581
582void MyTreeCtrl::GetItemsRecursively(const wxTreeItemId& idParent, long cookie)
583{
584 wxTreeItemId id;
585
586 if( cookie == -1 )
587 id = GetFirstChild(idParent, cookie);
588 else
589 id = GetNextChild(idParent, cookie);
590
591 if(id <= 0)
592 return;
593
594 wxString text = GetItemText(id);
595 wxLogMessage(text);
596
597 if (ItemHasChildren(id))
598 GetItemsRecursively(id,-1);
599
600 GetItemsRecursively(idParent, cookie);
601}
602
603void MyTreeCtrl::DoToggleIcon(const wxTreeItemId& item)
604{
605 int image = GetItemImage(item) == TreeCtrlIcon_Folder ? TreeCtrlIcon_File
606 : TreeCtrlIcon_Folder;
607
608 SetItemImage(item, image);
609}
610
611
612// avoid repetition
613#define TREE_EVENT_HANDLER(name) \
614void MyTreeCtrl::name(wxTreeEvent& WXUNUSED(event)) \
615{ \
616 wxLogMessage(#name); \
617}
618
619TREE_EVENT_HANDLER(OnBeginDrag)
620TREE_EVENT_HANDLER(OnBeginRDrag)
621TREE_EVENT_HANDLER(OnDeleteItem)
622TREE_EVENT_HANDLER(OnGetInfo)
623TREE_EVENT_HANDLER(OnSetInfo)
624TREE_EVENT_HANDLER(OnItemExpanded)
625TREE_EVENT_HANDLER(OnItemExpanding)
626TREE_EVENT_HANDLER(OnItemCollapsed)
627TREE_EVENT_HANDLER(OnSelChanged)
628TREE_EVENT_HANDLER(OnSelChanging)
629
630#undef TREE_EVENT_HANDLER
631
632void MyTreeCtrl::OnBeginLabelEdit(wxTreeEvent& event)
633{
634 wxLogMessage("OnBeginLabelEdit");
635
636 // for testing, prevent this items label editing
637 wxTreeItemId itemId = event.GetItem();
638 if ( IsTestItem(itemId) )
639 {
640 wxMessageBox("You can't edit this item.");
641
642 event.Veto();
643 }
644}
645
646void MyTreeCtrl::OnEndLabelEdit(wxTreeEvent& event)
647{
648 wxLogMessage("OnEndLabelEdit");
649
650 // don't allow anything except letters in the labels
651 if ( !event.GetLabel().IsWord() )
652 {
653 wxMessageBox("The label should contain only letters.");
654
655 event.Veto();
656 }
657}
658
659void MyTreeCtrl::OnItemCollapsing(wxTreeEvent& event)
660{
661 wxLogMessage("OnItemCollapsing");
662
663 // for testing, prevent the user from collapsing the first child folder
664 wxTreeItemId itemId = event.GetItem();
665 if ( IsTestItem(itemId) )
666 {
667 wxMessageBox("You can't collapse this item.");
668
669 event.Veto();
670 }
671}
672
673void MyTreeCtrl::OnTreeKeyDown(wxTreeEvent&WXUNUSED(event))
674{
675 wxLogMessage("OnTreeKeyDown");
676}
677
678void MyTreeCtrl::OnItemActivated(wxTreeEvent&WXUNUSED(event))
679{
680 // show some info about this item
681 wxTreeItemId itemId = GetSelection();
682 MyTreeItemData *item = (MyTreeItemData *)GetItemData(itemId);
683
684 if ( item != NULL )
685 {
686 item->ShowInfo(this);
687 }
688
689 wxLogMessage("OnItemActivated");
690}
691
692void MyTreeCtrl::OnRMouseDClick(wxMouseEvent& event)
693{
694 wxTreeItemId id = HitTest(event.GetPosition());
695 if ( !id )
696 wxLogMessage("No item under mouse");
697 else
698 {
699 MyTreeItemData *item = (MyTreeItemData *)GetItemData(id);
700 if ( item )
701 wxLogMessage("Item '%s' under mouse", item->GetDesc());
702 }
703}
704
705static inline const char *Bool2String(bool b)
706{
707 return b ? "" : "not ";
708}
709
710void MyTreeItemData::ShowInfo(wxTreeCtrl *tree)
711{
712 wxLogMessage("Item '%s': %sselected, %sexpanded, %sbold,\n"
713 "%u children (%u immediately under this item).",
714 m_desc.c_str(),
715 Bool2String(tree->IsSelected(GetId())),
716 Bool2String(tree->IsExpanded(GetId())),
717 Bool2String(tree->IsBold(GetId())),
718 tree->GetChildrenCount(GetId()),
719 tree->GetChildrenCount(GetId(), FALSE));
720}