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