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