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