src/generic/treectrl.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        treectrl.cpp
   3 // Purpose:
   4 // Author:      Robert Roebling
   5 // Created:     01/02/97
   6 // 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
  18 //-----------------------------------------------------------------------------
  19 // wxTreeItem
  20 //-----------------------------------------------------------------------------
  21
  22 IMPLEMENT_DYNAMIC_CLASS(wxTreeItem, wxObject)
  23
  24 wxTreeItem::wxTreeItem(void)
  25 {
  26   m_mask = 0;
  27   m_itemId = 0;
  28   m_state = 0;
  29   m_stateMask = 0;
  30   m_image = -1;
  31   m_selectedImage = -1;
  32   m_children = 0;
  33   m_data = 0;
  34 };
  35
  36 //-----------------------------------------------------------------------------
  37 // wxTreeEvent
  38 //-----------------------------------------------------------------------------
  39
  40 IMPLEMENT_DYNAMIC_CLASS(wxTreeEvent,wxCommandEvent)
  41
  42 wxTreeEvent::wxTreeEvent( WXTYPE commandType, int id ) :
  43   wxCommandEvent( commandType, id )
  44 {
  45   m_code = 0;
  46   m_oldItem = 0;
  47 };
  48
  49 //-----------------------------------------------------------------------------
  50 // wxGenericTreeItem
  51 //-----------------------------------------------------------------------------
  52
  53 IMPLEMENT_DYNAMIC_CLASS(wxGenericTreeItem,wxObject)
  54
  55 wxGenericTreeItem::wxGenericTreeItem( wxGenericTreeItem *parent )
  56 {
  57   Reset();
  58   m_parent = parent;
  59   m_hasHilight = FALSE;
  60 };
  61
  62 wxGenericTreeItem::wxGenericTreeItem( wxGenericTreeItem *parent, const wxTreeItem& item, wxDC *dc )
  63 {
  64   Reset();
  65   SetItem( item, dc );
  66   m_parent = parent;
  67   m_hasHilight = FALSE;
  68 };
  69
  70 void wxGenericTreeItem::SetItem( const wxTreeItem &item, wxDC *dc )
  71 {
  72   if ((item.m_mask & wxTREE_MASK_HANDLE) == wxTREE_MASK_HANDLE)
  73     m_itemId = item.m_itemId;
  74   if ((item.m_mask & wxTREE_MASK_STATE) == wxTREE_MASK_STATE)
  75     m_state = item.m_state;
  76   if ((item.m_mask & wxTREE_MASK_TEXT) == wxTREE_MASK_TEXT)
  77     m_text = item.m_text;
  78   if ((item.m_mask & wxTREE_MASK_IMAGE) == wxTREE_MASK_IMAGE)
  79     m_image = item.m_image;
  80   if ((item.m_mask & wxTREE_MASK_SELECTED_IMAGE) == wxTREE_MASK_SELECTED_IMAGE)
  81     m_selectedImage = item.m_selectedImage;
  82   if ((item.m_mask & wxTREE_MASK_CHILDREN) == wxTREE_MASK_CHILDREN)
  83     m_hasChildren = (item.m_children > 0);
  84   if ((item.m_mask & wxTREE_MASK_DATA) == wxTREE_MASK_DATA)
  85     m_data = item.m_data;
  86   long lw = 0;
  87   long lh = 0;
  88   dc->GetTextExtent( m_text, &lw, &lh );
  89   m_width = lw;
  90   m_height = lh;
  91 };
  92
  93 void wxGenericTreeItem::SetText( const wxString &text, wxDC *dc )
  94 {
  95   m_text = text;
  96   long lw = 0;
  97   long lh = 0;
  98   dc->GetTextExtent( m_text, &lw, &lh );
  99   m_width = lw;
 100   m_height = lh;
 101 };
 102
 103 void wxGenericTreeItem::Reset(void)
 104 {
 105   m_itemId = -1;
 106   m_state = 0;
 107   m_text = "";
 108   m_image = -1;
 109   m_selectedImage = -1;
 110 //  m_children = 0;
 111   m_hasChildren = FALSE;
 112   m_data = 0;
 113   m_x = 0;
 114   m_y = 0;
 115   m_height = 0;
 116   m_width = 0;
 117   m_xCross = 0;
 118   m_yCross = 0;
 119   m_level = 0;
 120   m_children.DeleteContents( TRUE );
 121   m_parent = NULL;
 122 };
 123
 124 void wxGenericTreeItem::GetItem( wxTreeItem &item ) const
 125 {
 126   if ((item.m_mask & wxTREE_MASK_STATE) == wxTREE_MASK_STATE)
 127     item.m_state = m_state;
 128   if ((item.m_mask & wxTREE_MASK_TEXT) == wxTREE_MASK_TEXT)
 129     item.m_text = m_text;
 130   if ((item.m_mask & wxTREE_MASK_IMAGE) == wxTREE_MASK_IMAGE)
 131     item.m_image = m_image;
 132   if ((item.m_mask & wxTREE_MASK_SELECTED_IMAGE) == wxTREE_MASK_SELECTED_IMAGE)
 133     item.m_selectedImage = m_selectedImage;
 134   if ((item.m_mask & wxTREE_MASK_CHILDREN) == wxTREE_MASK_CHILDREN)
 135     item.m_children = (int)m_hasChildren;
 136   if ((item.m_mask & wxTREE_MASK_DATA) == wxTREE_MASK_DATA)
 137     item.m_data = m_data;
 138 };
 139
 140 bool wxGenericTreeItem::HasChildren(void)
 141 {
 142   return m_hasChildren;
 143 };
 144
 145 bool wxGenericTreeItem::HasPlus(void)
 146 {
 147   return (m_hasChildren && (m_children.Number() == 0));
 148 };
 149
 150 int wxGenericTreeItem::NumberOfVisibleDescendents(void)
 151 {
 152   int ret = m_children.Number();
 153   wxNode *node = m_children.First();
 154   while (node)
 155   {
 156     wxGenericTreeItem *item = (wxGenericTreeItem*)node->Data();
 157     ret += item->NumberOfVisibleDescendents();
 158     node = node->Next();
 159   };
 160   return ret;
 161 };
 162
 163 int wxGenericTreeItem::NumberOfVisibleChildren(void)
 164 {
 165   return m_children.Number();
 166 };
 167
 168 wxGenericTreeItem *wxGenericTreeItem::FindItem( long itemId ) const
 169 {
 170   if (m_itemId == itemId) return (wxGenericTreeItem*)(this);
 171   wxNode *node = m_children.First();
 172   while (node)
 173   {
 174     wxGenericTreeItem *item = (wxGenericTreeItem*)node->Data();
 175     wxGenericTreeItem *res = item->FindItem( itemId );
 176     if (res) return (wxGenericTreeItem*)(res);
 177     node = node->Next();
 178   };
 179   return NULL;
 180 };
 181
 182 void wxGenericTreeItem::AddChild( wxGenericTreeItem *child )
 183 {
 184   m_children.Append( child );
 185 };
 186
 187 void wxGenericTreeItem::SetCross( int x, int y )
 188 {
 189   m_xCross = x;
 190   m_yCross = y;
 191 };
 192
 193 void wxGenericTreeItem::GetSize( int &x, int &y )
 194 {
 195   if (y < m_y + 10) y = m_y +10;
 196   int width = m_x +  m_width;
 197   if (width > x) x = width;
 198   wxNode *node = m_children.First();
 199   while (node)
 200   {
 201     wxGenericTreeItem *item = (wxGenericTreeItem*)node->Data();
 202     item->GetSize( x, y );
 203     node = node->Next();
 204   };
 205 };
 206
 207 long wxGenericTreeItem::HitTest( const wxPoint& point, int &flags )
 208 {
 209   if (m_parent && ((point.y > m_y) && (point.y < m_y+m_height)))
 210   {
 211     if ((point.x > m_xCross-5) &&
 212         (point.x < m_xCross+5) &&
 213         (point.y > m_yCross-5) &&
 214         (point.y < m_yCross+5) &&
 215         (m_hasChildren)
 216        )
 217     {
 218       flags = wxTREE_HITTEST_ONITEMBUTTON;
 219       return m_itemId;
 220     };
 221     if ((point.x > m_x) && (point.x < m_x+m_width))
 222     {
 223       flags = wxTREE_HITTEST_ONITEMLABEL;
 224       return m_itemId;
 225     };
 226     if (point.x > m_x)
 227     {
 228       flags = wxTREE_HITTEST_ONITEMRIGHT;
 229       return m_itemId;
 230     };
 231     flags = wxTREE_HITTEST_ONITEMINDENT;
 232     return m_itemId;
 233   }
 234   else
 235   {
 236     wxNode *node = m_children.First();
 237     while (node)
 238     {
 239       wxGenericTreeItem *child = (wxGenericTreeItem*)node->Data();
 240       long res = child->HitTest( point, flags );
 241       if (res != -1) return res;
 242       node = node->Next();
 243     };
 244   };
 245   return -1;
 246 };
 247
 248 void wxGenericTreeItem::PrepareEvent( wxTreeEvent &event )
 249 {
 250   event.m_item.m_itemId = m_itemId;
 251   event.m_item.m_state = m_state;
 252   event.m_item.m_text = m_text;
 253   event.m_item.m_image = m_image;
 254   event.m_item.m_selectedImage = m_selectedImage;
 255   event.m_item.m_children = (int)m_hasChildren;
 256   event.m_item.m_data = m_data;
 257   event.m_oldItem = 0;
 258   event.m_code = 0;
 259   event.m_pointDrag.x = 0;
 260   event.m_pointDrag.y = 0;
 261 };
 262
 263 void wxGenericTreeItem::SendKeyDown( wxWindow *target )
 264 {
 265   wxTreeEvent event( wxEVT_COMMAND_TREE_KEY_DOWN, target->GetId() );
 266   PrepareEvent( event );
 267   event.SetEventObject( target );
 268   target->ProcessEvent( event );
 269 };
 270
 271 void wxGenericTreeItem::SendSelected( wxWindow *target )
 272 {
 273   wxTreeEvent event( wxEVT_COMMAND_TREE_SEL_CHANGED, target->GetId() );
 274   PrepareEvent( event );
 275   event.SetEventObject( target );
 276   target->ProcessEvent( event );
 277 };
 278
 279 void wxGenericTreeItem::SendDelete( wxWindow *target )
 280 {
 281   wxTreeEvent event( wxEVT_COMMAND_TREE_DELETE_ITEM,
 282  target->GetId() );
 283   PrepareEvent( event );
 284   event.SetEventObject( target );
 285   target->ProcessEvent( event );
 286 };
 287
 288 void wxGenericTreeItem::SendExpand( wxWindow *target )
 289 {
 290   wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_EXPANDED,
 291  target->GetId() );
 292   PrepareEvent( event );
 293   event.SetEventObject( target );
 294   target->ProcessEvent( event );
 295 };
 296
 297 void wxGenericTreeItem::SetHilight( bool set )
 298 {
 299   m_hasHilight = set;
 300 };
 301
 302 bool wxGenericTreeItem::HasHilight(void)
 303 {
 304   return m_hasHilight;
 305 };
 306
 307 //-----------------------------------------------------------------------------
 308 // wxTreeCtrl
 309 //-----------------------------------------------------------------------------
 310
 311 IMPLEMENT_DYNAMIC_CLASS(wxTreeCtrl,wxScrolledWindow
 312 )
 313
 314 BEGIN_EVENT_TABLE(wxTreeCtrl,wxScrolledWindow)
 315   EVT_PAINT          (wxTreeCtrl::OnPaint)
 316   EVT_MOUSE_EVENTS   (wxTreeCtrl::OnMouse)
 317   EVT_CHAR           (wxTreeCtrl::OnChar)
 318   EVT_SET_FOCUS      (wxTreeCtrl::OnSetFocus)
 319   EVT_KILL_FOCUS     (wxTreeCtrl::OnKillFocus)
 320 END_EVENT_TABLE()
 321
 322 wxTreeCtrl::wxTreeCtrl(void)
 323 {
 324   m_current = NULL;
 325   m_anchor = NULL;
 326   m_hasFocus = FALSE;
 327   m_xScroll = 0;
 328   m_yScroll = 0;
 329   m_lastId = 0;
 330   m_lineHeight = 10;
 331   m_indent = 15;
 332   m_isCreated = FALSE;
 333   m_dc = NULL;
 334   m_hilightBrush = new wxBrush( wxSystemSettings::GetSystemColour(wxSYS_COLOUR_HIGHLIGHT), wxSOLID );
 335 };
 336
 337 wxTreeCtrl::wxTreeCtrl(wxWindow *parent, const wxWindowID id,
 338             const wxPoint& pos,
 339  const wxSize& size,
 340             const long style, const wxString& name )
 341 {
 342   m_current = NULL;
 343   m_anchor = NULL;
 344   m_hasFocus = FALSE;
 345   m_xScroll = 0;
 346   m_yScroll = 0;
 347   m_lastId = 0;
 348   m_lineHeight = 10;
 349   m_indent = 15;
 350   m_isCreated = FALSE;
 351   m_dc = NULL;
 352   m_hilightBrush = new wxBrush( wxSystemSettings::GetSystemColour(wxSYS_COLOUR_HIGHLIGHT), wxSOLID );
 353   Create( parent, id, pos, size, style, name );
 354 };
 355
 356 wxTreeCtrl::~wxTreeCtrl(void)
 357 {
 358   if (m_dc) delete m_dc;
 359 };
 360
 361 bool wxTreeCtrl::Create(wxWindow *parent, const wxWindowID id,
 362             const wxPoint& pos,
 363  const wxSize& size,
 364             const long style
 365 , const wxString& name )
 366 {
 367   wxScrolledWindow::Create( parent, id, pos, size, style, name );
 368   SetBackgroundColour( *wxWHITE );
 369   m_dottedPen = wxPen( *wxBLACK, 0, 0 );
 370   return TRUE;
 371 };
 372
 373 int wxTreeCtrl::GetCount(void) const
 374 {
 375   if (!m_anchor) return 0;
 376   return m_anchor->NumberOfVisibleDescendents();
 377 };
 378
 379 long wxTreeCtrl::InsertItem( const long parent, const wxString& label, const int image,
 380       const int selImage, const long WXUNUSED(insertAfter) )
 381 {
 382   wxGenericTreeItem *p = NULL;
 383   if (parent == 0)
 384   {
 385     if (m_anchor) return -1;
 386   }
 387   else
 388   {
 389     p = FindItem( parent );
 390     if (!p) return -1;
 391   };
 392   wxTreeItem item;
 393   m_lastId++;
 394   item.m_mask = wxTREE_MASK_HANDLE;
 395   item.m_itemId = m_lastId;
 396   if (!label.IsNull() || (label == ""))
 397   {
 398     item.m_text = label;
 399     item.m_mask |= wxTREE_MASK_TEXT;
 400   };
 401   if (image >= 0)
 402   {
 403     item.m_image = image;
 404     item.m_mask |= wxTREE_MASK_IMAGE;
 405   };
 406   if (selImage >= 0)
 407   {
 408     item.m_selectedImage = selImage;
 409     item.m_mask |= wxTREE_MASK_SELECTED_IMAGE;
 410   };
 411
 412   wxClientDC dc(this);
 413   wxGenericTreeItem *new_child = new wxGenericTreeItem( p, item, &dc );
 414   if (p)
 415     p->AddChild( new_child );
 416   else
 417     m_anchor = new_child;
 418
 419   if (p)
 420   {
 421     CalculatePositions();
 422
 423     int ch = 0;
 424     GetClientSize( NULL, &ch );
 425
 426     wxRectangle rect;
 427     rect.x = 0; rect.y = 0;
 428     rect.width = 10000; rect.height = ch;
 429
 430     PrepareDC( dc );
 431     if (p->m_children.Number() == 1)
 432     {
 433       rect.y = dc.LogicalToDeviceY( p->m_y );
 434     }
 435     else
 436     {
 437       wxNode *node = p->m_children.Member( new_child )->Previous();
 438       wxGenericTreeItem* last_child = (wxGenericTreeItem*)node->Data();
 439       rect.y = dc.LogicalToDeviceY( last_child->m_y );
 440     };
 441
 442     long doX = 0;
 443     long doY = 0;
 444     dc.GetDeviceOrigin( &doX, &doY );
 445     rect.height = ch-rect.y-doY;
 446
 447     AdjustMyScrollbars();
 448
 449     if (rect.height > 0) Refresh( FALSE, &rect);
 450   }
 451   else
 452   {
 453     AdjustMyScrollbars();
 454
 455     Refresh();
 456   };
 457
 458   return m_lastId;
 459 };
 460
 461 long wxTreeCtrl::InsertItem( const long parent, wxTreeItem &info, const long WXUNUSED(insertAfter) )
 462 {
 463   int oldMask = info.m_mask;
 464   wxGenericTreeItem *p = NULL;
 465   if (parent == 0)
 466   {
 467     if (m_anchor) return -1;
 468   }
 469   else
 470   {
 471     p = FindItem( parent );
 472     if (!p)
 473     {
 474       printf( "TreeItem not found.\n" );
 475       return -1;
 476     };
 477   };
 478   long ret = 0;
 479   if ((info.m_mask & wxTREE_MASK_HANDLE) == 0)
 480   {
 481     m_lastId++;
 482     info.m_itemId = m_lastId;
 483     info.m_mask |= wxTREE_MASK_HANDLE;
 484     ret = m_lastId;
 485   }
 486   else
 487   {
 488     ret = info.m_itemId;
 489   };
 490
 491   wxClientDC dc(this);
 492   wxGenericTreeItem *new_child = new wxGenericTreeItem( p, info, &dc );
 493   if (p)
 494     p->AddChild( new_child );
 495   else
 496     m_anchor = new_child;
 497
 498   if (p)
 499   {
 500     CalculatePositions();
 501
 502     int ch = 0;
 503     GetClientSize( NULL, &ch );
 504
 505     wxRectangle rect;
 506     rect.x = 0; rect.y = 0;
 507     rect.width = 10000; rect.height = ch;
 508
 509     PrepareDC( dc );
 510     if (p->m_children.Number() == 1)
 511     {
 512       rect.y = dc.LogicalToDeviceY( p->m_y );
 513     }
 514     else
 515     {
 516       wxNode *node = p->m_children.Member( new_child )->Previous();
 517       wxGenericTreeItem* last_child = (wxGenericTreeItem*)node->Data();
 518       rect.y = dc.LogicalToDeviceY( last_child->m_y );
 519     };
 520
 521     long doX = 0;
 522     long doY = 0;
 523     dc.GetDeviceOrigin( &doX, &doY );
 524     rect.height = ch-rect.y-doY;
 525
 526     AdjustMyScrollbars();
 527
 528     if (rect.height > 0) Refresh( FALSE, &rect);
 529   }
 530   else
 531   {
 532     AdjustMyScrollbars();
 533
 534     Refresh();
 535   };
 536
 537   info.m_mask = oldMask;
 538   return ret;
 539 };
 540
 541 bool wxTreeCtrl::ExpandItem( const long item, const int action )
 542 {
 543   wxGenericTreeItem *i = FindItem( item );
 544   if (!i) return FALSE;
 545   switch (action)
 546   {
 547     case wxTREE_EXPAND_EXPAND:
 548     {
 549       i->SendExpand( this );
 550       break;
 551     };
 552     case wxTREE_EXPAND_COLLAPSE_RESET:
 553     case wxTREE_EXPAND_COLLAPSE:
 554     {
 555       wxNode *node = i->m_children.First();
 556       while (node)
 557       {
 558         wxGenericTreeItem *child = (wxGenericTreeItem*)node->Data();
 559         child->SendDelete( this );
 560         delete node;
 561         node = i->m_children.First();
 562       };
 563
 564       int cw = 0;
 565       int ch = 0;
 566       GetClientSize( &cw, &ch );
 567       wxRect rect;
 568       rect.x = 0;
 569       rect.width = cw;
 570       wxClientDC dc(this);
 571       PrepareDC(dc);
 572       rect.y = dc.LogicalToDeviceY( i->m_y );
 573
 574       long doY = 0;
 575       dc.GetDeviceOrigin( NULL, &doY );
 576       rect.height = ch-rect.y-doY;
 577       Refresh( TRUE, &rect );
 578
 579       AdjustMyScrollbars();
 580       break;
 581     };
 582     case wxTREE_EXPAND_TOGGLE:
 583     {
 584       if (i->HasPlus())
 585         ExpandItem( item, wxTREE_EXPAND_EXPAND );
 586       else
 587         ExpandItem( item, wxTREE_EXPAND_COLLAPSE );
 588       break;
 589     };
 590   };
 591   return TRUE;
 592 };
 593
 594 bool wxTreeCtrl::DeleteAllItems(void)
 595 {
 596   delete m_anchor;
 597   m_anchor = NULL;
 598   Refresh();
 599   return TRUE;
 600 };
 601
 602 bool wxTreeCtrl::GetItem( wxTreeItem &info ) const
 603 {
 604   wxGenericTreeItem *i = FindItem( info.m_itemId );
 605   if (!i) return FALSE;
 606   i->GetItem( info );
 607   return TRUE;
 608 };
 609
 610 long wxTreeCtrl::GetItemData( const long item ) const
 611 {
 612   wxGenericTreeItem *i = FindItem( item );
 613   if (!i) return 0;
 614   return i->m_data;
 615 };
 616
 617 wxString wxTreeCtrl::GetItemText( const long item ) const
 618 {
 619   wxGenericTreeItem *i = FindItem( item );
 620   if (!i) return "";
 621   return i->m_text;
 622 };
 623
 624 long wxTreeCtrl::GetParent( const long item ) const
 625 {
 626   wxGenericTreeItem *i = FindItem( item );
 627   if (!i) return -1;
 628   i = i->m_parent;
 629   if (!i) return -1;
 630   return i->m_parent->m_itemId;
 631 };
 632
 633 long wxTreeCtrl::GetRootItem(void) const
 634 {
 635   if (m_anchor) return m_anchor->m_itemId;
 636   return -1;
 637 };
 638
 639 long wxTreeCtrl::GetSelection(void) const
 640 {
 641   return 0;
 642 };
 643
 644 bool wxTreeCtrl::SelectItem( const long WXUNUSED(item) ) const
 645 {
 646   return FALSE;
 647 };
 648
 649 bool wxTreeCtrl::ItemHasChildren( const long item ) const
 650 {
 651   wxGenericTreeItem *i = FindItem( item );
 652   if (!i) return FALSE;
 653   return i->m_hasChildren;
 654 };
 655
 656 void wxTreeCtrl::SetIndent( const int indent )
 657 {
 658   m_indent = indent;
 659   Refresh();
 660 };
 661
 662 int wxTreeCtrl::GetIndent(void) const
 663 {
 664   return m_indent;
 665 };
 666
 667 bool wxTreeCtrl::SetItem( wxTreeItem &info )
 668 {
 669   wxGenericTreeItem *i = FindItem( info.m_itemId );
 670   if (!i) return FALSE;
 671   wxClientDC dc(this);
 672   i->SetItem( info, &dc );
 673   return TRUE;
 674 };
 675
 676 bool wxTreeCtrl::SetItemData( const long item, const long data )
 677 {
 678   wxGenericTreeItem *i = FindItem( item );
 679   if (!i) return FALSE;
 680   i->m_data = data;
 681   return TRUE;
 682 };
 683
 684 bool wxTreeCtrl::SetItemText( const long item, const wxString &text )
 685 {
 686   wxGenericTreeItem *i = FindItem( item );
 687   if (!i) return FALSE;
 688   wxClientDC dc(this);
 689   i->SetText( text, &dc );
 690   return TRUE;
 691 };
 692
 693 long wxTreeCtrl::HitTest( const wxPoint& point, int &flags )
 694 {
 695   flags = 0;
 696   if (!m_anchor) return -1;
 697   return m_anchor->HitTest( point, flags );
 698 };
 699
 700 void wxTreeCtrl::AdjustMyScrollbars(void)
 701 {
 702   if (m_anchor)
 703   {
 704     int x = 0;
 705     int y = 0;
 706     m_anchor->GetSize( x, y );
 707     y += 2*m_lineHeight;
 708     int x_pos = GetScrollPos( wxHORIZONTAL );
 709     int y_pos = GetScrollPos( wxVERTICAL );
 710     SetScrollbars( 10, 10, x/10, y/10, x_pos, y_pos );
 711   }
 712   else
 713   {
 714     SetScrollbars( 0, 0, 0, 0 );
 715   };
 716 };
 717
 718 void wxTreeCtrl::PaintLevel( wxGenericTreeItem *item, wxPaintDC &dc, int level, int &y )
 719 {
 720   int horizX = level*m_indent+10;
 721   int oldY = y;
 722   wxNode *node = item->m_children.First();
 723   while (node)
 724   {
 725     wxGenericTreeItem *child = (wxGenericTreeItem *)node->Data();
 726     dc.SetPen( m_dottedPen );
 727
 728     child->SetCross( horizX+15, y );
 729
 730     if (!node->Next())
 731     {
 732       if (level != 0) oldY -= (m_lineHeight-5);
 733       dc.DrawLine( horizX, oldY, horizX, y );
 734     };
 735
 736     child->m_x = horizX+33;
 737     child->m_y = y-m_lineHeight/3;
 738     child->m_height = m_lineHeight;
 739
 740     if (IsExposed( 0, child->m_y-2, 10000, m_lineHeight+4 ))
 741     {
 742       int startX,endX;
 743       if ((node->Previous()) || (level != 0))
 744         startX = horizX; else startX = horizX-10;
 745       if (child->HasChildren())
 746         endX = horizX+10; else endX = horizX+30;
 747       dc.DrawLine( startX, y, endX, y );
 748
 749       if  (child->HasChildren())
 750       {
 751         dc.DrawLine( horizX+20, y, horizX+30, y );
 752         dc.SetPen( *wxGREY_PEN );
 753         dc.DrawRectangle( horizX+10, y-4, 11, 9 );
 754         dc.SetPen( *wxBLACK_PEN );
 755         dc.DrawLine( horizX+13, y, horizX+17, y );
 756         if (child->HasPlus())
 757           dc.DrawLine( horizX+15, y-2, horizX+15, y+2 );
 758       };
 759
 760       if (child->HasHilight())
 761       {
 762         dc.SetTextForeground( wxSystemSettings::GetSystemColour( wxSYS_COLOUR_HIGHLIGHTTEXT ) );
 763         dc.SetBrush( *m_hilightBrush );
 764         if (m_hasFocus)
 765           dc.SetPen( wxBLACK_PEN );
 766         else
 767           dc.SetPen( wxTRANSPARENT_PEN );
 768         long tw = 0;
 769         long th = 0;
 770         dc.GetTextExtent( child->m_text, &tw, &th );
 771         dc.DrawRectangle( child->m_x-2, child->m_y-2, tw+4, th+4 );
 772       };
 773
 774       dc.DrawText( child->m_text, child->m_x, child->m_y );
 775
 776       if (child->HasHilight())
 777       {
 778         dc.SetTextForeground( *wxBLACK );
 779         dc.SetBrush( *wxWHITE_BRUSH );
 780         dc.SetPen( *wxBLACK_PEN );
 781       };
 782
 783     };
 784
 785     y += m_lineHeight;
 786     if (child->NumberOfVisibleChildren() > 0)
 787       PaintLevel( child, dc, level+1, y );
 788     node = node->Next();
 789   };
 790 };
 791
 792 void wxTreeCtrl::OnPaint( const wxPaintEvent &WXUNUSED(event) )
 793 {
 794   if (!m_anchor) return;
 795
 796   if (!m_dc)
 797   {
 798     m_dc = new wxPaintDC(this);
 799     PrepareDC( *m_dc );
 800   };
 801
 802   m_dc->SetFont( wxSystemSettings::GetSystemFont( wxSYS_SYSTEM_FONT ) );
 803
 804   m_dc->SetPen( m_dottedPen );
 805   m_lineHeight = (int)(m_dc->GetCharHeight() + 4);
 806
 807   int y = m_lineHeight / 2 + 2;
 808   PaintLevel( m_anchor, *m_dc, 0, y );
 809 };
 810
 811 void wxTreeCtrl::OnSetFocus( const wxFocusEvent &WXUNUSED(event) )
 812 {
 813   m_hasFocus = TRUE;
 814   if (m_current) RefreshLine( m_current );
 815 };
 816
 817 void wxTreeCtrl::OnKillFocus( const wxFocusEvent &WXUNUSED(event) )
 818 {
 819   m_hasFocus = FALSE;
 820   if (m_current) RefreshLine( m_current );
 821 };
 822
 823 void wxTreeCtrl::OnChar( wxKeyEvent &event )
 824 {
 825   event.Skip();
 826 };
 827
 828 void wxTreeCtrl::OnMouse( const wxMouseEvent &event )
 829 {
 830   if (!event.LeftDown() &&
 831       !event.LeftDClick()) return;
 832
 833   wxClientDC dc(this);
 834   PrepareDC(dc);
 835   long x = dc.DeviceToLogicalX( (long)event.GetX() );
 836   long y = dc.DeviceToLogicalY( (long)event.GetY() );
 837
 838   int flag = 0;
 839   long id = HitTest( wxPoint(x,y), flag );
 840   if (id == -1) return;
 841   wxGenericTreeItem *item = FindItem( id );
 842
 843   if (!item) return;
 844   if ((flag != wxTREE_HITTEST_ONITEMBUTTON) &&
 845       (flag != wxTREE_HITTEST_ONITEMLABEL)) return;
 846
 847   if (m_current != item)
 848   {
 849     if (m_current)
 850     {
 851       m_current->SetHilight( FALSE );
 852       RefreshLine( m_current );
 853     };
 854     m_current = item;
 855     m_current->SetHilight( TRUE );
 856     RefreshLine( m_current );
 857     m_current->SendSelected( this );
 858   };
 859
 860   if (event.LeftDClick()) m_current->SendKeyDown( this );
 861
 862   if (flag == wxTREE_HITTEST_ONITEMBUTTON)
 863   {
 864     ExpandItem( item->m_itemId, wxTREE_EXPAND_TOGGLE );
 865     return;
 866   };
 867 };
 868
 869 void wxTreeCtrl::CalculateLevel( wxGenericTreeItem *item, wxPaintDC &dc, int level, int &y )
 870 {
 871   int horizX = level*m_indent+10;
 872   wxNode *node = item->m_children.First();
 873   while (node)
 874   {
 875     wxGenericTreeItem *child = (wxGenericTreeItem *)node->Data();
 876     dc.SetPen( m_dottedPen );
 877
 878     int startX,endX;
 879     if ((node->Previous()) || (level != 0))
 880       startX = horizX; else startX = horizX-10;
 881     if (child->HasChildren())
 882       endX = horizX+10; else endX = horizX+30;
 883
 884     child->m_x = horizX+33;
 885     child->m_y = y-m_lineHeight/3-2;
 886     child->m_height = m_lineHeight;
 887
 888     y += m_lineHeight;
 889     if (child->NumberOfVisibleChildren() > 0)
 890       CalculateLevel( child, dc, level+1, y );
 891
 892     node = node->Next();
 893   };
 894 };
 895
 896 void wxTreeCtrl::CalculatePositions(void)
 897 {
 898   if (!m_anchor) return;
 899
 900   wxClientDC dc(this);
 901   PrepareDC( dc );
 902
 903   dc.SetFont( wxSystemSettings::GetSystemFont( wxSYS_SYSTEM_FONT ) );
 904
 905   dc.SetPen( m_dottedPen );
 906   m_lineHeight = (int)(dc.GetCharHeight() + 4);
 907
 908   int y = m_lineHeight / 2 + 2;
 909   CalculateLevel( m_anchor, dc, 0, y );
 910 };
 911
 912 wxGenericTreeItem *wxTreeCtrl::FindItem( long itemId ) const
 913 {
 914   if (!m_anchor) return NULL;
 915   return m_anchor->FindItem( itemId );
 916 };
 917
 918 void wxTreeCtrl::RefreshLine( wxGenericTreeItem *item )
 919 {
 920   if (!item) return;
 921   wxClientDC dc(this);
 922   PrepareDC( dc );
 923   wxRect rect;
 924   rect.x = dc.LogicalToDeviceX( item->m_x-2 );
 925   rect.y = dc.LogicalToDeviceY( item->m_y-2 );
 926   rect.width = 1000;
 927   rect.height = dc.GetCharHeight()+4;
 928   Refresh( TRUE, &rect );
 929 };
 930
 931
 932