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