src/generic/treectrl.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        treectrl.cpp
   3 // Purpose:     generic tree control implementation
   4 // Author:      Robert Roebling
   5 // Created:     01/02/97
   6 // Modified:    22/10/98 - almost total rewrite, simpler interface (VZ)
   7 // Id:          $Id$
   8 // Copyright:   (c) 1998 Robert Roebling, Julian Smart and Markus Holzem
   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 #include "wx/treectrl.h"
  32 #include "wx/generic/imaglist.h"
  33 #include "wx/settings.h"
  34 #include "wx/log.h"
  35 #include "wx/intl.h"
  36 #include "wx/dynarray.h"
  37 #include "wx/arrimpl.cpp"
  38 #include "wx/dcclient.h"
  39 #include "wx/msgdlg.h"
  40
  41 // -----------------------------------------------------------------------------
  42 // array types
  43 // -----------------------------------------------------------------------------
  44
  45 class WXDLLEXPORT wxGenericTreeItem;
  46
  47 WX_DEFINE_ARRAY(wxGenericTreeItem *, wxArrayGenericTreeItems);
  48 WX_DEFINE_OBJARRAY(wxArrayTreeItemIds);
  49
  50 // -----------------------------------------------------------------------------
  51 // private classes
  52 // -----------------------------------------------------------------------------
  53
  54 // a tree item
  55 class WXDLLEXPORT wxGenericTreeItem
  56 {
  57 public:
  58   // ctors & dtor
  59   wxGenericTreeItem() { m_data = NULL; }
  60   wxGenericTreeItem( wxGenericTreeItem *parent,
  61                      const wxString& text,
  62                      wxDC& dc,
  63                      int image, int selImage,
  64                      wxTreeItemData *data );
  65
  66   ~wxGenericTreeItem();
  67
  68   // trivial accessors
  69   wxArrayGenericTreeItems& GetChildren() { return m_children; }
  70
  71   const wxString& GetText() const { return m_text; }
  72   int GetImage() const { return m_image; }
  73   int GetSelectedImage() const { return m_selImage; }
  74   wxTreeItemData *GetData() const { return m_data; }
  75
  76   void SetText( const wxString &text );
  77   void SetImage(int image) { m_image = image; }
  78   void SetSelectedImage(int image) { m_selImage = image; }
  79   void SetData(wxTreeItemData *data) { m_data = data; }
  80
  81   void SetHasPlus(bool has = TRUE) { m_hasPlus = has; }
  82
  83   void SetBold(bool bold) { m_isBold = bold; }
  84
  85   int GetX() const { return m_x; }
  86   int GetY() const { return m_y; }
  87
  88   void SetX(int x) { m_x = x; }
  89   void SetY(int y) { m_y = y; }
  90
  91   int  GetHeight() const { return m_height; }
  92   int  GetWidth()  const { return m_width; }
  93
  94   void SetHeight(int h) { m_height = h; }
  95   void SetWidth(int w) { m_width = w; }
  96
  97
  98   wxGenericTreeItem *GetParent() const { return m_parent; }
  99
 100   // operations
 101     // deletes all children notifying the treectrl about it if !NULL pointer
 102     // given
 103   void DeleteChildren(wxTreeCtrl *tree = NULL);
 104     // FIXME don't know what is it for
 105   void Reset();
 106
 107   // get count of all children (and grand children if 'recursively')
 108   size_t GetChildrenCount(bool recursively = TRUE) const;
 109
 110   void Insert(wxGenericTreeItem *child, size_t index)
 111     { m_children.Insert(child, index); }
 112
 113   void SetCross( int x, int y );
 114   void GetSize( int &x, int &y, const wxTreeCtrl* );
 115
 116   // return the item at given position (or NULL if no item), onButton is TRUE
 117   // if the point belongs to the item's button, otherwise it lies on the
 118   // button's label
 119   wxGenericTreeItem *HitTest( const wxPoint& point, const wxTreeCtrl *, int &flags);
 120
 121   void Expand() { m_isCollapsed = FALSE; }
 122   void Collapse() { m_isCollapsed = TRUE; }
 123
 124   void SetHilight( bool set = TRUE ) { m_hasHilight = set; }
 125
 126   // status inquiries
 127   bool HasChildren() const { return !m_children.IsEmpty(); }
 128   bool HasHilight()  const { return m_hasHilight; }
 129   bool IsExpanded()  const { return !m_isCollapsed; }
 130   bool HasPlus()     const { return m_hasPlus || HasChildren(); }
 131   bool IsBold()      const { return m_isBold; }
 132
 133 private:
 134   wxString            m_text;
 135
 136   int                 m_image,
 137                       m_selImage;
 138
 139   wxTreeItemData     *m_data;
 140
 141   // use bitfields to save size
 142   int                 m_isCollapsed :1;
 143   int                 m_hasHilight  :1; // same as focused
 144   int                 m_hasPlus     :1; // used for item which doesn't have
 145                                         // children but still has a [+] button
 146   int                 m_isBold      :1; // render the label in bold font
 147
 148   int                 m_x, m_y;
 149   long                m_height, m_width;
 150   int                 m_xCross, m_yCross;
 151   int                 m_level;
 152   wxArrayGenericTreeItems m_children;
 153   wxGenericTreeItem  *m_parent;
 154 };
 155
 156 // =============================================================================
 157 // implementation
 158 // =============================================================================
 159
 160
 161 // -----------------------------------------------------------------------------
 162 // wxTreeRenameTimer (internal)
 163 // -----------------------------------------------------------------------------
 164
 165 wxTreeRenameTimer::wxTreeRenameTimer( wxTreeCtrl *owner )
 166 {
 167     m_owner = owner;
 168 }
 169
 170 void wxTreeRenameTimer::Notify()
 171 {
 172     m_owner->OnRenameTimer();
 173 }
 174
 175 //-----------------------------------------------------------------------------
 176 // wxTreeTextCtrl (internal)
 177 //-----------------------------------------------------------------------------
 178
 179 IMPLEMENT_DYNAMIC_CLASS(wxTreeTextCtrl,wxTextCtrl);
 180
 181 BEGIN_EVENT_TABLE(wxTreeTextCtrl,wxTextCtrl)
 182     EVT_CHAR           (wxTreeTextCtrl::OnChar)
 183     EVT_KILL_FOCUS     (wxTreeTextCtrl::OnKillFocus)
 184 END_EVENT_TABLE()
 185
 186 wxTreeTextCtrl::wxTreeTextCtrl( wxWindow *parent, const wxWindowID id,
 187     bool *accept, wxString *res, wxTreeCtrl *owner,
 188     const wxString &value, const wxPoint &pos, const wxSize &size,
 189     int style, const wxValidator& validator, const wxString &name ) :
 190   wxTextCtrl( parent, id, value, pos, size, style, validator, name )
 191 {
 192     m_res = res;
 193     m_accept = accept;
 194     m_owner = owner;
 195     (*m_accept) = FALSE;
 196     (*m_res) = "";
 197     m_startValue = value;
 198 }
 199
 200 void wxTreeTextCtrl::OnChar( wxKeyEvent &event )
 201 {
 202     if (event.m_keyCode == WXK_RETURN)
 203     {
 204         (*m_accept) = TRUE;
 205         (*m_res) = GetValue();
 206         m_owner->SetFocus();
 207         return;
 208     }
 209     if (event.m_keyCode == WXK_ESCAPE)
 210     {
 211         (*m_accept) = FALSE;
 212         (*m_res) = "";
 213         m_owner->SetFocus();
 214         return;
 215     }
 216     event.Skip();
 217 }
 218
 219 void wxTreeTextCtrl::OnKillFocus( wxFocusEvent &WXUNUSED(event) )
 220 {
 221     if (wxPendingDelete.Member(this)) return;
 222
 223     wxPendingDelete.Append(this);
 224
 225     if ((*m_accept) && ((*m_res) != m_startValue))
 226         m_owner->OnRenameAccept();
 227 }
 228
 229 #define PIXELS_PER_UNIT 10
 230 // -----------------------------------------------------------------------------
 231 // wxTreeEvent
 232 // -----------------------------------------------------------------------------
 233
 234 IMPLEMENT_DYNAMIC_CLASS(wxTreeEvent, wxNotifyEvent)
 235
 236 wxTreeEvent::wxTreeEvent( wxEventType commandType, int id )
 237            : wxNotifyEvent( commandType, id )
 238 {
 239   m_code = 0;
 240   m_itemOld = (wxGenericTreeItem *)NULL;
 241 }
 242
 243 // -----------------------------------------------------------------------------
 244 // wxGenericTreeItem
 245 // -----------------------------------------------------------------------------
 246
 247 wxGenericTreeItem::wxGenericTreeItem(wxGenericTreeItem *parent,
 248                                      const wxString& text,
 249                                      wxDC& dc,
 250                                      int image, int selImage,
 251                                      wxTreeItemData *data)
 252                  : m_text(text)
 253 {
 254   m_image = image;
 255   m_selImage = selImage;
 256   m_data = data;
 257   m_x = m_y = 0;
 258   m_xCross = m_yCross = 0;
 259
 260   m_level = 0;
 261
 262   m_isCollapsed = TRUE;
 263   m_hasHilight = FALSE;
 264   m_hasPlus = FALSE;
 265   m_isBold = FALSE;
 266
 267   m_parent = parent;
 268
 269   dc.GetTextExtent( m_text, &m_width, &m_height );
 270   // TODO : Add the width of the image
 271   // PB   : We don't know which image is shown (image, selImage)
 272   //        We don't even know imageList from the treectrl this item belongs to !!!
 273   // At this point m_width doesn't mean much, this can be remove !
 274 }
 275
 276 wxGenericTreeItem::~wxGenericTreeItem()
 277 {
 278   delete m_data;
 279
 280   wxASSERT_MSG( m_children.IsEmpty(),
 281                 _T("please call DeleteChildren() before deleting the item") );
 282 }
 283
 284 void wxGenericTreeItem::DeleteChildren(wxTreeCtrl *tree)
 285 {
 286   size_t count = m_children.Count();
 287   for ( size_t n = 0; n < count; n++ )
 288   {
 289     wxGenericTreeItem *child = m_children[n];
 290     if ( tree )
 291     {
 292       tree->SendDeleteEvent(child);
 293     }
 294
 295     child->DeleteChildren(tree);
 296     delete child;
 297   }
 298
 299   m_children.Empty();
 300 }
 301
 302 void wxGenericTreeItem::SetText( const wxString &text )
 303 {
 304   m_text = text;
 305 }
 306
 307 void wxGenericTreeItem::Reset()
 308 {
 309   m_text.Empty();
 310   m_image =
 311   m_selImage = -1;
 312   m_data = NULL;
 313   m_x = m_y =
 314   m_height = m_width = 0;
 315   m_xCross =
 316   m_yCross = 0;
 317
 318   m_level = 0;
 319
 320   DeleteChildren();
 321   m_isCollapsed = TRUE;
 322
 323   m_parent = (wxGenericTreeItem *)NULL;
 324 }
 325
 326 size_t wxGenericTreeItem::GetChildrenCount(bool recursively) const
 327 {
 328   size_t count = m_children.Count();
 329   if ( !recursively )
 330     return count;
 331
 332   size_t total = count;
 333   for ( size_t n = 0; n < count; ++n )
 334   {
 335     total += m_children[n]->GetChildrenCount();
 336   }
 337
 338   return total;
 339 }
 340
 341 void wxGenericTreeItem::SetCross( int x, int y )
 342 {
 343   m_xCross = x;
 344   m_yCross = y;
 345 }
 346
 347 void wxGenericTreeItem::GetSize( int &x, int &y, const wxTreeCtrl *theTree )
 348 {
 349   int bottomY=m_y+theTree->GetLineHeight(this);
 350   if ( y < bottomY ) y = bottomY;
 351   int width = m_x +  m_width;
 352   if ( x < width ) x = width;
 353
 354   if (IsExpanded())
 355   {
 356     size_t count = m_children.Count();
 357     for ( size_t n = 0; n < count; ++n )
 358     {
 359       m_children[n]->GetSize( x, y, theTree );
 360     }
 361   }
 362 }
 363
 364 wxGenericTreeItem *wxGenericTreeItem::HitTest( const wxPoint& point,
 365                                                const wxTreeCtrl *theTree,
 366                                                int &flags)
 367 {
 368   if ((point.y > m_y) && (point.y < m_y + theTree->GetLineHeight(this)))
 369   {
 370     if (point.y<m_y+theTree->GetLineHeight(this)/2) flags|=wxTREE_HITTEST_ONITEMUPPERPART;
 371     else flags|=wxTREE_HITTEST_ONITEMLOWERPART;
 372
 373     // FIXME why +5?
 374     //  Because that is the size of the plus sign, RR
 375     if ((point.x > m_xCross-5) && (point.x < m_xCross+5) &&
 376         (point.y > m_yCross-5) && (point.y < m_yCross+5) &&
 377         (IsExpanded() || HasPlus()))
 378     {
 379       flags|=wxTREE_HITTEST_ONITEMBUTTON;
 380       return this;
 381     }
 382
 383     if ((point.x >= m_x) && (point.x <= m_x+m_width))
 384     {
 385       int image_w = -1;
 386       int image_h;
 387
 388       // assuming every image (normal and selected ) has the same size !
 389       if ((m_image!=-1) && theTree->m_imageListNormal)
 390           theTree->m_imageListNormal->GetSize(m_image, image_w, image_h);
 391
 392       if ((image_w != -1) && (point.x <= m_x + image_w + 1))
 393         flags|=wxTREE_HITTEST_ONITEMICON;
 394       else
 395         flags|=wxTREE_HITTEST_ONITEMLABEL;
 396
 397       return this;
 398     }
 399
 400     if (point.x < m_x)         flags|=wxTREE_HITTEST_ONITEMIDENT;
 401     if (point.x > m_x+m_width) flags|=wxTREE_HITTEST_ONITEMRIGHT;
 402
 403     return this;
 404   }
 405   else
 406   {
 407     if (!m_isCollapsed)
 408     {
 409       size_t count = m_children.Count();
 410       for ( size_t n = 0; n < count; n++ )
 411       {
 412         wxGenericTreeItem *res = m_children[n]->HitTest( point, theTree, flags );
 413         if ( res != NULL )
 414           return res;
 415       }
 416     }
 417   }
 418
 419   flags|=wxTREE_HITTEST_NOWHERE;
 420   return NULL;
 421 }
 422
 423 // -----------------------------------------------------------------------------
 424 // wxTreeCtrl implementation
 425 // -----------------------------------------------------------------------------
 426
 427 IMPLEMENT_DYNAMIC_CLASS(wxTreeCtrl, wxScrolledWindow)
 428
 429 BEGIN_EVENT_TABLE(wxTreeCtrl,wxScrolledWindow)
 430   EVT_PAINT          (wxTreeCtrl::OnPaint)
 431   EVT_MOUSE_EVENTS   (wxTreeCtrl::OnMouse)
 432   EVT_CHAR           (wxTreeCtrl::OnChar)
 433   EVT_SET_FOCUS      (wxTreeCtrl::OnSetFocus)
 434   EVT_KILL_FOCUS     (wxTreeCtrl::OnKillFocus)
 435   EVT_IDLE           (wxTreeCtrl::OnIdle)
 436 END_EVENT_TABLE()
 437
 438 // -----------------------------------------------------------------------------
 439 // construction/destruction
 440 // -----------------------------------------------------------------------------
 441
 442 void wxTreeCtrl::Init()
 443 {
 444   m_current =
 445   m_key_current =
 446   m_anchor = (wxGenericTreeItem *) NULL;
 447   m_hasFocus = FALSE;
 448   m_dirty = FALSE;
 449
 450   m_xScroll = 0;
 451   m_yScroll = 0;
 452   m_lineHeight = 10;
 453   m_indent = 15;
 454   m_spacing = 18;
 455
 456   m_hilightBrush = new wxBrush
 457     (
 458       wxSystemSettings::GetSystemColour(wxSYS_COLOUR_HIGHLIGHT),
 459       wxSOLID
 460     );
 461
 462   m_imageListNormal =
 463   m_imageListState = (wxImageList *) NULL;
 464
 465   m_dragCount = 0;
 466
 467   m_renameTimer = new wxTreeRenameTimer( this );
 468 }
 469
 470 bool wxTreeCtrl::Create(wxWindow *parent, wxWindowID id,
 471                         const wxPoint& pos, const wxSize& size,
 472                         long style,
 473             const wxValidator &validator,
 474             const wxString& name )
 475 {
 476   Init();
 477
 478   wxScrolledWindow::Create( parent, id, pos, size, style|wxHSCROLL|wxVSCROLL, name );
 479
 480 #if wxUSE_VALIDATORS
 481   SetValidator( validator );
 482 #endif
 483
 484   SetBackgroundColour( *wxWHITE );
 485 //  m_dottedPen = wxPen( "grey", 0, wxDOT );
 486   m_dottedPen = wxPen( "grey", 0, 0 );
 487
 488   return TRUE;
 489 }
 490
 491 wxTreeCtrl::~wxTreeCtrl()
 492 {
 493   wxDELETE( m_hilightBrush );
 494
 495   DeleteAllItems();
 496
 497   delete m_renameTimer;
 498 }
 499
 500 // -----------------------------------------------------------------------------
 501 // accessors
 502 // -----------------------------------------------------------------------------
 503
 504 size_t wxTreeCtrl::GetCount() const
 505 {
 506   return m_anchor == NULL ? 0u : m_anchor->GetChildrenCount();
 507 }
 508
 509 void wxTreeCtrl::SetIndent(unsigned int indent)
 510 {
 511   m_indent = indent;
 512   m_dirty = TRUE;
 513   Refresh();
 514 }
 515
 516 void wxTreeCtrl::SetSpacing(unsigned int spacing)
 517 {
 518   m_spacing = spacing;
 519   m_dirty = TRUE;
 520   Refresh();
 521 }
 522
 523 size_t wxTreeCtrl::GetChildrenCount(const wxTreeItemId& item, bool recursively)
 524 {
 525   wxCHECK_MSG( item.IsOk(), 0u, _T("invalid tree item") );
 526
 527   return item.m_pItem->GetChildrenCount(recursively);
 528 }
 529
 530 // -----------------------------------------------------------------------------
 531 // functions to work with tree items
 532 // -----------------------------------------------------------------------------
 533
 534 wxString wxTreeCtrl::GetItemText(const wxTreeItemId& item) const
 535 {
 536   wxCHECK_MSG( item.IsOk(), _T(""), _T("invalid tree item") );
 537
 538   return item.m_pItem->GetText();
 539 }
 540
 541 int wxTreeCtrl::GetItemImage(const wxTreeItemId& item) const
 542 {
 543   wxCHECK_MSG( item.IsOk(), -1, _T("invalid tree item") );
 544
 545   return item.m_pItem->GetImage();
 546 }
 547
 548 int wxTreeCtrl::GetItemSelectedImage(const wxTreeItemId& item) const
 549 {
 550   wxCHECK_MSG( item.IsOk(), -1, _T("invalid tree item") );
 551
 552   return item.m_pItem->GetSelectedImage();
 553 }
 554
 555 wxTreeItemData *wxTreeCtrl::GetItemData(const wxTreeItemId& item) const
 556 {
 557   wxCHECK_MSG( item.IsOk(), NULL, _T("invalid tree item") );
 558
 559   return item.m_pItem->GetData();
 560 }
 561
 562 void wxTreeCtrl::SetItemText(const wxTreeItemId& item, const wxString& text)
 563 {
 564   wxCHECK_RET( item.IsOk(), _T("invalid tree item") );
 565
 566   wxClientDC dc(this);
 567   wxGenericTreeItem *pItem = item.m_pItem;
 568   pItem->SetText(text);
 569   CalculateSize(pItem, dc);
 570   RefreshLine(pItem);
 571 }
 572
 573 void wxTreeCtrl::SetItemImage(const wxTreeItemId& item, int image)
 574 {
 575   wxCHECK_RET( item.IsOk(), _T("invalid tree item") );
 576
 577   wxClientDC dc(this);
 578   wxGenericTreeItem *pItem = item.m_pItem;
 579   pItem->SetImage(image);
 580   CalculateSize(pItem, dc);
 581   RefreshLine(pItem);
 582 }
 583
 584 void wxTreeCtrl::SetItemSelectedImage(const wxTreeItemId& item, int image)
 585 {
 586   wxCHECK_RET( item.IsOk(), _T("invalid tree item") );
 587
 588   wxClientDC dc(this);
 589   wxGenericTreeItem *pItem = item.m_pItem;
 590   pItem->SetSelectedImage(image);
 591   CalculateSize(pItem, dc);
 592   RefreshLine(pItem);
 593 }
 594
 595 void wxTreeCtrl::SetItemData(const wxTreeItemId& item, wxTreeItemData *data)
 596 {
 597   wxCHECK_RET( item.IsOk(), _T("invalid tree item") );
 598
 599   item.m_pItem->SetData(data);
 600 }
 601
 602 void wxTreeCtrl::SetItemHasChildren(const wxTreeItemId& item, bool has)
 603 {
 604   wxCHECK_RET( item.IsOk(), _T("invalid tree item") );
 605
 606   wxGenericTreeItem *pItem = item.m_pItem;
 607   pItem->SetHasPlus(has);
 608   RefreshLine(pItem);
 609 }
 610
 611 void wxTreeCtrl::SetItemBold(const wxTreeItemId& item, bool bold)
 612 {
 613   wxCHECK_RET( item.IsOk(), _T("invalid tree item") );
 614
 615   // avoid redrawing the tree if no real change
 616   wxGenericTreeItem *pItem = item.m_pItem;
 617   if ( pItem->IsBold() != bold )
 618   {
 619     pItem->SetBold(bold);
 620     RefreshLine(pItem);
 621   }
 622 }
 623
 624 // -----------------------------------------------------------------------------
 625 // item status inquiries
 626 // -----------------------------------------------------------------------------
 627
 628 bool wxTreeCtrl::IsVisible(const wxTreeItemId& WXUNUSED(item)) const
 629 {
 630   wxFAIL_MSG(_T("not implemented"));
 631
 632   return TRUE;
 633 }
 634
 635 bool wxTreeCtrl::ItemHasChildren(const wxTreeItemId& item) const
 636 {
 637   wxCHECK_MSG( item.IsOk(), FALSE, _T("invalid tree item") );
 638
 639   return !item.m_pItem->GetChildren().IsEmpty();
 640 }
 641
 642 bool wxTreeCtrl::IsExpanded(const wxTreeItemId& item) const
 643 {
 644   wxCHECK_MSG( item.IsOk(), FALSE, _T("invalid tree item") );
 645
 646   return item.m_pItem->IsExpanded();
 647 }
 648
 649 bool wxTreeCtrl::IsSelected(const wxTreeItemId& item) const
 650 {
 651   wxCHECK_MSG( item.IsOk(), FALSE, _T("invalid tree item") );
 652
 653   return item.m_pItem->HasHilight();
 654 }
 655
 656 bool wxTreeCtrl::IsBold(const wxTreeItemId& item) const
 657 {
 658   wxCHECK_MSG( item.IsOk(), FALSE, _T("invalid tree item") );
 659
 660   return item.m_pItem->IsBold();
 661 }
 662
 663 // -----------------------------------------------------------------------------
 664 // navigation
 665 // -----------------------------------------------------------------------------
 666
 667 wxTreeItemId wxTreeCtrl::GetParent(const wxTreeItemId& item) const
 668 {
 669   wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
 670
 671   return item.m_pItem->GetParent();
 672 }
 673
 674 wxTreeItemId wxTreeCtrl::GetFirstChild(const wxTreeItemId& item, long& cookie) const
 675 {
 676   wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
 677
 678   cookie = 0;
 679   return GetNextChild(item, cookie);
 680 }
 681
 682 wxTreeItemId wxTreeCtrl::GetNextChild(const wxTreeItemId& item, long& cookie) const
 683 {
 684   wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
 685
 686   wxArrayGenericTreeItems& children = item.m_pItem->GetChildren();
 687   if ( (size_t)cookie < children.Count() )
 688   {
 689     return children.Item(cookie++);
 690   }
 691   else
 692   {
 693     // there are no more of them
 694     return wxTreeItemId();
 695   }
 696 }
 697
 698 wxTreeItemId wxTreeCtrl::GetLastChild(const wxTreeItemId& item) const
 699 {
 700   wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
 701
 702   wxArrayGenericTreeItems& children = item.m_pItem->GetChildren();
 703   return (children.IsEmpty() ? wxTreeItemId() : wxTreeItemId(children.Last()));
 704 }
 705
 706 wxTreeItemId wxTreeCtrl::GetNextSibling(const wxTreeItemId& item) const
 707 {
 708   wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
 709
 710   wxGenericTreeItem *i = item.m_pItem;
 711   wxGenericTreeItem *parent = i->GetParent();
 712   if ( parent == NULL )
 713   {
 714     // root item doesn't have any siblings
 715     return wxTreeItemId();
 716   }
 717
 718   wxArrayGenericTreeItems& siblings = parent->GetChildren();
 719   int index = siblings.Index(i);
 720   wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent?
 721
 722   size_t n = (size_t)(index + 1);
 723   return n == siblings.Count() ? wxTreeItemId() : wxTreeItemId(siblings[n]);
 724 }
 725
 726 wxTreeItemId wxTreeCtrl::GetPrevSibling(const wxTreeItemId& item) const
 727 {
 728   wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
 729
 730   wxGenericTreeItem *i = item.m_pItem;
 731   wxGenericTreeItem *parent = i->GetParent();
 732   if ( parent == NULL )
 733   {
 734     // root item doesn't have any siblings
 735     return wxTreeItemId();
 736   }
 737
 738   wxArrayGenericTreeItems& siblings = parent->GetChildren();
 739   int index = siblings.Index(i);
 740   wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent?
 741
 742   return index == 0 ? wxTreeItemId()
 743                     : wxTreeItemId(siblings[(size_t)(index - 1)]);
 744 }
 745
 746 wxTreeItemId wxTreeCtrl::GetFirstVisibleItem() const
 747 {
 748   wxFAIL_MSG(_T("not implemented"));
 749
 750   return wxTreeItemId();
 751 }
 752
 753 wxTreeItemId wxTreeCtrl::GetNextVisible(const wxTreeItemId& item) const
 754 {
 755   wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
 756
 757   wxFAIL_MSG(_T("not implemented"));
 758
 759   return wxTreeItemId();
 760 }
 761
 762 wxTreeItemId wxTreeCtrl::GetPrevVisible(const wxTreeItemId& item) const
 763 {
 764   wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
 765
 766   wxFAIL_MSG(_T("not implemented"));
 767
 768   return wxTreeItemId();
 769 }
 770
 771 // -----------------------------------------------------------------------------
 772 // operations
 773 // -----------------------------------------------------------------------------
 774
 775 wxTreeItemId wxTreeCtrl::DoInsertItem(const wxTreeItemId& parentId,
 776                                       size_t previous,
 777                                       const wxString& text,
 778                                       int image, int selImage,
 779                                       wxTreeItemData *data)
 780 {
 781   wxGenericTreeItem *parent = parentId.m_pItem;
 782   if ( !parent )
 783   {
 784     // should we give a warning here?
 785     return AddRoot(text, image, selImage, data);
 786   }
 787
 788   wxClientDC dc(this);
 789   wxGenericTreeItem *item = new wxGenericTreeItem(parent,
 790                                                   text, dc,
 791                                                   image, selImage,
 792                                                   data);
 793
 794   if ( data != NULL )
 795   {
 796     data->m_pItem = item;
 797   }
 798
 799   parent->Insert( item, previous );
 800
 801   m_dirty = TRUE;
 802
 803   return item;
 804 }
 805
 806 wxTreeItemId wxTreeCtrl::AddRoot(const wxString& text,
 807                                  int image, int selImage,
 808                                  wxTreeItemData *data)
 809 {
 810   wxCHECK_MSG( !m_anchor, wxTreeItemId(), _T("tree can have only one root") );
 811
 812   wxClientDC dc(this);
 813   m_anchor = new wxGenericTreeItem((wxGenericTreeItem *)NULL, text, dc,
 814                                    image, selImage, data);
 815   if ( data != NULL )
 816   {
 817     data->m_pItem = m_anchor;
 818   }
 819
 820   Refresh();
 821   AdjustMyScrollbars();
 822
 823   return m_anchor;
 824 }
 825
 826 wxTreeItemId wxTreeCtrl::PrependItem(const wxTreeItemId& parent,
 827                                      const wxString& text,
 828                                      int image, int selImage,
 829                                      wxTreeItemData *data)
 830 {
 831   return DoInsertItem(parent, 0u, text, image, selImage, data);
 832 }
 833
 834 wxTreeItemId wxTreeCtrl::InsertItem(const wxTreeItemId& parentId,
 835                                     const wxTreeItemId& idPrevious,
 836                                     const wxString& text,
 837                                     int image, int selImage,
 838                                     wxTreeItemData *data)
 839 {
 840   wxGenericTreeItem *parent = parentId.m_pItem;
 841   if ( !parent )
 842   {
 843     // should we give a warning here?
 844     return AddRoot(text, image, selImage, data);
 845   }
 846
 847   int index = parent->GetChildren().Index(idPrevious.m_pItem);
 848   wxASSERT_MSG( index != wxNOT_FOUND,
 849                 _T("previous item in wxTreeCtrl::InsertItem() is not a sibling") );
 850   return DoInsertItem(parentId, (size_t)++index, text, image, selImage, data);
 851 }
 852
 853 wxTreeItemId wxTreeCtrl::AppendItem(const wxTreeItemId& parentId,
 854                                     const wxString& text,
 855                                     int image, int selImage,
 856                                     wxTreeItemData *data)
 857 {
 858   wxGenericTreeItem *parent = parentId.m_pItem;
 859   if ( !parent )
 860   {
 861     // should we give a warning here?
 862     return AddRoot(text, image, selImage, data);
 863   }
 864
 865   return DoInsertItem(parent, parent->GetChildren().Count(), text,
 866                       image, selImage, data);
 867 }
 868
 869 void wxTreeCtrl::SendDeleteEvent(wxGenericTreeItem *item)
 870 {
 871   wxTreeEvent event( wxEVT_COMMAND_TREE_DELETE_ITEM, GetId() );
 872   event.m_item = item;
 873   event.SetEventObject( this );
 874   ProcessEvent( event );
 875 }
 876
 877 void wxTreeCtrl::DeleteChildren(const wxTreeItemId& itemId)
 878 {
 879     wxGenericTreeItem *item = itemId.m_pItem;
 880     item->DeleteChildren(this);
 881
 882     m_dirty = TRUE;
 883 }
 884
 885 void wxTreeCtrl::Delete(const wxTreeItemId& itemId)
 886 {
 887   wxGenericTreeItem *item = itemId.m_pItem;
 888   wxGenericTreeItem *parent = item->GetParent();
 889
 890   if ( parent )
 891   {
 892     parent->GetChildren().Remove(item);
 893   }
 894
 895   item->DeleteChildren(this);
 896   SendDeleteEvent(item);
 897   delete item;
 898
 899   m_dirty = TRUE;
 900 }
 901
 902 void wxTreeCtrl::DeleteAllItems()
 903 {
 904   if ( m_anchor )
 905   {
 906     m_anchor->DeleteChildren(this);
 907     delete m_anchor;
 908
 909     m_anchor = NULL;
 910
 911     m_dirty = TRUE;
 912   }
 913 }
 914
 915 void wxTreeCtrl::Expand(const wxTreeItemId& itemId)
 916 {
 917   wxGenericTreeItem *item = itemId.m_pItem;
 918
 919   if ( !item->HasPlus() )
 920     return;
 921
 922   if ( item->IsExpanded() )
 923     return;
 924
 925   wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_EXPANDING, GetId() );
 926   event.m_item = item;
 927   event.SetEventObject( this );
 928   if ( ProcessEvent( event ) && event.m_code )
 929   {
 930     // cancelled by program
 931     return;
 932   }
 933
 934   item->Expand();
 935   CalculatePositions();
 936
 937   RefreshSubtree(item);
 938
 939   event.SetEventType(wxEVT_COMMAND_TREE_ITEM_EXPANDED);
 940   ProcessEvent( event );
 941 }
 942
 943 void wxTreeCtrl::Collapse(const wxTreeItemId& itemId)
 944 {
 945   wxGenericTreeItem *item = itemId.m_pItem;
 946
 947   if ( !item->IsExpanded() )
 948     return;
 949
 950   wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_COLLAPSING, GetId() );
 951   event.m_item = item;
 952   event.SetEventObject( this );
 953   if ( ProcessEvent( event ) && event.m_code )
 954   {
 955     // cancelled by program
 956     return;
 957   }
 958
 959   item->Collapse();
 960
 961   wxArrayGenericTreeItems& children = item->GetChildren();
 962   size_t count = children.Count();
 963   for ( size_t n = 0; n < count; n++ )
 964   {
 965     Collapse(children[n]);
 966   }
 967
 968   CalculatePositions();
 969
 970   RefreshSubtree(item);
 971
 972   event.SetEventType(wxEVT_COMMAND_TREE_ITEM_COLLAPSED);
 973   ProcessEvent( event );
 974 }
 975
 976 void wxTreeCtrl::CollapseAndReset(const wxTreeItemId& item)
 977 {
 978   Collapse(item);
 979   DeleteChildren(item);
 980 }
 981
 982 void wxTreeCtrl::Toggle(const wxTreeItemId& itemId)
 983 {
 984   wxGenericTreeItem *item = itemId.m_pItem;
 985
 986   if ( item->IsExpanded() )
 987     Collapse(itemId);
 988   else
 989     Expand(itemId);
 990 }
 991
 992 void wxTreeCtrl::Unselect()
 993 {
 994   if ( m_current )
 995   {
 996     m_current->SetHilight( FALSE );
 997     RefreshLine( m_current );
 998   }
 999 }
1000
1001 void wxTreeCtrl::UnselectAllChildren(wxGenericTreeItem *item)
1002 {
1003   item->SetHilight(FALSE);
1004   RefreshLine(item);
1005
1006   if (item->HasChildren())
1007     {
1008       wxArrayGenericTreeItems& children = item->GetChildren();
1009       size_t count = children.Count();
1010       for ( size_t n = 0; n < count; ++n )
1011         UnselectAllChildren(children[n]);
1012     }
1013 }
1014
1015 void wxTreeCtrl::UnselectAll()
1016 {
1017   UnselectAllChildren(GetRootItem().m_pItem);
1018 }
1019
1020 // Recursive function !
1021 // To stop we must have crt_item<last_item
1022 // Algorithm :
1023 // Tag all next children, when no more children,
1024 // Move to parent (not to tag)
1025 // Keep going... if we found last_item, we stop.
1026 bool wxTreeCtrl::TagNextChildren(wxGenericTreeItem *crt_item, wxGenericTreeItem *last_item, bool select)
1027 {
1028     wxGenericTreeItem *parent = crt_item->GetParent();
1029
1030     if ( parent == NULL ) // This is root item
1031       return TagAllChildrenUntilLast(crt_item, last_item, select);
1032
1033     wxArrayGenericTreeItems& children = parent->GetChildren();
1034     int index = children.Index(crt_item);
1035     wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent?
1036
1037     size_t count = children.Count();
1038     for (size_t n=(size_t)(index+1); n<count; ++n)
1039       if (TagAllChildrenUntilLast(children[n], last_item, select)) return TRUE;
1040
1041     return TagNextChildren(parent, last_item, select);
1042 }
1043
1044 bool wxTreeCtrl::TagAllChildrenUntilLast(wxGenericTreeItem *crt_item, wxGenericTreeItem *last_item, bool select)
1045 {
1046   crt_item->SetHilight(select);
1047   RefreshLine(crt_item);
1048
1049   if (crt_item==last_item) return TRUE;
1050
1051   if (crt_item->HasChildren())
1052     {
1053       wxArrayGenericTreeItems& children = crt_item->GetChildren();
1054       size_t count = children.Count();
1055       for ( size_t n = 0; n < count; ++n )
1056         if (TagAllChildrenUntilLast(children[n], last_item, select)) return TRUE;
1057     }
1058
1059   return FALSE;
1060 }
1061
1062 void wxTreeCtrl::SelectItemRange(wxGenericTreeItem *item1, wxGenericTreeItem *item2)
1063 {
1064   // item2 is not necessary after item1
1065   wxGenericTreeItem *first=NULL, *last=NULL;
1066
1067   // choice first' and 'last' between item1 and item2
1068   if (item1->GetY()<item2->GetY())
1069     {
1070       first=item1;
1071       last=item2;
1072     }
1073   else
1074     {
1075       first=item2;
1076       last=item1;
1077     }
1078
1079   bool select=m_current->HasHilight();
1080
1081   if (TagAllChildrenUntilLast(first,last,select)) return;
1082
1083   TagNextChildren(first,last,select);
1084 }
1085
1086 void wxTreeCtrl::SelectItem(const wxTreeItemId& itemId,
1087                             bool unselect_others,
1088                             bool extended_select)
1089 {
1090     wxCHECK_RET( itemId.IsOk(), _T("invalid tree item") );
1091
1092     bool is_single=!(GetWindowStyleFlag() & wxTR_MULTIPLE);
1093     wxGenericTreeItem *item = itemId.m_pItem;
1094
1095     //wxCHECK_RET( ( (!unselect_others) && is_single),
1096     //           _T("this is a single selection tree") );
1097
1098     // to keep going anyhow !!!
1099     if (is_single)
1100       {
1101         if (item->HasHilight()) return; // nothing to do
1102         unselect_others=TRUE;
1103         extended_select=FALSE;
1104       }
1105     else // check if selection will really change
1106       if (unselect_others && item->HasHilight())
1107       {
1108         // selection change if there is more than one item currently selected
1109         wxArrayTreeItemIds selected_items;
1110         if (GetSelections(selected_items)==1) return;
1111       }
1112
1113     wxTreeEvent event( wxEVT_COMMAND_TREE_SEL_CHANGING, GetId() );
1114     event.m_item = item;
1115     event.m_itemOld = m_current;
1116     event.SetEventObject( this );
1117     // TODO : Here we don't send any selection mode yet !
1118
1119     if ( GetEventHandler()->ProcessEvent( event ) && !event.IsAllowed() )
1120       return;
1121
1122     // ctrl press
1123     if (unselect_others)
1124     {
1125         if (is_single) Unselect(); // to speed up thing
1126         else UnselectAll();
1127     }
1128
1129     // shift press
1130     if (extended_select)
1131     {
1132         if (m_current == NULL) m_current=m_key_current=GetRootItem().m_pItem;
1133         // don't change the mark (m_current)
1134         SelectItemRange(m_current, item);
1135     }
1136     else
1137     {
1138         bool select=TRUE; // the default
1139
1140         // Check if we need to toggle hilight (ctrl mode)
1141         if (!unselect_others)
1142           select=!item->HasHilight();
1143
1144         m_current = m_key_current = item;
1145         m_current->SetHilight(select);
1146         RefreshLine( m_current );
1147     }
1148
1149     event.SetEventType(wxEVT_COMMAND_TREE_SEL_CHANGED);
1150     GetEventHandler()->ProcessEvent( event );
1151 }
1152
1153 void wxTreeCtrl::FillArray(wxGenericTreeItem *item,
1154                            wxArrayTreeItemIds &array) const
1155 {
1156     if ( item->HasHilight() )
1157         array.Add(wxTreeItemId(item));
1158
1159     if ( item->HasChildren() )
1160     {
1161         wxArrayGenericTreeItems& children = item->GetChildren();
1162         size_t count = children.GetCount();
1163         for ( size_t n = 0; n < count; ++n )
1164             FillArray(children[n],array);
1165     }
1166 }
1167
1168 size_t wxTreeCtrl::GetSelections(wxArrayTreeItemIds &array) const
1169 {
1170   array.Empty();
1171   FillArray(GetRootItem().m_pItem, array);
1172
1173   return array.Count();
1174 }
1175
1176 void wxTreeCtrl::EnsureVisible(const wxTreeItemId& item)
1177 {
1178     if (!item.IsOk()) return;
1179
1180     wxGenericTreeItem *gitem = item.m_pItem;
1181
1182     // first expand all parent branches
1183     wxGenericTreeItem *parent = gitem->GetParent();
1184     while ( parent )
1185     {
1186         Expand(parent);
1187         parent = parent->GetParent();
1188     }
1189
1190     //if (parent) CalculatePositions();
1191
1192     ScrollTo(item);
1193 }
1194
1195 void wxTreeCtrl::ScrollTo(const wxTreeItemId &item)
1196 {
1197     if (!item.IsOk()) return;
1198
1199     wxGenericTreeItem *gitem = item.m_pItem;
1200
1201     // now scroll to the item
1202     int item_y = gitem->GetY();
1203
1204     int start_x = 0;
1205     int start_y = 0;
1206     ViewStart( &start_x, &start_y );
1207     start_y *= PIXELS_PER_UNIT;
1208
1209     int client_h = 0;
1210     int client_w = 0;
1211     GetClientSize( &client_w, &client_h );
1212
1213     if (item_y < start_y+3)
1214     {
1215         // going down
1216         int x = 0;
1217         int y = 0;
1218         m_anchor->GetSize( x, y, this );
1219         y += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels
1220         int x_pos = GetScrollPos( wxHORIZONTAL );
1221         // Item should appear at top
1222         SetScrollbars( PIXELS_PER_UNIT, PIXELS_PER_UNIT, x/PIXELS_PER_UNIT, y/PIXELS_PER_UNIT, x_pos, item_y/PIXELS_PER_UNIT );
1223     }
1224     else if (item_y+GetLineHeight(gitem) > start_y+client_h)
1225     {
1226        // going up
1227        int x = 0;
1228        int y = 0;
1229        m_anchor->GetSize( x, y, this );
1230        y += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels
1231        item_y += PIXELS_PER_UNIT+2;
1232        int x_pos = GetScrollPos( wxHORIZONTAL );
1233         // Item should appear at bottom
1234        SetScrollbars( PIXELS_PER_UNIT, PIXELS_PER_UNIT, x/PIXELS_PER_UNIT, y/PIXELS_PER_UNIT, x_pos, (item_y+GetLineHeight(gitem)-client_h)/PIXELS_PER_UNIT );
1235     }
1236 }
1237
1238 // FIXME: tree sorting functions are not reentrant and not MT-safe!
1239 static wxTreeCtrl *s_treeBeingSorted = NULL;
1240
1241 static int tree_ctrl_compare_func(wxGenericTreeItem **item1,
1242                                   wxGenericTreeItem **item2)
1243 {
1244     wxCHECK_MSG( s_treeBeingSorted, 0, _T("bug in wxTreeCtrl::SortChildren()") );
1245
1246     return s_treeBeingSorted->OnCompareItems(*item1, *item2);
1247 }
1248
1249 int wxTreeCtrl::OnCompareItems(const wxTreeItemId& item1,
1250                                const wxTreeItemId& item2)
1251 {
1252     return wxStrcmp(GetItemText(item1), GetItemText(item2));
1253 }
1254
1255 void wxTreeCtrl::SortChildren(const wxTreeItemId& itemId)
1256 {
1257     wxCHECK_RET( itemId.IsOk(), _T("invalid tree item") );
1258
1259     wxGenericTreeItem *item = itemId.m_pItem;
1260
1261     wxCHECK_RET( !s_treeBeingSorted,
1262                  _T("wxTreeCtrl::SortChildren is not reentrant") );
1263
1264     wxArrayGenericTreeItems& children = item->GetChildren();
1265     if ( children.Count() > 1 )
1266     {
1267         s_treeBeingSorted = this;
1268         children.Sort(tree_ctrl_compare_func);
1269         s_treeBeingSorted = NULL;
1270
1271         m_dirty = TRUE;
1272     }
1273     //else: don't make the tree dirty as nothing changed
1274 }
1275
1276 wxImageList *wxTreeCtrl::GetImageList() const
1277 {
1278     return m_imageListNormal;
1279 }
1280
1281 wxImageList *wxTreeCtrl::GetStateImageList() const
1282 {
1283     return m_imageListState;
1284 }
1285
1286 void wxTreeCtrl::SetImageList(wxImageList *imageList)
1287 {
1288    m_imageListNormal = imageList;
1289
1290    // Calculate a m_lineHeight value from the image sizes.
1291    // May be toggle off. Then wxTreeCtrl will spread when
1292    // necessary (which might look ugly).
1293 #if 1
1294    wxPaintDC dc(this);
1295    m_lineHeight = (int)(dc.GetCharHeight() + 4);
1296    int
1297       width = 0,
1298       height = 0,
1299       n = m_imageListNormal->GetImageCount();
1300    for(int i = 0; i < n ; i++)
1301    {
1302       m_imageListNormal->GetSize(i, width, height);
1303       if(height > m_lineHeight) m_lineHeight = height;
1304    }
1305
1306    if (m_lineHeight<40) m_lineHeight+=4; // at least 4 pixels (odd such that a line can be drawn in between)
1307    else m_lineHeight+=m_lineHeight/10;   // otherwise 10% extra spacing
1308
1309 #endif
1310 }
1311
1312 void wxTreeCtrl::SetStateImageList(wxImageList *imageList)
1313 {
1314     m_imageListState = imageList;
1315 }
1316
1317 // -----------------------------------------------------------------------------
1318 // helpers
1319 // -----------------------------------------------------------------------------
1320
1321 void wxTreeCtrl::AdjustMyScrollbars()
1322 {
1323     if (m_anchor)
1324     {
1325         int x = 0;
1326         int y = 0;
1327         m_anchor->GetSize( x, y, this );
1328         //y += GetLineHeight(m_anchor);
1329         y += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels
1330         int x_pos = GetScrollPos( wxHORIZONTAL );
1331         int y_pos = GetScrollPos( wxVERTICAL );
1332         SetScrollbars( PIXELS_PER_UNIT, PIXELS_PER_UNIT, x/PIXELS_PER_UNIT, y/PIXELS_PER_UNIT, x_pos, y_pos );
1333     }
1334     else
1335     {
1336         SetScrollbars( 0, 0, 0, 0 );
1337     }
1338 }
1339
1340 int wxTreeCtrl::GetLineHeight(wxGenericTreeItem *item) const
1341 {
1342   if (GetWindowStyleFlag() & wxTR_HAS_VARIABLE_ROW_HEIGHT)
1343     return item->GetHeight();
1344   else
1345     return m_lineHeight;
1346 }
1347
1348 void wxTreeCtrl::PaintItem(wxGenericTreeItem *item, wxDC& dc)
1349 {
1350     // render bold items in bold
1351     wxFont fontOld;
1352     wxFont fontNew;
1353
1354     if (item->IsBold())
1355     {
1356         fontOld = dc.GetFont();
1357         if (fontOld.Ok())
1358         {
1359           // VZ: is there any better way to make a bold variant of old font?
1360           fontNew = wxFont( fontOld.GetPointSize(),
1361                             fontOld.GetFamily(),
1362                             fontOld.GetStyle(),
1363                             wxBOLD,
1364                             fontOld.GetUnderlined());
1365           dc.SetFont(fontNew);
1366         }
1367         else
1368         {
1369             wxFAIL_MSG(_T("wxDC::GetFont() failed!"));
1370         }
1371     }
1372
1373     long text_w = 0;
1374     long text_h = 0;
1375     dc.GetTextExtent( item->GetText(), &text_w, &text_h );
1376
1377     int image_h = 0;
1378     int image_w = 0;
1379     if ((item->IsExpanded()) && (item->GetSelectedImage() != -1))
1380     {
1381         m_imageListNormal->GetSize( item->GetSelectedImage(), image_w, image_h );
1382         image_w += 4;
1383     }
1384     else if (item->GetImage() != -1)
1385     {
1386         m_imageListNormal->GetSize( item->GetImage(), image_w, image_h );
1387         image_w += 4;
1388     }
1389
1390     int total_h = GetLineHeight(item);
1391
1392     dc.DrawRectangle( item->GetX()-2, item->GetY(), item->GetWidth()+2, total_h );
1393
1394     if ((item->IsExpanded()) && (item->GetSelectedImage() != -1))
1395     {
1396         dc.SetClippingRegion( item->GetX(), item->GetY(), image_w-2, total_h );
1397         m_imageListNormal->Draw( item->GetSelectedImage(), dc,
1398                                  item->GetX(),
1399                                  item->GetY() +((total_h > image_h)?((total_h-image_h)/2):0),
1400                                  wxIMAGELIST_DRAW_TRANSPARENT );
1401         dc.DestroyClippingRegion();
1402     }
1403     else if (item->GetImage() != -1)
1404     {
1405         dc.SetClippingRegion( item->GetX(), item->GetY(), image_w-2, total_h );
1406         m_imageListNormal->Draw( item->GetImage(), dc,
1407                                  item->GetX(),
1408                                  item->GetY() +((total_h > image_h)?((total_h-image_h)/2):0),
1409                                  wxIMAGELIST_DRAW_TRANSPARENT );
1410         dc.DestroyClippingRegion();
1411     }
1412
1413     dc.SetBackgroundMode(wxTRANSPARENT);
1414     dc.DrawText( item->GetText(), image_w + item->GetX(), item->GetY()
1415                  + ((total_h > text_h) ? (total_h - text_h)/2 : 0));
1416
1417     // restore normal font for bold items
1418     if (fontOld.Ok())
1419     {
1420         dc.SetFont( fontOld);
1421     }
1422 }
1423
1424 // Now y stands for the top of the item, whereas it used to stand for middle !
1425 void wxTreeCtrl::PaintLevel( wxGenericTreeItem *item, wxDC &dc, int level, int &y )
1426 {
1427     int horizX = level*m_indent;
1428
1429     item->SetX( horizX+m_indent+m_spacing );
1430     item->SetY( y );
1431
1432     int oldY = y;
1433     y+=GetLineHeight(item)/2;
1434
1435     item->SetCross( horizX+m_indent, y );
1436
1437     int exposed_x = dc.LogicalToDeviceX( 0 );
1438     int exposed_y = dc.LogicalToDeviceY( item->GetY() );
1439
1440     if (IsExposed( exposed_x, exposed_y, 10000, GetLineHeight(item) ))  // 10000 = very much
1441     {
1442         int startX = horizX;
1443         int endX = horizX + (m_indent-5);
1444
1445 //        if (!item->HasChildren()) endX += (m_indent+5);
1446         if (!item->HasChildren()) endX += 20;
1447
1448         dc.DrawLine( startX, y, endX, y );
1449
1450         if (item->HasPlus())
1451         {
1452             dc.DrawLine( horizX+(m_indent+5), y, horizX+(m_indent+15), y );
1453             dc.SetPen( *wxGREY_PEN );
1454             dc.SetBrush( *wxWHITE_BRUSH );
1455             dc.DrawRectangle( horizX+(m_indent-5), y-4, 11, 9 );
1456
1457             dc.SetPen( *wxBLACK_PEN );
1458             dc.DrawLine( horizX+(m_indent-2), y, horizX+(m_indent+3), y );
1459             if (!item->IsExpanded())
1460                 dc.DrawLine( horizX+m_indent, y-2, horizX+m_indent, y+3 );
1461
1462             dc.SetPen( m_dottedPen );
1463         }
1464
1465         if (item->HasHilight())
1466         {
1467             dc.SetTextForeground( wxSystemSettings::GetSystemColour( wxSYS_COLOUR_HIGHLIGHTTEXT ) );
1468
1469             dc.SetBrush( *m_hilightBrush );
1470
1471             if (m_hasFocus)
1472                 dc.SetPen( *wxBLACK_PEN );
1473             else
1474                 dc.SetPen( *wxTRANSPARENT_PEN );
1475
1476             PaintItem(item, dc);
1477
1478             dc.SetPen( m_dottedPen );
1479             dc.SetTextForeground( *wxBLACK );
1480             dc.SetBrush( *wxWHITE_BRUSH );
1481         }
1482         else
1483         {
1484             dc.SetBrush( *wxWHITE_BRUSH );
1485             dc.SetPen( *wxTRANSPARENT_PEN );
1486
1487             PaintItem(item, dc);
1488
1489             dc.SetPen( m_dottedPen );
1490         }
1491     }
1492
1493     y = oldY+GetLineHeight(item);
1494
1495     if (item->IsExpanded())
1496     {
1497         oldY+=GetLineHeight(item)/2;
1498         int semiOldY=y; // (=y) for stupid compilator
1499
1500         wxArrayGenericTreeItems& children = item->GetChildren();
1501         size_t n, count = children.Count();
1502         for ( n = 0; n < count; ++n )
1503         {
1504             semiOldY=y;
1505             PaintLevel( children[n], dc, level+1, y );
1506         }
1507
1508         // it may happen that the item is expanded but has no items (when you
1509         // delete all its children for example) - don't draw the vertical line
1510         // in this case
1511         if (count > 0)
1512           {
1513             semiOldY+=GetLineHeight(children[--n])/2;
1514             dc.DrawLine( horizX+m_indent, oldY+5, horizX+m_indent, semiOldY );
1515           }
1516     }
1517 }
1518
1519 void wxTreeCtrl::DrawBorder(wxTreeItemId &item)
1520 {
1521   if (!item) return;
1522
1523     wxGenericTreeItem *i=item.m_pItem;
1524
1525     wxPaintDC dc(this);
1526     PrepareDC( dc );
1527     dc.SetLogicalFunction(wxINVERT);
1528
1529     int w,h,x;
1530     ViewStart(&x,&h);     // we only need x
1531     GetClientSize(&w,&h); // we only need w
1532
1533     h=GetLineHeight(i)+1;
1534     // 2 white column at border
1535     dc.DrawRectangle( PIXELS_PER_UNIT*x+2, i->GetY()-1, w-6, h);
1536 }
1537
1538 void wxTreeCtrl::DrawLine(wxTreeItemId &item, bool below)
1539 {
1540   if (!item) return;
1541
1542     wxGenericTreeItem *i=item.m_pItem;
1543
1544     wxPaintDC dc(this);
1545     PrepareDC( dc );
1546     dc.SetLogicalFunction(wxINVERT);
1547
1548     int w,h,y;
1549     GetSize(&w,&h);
1550
1551     if (below) y=i->GetY()+GetLineHeight(i)-1;
1552     else y=i->GetY();
1553
1554     dc.DrawLine( 0, y, w, y);
1555 }
1556
1557 // -----------------------------------------------------------------------------
1558 // wxWindows callbacks
1559 // -----------------------------------------------------------------------------
1560
1561 void wxTreeCtrl::OnPaint( wxPaintEvent &WXUNUSED(event) )
1562 {
1563     if ( !m_anchor)
1564         return;
1565
1566     wxPaintDC dc(this);
1567     PrepareDC( dc );
1568
1569     dc.SetFont( wxSystemSettings::GetSystemFont( wxSYS_DEFAULT_GUI_FONT ) );
1570
1571     dc.SetPen( m_dottedPen );
1572     //if(GetImageList() == NULL)
1573     // m_lineHeight = (int)(dc.GetCharHeight() + 4);
1574
1575     int y = 2;
1576     PaintLevel( m_anchor, dc, 0, y );
1577 }
1578
1579 void wxTreeCtrl::OnSetFocus( wxFocusEvent &WXUNUSED(event) )
1580 {
1581     m_hasFocus = TRUE;
1582
1583     if (m_current) RefreshLine( m_current );
1584 }
1585
1586 void wxTreeCtrl::OnKillFocus( wxFocusEvent &WXUNUSED(event) )
1587 {
1588     m_hasFocus = FALSE;
1589
1590     if (m_current) RefreshLine( m_current );
1591 }
1592
1593 void wxTreeCtrl::OnChar( wxKeyEvent &event )
1594 {
1595     wxTreeEvent te( wxEVT_COMMAND_TREE_KEY_DOWN, GetId() );
1596     te.m_code = event.KeyCode();
1597     te.SetEventObject( this );
1598     GetEventHandler()->ProcessEvent( te );
1599
1600     if ( (m_current == 0) || (m_key_current == 0) )
1601     {
1602         event.Skip();
1603         return;
1604     }
1605
1606     bool is_multiple=(GetWindowStyleFlag() & wxTR_MULTIPLE);
1607     bool extended_select=(event.ShiftDown() && is_multiple);
1608     bool unselect_others=!(extended_select || (event.ControlDown() && is_multiple));
1609
1610     switch (event.KeyCode())
1611     {
1612         case '+':
1613         case WXK_ADD:
1614             if (m_current->HasPlus() && !IsExpanded(m_current))
1615             {
1616                 Expand(m_current);
1617             }
1618             break;
1619
1620         case '-':
1621         case WXK_SUBTRACT:
1622             if (IsExpanded(m_current))
1623             {
1624                 Collapse(m_current);
1625             }
1626             break;
1627
1628         case '*':
1629         case WXK_MULTIPLY:
1630             Toggle(m_current);
1631             break;
1632
1633         case ' ':
1634         case WXK_RETURN:
1635             {
1636                 wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_ACTIVATED, GetId() );
1637                 event.m_item = m_current;
1638                 event.m_code = 0;
1639                 event.SetEventObject( this );
1640                 GetEventHandler()->ProcessEvent( event );
1641             }
1642             break;
1643
1644             // up goes to the previous sibling or to the last of its children if
1645             // it's expanded
1646         case WXK_UP:
1647             {
1648                 wxTreeItemId prev = GetPrevSibling( m_key_current );
1649                 if (!prev)
1650                 {
1651                     prev = GetParent( m_key_current );
1652                     if (prev)
1653                     {
1654                         long cockie = 0;
1655                         wxTreeItemId current = m_key_current;
1656                         if (current == GetFirstChild( prev, cockie ))
1657                         {
1658                             // otherwise we return to where we came from
1659                             SelectItem( prev, unselect_others, extended_select );
1660                             m_key_current=prev.m_pItem;
1661                             EnsureVisible( prev );
1662                             break;
1663                         }
1664                     }
1665                 }
1666                 if (prev)
1667                 {
1668                     while ( IsExpanded(prev) && HasChildren(prev) )
1669                     {
1670                         wxTreeItemId child = GetLastChild(prev);
1671                         if ( child )
1672                         {
1673                             prev = child;
1674                         }
1675                     }
1676
1677                     SelectItem( prev, unselect_others, extended_select );
1678                     m_key_current=prev.m_pItem;
1679                     EnsureVisible( prev );
1680                 }
1681             }
1682             break;
1683
1684             // left arrow goes to the parent
1685         case WXK_LEFT:
1686             {
1687                 wxTreeItemId prev = GetParent( m_current );
1688                 if (prev)
1689                 {
1690                     EnsureVisible( prev );
1691                     SelectItem( prev, unselect_others, extended_select );
1692                 }
1693             }
1694             break;
1695
1696         case WXK_RIGHT:
1697             // this works the same as the down arrow except that we also expand the
1698             // item if it wasn't expanded yet
1699             Expand(m_current);
1700             // fall through
1701
1702         case WXK_DOWN:
1703             {
1704                 if (IsExpanded(m_key_current) && HasChildren(m_key_current))
1705                 {
1706                     long cookie = 0;
1707                     wxTreeItemId child = GetFirstChild( m_key_current, cookie );
1708                     SelectItem( child, unselect_others, extended_select );
1709                     m_key_current=child.m_pItem;
1710                     EnsureVisible( child );
1711                 }
1712                 else
1713                 {
1714                     wxTreeItemId next = GetNextSibling( m_key_current );
1715 //                    if (next == 0)
1716                     if (!next)
1717                     {
1718                         wxTreeItemId current = m_key_current;
1719                         while (current && !next)
1720                         {
1721                             current = GetParent( current );
1722                             if (current) next = GetNextSibling( current );
1723                         }
1724                     }
1725 //                    if (next != 0)
1726                     if (next)
1727                     {
1728                         SelectItem( next, unselect_others, extended_select );
1729                         m_key_current=next.m_pItem;
1730                         EnsureVisible( next );
1731                     }
1732                 }
1733             }
1734             break;
1735
1736             // <End> selects the last visible tree item
1737         case WXK_END:
1738             {
1739                 wxTreeItemId last = GetRootItem();
1740
1741                 while ( last.IsOk() && IsExpanded(last) )
1742                 {
1743                     wxTreeItemId lastChild = GetLastChild(last);
1744
1745                     // it may happen if the item was expanded but then all of
1746                     // its children have been deleted - so IsExpanded() returned
1747                     // TRUE, but GetLastChild() returned invalid item
1748                     if ( !lastChild )
1749                         break;
1750
1751                     last = lastChild;
1752                 }
1753
1754                 if ( last.IsOk() )
1755                 {
1756                     EnsureVisible( last );
1757                     SelectItem( last, unselect_others, extended_select );
1758                 }
1759             }
1760             break;
1761
1762             // <Home> selects the root item
1763         case WXK_HOME:
1764             {
1765                 wxTreeItemId prev = GetRootItem();
1766                 if (prev)
1767                 {
1768                     EnsureVisible( prev );
1769                     SelectItem( prev, unselect_others, extended_select );
1770                 }
1771             }
1772             break;
1773
1774         default:
1775             event.Skip();
1776     }
1777 }
1778
1779 wxTreeItemId wxTreeCtrl::HitTest(const wxPoint& point, int& flags)
1780 {
1781     wxClientDC dc(this);
1782     PrepareDC(dc);
1783     long x = dc.DeviceToLogicalX( (long)point.x );
1784     long y = dc.DeviceToLogicalY( (long)point.y );
1785     int w, h;
1786     GetSize(&w, &h);
1787
1788     flags=0;
1789     if (point.x<0) flags|=wxTREE_HITTEST_TOLEFT;
1790     if (point.x>w) flags|=wxTREE_HITTEST_TORIGHT;
1791     if (point.y<0) flags|=wxTREE_HITTEST_ABOVE;
1792     if (point.y>h) flags|=wxTREE_HITTEST_BELOW;
1793
1794     return m_anchor->HitTest( wxPoint(x, y), this, flags);
1795 }
1796
1797 /* **** */
1798
1799 void wxTreeCtrl::Edit( const wxTreeItemId& item )
1800 {
1801     if (!item.IsOk()) return;
1802
1803     m_currentEdit = item.m_pItem;
1804
1805     wxTreeEvent te( wxEVT_COMMAND_TREE_BEGIN_LABEL_EDIT, GetId() );
1806     te.m_item = m_currentEdit;
1807     te.SetEventObject( this );
1808     GetEventHandler()->ProcessEvent( te );
1809
1810     if (!te.IsAllowed()) return;
1811
1812     wxString s = m_currentEdit->GetText();
1813     int x = m_currentEdit->GetX();
1814     int y = m_currentEdit->GetY();
1815     int w = m_currentEdit->GetWidth();
1816     int h = m_currentEdit->GetHeight();
1817
1818     int image_h = 0;
1819     int image_w = 0;
1820     if ((m_currentEdit->IsExpanded()) && (m_currentEdit->GetSelectedImage() != -1))
1821     {
1822         m_imageListNormal->GetSize( m_currentEdit->GetSelectedImage(), image_w, image_h );
1823         image_w += 4;
1824     }
1825     else if (m_currentEdit->GetImage() != -1)
1826     {
1827         m_imageListNormal->GetSize( m_currentEdit->GetImage(), image_w, image_h );
1828         image_w += 4;
1829     }
1830     x += image_w;
1831     w -= image_w + 4; // I don't know why +4 is needed
1832
1833     wxClientDC dc(this);
1834     PrepareDC( dc );
1835     x = dc.LogicalToDeviceX( x );
1836     y = dc.LogicalToDeviceY( y );
1837
1838     wxTreeTextCtrl *text = new wxTreeTextCtrl(
1839       this, -1, &m_renameAccept, &m_renameRes, this, s, wxPoint(x-4,y-4), wxSize(w+11,h+8) );
1840     text->SetFocus();
1841 }
1842
1843 void wxTreeCtrl::OnRenameTimer()
1844 {
1845     Edit( m_current );
1846 }
1847
1848 void wxTreeCtrl::OnRenameAccept()
1849 {
1850     wxTreeEvent le( wxEVT_COMMAND_TREE_END_LABEL_EDIT, GetId() );
1851     le.m_item = m_currentEdit;
1852     le.SetEventObject( this );
1853     le.m_label = m_renameRes;
1854     GetEventHandler()->ProcessEvent( le );
1855
1856     if (!le.IsAllowed()) return;
1857
1858     SetItemText( m_currentEdit, m_renameRes );
1859 }
1860
1861 void wxTreeCtrl::OnMouse( wxMouseEvent &event )
1862 {
1863     if (!event.LeftIsDown()) m_dragCount = 0;
1864
1865     if ( !(event.LeftUp() || event.LeftDClick() || event.Dragging()) ) return;
1866
1867     if ( !m_anchor ) return;
1868
1869     wxClientDC dc(this);
1870     PrepareDC(dc);
1871     long x = dc.DeviceToLogicalX( (long)event.GetX() );
1872     long y = dc.DeviceToLogicalY( (long)event.GetY() );
1873
1874     int flags=0;
1875     wxGenericTreeItem *item = m_anchor->HitTest( wxPoint(x,y), this, flags);
1876     bool onButton = flags & wxTREE_HITTEST_ONITEMBUTTON;
1877
1878     if (item == NULL) return;  /* we hit the blank area */
1879
1880     if (event.Dragging())
1881     {
1882         if (m_dragCount == 2)  /* small drag latency (3?) */
1883         {
1884             m_dragCount = 0;
1885
1886             wxTreeEvent nevent(wxEVT_COMMAND_TREE_BEGIN_DRAG, GetId());
1887             nevent.m_item = m_current;
1888             nevent.SetEventObject(this);
1889             GetEventHandler()->ProcessEvent(nevent);
1890         }
1891         else
1892         {
1893             m_dragCount++;
1894         }
1895         return;
1896     }
1897
1898     if (event.LeftUp() && (item == m_current) &&
1899         (flags & wxTREE_HITTEST_ONITEMLABEL) &&
1900         HasFlag(wxTR_EDIT_LABELS) )
1901     {
1902         m_renameTimer->Start( 100, TRUE );
1903         return;
1904     }
1905
1906     bool is_multiple=(GetWindowStyleFlag() & wxTR_MULTIPLE);
1907     bool extended_select=(event.ShiftDown() && is_multiple);
1908     bool unselect_others=!(extended_select || (event.ControlDown() && is_multiple));
1909
1910     if (onButton)
1911     {
1912         Toggle( item );
1913         if (is_multiple)
1914             return;
1915     }
1916
1917     SelectItem(item, unselect_others, extended_select);
1918
1919     if (event.LeftDClick())
1920     {
1921         wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_ACTIVATED, GetId() );
1922         event.m_item = item;
1923         event.m_code = 0;
1924         event.SetEventObject( this );
1925         GetEventHandler()->ProcessEvent( event );
1926     }
1927 }
1928
1929 void wxTreeCtrl::OnIdle( wxIdleEvent &WXUNUSED(event) )
1930 {
1931     /* after all changes have been done to the tree control,
1932      * we actually redraw the tree when everything is over */
1933
1934     if (!m_dirty)
1935         return;
1936
1937     m_dirty = FALSE;
1938
1939     CalculatePositions();
1940     Refresh();
1941     AdjustMyScrollbars();
1942 }
1943
1944 void wxTreeCtrl::CalculateSize( wxGenericTreeItem *item, wxDC &dc )
1945 {
1946     long text_w = 0;
1947     long text_h = 0;
1948
1949     wxFont fontOld;
1950     wxFont fontNew;
1951     if (item->IsBold())
1952     {
1953         fontOld = dc.GetFont();
1954         if (fontOld.Ok())
1955         {
1956           // VZ: is there any better way to make a bold variant of old font?
1957           fontNew = wxFont( fontOld.GetPointSize(),
1958                             fontOld.GetFamily(),
1959                             fontOld.GetStyle(),
1960                             wxBOLD,
1961                             fontOld.GetUnderlined());
1962           dc.SetFont(fontNew);
1963         }
1964         else
1965         {
1966             wxFAIL_MSG(_T("wxDC::GetFont() failed!"));
1967         }
1968     }
1969
1970     dc.GetTextExtent( item->GetText(), &text_w, &text_h );
1971     text_h+=2;
1972
1973     // restore normal font for bold items
1974     if (fontOld.Ok())
1975         dc.SetFont( fontOld);
1976
1977     int image_h = 0;
1978     int image_w = 0;
1979     if ((item->IsExpanded()) && (item->GetSelectedImage() != -1))
1980     {
1981         m_imageListNormal->GetSize( item->GetSelectedImage(), image_w, image_h );
1982         image_w += 4;
1983     }
1984     else if (item->GetImage() != -1)
1985     {
1986         m_imageListNormal->GetSize( item->GetImage(), image_w, image_h );
1987         image_w += 4;
1988     }
1989
1990     int total_h = (image_h > text_h) ? image_h : text_h;
1991
1992     if (total_h<40) total_h+=4; // at least 4 pixels
1993     else total_h+=total_h/10;   // otherwise 10% extra spacing
1994
1995     item->SetHeight(total_h);
1996     if (total_h>m_lineHeight) m_lineHeight=total_h;
1997
1998     item->SetWidth(image_w+text_w+2);
1999 }
2000
2001 // -----------------------------------------------------------------------------
2002 // for developper : y is now the top of the level
2003 // not the middle of it !
2004 void wxTreeCtrl::CalculateLevel( wxGenericTreeItem *item, wxDC &dc, int level, int &y )
2005 {
2006     int horizX = level*m_indent;
2007
2008     CalculateSize( item, dc );
2009
2010     // set its position
2011     item->SetX( horizX+m_indent+m_spacing );
2012     item->SetY( y );
2013     y+=GetLineHeight(item);
2014
2015     if ( !item->IsExpanded() )
2016     {
2017         // we dont need to calculate collapsed branches
2018         return;
2019     }
2020
2021     wxArrayGenericTreeItems& children = item->GetChildren();
2022     size_t n, count = children.Count();
2023     for (n = 0; n < count; ++n )
2024       CalculateLevel( children[n], dc, level+1, y );  // recurse
2025 }
2026
2027 void wxTreeCtrl::CalculatePositions()
2028 {
2029     if ( !m_anchor ) return;
2030
2031     wxClientDC dc(this);
2032     PrepareDC( dc );
2033
2034     dc.SetFont( wxSystemSettings::GetSystemFont( wxSYS_DEFAULT_GUI_FONT ) );
2035
2036     dc.SetPen( m_dottedPen );
2037     //if(GetImageList() == NULL)
2038     // m_lineHeight = (int)(dc.GetCharHeight() + 4);
2039
2040     int y = 2;
2041     CalculateLevel( m_anchor, dc, 0, y ); // start recursion
2042 }
2043
2044 void wxTreeCtrl::RefreshSubtree(wxGenericTreeItem *item)
2045 {
2046     wxClientDC dc(this);
2047     PrepareDC(dc);
2048
2049     int cw = 0;
2050     int ch = 0;
2051     GetClientSize( &cw, &ch );
2052
2053     wxRect rect;
2054     rect.x = dc.LogicalToDeviceX( 0 );
2055     rect.width = cw;
2056     rect.y = dc.LogicalToDeviceY( item->GetY() );
2057     rect.height = ch;
2058
2059     Refresh( TRUE, &rect );
2060
2061     AdjustMyScrollbars();
2062 }
2063
2064 void wxTreeCtrl::RefreshLine( wxGenericTreeItem *item )
2065 {
2066     wxClientDC dc(this);
2067     PrepareDC( dc );
2068
2069     int cw = 0;
2070     int ch = 0;
2071     GetClientSize( &cw, &ch );
2072
2073     wxRect rect;
2074     rect.x = dc.LogicalToDeviceX( 0 );
2075     rect.y = dc.LogicalToDeviceY( item->GetY() );
2076     rect.width = cw;
2077     rect.height = GetLineHeight(item); //dc.GetCharHeight() + 6;
2078
2079     Refresh( TRUE, &rect );
2080 }
2081