src/generic/treectrl.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        treectrl.cpp
   3 // Purpose:
   4 // Author:      Robert Roebling
   5 // Created:     01/02/97
   6 // Id:          $Id$
   7 // Copyright:   (c) 1998 Robert Roebling, Julian Smart and Markus Holzem
   8 // Licence:     wxWindows licence
   9 /////////////////////////////////////////////////////////////////////////////
  10
  11 #ifdef __GNUG__
  12 #pragma implementation "treectrl.h"
  13 #endif
  14
  15 #include "wx/treectrl.h"
  16 #include "wx/settings.h"
  17 #include "wx/log.h"
  18 #include <wx/intl.h>
  19
  20 //-----------------------------------------------------------------------------
  21 // wxTreeItem
  22 //-----------------------------------------------------------------------------
  23
  24 IMPLEMENT_DYNAMIC_CLASS(wxTreeItem, wxObject)
  25
  26 wxTreeItem::wxTreeItem()
  27 {
  28   m_mask = 0;
  29   m_itemId = 0;
  30   m_state = 0;
  31   m_stateMask = 0;
  32   m_image = -1;
  33   m_selectedImage = -1;
  34   m_children = 0;
  35   m_data = 0;
  36 }
  37
  38 //-----------------------------------------------------------------------------
  39 // wxTreeEvent
  40 //-----------------------------------------------------------------------------
  41
  42 IMPLEMENT_DYNAMIC_CLASS(wxTreeEvent,wxCommandEvent)
  43
  44 wxTreeEvent::wxTreeEvent( wxEventType commandType, int id ) :
  45   wxCommandEvent( commandType, id )
  46 {
  47   m_code = 0;
  48   m_oldItem = 0;
  49 }
  50
  51 //-----------------------------------------------------------------------------
  52 // wxGenericTreeItem
  53 //-----------------------------------------------------------------------------
  54
  55 IMPLEMENT_DYNAMIC_CLASS(wxGenericTreeItem,wxObject)
  56
  57 wxGenericTreeItem::wxGenericTreeItem( wxGenericTreeItem *parent )
  58 {
  59   Reset();
  60   m_parent = parent;
  61   m_hasHilight = FALSE;
  62 }
  63
  64 wxGenericTreeItem::wxGenericTreeItem( wxGenericTreeItem *parent, const wxTreeItem& item, wxDC *dc )
  65 {
  66   Reset();
  67   SetItem( item, dc );
  68   m_parent = parent;
  69   m_hasHilight = FALSE;
  70 }
  71
  72 void wxGenericTreeItem::SetItem( const wxTreeItem &item, wxDC *dc )
  73 {
  74   if ((item.m_mask & wxTREE_MASK_HANDLE) == wxTREE_MASK_HANDLE)
  75     m_itemId = item.m_itemId;
  76   if ((item.m_mask & wxTREE_MASK_STATE) == wxTREE_MASK_STATE)
  77     m_state = item.m_state;
  78   if ((item.m_mask & wxTREE_MASK_TEXT) == wxTREE_MASK_TEXT)
  79     m_text = item.m_text;
  80   if ((item.m_mask & wxTREE_MASK_IMAGE) == wxTREE_MASK_IMAGE)
  81     m_image = item.m_image;
  82   if ((item.m_mask & wxTREE_MASK_SELECTED_IMAGE) == wxTREE_MASK_SELECTED_IMAGE)
  83     m_selectedImage = item.m_selectedImage;
  84   if ((item.m_mask & wxTREE_MASK_CHILDREN) == wxTREE_MASK_CHILDREN)
  85     m_hasChildren = (item.m_children > 0);
  86   if ((item.m_mask & wxTREE_MASK_DATA) == wxTREE_MASK_DATA)
  87     m_data = item.m_data;
  88   long lw = 0;
  89   long lh = 0;
  90   dc->GetTextExtent( m_text, &lw, &lh );
  91   m_width = lw;
  92   m_height = lh;
  93 }
  94
  95 void wxGenericTreeItem::SetText( const wxString &text, wxDC *dc )
  96 {
  97   m_text = text;
  98   long lw = 0;
  99   long lh = 0;
 100   dc->GetTextExtent( m_text, &lw, &lh );
 101   m_width = lw;
 102   m_height = lh;
 103 }
 104
 105 void wxGenericTreeItem::Reset()
 106 {
 107   m_itemId = -1;
 108   m_state = 0;
 109   m_text = "";
 110   m_image = -1;
 111   m_selectedImage = -1;
 112 //  m_children = 0;
 113   m_hasChildren = FALSE;
 114   m_data = 0;
 115   m_x = 0;
 116   m_y = 0;
 117   m_height = 0;
 118   m_width = 0;
 119   m_xCross = 0;
 120   m_yCross = 0;
 121   m_level = 0;
 122   m_children.DeleteContents( TRUE );
 123   m_isCollapsed = TRUE;
 124   m_parent = (wxGenericTreeItem *) NULL;
 125 }
 126
 127 void wxGenericTreeItem::GetItem( wxTreeItem &item ) const
 128 {
 129   if ((item.m_mask & wxTREE_MASK_STATE) == wxTREE_MASK_STATE)
 130     item.m_state = m_state;
 131   if ((item.m_mask & wxTREE_MASK_TEXT) == wxTREE_MASK_TEXT)
 132     item.m_text = m_text;
 133   if ((item.m_mask & wxTREE_MASK_IMAGE) == wxTREE_MASK_IMAGE)
 134     item.m_image = m_image;
 135   if ((item.m_mask & wxTREE_MASK_SELECTED_IMAGE) == wxTREE_MASK_SELECTED_IMAGE)
 136     item.m_selectedImage = m_selectedImage;
 137   if ((item.m_mask & wxTREE_MASK_CHILDREN) == wxTREE_MASK_CHILDREN)
 138     item.m_children = (int)m_hasChildren;
 139   if ((item.m_mask & wxTREE_MASK_DATA) == wxTREE_MASK_DATA)
 140     item.m_data = m_data;
 141 }
 142
 143 bool wxGenericTreeItem::HasChildren()
 144 {
 145   return m_hasChildren;
 146 }
 147
 148 bool wxGenericTreeItem::HasPlus()
 149 {
 150   if ( !HasChildren() )
 151     return FALSE;
 152
 153   return !IsExpanded();
 154 }
 155
 156 int wxGenericTreeItem::NumberOfVisibleDescendents()
 157 {
 158   int ret = m_children.Number();
 159   wxNode *node = m_children.First();
 160   while (node)
 161   {
 162     wxGenericTreeItem *item = (wxGenericTreeItem*)node->Data();
 163     ret += item->NumberOfVisibleDescendents();
 164     node = node->Next();
 165   }
 166   return ret;
 167 }
 168
 169 int wxGenericTreeItem::NumberOfVisibleChildren()
 170 {
 171   return m_isCollapsed ? 0 : m_children.Number();
 172 }
 173
 174 wxGenericTreeItem *wxGenericTreeItem::FindItem( long itemId ) const
 175 {
 176   if (m_itemId == itemId) return (wxGenericTreeItem*)(this);
 177   wxNode *node = m_children.First();
 178   while (node)
 179   {
 180     wxGenericTreeItem *item = (wxGenericTreeItem*)node->Data();
 181     wxGenericTreeItem *res = item->FindItem( itemId );
 182     if (res) return (wxGenericTreeItem*)(res);
 183     node = node->Next();
 184   }
 185   return (wxGenericTreeItem *) NULL;
 186 }
 187
 188 void wxGenericTreeItem::AddChild( wxGenericTreeItem *child )
 189 {
 190   m_children.Append( child );
 191 }
 192
 193 void wxGenericTreeItem::SetCross( int x, int y )
 194 {
 195   m_xCross = x;
 196   m_yCross = y;
 197 }
 198
 199 void wxGenericTreeItem::GetSize( int &x, int &y )
 200 {
 201   if (y < m_y + 10) y = m_y +10;
 202   int width = m_x +  m_width;
 203   if (width > x) x = width;
 204   wxNode *node = m_children.First();
 205   while (node)
 206   {
 207     wxGenericTreeItem *item = (wxGenericTreeItem*)node->Data();
 208     item->GetSize( x, y );
 209     node = node->Next();
 210   }
 211 }
 212
 213 long wxGenericTreeItem::HitTest( const wxPoint& point, int &flags )
 214 {
 215   if ((point.y > m_y) && (point.y < m_y+m_height))
 216   {
 217     if ((point.x > m_xCross-5) &&
 218         (point.x < m_xCross+5) &&
 219         (point.y > m_yCross-5) &&
 220         (point.y < m_yCross+5) &&
 221         (m_hasChildren))
 222     {
 223       flags = wxTREE_HITTEST_ONITEMBUTTON;
 224       return m_itemId;
 225     }
 226     if ((point.x > m_x) && (point.x < m_x+m_width))
 227     {
 228       flags = wxTREE_HITTEST_ONITEMLABEL;
 229       return m_itemId;
 230     }
 231     if (point.x > m_x)
 232     {
 233       flags = wxTREE_HITTEST_ONITEMRIGHT;
 234       return m_itemId;
 235     }
 236     flags = wxTREE_HITTEST_ONITEMINDENT;
 237     return m_itemId;
 238   }
 239   else
 240   {
 241     if (!m_isCollapsed)
 242     {
 243       wxNode *node = m_children.First();
 244       while (node)
 245       {
 246         wxGenericTreeItem *child = (wxGenericTreeItem*)node->Data();
 247         long res = child->HitTest( point, flags );
 248         if (res != -1) return res;
 249         node = node->Next();
 250       }
 251     }
 252   }
 253   return -1;
 254 }
 255
 256 void wxGenericTreeItem::PrepareEvent( wxTreeEvent &event )
 257 {
 258   event.m_item.m_itemId = m_itemId;
 259   event.m_item.m_state = m_state;
 260   event.m_item.m_text = m_text;
 261   event.m_item.m_image = m_image;
 262   event.m_item.m_selectedImage = m_selectedImage;
 263   event.m_item.m_children = (int)m_hasChildren;
 264   event.m_item.m_data = m_data;
 265   event.m_oldItem = 0;
 266   event.m_code = 0;
 267   event.m_pointDrag.x = 0;
 268   event.m_pointDrag.y = 0;
 269 }
 270
 271 void wxGenericTreeItem::SendKeyDown( wxWindow *target )
 272 {
 273   wxTreeEvent event( wxEVT_COMMAND_TREE_KEY_DOWN, target->GetId() );
 274   PrepareEvent( event );
 275   event.SetEventObject( target );
 276   target->ProcessEvent( event );
 277 }
 278
 279 void wxGenericTreeItem::SendSelected( wxWindow *target )
 280 {
 281   wxTreeEvent event( wxEVT_COMMAND_TREE_SEL_CHANGED, target->GetId() );
 282   PrepareEvent( event );
 283   event.SetEventObject( target );
 284   target->ProcessEvent( event );
 285 }
 286
 287 void wxGenericTreeItem::SendDelete( wxWindow *target )
 288 {
 289   wxTreeEvent event( wxEVT_COMMAND_TREE_DELETE_ITEM, target->GetId() );
 290   PrepareEvent( event );
 291   event.SetEventObject( target );
 292   target->ProcessEvent( event );
 293 }
 294
 295 void wxGenericTreeItem::SendExpand( wxWindow *target )
 296 {
 297   m_isCollapsed = FALSE;
 298
 299   wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_EXPANDING, target->GetId() );
 300   event.SetEventObject( target );
 301   PrepareEvent( event );
 302   target->ProcessEvent( event );
 303
 304   event.SetEventType(wxEVT_COMMAND_TREE_ITEM_EXPANDED);
 305   PrepareEvent( event );
 306   target->ProcessEvent( event );
 307 }
 308
 309 void wxGenericTreeItem::SendCollapse( wxWindow *target )
 310 {
 311   m_isCollapsed = TRUE;
 312
 313   wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_COLLAPSING, target->GetId() );
 314   event.SetEventObject( target );
 315   PrepareEvent( event );
 316   target->ProcessEvent( event );
 317
 318   event.SetEventType(wxEVT_COMMAND_TREE_ITEM_COLLAPSED);
 319   PrepareEvent( event );
 320   target->ProcessEvent( event );
 321 }
 322
 323 void wxGenericTreeItem::SetHilight( bool set )
 324 {
 325   m_hasHilight = set;
 326 }
 327
 328 bool wxGenericTreeItem::HasHilight()
 329 {
 330   return m_hasHilight;
 331 }
 332
 333 //-----------------------------------------------------------------------------
 334 // wxTreeCtrl
 335 //-----------------------------------------------------------------------------
 336
 337 IMPLEMENT_DYNAMIC_CLASS(wxTreeCtrl,wxScrolledWindow)
 338
 339 BEGIN_EVENT_TABLE(wxTreeCtrl,wxScrolledWindow)
 340   EVT_PAINT          (wxTreeCtrl::OnPaint)
 341   EVT_MOUSE_EVENTS   (wxTreeCtrl::OnMouse)
 342   EVT_CHAR           (wxTreeCtrl::OnChar)
 343   EVT_SET_FOCUS      (wxTreeCtrl::OnSetFocus)
 344   EVT_KILL_FOCUS     (wxTreeCtrl::OnKillFocus)
 345 END_EVENT_TABLE()
 346
 347 wxTreeCtrl::wxTreeCtrl()
 348 {
 349   m_current = (wxGenericTreeItem *) NULL;
 350   m_anchor = (wxGenericTreeItem *) NULL;
 351   m_hasFocus = FALSE;
 352   m_xScroll = 0;
 353   m_yScroll = 0;
 354   m_lastId = 0;
 355   m_lineHeight = 10;
 356   m_indent = 15;
 357   m_isCreated = FALSE;
 358   m_hilightBrush = new wxBrush( wxSystemSettings::GetSystemColour(wxSYS_COLOUR_HIGHLIGHT), wxSOLID );
 359   m_imageList = (wxImageList *) NULL;
 360   m_smallImageList = (wxImageList *) NULL;
 361 }
 362
 363 wxTreeCtrl::wxTreeCtrl(wxWindow *parent, wxWindowID id,
 364             const wxPoint& pos, const wxSize& size,
 365             long style, const wxString& name )
 366 {
 367   m_current = (wxGenericTreeItem *) NULL;
 368   m_anchor = (wxGenericTreeItem *) NULL;
 369   m_hasFocus = FALSE;
 370   m_xScroll = 0;
 371   m_yScroll = 0;
 372   m_lastId = 0;
 373   m_lineHeight = 10;
 374   m_indent = 15;
 375   m_isCreated = FALSE;
 376   m_hilightBrush = new wxBrush( wxSystemSettings::GetSystemColour(wxSYS_COLOUR_HIGHLIGHT), wxSOLID );
 377   m_imageList = (wxImageList *) NULL;
 378   m_smallImageList = (wxImageList *) NULL;
 379   Create( parent, id, pos, size, style, name );
 380 }
 381
 382 wxTreeCtrl::~wxTreeCtrl()
 383 {
 384   if (m_hilightBrush) delete m_hilightBrush;
 385   if (m_anchor) delete m_anchor;
 386 }
 387
 388 bool wxTreeCtrl::Create(wxWindow *parent, wxWindowID id,
 389             const wxPoint& pos, const wxSize& size,
 390             long style, const wxString& name )
 391 {
 392   wxScrolledWindow::Create( parent, id, pos, size, style, name );
 393   SetBackgroundColour( *wxWHITE );
 394   m_dottedPen = wxPen( *wxBLACK, 0, 0 );
 395   return TRUE;
 396 }
 397
 398 int wxTreeCtrl::GetCount() const
 399 {
 400   if (!m_anchor) return 0;
 401   return m_anchor->NumberOfVisibleDescendents();
 402 }
 403
 404 long wxTreeCtrl::InsertItem( long parent, const wxString& label, int image,
 405       int selImage, long WXUNUSED(insertAfter) )
 406 {
 407   wxGenericTreeItem *p = (wxGenericTreeItem *) NULL;
 408   if (parent == 0)
 409   {
 410     if (m_anchor) return -1;
 411   }
 412   else
 413   {
 414     p = FindItem( parent );
 415     if (!p) return -1;
 416   }
 417   wxTreeItem item;
 418   m_lastId++;
 419   item.m_mask = wxTREE_MASK_HANDLE;
 420   item.m_itemId = m_lastId;
 421   if (!label.IsNull() || (label == ""))
 422   {
 423     item.m_text = label;
 424     item.m_mask |= wxTREE_MASK_TEXT;
 425   }
 426   if (image >= 0)
 427   {
 428     item.m_image = image;
 429     item.m_mask |= wxTREE_MASK_IMAGE;
 430   }
 431   if (selImage >= 0)
 432   {
 433     item.m_selectedImage = selImage;
 434     item.m_mask |= wxTREE_MASK_SELECTED_IMAGE;
 435   }
 436
 437   wxClientDC dc(this);
 438   wxGenericTreeItem *new_child = new wxGenericTreeItem( p, item, &dc );
 439   if (p)
 440     p->AddChild( new_child );
 441   else
 442     m_anchor = new_child;
 443
 444   if (p)
 445   {
 446     CalculatePositions();
 447
 448     if (!p->HasChildren()) p->m_hasChildren = TRUE;
 449
 450     int ch = 0;
 451     GetClientSize( (int *) NULL, &ch );
 452
 453     PrepareDC( dc );
 454
 455     wxRectangle rect;
 456     rect.x = dc.LogicalToDeviceX( 0 );
 457     rect.y = 0;
 458     rect.width = 10000;
 459     rect.height = ch;
 460
 461     if (p->m_children.Number() == 1)
 462     {
 463       rect.y = dc.LogicalToDeviceY( p->m_y );
 464     }
 465     else
 466     {
 467       wxNode *node = p->m_children.Member( new_child )->Previous();
 468       wxGenericTreeItem* last_child = (wxGenericTreeItem*)node->Data();
 469       rect.y = dc.LogicalToDeviceY( last_child->m_y );
 470     }
 471
 472     AdjustMyScrollbars();
 473
 474     if (rect.height > 0) Refresh( FALSE, &rect);
 475   }
 476   else
 477   {
 478     AdjustMyScrollbars();
 479
 480     Refresh();
 481   }
 482
 483   return m_lastId;
 484 }
 485
 486 long wxTreeCtrl::InsertItem( long parent, wxTreeItem &info, long WXUNUSED(insertAfter) )
 487 {
 488   int oldMask = info.m_mask;
 489   wxGenericTreeItem *p = (wxGenericTreeItem *) NULL;
 490   if (parent == 0)
 491   {
 492     if (m_anchor) return -1;
 493   }
 494   else
 495   {
 496     p = FindItem( parent );
 497     if (!p)
 498     {
 499       printf( "TreeItem not found.\n" );
 500       return -1;
 501     }
 502   }
 503   long ret = 0;
 504   if ((info.m_mask & wxTREE_MASK_HANDLE) == 0)
 505   {
 506     m_lastId++;
 507     info.m_itemId = m_lastId;
 508     info.m_mask |= wxTREE_MASK_HANDLE;
 509     ret = m_lastId;
 510   }
 511   else
 512   {
 513     ret = info.m_itemId;
 514   }
 515
 516   wxClientDC dc(this);
 517   wxGenericTreeItem *new_child = new wxGenericTreeItem( p, info, &dc );
 518   if (p)
 519     p->AddChild( new_child );
 520   else
 521     m_anchor = new_child;
 522
 523   if (p)
 524   {
 525     CalculatePositions();
 526
 527     if (!p->HasChildren()) p->m_hasChildren = TRUE;
 528
 529     int ch = 0;
 530     GetClientSize( (int *) NULL, &ch );
 531
 532     PrepareDC( dc );
 533
 534     wxRectangle rect;
 535     rect.x = dc.LogicalToDeviceX( 0 );
 536     rect.y = 0;
 537     rect.width = 10000;
 538     rect.height = ch;
 539
 540     if (p->m_children.Number() == 1)
 541     {
 542       rect.y = dc.LogicalToDeviceY( p->m_y );
 543     }
 544     else
 545     {
 546       wxNode *node = p->m_children.Member( new_child )->Previous();
 547       wxGenericTreeItem* last_child = (wxGenericTreeItem*)node->Data();
 548       rect.y = dc.LogicalToDeviceY( last_child->m_y );
 549     }
 550
 551     AdjustMyScrollbars();
 552
 553     if (rect.height > 0) Refresh( FALSE, &rect);
 554   }
 555   else
 556   {
 557     AdjustMyScrollbars();
 558
 559     Refresh();
 560   }
 561
 562   info.m_mask = oldMask;
 563   return ret;
 564 }
 565
 566 bool wxTreeCtrl::ExpandItem( long item, int action )
 567 {
 568   wxGenericTreeItem *i = FindItem( item );
 569   if (!i)
 570     return FALSE;
 571
 572   switch (action)
 573   {
 574     case wxTREE_EXPAND_EXPAND:
 575     {
 576       i->SendExpand( this );
 577       break;
 578     }
 579
 580     case wxTREE_EXPAND_COLLAPSE_RESET:
 581     case wxTREE_EXPAND_COLLAPSE:
 582     {
 583       wxNode *node = i->m_children.First();
 584       while (node)
 585       {
 586         wxGenericTreeItem *child = (wxGenericTreeItem*)node->Data();
 587         if ( child->IsExpanded() )
 588           ExpandItem( child->m_itemId, wxTREE_EXPAND_COLLAPSE );
 589         node = node->Next();
 590       }
 591
 592       CalculatePositions();
 593
 594       i->SendCollapse( this );
 595       break;
 596     }
 597
 598     case wxTREE_EXPAND_TOGGLE:
 599     {
 600       if ( i->IsExpanded() )
 601         ExpandItem( item, wxTREE_EXPAND_COLLAPSE );
 602       else
 603         ExpandItem( item, wxTREE_EXPAND_EXPAND );
 604       return TRUE;
 605     }
 606   }
 607
 608   wxClientDC dc(this);
 609   PrepareDC(dc);
 610
 611   int cw = 0;
 612   int ch = 0;
 613   GetClientSize( &cw, &ch );
 614
 615   wxRect rect;
 616   rect.x = dc.LogicalToDeviceX( 0 );
 617   rect.width = cw;
 618   rect.y = dc.LogicalToDeviceY( i->m_y );
 619   rect.height = ch;
 620   Refresh( TRUE, &rect );
 621
 622   AdjustMyScrollbars();
 623
 624   return TRUE;
 625 }
 626
 627 void wxTreeCtrl::DeleteItem( long item )
 628 {
 629   wxGenericTreeItem *pItem = FindItem( item );
 630   wxCHECK_RET( pItem != NULL, _("wxTreeCtrl::DeleteItem: no such pItem.") );
 631
 632   pItem->m_parent->m_children.DeleteObject(pItem);
 633
 634   Refresh();
 635 }
 636
 637 void wxTreeCtrl::DeleteChildren( long item )
 638 {
 639   wxGenericTreeItem *pItem = FindItem( item );
 640   wxCHECK_RET( pItem != NULL, _("wxTreeCtrl::DeleteChildren: no such pItem.") );
 641
 642   pItem->m_children.Clear();
 643
 644   Refresh();
 645 }
 646
 647 bool wxTreeCtrl::DeleteAllItems()
 648 {
 649   delete m_anchor;
 650   m_anchor = (wxGenericTreeItem *) NULL;
 651   Refresh();
 652   return TRUE;
 653 }
 654
 655 bool wxTreeCtrl::GetItem( wxTreeItem &info ) const
 656 {
 657   wxGenericTreeItem *i = FindItem( info.m_itemId );
 658   if (!i) return FALSE;
 659   i->GetItem( info );
 660   return TRUE;
 661 }
 662
 663 long wxTreeCtrl::GetItemData( long item ) const
 664 {
 665   wxGenericTreeItem *i = FindItem( item );
 666   if (!i) return 0;
 667   return i->m_data;
 668 }
 669
 670 wxString wxTreeCtrl::GetItemText( long item ) const
 671 {
 672   wxGenericTreeItem *i = FindItem( item );
 673   if (!i) return "";
 674   return i->m_text;
 675 }
 676
 677 int wxTreeCtrl::GetItemImage(long item) const
 678 {
 679   wxGenericTreeItem *i = FindItem( item );
 680   return i == 0 ? -1 : i->GetImage();
 681 }
 682
 683 long wxTreeCtrl::GetParent( long item ) const
 684 {
 685   wxGenericTreeItem *i = FindItem( item );
 686   if (!i) return -1;
 687   i = i->m_parent;
 688   if (!i) return -1;
 689   return i->m_parent->m_itemId;
 690 }
 691
 692 long wxTreeCtrl::GetRootItem() const
 693 {
 694   if (m_anchor) return m_anchor->m_itemId;
 695   return -1;
 696 }
 697
 698 long wxTreeCtrl::GetChild( long item ) const
 699 {
 700   wxGenericTreeItem *i = FindItem( item );
 701   if (!i) return -1;
 702   if (!i->HasChildren()) return -1;
 703   wxNode *node = i->m_children.First();
 704   i = (wxGenericTreeItem *)node->Data();
 705   return i->m_itemId;
 706 }
 707
 708 long wxTreeCtrl::GetNextItem( long item, int code ) const
 709 {
 710   switch (code)
 711   {
 712     case wxTREE_NEXT_CARET:  return GetSelection();
 713     case wxTREE_NEXT_CHILD:  return GetChild( item );
 714     case wxTREE_NEXT_ROOT:   return GetRootItem();
 715     case wxTREE_NEXT_PARENT: return GetParent( item );
 716     case wxTREE_NEXT_NEXT:
 717     {
 718       wxGenericTreeItem *i = FindItem( item );
 719       if (!i) return -1;
 720       if (i->m_parent)
 721       {
 722         wxNode *node = i->m_parent->m_children.Member( i );
 723         if (!node) return -1;
 724         node = node->Next();
 725         if (!node) return -1;
 726         i = (wxGenericTreeItem *)node->Data();
 727         return i->m_itemId;
 728       }
 729       break;
 730     }
 731     case wxTREE_NEXT_PREVIOUS:
 732     {
 733       wxGenericTreeItem *i = FindItem( item );
 734       if (!i) return -1;
 735       if (i->m_parent)
 736       {
 737         wxNode *node = i->m_parent->m_children.Member( i );
 738         if (!node) return -1;
 739         node = node->Previous();
 740         if (!node) return -1;
 741         i = (wxGenericTreeItem *)node->Data();
 742         return i->m_itemId;
 743       }
 744       break;
 745     }
 746   }
 747   return -1;
 748 }
 749
 750 long wxTreeCtrl::GetSelection() const
 751 {
 752   return m_current ? m_current->GetItemId() : -1;
 753 }
 754
 755 bool wxTreeCtrl::SelectItem( long itemId )
 756 {
 757   wxGenericTreeItem *pItem = FindItem(itemId);
 758   if ( !pItem ) {
 759     wxLogDebug(_("Can't select an item %d which doesn't exist."), itemId);
 760
 761     return FALSE;
 762   }
 763
 764   SelectItem(pItem);
 765
 766   return TRUE;
 767 }
 768
 769 void wxTreeCtrl::SelectItem(wxGenericTreeItem *item)
 770 {
 771   if (m_current != item)
 772   {
 773     if (m_current)
 774     {
 775       m_current->SetHilight( FALSE );
 776       RefreshLine( m_current );
 777     }
 778     m_current = item;
 779     m_current->SetHilight( TRUE );
 780     RefreshLine( m_current );
 781
 782     m_current->SendSelected( this );
 783   }
 784 }
 785
 786 bool wxTreeCtrl::ItemHasChildren( long item ) const
 787 {
 788   wxGenericTreeItem *i = FindItem( item );
 789   if (!i) return FALSE;
 790   return i->m_hasChildren;
 791 }
 792
 793 void wxTreeCtrl::SetIndent( int indent )
 794 {
 795   m_indent = indent;
 796   Refresh();
 797 }
 798
 799 int wxTreeCtrl::GetIndent() const
 800 {
 801   return m_indent;
 802 }
 803
 804 bool wxTreeCtrl::SetItem( wxTreeItem &info )
 805 {
 806   wxGenericTreeItem *i = FindItem( info.m_itemId );
 807   if (!i) return FALSE;
 808   wxClientDC dc(this);
 809   i->SetItem( info, &dc );
 810   Refresh();
 811   return TRUE;
 812 }
 813
 814 bool wxTreeCtrl::SetItemData( long item, long data )
 815 {
 816   wxGenericTreeItem *i = FindItem( item );
 817   if (!i) return FALSE;
 818   i->m_data = data;
 819   return TRUE;
 820 }
 821
 822 bool wxTreeCtrl::SetItemText( long item, const wxString &text )
 823 {
 824   wxGenericTreeItem *i = FindItem( item );
 825   if (!i) return FALSE;
 826   wxClientDC dc(this);
 827   i->SetText( text, &dc );
 828   return TRUE;
 829 }
 830
 831 void wxTreeCtrl::SetItemImage(long item, int image, int imageSel) const
 832 {
 833   wxGenericTreeItem *i = FindItem( item );
 834   if ( i != 0 )
 835   {
 836     i->SetImage(image);
 837     i->SetSelectedImage(imageSel);
 838   }
 839 }
 840
 841 int wxTreeCtrl::GetItemState( long item, long stateMask ) const
 842 {
 843   wxGenericTreeItem *i = FindItem( item );
 844   if (!i) return 0;
 845   int ret = 0;
 846
 847   // Currently, an item is always selected _and_ focussed in wxGTK
 848
 849   if (stateMask & wxTREE_STATE_FOCUSED)
 850     if (m_current == i)
 851       ret += wxTREE_STATE_FOCUSED;
 852
 853   if (stateMask & wxTREE_STATE_SELECTED)
 854     if (m_current == i)
 855       ret += wxTREE_STATE_SELECTED;
 856
 857   if (stateMask & wxTREE_STATE_EXPANDED)
 858     if (i->IsExpanded())
 859       ret += wxTREE_STATE_EXPANDED;
 860
 861   return ret;
 862 }
 863
 864 bool wxTreeCtrl::SetItemState( long item, long state, long stateMask )
 865 {
 866   wxGenericTreeItem *i = FindItem( item );
 867   if (!i) return FALSE;
 868
 869   if (stateMask & wxTREE_STATE_FOCUSED)
 870   {
 871     if (state & wxTREE_STATE_FOCUSED)
 872       SelectItem( i );
 873   }
 874
 875   if (stateMask & wxTREE_STATE_SELECTED)
 876   {
 877     if (state & wxTREE_STATE_SELECTED)
 878       SelectItem( i );
 879   }
 880
 881   if (stateMask & wxTREE_STATE_EXPANDED)
 882   {
 883     if (state & wxTREE_STATE_EXPANDED)
 884       ExpandItem( i->m_itemId, wxTREE_EXPAND_EXPAND );
 885     else
 886       ExpandItem( i->m_itemId, wxTREE_EXPAND_COLLAPSE );
 887   }
 888
 889   return TRUE;
 890 }
 891
 892 long wxTreeCtrl::HitTest( const wxPoint& point, int &flags )
 893 {
 894   flags = 0;
 895   if (!m_anchor) return -1;
 896   return m_anchor->HitTest( point, flags );
 897 }
 898
 899 wxImageList *wxTreeCtrl::GetImageList( int which ) const
 900 {
 901   if (which == wxIMAGE_LIST_NORMAL) return m_imageList;
 902   return m_smallImageList;
 903 }
 904
 905 void wxTreeCtrl::SetImageList( wxImageList *imageList, int which )
 906 {
 907   if (which == wxIMAGE_LIST_NORMAL)
 908   {
 909     if (m_imageList) delete m_imageList;
 910     m_imageList = imageList;
 911   }
 912   else
 913   {
 914     if (m_smallImageList) delete m_smallImageList;
 915     m_smallImageList = imageList;
 916   }
 917 }
 918
 919 void wxTreeCtrl::AdjustMyScrollbars()
 920 {
 921   if (m_anchor)
 922   {
 923     int x = 0;
 924     int y = 0;
 925     m_anchor->GetSize( x, y );
 926     y += 2*m_lineHeight;
 927     int x_pos = GetScrollPos( wxHORIZONTAL );
 928     int y_pos = GetScrollPos( wxVERTICAL );
 929     SetScrollbars( 10, 10, x/10, y/10, x_pos, y_pos );
 930   }
 931   else
 932   {
 933     SetScrollbars( 0, 0, 0, 0 );
 934   }
 935 }
 936
 937 void wxTreeCtrl::PaintLevel( wxGenericTreeItem *item, wxPaintDC &dc, int level, int &y )
 938 {
 939   int horizX = level*m_indent;
 940
 941   item->m_x = horizX+33;
 942   item->m_y = y-m_lineHeight/3;
 943   item->m_height = m_lineHeight;
 944
 945   item->SetCross( horizX+15, y );
 946
 947   int oldY = y;
 948
 949   int exposed_x = dc.LogicalToDeviceX( 0 );
 950   int exposed_y = dc.LogicalToDeviceY( item->m_y-2 );
 951
 952   if (IsExposed( exposed_x, exposed_y, 1000, m_lineHeight+4 ))
 953   {
 954     int startX = horizX;
 955     int endX = horizX + 10;
 956
 957     if (!item->HasChildren()) endX += 20;
 958
 959     dc.DrawLine( startX, y, endX, y );
 960
 961       if (item->HasChildren())
 962       {
 963         dc.DrawLine( horizX+20, y, horizX+30, y );
 964         dc.SetPen( *wxGREY_PEN );
 965         dc.DrawRectangle( horizX+10, y-4, 11, 9 );
 966         dc.SetPen( *wxBLACK_PEN );
 967         dc.DrawLine( horizX+13, y, horizX+18, y );
 968         if (item->HasPlus())
 969           dc.DrawLine( horizX+15, y-2, horizX+15, y+3 );
 970       }
 971
 972       if (item->HasHilight())
 973       {
 974         dc.SetTextForeground( wxSystemSettings::GetSystemColour( wxSYS_COLOUR_HIGHLIGHTTEXT ) );
 975         dc.SetBrush( *m_hilightBrush );
 976         long tw, th;
 977         dc.GetTextExtent( item->m_text, &tw, &th );
 978         if (m_hasFocus)
 979         {
 980           dc.SetPen( *wxBLACK_PEN );
 981           dc.DrawRectangle( item->m_x-2, item->m_y-2, tw+4, th+4 );
 982         }
 983         else
 984         {
 985           dc.SetPen( *wxTRANSPARENT_PEN );
 986           dc.DrawRectangle( item->m_x-2, item->m_y-2, tw+4, th+4 );
 987         }
 988         dc.DrawText( item->m_text, item->m_x, item->m_y );
 989
 990         dc.SetPen( *wxBLACK_PEN );
 991         dc.SetTextForeground( *wxBLACK );
 992         dc.SetBrush( *wxWHITE_BRUSH );
 993       }
 994       else
 995       {
 996         dc.SetBrush( *wxWHITE_BRUSH );
 997         dc.SetPen( *wxTRANSPARENT_PEN );
 998         long tw, th;
 999         dc.GetTextExtent( item->m_text, &tw, &th );
1000         dc.DrawRectangle( item->m_x-2, item->m_y-2, tw+4, th+4 );
1001         dc.DrawText( item->m_text, item->m_x, item->m_y );
1002         dc.SetPen( *wxBLACK_PEN );
1003       }
1004   }
1005
1006   if (item->NumberOfVisibleChildren() == 0) return;
1007
1008   int semiOldY = y;
1009
1010   wxNode *node = item->m_children.First();
1011   while (node)
1012   {
1013     wxGenericTreeItem *child = (wxGenericTreeItem *)node->Data();
1014
1015     y += m_lineHeight;
1016     semiOldY = y;
1017
1018     PaintLevel( child, dc, level+1, y );
1019
1020     node = node->Next();
1021   }
1022
1023   dc.DrawLine( horizX+15, oldY+5, horizX+15, semiOldY );
1024 }
1025
1026 void wxTreeCtrl::OnPaint( const wxPaintEvent &WXUNUSED(event) )
1027 {
1028   if (!m_anchor) return;
1029
1030   wxPaintDC dc(this);
1031   PrepareDC( dc );
1032
1033   dc.SetFont( wxSystemSettings::GetSystemFont( wxSYS_SYSTEM_FONT ) );
1034
1035   dc.SetPen( m_dottedPen );
1036   m_lineHeight = (int)(dc.GetCharHeight() + 4);
1037
1038   int y = m_lineHeight / 2 + 2;
1039   PaintLevel( m_anchor, dc, 0, y );
1040 }
1041
1042 void wxTreeCtrl::OnSetFocus( const wxFocusEvent &WXUNUSED(event) )
1043 {
1044   m_hasFocus = TRUE;
1045   if (m_current) RefreshLine( m_current );
1046 }
1047
1048 void wxTreeCtrl::OnKillFocus( const wxFocusEvent &WXUNUSED(event) )
1049 {
1050   m_hasFocus = FALSE;
1051   if (m_current) RefreshLine( m_current );
1052 }
1053
1054 void wxTreeCtrl::OnChar( wxKeyEvent &event )
1055 {
1056   event.Skip();
1057 }
1058
1059 void wxTreeCtrl::OnMouse( const wxMouseEvent &event )
1060 {
1061   if (!event.LeftDown() &&
1062       !event.LeftDClick()) return;
1063
1064   wxClientDC dc(this);
1065   PrepareDC(dc);
1066   long x = dc.DeviceToLogicalX( (long)event.GetX() );
1067   long y = dc.DeviceToLogicalY( (long)event.GetY() );
1068
1069   int flag = 0;
1070   long id = HitTest( wxPoint(x,y), flag );
1071   if (id == -1)
1072     return;
1073   wxGenericTreeItem *item = FindItem( id );
1074
1075   if (!item) return;
1076   if ((flag != wxTREE_HITTEST_ONITEMBUTTON) &&
1077       (flag != wxTREE_HITTEST_ONITEMLABEL)) return;
1078
1079   SelectItem(item);
1080
1081   if (event.LeftDClick())
1082     m_current->SendKeyDown( this );
1083
1084   if (flag == wxTREE_HITTEST_ONITEMBUTTON)
1085   {
1086     ExpandItem( item->m_itemId, wxTREE_EXPAND_TOGGLE );
1087     return;
1088   }
1089 }
1090
1091 void wxTreeCtrl::CalculateLevel( wxGenericTreeItem *item, wxPaintDC &dc, int level, int &y )
1092 {
1093   int horizX = level*m_indent;
1094
1095   item->m_x = horizX+33;
1096   item->m_y = y-m_lineHeight/3-2;
1097   item->m_height = m_lineHeight;
1098
1099   if (item->NumberOfVisibleChildren() == 0) return;
1100
1101   wxNode *node = item->m_children.First();
1102   while (node)
1103   {
1104     wxGenericTreeItem *child = (wxGenericTreeItem *)node->Data();
1105
1106     y += m_lineHeight;
1107     CalculateLevel( child, dc, level+1, y );
1108
1109     node = node->Next();
1110   }
1111 }
1112
1113 void wxTreeCtrl::CalculatePositions()
1114 {
1115   if (!m_anchor)
1116     return;
1117
1118   wxClientDC dc(this);
1119   PrepareDC( dc );
1120
1121   dc.SetFont( wxSystemSettings::GetSystemFont( wxSYS_SYSTEM_FONT ) );
1122
1123   dc.SetPen( m_dottedPen );
1124   m_lineHeight = (int)(dc.GetCharHeight() + 4);
1125
1126   int y = m_lineHeight / 2 + 2;
1127   CalculateLevel( m_anchor, dc, 0, y );
1128 }
1129
1130 wxGenericTreeItem *wxTreeCtrl::FindItem( long itemId ) const
1131 {
1132   if (!m_anchor) return (wxGenericTreeItem *) NULL;
1133   return m_anchor->FindItem( itemId );
1134 }
1135
1136 void wxTreeCtrl::RefreshLine( wxGenericTreeItem *item )
1137 {
1138   if (!item) return;
1139   wxClientDC dc(this);
1140   PrepareDC( dc );
1141   wxRect rect;
1142   rect.x = dc.LogicalToDeviceX( item->m_x-2 );
1143   rect.y = dc.LogicalToDeviceY( item->m_y-2 );
1144   rect.width = 1000;
1145   rect.height = dc.GetCharHeight()+6;
1146   Refresh( TRUE, &rect );
1147 }
1148
1149
1150