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 #ifndef WX_PRECOMP
  23     #include "wx/dc.h"
  24     #include "wx/settings.h"
  25     #include "wx/log.h"
  26     #include "wx/crt.h"
  27 #endif
  28
  29 #include "wx/spinctrl.h"
  30 #include "wx/choice.h"
  31 #include "wx/imaglist.h"
  32
  33 const char wxDataViewCtrlNameStr[] = "dataviewCtrl";
  34
  35 namespace
  36 {
  37
  38 // Custom handler pushed on top of the edit control used by wxDataViewCtrl to
  39 // forward some events to the main control itself.
  40 class wxDataViewEditorCtrlEvtHandler: public wxEvtHandler
  41 {
  42 public:
  43     wxDataViewEditorCtrlEvtHandler(wxControl *editor, wxDataViewRenderer *owner)
  44     {
  45         m_editorCtrl = editor;
  46         m_owner = owner;
  47
  48         m_finished = false;
  49     }
  50
  51     void AcceptChangesAndFinish();
  52     void SetFocusOnIdle( bool focus = true ) { m_focusOnIdle = focus; }
  53
  54 protected:
  55     void OnChar( wxKeyEvent &event );
  56     void OnTextEnter( wxCommandEvent &event );
  57     void OnKillFocus( wxFocusEvent &event );
  58     void OnIdle( wxIdleEvent &event );
  59
  60 private:
  61     wxDataViewRenderer     *m_owner;
  62     wxControl              *m_editorCtrl;
  63     bool                    m_finished;
  64     bool                    m_focusOnIdle;
  65
  66 private:
  67     DECLARE_EVENT_TABLE()
  68 };
  69
  70 } // anonymous namespace
  71
  72 // ---------------------------------------------------------
  73 // wxDataViewModelNotifier
  74 // ---------------------------------------------------------
  75
  76 #include "wx/listimpl.cpp"
  77 WX_DEFINE_LIST(wxDataViewModelNotifiers)
  78
  79 bool wxDataViewModelNotifier::ItemsAdded( const wxDataViewItem &parent, const wxDataViewItemArray &items )
  80 {
  81     size_t count = items.GetCount();
  82     size_t i;
  83     for (i = 0; i < count; i++)
  84         if (!ItemAdded( parent, items[i] )) return false;
  85
  86     return true;
  87 }
  88
  89 bool wxDataViewModelNotifier::ItemsDeleted( const wxDataViewItem &parent, const wxDataViewItemArray &items )
  90 {
  91     size_t count = items.GetCount();
  92     size_t i;
  93     for (i = 0; i < count; i++)
  94         if (!ItemDeleted( parent, items[i] )) return false;
  95
  96     return true;
  97 }
  98
  99 bool wxDataViewModelNotifier::ItemsChanged( const wxDataViewItemArray &items )
 100 {
 101     size_t count = items.GetCount();
 102     size_t i;
 103     for (i = 0; i < count; i++)
 104         if (!ItemChanged( items[i] )) return false;
 105
 106     return true;
 107 }
 108
 109 // ---------------------------------------------------------
 110 // wxDataViewModel
 111 // ---------------------------------------------------------
 112
 113 wxDataViewModel::wxDataViewModel()
 114 {
 115     m_notifiers.DeleteContents( true );
 116 }
 117
 118 bool wxDataViewModel::ItemAdded( const wxDataViewItem &parent, const wxDataViewItem &item )
 119 {
 120     bool ret = true;
 121
 122     wxDataViewModelNotifiers::iterator iter;
 123     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 124     {
 125         wxDataViewModelNotifier* notifier = *iter;
 126         if (!notifier->ItemAdded( parent, item ))
 127             ret = false;
 128     }
 129
 130     return ret;
 131 }
 132
 133 bool wxDataViewModel::ItemDeleted( const wxDataViewItem &parent, const wxDataViewItem &item )
 134 {
 135     bool ret = true;
 136
 137     wxDataViewModelNotifiers::iterator iter;
 138     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 139     {
 140         wxDataViewModelNotifier* notifier = *iter;
 141         if (!notifier->ItemDeleted( parent, item ))
 142             ret = false;
 143     }
 144
 145     return ret;
 146 }
 147
 148 bool wxDataViewModel::ItemChanged( const wxDataViewItem &item )
 149 {
 150     bool ret = true;
 151
 152     wxDataViewModelNotifiers::iterator iter;
 153     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 154     {
 155         wxDataViewModelNotifier* notifier = *iter;
 156         if (!notifier->ItemChanged( item ))
 157             ret = false;
 158     }
 159
 160     return ret;
 161 }
 162
 163 bool wxDataViewModel::ItemsAdded( const wxDataViewItem &parent, const wxDataViewItemArray &items )
 164 {
 165     bool ret = true;
 166
 167     wxDataViewModelNotifiers::iterator iter;
 168     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 169     {
 170         wxDataViewModelNotifier* notifier = *iter;
 171         if (!notifier->ItemsAdded( parent, items ))
 172             ret = false;
 173     }
 174
 175     return ret;
 176 }
 177
 178 bool wxDataViewModel::ItemsDeleted( const wxDataViewItem &parent, const wxDataViewItemArray &items )
 179 {
 180     bool ret = true;
 181
 182     wxDataViewModelNotifiers::iterator iter;
 183     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 184     {
 185         wxDataViewModelNotifier* notifier = *iter;
 186         if (!notifier->ItemsDeleted( parent, items ))
 187             ret = false;
 188     }
 189
 190     return ret;
 191 }
 192
 193 bool wxDataViewModel::ItemsChanged( const wxDataViewItemArray &items )
 194 {
 195     bool ret = true;
 196
 197     wxDataViewModelNotifiers::iterator iter;
 198     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 199     {
 200         wxDataViewModelNotifier* notifier = *iter;
 201         if (!notifier->ItemsChanged( items ))
 202             ret = false;
 203     }
 204
 205     return ret;
 206 }
 207
 208 bool wxDataViewModel::ValueChanged( const wxDataViewItem &item, unsigned int col )
 209 {
 210     bool ret = true;
 211
 212     wxDataViewModelNotifiers::iterator iter;
 213     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 214     {
 215         wxDataViewModelNotifier* notifier = *iter;
 216         if (!notifier->ValueChanged( item, col ))
 217             ret = false;
 218     }
 219
 220     return ret;
 221 }
 222
 223 bool wxDataViewModel::Cleared()
 224 {
 225     bool ret = true;
 226
 227     wxDataViewModelNotifiers::iterator iter;
 228     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 229     {
 230         wxDataViewModelNotifier* notifier = *iter;
 231         if (!notifier->Cleared())
 232             ret = false;
 233     }
 234
 235     return ret;
 236 }
 237
 238 void wxDataViewModel::Resort()
 239 {
 240     wxDataViewModelNotifiers::iterator iter;
 241     for (iter = m_notifiers.begin(); iter != m_notifiers.end(); ++iter)
 242     {
 243         wxDataViewModelNotifier* notifier = *iter;
 244         notifier->Resort();
 245     }
 246 }
 247
 248 void wxDataViewModel::AddNotifier( wxDataViewModelNotifier *notifier )
 249 {
 250     m_notifiers.push_back( notifier );
 251     notifier->SetOwner( this );
 252 }
 253
 254 void wxDataViewModel::RemoveNotifier( wxDataViewModelNotifier *notifier )
 255 {
 256     m_notifiers.DeleteObject( notifier );
 257 }
 258
 259 int wxDataViewModel::Compare( const wxDataViewItem &item1, const wxDataViewItem &item2,
 260                               unsigned int column, bool ascending ) const
 261 {
 262     // sort branches before leaves
 263     bool item1_is_container = IsContainer(item1);
 264     bool item2_is_container = IsContainer(item2);
 265
 266     if (item1_is_container && !item2_is_container)
 267         return 1;
 268     if (item2_is_container && !item1_is_container)
 269         return -1;
 270
 271     wxVariant value1,value2;
 272     GetValue( value1, item1, column );
 273     GetValue( value2, item2, column );
 274
 275     if (!ascending)
 276     {
 277         wxVariant temp = value1;
 278         value1 = value2;
 279         value2 = temp;
 280     }
 281
 282     if (value1.GetType() == wxT("string"))
 283     {
 284         wxString str1 = value1.GetString();
 285         wxString str2 = value2.GetString();
 286         int res = str1.Cmp( str2 );
 287         if (res)
 288             return res;
 289     }
 290     else if (value1.GetType() == wxT("long"))
 291     {
 292         long l1 = value1.GetLong();
 293         long l2 = value2.GetLong();
 294         long res = l1-l2;
 295         if (res)
 296             return res;
 297     }
 298     else if (value1.GetType() == wxT("double"))
 299     {
 300         double d1 = value1.GetDouble();
 301         double d2 = value2.GetDouble();
 302         if (d1 < d2)
 303             return 1;
 304         if (d1 > d2)
 305             return -1;
 306     }
 307     else if (value1.GetType() == wxT("datetime"))
 308     {
 309         wxDateTime dt1 = value1.GetDateTime();
 310         wxDateTime dt2 = value2.GetDateTime();
 311         if (dt1.IsEarlierThan(dt2))
 312             return 1;
 313         if (dt2.IsEarlierThan(dt1))
 314             return -1;
 315     }
 316
 317     // items must be different
 318     wxUIntPtr id1 = wxPtrToUInt(item1.GetID()),
 319               id2 = wxPtrToUInt(item2.GetID());
 320
 321     return ascending ? id1 - id2 : id2 - id1;
 322 }
 323
 324 // ---------------------------------------------------------
 325 // wxDataViewIndexListModel
 326 // ---------------------------------------------------------
 327
 328 static int my_sort( int *v1, int *v2 )
 329 {
 330    return *v2-*v1;
 331 }
 332
 333
 334 wxDataViewIndexListModel::wxDataViewIndexListModel( unsigned int initial_size )
 335 {
 336     // IDs are ordered until an item gets deleted or inserted
 337     m_ordered = true;
 338
 339     // build initial index
 340     unsigned int i;
 341     for (i = 1; i < initial_size+1; i++)
 342             m_hash.Add( wxUIntToPtr(i) );
 343     m_nextFreeID = initial_size + 1;
 344 }
 345
 346 void wxDataViewIndexListModel::Reset( unsigned int new_size )
 347 {
 348     m_hash.Clear();
 349
 350     // IDs are ordered until an item gets deleted or inserted
 351     m_ordered = true;
 352
 353     // build initial index
 354     unsigned int i;
 355     for (i = 1; i < new_size+1; i++)
 356             m_hash.Add( wxUIntToPtr(i) );
 357
 358     m_nextFreeID = new_size + 1;
 359
 360     /* wxDataViewModel:: */ Cleared();
 361 }
 362
 363 void wxDataViewIndexListModel::RowPrepended()
 364 {
 365     m_ordered = false;
 366
 367     unsigned int id = m_nextFreeID;
 368     m_nextFreeID++;
 369
 370     m_hash.Insert( wxUIntToPtr(id), 0 );
 371     wxDataViewItem item( wxUIntToPtr(id) );
 372     ItemAdded( wxDataViewItem(0), item );
 373
 374 }
 375
 376 void wxDataViewIndexListModel::RowInserted( unsigned int before )
 377 {
 378     m_ordered = false;
 379
 380     unsigned int id = m_nextFreeID;
 381     m_nextFreeID++;
 382
 383     m_hash.Insert( wxUIntToPtr(id), before );
 384     wxDataViewItem item( wxUIntToPtr(id) );
 385     ItemAdded( wxDataViewItem(0), item );
 386 }
 387
 388 void wxDataViewIndexListModel::RowAppended()
 389 {
 390     unsigned int id = m_nextFreeID;
 391     m_nextFreeID++;
 392
 393     m_hash.Add( wxUIntToPtr(id) );
 394     wxDataViewItem item( wxUIntToPtr(id) );
 395     ItemAdded( wxDataViewItem(0), item );
 396 }
 397
 398 void wxDataViewIndexListModel::RowDeleted( unsigned int row )
 399 {
 400     m_ordered = false;
 401
 402     wxDataViewItem item( m_hash[row] );
 403     /* wxDataViewModel:: */ ItemDeleted( wxDataViewItem(0), item );
 404     m_hash.RemoveAt( row );
 405 }
 406
 407 void wxDataViewIndexListModel::RowsDeleted( const wxArrayInt &rows )
 408 {
 409     wxArrayInt sorted = rows;
 410     sorted.Sort( my_sort );
 411
 412     m_ordered = false;
 413
 414     wxDataViewItemArray array;
 415     unsigned int i;
 416     for (i = 0; i < rows.GetCount(); i++)
 417     {
 418             wxDataViewItem item( m_hash[rows[i]] );
 419             array.Add( item );
 420     }
 421     /* wxDataViewModel:: */ ItemsDeleted( wxDataViewItem(0), array );
 422
 423     for (i = 0; i < sorted.GetCount(); i++)
 424            m_hash.RemoveAt( sorted[i] );
 425 }
 426
 427 void wxDataViewIndexListModel::RowChanged( unsigned int row )
 428 {
 429     /* wxDataViewModel:: */ ItemChanged( GetItem(row) );
 430 }
 431
 432 void wxDataViewIndexListModel::RowValueChanged( unsigned int row, unsigned int col )
 433 {
 434     /* wxDataViewModel:: */ ValueChanged( GetItem(row), col );
 435 }
 436
 437 unsigned int wxDataViewIndexListModel::GetRow( const wxDataViewItem &item ) const
 438 {
 439     if (m_ordered)
 440         return wxPtrToUInt(item.GetID())-1;
 441
 442     // assert for not found
 443     return (unsigned int) m_hash.Index( item.GetID() );
 444 }
 445
 446 wxDataViewItem wxDataViewIndexListModel::GetItem( unsigned int row ) const
 447 {
 448     wxASSERT( row < m_hash.GetCount() );
 449     return wxDataViewItem( m_hash[row] );
 450 }
 451
 452 bool wxDataViewIndexListModel::HasDefaultCompare() const
 453 {
 454     return !m_ordered;
 455 }
 456
 457 int wxDataViewIndexListModel::Compare(const wxDataViewItem& item1,
 458                                       const wxDataViewItem& item2,
 459                                       unsigned int WXUNUSED(column),
 460                                       bool ascending) const
 461 {
 462     if (m_ordered)
 463     {
 464         unsigned int pos1 = wxPtrToUInt(item1.GetID());  // -1 not needed here
 465         unsigned int pos2 = wxPtrToUInt(item2.GetID());  // -1 not needed here
 466
 467         if (ascending)
 468             return pos1 - pos2;
 469         else
 470             return pos2 - pos1;
 471     }
 472
 473     if (ascending)
 474         return GetRow(item1) - GetRow(item2);
 475
 476     return GetRow(item2) - GetRow(item1);
 477 }
 478
 479 unsigned int wxDataViewIndexListModel::GetChildren( const wxDataViewItem &item, wxDataViewItemArray &children ) const
 480 {
 481     if (item.IsOk())
 482         return 0;
 483
 484     children = m_hash;
 485
 486     return m_hash.GetCount();
 487 }
 488
 489 // ---------------------------------------------------------
 490 // wxDataViewVirtualListModel
 491 // ---------------------------------------------------------
 492
 493 #ifndef __WXMAC__
 494
 495 wxDataViewVirtualListModel::wxDataViewVirtualListModel( unsigned int initial_size )
 496 {
 497     m_size = initial_size;
 498 }
 499
 500 void wxDataViewVirtualListModel::Reset( unsigned int new_size )
 501 {
 502     m_size = new_size;
 503
 504     /* wxDataViewModel:: */ Cleared();
 505 }
 506
 507 void wxDataViewVirtualListModel::RowPrepended()
 508 {
 509     m_size++;
 510     wxDataViewItem item( wxUIntToPtr(1) );
 511     ItemAdded( wxDataViewItem(0), item );
 512 }
 513
 514 void wxDataViewVirtualListModel::RowInserted( unsigned int before )
 515 {
 516     m_size++;
 517     wxDataViewItem item( wxUIntToPtr(before+1) );
 518     ItemAdded( wxDataViewItem(0), item );
 519 }
 520
 521 void wxDataViewVirtualListModel::RowAppended()
 522 {
 523     m_size++;
 524     wxDataViewItem item( wxUIntToPtr(m_size) );
 525     ItemAdded( wxDataViewItem(0), item );
 526 }
 527
 528 void wxDataViewVirtualListModel::RowDeleted( unsigned int row )
 529 {
 530     m_size--;
 531     wxDataViewItem item( wxUIntToPtr(row+1) );
 532     /* wxDataViewModel:: */ ItemDeleted( wxDataViewItem(0), item );
 533 }
 534
 535 void wxDataViewVirtualListModel::RowsDeleted( const wxArrayInt &rows )
 536 {
 537     m_size -= rows.GetCount();
 538
 539     wxArrayInt sorted = rows;
 540     sorted.Sort( my_sort );
 541
 542     wxDataViewItemArray array;
 543     unsigned int i;
 544     for (i = 0; i < sorted.GetCount(); i++)
 545     {
 546         wxDataViewItem item( wxUIntToPtr(sorted[i]+1) );
 547         array.Add( item );
 548     }
 549     /* wxDataViewModel:: */ ItemsDeleted( wxDataViewItem(0), array );
 550 }
 551
 552 void wxDataViewVirtualListModel::RowChanged( unsigned int row )
 553 {
 554     /* wxDataViewModel:: */ ItemChanged( GetItem(row) );
 555 }
 556
 557 void wxDataViewVirtualListModel::RowValueChanged( unsigned int row, unsigned int col )
 558 {
 559     /* wxDataViewModel:: */ ValueChanged( GetItem(row), col );
 560 }
 561
 562 unsigned int wxDataViewVirtualListModel::GetRow( const wxDataViewItem &item ) const
 563 {
 564     return wxPtrToUInt( item.GetID() ) -1;
 565 }
 566
 567 wxDataViewItem wxDataViewVirtualListModel::GetItem( unsigned int row ) const
 568 {
 569     return wxDataViewItem( wxUIntToPtr(row+1) );
 570 }
 571
 572 bool wxDataViewVirtualListModel::HasDefaultCompare() const
 573 {
 574     return true;
 575 }
 576
 577 int wxDataViewVirtualListModel::Compare(const wxDataViewItem& item1,
 578                                       const wxDataViewItem& item2,
 579                                       unsigned int WXUNUSED(column),
 580                                       bool ascending) const
 581 {
 582     unsigned int pos1 = wxPtrToUInt(item1.GetID());  // -1 not needed here
 583     unsigned int pos2 = wxPtrToUInt(item2.GetID());  // -1 not needed here
 584
 585     if (ascending)
 586        return pos1 - pos2;
 587     else
 588        return pos2 - pos1;
 589 }
 590
 591 unsigned int wxDataViewVirtualListModel::GetChildren( const wxDataViewItem &WXUNUSED(item), wxDataViewItemArray &WXUNUSED(children) ) const
 592 {
 593     return 0;  // should we report an error ?
 594 }
 595
 596 #endif  // __WXMAC__
 597
 598 //-----------------------------------------------------------------------------
 599 // wxDataViewIconText
 600 //-----------------------------------------------------------------------------
 601
 602 IMPLEMENT_DYNAMIC_CLASS(wxDataViewIconText,wxObject)
 603
 604 IMPLEMENT_VARIANT_OBJECT_EXPORTED(wxDataViewIconText, WXDLLIMPEXP_ADV)
 605
 606 // ---------------------------------------------------------
 607 // wxDataViewRendererBase
 608 // ---------------------------------------------------------
 609
 610 IMPLEMENT_ABSTRACT_CLASS(wxDataViewRendererBase, wxObject)
 611
 612 wxDataViewRendererBase::wxDataViewRendererBase( const wxString &varianttype,
 613                                                 wxDataViewCellMode WXUNUSED(mode),
 614                                                 int WXUNUSED(align) )
 615 {
 616     m_variantType = varianttype;
 617     m_owner = NULL;
 618 }
 619
 620 wxDataViewRendererBase::~wxDataViewRendererBase()
 621 {
 622 }
 623
 624 const wxDataViewCtrl* wxDataViewRendererBase::GetView() const
 625 {
 626     return const_cast<wxDataViewRendererBase*>(this)->GetOwner()->GetOwner();
 627 }
 628
 629 bool wxDataViewRendererBase::StartEditing( const wxDataViewItem &item, wxRect labelRect )
 630 {
 631     wxDataViewCtrl* dv_ctrl = GetOwner()->GetOwner();
 632
 633     // Before doing anything we send an event asking if editing of this item is really wanted.
 634     wxDataViewEvent start_event( wxEVT_COMMAND_DATAVIEW_ITEM_START_EDITING, dv_ctrl->GetId() );
 635     start_event.SetDataViewColumn( GetOwner() );
 636     start_event.SetModel( dv_ctrl->GetModel() );
 637     start_event.SetItem( item );
 638     start_event.SetEventObject( dv_ctrl );
 639     dv_ctrl->GetEventHandler()->ProcessEvent( start_event );
 640     if( !start_event.IsAllowed() )
 641         return false;
 642
 643     m_item = item; // remember for later
 644
 645     unsigned int col = GetOwner()->GetModelColumn();
 646     wxVariant value;
 647     dv_ctrl->GetModel()->GetValue( value, item, col );
 648
 649     m_editorCtrl = CreateEditorCtrl( dv_ctrl->GetMainWindow(), labelRect, value );
 650
 651     // there might be no editor control for the given item
 652     if(!m_editorCtrl)
 653         return false;
 654
 655     wxDataViewEditorCtrlEvtHandler *handler =
 656         new wxDataViewEditorCtrlEvtHandler( m_editorCtrl, (wxDataViewRenderer*) this );
 657
 658     m_editorCtrl->PushEventHandler( handler );
 659
 660 #if defined(__WXGTK20__) && !defined(wxUSE_GENERICDATAVIEWCTRL)
 661     handler->SetFocusOnIdle();
 662 #else
 663     m_editorCtrl->SetFocus();
 664 #endif
 665
 666     // Now we should send Editing Started event
 667     wxDataViewEvent event( wxEVT_COMMAND_DATAVIEW_ITEM_EDITING_STARTED, dv_ctrl->GetId() );
 668     event.SetDataViewColumn( GetOwner() );
 669     event.SetModel( dv_ctrl->GetModel() );
 670     event.SetItem( item );
 671     event.SetEventObject( dv_ctrl );
 672     dv_ctrl->GetEventHandler()->ProcessEvent( event );
 673
 674     return true;
 675 }
 676
 677 void wxDataViewRendererBase::DestroyEditControl()
 678 {
 679     // Hide the control immediately but don't delete it yet as there could be
 680     // some pending messages for it.
 681     m_editorCtrl->Hide();
 682
 683     wxEvtHandler * const handler = m_editorCtrl->PopEventHandler();
 684
 685     wxPendingDelete.Append(handler);
 686     wxPendingDelete.Append(m_editorCtrl);
 687 }
 688
 689 void wxDataViewRendererBase::CancelEditing()
 690 {
 691     if (!m_editorCtrl)
 692         return;
 693
 694     GetOwner()->GetOwner()->GetMainWindow()->SetFocus();
 695
 696     DestroyEditControl();
 697 }
 698
 699 bool wxDataViewRendererBase::FinishEditing()
 700 {
 701     if (!m_editorCtrl)
 702         return true;
 703
 704     wxVariant value;
 705     GetValueFromEditorCtrl( m_editorCtrl, value );
 706
 707     wxDataViewCtrl* dv_ctrl = GetOwner()->GetOwner();
 708
 709     dv_ctrl->GetMainWindow()->SetFocus();
 710
 711     DestroyEditControl();
 712
 713     if (!Validate(value))
 714         return false;
 715
 716     unsigned int col = GetOwner()->GetModelColumn();
 717     dv_ctrl->GetModel()->ChangeValue(value, m_item, col);
 718
 719     // Now we should send Editing Done event
 720     wxDataViewEvent event( wxEVT_COMMAND_DATAVIEW_ITEM_EDITING_DONE, dv_ctrl->GetId() );
 721     event.SetDataViewColumn( GetOwner() );
 722     event.SetModel( dv_ctrl->GetModel() );
 723     event.SetItem( m_item );
 724     event.SetEventObject( dv_ctrl );
 725     dv_ctrl->GetEventHandler()->ProcessEvent( event );
 726
 727     return true;
 728 }
 729
 730 // ----------------------------------------------------------------------------
 731 // wxDataViewCustomRendererBase
 732 // ----------------------------------------------------------------------------
 733
 734 void
 735 wxDataViewCustomRendererBase::WXCallRender(wxRect rectCell, wxDC *dc, int state)
 736 {
 737     wxCHECK_RET( dc, "no DC to draw on in custom renderer?" );
 738
 739     // adjust the rectangle ourselves to account for the alignment
 740     wxRect rectItem = rectCell;
 741     const int align = GetAlignment();
 742     if ( align != wxDVR_DEFAULT_ALIGNMENT )
 743     {
 744         const wxSize size = GetSize();
 745
 746         // take alignment into account only if there is enough space, otherwise
 747         // show as much contents as possible
 748         //
 749         // notice that many existing renderers (e.g. wxDataViewSpinRenderer)
 750         // return hard-coded size which can be more than they need and if we
 751         // trusted their GetSize() we'd draw the text out of cell bounds
 752         // entirely
 753
 754         if ( size.x >= 0 && size.x < rectCell.width )
 755         {
 756             if ( align & wxALIGN_CENTER_HORIZONTAL )
 757                 rectItem.x += (rectCell.width - size.x)/2;
 758             else if ( align & wxALIGN_RIGHT )
 759                 rectItem.x += rectCell.width - size.x;
 760             // else: wxALIGN_LEFT is the default
 761
 762             rectItem.width = size.x;
 763         }
 764
 765         if ( size.y >= 0 && size.y < rectCell.height )
 766         {
 767             if ( align & wxALIGN_CENTER_VERTICAL )
 768                 rectItem.y += (rectCell.height - size.y)/2;
 769             else if ( align & wxALIGN_BOTTOM )
 770                 rectItem.y += rectCell.height - size.y;
 771             // else: wxALIGN_TOP is the default
 772
 773             rectItem.height = size.y;
 774         }
 775     }
 776
 777
 778     // set up the DC attributes
 779
 780     // override custom foreground with the standard one for the selected items
 781     // because we currently don't allow changing the selection background and
 782     // custom colours may be unreadable on it
 783     wxColour col;
 784     if ( state & wxDATAVIEW_CELL_SELECTED )
 785         col = wxSystemSettings::GetColour(wxSYS_COLOUR_HIGHLIGHTTEXT);
 786     else if ( m_attr.HasColour() )
 787         col = m_attr.GetColour();
 788     else // use default foreground
 789         col = GetOwner()->GetOwner()->GetForegroundColour();
 790
 791     wxDCTextColourChanger changeFg(*dc, col);
 792
 793     wxDCFontChanger changeFont(*dc);
 794     if ( m_attr.HasFont() )
 795     {
 796         wxFont font(dc->GetFont());
 797         if ( m_attr.GetBold() )
 798             font.MakeBold();
 799         if ( m_attr.GetItalic() )
 800             font.MakeItalic();
 801
 802         changeFont.Set(font);
 803     }
 804
 805     Render(rectItem, dc, state);
 806 }
 807
 808 void
 809 wxDataViewCustomRendererBase::RenderText(const wxString& text,
 810                                          int xoffset,
 811                                          wxRect rect,
 812                                          wxDC *dc,
 813                                          int WXUNUSED(state))
 814 {
 815     wxRect rectText = rect;
 816     rectText.x += xoffset;
 817     rectText.width -= xoffset;
 818
 819     // check if we want to ellipsize the text if it doesn't fit
 820     wxString ellipsizedText;
 821     if ( GetEllipsizeMode() != wxELLIPSIZE_NONE )
 822     {
 823         ellipsizedText = wxControl::Ellipsize
 824                                     (
 825                                         text,
 826                                         *dc,
 827                                         GetEllipsizeMode(),
 828                                         rectText.width,
 829                                         wxELLIPSIZE_FLAGS_NONE
 830                                     );
 831     }
 832
 833     // get the alignment to use
 834     int align = GetAlignment();
 835     if ( align == wxDVR_DEFAULT_ALIGNMENT )
 836     {
 837         // if we don't have an explicit alignment ourselves, use that of the
 838         // column in horizontal direction and default vertical alignment
 839         align = GetOwner()->GetAlignment() | wxALIGN_CENTRE_VERTICAL;
 840     }
 841
 842     dc->DrawLabel(ellipsizedText.empty() ? text : ellipsizedText,
 843                   rectText, align);
 844 }
 845
 846 //-----------------------------------------------------------------------------
 847 // wxDataViewEditorCtrlEvtHandler
 848 //-----------------------------------------------------------------------------
 849
 850 BEGIN_EVENT_TABLE(wxDataViewEditorCtrlEvtHandler, wxEvtHandler)
 851     EVT_CHAR           (wxDataViewEditorCtrlEvtHandler::OnChar)
 852     EVT_KILL_FOCUS     (wxDataViewEditorCtrlEvtHandler::OnKillFocus)
 853     EVT_IDLE           (wxDataViewEditorCtrlEvtHandler::OnIdle)
 854     EVT_TEXT_ENTER     (-1, wxDataViewEditorCtrlEvtHandler::OnTextEnter)
 855 END_EVENT_TABLE()
 856
 857 void wxDataViewEditorCtrlEvtHandler::OnIdle( wxIdleEvent &event )
 858 {
 859     if (m_focusOnIdle)
 860     {
 861         m_focusOnIdle = false;
 862         if (wxWindow::FindFocus() != m_editorCtrl)
 863             m_editorCtrl->SetFocus();
 864     }
 865
 866     event.Skip();
 867 }
 868
 869 void wxDataViewEditorCtrlEvtHandler::OnTextEnter( wxCommandEvent &WXUNUSED(event) )
 870 {
 871     m_finished = true;
 872     m_owner->FinishEditing();
 873 }
 874
 875 void wxDataViewEditorCtrlEvtHandler::OnChar( wxKeyEvent &event )
 876 {
 877     switch ( event.m_keyCode )
 878     {
 879         case WXK_RETURN:
 880             m_finished = true;
 881             m_owner->FinishEditing();
 882             break;
 883
 884         case WXK_ESCAPE:
 885         {
 886             m_finished = true;
 887             m_owner->CancelEditing();
 888             break;
 889         }
 890         default:
 891             event.Skip();
 892     }
 893 }
 894
 895 void wxDataViewEditorCtrlEvtHandler::OnKillFocus( wxFocusEvent &event )
 896 {
 897     if (!m_finished)
 898     {
 899         m_finished = true;
 900         m_owner->FinishEditing();
 901     }
 902
 903     event.Skip();
 904 }
 905
 906 // ---------------------------------------------------------
 907 // wxDataViewColumnBase
 908 // ---------------------------------------------------------
 909
 910 void wxDataViewColumnBase::Init(wxDataViewRenderer *renderer,
 911                                 unsigned int model_column)
 912 {
 913     m_renderer = renderer;
 914     m_model_column = model_column;
 915     m_owner = NULL;
 916     m_renderer->SetOwner( (wxDataViewColumn*) this );
 917 }
 918
 919 wxDataViewColumnBase::~wxDataViewColumnBase()
 920 {
 921     delete m_renderer;
 922 }
 923
 924 // ---------------------------------------------------------
 925 // wxDataViewCtrlBase
 926 // ---------------------------------------------------------
 927
 928 IMPLEMENT_ABSTRACT_CLASS(wxDataViewCtrlBase, wxControl)
 929
 930 wxDataViewCtrlBase::wxDataViewCtrlBase()
 931 {
 932     m_model = NULL;
 933     m_expander_column = 0;
 934     m_indent = 8;
 935 }
 936
 937 wxDataViewCtrlBase::~wxDataViewCtrlBase()
 938 {
 939     if (m_model)
 940     {
 941         m_model->DecRef();
 942         m_model = NULL;
 943     }
 944 }
 945
 946 bool wxDataViewCtrlBase::AssociateModel( wxDataViewModel *model )
 947 {
 948     if (m_model)
 949     {
 950         m_model->DecRef();   // discard old model, if any
 951     }
 952
 953     // add our own reference to the new model:
 954     m_model = model;
 955     if (m_model)
 956     {
 957         m_model->IncRef();
 958     }
 959
 960     return true;
 961 }
 962
 963 wxDataViewModel* wxDataViewCtrlBase::GetModel()
 964 {
 965     return m_model;
 966 }
 967
 968 const wxDataViewModel* wxDataViewCtrlBase::GetModel() const
 969 {
 970     return m_model;
 971 }
 972
 973 void wxDataViewCtrlBase::ExpandAncestors( const wxDataViewItem & item )
 974 {
 975     if (!m_model) return;
 976
 977     if (!item.IsOk()) return;
 978
 979     wxVector<wxDataViewItem> parentChain;
 980
 981     // at first we get all the parents of the selected item
 982     wxDataViewItem parent = m_model->GetParent(item);
 983     while (parent.IsOk())
 984     {
 985         parentChain.push_back(parent);
 986         parent = m_model->GetParent(parent);
 987     }
 988
 989     // then we expand the parents, starting at the root
 990     while (!parentChain.empty())
 991     {
 992          Expand(parentChain.back());
 993          parentChain.pop_back();
 994     }
 995 }
 996
 997 wxDataViewColumn *
 998 wxDataViewCtrlBase::AppendTextColumn( const wxString &label, unsigned int model_column,
 999                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1000 {
1001     wxDataViewColumn *ret = new wxDataViewColumn( label,
1002         new wxDataViewTextRenderer( wxT("string"), mode ),
1003         model_column, width, align, flags );
1004     AppendColumn( ret );
1005     return ret;
1006 }
1007
1008 wxDataViewColumn *
1009 wxDataViewCtrlBase::AppendIconTextColumn( const wxString &label, unsigned int model_column,
1010                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1011 {
1012     wxDataViewColumn *ret = new wxDataViewColumn( label,
1013         new wxDataViewIconTextRenderer( wxT("wxDataViewIconText"), mode ),
1014         model_column, width, align, flags );
1015     AppendColumn( ret );
1016     return ret;
1017 }
1018
1019 wxDataViewColumn *
1020 wxDataViewCtrlBase::AppendToggleColumn( const wxString &label, unsigned int model_column,
1021                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1022 {
1023     wxDataViewColumn *ret = new wxDataViewColumn( label,
1024         new wxDataViewToggleRenderer( wxT("bool"), mode ),
1025         model_column, width, align, flags );
1026     AppendColumn( ret );
1027     return ret;
1028 }
1029
1030 wxDataViewColumn *
1031 wxDataViewCtrlBase::AppendProgressColumn( const wxString &label, unsigned int model_column,
1032                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1033 {
1034     wxDataViewColumn *ret = new wxDataViewColumn( label,
1035         new wxDataViewProgressRenderer( wxEmptyString, wxT("long"), mode ),
1036         model_column, width, align, flags );
1037     AppendColumn( ret );
1038     return ret;
1039 }
1040
1041 wxDataViewColumn *
1042 wxDataViewCtrlBase::AppendDateColumn( const wxString &label, unsigned int model_column,
1043                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1044 {
1045     wxDataViewColumn *ret = new wxDataViewColumn( label,
1046         new wxDataViewDateRenderer( wxT("datetime"), mode ),
1047         model_column, width, align, flags );
1048     AppendColumn( ret );
1049     return ret;
1050 }
1051
1052 wxDataViewColumn *
1053 wxDataViewCtrlBase::AppendBitmapColumn( const wxString &label, unsigned int model_column,
1054                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1055 {
1056     wxDataViewColumn *ret = new wxDataViewColumn( label,
1057         new wxDataViewBitmapRenderer( wxT("wxBitmap"), mode ),
1058         model_column, width, align, flags );
1059     AppendColumn( ret );
1060     return ret;
1061 }
1062
1063 wxDataViewColumn *
1064 wxDataViewCtrlBase::AppendTextColumn( const wxBitmap &label, unsigned int model_column,
1065                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1066 {
1067     wxDataViewColumn *ret = new wxDataViewColumn( label,
1068         new wxDataViewTextRenderer( wxT("string"), mode ),
1069         model_column, width, align, flags );
1070     AppendColumn( ret );
1071     return ret;
1072 }
1073
1074 wxDataViewColumn *
1075 wxDataViewCtrlBase::AppendIconTextColumn( const wxBitmap &label, unsigned int model_column,
1076                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1077 {
1078     wxDataViewColumn *ret = new wxDataViewColumn( label,
1079         new wxDataViewIconTextRenderer( wxT("wxDataViewIconText"), mode ),
1080         model_column, width, align, flags );
1081     AppendColumn( ret );
1082     return ret;
1083 }
1084
1085 wxDataViewColumn *
1086 wxDataViewCtrlBase::AppendToggleColumn( const wxBitmap &label, unsigned int model_column,
1087                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1088 {
1089     wxDataViewColumn *ret = new wxDataViewColumn( label,
1090         new wxDataViewToggleRenderer( wxT("bool"), mode ),
1091         model_column, width, align, flags );
1092     AppendColumn( ret );
1093     return ret;
1094 }
1095
1096 wxDataViewColumn *
1097 wxDataViewCtrlBase::AppendProgressColumn( const wxBitmap &label, unsigned int model_column,
1098                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1099 {
1100     wxDataViewColumn *ret = new wxDataViewColumn( label,
1101         new wxDataViewProgressRenderer( wxEmptyString, wxT("long"), mode ),
1102         model_column, width, align, flags );
1103     AppendColumn( ret );
1104     return ret;
1105 }
1106
1107 wxDataViewColumn *
1108 wxDataViewCtrlBase::AppendDateColumn( const wxBitmap &label, unsigned int model_column,
1109                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1110 {
1111     wxDataViewColumn *ret = new wxDataViewColumn( label,
1112         new wxDataViewDateRenderer( wxT("datetime"), mode ),
1113         model_column, width, align, flags );
1114     AppendColumn( ret );
1115     return ret;
1116 }
1117
1118 wxDataViewColumn *
1119 wxDataViewCtrlBase::AppendBitmapColumn( const wxBitmap &label, unsigned int model_column,
1120                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1121 {
1122     wxDataViewColumn *ret = new wxDataViewColumn( label,
1123         new wxDataViewBitmapRenderer( wxT("wxBitmap"), mode ),
1124         model_column, width, align, flags );
1125     AppendColumn( ret );
1126     return ret;
1127 }
1128
1129 wxDataViewColumn *
1130 wxDataViewCtrlBase::PrependTextColumn( const wxString &label, unsigned int model_column,
1131                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1132 {
1133     wxDataViewColumn *ret = new wxDataViewColumn( label,
1134         new wxDataViewTextRenderer( wxT("string"), mode ),
1135         model_column, width, align, flags );
1136     PrependColumn( ret );
1137     return ret;
1138 }
1139
1140 wxDataViewColumn *
1141 wxDataViewCtrlBase::PrependIconTextColumn( const wxString &label, unsigned int model_column,
1142                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1143 {
1144     wxDataViewColumn *ret = new wxDataViewColumn( label,
1145         new wxDataViewIconTextRenderer( wxT("wxDataViewIconText"), mode ),
1146         model_column, width, align, flags );
1147     PrependColumn( ret );
1148     return ret;
1149 }
1150
1151 wxDataViewColumn *
1152 wxDataViewCtrlBase::PrependToggleColumn( const wxString &label, unsigned int model_column,
1153                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1154 {
1155
1156     wxDataViewColumn *ret = new wxDataViewColumn( label,
1157         new wxDataViewToggleRenderer( wxT("bool"), mode ),
1158         model_column, width, align, flags );
1159     PrependColumn( ret );
1160     return ret;
1161 }
1162
1163 wxDataViewColumn *
1164 wxDataViewCtrlBase::PrependProgressColumn( const wxString &label, unsigned int model_column,
1165                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1166 {
1167     wxDataViewColumn *ret = new wxDataViewColumn( label,
1168         new wxDataViewProgressRenderer( wxEmptyString, wxT("long"), mode ),
1169         model_column, width, align, flags );
1170     PrependColumn( ret );
1171     return ret;
1172 }
1173
1174 wxDataViewColumn *
1175 wxDataViewCtrlBase::PrependDateColumn( const wxString &label, unsigned int model_column,
1176                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1177 {
1178     wxDataViewColumn *ret = new wxDataViewColumn( label,
1179         new wxDataViewDateRenderer( wxT("datetime"), mode ),
1180         model_column, width, align, flags );
1181     PrependColumn( ret );
1182     return ret;
1183 }
1184
1185 wxDataViewColumn *
1186 wxDataViewCtrlBase::PrependBitmapColumn( const wxString &label, unsigned int model_column,
1187                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1188 {
1189     wxDataViewColumn *ret = new wxDataViewColumn( label,
1190         new wxDataViewBitmapRenderer( wxT("wxBitmap"), mode ),
1191         model_column, width, align, flags );
1192     PrependColumn( ret );
1193     return ret;
1194 }
1195
1196 wxDataViewColumn *
1197 wxDataViewCtrlBase::PrependTextColumn( const wxBitmap &label, unsigned int model_column,
1198                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1199 {
1200     wxDataViewColumn *ret = new wxDataViewColumn( label,
1201         new wxDataViewTextRenderer( wxT("string"), mode ),
1202         model_column, width, align, flags );
1203     PrependColumn( ret );
1204     return ret;
1205 }
1206
1207 wxDataViewColumn *
1208 wxDataViewCtrlBase::PrependIconTextColumn( const wxBitmap &label, unsigned int model_column,
1209                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1210 {
1211     wxDataViewColumn *ret = new wxDataViewColumn( label,
1212         new wxDataViewIconTextRenderer( wxT("wxDataViewIconText"), mode ),
1213         model_column, width, align, flags );
1214     PrependColumn( ret );
1215     return ret;
1216 }
1217
1218 wxDataViewColumn *
1219 wxDataViewCtrlBase::PrependToggleColumn( const wxBitmap &label, unsigned int model_column,
1220                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1221 {
1222     wxDataViewColumn *ret = new wxDataViewColumn( label,
1223         new wxDataViewToggleRenderer( wxT("bool"), mode ),
1224         model_column, width, align, flags );
1225     PrependColumn( ret );
1226     return ret;
1227 }
1228
1229 wxDataViewColumn *
1230 wxDataViewCtrlBase::PrependProgressColumn( const wxBitmap &label, unsigned int model_column,
1231                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1232 {
1233     wxDataViewColumn *ret = new wxDataViewColumn( label,
1234         new wxDataViewProgressRenderer( wxEmptyString, wxT("long"), mode ),
1235         model_column, width, align, flags );
1236     PrependColumn( ret );
1237     return ret;
1238 }
1239
1240 wxDataViewColumn *
1241 wxDataViewCtrlBase::PrependDateColumn( const wxBitmap &label, unsigned int model_column,
1242                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1243 {
1244     wxDataViewColumn *ret = new wxDataViewColumn( label,
1245         new wxDataViewDateRenderer( wxT("datetime"), mode ),
1246         model_column, width, align, flags );
1247     PrependColumn( ret );
1248     return ret;
1249 }
1250
1251 wxDataViewColumn *
1252 wxDataViewCtrlBase::PrependBitmapColumn( const wxBitmap &label, unsigned int model_column,
1253                             wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1254 {
1255     wxDataViewColumn *ret = new wxDataViewColumn( label,
1256         new wxDataViewBitmapRenderer( wxT("wxBitmap"), mode ),
1257         model_column, width, align, flags );
1258     PrependColumn( ret );
1259     return ret;
1260 }
1261
1262 bool
1263 wxDataViewCtrlBase::AppendColumn( wxDataViewColumn *col )
1264 {
1265     col->SetOwner( (wxDataViewCtrl*) this );
1266     return true;
1267 }
1268
1269 bool
1270 wxDataViewCtrlBase::PrependColumn( wxDataViewColumn *col )
1271 {
1272     col->SetOwner( (wxDataViewCtrl*) this );
1273     return true;
1274 }
1275
1276 bool
1277 wxDataViewCtrlBase::InsertColumn( unsigned int WXUNUSED(pos), wxDataViewColumn *col )
1278 {
1279     col->SetOwner( (wxDataViewCtrl*) this );
1280     return true;
1281 }
1282
1283 // ---------------------------------------------------------
1284 // wxDataViewEvent
1285 // ---------------------------------------------------------
1286
1287 IMPLEMENT_DYNAMIC_CLASS(wxDataViewEvent,wxNotifyEvent)
1288
1289 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_SELECTION_CHANGED, wxDataViewEvent );
1290
1291 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_ITEM_ACTIVATED, wxDataViewEvent );
1292 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_ITEM_COLLAPSING, wxDataViewEvent );
1293 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_ITEM_COLLAPSED, wxDataViewEvent );
1294 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_ITEM_EXPANDING, wxDataViewEvent );
1295 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_ITEM_EXPANDED, wxDataViewEvent );
1296 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_ITEM_EDITING_STARTED, wxDataViewEvent );
1297 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_ITEM_START_EDITING, wxDataViewEvent );
1298 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_ITEM_EDITING_DONE, wxDataViewEvent );
1299 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_ITEM_VALUE_CHANGED, wxDataViewEvent );
1300
1301 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_ITEM_CONTEXT_MENU, wxDataViewEvent );
1302
1303 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_CLICK, wxDataViewEvent );
1304 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK, wxDataViewEvent );
1305 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_COLUMN_SORTED, wxDataViewEvent );
1306 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_COLUMN_REORDERED, wxDataViewEvent );
1307
1308 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_CACHE_HINT, wxDataViewEvent );
1309
1310 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_ITEM_BEGIN_DRAG, wxDataViewEvent );
1311 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_ITEM_DROP_POSSIBLE, wxDataViewEvent );
1312 wxDEFINE_EVENT( wxEVT_COMMAND_DATAVIEW_ITEM_DROP, wxDataViewEvent );
1313
1314
1315
1316 // -------------------------------------
1317 // wxDataViewSpinRenderer
1318 // -------------------------------------
1319
1320 wxDataViewSpinRenderer::wxDataViewSpinRenderer( int min, int max, wxDataViewCellMode mode, int alignment ) :
1321    wxDataViewCustomRenderer(wxT("long"), mode, alignment )
1322 {
1323     m_min = min;
1324     m_max = max;
1325 }
1326
1327 wxControl* wxDataViewSpinRenderer::CreateEditorCtrl( wxWindow *parent, wxRect labelRect, const wxVariant &value )
1328 {
1329     long l = value;
1330     wxSize size = labelRect.GetSize();
1331 #ifdef __WXMAC__
1332     size = wxSize( wxMax(70,labelRect.width ), -1 );
1333 #endif
1334     wxString str;
1335     str.Printf( wxT("%d"), (int) l );
1336     wxSpinCtrl *sc = new wxSpinCtrl( parent, wxID_ANY, str,
1337                labelRect.GetTopLeft(), size, wxSP_ARROW_KEYS|wxTE_PROCESS_ENTER, m_min, m_max, l );
1338 #ifdef __WXMAC__
1339     size = sc->GetSize();
1340     wxPoint pt = sc->GetPosition();
1341     sc->SetSize( pt.x - 4, pt.y - 4, size.x, size.y );
1342 #endif
1343
1344     return sc;
1345 }
1346
1347 bool wxDataViewSpinRenderer::GetValueFromEditorCtrl( wxControl* editor, wxVariant &value )
1348 {
1349     wxSpinCtrl *sc = (wxSpinCtrl*) editor;
1350     long l = sc->GetValue();
1351     value = l;
1352     return true;
1353 }
1354
1355 bool wxDataViewSpinRenderer::Render( wxRect rect, wxDC *dc, int state )
1356 {
1357     wxString str;
1358     str.Printf(wxT("%d"), (int) m_data );
1359     RenderText( str, 0, rect, dc, state );
1360     return true;
1361 }
1362
1363 wxSize wxDataViewSpinRenderer::GetSize() const
1364 {
1365     return wxSize(80,16);
1366 }
1367
1368 bool wxDataViewSpinRenderer::SetValue( const wxVariant &value )
1369 {
1370     m_data = value.GetLong();
1371     return true;
1372 }
1373
1374 bool wxDataViewSpinRenderer::GetValue( wxVariant &value ) const
1375 {
1376     value = m_data;
1377     return true;
1378 }
1379
1380 // -------------------------------------
1381 // wxDataViewChoiceRenderer
1382 // -------------------------------------
1383
1384 #if defined(wxHAS_GENERIC_DATAVIEWCTRL) || defined(__WXOSX_CARBON__)
1385
1386 wxDataViewChoiceRenderer::wxDataViewChoiceRenderer( const wxArrayString& choices, wxDataViewCellMode mode, int alignment ) :
1387    wxDataViewCustomRenderer(wxT("string"), mode, alignment )
1388 {
1389     m_choices = choices;
1390 }
1391
1392 wxControl* wxDataViewChoiceRenderer::CreateEditorCtrl( wxWindow *parent, wxRect labelRect, const wxVariant &value )
1393 {
1394     wxChoice* c = new wxChoice(parent, wxID_ANY, labelRect.GetTopLeft(), wxDefaultSize, m_choices );
1395     c->Move(labelRect.GetRight() - c->GetRect().width, wxDefaultCoord);
1396     c->SetStringSelection( value.GetString() );
1397     return c;
1398 }
1399
1400 bool wxDataViewChoiceRenderer::GetValueFromEditorCtrl( wxControl* editor, wxVariant &value )
1401 {
1402     wxChoice *c = (wxChoice*) editor;
1403     wxString s = c->GetStringSelection();
1404     value = s;
1405     return true;
1406 }
1407
1408 bool wxDataViewChoiceRenderer::Render( wxRect rect, wxDC *dc, int state )
1409 {
1410     RenderText( m_data, 0, rect, dc, state );
1411     return true;
1412 }
1413
1414 wxSize wxDataViewChoiceRenderer::GetSize() const
1415 {
1416     return wxSize(80,16);
1417 }
1418
1419 bool wxDataViewChoiceRenderer::SetValue( const wxVariant &value )
1420 {
1421     m_data = value.GetString();
1422     return true;
1423 }
1424
1425 bool wxDataViewChoiceRenderer::GetValue( wxVariant &value ) const
1426 {
1427     value = m_data;
1428     return true;
1429 }
1430
1431 // ----------------------------------------------------------------------------
1432 // wxDataViewChoiceByIndexRenderer
1433 // ----------------------------------------------------------------------------
1434
1435 wxDataViewChoiceByIndexRenderer::wxDataViewChoiceByIndexRenderer( const wxArrayString &choices,
1436                                   wxDataViewCellMode mode, int alignment ) :
1437       wxDataViewChoiceRenderer( choices, mode, alignment )
1438 {
1439 }
1440
1441 wxControl* wxDataViewChoiceByIndexRenderer::CreateEditorCtrl( wxWindow *parent, wxRect labelRect, const wxVariant &value )
1442 {
1443     wxVariant string_value = GetChoice( value.GetLong() );
1444
1445     return wxDataViewChoiceRenderer::CreateEditorCtrl( parent, labelRect, string_value );
1446 }
1447
1448 bool wxDataViewChoiceByIndexRenderer::GetValueFromEditorCtrl( wxControl* editor, wxVariant &value )
1449 {
1450     wxVariant string_value;
1451     if (!wxDataViewChoiceRenderer::GetValueFromEditorCtrl( editor, string_value ))
1452         return false;
1453
1454     value = (long) GetChoices().Index( string_value.GetString() );
1455     return true;
1456 }
1457
1458 bool wxDataViewChoiceByIndexRenderer::SetValue( const wxVariant &value )
1459 {
1460     wxVariant string_value = GetChoice( value.GetLong() );
1461     return wxDataViewChoiceRenderer::SetValue( string_value );
1462 }
1463
1464 bool wxDataViewChoiceByIndexRenderer::GetValue( wxVariant &value ) const
1465 {
1466     wxVariant string_value;
1467     if (!wxDataViewChoiceRenderer::GetValue( string_value ))
1468         return false;
1469
1470     value = (long) GetChoices().Index( string_value.GetString() );
1471     return true;
1472 }
1473
1474 #endif
1475
1476 //-----------------------------------------------------------------------------
1477 // wxDataViewListStore
1478 //-----------------------------------------------------------------------------
1479
1480 wxDataViewListStore::wxDataViewListStore()
1481 {
1482 }
1483
1484 wxDataViewListStore::~wxDataViewListStore()
1485 {
1486     wxVector<wxDataViewListStoreLine*>::iterator it;
1487     for (it = m_data.begin(); it != m_data.end(); ++it)
1488     {
1489         wxDataViewListStoreLine* line = *it;
1490         delete line;
1491     }
1492 }
1493
1494 void wxDataViewListStore::PrependColumn( const wxString &varianttype )
1495 {
1496     m_cols.Insert( varianttype, 0 );
1497 }
1498
1499 void wxDataViewListStore::InsertColumn( unsigned int pos, const wxString &varianttype )
1500 {
1501     m_cols.Insert( varianttype, pos );
1502 }
1503
1504 void wxDataViewListStore::AppendColumn( const wxString &varianttype )
1505 {
1506     m_cols.Add( varianttype );
1507 }
1508
1509 unsigned int wxDataViewListStore::GetColumnCount() const
1510 {
1511     return m_cols.GetCount();
1512 }
1513
1514 wxString wxDataViewListStore::GetColumnType( unsigned int pos ) const
1515 {
1516     return m_cols[pos];
1517 }
1518
1519 void wxDataViewListStore::AppendItem( const wxVector<wxVariant> &values, wxClientData *data )
1520 {
1521     wxDataViewListStoreLine *line = new wxDataViewListStoreLine( data );
1522     line->m_values = values;
1523     m_data.push_back( line );
1524
1525     RowAppended();
1526 }
1527
1528 void wxDataViewListStore::PrependItem( const wxVector<wxVariant> &values, wxClientData *data )
1529 {
1530     wxDataViewListStoreLine *line = new wxDataViewListStoreLine( data );
1531     line->m_values = values;
1532     m_data.insert( m_data.begin(), line );
1533
1534     RowPrepended();
1535 }
1536
1537 void wxDataViewListStore::InsertItem(  unsigned int row, const wxVector<wxVariant> &values,
1538                                        wxClientData *data )
1539 {
1540     wxDataViewListStoreLine *line = new wxDataViewListStoreLine( data );
1541     line->m_values = values;
1542     m_data.insert( m_data.begin()+row, line );
1543
1544     RowInserted( row );
1545 }
1546
1547 void wxDataViewListStore::DeleteItem( unsigned int row )
1548 {
1549     wxVector<wxDataViewListStoreLine*>::iterator it = m_data.begin() + row;
1550     delete *it;
1551     m_data.erase( it );
1552
1553     RowDeleted( row );
1554 }
1555
1556 void wxDataViewListStore::DeleteAllItems()
1557 {
1558     wxVector<wxDataViewListStoreLine*>::iterator it;
1559     for (it = m_data.begin(); it != m_data.end(); ++it)
1560     {
1561         wxDataViewListStoreLine* line = *it;
1562         delete line;
1563     }
1564
1565     m_data.clear();
1566
1567     Reset( 0 );
1568 }
1569
1570 void wxDataViewListStore::GetValueByRow( wxVariant &value, unsigned int row, unsigned int col ) const
1571 {
1572     wxDataViewListStoreLine *line = m_data[row];
1573     value = line->m_values[col];
1574 }
1575
1576 bool wxDataViewListStore::SetValueByRow( const wxVariant &value, unsigned int row, unsigned int col )
1577 {
1578     wxDataViewListStoreLine *line = m_data[row];
1579     line->m_values[col] = value;
1580
1581     return true;
1582 }
1583
1584 //-----------------------------------------------------------------------------
1585 // wxDataViewListCtrl
1586 //-----------------------------------------------------------------------------
1587
1588 IMPLEMENT_DYNAMIC_CLASS(wxDataViewListCtrl,wxDataViewCtrl)
1589
1590 BEGIN_EVENT_TABLE(wxDataViewListCtrl,wxDataViewCtrl)
1591    EVT_SIZE( wxDataViewListCtrl::OnSize )
1592 END_EVENT_TABLE()
1593
1594 wxDataViewListCtrl::wxDataViewListCtrl()
1595 {
1596 }
1597
1598 wxDataViewListCtrl::wxDataViewListCtrl( wxWindow *parent, wxWindowID id,
1599            const wxPoint& pos, const wxSize& size, long style,
1600            const wxValidator& validator )
1601 {
1602     Create( parent, id, pos, size, style, validator );
1603
1604     wxDataViewListStore *store = new wxDataViewListStore;
1605     AssociateModel( store );
1606     store->DecRef();
1607 }
1608
1609 wxDataViewListCtrl::~wxDataViewListCtrl()
1610 {
1611 }
1612
1613
1614 bool wxDataViewListCtrl::Create( wxWindow *parent, wxWindowID id,
1615            const wxPoint& pos, const wxSize& size, long style,
1616            const wxValidator& validator )
1617 {
1618     return wxDataViewCtrl::Create( parent, id, pos, size, style, validator );
1619 }
1620
1621 bool wxDataViewListCtrl::AppendColumn( wxDataViewColumn *column, const wxString &varianttype )
1622 {
1623     GetStore()->AppendColumn( varianttype );
1624     return wxDataViewCtrl::AppendColumn( column );
1625 }
1626
1627 bool wxDataViewListCtrl::PrependColumn( wxDataViewColumn *column, const wxString &varianttype )
1628 {
1629     GetStore()->PrependColumn( varianttype );
1630     return wxDataViewCtrl::PrependColumn( column );
1631 }
1632
1633 bool wxDataViewListCtrl::InsertColumn( unsigned int pos, wxDataViewColumn *column, const wxString &varianttype )
1634 {
1635     GetStore()->InsertColumn( pos, varianttype );
1636     return wxDataViewCtrl::InsertColumn( pos, column );
1637 }
1638
1639 bool wxDataViewListCtrl::PrependColumn( wxDataViewColumn *col )
1640 {
1641     return PrependColumn( col, "string" );
1642 }
1643
1644 bool wxDataViewListCtrl::InsertColumn( unsigned int pos, wxDataViewColumn *col )
1645 {
1646     return InsertColumn( pos, col, "string" );
1647 }
1648
1649 bool wxDataViewListCtrl::AppendColumn( wxDataViewColumn *col )
1650 {
1651     return AppendColumn( col, "string" );
1652 }
1653
1654 wxDataViewColumn *wxDataViewListCtrl::AppendTextColumn( const wxString &label,
1655           wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1656 {
1657     GetStore()->AppendColumn( wxT("string") );
1658
1659     wxDataViewColumn *ret = new wxDataViewColumn( label,
1660         new wxDataViewTextRenderer( wxT("string"), mode ),
1661         GetStore()->GetColumnCount()-1, width, align, flags );
1662
1663     wxDataViewCtrl::AppendColumn( ret );
1664
1665     return ret;
1666 }
1667
1668 wxDataViewColumn *wxDataViewListCtrl::AppendToggleColumn( const wxString &label,
1669           wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1670 {
1671     GetStore()->AppendColumn( wxT("bool") );
1672
1673     wxDataViewColumn *ret = new wxDataViewColumn( label,
1674         new wxDataViewToggleRenderer( wxT("bool"), mode ),
1675         GetStore()->GetColumnCount()-1, width, align, flags );
1676
1677     wxDataViewCtrl::AppendColumn( ret );
1678
1679     return ret;
1680 }
1681
1682 wxDataViewColumn *wxDataViewListCtrl::AppendProgressColumn( const wxString &label,
1683           wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1684 {
1685     GetStore()->AppendColumn( wxT("long") );
1686
1687     wxDataViewColumn *ret = new wxDataViewColumn( label,
1688         new wxDataViewProgressRenderer( wxEmptyString, wxT("long"), mode ),
1689         GetStore()->GetColumnCount()-1, width, align, flags );
1690
1691     wxDataViewCtrl::AppendColumn( ret );
1692
1693     return ret;
1694 }
1695
1696 wxDataViewColumn *wxDataViewListCtrl::AppendIconTextColumn( const wxString &label,
1697           wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1698 {
1699     GetStore()->AppendColumn( wxT("wxDataViewIconText") );
1700
1701     wxDataViewColumn *ret = new wxDataViewColumn( label,
1702         new wxDataViewIconTextRenderer( wxT("wxDataViewIconText"), mode ),
1703         GetStore()->GetColumnCount()-1, width, align, flags );
1704
1705     wxDataViewCtrl::AppendColumn( ret );
1706
1707     return ret;
1708 }
1709
1710 void wxDataViewListCtrl::OnSize( wxSizeEvent &event )
1711 {
1712     event.Skip( true );
1713 }
1714
1715 //-----------------------------------------------------------------------------
1716 // wxDataViewTreeStore
1717 //-----------------------------------------------------------------------------
1718
1719 wxDataViewTreeStoreNode::wxDataViewTreeStoreNode(
1720         wxDataViewTreeStoreNode *parent,
1721         const wxString &text, const wxIcon &icon, wxClientData *data )
1722 {
1723     m_parent = parent;
1724     m_text = text;
1725     m_icon = icon;
1726     m_data = data;
1727 }
1728
1729 wxDataViewTreeStoreNode::~wxDataViewTreeStoreNode()
1730 {
1731     if (m_data)
1732         delete m_data;
1733 }
1734
1735 #include "wx/listimpl.cpp"
1736 WX_DEFINE_LIST(wxDataViewTreeStoreNodeList)
1737
1738 wxDataViewTreeStoreContainerNode::wxDataViewTreeStoreContainerNode(
1739         wxDataViewTreeStoreNode *parent, const wxString &text,
1740         const wxIcon &icon, const wxIcon &expanded, wxClientData *data ) :
1741     wxDataViewTreeStoreNode( parent, text, icon, data )
1742 {
1743     m_iconExpanded = expanded;
1744     m_isExpanded = false;
1745     m_children.DeleteContents(true);
1746 }
1747
1748 wxDataViewTreeStoreContainerNode::~wxDataViewTreeStoreContainerNode()
1749 {
1750 }
1751
1752 //-----------------------------------------------------------------------------
1753
1754 wxDataViewTreeStore::wxDataViewTreeStore()
1755 {
1756     m_root = new wxDataViewTreeStoreContainerNode( NULL, wxEmptyString );
1757 }
1758
1759 wxDataViewTreeStore::~wxDataViewTreeStore()
1760 {
1761     delete m_root;
1762 }
1763
1764 wxDataViewItem wxDataViewTreeStore::AppendItem( const wxDataViewItem& parent,
1765         const wxString &text, const wxIcon &icon, wxClientData *data )
1766 {
1767     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1768     if (!parent_node) return wxDataViewItem(0);
1769
1770     wxDataViewTreeStoreNode *node =
1771         new wxDataViewTreeStoreNode( parent_node, text, icon, data );
1772     parent_node->GetChildren().Append( node );
1773
1774     return node->GetItem();
1775 }
1776
1777 wxDataViewItem wxDataViewTreeStore::PrependItem( const wxDataViewItem& parent,
1778         const wxString &text, const wxIcon &icon, wxClientData *data )
1779 {
1780     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1781     if (!parent_node) return wxDataViewItem(0);
1782
1783     wxDataViewTreeStoreNode *node =
1784         new wxDataViewTreeStoreNode( parent_node, text, icon, data );
1785     parent_node->GetChildren().Insert( node );
1786
1787     return node->GetItem();
1788 }
1789
1790 wxDataViewItem
1791 wxDataViewTreeStore::InsertItem(const wxDataViewItem& parent,
1792                                 const wxDataViewItem& previous,
1793                                 const wxString& text,
1794                                 const wxIcon& icon,
1795                                 wxClientData *data)
1796 {
1797     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1798     if (!parent_node) return wxDataViewItem(0);
1799
1800     wxDataViewTreeStoreNode *previous_node = FindNode( previous );
1801     int pos = parent_node->GetChildren().IndexOf( previous_node );
1802     if (pos == wxNOT_FOUND) return wxDataViewItem(0);
1803
1804     wxDataViewTreeStoreNode *node =
1805         new wxDataViewTreeStoreNode( parent_node, text, icon, data );
1806     parent_node->GetChildren().Insert( (size_t) pos, node );
1807
1808     return node->GetItem();
1809 }
1810
1811 wxDataViewItem wxDataViewTreeStore::PrependContainer( const wxDataViewItem& parent,
1812         const wxString &text, const wxIcon &icon, const wxIcon &expanded,
1813         wxClientData *data )
1814 {
1815     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1816     if (!parent_node) return wxDataViewItem(0);
1817
1818     wxDataViewTreeStoreContainerNode *node =
1819         new wxDataViewTreeStoreContainerNode( parent_node, text, icon, expanded, data );
1820     parent_node->GetChildren().Insert( node );
1821
1822     return node->GetItem();
1823 }
1824
1825 wxDataViewItem
1826 wxDataViewTreeStore::AppendContainer(const wxDataViewItem& parent,
1827                                      const wxString &text,
1828                                      const wxIcon& icon,
1829                                      const wxIcon& expanded,
1830                                      wxClientData * data)
1831 {
1832     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1833     if (!parent_node) return wxDataViewItem(0);
1834
1835     wxDataViewTreeStoreContainerNode *node =
1836         new wxDataViewTreeStoreContainerNode( parent_node, text, icon, expanded, data );
1837     parent_node->GetChildren().Append( node );
1838
1839     return node->GetItem();
1840 }
1841
1842 wxDataViewItem
1843 wxDataViewTreeStore::InsertContainer(const wxDataViewItem& parent,
1844                                      const wxDataViewItem& previous,
1845                                      const wxString& text,
1846                                      const wxIcon& icon,
1847                                      const wxIcon& expanded,
1848                                      wxClientData * data)
1849 {
1850     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1851     if (!parent_node) return wxDataViewItem(0);
1852
1853     wxDataViewTreeStoreNode *previous_node = FindNode( previous );
1854     int pos = parent_node->GetChildren().IndexOf( previous_node );
1855     if (pos == wxNOT_FOUND) return wxDataViewItem(0);
1856
1857     wxDataViewTreeStoreContainerNode *node =
1858         new wxDataViewTreeStoreContainerNode( parent_node, text, icon, expanded, data );
1859     parent_node->GetChildren().Insert( (size_t) pos, node );
1860
1861     return node->GetItem();
1862 }
1863
1864 bool wxDataViewTreeStore::IsContainer( const wxDataViewItem& item ) const
1865 {
1866     wxDataViewTreeStoreNode *node = FindNode( item );
1867     if (!node) return false;
1868
1869     return node->IsContainer();
1870 }
1871
1872 wxDataViewItem wxDataViewTreeStore::GetNthChild( const wxDataViewItem& parent, unsigned int pos ) const
1873 {
1874     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1875     if (!parent_node) return wxDataViewItem(0);
1876
1877     wxDataViewTreeStoreNodeList::compatibility_iterator node = parent_node->GetChildren().Item( pos );
1878     if (node)
1879         return node->GetData();
1880
1881     return wxDataViewItem(0);
1882 }
1883
1884 int wxDataViewTreeStore::GetChildCount( const wxDataViewItem& parent ) const
1885 {
1886     wxDataViewTreeStoreNode *node = FindNode( parent );
1887     if (!node) return -1;
1888
1889     if (!node->IsContainer())
1890         return 0;
1891
1892     wxDataViewTreeStoreContainerNode *container_node = (wxDataViewTreeStoreContainerNode*) node;
1893     return (int) container_node->GetChildren().GetCount();
1894 }
1895
1896 void wxDataViewTreeStore::SetItemText( const wxDataViewItem& item, const wxString &text )
1897 {
1898     wxDataViewTreeStoreNode *node = FindNode( item );
1899     if (!node) return;
1900
1901     node->SetText( text );
1902 }
1903
1904 wxString wxDataViewTreeStore::GetItemText( const wxDataViewItem& item ) const
1905 {
1906     wxDataViewTreeStoreNode *node = FindNode( item );
1907     if (!node) return wxEmptyString;
1908
1909     return node->GetText();
1910 }
1911
1912 void wxDataViewTreeStore::SetItemIcon( const wxDataViewItem& item, const wxIcon &icon )
1913 {
1914     wxDataViewTreeStoreNode *node = FindNode( item );
1915     if (!node) return;
1916
1917     node->SetIcon( icon );
1918 }
1919
1920 const wxIcon &wxDataViewTreeStore::GetItemIcon( const wxDataViewItem& item ) const
1921 {
1922     wxDataViewTreeStoreNode *node = FindNode( item );
1923     if (!node) return wxNullIcon;
1924
1925     return node->GetIcon();
1926 }
1927
1928 void wxDataViewTreeStore::SetItemExpandedIcon( const wxDataViewItem& item, const wxIcon &icon )
1929 {
1930     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1931     if (!node) return;
1932
1933     node->SetExpandedIcon( icon );
1934 }
1935
1936 const wxIcon &wxDataViewTreeStore::GetItemExpandedIcon( const wxDataViewItem& item ) const
1937 {
1938     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1939     if (!node) return wxNullIcon;
1940
1941     return node->GetExpandedIcon();
1942 }
1943
1944 void wxDataViewTreeStore::SetItemData( const wxDataViewItem& item, wxClientData *data )
1945 {
1946     wxDataViewTreeStoreNode *node = FindNode( item );
1947     if (!node) return;
1948
1949     node->SetData( data );
1950 }
1951
1952 wxClientData *wxDataViewTreeStore::GetItemData( const wxDataViewItem& item ) const
1953 {
1954     wxDataViewTreeStoreNode *node = FindNode( item );
1955     if (!node) return NULL;
1956
1957     return node->GetData();
1958 }
1959
1960 void wxDataViewTreeStore::DeleteItem( const wxDataViewItem& item )
1961 {
1962     if (!item.IsOk()) return;
1963
1964     wxDataViewItem parent_item = GetParent( item );
1965
1966     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent_item );
1967     if (!parent_node) return;
1968
1969     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1970     if (!node) return;
1971
1972     parent_node->GetChildren().DeleteObject( node );
1973 }
1974
1975 void wxDataViewTreeStore::DeleteChildren( const wxDataViewItem& item )
1976 {
1977     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1978     if (!node) return;
1979
1980     node->GetChildren().clear();
1981 }
1982
1983 void wxDataViewTreeStore::DeleteAllItems()
1984 {
1985     DeleteChildren(m_root);
1986 }
1987
1988 void
1989 wxDataViewTreeStore::GetValue(wxVariant &variant,
1990                               const wxDataViewItem &item,
1991                               unsigned int WXUNUSED(col)) const
1992 {
1993     // if (col != 0) return;
1994
1995     wxDataViewTreeStoreNode *node = FindNode( item );
1996     if (!node) return;
1997
1998     wxIcon icon( node->GetIcon());
1999     if (node->IsContainer())
2000     {
2001         wxDataViewTreeStoreContainerNode *container = (wxDataViewTreeStoreContainerNode*) node;
2002         if (container->IsExpanded() && container->GetExpandedIcon().IsOk())
2003            icon = container->GetExpandedIcon();
2004     }
2005
2006     wxDataViewIconText data( node->GetText(), icon );
2007
2008     variant << data;
2009 }
2010
2011 bool
2012 wxDataViewTreeStore::SetValue(const wxVariant& variant,
2013                               const wxDataViewItem& item,
2014                               unsigned int WXUNUSED(col))
2015 {
2016     // if (col != 0) return false;
2017
2018     wxDataViewTreeStoreNode *node = FindNode( item );
2019     if (!node) return false;
2020
2021     wxDataViewIconText data;
2022
2023     data << variant;
2024
2025     node->SetText( data.GetText() );
2026     node->SetIcon( data.GetIcon() );
2027
2028     return true;
2029 }
2030
2031 wxDataViewItem wxDataViewTreeStore::GetParent( const wxDataViewItem &item ) const
2032 {
2033     wxDataViewTreeStoreNode *node = FindNode( item );
2034     if (!node) return wxDataViewItem(0);
2035
2036     wxDataViewTreeStoreNode *parent = node->GetParent();
2037     if (!parent) return wxDataViewItem(0);
2038
2039     if (parent == m_root)
2040         return wxDataViewItem(0);
2041
2042     return parent->GetItem();
2043 }
2044
2045 unsigned int wxDataViewTreeStore::GetChildren( const wxDataViewItem &item, wxDataViewItemArray &children ) const
2046 {
2047     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
2048     if (!node) return 0;
2049
2050     wxDataViewTreeStoreNodeList::iterator iter;
2051     for (iter = node->GetChildren().begin(); iter != node->GetChildren().end(); iter++)
2052     {
2053         wxDataViewTreeStoreNode* child = *iter;
2054         children.Add( child->GetItem() );
2055     }
2056
2057     return node->GetChildren().GetCount();
2058 }
2059
2060 int wxDataViewTreeStore::Compare( const wxDataViewItem &item1, const wxDataViewItem &item2,
2061                          unsigned int WXUNUSED(column), bool WXUNUSED(ascending) ) const
2062 {
2063     wxDataViewTreeStoreNode *node1 = FindNode( item1 );
2064     wxDataViewTreeStoreNode *node2 = FindNode( item2 );
2065
2066     if (!node1 || !node2)
2067         return 0;
2068
2069     wxDataViewTreeStoreContainerNode* parent1 =
2070         (wxDataViewTreeStoreContainerNode*) node1->GetParent();
2071     wxDataViewTreeStoreContainerNode* parent2 =
2072         (wxDataViewTreeStoreContainerNode*) node2->GetParent();
2073
2074     if (parent1 != parent2)
2075     {
2076         wxLogError( wxT("Comparing items with different parent.") );
2077         return 0;
2078     }
2079
2080     if (node1->IsContainer() && !node2->IsContainer())
2081         return -1;
2082
2083     if (node2->IsContainer() && !node1->IsContainer())
2084         return 1;
2085
2086     return parent1->GetChildren().IndexOf( node1 ) - parent2->GetChildren().IndexOf( node2 );
2087 }
2088
2089 wxDataViewTreeStoreNode *wxDataViewTreeStore::FindNode( const wxDataViewItem &item ) const
2090 {
2091     if (!item.IsOk())
2092         return m_root;
2093
2094     return (wxDataViewTreeStoreNode*) item.GetID();
2095 }
2096
2097 wxDataViewTreeStoreContainerNode *wxDataViewTreeStore::FindContainerNode( const wxDataViewItem &item ) const
2098 {
2099     if (!item.IsOk())
2100         return (wxDataViewTreeStoreContainerNode*) m_root;
2101
2102     wxDataViewTreeStoreNode* node = (wxDataViewTreeStoreNode*) item.GetID();
2103
2104     if (!node->IsContainer())
2105         return NULL;
2106
2107     return (wxDataViewTreeStoreContainerNode*) node;
2108 }
2109
2110 //-----------------------------------------------------------------------------
2111 // wxDataViewTreeCtrl
2112 //-----------------------------------------------------------------------------
2113
2114 IMPLEMENT_DYNAMIC_CLASS(wxDataViewTreeCtrl,wxDataViewCtrl)
2115
2116 BEGIN_EVENT_TABLE(wxDataViewTreeCtrl,wxDataViewCtrl)
2117    EVT_DATAVIEW_ITEM_EXPANDED(-1, wxDataViewTreeCtrl::OnExpanded)
2118    EVT_DATAVIEW_ITEM_COLLAPSED(-1, wxDataViewTreeCtrl::OnCollapsed)
2119    EVT_SIZE( wxDataViewTreeCtrl::OnSize )
2120 END_EVENT_TABLE()
2121
2122 wxDataViewTreeCtrl::~wxDataViewTreeCtrl()
2123 {
2124     delete m_imageList;
2125 }
2126
2127 bool wxDataViewTreeCtrl::Create( wxWindow *parent, wxWindowID id,
2128            const wxPoint& pos, const wxSize& size, long style, const wxValidator& validator )
2129 {
2130     if ( !wxDataViewCtrl::Create( parent, id, pos, size, style, validator ) )
2131         return false;
2132
2133     // create the standard model and a column in the tree
2134     wxDataViewTreeStore *store = new wxDataViewTreeStore;
2135     AssociateModel( store );
2136     store->DecRef();
2137
2138     AppendIconTextColumn
2139     (
2140         wxString(),                 // no label (header is not shown anyhow)
2141         0,                          // the only model column
2142         wxDATAVIEW_CELL_EDITABLE,
2143         -1,                         // default width
2144         wxALIGN_NOT,                //  and alignment
2145         0                           // not resizeable
2146     );
2147
2148     return true;
2149 }
2150
2151 void wxDataViewTreeCtrl::SetImageList( wxImageList *imagelist )
2152 {
2153     delete m_imageList;
2154
2155     m_imageList = imagelist;
2156 }
2157
2158 wxDataViewItem wxDataViewTreeCtrl::AppendItem( const wxDataViewItem& parent,
2159         const wxString &text, int iconIndex, wxClientData *data )
2160 {
2161     wxIcon icon = wxNullIcon;
2162     if (m_imageList && (iconIndex != -1))
2163         icon = m_imageList->GetIcon( iconIndex );
2164
2165     wxDataViewItem res = GetStore()->AppendItem( parent, text, icon, data );
2166
2167     GetStore()->ItemAdded( parent, res );
2168
2169     return res;
2170 }
2171
2172 wxDataViewItem wxDataViewTreeCtrl::PrependItem( const wxDataViewItem& parent,
2173         const wxString &text, int iconIndex, wxClientData *data )
2174 {
2175     wxIcon icon = wxNullIcon;
2176     if (m_imageList && (iconIndex != -1))
2177         icon = m_imageList->GetIcon( iconIndex );
2178
2179     wxDataViewItem res = GetStore()->PrependItem( parent, text, icon, data );
2180
2181     GetStore()->ItemAdded( parent, res );
2182
2183     return res;
2184 }
2185
2186 wxDataViewItem wxDataViewTreeCtrl::InsertItem( const wxDataViewItem& parent, const wxDataViewItem& previous,
2187         const wxString &text, int iconIndex, wxClientData *data )
2188 {
2189     wxIcon icon = wxNullIcon;
2190     if (m_imageList && (iconIndex != -1))
2191         icon = m_imageList->GetIcon( iconIndex );
2192
2193     wxDataViewItem res = GetStore()->InsertItem( parent, previous, text, icon, data );
2194
2195     GetStore()->ItemAdded( parent, res );
2196
2197     return res;
2198 }
2199
2200 wxDataViewItem wxDataViewTreeCtrl::PrependContainer( const wxDataViewItem& parent,
2201         const wxString &text, int iconIndex, int expandedIndex, wxClientData *data )
2202 {
2203     wxIcon icon = wxNullIcon;
2204     if (m_imageList && (iconIndex != -1))
2205         icon = m_imageList->GetIcon( iconIndex );
2206
2207     wxIcon expanded = wxNullIcon;
2208     if (m_imageList && (expandedIndex != -1))
2209         expanded = m_imageList->GetIcon( expandedIndex );
2210
2211     wxDataViewItem res = GetStore()->PrependContainer( parent, text, icon, expanded, data );
2212
2213     GetStore()->ItemAdded( parent, res );
2214
2215     return res;
2216 }
2217
2218 wxDataViewItem wxDataViewTreeCtrl::AppendContainer( const wxDataViewItem& parent,
2219         const wxString &text, int iconIndex, int expandedIndex, wxClientData *data )
2220 {
2221     wxIcon icon = wxNullIcon;
2222     if (m_imageList && (iconIndex != -1))
2223         icon = m_imageList->GetIcon( iconIndex );
2224
2225     wxIcon expanded = wxNullIcon;
2226     if (m_imageList && (expandedIndex != -1))
2227         expanded = m_imageList->GetIcon( expandedIndex );
2228
2229     wxDataViewItem res = GetStore()->AppendContainer( parent, text, icon, expanded, data );
2230
2231     GetStore()->ItemAdded( parent, res );
2232
2233     return res;
2234 }
2235
2236 wxDataViewItem wxDataViewTreeCtrl::InsertContainer( const wxDataViewItem& parent, const wxDataViewItem& previous,
2237         const wxString &text, int iconIndex, int expandedIndex, wxClientData *data )
2238 {
2239     wxIcon icon = wxNullIcon;
2240     if (m_imageList && (iconIndex != -1))
2241         icon = m_imageList->GetIcon( iconIndex );
2242
2243     wxIcon expanded = wxNullIcon;
2244     if (m_imageList && (expandedIndex != -1))
2245         expanded = m_imageList->GetIcon( expandedIndex );
2246
2247     wxDataViewItem res = GetStore()->InsertContainer( parent, previous, text, icon, expanded, data );
2248
2249     GetStore()->ItemAdded( parent, res );
2250
2251     return res;
2252 }
2253
2254 void wxDataViewTreeCtrl::SetItemText( const wxDataViewItem& item, const wxString &text )
2255 {
2256     GetStore()->SetItemText(item,text);
2257
2258     // notify control
2259     GetStore()->ValueChanged( item, 0 );
2260 }
2261
2262 void wxDataViewTreeCtrl::SetItemIcon( const wxDataViewItem& item, const wxIcon &icon )
2263 {
2264     GetStore()->SetItemIcon(item,icon);
2265
2266     // notify control
2267     GetStore()->ValueChanged( item, 0 );
2268 }
2269
2270 void wxDataViewTreeCtrl::SetItemExpandedIcon( const wxDataViewItem& item, const wxIcon &icon )
2271 {
2272     GetStore()->SetItemExpandedIcon(item,icon);
2273
2274     // notify control
2275     GetStore()->ValueChanged( item, 0 );
2276 }
2277
2278 void wxDataViewTreeCtrl::DeleteItem( const wxDataViewItem& item )
2279 {
2280     wxDataViewItem parent_item = GetStore()->GetParent( item );
2281
2282     GetStore()->DeleteItem(item);
2283
2284     // notify control
2285     GetStore()->ItemDeleted( parent_item, item );
2286 }
2287
2288 void wxDataViewTreeCtrl::DeleteChildren( const wxDataViewItem& item )
2289 {
2290     wxDataViewTreeStoreContainerNode *node = GetStore()->FindContainerNode( item );
2291     if (!node) return;
2292
2293     wxDataViewItemArray array;
2294     wxDataViewTreeStoreNodeList::iterator iter;
2295     for (iter = node->GetChildren().begin(); iter != node->GetChildren().end(); iter++)
2296     {
2297         wxDataViewTreeStoreNode* child = *iter;
2298         array.Add( child->GetItem() );
2299     }
2300
2301     GetStore()->DeleteChildren( item );
2302
2303     // notify control
2304     GetStore()->ItemsDeleted( item, array );
2305 }
2306
2307 void  wxDataViewTreeCtrl::DeleteAllItems()
2308 {
2309     GetStore()->DeleteAllItems();
2310
2311     GetStore()->Cleared();
2312 }
2313
2314 void wxDataViewTreeCtrl::OnExpanded( wxDataViewEvent &event )
2315 {
2316     if (m_imageList) return;
2317
2318     wxDataViewTreeStoreContainerNode* container = GetStore()->FindContainerNode( event.GetItem() );
2319     if (!container) return;
2320
2321     container->SetExpanded( true );
2322
2323     GetStore()->ItemChanged( event.GetItem() );
2324 }
2325
2326 void wxDataViewTreeCtrl::OnCollapsed( wxDataViewEvent &event )
2327 {
2328     if (m_imageList) return;
2329
2330     wxDataViewTreeStoreContainerNode* container = GetStore()->FindContainerNode( event.GetItem() );
2331     if (!container) return;
2332
2333     container->SetExpanded( false );
2334
2335     GetStore()->ItemChanged( event.GetItem() );
2336 }
2337
2338 void wxDataViewTreeCtrl::OnSize( wxSizeEvent &event )
2339 {
2340 #if defined(wxUSE_GENERICDATAVIEWCTRL)
2341     // automatically resize our only column to take the entire control width
2342     if ( GetColumnCount() )
2343     {
2344         wxSize size = GetClientSize();
2345         GetColumn(0)->SetWidth(size.x);
2346     }
2347 #endif
2348     event.Skip( true );
2349 }
2350
2351 #endif // wxUSE_DATAVIEWCTRL
2352