src/generic/datavgen.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/generic/datavgen.cpp
   3 // Purpose:     wxDataViewCtrl generic implementation
   4 // Author:      Robert Roebling
   5 // Modified by: Francesco Montorsi, Guru Kathiresan, Bo Yang
   6 // Id:          $Id$
   7 // Copyright:   (c) 1998 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 #ifdef wxUSE_GENERICDATAVIEWCTRL
  23
  24 #ifndef WX_PRECOMP
  25     #ifdef __WXMSW__
  26         #include "wx/msw/private.h"
  27         #include "wx/msw/wrapwin.h"
  28         #include "wx/msw/wrapcctl.h" // include <commctrl.h> "properly"
  29     #endif
  30     #include "wx/sizer.h"
  31     #include "wx/log.h"
  32     #include "wx/dcclient.h"
  33     #include "wx/timer.h"
  34     #include "wx/settings.h"
  35     #include "wx/msgdlg.h"
  36     #include "wx/dcscreen.h"
  37     #include "wx/frame.h"
  38 #endif
  39
  40 #include "wx/stockitem.h"
  41 #include "wx/popupwin.h"
  42 #include "wx/renderer.h"
  43 #include "wx/dcbuffer.h"
  44 #include "wx/icon.h"
  45 #include "wx/list.h"
  46 #include "wx/listimpl.cpp"
  47 #include "wx/imaglist.h"
  48 #include "wx/headerctrl.h"
  49 #include "wx/dnd.h"
  50 #include "wx/stopwatch.h"
  51 #include "wx/weakref.h"
  52
  53 //-----------------------------------------------------------------------------
  54 // classes
  55 //-----------------------------------------------------------------------------
  56
  57 class wxDataViewColumn;
  58 class wxDataViewHeaderWindow;
  59 class wxDataViewCtrl;
  60
  61 //-----------------------------------------------------------------------------
  62 // classes
  63 //-----------------------------------------------------------------------------
  64
  65 static const int SCROLL_UNIT_X = 15;
  66
  67 // the cell padding on the left/right
  68 static const int PADDING_RIGHTLEFT = 3;
  69
  70 // the expander space margin
  71 static const int EXPANDER_MARGIN = 4;
  72
  73 #ifdef __WXMSW__
  74 static const int EXPANDER_OFFSET = 4;
  75 #else
  76 static const int EXPANDER_OFFSET = 1;
  77 #endif
  78
  79 // Below is the compare stuff.
  80 // For the generic implementation, both the leaf nodes and the nodes are sorted for
  81 // fast search when needed
  82 static wxDataViewModel* g_model;
  83 static int g_column = -2;
  84 static bool g_asending = true;
  85
  86 // ----------------------------------------------------------------------------
  87 // helper functions
  88 // ----------------------------------------------------------------------------
  89
  90 namespace
  91 {
  92
  93 // Return the expander column or, if it is not set, the first column and also
  94 // set it as the expander one for the future.
  95 wxDataViewColumn* GetExpanderColumnOrFirstOne(wxDataViewCtrl* dataview)
  96 {
  97     wxDataViewColumn* expander = dataview->GetExpanderColumn();
  98     if (!expander)
  99     {
 100         // TODO-RTL: last column for RTL support
 101         expander = dataview->GetColumnAt( 0 );
 102         dataview->SetExpanderColumn(expander);
 103     }
 104
 105     return expander;
 106 }
 107
 108 } // anonymous namespace
 109
 110 //-----------------------------------------------------------------------------
 111 // wxDataViewColumn
 112 //-----------------------------------------------------------------------------
 113
 114 void wxDataViewColumn::Init(int width, wxAlignment align, int flags)
 115 {
 116     m_width = width;
 117     m_minWidth = 0;
 118     m_align = align;
 119     m_flags = flags;
 120     m_sort = false;
 121     m_sortAscending = true;
 122 }
 123
 124 int wxDataViewColumn::GetWidth() const
 125 {
 126     switch ( m_width )
 127     {
 128         case wxCOL_WIDTH_DEFAULT:
 129             return wxDVC_DEFAULT_WIDTH;
 130
 131         case wxCOL_WIDTH_AUTOSIZE:
 132             wxCHECK_MSG( m_owner, wxDVC_DEFAULT_WIDTH, "no owner control" );
 133             return m_owner->GetBestColumnWidth(m_owner->GetColumnIndex(this));
 134
 135         default:
 136             return m_width;
 137     }
 138 }
 139
 140 void wxDataViewColumn::UpdateDisplay()
 141 {
 142     if (m_owner)
 143     {
 144         int idx = m_owner->GetColumnIndex( this );
 145         m_owner->OnColumnChange( idx );
 146     }
 147 }
 148
 149 void wxDataViewColumn::SetSortOrder(bool ascending)
 150 {
 151     if ( !m_owner )
 152         return;
 153
 154     // First unset the old sort column if any.
 155     int oldSortKey = m_owner->GetSortingColumnIndex();
 156     if ( oldSortKey != wxNOT_FOUND )
 157     {
 158         m_owner->GetColumn(oldSortKey)->UnsetAsSortKey();
 159     }
 160
 161     // Now set this one as the new sort column.
 162     const int idx = m_owner->GetColumnIndex(this);
 163     m_owner->SetSortingColumnIndex(idx);
 164
 165     m_sort = true;
 166     m_sortAscending = ascending;
 167
 168     // Call this directly instead of using UpdateDisplay() as we already have
 169     // the column index, no need to look it up again.
 170     m_owner->OnColumnChange(idx);
 171 }
 172
 173 //-----------------------------------------------------------------------------
 174 // wxDataViewHeaderWindow
 175 //-----------------------------------------------------------------------------
 176
 177 class wxDataViewHeaderWindow : public wxHeaderCtrl
 178 {
 179 public:
 180     wxDataViewHeaderWindow(wxDataViewCtrl *parent)
 181         : wxHeaderCtrl(parent)
 182     {
 183     }
 184
 185     wxDataViewCtrl *GetOwner() const
 186         { return static_cast<wxDataViewCtrl *>(GetParent()); }
 187
 188 protected:
 189     // implement/override wxHeaderCtrl functions by forwarding them to the main
 190     // control
 191     virtual const wxHeaderColumn& GetColumn(unsigned int idx) const
 192     {
 193         return *(GetOwner()->GetColumn(idx));
 194     }
 195
 196     virtual bool UpdateColumnWidthToFit(unsigned int idx, int widthTitle)
 197     {
 198         wxDataViewCtrl * const owner = GetOwner();
 199
 200         int widthContents = owner->GetBestColumnWidth(idx);
 201         owner->GetColumn(idx)->SetWidth(wxMax(widthTitle, widthContents));
 202         owner->OnColumnChange(idx);
 203
 204         return true;
 205     }
 206
 207 private:
 208     bool SendEvent(wxEventType type, unsigned int n)
 209     {
 210         wxDataViewCtrl * const owner = GetOwner();
 211         wxDataViewEvent event(type, owner->GetId());
 212
 213         event.SetEventObject(owner);
 214         event.SetColumn(n);
 215         event.SetDataViewColumn(owner->GetColumn(n));
 216         event.SetModel(owner->GetModel());
 217
 218         // for events created by wxDataViewHeaderWindow the
 219         // row / value fields are not valid
 220         return owner->ProcessWindowEvent(event);
 221     }
 222
 223     void OnClick(wxHeaderCtrlEvent& event)
 224     {
 225         const unsigned idx = event.GetColumn();
 226
 227         if ( SendEvent(wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_CLICK, idx) )
 228             return;
 229
 230         // default handling for the column click is to sort by this column or
 231         // toggle its sort order
 232         wxDataViewCtrl * const owner = GetOwner();
 233         wxDataViewColumn * const col = owner->GetColumn(idx);
 234         if ( !col->IsSortable() )
 235         {
 236             // no default handling for non-sortable columns
 237             event.Skip();
 238             return;
 239         }
 240
 241         if ( col->IsSortKey() )
 242         {
 243             // already using this column for sorting, just change the order
 244             col->ToggleSortOrder();
 245         }
 246         else // not using this column for sorting yet
 247         {
 248             col->SetSortOrder(true);
 249         }
 250
 251         wxDataViewModel * const model = owner->GetModel();
 252         if ( model )
 253             model->Resort();
 254
 255         owner->OnColumnChange(idx);
 256
 257         SendEvent(wxEVT_COMMAND_DATAVIEW_COLUMN_SORTED, idx);
 258     }
 259
 260     void OnRClick(wxHeaderCtrlEvent& event)
 261     {
 262         if ( !SendEvent(wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK,
 263                         event.GetColumn()) )
 264             event.Skip();
 265     }
 266
 267     void OnResize(wxHeaderCtrlEvent& event)
 268     {
 269         wxDataViewCtrl * const owner = GetOwner();
 270
 271         const unsigned col = event.GetColumn();
 272         owner->GetColumn(col)->SetWidth(event.GetWidth());
 273         GetOwner()->OnColumnChange(col);
 274     }
 275
 276     void OnEndReorder(wxHeaderCtrlEvent& event)
 277     {
 278         wxDataViewCtrl * const owner = GetOwner();
 279         owner->ColumnMoved(owner->GetColumn(event.GetColumn()),
 280                         event.GetNewOrder());
 281     }
 282
 283     DECLARE_EVENT_TABLE()
 284     wxDECLARE_NO_COPY_CLASS(wxDataViewHeaderWindow);
 285 };
 286
 287 BEGIN_EVENT_TABLE(wxDataViewHeaderWindow, wxHeaderCtrl)
 288     EVT_HEADER_CLICK(wxID_ANY, wxDataViewHeaderWindow::OnClick)
 289     EVT_HEADER_RIGHT_CLICK(wxID_ANY, wxDataViewHeaderWindow::OnRClick)
 290
 291     EVT_HEADER_RESIZING(wxID_ANY, wxDataViewHeaderWindow::OnResize)
 292     EVT_HEADER_END_RESIZE(wxID_ANY, wxDataViewHeaderWindow::OnResize)
 293
 294     EVT_HEADER_END_REORDER(wxID_ANY, wxDataViewHeaderWindow::OnEndReorder)
 295 END_EVENT_TABLE()
 296
 297 //-----------------------------------------------------------------------------
 298 // wxDataViewRenameTimer
 299 //-----------------------------------------------------------------------------
 300
 301 class wxDataViewRenameTimer: public wxTimer
 302 {
 303 private:
 304     wxDataViewMainWindow *m_owner;
 305
 306 public:
 307     wxDataViewRenameTimer( wxDataViewMainWindow *owner );
 308     void Notify();
 309 };
 310
 311 //-----------------------------------------------------------------------------
 312 // wxDataViewTreeNode
 313 //-----------------------------------------------------------------------------
 314
 315 class wxDataViewTreeNode;
 316 WX_DEFINE_ARRAY( wxDataViewTreeNode *, wxDataViewTreeNodes );
 317
 318 int LINKAGEMODE wxGenericTreeModelNodeCmp( wxDataViewTreeNode ** node1,
 319                                            wxDataViewTreeNode ** node2);
 320
 321 class wxDataViewTreeNode
 322 {
 323 public:
 324     wxDataViewTreeNode(wxDataViewTreeNode *parent, const wxDataViewItem& item)
 325         : m_parent(parent),
 326           m_item(item),
 327           m_branchData(NULL)
 328     {
 329     }
 330
 331     ~wxDataViewTreeNode()
 332     {
 333         if ( m_branchData )
 334         {
 335             wxDataViewTreeNodes& nodes = m_branchData->children;
 336             for ( wxDataViewTreeNodes::iterator i = nodes.begin();
 337                   i != nodes.end();
 338                   ++i )
 339             {
 340                 delete *i;
 341             }
 342
 343             delete m_branchData;
 344         }
 345     }
 346
 347     static wxDataViewTreeNode* CreateRootNode()
 348     {
 349         wxDataViewTreeNode *n = new wxDataViewTreeNode(NULL, wxDataViewItem());
 350         n->m_branchData = new BranchNodeData;
 351         n->m_branchData->open = true;
 352         return n;
 353     }
 354
 355     wxDataViewTreeNode * GetParent() const { return m_parent; }
 356
 357     const wxDataViewTreeNodes& GetChildNodes() const
 358     {
 359         wxASSERT( m_branchData != NULL );
 360         return m_branchData->children;
 361     }
 362
 363     void InsertChild(wxDataViewTreeNode *node, unsigned index)
 364     {
 365         if ( !m_branchData )
 366             m_branchData = new BranchNodeData;
 367
 368         m_branchData->children.Insert(node, index);
 369
 370         // TODO: insert into sorted array directly in O(log n) instead of resorting in O(n log n)
 371         if (g_column >= -1)
 372             m_branchData->children.Sort( &wxGenericTreeModelNodeCmp );
 373     }
 374
 375     void RemoveChild(wxDataViewTreeNode *node)
 376     {
 377         wxCHECK_RET( m_branchData != NULL, "leaf node doesn't have children" );
 378         m_branchData->children.Remove(node);
 379     }
 380
 381     // returns position of child node for given item in children list or wxNOT_FOUND
 382     int FindChildByItem(const wxDataViewItem& item) const
 383     {
 384         if ( !m_branchData )
 385             return wxNOT_FOUND;
 386
 387         const wxDataViewTreeNodes& nodes = m_branchData->children;
 388         const int len = nodes.size();
 389         for ( int i = 0; i < len; i++ )
 390         {
 391             if ( nodes[i]->m_item == item )
 392                 return i;
 393         }
 394         return wxNOT_FOUND;
 395     }
 396
 397     const wxDataViewItem & GetItem() const { return m_item; }
 398     void SetItem( const wxDataViewItem & item ) { m_item = item; }
 399
 400     int GetIndentLevel() const
 401     {
 402         int ret = 0;
 403         const wxDataViewTreeNode * node = this;
 404         while( node->GetParent()->GetParent() != NULL )
 405         {
 406             node = node->GetParent();
 407             ret ++;
 408         }
 409         return ret;
 410     }
 411
 412     bool IsOpen() const
 413     {
 414         return m_branchData && m_branchData->open;
 415     }
 416
 417     void ToggleOpen()
 418     {
 419         // We do not allow the (invisible) root node to be collapsed because
 420         // there is no way to expand it again.
 421         if ( !m_parent )
 422             return;
 423
 424         wxCHECK_RET( m_branchData != NULL, "can't open leaf node" );
 425
 426         int sum = 0;
 427
 428         const wxDataViewTreeNodes& nodes = m_branchData->children;
 429         const int len = nodes.GetCount();
 430         for ( int i = 0;i < len; i ++)
 431             sum += 1 + nodes[i]->GetSubTreeCount();
 432
 433         if (m_branchData->open)
 434         {
 435             ChangeSubTreeCount(-sum);
 436             m_branchData->open = !m_branchData->open;
 437         }
 438         else
 439         {
 440             m_branchData->open = !m_branchData->open;
 441             ChangeSubTreeCount(+sum);
 442         }
 443     }
 444
 445     // "HasChildren" property corresponds to model's IsContainer(). Note that it may be true
 446     // even if GetChildNodes() is empty; see below.
 447     bool HasChildren() const
 448     {
 449         return m_branchData != NULL;
 450     }
 451
 452     void SetHasChildren(bool has)
 453     {
 454         // The invisible root item always has children, so ignore any attempts
 455         // to change this.
 456         if ( !m_parent )
 457             return;
 458
 459         if ( !has )
 460         {
 461             wxDELETE(m_branchData);
 462         }
 463         else if ( m_branchData == NULL )
 464         {
 465             m_branchData = new BranchNodeData;
 466         }
 467     }
 468
 469     int GetSubTreeCount() const
 470     {
 471         return m_branchData ? m_branchData->subTreeCount : 0;
 472     }
 473
 474     void ChangeSubTreeCount( int num )
 475     {
 476         wxASSERT( m_branchData != NULL );
 477
 478         if( !m_branchData->open )
 479             return;
 480
 481         m_branchData->subTreeCount += num;
 482         wxASSERT( m_branchData->subTreeCount >= 0 );
 483
 484         if( m_parent )
 485             m_parent->ChangeSubTreeCount(num);
 486     }
 487
 488     void Resort()
 489     {
 490         if ( !m_branchData )
 491             return;
 492
 493         if (g_column >= -1)
 494         {
 495             wxDataViewTreeNodes& nodes = m_branchData->children;
 496
 497             nodes.Sort( &wxGenericTreeModelNodeCmp );
 498             int len = nodes.GetCount();
 499             for (int i = 0; i < len; i ++)
 500             {
 501                 if ( nodes[i]->HasChildren() )
 502                     nodes[i]->Resort();
 503             }
 504         }
 505     }
 506
 507
 508 private:
 509     wxDataViewTreeNode  *m_parent;
 510
 511     // Corresponding model item.
 512     wxDataViewItem       m_item;
 513
 514     // Data specific to non-leaf (branch, inner) nodes. They are kept in a
 515     // separate struct in order to conserve memory.
 516     struct BranchNodeData
 517     {
 518         BranchNodeData()
 519             : open(false),
 520               subTreeCount(0)
 521         {
 522         }
 523
 524         // Child nodes. Note that this may be empty even if m_hasChildren in
 525         // case this branch of the tree wasn't expanded and realized yet.
 526         wxDataViewTreeNodes  children;
 527
 528         // Is the branch node currently open (expanded)?
 529         bool                 open;
 530
 531         // Total count of expanded (i.e. visible with the help of some
 532         // scrolling) items in the subtree, but excluding this node. I.e. it is
 533         // 0 for leaves and is the number of rows the subtree occupies for
 534         // branch nodes.
 535         int                  subTreeCount;
 536     };
 537
 538     BranchNodeData *m_branchData;
 539 };
 540
 541
 542 int LINKAGEMODE wxGenericTreeModelNodeCmp( wxDataViewTreeNode ** node1,
 543                                            wxDataViewTreeNode ** node2)
 544 {
 545     return g_model->Compare( (*node1)->GetItem(), (*node2)->GetItem(), g_column, g_asending );
 546 }
 547
 548
 549 //-----------------------------------------------------------------------------
 550 // wxDataViewMainWindow
 551 //-----------------------------------------------------------------------------
 552
 553 WX_DEFINE_SORTED_ARRAY_SIZE_T(unsigned int, wxDataViewSelection);
 554
 555 class wxDataViewMainWindow: public wxWindow
 556 {
 557 public:
 558     wxDataViewMainWindow( wxDataViewCtrl *parent,
 559                             wxWindowID id,
 560                             const wxPoint &pos = wxDefaultPosition,
 561                             const wxSize &size = wxDefaultSize,
 562                             const wxString &name = wxT("wxdataviewctrlmainwindow") );
 563     virtual ~wxDataViewMainWindow();
 564
 565     bool IsList() const { return GetModel()->IsListModel(); }
 566     bool IsVirtualList() const { return m_root == NULL; }
 567
 568     // notifications from wxDataViewModel
 569     bool ItemAdded( const wxDataViewItem &parent, const wxDataViewItem &item );
 570     bool ItemDeleted( const wxDataViewItem &parent, const wxDataViewItem &item );
 571     bool ItemChanged( const wxDataViewItem &item );
 572     bool ValueChanged( const wxDataViewItem &item, unsigned int model_column );
 573     bool Cleared();
 574     void Resort()
 575     {
 576         if (!IsVirtualList())
 577         {
 578             SortPrepare();
 579             m_root->Resort();
 580         }
 581         UpdateDisplay();
 582     }
 583
 584     void SortPrepare()
 585     {
 586         g_model = GetModel();
 587         wxDataViewColumn* col = GetOwner()->GetSortingColumn();
 588         if( !col )
 589         {
 590             if (g_model->HasDefaultCompare())
 591                 g_column = -1;
 592             else
 593                 g_column = -2;
 594
 595             g_asending = true;
 596             return;
 597         }
 598         g_column = col->GetModelColumn();
 599         g_asending = col->IsSortOrderAscending();
 600     }
 601
 602     void SetOwner( wxDataViewCtrl* owner ) { m_owner = owner; }
 603     wxDataViewCtrl *GetOwner() { return m_owner; }
 604     const wxDataViewCtrl *GetOwner() const { return m_owner; }
 605
 606     wxDataViewModel* GetModel() { return GetOwner()->GetModel(); }
 607     const wxDataViewModel* GetModel() const { return GetOwner()->GetModel(); }
 608
 609 #if wxUSE_DRAG_AND_DROP
 610     wxBitmap CreateItemBitmap( unsigned int row, int &indent );
 611 #endif // wxUSE_DRAG_AND_DROP
 612     void OnPaint( wxPaintEvent &event );
 613     void OnCharHook( wxKeyEvent &event );
 614     void OnChar( wxKeyEvent &event );
 615     void OnVerticalNavigation(unsigned int newCurrent, const wxKeyEvent& event);
 616     void OnLeftKey();
 617     void OnRightKey();
 618     void OnMouse( wxMouseEvent &event );
 619     void OnSetFocus( wxFocusEvent &event );
 620     void OnKillFocus( wxFocusEvent &event );
 621
 622     void UpdateDisplay();
 623     void RecalculateDisplay();
 624     void OnInternalIdle();
 625
 626     void OnRenameTimer();
 627
 628     void ScrollWindow( int dx, int dy, const wxRect *rect = NULL );
 629     void ScrollTo( int rows, int column );
 630
 631     unsigned GetCurrentRow() const { return m_currentRow; }
 632     bool HasCurrentRow() { return m_currentRow != (unsigned int)-1; }
 633     void ChangeCurrentRow( unsigned int row );
 634     bool TryAdvanceCurrentColumn(wxDataViewTreeNode *node, bool forward);
 635
 636     wxDataViewColumn *GetCurrentColumn() const { return m_currentCol; }
 637     void ClearCurrentColumn() { m_currentCol = NULL; }
 638
 639     bool IsSingleSel() const { return !GetParent()->HasFlag(wxDV_MULTIPLE); }
 640     bool IsEmpty() { return GetRowCount() == 0; }
 641
 642     int GetCountPerPage() const;
 643     int GetEndOfLastCol() const;
 644     unsigned int GetFirstVisibleRow() const;
 645
 646     // I change this method to un const because in the tree view,
 647     // the displaying number of the tree are changing along with the
 648     // expanding/collapsing of the tree nodes
 649     unsigned int GetLastVisibleRow();
 650     unsigned int GetRowCount() const;
 651
 652     const wxDataViewSelection& GetSelections() const { return m_selection; }
 653     void SetSelections( const wxDataViewSelection & sel )
 654         { m_selection = sel; UpdateDisplay(); }
 655     void Select( const wxArrayInt& aSelections );
 656     void SelectAllRows( bool on );
 657     void SelectRow( unsigned int row, bool on );
 658     void SelectRows( unsigned int from, unsigned int to, bool on );
 659     void ReverseRowSelection( unsigned int row );
 660     bool IsRowSelected( unsigned int row );
 661     void SendSelectionChangedEvent( const wxDataViewItem& item);
 662
 663     void RefreshRow( unsigned int row );
 664     void RefreshRows( unsigned int from, unsigned int to );
 665     void RefreshRowsAfter( unsigned int firstRow );
 666
 667     // returns the colour to be used for drawing the rules
 668     wxColour GetRuleColour() const
 669     {
 670         return wxSystemSettings::GetColour(wxSYS_COLOUR_3DLIGHT);
 671     }
 672
 673     wxRect GetLineRect( unsigned int row ) const;
 674
 675     int GetLineStart( unsigned int row ) const;  // row * m_lineHeight in fixed mode
 676     int GetLineHeight( unsigned int row ) const; // m_lineHeight in fixed mode
 677     int GetLineAt( unsigned int y ) const;       // y / m_lineHeight in fixed mode
 678
 679     void SetRowHeight( int lineHeight ) { m_lineHeight = lineHeight; }
 680     int GetRowHeight() const { return m_lineHeight; }
 681
 682     // Some useful functions for row and item mapping
 683     wxDataViewItem GetItemByRow( unsigned int row ) const;
 684     int GetRowByItem( const wxDataViewItem & item ) const;
 685
 686     wxDataViewTreeNode * GetTreeNodeByRow( unsigned int row ) const;
 687     // We did not need this temporarily
 688     // wxDataViewTreeNode * GetTreeNodeByItem( const wxDataViewItem & item );
 689
 690     // Methods for building the mapping tree
 691     void BuildTree( wxDataViewModel  * model );
 692     void DestroyTree();
 693     void HitTest( const wxPoint & point, wxDataViewItem & item, wxDataViewColumn* &column );
 694     wxRect GetItemRect( const wxDataViewItem & item, const wxDataViewColumn* column );
 695
 696     void Expand( unsigned int row );
 697     void Collapse( unsigned int row );
 698     bool IsExpanded( unsigned int row ) const;
 699     bool HasChildren( unsigned int row ) const;
 700
 701 #if wxUSE_DRAG_AND_DROP
 702     bool EnableDragSource( const wxDataFormat &format );
 703     bool EnableDropTarget( const wxDataFormat &format );
 704
 705     void RemoveDropHint();
 706     wxDragResult OnDragOver( wxDataFormat format, wxCoord x, wxCoord y, wxDragResult def );
 707     bool OnDrop( wxDataFormat format, wxCoord x, wxCoord y );
 708     wxDragResult OnData( wxDataFormat format, wxCoord x, wxCoord y, wxDragResult def );
 709     void OnLeave();
 710 #endif // wxUSE_DRAG_AND_DROP
 711
 712     void OnColumnsCountChanged();
 713
 714     // Called by wxDataViewCtrl and our own OnRenameTimer() to start edit the
 715     // specified item in the given column.
 716     void StartEditing(const wxDataViewItem& item, const wxDataViewColumn* col);
 717
 718 private:
 719     int RecalculateCount() const;
 720
 721     // Return false only if the event was vetoed by its handler.
 722     bool SendExpanderEvent(wxEventType type, const wxDataViewItem& item);
 723
 724     wxDataViewTreeNode * FindNode( const wxDataViewItem & item );
 725
 726     wxDataViewColumn *FindColumnForEditing(const wxDataViewItem& item, wxDataViewCellMode mode);
 727
 728     bool IsCellEditableInMode(const wxDataViewItem& item, const wxDataViewColumn *col, wxDataViewCellMode mode) const;
 729
 730     void DrawCellBackground( wxDataViewRenderer* cell, wxDC& dc, const wxRect& rect );
 731
 732 private:
 733     wxDataViewCtrl             *m_owner;
 734     int                         m_lineHeight;
 735     bool                        m_dirty;
 736
 737     wxDataViewColumn           *m_currentCol;
 738     unsigned int                m_currentRow;
 739     wxDataViewSelection         m_selection;
 740
 741     wxDataViewRenameTimer      *m_renameTimer;
 742     bool                        m_lastOnSame;
 743
 744     bool                        m_hasFocus;
 745     bool                        m_useCellFocus;
 746     bool                        m_currentColSetByKeyboard;
 747
 748 #if wxUSE_DRAG_AND_DROP
 749     int                         m_dragCount;
 750     wxPoint                     m_dragStart;
 751
 752     bool                        m_dragEnabled;
 753     wxDataFormat                m_dragFormat;
 754
 755     bool                        m_dropEnabled;
 756     wxDataFormat                m_dropFormat;
 757     bool                        m_dropHint;
 758     unsigned int                m_dropHintLine;
 759 #endif // wxUSE_DRAG_AND_DROP
 760
 761     // for double click logic
 762     unsigned int m_lineLastClicked,
 763         m_lineBeforeLastClicked,
 764         m_lineSelectSingleOnUp;
 765
 766     // the pen used to draw horiz/vertical rules
 767     wxPen m_penRule;
 768
 769     // the pen used to draw the expander and the lines
 770     wxPen m_penExpander;
 771
 772     // This is the tree structure of the model
 773     wxDataViewTreeNode * m_root;
 774     int m_count;
 775
 776     // This is the tree node under the cursor
 777     wxDataViewTreeNode * m_underMouse;
 778
 779     // The control used for editing or NULL.
 780     wxWeakRef<wxWindow> m_editorCtrl;
 781
 782     // Id m_editorCtrl is non-NULL, pointer to the associated renderer.
 783     wxDataViewRenderer* m_editorRenderer;
 784
 785 private:
 786     DECLARE_DYNAMIC_CLASS(wxDataViewMainWindow)
 787     DECLARE_EVENT_TABLE()
 788 };
 789
 790 // ---------------------------------------------------------
 791 // wxGenericDataViewModelNotifier
 792 // ---------------------------------------------------------
 793
 794 class wxGenericDataViewModelNotifier: public wxDataViewModelNotifier
 795 {
 796 public:
 797     wxGenericDataViewModelNotifier( wxDataViewMainWindow *mainWindow )
 798         { m_mainWindow = mainWindow; }
 799
 800     virtual bool ItemAdded( const wxDataViewItem & parent, const wxDataViewItem & item )
 801         { return m_mainWindow->ItemAdded( parent , item ); }
 802     virtual bool ItemDeleted( const wxDataViewItem &parent, const wxDataViewItem &item )
 803         { return m_mainWindow->ItemDeleted( parent, item ); }
 804     virtual bool ItemChanged( const wxDataViewItem & item )
 805         { return m_mainWindow->ItemChanged(item);  }
 806     virtual bool ValueChanged( const wxDataViewItem & item , unsigned int col )
 807         { return m_mainWindow->ValueChanged( item, col ); }
 808     virtual bool Cleared()
 809         { return m_mainWindow->Cleared(); }
 810     virtual void Resort()
 811         { m_mainWindow->Resort(); }
 812
 813     wxDataViewMainWindow    *m_mainWindow;
 814 };
 815
 816 // ---------------------------------------------------------
 817 // wxDataViewRenderer
 818 // ---------------------------------------------------------
 819
 820 IMPLEMENT_ABSTRACT_CLASS(wxDataViewRenderer, wxDataViewRendererBase)
 821
 822 wxDataViewRenderer::wxDataViewRenderer( const wxString &varianttype,
 823                                         wxDataViewCellMode mode,
 824                                         int align) :
 825     wxDataViewCustomRendererBase( varianttype, mode, align )
 826 {
 827     m_align = align;
 828     m_mode = mode;
 829     m_ellipsizeMode = wxELLIPSIZE_MIDDLE;
 830     m_dc = NULL;
 831 }
 832
 833 wxDataViewRenderer::~wxDataViewRenderer()
 834 {
 835     delete m_dc;
 836 }
 837
 838 wxDC *wxDataViewRenderer::GetDC()
 839 {
 840     if (m_dc == NULL)
 841     {
 842         if (GetOwner() == NULL)
 843             return NULL;
 844         if (GetOwner()->GetOwner() == NULL)
 845             return NULL;
 846         m_dc = new wxClientDC( GetOwner()->GetOwner() );
 847     }
 848
 849     return m_dc;
 850 }
 851
 852 void wxDataViewRenderer::SetAlignment( int align )
 853 {
 854     m_align=align;
 855 }
 856
 857 int wxDataViewRenderer::GetAlignment() const
 858 {
 859     return m_align;
 860 }
 861
 862 // ---------------------------------------------------------
 863 // wxDataViewCustomRenderer
 864 // ---------------------------------------------------------
 865
 866 IMPLEMENT_ABSTRACT_CLASS(wxDataViewCustomRenderer, wxDataViewRenderer)
 867
 868 wxDataViewCustomRenderer::wxDataViewCustomRenderer( const wxString &varianttype,
 869                         wxDataViewCellMode mode, int align ) :
 870     wxDataViewRenderer( varianttype, mode, align )
 871 {
 872 }
 873
 874 // ---------------------------------------------------------
 875 // wxDataViewTextRenderer
 876 // ---------------------------------------------------------
 877
 878 IMPLEMENT_CLASS(wxDataViewTextRenderer, wxDataViewRenderer)
 879
 880 wxDataViewTextRenderer::wxDataViewTextRenderer( const wxString &varianttype,
 881                                                 wxDataViewCellMode mode, int align ) :
 882     wxDataViewRenderer( varianttype, mode, align )
 883 {
 884 }
 885
 886 bool wxDataViewTextRenderer::SetValue( const wxVariant &value )
 887 {
 888     m_text = value.GetString();
 889
 890     return true;
 891 }
 892
 893 bool wxDataViewTextRenderer::GetValue( wxVariant& WXUNUSED(value) ) const
 894 {
 895     return false;
 896 }
 897
 898 bool wxDataViewTextRenderer::HasEditorCtrl() const
 899 {
 900     return true;
 901 }
 902
 903 wxWindow* wxDataViewTextRenderer::CreateEditorCtrl( wxWindow *parent,
 904         wxRect labelRect, const wxVariant &value )
 905 {
 906     wxTextCtrl* ctrl = new wxTextCtrl( parent, wxID_ANY, value,
 907                                        wxPoint(labelRect.x,labelRect.y),
 908                                        wxSize(labelRect.width,labelRect.height),
 909                                        wxTE_PROCESS_ENTER );
 910
 911     // select the text in the control an place the cursor at the end
 912     ctrl->SetInsertionPointEnd();
 913     ctrl->SelectAll();
 914
 915     return ctrl;
 916 }
 917
 918 bool wxDataViewTextRenderer::GetValueFromEditorCtrl( wxWindow *editor, wxVariant &value )
 919 {
 920     wxTextCtrl *text = (wxTextCtrl*) editor;
 921     value = text->GetValue();
 922     return true;
 923 }
 924
 925 bool wxDataViewTextRenderer::Render(wxRect rect, wxDC *dc, int state)
 926 {
 927     RenderText(m_text, 0, rect, dc, state);
 928     return true;
 929 }
 930
 931 wxSize wxDataViewTextRenderer::GetSize() const
 932 {
 933     if (!m_text.empty())
 934         return GetTextExtent(m_text);
 935     else
 936         return wxSize(wxDVC_DEFAULT_RENDERER_SIZE,wxDVC_DEFAULT_RENDERER_SIZE);
 937 }
 938
 939 // ---------------------------------------------------------
 940 // wxDataViewBitmapRenderer
 941 // ---------------------------------------------------------
 942
 943 IMPLEMENT_CLASS(wxDataViewBitmapRenderer, wxDataViewRenderer)
 944
 945 wxDataViewBitmapRenderer::wxDataViewBitmapRenderer( const wxString &varianttype,
 946                                                     wxDataViewCellMode mode, int align ) :
 947     wxDataViewRenderer( varianttype, mode, align )
 948 {
 949 }
 950
 951 bool wxDataViewBitmapRenderer::SetValue( const wxVariant &value )
 952 {
 953     if (value.GetType() == wxT("wxBitmap"))
 954         m_bitmap << value;
 955     if (value.GetType() == wxT("wxIcon"))
 956         m_icon << value;
 957
 958     return true;
 959 }
 960
 961 bool wxDataViewBitmapRenderer::GetValue( wxVariant& WXUNUSED(value) ) const
 962 {
 963     return false;
 964 }
 965
 966 bool wxDataViewBitmapRenderer::Render( wxRect cell, wxDC *dc, int WXUNUSED(state) )
 967 {
 968     if (m_bitmap.IsOk())
 969         dc->DrawBitmap( m_bitmap, cell.x, cell.y );
 970     else if (m_icon.IsOk())
 971         dc->DrawIcon( m_icon, cell.x, cell.y );
 972
 973     return true;
 974 }
 975
 976 wxSize wxDataViewBitmapRenderer::GetSize() const
 977 {
 978     if (m_bitmap.IsOk())
 979         return wxSize( m_bitmap.GetWidth(), m_bitmap.GetHeight() );
 980     else if (m_icon.IsOk())
 981         return wxSize( m_icon.GetWidth(), m_icon.GetHeight() );
 982
 983     return wxSize(wxDVC_DEFAULT_RENDERER_SIZE,wxDVC_DEFAULT_RENDERER_SIZE);
 984 }
 985
 986 // ---------------------------------------------------------
 987 // wxDataViewToggleRenderer
 988 // ---------------------------------------------------------
 989
 990 IMPLEMENT_ABSTRACT_CLASS(wxDataViewToggleRenderer, wxDataViewRenderer)
 991
 992 wxDataViewToggleRenderer::wxDataViewToggleRenderer( const wxString &varianttype,
 993                         wxDataViewCellMode mode, int align ) :
 994     wxDataViewRenderer( varianttype, mode, align )
 995 {
 996     m_toggle = false;
 997 }
 998
 999 bool wxDataViewToggleRenderer::SetValue( const wxVariant &value )
1000 {
1001     m_toggle = value.GetBool();
1002
1003     return true;
1004 }
1005
1006 bool wxDataViewToggleRenderer::GetValue( wxVariant &WXUNUSED(value) ) const
1007 {
1008     return false;
1009 }
1010
1011 bool wxDataViewToggleRenderer::Render( wxRect cell, wxDC *dc, int WXUNUSED(state) )
1012 {
1013     int flags = 0;
1014     if (m_toggle)
1015         flags |= wxCONTROL_CHECKED;
1016     if (GetMode() != wxDATAVIEW_CELL_ACTIVATABLE ||
1017         GetEnabled() == false)
1018         flags |= wxCONTROL_DISABLED;
1019
1020     // Ensure that the check boxes always have at least the minimal required
1021     // size, otherwise DrawCheckBox() doesn't really work well. If this size is
1022     // greater than the cell size, the checkbox will be truncated but this is a
1023     // lesser evil.
1024     wxSize size = cell.GetSize();
1025     size.IncTo(GetSize());
1026     cell.SetSize(size);
1027
1028     wxRendererNative::Get().DrawCheckBox(
1029             GetOwner()->GetOwner(),
1030             *dc,
1031             cell,
1032             flags );
1033
1034     return true;
1035 }
1036
1037 bool wxDataViewToggleRenderer::WXActivateCell(const wxRect& WXUNUSED(cell),
1038                                               wxDataViewModel *model,
1039                                               const wxDataViewItem& item,
1040                                               unsigned int col,
1041                                               const wxMouseEvent *mouseEvent)
1042 {
1043     if ( mouseEvent )
1044     {
1045         // only react to clicks directly on the checkbox, not elsewhere in the same cell:
1046         if ( !wxRect(GetSize()).Contains(mouseEvent->GetPosition()) )
1047             return false;
1048     }
1049
1050     model->ChangeValue(!m_toggle, item, col);
1051     return true;
1052 }
1053
1054 wxSize wxDataViewToggleRenderer::GetSize() const
1055 {
1056     // the window parameter is not used by GetCheckBoxSize() so it's
1057     // safe to pass NULL
1058     return wxRendererNative::Get().GetCheckBoxSize(NULL);
1059 }
1060
1061 // ---------------------------------------------------------
1062 // wxDataViewProgressRenderer
1063 // ---------------------------------------------------------
1064
1065 IMPLEMENT_ABSTRACT_CLASS(wxDataViewProgressRenderer, wxDataViewRenderer)
1066
1067 wxDataViewProgressRenderer::wxDataViewProgressRenderer( const wxString &label,
1068     const wxString &varianttype, wxDataViewCellMode mode, int align ) :
1069     wxDataViewRenderer( varianttype, mode, align )
1070 {
1071     m_label = label;
1072     m_value = 0;
1073 }
1074
1075 bool wxDataViewProgressRenderer::SetValue( const wxVariant &value )
1076 {
1077     m_value = (long) value;
1078
1079     if (m_value < 0) m_value = 0;
1080     if (m_value > 100) m_value = 100;
1081
1082     return true;
1083 }
1084
1085 bool wxDataViewProgressRenderer::GetValue( wxVariant &value ) const
1086 {
1087     value = (long) m_value;
1088     return true;
1089 }
1090
1091 bool
1092 wxDataViewProgressRenderer::Render(wxRect rect, wxDC *dc, int WXUNUSED(state))
1093 {
1094     // deflate the rect to leave a small border between bars in adjacent rows
1095     wxRect bar = rect.Deflate(0, 1);
1096
1097     dc->SetBrush( *wxTRANSPARENT_BRUSH );
1098     dc->SetPen( *wxBLACK_PEN );
1099     dc->DrawRectangle( bar );
1100
1101     bar.width = (int)(bar.width * m_value / 100.);
1102     dc->SetPen( *wxTRANSPARENT_PEN );
1103
1104     const wxDataViewItemAttr& attr = GetAttr();
1105     dc->SetBrush( attr.HasColour() ? wxBrush(attr.GetColour())
1106                                   : *wxBLUE_BRUSH );
1107     dc->DrawRectangle( bar );
1108
1109     return true;
1110 }
1111
1112 wxSize wxDataViewProgressRenderer::GetSize() const
1113 {
1114     return wxSize(40,12);
1115 }
1116
1117 // ---------------------------------------------------------
1118 // wxDataViewIconTextRenderer
1119 // ---------------------------------------------------------
1120
1121 IMPLEMENT_CLASS(wxDataViewIconTextRenderer, wxDataViewRenderer)
1122
1123 wxDataViewIconTextRenderer::wxDataViewIconTextRenderer(
1124 const wxString &varianttype, wxDataViewCellMode mode, int align ) :
1125     wxDataViewRenderer( varianttype, mode, align )
1126 {
1127     SetMode(mode);
1128     SetAlignment(align);
1129 }
1130
1131 bool wxDataViewIconTextRenderer::SetValue( const wxVariant &value )
1132 {
1133     m_value << value;
1134     return true;
1135 }
1136
1137 bool wxDataViewIconTextRenderer::GetValue( wxVariant& WXUNUSED(value) ) const
1138 {
1139     return false;
1140 }
1141
1142 bool wxDataViewIconTextRenderer::Render(wxRect rect, wxDC *dc, int state)
1143 {
1144     int xoffset = 0;
1145
1146     const wxIcon& icon = m_value.GetIcon();
1147     if ( icon.IsOk() )
1148     {
1149         dc->DrawIcon(icon, rect.x, rect.y + (rect.height - icon.GetHeight())/2);
1150         xoffset = icon.GetWidth()+4;
1151     }
1152
1153     RenderText(m_value.GetText(), xoffset, rect, dc, state);
1154
1155     return true;
1156 }
1157
1158 wxSize wxDataViewIconTextRenderer::GetSize() const
1159 {
1160     if (!m_value.GetText().empty())
1161     {
1162         wxSize size = GetTextExtent(m_value.GetText());
1163
1164         if (m_value.GetIcon().IsOk())
1165             size.x += m_value.GetIcon().GetWidth() + 4;
1166         return size;
1167     }
1168     return wxSize(80,20);
1169 }
1170
1171 wxWindow* wxDataViewIconTextRenderer::CreateEditorCtrl(wxWindow *parent, wxRect labelRect, const wxVariant& value)
1172 {
1173     wxDataViewIconText iconText;
1174     iconText << value;
1175
1176     wxString text = iconText.GetText();
1177
1178     // adjust the label rect to take the width of the icon into account
1179     if (iconText.GetIcon().IsOk())
1180     {
1181         int w = iconText.GetIcon().GetWidth() + 4;
1182         labelRect.x += w;
1183         labelRect.width -= w;
1184     }
1185
1186     wxTextCtrl* ctrl = new wxTextCtrl( parent, wxID_ANY, text,
1187                                        wxPoint(labelRect.x,labelRect.y),
1188                                        wxSize(labelRect.width,labelRect.height),
1189                                        wxTE_PROCESS_ENTER );
1190
1191     // select the text in the control an place the cursor at the end
1192     ctrl->SetInsertionPointEnd();
1193     ctrl->SelectAll();
1194
1195     return ctrl;
1196 }
1197
1198 bool wxDataViewIconTextRenderer::GetValueFromEditorCtrl( wxWindow *editor, wxVariant& value )
1199 {
1200     wxTextCtrl *text = (wxTextCtrl*) editor;
1201
1202     // The icon can't be edited so get its old value and reuse it.
1203     wxVariant valueOld;
1204     wxDataViewColumn* const col = GetOwner();
1205     GetView()->GetModel()->GetValue(valueOld, m_item, col->GetModelColumn());
1206
1207     wxDataViewIconText iconText;
1208     iconText << valueOld;
1209
1210     // But replace the text with the value entered by user.
1211     iconText.SetText(text->GetValue());
1212
1213     value << iconText;
1214     return true;
1215 }
1216
1217 //-----------------------------------------------------------------------------
1218 // wxDataViewDropTarget
1219 //-----------------------------------------------------------------------------
1220
1221 #if wxUSE_DRAG_AND_DROP
1222
1223 class wxBitmapCanvas: public wxWindow
1224 {
1225 public:
1226     wxBitmapCanvas( wxWindow *parent, const wxBitmap &bitmap, const wxSize &size ) :
1227     wxWindow( parent, wxID_ANY, wxPoint(0,0), size )
1228     {
1229         m_bitmap = bitmap;
1230         Connect( wxEVT_PAINT, wxPaintEventHandler(wxBitmapCanvas::OnPaint) );
1231     }
1232
1233     void OnPaint( wxPaintEvent &WXUNUSED(event) )
1234     {
1235         wxPaintDC dc(this);
1236         dc.DrawBitmap( m_bitmap, 0, 0);
1237     }
1238
1239     wxBitmap m_bitmap;
1240 };
1241
1242 class wxDataViewDropSource: public wxDropSource
1243 {
1244 public:
1245     wxDataViewDropSource( wxDataViewMainWindow *win, unsigned int row ) :
1246         wxDropSource( win )
1247     {
1248         m_win = win;
1249         m_row = row;
1250         m_hint = NULL;
1251     }
1252
1253     ~wxDataViewDropSource()
1254     {
1255         delete m_hint;
1256     }
1257
1258     virtual bool GiveFeedback( wxDragResult WXUNUSED(effect) )
1259     {
1260         wxPoint pos = wxGetMousePosition();
1261
1262         if (!m_hint)
1263         {
1264             int liney = m_win->GetLineStart( m_row );
1265             int linex = 0;
1266             m_win->GetOwner()->CalcUnscrolledPosition( 0, liney, NULL, &liney );
1267             m_win->ClientToScreen( &linex, &liney );
1268             m_dist_x = pos.x - linex;
1269             m_dist_y = pos.y - liney;
1270
1271             int indent = 0;
1272             wxBitmap ib = m_win->CreateItemBitmap( m_row, indent );
1273             m_dist_x -= indent;
1274             m_hint = new wxFrame( m_win->GetParent(), wxID_ANY, wxEmptyString,
1275                                         wxPoint(pos.x - m_dist_x, pos.y + 5 ),
1276                                         ib.GetSize(),
1277                                         wxFRAME_TOOL_WINDOW |
1278                                         wxFRAME_FLOAT_ON_PARENT |
1279                                         wxFRAME_NO_TASKBAR |
1280                                         wxNO_BORDER );
1281             new wxBitmapCanvas( m_hint, ib, ib.GetSize() );
1282             m_hint->Show();
1283         }
1284         else
1285         {
1286             m_hint->Move( pos.x - m_dist_x, pos.y + 5  );
1287             m_hint->SetTransparent( 128 );
1288         }
1289
1290         return false;
1291     }
1292
1293     wxDataViewMainWindow   *m_win;
1294     unsigned int            m_row;
1295     wxFrame                *m_hint;
1296     int m_dist_x,m_dist_y;
1297 };
1298
1299
1300 class wxDataViewDropTarget: public wxDropTarget
1301 {
1302 public:
1303     wxDataViewDropTarget( wxDataObject *obj, wxDataViewMainWindow *win ) :
1304         wxDropTarget( obj )
1305     {
1306         m_win = win;
1307     }
1308
1309     virtual wxDragResult OnDragOver( wxCoord x, wxCoord y, wxDragResult def )
1310     {
1311         wxDataFormat format = GetMatchingPair();
1312         if (format == wxDF_INVALID)
1313             return wxDragNone;
1314         return m_win->OnDragOver( format, x, y, def);
1315     }
1316
1317     virtual bool OnDrop( wxCoord x, wxCoord y )
1318     {
1319         wxDataFormat format = GetMatchingPair();
1320         if (format == wxDF_INVALID)
1321             return false;
1322         return m_win->OnDrop( format, x, y );
1323     }
1324
1325     virtual wxDragResult OnData( wxCoord x, wxCoord y, wxDragResult def )
1326     {
1327         wxDataFormat format = GetMatchingPair();
1328         if (format == wxDF_INVALID)
1329             return wxDragNone;
1330         if (!GetData())
1331             return wxDragNone;
1332         return m_win->OnData( format, x, y, def );
1333     }
1334
1335     virtual void OnLeave()
1336         { m_win->OnLeave(); }
1337
1338     wxDataViewMainWindow   *m_win;
1339 };
1340
1341 #endif // wxUSE_DRAG_AND_DROP
1342
1343 //-----------------------------------------------------------------------------
1344 // wxDataViewRenameTimer
1345 //-----------------------------------------------------------------------------
1346
1347 wxDataViewRenameTimer::wxDataViewRenameTimer( wxDataViewMainWindow *owner )
1348 {
1349     m_owner = owner;
1350 }
1351
1352 void wxDataViewRenameTimer::Notify()
1353 {
1354     m_owner->OnRenameTimer();
1355 }
1356
1357 //-----------------------------------------------------------------------------
1358 // wxDataViewMainWindow
1359 //-----------------------------------------------------------------------------
1360
1361 // The tree building helper, declared firstly
1362 static void BuildTreeHelper( const wxDataViewModel * model,  const wxDataViewItem & item,
1363                              wxDataViewTreeNode * node);
1364
1365 int LINKAGEMODE wxDataViewSelectionCmp( unsigned int row1, unsigned int row2 )
1366 {
1367     if (row1 > row2) return 1;
1368     if (row1 == row2) return 0;
1369     return -1;
1370 }
1371
1372 IMPLEMENT_ABSTRACT_CLASS(wxDataViewMainWindow, wxWindow)
1373
1374 BEGIN_EVENT_TABLE(wxDataViewMainWindow,wxWindow)
1375     EVT_PAINT         (wxDataViewMainWindow::OnPaint)
1376     EVT_MOUSE_EVENTS  (wxDataViewMainWindow::OnMouse)
1377     EVT_SET_FOCUS     (wxDataViewMainWindow::OnSetFocus)
1378     EVT_KILL_FOCUS    (wxDataViewMainWindow::OnKillFocus)
1379     EVT_CHAR_HOOK     (wxDataViewMainWindow::OnCharHook)
1380     EVT_CHAR          (wxDataViewMainWindow::OnChar)
1381 END_EVENT_TABLE()
1382
1383 wxDataViewMainWindow::wxDataViewMainWindow( wxDataViewCtrl *parent, wxWindowID id,
1384     const wxPoint &pos, const wxSize &size, const wxString &name ) :
1385     wxWindow( parent, id, pos, size, wxWANTS_CHARS|wxBORDER_NONE, name ),
1386     m_selection( wxDataViewSelectionCmp )
1387
1388 {
1389     SetOwner( parent );
1390
1391     m_editorRenderer = NULL;
1392
1393     m_lastOnSame = false;
1394     m_renameTimer = new wxDataViewRenameTimer( this );
1395
1396     // TODO: user better initial values/nothing selected
1397     m_currentCol = NULL;
1398     m_currentColSetByKeyboard = false;
1399     m_useCellFocus = false;
1400     m_currentRow = 0;
1401
1402 #ifdef __WXMSW__
1403     // We would like to use the same line height that Explorer uses. This is
1404     // different from standard ListView control since Vista.
1405     if ( wxGetWinVersion() >= wxWinVersion_Vista )
1406         m_lineHeight = wxMax(16, GetCharHeight()) + 6; // 16 = mini icon height
1407     else
1408 #endif // __WXMSW__
1409         m_lineHeight = wxMax(16, GetCharHeight()) + 1; // 16 = mini icon height
1410
1411 #if wxUSE_DRAG_AND_DROP
1412     m_dragCount = 0;
1413     m_dragStart = wxPoint(0,0);
1414
1415     m_dragEnabled = false;
1416     m_dropEnabled = false;
1417     m_dropHint = false;
1418     m_dropHintLine = (unsigned int) -1;
1419 #endif // wxUSE_DRAG_AND_DROP
1420
1421     m_lineLastClicked = (unsigned int) -1;
1422     m_lineBeforeLastClicked = (unsigned int) -1;
1423     m_lineSelectSingleOnUp = (unsigned int) -1;
1424
1425     m_hasFocus = false;
1426
1427     SetBackgroundColour( *wxWHITE );
1428
1429     SetBackgroundStyle(wxBG_STYLE_CUSTOM);
1430
1431     m_penRule = wxPen(GetRuleColour());
1432
1433     // compose a pen whichcan draw black lines
1434     // TODO: maybe there is something system colour to use
1435     m_penExpander = wxPen(wxColour(0,0,0));
1436
1437     m_root = wxDataViewTreeNode::CreateRootNode();
1438
1439     // Make m_count = -1 will cause the class recaculate the real displaying number of rows.
1440     m_count = -1;
1441     m_underMouse = NULL;
1442     UpdateDisplay();
1443 }
1444
1445 wxDataViewMainWindow::~wxDataViewMainWindow()
1446 {
1447     DestroyTree();
1448     delete m_renameTimer;
1449 }
1450
1451
1452 #if wxUSE_DRAG_AND_DROP
1453 bool wxDataViewMainWindow::EnableDragSource( const wxDataFormat &format )
1454 {
1455     m_dragFormat = format;
1456     m_dragEnabled = format != wxDF_INVALID;
1457
1458     return true;
1459 }
1460
1461 bool wxDataViewMainWindow::EnableDropTarget( const wxDataFormat &format )
1462 {
1463     m_dropFormat = format;
1464     m_dropEnabled = format != wxDF_INVALID;
1465
1466     if (m_dropEnabled)
1467         SetDropTarget( new wxDataViewDropTarget( new wxCustomDataObject( format ), this ) );
1468
1469     return true;
1470 }
1471
1472 void wxDataViewMainWindow::RemoveDropHint()
1473 {
1474     if (m_dropHint)
1475     {
1476             m_dropHint = false;
1477             RefreshRow( m_dropHintLine );
1478             m_dropHintLine = (unsigned int) -1;
1479     }
1480 }
1481
1482 wxDragResult wxDataViewMainWindow::OnDragOver( wxDataFormat format, wxCoord x,
1483                                                wxCoord y, wxDragResult def )
1484 {
1485     int xx = x;
1486     int yy = y;
1487     m_owner->CalcUnscrolledPosition( xx, yy, &xx, &yy );
1488     unsigned int row = GetLineAt( yy );
1489
1490     if ((row >= GetRowCount()) || (xx > GetEndOfLastCol()))
1491     {
1492         RemoveDropHint();
1493         return wxDragNone;
1494     }
1495
1496     wxDataViewItem item = GetItemByRow( row );
1497
1498     wxDataViewModel *model = GetModel();
1499
1500     wxDataViewEvent event( wxEVT_COMMAND_DATAVIEW_ITEM_DROP_POSSIBLE, m_owner->GetId() );
1501     event.SetEventObject( m_owner );
1502     event.SetItem( item );
1503     event.SetModel( model );
1504     event.SetDataFormat( format );
1505     event.SetDropEffect( def );
1506     if (!m_owner->HandleWindowEvent( event ))
1507     {
1508         RemoveDropHint();
1509         return wxDragNone;
1510     }
1511
1512     if (!event.IsAllowed())
1513     {
1514         RemoveDropHint();
1515         return wxDragNone;
1516     }
1517
1518
1519     if (m_dropHint && (row != m_dropHintLine))
1520         RefreshRow( m_dropHintLine );
1521     m_dropHint = true;
1522     m_dropHintLine = row;
1523     RefreshRow( row );
1524
1525     return def;
1526 }
1527
1528 bool wxDataViewMainWindow::OnDrop( wxDataFormat format, wxCoord x, wxCoord y )
1529 {
1530     RemoveDropHint();
1531
1532     int xx = x;
1533     int yy = y;
1534     m_owner->CalcUnscrolledPosition( xx, yy, &xx, &yy );
1535     unsigned int row = GetLineAt( yy );
1536
1537     if ((row >= GetRowCount()) || (xx > GetEndOfLastCol()))
1538         return false;
1539
1540     wxDataViewItem item = GetItemByRow( row );
1541
1542     wxDataViewModel *model = GetModel();
1543
1544     wxDataViewEvent event( wxEVT_COMMAND_DATAVIEW_ITEM_DROP_POSSIBLE, m_owner->GetId() );
1545     event.SetEventObject( m_owner );
1546     event.SetItem( item );
1547     event.SetModel( model );
1548     event.SetDataFormat( format );
1549     if (!m_owner->HandleWindowEvent( event ))
1550         return false;
1551
1552     if (!event.IsAllowed())
1553         return false;
1554
1555     return true;
1556 }
1557
1558 wxDragResult wxDataViewMainWindow::OnData( wxDataFormat format, wxCoord x, wxCoord y,
1559                                            wxDragResult def )
1560 {
1561     int xx = x;
1562     int yy = y;
1563     m_owner->CalcUnscrolledPosition( xx, yy, &xx, &yy );
1564     unsigned int row = GetLineAt( yy );
1565
1566     if ((row >= GetRowCount()) || (xx > GetEndOfLastCol()))
1567         return wxDragNone;
1568
1569     wxDataViewItem item = GetItemByRow( row );
1570
1571     wxDataViewModel *model = GetModel();
1572
1573     wxCustomDataObject *obj = (wxCustomDataObject *) GetDropTarget()->GetDataObject();
1574
1575     wxDataViewEvent event( wxEVT_COMMAND_DATAVIEW_ITEM_DROP, m_owner->GetId() );
1576     event.SetEventObject( m_owner );
1577     event.SetItem( item );
1578     event.SetModel( model );
1579     event.SetDataFormat( format );
1580     event.SetDataSize( obj->GetSize() );
1581     event.SetDataBuffer( obj->GetData() );
1582     event.SetDropEffect( def );
1583     if (!m_owner->HandleWindowEvent( event ))
1584         return wxDragNone;
1585
1586     if (!event.IsAllowed())
1587         return wxDragNone;
1588
1589     return def;
1590 }
1591
1592 void wxDataViewMainWindow::OnLeave()
1593 {
1594     RemoveDropHint();
1595 }
1596
1597 wxBitmap wxDataViewMainWindow::CreateItemBitmap( unsigned int row, int &indent )
1598 {
1599     int height = GetLineHeight( row );
1600     int width = 0;
1601     unsigned int cols = GetOwner()->GetColumnCount();
1602     unsigned int col;
1603     for (col = 0; col < cols; col++)
1604     {
1605         wxDataViewColumn *column = GetOwner()->GetColumnAt(col);
1606         if (column->IsHidden())
1607             continue;      // skip it!
1608         width += column->GetWidth();
1609     }
1610
1611     indent = 0;
1612     if (!IsList())
1613     {
1614         wxDataViewTreeNode *node = GetTreeNodeByRow(row);
1615         indent = GetOwner()->GetIndent() * node->GetIndentLevel();
1616         indent = indent + m_lineHeight;
1617             // try to use the m_lineHeight as the expander space
1618     }
1619     width -= indent;
1620
1621     wxBitmap bitmap( width, height );
1622     wxMemoryDC dc( bitmap );
1623     dc.SetFont( GetFont() );
1624     dc.SetPen( *wxBLACK_PEN );
1625     dc.SetBrush( *wxWHITE_BRUSH );
1626     dc.DrawRectangle( 0,0,width,height );
1627
1628     wxDataViewModel *model = m_owner->GetModel();
1629
1630     wxDataViewColumn * const
1631         expander = GetExpanderColumnOrFirstOne(GetOwner());
1632
1633     int x = 0;
1634     for (col = 0; col < cols; col++)
1635     {
1636         wxDataViewColumn *column = GetOwner()->GetColumnAt( col );
1637         wxDataViewRenderer *cell = column->GetRenderer();
1638
1639         if (column->IsHidden())
1640             continue;       // skip it!
1641
1642         width = column->GetWidth();
1643
1644         if (column == expander)
1645             width -= indent;
1646
1647         wxDataViewItem item = GetItemByRow( row );
1648         cell->PrepareForItem(model, item, column->GetModelColumn());
1649
1650         wxRect item_rect(x, 0, width, height);
1651         item_rect.Deflate(PADDING_RIGHTLEFT, 0);
1652
1653         // dc.SetClippingRegion( item_rect );
1654         cell->WXCallRender(item_rect, &dc, 0);
1655         // dc.DestroyClippingRegion();
1656
1657         x += width;
1658     }
1659
1660     return bitmap;
1661 }
1662
1663 #endif // wxUSE_DRAG_AND_DROP
1664
1665
1666 // Draw focus rect for individual cell. Unlike native focus rect, we render
1667 // this in foreground text color (typically white) to enhance contrast and
1668 // make it visible.
1669 static void DrawSelectedCellFocusRect(wxDC& dc, const wxRect& rect)
1670 {
1671     // (This code is based on wxRendererGeneric::DrawFocusRect and modified.)
1672
1673     // draw the pixels manually because the "dots" in wxPen with wxDOT style
1674     // may be short traits and not really dots
1675     //
1676     // note that to behave in the same manner as DrawRect(), we must exclude
1677     // the bottom and right borders from the rectangle
1678     wxCoord x1 = rect.GetLeft(),
1679             y1 = rect.GetTop(),
1680             x2 = rect.GetRight(),
1681             y2 = rect.GetBottom();
1682
1683     wxDCPenChanger pen(dc, wxSystemSettings::GetColour(wxSYS_COLOUR_HIGHLIGHTTEXT));
1684
1685     wxCoord z;
1686     for ( z = x1 + 1; z < x2; z += 2 )
1687         dc.DrawPoint(z, rect.GetTop());
1688
1689     wxCoord shift = z == x2 ? 0 : 1;
1690     for ( z = y1 + shift; z < y2; z += 2 )
1691         dc.DrawPoint(x2, z);
1692
1693     shift = z == y2 ? 0 : 1;
1694     for ( z = x2 - shift; z > x1; z -= 2 )
1695         dc.DrawPoint(z, y2);
1696
1697     shift = z == x1 ? 0 : 1;
1698     for ( z = y2 - shift; z > y1; z -= 2 )
1699         dc.DrawPoint(x1, z);
1700 }
1701
1702
1703 void wxDataViewMainWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
1704 {
1705     wxDataViewModel *model = GetModel();
1706     wxAutoBufferedPaintDC dc( this );
1707
1708     dc.SetBrush(GetOwner()->GetBackgroundColour());
1709     dc.SetPen( *wxTRANSPARENT_PEN );
1710     dc.DrawRectangle(GetClientSize());
1711
1712     if ( IsEmpty() )
1713     {
1714         // No items to draw.
1715         return;
1716     }
1717
1718     // prepare the DC
1719     GetOwner()->PrepareDC( dc );
1720     dc.SetFont( GetFont() );
1721
1722     wxRect update = GetUpdateRegion().GetBox();
1723     m_owner->CalcUnscrolledPosition( update.x, update.y, &update.x, &update.y );
1724
1725     // compute which items needs to be redrawn
1726     unsigned int item_start = GetLineAt( wxMax(0,update.y) );
1727     unsigned int item_count =
1728         wxMin( (int)(  GetLineAt( wxMax(0,update.y+update.height) ) - item_start + 1),
1729             (int)(GetRowCount( ) - item_start));
1730     unsigned int item_last = item_start + item_count;
1731
1732     // Send the event to wxDataViewCtrl itself.
1733     wxWindow * const parent = GetParent();
1734     wxDataViewEvent cache_event(wxEVT_COMMAND_DATAVIEW_CACHE_HINT, parent->GetId());
1735     cache_event.SetEventObject(parent);
1736     cache_event.SetCache(item_start, item_last - 1);
1737     parent->ProcessWindowEvent(cache_event);
1738
1739     // compute which columns needs to be redrawn
1740     unsigned int cols = GetOwner()->GetColumnCount();
1741     if ( !cols )
1742     {
1743         // we assume that we have at least one column below and painting an
1744         // empty control is unnecessary anyhow
1745         return;
1746     }
1747
1748     unsigned int col_start = 0;
1749     unsigned int x_start;
1750     for (x_start = 0; col_start < cols; col_start++)
1751     {
1752         wxDataViewColumn *col = GetOwner()->GetColumnAt(col_start);
1753         if (col->IsHidden())
1754             continue;      // skip it!
1755
1756         unsigned int w = col->GetWidth();
1757         if (x_start+w >= (unsigned int)update.x)
1758             break;
1759
1760         x_start += w;
1761     }
1762
1763     unsigned int col_last = col_start;
1764     unsigned int x_last = x_start;
1765     for (; col_last < cols; col_last++)
1766     {
1767         wxDataViewColumn *col = GetOwner()->GetColumnAt(col_last);
1768         if (col->IsHidden())
1769             continue;      // skip it!
1770
1771         if (x_last > (unsigned int)update.GetRight())
1772             break;
1773
1774         x_last += col->GetWidth();
1775     }
1776
1777     // Draw background of alternate rows specially if required
1778     if ( m_owner->HasFlag(wxDV_ROW_LINES) )
1779     {
1780         wxColour altRowColour = m_owner->m_alternateRowColour;
1781         if ( !altRowColour.IsOk() )
1782         {
1783             // Determine the alternate rows colour automatically from the
1784             // background colour.
1785             const wxColour bgColour = m_owner->GetBackgroundColour();
1786
1787             // Depending on the background, alternate row color
1788             // will be 3% more dark or 50% brighter.
1789             int alpha = bgColour.GetRGB() > 0x808080 ? 97 : 150;
1790             altRowColour = bgColour.ChangeLightness(alpha);
1791         }
1792
1793         dc.SetPen(*wxTRANSPARENT_PEN);
1794         dc.SetBrush(wxBrush(altRowColour));
1795
1796         for (unsigned int item = item_start; item < item_last; item++)
1797         {
1798             if ( item % 2 )
1799             {
1800                 dc.DrawRectangle(x_start,
1801                                  GetLineStart(item),
1802                                  GetClientSize().GetWidth(),
1803                                  GetLineHeight(item));
1804             }
1805         }
1806     }
1807
1808     // Draw horizontal rules if required
1809     if ( m_owner->HasFlag(wxDV_HORIZ_RULES) )
1810     {
1811         dc.SetPen(m_penRule);
1812         dc.SetBrush(*wxTRANSPARENT_BRUSH);
1813
1814         for (unsigned int i = item_start; i <= item_last; i++)
1815         {
1816             int y = GetLineStart( i );
1817             dc.DrawLine(x_start, y, x_last, y);
1818         }
1819     }
1820
1821     // Draw vertical rules if required
1822     if ( m_owner->HasFlag(wxDV_VERT_RULES) )
1823     {
1824         dc.SetPen(m_penRule);
1825         dc.SetBrush(*wxTRANSPARENT_BRUSH);
1826
1827         // NB: Vertical rules are drawn in the last pixel of a column so that
1828         //     they align perfectly with native MSW wxHeaderCtrl as well as for
1829         //     consistency with MSW native list control. There's no vertical
1830         //     rule at the most-left side of the control.
1831
1832         int x = x_start - 1;
1833         for (unsigned int i = col_start; i < col_last; i++)
1834         {
1835             wxDataViewColumn *col = GetOwner()->GetColumnAt(i);
1836             if (col->IsHidden())
1837                 continue;       // skip it
1838
1839             x += col->GetWidth();
1840
1841             dc.DrawLine(x, GetLineStart( item_start ),
1842                         x, GetLineStart( item_last ) );
1843         }
1844     }
1845
1846     // redraw the background for the items which are selected/current
1847     for (unsigned int item = item_start; item < item_last; item++)
1848     {
1849         bool selected = m_selection.Index( item ) != wxNOT_FOUND;
1850
1851         if (selected || item == m_currentRow)
1852         {
1853             wxRect rect( x_start, GetLineStart( item ),
1854                          x_last - x_start, GetLineHeight( item ) );
1855
1856             // draw selection and whole-item focus:
1857             if ( selected )
1858             {
1859                 int flags = wxCONTROL_SELECTED;
1860                 if (m_hasFocus)
1861                     flags |= wxCONTROL_FOCUSED;
1862
1863                 wxRendererNative::Get().DrawItemSelectionRect
1864                                     (
1865                                         this,
1866                                         dc,
1867                                         rect,
1868                                         flags
1869                                     );
1870             }
1871
1872             // draw keyboard focus rect if applicable
1873             if ( item == m_currentRow && m_hasFocus )
1874             {
1875                 bool renderColumnFocus = false;
1876
1877                 if ( m_useCellFocus && m_currentCol && m_currentColSetByKeyboard )
1878                 {
1879                     renderColumnFocus = true;
1880
1881                     // If this is container node without columns, render full-row focus:
1882                     if ( !IsList() )
1883                     {
1884                         wxDataViewTreeNode *node = GetTreeNodeByRow(item);
1885                         if ( node->HasChildren() && !model->HasContainerColumns(node->GetItem()) )
1886                             renderColumnFocus = false;
1887                     }
1888                 }
1889
1890                 if ( renderColumnFocus )
1891                 {
1892                     for ( unsigned int i = col_start; i < col_last; i++ )
1893                     {
1894                         wxDataViewColumn *col = GetOwner()->GetColumnAt(i);
1895                         if ( col->IsHidden() )
1896                             continue;
1897
1898                         rect.width = col->GetWidth();
1899
1900                         if ( col == m_currentCol )
1901                         {
1902                             // make the rect more visible by adding a small
1903                             // margin around it:
1904                             rect.Deflate(1, 1);
1905
1906                             if ( selected )
1907                             {
1908                                 // DrawFocusRect() uses XOR and is all but
1909                                 // invisible against dark-blue background. Use
1910                                 // the same color used for selected text.
1911                                 DrawSelectedCellFocusRect(dc, rect);
1912                             }
1913                             else
1914                             {
1915                                 wxRendererNative::Get().DrawFocusRect
1916                                                     (
1917                                                         this,
1918                                                         dc,
1919                                                         rect,
1920                                                         0
1921                                                 );
1922                             }
1923                             break;
1924                         }
1925
1926                         rect.x += rect.width;
1927                     }
1928                 }
1929                 else
1930                 {
1931                     // render focus rectangle for the whole row
1932                     wxRendererNative::Get().DrawFocusRect
1933                                         (
1934                                             this,
1935                                             dc,
1936                                             rect,
1937                                             selected ? (int)wxCONTROL_SELECTED : 0
1938                                         );
1939                 }
1940             }
1941         }
1942     }
1943
1944 #if wxUSE_DRAG_AND_DROP
1945     if (m_dropHint)
1946     {
1947         wxRect rect( x_start, GetLineStart( m_dropHintLine ),
1948                      x_last - x_start, GetLineHeight( m_dropHintLine ) );
1949         dc.SetPen( *wxBLACK_PEN );
1950         dc.SetBrush( *wxTRANSPARENT_BRUSH );
1951         dc.DrawRectangle( rect );
1952     }
1953 #endif // wxUSE_DRAG_AND_DROP
1954
1955     wxDataViewColumn * const
1956         expander = GetExpanderColumnOrFirstOne(GetOwner());
1957
1958     // redraw all cells for all rows which must be repainted and all columns
1959     wxRect cell_rect;
1960     cell_rect.x = x_start;
1961     for (unsigned int i = col_start; i < col_last; i++)
1962     {
1963         wxDataViewColumn *col = GetOwner()->GetColumnAt( i );
1964         wxDataViewRenderer *cell = col->GetRenderer();
1965         cell_rect.width = col->GetWidth();
1966
1967         if ( col->IsHidden() || cell_rect.width <= 0 )
1968             continue;       // skip it!
1969
1970         for (unsigned int item = item_start; item < item_last; item++)
1971         {
1972             // get the cell value and set it into the renderer
1973             wxDataViewTreeNode *node = NULL;
1974             wxDataViewItem dataitem;
1975
1976             if (!IsVirtualList())
1977             {
1978                 node = GetTreeNodeByRow(item);
1979                 if( node == NULL )
1980                     continue;
1981
1982                 dataitem = node->GetItem();
1983
1984                 // Skip all columns of "container" rows except the expander
1985                 // column itself unless HasContainerColumns() overrides this.
1986                 if ( col != expander &&
1987                         model->IsContainer(dataitem) &&
1988                             !model->HasContainerColumns(dataitem) )
1989                     continue;
1990             }
1991             else
1992             {
1993                 dataitem = wxDataViewItem( wxUIntToPtr(item+1) );
1994             }
1995
1996             cell->PrepareForItem(model, dataitem, col->GetModelColumn());
1997
1998             // update cell_rect
1999             cell_rect.y = GetLineStart( item );
2000             cell_rect.height = GetLineHeight( item );
2001
2002             // draw the background
2003             bool selected = m_selection.Index( item ) != wxNOT_FOUND;
2004             if ( !selected )
2005                 DrawCellBackground( cell, dc, cell_rect );
2006
2007             // deal with the expander
2008             int indent = 0;
2009             if ((!IsList()) && (col == expander))
2010             {
2011                 // Calculate the indent first
2012                 indent = GetOwner()->GetIndent() * node->GetIndentLevel();
2013
2014                 // we reserve m_lineHeight of horizontal space for the expander
2015                 // but leave EXPANDER_MARGIN around the expander itself
2016                 int exp_x = cell_rect.x + indent + EXPANDER_MARGIN;
2017
2018                 indent += m_lineHeight;
2019
2020                 // draw expander if needed and visible
2021                 if ( node->HasChildren() && exp_x < cell_rect.GetRight() )
2022                 {
2023                     dc.SetPen( m_penExpander );
2024                     dc.SetBrush( wxNullBrush );
2025
2026                     int exp_size = m_lineHeight - 2*EXPANDER_MARGIN;
2027                     int exp_y = cell_rect.y + (cell_rect.height - exp_size)/2
2028                                    + EXPANDER_MARGIN - EXPANDER_OFFSET;
2029
2030                     const wxRect rect(exp_x, exp_y, exp_size, exp_size);
2031
2032                     int flag = 0;
2033                     if ( m_underMouse == node )
2034                         flag |= wxCONTROL_CURRENT;
2035                     if ( node->IsOpen() )
2036                         flag |= wxCONTROL_EXPANDED;
2037
2038                     // ensure that we don't overflow the cell (which might
2039                     // happen if the column is very narrow)
2040                     wxDCClipper clip(dc, cell_rect);
2041
2042                     wxRendererNative::Get().DrawTreeItemButton( this, dc, rect, flag);
2043                 }
2044
2045                 // force the expander column to left-center align
2046                 cell->SetAlignment( wxALIGN_CENTER_VERTICAL );
2047             }
2048
2049             wxRect item_rect = cell_rect;
2050             item_rect.Deflate(PADDING_RIGHTLEFT, 0);
2051
2052             // account for the tree indent (harmless if we're not indented)
2053             item_rect.x += indent;
2054             item_rect.width -= indent;
2055
2056             if ( item_rect.width <= 0 )
2057                 continue;
2058
2059             int state = 0;
2060             if (m_hasFocus && selected)
2061                 state |= wxDATAVIEW_CELL_SELECTED;
2062
2063             // TODO: it would be much more efficient to create a clipping
2064             //       region for the entire column being rendered (in the OnPaint
2065             //       of wxDataViewMainWindow) instead of a single clip region for
2066             //       each cell. However it would mean that each renderer should
2067             //       respect the given wxRect's top & bottom coords, eventually
2068             //       violating only the left & right coords - however the user can
2069             //       make its own renderer and thus we cannot be sure of that.
2070             wxDCClipper clip(dc, item_rect);
2071
2072             cell->WXCallRender(item_rect, &dc, state);
2073         }
2074
2075         cell_rect.x += cell_rect.width;
2076     }
2077 }
2078
2079
2080 void wxDataViewMainWindow::DrawCellBackground( wxDataViewRenderer* cell, wxDC& dc, const wxRect& rect )
2081 {
2082     wxRect rectBg( rect );
2083
2084     // don't overlap the horizontal rules
2085     if ( m_owner->HasFlag(wxDV_HORIZ_RULES) )
2086     {
2087         rectBg.x++;
2088         rectBg.width--;
2089     }
2090
2091     // don't overlap the vertical rules
2092     if ( m_owner->HasFlag(wxDV_VERT_RULES) )
2093     {
2094         rectBg.y++;
2095         rectBg.height--;
2096     }
2097
2098     cell->RenderBackground(&dc, rectBg);
2099 }
2100
2101 void wxDataViewMainWindow::OnRenameTimer()
2102 {
2103     // We have to call this here because changes may just have
2104     // been made and no screen update taken place.
2105     if ( m_dirty )
2106     {
2107         // TODO: use wxTheApp->SafeYieldFor(NULL, wxEVT_CATEGORY_UI) instead
2108         //       (needs to be tested!)
2109         wxSafeYield();
2110     }
2111
2112     wxDataViewItem item = GetItemByRow( m_currentRow );
2113
2114     StartEditing( item, m_currentCol );
2115 }
2116
2117 void
2118 wxDataViewMainWindow::StartEditing(const wxDataViewItem& item,
2119                                    const wxDataViewColumn* col)
2120 {
2121     wxDataViewRenderer* renderer = col->GetRenderer();
2122     if ( !IsCellEditableInMode(item, col, wxDATAVIEW_CELL_EDITABLE) )
2123         return;
2124
2125     const wxRect itemRect = GetItemRect(item, col);
2126     if ( renderer->StartEditing(item, itemRect) )
2127     {
2128         // Save the renderer to be able to finish/cancel editing it later and
2129         // save the control to be able to detect if we're still editing it.
2130         m_editorRenderer = renderer;
2131         m_editorCtrl = renderer->GetEditorCtrl();
2132     }
2133 }
2134
2135 //-----------------------------------------------------------------------------
2136 // Helper class for do operation on the tree node
2137 //-----------------------------------------------------------------------------
2138 class DoJob
2139 {
2140 public:
2141     DoJob() { }
2142     virtual ~DoJob() { }
2143
2144     // The return value control how the tree-walker tranverse the tree
2145     enum
2146     {
2147         DONE,          // Job done, stop traversing and return
2148         SKIP_SUBTREE,  // Ignore the current node's subtree and continue
2149         CONTINUE       // Job not done, continue
2150     };
2151
2152     virtual int operator() ( wxDataViewTreeNode * node ) = 0;
2153 };
2154
2155 bool Walker( wxDataViewTreeNode * node, DoJob & func )
2156 {
2157     wxCHECK_MSG( node, false, "can't walk NULL node" );
2158
2159     switch( func( node ) )
2160     {
2161         case DoJob::DONE:
2162             return true;
2163         case DoJob::SKIP_SUBTREE:
2164             return false;
2165         case DoJob::CONTINUE:
2166             break;
2167     }
2168
2169     if ( node->HasChildren() )
2170     {
2171         const wxDataViewTreeNodes& nodes = node->GetChildNodes();
2172
2173         for ( wxDataViewTreeNodes::const_iterator i = nodes.begin();
2174               i != nodes.end();
2175               ++i )
2176         {
2177             if ( Walker(*i, func) )
2178                 return true;
2179         }
2180     }
2181
2182     return false;
2183 }
2184
2185 bool wxDataViewMainWindow::ItemAdded(const wxDataViewItem & parent, const wxDataViewItem & item)
2186 {
2187     if (IsVirtualList())
2188     {
2189         wxDataViewVirtualListModel *list_model =
2190             (wxDataViewVirtualListModel*) GetModel();
2191         m_count = list_model->GetCount();
2192     }
2193     else
2194     {
2195         SortPrepare();
2196
2197         wxDataViewTreeNode *parentNode = FindNode(parent);
2198
2199         if ( !parentNode )
2200             return false;
2201
2202         wxDataViewItemArray modelSiblings;
2203         GetModel()->GetChildren(parent, modelSiblings);
2204         const int modelSiblingsSize = modelSiblings.size();
2205
2206         int posInModel = modelSiblings.Index(item, /*fromEnd=*/true);
2207         wxCHECK_MSG( posInModel != wxNOT_FOUND, false, "adding non-existent item?" );
2208
2209         wxDataViewTreeNode *itemNode = new wxDataViewTreeNode(parentNode, item);
2210         itemNode->SetHasChildren(GetModel()->IsContainer(item));
2211
2212         parentNode->SetHasChildren(true);
2213
2214         const wxDataViewTreeNodes& nodeSiblings = parentNode->GetChildNodes();
2215         const int nodeSiblingsSize = nodeSiblings.size();
2216
2217         int nodePos = 0;
2218
2219         if ( posInModel == modelSiblingsSize - 1 )
2220         {
2221             nodePos = nodeSiblingsSize;
2222         }
2223         else if ( modelSiblingsSize == nodeSiblingsSize + 1 )
2224         {
2225             // This is the simple case when our node tree already matches the
2226             // model and only this one item is missing.
2227             nodePos = posInModel;
2228         }
2229         else
2230         {
2231             // It's possible that a larger discrepancy between the model and
2232             // our realization exists. This can happen e.g. when adding a bunch
2233             // of items to the model and then calling ItemsAdded() just once
2234             // afterwards. In this case, we must find the right position by
2235             // looking at sibling items.
2236
2237             // append to the end if we won't find a better position:
2238             nodePos = nodeSiblingsSize;
2239
2240             for ( int nextItemPos = posInModel + 1;
2241                   nextItemPos < modelSiblingsSize;
2242                   nextItemPos++ )
2243             {
2244                 int nextNodePos = parentNode->FindChildByItem(modelSiblings[nextItemPos]);
2245                 if ( nextNodePos != wxNOT_FOUND )
2246                 {
2247                     nodePos = nextNodePos;
2248                     break;
2249                 }
2250             }
2251         }
2252
2253         parentNode->ChangeSubTreeCount(+1);
2254         parentNode->InsertChild(itemNode, nodePos);
2255
2256         m_count = -1;
2257     }
2258
2259     GetOwner()->InvalidateColBestWidths();
2260     UpdateDisplay();
2261
2262     return true;
2263 }
2264
2265 bool wxDataViewMainWindow::ItemDeleted(const wxDataViewItem& parent,
2266                                        const wxDataViewItem& item)
2267 {
2268     if (IsVirtualList())
2269     {
2270         wxDataViewVirtualListModel *list_model =
2271             (wxDataViewVirtualListModel*) GetModel();
2272         m_count = list_model->GetCount();
2273
2274         if ( !m_selection.empty() )
2275         {
2276             const int row = GetRowByItem(item);
2277
2278             int rowIndexInSelection = wxNOT_FOUND;
2279
2280             const size_t selCount = m_selection.size();
2281             for ( size_t i = 0; i < selCount; i++ )
2282             {
2283                 if ( m_selection[i] == (unsigned)row )
2284                     rowIndexInSelection = i;
2285                 else if ( m_selection[i] > (unsigned)row )
2286                     m_selection[i]--;
2287             }
2288
2289             if ( rowIndexInSelection != wxNOT_FOUND )
2290                 m_selection.RemoveAt(rowIndexInSelection);
2291         }
2292
2293     }
2294     else // general case
2295     {
2296         wxDataViewTreeNode *parentNode = FindNode(parent);
2297
2298         // Notice that it is possible that the item being deleted is not in the
2299         // tree at all, for example we could be deleting a never shown (because
2300         // collapsed) item in a tree model. So it's not an error if we don't know
2301         // about this item, just return without doing anything then.
2302         if ( !parentNode )
2303             return true;
2304
2305         wxCHECK_MSG( parentNode->HasChildren(), false, "parent node doesn't have children?" );
2306         const wxDataViewTreeNodes& parentsChildren = parentNode->GetChildNodes();
2307
2308         // We can't use FindNode() to find 'item', because it was already
2309         // removed from the model by the time ItemDeleted() is called, so we
2310         // have to do it manually. We keep track of its position as well for
2311         // later use.
2312         int itemPosInNode = 0;
2313         wxDataViewTreeNode *itemNode = NULL;
2314         for ( wxDataViewTreeNodes::const_iterator i = parentsChildren.begin();
2315               i != parentsChildren.end();
2316               ++i, ++itemPosInNode )
2317         {
2318             if( (*i)->GetItem() == item )
2319             {
2320                 itemNode = *i;
2321                 break;
2322             }
2323         }
2324
2325         // If the parent wasn't expanded, it's possible that we didn't have a
2326         // node corresponding to 'item' and so there's nothing left to do.
2327         if ( !itemNode )
2328         {
2329             // If this was the last child to be removed, it's possible the parent
2330             // node became a leaf. Let's ask the model about it.
2331             if ( parentNode->GetChildNodes().empty() )
2332                 parentNode->SetHasChildren(GetModel()->IsContainer(parent));
2333
2334             return true;
2335         }
2336
2337         // Delete the item from wxDataViewTreeNode representation:
2338         const int itemsDeleted = 1 + itemNode->GetSubTreeCount();
2339
2340         parentNode->RemoveChild(itemNode);
2341         delete itemNode;
2342         parentNode->ChangeSubTreeCount(-itemsDeleted);
2343
2344         // Make the row number invalid and get a new valid one when user call GetRowCount
2345         m_count = -1;
2346
2347         // If this was the last child to be removed, it's possible the parent
2348         // node became a leaf. Let's ask the model about it.
2349         if ( parentNode->GetChildNodes().empty() )
2350         {
2351             bool isContainer = GetModel()->IsContainer(parent);
2352             parentNode->SetHasChildren(isContainer);
2353             if ( isContainer )
2354             {
2355                 // If it's still a container, make sure we show "+" icon for it
2356                 // and not "-" one as there is nothing to collapse any more.
2357                 if ( parentNode->IsOpen() )
2358                     parentNode->ToggleOpen();
2359             }
2360         }
2361
2362         // Update selection by removing 'item' and its entire children tree from the selection.
2363         if ( !m_selection.empty() )
2364         {
2365             // we can't call GetRowByItem() on 'item', as it's already deleted, so compute it from
2366             // the parent ('parentNode') and position in its list of children
2367             int itemRow;
2368             if ( itemPosInNode == 0 )
2369             {
2370                 // 1st child, row number is that of the parent parentNode + 1
2371                 itemRow = GetRowByItem(parentNode->GetItem()) + 1;
2372             }
2373             else
2374             {
2375                 // row number is that of the sibling above 'item' + its subtree if any + 1
2376                 const wxDataViewTreeNode *siblingNode = parentNode->GetChildNodes()[itemPosInNode - 1];
2377
2378                 itemRow = GetRowByItem(siblingNode->GetItem()) +
2379                           siblingNode->GetSubTreeCount() +
2380                           1;
2381             }
2382
2383             wxDataViewSelection newsel(wxDataViewSelectionCmp);
2384
2385             const size_t numSelections = m_selection.size();
2386             for ( size_t i = 0; i < numSelections; ++i )
2387             {
2388                 const int s = m_selection[i];
2389                 if ( s < itemRow )
2390                     newsel.push_back(s);
2391                 else if ( s >= itemRow + itemsDeleted )
2392                     newsel.push_back(s - itemsDeleted);
2393                 // else: deleted item, remove from selection
2394             }
2395
2396             m_selection = newsel;
2397         }
2398     }
2399
2400     // Change the current row to the last row if the current exceed the max row number
2401     if( m_currentRow > GetRowCount() )
2402         ChangeCurrentRow(m_count - 1);
2403
2404     GetOwner()->InvalidateColBestWidths();
2405     UpdateDisplay();
2406
2407     return true;
2408 }
2409
2410 bool wxDataViewMainWindow::ItemChanged(const wxDataViewItem & item)
2411 {
2412     SortPrepare();
2413     g_model->Resort();
2414
2415     GetOwner()->InvalidateColBestWidths();
2416
2417     // Send event
2418     wxWindow *parent = GetParent();
2419     wxDataViewEvent le(wxEVT_COMMAND_DATAVIEW_ITEM_VALUE_CHANGED, parent->GetId());
2420     le.SetEventObject(parent);
2421     le.SetModel(GetModel());
2422     le.SetItem(item);
2423     parent->ProcessWindowEvent(le);
2424
2425     return true;
2426 }
2427
2428 bool wxDataViewMainWindow::ValueChanged( const wxDataViewItem & item, unsigned int model_column )
2429 {
2430     int view_column = -1;
2431     unsigned int n_col = m_owner->GetColumnCount();
2432     for (unsigned i = 0; i < n_col; i++)
2433     {
2434         wxDataViewColumn *column = m_owner->GetColumn( i );
2435         if (column->GetModelColumn() == model_column)
2436         {
2437             view_column = (int) i;
2438             break;
2439         }
2440     }
2441     if (view_column == -1)
2442         return false;
2443
2444     // NOTE: to be valid, we cannot use e.g. INT_MAX - 1
2445 /*#define MAX_VIRTUAL_WIDTH       100000
2446
2447     wxRect rect( 0, row*m_lineHeight, MAX_VIRTUAL_WIDTH, m_lineHeight );
2448     m_owner->CalcScrolledPosition( rect.x, rect.y, &rect.x, &rect.y );
2449     Refresh( true, &rect );
2450
2451     return true;
2452 */
2453     SortPrepare();
2454     g_model->Resort();
2455
2456     GetOwner()->InvalidateColBestWidth(view_column);
2457
2458     // Send event
2459     wxWindow *parent = GetParent();
2460     wxDataViewEvent le(wxEVT_COMMAND_DATAVIEW_ITEM_VALUE_CHANGED, parent->GetId());
2461     le.SetEventObject(parent);
2462     le.SetModel(GetModel());
2463     le.SetItem(item);
2464     le.SetColumn(view_column);
2465     le.SetDataViewColumn(GetOwner()->GetColumn(view_column));
2466     parent->ProcessWindowEvent(le);
2467
2468     return true;
2469 }
2470
2471 bool wxDataViewMainWindow::Cleared()
2472 {
2473     DestroyTree();
2474     m_selection.Clear();
2475
2476     if (GetModel())
2477     {
2478         SortPrepare();
2479         BuildTree( GetModel() );
2480     }
2481     else
2482     {
2483         m_count = 0;
2484     }
2485
2486     GetOwner()->InvalidateColBestWidths();
2487     UpdateDisplay();
2488
2489     return true;
2490 }
2491
2492 void wxDataViewMainWindow::UpdateDisplay()
2493 {
2494     m_dirty = true;
2495     m_underMouse = NULL;
2496 }
2497
2498 void wxDataViewMainWindow::OnInternalIdle()
2499 {
2500     wxWindow::OnInternalIdle();
2501
2502     if (m_dirty)
2503     {
2504         RecalculateDisplay();
2505         m_dirty = false;
2506     }
2507 }
2508
2509 void wxDataViewMainWindow::RecalculateDisplay()
2510 {
2511     wxDataViewModel *model = GetModel();
2512     if (!model)
2513     {
2514         Refresh();
2515         return;
2516     }
2517
2518     int width = GetEndOfLastCol();
2519     int height = GetLineStart( GetRowCount() );
2520
2521     SetVirtualSize( width, height );
2522     GetOwner()->SetScrollRate( 10, m_lineHeight );
2523
2524     Refresh();
2525 }
2526
2527 void wxDataViewMainWindow::ScrollWindow( int dx, int dy, const wxRect *rect )
2528 {
2529     m_underMouse = NULL;
2530
2531     wxWindow::ScrollWindow( dx, dy, rect );
2532
2533     if (GetOwner()->m_headerArea)
2534         GetOwner()->m_headerArea->ScrollWindow( dx, 0 );
2535 }
2536
2537 void wxDataViewMainWindow::ScrollTo( int rows, int column )
2538 {
2539     m_underMouse = NULL;
2540
2541     int x, y;
2542     m_owner->GetScrollPixelsPerUnit( &x, &y );
2543     int sy = GetLineStart( rows )/y;
2544     int sx = -1;
2545     if( column != -1 )
2546     {
2547         wxRect rect = GetClientRect();
2548         int colnum = 0;
2549         int x_start, w = 0;
2550         int xx, yy, xe;
2551         m_owner->CalcUnscrolledPosition( rect.x, rect.y, &xx, &yy );
2552         for (x_start = 0; colnum < column; colnum++)
2553         {
2554             wxDataViewColumn *col = GetOwner()->GetColumnAt(colnum);
2555             if (col->IsHidden())
2556                 continue;      // skip it!
2557
2558             w = col->GetWidth();
2559             x_start += w;
2560         }
2561
2562         int x_end = x_start + w;
2563         xe = xx + rect.width;
2564         if( x_end > xe )
2565         {
2566             sx = ( xx + x_end - xe )/x;
2567         }
2568         if( x_start < xx )
2569         {
2570             sx = x_start/x;
2571         }
2572     }
2573     m_owner->Scroll( sx, sy );
2574 }
2575
2576 int wxDataViewMainWindow::GetCountPerPage() const
2577 {
2578     wxSize size = GetClientSize();
2579     return size.y / m_lineHeight;
2580 }
2581
2582 int wxDataViewMainWindow::GetEndOfLastCol() const
2583 {
2584     int width = 0;
2585     unsigned int i;
2586     for (i = 0; i < GetOwner()->GetColumnCount(); i++)
2587     {
2588         const wxDataViewColumn *c =
2589             const_cast<wxDataViewCtrl*>(GetOwner())->GetColumnAt( i );
2590
2591         if (!c->IsHidden())
2592             width += c->GetWidth();
2593     }
2594     return width;
2595 }
2596
2597 unsigned int wxDataViewMainWindow::GetFirstVisibleRow() const
2598 {
2599     int x = 0;
2600     int y = 0;
2601     m_owner->CalcUnscrolledPosition( x, y, &x, &y );
2602
2603     return GetLineAt( y );
2604 }
2605
2606 unsigned int wxDataViewMainWindow::GetLastVisibleRow()
2607 {
2608     wxSize client_size = GetClientSize();
2609     m_owner->CalcUnscrolledPosition( client_size.x, client_size.y,
2610                                     &client_size.x, &client_size.y );
2611
2612     // we should deal with the pixel here
2613     unsigned int row = GetLineAt(client_size.y) - 1;
2614
2615     return wxMin( GetRowCount()-1, row );
2616 }
2617
2618 unsigned int wxDataViewMainWindow::GetRowCount() const
2619 {
2620     if ( m_count == -1 )
2621     {
2622         wxDataViewMainWindow* const
2623             self = const_cast<wxDataViewMainWindow*>(this);
2624         self->m_count = RecalculateCount();
2625         self->UpdateDisplay();
2626     }
2627     return m_count;
2628 }
2629
2630 void wxDataViewMainWindow::ChangeCurrentRow( unsigned int row )
2631 {
2632     m_currentRow = row;
2633
2634     // send event
2635 }
2636
2637 void wxDataViewMainWindow::SelectAllRows( bool on )
2638 {
2639     if (IsEmpty())
2640         return;
2641
2642     if (on)
2643     {
2644         m_selection.Clear();
2645         for (unsigned int i = 0; i < GetRowCount(); i++)
2646             m_selection.Add( i );
2647         Refresh();
2648     }
2649     else
2650     {
2651         unsigned int first_visible = GetFirstVisibleRow();
2652         unsigned int last_visible = GetLastVisibleRow();
2653         unsigned int i;
2654         for (i = 0; i < m_selection.GetCount(); i++)
2655         {
2656             unsigned int row = m_selection[i];
2657             if ((row >= first_visible) && (row <= last_visible))
2658                 RefreshRow( row );
2659         }
2660         m_selection.Clear();
2661     }
2662 }
2663
2664 void wxDataViewMainWindow::SelectRow( unsigned int row, bool on )
2665 {
2666     if (m_selection.Index( row ) == wxNOT_FOUND)
2667     {
2668         if (on)
2669         {
2670             m_selection.Add( row );
2671             RefreshRow( row );
2672         }
2673     }
2674     else
2675     {
2676         if (!on)
2677         {
2678             m_selection.Remove( row );
2679             RefreshRow( row );
2680         }
2681     }
2682 }
2683
2684 void wxDataViewMainWindow::SelectRows( unsigned int from, unsigned int to, bool on )
2685 {
2686     if (from > to)
2687     {
2688         unsigned int tmp = from;
2689         from = to;
2690         to = tmp;
2691     }
2692
2693     unsigned int i;
2694     for (i = from; i <= to; i++)
2695     {
2696         if (m_selection.Index( i ) == wxNOT_FOUND)
2697         {
2698             if (on)
2699                 m_selection.Add( i );
2700         }
2701         else
2702         {
2703             if (!on)
2704                 m_selection.Remove( i );
2705         }
2706     }
2707     RefreshRows( from, to );
2708 }
2709
2710 void wxDataViewMainWindow::Select( const wxArrayInt& aSelections )
2711 {
2712     for (size_t i=0; i < aSelections.GetCount(); i++)
2713     {
2714         int n = aSelections[i];
2715
2716         m_selection.Add( n );
2717         RefreshRow( n );
2718     }
2719 }
2720
2721 void wxDataViewMainWindow::ReverseRowSelection( unsigned int row )
2722 {
2723     if (m_selection.Index( row ) == wxNOT_FOUND)
2724         m_selection.Add( row );
2725     else
2726         m_selection.Remove( row );
2727     RefreshRow( row );
2728 }
2729
2730 bool wxDataViewMainWindow::IsRowSelected( unsigned int row )
2731 {
2732     return (m_selection.Index( row ) != wxNOT_FOUND);
2733 }
2734
2735 void wxDataViewMainWindow::SendSelectionChangedEvent( const wxDataViewItem& item)
2736 {
2737     wxWindow *parent = GetParent();
2738     wxDataViewEvent le(wxEVT_COMMAND_DATAVIEW_SELECTION_CHANGED, parent->GetId());
2739
2740     le.SetEventObject(parent);
2741     le.SetModel(GetModel());
2742     le.SetItem( item );
2743
2744     parent->ProcessWindowEvent(le);
2745 }
2746
2747 void wxDataViewMainWindow::RefreshRow( unsigned int row )
2748 {
2749     wxRect rect( 0, GetLineStart( row ), GetEndOfLastCol(), GetLineHeight( row ) );
2750     m_owner->CalcScrolledPosition( rect.x, rect.y, &rect.x, &rect.y );
2751
2752     wxSize client_size = GetClientSize();
2753     wxRect client_rect( 0, 0, client_size.x, client_size.y );
2754     wxRect intersect_rect = client_rect.Intersect( rect );
2755     if (intersect_rect.width > 0)
2756         Refresh( true, &intersect_rect );
2757 }
2758
2759 void wxDataViewMainWindow::RefreshRows( unsigned int from, unsigned int to )
2760 {
2761     if (from > to)
2762     {
2763         unsigned int tmp = to;
2764         to = from;
2765         from = tmp;
2766     }
2767
2768     wxRect rect( 0, GetLineStart( from ), GetEndOfLastCol(), GetLineStart( (to-from+1) ) );
2769     m_owner->CalcScrolledPosition( rect.x, rect.y, &rect.x, &rect.y );
2770
2771     wxSize client_size = GetClientSize();
2772     wxRect client_rect( 0, 0, client_size.x, client_size.y );
2773     wxRect intersect_rect = client_rect.Intersect( rect );
2774     if (intersect_rect.width > 0)
2775         Refresh( true, &intersect_rect );
2776 }
2777
2778 void wxDataViewMainWindow::RefreshRowsAfter( unsigned int firstRow )
2779 {
2780     wxSize client_size = GetClientSize();
2781     int start = GetLineStart( firstRow );
2782     m_owner->CalcScrolledPosition( start, 0, &start, NULL );
2783     if (start > client_size.y) return;
2784
2785     wxRect rect( 0, start, client_size.x, client_size.y - start );
2786
2787     Refresh( true, &rect );
2788 }
2789
2790 wxRect wxDataViewMainWindow::GetLineRect( unsigned int row ) const
2791 {
2792     wxRect rect;
2793     rect.x = 0;
2794     rect.y = GetLineStart( row );
2795     rect.width = GetEndOfLastCol();
2796     rect.height = GetLineHeight( row );
2797
2798     return rect;
2799 }
2800
2801 int wxDataViewMainWindow::GetLineStart( unsigned int row ) const
2802 {
2803     const wxDataViewModel *model = GetModel();
2804
2805     if (GetOwner()->GetWindowStyle() & wxDV_VARIABLE_LINE_HEIGHT)
2806     {
2807         // TODO make more efficient
2808
2809         int start = 0;
2810
2811         unsigned int r;
2812         for (r = 0; r < row; r++)
2813         {
2814             const wxDataViewTreeNode* node = GetTreeNodeByRow(r);
2815             if (!node) return start;
2816
2817             wxDataViewItem item = node->GetItem();
2818
2819             unsigned int cols = GetOwner()->GetColumnCount();
2820             unsigned int col;
2821             int height = m_lineHeight;
2822             for (col = 0; col < cols; col++)
2823             {
2824                 const wxDataViewColumn *column = GetOwner()->GetColumn(col);
2825                 if (column->IsHidden())
2826                     continue;      // skip it!
2827
2828                 if ((col != 0) &&
2829                     model->IsContainer(item) &&
2830                     !model->HasContainerColumns(item))
2831                     continue;      // skip it!
2832
2833                 wxDataViewRenderer *renderer =
2834                     const_cast<wxDataViewRenderer*>(column->GetRenderer());
2835                 renderer->PrepareForItem(model, item, column->GetModelColumn());
2836
2837                 height = wxMax( height, renderer->GetSize().y );
2838             }
2839
2840             start += height;
2841         }
2842
2843         return start;
2844     }
2845     else
2846     {
2847         return row * m_lineHeight;
2848     }
2849 }
2850
2851 int wxDataViewMainWindow::GetLineAt( unsigned int y ) const
2852 {
2853     const wxDataViewModel *model = GetModel();
2854
2855     // check for the easy case first
2856     if ( !GetOwner()->HasFlag(wxDV_VARIABLE_LINE_HEIGHT) )
2857         return y / m_lineHeight;
2858
2859     // TODO make more efficient
2860     unsigned int row = 0;
2861     unsigned int yy = 0;
2862     for (;;)
2863     {
2864         const wxDataViewTreeNode* node = GetTreeNodeByRow(row);
2865         if (!node)
2866         {
2867             // not really correct...
2868             return row + ((y-yy) / m_lineHeight);
2869         }
2870
2871         wxDataViewItem item = node->GetItem();
2872
2873         unsigned int cols = GetOwner()->GetColumnCount();
2874         unsigned int col;
2875         int height = m_lineHeight;
2876         for (col = 0; col < cols; col++)
2877         {
2878             const wxDataViewColumn *column = GetOwner()->GetColumn(col);
2879             if (column->IsHidden())
2880                 continue;      // skip it!
2881
2882             if ((col != 0) &&
2883                 model->IsContainer(item) &&
2884                 !model->HasContainerColumns(item))
2885                 continue;      // skip it!
2886
2887             wxDataViewRenderer *renderer =
2888                 const_cast<wxDataViewRenderer*>(column->GetRenderer());
2889             renderer->PrepareForItem(model, item, column->GetModelColumn());
2890
2891             height = wxMax( height, renderer->GetSize().y );
2892         }
2893
2894         yy += height;
2895         if (y < yy)
2896             return row;
2897
2898         row++;
2899     }
2900 }
2901
2902 int wxDataViewMainWindow::GetLineHeight( unsigned int row ) const
2903 {
2904     const wxDataViewModel *model = GetModel();
2905
2906     if (GetOwner()->GetWindowStyle() & wxDV_VARIABLE_LINE_HEIGHT)
2907     {
2908         wxASSERT( !IsVirtualList() );
2909
2910         const wxDataViewTreeNode* node = GetTreeNodeByRow(row);
2911         // wxASSERT( node );
2912         if (!node) return m_lineHeight;
2913
2914         wxDataViewItem item = node->GetItem();
2915
2916         int height = m_lineHeight;
2917
2918         unsigned int cols = GetOwner()->GetColumnCount();
2919         unsigned int col;
2920         for (col = 0; col < cols; col++)
2921         {
2922             const wxDataViewColumn *column = GetOwner()->GetColumn(col);
2923             if (column->IsHidden())
2924                 continue;      // skip it!
2925
2926             if ((col != 0) &&
2927                 model->IsContainer(item) &&
2928                 !model->HasContainerColumns(item))
2929                 continue;      // skip it!
2930
2931             wxDataViewRenderer *renderer =
2932                 const_cast<wxDataViewRenderer*>(column->GetRenderer());
2933             renderer->PrepareForItem(model, item, column->GetModelColumn());
2934
2935             height = wxMax( height, renderer->GetSize().y );
2936         }
2937
2938         return height;
2939     }
2940     else
2941     {
2942         return m_lineHeight;
2943     }
2944 }
2945
2946
2947 class RowToTreeNodeJob: public DoJob
2948 {
2949 public:
2950     RowToTreeNodeJob( unsigned int row , int current, wxDataViewTreeNode * node )
2951     {
2952         this->row = row;
2953         this->current = current;
2954         ret = NULL;
2955         parent = node;
2956     }
2957
2958     virtual int operator() ( wxDataViewTreeNode * node )
2959     {
2960         current ++;
2961         if( current == static_cast<int>(row))
2962         {
2963             ret = node;
2964             return DoJob::DONE;
2965         }
2966
2967         if( node->GetSubTreeCount() + current < static_cast<int>(row) )
2968         {
2969             current += node->GetSubTreeCount();
2970             return  DoJob::SKIP_SUBTREE;
2971         }
2972         else
2973         {
2974             parent = node;
2975
2976             // If the current node has only leaf children, we can find the
2977             // desired node directly. This can speed up finding the node
2978             // in some cases, and will have a very good effect for list views.
2979             if ( node->HasChildren() &&
2980                  (int)node->GetChildNodes().size() == node->GetSubTreeCount() )
2981             {
2982                 const int index = static_cast<int>(row) - current - 1;
2983                 ret = node->GetChildNodes()[index];
2984                 return DoJob::DONE;
2985             }
2986
2987             return DoJob::CONTINUE;
2988         }
2989     }
2990
2991     wxDataViewTreeNode * GetResult() const
2992         { return ret; }
2993
2994 private:
2995     unsigned int row;
2996     int current;
2997     wxDataViewTreeNode * ret;
2998     wxDataViewTreeNode * parent;
2999 };
3000
3001 wxDataViewTreeNode * wxDataViewMainWindow::GetTreeNodeByRow(unsigned int row) const
3002 {
3003     wxASSERT( !IsVirtualList() );
3004
3005     RowToTreeNodeJob job( row , -2, m_root );
3006     Walker( m_root , job );
3007     return job.GetResult();
3008 }
3009
3010 wxDataViewItem wxDataViewMainWindow::GetItemByRow(unsigned int row) const
3011 {
3012     wxDataViewItem item;
3013     if (IsVirtualList())
3014     {
3015         if ( row < GetRowCount() )
3016             item = wxDataViewItem(wxUIntToPtr(row+1));
3017     }
3018     else
3019     {
3020         wxDataViewTreeNode *node = GetTreeNodeByRow(row);
3021         if ( node )
3022             item = node->GetItem();
3023     }
3024
3025     return item;
3026 }
3027
3028 bool
3029 wxDataViewMainWindow::SendExpanderEvent(wxEventType type,
3030                                         const wxDataViewItem& item)
3031 {
3032     wxWindow *parent = GetParent();
3033     wxDataViewEvent le(type, parent->GetId());
3034
3035     le.SetEventObject(parent);
3036     le.SetModel(GetModel());
3037     le.SetItem( item );
3038
3039     return !parent->ProcessWindowEvent(le) || le.IsAllowed();
3040 }
3041
3042 bool wxDataViewMainWindow::IsExpanded( unsigned int row ) const
3043 {
3044     if (IsList())
3045         return false;
3046
3047     wxDataViewTreeNode * node = GetTreeNodeByRow(row);
3048     if (!node)
3049         return false;
3050
3051     if (!node->HasChildren())
3052         return false;
3053
3054     return node->IsOpen();
3055 }
3056
3057 bool wxDataViewMainWindow::HasChildren( unsigned int row ) const
3058 {
3059     if (IsList())
3060         return false;
3061
3062     wxDataViewTreeNode * node = GetTreeNodeByRow(row);
3063     if (!node)
3064         return false;
3065
3066     if (!node->HasChildren())
3067         return false;
3068
3069     return true;
3070 }
3071
3072 void wxDataViewMainWindow::Expand( unsigned int row )
3073 {
3074     if (IsList())
3075         return;
3076
3077     wxDataViewTreeNode * node = GetTreeNodeByRow(row);
3078     if (!node)
3079         return;
3080
3081     if (!node->HasChildren())
3082         return;
3083
3084             if (!node->IsOpen())
3085             {
3086                 if ( !SendExpanderEvent(wxEVT_COMMAND_DATAVIEW_ITEM_EXPANDING, node->GetItem()) )
3087                 {
3088                     // Vetoed by the event handler.
3089                     return;
3090                 }
3091
3092                 node->ToggleOpen();
3093
3094                 // build the children of current node
3095                 if( node->GetChildNodes().empty() )
3096                 {
3097                     SortPrepare();
3098                     ::BuildTreeHelper(GetModel(), node->GetItem(), node);
3099                 }
3100
3101                 // By expanding the node all row indices that are currently in the selection list
3102                 // and are greater than our node have become invalid. So we have to correct that now.
3103                 const unsigned rowAdjustment = node->GetSubTreeCount();
3104                 for(unsigned i=0; i<m_selection.size(); ++i)
3105                 {
3106                     const unsigned testRow = m_selection[i];
3107                     // all rows above us are not affected, so skip them
3108                     if(testRow <= row)
3109                         continue;
3110
3111                     m_selection[i] += rowAdjustment;
3112                 }
3113
3114                 if(m_currentRow > row)
3115                     ChangeCurrentRow(m_currentRow + rowAdjustment);
3116
3117                 m_count = -1;
3118                 UpdateDisplay();
3119                 // Send the expanded event
3120                 SendExpanderEvent(wxEVT_COMMAND_DATAVIEW_ITEM_EXPANDED,node->GetItem());
3121             }
3122 }
3123
3124 void wxDataViewMainWindow::Collapse(unsigned int row)
3125 {
3126     if (IsList())
3127         return;
3128
3129     wxDataViewTreeNode *node = GetTreeNodeByRow(row);
3130     if (!node)
3131         return;
3132
3133     if (!node->HasChildren())
3134         return;
3135
3136         if (node->IsOpen())
3137         {
3138             if ( !SendExpanderEvent(wxEVT_COMMAND_DATAVIEW_ITEM_COLLAPSING,node->GetItem()) )
3139             {
3140                 // Vetoed by the event handler.
3141                 return;
3142             }
3143
3144             // Find out if there are selected items below the current node.
3145             bool selectCollapsingRow = false;
3146             const unsigned rowAdjustment = node->GetSubTreeCount();
3147             unsigned maxRowToBeTested = row + rowAdjustment;
3148             for(unsigned i=0; i<m_selection.size(); ++i)
3149             {
3150                 const unsigned testRow = m_selection[i];
3151                 if(testRow > row && testRow <= maxRowToBeTested)
3152                 {
3153                     selectCollapsingRow = true;
3154                     // get out as soon as we have found a node that is selected
3155                     break;
3156                 }
3157             }
3158
3159             node->ToggleOpen();
3160
3161             // If the node to be closed has selected items the user won't see those any longer.
3162             // We select the collapsing node in this case.
3163             if(selectCollapsingRow)
3164             {
3165                 SelectAllRows(false);
3166                 ChangeCurrentRow(row);
3167                 SelectRow(row, true);
3168                 SendSelectionChangedEvent(GetItemByRow(row));
3169             }
3170             else
3171             {
3172                 // if there were no selected items below our node we still need to "fix" the
3173                 // selection list to adjust for the changing of the row indices.
3174                 // We actually do the opposite of what we are doing in Expand().
3175                 for(unsigned i=0; i<m_selection.size(); ++i)
3176                 {
3177                     const unsigned testRow = m_selection[i];
3178                     // all rows above us are not affected, so skip them
3179                     if(testRow <= row)
3180                         continue;
3181
3182                     m_selection[i] -= rowAdjustment;
3183                 }
3184
3185                 // if the "current row" is being collapsed away we change it to the current row ;-)
3186                 if(m_currentRow > row && m_currentRow <= maxRowToBeTested)
3187                     ChangeCurrentRow(row);
3188                 else if(m_currentRow > row)
3189                     ChangeCurrentRow(m_currentRow - rowAdjustment);
3190             }
3191
3192             m_count = -1;
3193             UpdateDisplay();
3194             SendExpanderEvent(wxEVT_COMMAND_DATAVIEW_ITEM_COLLAPSED,node->GetItem());
3195         }
3196 }
3197
3198 wxDataViewTreeNode * wxDataViewMainWindow::FindNode( const wxDataViewItem & item )
3199 {
3200     const wxDataViewModel * model = GetModel();
3201     if( model == NULL )
3202         return NULL;
3203
3204     if (!item.IsOk())
3205         return m_root;
3206
3207     // Compose the parent-chain for the item we are looking for
3208     wxVector<wxDataViewItem> parentChain;
3209     wxDataViewItem it( item );
3210     while( it.IsOk() )
3211     {
3212         parentChain.push_back(it);
3213         it = model->GetParent(it);
3214     }
3215
3216     // Find the item along the parent-chain.
3217     // This algorithm is designed to speed up the node-finding method
3218     wxDataViewTreeNode* node = m_root;
3219     for( unsigned iter = parentChain.size()-1; ; --iter )
3220     {
3221         if( node->HasChildren() )
3222         {
3223             if( node->GetChildNodes().empty() )
3224             {
3225                 // Even though the item is a container, it doesn't have any
3226                 // child nodes in the control's representation yet. We have
3227                 // to realize its subtree now.
3228                 SortPrepare();
3229                 ::BuildTreeHelper(model, node->GetItem(), node);
3230             }
3231
3232             const wxDataViewTreeNodes& nodes = node->GetChildNodes();
3233             bool found = false;
3234
3235             for (unsigned i = 0; i < nodes.GetCount(); ++i)
3236             {
3237                 wxDataViewTreeNode* currentNode = nodes[i];
3238                 if (currentNode->GetItem() == parentChain[iter])
3239                 {
3240                     if (currentNode->GetItem() == item)
3241                         return currentNode;
3242
3243                     node = currentNode;
3244                     found = true;
3245                     break;
3246                 }
3247             }
3248             if (!found)
3249                 return NULL;
3250         }
3251         else
3252             return NULL;
3253
3254         if ( !iter )
3255             break;
3256     }
3257     return NULL;
3258 }
3259
3260 void wxDataViewMainWindow::HitTest( const wxPoint & point, wxDataViewItem & item,
3261                                     wxDataViewColumn* &column )
3262 {
3263     wxDataViewColumn *col = NULL;
3264     unsigned int cols = GetOwner()->GetColumnCount();
3265     unsigned int colnum = 0;
3266     int x, y;
3267     m_owner->CalcUnscrolledPosition( point.x, point.y, &x, &y );
3268     for (unsigned x_start = 0; colnum < cols; colnum++)
3269     {
3270         col = GetOwner()->GetColumnAt(colnum);
3271         if (col->IsHidden())
3272             continue;      // skip it!
3273
3274         unsigned int w = col->GetWidth();
3275         if (x_start+w >= (unsigned int)x)
3276             break;
3277
3278         x_start += w;
3279     }
3280
3281     column = col;
3282     item = GetItemByRow( GetLineAt( y ) );
3283 }
3284
3285 wxRect wxDataViewMainWindow::GetItemRect( const wxDataViewItem & item,
3286                                           const wxDataViewColumn* column )
3287 {
3288     int xpos = 0;
3289     int width = 0;
3290
3291     unsigned int cols = GetOwner()->GetColumnCount();
3292     // If column is null the loop will compute the combined width of all columns.
3293     // Otherwise, it will compute the x position of the column we are looking for.
3294     for (unsigned int i = 0; i < cols; i++)
3295     {
3296         wxDataViewColumn* col = GetOwner()->GetColumnAt( i );
3297
3298         if (col == column)
3299             break;
3300
3301         if (col->IsHidden())
3302             continue;      // skip it!
3303
3304         xpos += col->GetWidth();
3305         width += col->GetWidth();
3306     }
3307
3308     if(column != 0)
3309     {
3310         // If we have a column, we need can get its width directly.
3311         if(column->IsHidden())
3312             width = 0;
3313         else
3314             width = column->GetWidth();
3315
3316     }
3317     else
3318     {
3319         // If we have no column, we reset the x position back to zero.
3320         xpos = 0;
3321     }
3322
3323     // we have to take an expander column into account and compute its indentation
3324     // to get the correct x position where the actual text is
3325     int indent = 0;
3326     int row = GetRowByItem(item);
3327     if (!IsList() &&
3328             (column == 0 || GetExpanderColumnOrFirstOne(GetOwner()) == column) )
3329     {
3330         wxDataViewTreeNode* node = GetTreeNodeByRow(row);
3331         indent = GetOwner()->GetIndent() * node->GetIndentLevel();
3332         indent = indent + m_lineHeight; // use m_lineHeight as the width of the expander
3333     }
3334
3335     wxRect itemRect( xpos + indent,
3336                      GetLineStart( row ),
3337                      width - indent,
3338                      GetLineHeight( row ) );
3339
3340     GetOwner()->CalcScrolledPosition(  itemRect.x,  itemRect.y,
3341                                       &itemRect.x, &itemRect.y );
3342
3343     return itemRect;
3344 }
3345
3346 int wxDataViewMainWindow::RecalculateCount() const
3347 {
3348     if (IsVirtualList())
3349     {
3350         wxDataViewVirtualListModel *list_model =
3351             (wxDataViewVirtualListModel*) GetModel();
3352
3353         return list_model->GetCount();
3354     }
3355     else
3356     {
3357         return m_root->GetSubTreeCount();
3358     }
3359 }
3360
3361 class ItemToRowJob : public DoJob
3362 {
3363 public:
3364     ItemToRowJob(const wxDataViewItem& item_, wxVector<wxDataViewItem>::reverse_iterator iter)
3365         : m_iter(iter),
3366         item(item_)
3367     {
3368         ret = -1;
3369     }
3370
3371     // Maybe binary search will help to speed up this process
3372     virtual int operator() ( wxDataViewTreeNode * node)
3373     {
3374         ret ++;
3375         if( node->GetItem() == item )
3376         {
3377             return DoJob::DONE;
3378         }
3379
3380         if( node->GetItem() == *m_iter )
3381         {
3382             m_iter++;
3383             return DoJob::CONTINUE;
3384         }
3385         else
3386         {
3387             ret += node->GetSubTreeCount();
3388             return DoJob::SKIP_SUBTREE;
3389         }
3390
3391     }
3392
3393     // the row number is begin from zero
3394     int GetResult() const
3395         { return ret -1; }
3396
3397 private:
3398     wxVector<wxDataViewItem>::reverse_iterator m_iter;
3399     wxDataViewItem item;
3400     int ret;
3401
3402 };
3403
3404 int wxDataViewMainWindow::GetRowByItem(const wxDataViewItem & item) const
3405 {
3406     const wxDataViewModel * model = GetModel();
3407     if( model == NULL )
3408         return -1;
3409
3410     if (IsVirtualList())
3411     {
3412         return wxPtrToUInt( item.GetID() ) -1;
3413     }
3414     else
3415     {
3416         if( !item.IsOk() )
3417             return -1;
3418
3419         // Compose the parent-chain of the item we are looking for
3420         wxVector<wxDataViewItem> parentChain;
3421         wxDataViewItem it( item );
3422         while( it.IsOk() )
3423         {
3424             parentChain.push_back(it);
3425             it = model->GetParent(it);
3426         }
3427
3428         // add an 'invalid' item to represent our 'invisible' root node
3429         parentChain.push_back(wxDataViewItem());
3430
3431         // the parent chain was created by adding the deepest parent first.
3432         // so if we want to start at the root node, we have to iterate backwards through the vector
3433         ItemToRowJob job( item, parentChain.rbegin() );
3434         Walker( m_root, job );
3435         return job.GetResult();
3436     }
3437 }
3438
3439 static void BuildTreeHelper( const wxDataViewModel * model,  const wxDataViewItem & item,
3440                              wxDataViewTreeNode * node)
3441 {
3442     if( !model->IsContainer( item ) )
3443         return;
3444
3445     wxDataViewItemArray children;
3446     unsigned int num = model->GetChildren( item, children);
3447
3448     for ( unsigned int index = 0; index < num; index++ )
3449     {
3450         wxDataViewTreeNode *n = new wxDataViewTreeNode(node, children[index]);
3451
3452         if( model->IsContainer(children[index]) )
3453             n->SetHasChildren( true );
3454
3455         node->InsertChild(n, index);
3456     }
3457
3458     wxASSERT( node->IsOpen() );
3459     node->ChangeSubTreeCount(+num);
3460 }
3461
3462 void wxDataViewMainWindow::BuildTree(wxDataViewModel * model)
3463 {
3464     DestroyTree();
3465
3466     if (GetModel()->IsVirtualListModel())
3467     {
3468         m_count = -1;
3469         return;
3470     }
3471
3472     m_root = wxDataViewTreeNode::CreateRootNode();
3473
3474     // First we define a invalid item to fetch the top-level elements
3475     wxDataViewItem item;
3476     SortPrepare();
3477     BuildTreeHelper( model, item, m_root);
3478     m_count = -1;
3479 }
3480
3481 void wxDataViewMainWindow::DestroyTree()
3482 {
3483     if (!IsVirtualList())
3484     {
3485         wxDELETE(m_root);
3486         m_count = 0;
3487     }
3488 }
3489
3490 wxDataViewColumn*
3491 wxDataViewMainWindow::FindColumnForEditing(const wxDataViewItem& item, wxDataViewCellMode mode)
3492 {
3493     // Edit the current column editable in 'mode'. If no column is focused
3494     // (typically because the user has full row selected), try to find the
3495     // first editable column (this would typically be a checkbox for
3496     // wxDATAVIEW_CELL_ACTIVATABLE and we don't want to force the user to set
3497     // focus on the checkbox column; or on the only editable text column).
3498
3499     wxDataViewColumn *candidate = m_currentCol;
3500
3501     if ( candidate &&
3502          !IsCellEditableInMode(item, candidate, mode) &&
3503          !m_currentColSetByKeyboard )
3504     {
3505         // If current column was set by mouse to something not editable (in
3506         // 'mode') and the user pressed Space/F2 to edit it, treat the
3507         // situation as if there was whole-row focus, because that's what is
3508         // visually indicated and the mouse click could very well be targeted
3509         // on the row rather than on an individual cell.
3510         //
3511         // But if it was done by keyboard, respect that even if the column
3512         // isn't editable, because focus is visually on that column and editing
3513         // something else would be surprising.
3514         candidate = NULL;
3515     }
3516
3517     if ( !candidate )
3518     {
3519         const unsigned cols = GetOwner()->GetColumnCount();
3520         for ( unsigned i = 0; i < cols; i++ )
3521         {
3522             wxDataViewColumn *c = GetOwner()->GetColumnAt(i);
3523             if ( c->IsHidden() )
3524                 continue;
3525
3526             if ( IsCellEditableInMode(item, c, mode) )
3527             {
3528                 candidate = c;
3529                 break;
3530             }
3531         }
3532     }
3533
3534     // If on container item without columns, only the expander column
3535     // may be directly editable:
3536     if ( candidate &&
3537          GetOwner()->GetExpanderColumn() != candidate &&
3538          GetModel()->IsContainer(item) &&
3539          !GetModel()->HasContainerColumns(item) )
3540     {
3541         candidate = GetOwner()->GetExpanderColumn();
3542     }
3543
3544     if ( !candidate )
3545        return NULL;
3546
3547    if ( !IsCellEditableInMode(item, candidate, mode) )
3548        return NULL;
3549
3550    return candidate;
3551 }
3552
3553 bool wxDataViewMainWindow::IsCellEditableInMode(const wxDataViewItem& item,
3554                                                 const wxDataViewColumn *col,
3555                                                 wxDataViewCellMode mode) const
3556 {
3557     if ( col->GetRenderer()->GetMode() != mode )
3558         return false;
3559
3560     if ( !GetModel()->IsEnabled(item, col->GetModelColumn()) )
3561         return false;
3562
3563     return true;
3564 }
3565
3566 void wxDataViewMainWindow::OnCharHook(wxKeyEvent& event)
3567 {
3568     if ( m_editorCtrl )
3569     {
3570         // Handle any keys special for the in-place editor and return without
3571         // calling Skip() below.
3572         switch ( event.GetKeyCode() )
3573         {
3574             case WXK_ESCAPE:
3575                 m_editorRenderer->CancelEditing();
3576                 return;
3577
3578             case WXK_RETURN:
3579                 m_editorRenderer->FinishEditing();
3580                 return;
3581         }
3582     }
3583
3584     event.Skip();
3585 }
3586
3587 void wxDataViewMainWindow::OnChar( wxKeyEvent &event )
3588 {
3589     wxWindow * const parent = GetParent();
3590
3591     // propagate the char event upwards
3592     wxKeyEvent eventForParent(event);
3593     eventForParent.SetEventObject(parent);
3594     if ( parent->ProcessWindowEvent(eventForParent) )
3595         return;
3596
3597     if ( parent->HandleAsNavigationKey(event) )
3598         return;
3599
3600     // no item -> nothing to do
3601     if (!HasCurrentRow())
3602     {
3603         event.Skip();
3604         return;
3605     }
3606
3607     // don't use m_linesPerPage directly as it might not be computed yet
3608     const int pageSize = GetCountPerPage();
3609     wxCHECK_RET( pageSize, wxT("should have non zero page size") );
3610
3611     switch ( event.GetKeyCode() )
3612     {
3613         case WXK_RETURN:
3614             if ( event.HasModifiers() )
3615             {
3616                 event.Skip();
3617                 break;
3618             }
3619             else
3620             {
3621                 // Enter activates the item, i.e. sends wxEVT_COMMAND_DATAVIEW_ITEM_ACTIVATED to
3622                 // it. Only if that event is not handled do we activate column renderer (which
3623                 // is normally done by Space) or even inline editing.
3624
3625                 const wxDataViewItem item = GetItemByRow(m_currentRow);
3626
3627                 wxDataViewEvent le(wxEVT_COMMAND_DATAVIEW_ITEM_ACTIVATED,
3628                                    parent->GetId());
3629                 le.SetItem(item);
3630                 le.SetEventObject(parent);
3631                 le.SetModel(GetModel());
3632
3633                 if ( parent->ProcessWindowEvent(le) )
3634                     break;
3635                 // else: fall through to WXK_SPACE handling
3636             }
3637
3638         case WXK_SPACE:
3639             if ( event.HasModifiers() )
3640             {
3641                 event.Skip();
3642                 break;
3643             }
3644             else
3645             {
3646                 // Space toggles activatable items or -- if not activatable --
3647                 // starts inline editing (this is normally done using F2 on
3648                 // Windows, but Space is common everywhere else, so use it too
3649                 // for greater cross-platform compatibility).
3650
3651                 const wxDataViewItem item = GetItemByRow(m_currentRow);
3652
3653                 // Activate the current activatable column. If not column is focused (typically
3654                 // because the user has full row selected), try to find the first activatable
3655                 // column (this would typically be a checkbox and we don't want to force the user
3656                 // to set focus on the checkbox column).
3657                 wxDataViewColumn *activatableCol = FindColumnForEditing(item, wxDATAVIEW_CELL_ACTIVATABLE);
3658
3659                 if ( activatableCol )
3660                 {
3661                     const unsigned colIdx = activatableCol->GetModelColumn();
3662                     const wxRect cell_rect = GetOwner()->GetItemRect(item, activatableCol);
3663
3664                     wxDataViewRenderer *cell = activatableCol->GetRenderer();
3665                     cell->PrepareForItem(GetModel(), item, colIdx);
3666                     cell->WXActivateCell(cell_rect, GetModel(), item, colIdx, NULL);
3667
3668                     break;
3669                 }
3670                 // else: fall through to WXK_F2 handling
3671             }
3672
3673         case WXK_F2:
3674             if ( event.HasModifiers() )
3675             {
3676                 event.Skip();
3677                 break;
3678             }
3679             else
3680             {
3681                 if( !m_selection.empty() )
3682                 {
3683                     // Mimic Windows 7 behavior: edit the item that has focus
3684                     // if it is selected and the first selected item if focus
3685                     // is out of selection.
3686                     int sel;
3687                     if ( m_selection.Index(m_currentRow) != wxNOT_FOUND )
3688                         sel = m_currentRow;
3689                     else
3690                         sel = m_selection[0];
3691
3692
3693                     const wxDataViewItem item = GetItemByRow(sel);
3694
3695                     // Edit the current column. If no column is focused
3696                     // (typically because the user has full row selected), try
3697                     // to find the first editable column.
3698                     wxDataViewColumn *editableCol = FindColumnForEditing(item, wxDATAVIEW_CELL_EDITABLE);
3699
3700                     if ( editableCol )
3701                         GetOwner()->EditItem(item, editableCol);
3702                 }
3703             }
3704             break;
3705
3706         case WXK_UP:
3707             if ( m_currentRow > 0 )
3708                 OnVerticalNavigation( m_currentRow - 1, event );
3709             break;
3710
3711         case WXK_DOWN:
3712             if ( m_currentRow + 1 < GetRowCount() )
3713                 OnVerticalNavigation( m_currentRow + 1, event );
3714             break;
3715         // Add the process for tree expanding/collapsing
3716         case WXK_LEFT:
3717             OnLeftKey();
3718             break;
3719
3720         case WXK_RIGHT:
3721             OnRightKey();
3722             break;
3723
3724         case WXK_END:
3725         {
3726             if (!IsEmpty())
3727                 OnVerticalNavigation( GetRowCount() - 1, event );
3728             break;
3729         }
3730         case WXK_HOME:
3731             if (!IsEmpty())
3732                 OnVerticalNavigation( 0, event );
3733             break;
3734
3735         case WXK_PAGEUP:
3736             {
3737                 int steps = pageSize - 1;
3738                 int index = m_currentRow - steps;
3739                 if (index < 0)
3740                     index = 0;
3741
3742                 OnVerticalNavigation( index, event );
3743             }
3744             break;
3745
3746         case WXK_PAGEDOWN:
3747             {
3748                 int steps = pageSize - 1;
3749                 unsigned int index = m_currentRow + steps;
3750                 unsigned int count = GetRowCount();
3751                 if ( index >= count )
3752                     index = count - 1;
3753
3754                 OnVerticalNavigation( index, event );
3755             }
3756             break;
3757
3758         default:
3759             event.Skip();
3760     }
3761 }
3762
3763 void wxDataViewMainWindow::OnVerticalNavigation(unsigned int newCurrent, const wxKeyEvent& event)
3764 {
3765     wxCHECK_RET( newCurrent < GetRowCount(),
3766                 wxT("invalid item index in OnVerticalNavigation()") );
3767
3768     // if there is no selection, we cannot move it anywhere
3769     if (!HasCurrentRow())
3770         return;
3771
3772     unsigned int oldCurrent = m_currentRow;
3773
3774     // in single selection we just ignore Shift as we can't select several
3775     // items anyhow
3776     if ( event.ShiftDown() && !IsSingleSel() )
3777     {
3778         RefreshRow( oldCurrent );
3779
3780         ChangeCurrentRow( newCurrent );
3781
3782         // select all the items between the old and the new one
3783         if ( oldCurrent > newCurrent )
3784         {
3785             newCurrent = oldCurrent;
3786             oldCurrent = m_currentRow;
3787         }
3788
3789         SelectRows( oldCurrent, newCurrent, true );
3790         if (oldCurrent!=newCurrent)
3791             SendSelectionChangedEvent(GetItemByRow(m_selection[0]));
3792     }
3793     else // !shift
3794     {
3795         RefreshRow( oldCurrent );
3796
3797         // all previously selected items are unselected unless ctrl is held
3798         if ( !event.ControlDown() )
3799             SelectAllRows(false);
3800
3801         ChangeCurrentRow( newCurrent );
3802
3803         if ( !event.ControlDown() )
3804         {
3805             SelectRow( m_currentRow, true );
3806             SendSelectionChangedEvent(GetItemByRow(m_currentRow));
3807         }
3808         else
3809             RefreshRow( m_currentRow );
3810     }
3811
3812     GetOwner()->EnsureVisible( m_currentRow, -1 );
3813 }
3814
3815 void wxDataViewMainWindow::OnLeftKey()
3816 {
3817     if ( IsList() )
3818     {
3819         TryAdvanceCurrentColumn(NULL, /*forward=*/false);
3820     }
3821     else
3822     {
3823         wxDataViewTreeNode* node = GetTreeNodeByRow(m_currentRow);
3824
3825         if ( TryAdvanceCurrentColumn(node, /*forward=*/false) )
3826             return;
3827
3828         // Because TryAdvanceCurrentColumn() return false, we are at the first
3829         // column or using whole-row selection. In this situation, we can use
3830         // the standard TreeView handling of the left key.
3831         if (node->HasChildren() && node->IsOpen())
3832         {
3833             Collapse(m_currentRow);
3834         }
3835         else
3836         {
3837             // if the node is already closed, we move the selection to its parent
3838             wxDataViewTreeNode *parent_node = node->GetParent();
3839
3840             if (parent_node)
3841             {
3842                 int parent = GetRowByItem( parent_node->GetItem() );
3843                 if ( parent >= 0 )
3844                 {
3845                     unsigned int row = m_currentRow;
3846                     SelectRow( row, false);
3847                     SelectRow( parent, true );
3848                     ChangeCurrentRow( parent );
3849                     GetOwner()->EnsureVisible( parent, -1 );
3850                     SendSelectionChangedEvent( parent_node->GetItem() );
3851                 }
3852             }
3853         }
3854     }
3855 }
3856
3857 void wxDataViewMainWindow::OnRightKey()
3858 {
3859     if ( IsList() )
3860     {
3861         TryAdvanceCurrentColumn(NULL, /*forward=*/true);
3862     }
3863     else
3864     {
3865         wxDataViewTreeNode* node = GetTreeNodeByRow(m_currentRow);
3866
3867         if ( node->HasChildren() )
3868         {
3869             if ( !node->IsOpen() )
3870             {
3871                 Expand( m_currentRow );
3872             }
3873             else
3874             {
3875                 // if the node is already open, we move the selection to the first child
3876                 unsigned int row = m_currentRow;
3877                 SelectRow( row, false );
3878                 SelectRow( row + 1, true );
3879                 ChangeCurrentRow( row + 1 );
3880                 GetOwner()->EnsureVisible( row + 1, -1 );
3881                 SendSelectionChangedEvent( GetItemByRow(row+1) );
3882             }
3883         }
3884         else
3885         {
3886             TryAdvanceCurrentColumn(node, /*forward=*/true);
3887         }
3888     }
3889 }
3890
3891 bool wxDataViewMainWindow::TryAdvanceCurrentColumn(wxDataViewTreeNode *node, bool forward)
3892 {
3893     if ( GetOwner()->GetColumnCount() == 0 )
3894         return false;
3895
3896     if ( !m_useCellFocus )
3897         return false;
3898
3899     if ( node )
3900     {
3901         // navigation shouldn't work in branch nodes without other columns:
3902         if ( node->HasChildren() && !GetModel()->HasContainerColumns(node->GetItem()) )
3903             return false;
3904     }
3905
3906     if ( m_currentCol == NULL || !m_currentColSetByKeyboard )
3907     {
3908         if ( forward )
3909         {
3910             m_currentCol = GetOwner()->GetColumnAt(1);
3911             m_currentColSetByKeyboard = true;
3912             RefreshRow(m_currentRow);
3913             return true;
3914         }
3915         else
3916             return false;
3917     }
3918
3919     int idx = GetOwner()->GetColumnIndex(m_currentCol) + (forward ? +1 : -1);
3920
3921     if ( idx >= (int)GetOwner()->GetColumnCount() )
3922         return false;
3923
3924     GetOwner()->EnsureVisible(m_currentRow, idx);
3925
3926     if ( idx < 1 )
3927     {
3928         // We are going to the left of the second column. Reset to whole-row
3929         // focus (which means first column would be edited).
3930         m_currentCol = NULL;
3931         RefreshRow(m_currentRow);
3932         return true;
3933     }
3934
3935     m_currentCol = GetOwner()->GetColumnAt(idx);
3936     m_currentColSetByKeyboard = true;
3937     RefreshRow(m_currentRow);
3938     return true;
3939 }
3940
3941 void wxDataViewMainWindow::OnMouse( wxMouseEvent &event )
3942 {
3943     if (event.GetEventType() == wxEVT_MOUSEWHEEL)
3944     {
3945         // let the base handle mouse wheel events.
3946         event.Skip();
3947         return;
3948     }
3949
3950     if(event.LeftDown())
3951     {
3952         // Not skipping this event would prevent the system from setting focus
3953         // to this window.
3954         event.Skip();
3955     }
3956
3957     int x = event.GetX();
3958     int y = event.GetY();
3959     m_owner->CalcUnscrolledPosition( x, y, &x, &y );
3960     wxDataViewColumn *col = NULL;
3961
3962     int xpos = 0;
3963     unsigned int cols = GetOwner()->GetColumnCount();
3964     unsigned int i;
3965     for (i = 0; i < cols; i++)
3966     {
3967         wxDataViewColumn *c = GetOwner()->GetColumnAt( i );
3968         if (c->IsHidden())
3969             continue;      // skip it!
3970
3971         if (x < xpos + c->GetWidth())
3972         {
3973             col = c;
3974             break;
3975         }
3976         xpos += c->GetWidth();
3977     }
3978
3979     wxDataViewModel* const model = GetModel();
3980
3981     const unsigned int current = GetLineAt( y );
3982     const wxDataViewItem item = GetItemByRow(current);
3983
3984     // Handle right clicking here, before everything else as context menu
3985     // events should be sent even when we click outside of any item, unlike all
3986     // the other ones.
3987     if (event.RightUp())
3988     {
3989         wxWindow *parent = GetParent();
3990         wxDataViewEvent le(wxEVT_COMMAND_DATAVIEW_ITEM_CONTEXT_MENU, parent->GetId());
3991         le.SetEventObject(parent);
3992         le.SetModel(model);
3993
3994         if ( item.IsOk() && col )
3995         {
3996             le.SetItem( item );
3997             le.SetColumn( col->GetModelColumn() );
3998             le.SetDataViewColumn( col );
3999
4000             wxVariant value;
4001             model->GetValue( value, item, col->GetModelColumn() );
4002             le.SetValue(value);
4003         }
4004
4005         parent->ProcessWindowEvent(le);
4006         return;
4007     }
4008
4009     if (!col)
4010     {
4011         event.Skip();
4012         return;
4013     }
4014
4015     wxDataViewRenderer *cell = col->GetRenderer();
4016     if ((current >= GetRowCount()) || (x > GetEndOfLastCol()))
4017     {
4018         // Unselect all if below the last row ?
4019         event.Skip();
4020         return;
4021     }
4022
4023     wxDataViewColumn* const
4024         expander = GetExpanderColumnOrFirstOne(GetOwner());
4025
4026     // Test whether the mouse is hovering over the expander (a.k.a tree "+"
4027     // button) and also determine the offset of the real cell start, skipping
4028     // the indentation and the expander itself.
4029     bool hoverOverExpander = false;
4030     int itemOffset = 0;
4031     if ((!IsList()) && (expander == col))
4032     {
4033         wxDataViewTreeNode * node = GetTreeNodeByRow(current);
4034
4035         int indent = node->GetIndentLevel();
4036         itemOffset = GetOwner()->GetIndent()*indent;
4037
4038         if ( node->HasChildren() )
4039         {
4040             // we make the rectangle we are looking in a bit bigger than the actual
4041             // visual expander so the user can hit that little thing reliably
4042             wxRect rect(itemOffset,
4043                         GetLineStart( current ) + (GetLineHeight(current) - m_lineHeight)/2,
4044                         m_lineHeight, m_lineHeight);
4045
4046             if( rect.Contains(x, y) )
4047             {
4048                 // So the mouse is over the expander
4049                 hoverOverExpander = true;
4050                 if (m_underMouse && m_underMouse != node)
4051                 {
4052                     // wxLogMessage("Undo the row: %d", GetRowByItem(m_underMouse->GetItem()));
4053                     RefreshRow(GetRowByItem(m_underMouse->GetItem()));
4054                 }
4055                 if (m_underMouse != node)
4056                 {
4057                     // wxLogMessage("Do the row: %d", current);
4058                     RefreshRow(current);
4059                 }
4060                 m_underMouse = node;
4061             }
4062         }
4063
4064         // Account for the expander as well, even if this item doesn't have it,
4065         // its parent does so it still counts for the offset.
4066         itemOffset += m_lineHeight;
4067     }
4068     if (!hoverOverExpander)
4069     {
4070         if (m_underMouse != NULL)
4071         {
4072             // wxLogMessage("Undo the row: %d", GetRowByItem(m_underMouse->GetItem()));
4073             RefreshRow(GetRowByItem(m_underMouse->GetItem()));
4074             m_underMouse = NULL;
4075         }
4076     }
4077
4078 #if wxUSE_DRAG_AND_DROP
4079     if (event.Dragging())
4080     {
4081         if (m_dragCount == 0)
4082         {
4083             // we have to report the raw, physical coords as we want to be
4084             // able to call HitTest(event.m_pointDrag) from the user code to
4085             // get the item being dragged
4086             m_dragStart = event.GetPosition();
4087         }
4088
4089         m_dragCount++;
4090
4091         if (m_dragCount != 3)
4092             return;
4093
4094         if (event.LeftIsDown())
4095         {
4096             m_owner->CalcUnscrolledPosition( m_dragStart.x, m_dragStart.y,
4097                                              &m_dragStart.x, &m_dragStart.y );
4098             unsigned int drag_item_row = GetLineAt( m_dragStart.y );
4099             wxDataViewItem itemDragged = GetItemByRow( drag_item_row );
4100
4101             // Notify cell about drag
4102             wxDataViewEvent event( wxEVT_COMMAND_DATAVIEW_ITEM_BEGIN_DRAG, m_owner->GetId() );
4103             event.SetEventObject( m_owner );
4104             event.SetItem( itemDragged );
4105             event.SetModel( model );
4106             if (!m_owner->HandleWindowEvent( event ))
4107                 return;
4108
4109             if (!event.IsAllowed())
4110                 return;
4111
4112             wxDataObject *obj = event.GetDataObject();
4113             if (!obj)
4114                 return;
4115
4116             wxDataViewDropSource drag( this, drag_item_row );
4117             drag.SetData( *obj );
4118             /* wxDragResult res = */ drag.DoDragDrop(event.GetDragFlags());
4119             delete obj;
4120         }
4121         return;
4122     }
4123     else
4124     {
4125         m_dragCount = 0;
4126     }
4127 #endif // wxUSE_DRAG_AND_DROP
4128
4129     bool simulateClick = false;
4130
4131     if (event.ButtonDClick())
4132     {
4133         m_renameTimer->Stop();
4134         m_lastOnSame = false;
4135     }
4136
4137     bool ignore_other_columns =
4138         ((expander != col) &&
4139         (model->IsContainer(item)) &&
4140         (!model->HasContainerColumns(item)));
4141
4142     if (event.LeftDClick())
4143     {
4144         if(hoverOverExpander)
4145         {
4146             // a double click on the expander will be converted into a "simulated" normal click
4147             simulateClick = true;
4148         }
4149         else if ( current == m_lineLastClicked )
4150         {
4151             wxWindow *parent = GetParent();
4152             wxDataViewEvent le(wxEVT_COMMAND_DATAVIEW_ITEM_ACTIVATED, parent->GetId());
4153             le.SetItem( item );
4154             le.SetColumn( col->GetModelColumn() );
4155             le.SetDataViewColumn( col );
4156             le.SetEventObject(parent);
4157             le.SetModel(GetModel());
4158
4159             parent->ProcessWindowEvent(le);
4160             return;
4161         }
4162         else
4163         {
4164             // The first click was on another item, so don't interpret this as
4165             // a double click, but as a simple click instead
4166             simulateClick = true;
4167         }
4168     }
4169
4170     if (event.LeftUp() && !hoverOverExpander)
4171     {
4172         if (m_lineSelectSingleOnUp != (unsigned int)-1)
4173         {
4174             // select single line
4175             SelectAllRows( false );
4176             SelectRow( m_lineSelectSingleOnUp, true );
4177             SendSelectionChangedEvent( GetItemByRow(m_lineSelectSingleOnUp) );
4178         }
4179
4180         // If the user click the expander, we do not do editing even if the column
4181         // with expander are editable
4182         if (m_lastOnSame && !ignore_other_columns)
4183         {
4184             if ((col == m_currentCol) && (current == m_currentRow) &&
4185                 IsCellEditableInMode(item, col, wxDATAVIEW_CELL_EDITABLE) )
4186             {
4187                 m_renameTimer->Start( 100, true );
4188             }
4189         }
4190
4191         m_lastOnSame = false;
4192         m_lineSelectSingleOnUp = (unsigned int)-1;
4193     }
4194     else if(!event.LeftUp())
4195     {
4196         // This is necessary, because after a DnD operation in
4197         // from and to ourself, the up event is swallowed by the
4198         // DnD code. So on next non-up event (which means here and
4199         // now) m_lineSelectSingleOnUp should be reset.
4200         m_lineSelectSingleOnUp = (unsigned int)-1;
4201     }
4202
4203     if (event.RightDown())
4204     {
4205         m_lineBeforeLastClicked = m_lineLastClicked;
4206         m_lineLastClicked = current;
4207
4208         // If the item is already selected, do not update the selection.
4209         // Multi-selections should not be cleared if a selected item is clicked.
4210         if (!IsRowSelected(current))
4211         {
4212             SelectAllRows(false);
4213             const unsigned oldCurrent = m_currentRow;
4214             ChangeCurrentRow(current);
4215             SelectRow(m_currentRow,true);
4216             RefreshRow(oldCurrent);
4217             SendSelectionChangedEvent(GetItemByRow( m_currentRow ) );
4218         }
4219     }
4220     else if (event.MiddleDown())
4221     {
4222     }
4223
4224     if((event.LeftDown() || simulateClick) && hoverOverExpander)
4225     {
4226         wxDataViewTreeNode* node = GetTreeNodeByRow(current);
4227
4228         // hoverOverExpander being true tells us that our node must be
4229         // valid and have children.
4230         // So we don't need any extra checks.
4231         if( node->IsOpen() )
4232             Collapse(current);
4233         else
4234             Expand(current);
4235     }
4236     else if ((event.LeftDown() || simulateClick) && !hoverOverExpander)
4237     {
4238         m_lineBeforeLastClicked = m_lineLastClicked;
4239         m_lineLastClicked = current;
4240
4241         unsigned int oldCurrentRow = m_currentRow;
4242         bool oldWasSelected = IsRowSelected(m_currentRow);
4243
4244         bool cmdModifierDown = event.CmdDown();
4245         if ( IsSingleSel() || !(cmdModifierDown || event.ShiftDown()) )
4246         {
4247             if ( IsSingleSel() || !IsRowSelected(current) )
4248             {
4249                 SelectAllRows( false );
4250                 ChangeCurrentRow(current);
4251                 SelectRow(m_currentRow,true);
4252                 SendSelectionChangedEvent(GetItemByRow( m_currentRow ) );
4253             }
4254             else // multi sel & current is highlighted & no mod keys
4255             {
4256                 m_lineSelectSingleOnUp = current;
4257                 ChangeCurrentRow(current); // change focus
4258             }
4259         }
4260         else // multi sel & either ctrl or shift is down
4261         {
4262             if (cmdModifierDown)
4263             {
4264                 ChangeCurrentRow(current);
4265                 ReverseRowSelection(m_currentRow);
4266                 SendSelectionChangedEvent(GetItemByRow(m_currentRow));
4267             }
4268             else if (event.ShiftDown())
4269             {
4270                 ChangeCurrentRow(current);
4271
4272                 unsigned int lineFrom = oldCurrentRow,
4273                     lineTo = current;
4274
4275                 if ( lineTo < lineFrom )
4276                 {
4277                     lineTo = lineFrom;
4278                     lineFrom = m_currentRow;
4279                 }
4280
4281                 SelectRows(lineFrom, lineTo, true);
4282                 SendSelectionChangedEvent(GetItemByRow(m_selection[0]) );
4283             }
4284             else // !ctrl, !shift
4285             {
4286                 // test in the enclosing if should make it impossible
4287                 wxFAIL_MSG( wxT("how did we get here?") );
4288             }
4289         }
4290
4291         if (m_currentRow != oldCurrentRow)
4292             RefreshRow( oldCurrentRow );
4293
4294         wxDataViewColumn *oldCurrentCol = m_currentCol;
4295
4296         // Update selection here...
4297         m_currentCol = col;
4298         m_currentColSetByKeyboard = false;
4299
4300         // This flag is used to decide whether we should start editing the item
4301         // label. We do it if the user clicks twice (but not double clicks,
4302         // i.e. simulateClick is false) on the same item but not if the click
4303         // was used for something else already, e.g. selecting the item (so it
4304         // must have been already selected) or giving the focus to the control
4305         // (so it must have had focus already).
4306         m_lastOnSame = !simulateClick && ((col == oldCurrentCol) &&
4307                         (current == oldCurrentRow)) && oldWasSelected &&
4308                         HasFocus();
4309
4310         // Call ActivateCell() after everything else as under GTK+
4311         if ( IsCellEditableInMode(item, col, wxDATAVIEW_CELL_ACTIVATABLE) )
4312         {
4313             // notify cell about click
4314             cell->PrepareForItem(model, item, col->GetModelColumn());
4315
4316             wxRect cell_rect( xpos + itemOffset,
4317                               GetLineStart( current ),
4318                               col->GetWidth() - itemOffset,
4319                               GetLineHeight( current ) );
4320
4321             // Report position relative to the cell's custom area, i.e.
4322             // no the entire space as given by the control but the one
4323             // used by the renderer after calculation of alignment etc.
4324
4325             // adjust the rectangle ourselves to account for the alignment
4326             wxRect rectItem = cell_rect;
4327             const int align = cell->GetAlignment();
4328             if ( align != wxDVR_DEFAULT_ALIGNMENT )
4329             {
4330                 const wxSize size = cell->GetSize();
4331
4332                 if ( size.x >= 0 && size.x < cell_rect.width )
4333                 {
4334                     if ( align & wxALIGN_CENTER_HORIZONTAL )
4335                         rectItem.x += (cell_rect.width - size.x)/2;
4336                     else if ( align & wxALIGN_RIGHT )
4337                         rectItem.x += cell_rect.width - size.x;
4338                     // else: wxALIGN_LEFT is the default
4339                 }
4340
4341                 if ( size.y >= 0 && size.y < cell_rect.height )
4342                 {
4343                     if ( align & wxALIGN_CENTER_VERTICAL )
4344                         rectItem.y += (cell_rect.height - size.y)/2;
4345                     else if ( align & wxALIGN_BOTTOM )
4346                         rectItem.y += cell_rect.height - size.y;
4347                     // else: wxALIGN_TOP is the default
4348                 }
4349             }
4350
4351             wxMouseEvent event2(event);
4352             event2.m_x -= rectItem.x;
4353             event2.m_y -= rectItem.y;
4354             m_owner->CalcUnscrolledPosition(event2.m_x, event2.m_y, &event2.m_x, &event2.m_y);
4355
4356              /* ignore ret */ cell->WXActivateCell
4357                                     (
4358                                         cell_rect,
4359                                         model,
4360                                         item,
4361                                         col->GetModelColumn(),
4362                                         &event2
4363                                     );
4364         }
4365     }
4366 }
4367
4368 void wxDataViewMainWindow::OnSetFocus( wxFocusEvent &event )
4369 {
4370     m_hasFocus = true;
4371
4372     if (HasCurrentRow())
4373         Refresh();
4374
4375     event.Skip();
4376 }
4377
4378 void wxDataViewMainWindow::OnKillFocus( wxFocusEvent &event )
4379 {
4380     m_hasFocus = false;
4381
4382     if (HasCurrentRow())
4383         Refresh();
4384
4385     event.Skip();
4386 }
4387
4388 void wxDataViewMainWindow::OnColumnsCountChanged()
4389 {
4390     int editableCount = 0;
4391
4392     const unsigned cols = GetOwner()->GetColumnCount();
4393     for ( unsigned i = 0; i < cols; i++ )
4394     {
4395         wxDataViewColumn *c = GetOwner()->GetColumnAt(i);
4396         if ( c->IsHidden() )
4397             continue;
4398         if ( c->GetRenderer()->GetMode() != wxDATAVIEW_CELL_INERT )
4399             editableCount++;
4400     }
4401
4402     m_useCellFocus = (editableCount > 1);
4403
4404     UpdateDisplay();
4405 }
4406
4407 //-----------------------------------------------------------------------------
4408 // wxDataViewCtrl
4409 //-----------------------------------------------------------------------------
4410
4411 WX_DEFINE_LIST(wxDataViewColumnList)
4412
4413 IMPLEMENT_DYNAMIC_CLASS(wxDataViewCtrl, wxDataViewCtrlBase)
4414 BEGIN_EVENT_TABLE(wxDataViewCtrl, wxDataViewCtrlBase)
4415     EVT_SIZE(wxDataViewCtrl::OnSize)
4416 END_EVENT_TABLE()
4417
4418 wxDataViewCtrl::~wxDataViewCtrl()
4419 {
4420     if (m_notifier)
4421         GetModel()->RemoveNotifier( m_notifier );
4422
4423     m_cols.Clear();
4424     m_colsBestWidths.clear();
4425 }
4426
4427 void wxDataViewCtrl::Init()
4428 {
4429     m_cols.DeleteContents(true);
4430     m_notifier = NULL;
4431
4432     // No sorting column at start
4433     m_sortingColumnIdx = wxNOT_FOUND;
4434
4435     m_headerArea = NULL;
4436
4437     m_colsDirty = false;
4438 }
4439
4440 bool wxDataViewCtrl::Create(wxWindow *parent,
4441                             wxWindowID id,
4442                             const wxPoint& pos,
4443                             const wxSize& size,
4444                             long style,
4445                             const wxValidator& validator,
4446                             const wxString& name)
4447 {
4448 //    if ( (style & wxBORDER_MASK) == 0)
4449 //        style |= wxBORDER_SUNKEN;
4450
4451     Init();
4452
4453     if (!wxControl::Create( parent, id, pos, size,
4454                             style | wxScrolledWindowStyle, validator, name))
4455         return false;
4456
4457     SetInitialSize(size);
4458
4459 #ifdef __WXMAC__
4460     MacSetClipChildren( true );
4461 #endif
4462
4463     m_clientArea = new wxDataViewMainWindow( this, wxID_ANY );
4464
4465     // We use the cursor keys for moving the selection, not scrolling, so call
4466     // this method to ensure wxScrollHelperEvtHandler doesn't catch all
4467     // keyboard events forwarded to us from wxListMainWindow.
4468     DisableKeyboardScrolling();
4469
4470     if (HasFlag(wxDV_NO_HEADER))
4471         m_headerArea = NULL;
4472     else
4473         m_headerArea = new wxDataViewHeaderWindow(this);
4474
4475     SetTargetWindow( m_clientArea );
4476
4477     wxBoxSizer *sizer = new wxBoxSizer( wxVERTICAL );
4478     if (m_headerArea)
4479         sizer->Add( m_headerArea, 0, wxGROW );
4480     sizer->Add( m_clientArea, 1, wxGROW );
4481     SetSizer( sizer );
4482
4483     return true;
4484 }
4485
4486 wxBorder wxDataViewCtrl::GetDefaultBorder() const
4487 {
4488     return wxBORDER_THEME;
4489 }
4490
4491 #ifdef __WXMSW__
4492 WXLRESULT wxDataViewCtrl::MSWWindowProc(WXUINT nMsg,
4493                                         WXWPARAM wParam,
4494                                         WXLPARAM lParam)
4495 {
4496     WXLRESULT rc = wxDataViewCtrlBase::MSWWindowProc(nMsg, wParam, lParam);
4497
4498 #ifndef __WXWINCE__
4499     // we need to process arrows ourselves for scrolling
4500     if ( nMsg == WM_GETDLGCODE )
4501     {
4502         rc |= DLGC_WANTARROWS;
4503     }
4504 #endif
4505
4506     return rc;
4507 }
4508 #endif
4509
4510 wxSize wxDataViewCtrl::GetSizeAvailableForScrollTarget(const wxSize& size)
4511 {
4512     wxSize newsize = size;
4513     if (!HasFlag(wxDV_NO_HEADER) && (m_headerArea))
4514     newsize.y -= m_headerArea->GetSize().y;
4515
4516     return newsize;
4517 }
4518
4519 void wxDataViewCtrl::OnSize( wxSizeEvent &WXUNUSED(event) )
4520 {
4521     // We need to override OnSize so that our scrolled
4522     // window a) does call Layout() to use sizers for
4523     // positioning the controls but b) does not query
4524     // the sizer for their size and use that for setting
4525     // the scrollable area as set that ourselves by
4526     // calling SetScrollbar() further down.
4527
4528     Layout();
4529
4530     AdjustScrollbars();
4531
4532     // We must redraw the headers if their height changed. Normally this
4533     // shouldn't happen as the control shouldn't let itself be resized beneath
4534     // its minimal height but avoid the display artefacts that appear if it
4535     // does happen, e.g. because there is really not enough vertical space.
4536     if ( !HasFlag(wxDV_NO_HEADER) && m_headerArea &&
4537             m_headerArea->GetSize().y <= m_headerArea->GetBestSize(). y )
4538     {
4539         m_headerArea->Refresh();
4540     }
4541 }
4542
4543 void wxDataViewCtrl::SetFocus()
4544 {
4545     if (m_clientArea)
4546         m_clientArea->SetFocus();
4547 }
4548
4549 bool wxDataViewCtrl::AssociateModel( wxDataViewModel *model )
4550 {
4551     if (!wxDataViewCtrlBase::AssociateModel( model ))
4552         return false;
4553
4554     if (model)
4555     {
4556         m_notifier = new wxGenericDataViewModelNotifier( m_clientArea );
4557         model->AddNotifier( m_notifier );
4558     }
4559     else if (m_notifier)
4560     {
4561         m_notifier->Cleared();
4562         m_notifier = NULL;
4563     }
4564
4565     m_clientArea->DestroyTree();
4566
4567     if (model)
4568     {
4569         m_clientArea->BuildTree(model);
4570     }
4571
4572     m_clientArea->UpdateDisplay();
4573
4574     return true;
4575 }
4576
4577 #if wxUSE_DRAG_AND_DROP
4578
4579 bool wxDataViewCtrl::EnableDragSource( const wxDataFormat &format )
4580 {
4581     return m_clientArea->EnableDragSource( format );
4582 }
4583
4584 bool wxDataViewCtrl::EnableDropTarget( const wxDataFormat &format )
4585 {
4586     return m_clientArea->EnableDropTarget( format );
4587 }
4588
4589 #endif // wxUSE_DRAG_AND_DROP
4590
4591 bool wxDataViewCtrl::AppendColumn( wxDataViewColumn *col )
4592 {
4593     if (!wxDataViewCtrlBase::AppendColumn(col))
4594         return false;
4595
4596     m_cols.Append( col );
4597     m_colsBestWidths.push_back(CachedColWidthInfo());
4598     OnColumnsCountChanged();
4599     return true;
4600 }
4601
4602 bool wxDataViewCtrl::PrependColumn( wxDataViewColumn *col )
4603 {
4604     if (!wxDataViewCtrlBase::PrependColumn(col))
4605         return false;
4606
4607     m_cols.Insert( col );
4608     m_colsBestWidths.insert(m_colsBestWidths.begin(), CachedColWidthInfo());
4609     OnColumnsCountChanged();
4610     return true;
4611 }
4612
4613 bool wxDataViewCtrl::InsertColumn( unsigned int pos, wxDataViewColumn *col )
4614 {
4615     if (!wxDataViewCtrlBase::InsertColumn(pos,col))
4616         return false;
4617
4618     m_cols.Insert( pos, col );
4619     m_colsBestWidths.insert(m_colsBestWidths.begin() + pos, CachedColWidthInfo());
4620     OnColumnsCountChanged();
4621     return true;
4622 }
4623
4624 void wxDataViewCtrl::OnColumnChange(unsigned int idx)
4625 {
4626     if ( m_headerArea )
4627         m_headerArea->UpdateColumn(idx);
4628
4629     m_clientArea->UpdateDisplay();
4630 }
4631
4632 void wxDataViewCtrl::OnColumnsCountChanged()
4633 {
4634     if (m_headerArea)
4635         m_headerArea->SetColumnCount(GetColumnCount());
4636
4637     m_clientArea->OnColumnsCountChanged();
4638 }
4639
4640 void wxDataViewCtrl::DoSetExpanderColumn()
4641 {
4642     m_clientArea->UpdateDisplay();
4643 }
4644
4645 void wxDataViewCtrl::DoSetIndent()
4646 {
4647     m_clientArea->UpdateDisplay();
4648 }
4649
4650 unsigned int wxDataViewCtrl::GetColumnCount() const
4651 {
4652     return m_cols.GetCount();
4653 }
4654
4655 bool wxDataViewCtrl::SetRowHeight( int lineHeight )
4656 {
4657     if ( !m_clientArea )
4658         return false;
4659
4660     m_clientArea->SetRowHeight(lineHeight);
4661
4662     return true;
4663 }
4664
4665 wxDataViewColumn* wxDataViewCtrl::GetColumn( unsigned int idx ) const
4666 {
4667     return m_cols[idx];
4668 }
4669
4670 wxDataViewColumn *wxDataViewCtrl::GetColumnAt(unsigned int pos) const
4671 {
4672     // columns can't be reordered if there is no header window which allows
4673     // to do this
4674     const unsigned idx = m_headerArea ? m_headerArea->GetColumnsOrder()[pos]
4675                                     : pos;
4676
4677     return GetColumn(idx);
4678 }
4679
4680 int wxDataViewCtrl::GetColumnIndex(const wxDataViewColumn *column) const
4681 {
4682     const unsigned count = m_cols.size();
4683     for ( unsigned n = 0; n < count; n++ )
4684     {
4685         if ( m_cols[n] == column )
4686             return n;
4687     }
4688
4689     return wxNOT_FOUND;
4690 }
4691
4692 unsigned int wxDataViewCtrl::GetBestColumnWidth(int idx) const
4693 {
4694     if ( m_colsBestWidths[idx].width != 0 )
4695         return m_colsBestWidths[idx].width;
4696
4697     const int count = m_clientArea->GetRowCount();
4698     wxDataViewColumn *column = GetColumn(idx);
4699     wxDataViewRenderer *renderer =
4700         const_cast<wxDataViewRenderer*>(column->GetRenderer());
4701
4702     class MaxWidthCalculator
4703     {
4704     public:
4705         MaxWidthCalculator(const wxDataViewCtrl *dvc,
4706                            wxDataViewMainWindow *clientArea,
4707                            wxDataViewRenderer *renderer,
4708                            const wxDataViewModel *model,
4709                            unsigned column,
4710                            int expanderSize)
4711             : m_width(0),
4712               m_dvc(dvc),
4713               m_clientArea(clientArea),
4714               m_renderer(renderer),
4715               m_model(model),
4716               m_column(column),
4717               m_expanderSize(expanderSize)
4718
4719         {
4720             m_isExpanderCol =
4721                 !clientArea->IsList() &&
4722                 (column == 0 ||
4723                  GetExpanderColumnOrFirstOne(const_cast<wxDataViewCtrl*>(dvc)) == dvc->GetColumnAt(column));
4724         }
4725
4726         void UpdateWithWidth(int width)
4727         {
4728             m_width = wxMax(m_width, width);
4729         }
4730
4731         void UpdateWithRow(int row)
4732         {
4733             int indent = 0;
4734             wxDataViewItem item;
4735
4736             if ( m_isExpanderCol )
4737             {
4738                 wxDataViewTreeNode *node = m_clientArea->GetTreeNodeByRow(row);
4739                 item = node->GetItem();
4740                 indent = m_dvc->GetIndent() * node->GetIndentLevel() + m_expanderSize;
4741             }
4742             else
4743             {
4744                 item = m_clientArea->GetItemByRow(row);
4745             }
4746
4747             m_renderer->PrepareForItem(m_model, item, m_column);
4748             m_width = wxMax(m_width, m_renderer->GetSize().x + indent);
4749         }
4750
4751         int GetMaxWidth() const { return m_width; }
4752
4753     private:
4754         int m_width;
4755         const wxDataViewCtrl *m_dvc;
4756         wxDataViewMainWindow *m_clientArea;
4757         wxDataViewRenderer *m_renderer;
4758         const wxDataViewModel *m_model;
4759         unsigned m_column;
4760         bool m_isExpanderCol;
4761         int m_expanderSize;
4762     };
4763
4764     MaxWidthCalculator calculator(this, m_clientArea, renderer,
4765                                   GetModel(), column->GetModelColumn(),
4766                                   m_clientArea->GetRowHeight());
4767
4768     calculator.UpdateWithWidth(column->GetMinWidth());
4769
4770     if ( m_headerArea )
4771         calculator.UpdateWithWidth(m_headerArea->GetColumnTitleWidth(*column));
4772
4773     // The code below deserves some explanation. For very large controls, we
4774     // simply can't afford to calculate sizes for all items, it takes too
4775     // long. So the best we can do is to check the first and the last N/2
4776     // items in the control for some sufficiently large N and calculate best
4777     // sizes from that. That can result in the calculated best width being too
4778     // small for some outliers, but it's better to get slightly imperfect
4779     // result than to wait several seconds after every update. To avoid highly
4780     // visible miscalculations, we also include all currently visible items
4781     // no matter what.  Finally, the value of N is determined dynamically by
4782     // measuring how much time we spent on the determining item widths so far.
4783
4784 #if wxUSE_STOPWATCH
4785     int top_part_end = count;
4786     static const long CALC_TIMEOUT = 20/*ms*/;
4787     // don't call wxStopWatch::Time() too often
4788     static const unsigned CALC_CHECK_FREQ = 100;
4789     wxStopWatch timer;
4790 #else
4791     // use some hard-coded limit, that's the best we can do without timer
4792     int top_part_end = wxMin(500, count);
4793 #endif // wxUSE_STOPWATCH/!wxUSE_STOPWATCH
4794
4795     int row = 0;
4796
4797     for ( row = 0; row < top_part_end; row++ )
4798     {
4799 #if wxUSE_STOPWATCH
4800         if ( row % CALC_CHECK_FREQ == CALC_CHECK_FREQ-1 &&
4801              timer.Time() > CALC_TIMEOUT )
4802             break;
4803 #endif // wxUSE_STOPWATCH
4804         calculator.UpdateWithRow(row);
4805     }
4806
4807     // row is the first unmeasured item now; that's our value of N/2
4808
4809     if ( row < count )
4810     {
4811         top_part_end = row;
4812
4813         // add bottom N/2 items now:
4814         const int bottom_part_start = wxMax(row, count - row);
4815         for ( row = bottom_part_start; row < count; row++ )
4816         {
4817             calculator.UpdateWithRow(row);
4818         }
4819
4820         // finally, include currently visible items in the calculation:
4821         const wxPoint origin = CalcUnscrolledPosition(wxPoint(0, 0));
4822         int first_visible = m_clientArea->GetLineAt(origin.y);
4823         int last_visible = m_clientArea->GetLineAt(origin.y + GetClientSize().y);
4824
4825         first_visible = wxMax(first_visible, top_part_end);
4826         last_visible = wxMin(bottom_part_start, last_visible);
4827
4828         for ( row = first_visible; row < last_visible; row++ )
4829         {
4830             calculator.UpdateWithRow(row);
4831         }
4832
4833         wxLogTrace("dataview",
4834                    "determined best size from %d top, %d bottom plus %d more visible items out of %d total",
4835                    top_part_end,
4836                    count - bottom_part_start,
4837                    wxMax(0, last_visible - first_visible),
4838                    count);
4839     }
4840
4841     int max_width = calculator.GetMaxWidth();
4842     if ( max_width > 0 )
4843         max_width += 2 * PADDING_RIGHTLEFT;
4844
4845     const_cast<wxDataViewCtrl*>(this)->m_colsBestWidths[idx].width = max_width;
4846     return max_width;
4847 }
4848
4849 void wxDataViewCtrl::ColumnMoved(wxDataViewColumn * WXUNUSED(col),
4850                                 unsigned int WXUNUSED(new_pos))
4851 {
4852     // do _not_ reorder m_cols elements here, they should always be in the
4853     // order in which columns were added, we only display the columns in
4854     // different order
4855     m_clientArea->UpdateDisplay();
4856 }
4857
4858 bool wxDataViewCtrl::DeleteColumn( wxDataViewColumn *column )
4859 {
4860     wxDataViewColumnList::compatibility_iterator ret = m_cols.Find( column );
4861     if (!ret)
4862         return false;
4863
4864     m_colsBestWidths.erase(m_colsBestWidths.begin() + GetColumnIndex(column));
4865     m_cols.Erase(ret);
4866
4867     if ( m_clientArea->GetCurrentColumn() == column )
4868         m_clientArea->ClearCurrentColumn();
4869
4870     OnColumnsCountChanged();
4871
4872     return true;
4873 }
4874
4875 bool wxDataViewCtrl::ClearColumns()
4876 {
4877     SetExpanderColumn(NULL);
4878     m_cols.Clear();
4879     m_colsBestWidths.clear();
4880
4881     m_clientArea->ClearCurrentColumn();
4882
4883     OnColumnsCountChanged();
4884
4885     return true;
4886 }
4887
4888 void wxDataViewCtrl::InvalidateColBestWidth(int idx)
4889 {
4890     m_colsBestWidths[idx].width = 0;
4891     m_colsBestWidths[idx].dirty = true;
4892     m_colsDirty = true;
4893 }
4894
4895 void wxDataViewCtrl::InvalidateColBestWidths()
4896 {
4897     // mark all columns as dirty:
4898     m_colsBestWidths.clear();
4899     m_colsBestWidths.resize(m_cols.size());
4900     m_colsDirty = true;
4901 }
4902
4903 void wxDataViewCtrl::UpdateColWidths()
4904 {
4905     m_colsDirty = false;
4906
4907     if ( !m_headerArea )
4908         return;
4909
4910     const unsigned len = m_colsBestWidths.size();
4911     for ( unsigned i = 0; i < len; i++ )
4912     {
4913         // Note that we have to have an explicit 'dirty' flag here instead of
4914         // checking if the width==0, as is done in GetBestColumnWidth().
4915         //
4916         // Testing width==0 wouldn't work correctly if some code called
4917         // GetWidth() after col. width invalidation but before
4918         // wxDataViewCtrl::UpdateColWidths() was called at idle time. This
4919         // would result in the header's column width getting out of sync with
4920         // the control itself.
4921         if ( m_colsBestWidths[i].dirty )
4922         {
4923             m_headerArea->UpdateColumn(i);
4924             m_colsBestWidths[i].dirty = false;
4925         }
4926     }
4927 }
4928
4929 void wxDataViewCtrl::OnInternalIdle()
4930 {
4931     wxDataViewCtrlBase::OnInternalIdle();
4932
4933     if ( m_colsDirty )
4934         UpdateColWidths();
4935 }
4936
4937 int wxDataViewCtrl::GetColumnPosition( const wxDataViewColumn *column ) const
4938 {
4939     unsigned int len = GetColumnCount();
4940     for ( unsigned int i = 0; i < len; i++ )
4941     {
4942         wxDataViewColumn * col = GetColumnAt(i);
4943         if (column==col)
4944             return i;
4945     }
4946
4947     return wxNOT_FOUND;
4948 }
4949
4950 wxDataViewColumn *wxDataViewCtrl::GetSortingColumn() const
4951 {
4952     return m_sortingColumnIdx == wxNOT_FOUND ? NULL
4953                                             : GetColumn(m_sortingColumnIdx);
4954 }
4955
4956 wxDataViewItem wxDataViewCtrl::DoGetCurrentItem() const
4957 {
4958     return GetItemByRow(m_clientArea->GetCurrentRow());
4959 }
4960
4961 void wxDataViewCtrl::DoSetCurrentItem(const wxDataViewItem& item)
4962 {
4963     const int row = m_clientArea->GetRowByItem(item);
4964
4965     const unsigned oldCurrent = m_clientArea->GetCurrentRow();
4966     if ( static_cast<unsigned>(row) != oldCurrent )
4967     {
4968         m_clientArea->ChangeCurrentRow(row);
4969         m_clientArea->RefreshRow(oldCurrent);
4970         m_clientArea->RefreshRow(row);
4971     }
4972 }
4973
4974 wxDataViewColumn *wxDataViewCtrl::GetCurrentColumn() const
4975 {
4976     return m_clientArea->GetCurrentColumn();
4977 }
4978
4979 int wxDataViewCtrl::GetSelectedItemsCount() const
4980 {
4981     return m_clientArea->GetSelections().size();
4982 }
4983
4984 int wxDataViewCtrl::GetSelections( wxDataViewItemArray & sel ) const
4985 {
4986     sel.Empty();
4987     const wxDataViewSelection& selections = m_clientArea->GetSelections();
4988
4989     const size_t len = selections.size();
4990     for ( size_t i = 0; i < len; i++ )
4991     {
4992         wxDataViewItem item = m_clientArea->GetItemByRow(selections[i]);
4993         if ( item.IsOk() )
4994         {
4995             sel.Add(item);
4996         }
4997         else
4998         {
4999             wxFAIL_MSG( "invalid item in selection - bad internal state" );
5000         }
5001     }
5002
5003     return sel.size();
5004 }
5005
5006 void wxDataViewCtrl::SetSelections( const wxDataViewItemArray & sel )
5007 {
5008     wxDataViewSelection selection(wxDataViewSelectionCmp);
5009
5010     wxDataViewItem last_parent;
5011
5012     int len = sel.GetCount();
5013     for( int i = 0; i < len; i ++ )
5014     {
5015         wxDataViewItem item = sel[i];
5016         wxDataViewItem parent = GetModel()->GetParent( item );
5017         if (parent)
5018         {
5019             if (parent != last_parent)
5020                 ExpandAncestors(item);
5021         }
5022
5023         last_parent = parent;
5024         int row = m_clientArea->GetRowByItem( item );
5025         if( row >= 0 )
5026             selection.Add( static_cast<unsigned int>(row) );
5027     }
5028
5029     m_clientArea->SetSelections( selection );
5030 }
5031
5032 void wxDataViewCtrl::Select( const wxDataViewItem & item )
5033 {
5034     ExpandAncestors( item );
5035
5036     int row = m_clientArea->GetRowByItem( item );
5037     if( row >= 0 )
5038     {
5039         // Unselect all rows before select another in the single select mode
5040         if (m_clientArea->IsSingleSel())
5041             m_clientArea->SelectAllRows(false);
5042
5043         m_clientArea->SelectRow(row, true);
5044
5045         // Also set focus to the selected item
5046         m_clientArea->ChangeCurrentRow( row );
5047     }
5048 }
5049
5050 void wxDataViewCtrl::Unselect( const wxDataViewItem & item )
5051 {
5052     int row = m_clientArea->GetRowByItem( item );
5053     if( row >= 0 )
5054         m_clientArea->SelectRow(row, false);
5055 }
5056
5057 bool wxDataViewCtrl::IsSelected( const wxDataViewItem & item ) const
5058 {
5059     int row = m_clientArea->GetRowByItem( item );
5060     if( row >= 0 )
5061     {
5062         return m_clientArea->IsRowSelected(row);
5063     }
5064     return false;
5065 }
5066
5067 void wxDataViewCtrl::SetAlternateRowColour(const wxColour& colour)
5068 {
5069     m_alternateRowColour = colour;
5070 }
5071
5072 void wxDataViewCtrl::SelectAll()
5073 {
5074     m_clientArea->SelectAllRows(true);
5075 }
5076
5077 void wxDataViewCtrl::UnselectAll()
5078 {
5079     m_clientArea->SelectAllRows(false);
5080 }
5081
5082 void wxDataViewCtrl::EnsureVisible( int row, int column )
5083 {
5084     if( row < 0 )
5085         row = 0;
5086     if( row > (int) m_clientArea->GetRowCount() )
5087         row = m_clientArea->GetRowCount();
5088
5089     int first = m_clientArea->GetFirstVisibleRow();
5090     int last = m_clientArea->GetLastVisibleRow();
5091     if( row < first )
5092         m_clientArea->ScrollTo( row, column );
5093     else if( row > last )
5094         m_clientArea->ScrollTo( row - last + first, column );
5095     else
5096         m_clientArea->ScrollTo( first, column );
5097 }
5098
5099 void wxDataViewCtrl::EnsureVisible( const wxDataViewItem & item, const wxDataViewColumn * column )
5100 {
5101     ExpandAncestors( item );
5102
5103     m_clientArea->RecalculateDisplay();
5104
5105     int row = m_clientArea->GetRowByItem(item);
5106     if( row >= 0 )
5107     {
5108         if( column == NULL )
5109             EnsureVisible(row, -1);
5110         else
5111             EnsureVisible( row, GetColumnIndex(column) );
5112     }
5113
5114 }
5115
5116 void wxDataViewCtrl::HitTest( const wxPoint & point, wxDataViewItem & item,
5117                               wxDataViewColumn* &column ) const
5118 {
5119     m_clientArea->HitTest(point, item, column);
5120 }
5121
5122 wxRect wxDataViewCtrl::GetItemRect( const wxDataViewItem & item,
5123                                     const wxDataViewColumn* column ) const
5124 {
5125     return m_clientArea->GetItemRect(item, column);
5126 }
5127
5128 wxDataViewItem wxDataViewCtrl::GetItemByRow( unsigned int row ) const
5129 {
5130     return m_clientArea->GetItemByRow( row );
5131 }
5132
5133 int wxDataViewCtrl::GetRowByItem( const wxDataViewItem & item ) const
5134 {
5135     return m_clientArea->GetRowByItem( item );
5136 }
5137
5138 void wxDataViewCtrl::Expand( const wxDataViewItem & item )
5139 {
5140     ExpandAncestors( item );
5141
5142     int row = m_clientArea->GetRowByItem( item );
5143     if (row != -1)
5144     {
5145         m_clientArea->Expand(row);
5146         InvalidateColBestWidths();
5147     }
5148 }
5149
5150 void wxDataViewCtrl::Collapse( const wxDataViewItem & item )
5151 {
5152     int row = m_clientArea->GetRowByItem( item );
5153     if (row != -1)
5154     {
5155         m_clientArea->Collapse(row);
5156         InvalidateColBestWidths();
5157     }
5158 }
5159
5160 bool wxDataViewCtrl::IsExpanded( const wxDataViewItem & item ) const
5161 {
5162     int row = m_clientArea->GetRowByItem( item );
5163     if (row != -1)
5164         return m_clientArea->IsExpanded(row);
5165     return false;
5166 }
5167
5168 void wxDataViewCtrl::EditItem(const wxDataViewItem& item, const wxDataViewColumn *column)
5169 {
5170     wxCHECK_RET( item.IsOk(), "invalid item" );
5171     wxCHECK_RET( column, "no column provided" );
5172
5173     m_clientArea->StartEditing(item, column);
5174 }
5175
5176 #endif // !wxUSE_GENERICDATAVIEWCTRL
5177
5178 #endif // wxUSE_DATAVIEWCTRL