samples/treectrl/treectrl.cpp

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