src/common/datavcmn.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/common/datavcmn.cpp
   3 // Purpose:     wxDataViewCtrl base classes and common parts
   4 // Author:      Robert Roebling
   5 // Created:     2006/02/20
   6 // RCS-ID:      $Id$
   7 // Copyright:   (c) 2006, Robert Roebling
   8 // Licence:     wxWindows licence
   9 /////////////////////////////////////////////////////////////////////////////
  10
  11 // For compilers that support precompilation, includes "wx.h".
  12 #include "wx/wxprec.h"
  13
  14 #ifdef __BORLANDC__
  15     #pragma hdrstop
  16 #endif
  17
  18 #if wxUSE_DATAVIEWCTRL
  19
  20 #include "wx/dataview.h"
  21
  22 #include "wx/spinctrl.h"
  23
  24 #ifndef WX_PRECOMP
  25     #include "wx/dc.h"
  26     #include "wx/settings.h"
  27     #include "wx/log.h"
  28     #include "wx/icon.h"
  29     #include "wx/crt.h"
  30 #endif
  31
  32 const wxChar wxDataViewCtrlNameStr[] = wxT("dataviewCtrl");
  33
  34
  35 bool operator == (const wxDataViewItem &left, const wxDataViewItem &right)
  36 {
  37     return (left.GetID() == right.GetID() );
  38 }
  39
  40 #ifdef __WXDEBUG__
  41 void wxDataViewItem::Print(const wxString& text) const
  42 {
  43     wxPrintf("item %s: %l\n", text, (long)m_id);
  44 }
  45 #endif
  46
  47 // ---------------------------------------------------------
  48 // wxDataViewModelNotifier
  49 // ---------------------------------------------------------
  50
  51 #include "wx/listimpl.cpp"
  52 WX_DEFINE_LIST(wxDataViewModelNotifiers)
  53
  54 bool wxDataViewModelNotifier::ItemsAdded( const wxDataViewItem &parent, const wxDataViewItemArray &items )
  55 {
  56     size_t count = items.GetCount();
  57     size_t i;
  58     for (i = 0; i < count; i++)
  59         if (!ItemAdded( parent, items[i] )) return false;
  60
  61     return true;
  62 }
  63
  64 bool wxDataViewModelNotifier::ItemsDeleted( const wxDataViewItem &parent, const wxDataViewItemArray &items )
  65 {
  66     size_t count = items.GetCount();
  67     size_t i;
  68     for (i = 0; i < count; i++)
  69         if (!ItemDeleted( parent, items[i] )) return false;
  70
  71     return true;
  72 }
  73
  74 bool wxDataViewModelNotifier::ItemsChanged( const wxDataViewItemArray &items )
  75 {
  76     size_t count = items.GetCount();
  77     size_t i;
  78     for (i = 0; i < count; i++)
  79         if (!ItemChanged( items[i] )) return false;
  80
  81     return true;
  82 }
  83
  84 // ---------------------------------------------------------
  85 // wxDataViewModel
  86 // ---------------------------------------------------------
  87
  88 wxDataViewModel::wxDataViewModel()
  89 {
  90     m_notifiers.DeleteContents( true );
  91 }
  92
  93 bool wxDataViewModel::ItemAdded( const wxDataViewItem &parent, const wxDataViewItem &item )
  94 {
  95     bool ret = true;
  96
  97     wxDataViewModelNotifiers::iterator iter;
  98     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
  99     {
 100         wxDataViewModelNotifier* notifier = *iter;
 101         if (!notifier->ItemAdded( parent, item ))
 102             ret = false;
 103     }
 104
 105     return ret;
 106 }
 107
 108 bool wxDataViewModel::ItemDeleted( const wxDataViewItem &parent, const wxDataViewItem &item )
 109 {
 110     bool ret = true;
 111
 112     wxDataViewModelNotifiers::iterator iter;
 113     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 114     {
 115         wxDataViewModelNotifier* notifier = *iter;
 116         if (!notifier->ItemDeleted( parent, item ))
 117             ret = false;
 118     }
 119
 120     return ret;
 121 }
 122
 123 bool wxDataViewModel::ItemChanged( const wxDataViewItem &item )
 124 {
 125     bool ret = true;
 126
 127     wxDataViewModelNotifiers::iterator iter;
 128     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 129     {
 130         wxDataViewModelNotifier* notifier = *iter;
 131         if (!notifier->ItemChanged( item ))
 132             ret = false;
 133     }
 134
 135     return ret;
 136 }
 137
 138 bool wxDataViewModel::ItemsAdded( const wxDataViewItem &parent, const wxDataViewItemArray &items )
 139 {
 140     bool ret = true;
 141
 142     wxDataViewModelNotifiers::iterator iter;
 143     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 144     {
 145         wxDataViewModelNotifier* notifier = *iter;
 146         if (!notifier->ItemsAdded( parent, items ))
 147             ret = false;
 148     }
 149
 150     return ret;
 151 }
 152
 153 bool wxDataViewModel::ItemsDeleted( const wxDataViewItem &parent, const wxDataViewItemArray &items )
 154 {
 155     bool ret = true;
 156
 157     wxDataViewModelNotifiers::iterator iter;
 158     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 159     {
 160         wxDataViewModelNotifier* notifier = *iter;
 161         if (!notifier->ItemsDeleted( parent, items ))
 162             ret = false;
 163     }
 164
 165     return ret;
 166 }
 167
 168 bool wxDataViewModel::ItemsChanged( const wxDataViewItemArray &items )
 169 {
 170     bool ret = true;
 171
 172     wxDataViewModelNotifiers::iterator iter;
 173     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 174     {
 175         wxDataViewModelNotifier* notifier = *iter;
 176         if (!notifier->ItemsChanged( items ))
 177             ret = false;
 178     }
 179
 180     return ret;
 181 }
 182
 183 bool wxDataViewModel::ValueChanged( const wxDataViewItem &item, unsigned int col )
 184 {
 185     bool ret = true;
 186
 187     wxDataViewModelNotifiers::iterator iter;
 188     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 189     {
 190         wxDataViewModelNotifier* notifier = *iter;
 191         if (!notifier->ValueChanged( item, col ))
 192             ret = false;
 193     }
 194
 195     return ret;
 196 }
 197
 198 bool wxDataViewModel::Cleared()
 199 {
 200     bool ret = true;
 201
 202     wxDataViewModelNotifiers::iterator iter;
 203     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 204     {
 205         wxDataViewModelNotifier* notifier = *iter;
 206         if (!notifier->Cleared())
 207             ret = false;
 208     }
 209
 210     return ret;
 211 }
 212
 213 void wxDataViewModel::Resort()
 214 {
 215     wxDataViewModelNotifiers::iterator iter;
 216     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 217     {
 218         wxDataViewModelNotifier* notifier = *iter;
 219         notifier->Resort();
 220     }
 221 }
 222
 223 void wxDataViewModel::AddNotifier( wxDataViewModelNotifier *notifier )
 224 {
 225     m_notifiers.push_back( notifier );
 226     notifier->SetOwner( this );
 227 }
 228
 229 void wxDataViewModel::RemoveNotifier( wxDataViewModelNotifier *notifier )
 230 {
 231     m_notifiers.DeleteObject( notifier );
 232 }
 233
 234 int wxDataViewModel::Compare( const wxDataViewItem &item1, const wxDataViewItem &item2,
 235                               unsigned int column, bool ascending )
 236 {
 237     // sort branches before leaves
 238     bool item1_is_container = IsContainer(item1);
 239     bool item2_is_container = IsContainer(item2);
 240
 241     if (item1_is_container && !item2_is_container)
 242         return 1;
 243     if (item2_is_container && !item1_is_container)
 244         return -1;
 245
 246     wxVariant value1,value2;
 247     GetValue( value1, item1, column );
 248     GetValue( value2, item2, column );
 249
 250     if (!ascending)
 251     {
 252         wxVariant temp = value1;
 253         value1 = value2;
 254         value2 = temp;
 255     }
 256
 257     if (value1.GetType() == wxT("string"))
 258     {
 259         wxString str1 = value1.GetString();
 260         wxString str2 = value2.GetString();
 261         int res = str1.Cmp( str2 );
 262         if (res) return res;
 263     } else
 264     if (value1.GetType() == wxT("long"))
 265     {
 266         long l1 = value1.GetLong();
 267         long l2 = value2.GetLong();
 268         long res = l1-l2;
 269         if (res) return res;
 270     } else
 271     if (value1.GetType() == wxT("double"))
 272     {
 273         double d1 = value1.GetDouble();
 274         double d2 = value2.GetDouble();
 275         if (d1 < d2) return 1;
 276         if (d1 > d2) return -1;
 277     } else
 278     if (value1.GetType() == wxT("datetime"))
 279     {
 280         wxDateTime dt1 = value1.GetDateTime();
 281         wxDateTime dt2 = value2.GetDateTime();
 282         if (dt1.IsEarlierThan(dt2)) return 1;
 283         if (dt2.IsEarlierThan(dt1)) return -11;
 284     }
 285
 286     // items must be different
 287     unsigned long litem1 = (unsigned long) item1.GetID();
 288     unsigned long litem2 = (unsigned long) item2.GetID();
 289
 290     if (!ascending)
 291         return litem2-litem2;
 292
 293     return litem1-litem2;
 294 }
 295
 296 // ---------------------------------------------------------
 297 // wxDataViewIndexListModel
 298 // ---------------------------------------------------------
 299
 300 wxDataViewIndexListModel::wxDataViewIndexListModel( unsigned int initial_size )
 301 {
 302 #ifndef __WXMAC__
 303     m_useHash = false;
 304 #else
 305     m_useHash = true;
 306 #endif
 307
 308     if (m_useHash)
 309     {
 310         // IDs are ordered until an item gets deleted or inserted
 311         m_ordered = true;
 312
 313         // build initial index
 314         unsigned int i;
 315         for (i = 1; i < initial_size+1; i++)
 316             m_hash.Add( (void*) i );
 317         m_lastIndex = initial_size + 1;
 318     }
 319     else
 320     {
 321         m_lastIndex = initial_size-1;
 322     }
 323 }
 324
 325 wxDataViewIndexListModel::~wxDataViewIndexListModel()
 326 {
 327 }
 328
 329 void wxDataViewIndexListModel::Reset( unsigned int new_size )
 330 {
 331     if (m_useHash)
 332     {
 333         m_hash.Clear();
 334
 335         // IDs are ordered until an item gets deleted or inserted
 336         m_ordered = true;
 337
 338         // build initial index
 339         unsigned int i;
 340         for (i = 1; i < new_size+1; i++)
 341             m_hash.Add( (void*) i );
 342         m_lastIndex = new_size + 1;
 343     }
 344     else
 345     {
 346         m_lastIndex = new_size-1;
 347     }
 348
 349     wxDataViewModel::Cleared();
 350 }
 351
 352 void wxDataViewIndexListModel::RowPrepended()
 353 {
 354     if (m_useHash)
 355     {
 356         m_ordered = false;
 357
 358         unsigned int id = m_lastIndex++;
 359         m_hash.Insert( (void*) id, 0 );
 360         wxDataViewItem item( (void*) id );
 361         ItemAdded( wxDataViewItem(0), item );
 362     }
 363     else
 364     {
 365         m_lastIndex++;
 366         wxDataViewItem item( (void*) 0 );
 367         ItemAdded( wxDataViewItem(0), item );
 368     }
 369 }
 370
 371 void wxDataViewIndexListModel::RowInserted( unsigned int before )
 372 {
 373     if (m_useHash)
 374     {
 375         m_ordered = false;
 376
 377         unsigned int id = m_lastIndex++;
 378         m_hash.Insert( (void*) id, before );
 379         wxDataViewItem item( (void*) id );
 380         ItemAdded( wxDataViewItem(0), item );
 381     }
 382     else
 383     {
 384         m_lastIndex++;
 385         wxDataViewItem item( (void*) before );
 386         ItemAdded( wxDataViewItem(0), item );
 387     }
 388 }
 389
 390 void wxDataViewIndexListModel::RowAppended()
 391 {
 392     if (m_useHash)
 393     {
 394         unsigned int id = m_lastIndex++;
 395         m_hash.Add( (void*) id );
 396         wxDataViewItem item( (void*) id );
 397         ItemAdded( wxDataViewItem(0), item );
 398     }
 399     else
 400     {
 401         m_lastIndex++;
 402         wxDataViewItem item( (void*) m_lastIndex );
 403         ItemAdded( wxDataViewItem(0), item );
 404     }
 405 }
 406
 407 void wxDataViewIndexListModel::RowDeleted( unsigned int row )
 408 {
 409     if (m_useHash)
 410     {
 411         m_ordered = false;
 412
 413         wxDataViewItem item( m_hash[row] );
 414         wxDataViewModel::ItemDeleted( wxDataViewItem(0), item );
 415         m_hash.RemoveAt( row );
 416     }
 417     else
 418     {
 419         wxDataViewItem item( (void*) row );
 420         wxDataViewModel::ItemDeleted( wxDataViewItem(0), item );
 421         m_lastIndex++;
 422     }
 423 }
 424
 425 static int my_sort( int *v1, int *v2 )
 426 {
 427    return *v2-*v1;
 428 }
 429
 430 void wxDataViewIndexListModel::RowsDeleted( const wxArrayInt &rows )
 431 {
 432     wxArrayInt sorted = rows;
 433     sorted.Sort( my_sort );
 434
 435     if (m_useHash)
 436     {
 437         m_ordered = false;
 438
 439         wxDataViewItemArray array;
 440         unsigned int i;
 441         for (i = 0; i < rows.GetCount(); i++)
 442         {
 443             wxDataViewItem item( m_hash[rows[i]] );
 444             array.Add( item );
 445         }
 446         wxDataViewModel::ItemsDeleted( wxDataViewItem(0), array );
 447
 448         for (i = 0; i < sorted.GetCount(); i++)
 449            m_hash.RemoveAt( sorted[i] );
 450     }
 451     else
 452     {
 453         wxDataViewItemArray array;
 454         unsigned int i;
 455         for (i = 0; i < sorted.GetCount(); i++)
 456         {
 457             wxDataViewItem item( (void*) sorted[i] );
 458             array.Add( item );
 459         }
 460         wxDataViewModel::ItemsDeleted( wxDataViewItem(0), array );
 461
 462         m_lastIndex -= rows.GetCount();
 463     }
 464 }
 465
 466 void wxDataViewIndexListModel::RowChanged( unsigned int row )
 467 {
 468     wxDataViewModel::ItemChanged( GetItem(row) );
 469 }
 470
 471 void wxDataViewIndexListModel::RowValueChanged( unsigned int row, unsigned int col )
 472 {
 473     wxDataViewModel::ValueChanged( GetItem(row), col );
 474 }
 475
 476 unsigned int wxDataViewIndexListModel::GetRow( const wxDataViewItem &item ) const
 477 {
 478     if (m_useHash)
 479     {
 480         if (m_ordered)
 481         {
 482             unsigned int pos = wxPtrToUInt( item.GetID() );
 483             return pos-1;
 484         }
 485
 486         // assert for not found
 487         return (unsigned int) m_hash.Index( item.GetID() );
 488     }
 489     else
 490     {
 491         return wxPtrToUInt( item.GetID() );
 492     }
 493 }
 494
 495 wxDataViewItem wxDataViewIndexListModel::GetItem( unsigned int row ) const
 496 {
 497     if (m_useHash)
 498     {
 499         wxASSERT( row < m_hash.GetCount() );
 500         return wxDataViewItem( m_hash[row] );
 501     }
 502     else
 503     {
 504         return wxDataViewItem( (void*) row  );
 505     }
 506 }
 507
 508 bool wxDataViewIndexListModel::HasDefaultCompare() const
 509 {
 510     return !m_ordered;
 511 }
 512
 513 int wxDataViewIndexListModel::Compare(const wxDataViewItem& item1,
 514                                       const wxDataViewItem& item2,
 515                                       unsigned int WXUNUSED(column),
 516                                       bool ascending)
 517 {
 518     if (m_ordered || !m_useHash)
 519     {
 520         unsigned int pos1 = wxPtrToUInt(item1.GetID());
 521         unsigned int pos2 = wxPtrToUInt(item2.GetID());
 522
 523         if (ascending)
 524             return pos1 - pos2;
 525         else
 526             return pos2 - pos1;
 527     }
 528
 529     if (ascending)
 530         return GetRow(item1) - GetRow(item2);
 531
 532     return GetRow(item2) - GetRow(item1);
 533 }
 534
 535 void wxDataViewIndexListModel::GetValue( wxVariant &variant,
 536                            const wxDataViewItem &item, unsigned int col ) const
 537 {
 538     GetValue( variant, GetRow(item), col );
 539 }
 540
 541 bool wxDataViewIndexListModel::SetValue( const wxVariant &variant,
 542                            const wxDataViewItem &item, unsigned int col )
 543 {
 544     return SetValue( variant, GetRow(item), col );
 545 }
 546
 547 bool wxDataViewIndexListModel::GetAttr( const wxDataViewItem &item, unsigned int col, wxDataViewItemAttr &attr )
 548 {
 549     return GetAttr( GetRow(item), col, attr );
 550 }
 551
 552 wxDataViewItem wxDataViewIndexListModel::GetParent( const wxDataViewItem & WXUNUSED(item) ) const
 553 {
 554     return wxDataViewItem(0);
 555 }
 556
 557 bool wxDataViewIndexListModel::IsContainer( const wxDataViewItem &item ) const
 558 {
 559     // only the invisible root item has children
 560     if (!item.IsOk())
 561         return true;
 562
 563     return false;
 564 }
 565
 566 unsigned int wxDataViewIndexListModel::GetChildren( const wxDataViewItem &item, wxDataViewItemArray &children ) const
 567 {
 568     if (!m_useHash)
 569         return 0;  // error
 570
 571     if (item.IsOk())
 572         return 0;
 573
 574     children = m_hash;
 575
 576     return m_hash.GetCount();
 577 }
 578
 579 //-----------------------------------------------------------------------------
 580 // wxDataViewIconText
 581 //-----------------------------------------------------------------------------
 582
 583 IMPLEMENT_DYNAMIC_CLASS(wxDataViewIconText,wxObject)
 584
 585 IMPLEMENT_VARIANT_OBJECT_EXPORTED(wxDataViewIconText, WXDLLIMPEXP_ADV)
 586
 587 bool operator == (const wxDataViewIconText &one, const wxDataViewIconText &two)
 588 {
 589     if (one.GetText() != two.GetText()) return false;
 590     if (one.IsSameAs(two)) return false;
 591     return true;
 592 }
 593
 594 // ---------------------------------------------------------
 595 // wxDataViewRendererBase
 596 // ---------------------------------------------------------
 597
 598 IMPLEMENT_ABSTRACT_CLASS(wxDataViewRendererBase, wxObject)
 599
 600 wxDataViewRendererBase::wxDataViewRendererBase( const wxString &varianttype,
 601                                                 wxDataViewCellMode WXUNUSED(mode),
 602                                                 int WXUNUSED(align) )
 603 {
 604     m_variantType = varianttype;
 605     m_editorCtrl = NULL;
 606 }
 607
 608 const wxDataViewCtrl* wxDataViewRendererBase::GetView() const
 609 {
 610     return wx_const_cast(wxDataViewRendererBase*, this)->GetOwner()->GetOwner();
 611 }
 612
 613 bool wxDataViewRendererBase::StartEditing( const wxDataViewItem &item, wxRect labelRect )
 614 {
 615     m_item = item; // remember for later
 616
 617     unsigned int col = GetOwner()->GetModelColumn();
 618     wxVariant value;
 619     GetOwner()->GetOwner()->GetModel()->GetValue( value, item, col );
 620
 621     m_editorCtrl = CreateEditorCtrl( GetOwner()->GetOwner()->GetMainWindow(), labelRect, value );
 622
 623     wxDataViewEditorCtrlEvtHandler *handler =
 624         new wxDataViewEditorCtrlEvtHandler( m_editorCtrl, (wxDataViewRenderer*) this );
 625
 626     m_editorCtrl->PushEventHandler( handler );
 627
 628 #if defined(__WXGTK20__) && !defined(wxUSE_GENERICDATAVIEWCTRL)
 629     handler->SetFocusOnIdle();
 630 #else
 631     m_editorCtrl->SetFocus();
 632 #endif
 633
 634     // Now we should send Editing Started event
 635     wxDataViewEvent event( wxEVT_COMMAND_DATAVIEW_ITEM_EDITING_STARTED, GetOwner()->GetOwner()->GetId() );
 636     event.SetDataViewColumn( GetOwner() );
 637     event.SetModel( GetOwner()->GetOwner()->GetModel() );
 638     event.SetItem( item );
 639     GetOwner()->GetOwner()->GetEventHandler()->ProcessEvent( event );
 640
 641     return true;
 642 }
 643
 644 void wxDataViewRendererBase::CancelEditing()
 645 {
 646     wxPendingDelete.Append( m_editorCtrl );
 647
 648     GetOwner()->GetOwner()->GetMainWindow()->SetFocus();
 649
 650     // m_editorCtrl->PopEventHandler( true );
 651 }
 652
 653 bool wxDataViewRendererBase::FinishEditing()
 654 {
 655     wxVariant value;
 656     GetValueFromEditorCtrl( m_editorCtrl, value );
 657
 658     wxPendingDelete.Append( m_editorCtrl );
 659
 660     GetOwner()->GetOwner()->GetMainWindow()->SetFocus();
 661
 662     if (!Validate(value))
 663         return false;
 664
 665     unsigned int col = GetOwner()->GetModelColumn();
 666     GetOwner()->GetOwner()->GetModel()->SetValue( value, m_item, col );
 667     GetOwner()->GetOwner()->GetModel()->ValueChanged( m_item, col );
 668
 669     // m_editorCtrl->PopEventHandler( true );
 670
 671     // Now we should send Editing Done event
 672     wxDataViewEvent event( wxEVT_COMMAND_DATAVIEW_ITEM_EDITING_DONE, GetOwner()->GetOwner()->GetId() );
 673     event.SetDataViewColumn( GetOwner() );
 674     event.SetModel( GetOwner()->GetOwner()->GetModel() );
 675     event.SetItem( m_item );
 676     GetOwner()->GetOwner()->GetEventHandler()->ProcessEvent( event );
 677
 678     return true;
 679 }
 680
 681 //-----------------------------------------------------------------------------
 682 // wxDataViewEditorCtrlEvtHandler
 683 //-----------------------------------------------------------------------------
 684
 685 BEGIN_EVENT_TABLE(wxDataViewEditorCtrlEvtHandler, wxEvtHandler)
 686     EVT_CHAR           (wxDataViewEditorCtrlEvtHandler::OnChar)
 687     EVT_KILL_FOCUS     (wxDataViewEditorCtrlEvtHandler::OnKillFocus)
 688     EVT_IDLE           (wxDataViewEditorCtrlEvtHandler::OnIdle)
 689 END_EVENT_TABLE()
 690
 691 wxDataViewEditorCtrlEvtHandler::wxDataViewEditorCtrlEvtHandler(
 692                                 wxControl *editorCtrl,
 693                                 wxDataViewRenderer *owner )
 694 {
 695     m_owner = owner;
 696     m_editorCtrl = editorCtrl;
 697
 698     m_finished = false;
 699 }
 700
 701 void wxDataViewEditorCtrlEvtHandler::OnIdle( wxIdleEvent &event )
 702 {
 703     if (m_focusOnIdle)
 704     {
 705         m_focusOnIdle = false;
 706         if (wxWindow::FindFocus() != m_editorCtrl)
 707             m_editorCtrl->SetFocus();
 708     }
 709
 710     event.Skip();
 711 }
 712
 713 void wxDataViewEditorCtrlEvtHandler::OnChar( wxKeyEvent &event )
 714 {
 715     switch ( event.m_keyCode )
 716     {
 717         case WXK_RETURN:
 718             m_finished = true;
 719             m_owner->FinishEditing();
 720             break;
 721
 722         case WXK_ESCAPE:
 723             m_finished = true;
 724             m_owner->CancelEditing();
 725             break;
 726
 727         default:
 728             event.Skip();
 729     }
 730 }
 731
 732 void wxDataViewEditorCtrlEvtHandler::OnKillFocus( wxFocusEvent &event )
 733 {
 734     if (!m_finished)
 735     {
 736         m_finished = true;
 737         m_owner->FinishEditing();
 738     }
 739
 740     event.Skip();
 741 }
 742
 743 // ---------------------------------------------------------
 744 // wxDataViewColumnBase
 745 // ---------------------------------------------------------
 746
 747 IMPLEMENT_ABSTRACT_CLASS(wxDataViewColumnBase, wxObject)
 748
 749 wxDataViewColumnBase::wxDataViewColumnBase(const wxString& WXUNUSED(title),
 750                                            wxDataViewRenderer *renderer,
 751                                            unsigned int model_column,
 752                                            int WXUNUSED(width),
 753                                            wxAlignment WXUNUSED(align),
 754                                            int WXUNUSED(flags))
 755 {
 756     m_renderer = renderer;
 757     m_model_column = model_column;
 758     m_owner = NULL;
 759     m_renderer->SetOwner( (wxDataViewColumn*) this );
 760
 761     // NOTE: the wxDataViewColumn's ctor must store the width, align, flags
 762     //       parameters inside the native control!
 763 }
 764
 765 wxDataViewColumnBase::wxDataViewColumnBase(const wxBitmap& bitmap,
 766                                            wxDataViewRenderer *renderer,
 767                                            unsigned int model_column,
 768                                            int WXUNUSED(width),
 769                                            wxAlignment WXUNUSED(align),
 770                                            int WXUNUSED(flags) )
 771 {
 772     m_renderer = renderer;
 773     m_model_column = model_column;
 774     m_bitmap = bitmap;
 775     m_owner = NULL;
 776     m_renderer->SetOwner( (wxDataViewColumn*) this );
 777 }
 778
 779 wxDataViewColumnBase::~wxDataViewColumnBase()
 780 {
 781     if (m_renderer)
 782         delete m_renderer;
 783 }
 784
 785 int wxDataViewColumnBase::GetFlags() const
 786 {
 787     int ret = 0;
 788
 789     if (IsSortable())
 790         ret |= wxDATAVIEW_COL_SORTABLE;
 791     if (IsResizeable())
 792         ret |= wxDATAVIEW_COL_RESIZABLE;
 793     if (IsHidden())
 794         ret |= wxDATAVIEW_COL_HIDDEN;
 795
 796     return ret;
 797 }
 798
 799 void wxDataViewColumnBase::SetFlags(int flags)
 800 {
 801     SetSortable((flags & wxDATAVIEW_COL_SORTABLE) != 0);
 802     SetResizeable((flags & wxDATAVIEW_COL_RESIZABLE) != 0);
 803     SetHidden((flags & wxDATAVIEW_COL_HIDDEN) != 0);
 804 }
 805
 806 // ---------------------------------------------------------
 807 // wxDataViewCtrlBase
 808 // ---------------------------------------------------------
 809
 810 IMPLEMENT_ABSTRACT_CLASS(wxDataViewCtrlBase, wxControl)
 811
 812 wxDataViewCtrlBase::wxDataViewCtrlBase()
 813 {
 814     m_model = NULL;
 815     m_expander_column = 0;
 816     m_indent = 8;
 817 }
 818
 819 wxDataViewCtrlBase::~wxDataViewCtrlBase()
 820 {
 821     if (m_model)
 822     {
 823         m_model->DecRef();
 824         m_model = NULL;
 825     }
 826 }
 827
 828 bool wxDataViewCtrlBase::AssociateModel( wxDataViewModel *model )
 829 {
 830     if (m_model)
 831     {
 832         m_model->DecRef();   // discard old model, if any
 833     }
 834
 835     // add our own reference to the new model:
 836     m_model = model;
 837     if (m_model)
 838     {
 839         m_model->IncRef();
 840     }
 841
 842     return true;
 843 }
 844
 845 wxDataViewModel* wxDataViewCtrlBase::GetModel()
 846 {
 847     return m_model;
 848 }
 849
 850 const wxDataViewModel* wxDataViewCtrlBase::GetModel() const
 851 {
 852     return m_model;
 853 }
 854
 855 wxDataViewColumn *
 856 wxDataViewCtrlBase::AppendTextColumn( const wxString &label, unsigned int model_column,
 857                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
 858 {
 859     wxDataViewColumn *ret = new wxDataViewColumn( label,
 860         new wxDataViewTextRenderer( wxT("string"), mode, (int)align ),
 861         model_column, width, align, flags );
 862     AppendColumn( ret );
 863     return ret;
 864 }
 865
 866 wxDataViewColumn *
 867 wxDataViewCtrlBase::AppendIconTextColumn( const wxString &label, unsigned int model_column,
 868                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
 869 {
 870     wxDataViewColumn *ret = new wxDataViewColumn( label,
 871         new wxDataViewIconTextRenderer( wxT("wxDataViewIconText"), mode, (int)align ),
 872         model_column, width, align, flags );
 873     AppendColumn( ret );
 874     return ret;
 875 }
 876
 877 wxDataViewColumn *
 878 wxDataViewCtrlBase::AppendToggleColumn( const wxString &label, unsigned int model_column,
 879                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
 880 {
 881
 882     wxDataViewColumn *ret = new wxDataViewColumn( label,
 883         new wxDataViewToggleRenderer( wxT("bool"), mode, (int)align ),
 884         model_column, width, align, flags );
 885     AppendColumn( ret );
 886     return ret;
 887 }
 888
 889 wxDataViewColumn *
 890 wxDataViewCtrlBase::AppendProgressColumn( const wxString &label, unsigned int model_column,
 891                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
 892 {
 893     wxDataViewColumn *ret = new wxDataViewColumn( label,
 894         new wxDataViewProgressRenderer( wxEmptyString, wxT("long"), mode, (int)align ),
 895         model_column, width, align, flags );
 896     AppendColumn( ret );
 897     return ret;
 898 }
 899
 900 wxDataViewColumn *
 901 wxDataViewCtrlBase::AppendDateColumn( const wxString &label, unsigned int model_column,
 902                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
 903 {
 904     wxDataViewColumn *ret = new wxDataViewColumn( label,
 905         new wxDataViewDateRenderer( wxT("datetime"), mode, (int)align ),
 906         model_column, width, align, flags );
 907     AppendColumn( ret );
 908     return ret;
 909 }
 910
 911 wxDataViewColumn *
 912 wxDataViewCtrlBase::AppendBitmapColumn( const wxString &label, unsigned int model_column,
 913                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
 914 {
 915     wxDataViewColumn *ret = new wxDataViewColumn( label,
 916         new wxDataViewBitmapRenderer( wxT("wxBitmap"), mode, (int)align ),
 917         model_column, width, align, flags );
 918     AppendColumn( ret );
 919     return ret;
 920 }
 921
 922 wxDataViewColumn *
 923 wxDataViewCtrlBase::AppendTextColumn( const wxBitmap &label, unsigned int model_column,
 924                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
 925 {
 926     wxDataViewColumn *ret = new wxDataViewColumn( label,
 927         new wxDataViewTextRenderer( wxT("string"), mode, (int)align ),
 928         model_column, width, align, flags );
 929     AppendColumn( ret );
 930     return ret;
 931 }
 932
 933 wxDataViewColumn *
 934 wxDataViewCtrlBase::AppendIconTextColumn( const wxBitmap &label, unsigned int model_column,
 935                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
 936 {
 937     wxDataViewColumn *ret = new wxDataViewColumn( label,
 938         new wxDataViewIconTextRenderer( wxT("wxDataViewIconText"), mode, (int)align ),
 939         model_column, width, align, flags );
 940     AppendColumn( ret );
 941     return ret;
 942 }
 943
 944 wxDataViewColumn *
 945 wxDataViewCtrlBase::AppendToggleColumn( const wxBitmap &label, unsigned int model_column,
 946                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
 947 {
 948     wxDataViewColumn *ret = new wxDataViewColumn( label,
 949         new wxDataViewToggleRenderer( wxT("bool"), mode, (int)align ),
 950         model_column, width, align, flags );
 951     AppendColumn( ret );
 952     return ret;
 953 }
 954
 955 wxDataViewColumn *
 956 wxDataViewCtrlBase::AppendProgressColumn( const wxBitmap &label, unsigned int model_column,
 957                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
 958 {
 959     wxDataViewColumn *ret = new wxDataViewColumn( label,
 960         new wxDataViewProgressRenderer( wxEmptyString, wxT("long"), mode, (int)align ),
 961         model_column, width, align, flags );
 962     AppendColumn( ret );
 963     return ret;
 964 }
 965
 966 wxDataViewColumn *
 967 wxDataViewCtrlBase::AppendDateColumn( const wxBitmap &label, unsigned int model_column,
 968                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
 969 {
 970     wxDataViewColumn *ret = new wxDataViewColumn( label,
 971         new wxDataViewDateRenderer( wxT("datetime"), mode, (int)align ),
 972         model_column, width, align, flags );
 973     AppendColumn( ret );
 974     return ret;
 975 }
 976
 977 wxDataViewColumn *
 978 wxDataViewCtrlBase::AppendBitmapColumn( const wxBitmap &label, unsigned int model_column,
 979                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
 980 {
 981     wxDataViewColumn *ret = new wxDataViewColumn( label,
 982         new wxDataViewBitmapRenderer( wxT("wxBitmap"), mode, (int)align ),
 983         model_column, width, align, flags );
 984     AppendColumn( ret );
 985     return ret;
 986 }
 987
 988 wxDataViewColumn *
 989 wxDataViewCtrlBase::PrependTextColumn( const wxString &label, unsigned int model_column,
 990                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
 991 {
 992     wxDataViewColumn *ret = new wxDataViewColumn( label,
 993         new wxDataViewTextRenderer( wxT("string"), mode, (int)align ),
 994         model_column, width, align, flags );
 995     PrependColumn( ret );
 996     return ret;
 997 }
 998
 999 wxDataViewColumn *
1000 wxDataViewCtrlBase::PrependIconTextColumn( const wxString &label, unsigned int model_column,
1001                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1002 {
1003     wxDataViewColumn *ret = new wxDataViewColumn( label,
1004         new wxDataViewIconTextRenderer( wxT("wxDataViewIconText"), mode, (int)align ),
1005         model_column, width, align, flags );
1006     PrependColumn( ret );
1007     return ret;
1008 }
1009
1010 wxDataViewColumn *
1011 wxDataViewCtrlBase::PrependToggleColumn( const wxString &label, unsigned int model_column,
1012                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1013 {
1014
1015     wxDataViewColumn *ret = new wxDataViewColumn( label,
1016         new wxDataViewToggleRenderer( wxT("bool"), mode, (int)align ),
1017         model_column, width, align, flags );
1018     PrependColumn( ret );
1019     return ret;
1020 }
1021
1022 wxDataViewColumn *
1023 wxDataViewCtrlBase::PrependProgressColumn( const wxString &label, unsigned int model_column,
1024                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1025 {
1026     wxDataViewColumn *ret = new wxDataViewColumn( label,
1027         new wxDataViewProgressRenderer( wxEmptyString, wxT("long"), mode, (int)align ),
1028         model_column, width, align, flags );
1029     PrependColumn( ret );
1030     return ret;
1031 }
1032
1033 wxDataViewColumn *
1034 wxDataViewCtrlBase::PrependDateColumn( const wxString &label, unsigned int model_column,
1035                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1036 {
1037     wxDataViewColumn *ret = new wxDataViewColumn( label,
1038         new wxDataViewDateRenderer( wxT("datetime"), mode, (int)align ),
1039         model_column, width, align, flags );
1040     PrependColumn( ret );
1041     return ret;
1042 }
1043
1044 wxDataViewColumn *
1045 wxDataViewCtrlBase::PrependBitmapColumn( const wxString &label, unsigned int model_column,
1046                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1047 {
1048     wxDataViewColumn *ret = new wxDataViewColumn( label,
1049         new wxDataViewBitmapRenderer( wxT("wxBitmap"), mode, (int)align ),
1050         model_column, width, align, flags );
1051     PrependColumn( ret );
1052     return ret;
1053 }
1054
1055 wxDataViewColumn *
1056 wxDataViewCtrlBase::PrependTextColumn( const wxBitmap &label, unsigned int model_column,
1057                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1058 {
1059     wxDataViewColumn *ret = new wxDataViewColumn( label,
1060         new wxDataViewTextRenderer( wxT("string"), mode, (int)align ),
1061         model_column, width, align, flags );
1062     PrependColumn( ret );
1063     return ret;
1064 }
1065
1066 wxDataViewColumn *
1067 wxDataViewCtrlBase::PrependIconTextColumn( const wxBitmap &label, unsigned int model_column,
1068                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1069 {
1070     wxDataViewColumn *ret = new wxDataViewColumn( label,
1071         new wxDataViewIconTextRenderer( wxT("wxDataViewIconText"), mode, (int)align ),
1072         model_column, width, align, flags );
1073     PrependColumn( ret );
1074     return ret;
1075 }
1076
1077 wxDataViewColumn *
1078 wxDataViewCtrlBase::PrependToggleColumn( const wxBitmap &label, unsigned int model_column,
1079                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1080 {
1081     wxDataViewColumn *ret = new wxDataViewColumn( label,
1082         new wxDataViewToggleRenderer( wxT("bool"), mode, (int)align ),
1083         model_column, width, align, flags );
1084     PrependColumn( ret );
1085     return ret;
1086 }
1087
1088 wxDataViewColumn *
1089 wxDataViewCtrlBase::PrependProgressColumn( const wxBitmap &label, unsigned int model_column,
1090                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1091 {
1092     wxDataViewColumn *ret = new wxDataViewColumn( label,
1093         new wxDataViewProgressRenderer( wxEmptyString, wxT("long"), mode, (int)align ),
1094         model_column, width, align, flags );
1095     PrependColumn( ret );
1096     return ret;
1097 }
1098
1099 wxDataViewColumn *
1100 wxDataViewCtrlBase::PrependDateColumn( const wxBitmap &label, unsigned int model_column,
1101                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1102 {
1103     wxDataViewColumn *ret = new wxDataViewColumn( label,
1104         new wxDataViewDateRenderer( wxT("datetime"), mode, (int)align ),
1105         model_column, width, align, flags );
1106     PrependColumn( ret );
1107     return ret;
1108 }
1109
1110 wxDataViewColumn *
1111 wxDataViewCtrlBase::PrependBitmapColumn( const wxBitmap &label, unsigned int model_column,
1112                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1113 {
1114     wxDataViewColumn *ret = new wxDataViewColumn( label,
1115         new wxDataViewBitmapRenderer( wxT("wxBitmap"), mode, (int)align ),
1116         model_column, width, align, flags );
1117     PrependColumn( ret );
1118     return ret;
1119 }
1120
1121 bool
1122 wxDataViewCtrlBase::AppendColumn( wxDataViewColumn *col )
1123 {
1124     col->SetOwner( (wxDataViewCtrl*) this );
1125     return true;
1126 }
1127
1128 bool
1129 wxDataViewCtrlBase::PrependColumn( wxDataViewColumn *col )
1130 {
1131     col->SetOwner( (wxDataViewCtrl*) this );
1132     return true;
1133 }
1134
1135 // ---------------------------------------------------------
1136 // wxDataViewEvent
1137 // ---------------------------------------------------------
1138
1139 IMPLEMENT_DYNAMIC_CLASS(wxDataViewEvent,wxNotifyEvent)
1140
1141 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_SELECTION_CHANGED)
1142
1143 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_ITEM_ACTIVATED)
1144 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_ITEM_COLLAPSING)
1145 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_ITEM_COLLAPSED)
1146 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_ITEM_EXPANDING)
1147 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_ITEM_EXPANDED)
1148 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_ITEM_EDITING_STARTED)
1149 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_ITEM_EDITING_DONE)
1150 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_ITEM_VALUE_CHANGED)
1151
1152 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_ITEM_CONTEXT_MENU)
1153
1154 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_CLICK)
1155 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK)
1156 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_COLUMN_SORTED)
1157 DEFINE_EVENT_TYPE(wxEVT_COMMAND_DATAVIEW_COLUMN_REORDERED)
1158
1159
1160 // -------------------------------------
1161 // wxDataViewSpinRenderer
1162 // -------------------------------------
1163
1164 wxDataViewSpinRenderer::wxDataViewSpinRenderer( int min, int max, wxDataViewCellMode mode, int alignment ) :
1165    wxDataViewCustomRenderer( "long", mode, alignment )
1166 {
1167     m_min = min;
1168     m_max = max;
1169 }
1170
1171 wxControl* wxDataViewSpinRenderer::CreateEditorCtrl( wxWindow *parent, wxRect labelRect, const wxVariant &value )
1172 {
1173     long l = value;
1174     return new wxSpinCtrl( parent, wxID_ANY, wxEmptyString,
1175                labelRect.GetTopLeft(), labelRect.GetSize(), wxSP_ARROW_KEYS, m_min, m_max, l );
1176 }
1177
1178 bool wxDataViewSpinRenderer::GetValueFromEditorCtrl( wxControl* editor, wxVariant &value )
1179 {
1180     wxSpinCtrl *sc = (wxSpinCtrl*) editor;
1181     long l = sc->GetValue();
1182     value = l;
1183     return true;
1184 }
1185
1186 bool wxDataViewSpinRenderer::Render( wxRect rect, wxDC *dc, int state )
1187 {
1188     wxString str;
1189     str.Printf( "%d", (int) m_data );
1190     RenderText( str, 0, rect, dc, state );
1191     return true;
1192 }
1193
1194 wxSize wxDataViewSpinRenderer::GetSize() const
1195 {
1196     return wxSize(80,16);
1197 }
1198
1199 bool wxDataViewSpinRenderer::SetValue( const wxVariant &value )
1200 {
1201     m_data = value.GetLong();
1202     return true;
1203 }
1204
1205 bool wxDataViewSpinRenderer::GetValue( wxVariant &value ) const
1206 {
1207     value = m_data;
1208     return true;
1209 }
1210
1211 //-----------------------------------------------------------------------------
1212 // wxDataViewTreeStore
1213 //-----------------------------------------------------------------------------
1214
1215 wxDataViewTreeStoreNode::wxDataViewTreeStoreNode(
1216         wxDataViewTreeStoreNode *parent,
1217         const wxString &text, const wxIcon &icon, wxClientData *data )
1218 {
1219     m_parent = parent;
1220     m_text = text;
1221     m_icon = icon;
1222     m_data = data;
1223 }
1224
1225 wxDataViewTreeStoreNode::~wxDataViewTreeStoreNode()
1226 {
1227     if (m_data)
1228         delete m_data;
1229 }
1230
1231 #include "wx/listimpl.cpp"
1232 WX_DEFINE_LIST(wxDataViewTreeStoreNodeList)
1233
1234 wxDataViewTreeStoreContainerNode::wxDataViewTreeStoreContainerNode(
1235         wxDataViewTreeStoreNode *parent, const wxString &text,
1236         const wxIcon &icon, const wxIcon &expanded, wxClientData *data ) :
1237     wxDataViewTreeStoreNode( parent, text, icon, data )
1238 {
1239     m_iconExpanded = expanded;
1240     m_isExpanded = false;
1241     m_children.DeleteContents(true);
1242 }
1243
1244 wxDataViewTreeStoreContainerNode::~wxDataViewTreeStoreContainerNode()
1245 {
1246 }
1247
1248 //-----------------------------------------------------------------------------
1249
1250 wxDataViewTreeStore::wxDataViewTreeStore()
1251 {
1252     m_root = new wxDataViewTreeStoreContainerNode( NULL, wxEmptyString );
1253 }
1254
1255 wxDataViewTreeStore::~wxDataViewTreeStore()
1256 {
1257     delete m_root;
1258 }
1259
1260 wxDataViewItem wxDataViewTreeStore::AppendItem( const wxDataViewItem& parent,
1261         const wxString &text, const wxIcon &icon, wxClientData *data )
1262 {
1263     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1264     if (!parent_node) return wxDataViewItem(0);
1265
1266     wxDataViewTreeStoreNode *node =
1267         new wxDataViewTreeStoreNode( parent_node, text, icon, data );
1268     parent_node->GetChildren().Append( node );
1269
1270     // notify control
1271     ItemAdded( parent, node->GetItem() );
1272
1273     return node->GetItem();
1274 }
1275
1276 wxDataViewItem wxDataViewTreeStore::PrependItem( const wxDataViewItem& parent,
1277         const wxString &text, const wxIcon &icon, wxClientData *data )
1278 {
1279     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1280     if (!parent_node) return wxDataViewItem(0);
1281
1282     wxDataViewTreeStoreNode *node =
1283         new wxDataViewTreeStoreNode( parent_node, text, icon, data );
1284     parent_node->GetChildren().Insert( node );
1285
1286     // notify control
1287     ItemAdded( parent, node->GetItem() );
1288
1289     return node->GetItem();
1290 }
1291
1292 wxDataViewItem
1293 wxDataViewTreeStore::InsertItem(const wxDataViewItem& WXUNUSED(parent),
1294                                 const wxDataViewItem& WXUNUSED(previous),
1295                                 const wxString& WXUNUSED(text),
1296                                 const wxIcon& WXUNUSED(icon),
1297                                 wxClientData * WXUNUSED(data))
1298 {
1299     return wxDataViewItem(0);
1300 }
1301
1302 wxDataViewItem wxDataViewTreeStore::PrependContainer( const wxDataViewItem& parent,
1303         const wxString &text, const wxIcon &icon, const wxIcon &expanded,
1304         wxClientData *data )
1305 {
1306     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1307     if (!parent_node) return wxDataViewItem(0);
1308
1309     wxDataViewTreeStoreContainerNode *node =
1310         new wxDataViewTreeStoreContainerNode( parent_node, text, icon, expanded, data );
1311     parent_node->GetChildren().Insert( node );
1312
1313     // notify control
1314     ItemAdded( parent, node->GetItem() );
1315
1316     return node->GetItem();
1317 }
1318
1319 wxDataViewItem
1320 wxDataViewTreeStore::AppendContainer(const wxDataViewItem& parent,
1321                                      const wxString &text,
1322                                      const wxIcon& icon,
1323                                      const wxIcon& expanded,
1324                                      wxClientData * data)
1325 {
1326     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1327     if (!parent_node) return wxDataViewItem(0);
1328
1329     wxDataViewTreeStoreContainerNode *node =
1330         new wxDataViewTreeStoreContainerNode( parent_node, text, icon, expanded, data );
1331     parent_node->GetChildren().Append( node );
1332
1333     // notify control
1334     ItemAdded( parent, node->GetItem() );
1335
1336     return node->GetItem();
1337 }
1338
1339 wxDataViewItem
1340 wxDataViewTreeStore::InsertContainer(const wxDataViewItem& WXUNUSED(parent),
1341                                      const wxDataViewItem& WXUNUSED(previous),
1342                                      const wxString& WXUNUSED(text),
1343                                      const wxIcon& WXUNUSED(icon),
1344                                      const wxIcon& WXUNUSED(expanded),
1345                                      wxClientData * WXUNUSED(data))
1346 {
1347     return wxDataViewItem(0);
1348 }
1349
1350 wxDataViewItem wxDataViewTreeStore::GetNthChild( const wxDataViewItem& parent, unsigned int pos ) const
1351 {
1352     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1353     if (!parent_node) return wxDataViewItem(0);
1354
1355     wxDataViewTreeStoreNodeList::compatibility_iterator node = parent_node->GetChildren().Item( pos );
1356     if (node)
1357         return node->GetData();
1358
1359     return wxDataViewItem(0);
1360 }
1361
1362 int wxDataViewTreeStore::GetChildCount( const wxDataViewItem& parent ) const
1363 {
1364     wxDataViewTreeStoreNode *node = FindNode( parent );
1365     if (!node) return -1;
1366
1367     if (!node->IsContainer())
1368         return 0;
1369
1370     wxDataViewTreeStoreContainerNode *container_node = (wxDataViewTreeStoreContainerNode*) node;
1371     return (int) container_node->GetChildren().GetCount();
1372 }
1373
1374 void wxDataViewTreeStore::SetItemText( const wxDataViewItem& item, const wxString &text )
1375 {
1376     wxDataViewTreeStoreNode *node = FindNode( item );
1377     if (!node) return;
1378
1379     node->SetText( text );
1380
1381     // notify control
1382     ValueChanged( item, 0 );
1383 }
1384
1385 wxString wxDataViewTreeStore::GetItemText( const wxDataViewItem& item ) const
1386 {
1387     wxDataViewTreeStoreNode *node = FindNode( item );
1388     if (!node) return wxEmptyString;
1389
1390     return node->GetText();
1391 }
1392
1393 void wxDataViewTreeStore::SetItemIcon( const wxDataViewItem& item, const wxIcon &icon )
1394 {
1395     wxDataViewTreeStoreNode *node = FindNode( item );
1396     if (!node) return;
1397
1398     node->SetIcon( icon );
1399
1400     // notify control
1401     ValueChanged( item, 0 );
1402 }
1403
1404 const wxIcon &wxDataViewTreeStore::GetItemIcon( const wxDataViewItem& item ) const
1405 {
1406     wxDataViewTreeStoreNode *node = FindNode( item );
1407     if (!node) return wxNullIcon;
1408
1409     return node->GetIcon();
1410 }
1411
1412 void wxDataViewTreeStore::SetItemExpandedIcon( const wxDataViewItem& item, const wxIcon &icon )
1413 {
1414     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1415     if (!node) return;
1416
1417     node->SetExpandedIcon( icon );
1418
1419     // notify control
1420     ValueChanged( item, 0 );
1421 }
1422
1423 const wxIcon &wxDataViewTreeStore::GetItemExpandedIcon( const wxDataViewItem& item ) const
1424 {
1425     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1426     if (!node) return wxNullIcon;
1427
1428     return node->GetExpandedIcon();
1429 }
1430
1431 void wxDataViewTreeStore::SetItemData( const wxDataViewItem& item, wxClientData *data )
1432 {
1433     wxDataViewTreeStoreNode *node = FindNode( item );
1434     if (!node) return;
1435
1436     node->SetData( data );
1437
1438     // notify control? only sensible when sorting on client data
1439     // ValueChanged( item, 0 );
1440 }
1441
1442 wxClientData *wxDataViewTreeStore::GetItemData( const wxDataViewItem& item ) const
1443 {
1444     wxDataViewTreeStoreNode *node = FindNode( item );
1445     if (!node) return NULL;
1446
1447     return node->GetData();
1448 }
1449
1450 void wxDataViewTreeStore::DeleteItem( const wxDataViewItem& item )
1451 {
1452     if (!item.IsOk()) return;
1453
1454     wxDataViewItem parent_item = GetParent( item );
1455
1456     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent_item );
1457     if (!parent_node) return;
1458
1459     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1460     if (!node) return;
1461
1462     parent_node->GetChildren().DeleteObject( node );
1463
1464     // notify control
1465     ItemDeleted( parent_item, item );
1466 }
1467
1468 void wxDataViewTreeStore::DeleteChildren( const wxDataViewItem& item )
1469 {
1470     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1471     if (!node) return;
1472
1473     wxDataViewItemArray array;
1474     wxDataViewTreeStoreNodeList::iterator iter;
1475     for (iter = node->GetChildren().begin(); iter != node->GetChildren().end(); iter++)
1476     {
1477         wxDataViewTreeStoreNode* child = *iter;
1478         array.Add( child->GetItem() );
1479     }
1480
1481     node->GetChildren().clear();
1482
1483     // notify control
1484     ItemsDeleted( item, array );
1485 }
1486
1487 void wxDataViewTreeStore::DeleteAllItems()
1488 {
1489     // TODO
1490 }
1491
1492 void
1493 wxDataViewTreeStore::GetValue(wxVariant &variant,
1494                               const wxDataViewItem &item,
1495                               unsigned int WXUNUSED(col)) const
1496 {
1497     // if (col != 0) return;
1498
1499     wxDataViewTreeStoreNode *node = FindNode( item );
1500     if (!node) return;
1501
1502     wxIcon icon( node->GetIcon());
1503     if (node->IsContainer())
1504     {
1505         wxDataViewTreeStoreContainerNode *container = (wxDataViewTreeStoreContainerNode*) node;
1506         if (container->IsExpanded() && container->GetExpandedIcon().IsOk())
1507            icon = container->GetExpandedIcon();
1508     }
1509
1510     wxDataViewIconText data( node->GetText(), icon );
1511
1512     variant << data;
1513 }
1514
1515 bool
1516 wxDataViewTreeStore::SetValue(const wxVariant& variant,
1517                               const wxDataViewItem& item,
1518                               unsigned int WXUNUSED(col))
1519 {
1520     // if (col != 0) return false;
1521
1522     wxDataViewTreeStoreNode *node = FindNode( item );
1523     if (!node) return false;
1524
1525     wxDataViewIconText data;
1526
1527     data << variant;
1528
1529     node->SetText( data.GetText() );
1530     node->SetIcon( data.GetIcon() );
1531
1532     return true;
1533 }
1534
1535 wxDataViewItem wxDataViewTreeStore::GetParent( const wxDataViewItem &item ) const
1536 {
1537     wxDataViewTreeStoreNode *node = FindNode( item );
1538     if (!node) return wxDataViewItem(0);
1539
1540     wxDataViewTreeStoreNode *parent = node->GetParent();
1541     if (!parent) return wxDataViewItem(0);
1542
1543     if (parent == m_root)
1544         return wxDataViewItem(0);
1545
1546     return parent->GetItem();
1547 }
1548
1549 bool wxDataViewTreeStore::IsContainer( const wxDataViewItem &item ) const
1550 {
1551     wxDataViewTreeStoreNode *node = FindNode( item );
1552     if (!node) return false;
1553
1554     return node->IsContainer();
1555 }
1556
1557 unsigned int wxDataViewTreeStore::GetChildren( const wxDataViewItem &item, wxDataViewItemArray &children ) const
1558 {
1559     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1560     if (!node) return 0;
1561
1562     wxDataViewTreeStoreNodeList::iterator iter;
1563     for (iter = node->GetChildren().begin(); iter != node->GetChildren().end(); iter++)
1564     {
1565         wxDataViewTreeStoreNode* child = *iter;
1566         children.Add( child->GetItem() );
1567     }
1568
1569     return node->GetChildren().GetCount();
1570 }
1571
1572 int wxDataViewTreeStore::Compare( const wxDataViewItem &item1, const wxDataViewItem &item2,
1573                          unsigned int WXUNUSED(column), bool WXUNUSED(ascending) )
1574 {
1575     wxDataViewTreeStoreNode *node1 = FindNode( item1 );
1576     wxDataViewTreeStoreNode *node2 = FindNode( item2 );
1577
1578     if (!node1 || !node2)
1579         return 0;
1580
1581     wxDataViewTreeStoreContainerNode* parent1 =
1582         (wxDataViewTreeStoreContainerNode*) node1->GetParent();
1583     wxDataViewTreeStoreContainerNode* parent2 =
1584         (wxDataViewTreeStoreContainerNode*) node2->GetParent();
1585
1586     if (parent1 != parent2)
1587     {
1588         wxLogError( wxT("Comparing items with different parent.") );
1589         return 0;
1590     }
1591
1592     if (node1->IsContainer() && !!node2->IsContainer())
1593         return 1;
1594
1595     if (node2->IsContainer() && !!node1->IsContainer())
1596         return -1;
1597
1598     return parent1->GetChildren().IndexOf( node1 ) - parent1->GetChildren().IndexOf( node2 );
1599 }
1600
1601 wxDataViewTreeStoreNode *wxDataViewTreeStore::FindNode( const wxDataViewItem &item ) const
1602 {
1603     if (!item.IsOk())
1604         return m_root;
1605
1606     return (wxDataViewTreeStoreNode*) item.GetID();
1607 }
1608
1609 wxDataViewTreeStoreContainerNode *wxDataViewTreeStore::FindContainerNode( const wxDataViewItem &item ) const
1610 {
1611     if (!item.IsOk())
1612         return (wxDataViewTreeStoreContainerNode*) m_root;
1613
1614     wxDataViewTreeStoreNode* node = (wxDataViewTreeStoreNode*) item.GetID();
1615
1616     if (!node->IsContainer())
1617         return NULL;
1618
1619     return (wxDataViewTreeStoreContainerNode*) node;
1620 }
1621
1622 //-----------------------------------------------------------------------------
1623 // wxDataViewTreeCtrl
1624 //-----------------------------------------------------------------------------
1625
1626 IMPLEMENT_DYNAMIC_CLASS(wxDataViewTreeCtrl,wxDataViewCtrl)
1627
1628 BEGIN_EVENT_TABLE(wxDataViewTreeCtrl,wxDataViewCtrl)
1629    EVT_DATAVIEW_ITEM_EXPANDED(-1, wxDataViewTreeCtrl::OnExpanded)
1630    EVT_DATAVIEW_ITEM_COLLAPSED(-1, wxDataViewTreeCtrl::OnCollapsed)
1631    EVT_SIZE( wxDataViewTreeCtrl::OnSize )
1632 END_EVENT_TABLE()
1633
1634 wxDataViewTreeCtrl::wxDataViewTreeCtrl()
1635 {
1636     m_imageList = NULL;
1637 }
1638
1639 wxDataViewTreeCtrl::wxDataViewTreeCtrl( wxWindow *parent, wxWindowID id,
1640            const wxPoint& pos, const wxSize& size, long style, const wxValidator& validator )
1641 {
1642     m_imageList = NULL;
1643     Create( parent, id, pos, size, style, validator );
1644
1645     wxDataViewTreeStore *store = new wxDataViewTreeStore;
1646     AssociateModel( store );
1647     store->DecRef();
1648
1649 #if !defined(__WXGTK20__) || defined(wxUSE_GENERICDATAVIEWCTRL)
1650     wxDataViewColumn *col = AppendIconTextColumn( "", 0, wxDATAVIEW_CELL_INERT, 40 );
1651
1652     wxSize cient_size = GetClientSize();
1653     col->SetWidth( size.x );
1654 #else
1655     AppendIconTextColumn( "", 0, wxDATAVIEW_CELL_INERT, 40 );
1656 #endif
1657 }
1658
1659 wxDataViewTreeCtrl::~wxDataViewTreeCtrl()
1660 {
1661     if (m_imageList)
1662         delete m_imageList;
1663 }
1664
1665 bool wxDataViewTreeCtrl::Create( wxWindow *parent, wxWindowID id,
1666            const wxPoint& pos, const wxSize& size, long style, const wxValidator& validator )
1667 {
1668     return wxDataViewCtrl::Create( parent, id, pos, size, style, validator );
1669 }
1670
1671 void wxDataViewTreeCtrl::SetImageList( wxImageList *imagelist )
1672 {
1673     if (m_imageList)
1674         delete m_imageList;
1675
1676     m_imageList = imagelist;
1677 }
1678
1679 wxDataViewItem wxDataViewTreeCtrl::AppendItem( const wxDataViewItem& parent,
1680         const wxString &text, int iconIndex, wxClientData *data )
1681 {
1682     wxIcon icon = wxNullIcon;
1683     if (m_imageList && (iconIndex != -1))
1684         icon = m_imageList->GetIcon( iconIndex );
1685
1686     return GetStore()->AppendItem( parent, text, icon, data );
1687 }
1688
1689 wxDataViewItem wxDataViewTreeCtrl::PrependItem( const wxDataViewItem& parent,
1690         const wxString &text, int iconIndex, wxClientData *data )
1691 {
1692     wxIcon icon = wxNullIcon;
1693     if (m_imageList && (iconIndex != -1))
1694         icon = m_imageList->GetIcon( iconIndex );
1695
1696     return GetStore()->PrependItem( parent, text, icon, data );
1697 }
1698
1699 wxDataViewItem wxDataViewTreeCtrl::InsertItem( const wxDataViewItem& parent, const wxDataViewItem& previous,
1700         const wxString &text, int iconIndex, wxClientData *data )
1701 {
1702     wxIcon icon = wxNullIcon;
1703     if (m_imageList && (iconIndex != -1))
1704         icon = m_imageList->GetIcon( iconIndex );
1705
1706     return GetStore()->InsertItem( parent, previous, text, icon, data );
1707 }
1708
1709 wxDataViewItem wxDataViewTreeCtrl::PrependContainer( const wxDataViewItem& parent,
1710         const wxString &text, int iconIndex, int expandedIndex, wxClientData *data )
1711 {
1712     wxIcon icon = wxNullIcon;
1713     if (m_imageList && (iconIndex != -1))
1714         icon = m_imageList->GetIcon( iconIndex );
1715
1716     wxIcon expanded = wxNullIcon;
1717     if (m_imageList && (expandedIndex != -1))
1718         expanded = m_imageList->GetIcon( expandedIndex );
1719
1720     return GetStore()->PrependContainer( parent, text, icon, expanded, data );
1721 }
1722
1723 wxDataViewItem wxDataViewTreeCtrl::AppendContainer( const wxDataViewItem& parent,
1724         const wxString &text, int iconIndex, int expandedIndex, wxClientData *data )
1725 {
1726     wxIcon icon = wxNullIcon;
1727     if (m_imageList && (iconIndex != -1))
1728         icon = m_imageList->GetIcon( iconIndex );
1729
1730     wxIcon expanded = wxNullIcon;
1731     if (m_imageList && (expandedIndex != -1))
1732         expanded = m_imageList->GetIcon( expandedIndex );
1733
1734     return GetStore()->AppendContainer( parent, text, icon, expanded, data );
1735 }
1736
1737 wxDataViewItem wxDataViewTreeCtrl::InsertContainer( const wxDataViewItem& parent, const wxDataViewItem& previous,
1738         const wxString &text, int iconIndex, int expandedIndex, wxClientData *data )
1739 {
1740     wxIcon icon = wxNullIcon;
1741     if (m_imageList && (iconIndex != -1))
1742         icon = m_imageList->GetIcon( iconIndex );
1743
1744     wxIcon expanded = wxNullIcon;
1745     if (m_imageList && (expandedIndex != -1))
1746         expanded = m_imageList->GetIcon( expandedIndex );
1747
1748     return GetStore()->InsertContainer( parent, previous, text, icon, expanded, data );
1749 }
1750
1751 void wxDataViewTreeCtrl::OnExpanded( wxDataViewEvent &event )
1752 {
1753     if (m_imageList) return;
1754
1755     wxDataViewTreeStoreContainerNode* container = GetStore()->FindContainerNode( event.GetItem() );
1756     if (!container) return;
1757
1758     container->SetExpanded( true );
1759     GetStore()->ItemChanged( event.GetItem() );
1760 }
1761
1762 void wxDataViewTreeCtrl::OnCollapsed( wxDataViewEvent &event )
1763 {
1764     if (m_imageList) return;
1765
1766     wxDataViewTreeStoreContainerNode* container = GetStore()->FindContainerNode( event.GetItem() );
1767     if (!container) return;
1768
1769     container->SetExpanded( false );
1770     GetStore()->ItemChanged( event.GetItem() );
1771 }
1772
1773 void wxDataViewTreeCtrl::OnSize( wxSizeEvent &event )
1774 {
1775 #if !defined(__WXGTK20__) || defined(wxUSE_GENERICDATAVIEWCTRL)
1776     wxSize size = GetClientSize();
1777     wxDataViewColumn *col = GetColumn( 0 );
1778     if (col) col->SetWidth( size.x );
1779 #endif
1780     event.Skip( true );
1781 }
1782
1783 #endif // wxUSE_DATAVIEWCTRL
1784