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
1158
1159 // -------------------------------------
1160 // wxDataViewSpinRenderer
1161 // -------------------------------------
1162
1163 wxDataViewSpinRenderer::wxDataViewSpinRenderer( int min, int max, wxDataViewCellMode mode, int alignment ) :
1164    wxDataViewCustomRenderer( "long", mode, alignment )
1165 {
1166     m_min = min;
1167     m_max = max;
1168 }
1169
1170 wxControl* wxDataViewSpinRenderer::CreateEditorCtrl( wxWindow *parent, wxRect labelRect, const wxVariant &value )
1171 {
1172     long l = value;
1173     return new wxSpinCtrl( parent, wxID_ANY, wxEmptyString,
1174                labelRect.GetTopLeft(), labelRect.GetSize(), wxSP_ARROW_KEYS, m_min, m_max, l );
1175 }
1176
1177 bool wxDataViewSpinRenderer::GetValueFromEditorCtrl( wxControl* editor, wxVariant &value )
1178 {
1179     wxSpinCtrl *sc = (wxSpinCtrl*) editor;
1180     long l = sc->GetValue();
1181     value = l;
1182     return true;
1183 }
1184
1185 bool wxDataViewSpinRenderer::Render( wxRect rect, wxDC *dc, int state )
1186 {
1187     wxString str;
1188     str.Printf( "%d", (int) m_data );
1189     RenderText( str, 0, rect, dc, state );
1190     return true;
1191 }
1192
1193 wxSize wxDataViewSpinRenderer::GetSize() const
1194 {
1195     return wxSize(80,16);
1196 }
1197
1198 bool wxDataViewSpinRenderer::SetValue( const wxVariant &value )
1199 {
1200     m_data = value.GetLong();
1201     return true;
1202 }
1203
1204 bool wxDataViewSpinRenderer::GetValue( wxVariant &value ) const
1205 {
1206     value = m_data;
1207     return true;
1208 }
1209
1210 //-----------------------------------------------------------------------------
1211 // wxDataViewTreeStore
1212 //-----------------------------------------------------------------------------
1213
1214 wxDataViewTreeStoreNode::wxDataViewTreeStoreNode(
1215         wxDataViewTreeStoreNode *parent,
1216         const wxString &text, const wxIcon &icon, wxClientData *data )
1217 {
1218     m_parent = parent;
1219     m_text = text;
1220     m_icon = icon;
1221     m_data = data;
1222 }
1223
1224 wxDataViewTreeStoreNode::~wxDataViewTreeStoreNode()
1225 {
1226     if (m_data)
1227         delete m_data;
1228 }
1229
1230 #include "wx/listimpl.cpp"
1231 WX_DEFINE_LIST(wxDataViewTreeStoreNodeList)
1232
1233 wxDataViewTreeStoreContainerNode::wxDataViewTreeStoreContainerNode(
1234         wxDataViewTreeStoreNode *parent, const wxString &text,
1235         const wxIcon &icon, const wxIcon &expanded, wxClientData *data ) :
1236     wxDataViewTreeStoreNode( parent, text, icon, data )
1237 {
1238     m_iconExpanded = expanded;
1239     m_isExpanded = false;
1240     m_children.DeleteContents(true);
1241 }
1242
1243 wxDataViewTreeStoreContainerNode::~wxDataViewTreeStoreContainerNode()
1244 {
1245 }
1246
1247 //-----------------------------------------------------------------------------
1248
1249 wxDataViewTreeStore::wxDataViewTreeStore()
1250 {
1251     m_root = new wxDataViewTreeStoreContainerNode( NULL, wxEmptyString );
1252 }
1253
1254 wxDataViewTreeStore::~wxDataViewTreeStore()
1255 {
1256     delete m_root;
1257 }
1258
1259 wxDataViewItem wxDataViewTreeStore::AppendItem( const wxDataViewItem& parent,
1260         const wxString &text, const wxIcon &icon, wxClientData *data )
1261 {
1262     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1263     if (!parent_node) return wxDataViewItem(0);
1264
1265     wxDataViewTreeStoreNode *node =
1266         new wxDataViewTreeStoreNode( parent_node, text, icon, data );
1267     parent_node->GetChildren().Append( node );
1268
1269     // notify control
1270     ItemAdded( parent, node->GetItem() );
1271
1272     return node->GetItem();
1273 }
1274
1275 wxDataViewItem wxDataViewTreeStore::PrependItem( const wxDataViewItem& parent,
1276         const wxString &text, const wxIcon &icon, wxClientData *data )
1277 {
1278     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1279     if (!parent_node) return wxDataViewItem(0);
1280
1281     wxDataViewTreeStoreNode *node =
1282         new wxDataViewTreeStoreNode( parent_node, text, icon, data );
1283     parent_node->GetChildren().Insert( node );
1284
1285     // notify control
1286     ItemAdded( parent, node->GetItem() );
1287
1288     return node->GetItem();
1289 }
1290
1291 wxDataViewItem
1292 wxDataViewTreeStore::InsertItem(const wxDataViewItem& WXUNUSED(parent),
1293                                 const wxDataViewItem& WXUNUSED(previous),
1294                                 const wxString& WXUNUSED(text),
1295                                 const wxIcon& WXUNUSED(icon),
1296                                 wxClientData * WXUNUSED(data))
1297 {
1298     return wxDataViewItem(0);
1299 }
1300
1301 wxDataViewItem wxDataViewTreeStore::PrependContainer( const wxDataViewItem& parent,
1302         const wxString &text, const wxIcon &icon, const wxIcon &expanded,
1303         wxClientData *data )
1304 {
1305     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1306     if (!parent_node) return wxDataViewItem(0);
1307
1308     wxDataViewTreeStoreContainerNode *node =
1309         new wxDataViewTreeStoreContainerNode( parent_node, text, icon, expanded, data );
1310     parent_node->GetChildren().Insert( node );
1311
1312     // notify control
1313     ItemAdded( parent, node->GetItem() );
1314
1315     return node->GetItem();
1316 }
1317
1318 wxDataViewItem
1319 wxDataViewTreeStore::AppendContainer(const wxDataViewItem& parent,
1320                                      const wxString &text,
1321                                      const wxIcon& icon,
1322                                      const wxIcon& expanded,
1323                                      wxClientData * data)
1324 {
1325     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1326     if (!parent_node) return wxDataViewItem(0);
1327
1328     wxDataViewTreeStoreContainerNode *node =
1329         new wxDataViewTreeStoreContainerNode( parent_node, text, icon, expanded, data );
1330     parent_node->GetChildren().Append( node );
1331
1332     // notify control
1333     ItemAdded( parent, node->GetItem() );
1334
1335     return node->GetItem();
1336 }
1337
1338 wxDataViewItem
1339 wxDataViewTreeStore::InsertContainer(const wxDataViewItem& WXUNUSED(parent),
1340                                      const wxDataViewItem& WXUNUSED(previous),
1341                                      const wxString& WXUNUSED(text),
1342                                      const wxIcon& WXUNUSED(icon),
1343                                      const wxIcon& WXUNUSED(expanded),
1344                                      wxClientData * WXUNUSED(data))
1345 {
1346     return wxDataViewItem(0);
1347 }
1348
1349 wxDataViewItem wxDataViewTreeStore::GetNthChild( const wxDataViewItem& parent, unsigned int pos ) const
1350 {
1351     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1352     if (!parent_node) return wxDataViewItem(0);
1353
1354     wxDataViewTreeStoreNodeList::compatibility_iterator node = parent_node->GetChildren().Item( pos );
1355     if (node)
1356         return node->GetData();
1357
1358     return wxDataViewItem(0);
1359 }
1360
1361 int wxDataViewTreeStore::GetChildCount( const wxDataViewItem& parent ) const
1362 {
1363     wxDataViewTreeStoreNode *node = FindNode( parent );
1364     if (!node) return -1;
1365
1366     if (!node->IsContainer())
1367         return 0;
1368
1369     wxDataViewTreeStoreContainerNode *container_node = (wxDataViewTreeStoreContainerNode*) node;
1370     return (int) container_node->GetChildren().GetCount();
1371 }
1372
1373 void wxDataViewTreeStore::SetItemText( const wxDataViewItem& item, const wxString &text )
1374 {
1375     wxDataViewTreeStoreNode *node = FindNode( item );
1376     if (!node) return;
1377
1378     node->SetText( text );
1379
1380     // notify control
1381     ValueChanged( item, 0 );
1382 }
1383
1384 wxString wxDataViewTreeStore::GetItemText( const wxDataViewItem& item ) const
1385 {
1386     wxDataViewTreeStoreNode *node = FindNode( item );
1387     if (!node) return wxEmptyString;
1388
1389     return node->GetText();
1390 }
1391
1392 void wxDataViewTreeStore::SetItemIcon( const wxDataViewItem& item, const wxIcon &icon )
1393 {
1394     wxDataViewTreeStoreNode *node = FindNode( item );
1395     if (!node) return;
1396
1397     node->SetIcon( icon );
1398
1399     // notify control
1400     ValueChanged( item, 0 );
1401 }
1402
1403 const wxIcon &wxDataViewTreeStore::GetItemIcon( const wxDataViewItem& item ) const
1404 {
1405     wxDataViewTreeStoreNode *node = FindNode( item );
1406     if (!node) return wxNullIcon;
1407
1408     return node->GetIcon();
1409 }
1410
1411 void wxDataViewTreeStore::SetItemExpandedIcon( const wxDataViewItem& item, const wxIcon &icon )
1412 {
1413     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1414     if (!node) return;
1415
1416     node->SetExpandedIcon( icon );
1417
1418     // notify control
1419     ValueChanged( item, 0 );
1420 }
1421
1422 const wxIcon &wxDataViewTreeStore::GetItemExpandedIcon( const wxDataViewItem& item ) const
1423 {
1424     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1425     if (!node) return wxNullIcon;
1426
1427     return node->GetExpandedIcon();
1428 }
1429
1430 void wxDataViewTreeStore::SetItemData( const wxDataViewItem& item, wxClientData *data )
1431 {
1432     wxDataViewTreeStoreNode *node = FindNode( item );
1433     if (!node) return;
1434
1435     node->SetData( data );
1436
1437     // notify control? only sensible when sorting on client data
1438     // ValueChanged( item, 0 );
1439 }
1440
1441 wxClientData *wxDataViewTreeStore::GetItemData( const wxDataViewItem& item ) const
1442 {
1443     wxDataViewTreeStoreNode *node = FindNode( item );
1444     if (!node) return NULL;
1445
1446     return node->GetData();
1447 }
1448
1449 void wxDataViewTreeStore::DeleteItem( const wxDataViewItem& item )
1450 {
1451     if (!item.IsOk()) return;
1452
1453     wxDataViewItem parent_item = GetParent( item );
1454
1455     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent_item );
1456     if (!parent_node) return;
1457
1458     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1459     if (!node) return;
1460
1461     parent_node->GetChildren().DeleteObject( node );
1462
1463     // notify control
1464     ItemDeleted( parent_item, item );
1465 }
1466
1467 void wxDataViewTreeStore::DeleteChildren( const wxDataViewItem& item )
1468 {
1469     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1470     if (!node) return;
1471
1472     wxDataViewItemArray array;
1473     wxDataViewTreeStoreNodeList::iterator iter;
1474     for (iter = node->GetChildren().begin(); iter != node->GetChildren().end(); iter++)
1475     {
1476         wxDataViewTreeStoreNode* child = *iter;
1477         array.Add( child->GetItem() );
1478     }
1479
1480     node->GetChildren().clear();
1481
1482     // notify control
1483     ItemsDeleted( item, array );
1484 }
1485
1486 void wxDataViewTreeStore::DeleteAllItems()
1487 {
1488     // TODO
1489 }
1490
1491 void
1492 wxDataViewTreeStore::GetValue(wxVariant &variant,
1493                               const wxDataViewItem &item,
1494                               unsigned int WXUNUSED(col)) const
1495 {
1496     // if (col != 0) return;
1497
1498     wxDataViewTreeStoreNode *node = FindNode( item );
1499     if (!node) return;
1500
1501     wxIcon icon( node->GetIcon());
1502     if (node->IsContainer())
1503     {
1504         wxDataViewTreeStoreContainerNode *container = (wxDataViewTreeStoreContainerNode*) node;
1505         if (container->IsExpanded() && container->GetExpandedIcon().IsOk())
1506            icon = container->GetExpandedIcon();
1507     }
1508
1509     wxDataViewIconText data( node->GetText(), icon );
1510
1511     variant << data;
1512 }
1513
1514 bool
1515 wxDataViewTreeStore::SetValue(const wxVariant& variant,
1516                               const wxDataViewItem& item,
1517                               unsigned int WXUNUSED(col))
1518 {
1519     // if (col != 0) return false;
1520
1521     wxDataViewTreeStoreNode *node = FindNode( item );
1522     if (!node) return false;
1523
1524     wxDataViewIconText data;
1525
1526     data << variant;
1527
1528     node->SetText( data.GetText() );
1529     node->SetIcon( data.GetIcon() );
1530
1531     return true;
1532 }
1533
1534 wxDataViewItem wxDataViewTreeStore::GetParent( const wxDataViewItem &item ) const
1535 {
1536     wxDataViewTreeStoreNode *node = FindNode( item );
1537     if (!node) return wxDataViewItem(0);
1538
1539     wxDataViewTreeStoreNode *parent = node->GetParent();
1540     if (!parent) return wxDataViewItem(0);
1541
1542     if (parent == m_root)
1543         return wxDataViewItem(0);
1544
1545     return parent->GetItem();
1546 }
1547
1548 bool wxDataViewTreeStore::IsContainer( const wxDataViewItem &item ) const
1549 {
1550     wxDataViewTreeStoreNode *node = FindNode( item );
1551     if (!node) return false;
1552
1553     return node->IsContainer();
1554 }
1555
1556 unsigned int wxDataViewTreeStore::GetChildren( const wxDataViewItem &item, wxDataViewItemArray &children ) const
1557 {
1558     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1559     if (!node) return 0;
1560
1561     wxDataViewTreeStoreNodeList::iterator iter;
1562     for (iter = node->GetChildren().begin(); iter != node->GetChildren().end(); iter++)
1563     {
1564         wxDataViewTreeStoreNode* child = *iter;
1565         children.Add( child->GetItem() );
1566     }
1567
1568     return node->GetChildren().GetCount();
1569 }
1570
1571 int wxDataViewTreeStore::Compare( const wxDataViewItem &item1, const wxDataViewItem &item2,
1572                          unsigned int WXUNUSED(column), bool WXUNUSED(ascending) )
1573 {
1574     wxDataViewTreeStoreNode *node1 = FindNode( item1 );
1575     wxDataViewTreeStoreNode *node2 = FindNode( item2 );
1576
1577     if (!node1 || !node2)
1578         return 0;
1579
1580     wxDataViewTreeStoreContainerNode* parent1 =
1581         (wxDataViewTreeStoreContainerNode*) node1->GetParent();
1582     wxDataViewTreeStoreContainerNode* parent2 =
1583         (wxDataViewTreeStoreContainerNode*) node2->GetParent();
1584
1585     if (parent1 != parent2)
1586     {
1587         wxLogError( wxT("Comparing items with different parent.") );
1588         return 0;
1589     }
1590
1591     if (node1->IsContainer() && !!node2->IsContainer())
1592         return 1;
1593
1594     if (node2->IsContainer() && !!node1->IsContainer())
1595         return -1;
1596
1597     return parent1->GetChildren().IndexOf( node1 ) - parent1->GetChildren().IndexOf( node2 );
1598 }
1599
1600 wxDataViewTreeStoreNode *wxDataViewTreeStore::FindNode( const wxDataViewItem &item ) const
1601 {
1602     if (!item.IsOk())
1603         return m_root;
1604
1605     return (wxDataViewTreeStoreNode*) item.GetID();
1606 }
1607
1608 wxDataViewTreeStoreContainerNode *wxDataViewTreeStore::FindContainerNode( const wxDataViewItem &item ) const
1609 {
1610     if (!item.IsOk())
1611         return (wxDataViewTreeStoreContainerNode*) m_root;
1612
1613     wxDataViewTreeStoreNode* node = (wxDataViewTreeStoreNode*) item.GetID();
1614
1615     if (!node->IsContainer())
1616         return NULL;
1617
1618     return (wxDataViewTreeStoreContainerNode*) node;
1619 }
1620
1621 //-----------------------------------------------------------------------------
1622 // wxDataViewTreeCtrl
1623 //-----------------------------------------------------------------------------
1624
1625 IMPLEMENT_DYNAMIC_CLASS(wxDataViewTreeCtrl,wxDataViewCtrl)
1626
1627 BEGIN_EVENT_TABLE(wxDataViewTreeCtrl,wxDataViewCtrl)
1628    EVT_DATAVIEW_ITEM_EXPANDED(-1, wxDataViewTreeCtrl::OnExpanded)
1629    EVT_DATAVIEW_ITEM_COLLAPSED(-1, wxDataViewTreeCtrl::OnCollapsed)
1630    EVT_SIZE( wxDataViewTreeCtrl::OnSize )
1631 END_EVENT_TABLE()
1632
1633 wxDataViewTreeCtrl::wxDataViewTreeCtrl()
1634 {
1635     m_imageList = NULL;
1636 }
1637
1638 wxDataViewTreeCtrl::wxDataViewTreeCtrl( wxWindow *parent, wxWindowID id,
1639            const wxPoint& pos, const wxSize& size, long style, const wxValidator& validator )
1640 {
1641     m_imageList = NULL;
1642     Create( parent, id, pos, size, style, validator );
1643
1644     wxDataViewTreeStore *store = new wxDataViewTreeStore;
1645     AssociateModel( store );
1646     store->DecRef();
1647
1648 #if !defined(__WXGTK20__) || defined(wxUSE_GENERICDATAVIEWCTRL)
1649     wxDataViewColumn *col = AppendIconTextColumn( "", 0, wxDATAVIEW_CELL_INERT, 40 );
1650
1651     wxSize cient_size = GetClientSize();
1652     col->SetWidth( size.x );
1653 #else
1654     AppendIconTextColumn( "", 0, wxDATAVIEW_CELL_INERT, 40 );
1655 #endif
1656 }
1657
1658 wxDataViewTreeCtrl::~wxDataViewTreeCtrl()
1659 {
1660     if (m_imageList)
1661         delete m_imageList;
1662 }
1663
1664 bool wxDataViewTreeCtrl::Create( wxWindow *parent, wxWindowID id,
1665            const wxPoint& pos, const wxSize& size, long style, const wxValidator& validator )
1666 {
1667     return wxDataViewCtrl::Create( parent, id, pos, size, style, validator );
1668 }
1669
1670 void wxDataViewTreeCtrl::SetImageList( wxImageList *imagelist )
1671 {
1672     if (m_imageList)
1673         delete m_imageList;
1674
1675     m_imageList = imagelist;
1676 }
1677
1678 wxDataViewItem wxDataViewTreeCtrl::AppendItem( const wxDataViewItem& parent,
1679         const wxString &text, int iconIndex, wxClientData *data )
1680 {
1681     wxIcon icon = wxNullIcon;
1682     if (m_imageList && (iconIndex != -1))
1683         icon = m_imageList->GetIcon( iconIndex );
1684
1685     return GetStore()->AppendItem( parent, text, icon, data );
1686 }
1687
1688 wxDataViewItem wxDataViewTreeCtrl::PrependItem( const wxDataViewItem& parent,
1689         const wxString &text, int iconIndex, wxClientData *data )
1690 {
1691     wxIcon icon = wxNullIcon;
1692     if (m_imageList && (iconIndex != -1))
1693         icon = m_imageList->GetIcon( iconIndex );
1694
1695     return GetStore()->PrependItem( parent, text, icon, data );
1696 }
1697
1698 wxDataViewItem wxDataViewTreeCtrl::InsertItem( const wxDataViewItem& parent, const wxDataViewItem& previous,
1699         const wxString &text, int iconIndex, wxClientData *data )
1700 {
1701     wxIcon icon = wxNullIcon;
1702     if (m_imageList && (iconIndex != -1))
1703         icon = m_imageList->GetIcon( iconIndex );
1704
1705     return GetStore()->InsertItem( parent, previous, text, icon, data );
1706 }
1707
1708 wxDataViewItem wxDataViewTreeCtrl::PrependContainer( const wxDataViewItem& parent,
1709         const wxString &text, int iconIndex, int expandedIndex, wxClientData *data )
1710 {
1711     wxIcon icon = wxNullIcon;
1712     if (m_imageList && (iconIndex != -1))
1713         icon = m_imageList->GetIcon( iconIndex );
1714
1715     wxIcon expanded = wxNullIcon;
1716     if (m_imageList && (expandedIndex != -1))
1717         expanded = m_imageList->GetIcon( expandedIndex );
1718
1719     return GetStore()->PrependContainer( parent, text, icon, expanded, data );
1720 }
1721
1722 wxDataViewItem wxDataViewTreeCtrl::AppendContainer( const wxDataViewItem& parent,
1723         const wxString &text, int iconIndex, int expandedIndex, wxClientData *data )
1724 {
1725     wxIcon icon = wxNullIcon;
1726     if (m_imageList && (iconIndex != -1))
1727         icon = m_imageList->GetIcon( iconIndex );
1728
1729     wxIcon expanded = wxNullIcon;
1730     if (m_imageList && (expandedIndex != -1))
1731         expanded = m_imageList->GetIcon( expandedIndex );
1732
1733     return GetStore()->AppendContainer( parent, text, icon, expanded, data );
1734 }
1735
1736 wxDataViewItem wxDataViewTreeCtrl::InsertContainer( const wxDataViewItem& parent, const wxDataViewItem& previous,
1737         const wxString &text, int iconIndex, int expandedIndex, wxClientData *data )
1738 {
1739     wxIcon icon = wxNullIcon;
1740     if (m_imageList && (iconIndex != -1))
1741         icon = m_imageList->GetIcon( iconIndex );
1742
1743     wxIcon expanded = wxNullIcon;
1744     if (m_imageList && (expandedIndex != -1))
1745         expanded = m_imageList->GetIcon( expandedIndex );
1746
1747     return GetStore()->InsertContainer( parent, previous, text, icon, expanded, data );
1748 }
1749
1750 void wxDataViewTreeCtrl::OnExpanded( wxDataViewEvent &event )
1751 {
1752     if (m_imageList) return;
1753
1754     wxDataViewTreeStoreContainerNode* container = GetStore()->FindContainerNode( event.GetItem() );
1755     if (!container) return;
1756
1757     container->SetExpanded( true );
1758     GetStore()->ItemChanged( event.GetItem() );
1759 }
1760
1761 void wxDataViewTreeCtrl::OnCollapsed( wxDataViewEvent &event )
1762 {
1763     if (m_imageList) return;
1764
1765     wxDataViewTreeStoreContainerNode* container = GetStore()->FindContainerNode( event.GetItem() );
1766     if (!container) return;
1767
1768     container->SetExpanded( false );
1769     GetStore()->ItemChanged( event.GetItem() );
1770 }
1771
1772 void wxDataViewTreeCtrl::OnSize( wxSizeEvent &event )
1773 {
1774 #if !defined(__WXGTK20__) || defined(wxUSE_GENERICDATAVIEWCTRL)
1775     wxSize size = GetClientSize();
1776     wxDataViewColumn *col = GetColumn( 0 );
1777     if (col) col->SetWidth( size.x );
1778 #endif
1779     event.Skip( true );
1780 }
1781
1782 #endif // wxUSE_DATAVIEWCTRL
1783