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