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