]> git.saurik.com Git - wxWidgets.git/blob - src/generic/datavgen.cpp
projects / wxWidgets.git / blob
? search:


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