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