src/msw/treectrl.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/msw/treectrl.cpp
   3 // Purpose:     wxTreeCtrl
   4 // Author:      Julian Smart
   5 // Modified by: Vadim Zeitlin to be less MSW-specific on 10.10.98
   6 // Created:     1997
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) Julian Smart
   9 // Licence:     wxWindows licence
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 // ============================================================================
  13 // declarations
  14 // ============================================================================
  15
  16 // ----------------------------------------------------------------------------
  17 // headers
  18 // ----------------------------------------------------------------------------
  19
  20 #ifdef __GNUG__
  21     #pragma implementation "treectrl.h"
  22 #endif
  23
  24 // For compilers that support precompilation, includes "wx.h".
  25 #include "wx/wxprec.h"
  26
  27 #ifdef __BORLANDC__
  28     #pragma hdrstop
  29 #endif
  30
  31 #if wxUSE_TREECTRL
  32
  33 #include "wx/msw/private.h"
  34
  35 // Set this to 1 to be _absolutely_ sure that repainting will work for all
  36 // comctl32.dll versions
  37 #define wxUSE_COMCTL32_SAFELY 0
  38
  39 // Mingw32 is a bit mental even though this is done in winundef
  40 #ifdef GetFirstChild
  41     #undef GetFirstChild
  42 #endif
  43
  44 #ifdef GetNextSibling
  45     #undef GetNextSibling
  46 #endif
  47
  48 #if defined(__WIN95__)
  49
  50 #include "wx/app.h"
  51 #include "wx/log.h"
  52 #include "wx/dynarray.h"
  53 #include "wx/imaglist.h"
  54 #include "wx/settings.h"
  55 #include "wx/msw/treectrl.h"
  56 #include "wx/msw/dragimag.h"
  57
  58 #ifdef __GNUWIN32_OLD__
  59     #include "wx/msw/gnuwin32/extra.h"
  60 #endif
  61
  62 #if defined(__WIN95__) && !((defined(__GNUWIN32_OLD__) || defined(__TWIN32__)) && !defined(__CYGWIN10__))
  63     #include <commctrl.h>
  64 #endif
  65
  66 // Bug in headers, sometimes
  67 #ifndef TVIS_FOCUSED
  68     #define TVIS_FOCUSED            0x0001
  69 #endif
  70
  71 #ifndef TV_FIRST
  72     #define TV_FIRST                0x1100
  73 #endif
  74
  75 #ifndef TVS_CHECKBOXES
  76     #define TVS_CHECKBOXES          0x0100
  77 #endif
  78
  79 #ifndef TVS_FULLROWSELECT
  80     #define TVS_FULLROWSELECT       0x1000
  81 #endif
  82
  83 // old headers might miss these messages (comctl32.dll 4.71+ only)
  84 #ifndef TVM_SETBKCOLOR
  85     #define TVM_SETBKCOLOR          (TV_FIRST + 29)
  86     #define TVM_SETTEXTCOLOR        (TV_FIRST + 30)
  87 #endif
  88
  89 // a macro to hide the ugliness of nested casts
  90 #define HITEM(item)     (HTREEITEM)(WXHTREEITEM)(item)
  91
  92 // the native control doesn't support multiple selections under MSW and we
  93 // have 2 ways to emulate them: either using TVS_CHECKBOXES style and let
  94 // checkboxes be the selection status (checked == selected) or by really
  95 // emulating everything, i.e. intercepting mouse and key events &c. The first
  96 // approach is much easier but doesn't work with comctl32.dll < 4.71 and also
  97 // looks quite ugly.
  98 #define wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE 0
  99
 100
 101 // ----------------------------------------------------------------------------
 102 // private functions
 103 // ----------------------------------------------------------------------------
 104
 105 // wrapper for TreeView_HitTest
 106 static HTREEITEM GetItemFromPoint(HWND hwndTV, int x, int y)
 107 {
 108     TV_HITTESTINFO tvht;
 109     tvht.pt.x = x;
 110     tvht.pt.y = y;
 111
 112     return (HTREEITEM)TreeView_HitTest(hwndTV, &tvht);
 113 }
 114
 115 #if !wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
 116
 117 // wrappers for TreeView_GetItem/TreeView_SetItem
 118 static bool IsItemSelected(HWND hwndTV, HTREEITEM hItem)
 119 {
 120     TV_ITEM tvi;
 121     tvi.mask = TVIF_STATE | TVIF_HANDLE;
 122     tvi.stateMask = TVIS_SELECTED;
 123     tvi.hItem = hItem;
 124
 125     if ( !TreeView_GetItem(hwndTV, &tvi) )
 126     {
 127         wxLogLastError(wxT("TreeView_GetItem"));
 128     }
 129
 130     return (tvi.state & TVIS_SELECTED) != 0;
 131 }
 132
 133 static void SelectItem(HWND hwndTV, HTREEITEM hItem, bool select = TRUE)
 134 {
 135     TV_ITEM tvi;
 136     tvi.mask = TVIF_STATE | TVIF_HANDLE;
 137     tvi.stateMask = TVIS_SELECTED;
 138     tvi.state = select ? TVIS_SELECTED : 0;
 139     tvi.hItem = hItem;
 140
 141     if ( TreeView_SetItem(hwndTV, &tvi) == -1 )
 142     {
 143         wxLogLastError(wxT("TreeView_SetItem"));
 144     }
 145 }
 146
 147 static inline void UnselectItem(HWND hwndTV, HTREEITEM htItem)
 148 {
 149     SelectItem(hwndTV, htItem, FALSE);
 150 }
 151
 152 static inline void ToggleItemSelection(HWND hwndTV, HTREEITEM htItem)
 153 {
 154     SelectItem(hwndTV, htItem, !IsItemSelected(hwndTV, htItem));
 155 }
 156
 157 // helper function which selects all items in a range and, optionally,
 158 // unselects all others
 159 static void SelectRange(HWND hwndTV,
 160                         HTREEITEM htFirst,
 161                         HTREEITEM htLast,
 162                         bool unselectOthers = TRUE)
 163 {
 164     // find the first (or last) item and select it
 165     bool cont = TRUE;
 166     HTREEITEM htItem = (HTREEITEM)TreeView_GetRoot(hwndTV);
 167     while ( htItem && cont )
 168     {
 169         if ( (htItem == htFirst) || (htItem == htLast) )
 170         {
 171             if ( !IsItemSelected(hwndTV, htItem) )
 172             {
 173                 SelectItem(hwndTV, htItem);
 174             }
 175
 176             cont = FALSE;
 177         }
 178         else
 179         {
 180             if ( unselectOthers && IsItemSelected(hwndTV, htItem) )
 181             {
 182                 UnselectItem(hwndTV, htItem);
 183             }
 184         }
 185
 186         htItem = (HTREEITEM)TreeView_GetNextVisible(hwndTV, htItem);
 187     }
 188
 189     // select the items in range
 190     cont = htFirst != htLast;
 191     while ( htItem && cont )
 192     {
 193         if ( !IsItemSelected(hwndTV, htItem) )
 194         {
 195             SelectItem(hwndTV, htItem);
 196         }
 197
 198         cont = (htItem != htFirst) && (htItem != htLast);
 199
 200         htItem = (HTREEITEM)TreeView_GetNextVisible(hwndTV, htItem);
 201     }
 202
 203     // unselect the rest
 204     if ( unselectOthers )
 205     {
 206         while ( htItem )
 207         {
 208             if ( IsItemSelected(hwndTV, htItem) )
 209             {
 210                 UnselectItem(hwndTV, htItem);
 211             }
 212
 213             htItem = (HTREEITEM)TreeView_GetNextVisible(hwndTV, htItem);
 214         }
 215     }
 216
 217     // seems to be necessary - otherwise the just selected items don't always
 218     // appear as selected
 219     UpdateWindow(hwndTV);
 220 }
 221
 222 // helper function which tricks the standard control into changing the focused
 223 // item without changing anything else (if someone knows why Microsoft doesn't
 224 // allow to do it by just setting TVIS_FOCUSED flag, please tell me!)
 225 static void SetFocus(HWND hwndTV, HTREEITEM htItem)
 226 {
 227     // the current focus
 228     HTREEITEM htFocus = (HTREEITEM)TreeView_GetSelection(hwndTV);
 229
 230     if ( htItem )
 231     {
 232         // set the focus
 233         if ( htItem != htFocus )
 234         {
 235             // remember the selection state of the item
 236             bool wasSelected = IsItemSelected(hwndTV, htItem);
 237
 238             if ( htFocus && IsItemSelected(hwndTV, htFocus) )
 239             {
 240                 // prevent the tree from unselecting the old focus which it
 241                 // would do by default (TreeView_SelectItem unselects the
 242                 // focused item)
 243                 TreeView_SelectItem(hwndTV, 0);
 244                 SelectItem(hwndTV, htFocus);
 245             }
 246
 247             TreeView_SelectItem(hwndTV, htItem);
 248
 249             if ( !wasSelected )
 250             {
 251                 // need to clear the selection which TreeView_SelectItem() gave
 252                 // us
 253                 UnselectItem(hwndTV, htItem);
 254             }
 255             //else: was selected, still selected - ok
 256         }
 257         //else: nothing to do, focus already there
 258     }
 259     else
 260     {
 261         if ( htFocus )
 262         {
 263             bool wasFocusSelected = IsItemSelected(hwndTV, htFocus);
 264
 265             // just clear the focus
 266             TreeView_SelectItem(hwndTV, 0);
 267
 268             if ( wasFocusSelected )
 269             {
 270                 // restore the selection state
 271                 SelectItem(hwndTV, htFocus);
 272             }
 273         }
 274         //else: nothing to do, no focus already
 275     }
 276 }
 277
 278 #endif // wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
 279
 280 // ----------------------------------------------------------------------------
 281 // private classes
 282 // ----------------------------------------------------------------------------
 283
 284 // a convenient wrapper around TV_ITEM struct which adds a ctor
 285 #ifdef __VISUALC__
 286 #pragma warning( disable : 4097 ) // inheriting from typedef
 287 #endif
 288
 289 struct wxTreeViewItem : public TV_ITEM
 290 {
 291     wxTreeViewItem(const wxTreeItemId& item,    // the item handle
 292                    UINT mask_,                  // fields which are valid
 293                    UINT stateMask_ = 0)         // for TVIF_STATE only
 294     {
 295         // hItem member is always valid
 296         mask = mask_ | TVIF_HANDLE;
 297         stateMask = stateMask_;
 298         hItem = HITEM(item);
 299     }
 300 };
 301
 302 #ifdef __VISUALC__
 303 #pragma warning( default : 4097 )
 304 #endif
 305
 306 // a class which encapsulates the tree traversal logic: it vists all (unless
 307 // OnVisit() returns FALSE) items under the given one
 308 class wxTreeTraversal
 309 {
 310 public:
 311     wxTreeTraversal(const wxTreeCtrl *tree)
 312     {
 313         m_tree = tree;
 314     }
 315
 316     // do traverse the tree: visit all items (recursively by default) under the
 317     // given one; return TRUE if all items were traversed or FALSE if the
 318     // traversal was aborted because OnVisit returned FALSE
 319     bool DoTraverse(const wxTreeItemId& root, bool recursively = TRUE);
 320
 321     // override this function to do whatever is needed for each item, return
 322     // FALSE to stop traversing
 323     virtual bool OnVisit(const wxTreeItemId& item) = 0;
 324
 325 protected:
 326     const wxTreeCtrl *GetTree() const { return m_tree; }
 327
 328 private:
 329     bool Traverse(const wxTreeItemId& root, bool recursively);
 330
 331     const wxTreeCtrl *m_tree;
 332 };
 333
 334 // internal class for getting the selected items
 335 class TraverseSelections : public wxTreeTraversal
 336 {
 337 public:
 338     TraverseSelections(const wxTreeCtrl *tree,
 339                        wxArrayTreeItemIds& selections)
 340         : wxTreeTraversal(tree), m_selections(selections)
 341         {
 342             m_selections.Empty();
 343
 344             DoTraverse(tree->GetRootItem());
 345         }
 346
 347     virtual bool OnVisit(const wxTreeItemId& item)
 348     {
 349 #if wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
 350         if ( GetTree()->IsItemChecked(item) )
 351 #else
 352         if ( ::IsItemSelected(GetHwndOf(GetTree()), HITEM(item)) )
 353 #endif
 354         {
 355             m_selections.Add(item);
 356         }
 357
 358         return TRUE;
 359     }
 360
 361     size_t GetCount() const { return m_selections.GetCount(); }
 362
 363 private:
 364     wxArrayTreeItemIds& m_selections;
 365 };
 366
 367 // internal class for counting tree items
 368 class TraverseCounter : public wxTreeTraversal
 369 {
 370 public:
 371     TraverseCounter(const wxTreeCtrl *tree,
 372                     const wxTreeItemId& root,
 373                     bool recursively)
 374         : wxTreeTraversal(tree)
 375         {
 376             m_count = 0;
 377
 378             DoTraverse(root, recursively);
 379         }
 380
 381     virtual bool OnVisit(const wxTreeItemId& WXUNUSED(item))
 382     {
 383         m_count++;
 384
 385         return TRUE;
 386     }
 387
 388     size_t GetCount() const { return m_count; }
 389
 390 private:
 391     size_t m_count;
 392 };
 393
 394 // ----------------------------------------------------------------------------
 395 // This class is needed for support of different images: the Win32 common
 396 // control natively supports only 2 images (the normal one and another for the
 397 // selected state). We wish to provide support for 2 more of them for folder
 398 // items (i.e. those which have children): for expanded state and for expanded
 399 // selected state. For this we use this structure to store the additional items
 400 // images.
 401 //
 402 // There is only one problem with this: when we retrieve the item's data, we
 403 // don't know whether we get a pointer to wxTreeItemData or
 404 // wxTreeItemIndirectData. So we always set the item id to an invalid value
 405 // in this class and the code using the client data checks for it and retrieves
 406 // the real client data in this case.
 407 // ----------------------------------------------------------------------------
 408
 409 class wxTreeItemIndirectData : public wxTreeItemData
 410 {
 411 public:
 412     // ctor associates this data with the item and the real item data becomes
 413     // available through our GetData() method
 414     wxTreeItemIndirectData(wxTreeCtrl *tree, const wxTreeItemId& item)
 415     {
 416         for ( size_t n = 0; n < WXSIZEOF(m_images); n++ )
 417         {
 418             m_images[n] = -1;
 419         }
 420
 421         // save the old data
 422         m_data = tree->GetItemData(item);
 423
 424         // and set ourselves as the new one
 425         tree->SetIndirectItemData(item, this);
 426
 427         // we must have the invalid value for the item
 428         m_pItem = 0l;
 429     }
 430
 431     // dtor deletes the associated data as well
 432     virtual ~wxTreeItemIndirectData() { delete m_data; }
 433
 434     // accessors
 435         // get the real data associated with the item
 436     wxTreeItemData *GetData() const { return m_data; }
 437         // change it
 438     void SetData(wxTreeItemData *data) { m_data = data; }
 439
 440         // do we have such image?
 441     bool HasImage(wxTreeItemIcon which) const { return m_images[which] != -1; }
 442         // get image
 443     int GetImage(wxTreeItemIcon which) const { return m_images[which]; }
 444         // change it
 445     void SetImage(int image, wxTreeItemIcon which) { m_images[which] = image; }
 446
 447 private:
 448     // all the images associated with the item
 449     int m_images[wxTreeItemIcon_Max];
 450
 451     // the real client data
 452     wxTreeItemData *m_data;
 453 };
 454
 455 // ----------------------------------------------------------------------------
 456 // wxWin macros
 457 // ----------------------------------------------------------------------------
 458
 459 IMPLEMENT_DYNAMIC_CLASS(wxTreeCtrl, wxControl)
 460
 461 // ----------------------------------------------------------------------------
 462 // variables
 463 // ----------------------------------------------------------------------------
 464
 465 // handy table for sending events
 466 static wxEventType g_events[2][2] =
 467 {
 468     { wxEVT_COMMAND_TREE_ITEM_COLLAPSED, wxEVT_COMMAND_TREE_ITEM_COLLAPSING },
 469     { wxEVT_COMMAND_TREE_ITEM_EXPANDED,  wxEVT_COMMAND_TREE_ITEM_EXPANDING  }
 470 };
 471
 472 // ============================================================================
 473 // implementation
 474 // ============================================================================
 475
 476 // ----------------------------------------------------------------------------
 477 // tree traversal
 478 // ----------------------------------------------------------------------------
 479
 480 bool wxTreeTraversal::DoTraverse(const wxTreeItemId& root, bool recursively)
 481 {
 482     if ( !OnVisit(root) )
 483         return FALSE;
 484
 485     return Traverse(root, recursively);
 486 }
 487
 488 bool wxTreeTraversal::Traverse(const wxTreeItemId& root, bool recursively)
 489 {
 490     long cookie;
 491     wxTreeItemId child = m_tree->GetFirstChild(root, cookie);
 492     while ( child.IsOk() )
 493     {
 494         // depth first traversal
 495         if ( recursively && !Traverse(child, TRUE) )
 496             return FALSE;
 497
 498         if ( !OnVisit(child) )
 499             return FALSE;
 500
 501         child = m_tree->GetNextChild(root, cookie);
 502     }
 503
 504     return TRUE;
 505 }
 506
 507 // ----------------------------------------------------------------------------
 508 // construction and destruction
 509 // ----------------------------------------------------------------------------
 510
 511 void wxTreeCtrl::Init()
 512 {
 513     m_imageListNormal = NULL;
 514     m_imageListState = NULL;
 515     m_ownsImageListNormal = m_ownsImageListState = FALSE;
 516     m_textCtrl = NULL;
 517     m_hasAnyAttr = FALSE;
 518     m_dragImage = NULL;
 519     m_htSelStart = 0;
 520
 521     // Initialize static array of events, because with the new event system,
 522     // they may not be initialized yet.
 523
 524     g_events[0][0] = wxEVT_COMMAND_TREE_ITEM_COLLAPSED;
 525     g_events[0][1] = wxEVT_COMMAND_TREE_ITEM_COLLAPSING;
 526     g_events[1][0] = wxEVT_COMMAND_TREE_ITEM_EXPANDED;
 527     g_events[1][1] = wxEVT_COMMAND_TREE_ITEM_EXPANDING;
 528 }
 529
 530 bool wxTreeCtrl::Create(wxWindow *parent,
 531                         wxWindowID id,
 532                         const wxPoint& pos,
 533                         const wxSize& size,
 534                         long style,
 535                         const wxValidator& validator,
 536                         const wxString& name)
 537 {
 538     Init();
 539
 540     if ( !CreateControl(parent, id, pos, size, style, validator, name) )
 541         return FALSE;
 542
 543     DWORD wstyle = WS_VISIBLE | WS_CHILD | WS_TABSTOP |
 544                    TVS_SHOWSELALWAYS;
 545
 546     if ( m_windowStyle & wxCLIP_SIBLINGS )
 547         wstyle |= WS_CLIPSIBLINGS;
 548
 549     if ((m_windowStyle & wxTR_NO_LINES) == 0)
 550         wstyle |= TVS_HASLINES;
 551     if ( m_windowStyle & wxTR_HAS_BUTTONS )
 552         wstyle |= TVS_HASBUTTONS;
 553
 554     if ( m_windowStyle & wxTR_EDIT_LABELS )
 555         wstyle |= TVS_EDITLABELS;
 556
 557     if ( m_windowStyle & wxTR_LINES_AT_ROOT )
 558         wstyle |= TVS_LINESATROOT;
 559
 560     if ( m_windowStyle & wxTR_FULL_ROW_HIGHLIGHT )
 561     {
 562         if ( wxTheApp->GetComCtl32Version() >= 471 )
 563             wstyle |= TVS_FULLROWSELECT;
 564     }
 565
 566
 567     // using TVS_CHECKBOXES for emulation of a multiselection tree control
 568     // doesn't work without the new enough headers
 569 #if wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE && \
 570     !defined( __GNUWIN32_OLD__ ) && \
 571     !defined( __BORLANDC__ ) && \
 572     !defined( __WATCOMC__ ) && \
 573     (!defined(__VISUALC__) || (__VISUALC__ > 1010))
 574
 575     // we emulate the multiple selection tree controls by using checkboxes: set
 576     // up the image list we need for this if we do have multiple selections
 577     if ( m_windowStyle & wxTR_MULTIPLE )
 578         wstyle |= TVS_CHECKBOXES;
 579 #endif // wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
 580
 581     // Create the tree control.
 582     if ( !MSWCreateControl(WC_TREEVIEW, wstyle) )
 583         return FALSE;
 584
 585 #if wxUSE_COMCTL32_SAFELY
 586     wxWindow::SetBackgroundColour(wxSystemSettings::GetSystemColour(wxSYS_COLOUR_WINDOW));
 587     wxWindow::SetForegroundColour(wxWindow::GetParent()->GetForegroundColour());
 588 #elif 1
 589     SetBackgroundColour(wxSystemSettings::GetSystemColour(wxSYS_COLOUR_WINDOW));
 590     SetForegroundColour(wxWindow::GetParent()->GetForegroundColour());
 591 #else
 592     // This works around a bug in the Windows tree control whereby for some versions
 593     // of comctrl32, setting any colour actually draws the background in black.
 594     // This will initialise the background to the system colour.
 595     // THIS FIX NOW REVERTED since it caused problems on _other_ systems.
 596     // Assume the user has an updated comctl32.dll.
 597     ::SendMessage(GetHwnd(), TVM_SETBKCOLOR, 0,-1);
 598     wxWindow::SetBackgroundColour(wxSystemSettings::GetSystemColour(wxSYS_COLOUR_WINDOW));
 599     SetForegroundColour(wxWindow::GetParent()->GetForegroundColour());
 600 #endif
 601
 602
 603     // VZ: this is some experimental code which may be used to get the
 604     //     TVS_CHECKBOXES style functionality for comctl32.dll < 4.71.
 605     //     AFAIK, the standard DLL does about the same thing anyhow.
 606 #if 0
 607     if ( m_windowStyle & wxTR_MULTIPLE )
 608     {
 609         wxBitmap bmp;
 610
 611         // create the DC compatible with the current screen
 612         HDC hdcMem = CreateCompatibleDC(NULL);
 613
 614         // create a mono bitmap of the standard size
 615         int x = GetSystemMetrics(SM_CXMENUCHECK);
 616         int y = GetSystemMetrics(SM_CYMENUCHECK);
 617         wxImageList imagelistCheckboxes(x, y, FALSE, 2);
 618         HBITMAP hbmpCheck = CreateBitmap(x, y,   // bitmap size
 619                                          1,      // # of color planes
 620                                          1,      // # bits needed for one pixel
 621                                          0);     // array containing colour data
 622         SelectObject(hdcMem, hbmpCheck);
 623
 624         // then draw a check mark into it
 625         RECT rect = { 0, 0, x, y };
 626         if ( !::DrawFrameControl(hdcMem, &rect,
 627                                  DFC_BUTTON,
 628                                  DFCS_BUTTONCHECK | DFCS_CHECKED) )
 629         {
 630             wxLogLastError(wxT("DrawFrameControl(check)"));
 631         }
 632
 633         bmp.SetHBITMAP((WXHBITMAP)hbmpCheck);
 634         imagelistCheckboxes.Add(bmp);
 635
 636         if ( !::DrawFrameControl(hdcMem, &rect,
 637                                  DFC_BUTTON,
 638                                  DFCS_BUTTONCHECK) )
 639         {
 640             wxLogLastError(wxT("DrawFrameControl(uncheck)"));
 641         }
 642
 643         bmp.SetHBITMAP((WXHBITMAP)hbmpCheck);
 644         imagelistCheckboxes.Add(bmp);
 645
 646         // clean up
 647         ::DeleteDC(hdcMem);
 648
 649         // set the imagelist
 650         SetStateImageList(&imagelistCheckboxes);
 651     }
 652 #endif // 0
 653
 654     SetSize(pos.x, pos.y, size.x, size.y);
 655
 656     return TRUE;
 657 }
 658
 659 wxTreeCtrl::~wxTreeCtrl()
 660 {
 661     // delete any attributes
 662     if ( m_hasAnyAttr )
 663     {
 664         for ( wxNode *node = m_attrs.Next(); node; node = m_attrs.Next() )
 665         {
 666             delete (wxTreeItemAttr *)node->Data();
 667         }
 668
 669         // prevent TVN_DELETEITEM handler from deleting the attributes again!
 670         m_hasAnyAttr = FALSE;
 671     }
 672
 673     DeleteTextCtrl();
 674
 675     // delete user data to prevent memory leaks
 676     DeleteAllItems();
 677
 678     if (m_ownsImageListNormal) delete m_imageListNormal;
 679     if (m_ownsImageListState) delete m_imageListState;
 680 }
 681
 682 // ----------------------------------------------------------------------------
 683 // accessors
 684 // ----------------------------------------------------------------------------
 685
 686 // simple wrappers which add error checking in debug mode
 687
 688 bool wxTreeCtrl::DoGetItem(wxTreeViewItem* tvItem) const
 689 {
 690     if ( !TreeView_GetItem(GetHwnd(), tvItem) )
 691     {
 692         wxLogLastError(wxT("TreeView_GetItem"));
 693
 694         return FALSE;
 695     }
 696
 697     return TRUE;
 698 }
 699
 700 void wxTreeCtrl::DoSetItem(wxTreeViewItem* tvItem)
 701 {
 702     if ( TreeView_SetItem(GetHwnd(), tvItem) == -1 )
 703     {
 704         wxLogLastError(wxT("TreeView_SetItem"));
 705     }
 706 }
 707
 708 size_t wxTreeCtrl::GetCount() const
 709 {
 710     return (size_t)TreeView_GetCount(GetHwnd());
 711 }
 712
 713 unsigned int wxTreeCtrl::GetIndent() const
 714 {
 715     return TreeView_GetIndent(GetHwnd());
 716 }
 717
 718 void wxTreeCtrl::SetIndent(unsigned int indent)
 719 {
 720     TreeView_SetIndent(GetHwnd(), indent);
 721 }
 722
 723 wxImageList *wxTreeCtrl::GetImageList() const
 724 {
 725     return m_imageListNormal;
 726 }
 727
 728 wxImageList *wxTreeCtrl::GetStateImageList() const
 729 {
 730     return m_imageListNormal;
 731 }
 732
 733 void wxTreeCtrl::SetAnyImageList(wxImageList *imageList, int which)
 734 {
 735     // no error return
 736     TreeView_SetImageList(GetHwnd(),
 737                           imageList ? imageList->GetHIMAGELIST() : 0,
 738                           which);
 739 }
 740
 741 void wxTreeCtrl::SetImageList(wxImageList *imageList)
 742 {
 743     if (m_ownsImageListNormal) delete m_imageListNormal;
 744     SetAnyImageList(m_imageListNormal = imageList, TVSIL_NORMAL);
 745     m_ownsImageListNormal = FALSE;
 746 }
 747
 748 void wxTreeCtrl::SetStateImageList(wxImageList *imageList)
 749 {
 750     if (m_ownsImageListState) delete m_imageListState;
 751     SetAnyImageList(m_imageListState = imageList, TVSIL_STATE);
 752     m_ownsImageListState = FALSE;
 753 }
 754
 755 void wxTreeCtrl::AssignImageList(wxImageList *imageList)
 756 {
 757     SetImageList(imageList);
 758     m_ownsImageListNormal = TRUE;
 759 }
 760
 761 void wxTreeCtrl::AssignStateImageList(wxImageList *imageList)
 762 {
 763     SetStateImageList(imageList);
 764     m_ownsImageListState = TRUE;
 765 }
 766
 767 size_t wxTreeCtrl::GetChildrenCount(const wxTreeItemId& item,
 768                                     bool recursively) const
 769 {
 770     TraverseCounter counter(this, item, recursively);
 771
 772     return counter.GetCount() - 1;
 773 }
 774
 775 // ----------------------------------------------------------------------------
 776 // control colours
 777 // ----------------------------------------------------------------------------
 778
 779 bool wxTreeCtrl::SetBackgroundColour(const wxColour &colour)
 780 {
 781 #if !wxUSE_COMCTL32_SAFELY
 782     if ( !wxWindowBase::SetBackgroundColour(colour) )
 783         return FALSE;
 784
 785     SendMessage(GetHwnd(), TVM_SETBKCOLOR, 0, colour.GetPixel());
 786 #endif
 787
 788     return TRUE;
 789 }
 790
 791 bool wxTreeCtrl::SetForegroundColour(const wxColour &colour)
 792 {
 793 #if !wxUSE_COMCTL32_SAFELY
 794     if ( !wxWindowBase::SetForegroundColour(colour) )
 795         return FALSE;
 796
 797     SendMessage(GetHwnd(), TVM_SETTEXTCOLOR, 0, colour.GetPixel());
 798 #endif
 799
 800     return TRUE;
 801 }
 802
 803 // ----------------------------------------------------------------------------
 804 // Item access
 805 // ----------------------------------------------------------------------------
 806
 807 wxString wxTreeCtrl::GetItemText(const wxTreeItemId& item) const
 808 {
 809     wxChar buf[512];  // the size is arbitrary...
 810
 811     wxTreeViewItem tvItem(item, TVIF_TEXT);
 812     tvItem.pszText = buf;
 813     tvItem.cchTextMax = WXSIZEOF(buf);
 814     if ( !DoGetItem(&tvItem) )
 815     {
 816         // don't return some garbage which was on stack, but an empty string
 817         buf[0] = wxT('\0');
 818     }
 819
 820     return wxString(buf);
 821 }
 822
 823 void wxTreeCtrl::SetItemText(const wxTreeItemId& item, const wxString& text)
 824 {
 825     wxTreeViewItem tvItem(item, TVIF_TEXT);
 826     tvItem.pszText = (wxChar *)text.c_str();  // conversion is ok
 827     DoSetItem(&tvItem);
 828 }
 829
 830 int wxTreeCtrl::DoGetItemImageFromData(const wxTreeItemId& item,
 831                                        wxTreeItemIcon which) const
 832 {
 833     wxTreeViewItem tvItem(item, TVIF_PARAM);
 834     if ( !DoGetItem(&tvItem) )
 835     {
 836         return -1;
 837     }
 838
 839     return ((wxTreeItemIndirectData *)tvItem.lParam)->GetImage(which);
 840 }
 841
 842 void wxTreeCtrl::DoSetItemImageFromData(const wxTreeItemId& item,
 843                                         int image,
 844                                         wxTreeItemIcon which) const
 845 {
 846     wxTreeViewItem tvItem(item, TVIF_PARAM);
 847     if ( !DoGetItem(&tvItem) )
 848     {
 849         return;
 850     }
 851
 852     wxTreeItemIndirectData *data = ((wxTreeItemIndirectData *)tvItem.lParam);
 853
 854     data->SetImage(image, which);
 855
 856     // make sure that we have selected images as well
 857     if ( which == wxTreeItemIcon_Normal &&
 858          !data->HasImage(wxTreeItemIcon_Selected) )
 859     {
 860         data->SetImage(image, wxTreeItemIcon_Selected);
 861     }
 862
 863     if ( which == wxTreeItemIcon_Expanded &&
 864          !data->HasImage(wxTreeItemIcon_SelectedExpanded) )
 865     {
 866         data->SetImage(image, wxTreeItemIcon_SelectedExpanded);
 867     }
 868 }
 869
 870 void wxTreeCtrl::DoSetItemImages(const wxTreeItemId& item,
 871                                  int image,
 872                                  int imageSel)
 873 {
 874     wxTreeViewItem tvItem(item, TVIF_IMAGE | TVIF_SELECTEDIMAGE);
 875     tvItem.iSelectedImage = imageSel;
 876     tvItem.iImage = image;
 877     DoSetItem(&tvItem);
 878 }
 879
 880 int wxTreeCtrl::GetItemImage(const wxTreeItemId& item,
 881                              wxTreeItemIcon which) const
 882 {
 883     if ( HasIndirectData(item) )
 884     {
 885         return DoGetItemImageFromData(item, which);
 886     }
 887
 888     UINT mask;
 889     switch ( which )
 890     {
 891         default:
 892             wxFAIL_MSG( wxT("unknown tree item image type") );
 893
 894         case wxTreeItemIcon_Normal:
 895             mask = TVIF_IMAGE;
 896             break;
 897
 898         case wxTreeItemIcon_Selected:
 899             mask = TVIF_SELECTEDIMAGE;
 900             break;
 901
 902         case wxTreeItemIcon_Expanded:
 903         case wxTreeItemIcon_SelectedExpanded:
 904             return -1;
 905     }
 906
 907     wxTreeViewItem tvItem(item, mask);
 908     DoGetItem(&tvItem);
 909
 910     return mask == TVIF_IMAGE ? tvItem.iImage : tvItem.iSelectedImage;
 911 }
 912
 913 void wxTreeCtrl::SetItemImage(const wxTreeItemId& item, int image,
 914                               wxTreeItemIcon which)
 915 {
 916     int imageNormal, imageSel;
 917     switch ( which )
 918     {
 919         default:
 920             wxFAIL_MSG( wxT("unknown tree item image type") );
 921
 922         case wxTreeItemIcon_Normal:
 923             imageNormal = image;
 924             imageSel = GetItemSelectedImage(item);
 925             break;
 926
 927         case wxTreeItemIcon_Selected:
 928             imageNormal = GetItemImage(item);
 929             imageSel = image;
 930             break;
 931
 932         case wxTreeItemIcon_Expanded:
 933         case wxTreeItemIcon_SelectedExpanded:
 934             if ( !HasIndirectData(item) )
 935             {
 936                 // we need to get the old images first, because after we create
 937                 // the wxTreeItemIndirectData GetItemXXXImage() will use it to
 938                 // get the images
 939                 imageNormal = GetItemImage(item);
 940                 imageSel = GetItemSelectedImage(item);
 941
 942                 // if it doesn't have it yet, add it
 943                 wxTreeItemIndirectData *data = new
 944                     wxTreeItemIndirectData(this, item);
 945
 946                 // copy the data to the new location
 947                 data->SetImage(imageNormal, wxTreeItemIcon_Normal);
 948                 data->SetImage(imageSel, wxTreeItemIcon_Selected);
 949             }
 950
 951             DoSetItemImageFromData(item, image, which);
 952
 953             // reset the normal/selected images because we won't use them any
 954             // more - now they're stored inside the indirect data
 955             imageNormal =
 956             imageSel = I_IMAGECALLBACK;
 957             break;
 958     }
 959
 960     // NB: at least in version 5.00.0518.9 of comctl32.dll we need to always
 961     //     change both normal and selected image - otherwise the change simply
 962     //     doesn't take place!
 963     DoSetItemImages(item, imageNormal, imageSel);
 964 }
 965
 966 wxTreeItemData *wxTreeCtrl::GetItemData(const wxTreeItemId& item) const
 967 {
 968     wxTreeViewItem tvItem(item, TVIF_PARAM);
 969     if ( !DoGetItem(&tvItem) )
 970     {
 971         return NULL;
 972     }
 973
 974     wxTreeItemData *data = (wxTreeItemData *)tvItem.lParam;
 975     if ( IsDataIndirect(data) )
 976     {
 977         data = ((wxTreeItemIndirectData *)data)->GetData();
 978     }
 979
 980     return data;
 981 }
 982
 983 void wxTreeCtrl::SetItemData(const wxTreeItemId& item, wxTreeItemData *data)
 984 {
 985     // first, associate this piece of data with this item
 986     if ( data )
 987     {
 988         data->SetId(item);
 989     }
 990
 991     wxTreeViewItem tvItem(item, TVIF_PARAM);
 992
 993     if ( HasIndirectData(item) )
 994     {
 995         if ( DoGetItem(&tvItem) )
 996         {
 997             ((wxTreeItemIndirectData *)tvItem.lParam)->SetData(data);
 998         }
 999         else
1000         {
1001             wxFAIL_MSG( wxT("failed to change tree items data") );
1002         }
1003     }
1004     else
1005     {
1006         tvItem.lParam = (LPARAM)data;
1007         DoSetItem(&tvItem);
1008     }
1009 }
1010
1011 void wxTreeCtrl::SetIndirectItemData(const wxTreeItemId& item,
1012                                      wxTreeItemIndirectData *data)
1013 {
1014     // this should never happen because it's unnecessary and will probably lead
1015     // to crash too because the code elsewhere supposes that the pointer the
1016     // wxTreeItemIndirectData has is a real wxItemData and not
1017     // wxTreeItemIndirectData as well
1018     wxASSERT_MSG( !HasIndirectData(item), wxT("setting indirect data twice?") );
1019
1020     SetItemData(item, data);
1021 }
1022
1023 bool wxTreeCtrl::HasIndirectData(const wxTreeItemId& item) const
1024 {
1025     // query the item itself
1026     wxTreeViewItem tvItem(item, TVIF_PARAM);
1027     if ( !DoGetItem(&tvItem) )
1028     {
1029         return FALSE;
1030     }
1031
1032     wxTreeItemData *data = (wxTreeItemData *)tvItem.lParam;
1033
1034     return data && IsDataIndirect(data);
1035 }
1036
1037 void wxTreeCtrl::SetItemHasChildren(const wxTreeItemId& item, bool has)
1038 {
1039     wxTreeViewItem tvItem(item, TVIF_CHILDREN);
1040     tvItem.cChildren = (int)has;
1041     DoSetItem(&tvItem);
1042 }
1043
1044 void wxTreeCtrl::SetItemBold(const wxTreeItemId& item, bool bold)
1045 {
1046     wxTreeViewItem tvItem(item, TVIF_STATE, TVIS_BOLD);
1047     tvItem.state = bold ? TVIS_BOLD : 0;
1048     DoSetItem(&tvItem);
1049 }
1050
1051 void wxTreeCtrl::SetItemDropHighlight(const wxTreeItemId& item, bool highlight)
1052 {
1053     wxTreeViewItem tvItem(item, TVIF_STATE, TVIS_DROPHILITED);
1054     tvItem.state = highlight ? TVIS_DROPHILITED : 0;
1055     DoSetItem(&tvItem);
1056 }
1057
1058 void wxTreeCtrl::RefreshItem(const wxTreeItemId& item)
1059 {
1060     wxRect rect;
1061     if ( GetBoundingRect(item, rect) )
1062     {
1063         RefreshRect(rect);
1064     }
1065 }
1066
1067 void wxTreeCtrl::SetItemTextColour(const wxTreeItemId& item,
1068                                    const wxColour& col)
1069 {
1070     m_hasAnyAttr = TRUE;
1071
1072     long id = (long)(WXHTREEITEM)item;
1073     wxTreeItemAttr *attr = (wxTreeItemAttr *)m_attrs.Get(id);
1074     if ( !attr )
1075     {
1076         attr = new wxTreeItemAttr;
1077         m_attrs.Put(id, (wxObject *)attr);
1078     }
1079
1080     attr->SetTextColour(col);
1081
1082     RefreshItem(item);
1083 }
1084
1085 void wxTreeCtrl::SetItemBackgroundColour(const wxTreeItemId& item,
1086                                          const wxColour& col)
1087 {
1088     m_hasAnyAttr = TRUE;
1089
1090     long id = (long)(WXHTREEITEM)item;
1091     wxTreeItemAttr *attr = (wxTreeItemAttr *)m_attrs.Get(id);
1092     if ( !attr )
1093     {
1094         attr = new wxTreeItemAttr;
1095         m_attrs.Put(id, (wxObject *)attr);
1096     }
1097
1098     attr->SetBackgroundColour(col);
1099
1100     RefreshItem(item);
1101 }
1102
1103 void wxTreeCtrl::SetItemFont(const wxTreeItemId& item, const wxFont& font)
1104 {
1105     m_hasAnyAttr = TRUE;
1106
1107     long id = (long)(WXHTREEITEM)item;
1108     wxTreeItemAttr *attr = (wxTreeItemAttr *)m_attrs.Get(id);
1109     if ( !attr )
1110     {
1111         attr = new wxTreeItemAttr;
1112         m_attrs.Put(id, (wxObject *)attr);
1113     }
1114
1115     attr->SetFont(font);
1116
1117     RefreshItem(item);
1118 }
1119
1120 // ----------------------------------------------------------------------------
1121 // Item status
1122 // ----------------------------------------------------------------------------
1123
1124 bool wxTreeCtrl::IsVisible(const wxTreeItemId& item) const
1125 {
1126     // Bug in Gnu-Win32 headers, so don't use the macro TreeView_GetItemRect
1127     RECT rect;
1128
1129     // this ugliness comes directly from MSDN - it *is* the correct way to pass
1130     // the HTREEITEM with TVM_GETITEMRECT
1131     *(WXHTREEITEM *)&rect = (WXHTREEITEM)item;
1132
1133     // FALSE means get item rect for the whole item, not only text
1134     return SendMessage(GetHwnd(), TVM_GETITEMRECT, FALSE, (LPARAM)&rect) != 0;
1135 }
1136
1137 bool wxTreeCtrl::ItemHasChildren(const wxTreeItemId& item) const
1138 {
1139     wxTreeViewItem tvItem(item, TVIF_CHILDREN);
1140     DoGetItem(&tvItem);
1141
1142     return tvItem.cChildren != 0;
1143 }
1144
1145 bool wxTreeCtrl::IsExpanded(const wxTreeItemId& item) const
1146 {
1147     // probably not a good idea to put it here
1148     //wxASSERT( ItemHasChildren(item) );
1149
1150     wxTreeViewItem tvItem(item, TVIF_STATE, TVIS_EXPANDED);
1151     DoGetItem(&tvItem);
1152
1153     return (tvItem.state & TVIS_EXPANDED) != 0;
1154 }
1155
1156 bool wxTreeCtrl::IsSelected(const wxTreeItemId& item) const
1157 {
1158     wxTreeViewItem tvItem(item, TVIF_STATE, TVIS_SELECTED);
1159     DoGetItem(&tvItem);
1160
1161     return (tvItem.state & TVIS_SELECTED) != 0;
1162 }
1163
1164 bool wxTreeCtrl::IsBold(const wxTreeItemId& item) const
1165 {
1166     wxTreeViewItem tvItem(item, TVIF_STATE, TVIS_BOLD);
1167     DoGetItem(&tvItem);
1168
1169     return (tvItem.state & TVIS_BOLD) != 0;
1170 }
1171
1172 // ----------------------------------------------------------------------------
1173 // navigation
1174 // ----------------------------------------------------------------------------
1175
1176 wxTreeItemId wxTreeCtrl::GetRootItem() const
1177 {
1178     return wxTreeItemId((WXHTREEITEM) TreeView_GetRoot(GetHwnd()));
1179 }
1180
1181 wxTreeItemId wxTreeCtrl::GetSelection() const
1182 {
1183     wxCHECK_MSG( !(m_windowStyle & wxTR_MULTIPLE), (long)(WXHTREEITEM)0,
1184                  wxT("this only works with single selection controls") );
1185
1186     return wxTreeItemId((WXHTREEITEM) TreeView_GetSelection(GetHwnd()));
1187 }
1188
1189 wxTreeItemId wxTreeCtrl::GetParent(const wxTreeItemId& item) const
1190 {
1191     return wxTreeItemId((WXHTREEITEM) TreeView_GetParent(GetHwnd(), HITEM(item)));
1192 }
1193
1194 wxTreeItemId wxTreeCtrl::GetFirstChild(const wxTreeItemId& item,
1195                                        long& _cookie) const
1196 {
1197     // remember the last child returned in 'cookie'
1198     _cookie = (long)TreeView_GetChild(GetHwnd(), HITEM(item));
1199
1200     return wxTreeItemId((WXHTREEITEM)_cookie);
1201 }
1202
1203 wxTreeItemId wxTreeCtrl::GetNextChild(const wxTreeItemId& WXUNUSED(item),
1204                                       long& _cookie) const
1205 {
1206     wxTreeItemId l = wxTreeItemId((WXHTREEITEM)TreeView_GetNextSibling(GetHwnd(),
1207                                    HITEM(_cookie)));
1208     _cookie = (long)l;
1209
1210     return l;
1211 }
1212
1213 wxTreeItemId wxTreeCtrl::GetLastChild(const wxTreeItemId& item) const
1214 {
1215     // can this be done more efficiently?
1216     long cookie;
1217
1218     wxTreeItemId childLast,
1219     child = GetFirstChild(item, cookie);
1220     while ( child.IsOk() )
1221     {
1222         childLast = child;
1223         child = GetNextChild(item, cookie);
1224     }
1225
1226     return childLast;
1227 }
1228
1229 wxTreeItemId wxTreeCtrl::GetNextSibling(const wxTreeItemId& item) const
1230 {
1231     return wxTreeItemId((WXHTREEITEM) TreeView_GetNextSibling(GetHwnd(), HITEM(item)));
1232 }
1233
1234 wxTreeItemId wxTreeCtrl::GetPrevSibling(const wxTreeItemId& item) const
1235 {
1236     return wxTreeItemId((WXHTREEITEM) TreeView_GetPrevSibling(GetHwnd(), HITEM(item)));
1237 }
1238
1239 wxTreeItemId wxTreeCtrl::GetFirstVisibleItem() const
1240 {
1241     return wxTreeItemId((WXHTREEITEM) TreeView_GetFirstVisible(GetHwnd()));
1242 }
1243
1244 wxTreeItemId wxTreeCtrl::GetNextVisible(const wxTreeItemId& item) const
1245 {
1246     wxASSERT_MSG( IsVisible(item), wxT("The item you call GetNextVisible() for must be visible itself!"));
1247
1248     return wxTreeItemId((WXHTREEITEM) TreeView_GetNextVisible(GetHwnd(), HITEM(item)));
1249 }
1250
1251 wxTreeItemId wxTreeCtrl::GetPrevVisible(const wxTreeItemId& item) const
1252 {
1253     wxASSERT_MSG( IsVisible(item), wxT("The item you call GetPrevVisible() for must be visible itself!"));
1254
1255     return wxTreeItemId((WXHTREEITEM) TreeView_GetPrevVisible(GetHwnd(), HITEM(item)));
1256 }
1257
1258 // ----------------------------------------------------------------------------
1259 // multiple selections emulation
1260 // ----------------------------------------------------------------------------
1261
1262 bool wxTreeCtrl::IsItemChecked(const wxTreeItemId& item) const
1263 {
1264     // receive the desired information.
1265     wxTreeViewItem tvItem(item, TVIF_STATE, TVIS_STATEIMAGEMASK);
1266     DoGetItem(&tvItem);
1267
1268     // state image indices are 1 based
1269     return ((tvItem.state >> 12) - 1) == 1;
1270 }
1271
1272 void wxTreeCtrl::SetItemCheck(const wxTreeItemId& item, bool check)
1273 {
1274     // receive the desired information.
1275     wxTreeViewItem tvItem(item, TVIF_STATE, TVIS_STATEIMAGEMASK);
1276
1277     // state images are one-based
1278     tvItem.state = (check ? 2 : 1) << 12;
1279
1280     DoSetItem(&tvItem);
1281 }
1282
1283 size_t wxTreeCtrl::GetSelections(wxArrayTreeItemIds& selections) const
1284 {
1285     TraverseSelections selector(this, selections);
1286
1287     return selector.GetCount();
1288 }
1289
1290 // ----------------------------------------------------------------------------
1291 // Usual operations
1292 // ----------------------------------------------------------------------------
1293
1294 wxTreeItemId wxTreeCtrl::DoInsertItem(const wxTreeItemId& parent,
1295                                       wxTreeItemId hInsertAfter,
1296                                       const wxString& text,
1297                                       int image, int selectedImage,
1298                                       wxTreeItemData *data)
1299 {
1300     wxCHECK_MSG( parent.IsOk() || !TreeView_GetRoot(GetHwnd()),
1301                  wxTreeItemId(),
1302                  _T("can't have more than one root in the tree") );
1303
1304     TV_INSERTSTRUCT tvIns;
1305     tvIns.hParent = HITEM(parent);
1306     tvIns.hInsertAfter = HITEM(hInsertAfter);
1307
1308     // this is how we insert the item as the first child: supply a NULL
1309     // hInsertAfter
1310     if ( !tvIns.hInsertAfter )
1311     {
1312         tvIns.hInsertAfter = TVI_FIRST;
1313     }
1314
1315     UINT mask = 0;
1316     if ( !text.IsEmpty() )
1317     {
1318         mask |= TVIF_TEXT;
1319         tvIns.item.pszText = (wxChar *)text.c_str();  // cast is ok
1320     }
1321     else
1322     {
1323         tvIns.item.pszText = NULL;
1324         tvIns.item.cchTextMax = 0;
1325     }
1326
1327     if ( image != -1 )
1328     {
1329         mask |= TVIF_IMAGE;
1330         tvIns.item.iImage = image;
1331
1332         if ( selectedImage == -1 )
1333         {
1334             // take the same image for selected icon if not specified
1335             selectedImage = image;
1336         }
1337     }
1338
1339     if ( selectedImage != -1 )
1340     {
1341         mask |= TVIF_SELECTEDIMAGE;
1342         tvIns.item.iSelectedImage = selectedImage;
1343     }
1344
1345     if ( data != NULL )
1346     {
1347         mask |= TVIF_PARAM;
1348         tvIns.item.lParam = (LPARAM)data;
1349     }
1350
1351     tvIns.item.mask = mask;
1352
1353     HTREEITEM id = (HTREEITEM) TreeView_InsertItem(GetHwnd(), &tvIns);
1354     if ( id == 0 )
1355     {
1356         wxLogLastError(wxT("TreeView_InsertItem"));
1357     }
1358
1359     if ( data != NULL )
1360     {
1361         // associate the application tree item with Win32 tree item handle
1362         data->SetId((WXHTREEITEM)id);
1363     }
1364
1365     return wxTreeItemId((WXHTREEITEM)id);
1366 }
1367
1368 // for compatibility only
1369 wxTreeItemId wxTreeCtrl::InsertItem(const wxTreeItemId& parent,
1370                                     const wxString& text,
1371                                     int image, int selImage,
1372                                     long insertAfter)
1373 {
1374     return DoInsertItem(parent, (WXHTREEITEM)insertAfter, text,
1375                         image, selImage, NULL);
1376 }
1377
1378 wxTreeItemId wxTreeCtrl::AddRoot(const wxString& text,
1379                                  int image, int selectedImage,
1380                                  wxTreeItemData *data)
1381 {
1382     return DoInsertItem(wxTreeItemId((long) (WXHTREEITEM) 0), (long)(WXHTREEITEM) 0,
1383                         text, image, selectedImage, data);
1384 }
1385
1386 wxTreeItemId wxTreeCtrl::PrependItem(const wxTreeItemId& parent,
1387                                      const wxString& text,
1388                                      int image, int selectedImage,
1389                                      wxTreeItemData *data)
1390 {
1391     return DoInsertItem(parent, (WXHTREEITEM) TVI_FIRST,
1392                         text, image, selectedImage, data);
1393 }
1394
1395 wxTreeItemId wxTreeCtrl::InsertItem(const wxTreeItemId& parent,
1396                                     const wxTreeItemId& idPrevious,
1397                                     const wxString& text,
1398                                     int image, int selectedImage,
1399                                     wxTreeItemData *data)
1400 {
1401     return DoInsertItem(parent, idPrevious, text, image, selectedImage, data);
1402 }
1403
1404 wxTreeItemId wxTreeCtrl::InsertItem(const wxTreeItemId& parent,
1405                                     size_t index,
1406                                     const wxString& text,
1407                                     int image, int selectedImage,
1408                                     wxTreeItemData *data)
1409 {
1410     // find the item from index
1411     long cookie;
1412     wxTreeItemId idPrev, idCur = GetFirstChild(parent, cookie);
1413     while ( index != 0 && idCur.IsOk() )
1414     {
1415         index--;
1416
1417         idPrev = idCur;
1418         idCur = GetNextChild(parent, cookie);
1419     }
1420
1421     // assert, not check: if the index is invalid, we will append the item
1422     // to the end
1423     wxASSERT_MSG( index == 0, _T("bad index in wxTreeCtrl::InsertItem") );
1424
1425     return DoInsertItem(parent, idPrev, text, image, selectedImage, data);
1426 }
1427
1428 wxTreeItemId wxTreeCtrl::AppendItem(const wxTreeItemId& parent,
1429                                     const wxString& text,
1430                                     int image, int selectedImage,
1431                                     wxTreeItemData *data)
1432 {
1433     return DoInsertItem(parent, (WXHTREEITEM) TVI_LAST,
1434                         text, image, selectedImage, data);
1435 }
1436
1437 void wxTreeCtrl::Delete(const wxTreeItemId& item)
1438 {
1439     if ( !TreeView_DeleteItem(GetHwnd(), HITEM(item)) )
1440     {
1441         wxLogLastError(wxT("TreeView_DeleteItem"));
1442     }
1443 }
1444
1445 // delete all children (but don't delete the item itself)
1446 void wxTreeCtrl::DeleteChildren(const wxTreeItemId& item)
1447 {
1448     long cookie;
1449
1450     wxArrayLong children;
1451     wxTreeItemId child = GetFirstChild(item, cookie);
1452     while ( child.IsOk() )
1453     {
1454         children.Add((long)(WXHTREEITEM)child);
1455
1456         child = GetNextChild(item, cookie);
1457     }
1458
1459     size_t nCount = children.Count();
1460     for ( size_t n = 0; n < nCount; n++ )
1461     {
1462         if ( !TreeView_DeleteItem(GetHwnd(), (HTREEITEM)children[n]) )
1463         {
1464             wxLogLastError(wxT("TreeView_DeleteItem"));
1465         }
1466     }
1467 }
1468
1469 void wxTreeCtrl::DeleteAllItems()
1470 {
1471     if ( !TreeView_DeleteAllItems(GetHwnd()) )
1472     {
1473         wxLogLastError(wxT("TreeView_DeleteAllItems"));
1474     }
1475 }
1476
1477 void wxTreeCtrl::DoExpand(const wxTreeItemId& item, int flag)
1478 {
1479     wxASSERT_MSG( flag == TVE_COLLAPSE ||
1480                   flag == (TVE_COLLAPSE | TVE_COLLAPSERESET) ||
1481                   flag == TVE_EXPAND   ||
1482                   flag == TVE_TOGGLE,
1483                   wxT("Unknown flag in wxTreeCtrl::DoExpand") );
1484
1485     // TreeView_Expand doesn't send TVN_ITEMEXPAND(ING) messages, so we must
1486     // emulate them. This behaviour has changed slightly with comctl32.dll
1487     // v 4.70 - now it does send them but only the first time. To maintain
1488     // compatible behaviour and also in order to not have surprises with the
1489     // future versions, don't rely on this and still do everything ourselves.
1490     // To avoid that the messages be sent twice when the item is expanded for
1491     // the first time we must clear TVIS_EXPANDEDONCE style manually.
1492
1493     wxTreeViewItem tvItem(item, TVIF_STATE, TVIS_EXPANDEDONCE);
1494     tvItem.state = 0;
1495     DoSetItem(&tvItem);
1496
1497     if ( TreeView_Expand(GetHwnd(), HITEM(item), flag) != 0 )
1498     {
1499         wxTreeEvent event(wxEVT_NULL, m_windowId);
1500         event.m_item = item;
1501
1502         bool isExpanded = IsExpanded(item);
1503
1504         event.SetEventObject(this);
1505
1506         // FIXME return value of {EXPAND|COLLAPS}ING event handler is discarded
1507         event.SetEventType(g_events[isExpanded][TRUE]);
1508         GetEventHandler()->ProcessEvent(event);
1509
1510         event.SetEventType(g_events[isExpanded][FALSE]);
1511         GetEventHandler()->ProcessEvent(event);
1512     }
1513     //else: change didn't took place, so do nothing at all
1514 }
1515
1516 void wxTreeCtrl::Expand(const wxTreeItemId& item)
1517 {
1518     DoExpand(item, TVE_EXPAND);
1519 }
1520
1521 void wxTreeCtrl::Collapse(const wxTreeItemId& item)
1522 {
1523     DoExpand(item, TVE_COLLAPSE);
1524 }
1525
1526 void wxTreeCtrl::CollapseAndReset(const wxTreeItemId& item)
1527 {
1528     DoExpand(item, TVE_COLLAPSE | TVE_COLLAPSERESET);
1529 }
1530
1531 void wxTreeCtrl::Toggle(const wxTreeItemId& item)
1532 {
1533     DoExpand(item, TVE_TOGGLE);
1534 }
1535
1536 void wxTreeCtrl::ExpandItem(const wxTreeItemId& item, int action)
1537 {
1538     DoExpand(item, action);
1539 }
1540
1541 void wxTreeCtrl::Unselect()
1542 {
1543     wxASSERT_MSG( !(m_windowStyle & wxTR_MULTIPLE),
1544                   wxT("doesn't make sense, may be you want UnselectAll()?") );
1545
1546     // just remove the selection
1547     SelectItem(wxTreeItemId((long) (WXHTREEITEM) 0));
1548 }
1549
1550 void wxTreeCtrl::UnselectAll()
1551 {
1552     if ( m_windowStyle & wxTR_MULTIPLE )
1553     {
1554         wxArrayTreeItemIds selections;
1555         size_t count = GetSelections(selections);
1556         for ( size_t n = 0; n < count; n++ )
1557         {
1558 #if wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
1559             SetItemCheck(selections[n], FALSE);
1560 #else // !wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
1561             ::UnselectItem(GetHwnd(), HITEM(selections[n]));
1562 #endif // wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE/!wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
1563         }
1564     }
1565     else
1566     {
1567         // just remove the selection
1568         Unselect();
1569     }
1570 }
1571
1572 void wxTreeCtrl::SelectItem(const wxTreeItemId& item)
1573 {
1574     if ( m_windowStyle & wxTR_MULTIPLE )
1575     {
1576 #if wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
1577         // selecting the item means checking it
1578         SetItemCheck(item);
1579 #else // !wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
1580         ::SelectItem(GetHwnd(), HITEM(item));
1581 #endif // wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE/!wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
1582     }
1583     else
1584     {
1585         // inspite of the docs (MSDN Jan 99 edition), we don't seem to receive
1586         // the notification from the control (i.e. TVN_SELCHANG{ED|ING}), so
1587         // send them ourselves
1588
1589         wxTreeEvent event(wxEVT_NULL, m_windowId);
1590         event.m_item = item;
1591         event.SetEventObject(this);
1592
1593         event.SetEventType(wxEVT_COMMAND_TREE_SEL_CHANGING);
1594         if ( !GetEventHandler()->ProcessEvent(event) || event.IsAllowed() )
1595         {
1596             if ( !TreeView_SelectItem(GetHwnd(), HITEM(item)) )
1597             {
1598                 wxLogLastError(wxT("TreeView_SelectItem"));
1599             }
1600             else
1601             {
1602                 event.SetEventType(wxEVT_COMMAND_TREE_SEL_CHANGED);
1603                 (void)GetEventHandler()->ProcessEvent(event);
1604             }
1605         }
1606         //else: program vetoed the change
1607     }
1608 }
1609
1610 void wxTreeCtrl::EnsureVisible(const wxTreeItemId& item)
1611 {
1612     // no error return
1613     TreeView_EnsureVisible(GetHwnd(), HITEM(item));
1614 }
1615
1616 void wxTreeCtrl::ScrollTo(const wxTreeItemId& item)
1617 {
1618     if ( !TreeView_SelectSetFirstVisible(GetHwnd(), HITEM(item)) )
1619     {
1620         wxLogLastError(wxT("TreeView_SelectSetFirstVisible"));
1621     }
1622 }
1623
1624 wxTextCtrl* wxTreeCtrl::GetEditControl() const
1625 {
1626     // normally, we could try to do something like this to return something
1627     // even when the editing was started by the user and not by calling
1628     // EditLabel() - but as nobody has asked for this so far and there might be
1629     // problems in the code below, I leave it disabled for now (VZ)
1630 #if 0
1631     if ( !m_textCtrl )
1632     {
1633         HWND hwndText = TreeView_GetEditControl(GetHwnd());
1634         if ( hwndText )
1635         {
1636             m_textCtrl = new wxTextCtrl(this, -1);
1637             m_textCtrl->Hide();
1638             m_textCtrl->SetHWND((WXHWND)hwndText);
1639         }
1640         //else: not editing label right now
1641     }
1642 #endif // 0
1643
1644     return m_textCtrl;
1645 }
1646
1647 void wxTreeCtrl::DeleteTextCtrl()
1648 {
1649     if ( m_textCtrl )
1650     {
1651         // the HWND corresponding to this control is deleted by the tree
1652         // control itself and we don't know when exactly this happens, so check
1653         // if the window still exists before calling UnsubclassWin()
1654         if ( !::IsWindow(GetHwndOf(m_textCtrl)) )
1655         {
1656             m_textCtrl->SetHWND(0);
1657         }
1658
1659         m_textCtrl->UnsubclassWin();
1660         m_textCtrl->SetHWND(0);
1661         delete m_textCtrl;
1662         m_textCtrl = NULL;
1663     }
1664 }
1665
1666 wxTextCtrl* wxTreeCtrl::EditLabel(const wxTreeItemId& item,
1667                                   wxClassInfo* textControlClass)
1668 {
1669     wxASSERT( textControlClass->IsKindOf(CLASSINFO(wxTextCtrl)) );
1670
1671     DeleteTextCtrl();
1672
1673     HWND hWnd = (HWND) TreeView_EditLabel(GetHwnd(), HITEM(item));
1674
1675     // this is not an error - the TVN_BEGINLABELEDIT handler might have
1676     // returned FALSE
1677     if ( !hWnd )
1678     {
1679         return NULL;
1680     }
1681
1682     m_textCtrl = (wxTextCtrl *)textControlClass->CreateObject();
1683     m_textCtrl->SetParent(this);
1684     m_textCtrl->SetHWND((WXHWND)hWnd);
1685     m_textCtrl->SubclassWin((WXHWND)hWnd);
1686
1687     return m_textCtrl;
1688 }
1689
1690 // End label editing, optionally cancelling the edit
1691 void wxTreeCtrl::EndEditLabel(const wxTreeItemId& WXUNUSED(item), bool discardChanges)
1692 {
1693     TreeView_EndEditLabelNow(GetHwnd(), discardChanges);
1694
1695     DeleteTextCtrl();
1696 }
1697
1698 wxTreeItemId wxTreeCtrl::HitTest(const wxPoint& point, int& flags)
1699 {
1700     TV_HITTESTINFO hitTestInfo;
1701     hitTestInfo.pt.x = (int)point.x;
1702     hitTestInfo.pt.y = (int)point.y;
1703
1704     TreeView_HitTest(GetHwnd(), &hitTestInfo);
1705
1706     flags = 0;
1707
1708     // avoid repetition
1709     #define TRANSLATE_FLAG(flag) if ( hitTestInfo.flags & TVHT_##flag ) \
1710                                     flags |= wxTREE_HITTEST_##flag
1711
1712     TRANSLATE_FLAG(ABOVE);
1713     TRANSLATE_FLAG(BELOW);
1714     TRANSLATE_FLAG(NOWHERE);
1715     TRANSLATE_FLAG(ONITEMBUTTON);
1716     TRANSLATE_FLAG(ONITEMICON);
1717     TRANSLATE_FLAG(ONITEMINDENT);
1718     TRANSLATE_FLAG(ONITEMLABEL);
1719     TRANSLATE_FLAG(ONITEMRIGHT);
1720     TRANSLATE_FLAG(ONITEMSTATEICON);
1721     TRANSLATE_FLAG(TOLEFT);
1722     TRANSLATE_FLAG(TORIGHT);
1723
1724     #undef TRANSLATE_FLAG
1725
1726     return wxTreeItemId((WXHTREEITEM) hitTestInfo.hItem);
1727 }
1728
1729 bool wxTreeCtrl::GetBoundingRect(const wxTreeItemId& item,
1730                                  wxRect& rect,
1731                                  bool textOnly) const
1732 {
1733     RECT rc;
1734     if ( TreeView_GetItemRect(GetHwnd(), HITEM(item),
1735                               &rc, textOnly) )
1736     {
1737         rect = wxRect(wxPoint(rc.left, rc.top), wxPoint(rc.right, rc.bottom));
1738
1739         return TRUE;
1740     }
1741     else
1742     {
1743         // couldn't retrieve rect: for example, item isn't visible
1744         return FALSE;
1745     }
1746 }
1747
1748 // ----------------------------------------------------------------------------
1749 // sorting stuff
1750 // ----------------------------------------------------------------------------
1751
1752 // this is just a tiny namespace which is friend to wxTreeCtrl and so can use
1753 // functions such as IsDataIndirect()
1754 class wxTreeSortHelper
1755 {
1756 public:
1757     static int CALLBACK Compare(LPARAM data1, LPARAM data2, LPARAM tree);
1758
1759 private:
1760     static wxTreeItemId GetIdFromData(wxTreeCtrl *tree, LPARAM item)
1761     {
1762         wxTreeItemData *data = (wxTreeItemData *)item;
1763         if ( tree->IsDataIndirect(data) )
1764         {
1765             data = ((wxTreeItemIndirectData *)data)->GetData();
1766         }
1767
1768         return data->GetId();
1769     }
1770 };
1771
1772 int CALLBACK wxTreeSortHelper::Compare(LPARAM pItem1,
1773                                        LPARAM pItem2,
1774                                        LPARAM htree)
1775 {
1776     wxCHECK_MSG( pItem1 && pItem2, 0,
1777                  wxT("sorting tree without data doesn't make sense") );
1778
1779     wxTreeCtrl *tree = (wxTreeCtrl *)htree;
1780
1781     return tree->OnCompareItems(GetIdFromData(tree, pItem1),
1782                                 GetIdFromData(tree, pItem2));
1783 }
1784
1785 int wxTreeCtrl::OnCompareItems(const wxTreeItemId& item1,
1786                                const wxTreeItemId& item2)
1787 {
1788     return wxStrcmp(GetItemText(item1), GetItemText(item2));
1789 }
1790
1791 void wxTreeCtrl::SortChildren(const wxTreeItemId& item)
1792 {
1793     // rely on the fact that TreeView_SortChildren does the same thing as our
1794     // default behaviour, i.e. sorts items alphabetically and so call it
1795     // directly if we're not in derived class (much more efficient!)
1796     if ( GetClassInfo() == CLASSINFO(wxTreeCtrl) )
1797     {
1798         TreeView_SortChildren(GetHwnd(), HITEM(item), 0);
1799     }
1800     else
1801     {
1802         TV_SORTCB tvSort;
1803         tvSort.hParent = HITEM(item);
1804         tvSort.lpfnCompare = wxTreeSortHelper::Compare;
1805         tvSort.lParam = (LPARAM)this;
1806         TreeView_SortChildrenCB(GetHwnd(), &tvSort, 0 /* reserved */);
1807     }
1808 }
1809
1810 // ----------------------------------------------------------------------------
1811 // implementation
1812 // ----------------------------------------------------------------------------
1813
1814 bool wxTreeCtrl::MSWCommand(WXUINT cmd, WXWORD id)
1815 {
1816     if ( cmd == EN_UPDATE )
1817     {
1818         wxCommandEvent event(wxEVT_COMMAND_TEXT_UPDATED, id);
1819         event.SetEventObject( this );
1820         ProcessCommand(event);
1821     }
1822     else if ( cmd == EN_KILLFOCUS )
1823     {
1824         wxCommandEvent event(wxEVT_KILL_FOCUS, id);
1825         event.SetEventObject( this );
1826         ProcessCommand(event);
1827     }
1828     else
1829     {
1830         // nothing done
1831         return FALSE;
1832     }
1833
1834     // command processed
1835     return TRUE;
1836 }
1837
1838 // we hook into WndProc to process WM_MOUSEMOVE/WM_BUTTONUP messages - as we
1839 // only do it during dragging, minimize wxWin overhead (this is important for
1840 // WM_MOUSEMOVE as they're a lot of them) by catching Windows messages directly
1841 // instead of passing by wxWin events
1842 long wxTreeCtrl::MSWWindowProc(WXUINT nMsg, WXWPARAM wParam, WXLPARAM lParam)
1843 {
1844     bool processed = FALSE;
1845     long rc = 0;
1846     bool isMultiple = (GetWindowStyle() & wxTR_MULTIPLE) != 0;
1847
1848     if ( (nMsg >= WM_MOUSEFIRST) && (nMsg <= WM_MOUSELAST) )
1849     {
1850         // we only process mouse messages here and these parameters have the same
1851         // meaning for all of them
1852         int x = GET_X_LPARAM(lParam),
1853             y = GET_Y_LPARAM(lParam);
1854         HTREEITEM htItem = GetItemFromPoint(GetHwnd(), x, y);
1855
1856         switch ( nMsg )
1857         {
1858 #if !wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
1859             case WM_LBUTTONDOWN:
1860                 if ( htItem && isMultiple )
1861                 {
1862                     if ( wParam & MK_CONTROL )
1863                     {
1864                         SetFocus();
1865
1866                         // toggle selected state
1867                         ToggleItemSelection(GetHwnd(), htItem);
1868
1869                         ::SetFocus(GetHwnd(), htItem);
1870
1871                         // reset on any click without Shift
1872                         m_htSelStart = 0;
1873
1874                         processed = TRUE;
1875                     }
1876                     else if ( wParam & MK_SHIFT )
1877                     {
1878                         // this selects all items between the starting one and
1879                         // the current
1880
1881                         if ( !m_htSelStart )
1882                         {
1883                             // take the focused item
1884                             m_htSelStart = (WXHTREEITEM)
1885                                 TreeView_GetSelection(GetHwnd());
1886                         }
1887
1888                         SelectRange(GetHwnd(), HITEM(m_htSelStart), htItem,
1889                                     !(wParam & MK_CONTROL));
1890
1891                         ::SetFocus(GetHwnd(), htItem);
1892
1893                         processed = TRUE;
1894                     }
1895                     else // normal click
1896                     {
1897                         // clear the selection and then let the default handler
1898                         // do the job
1899                         UnselectAll();
1900
1901                         // prevent the click from starting in-place editing
1902                         // when there was no selection in the control
1903                         TreeView_SelectItem(GetHwnd(), 0);
1904
1905                         // reset on any click without Shift
1906                         m_htSelStart = 0;
1907                     }
1908                 }
1909                 break;
1910 #endif // wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
1911
1912             case WM_MOUSEMOVE:
1913                 if ( m_dragImage )
1914                 {
1915                     m_dragImage->Move(wxPoint(x, y));
1916                     if ( htItem )
1917                     {
1918                         // highlight the item as target (hiding drag image is
1919                         // necessary - otherwise the display will be corrupted)
1920                         m_dragImage->Hide();
1921                         TreeView_SelectDropTarget(GetHwnd(), htItem);
1922                         m_dragImage->Show();
1923                     }
1924                 }
1925                 break;
1926
1927             case WM_LBUTTONUP:
1928             case WM_RBUTTONUP:
1929                 if ( m_dragImage )
1930                 {
1931                     m_dragImage->EndDrag();
1932                     delete m_dragImage;
1933                     m_dragImage = NULL;
1934
1935                     // generate the drag end event
1936                     wxTreeEvent event(wxEVT_COMMAND_TREE_END_DRAG, m_windowId);
1937
1938                     event.m_item = (WXHTREEITEM)htItem;
1939                     event.m_pointDrag = wxPoint(x, y);
1940                     event.SetEventObject(this);
1941
1942                     (void)GetEventHandler()->ProcessEvent(event);
1943
1944                     // if we don't do it, the tree seems to think that 2 items
1945                     // are selected simultaneously which is quite weird
1946                     TreeView_SelectDropTarget(GetHwnd(), 0);
1947                 }
1948                 break;
1949         }
1950     }
1951 #if !wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
1952     else if ( (nMsg == WM_SETFOCUS || nMsg == WM_KILLFOCUS) && isMultiple )
1953     {
1954         // the tree control greys out the selected item when it loses focus and
1955         // paints it as selected again when it regains it, but it won't do it
1956         // for the other items itself - help it
1957         wxArrayTreeItemIds selections;
1958         size_t count = GetSelections(selections);
1959         RECT rect;
1960         for ( size_t n = 0; n < count; n++ )
1961         {
1962             // TreeView_GetItemRect() will return FALSE if item is not visible,
1963             // which may happen perfectly well
1964             if ( TreeView_GetItemRect(GetHwnd(), HITEM(selections[n]),
1965                                       &rect, TRUE) )
1966             {
1967                 ::InvalidateRect(GetHwnd(), &rect, FALSE);
1968             }
1969         }
1970     }
1971     else if ( nMsg == WM_KEYDOWN && isMultiple )
1972     {
1973         bool bCtrl = wxIsCtrlDown(),
1974              bShift = wxIsShiftDown();
1975
1976         // we handle.arrows and space, but not page up/down and home/end: the
1977         // latter should be easy, but not the former
1978
1979         HTREEITEM htSel = (HTREEITEM)TreeView_GetSelection(GetHwnd());
1980         if ( !m_htSelStart )
1981         {
1982             m_htSelStart = (WXHTREEITEM)htSel;
1983         }
1984
1985         if ( wParam == VK_SPACE )
1986         {
1987             if ( bCtrl )
1988             {
1989                 ToggleItemSelection(GetHwnd(), htSel);
1990             }
1991             else
1992             {
1993                 UnselectAll();
1994
1995                 ::SelectItem(GetHwnd(), htSel);
1996             }
1997
1998             processed = TRUE;
1999         }
2000         else if ( wParam == VK_UP || wParam == VK_DOWN )
2001         {
2002             if ( !bCtrl && !bShift )
2003             {
2004                 // no modifiers, just clear selection and then let the default
2005                 // processing to take place
2006                 UnselectAll();
2007             }
2008             else if ( htSel )
2009             {
2010                 (void)wxControl::MSWWindowProc(nMsg, wParam, lParam);
2011
2012                 HTREEITEM htNext = (HTREEITEM)(wParam == VK_UP
2013                                     ? TreeView_GetPrevVisible(GetHwnd(), htSel)
2014                                     : TreeView_GetNextVisible(GetHwnd(), htSel));
2015
2016                 if ( !htNext )
2017                 {
2018                     // at the top/bottom
2019                     htNext = htSel;
2020                 }
2021
2022                 if ( bShift )
2023                 {
2024                     SelectRange(GetHwnd(), HITEM(m_htSelStart), htNext);
2025                 }
2026                 else // bCtrl
2027                 {
2028                     // without changing selection
2029                     ::SetFocus(GetHwnd(), htNext);
2030                 }
2031
2032                 processed = TRUE;
2033             }
2034         }
2035     }
2036 #endif // !wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
2037     if ( !processed )
2038         rc = wxControl::MSWWindowProc(nMsg, wParam, lParam);
2039
2040     return rc;
2041 }
2042
2043 // process WM_NOTIFY Windows message
2044 bool wxTreeCtrl::MSWOnNotify(int idCtrl, WXLPARAM lParam, WXLPARAM *result)
2045 {
2046     wxTreeEvent event(wxEVT_NULL, m_windowId);
2047     wxEventType eventType = wxEVT_NULL;
2048     NMHDR *hdr = (NMHDR *)lParam;
2049
2050     switch ( hdr->code )
2051     {
2052         case TVN_BEGINDRAG:
2053             eventType = wxEVT_COMMAND_TREE_BEGIN_DRAG;
2054             // fall through
2055
2056         case TVN_BEGINRDRAG:
2057             {
2058                 if ( eventType == wxEVT_NULL )
2059                     eventType = wxEVT_COMMAND_TREE_BEGIN_RDRAG;
2060                 //else: left drag, already set above
2061
2062                 NM_TREEVIEW *tv = (NM_TREEVIEW *)lParam;
2063
2064                 event.m_item = (WXHTREEITEM) tv->itemNew.hItem;
2065                 event.m_pointDrag = wxPoint(tv->ptDrag.x, tv->ptDrag.y);
2066
2067                 // don't allow dragging by default: the user code must
2068                 // explicitly say that it wants to allow it to avoid breaking
2069                 // the old apps
2070                 event.Veto();
2071             }
2072             break;
2073
2074         case TVN_BEGINLABELEDIT:
2075             {
2076                 eventType = wxEVT_COMMAND_TREE_BEGIN_LABEL_EDIT;
2077                 TV_DISPINFO *info = (TV_DISPINFO *)lParam;
2078
2079                 event.m_item = (WXHTREEITEM) info->item.hItem;
2080                 event.m_label = info->item.pszText;
2081             }
2082             break;
2083
2084         case TVN_DELETEITEM:
2085             {
2086                 eventType = wxEVT_COMMAND_TREE_DELETE_ITEM;
2087                 NM_TREEVIEW *tv = (NM_TREEVIEW *)lParam;
2088
2089                 event.m_item = (WXHTREEITEM)tv->itemOld.hItem;
2090
2091                 if ( m_hasAnyAttr )
2092                 {
2093                     delete (wxTreeItemAttr *)m_attrs.
2094                                 Delete((long)tv->itemOld.hItem);
2095                 }
2096             }
2097             break;
2098
2099         case TVN_ENDLABELEDIT:
2100             {
2101                 eventType = wxEVT_COMMAND_TREE_END_LABEL_EDIT;
2102                 TV_DISPINFO *info = (TV_DISPINFO *)lParam;
2103
2104                 event.m_item = (WXHTREEITEM)info->item.hItem;
2105                 event.m_label = info->item.pszText;
2106                 if (info->item.pszText == NULL)
2107                     return FALSE;
2108                 break;
2109             }
2110
2111         case TVN_GETDISPINFO:
2112             eventType = wxEVT_COMMAND_TREE_GET_INFO;
2113             // fall through
2114
2115         case TVN_SETDISPINFO:
2116             {
2117                 if ( eventType == wxEVT_NULL )
2118                     eventType = wxEVT_COMMAND_TREE_SET_INFO;
2119                 //else: get, already set above
2120
2121                 TV_DISPINFO *info = (TV_DISPINFO *)lParam;
2122
2123                 event.m_item = (WXHTREEITEM) info->item.hItem;
2124                 break;
2125             }
2126
2127         case TVN_ITEMEXPANDING:
2128         case TVN_ITEMEXPANDED:
2129             {
2130                 NM_TREEVIEW* tv = (NM_TREEVIEW*)lParam;
2131
2132                 bool expand = FALSE;
2133                 switch ( tv->action )
2134                 {
2135                     case TVE_EXPAND:
2136                         expand = TRUE;
2137                         break;
2138
2139                     case TVE_COLLAPSE:
2140                         expand = FALSE;
2141                         break;
2142
2143                     default:
2144                         wxLogDebug(wxT("unexpected code %d in TVN_ITEMEXPAND message"), tv->action);
2145                 }
2146
2147                 bool ing = ((int)hdr->code == TVN_ITEMEXPANDING);
2148                 eventType = g_events[expand][ing];
2149
2150                 event.m_item = (WXHTREEITEM) tv->itemNew.hItem;
2151             }
2152             break;
2153
2154         case TVN_KEYDOWN:
2155             {
2156                 eventType = wxEVT_COMMAND_TREE_KEY_DOWN;
2157                 TV_KEYDOWN *info = (TV_KEYDOWN *)lParam;
2158
2159                 // we pass 0 as last CreateKeyEvent() parameter because we
2160                 // don't have access to the real key press flags here - but as
2161                 // it is only used to determin wxKeyEvent::m_altDown flag it's
2162                 // not too bad
2163                 event.m_evtKey = CreateKeyEvent(wxEVT_KEY_DOWN,
2164                                                 wxCharCodeMSWToWX(info->wVKey),
2165                                                 0);
2166
2167                 // a separate event for Space/Return
2168                 if ( !wxIsCtrlDown() && !wxIsShiftDown() &&
2169                      ((info->wVKey == VK_SPACE) || (info->wVKey == VK_RETURN)) )
2170                 {
2171                     wxTreeEvent event2(wxEVT_COMMAND_TREE_ITEM_ACTIVATED,
2172                                        m_windowId);
2173                     event2.SetEventObject(this);
2174                     if ( !(GetWindowStyle() & wxTR_MULTIPLE) )
2175                     {
2176                         event2.m_item = GetSelection();
2177                     }
2178                     //else: don't know how to get it
2179
2180                     (void)GetEventHandler()->ProcessEvent(event2);
2181                 }
2182             }
2183             break;
2184
2185         case TVN_SELCHANGED:
2186             eventType = wxEVT_COMMAND_TREE_SEL_CHANGED;
2187             // fall through
2188
2189         case TVN_SELCHANGING:
2190             {
2191                 if ( eventType == wxEVT_NULL )
2192                     eventType = wxEVT_COMMAND_TREE_SEL_CHANGING;
2193                 //else: already set above
2194
2195                 NM_TREEVIEW* tv = (NM_TREEVIEW *)lParam;
2196
2197                 event.m_item = (WXHTREEITEM) tv->itemNew.hItem;
2198                 event.m_itemOld = (WXHTREEITEM) tv->itemOld.hItem;
2199             }
2200             break;
2201
2202 #if defined(_WIN32_IE) && _WIN32_IE >= 0x300 && !wxUSE_COMCTL32_SAFELY && !( defined(__GNUWIN32__) && !wxCHECK_W32API_VERSION( 1, 0 ) )
2203         case NM_CUSTOMDRAW:
2204             {
2205                 LPNMTVCUSTOMDRAW lptvcd = (LPNMTVCUSTOMDRAW)lParam;
2206                 NMCUSTOMDRAW& nmcd = lptvcd->nmcd;
2207                 switch ( nmcd.dwDrawStage )
2208                 {
2209                     case CDDS_PREPAINT:
2210                         // if we've got any items with non standard attributes,
2211                         // notify us before painting each item
2212                         *result = m_hasAnyAttr ? CDRF_NOTIFYITEMDRAW
2213                                                : CDRF_DODEFAULT;
2214                         break;
2215
2216                     case CDDS_ITEMPREPAINT:
2217                         {
2218                             wxTreeItemAttr *attr =
2219                                 (wxTreeItemAttr *)m_attrs.Get(nmcd.dwItemSpec);
2220
2221                             if ( !attr )
2222                             {
2223                                 // nothing to do for this item
2224                                 *result = CDRF_DODEFAULT;
2225                                 break;
2226                             }
2227
2228                             HFONT hFont;
2229                             wxColour colText, colBack;
2230                             if ( attr->HasFont() )
2231                             {
2232                                 wxFont font = attr->GetFont();
2233                                 hFont = (HFONT)font.GetResourceHandle();
2234                             }
2235                             else
2236                             {
2237                                 hFont = 0;
2238                             }
2239
2240                             if ( attr->HasTextColour() )
2241                             {
2242                                 colText = attr->GetTextColour();
2243                             }
2244                             else
2245                             {
2246                                 colText = GetForegroundColour();
2247                             }
2248
2249                             // selection colours should override ours
2250                             if ( nmcd.uItemState & CDIS_SELECTED )
2251                             {
2252                                 DWORD clrBk = ::GetSysColor(COLOR_HIGHLIGHT);
2253                                 lptvcd->clrTextBk = clrBk;
2254
2255                                 // try to make the text visible
2256                                 lptvcd->clrText = wxColourToRGB(colText);
2257                                 lptvcd->clrText |= ~clrBk;
2258                                 lptvcd->clrText &= 0x00ffffff;
2259                             }
2260                             else
2261                             {
2262                                 if ( attr->HasBackgroundColour() )
2263                                 {
2264                                     colBack = attr->GetBackgroundColour();
2265                                 }
2266                                 else
2267                                 {
2268                                     colBack = GetBackgroundColour();
2269                                 }
2270
2271                                 lptvcd->clrText = wxColourToRGB(colText);
2272                                 lptvcd->clrTextBk = wxColourToRGB(colBack);
2273                             }
2274
2275                             // note that if we wanted to set colours for
2276                             // individual columns (subitems), we would have
2277                             // returned CDRF_NOTIFYSUBITEMREDRAW from here
2278                             if ( hFont )
2279                             {
2280                                 ::SelectObject(nmcd.hdc, hFont);
2281
2282                                 *result = CDRF_NEWFONT;
2283                             }
2284                             else
2285                             {
2286                                 *result = CDRF_DODEFAULT;
2287                             }
2288                         }
2289                         break;
2290
2291                     default:
2292                         *result = CDRF_DODEFAULT;
2293                 }
2294             }
2295
2296             // we always process it
2297             return TRUE;
2298 #endif // _WIN32_IE >= 0x300
2299
2300         case NM_DBLCLK:
2301         case NM_RCLICK:
2302             {
2303                 TV_HITTESTINFO tvhti;
2304                 ::GetCursorPos(&tvhti.pt);
2305                 ::ScreenToClient(GetHwnd(), &tvhti.pt);
2306                 if ( TreeView_HitTest(GetHwnd(), &tvhti) )
2307                 {
2308                     if ( tvhti.flags & TVHT_ONITEM )
2309                     {
2310                         event.m_item = (WXHTREEITEM) tvhti.hItem;
2311                         eventType = (int)hdr->code == NM_DBLCLK
2312                                     ? wxEVT_COMMAND_TREE_ITEM_ACTIVATED
2313                                     : wxEVT_COMMAND_TREE_ITEM_RIGHT_CLICK;
2314
2315                         event.m_pointDrag.x = tvhti.pt.x;
2316                         event.m_pointDrag.y = tvhti.pt.y;
2317                     }
2318
2319                     break;
2320                 }
2321             }
2322             // fall through
2323
2324         default:
2325             return wxControl::MSWOnNotify(idCtrl, lParam, result);
2326     }
2327
2328     event.SetEventObject(this);
2329     event.SetEventType(eventType);
2330
2331     bool processed = GetEventHandler()->ProcessEvent(event);
2332
2333     // post processing
2334     switch ( hdr->code )
2335     {
2336         case NM_DBLCLK:
2337             // we translate NM_DBLCLK into ACTIVATED event, so don't interpret
2338             // the return code of this event handler as the return value for
2339             // NM_DBLCLK - otherwise, double clicking the item to toggle its
2340             // expanded status would never work
2341             *result = FALSE;
2342             break;
2343
2344         case TVN_BEGINDRAG:
2345         case TVN_BEGINRDRAG:
2346             if ( event.IsAllowed() )
2347             {
2348                 // normally this is impossible because the m_dragImage is
2349                 // deleted once the drag operation is over
2350                 wxASSERT_MSG( !m_dragImage, _T("starting to drag once again?") );
2351
2352                 m_dragImage = new wxDragImage(*this, event.m_item);
2353                 m_dragImage->BeginDrag(wxPoint(0, 0), this);
2354                 m_dragImage->Show();
2355             }
2356             break;
2357
2358         case TVN_DELETEITEM:
2359             {
2360                 // NB: we might process this message using wxWindows event
2361                 //     tables, but due to overhead of wxWin event system we
2362                 //     prefer to do it here ourself (otherwise deleting a tree
2363                 //     with many items is just too slow)
2364                 NM_TREEVIEW* tv = (NM_TREEVIEW *)lParam;
2365
2366                 wxTreeItemId item = event.m_item;
2367                 if ( HasIndirectData(item) )
2368                 {
2369                     wxTreeItemIndirectData *data = (wxTreeItemIndirectData *)
2370                                                         tv->itemOld.lParam;
2371                     delete data; // can't be NULL here
2372                 }
2373                 else
2374                 {
2375                     wxTreeItemData *data = (wxTreeItemData *)tv->itemOld.lParam;
2376                     delete data; // may be NULL, ok
2377                 }
2378
2379                 processed = TRUE; // Make sure we don't get called twice
2380             }
2381             break;
2382
2383         case TVN_BEGINLABELEDIT:
2384             // return TRUE to cancel label editing
2385             *result = !event.IsAllowed();
2386             break;
2387
2388         case TVN_ENDLABELEDIT:
2389             // return TRUE to set the label to the new string: note that we
2390             // also must pretend that we did process the message or it is going
2391             // to be passed to DefWindowProc() which will happily return FALSE
2392             // cancelling the label change
2393             *result = event.IsAllowed();
2394             processed = TRUE;
2395
2396             // ensure that we don't have the text ctrl which is going to be
2397             // deleted any more
2398             DeleteTextCtrl();
2399             break;
2400
2401         case TVN_SELCHANGING:
2402         case TVN_ITEMEXPANDING:
2403             // return TRUE to prevent the action from happening
2404             *result = !event.IsAllowed();
2405             break;
2406
2407         case TVN_GETDISPINFO:
2408             // NB: so far the user can't set the image himself anyhow, so do it
2409             //     anyway - but this may change later
2410 //          if ( /* !processed && */ 1 )
2411             {
2412                 wxTreeItemId item = event.m_item;
2413                 TV_DISPINFO *info = (TV_DISPINFO *)lParam;
2414                 if ( info->item.mask & TVIF_IMAGE )
2415                 {
2416                     info->item.iImage =
2417                         DoGetItemImageFromData
2418                         (
2419                          item,
2420                          IsExpanded(item) ? wxTreeItemIcon_Expanded
2421                                           : wxTreeItemIcon_Normal
2422                         );
2423                 }
2424                 if ( info->item.mask & TVIF_SELECTEDIMAGE )
2425                 {
2426                     info->item.iSelectedImage =
2427                         DoGetItemImageFromData
2428                         (
2429                          item,
2430                          IsExpanded(item) ? wxTreeItemIcon_SelectedExpanded
2431                                           : wxTreeItemIcon_Selected
2432                         );
2433                 }
2434                         }
2435             break;
2436
2437         //default:
2438             // for the other messages the return value is ignored and there is
2439             // nothing special to do
2440     }
2441     return processed;
2442 }
2443
2444 #endif // __WIN95__
2445
2446 #endif // wxUSE_TREECTRL