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