added container class source
[wxWidgets.git] / samples / treectrl / treectrl.cpp
0 / 909 (  0%)
CommitLineData
1/////////////////////////////////////////////////////////////////////////////
2// Name: treectrl.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// important: the #pragma argument must be different from treectrl.cpp,
13// otherwise gcc gets confused (as there is also treectrl.cpp in the library
14// which has identical #pragma) and the sample crashes on startup!
15#ifdef __GNUG__
16 #pragma interface "treetest.cpp"
17 #pragma implementation "treetest.cpp"
18#endif
19
20// For compilers that support precompilation, includes "wx/wx.h".
21#include "wx/wxprec.h"
22
23#ifdef __BORLANDC__
24 #pragma hdrstop
25#endif
26
27#ifndef WX_PRECOMP
28 #include "wx/wx.h"
29#endif
30
31#include "wx/log.h"
32
33#include "wx/image.h"
34#include "wx/imaglist.h"
35#include "wx/treectrl.h"
36
37#include "math.h"
38
39#ifdef __WIN32__
40 // this is not supported by native control
41 #define NO_VARIABLE_HEIGHT
42#endif
43
44#include "treectrl.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_SIZE(MyFrame::OnSize)
67
68 EVT_MENU(TreeTest_Quit, MyFrame::OnQuit)
69 EVT_MENU(TreeTest_About, MyFrame::OnAbout)
70 EVT_MENU(TreeTest_Dump, MyFrame::OnDump)
71#ifndef NO_MULTIPLE_SELECTION
72 EVT_MENU(TreeTest_DumpSelected, MyFrame::OnDumpSelected)
73 EVT_MENU(TreeTest_Select, MyFrame::OnSelect)
74 EVT_MENU(TreeTest_Unselect, MyFrame::OnUnselect)
75 EVT_MENU(TreeTest_ToggleSel, MyFrame::OnToggleSel)
76#endif // NO_MULTIPLE_SELECTION
77 EVT_MENU(TreeTest_Rename, MyFrame::OnRename)
78 EVT_MENU(TreeTest_Count, MyFrame::OnCount)
79 EVT_MENU(TreeTest_CountRec, MyFrame::OnCountRec)
80 EVT_MENU(TreeTest_Sort, MyFrame::OnSort)
81 EVT_MENU(TreeTest_SortRev, MyFrame::OnSortRev)
82 EVT_MENU(TreeTest_Bold, MyFrame::OnSetBold)
83 EVT_MENU(TreeTest_UnBold, MyFrame::OnClearBold)
84 EVT_MENU(TreeTest_Delete, MyFrame::OnDelete)
85 EVT_MENU(TreeTest_DeleteChildren, MyFrame::OnDeleteChildren)
86 EVT_MENU(TreeTest_DeleteAll, MyFrame::OnDeleteAll)
87 EVT_MENU(TreeTest_Recreate, MyFrame::OnRecreate)
88 EVT_MENU(TreeTest_ToggleImages, MyFrame::OnToggleImages)
89 EVT_MENU(TreeTest_SetImageSize, MyFrame::OnSetImageSize)
90 EVT_MENU(TreeTest_CollapseAndReset, MyFrame::OnCollapseAndReset)
91 EVT_MENU(TreeTest_EnsureVisible, MyFrame::OnEnsureVisible)
92 EVT_MENU(TreeTest_AddItem, MyFrame::OnAddItem)
93 EVT_MENU(TreeTest_InsertItem, MyFrame::OnInsertItem)
94 EVT_MENU(TreeTest_IncIndent, MyFrame::OnIncIndent)
95 EVT_MENU(TreeTest_DecIndent, MyFrame::OnDecIndent)
96 EVT_MENU(TreeTest_IncSpacing, MyFrame::OnIncSpacing)
97 EVT_MENU(TreeTest_DecSpacing, MyFrame::OnDecSpacing)
98 EVT_MENU(TreeTest_ToggleIcon, MyFrame::OnToggleIcon)
99END_EVENT_TABLE()
100
101#if USE_GENERIC_TREECTRL
102BEGIN_EVENT_TABLE(MyTreeCtrl, wxGenericTreeCtrl)
103#else
104BEGIN_EVENT_TABLE(MyTreeCtrl, wxTreeCtrl)
105#endif
106 EVT_TREE_BEGIN_DRAG(TreeTest_Ctrl, MyTreeCtrl::OnBeginDrag)
107 EVT_TREE_BEGIN_RDRAG(TreeTest_Ctrl, MyTreeCtrl::OnBeginRDrag)
108 EVT_TREE_END_DRAG(TreeTest_Ctrl, MyTreeCtrl::OnEndDrag)
109 EVT_TREE_BEGIN_LABEL_EDIT(TreeTest_Ctrl, MyTreeCtrl::OnBeginLabelEdit)
110 EVT_TREE_END_LABEL_EDIT(TreeTest_Ctrl, MyTreeCtrl::OnEndLabelEdit)
111 EVT_TREE_DELETE_ITEM(TreeTest_Ctrl, MyTreeCtrl::OnDeleteItem)
112#if 0 // there are so many of those that logging them causes flicker
113 EVT_TREE_GET_INFO(TreeTest_Ctrl, MyTreeCtrl::OnGetInfo)
114#endif
115 EVT_TREE_SET_INFO(TreeTest_Ctrl, MyTreeCtrl::OnSetInfo)
116 EVT_TREE_ITEM_EXPANDED(TreeTest_Ctrl, MyTreeCtrl::OnItemExpanded)
117 EVT_TREE_ITEM_EXPANDING(TreeTest_Ctrl, MyTreeCtrl::OnItemExpanding)
118 EVT_TREE_ITEM_COLLAPSED(TreeTest_Ctrl, MyTreeCtrl::OnItemCollapsed)
119 EVT_TREE_ITEM_COLLAPSING(TreeTest_Ctrl, MyTreeCtrl::OnItemCollapsing)
120 EVT_TREE_ITEM_RIGHT_CLICK(TreeTest_Ctrl, MyTreeCtrl::OnItemRightClick)
121
122 EVT_RIGHT_UP(MyTreeCtrl::OnRMouseUp)
123 EVT_TREE_SEL_CHANGED(TreeTest_Ctrl, MyTreeCtrl::OnSelChanged)
124 EVT_TREE_SEL_CHANGING(TreeTest_Ctrl, MyTreeCtrl::OnSelChanging)
125 EVT_TREE_KEY_DOWN(TreeTest_Ctrl, MyTreeCtrl::OnTreeKeyDown)
126 EVT_TREE_ITEM_ACTIVATED(TreeTest_Ctrl, MyTreeCtrl::OnItemActivated)
127 EVT_RIGHT_DCLICK(MyTreeCtrl::OnRMouseDClick)
128END_EVENT_TABLE()
129
130IMPLEMENT_APP(MyApp)
131
132bool MyApp::OnInit()
133{
134 // Create the main frame window
135 MyFrame *frame = new MyFrame("wxTreeCtrl Test", 50, 50, 450, 600);
136
137 // Show the frame
138 frame->Show(TRUE);
139 SetTopWindow(frame);
140
141 return TRUE;
142}
143
144
145// My frame constructor
146MyFrame::MyFrame(const wxString& title, int x, int y, int w, int h)
147 : wxFrame((wxFrame *)NULL, -1, title, wxPoint(x, y), wxSize(w, h)),
148 m_treeCtrl(NULL), m_textCtrl(NULL)
149{
150 // This reduces flicker effects - even better would be to define
151 // OnEraseBackground to do nothing. When the tree control's scrollbars are
152 // show or hidden, the frame is sent a background erase event.
153 SetBackgroundColour(wxColour(255, 255, 255));
154
155 // Give it an icon
156 SetIcon(wxICON(mondrian));
157
158#if wxUSE_MENUS
159 // Make a menubar
160 wxMenu *file_menu = new wxMenu,
161 *tree_menu = new wxMenu,
162 *item_menu = new wxMenu;
163
164 file_menu->Append(TreeTest_About, "&About...");
165 file_menu->AppendSeparator();
166 file_menu->Append(TreeTest_Quit, "E&xit\tAlt-X");
167
168#ifndef NO_MULTIPLE_SELECTION
169 tree_menu->Append(TreeTest_ToggleSel, "&Toggle selection mode");
170#endif // NO_MULTIPLE_SELECTION
171 tree_menu->Append(TreeTest_ToggleImages, "&Show images", "", TRUE);
172 tree_menu->Append(TreeTest_SetImageSize, "Set image si&ze...");
173 tree_menu->Append(TreeTest_Recreate, "&Recreate the tree");
174 tree_menu->Append(TreeTest_CollapseAndReset, "C&ollapse and reset");
175 tree_menu->AppendSeparator();
176 tree_menu->Append(TreeTest_AddItem, "Append a &new item");
177 tree_menu->Append(TreeTest_InsertItem, "&Insert a new item");
178 tree_menu->Append(TreeTest_Delete, "&Delete this item");
179 tree_menu->Append(TreeTest_DeleteChildren, "Delete &children");
180 tree_menu->Append(TreeTest_DeleteAll, "Delete &all items");
181 tree_menu->AppendSeparator();
182 tree_menu->Append(TreeTest_Count, "Count children of current item");
183 tree_menu->Append(TreeTest_CountRec, "Recursively count children of current item");
184 tree_menu->AppendSeparator();
185 tree_menu->Append(TreeTest_Sort, "Sort children of current item");
186 tree_menu->Append(TreeTest_SortRev, "Sort in reversed order");
187 tree_menu->AppendSeparator();
188 tree_menu->Append(TreeTest_EnsureVisible, "Make the last item &visible");
189 tree_menu->AppendSeparator();
190 tree_menu->Append(TreeTest_IncIndent, "Add 5 points to indentation\tAlt-I");
191 tree_menu->Append(TreeTest_DecIndent, "Reduce indentation by 5 points\tAlt-R");
192 tree_menu->AppendSeparator();
193 tree_menu->Append(TreeTest_IncSpacing, "Add 5 points to spacing\tCtrl-I");
194 tree_menu->Append(TreeTest_DecSpacing, "Reduce spacing by 5 points\tCtrl-R");
195
196 item_menu->Append(TreeTest_Dump, "&Dump item children");
197 item_menu->Append(TreeTest_Rename, "&Rename item...");
198
199 item_menu->AppendSeparator();
200 item_menu->Append(TreeTest_Bold, "Make item &bold");
201 item_menu->Append(TreeTest_UnBold, "Make item &not bold");
202 item_menu->AppendSeparator();
203 item_menu->Append(TreeTest_ToggleIcon, "Toggle the items &icon");
204
205#ifndef NO_MULTIPLE_SELECTION
206 item_menu->AppendSeparator();
207 item_menu->Append(TreeTest_DumpSelected, "Dump selected items\tAlt-D");
208 item_menu->Append(TreeTest_Select, "Select current item\tAlt-S");
209 item_menu->Append(TreeTest_Unselect, "Unselect everything\tAlt-U");
210#endif // NO_MULTIPLE_SELECTION
211
212 wxMenuBar *menu_bar = new wxMenuBar;
213 menu_bar->Append(file_menu, "&File");
214 menu_bar->Append(tree_menu, "&Tree");
215 menu_bar->Append(item_menu, "&Item");
216 SetMenuBar(menu_bar);
217
218 menu_bar->Check(TreeTest_ToggleImages, TRUE);
219#endif // wxUSE_MENUS
220
221 m_treeCtrl = new MyTreeCtrl(this, TreeTest_Ctrl,
222 wxDefaultPosition, wxDefaultSize,
223 wxTR_MAC_BUTTONS | wxTR_NO_LINES |
224 wxTR_EDIT_LABELS |
225#ifndef NO_VARIABLE_HEIGHT
226 wxTR_HAS_VARIABLE_ROW_HEIGHT |
227#endif
228 wxSUNKEN_BORDER);
229
230// m_treeCtrl->SetBackgroundColour( *wxLIGHT_GREY );
231
232 m_textCtrl = new wxTextCtrl(this, -1, "",
233 wxDefaultPosition, wxDefaultSize,
234 wxTE_MULTILINE | wxSUNKEN_BORDER);
235
236 // create a status bar with 3 panes
237 CreateStatusBar(3);
238 SetStatusText("", 0);
239
240#ifdef __WXMOTIF__
241 // For some reason, we get a memcpy crash in wxLogStream::DoLogStream
242 // on gcc/wxMotif, if we use wxLogTextCtl. Maybe it's just gcc?
243 delete wxLog::SetActiveTarget(new wxLogStderr);
244#else
245 // set our text control as the log target
246 wxLogTextCtrl *logWindow = new wxLogTextCtrl(m_textCtrl);
247 delete wxLog::SetActiveTarget(logWindow);
248#endif
249}
250
251MyFrame::~MyFrame()
252{
253 delete wxLog::SetActiveTarget(NULL);
254}
255
256void MyFrame::OnSize(wxSizeEvent& event)
257{
258 if ( m_treeCtrl && m_textCtrl )
259 {
260 Resize(GetClientSize());
261 }
262
263 event.Skip();
264}
265
266void MyFrame::Resize(const wxSize& size)
267{
268 m_treeCtrl->SetSize(0, 0, size.x, 2*size.y/3);
269 m_textCtrl->SetSize(0, 2*size.y/3, size.x, size.y/3);
270}
271
272void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event))
273{
274 Close(TRUE);
275}
276
277void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event))
278{
279 wxMessageBox("Tree test sample\n"
280 "(c) Julian Smart 1997, Vadim Zeitlin 1998",
281 "About tree test",
282 wxOK | wxICON_INFORMATION, this);
283}
284
285void MyFrame::OnRename(wxCommandEvent& WXUNUSED(event))
286{
287 wxTreeItemId item = m_treeCtrl->GetSelection();
288
289 CHECK_ITEM( item );
290
291 // old code - now we edit in place
292#if 0
293 static wxString s_text;
294 s_text = wxGetTextFromUser("New name: ", "Tree sample question",
295 s_text, this);
296 if ( !s_text.IsEmpty() )
297 {
298 m_treeCtrl->SetItemText(item, s_text);
299 }
300#endif // 0
301
302 // TODO demonstrate creating a custom edit control...
303 (void)m_treeCtrl->EditLabel(item);
304}
305
306void MyFrame::OnCount(wxCommandEvent& WXUNUSED(event))
307{
308 wxTreeItemId item = m_treeCtrl->GetSelection();
309
310 CHECK_ITEM( item );
311
312 int i = m_treeCtrl->GetChildrenCount( item, FALSE );
313
314 wxLogMessage(wxT("%d children"), i);
315}
316
317void MyFrame::OnCountRec(wxCommandEvent& WXUNUSED(event))
318{
319 wxTreeItemId item = m_treeCtrl->GetSelection();
320
321 CHECK_ITEM( item );
322
323 int i = m_treeCtrl->GetChildrenCount( item );
324
325 wxLogMessage(wxT("%d children"), i);
326}
327
328void MyFrame::DoSort(bool reverse)
329{
330 wxTreeItemId item = m_treeCtrl->GetSelection();
331
332 CHECK_ITEM( item );
333
334 m_treeCtrl->DoSortChildren(item, reverse);
335}
336
337void MyFrame::OnDump(wxCommandEvent& WXUNUSED(event))
338{
339 wxTreeItemId root = m_treeCtrl->GetSelection();
340
341 CHECK_ITEM( root );
342
343 m_treeCtrl->GetItemsRecursively(root, -1);
344}
345
346#ifndef NO_MULTIPLE_SELECTION
347
348void MyFrame::OnToggleSel(wxCommandEvent& WXUNUSED(event))
349{
350 long style = m_treeCtrl->GetWindowStyle();
351 if ( style & wxTR_MULTIPLE )
352 style &= ~wxTR_MULTIPLE;
353 else
354 style |= wxTR_MULTIPLE;
355
356 delete m_treeCtrl;
357
358 m_treeCtrl = new MyTreeCtrl(this, TreeTest_Ctrl,
359 wxDefaultPosition, wxDefaultSize,
360 style);
361 Resize(GetClientSize());
362}
363
364void MyFrame::OnDumpSelected(wxCommandEvent& WXUNUSED(event))
365{
366 wxArrayTreeItemIds array;
367
368 size_t count = m_treeCtrl->GetSelections(array);
369 wxLogMessage(wxT("%u items selected"), count);
370
371 for ( size_t n = 0; n < count; n++ )
372 {
373 wxLogMessage("\t%s", m_treeCtrl->GetItemText(array.Item(n)).c_str());
374 }
375}
376
377void MyFrame::OnSelect(wxCommandEvent& event)
378{
379 m_treeCtrl->SelectItem(m_treeCtrl->GetSelection());
380}
381
382void MyFrame::OnUnselect(wxCommandEvent& event)
383{
384 m_treeCtrl->UnselectAll();
385}
386
387#endif // NO_MULTIPLE_SELECTION
388
389void MyFrame::DoSetBold(bool bold)
390{
391 wxTreeItemId item = m_treeCtrl->GetSelection();
392
393 CHECK_ITEM( item );
394
395 m_treeCtrl->SetItemBold(item, bold);
396}
397
398void MyFrame::OnDelete(wxCommandEvent& WXUNUSED(event))
399{
400 wxTreeItemId item = m_treeCtrl->GetSelection();
401
402 CHECK_ITEM( item );
403
404 m_treeCtrl->Delete(item);
405}
406
407void MyFrame::OnDeleteChildren(wxCommandEvent& WXUNUSED(event))
408{
409 wxTreeItemId item = m_treeCtrl->GetSelection();
410
411 CHECK_ITEM( item );
412
413 m_treeCtrl->DeleteChildren(item);
414}
415
416void MyFrame::OnDeleteAll(wxCommandEvent& WXUNUSED(event))
417{
418 m_treeCtrl->DeleteAllItems();
419}
420
421void MyFrame::OnRecreate(wxCommandEvent& event)
422{
423 OnDeleteAll(event);
424 m_treeCtrl->AddTestItemsToTree(3, 2);
425}
426
427void MyFrame::OnSetImageSize(wxCommandEvent& event)
428{
429 long size = wxGetNumberFromUser("Enter the size for the images to use",
430 "Size: ",
431 "TreeCtrl sample",
432 16);
433 if ( size == -1 )
434 return;
435
436 m_treeCtrl->CreateImageList((int)size);
437
438 OnRecreate(event);
439}
440
441void MyFrame::OnToggleImages(wxCommandEvent& event)
442{
443 if ( wxGetApp().ShowImages() )
444 {
445 m_treeCtrl->CreateImageList(-1);
446
447 wxGetApp().SetShowImages(FALSE);
448 }
449 else
450 {
451 m_treeCtrl->CreateImageList();
452
453 wxGetApp().SetShowImages(TRUE);
454 }
455
456 OnRecreate(event);
457}
458
459void MyFrame::OnCollapseAndReset(wxCommandEvent& event)
460{
461 m_treeCtrl->CollapseAndReset(m_treeCtrl->GetRootItem());
462}
463
464void MyFrame::OnEnsureVisible(wxCommandEvent& event)
465{
466 m_treeCtrl->DoEnsureVisible();
467}
468
469void MyFrame::OnInsertItem(wxCommandEvent& WXUNUSED(event))
470{
471 int image = wxGetApp().ShowImages() ? MyTreeCtrl::TreeCtrlIcon_File : -1;
472 m_treeCtrl->InsertItem(m_treeCtrl->GetRootItem(), image, "2nd item");
473}
474
475void MyFrame::OnAddItem(wxCommandEvent& WXUNUSED(event))
476{
477 static int s_num = 0;
478
479 wxString text;
480 text.Printf("Item #%d", ++s_num);
481
482 m_treeCtrl->AppendItem(m_treeCtrl->GetRootItem(),
483 text /*,
484 MyTreeCtrl::TreeCtrlIcon_File */ );
485}
486
487void MyFrame::OnIncIndent(wxCommandEvent& WXUNUSED(event))
488{
489 unsigned int indent = m_treeCtrl->GetIndent();
490 if (indent < 100)
491 m_treeCtrl->SetIndent( indent+5 );
492}
493
494void MyFrame::OnDecIndent(wxCommandEvent& WXUNUSED(event))
495{
496 unsigned int indent = m_treeCtrl->GetIndent();
497 if (indent > 10)
498 m_treeCtrl->SetIndent( indent-5 );
499}
500
501void MyFrame::OnIncSpacing(wxCommandEvent& WXUNUSED(event))
502{
503 unsigned int indent = m_treeCtrl->GetSpacing();
504 if (indent < 100)
505 m_treeCtrl->SetSpacing( indent+5 );
506}
507
508void MyFrame::OnDecSpacing(wxCommandEvent& WXUNUSED(event))
509{
510 unsigned int indent = m_treeCtrl->GetSpacing();
511 if (indent > 10)
512 m_treeCtrl->SetSpacing( indent-5 );
513}
514
515void MyFrame::OnToggleIcon(wxCommandEvent& WXUNUSED(event))
516{
517 wxTreeItemId item = m_treeCtrl->GetSelection();
518
519 CHECK_ITEM( item );
520
521 m_treeCtrl->DoToggleIcon(item);
522}
523
524// MyTreeCtrl implementation
525#if USE_GENERIC_TREECTRL
526IMPLEMENT_DYNAMIC_CLASS(MyTreeCtrl, wxGenericTreeCtrl)
527#else
528IMPLEMENT_DYNAMIC_CLASS(MyTreeCtrl, wxTreeCtrl)
529#endif
530
531MyTreeCtrl::MyTreeCtrl(wxWindow *parent, const wxWindowID id,
532 const wxPoint& pos, const wxSize& size,
533 long style)
534 : wxTreeCtrl(parent, id, pos, size, style)
535{
536 m_reverseSort = FALSE;
537 m_imageListNormal = NULL;
538
539 CreateImageList();
540
541 // Add some items to the tree
542 AddTestItemsToTree(3, 2);
543}
544
545void MyTreeCtrl::CreateImageList(int size)
546{
547 delete m_imageListNormal;
548
549 if ( size == -1 )
550 {
551 m_imageListNormal = NULL;
552 }
553 else
554 {
555 // Make an image list containing small icons
556 m_imageListNormal = new wxImageList(size, size, TRUE);
557
558 // should correspond to TreeCtrlIcon_xxx enum
559#if defined(__WXMSW__) && defined(__WIN16__)
560 m_imageListNormal->Add(wxBitmap("bitmap1", wxBITMAP_TYPE_BMP_RESOURCE));
561 m_imageListNormal->Add(wxBitmap("bitmap2", wxBITMAP_TYPE_BMP_RESOURCE));
562 m_imageListNormal->Add(wxBitmap("bitmap3", wxBITMAP_TYPE_BMP_RESOURCE));
563 m_imageListNormal->Add(wxBitmap("bitmap4", wxBITMAP_TYPE_BMP_RESOURCE));
564 m_imageListNormal->Add(wxBitmap("bitmap5", wxBITMAP_TYPE_BMP_RESOURCE));
565#else // !MSW
566 wxIcon icons[5];
567 icons[0] = wxICON(icon1);
568 icons[1] = wxICON(icon2);
569 icons[2] = wxICON(icon3);
570 icons[3] = wxICON(icon4);
571 icons[4] = wxICON(icon5);
572
573 int sizeOrig = icons[0].GetWidth();
574 for ( size_t i = 0; i < WXSIZEOF(icons); i++ )
575 {
576 if ( size == sizeOrig )
577 {
578 m_imageListNormal->Add(icons[i]);
579 }
580 else
581 {
582 m_imageListNormal->Add(wxImage(icons[i]).Rescale(size, size).
583 ConvertToBitmap());
584 }
585 }
586#endif // MSW/!MSW
587 }
588
589 SetImageList(m_imageListNormal);
590}
591
592MyTreeCtrl::~MyTreeCtrl()
593{
594 delete m_imageListNormal;
595}
596
597int MyTreeCtrl::OnCompareItems(const wxTreeItemId& item1,
598 const wxTreeItemId& item2)
599{
600 if ( m_reverseSort )
601 {
602 // just exchange 1st and 2nd items
603 return wxTreeCtrl::OnCompareItems(item2, item1);
604 }
605 else
606 {
607 return wxTreeCtrl::OnCompareItems(item1, item2);
608 }
609}
610
611void MyTreeCtrl::AddItemsRecursively(const wxTreeItemId& idParent,
612 size_t numChildren,
613 size_t depth,
614 size_t folder)
615{
616 if ( depth > 0 )
617 {
618 bool hasChildren = depth > 1;
619
620 wxString str;
621 for ( size_t n = 0; n < numChildren; n++ )
622 {
623 // at depth 1 elements won't have any more children
624 if ( hasChildren )
625 str.Printf("%s child %d", "Folder", n + 1);
626 else
627 str.Printf("%s child %d.%d", "File", folder, n + 1);
628
629 // here we pass to AppendItem() normal and selected item images (we
630 // suppose that selected image follows the normal one in the enum)
631 int image, imageSel;
632 if ( wxGetApp().ShowImages() )
633 {
634 image = depth == 1 ? TreeCtrlIcon_File : TreeCtrlIcon_Folder;
635 imageSel = image + 1;
636 }
637 else
638 {
639 image = imageSel = -1;
640 }
641 wxTreeItemId id = AppendItem(idParent, str, image, imageSel,
642 new MyTreeItemData(str));
643
644 // and now we also set the expanded one (only for the folders)
645 if ( hasChildren && wxGetApp().ShowImages() )
646 {
647 SetItemImage(id, TreeCtrlIcon_FolderOpened,
648 wxTreeItemIcon_Expanded);
649 }
650
651 // remember the last child for OnEnsureVisible()
652 if ( !hasChildren && n == numChildren - 1 )
653 {
654 m_lastItem = id;
655 }
656
657 AddItemsRecursively(id, numChildren, depth - 1, n + 1);
658 }
659 }
660 //else: done!
661}
662
663void MyTreeCtrl::AddTestItemsToTree(size_t numChildren,
664 size_t depth)
665{
666 int image = wxGetApp().ShowImages() ? MyTreeCtrl::TreeCtrlIcon_Folder : -1;
667 wxTreeItemId rootId = AddRoot("Root",
668 image, image,
669 new MyTreeItemData("Root item"));
670 if ( image != -1 )
671 {
672 SetItemImage(rootId, TreeCtrlIcon_FolderOpened, wxTreeItemIcon_Expanded);
673 }
674
675 AddItemsRecursively(rootId, numChildren, depth, 0);
676
677 // set some colours/fonts for testing
678 SetItemFont(rootId, *wxITALIC_FONT);
679
680 long cookie;
681 wxTreeItemId id = GetFirstChild(rootId, cookie);
682 SetItemTextColour(id, *wxBLUE);
683
684 id = GetNextChild(rootId, cookie);
685 id = GetNextChild(rootId, cookie);
686 SetItemTextColour(id, *wxRED);
687 SetItemBackgroundColour(id, *wxLIGHT_GREY);
688}
689
690void MyTreeCtrl::GetItemsRecursively(const wxTreeItemId& idParent, long cookie)
691{
692 wxTreeItemId id;
693
694 if( cookie == -1 )
695 id = GetFirstChild(idParent, cookie);
696 else
697 id = GetNextChild(idParent, cookie);
698
699 if(id <= 0)
700 return;
701
702 wxString text = GetItemText(id);
703 wxLogMessage(text);
704
705 if (ItemHasChildren(id))
706 GetItemsRecursively(id,-1);
707
708 GetItemsRecursively(idParent, cookie);
709}
710
711void MyTreeCtrl::DoToggleIcon(const wxTreeItemId& item)
712{
713 int image = GetItemImage(item) == TreeCtrlIcon_Folder ? TreeCtrlIcon_File
714 : TreeCtrlIcon_Folder;
715
716 SetItemImage(item, image);
717}
718
719
720// avoid repetition
721#define TREE_EVENT_HANDLER(name) \
722void MyTreeCtrl::name(wxTreeEvent& event) \
723{ \
724 wxLogMessage(#name); \
725 event.Skip(); \
726}
727
728TREE_EVENT_HANDLER(OnBeginRDrag)
729TREE_EVENT_HANDLER(OnDeleteItem)
730TREE_EVENT_HANDLER(OnGetInfo)
731TREE_EVENT_HANDLER(OnSetInfo)
732TREE_EVENT_HANDLER(OnItemExpanded)
733TREE_EVENT_HANDLER(OnItemExpanding)
734TREE_EVENT_HANDLER(OnItemCollapsed)
735TREE_EVENT_HANDLER(OnSelChanged)
736TREE_EVENT_HANDLER(OnSelChanging)
737TREE_EVENT_HANDLER(OnTreeKeyDown)
738
739#undef TREE_EVENT_HANDLER
740
741void MyTreeCtrl::OnBeginDrag(wxTreeEvent& event)
742{
743 // need to explicitly allow drag
744 if ( event.GetItem() != GetRootItem() )
745 {
746 m_draggedItem = event.GetItem();
747
748 wxLogMessage("OnBeginDrag: started dragging %s",
749 GetItemText(m_draggedItem).c_str());
750
751 event.Allow();
752 }
753 else
754 {
755 wxLogMessage("OnBeginDrag: this item can't be dragged.");
756 }
757}
758
759void MyTreeCtrl::OnEndDrag(wxTreeEvent& event)
760{
761 wxTreeItemId itemSrc = m_draggedItem,
762 itemDst = event.GetItem();
763 m_draggedItem = (wxTreeItemId)0l;
764
765 // where to copy the item?
766 if ( itemDst.IsOk() && !ItemHasChildren(itemDst) )
767 {
768 // copy to the parent then
769 itemDst = GetParent(itemDst);
770 }
771
772 if ( !itemDst.IsOk() )
773 {
774 wxLogMessage("OnEndDrag: can't drop here.");
775
776 return;
777 }
778
779 wxString text = GetItemText(itemSrc);
780 wxLogMessage("OnEndDrag: '%s' copied to '%s'.",
781 text.c_str(), GetItemText(itemDst).c_str());
782
783 // just do append here - we could also insert it just before/after the item
784 // on which it was dropped, but this requires slightly more work... we also
785 // completely ignore the client data and icon of the old item but could
786 // copy them as well.
787 //
788 // Finally, we only copy one item here but we might copy the entire tree if
789 // we were dragging a folder.
790 int image = wxGetApp().ShowImages() ? TreeCtrlIcon_File : -1;
791 AppendItem(itemDst, text, image);
792}
793
794void MyTreeCtrl::OnBeginLabelEdit(wxTreeEvent& event)
795{
796 wxLogMessage("OnBeginLabelEdit");
797
798 // for testing, prevent this items label editing
799 wxTreeItemId itemId = event.GetItem();
800 if ( IsTestItem(itemId) )
801 {
802 wxMessageBox("You can't edit this item.");
803
804 event.Veto();
805 }
806}
807
808void MyTreeCtrl::OnEndLabelEdit(wxTreeEvent& event)
809{
810 wxLogMessage("OnEndLabelEdit");
811
812 // don't allow anything except letters in the labels
813 if ( !event.GetLabel().IsWord() )
814 {
815 wxMessageBox("The label should contain only letters.");
816
817 event.Veto();
818 }
819}
820
821void MyTreeCtrl::OnItemCollapsing(wxTreeEvent& event)
822{
823 wxLogMessage("OnItemCollapsing");
824
825 // for testing, prevent the user from collapsing the first child folder
826 wxTreeItemId itemId = event.GetItem();
827 if ( IsTestItem(itemId) )
828 {
829 wxMessageBox("You can't collapse this item.");
830
831 event.Veto();
832 }
833}
834
835void MyTreeCtrl::OnItemActivated(wxTreeEvent& event)
836{
837 // show some info about this item
838 wxTreeItemId itemId = event.GetItem();
839 MyTreeItemData *item = (MyTreeItemData *)GetItemData(itemId);
840
841 if ( item != NULL )
842 {
843 item->ShowInfo(this);
844 }
845
846 wxLogMessage("OnItemActivated");
847}
848
849void MyTreeCtrl::OnItemRightClick(wxTreeEvent& event)
850{
851 ShowMenu(event.GetItem(), event.GetPoint());
852}
853
854void MyTreeCtrl::OnRMouseUp(wxMouseEvent& event)
855{
856 wxPoint pt = event.GetPosition();
857 ShowMenu(HitTest(pt), pt);
858}
859
860void MyTreeCtrl::ShowMenu(wxTreeItemId id, const wxPoint& pt)
861{
862 wxString title;
863 if ( id.IsOk() )
864 {
865 title << _T("Menu for ") << GetItemText(id);
866 }
867 else
868 {
869 title = _T("Menu for no particular item");
870 }
871
872#if wxUSE_MENUS
873 wxMenu menu(title);
874 menu.Append(TreeTest_About, _T("&About..."));
875 menu.Append(TreeTest_Dump, _T("&Dump"));
876
877 PopupMenu(&menu, pt);
878#endif // wxUSE_MENUS
879}
880
881void MyTreeCtrl::OnRMouseDClick(wxMouseEvent& event)
882{
883 wxTreeItemId id = HitTest(event.GetPosition());
884 if ( !id )
885 wxLogMessage("No item under mouse");
886 else
887 {
888 MyTreeItemData *item = (MyTreeItemData *)GetItemData(id);
889 if ( item )
890 wxLogMessage("Item '%s' under mouse", item->GetDesc());
891 }
892}
893
894static inline const char *Bool2String(bool b)
895{
896 return b ? "" : "not ";
897}
898
899void MyTreeItemData::ShowInfo(wxTreeCtrl *tree)
900{
901 wxLogMessage("Item '%s': %sselected, %sexpanded, %sbold,\n"
902 "%u children (%u immediately under this item).",
903 m_desc.c_str(),
904 Bool2String(tree->IsSelected(GetId())),
905 Bool2String(tree->IsExpanded(GetId())),
906 Bool2String(tree->IsBold(GetId())),
907 tree->GetChildrenCount(GetId()),
908 tree->GetChildrenCount(GetId(), FALSE));
909}