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