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