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