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 #endif
  38
  39 #include "wx/stockitem.h"
  40 #include "wx/calctrl.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
  48 //-----------------------------------------------------------------------------
  49 // classes
  50 //-----------------------------------------------------------------------------
  51
  52 class wxDataViewCtrl;
  53
  54 static const int SCROLL_UNIT_X = 15;
  55
  56 // the cell padding on the left/right
  57 static const int PADDING_RIGHTLEFT = 3;
  58
  59 // the expander space margin
  60 static const int EXPANDER_MARGIN = 4;
  61
  62 //-----------------------------------------------------------------------------
  63 // wxDataViewHeaderWindow
  64 //-----------------------------------------------------------------------------
  65
  66 // on wxMSW the header window (only that part however) can be made native!
  67 #if defined(__WXMSW__) && !defined(__WXUNIVERSAL__)
  68     #define USE_NATIVE_HEADER_WINDOW
  69 #endif
  70
  71 //Below is the compare stuff
  72 //For the generic implements, both the leaf nodes and the nodes are sorted for fast search when needed
  73 static wxDataViewModel * g_model;
  74 static int g_column = -2;
  75 static bool g_asending = true;
  76
  77 // NB: for some reason, this class must be dllexport'ed or we get warnings from
  78 //     MSVC in DLL build
  79 class WXDLLIMPEXP_ADV wxDataViewHeaderWindowBase : public wxControl
  80 {
  81 public:
  82     wxDataViewHeaderWindowBase()
  83         { m_owner = NULL; }
  84
  85     bool Create(wxDataViewCtrl *parent, wxWindowID id,
  86                 const wxPoint &pos, const wxSize &size,
  87                 const wxString &name)
  88     {
  89         return wxWindow::Create(parent, id, pos, size, wxNO_BORDER, name);
  90     }
  91
  92     void SetOwner( wxDataViewCtrl* owner ) { m_owner = owner; }
  93     wxDataViewCtrl *GetOwner() { return m_owner; }
  94
  95     // called on column addition/removal
  96     virtual void UpdateDisplay() { /* by default, do nothing */ }
  97
  98     // returns the n-th column
  99     virtual wxDataViewColumn *GetColumn(unsigned int n)
 100     {
 101         wxASSERT(m_owner);
 102         wxDataViewColumn *ret = m_owner->GetColumn(n);
 103         wxASSERT(ret);
 104
 105         return ret;
 106     }
 107
 108 protected:
 109     wxDataViewCtrl      *m_owner;
 110
 111     // sends an event generated from the n-th wxDataViewColumn
 112     void SendEvent(wxEventType type, unsigned int n);
 113 };
 114
 115 #ifdef USE_NATIVE_HEADER_WINDOW
 116
 117 #define COLUMN_WIDTH_OFFSET         2
 118 #define wxDataViewHeaderWindowMSW   wxDataViewHeaderWindow
 119
 120 class wxDataViewHeaderWindowMSW : public wxDataViewHeaderWindowBase
 121 {
 122 public:
 123
 124     wxDataViewHeaderWindowMSW( wxDataViewCtrl *parent,
 125                                wxWindowID id,
 126                                const wxPoint &pos = wxDefaultPosition,
 127                                const wxSize &size = wxDefaultSize,
 128                                const wxString &name = wxT("wxdataviewctrlheaderwindow") )
 129     {
 130         Create(parent, id, pos, size, name);
 131     }
 132
 133     bool Create(wxDataViewCtrl *parent, wxWindowID id,
 134                 const wxPoint &pos, const wxSize &size,
 135                 const wxString &name);
 136
 137     ~wxDataViewHeaderWindowMSW();
 138
 139     // called when any column setting is changed and/or changed
 140     // the column count
 141     virtual void UpdateDisplay();
 142
 143     virtual void OnInternalIdle();
 144
 145     // called Refresh afterwards
 146     virtual void ScrollWindow(int dx, int dy, const wxRect *rect = NULL);
 147
 148     virtual wxBorder GetDefaultBorder() const { return wxBORDER_NONE; }
 149
 150 protected:
 151     virtual bool MSWOnNotify(int idCtrl, WXLPARAM lParam, WXLPARAM *result);
 152
 153     virtual void DoSetSize(int x, int y, int width, int height, int sizeFlags);
 154
 155     wxSize DoGetBestSize() const;
 156
 157     unsigned int GetColumnIdxFromHeader(NMHEADER *nmHDR);
 158
 159     wxDataViewColumn *GetColumnFromHeader(NMHEADER *nmHDR)
 160         { return GetColumn(GetColumnIdxFromHeader(nmHDR)); }
 161
 162     int   m_scrollOffsetX;
 163     int   m_buttonHeight;
 164     bool  m_delayedUpdate;
 165
 166 private:
 167     DECLARE_DYNAMIC_CLASS(wxDataViewHeaderWindowMSW)
 168 };
 169
 170 #else       // !defined(__WXMSW__)
 171
 172 #define HEADER_WINDOW_HEIGHT            25
 173 #define HEADER_HORIZ_BORDER             5
 174 #define HEADER_VERT_BORDER              3
 175 #define wxGenericDataViewHeaderWindow   wxDataViewHeaderWindow
 176
 177 class wxGenericDataViewHeaderWindow : public wxDataViewHeaderWindowBase
 178 {
 179 public:
 180     wxGenericDataViewHeaderWindow( wxDataViewCtrl *parent,
 181                                     wxWindowID id,
 182                                     const wxPoint &pos = wxDefaultPosition,
 183                                     const wxSize &size = wxDefaultSize,
 184                                     const wxString &name = wxT("wxdataviewctrlheaderwindow") )
 185     {
 186         Init();
 187         Create(parent, id, pos, size, name);
 188     }
 189
 190     bool Create(wxDataViewCtrl *parent, wxWindowID id,
 191                 const wxPoint &pos, const wxSize &size,
 192                 const wxString &name);
 193
 194     ~wxGenericDataViewHeaderWindow()
 195     {
 196         delete m_resizeCursor;
 197     }
 198
 199     virtual void UpdateDisplay() { Refresh(); }
 200
 201     // event handlers:
 202
 203     void OnPaint( wxPaintEvent &event );
 204     void OnMouse( wxMouseEvent &event );
 205     void OnSetFocus( wxFocusEvent &event );
 206
 207
 208 protected:
 209
 210     // vars used for column resizing:
 211
 212     wxCursor            *m_resizeCursor;
 213     const wxCursor      *m_currentCursor;
 214     bool                 m_isDragging;
 215
 216     bool                 m_dirty;     // needs refresh?
 217     int                  m_hover;     // index of the column under the mouse
 218     int                  m_column;    // index of the column being resized
 219     int                  m_currentX;  // divider line position in logical (unscrolled) coords
 220     int                  m_minX;      // minimal position beyond which the divider line
 221                                       // can't be dragged in logical coords
 222
 223     // the pen used to draw the current column width drag line
 224     // when resizing the columsn
 225     wxPen m_penCurrent;
 226
 227
 228     // internal utilities:
 229
 230     void Init()
 231     {
 232         m_currentCursor = (wxCursor *) NULL;
 233         m_resizeCursor = new wxCursor( wxCURSOR_SIZEWE );
 234
 235         m_isDragging = false;
 236         m_dirty = false;
 237
 238         m_hover = wxNOT_FOUND;
 239         m_column = wxNOT_FOUND;
 240         m_currentX = 0;
 241         m_minX = 0;
 242
 243         wxColour col = wxSystemSettings::GetColour(wxSYS_COLOUR_3DLIGHT);
 244         m_penCurrent = wxPen(col, 1, wxSOLID);
 245     }
 246
 247     void AdjustDC(wxDC& dc);
 248
 249 private:
 250     DECLARE_DYNAMIC_CLASS(wxGenericDataViewHeaderWindow)
 251     DECLARE_EVENT_TABLE()
 252 };
 253
 254 #endif      // defined(__WXMSW__)
 255
 256 //-----------------------------------------------------------------------------
 257 // wxDataViewRenameTimer
 258 //-----------------------------------------------------------------------------
 259
 260 class wxDataViewRenameTimer: public wxTimer
 261 {
 262 private:
 263     wxDataViewMainWindow *m_owner;
 264
 265 public:
 266     wxDataViewRenameTimer( wxDataViewMainWindow *owner );
 267     void Notify();
 268 };
 269
 270 //-----------------------------------------------------------------------------
 271 // wxDataViewTreeNode
 272 //-----------------------------------------------------------------------------
 273 class wxDataViewTreeNode;
 274 WX_DEFINE_ARRAY( wxDataViewTreeNode *, wxDataViewTreeNodes );
 275 WX_DEFINE_ARRAY( void* , wxDataViewTreeLeaves);
 276
 277 int LINKAGEMODE wxGenericTreeModelNodeCmp( wxDataViewTreeNode ** node1, wxDataViewTreeNode ** node2);
 278 int LINKAGEMODE wxGenericTreeModelItemCmp( void ** id1, void ** id2);
 279
 280 class wxDataViewTreeNode
 281 {
 282 public:
 283     wxDataViewTreeNode( wxDataViewTreeNode * parent = NULL )
 284     {
 285         m_parent = parent;
 286         if (!parent)
 287             m_open = true;
 288         else
 289             m_open = false;
 290         m_hasChildren = false;
 291         m_subTreeCount  = 0;
 292     }
 293
 294     ~wxDataViewTreeNode()
 295     {
 296     }
 297
 298     wxDataViewTreeNode * GetParent() { return m_parent; }
 299     void SetParent( wxDataViewTreeNode * parent ) { m_parent = parent; }
 300     wxDataViewTreeNodes &  GetNodes() { return m_nodes; }
 301     wxDataViewTreeLeaves & GetChildren() { return m_leaves; }
 302
 303     void AddNode( wxDataViewTreeNode * node )
 304     {
 305         m_leaves.Add( node->GetItem().GetID() );
 306         if (g_column >= -1)
 307             m_leaves.Sort( &wxGenericTreeModelItemCmp );
 308         m_nodes.Add( node );
 309         if (g_column >= -1)
 310             m_nodes.Sort( &wxGenericTreeModelNodeCmp );
 311     }
 312     void AddLeaf( void * leaf )
 313     {
 314         m_leaves.Add( leaf );
 315         if (g_column >= -1)
 316             m_leaves.Sort( &wxGenericTreeModelItemCmp );
 317     }
 318
 319     wxDataViewItem & GetItem() { return m_item; }
 320     void SetItem( const wxDataViewItem & item ) { m_item = item; }
 321
 322     unsigned int GetChildrenNumber() { return m_leaves.GetCount(); }
 323     unsigned int GetNodeNumber() { return m_nodes.GetCount(); }
 324     int GetIndentLevel()
 325     {
 326         int ret = 0 ;
 327         wxDataViewTreeNode * node = this;
 328         while( node->GetParent()->GetParent() != NULL )
 329         {
 330             node = node->GetParent();
 331             ret ++;
 332         }
 333         return ret;
 334     }
 335
 336     bool IsOpen()
 337     {
 338         return m_open ;
 339     }
 340
 341     void ToggleOpen()
 342     {
 343         int len = m_nodes.GetCount();
 344         int sum = 0;
 345         for ( int i = 0 ;i < len ; i ++)
 346             sum += m_nodes[i]->GetSubTreeCount();
 347
 348         sum += m_leaves.GetCount();
 349         if (m_open)
 350         {
 351             ChangeSubTreeCount(-sum);
 352             m_open = !m_open;
 353         }
 354         else
 355         {
 356             m_open = !m_open;
 357             ChangeSubTreeCount(sum);
 358         }
 359     }
 360     bool HasChildren() { return m_hasChildren; }
 361     void SetHasChildren( bool has ){ m_hasChildren = has; }
 362
 363     void SetSubTreeCount( int num ) { m_subTreeCount = num; }
 364     int GetSubTreeCount() { return m_subTreeCount; }
 365     void ChangeSubTreeCount( int num )
 366     {
 367         if( !m_open )
 368             return ;
 369         m_subTreeCount += num;
 370         if( m_parent )
 371             m_parent->ChangeSubTreeCount(num);
 372     }
 373
 374     void Resort()
 375     {
 376         if (g_column >= -1)
 377         {
 378             m_nodes.Sort( &wxGenericTreeModelNodeCmp );
 379             int len = m_nodes.GetCount();
 380             for (int i = 0; i < len; i ++)
 381                 m_nodes[i]->Resort();
 382             m_leaves.Sort( &wxGenericTreeModelItemCmp );
 383         }
 384     }
 385
 386 private:
 387     wxDataViewTreeNode  *m_parent;
 388     wxDataViewTreeNodes  m_nodes;
 389     wxDataViewTreeLeaves m_leaves;
 390     wxDataViewItem       m_item;
 391     bool                 m_open;
 392     bool                 m_hasChildren;
 393     int                  m_subTreeCount;
 394 };
 395
 396 int LINKAGEMODE wxGenericTreeModelNodeCmp( wxDataViewTreeNode ** node1, wxDataViewTreeNode ** node2)
 397 {
 398     return g_model->Compare( (*node1)->GetItem(), (*node2)->GetItem(), g_column, g_asending );
 399 }
 400
 401 int LINKAGEMODE wxGenericTreeModelItemCmp( void ** id1, void ** id2)
 402 {
 403     return g_model->Compare( *id1, *id2, g_column, g_asending );
 404 }
 405
 406
 407
 408 //-----------------------------------------------------------------------------
 409 // wxDataViewMainWindow
 410 //-----------------------------------------------------------------------------
 411
 412 WX_DEFINE_SORTED_USER_EXPORTED_ARRAY_SIZE_T(unsigned int, wxDataViewSelection,
 413                                             WXDLLIMPEXP_ADV);
 414 WX_DECLARE_LIST(wxDataViewItem, ItemList);
 415 WX_DEFINE_LIST(ItemList)
 416
 417 class wxDataViewMainWindow: public wxWindow
 418 {
 419 public:
 420     wxDataViewMainWindow( wxDataViewCtrl *parent,
 421                             wxWindowID id,
 422                             const wxPoint &pos = wxDefaultPosition,
 423                             const wxSize &size = wxDefaultSize,
 424                             const wxString &name = wxT("wxdataviewctrlmainwindow") );
 425     virtual ~wxDataViewMainWindow();
 426
 427     // notifications from wxDataViewModel
 428     bool ItemAdded( const wxDataViewItem &parent, const wxDataViewItem &item );
 429     bool ItemDeleted( const wxDataViewItem &parent, const wxDataViewItem &item );
 430     bool ItemChanged( const wxDataViewItem &item );
 431     bool ValueChanged( const wxDataViewItem &item, unsigned int col );
 432     bool Cleared();
 433     void Resort()
 434     {
 435         if (m_root)
 436         {
 437             SortPrepare();
 438             m_root->Resort();
 439         }
 440         UpdateDisplay();
 441     }
 442
 443     void SortPrepare()
 444     {
 445         g_model = GetOwner()->GetModel();
 446         wxDataViewColumn* col = GetOwner()->GetSortingColumn();
 447         if( !col )
 448         {
 449             if (g_model->HasDefaultCompare())
 450                 g_column = -1;
 451             else
 452                 g_column = -2;
 453
 454             g_asending = true;
 455             return;
 456         }
 457         g_column = col->GetModelColumn();
 458         g_asending = col->IsSortOrderAscending();
 459     }
 460
 461     void SetOwner( wxDataViewCtrl* owner ) { m_owner = owner; }
 462     wxDataViewCtrl *GetOwner() { return m_owner; }
 463     const wxDataViewCtrl *GetOwner() const { return m_owner; }
 464
 465     void OnPaint( wxPaintEvent &event );
 466     void OnArrowChar(unsigned int newCurrent, const wxKeyEvent& event);
 467     void OnChar( wxKeyEvent &event );
 468     void OnMouse( wxMouseEvent &event );
 469     void OnSetFocus( wxFocusEvent &event );
 470     void OnKillFocus( wxFocusEvent &event );
 471
 472     void UpdateDisplay();
 473     void RecalculateDisplay();
 474     void OnInternalIdle();
 475
 476     void OnRenameTimer();
 477
 478     void ScrollWindow( int dx, int dy, const wxRect *rect = NULL );
 479     void ScrollTo( int rows, int column );
 480
 481     bool HasCurrentRow() { return m_currentRow != (unsigned int)-1; }
 482     void ChangeCurrentRow( unsigned int row );
 483
 484     bool IsSingleSel() const { return !GetParent()->HasFlag(wxDV_MULTIPLE); }
 485     bool IsEmpty() { return GetRowCount() == 0; }
 486
 487     int GetCountPerPage() const;
 488     int GetEndOfLastCol() const;
 489     unsigned int GetFirstVisibleRow() const;
 490     //I change this method to un const because in the tree view, the displaying number of the tree are changing along with the expanding/collapsing of the tree nodes
 491     unsigned int GetLastVisibleRow();
 492     unsigned int GetRowCount() ;
 493
 494     wxDataViewItem GetSelection() const;
 495     wxDataViewSelection GetSelections(){ return m_selection; }
 496     void SetSelections( const wxDataViewSelection & sel ) { m_selection = sel; UpdateDisplay(); }
 497     void Select( const wxArrayInt& aSelections );
 498     void SelectAllRows( bool on );
 499     void SelectRow( unsigned int row, bool on );
 500     void SelectRows( unsigned int from, unsigned int to, bool on );
 501     void ReverseRowSelection( unsigned int row );
 502     bool IsRowSelected( unsigned int row );
 503     void SendSelectionChangedEvent( const wxDataViewItem& item);
 504
 505     void RefreshRow( unsigned int row );
 506     void RefreshRows( unsigned int from, unsigned int to );
 507     void RefreshRowsAfter( unsigned int firstRow );
 508
 509     // returns the colour to be used for drawing the rules
 510     wxColour GetRuleColour() const
 511     {
 512         return wxSystemSettings::GetColour(wxSYS_COLOUR_3DLIGHT);
 513     }
 514
 515     wxRect GetLineRect( unsigned int row ) const;
 516
 517     //Some useful functions for row and item mapping
 518     wxDataViewItem GetItemByRow( unsigned int row ) const;
 519     int GetRowByItem( const wxDataViewItem & item );
 520
 521     //Methods for building the mapping tree
 522     void BuildTree( wxDataViewModel  * model );
 523     void DestroyTree();
 524     void HitTest( const wxPoint & point, wxDataViewItem & item, wxDataViewColumn* &column );
 525     wxRect GetItemRect( const wxDataViewItem & item, const wxDataViewColumn* column );
 526
 527     void Expand( unsigned int row ) { OnExpanding( row ); }
 528     void Collapse( unsigned int row ) { OnCollapsing( row ); }
 529 private:
 530     wxDataViewTreeNode * GetTreeNodeByRow( unsigned int row );
 531     //We did not need this temporarily
 532     //wxDataViewTreeNode * GetTreeNodeByItem( const wxDataViewItem & item );
 533
 534     int RecalculateCount() ;
 535
 536     wxDataViewEvent SendExpanderEvent( wxEventType type, const wxDataViewItem & item );
 537     void OnExpanding( unsigned int row );
 538     void OnCollapsing( unsigned int row );
 539
 540     wxDataViewTreeNode * FindNode( const wxDataViewItem & item );
 541
 542 private:
 543     wxDataViewCtrl             *m_owner;
 544     int                         m_lineHeight;
 545     bool                        m_dirty;
 546
 547     wxDataViewColumn           *m_currentCol;
 548     unsigned int                m_currentRow;
 549     wxDataViewSelection         m_selection;
 550
 551     wxDataViewRenameTimer      *m_renameTimer;
 552     bool                        m_lastOnSame;
 553
 554     bool                        m_hasFocus;
 555
 556     int                         m_dragCount;
 557     wxPoint                     m_dragStart;
 558
 559     // for double click logic
 560     unsigned int m_lineLastClicked,
 561            m_lineBeforeLastClicked,
 562            m_lineSelectSingleOnUp;
 563
 564     // the pen used to draw horiz/vertical rules
 565     wxPen m_penRule;
 566
 567     // the pen used to draw the expander and the lines
 568     wxPen m_penExpander;
 569
 570     //This is the tree structure of the model
 571     wxDataViewTreeNode * m_root;
 572     int m_count;
 573     //This is the tree node under the cursor
 574     wxDataViewTreeNode * m_underMouse;
 575 private:
 576     DECLARE_DYNAMIC_CLASS(wxDataViewMainWindow)
 577     DECLARE_EVENT_TABLE()
 578 };
 579
 580 // ---------------------------------------------------------
 581 // wxGenericDataViewModelNotifier
 582 // ---------------------------------------------------------
 583
 584 class wxGenericDataViewModelNotifier: public wxDataViewModelNotifier
 585 {
 586 public:
 587     wxGenericDataViewModelNotifier( wxDataViewMainWindow *mainWindow )
 588         { m_mainWindow = mainWindow; }
 589
 590     virtual bool ItemAdded( const wxDataViewItem & parent, const wxDataViewItem & item )
 591         { return m_mainWindow->ItemAdded( parent , item ); }
 592     virtual bool ItemDeleted( const wxDataViewItem &parent, const wxDataViewItem &item )
 593         { return m_mainWindow->ItemDeleted( parent, item ); }
 594     virtual bool ItemChanged( const wxDataViewItem & item )
 595         { return m_mainWindow->ItemChanged(item);  }
 596     virtual bool ValueChanged( const wxDataViewItem & item , unsigned int col )
 597         { return m_mainWindow->ValueChanged( item, col ); }
 598     virtual bool Cleared()
 599         { return m_mainWindow->Cleared(); }
 600     virtual void Resort()
 601          { m_mainWindow->Resort(); }
 602
 603     wxDataViewMainWindow    *m_mainWindow;
 604 };
 605
 606 // ---------------------------------------------------------
 607 // wxDataViewRenderer
 608 // ---------------------------------------------------------
 609
 610 IMPLEMENT_ABSTRACT_CLASS(wxDataViewRenderer, wxDataViewRendererBase)
 611
 612 wxDataViewRenderer::wxDataViewRenderer( const wxString &varianttype,
 613                                         wxDataViewCellMode mode,
 614                                         int align) :
 615     wxDataViewRendererBase( varianttype, mode, align )
 616 {
 617     m_dc = NULL;
 618     m_align = align;
 619     m_mode = mode;
 620     m_wantsAttr = false;
 621     m_hasAttr = false;
 622 }
 623
 624 wxDataViewRenderer::~wxDataViewRenderer()
 625 {
 626     if (m_dc)
 627         delete m_dc;
 628 }
 629
 630 wxDC *wxDataViewRenderer::GetDC()
 631 {
 632     if (m_dc == NULL)
 633     {
 634         if (GetOwner() == NULL)
 635             return NULL;
 636         if (GetOwner()->GetOwner() == NULL)
 637             return NULL;
 638         m_dc = new wxClientDC( GetOwner()->GetOwner() );
 639     }
 640
 641     return m_dc;
 642 }
 643
 644 void wxDataViewRenderer::SetAlignment( int align )
 645 {
 646     m_align=align;
 647 }
 648
 649 int wxDataViewRenderer::GetAlignment() const
 650 {
 651     return m_align;
 652 }
 653
 654 int wxDataViewRenderer::CalculateAlignment() const
 655 {
 656     if (m_align == wxDVR_DEFAULT_ALIGNMENT)
 657     {
 658         if (GetOwner() == NULL)
 659            return wxALIGN_LEFT | wxALIGN_CENTRE_VERTICAL;
 660
 661         return GetOwner()->GetAlignment() | wxALIGN_CENTRE_VERTICAL;
 662     }
 663
 664     return m_align;
 665 }
 666
 667 // ---------------------------------------------------------
 668 // wxDataViewCustomRenderer
 669 // ---------------------------------------------------------
 670
 671 IMPLEMENT_ABSTRACT_CLASS(wxDataViewCustomRenderer, wxDataViewRenderer)
 672
 673 wxDataViewCustomRenderer::wxDataViewCustomRenderer( const wxString &varianttype,
 674                           wxDataViewCellMode mode, int align ) :
 675     wxDataViewRenderer( varianttype, mode, align )
 676 {
 677 }
 678
 679 void wxDataViewCustomRenderer::RenderText( const wxString &text, int xoffset, wxRect cell, wxDC *dc, int state )
 680 {
 681     wxDataViewCtrl *view = GetOwner()->GetOwner();
 682     wxColour col = (state & wxDATAVIEW_CELL_SELECTED) ?
 683                         wxSystemSettings::GetColour(wxSYS_COLOUR_HIGHLIGHTTEXT) :
 684                         view->GetForegroundColour();
 685     dc->SetTextForeground(col);
 686     dc->DrawText( text, cell.x + xoffset, cell.y + ((cell.height - dc->GetCharHeight()) / 2));
 687 }
 688
 689 // ---------------------------------------------------------
 690 // wxDataViewTextRenderer
 691 // ---------------------------------------------------------
 692
 693 IMPLEMENT_CLASS(wxDataViewTextRenderer, wxDataViewCustomRenderer)
 694
 695 wxDataViewTextRenderer::wxDataViewTextRenderer( const wxString &varianttype,
 696                                                 wxDataViewCellMode mode, int align ) :
 697     wxDataViewCustomRenderer( varianttype, mode, align )
 698 {
 699 }
 700
 701 bool wxDataViewTextRenderer::SetValue( const wxVariant &value )
 702 {
 703     m_text = value.GetString();
 704
 705     return true;
 706 }
 707
 708 bool wxDataViewTextRenderer::GetValue( wxVariant& WXUNUSED(value) ) const
 709 {
 710     return false;
 711 }
 712
 713 bool wxDataViewTextRenderer::HasEditorCtrl()
 714 {
 715     return true;
 716 }
 717
 718 wxControl* wxDataViewTextRenderer::CreateEditorCtrl( wxWindow *parent,
 719         wxRect labelRect, const wxVariant &value )
 720 {
 721     return new wxTextCtrl( parent, wxID_ANY, value,
 722                            wxPoint(labelRect.x,labelRect.y),
 723                            wxSize(labelRect.width,labelRect.height) );
 724 }
 725
 726 bool wxDataViewTextRenderer::GetValueFromEditorCtrl( wxControl *editor, wxVariant &value )
 727 {
 728     wxTextCtrl *text = (wxTextCtrl*) editor;
 729     value = text->GetValue();
 730     return true;
 731 }
 732
 733 bool wxDataViewTextRenderer::Render( wxRect cell, wxDC *dc, int state )
 734 {
 735     RenderText( m_text, 0, cell, dc, state );
 736     return true;
 737 }
 738
 739 wxSize wxDataViewTextRenderer::GetSize() const
 740 {
 741     const wxDataViewCtrl *view = GetView();
 742     if (!m_text.empty())
 743     {
 744         int x,y;
 745         view->GetTextExtent( m_text, &x, &y );
 746         return wxSize( x, y );
 747     }
 748     return wxSize(80,20);
 749 }
 750
 751 // ---------------------------------------------------------
 752 // wxDataViewTextRendererAttr
 753 // ---------------------------------------------------------
 754
 755 IMPLEMENT_CLASS(wxDataViewTextRendererAttr, wxDataViewTextRenderer)
 756
 757 wxDataViewTextRendererAttr::wxDataViewTextRendererAttr( const wxString &varianttype,
 758                             wxDataViewCellMode mode, int align ) :
 759     wxDataViewTextRenderer( varianttype, mode, align )
 760 {
 761     m_wantsAttr = true;
 762 }
 763
 764 bool wxDataViewTextRendererAttr::Render( wxRect cell, wxDC *dc, int WXUNUSED(state) )
 765 {
 766     wxFont font;
 767     wxColour colour;
 768
 769     if (m_hasAttr)
 770     {
 771         if (m_attr.HasColour())
 772         {
 773             colour = dc->GetTextForeground();
 774             dc->SetTextForeground( m_attr.GetColour() );
 775         }
 776
 777         if (m_attr.GetBold() || m_attr.GetItalic())
 778         {
 779              font = dc->GetFont();
 780              wxFont myfont = font;
 781              if (m_attr.GetBold())
 782                  myfont.SetWeight( wxFONTWEIGHT_BOLD );
 783              if (m_attr.GetItalic())
 784                  myfont.SetStyle( wxFONTSTYLE_ITALIC );
 785              dc->SetFont( myfont );
 786         }
 787     }
 788
 789     dc->DrawText( m_text, cell.x, cell.y + ((cell.height - dc->GetCharHeight()) / 2));
 790
 791     // restore dc
 792     if (m_hasAttr)
 793     {
 794         if (m_attr.HasColour())
 795             dc->SetTextForeground( colour );
 796
 797         if (m_attr.GetBold() || m_attr.GetItalic())
 798             dc->SetFont( font );
 799     }
 800
 801     return true;
 802 }
 803
 804
 805 // ---------------------------------------------------------
 806 // wxDataViewBitmapRenderer
 807 // ---------------------------------------------------------
 808
 809 IMPLEMENT_CLASS(wxDataViewBitmapRenderer, wxDataViewCustomRenderer)
 810
 811 wxDataViewBitmapRenderer::wxDataViewBitmapRenderer( const wxString &varianttype,
 812                                                     wxDataViewCellMode mode, int align ) :
 813     wxDataViewCustomRenderer( varianttype, mode, align )
 814 {
 815 }
 816
 817 bool wxDataViewBitmapRenderer::SetValue( const wxVariant &value )
 818 {
 819     if (value.GetType() == wxT("wxBitmap"))
 820         m_bitmap << value;
 821     if (value.GetType() == wxT("wxIcon"))
 822         m_icon << value;
 823
 824     return true;
 825 }
 826
 827 bool wxDataViewBitmapRenderer::GetValue( wxVariant& WXUNUSED(value) ) const
 828 {
 829     return false;
 830 }
 831
 832 bool wxDataViewBitmapRenderer::Render( wxRect cell, wxDC *dc, int WXUNUSED(state) )
 833 {
 834     if (m_bitmap.Ok())
 835         dc->DrawBitmap( m_bitmap, cell.x, cell.y );
 836     else if (m_icon.Ok())
 837         dc->DrawIcon( m_icon, cell.x, cell.y );
 838
 839     return true;
 840 }
 841
 842 wxSize wxDataViewBitmapRenderer::GetSize() const
 843 {
 844     if (m_bitmap.Ok())
 845         return wxSize( m_bitmap.GetWidth(), m_bitmap.GetHeight() );
 846     else if (m_icon.Ok())
 847         return wxSize( m_icon.GetWidth(), m_icon.GetHeight() );
 848
 849     return wxSize(16,16);
 850 }
 851
 852 // ---------------------------------------------------------
 853 // wxDataViewToggleRenderer
 854 // ---------------------------------------------------------
 855
 856 IMPLEMENT_ABSTRACT_CLASS(wxDataViewToggleRenderer, wxDataViewCustomRenderer)
 857
 858 wxDataViewToggleRenderer::wxDataViewToggleRenderer( const wxString &varianttype,
 859                         wxDataViewCellMode mode, int align ) :
 860     wxDataViewCustomRenderer( varianttype, mode, align )
 861 {
 862     m_toggle = false;
 863 }
 864
 865 bool wxDataViewToggleRenderer::SetValue( const wxVariant &value )
 866 {
 867     m_toggle = value.GetBool();
 868
 869     return true;
 870 }
 871
 872 bool wxDataViewToggleRenderer::GetValue( wxVariant &WXUNUSED(value) ) const
 873 {
 874     return false;
 875 }
 876
 877 bool wxDataViewToggleRenderer::Render( wxRect cell, wxDC *dc, int WXUNUSED(state) )
 878 {
 879     // User wxRenderer here
 880
 881     wxRect rect;
 882     rect.x = cell.x + cell.width/2 - 10;
 883     rect.width = 20;
 884     rect.y = cell.y + cell.height/2 - 10;
 885     rect.height = 20;
 886
 887     int flags = 0;
 888     if (m_toggle)
 889         flags |= wxCONTROL_CHECKED;
 890     if (GetMode() != wxDATAVIEW_CELL_ACTIVATABLE)
 891         flags |= wxCONTROL_DISABLED;
 892
 893     wxRendererNative::Get().DrawCheckBox(
 894             GetOwner()->GetOwner(),
 895             *dc,
 896             rect,
 897             flags );
 898
 899     return true;
 900 }
 901
 902 bool wxDataViewToggleRenderer::Activate( wxRect WXUNUSED(cell),
 903                                          wxDataViewModel *model,
 904                                          const wxDataViewItem & item, unsigned int col)
 905 {
 906     bool value = !m_toggle;
 907     wxVariant variant = value;
 908     model->SetValue( variant, item, col);
 909     model->ValueChanged( item, col );
 910     return true;
 911 }
 912
 913 wxSize wxDataViewToggleRenderer::GetSize() const
 914 {
 915     return wxSize(20,20);
 916 }
 917
 918 // ---------------------------------------------------------
 919 // wxDataViewProgressRenderer
 920 // ---------------------------------------------------------
 921
 922 IMPLEMENT_ABSTRACT_CLASS(wxDataViewProgressRenderer, wxDataViewCustomRenderer)
 923
 924 wxDataViewProgressRenderer::wxDataViewProgressRenderer( const wxString &label,
 925     const wxString &varianttype, wxDataViewCellMode mode, int align ) :
 926     wxDataViewCustomRenderer( varianttype, mode, align )
 927 {
 928     m_label = label;
 929     m_value = 0;
 930 }
 931
 932 wxDataViewProgressRenderer::~wxDataViewProgressRenderer()
 933 {
 934 }
 935
 936 bool wxDataViewProgressRenderer::SetValue( const wxVariant &value )
 937 {
 938     m_value = (long) value;
 939
 940     if (m_value < 0) m_value = 0;
 941     if (m_value > 100) m_value = 100;
 942
 943     return true;
 944 }
 945
 946 bool wxDataViewProgressRenderer::GetValue( wxVariant &value ) const
 947 {
 948     value = (long) m_value;
 949     return true;
 950 }
 951
 952 bool wxDataViewProgressRenderer::Render( wxRect cell, wxDC *dc, int WXUNUSED(state) )
 953 {
 954     double pct = (double)m_value / 100.0;
 955     wxRect bar = cell;
 956     bar.width = (int)(cell.width * pct);
 957     dc->SetPen( *wxTRANSPARENT_PEN );
 958     dc->SetBrush( *wxBLUE_BRUSH );
 959     dc->DrawRectangle( bar );
 960
 961     dc->SetBrush( *wxTRANSPARENT_BRUSH );
 962     dc->SetPen( *wxBLACK_PEN );
 963     dc->DrawRectangle( cell );
 964
 965     return true;
 966 }
 967
 968 wxSize wxDataViewProgressRenderer::GetSize() const
 969 {
 970     return wxSize(40,12);
 971 }
 972
 973 // ---------------------------------------------------------
 974 // wxDataViewDateRenderer
 975 // ---------------------------------------------------------
 976
 977 #define wxUSE_DATE_RENDERER_POPUP (wxUSE_CALENDARCTRL && wxUSE_POPUPWIN)
 978
 979 #if wxUSE_DATE_RENDERER_POPUP
 980
 981 class wxDataViewDateRendererPopupTransient: public wxPopupTransientWindow
 982 {
 983 public:
 984     wxDataViewDateRendererPopupTransient( wxWindow* parent, wxDateTime *value,
 985         wxDataViewModel *model, const wxDataViewItem & item, unsigned int col) :
 986         wxPopupTransientWindow( parent, wxBORDER_SIMPLE ),
 987         m_item( item )
 988     {
 989         m_model = model;
 990         m_col = col;
 991         m_cal = new wxCalendarCtrl( this, wxID_ANY, *value );
 992         wxBoxSizer *sizer = new wxBoxSizer( wxHORIZONTAL );
 993         sizer->Add( m_cal, 1, wxGROW );
 994         SetSizer( sizer );
 995         sizer->Fit( this );
 996     }
 997
 998     void OnCalendar( wxCalendarEvent &event );
 999
1000     wxCalendarCtrl      *m_cal;
1001     wxDataViewModel *m_model;
1002     unsigned int               m_col;
1003     const wxDataViewItem &   m_item;
1004
1005 protected:
1006     virtual void OnDismiss()
1007     {
1008     }
1009
1010 private:
1011     DECLARE_EVENT_TABLE()
1012 };
1013
1014 BEGIN_EVENT_TABLE(wxDataViewDateRendererPopupTransient,wxPopupTransientWindow)
1015     EVT_CALENDAR( wxID_ANY, wxDataViewDateRendererPopupTransient::OnCalendar )
1016 END_EVENT_TABLE()
1017
1018 void wxDataViewDateRendererPopupTransient::OnCalendar( wxCalendarEvent &event )
1019 {
1020     wxDateTime date = event.GetDate();
1021     wxVariant value = date;
1022     m_model->SetValue( value, m_item, m_col );
1023     m_model->ValueChanged( m_item, m_col );
1024     DismissAndNotify();
1025 }
1026
1027 #endif // wxUSE_DATE_RENDERER_POPUP
1028
1029 IMPLEMENT_ABSTRACT_CLASS(wxDataViewDateRenderer, wxDataViewCustomRenderer)
1030
1031 wxDataViewDateRenderer::wxDataViewDateRenderer( const wxString &varianttype,
1032                         wxDataViewCellMode mode, int align ) :
1033     wxDataViewCustomRenderer( varianttype, mode, align )
1034 {
1035 }
1036
1037 bool wxDataViewDateRenderer::SetValue( const wxVariant &value )
1038 {
1039     m_date = value.GetDateTime();
1040
1041     return true;
1042 }
1043
1044 bool wxDataViewDateRenderer::GetValue( wxVariant &value ) const
1045 {
1046     value = m_date;
1047     return true;
1048 }
1049
1050 bool wxDataViewDateRenderer::Render( wxRect cell, wxDC *dc, int state )
1051 {
1052     wxString tmp = m_date.FormatDate();
1053     RenderText( tmp, 0, cell, dc, state );
1054     return true;
1055 }
1056
1057 wxSize wxDataViewDateRenderer::GetSize() const
1058 {
1059     const wxDataViewCtrl* view = GetView();
1060     wxString tmp = m_date.FormatDate();
1061     wxCoord x,y,d;
1062     view->GetTextExtent( tmp, &x, &y, &d );
1063     return wxSize(x,y+d);
1064 }
1065
1066 bool wxDataViewDateRenderer::Activate( wxRect WXUNUSED(cell), wxDataViewModel *model,
1067                                        const wxDataViewItem & item, unsigned int col )
1068 {
1069     wxVariant variant;
1070     model->GetValue( variant, item, col );
1071     wxDateTime value = variant.GetDateTime();
1072
1073 #if wxUSE_DATE_RENDERER_POPUP
1074     wxDataViewDateRendererPopupTransient *popup = new wxDataViewDateRendererPopupTransient(
1075         GetOwner()->GetOwner()->GetParent(), &value, model, item, col);
1076     wxPoint pos = wxGetMousePosition();
1077     popup->Move( pos );
1078     popup->Layout();
1079     popup->Popup( popup->m_cal );
1080 #else // !wxUSE_DATE_RENDERER_POPUP
1081     wxMessageBox(value.Format());
1082 #endif // wxUSE_DATE_RENDERER_POPUP/!wxUSE_DATE_RENDERER_POPUP
1083     return true;
1084 }
1085
1086 // ---------------------------------------------------------
1087 // wxDataViewIconTextRenderer
1088 // ---------------------------------------------------------
1089
1090 IMPLEMENT_CLASS(wxDataViewIconTextRenderer, wxDataViewCustomRenderer)
1091
1092 wxDataViewIconTextRenderer::wxDataViewIconTextRenderer(
1093   const wxString &varianttype, wxDataViewCellMode mode, int align ) :
1094     wxDataViewCustomRenderer( varianttype, mode, align )
1095 {
1096     SetMode(mode);
1097     SetAlignment(align);
1098 }
1099
1100 wxDataViewIconTextRenderer::~wxDataViewIconTextRenderer()
1101 {
1102 }
1103
1104 bool wxDataViewIconTextRenderer::SetValue( const wxVariant &value )
1105 {
1106     m_value << value;
1107     return true;
1108 }
1109
1110 bool wxDataViewIconTextRenderer::GetValue( wxVariant& WXUNUSED(value) ) const
1111 {
1112     return false;
1113 }
1114
1115 bool wxDataViewIconTextRenderer::Render( wxRect cell, wxDC *dc, int state )
1116 {
1117     int xoffset = 0;
1118     const wxIcon &icon = m_value.GetIcon();
1119     if (icon.IsOk())
1120     {
1121         dc->DrawIcon( icon, cell.x, cell.y + ((cell.height - icon.GetHeight()) / 2));
1122         xoffset =  icon.GetWidth()+4;
1123     }
1124
1125     RenderText( m_value.GetText(), xoffset, cell, dc, state );
1126
1127     return true;
1128 }
1129
1130 wxSize wxDataViewIconTextRenderer::GetSize() const
1131 {
1132     const wxDataViewCtrl *view = GetView();
1133     if (!m_value.GetText().empty())
1134     {
1135         int x,y;
1136         view->GetTextExtent( m_value.GetText(), &x, &y );
1137
1138         if (m_value.GetIcon().IsOk())
1139             x += m_value.GetIcon().GetWidth() + 4;
1140         return wxSize( x, y );
1141     }
1142     return wxSize(80,20);
1143 }
1144
1145 wxControl *
1146 wxDataViewIconTextRenderer::CreateEditorCtrl(wxWindow * WXUNUSED(parent),
1147                                              wxRect WXUNUSED(labelRect),
1148                                              const wxVariant& WXUNUSED(value))
1149 {
1150     return NULL;
1151 }
1152
1153 bool
1154 wxDataViewIconTextRenderer::GetValueFromEditorCtrl(wxControl* WXUNUSED(editor),
1155                                                    wxVariant& WXUNUSED(value))
1156 {
1157     return false;
1158 }
1159
1160 // ---------------------------------------------------------
1161 // wxDataViewColumn
1162 // ---------------------------------------------------------
1163
1164 IMPLEMENT_ABSTRACT_CLASS(wxDataViewColumn, wxDataViewColumnBase)
1165
1166 wxDataViewColumn::wxDataViewColumn( const wxString &title, wxDataViewRenderer *cell,
1167                                     unsigned int model_column,
1168                                     int width, wxAlignment align, int flags ) :
1169     wxDataViewColumnBase( title, cell, model_column, width, align, flags )
1170 {
1171     SetAlignment(align);
1172     SetTitle(title);
1173     SetFlags(flags);
1174
1175     m_autosize = width < 0;  // TODO
1176
1177     Init(width < 0 ? wxDVC_DEFAULT_WIDTH : width);
1178 }
1179
1180 wxDataViewColumn::wxDataViewColumn( const wxBitmap &bitmap, wxDataViewRenderer *cell,
1181                                     unsigned int model_column,
1182                                     int width, wxAlignment align, int flags ) :
1183     wxDataViewColumnBase( bitmap, cell, model_column, width, align, flags )
1184 {
1185     SetAlignment(align);
1186     SetFlags(flags);
1187
1188     Init(width < 0 ? wxDVC_TOGGLE_DEFAULT_WIDTH : width);
1189 }
1190
1191 wxDataViewColumn::~wxDataViewColumn()
1192 {
1193 }
1194
1195 void wxDataViewColumn::Init( int width )
1196 {
1197     m_width = width;
1198     m_minWidth = wxDVC_DEFAULT_MINWIDTH;
1199     m_ascending = true;
1200 }
1201
1202 void wxDataViewColumn::SetResizeable( bool resizeable )
1203 {
1204     if (resizeable)
1205         m_flags |= wxDATAVIEW_COL_RESIZABLE;
1206     else
1207         m_flags &= ~wxDATAVIEW_COL_RESIZABLE;
1208 }
1209
1210 void wxDataViewColumn::SetHidden( bool hidden )
1211 {
1212     if (hidden)
1213         m_flags |= wxDATAVIEW_COL_HIDDEN;
1214     else
1215         m_flags &= ~wxDATAVIEW_COL_HIDDEN;
1216
1217     // tell our owner to e.g. update its scrollbars:
1218     if (GetOwner())
1219         GetOwner()->OnColumnChange();
1220 }
1221
1222 void wxDataViewColumn::SetSortable( bool sortable )
1223 {
1224     if (sortable)
1225         m_flags |= wxDATAVIEW_COL_SORTABLE;
1226     else
1227         m_flags &= ~wxDATAVIEW_COL_SORTABLE;
1228
1229     // Update header button
1230     if (GetOwner())
1231         GetOwner()->OnColumnChange();
1232 }
1233
1234 void wxDataViewColumn::SetReorderable( bool reorderable )
1235 {
1236     if (reorderable)
1237         m_flags |= wxDATAVIEW_COL_REORDERABLE;
1238     else
1239         m_flags &= ~wxDATAVIEW_COL_REORDERABLE;
1240 }
1241
1242 void wxDataViewColumn::SetSortOrder( bool ascending )
1243 {
1244     m_ascending = ascending;
1245
1246     // Update header button
1247     if (GetOwner())
1248         GetOwner()->OnColumnChange();
1249 }
1250
1251 bool wxDataViewColumn::IsSortOrderAscending() const
1252 {
1253     return m_ascending;
1254 }
1255
1256 void wxDataViewColumn::SetInternalWidth( int width )
1257 {
1258     m_width = width;
1259
1260     // the scrollbars of the wxDataViewCtrl needs to be recalculated!
1261     if (m_owner && m_owner->m_clientArea)
1262         m_owner->m_clientArea->RecalculateDisplay();
1263 }
1264
1265 void wxDataViewColumn::SetWidth( int width )
1266 {
1267     if (m_owner->m_headerArea) m_owner->m_headerArea->UpdateDisplay();
1268
1269     SetInternalWidth(width);
1270 }
1271
1272
1273 //-----------------------------------------------------------------------------
1274 // wxDataViewHeaderWindowBase
1275 //-----------------------------------------------------------------------------
1276
1277 void wxDataViewHeaderWindowBase::SendEvent(wxEventType type, unsigned int n)
1278 {
1279     wxWindow *parent = GetParent();
1280     wxDataViewEvent le(type, parent->GetId());
1281
1282     le.SetEventObject(parent);
1283     le.SetColumn(n);
1284     le.SetDataViewColumn(GetColumn(n));
1285     le.SetModel(GetOwner()->GetModel());
1286
1287     // for events created by wxDataViewHeaderWindow the
1288     // row / value fields are not valid
1289
1290     parent->GetEventHandler()->ProcessEvent(le);
1291 }
1292
1293 #ifdef USE_NATIVE_HEADER_WINDOW
1294
1295 #ifndef HDS_DRAGDROP
1296     #define HDS_DRAGDROP 0x0040
1297 #endif
1298 #ifndef HDS_FULLDRAG
1299     #define HDS_FULLDRAG 0x0080
1300 #endif
1301
1302 // implemented in msw/listctrl.cpp:
1303 int WXDLLIMPEXP_CORE wxMSWGetColumnClicked(NMHDR *nmhdr, POINT *ptClick);
1304
1305 IMPLEMENT_ABSTRACT_CLASS(wxDataViewHeaderWindowMSW, wxWindow)
1306
1307 bool wxDataViewHeaderWindowMSW::Create( wxDataViewCtrl *parent, wxWindowID id,
1308                                         const wxPoint &pos, const wxSize &size,
1309                                         const wxString &name )
1310 {
1311     m_owner = parent;
1312
1313     m_scrollOffsetX = 0;
1314     m_delayedUpdate = false;
1315     m_buttonHeight = wxRendererNative::Get().GetHeaderButtonHeight( this );
1316
1317     int x = pos.x == wxDefaultCoord ? 0 : pos.x,
1318         y = pos.y == wxDefaultCoord ? 0 : pos.y,
1319         w = size.x == wxDefaultCoord ? 1 : size.x,
1320         h = m_buttonHeight;
1321
1322     wxSize new_size(w,h);
1323
1324     if ( !CreateControl(parent, id, pos, new_size, 0, wxDefaultValidator, name) )
1325         return false;
1326
1327     // create the native WC_HEADER window:
1328     WXHWND hwndParent = (HWND)parent->GetHandle();
1329     WXDWORD msStyle = WS_CHILD | HDS_DRAGDROP | HDS_BUTTONS | HDS_HORZ | HDS_HOTTRACK | HDS_FULLDRAG;
1330
1331     if ( m_isShown )
1332         msStyle |= WS_VISIBLE;
1333
1334     m_hWnd = CreateWindowEx(0,
1335                             WC_HEADER,
1336                             (LPCTSTR) NULL,
1337                             msStyle,
1338                             x, y, w, h,
1339                             (HWND)hwndParent,
1340                             (HMENU)-1,
1341                             wxGetInstance(),
1342                             (LPVOID) NULL);
1343     if (m_hWnd == NULL)
1344     {
1345         wxLogLastError(_T("CreateWindowEx"));
1346         return false;
1347     }
1348
1349     // we need to subclass the m_hWnd to force wxWindow::HandleNotify
1350     // to call wxDataViewHeaderWindow::MSWOnNotify
1351     SubclassWin(m_hWnd);
1352
1353     // the following is required to get the default win's font for
1354     // header windows and must be done befor sending the HDM_LAYOUT msg
1355     SetFont(GetFont());
1356
1357     return true;
1358 }
1359
1360 wxDataViewHeaderWindowMSW::~wxDataViewHeaderWindow()
1361 {
1362     UnsubclassWin();
1363 }
1364
1365 wxSize wxDataViewHeaderWindowMSW::DoGetBestSize() const
1366 {
1367     return wxSize( 80, m_buttonHeight+2 );
1368 }
1369
1370 void wxDataViewHeaderWindowMSW::OnInternalIdle()
1371 {
1372     if (m_delayedUpdate)
1373     {
1374         m_delayedUpdate = false;
1375         UpdateDisplay();
1376     }
1377 }
1378
1379 void wxDataViewHeaderWindowMSW::UpdateDisplay()
1380 {
1381     // remove old columns
1382     for (int j=0, max=Header_GetItemCount((HWND)m_hWnd); j < max; j++)
1383         Header_DeleteItem((HWND)m_hWnd, 0);
1384
1385     // add the updated array of columns to the header control
1386     unsigned int cols = GetOwner()->GetColumnCount();
1387     unsigned int added = 0;
1388     for (unsigned int i = 0; i < cols; i++)
1389     {
1390         wxDataViewColumn *col = GetColumn( i );
1391         if (col->IsHidden())
1392             continue;      // don't add it!
1393
1394         HDITEM hdi;
1395         hdi.mask = HDI_TEXT | HDI_FORMAT | HDI_WIDTH;
1396         if (col->GetBitmap().IsOk())
1397         {
1398            hdi.mask |= HDI_BITMAP;
1399            hdi.hbm = (HBITMAP) col->GetBitmap().GetHBITMAP();
1400         }
1401         hdi.pszText = (wxChar *) col->GetTitle().wx_str();
1402         hdi.cxy = col->GetWidth();
1403         hdi.cchTextMax = sizeof(hdi.pszText)/sizeof(hdi.pszText[0]);
1404         hdi.fmt = HDF_LEFT | HDF_STRING;
1405         if (col->GetBitmap().IsOk())
1406             hdi.fmt |= HDF_BITMAP;
1407
1408         //hdi.fmt &= ~(HDF_SORTDOWN|HDF_SORTUP);
1409
1410         if (col->IsSortable() && GetOwner()->GetSortingColumn() == col)
1411         {
1412             //The Microsoft Comctrl32.dll 6.0 support SORTUP/SORTDOWN, but they are not default
1413             //see http://msdn2.microsoft.com/en-us/library/ms649534.aspx for more detail
1414             // VZ: works with 5.81
1415             hdi.fmt |= col->IsSortOrderAscending() ? HDF_SORTUP : HDF_SORTDOWN;
1416         }
1417
1418         // lParam is reserved for application's use:
1419         // we store there the column index to use it later in MSWOnNotify
1420         // (since columns may have been hidden)
1421         hdi.lParam = (LPARAM)i;
1422
1423         // the native wxMSW implementation of the header window
1424         // draws the column separator COLUMN_WIDTH_OFFSET pixels
1425         // on the right: to correct this effect we make the column
1426         // exactly COLUMN_WIDTH_OFFSET wider (for the first column):
1427         if (i == 0)
1428             hdi.cxy += COLUMN_WIDTH_OFFSET;
1429
1430         switch (col->GetAlignment())
1431         {
1432         case wxALIGN_LEFT:
1433             hdi.fmt |= HDF_LEFT;
1434             break;
1435         case wxALIGN_CENTER:
1436         case wxALIGN_CENTER_HORIZONTAL:
1437             hdi.fmt |= HDF_CENTER;
1438             break;
1439         case wxALIGN_RIGHT:
1440             hdi.fmt |= HDF_RIGHT;
1441             break;
1442
1443         default:
1444             // such alignment is not allowed for the column header!
1445             break; // wxFAIL;
1446         }
1447
1448         SendMessage((HWND)m_hWnd, HDM_INSERTITEM,
1449                     (WPARAM)added, (LPARAM)&hdi);
1450         added++;
1451     }
1452 }
1453
1454 unsigned int wxDataViewHeaderWindowMSW::GetColumnIdxFromHeader(NMHEADER *nmHDR)
1455 {
1456     unsigned int idx;
1457
1458     // NOTE: we don't just return nmHDR->iItem because when there are
1459     //       hidden columns, nmHDR->iItem may be different from
1460     //       nmHDR->pitem->lParam
1461
1462     if (nmHDR->pitem && nmHDR->pitem->mask & HDI_LPARAM)
1463     {
1464         idx = (unsigned int)nmHDR->pitem->lParam;
1465         return idx;
1466     }
1467
1468     HDITEM item;
1469     item.mask = HDI_LPARAM;
1470     Header_GetItem((HWND)m_hWnd, nmHDR->iItem, &item);
1471
1472     return (unsigned int)item.lParam;
1473 }
1474
1475 bool wxDataViewHeaderWindowMSW::MSWOnNotify(int idCtrl, WXLPARAM lParam, WXLPARAM *result)
1476 {
1477     NMHDR *nmhdr = (NMHDR *)lParam;
1478
1479     // is it a message from the header?
1480     if ( nmhdr->hwndFrom != (HWND)m_hWnd )
1481         return wxWindow::MSWOnNotify(idCtrl, lParam, result);
1482
1483     NMHEADER *nmHDR = (NMHEADER *)nmhdr;
1484     switch ( nmhdr->code )
1485     {
1486         case HDN_BEGINTRACK:
1487             // user has started to resize a column:
1488             // do we need to veto it?
1489             if (!GetColumn(nmHDR->iItem)->IsResizeable())
1490             {
1491                 // veto it!
1492                 *result = TRUE;
1493             }
1494             break;
1495
1496         case HDN_BEGINDRAG:
1497             // user has started to reorder a column
1498             if ((nmHDR->iItem != -1) && (!GetColumn(nmHDR->iItem)->IsReorderable()))
1499             {
1500                 // veto it!
1501                 *result = TRUE;
1502             }
1503             break;
1504
1505         case HDN_ENDDRAG:       // user has finished reordering a column
1506             {
1507                wxDataViewColumn *col = GetColumn(nmHDR->iItem);
1508                unsigned int new_pos = nmHDR->pitem->iOrder;
1509                m_owner->ColumnMoved( col, new_pos );
1510                m_delayedUpdate = true;
1511             }
1512             break;
1513
1514         case HDN_ITEMCHANGING:
1515             if (nmHDR->pitem != NULL &&
1516                 (nmHDR->pitem->mask & HDI_WIDTH) != 0)
1517             {
1518                 int minWidth = GetColumnFromHeader(nmHDR)->GetMinWidth();
1519                 if (nmHDR->pitem->cxy < minWidth)
1520                 {
1521                     // do not allow the user to resize this column under
1522                     // its minimal width:
1523                     *result = TRUE;
1524                 }
1525             }
1526             break;
1527
1528         case HDN_ITEMCHANGED:   // user is resizing a column
1529         case HDN_ENDTRACK:      // user has finished resizing a column
1530
1531             // update the width of the modified column:
1532             if (nmHDR->pitem != NULL &&
1533                 (nmHDR->pitem->mask & HDI_WIDTH) != 0)
1534             {
1535                 unsigned int idx = GetColumnIdxFromHeader(nmHDR);
1536                 unsigned int w = nmHDR->pitem->cxy;
1537                 wxDataViewColumn *col = GetColumn(idx);
1538
1539                 // see UpdateDisplay() for more info about COLUMN_WIDTH_OFFSET
1540                 if (idx == 0 && w > COLUMN_WIDTH_OFFSET)
1541                     w -= COLUMN_WIDTH_OFFSET;
1542
1543                 if (w >= (unsigned)col->GetMinWidth())
1544                     col->SetInternalWidth(w);
1545             }
1546             break;
1547
1548         case HDN_ITEMCLICK:
1549             {
1550                 unsigned int idx = GetColumnIdxFromHeader(nmHDR);
1551                 wxDataViewModel * model = GetOwner()->GetModel();
1552
1553                 if(nmHDR->iButton == 0)
1554                 {
1555                     wxDataViewColumn *col = GetColumn(idx);
1556                     if(col->IsSortable())
1557                     {
1558                         if(model && m_owner->GetSortingColumn() == col)
1559                         {
1560                             bool order = col->IsSortOrderAscending();
1561                             col->SetSortOrder(!order);
1562                         }
1563                         else if(model)
1564                         {
1565                             m_owner->SetSortingColumn(col);
1566                         }
1567                     }
1568                     UpdateDisplay();
1569                     if(model)
1570                         model->Resort();
1571                 }
1572
1573                 wxEventType evt = nmHDR->iButton == 0 ?
1574                         wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_CLICK :
1575                         wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK;
1576                 SendEvent(evt, idx);
1577             }
1578             break;
1579
1580         case NM_RCLICK:
1581             {
1582                 // NOTE: for some reason (i.e. for a bug in Windows)
1583                 //       the HDN_ITEMCLICK notification is not sent on
1584                 //       right clicks, so we need to handle NM_RCLICK
1585
1586                 POINT ptClick;
1587                 int column = wxMSWGetColumnClicked(nmhdr, &ptClick);
1588                 if (column != wxNOT_FOUND)
1589                 {
1590                     HDITEM item;
1591                     item.mask = HDI_LPARAM;
1592                     Header_GetItem((HWND)m_hWnd, column, &item);
1593
1594                     // 'idx' may be different from 'column' if there are
1595                     // hidden columns...
1596                     unsigned int idx = (unsigned int)item.lParam;
1597                     SendEvent(wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK,
1598                               idx);
1599                 }
1600             }
1601             break;
1602
1603         case HDN_GETDISPINFOW:
1604             // see wxListCtrl::MSWOnNotify for more info!
1605             break;
1606
1607         case HDN_ITEMDBLCLICK:
1608             {
1609                 unsigned int idx = GetColumnIdxFromHeader(nmHDR);
1610                 int w = GetOwner()->GetBestColumnWidth(idx);
1611
1612                 // update the native control:
1613                 HDITEM hd;
1614                 ZeroMemory(&hd, sizeof(hd));
1615                 hd.mask = HDI_WIDTH;
1616                 hd.cxy = w;
1617                 Header_SetItem(GetHwnd(),
1618                                nmHDR->iItem,  // NOTE: we don't want 'idx' here!
1619                                &hd);
1620
1621                 // update the wxDataViewColumn class:
1622                 GetColumn(idx)->SetInternalWidth(w);
1623             }
1624             break;
1625
1626         default:
1627             return wxWindow::MSWOnNotify(idCtrl, lParam, result);
1628     }
1629
1630     return true;
1631 }
1632
1633 void wxDataViewHeaderWindowMSW::ScrollWindow(int dx, int WXUNUSED(dy),
1634                                              const wxRect * WXUNUSED(rect))
1635 {
1636     m_scrollOffsetX += dx;
1637
1638     GetParent()->Layout();
1639 }
1640
1641 void wxDataViewHeaderWindowMSW::DoSetSize(int x, int y,
1642                                           int w, int h,
1643                                           int f)
1644 {
1645     // TODO: why is there a border + 2px around it?
1646     wxControl::DoSetSize( x+m_scrollOffsetX+1, y+1, w-m_scrollOffsetX-2, h-2, f );
1647 }
1648
1649 #else       // !defined(__WXMSW__)
1650
1651 IMPLEMENT_ABSTRACT_CLASS(wxGenericDataViewHeaderWindow, wxWindow)
1652 BEGIN_EVENT_TABLE(wxGenericDataViewHeaderWindow, wxWindow)
1653     EVT_PAINT         (wxGenericDataViewHeaderWindow::OnPaint)
1654     EVT_MOUSE_EVENTS  (wxGenericDataViewHeaderWindow::OnMouse)
1655     EVT_SET_FOCUS     (wxGenericDataViewHeaderWindow::OnSetFocus)
1656 END_EVENT_TABLE()
1657
1658 bool wxGenericDataViewHeaderWindow::Create(wxDataViewCtrl *parent, wxWindowID id,
1659                                            const wxPoint &pos, const wxSize &size,
1660                                            const wxString &name )
1661 {
1662     m_owner = parent;
1663
1664     if (!wxDataViewHeaderWindowBase::Create(parent, id, pos, size, name))
1665         return false;
1666
1667     wxVisualAttributes attr = wxPanel::GetClassDefaultAttributes();
1668     SetBackgroundStyle( wxBG_STYLE_CUSTOM );
1669     SetOwnForegroundColour( attr.colFg );
1670     SetOwnBackgroundColour( attr.colBg );
1671     if (!m_hasFont)
1672         SetOwnFont( attr.font );
1673
1674     // set our size hints: wxDataViewCtrl will put this wxWindow inside
1675     // a wxBoxSizer and in order to avoid super-big header windows,
1676     // we need to set our height as fixed
1677     SetMinSize(wxSize(-1, HEADER_WINDOW_HEIGHT));
1678     SetMaxSize(wxSize(-1, HEADER_WINDOW_HEIGHT));
1679
1680     return true;
1681 }
1682
1683 void wxGenericDataViewHeaderWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
1684 {
1685     int w, h;
1686     GetClientSize( &w, &h );
1687
1688     wxAutoBufferedPaintDC dc( this );
1689
1690     dc.SetBackground(GetBackgroundColour());
1691     dc.Clear();
1692
1693     int xpix;
1694     m_owner->GetScrollPixelsPerUnit( &xpix, NULL );
1695
1696     int x;
1697     m_owner->GetViewStart( &x, NULL );
1698
1699     // account for the horz scrollbar offset
1700     dc.SetDeviceOrigin( -x * xpix, 0 );
1701
1702     dc.SetFont( GetFont() );
1703
1704     unsigned int cols = GetOwner()->GetColumnCount();
1705     unsigned int i;
1706     int xpos = 0;
1707     for (i = 0; i < cols; i++)
1708     {
1709         wxDataViewColumn *col = GetColumn( i );
1710         if (col->IsHidden())
1711             continue;      // skip it!
1712
1713         int cw = col->GetWidth();
1714         int ch = h;
1715
1716         wxHeaderSortIconType sortArrow = wxHDR_SORT_ICON_NONE;
1717         if (col->IsSortable() && GetOwner()->GetSortingColumn() == col)
1718         {
1719             if (col->IsSortOrderAscending())
1720                 sortArrow = wxHDR_SORT_ICON_UP;
1721             else
1722                 sortArrow = wxHDR_SORT_ICON_DOWN;
1723         }
1724
1725         int state = 0;
1726         if (m_parent->IsEnabled())
1727         {
1728             if ((int) i == m_hover)
1729                 state = wxCONTROL_CURRENT;
1730         }
1731         else
1732         {
1733             state = (int) wxCONTROL_DISABLED;
1734         }
1735
1736         wxRendererNative::Get().DrawHeaderButton
1737                                 (
1738                                     this,
1739                                     dc,
1740                                     wxRect(xpos, 0, cw, ch-1),
1741                                     state,
1742                                     sortArrow
1743                                 );
1744
1745         // align as required the column title:
1746         int x = xpos;
1747         wxSize titleSz = dc.GetTextExtent(col->GetTitle());
1748         switch (col->GetAlignment())
1749         {
1750         case wxALIGN_LEFT:
1751             x += HEADER_HORIZ_BORDER;
1752             break;
1753         case wxALIGN_RIGHT:
1754             x += cw - titleSz.GetWidth() - HEADER_HORIZ_BORDER;
1755             break;
1756         default:
1757         case wxALIGN_CENTER:
1758         case wxALIGN_CENTER_HORIZONTAL:
1759             x += (cw - titleSz.GetWidth() - 2 * HEADER_HORIZ_BORDER)/2;
1760             break;
1761         }
1762
1763         // always center the title vertically:
1764         int y = wxMax((ch - titleSz.GetHeight()) / 2, HEADER_VERT_BORDER);
1765
1766         dc.SetClippingRegion( xpos+HEADER_HORIZ_BORDER,
1767                               HEADER_VERT_BORDER,
1768                               wxMax(cw - 2 * HEADER_HORIZ_BORDER, 1),  // width
1769                               wxMax(ch - 2 * HEADER_VERT_BORDER, 1));  // height
1770         dc.DrawText( col->GetTitle(), x, y );
1771         dc.DestroyClippingRegion();
1772
1773         xpos += cw;
1774     }
1775 }
1776
1777 void wxGenericDataViewHeaderWindow::OnSetFocus( wxFocusEvent &event )
1778 {
1779     GetParent()->SetFocus();
1780     event.Skip();
1781 }
1782
1783 void wxGenericDataViewHeaderWindow::OnMouse( wxMouseEvent &event )
1784 {
1785     // we want to work with logical coords
1786     int x;
1787     m_owner->CalcUnscrolledPosition(event.GetX(), 0, &x, NULL);
1788     int y = event.GetY();
1789
1790     if (m_isDragging)
1791     {
1792         // we don't draw the line beyond our window,
1793         // but we allow dragging it there
1794         int w = 0;
1795         GetClientSize( &w, NULL );
1796         m_owner->CalcUnscrolledPosition(w, 0, &w, NULL);
1797         w -= 6;
1798
1799         if (event.ButtonUp())
1800         {
1801             m_isDragging = false;
1802             if (HasCapture())
1803                 ReleaseMouse();
1804
1805             m_dirty = true;
1806         }
1807         m_currentX = wxMax(m_minX + 7, x);
1808
1809         if (m_currentX < w)
1810         {
1811             GetColumn(m_column)->SetWidth(m_currentX - m_minX);
1812             Refresh();
1813             GetOwner()->Refresh();
1814         }
1815
1816     }
1817     else     // not dragging
1818     {
1819         m_minX = 0;
1820         m_column = wxNOT_FOUND;
1821
1822         bool hit_border = false;
1823
1824         // end of the current column
1825         int xpos = 0;
1826
1827         // find the column where this event occured
1828         int countCol = m_owner->GetColumnCount();
1829         for (int column = 0; column < countCol; column++)
1830         {
1831             wxDataViewColumn *p = GetColumn(column);
1832
1833             if (p->IsHidden())
1834                 continue;   // skip if not shown
1835
1836             xpos += p->GetWidth();
1837             m_column = column;
1838             if ((abs(x-xpos) < 3) && (y < 22))
1839             {
1840                 hit_border = true;
1841                 break;
1842             }
1843
1844             if (x < xpos)
1845             {
1846                 // inside the column
1847                 break;
1848             }
1849
1850             m_minX = xpos;
1851         }
1852
1853         int old_hover = m_hover;
1854         m_hover = m_column;
1855         if (event.Leaving())
1856             m_hover = wxNOT_FOUND;
1857         if (old_hover != m_hover)
1858             Refresh();
1859
1860         if (m_column == wxNOT_FOUND)
1861             return;
1862
1863         bool resizeable = GetColumn(m_column)->IsResizeable();
1864         if (event.LeftDClick() && resizeable)
1865         {
1866             GetColumn(m_column)->SetWidth(GetOwner()->GetBestColumnWidth(m_column));
1867             Refresh();
1868         }
1869         else if (event.LeftDown() || event.RightUp())
1870         {
1871             if (hit_border && event.LeftDown() && resizeable)
1872             {
1873                 m_isDragging = true;
1874                 CaptureMouse();
1875                 m_currentX = x;
1876             }
1877             else    // click on a column
1878             {
1879                 wxDataViewModel * model = GetOwner()->GetModel();
1880                 wxEventType evt = event.LeftDown() ?
1881                         wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_CLICK :
1882                         wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK;
1883                 SendEvent(evt, m_column);
1884
1885                 //Left click the header
1886                 if(event.LeftDown())
1887                 {
1888                     wxDataViewColumn *col = GetColumn(m_column);
1889                     if(col->IsSortable())
1890                     {
1891                         wxDataViewColumn* sortCol = m_owner->GetSortingColumn();
1892                         if(model && sortCol == col)
1893                         {
1894                             bool order = col->IsSortOrderAscending();
1895                             col->SetSortOrder(!order);
1896                         }
1897                         else if(model)
1898                         {
1899                             m_owner->SetSortingColumn(col);
1900                         }
1901                     }
1902                     UpdateDisplay();
1903                     if(model)
1904                         model->Resort();
1905                     //Send the column sorted event
1906                     SendEvent(wxEVT_COMMAND_DATAVIEW_COLUMN_SORTED, m_column);
1907                 }
1908             }
1909         }
1910         else if (event.Moving())
1911         {
1912             if (hit_border && resizeable)
1913                 m_currentCursor = m_resizeCursor;
1914             else
1915                 m_currentCursor = wxSTANDARD_CURSOR;
1916
1917             SetCursor(*m_currentCursor);
1918         }
1919     }
1920 }
1921
1922 void wxGenericDataViewHeaderWindow::AdjustDC(wxDC& dc)
1923 {
1924     int xpix, x;
1925
1926     m_owner->GetScrollPixelsPerUnit( &xpix, NULL );
1927     m_owner->GetViewStart( &x, NULL );
1928
1929     // shift the DC origin to match the position of the main window horizontal
1930     // scrollbar: this allows us to always use logical coords
1931     dc.SetDeviceOrigin( -x * xpix, 0 );
1932 }
1933
1934 #endif      // defined(__WXMSW__)
1935
1936 //-----------------------------------------------------------------------------
1937 // wxDataViewRenameTimer
1938 //-----------------------------------------------------------------------------
1939
1940 wxDataViewRenameTimer::wxDataViewRenameTimer( wxDataViewMainWindow *owner )
1941 {
1942     m_owner = owner;
1943 }
1944
1945 void wxDataViewRenameTimer::Notify()
1946 {
1947     m_owner->OnRenameTimer();
1948 }
1949
1950 //-----------------------------------------------------------------------------
1951 // wxDataViewMainWindow
1952 //-----------------------------------------------------------------------------
1953
1954 //The tree building helper, declared firstly
1955 static void BuildTreeHelper( wxDataViewModel * model,  wxDataViewItem & item, wxDataViewTreeNode * node);
1956
1957 int LINKAGEMODE wxDataViewSelectionCmp( unsigned int row1, unsigned int row2 )
1958 {
1959     if (row1 > row2) return 1;
1960     if (row1 == row2) return 0;
1961     return -1;
1962 }
1963
1964
1965 IMPLEMENT_ABSTRACT_CLASS(wxDataViewMainWindow, wxWindow)
1966
1967 BEGIN_EVENT_TABLE(wxDataViewMainWindow,wxWindow)
1968     EVT_PAINT         (wxDataViewMainWindow::OnPaint)
1969     EVT_MOUSE_EVENTS  (wxDataViewMainWindow::OnMouse)
1970     EVT_SET_FOCUS     (wxDataViewMainWindow::OnSetFocus)
1971     EVT_KILL_FOCUS    (wxDataViewMainWindow::OnKillFocus)
1972     EVT_CHAR          (wxDataViewMainWindow::OnChar)
1973 END_EVENT_TABLE()
1974
1975 wxDataViewMainWindow::wxDataViewMainWindow( wxDataViewCtrl *parent, wxWindowID id,
1976     const wxPoint &pos, const wxSize &size, const wxString &name ) :
1977     wxWindow( parent, id, pos, size, wxWANTS_CHARS|wxBORDER_NONE, name ),
1978     m_selection( wxDataViewSelectionCmp )
1979
1980 {
1981     SetOwner( parent );
1982
1983     m_lastOnSame = false;
1984     m_renameTimer = new wxDataViewRenameTimer( this );
1985
1986     // TODO: user better initial values/nothing selected
1987     m_currentCol = NULL;
1988     m_currentRow = 0;
1989
1990     m_lineHeight = wxMax( 17, GetCharHeight() + 2 ); // 17 = mini icon height + 1
1991
1992     m_dragCount = 0;
1993     m_dragStart = wxPoint(0,0);
1994     m_lineLastClicked = (unsigned int) -1;
1995     m_lineBeforeLastClicked = (unsigned int) -1;
1996     m_lineSelectSingleOnUp = (unsigned int) -1;
1997
1998     m_hasFocus = false;
1999
2000     SetBackgroundColour( *wxWHITE );
2001
2002     SetBackgroundStyle(wxBG_STYLE_CUSTOM);
2003
2004     m_penRule = wxPen(GetRuleColour());
2005
2006     //Here I compose a pen can draw black lines, maybe there are something system colour to use
2007     m_penExpander = wxPen(wxColour(0,0,0));
2008     //Some new added code to deal with the tree structure
2009     m_root = new wxDataViewTreeNode( NULL );
2010     m_root->SetHasChildren(true);
2011
2012     //Make m_count = -1 will cause the class recaculate the real displaying number of rows.
2013     m_count = -1 ;
2014     m_underMouse = NULL;
2015     UpdateDisplay();
2016 }
2017
2018 wxDataViewMainWindow::~wxDataViewMainWindow()
2019 {
2020     DestroyTree();
2021     delete m_renameTimer;
2022 }
2023
2024 void wxDataViewMainWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
2025 {
2026     wxDataViewModel *model = GetOwner()->GetModel();
2027     wxAutoBufferedPaintDC dc( this );
2028
2029 #ifdef __WXMSW__
2030     dc.SetPen( *wxTRANSPARENT_PEN );
2031     dc.SetBrush( wxBrush( GetBackgroundColour()) );
2032     dc.SetBrush( *wxWHITE_BRUSH );
2033     wxSize size( GetClientSize() );
2034     dc.DrawRectangle( 0,0,size.x,size.y );
2035 #endif
2036
2037     // prepare the DC
2038     GetOwner()->PrepareDC( dc );
2039     dc.SetFont( GetFont() );
2040
2041     wxRect update = GetUpdateRegion().GetBox();
2042     m_owner->CalcUnscrolledPosition( update.x, update.y, &update.x, &update.y );
2043
2044     // compute which items needs to be redrawn
2045     unsigned int item_start = wxMax( 0, (update.y / m_lineHeight) );
2046     unsigned int item_count =
2047         wxMin( (int)(((update.y + update.height) / m_lineHeight) - item_start + 1),
2048                (int)(GetRowCount( ) - item_start));
2049     unsigned int item_last = item_start + item_count;
2050
2051     // compute which columns needs to be redrawn
2052     unsigned int cols = GetOwner()->GetColumnCount();
2053     unsigned int col_start = 0;
2054     unsigned int x_start = 0;
2055     for (x_start = 0; col_start < cols; col_start++)
2056     {
2057         wxDataViewColumn *col = GetOwner()->GetColumn(col_start);
2058         if (col->IsHidden())
2059             continue;      // skip it!
2060
2061         unsigned int w = col->GetWidth();
2062         if (x_start+w >= (unsigned int)update.x)
2063             break;
2064
2065         x_start += w;
2066     }
2067
2068     unsigned int col_last = col_start;
2069     unsigned int x_last = x_start;
2070     for (; col_last < cols; col_last++)
2071     {
2072         wxDataViewColumn *col = GetOwner()->GetColumn(col_last);
2073         if (col->IsHidden())
2074             continue;      // skip it!
2075
2076         if (x_last > (unsigned int)update.GetRight())
2077             break;
2078
2079         x_last += col->GetWidth();
2080     }
2081
2082     // Draw horizontal rules if required
2083     if ( m_owner->HasFlag(wxDV_HORIZ_RULES) )
2084     {
2085         dc.SetPen(m_penRule);
2086         dc.SetBrush(*wxTRANSPARENT_BRUSH);
2087
2088         for (unsigned int i = item_start; i <= item_last+1; i++)
2089         {
2090             int y = i * m_lineHeight;
2091             dc.DrawLine(x_start, y, x_last, y);
2092         }
2093     }
2094
2095     // Draw vertical rules if required
2096     if ( m_owner->HasFlag(wxDV_VERT_RULES) )
2097     {
2098         dc.SetPen(m_penRule);
2099         dc.SetBrush(*wxTRANSPARENT_BRUSH);
2100
2101         int x = x_start;
2102         for (unsigned int i = col_start; i < col_last; i++)
2103         {
2104             wxDataViewColumn *col = GetOwner()->GetColumn(i);
2105             if (col->IsHidden())
2106                 continue;       // skip it
2107
2108             dc.DrawLine(x, item_start * m_lineHeight,
2109                         x, item_last * m_lineHeight);
2110
2111             x += col->GetWidth();
2112         }
2113
2114         // Draw last vertical rule
2115         dc.DrawLine(x, item_start * m_lineHeight,
2116                     x, item_last * m_lineHeight);
2117     }
2118
2119     // redraw the background for the items which are selected/current
2120     for (unsigned int item = item_start; item < item_last; item++)
2121     {
2122         bool selected = m_selection.Index( item ) != wxNOT_FOUND;
2123         if (selected || item == m_currentRow)
2124         {
2125             int flags = selected ? (int)wxCONTROL_SELECTED : 0;
2126             if (item == m_currentRow)
2127                 flags |= wxCONTROL_CURRENT;
2128             if (m_hasFocus)
2129                 flags |= wxCONTROL_FOCUSED;
2130
2131             wxRect rect( x_start, item*m_lineHeight, x_last, m_lineHeight );
2132             wxRendererNative::Get().DrawItemSelectionRect
2133                                 (
2134                                     this,
2135                                     dc,
2136                                     rect,
2137                                     flags
2138                                 );
2139         }
2140     }
2141
2142     wxDataViewColumn *expander = GetOwner()->GetExpanderColumn();
2143     if (!expander)
2144     {
2145         // TODO: last column for RTL support
2146         expander = GetOwner()->GetColumn( 0 );
2147         GetOwner()->SetExpanderColumn(expander);
2148     }
2149
2150     // redraw all cells for all rows which must be repainted and for all columns
2151     wxRect cell_rect;
2152     cell_rect.x = x_start;
2153     cell_rect.height = m_lineHeight;        // -1 is for the horizontal rules
2154     for (unsigned int i = col_start; i < col_last; i++)
2155     {
2156         wxDataViewColumn *col = GetOwner()->GetColumn( i );
2157         wxDataViewRenderer *cell = col->GetRenderer();
2158         cell_rect.width = col->GetWidth();
2159
2160         if (col->IsHidden())
2161             continue;       // skipt it!
2162
2163
2164         for (unsigned int item = item_start; item < item_last; item++)
2165         {
2166             // get the cell value and set it into the renderer
2167             wxVariant value;
2168             wxDataViewTreeNode *node = NULL;
2169             wxDataViewItem dataitem;
2170
2171             if (m_root)
2172             {
2173                 node = GetTreeNodeByRow(item);
2174                 if( node == NULL )
2175                     continue;
2176
2177                 dataitem = node->GetItem();
2178
2179                 if ((i > 0) && model->IsContainer(dataitem) && !model->HasContainerColumns(dataitem))
2180                     continue;
2181             }
2182             else
2183             {
2184                 dataitem = wxDataViewItem( (void*) item );
2185             }
2186
2187             model->GetValue( value, dataitem, col->GetModelColumn());
2188             cell->SetValue( value );
2189
2190             if (cell->GetWantsAttr())
2191             {
2192                 wxDataViewItemAttr attr;
2193                 bool ret = model->GetAttr( dataitem, col->GetModelColumn(), attr );
2194                 if (ret)
2195                     cell->SetAttr( attr );
2196                 cell->SetHasAttr( ret );
2197             }
2198
2199             // update the y offset
2200             cell_rect.y = item * m_lineHeight;
2201
2202             //Draw the expander here.
2203             int indent = 0;
2204             if ((m_root) && (col == expander))
2205             {
2206                 indent = node->GetIndentLevel();
2207
2208                 //Calculate the indent first
2209                 indent = cell_rect.x + GetOwner()->GetIndent() * indent;
2210
2211                 int expander_width = m_lineHeight - 2*EXPANDER_MARGIN;
2212                 // change the cell_rect.x to the appropriate pos
2213                 int  expander_x = indent + EXPANDER_MARGIN , expander_y = cell_rect.y + EXPANDER_MARGIN ;
2214                 indent = indent + m_lineHeight ;  //try to use the m_lineHeight as the expander space
2215                 dc.SetPen( m_penExpander );
2216                 dc.SetBrush( wxNullBrush );
2217                 if( node->HasChildren() )
2218                 {
2219                     wxRect rect( expander_x , expander_y, expander_width, expander_width);
2220                     int flag = 0;
2221                     if (m_underMouse == node)
2222                     {
2223                         flag |= wxCONTROL_CURRENT;
2224                     }
2225                     if( node->IsOpen() )
2226                         wxRendererNative::Get().DrawTreeItemButton( this, dc, rect, flag|wxCONTROL_EXPANDED );
2227                     else
2228                         wxRendererNative::Get().DrawTreeItemButton( this, dc, rect, flag);
2229                 }
2230                  //force the expander column to left-center align
2231                  cell->SetAlignment( wxALIGN_CENTER_VERTICAL );
2232             }
2233             if (node && !node->HasChildren())
2234             {
2235                 // Yes, if the node does not have any child, it must be a leaf which
2236                 // mean that it is a temporarily created by GetTreeNodeByRow
2237                 wxDELETE(node)
2238             }
2239
2240             // cannot be bigger than allocated space
2241             wxSize size = cell->GetSize();
2242             // Because of the tree structure indent, here we should minus the width of the cell for drawing
2243             size.x = wxMin( size.x + 2*PADDING_RIGHTLEFT, cell_rect.width - indent );
2244             // size.y = wxMin( size.y, cell_rect.height );
2245             size.y = cell_rect.height;
2246
2247             wxRect item_rect(cell_rect.GetTopLeft(), size);
2248             int align = cell->CalculateAlignment();
2249
2250             // horizontal alignment:
2251             item_rect.x = cell_rect.x;
2252             if (align & wxALIGN_CENTER_HORIZONTAL)
2253                 item_rect.x = cell_rect.x + (cell_rect.width / 2) - (size.x / 2);
2254             else if (align & wxALIGN_RIGHT)
2255                 item_rect.x = cell_rect.x + cell_rect.width - size.x;
2256             //else: wxALIGN_LEFT is the default
2257
2258             // vertical alignment:
2259             item_rect.y = cell_rect.y;
2260             if (align & wxALIGN_CENTER_VERTICAL)
2261                 item_rect.y = cell_rect.y + (cell_rect.height / 2) - (size.y / 2);
2262             else if (align & wxALIGN_BOTTOM)
2263                 item_rect.y = cell_rect.y + cell_rect.height - size.y;
2264             //else: wxALIGN_TOP is the default
2265
2266             // add padding
2267             item_rect.x += PADDING_RIGHTLEFT;
2268             item_rect.width = size.x - 2 * PADDING_RIGHTLEFT;
2269
2270             //Here we add the tree indent
2271             item_rect.x += indent;
2272
2273             int state = 0;
2274             if (m_hasFocus && (m_selection.Index(item) != wxNOT_FOUND))
2275                 state |= wxDATAVIEW_CELL_SELECTED;
2276
2277             // TODO: it would be much more efficient to create a clipping
2278             //       region for the entire column being rendered (in the OnPaint
2279             //       of wxDataViewMainWindow) instead of a single clip region for
2280             //       each cell. However it would mean that each renderer should
2281             //       respect the given wxRect's top & bottom coords, eventually
2282             //       violating only the left & right coords - however the user can
2283             //       make its own renderer and thus we cannot be sure of that.
2284             dc.SetClippingRegion( item_rect );
2285             cell->Render( item_rect, &dc, state );
2286             dc.DestroyClippingRegion();
2287         }
2288
2289         cell_rect.x += cell_rect.width;
2290     }
2291 }
2292
2293 void wxDataViewMainWindow::OnRenameTimer()
2294 {
2295     // We have to call this here because changes may just have
2296     // been made and no screen update taken place.
2297     if ( m_dirty )
2298         wxSafeYield();
2299
2300     int xpos = 0;
2301     unsigned int cols = GetOwner()->GetColumnCount();
2302     unsigned int i;
2303     for (i = 0; i < cols; i++)
2304     {
2305         wxDataViewColumn *c = GetOwner()->GetColumn( i );
2306         if (c->IsHidden())
2307             continue;      // skip it!
2308
2309         if (c == m_currentCol)
2310             break;
2311         xpos += c->GetWidth();
2312     }
2313     wxRect labelRect( xpos, m_currentRow * m_lineHeight,
2314                       m_currentCol->GetWidth(), m_lineHeight );
2315
2316     GetOwner()->CalcScrolledPosition( labelRect.x, labelRect.y,
2317                                      &labelRect.x, &labelRect.y);
2318
2319     wxDataViewItem item = GetItemByRow( m_currentRow );
2320     m_currentCol->GetRenderer()->StartEditing( item, labelRect );
2321
2322 }
2323
2324 //------------------------------------------------------------------
2325 // Helper class for do operation on the tree node
2326 //------------------------------------------------------------------
2327 class DoJob
2328 {
2329 public:
2330     DoJob(){};
2331     virtual ~DoJob(){};
2332
2333     //The return value control how the tree-walker tranverse the tree
2334     // 0: Job done, stop tranverse and return
2335     // 1: Ignore the current node's subtree and continue
2336     // 2: Job not done, continue
2337     enum  { OK = 0 , IGR = 1, CONT = 2 };
2338     virtual int operator() ( wxDataViewTreeNode * node ) = 0 ;
2339     virtual int operator() ( void * n ) = 0;
2340 };
2341
2342 bool Walker( wxDataViewTreeNode * node, DoJob & func )
2343 {
2344     if( node==NULL )
2345         return false;
2346
2347     switch( func( node ) )
2348     {
2349         case DoJob::OK :
2350             return true ;
2351         case DoJob::IGR:
2352             return false;
2353         case DoJob::CONT:
2354             default:
2355                 ;
2356     }
2357
2358     wxDataViewTreeNodes nodes = node->GetNodes();
2359     wxDataViewTreeLeaves leaves = node->GetChildren();
2360
2361     int len_nodes = nodes.GetCount();
2362     int len = leaves.GetCount();
2363     int i = 0, nodes_i = 0;
2364
2365     for( ; i < len ; i ++ )
2366     {
2367         void * n = leaves[i];
2368         if( nodes_i < len_nodes && n == nodes[nodes_i]->GetItem().GetID() )
2369         {
2370             wxDataViewTreeNode * nd = nodes[nodes_i];
2371             nodes_i++;
2372
2373             if( Walker( nd , func ) )
2374                 return true;
2375
2376         }
2377         else
2378             switch( func( n ) )
2379             {
2380                 case DoJob::OK :
2381                     return true ;
2382                 case DoJob::IGR:
2383                     continue;
2384                 case DoJob::CONT:
2385                 default:
2386                     ;
2387             }
2388     }
2389     return false;
2390 }
2391
2392 bool wxDataViewMainWindow::ItemAdded(const wxDataViewItem & parent, const wxDataViewItem & item)
2393 {
2394     if (!m_root)
2395     {
2396         m_count++;
2397         UpdateDisplay();
2398         return true;
2399     }
2400
2401     SortPrepare();
2402
2403     wxDataViewTreeNode * node;
2404     node = FindNode(parent);
2405
2406     if( node == NULL )
2407         return false;
2408
2409     node->SetHasChildren( true );
2410
2411     if( g_model->IsContainer( item ) )
2412     {
2413         wxDataViewTreeNode * newnode = new wxDataViewTreeNode( node );
2414         newnode->SetItem(item);
2415         newnode->SetHasChildren( true );
2416         node->AddNode( newnode);
2417     }
2418     else
2419         node->AddLeaf( item.GetID() );
2420
2421     node->ChangeSubTreeCount(1);
2422
2423     m_count = -1;
2424     UpdateDisplay();
2425
2426     return true;
2427 }
2428
2429 static void DestroyTreeHelper( wxDataViewTreeNode * node);
2430
2431 bool wxDataViewMainWindow::ItemDeleted(const wxDataViewItem& parent,
2432                                        const wxDataViewItem& item)
2433 {
2434     if (!m_root)
2435     {
2436         m_count--;
2437         if( m_currentRow > GetRowCount() )
2438             m_currentRow = m_count - 1;
2439
2440         m_selection.Empty();
2441
2442         UpdateDisplay();
2443
2444         return true;
2445     }
2446
2447     wxDataViewTreeNode * node = FindNode(parent);
2448
2449     wxCHECK_MSG( node != NULL, false, "item not found" );
2450     wxCHECK_MSG( node->GetChildren().Index( item.GetID() ) != wxNOT_FOUND, false, "item not found" );
2451
2452     int sub = -1;
2453     node->GetChildren().Remove( item.GetID() );
2454     //Manuplate selection
2455     if( m_selection.GetCount() > 1 )
2456     {
2457         m_selection.Empty();
2458     }
2459     bool isContainer = false;
2460     wxDataViewTreeNodes nds = node->GetNodes();
2461     for (size_t i = 0; i < nds.GetCount(); i ++)
2462     {
2463         if (nds[i]->GetItem() == item)
2464         {
2465             isContainer = true;
2466             break;
2467         }
2468     }
2469     if( isContainer )
2470     {
2471         wxDataViewTreeNode * n = NULL;
2472         wxDataViewTreeNodes nodes = node->GetNodes();
2473         int len = nodes.GetCount();
2474         for( int i = 0 ; i < len; i ++)
2475         {
2476             if( nodes[i]->GetItem() == item )
2477             {
2478                 n = nodes[i];
2479                 break;
2480             }
2481         }
2482
2483         wxCHECK_MSG( n != NULL, false, "item not found" );
2484
2485         node->GetNodes().Remove( n );
2486         sub -= n->GetSubTreeCount();
2487         ::DestroyTreeHelper(n);
2488     }
2489     //Make the row number invalid and get a new valid one when user call GetRowCount
2490     m_count = -1;
2491     node->ChangeSubTreeCount(sub);
2492
2493     //Change the current row to the last row if the current exceed the max row number
2494     if( m_currentRow > GetRowCount() )
2495         m_currentRow = m_count - 1;
2496
2497     UpdateDisplay();
2498
2499     return true;
2500 }
2501
2502 bool wxDataViewMainWindow::ItemChanged(const wxDataViewItem & item)
2503 {
2504     SortPrepare();
2505     g_model->Resort();
2506
2507     //Send event
2508     wxWindow *parent = GetParent();
2509     wxDataViewEvent le(wxEVT_COMMAND_DATAVIEW_ITEM_VALUE_CHANGED, parent->GetId());
2510     le.SetEventObject(parent);
2511     le.SetModel(GetOwner()->GetModel());
2512     le.SetItem(item);
2513     parent->GetEventHandler()->ProcessEvent(le);
2514
2515     return true;
2516 }
2517
2518 bool wxDataViewMainWindow::ValueChanged( const wxDataViewItem & item, unsigned int col )
2519 {
2520     // NOTE: to be valid, we cannot use e.g. INT_MAX - 1
2521 /*#define MAX_VIRTUAL_WIDTH       100000
2522
2523     wxRect rect( 0, row*m_lineHeight, MAX_VIRTUAL_WIDTH, m_lineHeight );
2524     m_owner->CalcScrolledPosition( rect.x, rect.y, &rect.x, &rect.y );
2525     Refresh( true, &rect );
2526
2527     return true;
2528 */
2529     SortPrepare();
2530     g_model->Resort();
2531
2532     //Send event
2533     wxWindow *parent = GetParent();
2534     wxDataViewEvent le(wxEVT_COMMAND_DATAVIEW_ITEM_VALUE_CHANGED, parent->GetId());
2535     le.SetEventObject(parent);
2536     le.SetModel(GetOwner()->GetModel());
2537     le.SetItem(item);
2538     le.SetColumn(col);
2539     le.SetDataViewColumn(GetOwner()->GetColumn(col));
2540     parent->GetEventHandler()->ProcessEvent(le);
2541
2542     return true;
2543 }
2544
2545 bool wxDataViewMainWindow::Cleared()
2546 {
2547     DestroyTree();
2548
2549     SortPrepare();
2550     BuildTree( GetOwner()->GetModel() );
2551
2552     UpdateDisplay();
2553
2554     return true;
2555 }
2556
2557 void wxDataViewMainWindow::UpdateDisplay()
2558 {
2559     m_dirty = true;
2560 }
2561
2562 void wxDataViewMainWindow::OnInternalIdle()
2563 {
2564     wxWindow::OnInternalIdle();
2565
2566     if (m_dirty)
2567     {
2568         RecalculateDisplay();
2569         m_dirty = false;
2570     }
2571 }
2572
2573 void wxDataViewMainWindow::RecalculateDisplay()
2574 {
2575     wxDataViewModel *model = GetOwner()->GetModel();
2576     if (!model)
2577     {
2578         Refresh();
2579         return;
2580     }
2581
2582     int width = GetEndOfLastCol();
2583     int height = GetRowCount() * m_lineHeight;
2584
2585     SetVirtualSize( width, height );
2586     GetOwner()->SetScrollRate( 10, m_lineHeight );
2587
2588     Refresh();
2589 }
2590
2591 void wxDataViewMainWindow::ScrollWindow( int dx, int dy, const wxRect *rect )
2592 {
2593     wxWindow::ScrollWindow( dx, dy, rect );
2594
2595     if (GetOwner()->m_headerArea)
2596         GetOwner()->m_headerArea->ScrollWindow( dx, 0 );
2597 }
2598
2599 void wxDataViewMainWindow::ScrollTo( int rows, int column )
2600 {
2601     int x, y;
2602     m_owner->GetScrollPixelsPerUnit( &x, &y );
2603     int sy = rows*m_lineHeight/y;
2604     int sx = 0;
2605     if( column != -1 )
2606     {
2607         wxRect rect = GetClientRect();
2608         int colnum = 0;
2609         int x_start = 0, x_end = 0, w = 0;
2610         int xx, yy, xe;
2611         m_owner->CalcUnscrolledPosition( rect.x, rect.y, &xx, &yy );
2612         for (x_start = 0; colnum < column; colnum++)
2613         {
2614             wxDataViewColumn *col = GetOwner()->GetColumn(colnum);
2615             if (col->IsHidden())
2616                 continue;      // skip it!
2617
2618             w = col->GetWidth();
2619             x_start += w;
2620         }
2621
2622         x_end = x_start + w;
2623         xe = xx + rect.width;
2624         if( x_end > xe )
2625         {
2626             sx = ( xx + x_end - xe )/x;
2627         }
2628         if( x_start < xx )
2629         {
2630             sx = x_start/x;
2631         }
2632     }
2633     m_owner->Scroll( sx, sy );
2634 }
2635
2636 int wxDataViewMainWindow::GetCountPerPage() const
2637 {
2638     wxSize size = GetClientSize();
2639     return size.y / m_lineHeight;
2640 }
2641
2642 int wxDataViewMainWindow::GetEndOfLastCol() const
2643 {
2644     int width = 0;
2645     unsigned int i;
2646     for (i = 0; i < GetOwner()->GetColumnCount(); i++)
2647     {
2648         const wxDataViewColumn *c =
2649             wx_const_cast(wxDataViewCtrl*, GetOwner())->GetColumn( i );
2650
2651         if (!c->IsHidden())
2652             width += c->GetWidth();
2653     }
2654     return width;
2655 }
2656
2657 unsigned int wxDataViewMainWindow::GetFirstVisibleRow() const
2658 {
2659     int x = 0;
2660     int y = 0;
2661     m_owner->CalcUnscrolledPosition( x, y, &x, &y );
2662
2663     return y / m_lineHeight;
2664 }
2665
2666 unsigned int wxDataViewMainWindow::GetLastVisibleRow()
2667 {
2668     wxSize client_size = GetClientSize();
2669     m_owner->CalcUnscrolledPosition( client_size.x, client_size.y,
2670                                      &client_size.x, &client_size.y );
2671
2672     //we should deal with the pixel here
2673     unsigned int row = ((client_size.y)/m_lineHeight) - 1;
2674
2675     return wxMin( GetRowCount()-1, row );
2676 }
2677
2678 unsigned int wxDataViewMainWindow::GetRowCount()
2679 {
2680     if ( m_count == -1 )
2681     {
2682         m_count = RecalculateCount();
2683         int width, height;
2684         GetVirtualSize( &width, &height );
2685         height = m_count * m_lineHeight;
2686
2687         SetVirtualSize( width, height );
2688     }
2689     return m_count;
2690 }
2691
2692 void wxDataViewMainWindow::ChangeCurrentRow( unsigned int row )
2693 {
2694     m_currentRow = row;
2695
2696     // send event
2697 }
2698
2699 void wxDataViewMainWindow::SelectAllRows( bool on )
2700 {
2701     if (IsEmpty())
2702         return;
2703
2704     if (on)
2705     {
2706         m_selection.Clear();
2707         for (unsigned int i = 0; i < GetRowCount(); i++)
2708             m_selection.Add( i );
2709         Refresh();
2710     }
2711     else
2712     {
2713         unsigned int first_visible = GetFirstVisibleRow();
2714         unsigned int last_visible = GetLastVisibleRow();
2715         unsigned int i;
2716         for (i = 0; i < m_selection.GetCount(); i++)
2717         {
2718             unsigned int row = m_selection[i];
2719             if ((row >= first_visible) && (row <= last_visible))
2720                 RefreshRow( row );
2721         }
2722         m_selection.Clear();
2723     }
2724 }
2725
2726 void wxDataViewMainWindow::SelectRow( unsigned int row, bool on )
2727 {
2728     if (m_selection.Index( row ) == wxNOT_FOUND)
2729     {
2730         if (on)
2731         {
2732             m_selection.Add( row );
2733             RefreshRow( row );
2734         }
2735     }
2736     else
2737     {
2738         if (!on)
2739         {
2740             m_selection.Remove( row );
2741             RefreshRow( row );
2742         }
2743     }
2744 }
2745
2746 void wxDataViewMainWindow::SelectRows( unsigned int from, unsigned int to, bool on )
2747 {
2748     if (from > to)
2749     {
2750         unsigned int tmp = from;
2751         from = to;
2752         to = tmp;
2753     }
2754
2755     unsigned int i;
2756     for (i = from; i <= to; i++)
2757     {
2758         if (m_selection.Index( i ) == wxNOT_FOUND)
2759         {
2760             if (on)
2761                 m_selection.Add( i );
2762         }
2763         else
2764         {
2765             if (!on)
2766                 m_selection.Remove( i );
2767         }
2768     }
2769     RefreshRows( from, to );
2770 }
2771
2772 void wxDataViewMainWindow::Select( const wxArrayInt& aSelections )
2773 {
2774     for (size_t i=0; i < aSelections.GetCount(); i++)
2775     {
2776         int n = aSelections[i];
2777
2778         m_selection.Add( n );
2779         RefreshRow( n );
2780     }
2781 }
2782
2783 void wxDataViewMainWindow::ReverseRowSelection( unsigned int row )
2784 {
2785     if (m_selection.Index( row ) == wxNOT_FOUND)
2786         m_selection.Add( row );
2787     else
2788         m_selection.Remove( row );
2789     RefreshRow( row );
2790 }
2791
2792 bool wxDataViewMainWindow::IsRowSelected( unsigned int row )
2793 {
2794     return (m_selection.Index( row ) != wxNOT_FOUND);
2795 }
2796
2797 void wxDataViewMainWindow::SendSelectionChangedEvent( const wxDataViewItem& item)
2798 {
2799     wxWindow *parent = GetParent();
2800     wxDataViewEvent le(wxEVT_COMMAND_DATAVIEW_SELECTION_CHANGED, parent->GetId());
2801
2802     le.SetEventObject(parent);
2803     le.SetModel(GetOwner()->GetModel());
2804     le.SetItem( item );
2805
2806     parent->GetEventHandler()->ProcessEvent(le);
2807 }
2808
2809 void wxDataViewMainWindow::RefreshRow( unsigned int row )
2810 {
2811     wxRect rect( 0, row*m_lineHeight, GetEndOfLastCol(), m_lineHeight );
2812     m_owner->CalcScrolledPosition( rect.x, rect.y, &rect.x, &rect.y );
2813
2814     wxSize client_size = GetClientSize();
2815     wxRect client_rect( 0, 0, client_size.x, client_size.y );
2816     wxRect intersect_rect = client_rect.Intersect( rect );
2817     if (intersect_rect.width > 0)
2818         Refresh( true, &intersect_rect );
2819 }
2820
2821 void wxDataViewMainWindow::RefreshRows( unsigned int from, unsigned int to )
2822 {
2823     if (from > to)
2824     {
2825         unsigned int tmp = to;
2826         to = from;
2827         from = tmp;
2828     }
2829
2830     wxRect rect( 0, from*m_lineHeight, GetEndOfLastCol(), (to-from+1) * m_lineHeight );
2831     m_owner->CalcScrolledPosition( rect.x, rect.y, &rect.x, &rect.y );
2832
2833     wxSize client_size = GetClientSize();
2834     wxRect client_rect( 0, 0, client_size.x, client_size.y );
2835     wxRect intersect_rect = client_rect.Intersect( rect );
2836     if (intersect_rect.width > 0)
2837         Refresh( true, &intersect_rect );
2838 }
2839
2840 void wxDataViewMainWindow::RefreshRowsAfter( unsigned int firstRow )
2841 {
2842     unsigned int count = GetRowCount();
2843     if (firstRow > count)
2844         return;
2845
2846     wxRect rect( 0, firstRow*m_lineHeight, GetEndOfLastCol(), count * m_lineHeight );
2847     m_owner->CalcScrolledPosition( rect.x, rect.y, &rect.x, &rect.y );
2848
2849     wxSize client_size = GetClientSize();
2850     wxRect client_rect( 0, 0, client_size.x, client_size.y );
2851     wxRect intersect_rect = client_rect.Intersect( rect );
2852     if (intersect_rect.width > 0)
2853         Refresh( true, &intersect_rect );
2854 }
2855
2856 void wxDataViewMainWindow::OnArrowChar(unsigned int newCurrent, const wxKeyEvent& event)
2857 {
2858     wxCHECK_RET( newCurrent < GetRowCount(),
2859                  _T("invalid item index in OnArrowChar()") );
2860
2861     // if there is no selection, we cannot move it anywhere
2862     if (!HasCurrentRow())
2863         return;
2864
2865     unsigned int oldCurrent = m_currentRow;
2866
2867     // in single selection we just ignore Shift as we can't select several
2868     // items anyhow
2869     if ( event.ShiftDown() && !IsSingleSel() )
2870     {
2871         RefreshRow( oldCurrent );
2872
2873         ChangeCurrentRow( newCurrent );
2874
2875         // select all the items between the old and the new one
2876         if ( oldCurrent > newCurrent )
2877         {
2878             newCurrent = oldCurrent;
2879             oldCurrent = m_currentRow;
2880         }
2881
2882         SelectRows( oldCurrent, newCurrent, true );
2883         if (oldCurrent!=newCurrent)
2884             SendSelectionChangedEvent(GetItemByRow(m_selection[0]));
2885     }
2886     else // !shift
2887     {
2888         RefreshRow( oldCurrent );
2889
2890         // all previously selected items are unselected unless ctrl is held
2891         if ( !event.ControlDown() )
2892             SelectAllRows(false);
2893
2894         ChangeCurrentRow( newCurrent );
2895
2896         if ( !event.ControlDown() )
2897         {
2898             SelectRow( m_currentRow, true );
2899             SendSelectionChangedEvent(GetItemByRow(m_currentRow));
2900         }
2901         else
2902             RefreshRow( m_currentRow );
2903     }
2904
2905     GetOwner()->EnsureVisible( m_currentRow, -1 );
2906 }
2907
2908 wxRect wxDataViewMainWindow::GetLineRect( unsigned int row ) const
2909 {
2910     wxRect rect;
2911     rect.x = 0;
2912     rect.y = m_lineHeight * row;
2913     rect.width = GetEndOfLastCol();
2914     rect.height = m_lineHeight;
2915
2916     return rect;
2917 }
2918
2919 class RowToItemJob: public DoJob
2920 {
2921 public:
2922     RowToItemJob( unsigned int row , int current ) { this->row = row; this->current = current ;}
2923     virtual ~RowToItemJob(){};
2924
2925     virtual int operator() ( wxDataViewTreeNode * node )
2926     {
2927         current ++;
2928         if( current == static_cast<int>(row))
2929          {
2930             ret = node->GetItem() ;
2931             return DoJob::OK;
2932         }
2933
2934         if( node->GetSubTreeCount() + current < static_cast<int>(row) )
2935         {
2936             current += node->GetSubTreeCount();
2937             return  DoJob::IGR;
2938         }
2939         else
2940         {
2941             //If the current has no child node, we can find the desired item of the row number directly.
2942             //This if can speed up finding in some case, and will has a very good effect when it comes to list view
2943             if( node->GetNodes().GetCount() == 0)
2944             {
2945                 int index = static_cast<int>(row) - current - 1;
2946                 ret = node->GetChildren().Item( index );
2947                 return DoJob::OK;
2948             }
2949             return DoJob::CONT;
2950         }
2951     }
2952
2953     virtual int operator() ( void * n )
2954     {
2955         current ++;
2956         if( current == static_cast<int>(row))
2957          {
2958             ret = wxDataViewItem( n ) ;
2959             return DoJob::OK;
2960         }
2961         return DoJob::CONT;
2962     }
2963     wxDataViewItem GetResult(){ return ret; }
2964 private:
2965     unsigned int row;
2966     int current ;
2967     wxDataViewItem ret;
2968 };
2969
2970 wxDataViewItem wxDataViewMainWindow::GetItemByRow(unsigned int row) const
2971 {
2972     if (!m_root)
2973     {
2974         return wxDataViewItem( (void*) row );
2975     }
2976     else
2977     {
2978         RowToItemJob job( row, -2 );
2979         Walker( m_root , job );
2980         return job.GetResult();
2981     }
2982 }
2983
2984 class RowToTreeNodeJob: public DoJob
2985 {
2986 public:
2987     RowToTreeNodeJob( unsigned int row , int current, wxDataViewTreeNode * node )
2988     {
2989         this->row = row;
2990         this->current = current ;
2991         ret = NULL ;
2992         parent = node;
2993     }
2994     virtual ~RowToTreeNodeJob(){};
2995
2996     virtual int operator() ( wxDataViewTreeNode * node )
2997     {
2998         current ++;
2999         if( current == static_cast<int>(row))
3000          {
3001             ret = node ;
3002             return DoJob::OK;
3003         }
3004
3005         if( node->GetSubTreeCount() + current < static_cast<int>(row) )
3006         {
3007             current += node->GetSubTreeCount();
3008             return  DoJob::IGR;
3009         }
3010         else
3011         {
3012             parent = node;
3013             //If the current has no child node, we can find the desired item of the row number directly.
3014             //This if can speed up finding in some case, and will has a very good effect when it comes to list view
3015             if( node->GetNodes().GetCount() == 0)
3016             {
3017                 int index = static_cast<int>(row) - current - 1;
3018                 void * n = node->GetChildren().Item( index );
3019                 ret = new wxDataViewTreeNode( parent ) ;
3020                 ret->SetItem( wxDataViewItem( n ));
3021                 ret->SetHasChildren(false);
3022                 return DoJob::OK;
3023             }
3024             return DoJob::CONT;
3025         }
3026
3027
3028     }
3029
3030     virtual int operator() ( void * n )
3031     {
3032         current ++;
3033         if( current == static_cast<int>(row))
3034          {
3035             ret = new wxDataViewTreeNode( parent ) ;
3036             ret->SetItem( wxDataViewItem( n ));
3037             ret->SetHasChildren(false);
3038             return DoJob::OK;
3039         }
3040
3041         return DoJob::CONT;
3042     }
3043     wxDataViewTreeNode * GetResult(){ return ret; }
3044 private:
3045     unsigned int row;
3046     int current ;
3047     wxDataViewTreeNode * ret;
3048     wxDataViewTreeNode * parent ;
3049 };
3050
3051
3052 wxDataViewTreeNode * wxDataViewMainWindow::GetTreeNodeByRow(unsigned int row)
3053 {
3054     if (!m_root)
3055     {
3056         return NULL;
3057     }
3058     else
3059     {
3060         RowToTreeNodeJob job( row , -2, m_root );
3061         Walker( m_root , job );
3062         return job.GetResult();
3063     }
3064 }
3065
3066 wxDataViewEvent wxDataViewMainWindow::SendExpanderEvent( wxEventType type, const wxDataViewItem & item )
3067 {
3068     wxWindow *parent = GetParent();
3069     wxDataViewEvent le(type, parent->GetId());
3070
3071     le.SetEventObject(parent);
3072     le.SetModel(GetOwner()->GetModel());
3073     le.SetItem( item );
3074
3075     parent->GetEventHandler()->ProcessEvent(le);
3076     return le;
3077 }
3078
3079 void wxDataViewMainWindow::OnExpanding( unsigned int row )
3080 {
3081     wxDataViewTreeNode * node = GetTreeNodeByRow(row);
3082     if( node != NULL )
3083     {
3084         if( node->HasChildren())
3085         {
3086             if( !node->IsOpen())
3087             {
3088                wxDataViewEvent e = SendExpanderEvent(wxEVT_COMMAND_DATAVIEW_ITEM_EXPANDING,node->GetItem());
3089                //Check if the user prevent expanding
3090                if( e.GetSkipped() )
3091                     return;
3092
3093                node->ToggleOpen();
3094                //Here I build the children of current node
3095                if( node->GetChildrenNumber() == 0 )
3096                {
3097                    SortPrepare();
3098                    ::BuildTreeHelper(GetOwner()->GetModel(), node->GetItem(), node);
3099                }
3100                m_count = -1;
3101                UpdateDisplay();
3102                //Send the expanded event
3103                SendExpanderEvent(wxEVT_COMMAND_DATAVIEW_ITEM_EXPANDED,node->GetItem());
3104             }
3105             else
3106             {
3107                 SelectRow( row, false );
3108                 SelectRow( row + 1, true );
3109                 ChangeCurrentRow( row + 1 );
3110                 SendSelectionChangedEvent( GetItemByRow(row+1));
3111             }
3112         }
3113         else
3114             delete node;
3115     }
3116 }
3117
3118 void wxDataViewMainWindow::OnCollapsing(unsigned int row)
3119 {
3120     wxDataViewTreeNode * node = GetTreeNodeByRow(row);
3121     if( node != NULL )
3122     {
3123         wxDataViewTreeNode * nd = node;
3124
3125         if( node->HasChildren() && node->IsOpen() )
3126         {
3127             wxDataViewEvent e = SendExpanderEvent(wxEVT_COMMAND_DATAVIEW_ITEM_COLLAPSING,node->GetItem());
3128             if( e.GetSkipped() )
3129                 return;
3130             node->ToggleOpen();
3131             m_count = -1;
3132             UpdateDisplay();
3133             SendExpanderEvent(wxEVT_COMMAND_DATAVIEW_ITEM_COLLAPSED,nd->GetItem());
3134         }
3135         else
3136         {
3137             node = node->GetParent();
3138             if( node != NULL )
3139             {
3140                 int  parent = GetRowByItem( node->GetItem() ) ;
3141                 if( parent >= 0 )
3142                 {
3143                     SelectRow( row, false);
3144                     SelectRow(parent , true );
3145                     ChangeCurrentRow( parent );
3146                     SendSelectionChangedEvent( node->GetItem() );
3147                 }
3148             }
3149         }
3150         if( !nd->HasChildren())
3151             delete nd;
3152     }
3153 }
3154
3155 wxDataViewTreeNode * wxDataViewMainWindow::FindNode( const wxDataViewItem & item )
3156 {
3157     wxDataViewModel * model = GetOwner()->GetModel();
3158     if( model == NULL )
3159         return NULL;
3160
3161     //Compose the a parent-chain of the finding item
3162     ItemList list;
3163     list.DeleteContents( true );
3164     wxDataViewItem it( item );
3165     while( it.IsOk() )
3166     {
3167         wxDataViewItem * pItem = new wxDataViewItem( it );
3168         list.Insert( pItem );
3169         it = model->GetParent( it );
3170     }
3171
3172     //Find the item along the parent-chain.
3173     //This algorithm is designed to speed up the node-finding method
3174     wxDataViewTreeNode * node = m_root;
3175     for( ItemList::const_iterator iter = list.begin(); iter !=list.end() ; iter++ )
3176     {
3177         if( node->HasChildren() )
3178         {
3179             if( node->GetChildrenNumber() == 0 )
3180             {
3181                 SortPrepare();
3182                 ::BuildTreeHelper(model, node->GetItem(), node);
3183             }
3184
3185             wxDataViewTreeNodes nodes = node->GetNodes();
3186             unsigned int i;
3187             bool found = false;
3188
3189             for (i = 0; i < nodes.GetCount(); i ++)
3190             {
3191                 if (nodes[i]->GetItem() == (**iter))
3192                 {
3193                     if (nodes[i]->GetItem() == item)
3194                        return nodes[i];
3195
3196                     node = nodes[i];
3197                     found = true;
3198                     break;
3199                 }
3200             }
3201             if (!found)
3202                 return NULL;
3203         }
3204         else
3205             return NULL;
3206     }
3207     return NULL;
3208 }
3209
3210 void wxDataViewMainWindow::HitTest( const wxPoint & point, wxDataViewItem & item, wxDataViewColumn* &column )
3211 {
3212     wxDataViewColumn *col = NULL;
3213     unsigned int cols = GetOwner()->GetColumnCount();
3214     unsigned int colnum = 0;
3215     unsigned int x_start = 0;
3216     int x, y;
3217     m_owner->CalcUnscrolledPosition( point.x, point.y, &x, &y );
3218     for (x_start = 0; colnum < cols; colnum++)
3219     {
3220         col = GetOwner()->GetColumn(colnum);
3221         if (col->IsHidden())
3222             continue;      // skip it!
3223
3224         unsigned int w = col->GetWidth();
3225         if (x_start+w >= (unsigned int)x)
3226             break;
3227
3228         x_start += w;
3229     }
3230
3231     column = col;
3232     item = GetItemByRow( y/m_lineHeight );
3233 }
3234
3235 wxRect wxDataViewMainWindow::GetItemRect( const wxDataViewItem & item, const wxDataViewColumn* column )
3236 {
3237     int row = GetRowByItem(item);
3238     int y = row*m_lineHeight;
3239     int h = m_lineHeight;
3240     int x = 0;
3241     wxDataViewColumn *col = NULL;
3242     for( int i = 0, cols = GetOwner()->GetColumnCount(); i < cols; i ++ )
3243     {
3244        col = GetOwner()->GetColumn( i );
3245        x += col->GetWidth();
3246        if( GetOwner()->GetColumn(i+1) == column )
3247            break;
3248     }
3249     int w = col->GetWidth();
3250     m_owner->CalcScrolledPosition( x, y, &x, &y );
3251     return wxRect(x, y, w, h);
3252 }
3253
3254 int wxDataViewMainWindow::RecalculateCount()
3255 {
3256     if (!m_root)
3257     {
3258         wxDataViewIndexListModel *list_model = (wxDataViewIndexListModel*) GetOwner()->GetModel();
3259 #ifndef __WXMAC__
3260         return list_model->GetLastIndex() + 1;
3261 #else
3262         return list_model->GetLastIndex() - 1;
3263 #endif
3264     }
3265     else
3266     {
3267         return m_root->GetSubTreeCount();
3268     }
3269 }
3270
3271 class ItemToRowJob : public DoJob
3272 {
3273 public:
3274     ItemToRowJob(const wxDataViewItem & item, ItemList::const_iterator iter )
3275     { this->item = item ; ret = -1 ; m_iter = iter ; }
3276     virtual ~ItemToRowJob(){};
3277
3278     //Maybe binary search will help to speed up this process
3279     virtual int operator() ( wxDataViewTreeNode * node)
3280     {
3281          ret ++;
3282          if( node->GetItem() == item )
3283          {
3284              return DoJob::OK;
3285          }
3286
3287          if( node->GetItem() == **m_iter )
3288          {
3289              m_iter++ ;
3290              return DoJob::CONT;
3291          }
3292          else
3293          {
3294              ret += node->GetSubTreeCount();
3295              return DoJob::IGR;
3296          }
3297
3298     }
3299
3300     virtual int operator() ( void * n )
3301     {
3302         ret ++;
3303         if( n == item.GetID() )
3304             return DoJob::OK;
3305         return DoJob::CONT;
3306     }
3307     //the row number is begin from zero
3308     int GetResult(){ return ret -1 ; }
3309 private:
3310     ItemList::const_iterator  m_iter;
3311     wxDataViewItem item;
3312     int ret;
3313
3314 };
3315
3316 int wxDataViewMainWindow::GetRowByItem(const wxDataViewItem & item)
3317 {
3318     wxDataViewModel * model = GetOwner()->GetModel();
3319     if( model == NULL )
3320         return -1;
3321
3322     if (!m_root)
3323     {
3324         return wxPtrToUInt( item.GetID() );
3325     }
3326     else
3327     {
3328         if( !item.IsOk() )
3329             return -1;
3330
3331         //Compose the a parent-chain of the finding item
3332         ItemList list;
3333         wxDataViewItem * pItem = NULL;
3334         list.DeleteContents( true );
3335         wxDataViewItem it( item );
3336         while( it.IsOk() )
3337         {
3338             pItem = new wxDataViewItem( it );
3339             list.Insert( pItem );
3340             it = model->GetParent( it );
3341         }
3342         pItem = new wxDataViewItem( );
3343         list.Insert( pItem );
3344
3345         ItemToRowJob job( item, list.begin() );
3346         Walker(m_root , job );
3347         return job.GetResult();
3348     }
3349 }
3350
3351 static void BuildTreeHelper( wxDataViewModel * model,  wxDataViewItem & item, wxDataViewTreeNode * node)
3352 {
3353     if( !model->IsContainer( item ) )
3354         return ;
3355
3356     wxDataViewItemArray children;
3357     unsigned int num = model->GetChildren( item, children);
3358     unsigned int index = 0;
3359     while( index < num )
3360     {
3361         if( model->IsContainer( children[index] ) )
3362         {
3363             wxDataViewTreeNode * n = new wxDataViewTreeNode( node );
3364             n->SetItem(children[index]);
3365             n->SetHasChildren( true ) ;
3366             node->AddNode( n );
3367         }
3368         else
3369         {
3370             node->AddLeaf( children[index].GetID() );
3371         }
3372         index ++;
3373     }
3374     node->SetSubTreeCount( num );
3375     wxDataViewTreeNode * n = node->GetParent();
3376     if( n != NULL)
3377         n->ChangeSubTreeCount(num);
3378
3379 }
3380
3381 void wxDataViewMainWindow::BuildTree(wxDataViewModel * model)
3382 {
3383     DestroyTree();
3384
3385     if (GetOwner()->GetModel()->IsVirtualListModel())
3386     {
3387         m_count = -1 ;
3388         return;
3389     }
3390
3391     m_root = new wxDataViewTreeNode( NULL );
3392     m_root->SetHasChildren(true);
3393
3394     //First we define a invalid item to fetch the top-level elements
3395     wxDataViewItem item;
3396     SortPrepare();
3397     BuildTreeHelper( model, item, m_root);
3398     m_count = -1 ;
3399 }
3400
3401 static void DestroyTreeHelper( wxDataViewTreeNode * node )
3402 {
3403     if( node->GetNodeNumber() != 0 )
3404     {
3405         int len = node->GetNodeNumber();
3406         int i = 0 ;
3407         wxDataViewTreeNodes& nodes = node->GetNodes();
3408         for( ; i < len; i ++ )
3409         {
3410             DestroyTreeHelper(nodes[i]);
3411         }
3412     }
3413     delete node;
3414 }
3415
3416 void wxDataViewMainWindow::DestroyTree()
3417 {
3418     if (m_root)
3419     {
3420        ::DestroyTreeHelper(m_root);
3421         m_count = 0;
3422         m_root = NULL;
3423     }
3424 }
3425
3426 void wxDataViewMainWindow::OnChar( wxKeyEvent &event )
3427 {
3428     if ( HandleAsNavigationKey(event) )
3429         return;
3430
3431     // no item -> nothing to do
3432     if (!HasCurrentRow())
3433     {
3434         event.Skip();
3435         return;
3436     }
3437
3438     // don't use m_linesPerPage directly as it might not be computed yet
3439     const int pageSize = GetCountPerPage();
3440     wxCHECK_RET( pageSize, _T("should have non zero page size") );
3441
3442     switch ( event.GetKeyCode() )
3443     {
3444         case WXK_RETURN:
3445         {
3446             if (m_currentRow > 0)
3447             {
3448                 wxWindow *parent = GetParent();
3449                 wxDataViewEvent le(wxEVT_COMMAND_DATAVIEW_ITEM_ACTIVATED, parent->GetId());
3450                 le.SetItem( GetItemByRow(m_currentRow) );
3451                 le.SetEventObject(parent);
3452                 le.SetModel(GetOwner()->GetModel());
3453
3454                 parent->GetEventHandler()->ProcessEvent(le);
3455             }
3456             break;
3457         }
3458         case WXK_UP:
3459             if ( m_currentRow > 0 )
3460                 OnArrowChar( m_currentRow - 1, event );
3461             break;
3462
3463         case WXK_DOWN:
3464             if ( m_currentRow < GetRowCount() - 1 )
3465                 OnArrowChar( m_currentRow + 1, event );
3466             break;
3467         //Add the process for tree expanding/collapsing
3468         case WXK_LEFT:
3469          OnCollapsing(m_currentRow);
3470          break;
3471      case WXK_RIGHT:
3472          OnExpanding( m_currentRow);
3473          break;
3474         case WXK_END:
3475             if (!IsEmpty())
3476                 OnArrowChar( GetRowCount() - 1, event );
3477             break;
3478
3479         case WXK_HOME:
3480             if (!IsEmpty())
3481                 OnArrowChar( 0, event );
3482             break;
3483
3484         case WXK_PAGEUP:
3485             {
3486                 int steps = pageSize - 1;
3487                 int index = m_currentRow - steps;
3488                 if (index < 0)
3489                     index = 0;
3490
3491                 OnArrowChar( index, event );
3492             }
3493             break;
3494
3495         case WXK_PAGEDOWN:
3496             {
3497                 int steps = pageSize - 1;
3498                 unsigned int index = m_currentRow + steps;
3499                 unsigned int count = GetRowCount();
3500                 if ( index >= count )
3501                     index = count - 1;
3502
3503                 OnArrowChar( index, event );
3504             }
3505             break;
3506
3507         default:
3508             event.Skip();
3509     }
3510 }
3511
3512 void wxDataViewMainWindow::OnMouse( wxMouseEvent &event )
3513 {
3514     if (event.GetEventType() == wxEVT_MOUSEWHEEL)
3515     {
3516         // let the base handle mouse wheel events.
3517         event.Skip();
3518         return;
3519     }
3520
3521     int x = event.GetX();
3522     int y = event.GetY();
3523     m_owner->CalcUnscrolledPosition( x, y, &x, &y );
3524     wxDataViewColumn *col = NULL;
3525
3526     int xpos = 0;
3527     unsigned int cols = GetOwner()->GetColumnCount();
3528     unsigned int i;
3529     for (i = 0; i < cols; i++)
3530     {
3531         wxDataViewColumn *c = GetOwner()->GetColumn( i );
3532         if (c->IsHidden())
3533             continue;      // skip it!
3534
3535         if (x < xpos + c->GetWidth())
3536         {
3537             col = c;
3538             break;
3539         }
3540         xpos += c->GetWidth();
3541     }
3542     if (!col)
3543         return;
3544
3545     wxDataViewRenderer *cell = col->GetRenderer();
3546     unsigned int current = y / m_lineHeight;
3547     if ((current > GetRowCount()) || (x > GetEndOfLastCol()))
3548     {
3549         // Unselect all if below the last row ?
3550         return;
3551     }
3552
3553     //Test whether the mouse is hovered on the tree item button
3554     bool hover = false;
3555     if ((m_root) && (GetOwner()->GetExpanderColumn() == col))
3556     {
3557         wxDataViewTreeNode * node = GetTreeNodeByRow(current);
3558         if( node!=NULL && node->HasChildren() )
3559         {
3560             int indent = node->GetIndentLevel();
3561             indent = GetOwner()->GetIndent()*indent;
3562             wxRect rect( xpos + indent + EXPANDER_MARGIN, current * m_lineHeight + EXPANDER_MARGIN, m_lineHeight-2*EXPANDER_MARGIN,m_lineHeight-2*EXPANDER_MARGIN);
3563             if( rect.Contains( x, y) )
3564             {
3565                 //So the mouse is over the expander
3566                 hover = true;
3567                 if (m_underMouse && m_underMouse != node)
3568                 {
3569                     //wxLogMessage("Undo the row: %d", GetRowByItem(m_underMouse->GetItem()));
3570                     RefreshRow(GetRowByItem(m_underMouse->GetItem()));
3571                 }
3572                 if (m_underMouse != node)
3573                 {
3574                     //wxLogMessage("Do the row: %d", current);
3575                     RefreshRow(current);
3576                 }
3577                 m_underMouse = node;
3578             }
3579         }
3580         if (node!=NULL && !node->HasChildren())
3581             delete node;
3582     }
3583     if (!hover)
3584     {
3585         if (m_underMouse != NULL)
3586         {
3587             //wxLogMessage("Undo the row: %d", GetRowByItem(m_underMouse->GetItem()));
3588             RefreshRow(GetRowByItem(m_underMouse->GetItem()));
3589             m_underMouse = NULL;
3590         }
3591     }
3592
3593     wxDataViewModel *model = GetOwner()->GetModel();
3594
3595     if (event.Dragging())
3596     {
3597         if (m_dragCount == 0)
3598         {
3599             // we have to report the raw, physical coords as we want to be
3600             // able to call HitTest(event.m_pointDrag) from the user code to
3601             // get the item being dragged
3602             m_dragStart = event.GetPosition();
3603         }
3604
3605         m_dragCount++;
3606
3607         if (m_dragCount != 3)
3608             return;
3609
3610         if (event.LeftIsDown())
3611         {
3612             // Notify cell about drag
3613         }
3614         return;
3615     }
3616     else
3617     {
3618         m_dragCount = 0;
3619     }
3620
3621     bool forceClick = false;
3622
3623     if (event.ButtonDClick())
3624     {
3625         m_renameTimer->Stop();
3626         m_lastOnSame = false;
3627     }
3628
3629     wxDataViewItem item = GetItemByRow(current);
3630     bool ignore_other_columns =
3631         ((GetOwner()->GetExpanderColumn() != col) &&
3632          (model->IsContainer(item)) &&
3633          (!model->HasContainerColumns(item)));
3634
3635     if (event.LeftDClick())
3636     {
3637         if ( current == m_lineLastClicked )
3638         {
3639             if ((!ignore_other_columns) && (cell->GetMode() == wxDATAVIEW_CELL_ACTIVATABLE))
3640             {
3641                 wxVariant value;
3642                 model->GetValue( value, item, col->GetModelColumn() );
3643                 cell->SetValue( value );
3644                 wxRect cell_rect( xpos, current * m_lineHeight,
3645                                   col->GetWidth(), m_lineHeight );
3646                 cell->Activate( cell_rect, model, item, col->GetModelColumn() );
3647
3648             }
3649             else
3650             {
3651                 wxWindow *parent = GetParent();
3652                 wxDataViewEvent le(wxEVT_COMMAND_DATAVIEW_ITEM_ACTIVATED, parent->GetId());
3653                 le.SetItem( item );
3654                 le.SetEventObject(parent);
3655                 le.SetModel(GetOwner()->GetModel());
3656
3657                 parent->GetEventHandler()->ProcessEvent(le);
3658             }
3659             return;
3660         }
3661         else
3662         {
3663             // The first click was on another item, so don't interpret this as
3664             // a double click, but as a simple click instead
3665             forceClick = true;
3666         }
3667     }
3668
3669     if (event.LeftUp())
3670     {
3671         if (m_lineSelectSingleOnUp != (unsigned int)-1)
3672         {
3673             // select single line
3674             SelectAllRows( false );
3675             SelectRow( m_lineSelectSingleOnUp, true );
3676         }
3677
3678         //Process the event of user clicking the expander
3679         bool expander = false;
3680         if ((m_root) && (GetOwner()->GetExpanderColumn() == col))
3681         {
3682             wxDataViewTreeNode * node = GetTreeNodeByRow(current);
3683             if( node!=NULL && node->HasChildren() )
3684             {
3685                 int indent = node->GetIndentLevel();
3686                 indent = GetOwner()->GetIndent()*indent;
3687                 wxRect rect( xpos + indent + EXPANDER_MARGIN, current * m_lineHeight + EXPANDER_MARGIN, m_lineHeight-2*EXPANDER_MARGIN,m_lineHeight-2*EXPANDER_MARGIN);
3688                 if( rect.Contains( x, y) )
3689                 {
3690                     expander = true;
3691                     if( node->IsOpen() )
3692                         OnCollapsing(current);
3693                     else
3694                         OnExpanding( current );
3695                 }
3696             }
3697             if (node && !node->HasChildren())
3698                delete node;
3699         }
3700         //If the user click the expander, we do not do editing even if the column with expander are editable
3701         if (m_lastOnSame && !expander && !ignore_other_columns)
3702         {
3703             if ((col == m_currentCol) && (current == m_currentRow) &&
3704                 (cell->GetMode() & wxDATAVIEW_CELL_EDITABLE) )
3705             {
3706                 m_renameTimer->Start( 100, true );
3707             }
3708         }
3709
3710         m_lastOnSame = false;
3711         m_lineSelectSingleOnUp = (unsigned int)-1;
3712     }
3713     else
3714     {
3715         // This is necessary, because after a DnD operation in
3716         // from and to ourself, the up event is swallowed by the
3717         // DnD code. So on next non-up event (which means here and
3718         // now) m_lineSelectSingleOnUp should be reset.
3719         m_lineSelectSingleOnUp = (unsigned int)-1;
3720     }
3721
3722     if (event.RightDown())
3723     {
3724         m_lineBeforeLastClicked = m_lineLastClicked;
3725         m_lineLastClicked = current;
3726
3727         // If the item is already selected, do not update the selection.
3728         // Multi-selections should not be cleared if a selected item is clicked.
3729         if (!IsRowSelected(current))
3730         {
3731             SelectAllRows(false);
3732             ChangeCurrentRow(current);
3733             SelectRow(m_currentRow,true);
3734             SendSelectionChangedEvent(GetItemByRow( m_currentRow ) );
3735         }
3736
3737         wxVariant value;
3738         model->GetValue( value, item, col->GetModelColumn() );
3739         wxWindow *parent = GetParent();
3740         wxDataViewEvent le(wxEVT_COMMAND_DATAVIEW_ITEM_CONTEXT_MENU, parent->GetId());
3741         le.SetItem( item );
3742         le.SetEventObject(parent);
3743         le.SetModel(GetOwner()->GetModel());
3744         le.SetValue(value);
3745         parent->GetEventHandler()->ProcessEvent(le);
3746     }
3747     else if (event.MiddleDown())
3748     {
3749     }
3750     if (event.LeftDown() || forceClick)
3751     {
3752         SetFocus();
3753
3754         m_lineBeforeLastClicked = m_lineLastClicked;
3755         m_lineLastClicked = current;
3756
3757         unsigned int oldCurrentRow = m_currentRow;
3758         bool oldWasSelected = IsRowSelected(m_currentRow);
3759
3760         bool cmdModifierDown = event.CmdDown();
3761         if ( IsSingleSel() || !(cmdModifierDown || event.ShiftDown()) )
3762         {
3763             if ( IsSingleSel() || !IsRowSelected(current) )
3764             {
3765                 SelectAllRows( false );
3766                 ChangeCurrentRow(current);
3767                 SelectRow(m_currentRow,true);
3768                 SendSelectionChangedEvent(GetItemByRow( m_currentRow ) );
3769             }
3770             else // multi sel & current is highlighted & no mod keys
3771             {
3772                 m_lineSelectSingleOnUp = current;
3773                 ChangeCurrentRow(current); // change focus
3774             }
3775         }
3776         else // multi sel & either ctrl or shift is down
3777         {
3778             if (cmdModifierDown)
3779             {
3780                 ChangeCurrentRow(current);
3781                 ReverseRowSelection(m_currentRow);
3782                 SendSelectionChangedEvent(GetItemByRow(m_selection[0]) );
3783             }
3784             else if (event.ShiftDown())
3785             {
3786                 ChangeCurrentRow(current);
3787
3788                 unsigned int lineFrom = oldCurrentRow,
3789                        lineTo = current;
3790
3791                 if ( lineTo < lineFrom )
3792                 {
3793                     lineTo = lineFrom;
3794                     lineFrom = m_currentRow;
3795                 }
3796
3797                 SelectRows(lineFrom, lineTo, true);
3798                 SendSelectionChangedEvent(GetItemByRow(m_selection[0]) );
3799             }
3800             else // !ctrl, !shift
3801             {
3802                 // test in the enclosing if should make it impossible
3803                 wxFAIL_MSG( _T("how did we get here?") );
3804             }
3805         }
3806
3807         if (m_currentRow != oldCurrentRow)
3808             RefreshRow( oldCurrentRow );
3809
3810         wxDataViewColumn *oldCurrentCol = m_currentCol;
3811
3812         // Update selection here...
3813         m_currentCol = col;
3814
3815         m_lastOnSame = !forceClick && ((col == oldCurrentCol) &&
3816                         (current == oldCurrentRow)) && oldWasSelected;
3817
3818         // Call LeftClick after everything else as under GTK+
3819         if (cell->GetMode() & wxDATAVIEW_CELL_ACTIVATABLE)
3820         {
3821             // notify cell about right click
3822             wxVariant value;
3823             model->GetValue( value, item, col->GetModelColumn() );
3824             cell->SetValue( value );
3825             wxRect cell_rect( xpos, current * m_lineHeight,
3826                           col->GetWidth(), m_lineHeight );
3827             /* ignore ret */ cell->LeftClick( event.GetPosition(), cell_rect, model, item, col->GetModelColumn());
3828         }
3829     }
3830 }
3831
3832 void wxDataViewMainWindow::OnSetFocus( wxFocusEvent &event )
3833 {
3834     m_hasFocus = true;
3835
3836     if (HasCurrentRow())
3837         Refresh();
3838
3839     event.Skip();
3840 }
3841
3842 void wxDataViewMainWindow::OnKillFocus( wxFocusEvent &event )
3843 {
3844     m_hasFocus = false;
3845
3846     if (HasCurrentRow())
3847         Refresh();
3848
3849     event.Skip();
3850 }
3851
3852 wxDataViewItem wxDataViewMainWindow::GetSelection() const
3853 {
3854     if( m_selection.GetCount() != 1 )
3855         return wxDataViewItem();
3856
3857     return GetItemByRow( m_selection.Item(0));
3858 }
3859
3860 //-----------------------------------------------------------------------------
3861 // wxDataViewCtrl
3862 //-----------------------------------------------------------------------------
3863 WX_DEFINE_LIST(wxDataViewColumnList)
3864
3865 IMPLEMENT_DYNAMIC_CLASS(wxDataViewCtrl, wxDataViewCtrlBase)
3866
3867 BEGIN_EVENT_TABLE(wxDataViewCtrl, wxDataViewCtrlBase)
3868     EVT_SIZE(wxDataViewCtrl::OnSize)
3869 END_EVENT_TABLE()
3870
3871 wxDataViewCtrl::~wxDataViewCtrl()
3872 {
3873     if (m_notifier)
3874         GetModel()->RemoveNotifier( m_notifier );
3875
3876     wxDataViewColumnList::const_iterator iter;
3877     for (iter = m_cols.begin(); iter!=m_cols.end(); iter++)
3878     {
3879         delete *iter;
3880     }
3881 }
3882
3883 void wxDataViewCtrl::Init()
3884 {
3885     m_notifier = NULL;
3886 }
3887
3888 bool wxDataViewCtrl::Create(wxWindow *parent, wxWindowID id,
3889            const wxPoint& pos, const wxSize& size,
3890            long style, const wxValidator& validator )
3891 {
3892     if ( (style & wxBORDER_MASK) == 0)
3893         style |= wxBORDER_SUNKEN;
3894
3895     if (!wxControl::Create( parent, id, pos, size,
3896                             style | wxScrolledWindowStyle, validator))
3897         return false;
3898
3899     SetInitialSize(size);
3900
3901     Init();
3902
3903 #ifdef __WXMAC__
3904     MacSetClipChildren( true ) ;
3905 #endif
3906
3907     m_clientArea = new wxDataViewMainWindow( this, wxID_ANY );
3908
3909     if (HasFlag(wxDV_NO_HEADER))
3910         m_headerArea = NULL;
3911     else
3912         m_headerArea = new wxDataViewHeaderWindow( this, wxID_ANY );
3913
3914     SetTargetWindow( m_clientArea );
3915
3916     wxBoxSizer *sizer = new wxBoxSizer( wxVERTICAL );
3917     if (m_headerArea)
3918         sizer->Add( m_headerArea, 0, wxGROW );
3919     sizer->Add( m_clientArea, 1, wxGROW );
3920     SetSizer( sizer );
3921
3922     return true;
3923 }
3924
3925 #ifdef __WXMSW__
3926 WXLRESULT wxDataViewCtrl::MSWWindowProc(WXUINT nMsg,
3927                                        WXWPARAM wParam,
3928                                        WXLPARAM lParam)
3929 {
3930     WXLRESULT rc = wxDataViewCtrlBase::MSWWindowProc(nMsg, wParam, lParam);
3931
3932 #ifndef __WXWINCE__
3933     // we need to process arrows ourselves for scrolling
3934     if ( nMsg == WM_GETDLGCODE )
3935     {
3936         rc |= DLGC_WANTARROWS;
3937     }
3938 #endif
3939
3940     return rc;
3941 }
3942 #endif
3943
3944 void wxDataViewCtrl::OnSize( wxSizeEvent &WXUNUSED(event) )
3945 {
3946     // We need to override OnSize so that our scrolled
3947     // window a) does call Layout() to use sizers for
3948     // positioning the controls but b) does not query
3949     // the sizer for their size and use that for setting
3950     // the scrollable area as set that ourselves by
3951     // calling SetScrollbar() further down.
3952
3953     Layout();
3954
3955     AdjustScrollbars();
3956 }
3957
3958 bool wxDataViewCtrl::AssociateModel( wxDataViewModel *model )
3959 {
3960     if (!wxDataViewCtrlBase::AssociateModel( model ))
3961         return false;
3962
3963     m_notifier = new wxGenericDataViewModelNotifier( m_clientArea );
3964
3965     model->AddNotifier( m_notifier );
3966
3967     m_clientArea->DestroyTree();
3968
3969     m_clientArea->BuildTree(model);
3970
3971     m_clientArea->UpdateDisplay();
3972
3973     return true;
3974 }
3975
3976 bool wxDataViewCtrl::AppendColumn( wxDataViewColumn *col )
3977 {
3978     if (!wxDataViewCtrlBase::AppendColumn(col))
3979         return false;
3980
3981     m_cols.Append( col );
3982     OnColumnChange();
3983     return true;
3984 }
3985
3986 bool wxDataViewCtrl::PrependColumn( wxDataViewColumn *col )
3987 {
3988     if (!wxDataViewCtrlBase::PrependColumn(col))
3989         return false;
3990
3991     m_cols.Insert( col );
3992     OnColumnChange();
3993     return true;
3994 }
3995
3996 void wxDataViewCtrl::OnColumnChange()
3997 {
3998     if (m_headerArea)
3999         m_headerArea->UpdateDisplay();
4000
4001     m_clientArea->UpdateDisplay();
4002 }
4003
4004 void wxDataViewCtrl::DoSetExpanderColumn()
4005 {
4006     m_clientArea->UpdateDisplay();
4007 }
4008
4009 void wxDataViewCtrl::DoSetIndent()
4010 {
4011     m_clientArea->UpdateDisplay();
4012 }
4013
4014 unsigned int wxDataViewCtrl::GetColumnCount() const
4015 {
4016     return m_cols.GetCount();
4017 }
4018
4019 wxDataViewColumn* wxDataViewCtrl::GetColumn( unsigned int pos ) const
4020 {
4021     wxDataViewColumnList::const_iterator iter;
4022     unsigned int i = 0;
4023     for (iter = m_cols.begin(); iter!=m_cols.end(); iter++)
4024     {
4025         if (i == pos)
4026             return *iter;
4027
4028         if ((*iter)->IsHidden())
4029             continue;
4030         i ++;
4031     }
4032     return NULL;
4033 }
4034
4035 void wxDataViewCtrl::ColumnMoved( wxDataViewColumn* col, unsigned int new_pos )
4036 {
4037     if (new_pos > m_cols.GetCount()) return;
4038     m_cols.DeleteObject( col );
4039     m_cols.Insert( new_pos, col );
4040
4041     m_clientArea->UpdateDisplay();
4042 }
4043
4044 bool wxDataViewCtrl::DeleteColumn( wxDataViewColumn *column )
4045 {
4046     wxDataViewColumnList::compatibility_iterator  ret = m_cols.Find( column );
4047     if (!ret)
4048         return false;
4049
4050     m_cols.Erase(ret);
4051     delete column;
4052     OnColumnChange();
4053
4054     return true;
4055 }
4056
4057 bool wxDataViewCtrl::ClearColumns()
4058 {
4059     m_cols.clear();
4060     OnColumnChange();
4061     return true;
4062 }
4063
4064 int wxDataViewCtrl::GetColumnPosition( const wxDataViewColumn *column ) const
4065 {
4066     int ret = 0, dead = 0;
4067     int len = GetColumnCount();
4068     for (int i=0; i<len; i++)
4069     {
4070         wxDataViewColumn * col = GetColumn(i);
4071         if (col->IsHidden())
4072             continue;
4073         ret += col->GetWidth();
4074         if (column==col)
4075         {
4076             CalcScrolledPosition( ret, dead, &ret, &dead );
4077             break;
4078         }
4079     }
4080     return ret;
4081 }
4082
4083 wxDataViewColumn *wxDataViewCtrl::GetSortingColumn() const
4084 {
4085     return NULL;
4086 }
4087
4088 //Selection code with wxDataViewItem as parameters
4089 wxDataViewItem wxDataViewCtrl::GetSelection() const
4090 {
4091     return m_clientArea->GetSelection();
4092 }
4093
4094 int wxDataViewCtrl::GetSelections( wxDataViewItemArray & sel ) const
4095 {
4096     sel.Empty();
4097     wxDataViewSelection selection = m_clientArea->GetSelections();
4098     int len = selection.GetCount();
4099     for( int i = 0; i < len; i ++)
4100     {
4101         unsigned int row = selection[i];
4102         sel.Add( m_clientArea->GetItemByRow( row ) );
4103     }
4104     return len;
4105 }
4106
4107 void wxDataViewCtrl::SetSelections( const wxDataViewItemArray & sel )
4108 {
4109     wxDataViewSelection selection(wxDataViewSelectionCmp) ;
4110     int len = sel.GetCount();
4111     for( int i = 0; i < len; i ++ )
4112     {
4113         int row = m_clientArea->GetRowByItem( sel[i] );
4114         if( row >= 0 )
4115             selection.Add( static_cast<unsigned int>(row) );
4116     }
4117     m_clientArea->SetSelections( selection );
4118 }
4119
4120 void wxDataViewCtrl::Select( const wxDataViewItem & item )
4121 {
4122     int row = m_clientArea->GetRowByItem( item );
4123     if( row >= 0 )
4124     {
4125         //Unselect all rows before select another in the single select mode
4126         if (m_clientArea->IsSingleSel())
4127             m_clientArea->SelectAllRows(false);
4128         m_clientArea->SelectRow(row, true);
4129     }
4130 }
4131
4132 void wxDataViewCtrl::Unselect( const wxDataViewItem & item )
4133 {
4134     int row = m_clientArea->GetRowByItem( item );
4135     if( row >= 0 )
4136         m_clientArea->SelectRow(row, false);
4137 }
4138
4139 bool wxDataViewCtrl::IsSelected( const wxDataViewItem & item ) const
4140 {
4141     int row = m_clientArea->GetRowByItem( item );
4142     if( row >= 0 )
4143     {
4144         return m_clientArea->IsRowSelected(row);
4145     }
4146     return false;
4147 }
4148
4149 //Selection code with row number as parameter
4150 int wxDataViewCtrl::GetSelections( wxArrayInt & sel ) const
4151 {
4152     sel.Empty();
4153     wxDataViewSelection selection = m_clientArea->GetSelections();
4154     int len = selection.GetCount();
4155     for( int i = 0; i < len; i ++)
4156     {
4157         unsigned int row = selection[i];
4158         sel.Add( row );
4159     }
4160     return len;
4161 }
4162
4163 void wxDataViewCtrl::SetSelections( const wxArrayInt & sel )
4164 {
4165     wxDataViewSelection selection(wxDataViewSelectionCmp) ;
4166     int len = sel.GetCount();
4167     for( int i = 0; i < len; i ++ )
4168     {
4169         int row = sel[i];
4170         if( row >= 0 )
4171             selection.Add( static_cast<unsigned int>(row) );
4172     }
4173     m_clientArea->SetSelections( selection );
4174 }
4175
4176 void wxDataViewCtrl::Select( int row )
4177 {
4178     if( row >= 0 )
4179     {
4180         if (m_clientArea->IsSingleSel())
4181             m_clientArea->SelectAllRows(false);
4182         m_clientArea->SelectRow( row, true );
4183     }
4184 }
4185
4186 void wxDataViewCtrl::Unselect( int row )
4187 {
4188     if( row >= 0 )
4189         m_clientArea->SelectRow(row, false);
4190 }
4191
4192 bool wxDataViewCtrl::IsSelected( int row ) const
4193 {
4194     if( row >= 0 )
4195         return m_clientArea->IsRowSelected(row);
4196     return false;
4197 }
4198
4199 void wxDataViewCtrl::SelectRange( int from, int to )
4200 {
4201     wxArrayInt sel;
4202     for( int i = from; i < to; i ++ )
4203         sel.Add( i );
4204     m_clientArea->Select(sel);
4205 }
4206
4207 void wxDataViewCtrl::UnselectRange( int from, int to )
4208 {
4209     wxDataViewSelection sel = m_clientArea->GetSelections();
4210     for( int i = from; i < to; i ++ )
4211         if( sel.Index( i ) != wxNOT_FOUND )
4212             sel.Remove( i );
4213     m_clientArea->SetSelections(sel);
4214 }
4215
4216 void wxDataViewCtrl::SelectAll()
4217 {
4218     m_clientArea->SelectAllRows(true);
4219 }
4220
4221 void wxDataViewCtrl::UnselectAll()
4222 {
4223     m_clientArea->SelectAllRows(false);
4224 }
4225
4226 void wxDataViewCtrl::EnsureVisible( int row, int column )
4227 {
4228     if( row < 0 )
4229         row = 0;
4230     if( row > (int) m_clientArea->GetRowCount() )
4231         row = m_clientArea->GetRowCount();
4232
4233     int first = m_clientArea->GetFirstVisibleRow();
4234     int last = m_clientArea->GetLastVisibleRow();
4235     if( row < first )
4236         m_clientArea->ScrollTo( row, column );
4237     else if( row > last )
4238         m_clientArea->ScrollTo( row - last + first, column );
4239     else
4240         m_clientArea->ScrollTo( first, column );
4241 }
4242
4243 void wxDataViewCtrl::EnsureVisible( const wxDataViewItem & item, const wxDataViewColumn * column )
4244 {
4245     int row = m_clientArea->GetRowByItem(item);
4246     if( row >= 0 )
4247     {
4248         if( column == NULL )
4249             EnsureVisible(row, -1);
4250         else
4251         {
4252             int col = 0;
4253             int len = GetColumnCount();
4254             for( int i = 0; i < len; i ++ )
4255                 if( GetColumn(i) == column )
4256                 {
4257                     col = i;
4258                     break;
4259                 }
4260             EnsureVisible( row, col );
4261         }
4262     }
4263
4264 }
4265
4266 void wxDataViewCtrl::HitTest( const wxPoint & point, wxDataViewItem & item, wxDataViewColumn* &column ) const
4267 {
4268     m_clientArea->HitTest(point, item, column);
4269 }
4270
4271 wxRect wxDataViewCtrl::GetItemRect( const wxDataViewItem & item, const wxDataViewColumn* column ) const
4272 {
4273     return m_clientArea->GetItemRect(item, column);
4274 }
4275
4276 wxDataViewItem wxDataViewCtrl::GetItemByRow( unsigned int row ) const
4277 {
4278     return m_clientArea->GetItemByRow( row );
4279 }
4280
4281 int wxDataViewCtrl::GetRowByItem( const wxDataViewItem & item ) const
4282 {
4283     return m_clientArea->GetRowByItem( item );
4284 }
4285
4286 void wxDataViewCtrl::Expand( const wxDataViewItem & item )
4287 {
4288     int row = m_clientArea->GetRowByItem( item );
4289     if (row != -1)
4290         m_clientArea->Expand(row);
4291 }
4292
4293 void wxDataViewCtrl::Collapse( const wxDataViewItem & item )
4294 {
4295     int row = m_clientArea->GetRowByItem( item );
4296     if (row != -1)
4297         m_clientArea->Collapse(row);
4298 }
4299
4300  #endif
4301     // !wxUSE_GENERICDATAVIEWCTRL
4302
4303 #endif
4304     // wxUSE_DATAVIEWCTRL