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