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     wxString s = value;
1395     wxSize size = labelRect.GetSize();
1396 #ifdef __WXMAC__
1397     size = wxSize( wxMax(70,labelRect.width ), -1 );
1398 #endif
1399     wxChoice *c = new wxChoice( parent, wxID_ANY, labelRect.GetTopLeft(), size, m_choices );
1400     c->SetStringSelection( value.GetString() );
1401
1402     return c;
1403 }
1404
1405 bool wxDataViewChoiceRenderer::GetValueFromEditorCtrl( wxControl* editor, wxVariant &value )
1406 {
1407     wxChoice *c = (wxChoice*) editor;
1408     wxString s = c->GetStringSelection();
1409     value = s;
1410     return true;
1411 }
1412
1413 bool wxDataViewChoiceRenderer::Render( wxRect rect, wxDC *dc, int state )
1414 {
1415     RenderText( m_data, 0, rect, dc, state );
1416     return true;
1417 }
1418
1419 wxSize wxDataViewChoiceRenderer::GetSize() const
1420 {
1421     return wxSize(80,16);
1422 }
1423
1424 bool wxDataViewChoiceRenderer::SetValue( const wxVariant &value )
1425 {
1426     m_data = value.GetString();
1427     return true;
1428 }
1429
1430 bool wxDataViewChoiceRenderer::GetValue( wxVariant &value ) const
1431 {
1432     value = m_data;
1433     return true;
1434 }
1435
1436 // ----------------------------------------------------------------------------
1437 // wxDataViewChoiceByIndexRenderer
1438 // ----------------------------------------------------------------------------
1439
1440 wxDataViewChoiceByIndexRenderer::wxDataViewChoiceByIndexRenderer( const wxArrayString &choices,
1441                                   wxDataViewCellMode mode, int alignment ) :
1442       wxDataViewChoiceRenderer( choices, mode, alignment )
1443 {
1444 }
1445
1446 wxControl* wxDataViewChoiceByIndexRenderer::CreateEditorCtrl( wxWindow *parent, wxRect labelRect, const wxVariant &value )
1447 {
1448     wxVariant string_value = GetChoice( value.GetLong() );
1449
1450     return wxDataViewChoiceRenderer::CreateEditorCtrl( parent, labelRect, string_value );
1451 }
1452
1453 bool wxDataViewChoiceByIndexRenderer::GetValueFromEditorCtrl( wxControl* editor, wxVariant &value )
1454 {
1455     wxVariant string_value;
1456     if (!wxDataViewChoiceRenderer::GetValueFromEditorCtrl( editor, string_value ))
1457         return false;
1458
1459     value = (long) GetChoices().Index( string_value.GetString() );
1460     return true;
1461 }
1462
1463 bool wxDataViewChoiceByIndexRenderer::SetValue( const wxVariant &value )
1464 {
1465     wxVariant string_value = GetChoice( value.GetLong() );
1466     return wxDataViewChoiceRenderer::SetValue( string_value );
1467 }
1468
1469 bool wxDataViewChoiceByIndexRenderer::GetValue( wxVariant &value ) const
1470 {
1471     wxVariant string_value;
1472     if (!wxDataViewChoiceRenderer::GetValue( string_value ))
1473         return false;
1474
1475     value = (long) GetChoices().Index( string_value.GetString() );
1476     return true;
1477 }
1478
1479 #endif
1480
1481 //-----------------------------------------------------------------------------
1482 // wxDataViewListStore
1483 //-----------------------------------------------------------------------------
1484
1485 wxDataViewListStore::wxDataViewListStore()
1486 {
1487 }
1488
1489 wxDataViewListStore::~wxDataViewListStore()
1490 {
1491     wxVector<wxDataViewListStoreLine*>::iterator it;
1492     for (it = m_data.begin(); it != m_data.end(); ++it)
1493     {
1494         wxDataViewListStoreLine* line = *it;
1495         delete line;
1496     }
1497 }
1498
1499 void wxDataViewListStore::PrependColumn( const wxString &varianttype )
1500 {
1501     m_cols.Insert( varianttype, 0 );
1502 }
1503
1504 void wxDataViewListStore::InsertColumn( unsigned int pos, const wxString &varianttype )
1505 {
1506     m_cols.Insert( varianttype, pos );
1507 }
1508
1509 void wxDataViewListStore::AppendColumn( const wxString &varianttype )
1510 {
1511     m_cols.Add( varianttype );
1512 }
1513
1514 unsigned int wxDataViewListStore::GetColumnCount() const
1515 {
1516     return m_cols.GetCount();
1517 }
1518
1519 wxString wxDataViewListStore::GetColumnType( unsigned int pos ) const
1520 {
1521     return m_cols[pos];
1522 }
1523
1524 void wxDataViewListStore::AppendItem( const wxVector<wxVariant> &values, wxClientData *data )
1525 {
1526     wxDataViewListStoreLine *line = new wxDataViewListStoreLine( data );
1527     line->m_values = values;
1528     m_data.push_back( line );
1529
1530     RowAppended();
1531 }
1532
1533 void wxDataViewListStore::PrependItem( const wxVector<wxVariant> &values, wxClientData *data )
1534 {
1535     wxDataViewListStoreLine *line = new wxDataViewListStoreLine( data );
1536     line->m_values = values;
1537     m_data.insert( m_data.begin(), line );
1538
1539     RowPrepended();
1540 }
1541
1542 void wxDataViewListStore::InsertItem(  unsigned int row, const wxVector<wxVariant> &values,
1543                                        wxClientData *data )
1544 {
1545     wxDataViewListStoreLine *line = new wxDataViewListStoreLine( data );
1546     line->m_values = values;
1547     m_data.insert( m_data.begin()+row, line );
1548
1549     RowInserted( row );
1550 }
1551
1552 void wxDataViewListStore::DeleteItem( unsigned int row )
1553 {
1554     wxVector<wxDataViewListStoreLine*>::iterator it = m_data.begin() + row;
1555     delete *it;
1556     m_data.erase( it );
1557
1558     RowDeleted( row );
1559 }
1560
1561 void wxDataViewListStore::DeleteAllItems()
1562 {
1563     wxVector<wxDataViewListStoreLine*>::iterator it;
1564     for (it = m_data.begin(); it != m_data.end(); ++it)
1565     {
1566         wxDataViewListStoreLine* line = *it;
1567         delete line;
1568     }
1569
1570     m_data.clear();
1571
1572     Reset( 0 );
1573 }
1574
1575 void wxDataViewListStore::GetValueByRow( wxVariant &value, unsigned int row, unsigned int col ) const
1576 {
1577     wxDataViewListStoreLine *line = m_data[row];
1578     value = line->m_values[col];
1579 }
1580
1581 bool wxDataViewListStore::SetValueByRow( const wxVariant &value, unsigned int row, unsigned int col )
1582 {
1583     wxDataViewListStoreLine *line = m_data[row];
1584     line->m_values[col] = value;
1585
1586     return true;
1587 }
1588
1589 //-----------------------------------------------------------------------------
1590 // wxDataViewListCtrl
1591 //-----------------------------------------------------------------------------
1592
1593 IMPLEMENT_DYNAMIC_CLASS(wxDataViewListCtrl,wxDataViewCtrl)
1594
1595 BEGIN_EVENT_TABLE(wxDataViewListCtrl,wxDataViewCtrl)
1596    EVT_SIZE( wxDataViewListCtrl::OnSize )
1597 END_EVENT_TABLE()
1598
1599 wxDataViewListCtrl::wxDataViewListCtrl()
1600 {
1601 }
1602
1603 wxDataViewListCtrl::wxDataViewListCtrl( wxWindow *parent, wxWindowID id,
1604            const wxPoint& pos, const wxSize& size, long style,
1605            const wxValidator& validator )
1606 {
1607     Create( parent, id, pos, size, style, validator );
1608
1609     wxDataViewListStore *store = new wxDataViewListStore;
1610     AssociateModel( store );
1611     store->DecRef();
1612 }
1613
1614 wxDataViewListCtrl::~wxDataViewListCtrl()
1615 {
1616 }
1617
1618
1619 bool wxDataViewListCtrl::Create( wxWindow *parent, wxWindowID id,
1620            const wxPoint& pos, const wxSize& size, long style,
1621            const wxValidator& validator )
1622 {
1623     return wxDataViewCtrl::Create( parent, id, pos, size, style, validator );
1624 }
1625
1626 bool wxDataViewListCtrl::AppendColumn( wxDataViewColumn *column, const wxString &varianttype )
1627 {
1628     GetStore()->AppendColumn( varianttype );
1629     return wxDataViewCtrl::AppendColumn( column );
1630 }
1631
1632 bool wxDataViewListCtrl::PrependColumn( wxDataViewColumn *column, const wxString &varianttype )
1633 {
1634     GetStore()->PrependColumn( varianttype );
1635     return wxDataViewCtrl::PrependColumn( column );
1636 }
1637
1638 bool wxDataViewListCtrl::InsertColumn( unsigned int pos, wxDataViewColumn *column, const wxString &varianttype )
1639 {
1640     GetStore()->InsertColumn( pos, varianttype );
1641     return wxDataViewCtrl::InsertColumn( pos, column );
1642 }
1643
1644 bool wxDataViewListCtrl::PrependColumn( wxDataViewColumn *col )
1645 {
1646     return PrependColumn( col, "string" );
1647 }
1648
1649 bool wxDataViewListCtrl::InsertColumn( unsigned int pos, wxDataViewColumn *col )
1650 {
1651     return InsertColumn( pos, col, "string" );
1652 }
1653
1654 bool wxDataViewListCtrl::AppendColumn( wxDataViewColumn *col )
1655 {
1656     return AppendColumn( col, "string" );
1657 }
1658
1659 wxDataViewColumn *wxDataViewListCtrl::AppendTextColumn( const wxString &label,
1660           wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1661 {
1662     GetStore()->AppendColumn( wxT("string") );
1663
1664     wxDataViewColumn *ret = new wxDataViewColumn( label,
1665         new wxDataViewTextRenderer( wxT("string"), mode ),
1666         GetStore()->GetColumnCount()-1, width, align, flags );
1667
1668     wxDataViewCtrl::AppendColumn( ret );
1669
1670     return ret;
1671 }
1672
1673 wxDataViewColumn *wxDataViewListCtrl::AppendToggleColumn( const wxString &label,
1674           wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1675 {
1676     GetStore()->AppendColumn( wxT("bool") );
1677
1678     wxDataViewColumn *ret = new wxDataViewColumn( label,
1679         new wxDataViewToggleRenderer( wxT("bool"), mode ),
1680         GetStore()->GetColumnCount()-1, width, align, flags );
1681
1682     wxDataViewCtrl::AppendColumn( ret );
1683
1684     return ret;
1685 }
1686
1687 wxDataViewColumn *wxDataViewListCtrl::AppendProgressColumn( const wxString &label,
1688           wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1689 {
1690     GetStore()->AppendColumn( wxT("long") );
1691
1692     wxDataViewColumn *ret = new wxDataViewColumn( label,
1693         new wxDataViewProgressRenderer( wxEmptyString, wxT("long"), mode ),
1694         GetStore()->GetColumnCount()-1, width, align, flags );
1695
1696     wxDataViewCtrl::AppendColumn( ret );
1697
1698     return ret;
1699 }
1700
1701 wxDataViewColumn *wxDataViewListCtrl::AppendIconTextColumn( const wxString &label,
1702           wxDataViewCellMode mode, int width, wxAlignment align, int flags )
1703 {
1704     GetStore()->AppendColumn( wxT("wxDataViewIconText") );
1705
1706     wxDataViewColumn *ret = new wxDataViewColumn( label,
1707         new wxDataViewIconTextRenderer( wxT("wxDataViewIconText"), mode ),
1708         GetStore()->GetColumnCount()-1, width, align, flags );
1709
1710     wxDataViewCtrl::AppendColumn( ret );
1711
1712     return ret;
1713 }
1714
1715 void wxDataViewListCtrl::OnSize( wxSizeEvent &event )
1716 {
1717     event.Skip( true );
1718 }
1719
1720 //-----------------------------------------------------------------------------
1721 // wxDataViewTreeStore
1722 //-----------------------------------------------------------------------------
1723
1724 wxDataViewTreeStoreNode::wxDataViewTreeStoreNode(
1725         wxDataViewTreeStoreNode *parent,
1726         const wxString &text, const wxIcon &icon, wxClientData *data )
1727 {
1728     m_parent = parent;
1729     m_text = text;
1730     m_icon = icon;
1731     m_data = data;
1732 }
1733
1734 wxDataViewTreeStoreNode::~wxDataViewTreeStoreNode()
1735 {
1736     if (m_data)
1737         delete m_data;
1738 }
1739
1740 #include "wx/listimpl.cpp"
1741 WX_DEFINE_LIST(wxDataViewTreeStoreNodeList)
1742
1743 wxDataViewTreeStoreContainerNode::wxDataViewTreeStoreContainerNode(
1744         wxDataViewTreeStoreNode *parent, const wxString &text,
1745         const wxIcon &icon, const wxIcon &expanded, wxClientData *data ) :
1746     wxDataViewTreeStoreNode( parent, text, icon, data )
1747 {
1748     m_iconExpanded = expanded;
1749     m_isExpanded = false;
1750     m_children.DeleteContents(true);
1751 }
1752
1753 wxDataViewTreeStoreContainerNode::~wxDataViewTreeStoreContainerNode()
1754 {
1755 }
1756
1757 //-----------------------------------------------------------------------------
1758
1759 wxDataViewTreeStore::wxDataViewTreeStore()
1760 {
1761     m_root = new wxDataViewTreeStoreContainerNode( NULL, wxEmptyString );
1762 }
1763
1764 wxDataViewTreeStore::~wxDataViewTreeStore()
1765 {
1766     delete m_root;
1767 }
1768
1769 wxDataViewItem wxDataViewTreeStore::AppendItem( const wxDataViewItem& parent,
1770         const wxString &text, const wxIcon &icon, wxClientData *data )
1771 {
1772     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1773     if (!parent_node) return wxDataViewItem(0);
1774
1775     wxDataViewTreeStoreNode *node =
1776         new wxDataViewTreeStoreNode( parent_node, text, icon, data );
1777     parent_node->GetChildren().Append( node );
1778
1779     return node->GetItem();
1780 }
1781
1782 wxDataViewItem wxDataViewTreeStore::PrependItem( const wxDataViewItem& parent,
1783         const wxString &text, const wxIcon &icon, wxClientData *data )
1784 {
1785     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1786     if (!parent_node) return wxDataViewItem(0);
1787
1788     wxDataViewTreeStoreNode *node =
1789         new wxDataViewTreeStoreNode( parent_node, text, icon, data );
1790     parent_node->GetChildren().Insert( node );
1791
1792     return node->GetItem();
1793 }
1794
1795 wxDataViewItem
1796 wxDataViewTreeStore::InsertItem(const wxDataViewItem& parent,
1797                                 const wxDataViewItem& previous,
1798                                 const wxString& text,
1799                                 const wxIcon& icon,
1800                                 wxClientData *data)
1801 {
1802     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1803     if (!parent_node) return wxDataViewItem(0);
1804
1805     wxDataViewTreeStoreNode *previous_node = FindNode( previous );
1806     int pos = parent_node->GetChildren().IndexOf( previous_node );
1807     if (pos == wxNOT_FOUND) return wxDataViewItem(0);
1808
1809     wxDataViewTreeStoreNode *node =
1810         new wxDataViewTreeStoreNode( parent_node, text, icon, data );
1811     parent_node->GetChildren().Insert( (size_t) pos, node );
1812
1813     return node->GetItem();
1814 }
1815
1816 wxDataViewItem wxDataViewTreeStore::PrependContainer( const wxDataViewItem& parent,
1817         const wxString &text, const wxIcon &icon, const wxIcon &expanded,
1818         wxClientData *data )
1819 {
1820     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1821     if (!parent_node) return wxDataViewItem(0);
1822
1823     wxDataViewTreeStoreContainerNode *node =
1824         new wxDataViewTreeStoreContainerNode( parent_node, text, icon, expanded, data );
1825     parent_node->GetChildren().Insert( node );
1826
1827     return node->GetItem();
1828 }
1829
1830 wxDataViewItem
1831 wxDataViewTreeStore::AppendContainer(const wxDataViewItem& parent,
1832                                      const wxString &text,
1833                                      const wxIcon& icon,
1834                                      const wxIcon& expanded,
1835                                      wxClientData * data)
1836 {
1837     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1838     if (!parent_node) return wxDataViewItem(0);
1839
1840     wxDataViewTreeStoreContainerNode *node =
1841         new wxDataViewTreeStoreContainerNode( parent_node, text, icon, expanded, data );
1842     parent_node->GetChildren().Append( node );
1843
1844     return node->GetItem();
1845 }
1846
1847 wxDataViewItem
1848 wxDataViewTreeStore::InsertContainer(const wxDataViewItem& parent,
1849                                      const wxDataViewItem& previous,
1850                                      const wxString& text,
1851                                      const wxIcon& icon,
1852                                      const wxIcon& expanded,
1853                                      wxClientData * data)
1854 {
1855     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1856     if (!parent_node) return wxDataViewItem(0);
1857
1858     wxDataViewTreeStoreNode *previous_node = FindNode( previous );
1859     int pos = parent_node->GetChildren().IndexOf( previous_node );
1860     if (pos == wxNOT_FOUND) return wxDataViewItem(0);
1861
1862     wxDataViewTreeStoreContainerNode *node =
1863         new wxDataViewTreeStoreContainerNode( parent_node, text, icon, expanded, data );
1864     parent_node->GetChildren().Insert( (size_t) pos, node );
1865
1866     return node->GetItem();
1867 }
1868
1869 bool wxDataViewTreeStore::IsContainer( const wxDataViewItem& item ) const
1870 {
1871     wxDataViewTreeStoreNode *node = FindNode( item );
1872     if (!node) return false;
1873
1874     return node->IsContainer();
1875 }
1876
1877 wxDataViewItem wxDataViewTreeStore::GetNthChild( const wxDataViewItem& parent, unsigned int pos ) const
1878 {
1879     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent );
1880     if (!parent_node) return wxDataViewItem(0);
1881
1882     wxDataViewTreeStoreNodeList::compatibility_iterator node = parent_node->GetChildren().Item( pos );
1883     if (node)
1884         return node->GetData();
1885
1886     return wxDataViewItem(0);
1887 }
1888
1889 int wxDataViewTreeStore::GetChildCount( const wxDataViewItem& parent ) const
1890 {
1891     wxDataViewTreeStoreNode *node = FindNode( parent );
1892     if (!node) return -1;
1893
1894     if (!node->IsContainer())
1895         return 0;
1896
1897     wxDataViewTreeStoreContainerNode *container_node = (wxDataViewTreeStoreContainerNode*) node;
1898     return (int) container_node->GetChildren().GetCount();
1899 }
1900
1901 void wxDataViewTreeStore::SetItemText( const wxDataViewItem& item, const wxString &text )
1902 {
1903     wxDataViewTreeStoreNode *node = FindNode( item );
1904     if (!node) return;
1905
1906     node->SetText( text );
1907 }
1908
1909 wxString wxDataViewTreeStore::GetItemText( const wxDataViewItem& item ) const
1910 {
1911     wxDataViewTreeStoreNode *node = FindNode( item );
1912     if (!node) return wxEmptyString;
1913
1914     return node->GetText();
1915 }
1916
1917 void wxDataViewTreeStore::SetItemIcon( const wxDataViewItem& item, const wxIcon &icon )
1918 {
1919     wxDataViewTreeStoreNode *node = FindNode( item );
1920     if (!node) return;
1921
1922     node->SetIcon( icon );
1923 }
1924
1925 const wxIcon &wxDataViewTreeStore::GetItemIcon( const wxDataViewItem& item ) const
1926 {
1927     wxDataViewTreeStoreNode *node = FindNode( item );
1928     if (!node) return wxNullIcon;
1929
1930     return node->GetIcon();
1931 }
1932
1933 void wxDataViewTreeStore::SetItemExpandedIcon( const wxDataViewItem& item, const wxIcon &icon )
1934 {
1935     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1936     if (!node) return;
1937
1938     node->SetExpandedIcon( icon );
1939 }
1940
1941 const wxIcon &wxDataViewTreeStore::GetItemExpandedIcon( const wxDataViewItem& item ) const
1942 {
1943     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1944     if (!node) return wxNullIcon;
1945
1946     return node->GetExpandedIcon();
1947 }
1948
1949 void wxDataViewTreeStore::SetItemData( const wxDataViewItem& item, wxClientData *data )
1950 {
1951     wxDataViewTreeStoreNode *node = FindNode( item );
1952     if (!node) return;
1953
1954     node->SetData( data );
1955 }
1956
1957 wxClientData *wxDataViewTreeStore::GetItemData( const wxDataViewItem& item ) const
1958 {
1959     wxDataViewTreeStoreNode *node = FindNode( item );
1960     if (!node) return NULL;
1961
1962     return node->GetData();
1963 }
1964
1965 void wxDataViewTreeStore::DeleteItem( const wxDataViewItem& item )
1966 {
1967     if (!item.IsOk()) return;
1968
1969     wxDataViewItem parent_item = GetParent( item );
1970
1971     wxDataViewTreeStoreContainerNode *parent_node = FindContainerNode( parent_item );
1972     if (!parent_node) return;
1973
1974     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1975     if (!node) return;
1976
1977     parent_node->GetChildren().DeleteObject( node );
1978 }
1979
1980 void wxDataViewTreeStore::DeleteChildren( const wxDataViewItem& item )
1981 {
1982     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
1983     if (!node) return;
1984
1985     node->GetChildren().clear();
1986 }
1987
1988 void wxDataViewTreeStore::DeleteAllItems()
1989 {
1990     DeleteChildren(m_root);
1991 }
1992
1993 void
1994 wxDataViewTreeStore::GetValue(wxVariant &variant,
1995                               const wxDataViewItem &item,
1996                               unsigned int WXUNUSED(col)) const
1997 {
1998     // if (col != 0) return;
1999
2000     wxDataViewTreeStoreNode *node = FindNode( item );
2001     if (!node) return;
2002
2003     wxIcon icon( node->GetIcon());
2004     if (node->IsContainer())
2005     {
2006         wxDataViewTreeStoreContainerNode *container = (wxDataViewTreeStoreContainerNode*) node;
2007         if (container->IsExpanded() && container->GetExpandedIcon().IsOk())
2008            icon = container->GetExpandedIcon();
2009     }
2010
2011     wxDataViewIconText data( node->GetText(), icon );
2012
2013     variant << data;
2014 }
2015
2016 bool
2017 wxDataViewTreeStore::SetValue(const wxVariant& variant,
2018                               const wxDataViewItem& item,
2019                               unsigned int WXUNUSED(col))
2020 {
2021     // if (col != 0) return false;
2022
2023     wxDataViewTreeStoreNode *node = FindNode( item );
2024     if (!node) return false;
2025
2026     wxDataViewIconText data;
2027
2028     data << variant;
2029
2030     node->SetText( data.GetText() );
2031     node->SetIcon( data.GetIcon() );
2032
2033     return true;
2034 }
2035
2036 wxDataViewItem wxDataViewTreeStore::GetParent( const wxDataViewItem &item ) const
2037 {
2038     wxDataViewTreeStoreNode *node = FindNode( item );
2039     if (!node) return wxDataViewItem(0);
2040
2041     wxDataViewTreeStoreNode *parent = node->GetParent();
2042     if (!parent) return wxDataViewItem(0);
2043
2044     if (parent == m_root)
2045         return wxDataViewItem(0);
2046
2047     return parent->GetItem();
2048 }
2049
2050 unsigned int wxDataViewTreeStore::GetChildren( const wxDataViewItem &item, wxDataViewItemArray &children ) const
2051 {
2052     wxDataViewTreeStoreContainerNode *node = FindContainerNode( item );
2053     if (!node) return 0;
2054
2055     wxDataViewTreeStoreNodeList::iterator iter;
2056     for (iter = node->GetChildren().begin(); iter != node->GetChildren().end(); iter++)
2057     {
2058         wxDataViewTreeStoreNode* child = *iter;
2059         children.Add( child->GetItem() );
2060     }
2061
2062     return node->GetChildren().GetCount();
2063 }
2064
2065 int wxDataViewTreeStore::Compare( const wxDataViewItem &item1, const wxDataViewItem &item2,
2066                          unsigned int WXUNUSED(column), bool WXUNUSED(ascending) ) const
2067 {
2068     wxDataViewTreeStoreNode *node1 = FindNode( item1 );
2069     wxDataViewTreeStoreNode *node2 = FindNode( item2 );
2070
2071     if (!node1 || !node2)
2072         return 0;
2073
2074     wxDataViewTreeStoreContainerNode* parent1 =
2075         (wxDataViewTreeStoreContainerNode*) node1->GetParent();
2076     wxDataViewTreeStoreContainerNode* parent2 =
2077         (wxDataViewTreeStoreContainerNode*) node2->GetParent();
2078
2079     if (parent1 != parent2)
2080     {
2081         wxLogError( wxT("Comparing items with different parent.") );
2082         return 0;
2083     }
2084
2085     if (node1->IsContainer() && !node2->IsContainer())
2086         return -1;
2087
2088     if (node2->IsContainer() && !node1->IsContainer())
2089         return 1;
2090
2091     return parent1->GetChildren().IndexOf( node1 ) - parent2->GetChildren().IndexOf( node2 );
2092 }
2093
2094 wxDataViewTreeStoreNode *wxDataViewTreeStore::FindNode( const wxDataViewItem &item ) const
2095 {
2096     if (!item.IsOk())
2097         return m_root;
2098
2099     return (wxDataViewTreeStoreNode*) item.GetID();
2100 }
2101
2102 wxDataViewTreeStoreContainerNode *wxDataViewTreeStore::FindContainerNode( const wxDataViewItem &item ) const
2103 {
2104     if (!item.IsOk())
2105         return (wxDataViewTreeStoreContainerNode*) m_root;
2106
2107     wxDataViewTreeStoreNode* node = (wxDataViewTreeStoreNode*) item.GetID();
2108
2109     if (!node->IsContainer())
2110         return NULL;
2111
2112     return (wxDataViewTreeStoreContainerNode*) node;
2113 }
2114
2115 //-----------------------------------------------------------------------------
2116 // wxDataViewTreeCtrl
2117 //-----------------------------------------------------------------------------
2118
2119 IMPLEMENT_DYNAMIC_CLASS(wxDataViewTreeCtrl,wxDataViewCtrl)
2120
2121 BEGIN_EVENT_TABLE(wxDataViewTreeCtrl,wxDataViewCtrl)
2122    EVT_DATAVIEW_ITEM_EXPANDED(-1, wxDataViewTreeCtrl::OnExpanded)
2123    EVT_DATAVIEW_ITEM_COLLAPSED(-1, wxDataViewTreeCtrl::OnCollapsed)
2124    EVT_SIZE( wxDataViewTreeCtrl::OnSize )
2125 END_EVENT_TABLE()
2126
2127 wxDataViewTreeCtrl::~wxDataViewTreeCtrl()
2128 {
2129     delete m_imageList;
2130 }
2131
2132 bool wxDataViewTreeCtrl::Create( wxWindow *parent, wxWindowID id,
2133            const wxPoint& pos, const wxSize& size, long style, const wxValidator& validator )
2134 {
2135     if ( !wxDataViewCtrl::Create( parent, id, pos, size, style, validator ) )
2136         return false;
2137
2138     // create the standard model and a column in the tree
2139     wxDataViewTreeStore *store = new wxDataViewTreeStore;
2140     AssociateModel( store );
2141     store->DecRef();
2142
2143     AppendIconTextColumn
2144     (
2145         wxString(),                 // no label (header is not shown anyhow)
2146         0,                          // the only model column
2147         wxDATAVIEW_CELL_EDITABLE,
2148         -1,                         // default width
2149         wxALIGN_NOT,                //  and alignment
2150         0                           // not resizeable
2151     );
2152
2153     return true;
2154 }
2155
2156 void wxDataViewTreeCtrl::SetImageList( wxImageList *imagelist )
2157 {
2158     delete m_imageList;
2159
2160     m_imageList = imagelist;
2161 }
2162
2163 wxDataViewItem wxDataViewTreeCtrl::AppendItem( const wxDataViewItem& parent,
2164         const wxString &text, int iconIndex, wxClientData *data )
2165 {
2166     wxIcon icon = wxNullIcon;
2167     if (m_imageList && (iconIndex != -1))
2168         icon = m_imageList->GetIcon( iconIndex );
2169
2170     wxDataViewItem res = GetStore()->AppendItem( parent, text, icon, data );
2171
2172     GetStore()->ItemAdded( parent, res );
2173
2174     return res;
2175 }
2176
2177 wxDataViewItem wxDataViewTreeCtrl::PrependItem( const wxDataViewItem& parent,
2178         const wxString &text, int iconIndex, wxClientData *data )
2179 {
2180     wxIcon icon = wxNullIcon;
2181     if (m_imageList && (iconIndex != -1))
2182         icon = m_imageList->GetIcon( iconIndex );
2183
2184     wxDataViewItem res = GetStore()->PrependItem( parent, text, icon, data );
2185
2186     GetStore()->ItemAdded( parent, res );
2187
2188     return res;
2189 }
2190
2191 wxDataViewItem wxDataViewTreeCtrl::InsertItem( const wxDataViewItem& parent, const wxDataViewItem& previous,
2192         const wxString &text, int iconIndex, wxClientData *data )
2193 {
2194     wxIcon icon = wxNullIcon;
2195     if (m_imageList && (iconIndex != -1))
2196         icon = m_imageList->GetIcon( iconIndex );
2197
2198     wxDataViewItem res = GetStore()->InsertItem( parent, previous, text, icon, data );
2199
2200     GetStore()->ItemAdded( parent, res );
2201
2202     return res;
2203 }
2204
2205 wxDataViewItem wxDataViewTreeCtrl::PrependContainer( const wxDataViewItem& parent,
2206         const wxString &text, int iconIndex, int expandedIndex, wxClientData *data )
2207 {
2208     wxIcon icon = wxNullIcon;
2209     if (m_imageList && (iconIndex != -1))
2210         icon = m_imageList->GetIcon( iconIndex );
2211
2212     wxIcon expanded = wxNullIcon;
2213     if (m_imageList && (expandedIndex != -1))
2214         expanded = m_imageList->GetIcon( expandedIndex );
2215
2216     wxDataViewItem res = GetStore()->PrependContainer( parent, text, icon, expanded, data );
2217
2218     GetStore()->ItemAdded( parent, res );
2219
2220     return res;
2221 }
2222
2223 wxDataViewItem wxDataViewTreeCtrl::AppendContainer( const wxDataViewItem& parent,
2224         const wxString &text, int iconIndex, int expandedIndex, wxClientData *data )
2225 {
2226     wxIcon icon = wxNullIcon;
2227     if (m_imageList && (iconIndex != -1))
2228         icon = m_imageList->GetIcon( iconIndex );
2229
2230     wxIcon expanded = wxNullIcon;
2231     if (m_imageList && (expandedIndex != -1))
2232         expanded = m_imageList->GetIcon( expandedIndex );
2233
2234     wxDataViewItem res = GetStore()->AppendContainer( parent, text, icon, expanded, data );
2235
2236     GetStore()->ItemAdded( parent, res );
2237
2238     return res;
2239 }
2240
2241 wxDataViewItem wxDataViewTreeCtrl::InsertContainer( const wxDataViewItem& parent, const wxDataViewItem& previous,
2242         const wxString &text, int iconIndex, int expandedIndex, wxClientData *data )
2243 {
2244     wxIcon icon = wxNullIcon;
2245     if (m_imageList && (iconIndex != -1))
2246         icon = m_imageList->GetIcon( iconIndex );
2247
2248     wxIcon expanded = wxNullIcon;
2249     if (m_imageList && (expandedIndex != -1))
2250         expanded = m_imageList->GetIcon( expandedIndex );
2251
2252     wxDataViewItem res = GetStore()->InsertContainer( parent, previous, text, icon, expanded, data );
2253
2254     GetStore()->ItemAdded( parent, res );
2255
2256     return res;
2257 }
2258
2259 void wxDataViewTreeCtrl::SetItemText( const wxDataViewItem& item, const wxString &text )
2260 {
2261     GetStore()->SetItemText(item,text);
2262
2263     // notify control
2264     GetStore()->ValueChanged( item, 0 );
2265 }
2266
2267 void wxDataViewTreeCtrl::SetItemIcon( const wxDataViewItem& item, const wxIcon &icon )
2268 {
2269     GetStore()->SetItemIcon(item,icon);
2270
2271     // notify control
2272     GetStore()->ValueChanged( item, 0 );
2273 }
2274
2275 void wxDataViewTreeCtrl::SetItemExpandedIcon( const wxDataViewItem& item, const wxIcon &icon )
2276 {
2277     GetStore()->SetItemExpandedIcon(item,icon);
2278
2279     // notify control
2280     GetStore()->ValueChanged( item, 0 );
2281 }
2282
2283 void wxDataViewTreeCtrl::DeleteItem( const wxDataViewItem& item )
2284 {
2285     wxDataViewItem parent_item = GetStore()->GetParent( item );
2286
2287     GetStore()->DeleteItem(item);
2288
2289     // notify control
2290     GetStore()->ItemDeleted( parent_item, item );
2291 }
2292
2293 void wxDataViewTreeCtrl::DeleteChildren( const wxDataViewItem& item )
2294 {
2295     wxDataViewTreeStoreContainerNode *node = GetStore()->FindContainerNode( item );
2296     if (!node) return;
2297
2298     wxDataViewItemArray array;
2299     wxDataViewTreeStoreNodeList::iterator iter;
2300     for (iter = node->GetChildren().begin(); iter != node->GetChildren().end(); iter++)
2301     {
2302         wxDataViewTreeStoreNode* child = *iter;
2303         array.Add( child->GetItem() );
2304     }
2305
2306     GetStore()->DeleteChildren( item );
2307
2308     // notify control
2309     GetStore()->ItemsDeleted( item, array );
2310 }
2311
2312 void  wxDataViewTreeCtrl::DeleteAllItems()
2313 {
2314     GetStore()->DeleteAllItems();
2315
2316     GetStore()->Cleared();
2317 }
2318
2319 void wxDataViewTreeCtrl::OnExpanded( wxDataViewEvent &event )
2320 {
2321     if (m_imageList) return;
2322
2323     wxDataViewTreeStoreContainerNode* container = GetStore()->FindContainerNode( event.GetItem() );
2324     if (!container) return;
2325
2326     container->SetExpanded( true );
2327
2328     GetStore()->ItemChanged( event.GetItem() );
2329 }
2330
2331 void wxDataViewTreeCtrl::OnCollapsed( wxDataViewEvent &event )
2332 {
2333     if (m_imageList) return;
2334
2335     wxDataViewTreeStoreContainerNode* container = GetStore()->FindContainerNode( event.GetItem() );
2336     if (!container) return;
2337
2338     container->SetExpanded( false );
2339
2340     GetStore()->ItemChanged( event.GetItem() );
2341 }
2342
2343 void wxDataViewTreeCtrl::OnSize( wxSizeEvent &event )
2344 {
2345 #if defined(wxUSE_GENERICDATAVIEWCTRL)
2346     // automatically resize our only column to take the entire control width
2347     if ( GetColumnCount() )
2348     {
2349         wxSize size = GetClientSize();
2350         GetColumn(0)->SetWidth(size.x);
2351     }
2352 #endif
2353     event.Skip( true );
2354 }
2355
2356 #endif // wxUSE_DATAVIEWCTRL
2357