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