///////////////////////////////////////////////////////////////////////////// // Name: src/generic/datavgen.cpp // Purpose: wxDataViewCtrl generic implementation // Author: Robert Roebling // Modified by: Francesco Montorsi, Guru Kathiresan, Bo Yang // Copyright: (c) 1998 Robert Roebling // Licence: wxWindows licence ///////////////////////////////////////////////////////////////////////////// // For compilers that support precompilation, includes "wx.h". #include "wx/wxprec.h" #ifdef __BORLANDC__ #pragma hdrstop #endif #if wxUSE_DATAVIEWCTRL #include "wx/dataview.h" #ifdef wxUSE_GENERICDATAVIEWCTRL #ifndef WX_PRECOMP #ifdef __WXMSW__ #include "wx/msw/private.h" #include "wx/msw/wrapwin.h" #include "wx/msw/wrapcctl.h" // include "properly" #endif #include "wx/sizer.h" #include "wx/log.h" #include "wx/dcclient.h" #include "wx/timer.h" #include "wx/settings.h" #include "wx/msgdlg.h" #include "wx/dcscreen.h" #include "wx/frame.h" #endif #include "wx/stockitem.h" #include "wx/popupwin.h" #include "wx/renderer.h" #include "wx/dcbuffer.h" #include "wx/icon.h" #include "wx/list.h" #include "wx/listimpl.cpp" #include "wx/imaglist.h" #include "wx/headerctrl.h" #include "wx/dnd.h" #include "wx/stopwatch.h" #include "wx/weakref.h" //----------------------------------------------------------------------------- // classes //----------------------------------------------------------------------------- class wxDataViewColumn; class wxDataViewHeaderWindow; class wxDataViewCtrl; //----------------------------------------------------------------------------- // classes //----------------------------------------------------------------------------- static const int SCROLL_UNIT_X = 15; // the cell padding on the left/right static const int PADDING_RIGHTLEFT = 3; // the expander space margin static const int EXPANDER_MARGIN = 4; #ifdef __WXMSW__ static const int EXPANDER_OFFSET = 4; #else static const int EXPANDER_OFFSET = 1; #endif // Below is the compare stuff. // For the generic implementation, both the leaf nodes and the nodes are sorted for // fast search when needed static wxDataViewModel* g_model; // The column is either the index of the column to be used for sorting or one // of the special values in this enum: enum { // Sort when we're thawed later. SortColumn_OnThaw = -3, // Don't sort at all. SortColumn_None = -2, // Sort using the model default sort order. SortColumn_Default = -1 }; static int g_column = SortColumn_None; static bool g_asending = true; // ---------------------------------------------------------------------------- // helper functions // ---------------------------------------------------------------------------- namespace { // Return the expander column or, if it is not set, the first column and also // set it as the expander one for the future. wxDataViewColumn* GetExpanderColumnOrFirstOne(wxDataViewCtrl* dataview) { wxDataViewColumn* expander = dataview->GetExpanderColumn(); if (!expander) { // TODO-RTL: last column for RTL support expander = dataview->GetColumnAt( 0 ); dataview->SetExpanderColumn(expander); } return expander; } wxTextCtrl *CreateEditorTextCtrl(wxWindow *parent, const wxRect& labelRect, const wxString& value) { wxTextCtrl* ctrl = new wxTextCtrl(parent, wxID_ANY, value, wxPoint(labelRect.x,labelRect.y), wxSize(labelRect.width,labelRect.height), wxTE_PROCESS_ENTER); // Adjust size of wxTextCtrl editor to fit text, even if it means being // wider than the corresponding column (this is how Explorer behaves). const int fitting = ctrl->GetSizeFromTextSize(ctrl->GetTextExtent(ctrl->GetValue())).x; const int current = ctrl->GetSize().x; const int maxwidth = ctrl->GetParent()->GetSize().x - ctrl->GetPosition().x; // Adjust size so that it fits all content. Don't change anything if the // allocated space is already larger than needed and don't extend wxDVC's // boundaries. int width = wxMin(wxMax(current, fitting), maxwidth); if ( width != current ) ctrl->SetSize(wxSize(width, -1)); // select the text in the control an place the cursor at the end ctrl->SetInsertionPointEnd(); ctrl->SelectAll(); return ctrl; } } // anonymous namespace //----------------------------------------------------------------------------- // wxDataViewColumn //----------------------------------------------------------------------------- void wxDataViewColumn::Init(int width, wxAlignment align, int flags) { m_width = width; m_minWidth = 0; m_align = align; m_flags = flags; m_sort = false; m_sortAscending = true; } int wxDataViewColumn::GetWidth() const { switch ( m_width ) { case wxCOL_WIDTH_DEFAULT: return wxDVC_DEFAULT_WIDTH; case wxCOL_WIDTH_AUTOSIZE: wxCHECK_MSG( m_owner, wxDVC_DEFAULT_WIDTH, "no owner control" ); return m_owner->GetBestColumnWidth(m_owner->GetColumnIndex(this)); default: return m_width; } } void wxDataViewColumn::UpdateDisplay() { if (m_owner) { int idx = m_owner->GetColumnIndex( this ); m_owner->OnColumnChange( idx ); } } void wxDataViewColumn::UnsetAsSortKey() { m_sort = false; if ( m_owner ) m_owner->SetSortingColumnIndex(wxNOT_FOUND); UpdateDisplay(); } void wxDataViewColumn::SetSortOrder(bool ascending) { if ( !m_owner ) return; // First unset the old sort column if any. int oldSortKey = m_owner->GetSortingColumnIndex(); if ( oldSortKey != wxNOT_FOUND ) { m_owner->GetColumn(oldSortKey)->UnsetAsSortKey(); } // Now set this one as the new sort column. const int idx = m_owner->GetColumnIndex(this); m_owner->SetSortingColumnIndex(idx); m_sort = true; m_sortAscending = ascending; // Call this directly instead of using UpdateDisplay() as we already have // the column index, no need to look it up again. m_owner->OnColumnChange(idx); } //----------------------------------------------------------------------------- // wxDataViewHeaderWindow //----------------------------------------------------------------------------- class wxDataViewHeaderWindow : public wxHeaderCtrl { public: wxDataViewHeaderWindow(wxDataViewCtrl *parent) : wxHeaderCtrl(parent) { } wxDataViewCtrl *GetOwner() const { return static_cast(GetParent()); } protected: // implement/override wxHeaderCtrl functions by forwarding them to the main // control virtual const wxHeaderColumn& GetColumn(unsigned int idx) const { return *(GetOwner()->GetColumn(idx)); } virtual bool UpdateColumnWidthToFit(unsigned int idx, int widthTitle) { wxDataViewCtrl * const owner = GetOwner(); int widthContents = owner->GetBestColumnWidth(idx); owner->GetColumn(idx)->SetWidth(wxMax(widthTitle, widthContents)); owner->OnColumnChange(idx); return true; } private: void FinishEditing(); bool SendEvent(wxEventType type, unsigned int n) { wxDataViewCtrl * const owner = GetOwner(); wxDataViewEvent event(type, owner->GetId()); event.SetEventObject(owner); event.SetColumn(n); event.SetDataViewColumn(owner->GetColumn(n)); event.SetModel(owner->GetModel()); // for events created by wxDataViewHeaderWindow the // row / value fields are not valid return owner->ProcessWindowEvent(event); } void OnClick(wxHeaderCtrlEvent& event) { FinishEditing(); const unsigned idx = event.GetColumn(); if ( SendEvent(wxEVT_DATAVIEW_COLUMN_HEADER_CLICK, idx) ) return; // default handling for the column click is to sort by this column or // toggle its sort order wxDataViewCtrl * const owner = GetOwner(); wxDataViewColumn * const col = owner->GetColumn(idx); if ( !col->IsSortable() ) { // no default handling for non-sortable columns event.Skip(); return; } if ( col->IsSortKey() ) { // already using this column for sorting, just change the order col->ToggleSortOrder(); } else // not using this column for sorting yet { col->SetSortOrder(true); } wxDataViewModel * const model = owner->GetModel(); if ( model ) model->Resort(); owner->OnColumnChange(idx); SendEvent(wxEVT_DATAVIEW_COLUMN_SORTED, idx); } void OnRClick(wxHeaderCtrlEvent& event) { if ( !SendEvent(wxEVT_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK, event.GetColumn()) ) event.Skip(); } void OnResize(wxHeaderCtrlEvent& event) { FinishEditing(); wxDataViewCtrl * const owner = GetOwner(); const unsigned col = event.GetColumn(); owner->GetColumn(col)->SetWidth(event.GetWidth()); GetOwner()->OnColumnChange(col); } void OnEndReorder(wxHeaderCtrlEvent& event) { FinishEditing(); wxDataViewCtrl * const owner = GetOwner(); owner->ColumnMoved(owner->GetColumn(event.GetColumn()), event.GetNewOrder()); } DECLARE_EVENT_TABLE() wxDECLARE_NO_COPY_CLASS(wxDataViewHeaderWindow); }; BEGIN_EVENT_TABLE(wxDataViewHeaderWindow, wxHeaderCtrl) EVT_HEADER_CLICK(wxID_ANY, wxDataViewHeaderWindow::OnClick) EVT_HEADER_RIGHT_CLICK(wxID_ANY, wxDataViewHeaderWindow::OnRClick) EVT_HEADER_RESIZING(wxID_ANY, wxDataViewHeaderWindow::OnResize) EVT_HEADER_END_RESIZE(wxID_ANY, wxDataViewHeaderWindow::OnResize) EVT_HEADER_END_REORDER(wxID_ANY, wxDataViewHeaderWindow::OnEndReorder) END_EVENT_TABLE() //----------------------------------------------------------------------------- // wxDataViewRenameTimer //----------------------------------------------------------------------------- class wxDataViewRenameTimer: public wxTimer { private: wxDataViewMainWindow *m_owner; public: wxDataViewRenameTimer( wxDataViewMainWindow *owner ); void Notify(); }; //----------------------------------------------------------------------------- // wxDataViewTreeNode //----------------------------------------------------------------------------- class wxDataViewTreeNode; WX_DEFINE_ARRAY( wxDataViewTreeNode *, wxDataViewTreeNodes ); int LINKAGEMODE wxGenericTreeModelNodeCmp( wxDataViewTreeNode ** node1, wxDataViewTreeNode ** node2); class wxDataViewTreeNode { public: wxDataViewTreeNode(wxDataViewTreeNode *parent, const wxDataViewItem& item) : m_parent(parent), m_item(item), m_branchData(NULL) { } ~wxDataViewTreeNode() { if ( m_branchData ) { wxDataViewTreeNodes& nodes = m_branchData->children; for ( wxDataViewTreeNodes::iterator i = nodes.begin(); i != nodes.end(); ++i ) { delete *i; } delete m_branchData; } } static wxDataViewTreeNode* CreateRootNode() { wxDataViewTreeNode *n = new wxDataViewTreeNode(NULL, wxDataViewItem()); n->m_branchData = new BranchNodeData; n->m_branchData->open = true; return n; } wxDataViewTreeNode * GetParent() const { return m_parent; } const wxDataViewTreeNodes& GetChildNodes() const { wxASSERT( m_branchData != NULL ); return m_branchData->children; } void InsertChild(wxDataViewTreeNode *node, unsigned index) { if ( !m_branchData ) m_branchData = new BranchNodeData; m_branchData->children.Insert(node, index); // TODO: insert into sorted array directly in O(log n) instead of resorting in O(n log n) if (g_column >= -1) m_branchData->children.Sort( &wxGenericTreeModelNodeCmp ); } void RemoveChild(wxDataViewTreeNode *node) { wxCHECK_RET( m_branchData != NULL, "leaf node doesn't have children" ); m_branchData->children.Remove(node); } // returns position of child node for given item in children list or wxNOT_FOUND int FindChildByItem(const wxDataViewItem& item) const { if ( !m_branchData ) return wxNOT_FOUND; const wxDataViewTreeNodes& nodes = m_branchData->children; const int len = nodes.size(); for ( int i = 0; i < len; i++ ) { if ( nodes[i]->m_item == item ) return i; } return wxNOT_FOUND; } const wxDataViewItem & GetItem() const { return m_item; } void SetItem( const wxDataViewItem & item ) { m_item = item; } int GetIndentLevel() const { int ret = 0; const wxDataViewTreeNode * node = this; while( node->GetParent()->GetParent() != NULL ) { node = node->GetParent(); ret ++; } return ret; } bool IsOpen() const { return m_branchData && m_branchData->open; } void ToggleOpen() { // We do not allow the (invisible) root node to be collapsed because // there is no way to expand it again. if ( !m_parent ) return; wxCHECK_RET( m_branchData != NULL, "can't open leaf node" ); int sum = 0; const wxDataViewTreeNodes& nodes = m_branchData->children; const int len = nodes.GetCount(); for ( int i = 0;i < len; i ++) sum += 1 + nodes[i]->GetSubTreeCount(); if (m_branchData->open) { ChangeSubTreeCount(-sum); m_branchData->open = !m_branchData->open; } else { m_branchData->open = !m_branchData->open; ChangeSubTreeCount(+sum); } } // "HasChildren" property corresponds to model's IsContainer(). Note that it may be true // even if GetChildNodes() is empty; see below. bool HasChildren() const { return m_branchData != NULL; } void SetHasChildren(bool has) { // The invisible root item always has children, so ignore any attempts // to change this. if ( !m_parent ) return; if ( !has ) { wxDELETE(m_branchData); } else if ( m_branchData == NULL ) { m_branchData = new BranchNodeData; } } int GetSubTreeCount() const { return m_branchData ? m_branchData->subTreeCount : 0; } void ChangeSubTreeCount( int num ) { wxASSERT( m_branchData != NULL ); if( !m_branchData->open ) return; m_branchData->subTreeCount += num; wxASSERT( m_branchData->subTreeCount >= 0 ); if( m_parent ) m_parent->ChangeSubTreeCount(num); } void Resort() { if ( !m_branchData ) return; if (g_column >= -1) { wxDataViewTreeNodes& nodes = m_branchData->children; nodes.Sort( &wxGenericTreeModelNodeCmp ); int len = nodes.GetCount(); for (int i = 0; i < len; i ++) { if ( nodes[i]->HasChildren() ) nodes[i]->Resort(); } } } private: wxDataViewTreeNode *m_parent; // Corresponding model item. wxDataViewItem m_item; // Data specific to non-leaf (branch, inner) nodes. They are kept in a // separate struct in order to conserve memory. struct BranchNodeData { BranchNodeData() : open(false), subTreeCount(0) { } // Child nodes. Note that this may be empty even if m_hasChildren in // case this branch of the tree wasn't expanded and realized yet. wxDataViewTreeNodes children; // Is the branch node currently open (expanded)? bool open; // Total count of expanded (i.e. visible with the help of some // scrolling) items in the subtree, but excluding this node. I.e. it is // 0 for leaves and is the number of rows the subtree occupies for // branch nodes. int subTreeCount; }; BranchNodeData *m_branchData; }; int LINKAGEMODE wxGenericTreeModelNodeCmp( wxDataViewTreeNode ** node1, wxDataViewTreeNode ** node2) { return g_model->Compare( (*node1)->GetItem(), (*node2)->GetItem(), g_column, g_asending ); } //----------------------------------------------------------------------------- // wxDataViewMainWindow //----------------------------------------------------------------------------- WX_DEFINE_SORTED_ARRAY_SIZE_T(unsigned int, wxDataViewSelection); class wxDataViewMainWindow: public wxWindow { public: wxDataViewMainWindow( wxDataViewCtrl *parent, wxWindowID id, const wxPoint &pos = wxDefaultPosition, const wxSize &size = wxDefaultSize, const wxString &name = wxT("wxdataviewctrlmainwindow") ); virtual ~wxDataViewMainWindow(); bool IsList() const { return GetModel()->IsListModel(); } bool IsVirtualList() const { return m_root == NULL; } // notifications from wxDataViewModel bool ItemAdded( const wxDataViewItem &parent, const wxDataViewItem &item ); bool ItemDeleted( const wxDataViewItem &parent, const wxDataViewItem &item ); bool ItemChanged( const wxDataViewItem &item ); bool ValueChanged( const wxDataViewItem &item, unsigned int model_column ); bool Cleared(); void Resort() { if (!IsVirtualList()) { SortPrepare(); m_root->Resort(); } UpdateDisplay(); } // Override the base class method to resort if needed, i.e. if // SortPrepare() was called -- and ignored -- while we were frozen. virtual void DoThaw() { if ( g_column == SortColumn_OnThaw ) { Resort(); g_column = SortColumn_None; } wxWindow::DoThaw(); } void SortPrepare() { g_model = GetModel(); wxDataViewColumn* col = GetOwner()->GetSortingColumn(); if( !col ) { if (g_model->HasDefaultCompare()) { // See below for the explanation of IsFrozen() test. if ( IsFrozen() ) g_column = SortColumn_OnThaw; else g_column = SortColumn_Default; } else g_column = SortColumn_None; g_asending = true; return; } // Avoid sorting while the window is frozen, this allows to quickly add // many items without resorting after each addition and only resort // them all at once when the window is finally thawed, see above. if ( IsFrozen() ) { g_column = SortColumn_OnThaw; return; } g_column = col->GetModelColumn(); g_asending = col->IsSortOrderAscending(); } void SetOwner( wxDataViewCtrl* owner ) { m_owner = owner; } wxDataViewCtrl *GetOwner() { return m_owner; } const wxDataViewCtrl *GetOwner() const { return m_owner; } wxDataViewModel* GetModel() { return GetOwner()->GetModel(); } const wxDataViewModel* GetModel() const { return GetOwner()->GetModel(); } #if wxUSE_DRAG_AND_DROP wxBitmap CreateItemBitmap( unsigned int row, int &indent ); #endif // wxUSE_DRAG_AND_DROP void OnPaint( wxPaintEvent &event ); void OnCharHook( wxKeyEvent &event ); void OnChar( wxKeyEvent &event ); void OnVerticalNavigation(const wxKeyEvent& event, int delta); void OnLeftKey(wxKeyEvent& event); void OnRightKey(wxKeyEvent& event); void OnMouse( wxMouseEvent &event ); void OnSetFocus( wxFocusEvent &event ); void OnKillFocus( wxFocusEvent &event ); void UpdateDisplay(); void RecalculateDisplay(); void OnInternalIdle(); void OnRenameTimer(); void ScrollWindow( int dx, int dy, const wxRect *rect = NULL ); void ScrollTo( int rows, int column ); unsigned GetCurrentRow() const { return m_currentRow; } bool HasCurrentRow() { return m_currentRow != (unsigned int)-1; } void ChangeCurrentRow( unsigned int row ); bool TryAdvanceCurrentColumn(wxDataViewTreeNode *node, wxKeyEvent& event, bool forward); wxDataViewColumn *GetCurrentColumn() const { return m_currentCol; } void ClearCurrentColumn() { m_currentCol = NULL; } bool IsSingleSel() const { return !GetParent()->HasFlag(wxDV_MULTIPLE); } bool IsEmpty() { return GetRowCount() == 0; } int GetCountPerPage() const; int GetEndOfLastCol() const; unsigned int GetFirstVisibleRow() const; // 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 unsigned int GetLastVisibleRow(); unsigned int GetRowCount() const; const wxDataViewSelection& GetSelections() const { return m_selection; } void SetSelections( const wxDataViewSelection & sel ) { m_selection = sel; UpdateDisplay(); } void Select( const wxArrayInt& aSelections ); void SelectAllRows( bool on ); void SelectRow( unsigned int row, bool on ); void SelectRows( unsigned int from, unsigned int to, bool on ); void ReverseRowSelection( unsigned int row ); bool IsRowSelected( unsigned int row ); void SendSelectionChangedEvent( const wxDataViewItem& item); void RefreshRow( unsigned int row ); void RefreshRows( unsigned int from, unsigned int to ); void RefreshRowsAfter( unsigned int firstRow ); // returns the colour to be used for drawing the rules wxColour GetRuleColour() const { return wxSystemSettings::GetColour(wxSYS_COLOUR_3DLIGHT); } wxRect GetLineRect( unsigned int row ) const; int GetLineStart( unsigned int row ) const; // row * m_lineHeight in fixed mode int GetLineHeight( unsigned int row ) const; // m_lineHeight in fixed mode int GetLineAt( unsigned int y ) const; // y / m_lineHeight in fixed mode void SetRowHeight( int lineHeight ) { m_lineHeight = lineHeight; } int GetRowHeight() const { return m_lineHeight; } int GetDefaultRowHeight() const; // Some useful functions for row and item mapping wxDataViewItem GetItemByRow( unsigned int row ) const; int GetRowByItem( const wxDataViewItem & item ) const; wxDataViewTreeNode * GetTreeNodeByRow( unsigned int row ) const; // We did not need this temporarily // wxDataViewTreeNode * GetTreeNodeByItem( const wxDataViewItem & item ); // Methods for building the mapping tree void BuildTree( wxDataViewModel * model ); void DestroyTree(); void HitTest( const wxPoint & point, wxDataViewItem & item, wxDataViewColumn* &column ); wxRect GetItemRect( const wxDataViewItem & item, const wxDataViewColumn* column ); void Expand( unsigned int row ); void Collapse( unsigned int row ); bool IsExpanded( unsigned int row ) const; bool HasChildren( unsigned int row ) const; #if wxUSE_DRAG_AND_DROP bool EnableDragSource( const wxDataFormat &format ); bool EnableDropTarget( const wxDataFormat &format ); void RemoveDropHint(); wxDragResult OnDragOver( wxDataFormat format, wxCoord x, wxCoord y, wxDragResult def ); bool OnDrop( wxDataFormat format, wxCoord x, wxCoord y ); wxDragResult OnData( wxDataFormat format, wxCoord x, wxCoord y, wxDragResult def ); void OnLeave(); #endif // wxUSE_DRAG_AND_DROP void OnColumnsCountChanged(); // Adjust last column to window size void UpdateColumnSizes(); // Called by wxDataViewCtrl and our own OnRenameTimer() to start edit the // specified item in the given column. void StartEditing(const wxDataViewItem& item, const wxDataViewColumn* col); void FinishEditing(); private: int RecalculateCount() const; // Return false only if the event was vetoed by its handler. bool SendExpanderEvent(wxEventType type, const wxDataViewItem& item); wxDataViewTreeNode * FindNode( const wxDataViewItem & item ); wxDataViewColumn *FindColumnForEditing(const wxDataViewItem& item, wxDataViewCellMode mode); bool IsCellEditableInMode(const wxDataViewItem& item, const wxDataViewColumn *col, wxDataViewCellMode mode) const; void DrawCellBackground( wxDataViewRenderer* cell, wxDC& dc, const wxRect& rect ); private: wxDataViewCtrl *m_owner; int m_lineHeight; bool m_dirty; wxDataViewColumn *m_currentCol; unsigned int m_currentRow; wxDataViewSelection m_selection; wxDataViewRenameTimer *m_renameTimer; bool m_lastOnSame; bool m_hasFocus; bool m_useCellFocus; bool m_currentColSetByKeyboard; #if wxUSE_DRAG_AND_DROP int m_dragCount; wxPoint m_dragStart; bool m_dragEnabled; wxDataFormat m_dragFormat; bool m_dropEnabled; wxDataFormat m_dropFormat; bool m_dropHint; unsigned int m_dropHintLine; #endif // wxUSE_DRAG_AND_DROP // for double click logic unsigned int m_lineLastClicked, m_lineBeforeLastClicked, m_lineSelectSingleOnUp; // the pen used to draw horiz/vertical rules wxPen m_penRule; // the pen used to draw the expander and the lines wxPen m_penExpander; // This is the tree structure of the model wxDataViewTreeNode * m_root; int m_count; // This is the tree node under the cursor wxDataViewTreeNode * m_underMouse; // The control used for editing or NULL. wxWeakRef m_editorCtrl; // Id m_editorCtrl is non-NULL, pointer to the associated renderer. wxDataViewRenderer* m_editorRenderer; private: DECLARE_DYNAMIC_CLASS(wxDataViewMainWindow) DECLARE_EVENT_TABLE() }; // --------------------------------------------------------- // wxGenericDataViewModelNotifier // --------------------------------------------------------- class wxGenericDataViewModelNotifier: public wxDataViewModelNotifier { public: wxGenericDataViewModelNotifier( wxDataViewMainWindow *mainWindow ) { m_mainWindow = mainWindow; } virtual bool ItemAdded( const wxDataViewItem & parent, const wxDataViewItem & item ) { return m_mainWindow->ItemAdded( parent , item ); } virtual bool ItemDeleted( const wxDataViewItem &parent, const wxDataViewItem &item ) { return m_mainWindow->ItemDeleted( parent, item ); } virtual bool ItemChanged( const wxDataViewItem & item ) { return m_mainWindow->ItemChanged(item); } virtual bool ValueChanged( const wxDataViewItem & item , unsigned int col ) { return m_mainWindow->ValueChanged( item, col ); } virtual bool Cleared() { return m_mainWindow->Cleared(); } virtual void Resort() { m_mainWindow->Resort(); } wxDataViewMainWindow *m_mainWindow; }; // --------------------------------------------------------- // wxDataViewRenderer // --------------------------------------------------------- IMPLEMENT_ABSTRACT_CLASS(wxDataViewRenderer, wxDataViewRendererBase) wxDataViewRenderer::wxDataViewRenderer( const wxString &varianttype, wxDataViewCellMode mode, int align) : wxDataViewCustomRendererBase( varianttype, mode, align ) { m_align = align; m_mode = mode; m_ellipsizeMode = wxELLIPSIZE_MIDDLE; m_dc = NULL; } wxDataViewRenderer::~wxDataViewRenderer() { delete m_dc; } wxDC *wxDataViewRenderer::GetDC() { if (m_dc == NULL) { if (GetOwner() == NULL) return NULL; if (GetOwner()->GetOwner() == NULL) return NULL; m_dc = new wxClientDC( GetOwner()->GetOwner() ); } return m_dc; } void wxDataViewRenderer::SetAlignment( int align ) { m_align=align; } int wxDataViewRenderer::GetAlignment() const { return m_align; } // --------------------------------------------------------- // wxDataViewCustomRenderer // --------------------------------------------------------- IMPLEMENT_ABSTRACT_CLASS(wxDataViewCustomRenderer, wxDataViewRenderer) wxDataViewCustomRenderer::wxDataViewCustomRenderer( const wxString &varianttype, wxDataViewCellMode mode, int align ) : wxDataViewRenderer( varianttype, mode, align ) { } // --------------------------------------------------------- // wxDataViewTextRenderer // --------------------------------------------------------- IMPLEMENT_CLASS(wxDataViewTextRenderer, wxDataViewRenderer) wxDataViewTextRenderer::wxDataViewTextRenderer( const wxString &varianttype, wxDataViewCellMode mode, int align ) : wxDataViewRenderer( varianttype, mode, align ) { } bool wxDataViewTextRenderer::SetValue( const wxVariant &value ) { m_text = value.GetString(); return true; } bool wxDataViewTextRenderer::GetValue( wxVariant& WXUNUSED(value) ) const { return false; } bool wxDataViewTextRenderer::HasEditorCtrl() const { return true; } wxWindow* wxDataViewTextRenderer::CreateEditorCtrl( wxWindow *parent, wxRect labelRect, const wxVariant &value ) { return CreateEditorTextCtrl(parent, labelRect, value); } bool wxDataViewTextRenderer::GetValueFromEditorCtrl( wxWindow *editor, wxVariant &value ) { wxTextCtrl *text = (wxTextCtrl*) editor; value = text->GetValue(); return true; } bool wxDataViewTextRenderer::Render(wxRect rect, wxDC *dc, int state) { RenderText(m_text, 0, rect, dc, state); return true; } wxSize wxDataViewTextRenderer::GetSize() const { if (!m_text.empty()) return GetTextExtent(m_text); else return wxSize(wxDVC_DEFAULT_RENDERER_SIZE,wxDVC_DEFAULT_RENDERER_SIZE); } // --------------------------------------------------------- // wxDataViewBitmapRenderer // --------------------------------------------------------- IMPLEMENT_CLASS(wxDataViewBitmapRenderer, wxDataViewRenderer) wxDataViewBitmapRenderer::wxDataViewBitmapRenderer( const wxString &varianttype, wxDataViewCellMode mode, int align ) : wxDataViewRenderer( varianttype, mode, align ) { } bool wxDataViewBitmapRenderer::SetValue( const wxVariant &value ) { if (value.GetType() == wxT("wxBitmap")) m_bitmap << value; if (value.GetType() == wxT("wxIcon")) m_icon << value; return true; } bool wxDataViewBitmapRenderer::GetValue( wxVariant& WXUNUSED(value) ) const { return false; } bool wxDataViewBitmapRenderer::Render( wxRect cell, wxDC *dc, int WXUNUSED(state) ) { if (m_bitmap.IsOk()) dc->DrawBitmap( m_bitmap, cell.x, cell.y, true /* use mask */ ); else if (m_icon.IsOk()) dc->DrawIcon( m_icon, cell.x, cell.y ); return true; } wxSize wxDataViewBitmapRenderer::GetSize() const { if (m_bitmap.IsOk()) return wxSize( m_bitmap.GetWidth(), m_bitmap.GetHeight() ); else if (m_icon.IsOk()) return wxSize( m_icon.GetWidth(), m_icon.GetHeight() ); return wxSize(wxDVC_DEFAULT_RENDERER_SIZE,wxDVC_DEFAULT_RENDERER_SIZE); } // --------------------------------------------------------- // wxDataViewToggleRenderer // --------------------------------------------------------- IMPLEMENT_ABSTRACT_CLASS(wxDataViewToggleRenderer, wxDataViewRenderer) wxDataViewToggleRenderer::wxDataViewToggleRenderer( const wxString &varianttype, wxDataViewCellMode mode, int align ) : wxDataViewRenderer( varianttype, mode, align ) { m_toggle = false; } bool wxDataViewToggleRenderer::SetValue( const wxVariant &value ) { m_toggle = value.GetBool(); return true; } bool wxDataViewToggleRenderer::GetValue( wxVariant &WXUNUSED(value) ) const { return false; } bool wxDataViewToggleRenderer::Render( wxRect cell, wxDC *dc, int WXUNUSED(state) ) { int flags = 0; if (m_toggle) flags |= wxCONTROL_CHECKED; if (GetMode() != wxDATAVIEW_CELL_ACTIVATABLE || GetEnabled() == false) flags |= wxCONTROL_DISABLED; // Ensure that the check boxes always have at least the minimal required // size, otherwise DrawCheckBox() doesn't really work well. If this size is // greater than the cell size, the checkbox will be truncated but this is a // lesser evil. wxSize size = cell.GetSize(); size.IncTo(GetSize()); cell.SetSize(size); wxRendererNative::Get().DrawCheckBox( GetOwner()->GetOwner(), *dc, cell, flags ); return true; } bool wxDataViewToggleRenderer::WXActivateCell(const wxRect& WXUNUSED(cellRect), wxDataViewModel *model, const wxDataViewItem& item, unsigned int col, const wxMouseEvent *mouseEvent) { if ( mouseEvent ) { // Only react to clicks directly on the checkbox, not elsewhere in the // same cell. if ( !wxRect(GetSize()).Contains(mouseEvent->GetPosition()) ) return false; } model->ChangeValue(!m_toggle, item, col); return true; } wxSize wxDataViewToggleRenderer::GetSize() const { // the window parameter is not used by GetCheckBoxSize() so it's // safe to pass NULL return wxRendererNative::Get().GetCheckBoxSize(NULL); } // --------------------------------------------------------- // wxDataViewProgressRenderer // --------------------------------------------------------- IMPLEMENT_ABSTRACT_CLASS(wxDataViewProgressRenderer, wxDataViewRenderer) wxDataViewProgressRenderer::wxDataViewProgressRenderer( const wxString &label, const wxString &varianttype, wxDataViewCellMode mode, int align ) : wxDataViewRenderer( varianttype, mode, align ) { m_label = label; m_value = 0; } bool wxDataViewProgressRenderer::SetValue( const wxVariant &value ) { m_value = (long) value; if (m_value < 0) m_value = 0; if (m_value > 100) m_value = 100; return true; } bool wxDataViewProgressRenderer::GetValue( wxVariant &value ) const { value = (long) m_value; return true; } bool wxDataViewProgressRenderer::Render(wxRect rect, wxDC *dc, int WXUNUSED(state)) { // deflate the rect to leave a small border between bars in adjacent rows wxRect bar = rect.Deflate(0, 1); dc->SetBrush( *wxTRANSPARENT_BRUSH ); dc->SetPen( *wxBLACK_PEN ); dc->DrawRectangle( bar ); bar.width = (int)(bar.width * m_value / 100.); dc->SetPen( *wxTRANSPARENT_PEN ); const wxDataViewItemAttr& attr = GetAttr(); dc->SetBrush( attr.HasColour() ? wxBrush(attr.GetColour()) : *wxBLUE_BRUSH ); dc->DrawRectangle( bar ); return true; } wxSize wxDataViewProgressRenderer::GetSize() const { // Return -1 width because a progress bar fits any width; unlike most // renderers, it doesn't have a "good" width for the content. This makes it // grow to the whole column, which is pretty much always the desired // behaviour. Keep the height fixed so that the progress bar isn't too fat. return wxSize(-1, 12); } // --------------------------------------------------------- // wxDataViewIconTextRenderer // --------------------------------------------------------- IMPLEMENT_CLASS(wxDataViewIconTextRenderer, wxDataViewRenderer) wxDataViewIconTextRenderer::wxDataViewIconTextRenderer( const wxString &varianttype, wxDataViewCellMode mode, int align ) : wxDataViewRenderer( varianttype, mode, align ) { SetMode(mode); SetAlignment(align); } bool wxDataViewIconTextRenderer::SetValue( const wxVariant &value ) { m_value << value; return true; } bool wxDataViewIconTextRenderer::GetValue( wxVariant& WXUNUSED(value) ) const { return false; } bool wxDataViewIconTextRenderer::Render(wxRect rect, wxDC *dc, int state) { int xoffset = 0; const wxIcon& icon = m_value.GetIcon(); if ( icon.IsOk() ) { dc->DrawIcon(icon, rect.x, rect.y + (rect.height - icon.GetHeight())/2); xoffset = icon.GetWidth()+4; } RenderText(m_value.GetText(), xoffset, rect, dc, state); return true; } wxSize wxDataViewIconTextRenderer::GetSize() const { if (!m_value.GetText().empty()) { wxSize size = GetTextExtent(m_value.GetText()); if (m_value.GetIcon().IsOk()) size.x += m_value.GetIcon().GetWidth() + 4; return size; } return wxSize(80,20); } wxWindow* wxDataViewIconTextRenderer::CreateEditorCtrl(wxWindow *parent, wxRect labelRect, const wxVariant& value) { wxDataViewIconText iconText; iconText << value; wxString text = iconText.GetText(); // adjust the label rect to take the width of the icon into account if (iconText.GetIcon().IsOk()) { int w = iconText.GetIcon().GetWidth() + 4; labelRect.x += w; labelRect.width -= w; } return CreateEditorTextCtrl(parent, labelRect, text); } bool wxDataViewIconTextRenderer::GetValueFromEditorCtrl( wxWindow *editor, wxVariant& value ) { wxTextCtrl *text = (wxTextCtrl*) editor; // The icon can't be edited so get its old value and reuse it. wxVariant valueOld; wxDataViewColumn* const col = GetOwner(); GetView()->GetModel()->GetValue(valueOld, m_item, col->GetModelColumn()); wxDataViewIconText iconText; iconText << valueOld; // But replace the text with the value entered by user. iconText.SetText(text->GetValue()); value << iconText; return true; } //----------------------------------------------------------------------------- // wxDataViewDropTarget //----------------------------------------------------------------------------- #if wxUSE_DRAG_AND_DROP class wxBitmapCanvas: public wxWindow { public: wxBitmapCanvas( wxWindow *parent, const wxBitmap &bitmap, const wxSize &size ) : wxWindow( parent, wxID_ANY, wxPoint(0,0), size ) { m_bitmap = bitmap; Connect( wxEVT_PAINT, wxPaintEventHandler(wxBitmapCanvas::OnPaint) ); } void OnPaint( wxPaintEvent &WXUNUSED(event) ) { wxPaintDC dc(this); dc.DrawBitmap( m_bitmap, 0, 0); } wxBitmap m_bitmap; }; class wxDataViewDropSource: public wxDropSource { public: wxDataViewDropSource( wxDataViewMainWindow *win, unsigned int row ) : wxDropSource( win ) { m_win = win; m_row = row; m_hint = NULL; } ~wxDataViewDropSource() { delete m_hint; } virtual bool GiveFeedback( wxDragResult WXUNUSED(effect) ) { wxPoint pos = wxGetMousePosition(); if (!m_hint) { int liney = m_win->GetLineStart( m_row ); int linex = 0; m_win->GetOwner()->CalcUnscrolledPosition( 0, liney, NULL, &liney ); m_win->ClientToScreen( &linex, &liney ); m_dist_x = pos.x - linex; m_dist_y = pos.y - liney; int indent = 0; wxBitmap ib = m_win->CreateItemBitmap( m_row, indent ); m_dist_x -= indent; m_hint = new wxFrame( m_win->GetParent(), wxID_ANY, wxEmptyString, wxPoint(pos.x - m_dist_x, pos.y + 5 ), ib.GetSize(), wxFRAME_TOOL_WINDOW | wxFRAME_FLOAT_ON_PARENT | wxFRAME_NO_TASKBAR | wxNO_BORDER ); new wxBitmapCanvas( m_hint, ib, ib.GetSize() ); m_hint->Show(); } else { m_hint->Move( pos.x - m_dist_x, pos.y + 5 ); m_hint->SetTransparent( 128 ); } return false; } wxDataViewMainWindow *m_win; unsigned int m_row; wxFrame *m_hint; int m_dist_x,m_dist_y; }; class wxDataViewDropTarget: public wxDropTarget { public: wxDataViewDropTarget( wxDataObject *obj, wxDataViewMainWindow *win ) : wxDropTarget( obj ) { m_win = win; } virtual wxDragResult OnDragOver( wxCoord x, wxCoord y, wxDragResult def ) { wxDataFormat format = GetMatchingPair(); if (format == wxDF_INVALID) return wxDragNone; return m_win->OnDragOver( format, x, y, def); } virtual bool OnDrop( wxCoord x, wxCoord y ) { wxDataFormat format = GetMatchingPair(); if (format == wxDF_INVALID) return false; return m_win->OnDrop( format, x, y ); } virtual wxDragResult OnData( wxCoord x, wxCoord y, wxDragResult def ) { wxDataFormat format = GetMatchingPair(); if (format == wxDF_INVALID) return wxDragNone; if (!GetData()) return wxDragNone; return m_win->OnData( format, x, y, def ); } virtual void OnLeave() { m_win->OnLeave(); } wxDataViewMainWindow *m_win; }; #endif // wxUSE_DRAG_AND_DROP //----------------------------------------------------------------------------- // wxDataViewRenameTimer //----------------------------------------------------------------------------- wxDataViewRenameTimer::wxDataViewRenameTimer( wxDataViewMainWindow *owner ) { m_owner = owner; } void wxDataViewRenameTimer::Notify() { m_owner->OnRenameTimer(); } //----------------------------------------------------------------------------- // wxDataViewMainWindow //----------------------------------------------------------------------------- // The tree building helper, declared firstly static void BuildTreeHelper( const wxDataViewModel * model, const wxDataViewItem & item, wxDataViewTreeNode * node); int LINKAGEMODE wxDataViewSelectionCmp( unsigned int row1, unsigned int row2 ) { if (row1 > row2) return 1; if (row1 == row2) return 0; return -1; } IMPLEMENT_ABSTRACT_CLASS(wxDataViewMainWindow, wxWindow) BEGIN_EVENT_TABLE(wxDataViewMainWindow,wxWindow) EVT_PAINT (wxDataViewMainWindow::OnPaint) EVT_MOUSE_EVENTS (wxDataViewMainWindow::OnMouse) EVT_SET_FOCUS (wxDataViewMainWindow::OnSetFocus) EVT_KILL_FOCUS (wxDataViewMainWindow::OnKillFocus) EVT_CHAR_HOOK (wxDataViewMainWindow::OnCharHook) EVT_CHAR (wxDataViewMainWindow::OnChar) END_EVENT_TABLE() wxDataViewMainWindow::wxDataViewMainWindow( wxDataViewCtrl *parent, wxWindowID id, const wxPoint &pos, const wxSize &size, const wxString &name ) : wxWindow( parent, id, pos, size, wxWANTS_CHARS|wxBORDER_NONE, name ), m_selection( wxDataViewSelectionCmp ) { SetOwner( parent ); m_editorRenderer = NULL; m_lastOnSame = false; m_renameTimer = new wxDataViewRenameTimer( this ); // TODO: user better initial values/nothing selected m_currentCol = NULL; m_currentColSetByKeyboard = false; m_useCellFocus = false; m_currentRow = (unsigned)-1; m_lineHeight = GetDefaultRowHeight(); #if wxUSE_DRAG_AND_DROP m_dragCount = 0; m_dragStart = wxPoint(0,0); m_dragEnabled = false; m_dropEnabled = false; m_dropHint = false; m_dropHintLine = (unsigned int) -1; #endif // wxUSE_DRAG_AND_DROP m_lineLastClicked = (unsigned int) -1; m_lineBeforeLastClicked = (unsigned int) -1; m_lineSelectSingleOnUp = (unsigned int) -1; m_hasFocus = false; SetBackgroundColour( *wxWHITE ); SetBackgroundStyle(wxBG_STYLE_CUSTOM); m_penRule = wxPen(GetRuleColour()); // compose a pen whichcan draw black lines // TODO: maybe there is something system colour to use m_penExpander = wxPen(wxColour(0,0,0)); m_root = wxDataViewTreeNode::CreateRootNode(); // Make m_count = -1 will cause the class recaculate the real displaying number of rows. m_count = -1; m_underMouse = NULL; UpdateDisplay(); } wxDataViewMainWindow::~wxDataViewMainWindow() { DestroyTree(); delete m_renameTimer; } int wxDataViewMainWindow::GetDefaultRowHeight() const { #ifdef __WXMSW__ // We would like to use the same line height that Explorer uses. This is // different from standard ListView control since Vista. if ( wxGetWinVersion() >= wxWinVersion_Vista ) return wxMax(16, GetCharHeight()) + 6; // 16 = mini icon height else #endif // __WXMSW__ return wxMax(16, GetCharHeight()) + 1; // 16 = mini icon height } #if wxUSE_DRAG_AND_DROP bool wxDataViewMainWindow::EnableDragSource( const wxDataFormat &format ) { m_dragFormat = format; m_dragEnabled = format != wxDF_INVALID; return true; } bool wxDataViewMainWindow::EnableDropTarget( const wxDataFormat &format ) { m_dropFormat = format; m_dropEnabled = format != wxDF_INVALID; if (m_dropEnabled) SetDropTarget( new wxDataViewDropTarget( new wxCustomDataObject( format ), this ) ); return true; } void wxDataViewMainWindow::RemoveDropHint() { if (m_dropHint) { m_dropHint = false; RefreshRow( m_dropHintLine ); m_dropHintLine = (unsigned int) -1; } } wxDragResult wxDataViewMainWindow::OnDragOver( wxDataFormat format, wxCoord x, wxCoord y, wxDragResult def ) { int xx = x; int yy = y; m_owner->CalcUnscrolledPosition( xx, yy, &xx, &yy ); unsigned int row = GetLineAt( yy ); if ((row >= GetRowCount()) || (xx > GetEndOfLastCol())) { RemoveDropHint(); return wxDragNone; } wxDataViewItem item = GetItemByRow( row ); wxDataViewModel *model = GetModel(); wxDataViewEvent event( wxEVT_DATAVIEW_ITEM_DROP_POSSIBLE, m_owner->GetId() ); event.SetEventObject( m_owner ); event.SetItem( item ); event.SetModel( model ); event.SetDataFormat( format ); event.SetDropEffect( def ); if (!m_owner->HandleWindowEvent( event )) { RemoveDropHint(); return wxDragNone; } if (!event.IsAllowed()) { RemoveDropHint(); return wxDragNone; } if (m_dropHint && (row != m_dropHintLine)) RefreshRow( m_dropHintLine ); m_dropHint = true; m_dropHintLine = row; RefreshRow( row ); return def; } bool wxDataViewMainWindow::OnDrop( wxDataFormat format, wxCoord x, wxCoord y ) { RemoveDropHint(); int xx = x; int yy = y; m_owner->CalcUnscrolledPosition( xx, yy, &xx, &yy ); unsigned int row = GetLineAt( yy ); if ((row >= GetRowCount()) || (xx > GetEndOfLastCol())) return false; wxDataViewItem item = GetItemByRow( row ); wxDataViewModel *model = GetModel(); wxDataViewEvent event( wxEVT_DATAVIEW_ITEM_DROP_POSSIBLE, m_owner->GetId() ); event.SetEventObject( m_owner ); event.SetItem( item ); event.SetModel( model ); event.SetDataFormat( format ); if (!m_owner->HandleWindowEvent( event )) return false; if (!event.IsAllowed()) return false; return true; } wxDragResult wxDataViewMainWindow::OnData( wxDataFormat format, wxCoord x, wxCoord y, wxDragResult def ) { int xx = x; int yy = y; m_owner->CalcUnscrolledPosition( xx, yy, &xx, &yy ); unsigned int row = GetLineAt( yy ); if ((row >= GetRowCount()) || (xx > GetEndOfLastCol())) return wxDragNone; wxDataViewItem item = GetItemByRow( row ); wxDataViewModel *model = GetModel(); wxCustomDataObject *obj = (wxCustomDataObject *) GetDropTarget()->GetDataObject(); wxDataViewEvent event( wxEVT_DATAVIEW_ITEM_DROP, m_owner->GetId() ); event.SetEventObject( m_owner ); event.SetItem( item ); event.SetModel( model ); event.SetDataFormat( format ); event.SetDataSize( obj->GetSize() ); event.SetDataBuffer( obj->GetData() ); event.SetDropEffect( def ); if (!m_owner->HandleWindowEvent( event )) return wxDragNone; if (!event.IsAllowed()) return wxDragNone; return def; } void wxDataViewMainWindow::OnLeave() { RemoveDropHint(); } wxBitmap wxDataViewMainWindow::CreateItemBitmap( unsigned int row, int &indent ) { int height = GetLineHeight( row ); int width = 0; unsigned int cols = GetOwner()->GetColumnCount(); unsigned int col; for (col = 0; col < cols; col++) { wxDataViewColumn *column = GetOwner()->GetColumnAt(col); if (column->IsHidden()) continue; // skip it! width += column->GetWidth(); } indent = 0; if (!IsList()) { wxDataViewTreeNode *node = GetTreeNodeByRow(row); indent = GetOwner()->GetIndent() * node->GetIndentLevel(); indent = indent + m_lineHeight; // try to use the m_lineHeight as the expander space } width -= indent; wxBitmap bitmap( width, height ); wxMemoryDC dc( bitmap ); dc.SetFont( GetFont() ); dc.SetPen( *wxBLACK_PEN ); dc.SetBrush( *wxWHITE_BRUSH ); dc.DrawRectangle( 0,0,width,height ); wxDataViewModel *model = m_owner->GetModel(); wxDataViewColumn * const expander = GetExpanderColumnOrFirstOne(GetOwner()); int x = 0; for (col = 0; col < cols; col++) { wxDataViewColumn *column = GetOwner()->GetColumnAt( col ); wxDataViewRenderer *cell = column->GetRenderer(); if (column->IsHidden()) continue; // skip it! width = column->GetWidth(); if (column == expander) width -= indent; wxDataViewItem item = GetItemByRow( row ); cell->PrepareForItem(model, item, column->GetModelColumn()); wxRect item_rect(x, 0, width, height); item_rect.Deflate(PADDING_RIGHTLEFT, 0); // dc.SetClippingRegion( item_rect ); cell->WXCallRender(item_rect, &dc, 0); // dc.DestroyClippingRegion(); x += width; } return bitmap; } #endif // wxUSE_DRAG_AND_DROP // Draw focus rect for individual cell. Unlike native focus rect, we render // this in foreground text color (typically white) to enhance contrast and // make it visible. static void DrawSelectedCellFocusRect(wxDC& dc, const wxRect& rect) { // (This code is based on wxRendererGeneric::DrawFocusRect and modified.) // draw the pixels manually because the "dots" in wxPen with wxDOT style // may be short traits and not really dots // // note that to behave in the same manner as DrawRect(), we must exclude // the bottom and right borders from the rectangle wxCoord x1 = rect.GetLeft(), y1 = rect.GetTop(), x2 = rect.GetRight(), y2 = rect.GetBottom(); wxDCPenChanger pen(dc, wxSystemSettings::GetColour(wxSYS_COLOUR_HIGHLIGHTTEXT)); wxCoord z; for ( z = x1 + 1; z < x2; z += 2 ) dc.DrawPoint(z, rect.GetTop()); wxCoord shift = z == x2 ? 0 : 1; for ( z = y1 + shift; z < y2; z += 2 ) dc.DrawPoint(x2, z); shift = z == y2 ? 0 : 1; for ( z = x2 - shift; z > x1; z -= 2 ) dc.DrawPoint(z, y2); shift = z == x1 ? 0 : 1; for ( z = y2 - shift; z > y1; z -= 2 ) dc.DrawPoint(x1, z); } void wxDataViewMainWindow::OnPaint( wxPaintEvent &WXUNUSED(event) ) { wxDataViewModel *model = GetModel(); wxAutoBufferedPaintDC dc( this ); dc.SetBrush(GetOwner()->GetBackgroundColour()); dc.SetPen( *wxTRANSPARENT_PEN ); dc.DrawRectangle(GetClientSize()); if ( IsEmpty() ) { // No items to draw. return; } // prepare the DC GetOwner()->PrepareDC( dc ); dc.SetFont( GetFont() ); wxRect update = GetUpdateRegion().GetBox(); m_owner->CalcUnscrolledPosition( update.x, update.y, &update.x, &update.y ); // compute which items needs to be redrawn unsigned int item_start = GetLineAt( wxMax(0,update.y) ); unsigned int item_count = wxMin( (int)( GetLineAt( wxMax(0,update.y+update.height) ) - item_start + 1), (int)(GetRowCount( ) - item_start)); unsigned int item_last = item_start + item_count; // Send the event to wxDataViewCtrl itself. wxWindow * const parent = GetParent(); wxDataViewEvent cache_event(wxEVT_DATAVIEW_CACHE_HINT, parent->GetId()); cache_event.SetEventObject(parent); cache_event.SetCache(item_start, item_last - 1); parent->ProcessWindowEvent(cache_event); // compute which columns needs to be redrawn unsigned int cols = GetOwner()->GetColumnCount(); if ( !cols ) { // we assume that we have at least one column below and painting an // empty control is unnecessary anyhow return; } unsigned int col_start = 0; unsigned int x_start; for (x_start = 0; col_start < cols; col_start++) { wxDataViewColumn *col = GetOwner()->GetColumnAt(col_start); if (col->IsHidden()) continue; // skip it! unsigned int w = col->GetWidth(); if (x_start+w >= (unsigned int)update.x) break; x_start += w; } unsigned int col_last = col_start; unsigned int x_last = x_start; for (; col_last < cols; col_last++) { wxDataViewColumn *col = GetOwner()->GetColumnAt(col_last); if (col->IsHidden()) continue; // skip it! if (x_last > (unsigned int)update.GetRight()) break; x_last += col->GetWidth(); } // Draw background of alternate rows specially if required if ( m_owner->HasFlag(wxDV_ROW_LINES) ) { wxColour altRowColour = m_owner->m_alternateRowColour; if ( !altRowColour.IsOk() ) { // Determine the alternate rows colour automatically from the // background colour. const wxColour bgColour = m_owner->GetBackgroundColour(); // Depending on the background, alternate row color // will be 3% more dark or 50% brighter. int alpha = bgColour.GetRGB() > 0x808080 ? 97 : 150; altRowColour = bgColour.ChangeLightness(alpha); } dc.SetPen(*wxTRANSPARENT_PEN); dc.SetBrush(wxBrush(altRowColour)); for (unsigned int item = item_start; item < item_last; item++) { if ( item % 2 ) { dc.DrawRectangle(x_start, GetLineStart(item), GetClientSize().GetWidth(), GetLineHeight(item)); } } } // Draw horizontal rules if required if ( m_owner->HasFlag(wxDV_HORIZ_RULES) ) { dc.SetPen(m_penRule); dc.SetBrush(*wxTRANSPARENT_BRUSH); for (unsigned int i = item_start; i <= item_last; i++) { int y = GetLineStart( i ); dc.DrawLine(x_start, y, x_last, y); } } // Draw vertical rules if required if ( m_owner->HasFlag(wxDV_VERT_RULES) ) { dc.SetPen(m_penRule); dc.SetBrush(*wxTRANSPARENT_BRUSH); // NB: Vertical rules are drawn in the last pixel of a column so that // they align perfectly with native MSW wxHeaderCtrl as well as for // consistency with MSW native list control. There's no vertical // rule at the most-left side of the control. int x = x_start - 1; for (unsigned int i = col_start; i < col_last; i++) { wxDataViewColumn *col = GetOwner()->GetColumnAt(i); if (col->IsHidden()) continue; // skip it x += col->GetWidth(); dc.DrawLine(x, GetLineStart( item_start ), x, GetLineStart( item_last ) ); } } // redraw the background for the items which are selected/current for (unsigned int item = item_start; item < item_last; item++) { bool selected = m_selection.Index( item ) != wxNOT_FOUND; if (selected || item == m_currentRow) { wxRect rect( x_start, GetLineStart( item ), x_last - x_start, GetLineHeight( item ) ); // draw selection and whole-item focus: if ( selected ) { int flags = wxCONTROL_SELECTED; if (m_hasFocus) flags |= wxCONTROL_FOCUSED; wxRendererNative::Get().DrawItemSelectionRect ( this, dc, rect, flags ); } // draw keyboard focus rect if applicable if ( item == m_currentRow && m_hasFocus ) { bool renderColumnFocus = false; if ( m_useCellFocus && m_currentCol && m_currentColSetByKeyboard ) { renderColumnFocus = true; // If this is container node without columns, render full-row focus: if ( !IsList() ) { wxDataViewTreeNode *node = GetTreeNodeByRow(item); if ( node->HasChildren() && !model->HasContainerColumns(node->GetItem()) ) renderColumnFocus = false; } } if ( renderColumnFocus ) { for ( unsigned int i = col_start; i < col_last; i++ ) { wxDataViewColumn *col = GetOwner()->GetColumnAt(i); if ( col->IsHidden() ) continue; rect.width = col->GetWidth(); if ( col == m_currentCol ) { // make the rect more visible by adding a small // margin around it: rect.Deflate(1, 1); if ( selected ) { // DrawFocusRect() uses XOR and is all but // invisible against dark-blue background. Use // the same color used for selected text. DrawSelectedCellFocusRect(dc, rect); } else { wxRendererNative::Get().DrawFocusRect ( this, dc, rect, 0 ); } break; } rect.x += rect.width; } } else { // render focus rectangle for the whole row wxRendererNative::Get().DrawFocusRect ( this, dc, rect, selected ? (int)wxCONTROL_SELECTED : 0 ); } } } } #if wxUSE_DRAG_AND_DROP if (m_dropHint) { wxRect rect( x_start, GetLineStart( m_dropHintLine ), x_last - x_start, GetLineHeight( m_dropHintLine ) ); dc.SetPen( *wxBLACK_PEN ); dc.SetBrush( *wxTRANSPARENT_BRUSH ); dc.DrawRectangle( rect ); } #endif // wxUSE_DRAG_AND_DROP wxDataViewColumn * const expander = GetExpanderColumnOrFirstOne(GetOwner()); // redraw all cells for all rows which must be repainted and all columns wxRect cell_rect; cell_rect.x = x_start; for (unsigned int i = col_start; i < col_last; i++) { wxDataViewColumn *col = GetOwner()->GetColumnAt( i ); wxDataViewRenderer *cell = col->GetRenderer(); cell_rect.width = col->GetWidth(); if ( col->IsHidden() || cell_rect.width <= 0 ) continue; // skip it! for (unsigned int item = item_start; item < item_last; item++) { // get the cell value and set it into the renderer wxDataViewTreeNode *node = NULL; wxDataViewItem dataitem; if (!IsVirtualList()) { node = GetTreeNodeByRow(item); if( node == NULL ) continue; dataitem = node->GetItem(); // Skip all columns of "container" rows except the expander // column itself unless HasContainerColumns() overrides this. if ( col != expander && model->IsContainer(dataitem) && !model->HasContainerColumns(dataitem) ) continue; } else { dataitem = wxDataViewItem( wxUIntToPtr(item+1) ); } cell->PrepareForItem(model, dataitem, col->GetModelColumn()); // update cell_rect cell_rect.y = GetLineStart( item ); cell_rect.height = GetLineHeight( item ); // draw the background bool selected = m_selection.Index( item ) != wxNOT_FOUND; if ( !selected ) DrawCellBackground( cell, dc, cell_rect ); // deal with the expander int indent = 0; if ((!IsList()) && (col == expander)) { // Calculate the indent first indent = GetOwner()->GetIndent() * node->GetIndentLevel(); // we reserve m_lineHeight of horizontal space for the expander // but leave EXPANDER_MARGIN around the expander itself int exp_x = cell_rect.x + indent + EXPANDER_MARGIN; indent += m_lineHeight; // draw expander if needed and visible if ( node->HasChildren() && exp_x < cell_rect.GetRight() ) { dc.SetPen( m_penExpander ); dc.SetBrush( wxNullBrush ); int exp_size = m_lineHeight - 2*EXPANDER_MARGIN; int exp_y = cell_rect.y + (cell_rect.height - exp_size)/2 + EXPANDER_MARGIN - EXPANDER_OFFSET; const wxRect rect(exp_x, exp_y, exp_size, exp_size); int flag = 0; if ( m_underMouse == node ) flag |= wxCONTROL_CURRENT; if ( node->IsOpen() ) flag |= wxCONTROL_EXPANDED; // ensure that we don't overflow the cell (which might // happen if the column is very narrow) wxDCClipper clip(dc, cell_rect); wxRendererNative::Get().DrawTreeItemButton( this, dc, rect, flag); } // force the expander column to left-center align cell->SetAlignment( wxALIGN_CENTER_VERTICAL ); } wxRect item_rect = cell_rect; item_rect.Deflate(PADDING_RIGHTLEFT, 0); // account for the tree indent (harmless if we're not indented) item_rect.x += indent; item_rect.width -= indent; if ( item_rect.width <= 0 ) continue; int state = 0; if (m_hasFocus && selected) state |= wxDATAVIEW_CELL_SELECTED; // TODO: it would be much more efficient to create a clipping // region for the entire column being rendered (in the OnPaint // of wxDataViewMainWindow) instead of a single clip region for // each cell. However it would mean that each renderer should // respect the given wxRect's top & bottom coords, eventually // violating only the left & right coords - however the user can // make its own renderer and thus we cannot be sure of that. wxDCClipper clip(dc, item_rect); cell->WXCallRender(item_rect, &dc, state); } cell_rect.x += cell_rect.width; } } void wxDataViewMainWindow::DrawCellBackground( wxDataViewRenderer* cell, wxDC& dc, const wxRect& rect ) { wxRect rectBg( rect ); // don't overlap the horizontal rules if ( m_owner->HasFlag(wxDV_HORIZ_RULES) ) { rectBg.x++; rectBg.width--; } // don't overlap the vertical rules if ( m_owner->HasFlag(wxDV_VERT_RULES) ) { rectBg.y++; rectBg.height--; } cell->RenderBackground(&dc, rectBg); } void wxDataViewMainWindow::OnRenameTimer() { // We have to call this here because changes may just have // been made and no screen update taken place. if ( m_dirty ) { // TODO: use wxTheApp->SafeYieldFor(NULL, wxEVT_CATEGORY_UI) instead // (needs to be tested!) wxSafeYield(); } wxDataViewItem item = GetItemByRow( m_currentRow ); StartEditing( item, m_currentCol ); } void wxDataViewMainWindow::StartEditing(const wxDataViewItem& item, const wxDataViewColumn* col) { wxDataViewRenderer* renderer = col->GetRenderer(); if ( !IsCellEditableInMode(item, col, wxDATAVIEW_CELL_EDITABLE) ) return; const wxRect itemRect = GetItemRect(item, col); if ( renderer->StartEditing(item, itemRect) ) { // Save the renderer to be able to finish/cancel editing it later and // save the control to be able to detect if we're still editing it. m_editorRenderer = renderer; m_editorCtrl = renderer->GetEditorCtrl(); } } void wxDataViewMainWindow::FinishEditing() { if ( m_editorCtrl ) { m_editorRenderer->FinishEditing(); } } void wxDataViewHeaderWindow::FinishEditing() { wxDataViewMainWindow *win = static_cast(GetOwner()->GetMainWindow()); win->FinishEditing(); } //----------------------------------------------------------------------------- // Helper class for do operation on the tree node //----------------------------------------------------------------------------- class DoJob { public: DoJob() { } virtual ~DoJob() { } // The return value control how the tree-walker tranverse the tree enum { DONE, // Job done, stop traversing and return SKIP_SUBTREE, // Ignore the current node's subtree and continue CONTINUE // Job not done, continue }; virtual int operator() ( wxDataViewTreeNode * node ) = 0; }; bool Walker( wxDataViewTreeNode * node, DoJob & func ) { wxCHECK_MSG( node, false, "can't walk NULL node" ); switch( func( node ) ) { case DoJob::DONE: return true; case DoJob::SKIP_SUBTREE: return false; case DoJob::CONTINUE: break; } if ( node->HasChildren() ) { const wxDataViewTreeNodes& nodes = node->GetChildNodes(); for ( wxDataViewTreeNodes::const_iterator i = nodes.begin(); i != nodes.end(); ++i ) { if ( Walker(*i, func) ) return true; } } return false; } bool wxDataViewMainWindow::ItemAdded(const wxDataViewItem & parent, const wxDataViewItem & item) { if (IsVirtualList()) { wxDataViewVirtualListModel *list_model = (wxDataViewVirtualListModel*) GetModel(); m_count = list_model->GetCount(); } else { SortPrepare(); wxDataViewTreeNode *parentNode = FindNode(parent); if ( !parentNode ) return false; wxDataViewItemArray modelSiblings; GetModel()->GetChildren(parent, modelSiblings); const int modelSiblingsSize = modelSiblings.size(); int posInModel = modelSiblings.Index(item, /*fromEnd=*/true); wxCHECK_MSG( posInModel != wxNOT_FOUND, false, "adding non-existent item?" ); wxDataViewTreeNode *itemNode = new wxDataViewTreeNode(parentNode, item); itemNode->SetHasChildren(GetModel()->IsContainer(item)); parentNode->SetHasChildren(true); const wxDataViewTreeNodes& nodeSiblings = parentNode->GetChildNodes(); const int nodeSiblingsSize = nodeSiblings.size(); int nodePos = 0; if ( posInModel == modelSiblingsSize - 1 ) { nodePos = nodeSiblingsSize; } else if ( modelSiblingsSize == nodeSiblingsSize + 1 ) { // This is the simple case when our node tree already matches the // model and only this one item is missing. nodePos = posInModel; } else { // It's possible that a larger discrepancy between the model and // our realization exists. This can happen e.g. when adding a bunch // of items to the model and then calling ItemsAdded() just once // afterwards. In this case, we must find the right position by // looking at sibling items. // append to the end if we won't find a better position: nodePos = nodeSiblingsSize; for ( int nextItemPos = posInModel + 1; nextItemPos < modelSiblingsSize; nextItemPos++ ) { int nextNodePos = parentNode->FindChildByItem(modelSiblings[nextItemPos]); if ( nextNodePos != wxNOT_FOUND ) { nodePos = nextNodePos; break; } } } parentNode->ChangeSubTreeCount(+1); parentNode->InsertChild(itemNode, nodePos); m_count = -1; } GetOwner()->InvalidateColBestWidths(); UpdateDisplay(); return true; } bool wxDataViewMainWindow::ItemDeleted(const wxDataViewItem& parent, const wxDataViewItem& item) { if (IsVirtualList()) { wxDataViewVirtualListModel *list_model = (wxDataViewVirtualListModel*) GetModel(); m_count = list_model->GetCount(); if ( !m_selection.empty() ) { const int row = GetRowByItem(item); int rowIndexInSelection = wxNOT_FOUND; const size_t selCount = m_selection.size(); for ( size_t i = 0; i < selCount; i++ ) { if ( m_selection[i] == (unsigned)row ) rowIndexInSelection = i; else if ( m_selection[i] > (unsigned)row ) m_selection[i]--; } if ( rowIndexInSelection != wxNOT_FOUND ) m_selection.RemoveAt(rowIndexInSelection); } } else // general case { wxDataViewTreeNode *parentNode = FindNode(parent); // Notice that it is possible that the item being deleted is not in the // tree at all, for example we could be deleting a never shown (because // collapsed) item in a tree model. So it's not an error if we don't know // about this item, just return without doing anything then. if ( !parentNode ) return true; wxCHECK_MSG( parentNode->HasChildren(), false, "parent node doesn't have children?" ); const wxDataViewTreeNodes& parentsChildren = parentNode->GetChildNodes(); // We can't use FindNode() to find 'item', because it was already // removed from the model by the time ItemDeleted() is called, so we // have to do it manually. We keep track of its position as well for // later use. int itemPosInNode = 0; wxDataViewTreeNode *itemNode = NULL; for ( wxDataViewTreeNodes::const_iterator i = parentsChildren.begin(); i != parentsChildren.end(); ++i, ++itemPosInNode ) { if( (*i)->GetItem() == item ) { itemNode = *i; break; } } // If the parent wasn't expanded, it's possible that we didn't have a // node corresponding to 'item' and so there's nothing left to do. if ( !itemNode ) { // If this was the last child to be removed, it's possible the parent // node became a leaf. Let's ask the model about it. if ( parentNode->GetChildNodes().empty() ) parentNode->SetHasChildren(GetModel()->IsContainer(parent)); return true; } // Delete the item from wxDataViewTreeNode representation: const int itemsDeleted = 1 + itemNode->GetSubTreeCount(); parentNode->RemoveChild(itemNode); delete itemNode; parentNode->ChangeSubTreeCount(-itemsDeleted); // Make the row number invalid and get a new valid one when user call GetRowCount m_count = -1; // If this was the last child to be removed, it's possible the parent // node became a leaf. Let's ask the model about it. if ( parentNode->GetChildNodes().empty() ) { bool isContainer = GetModel()->IsContainer(parent); parentNode->SetHasChildren(isContainer); if ( isContainer ) { // If it's still a container, make sure we show "+" icon for it // and not "-" one as there is nothing to collapse any more. if ( parentNode->IsOpen() ) parentNode->ToggleOpen(); } } // Update selection by removing 'item' and its entire children tree from the selection. if ( !m_selection.empty() ) { // we can't call GetRowByItem() on 'item', as it's already deleted, so compute it from // the parent ('parentNode') and position in its list of children int itemRow; if ( itemPosInNode == 0 ) { // 1st child, row number is that of the parent parentNode + 1 itemRow = GetRowByItem(parentNode->GetItem()) + 1; } else { // row number is that of the sibling above 'item' + its subtree if any + 1 const wxDataViewTreeNode *siblingNode = parentNode->GetChildNodes()[itemPosInNode - 1]; itemRow = GetRowByItem(siblingNode->GetItem()) + siblingNode->GetSubTreeCount() + 1; } wxDataViewSelection newsel(wxDataViewSelectionCmp); const size_t numSelections = m_selection.size(); for ( size_t i = 0; i < numSelections; ++i ) { const int s = m_selection[i]; if ( s < itemRow ) newsel.push_back(s); else if ( s >= itemRow + itemsDeleted ) newsel.push_back(s - itemsDeleted); // else: deleted item, remove from selection } m_selection = newsel; } } // Change the current row to the last row if the current exceed the max row number if ( m_currentRow >= GetRowCount() ) ChangeCurrentRow(m_count - 1); GetOwner()->InvalidateColBestWidths(); UpdateDisplay(); return true; } bool wxDataViewMainWindow::ItemChanged(const wxDataViewItem & item) { SortPrepare(); g_model->Resort(); GetOwner()->InvalidateColBestWidths(); // Send event wxWindow *parent = GetParent(); wxDataViewEvent le(wxEVT_DATAVIEW_ITEM_VALUE_CHANGED, parent->GetId()); le.SetEventObject(parent); le.SetModel(GetModel()); le.SetItem(item); parent->ProcessWindowEvent(le); return true; } bool wxDataViewMainWindow::ValueChanged( const wxDataViewItem & item, unsigned int model_column ) { int view_column = m_owner->GetModelColumnIndex(model_column); if ( view_column == wxNOT_FOUND ) return false; // NOTE: to be valid, we cannot use e.g. INT_MAX - 1 /*#define MAX_VIRTUAL_WIDTH 100000 wxRect rect( 0, row*m_lineHeight, MAX_VIRTUAL_WIDTH, m_lineHeight ); m_owner->CalcScrolledPosition( rect.x, rect.y, &rect.x, &rect.y ); Refresh( true, &rect ); return true; */ SortPrepare(); g_model->Resort(); GetOwner()->InvalidateColBestWidth(view_column); // Send event wxWindow *parent = GetParent(); wxDataViewEvent le(wxEVT_DATAVIEW_ITEM_VALUE_CHANGED, parent->GetId()); le.SetEventObject(parent); le.SetModel(GetModel()); le.SetItem(item); le.SetColumn(view_column); le.SetDataViewColumn(GetOwner()->GetColumn(view_column)); parent->ProcessWindowEvent(le); return true; } bool wxDataViewMainWindow::Cleared() { DestroyTree(); m_selection.Clear(); m_currentRow = (unsigned)-1; if (GetModel()) { SortPrepare(); BuildTree( GetModel() ); } else { m_count = 0; } GetOwner()->InvalidateColBestWidths(); UpdateDisplay(); return true; } void wxDataViewMainWindow::UpdateDisplay() { m_dirty = true; m_underMouse = NULL; } void wxDataViewMainWindow::OnInternalIdle() { wxWindow::OnInternalIdle(); if (m_dirty) { UpdateColumnSizes(); RecalculateDisplay(); m_dirty = false; } } void wxDataViewMainWindow::RecalculateDisplay() { wxDataViewModel *model = GetModel(); if (!model) { Refresh(); return; } int width = GetEndOfLastCol(); int height = GetLineStart( GetRowCount() ); SetVirtualSize( width, height ); GetOwner()->SetScrollRate( 10, m_lineHeight ); Refresh(); } void wxDataViewMainWindow::ScrollWindow( int dx, int dy, const wxRect *rect ) { m_underMouse = NULL; wxWindow::ScrollWindow( dx, dy, rect ); if (GetOwner()->m_headerArea) GetOwner()->m_headerArea->ScrollWindow( dx, 0 ); } void wxDataViewMainWindow::ScrollTo( int rows, int column ) { m_underMouse = NULL; int x, y; m_owner->GetScrollPixelsPerUnit( &x, &y ); int sy = GetLineStart( rows )/y; int sx = -1; if( column != -1 ) { wxRect rect = GetClientRect(); int colnum = 0; int x_start, w = 0; int xx, yy, xe; m_owner->CalcUnscrolledPosition( rect.x, rect.y, &xx, &yy ); for (x_start = 0; colnum < column; colnum++) { wxDataViewColumn *col = GetOwner()->GetColumnAt(colnum); if (col->IsHidden()) continue; // skip it! w = col->GetWidth(); x_start += w; } int x_end = x_start + w; xe = xx + rect.width; if( x_end > xe ) { sx = ( xx + x_end - xe )/x; } if( x_start < xx ) { sx = x_start/x; } } m_owner->Scroll( sx, sy ); } int wxDataViewMainWindow::GetCountPerPage() const { wxSize size = GetClientSize(); return size.y / m_lineHeight; } int wxDataViewMainWindow::GetEndOfLastCol() const { int width = 0; unsigned int i; for (i = 0; i < GetOwner()->GetColumnCount(); i++) { const wxDataViewColumn *c = const_cast(GetOwner())->GetColumnAt( i ); if (!c->IsHidden()) width += c->GetWidth(); } return width; } unsigned int wxDataViewMainWindow::GetFirstVisibleRow() const { int x = 0; int y = 0; m_owner->CalcUnscrolledPosition( x, y, &x, &y ); return GetLineAt( y ); } unsigned int wxDataViewMainWindow::GetLastVisibleRow() { wxSize client_size = GetClientSize(); m_owner->CalcUnscrolledPosition( client_size.x, client_size.y, &client_size.x, &client_size.y ); // we should deal with the pixel here unsigned int row = GetLineAt(client_size.y) - 1; return wxMin( GetRowCount()-1, row ); } unsigned int wxDataViewMainWindow::GetRowCount() const { if ( m_count == -1 ) { wxDataViewMainWindow* const self = const_cast(this); self->m_count = RecalculateCount(); self->UpdateDisplay(); } return m_count; } void wxDataViewMainWindow::ChangeCurrentRow( unsigned int row ) { m_currentRow = row; // send event } void wxDataViewMainWindow::SelectAllRows( bool on ) { if (IsEmpty()) return; if (on) { m_selection.Clear(); for (unsigned int i = 0; i < GetRowCount(); i++) m_selection.Add( i ); Refresh(); } else { unsigned int first_visible = GetFirstVisibleRow(); unsigned int last_visible = GetLastVisibleRow(); unsigned int i; for (i = 0; i < m_selection.GetCount(); i++) { unsigned int row = m_selection[i]; if ((row >= first_visible) && (row <= last_visible)) RefreshRow( row ); } m_selection.Clear(); } } void wxDataViewMainWindow::SelectRow( unsigned int row, bool on ) { if (m_selection.Index( row ) == wxNOT_FOUND) { if (on) { m_selection.Add( row ); RefreshRow( row ); } } else { if (!on) { m_selection.Remove( row ); RefreshRow( row ); } } } void wxDataViewMainWindow::SelectRows( unsigned int from, unsigned int to, bool on ) { if (from > to) { unsigned int tmp = from; from = to; to = tmp; } unsigned int i; for (i = from; i <= to; i++) { if (m_selection.Index( i ) == wxNOT_FOUND) { if (on) m_selection.Add( i ); } else { if (!on) m_selection.Remove( i ); } } RefreshRows( from, to ); } void wxDataViewMainWindow::Select( const wxArrayInt& aSelections ) { for (size_t i=0; i < aSelections.GetCount(); i++) { int n = aSelections[i]; m_selection.Add( n ); RefreshRow( n ); } } void wxDataViewMainWindow::ReverseRowSelection( unsigned int row ) { if (m_selection.Index( row ) == wxNOT_FOUND) m_selection.Add( row ); else m_selection.Remove( row ); RefreshRow( row ); } bool wxDataViewMainWindow::IsRowSelected( unsigned int row ) { return (m_selection.Index( row ) != wxNOT_FOUND); } void wxDataViewMainWindow::SendSelectionChangedEvent( const wxDataViewItem& item) { wxWindow *parent = GetParent(); wxDataViewEvent le(wxEVT_DATAVIEW_SELECTION_CHANGED, parent->GetId()); le.SetEventObject(parent); le.SetModel(GetModel()); le.SetItem( item ); parent->ProcessWindowEvent(le); } void wxDataViewMainWindow::RefreshRow( unsigned int row ) { wxRect rect( 0, GetLineStart( row ), GetEndOfLastCol(), GetLineHeight( row ) ); m_owner->CalcScrolledPosition( rect.x, rect.y, &rect.x, &rect.y ); wxSize client_size = GetClientSize(); wxRect client_rect( 0, 0, client_size.x, client_size.y ); wxRect intersect_rect = client_rect.Intersect( rect ); if (intersect_rect.width > 0) Refresh( true, &intersect_rect ); } void wxDataViewMainWindow::RefreshRows( unsigned int from, unsigned int to ) { if (from > to) { unsigned int tmp = to; to = from; from = tmp; } wxRect rect( 0, GetLineStart( from ), GetEndOfLastCol(), GetLineStart( (to-from+1) ) ); m_owner->CalcScrolledPosition( rect.x, rect.y, &rect.x, &rect.y ); wxSize client_size = GetClientSize(); wxRect client_rect( 0, 0, client_size.x, client_size.y ); wxRect intersect_rect = client_rect.Intersect( rect ); if (intersect_rect.width > 0) Refresh( true, &intersect_rect ); } void wxDataViewMainWindow::RefreshRowsAfter( unsigned int firstRow ) { wxSize client_size = GetClientSize(); int start = GetLineStart( firstRow ); m_owner->CalcScrolledPosition( start, 0, &start, NULL ); if (start > client_size.y) return; wxRect rect( 0, start, client_size.x, client_size.y - start ); Refresh( true, &rect ); } wxRect wxDataViewMainWindow::GetLineRect( unsigned int row ) const { wxRect rect; rect.x = 0; rect.y = GetLineStart( row ); rect.width = GetEndOfLastCol(); rect.height = GetLineHeight( row ); return rect; } int wxDataViewMainWindow::GetLineStart( unsigned int row ) const { const wxDataViewModel *model = GetModel(); if (GetOwner()->GetWindowStyle() & wxDV_VARIABLE_LINE_HEIGHT) { // TODO make more efficient int start = 0; unsigned int r; for (r = 0; r < row; r++) { const wxDataViewTreeNode* node = GetTreeNodeByRow(r); if (!node) return start; wxDataViewItem item = node->GetItem(); unsigned int cols = GetOwner()->GetColumnCount(); unsigned int col; int height = m_lineHeight; for (col = 0; col < cols; col++) { const wxDataViewColumn *column = GetOwner()->GetColumn(col); if (column->IsHidden()) continue; // skip it! if ((col != 0) && model->IsContainer(item) && !model->HasContainerColumns(item)) continue; // skip it! wxDataViewRenderer *renderer = const_cast(column->GetRenderer()); renderer->PrepareForItem(model, item, column->GetModelColumn()); height = wxMax( height, renderer->GetSize().y ); } start += height; } return start; } else { return row * m_lineHeight; } } int wxDataViewMainWindow::GetLineAt( unsigned int y ) const { const wxDataViewModel *model = GetModel(); // check for the easy case first if ( !GetOwner()->HasFlag(wxDV_VARIABLE_LINE_HEIGHT) ) return y / m_lineHeight; // TODO make more efficient unsigned int row = 0; unsigned int yy = 0; for (;;) { const wxDataViewTreeNode* node = GetTreeNodeByRow(row); if (!node) { // not really correct... return row + ((y-yy) / m_lineHeight); } wxDataViewItem item = node->GetItem(); unsigned int cols = GetOwner()->GetColumnCount(); unsigned int col; int height = m_lineHeight; for (col = 0; col < cols; col++) { const wxDataViewColumn *column = GetOwner()->GetColumn(col); if (column->IsHidden()) continue; // skip it! if ((col != 0) && model->IsContainer(item) && !model->HasContainerColumns(item)) continue; // skip it! wxDataViewRenderer *renderer = const_cast(column->GetRenderer()); renderer->PrepareForItem(model, item, column->GetModelColumn()); height = wxMax( height, renderer->GetSize().y ); } yy += height; if (y < yy) return row; row++; } } int wxDataViewMainWindow::GetLineHeight( unsigned int row ) const { const wxDataViewModel *model = GetModel(); if (GetOwner()->GetWindowStyle() & wxDV_VARIABLE_LINE_HEIGHT) { wxASSERT( !IsVirtualList() ); const wxDataViewTreeNode* node = GetTreeNodeByRow(row); // wxASSERT( node ); if (!node) return m_lineHeight; wxDataViewItem item = node->GetItem(); int height = m_lineHeight; unsigned int cols = GetOwner()->GetColumnCount(); unsigned int col; for (col = 0; col < cols; col++) { const wxDataViewColumn *column = GetOwner()->GetColumn(col); if (column->IsHidden()) continue; // skip it! if ((col != 0) && model->IsContainer(item) && !model->HasContainerColumns(item)) continue; // skip it! wxDataViewRenderer *renderer = const_cast(column->GetRenderer()); renderer->PrepareForItem(model, item, column->GetModelColumn()); height = wxMax( height, renderer->GetSize().y ); } return height; } else { return m_lineHeight; } } class RowToTreeNodeJob: public DoJob { public: RowToTreeNodeJob( unsigned int row , int current, wxDataViewTreeNode * node ) { this->row = row; this->current = current; ret = NULL; parent = node; } virtual int operator() ( wxDataViewTreeNode * node ) { current ++; if( current == static_cast(row)) { ret = node; return DoJob::DONE; } if( node->GetSubTreeCount() + current < static_cast(row) ) { current += node->GetSubTreeCount(); return DoJob::SKIP_SUBTREE; } else { parent = node; // If the current node has only leaf children, we can find the // desired node directly. This can speed up finding the node // in some cases, and will have a very good effect for list views. if ( node->HasChildren() && (int)node->GetChildNodes().size() == node->GetSubTreeCount() ) { const int index = static_cast(row) - current - 1; ret = node->GetChildNodes()[index]; return DoJob::DONE; } return DoJob::CONTINUE; } } wxDataViewTreeNode * GetResult() const { return ret; } private: unsigned int row; int current; wxDataViewTreeNode * ret; wxDataViewTreeNode * parent; }; wxDataViewTreeNode * wxDataViewMainWindow::GetTreeNodeByRow(unsigned int row) const { wxASSERT( !IsVirtualList() ); if ( row == (unsigned)-1 ) return NULL; RowToTreeNodeJob job( row , -2, m_root ); Walker( m_root , job ); return job.GetResult(); } wxDataViewItem wxDataViewMainWindow::GetItemByRow(unsigned int row) const { wxDataViewItem item; if (IsVirtualList()) { if ( row < GetRowCount() ) item = wxDataViewItem(wxUIntToPtr(row+1)); } else { wxDataViewTreeNode *node = GetTreeNodeByRow(row); if ( node ) item = node->GetItem(); } return item; } bool wxDataViewMainWindow::SendExpanderEvent(wxEventType type, const wxDataViewItem& item) { wxWindow *parent = GetParent(); wxDataViewEvent le(type, parent->GetId()); le.SetEventObject(parent); le.SetModel(GetModel()); le.SetItem( item ); return !parent->ProcessWindowEvent(le) || le.IsAllowed(); } bool wxDataViewMainWindow::IsExpanded( unsigned int row ) const { if (IsList()) return false; wxDataViewTreeNode * node = GetTreeNodeByRow(row); if (!node) return false; if (!node->HasChildren()) return false; return node->IsOpen(); } bool wxDataViewMainWindow::HasChildren( unsigned int row ) const { if (IsList()) return false; wxDataViewTreeNode * node = GetTreeNodeByRow(row); if (!node) return false; if (!node->HasChildren()) return false; return true; } void wxDataViewMainWindow::Expand( unsigned int row ) { if (IsList()) return; wxDataViewTreeNode * node = GetTreeNodeByRow(row); if (!node) return; if (!node->HasChildren()) return; if (!node->IsOpen()) { if ( !SendExpanderEvent(wxEVT_DATAVIEW_ITEM_EXPANDING, node->GetItem()) ) { // Vetoed by the event handler. return; } node->ToggleOpen(); // build the children of current node if( node->GetChildNodes().empty() ) { SortPrepare(); ::BuildTreeHelper(GetModel(), node->GetItem(), node); } // By expanding the node all row indices that are currently in the selection list // and are greater than our node have become invalid. So we have to correct that now. const unsigned rowAdjustment = node->GetSubTreeCount(); for(unsigned i=0; i row) ChangeCurrentRow(m_currentRow + rowAdjustment); m_count = -1; UpdateDisplay(); // Send the expanded event SendExpanderEvent(wxEVT_DATAVIEW_ITEM_EXPANDED,node->GetItem()); } } void wxDataViewMainWindow::Collapse(unsigned int row) { if (IsList()) return; wxDataViewTreeNode *node = GetTreeNodeByRow(row); if (!node) return; if (!node->HasChildren()) return; if (node->IsOpen()) { if ( !SendExpanderEvent(wxEVT_DATAVIEW_ITEM_COLLAPSING,node->GetItem()) ) { // Vetoed by the event handler. return; } // Find out if there are selected items below the current node. bool selectCollapsingRow = false; const unsigned rowAdjustment = node->GetSubTreeCount(); unsigned maxRowToBeTested = row + rowAdjustment; for(unsigned i=0; i row && testRow <= maxRowToBeTested) { selectCollapsingRow = true; // get out as soon as we have found a node that is selected break; } } node->ToggleOpen(); // If the node to be closed has selected items the user won't see those any longer. // We select the collapsing node in this case. if(selectCollapsingRow) { SelectAllRows(false); ChangeCurrentRow(row); SelectRow(row, true); SendSelectionChangedEvent(GetItemByRow(row)); } else { // if there were no selected items below our node we still need to "fix" the // selection list to adjust for the changing of the row indices. // We actually do the opposite of what we are doing in Expand(). for(unsigned i=0; i row && m_currentRow <= maxRowToBeTested) ChangeCurrentRow(row); else if(m_currentRow > row) ChangeCurrentRow(m_currentRow - rowAdjustment); } m_count = -1; UpdateDisplay(); SendExpanderEvent(wxEVT_DATAVIEW_ITEM_COLLAPSED,node->GetItem()); } } wxDataViewTreeNode * wxDataViewMainWindow::FindNode( const wxDataViewItem & item ) { const wxDataViewModel * model = GetModel(); if( model == NULL ) return NULL; if (!item.IsOk()) return m_root; // Compose the parent-chain for the item we are looking for wxVector parentChain; wxDataViewItem it( item ); while( it.IsOk() ) { parentChain.push_back(it); it = model->GetParent(it); } // Find the item along the parent-chain. // This algorithm is designed to speed up the node-finding method wxDataViewTreeNode* node = m_root; for( unsigned iter = parentChain.size()-1; ; --iter ) { if( node->HasChildren() ) { if( node->GetChildNodes().empty() ) { // Even though the item is a container, it doesn't have any // child nodes in the control's representation yet. We have // to realize its subtree now. SortPrepare(); ::BuildTreeHelper(model, node->GetItem(), node); } const wxDataViewTreeNodes& nodes = node->GetChildNodes(); bool found = false; for (unsigned i = 0; i < nodes.GetCount(); ++i) { wxDataViewTreeNode* currentNode = nodes[i]; if (currentNode->GetItem() == parentChain[iter]) { if (currentNode->GetItem() == item) return currentNode; node = currentNode; found = true; break; } } if (!found) return NULL; } else return NULL; if ( !iter ) break; } return NULL; } void wxDataViewMainWindow::HitTest( const wxPoint & point, wxDataViewItem & item, wxDataViewColumn* &column ) { wxDataViewColumn *col = NULL; unsigned int cols = GetOwner()->GetColumnCount(); unsigned int colnum = 0; int x, y; m_owner->CalcUnscrolledPosition( point.x, point.y, &x, &y ); for (unsigned x_start = 0; colnum < cols; colnum++) { col = GetOwner()->GetColumnAt(colnum); if (col->IsHidden()) continue; // skip it! unsigned int w = col->GetWidth(); if (x_start+w >= (unsigned int)x) break; x_start += w; } column = col; item = GetItemByRow( GetLineAt( y ) ); } wxRect wxDataViewMainWindow::GetItemRect( const wxDataViewItem & item, const wxDataViewColumn* column ) { int xpos = 0; int width = 0; unsigned int cols = GetOwner()->GetColumnCount(); // If column is null the loop will compute the combined width of all columns. // Otherwise, it will compute the x position of the column we are looking for. for (unsigned int i = 0; i < cols; i++) { wxDataViewColumn* col = GetOwner()->GetColumnAt( i ); if (col == column) break; if (col->IsHidden()) continue; // skip it! xpos += col->GetWidth(); width += col->GetWidth(); } if(column != 0) { // If we have a column, we need can get its width directly. if(column->IsHidden()) width = 0; else width = column->GetWidth(); } else { // If we have no column, we reset the x position back to zero. xpos = 0; } // we have to take an expander column into account and compute its indentation // to get the correct x position where the actual text is int indent = 0; int row = GetRowByItem(item); if (!IsList() && (column == 0 || GetExpanderColumnOrFirstOne(GetOwner()) == column) ) { wxDataViewTreeNode* node = GetTreeNodeByRow(row); indent = GetOwner()->GetIndent() * node->GetIndentLevel(); indent = indent + m_lineHeight; // use m_lineHeight as the width of the expander } wxRect itemRect( xpos + indent, GetLineStart( row ), width - indent, GetLineHeight( row ) ); GetOwner()->CalcScrolledPosition( itemRect.x, itemRect.y, &itemRect.x, &itemRect.y ); return itemRect; } int wxDataViewMainWindow::RecalculateCount() const { if (IsVirtualList()) { wxDataViewVirtualListModel *list_model = (wxDataViewVirtualListModel*) GetModel(); return list_model->GetCount(); } else { return m_root->GetSubTreeCount(); } } class ItemToRowJob : public DoJob { public: ItemToRowJob(const wxDataViewItem& item_, wxVector::reverse_iterator iter) : m_iter(iter), item(item_) { ret = -1; } // Maybe binary search will help to speed up this process virtual int operator() ( wxDataViewTreeNode * node) { ret ++; if( node->GetItem() == item ) { return DoJob::DONE; } if( node->GetItem() == *m_iter ) { m_iter++; return DoJob::CONTINUE; } else { ret += node->GetSubTreeCount(); return DoJob::SKIP_SUBTREE; } } // the row number is begin from zero int GetResult() const { return ret -1; } private: wxVector::reverse_iterator m_iter; wxDataViewItem item; int ret; }; int wxDataViewMainWindow::GetRowByItem(const wxDataViewItem & item) const { const wxDataViewModel * model = GetModel(); if( model == NULL ) return -1; if (IsVirtualList()) { return wxPtrToUInt( item.GetID() ) -1; } else { if( !item.IsOk() ) return -1; // Compose the parent-chain of the item we are looking for wxVector parentChain; wxDataViewItem it( item ); while( it.IsOk() ) { parentChain.push_back(it); it = model->GetParent(it); } // add an 'invalid' item to represent our 'invisible' root node parentChain.push_back(wxDataViewItem()); // the parent chain was created by adding the deepest parent first. // so if we want to start at the root node, we have to iterate backwards through the vector ItemToRowJob job( item, parentChain.rbegin() ); Walker( m_root, job ); return job.GetResult(); } } static void BuildTreeHelper( const wxDataViewModel * model, const wxDataViewItem & item, wxDataViewTreeNode * node) { if( !model->IsContainer( item ) ) return; wxDataViewItemArray children; unsigned int num = model->GetChildren( item, children); for ( unsigned int index = 0; index < num; index++ ) { wxDataViewTreeNode *n = new wxDataViewTreeNode(node, children[index]); if( model->IsContainer(children[index]) ) n->SetHasChildren( true ); node->InsertChild(n, index); } wxASSERT( node->IsOpen() ); node->ChangeSubTreeCount(+num); } void wxDataViewMainWindow::BuildTree(wxDataViewModel * model) { DestroyTree(); if (GetModel()->IsVirtualListModel()) { m_count = -1; return; } m_root = wxDataViewTreeNode::CreateRootNode(); // First we define a invalid item to fetch the top-level elements wxDataViewItem item; SortPrepare(); BuildTreeHelper( model, item, m_root); m_count = -1; } void wxDataViewMainWindow::DestroyTree() { if (!IsVirtualList()) { wxDELETE(m_root); m_count = 0; } } wxDataViewColumn* wxDataViewMainWindow::FindColumnForEditing(const wxDataViewItem& item, wxDataViewCellMode mode) { // Edit the current column editable in 'mode'. If no column is focused // (typically because the user has full row selected), try to find the // first editable column (this would typically be a checkbox for // wxDATAVIEW_CELL_ACTIVATABLE and we don't want to force the user to set // focus on the checkbox column; or on the only editable text column). wxDataViewColumn *candidate = m_currentCol; if ( candidate && !IsCellEditableInMode(item, candidate, mode) && !m_currentColSetByKeyboard ) { // If current column was set by mouse to something not editable (in // 'mode') and the user pressed Space/F2 to edit it, treat the // situation as if there was whole-row focus, because that's what is // visually indicated and the mouse click could very well be targeted // on the row rather than on an individual cell. // // But if it was done by keyboard, respect that even if the column // isn't editable, because focus is visually on that column and editing // something else would be surprising. candidate = NULL; } if ( !candidate ) { const unsigned cols = GetOwner()->GetColumnCount(); for ( unsigned i = 0; i < cols; i++ ) { wxDataViewColumn *c = GetOwner()->GetColumnAt(i); if ( c->IsHidden() ) continue; if ( IsCellEditableInMode(item, c, mode) ) { candidate = c; break; } } } // If on container item without columns, only the expander column // may be directly editable: if ( candidate && GetOwner()->GetExpanderColumn() != candidate && GetModel()->IsContainer(item) && !GetModel()->HasContainerColumns(item) ) { candidate = GetOwner()->GetExpanderColumn(); } if ( !candidate ) return NULL; if ( !IsCellEditableInMode(item, candidate, mode) ) return NULL; return candidate; } bool wxDataViewMainWindow::IsCellEditableInMode(const wxDataViewItem& item, const wxDataViewColumn *col, wxDataViewCellMode mode) const { if ( col->GetRenderer()->GetMode() != mode ) return false; if ( !GetModel()->IsEnabled(item, col->GetModelColumn()) ) return false; return true; } void wxDataViewMainWindow::OnCharHook(wxKeyEvent& event) { if ( m_editorCtrl ) { // Handle any keys special for the in-place editor and return without // calling Skip() below. switch ( event.GetKeyCode() ) { case WXK_ESCAPE: m_editorRenderer->CancelEditing(); return; case WXK_RETURN: case WXK_TAB: m_editorRenderer->FinishEditing(); return; } } else if ( m_useCellFocus ) { if ( event.GetKeyCode() == WXK_TAB ) { if ( event.ShiftDown() ) OnLeftKey(event); else OnRightKey(event); return; } } event.Skip(); } void wxDataViewMainWindow::OnChar( wxKeyEvent &event ) { wxWindow * const parent = GetParent(); // propagate the char event upwards wxKeyEvent eventForParent(event); eventForParent.SetEventObject(parent); if ( parent->ProcessWindowEvent(eventForParent) ) return; if ( parent->HandleAsNavigationKey(event) ) return; // no item -> nothing to do if (!HasCurrentRow()) { event.Skip(); return; } // don't use m_linesPerPage directly as it might not be computed yet const int pageSize = GetCountPerPage(); wxCHECK_RET( pageSize, wxT("should have non zero page size") ); switch ( event.GetKeyCode() ) { case WXK_RETURN: if ( event.HasModifiers() ) { event.Skip(); break; } else { // Enter activates the item, i.e. sends wxEVT_DATAVIEW_ITEM_ACTIVATED to // it. Only if that event is not handled do we activate column renderer (which // is normally done by Space) or even inline editing. const wxDataViewItem item = GetItemByRow(m_currentRow); wxDataViewEvent le(wxEVT_DATAVIEW_ITEM_ACTIVATED, parent->GetId()); le.SetItem(item); le.SetEventObject(parent); le.SetModel(GetModel()); if ( parent->ProcessWindowEvent(le) ) break; // else: fall through to WXK_SPACE handling } case WXK_SPACE: if ( event.HasModifiers() ) { event.Skip(); break; } else { // Space toggles activatable items or -- if not activatable -- // starts inline editing (this is normally done using F2 on // Windows, but Space is common everywhere else, so use it too // for greater cross-platform compatibility). const wxDataViewItem item = GetItemByRow(m_currentRow); // Activate the current activatable column. If not column is focused (typically // because the user has full row selected), try to find the first activatable // column (this would typically be a checkbox and we don't want to force the user // to set focus on the checkbox column). wxDataViewColumn *activatableCol = FindColumnForEditing(item, wxDATAVIEW_CELL_ACTIVATABLE); if ( activatableCol ) { const unsigned colIdx = activatableCol->GetModelColumn(); const wxRect cell_rect = GetOwner()->GetItemRect(item, activatableCol); wxDataViewRenderer *cell = activatableCol->GetRenderer(); cell->PrepareForItem(GetModel(), item, colIdx); cell->WXActivateCell(cell_rect, GetModel(), item, colIdx, NULL); break; } // else: fall through to WXK_F2 handling } case WXK_F2: if ( event.HasModifiers() ) { event.Skip(); break; } else { if( !m_selection.empty() ) { // Mimic Windows 7 behavior: edit the item that has focus // if it is selected and the first selected item if focus // is out of selection. int sel; if ( m_selection.Index(m_currentRow) != wxNOT_FOUND ) sel = m_currentRow; else sel = m_selection[0]; const wxDataViewItem item = GetItemByRow(sel); // Edit the current column. If no column is focused // (typically because the user has full row selected), try // to find the first editable column. wxDataViewColumn *editableCol = FindColumnForEditing(item, wxDATAVIEW_CELL_EDITABLE); if ( editableCol ) GetOwner()->EditItem(item, editableCol); } } break; case WXK_UP: OnVerticalNavigation(event, -1); break; case WXK_DOWN: OnVerticalNavigation(event, +1); break; // Add the process for tree expanding/collapsing case WXK_LEFT: OnLeftKey(event); break; case WXK_RIGHT: OnRightKey(event); break; case WXK_END: OnVerticalNavigation(event, +(int)GetRowCount()); break; case WXK_HOME: OnVerticalNavigation(event, -(int)GetRowCount()); break; case WXK_PAGEUP: OnVerticalNavigation(event, -(pageSize - 1)); break; case WXK_PAGEDOWN: OnVerticalNavigation(event, +(pageSize - 1)); break; default: event.Skip(); } } void wxDataViewMainWindow::OnVerticalNavigation(const wxKeyEvent& event, int delta) { // if there is no selection, we cannot move it anywhere if (!HasCurrentRow() || IsEmpty()) return; int newRow = (int)m_currentRow + delta; // let's keep the new row inside the allowed range if ( newRow < 0 ) newRow = 0; const int rowCount = (int)GetRowCount(); if ( newRow >= rowCount ) newRow = rowCount - 1; unsigned int oldCurrent = m_currentRow; unsigned int newCurrent = (unsigned int)newRow; // in single selection we just ignore Shift as we can't select several // items anyhow if ( event.ShiftDown() && !IsSingleSel() ) { RefreshRow( oldCurrent ); ChangeCurrentRow( newCurrent ); // select all the items between the old and the new one if ( oldCurrent > newCurrent ) { newCurrent = oldCurrent; oldCurrent = m_currentRow; } SelectRows( oldCurrent, newCurrent, true ); if (oldCurrent!=newCurrent) SendSelectionChangedEvent(GetItemByRow(m_selection[0])); } else // !shift { RefreshRow( oldCurrent ); // all previously selected items are unselected unless ctrl is held if ( !event.ControlDown() ) SelectAllRows(false); ChangeCurrentRow( newCurrent ); if ( !event.ControlDown() ) { SelectRow( m_currentRow, true ); SendSelectionChangedEvent(GetItemByRow(m_currentRow)); } else RefreshRow( m_currentRow ); } GetOwner()->EnsureVisible( m_currentRow, -1 ); } void wxDataViewMainWindow::OnLeftKey(wxKeyEvent& event) { if ( IsList() ) { TryAdvanceCurrentColumn(NULL, event, /*forward=*/false); } else { wxDataViewTreeNode* node = GetTreeNodeByRow(m_currentRow); if ( !node ) return; if ( TryAdvanceCurrentColumn(node, event, /*forward=*/false) ) return; const bool dontCollapseNodes = event.GetKeyCode() == WXK_TAB; if ( dontCollapseNodes ) { m_currentCol = NULL; // allow focus change event.Skip(); return; } // Because TryAdvanceCurrentColumn() return false, we are at the first // column or using whole-row selection. In this situation, we can use // the standard TreeView handling of the left key. if (node->HasChildren() && node->IsOpen()) { Collapse(m_currentRow); } else { // if the node is already closed, we move the selection to its parent wxDataViewTreeNode *parent_node = node->GetParent(); if (parent_node) { int parent = GetRowByItem( parent_node->GetItem() ); if ( parent >= 0 ) { unsigned int row = m_currentRow; SelectRow( row, false); SelectRow( parent, true ); ChangeCurrentRow( parent ); GetOwner()->EnsureVisible( parent, -1 ); SendSelectionChangedEvent( parent_node->GetItem() ); } } } } } void wxDataViewMainWindow::OnRightKey(wxKeyEvent& event) { if ( IsList() ) { TryAdvanceCurrentColumn(NULL, event, /*forward=*/true); } else { wxDataViewTreeNode* node = GetTreeNodeByRow(m_currentRow); if ( !node ) return; if ( node->HasChildren() ) { if ( !node->IsOpen() ) { Expand( m_currentRow ); } else { // if the node is already open, we move the selection to the first child unsigned int row = m_currentRow; SelectRow( row, false ); SelectRow( row + 1, true ); ChangeCurrentRow( row + 1 ); GetOwner()->EnsureVisible( row + 1, -1 ); SendSelectionChangedEvent( GetItemByRow(row+1) ); } } else { TryAdvanceCurrentColumn(node, event, /*forward=*/true); } } } bool wxDataViewMainWindow::TryAdvanceCurrentColumn(wxDataViewTreeNode *node, wxKeyEvent& event, bool forward) { if ( GetOwner()->GetColumnCount() == 0 ) return false; if ( !m_useCellFocus ) return false; const bool wrapAround = event.GetKeyCode() == WXK_TAB; if ( node ) { // navigation shouldn't work in branch nodes without other columns: if ( node->HasChildren() && !GetModel()->HasContainerColumns(node->GetItem()) ) return false; } if ( m_currentCol == NULL || !m_currentColSetByKeyboard ) { if ( forward ) { m_currentCol = GetOwner()->GetColumnAt(1); m_currentColSetByKeyboard = true; RefreshRow(m_currentRow); return true; } else { if ( !wrapAround ) return false; } } int idx = GetOwner()->GetColumnIndex(m_currentCol) + (forward ? +1 : -1); if ( idx >= (int)GetOwner()->GetColumnCount() ) { if ( !wrapAround ) return false; if ( GetCurrentRow() < GetRowCount() - 1 ) { // go to the first column of the next row: idx = 0; OnVerticalNavigation(wxKeyEvent()/*dummy*/, +1); } else { // allow focus change event.Skip(); return false; } } if ( idx < 0 && wrapAround ) { if ( GetCurrentRow() > 0 ) { // go to the last column of the previous row: idx = (int)GetOwner()->GetColumnCount() - 1; OnVerticalNavigation(wxKeyEvent()/*dummy*/, -1); } else { // allow focus change event.Skip(); return false; } } GetOwner()->EnsureVisible(m_currentRow, idx); if ( idx < 1 ) { // We are going to the left of the second column. Reset to whole-row // focus (which means first column would be edited). m_currentCol = NULL; RefreshRow(m_currentRow); return true; } m_currentCol = GetOwner()->GetColumnAt(idx); m_currentColSetByKeyboard = true; RefreshRow(m_currentRow); return true; } void wxDataViewMainWindow::OnMouse( wxMouseEvent &event ) { if (event.GetEventType() == wxEVT_MOUSEWHEEL) { // let the base handle mouse wheel events. event.Skip(); return; } if(event.ButtonDown()) { // Not skipping button down events would prevent the system from // setting focus to this window as most (all?) of them do by default, // so skip it to enable default handling. event.Skip(); } int x = event.GetX(); int y = event.GetY(); m_owner->CalcUnscrolledPosition( x, y, &x, &y ); wxDataViewColumn *col = NULL; int xpos = 0; unsigned int cols = GetOwner()->GetColumnCount(); unsigned int i; for (i = 0; i < cols; i++) { wxDataViewColumn *c = GetOwner()->GetColumnAt( i ); if (c->IsHidden()) continue; // skip it! if (x < xpos + c->GetWidth()) { col = c; break; } xpos += c->GetWidth(); } wxDataViewModel* const model = GetModel(); const unsigned int current = GetLineAt( y ); const wxDataViewItem item = GetItemByRow(current); // Handle right clicking here, before everything else as context menu // events should be sent even when we click outside of any item, unlike all // the other ones. if (event.RightUp()) { wxWindow *parent = GetParent(); wxDataViewEvent le(wxEVT_DATAVIEW_ITEM_CONTEXT_MENU, parent->GetId()); le.SetEventObject(parent); le.SetModel(model); if ( item.IsOk() && col ) { le.SetItem( item ); le.SetColumn( col->GetModelColumn() ); le.SetDataViewColumn( col ); } parent->ProcessWindowEvent(le); return; } #if wxUSE_DRAG_AND_DROP if (event.Dragging() || ((m_dragCount > 0) && event.Leaving())) { if (m_dragCount == 0) { // we have to report the raw, physical coords as we want to be // able to call HitTest(event.m_pointDrag) from the user code to // get the item being dragged m_dragStart = event.GetPosition(); } m_dragCount++; if ((m_dragCount < 3) && (event.Leaving())) m_dragCount = 3; else if (m_dragCount != 3) return; if (event.LeftIsDown()) { m_owner->CalcUnscrolledPosition( m_dragStart.x, m_dragStart.y, &m_dragStart.x, &m_dragStart.y ); unsigned int drag_item_row = GetLineAt( m_dragStart.y ); wxDataViewItem itemDragged = GetItemByRow( drag_item_row ); // Notify cell about drag wxDataViewEvent event( wxEVT_DATAVIEW_ITEM_BEGIN_DRAG, m_owner->GetId() ); event.SetEventObject( m_owner ); event.SetItem( itemDragged ); event.SetModel( model ); if (!m_owner->HandleWindowEvent( event )) return; if (!event.IsAllowed()) return; wxDataObject *obj = event.GetDataObject(); if (!obj) return; wxDataViewDropSource drag( this, drag_item_row ); drag.SetData( *obj ); /* wxDragResult res = */ drag.DoDragDrop(event.GetDragFlags()); delete obj; } return; } else { m_dragCount = 0; } #endif // wxUSE_DRAG_AND_DROP // Check if we clicked outside the item area. if ((current >= GetRowCount()) || !col) { // Follow Windows convention here: clicking either left or right (but // not middle) button clears the existing selection. if (m_owner && (event.LeftDown() || event.RightDown())) { if (!GetSelections().empty()) { m_owner->UnselectAll(); SendSelectionChangedEvent(wxDataViewItem()); } } event.Skip(); return; } wxDataViewRenderer *cell = col->GetRenderer(); wxDataViewColumn* const expander = GetExpanderColumnOrFirstOne(GetOwner()); // Test whether the mouse is hovering over the expander (a.k.a tree "+" // button) and also determine the offset of the real cell start, skipping // the indentation and the expander itself. bool hoverOverExpander = false; int itemOffset = 0; if ((!IsList()) && (expander == col)) { wxDataViewTreeNode * node = GetTreeNodeByRow(current); int indent = node->GetIndentLevel(); itemOffset = GetOwner()->GetIndent()*indent; if ( node->HasChildren() ) { // we make the rectangle we are looking in a bit bigger than the actual // visual expander so the user can hit that little thing reliably wxRect rect(itemOffset, GetLineStart( current ) + (GetLineHeight(current) - m_lineHeight)/2, m_lineHeight, m_lineHeight); if( rect.Contains(x, y) ) { // So the mouse is over the expander hoverOverExpander = true; if (m_underMouse && m_underMouse != node) { // wxLogMessage("Undo the row: %d", GetRowByItem(m_underMouse->GetItem())); RefreshRow(GetRowByItem(m_underMouse->GetItem())); } if (m_underMouse != node) { // wxLogMessage("Do the row: %d", current); RefreshRow(current); } m_underMouse = node; } } // Account for the expander as well, even if this item doesn't have it, // its parent does so it still counts for the offset. itemOffset += m_lineHeight; } if (!hoverOverExpander) { if (m_underMouse != NULL) { // wxLogMessage("Undo the row: %d", GetRowByItem(m_underMouse->GetItem())); RefreshRow(GetRowByItem(m_underMouse->GetItem())); m_underMouse = NULL; } } bool simulateClick = false; if (event.ButtonDClick()) { m_renameTimer->Stop(); m_lastOnSame = false; } bool ignore_other_columns = ((expander != col) && (model->IsContainer(item)) && (!model->HasContainerColumns(item))); if (event.LeftDClick()) { if(hoverOverExpander) { // a double click on the expander will be converted into a "simulated" normal click simulateClick = true; } else if ( current == m_lineLastClicked ) { wxWindow *parent = GetParent(); wxDataViewEvent le(wxEVT_DATAVIEW_ITEM_ACTIVATED, parent->GetId()); le.SetItem( item ); le.SetColumn( col->GetModelColumn() ); le.SetDataViewColumn( col ); le.SetEventObject(parent); le.SetModel(GetModel()); parent->ProcessWindowEvent(le); return; } else { // The first click was on another item, so don't interpret this as // a double click, but as a simple click instead simulateClick = true; } } if (event.LeftUp() && !hoverOverExpander) { if (m_lineSelectSingleOnUp != (unsigned int)-1) { // select single line SelectAllRows( false ); SelectRow( m_lineSelectSingleOnUp, true ); SendSelectionChangedEvent( GetItemByRow(m_lineSelectSingleOnUp) ); } // If the user click the expander, we do not do editing even if the column // with expander are editable if (m_lastOnSame && !ignore_other_columns) { if ((col == m_currentCol) && (current == m_currentRow) && IsCellEditableInMode(item, col, wxDATAVIEW_CELL_EDITABLE) ) { m_renameTimer->Start( 100, true ); } } m_lastOnSame = false; m_lineSelectSingleOnUp = (unsigned int)-1; } else if(!event.LeftUp()) { // This is necessary, because after a DnD operation in // from and to ourself, the up event is swallowed by the // DnD code. So on next non-up event (which means here and // now) m_lineSelectSingleOnUp should be reset. m_lineSelectSingleOnUp = (unsigned int)-1; } if (event.RightDown()) { m_lineBeforeLastClicked = m_lineLastClicked; m_lineLastClicked = current; // If the item is already selected, do not update the selection. // Multi-selections should not be cleared if a selected item is clicked. if (!IsRowSelected(current)) { SelectAllRows(false); const unsigned oldCurrent = m_currentRow; ChangeCurrentRow(current); SelectRow(m_currentRow,true); RefreshRow(oldCurrent); SendSelectionChangedEvent(GetItemByRow( m_currentRow ) ); } } else if (event.MiddleDown()) { } if((event.LeftDown() || simulateClick) && hoverOverExpander) { wxDataViewTreeNode* node = GetTreeNodeByRow(current); // hoverOverExpander being true tells us that our node must be // valid and have children. // So we don't need any extra checks. if( node->IsOpen() ) Collapse(current); else Expand(current); } else if ((event.LeftDown() || simulateClick) && !hoverOverExpander) { m_lineBeforeLastClicked = m_lineLastClicked; m_lineLastClicked = current; unsigned int oldCurrentRow = m_currentRow; bool oldWasSelected = IsRowSelected(m_currentRow); bool cmdModifierDown = event.CmdDown(); if ( IsSingleSel() || !(cmdModifierDown || event.ShiftDown()) ) { if ( IsSingleSel() || !IsRowSelected(current) ) { SelectAllRows( false ); ChangeCurrentRow(current); SelectRow(m_currentRow,true); SendSelectionChangedEvent(GetItemByRow( m_currentRow ) ); } else // multi sel & current is highlighted & no mod keys { m_lineSelectSingleOnUp = current; ChangeCurrentRow(current); // change focus } } else // multi sel & either ctrl or shift is down { if (cmdModifierDown) { ChangeCurrentRow(current); ReverseRowSelection(m_currentRow); SendSelectionChangedEvent(GetItemByRow(m_currentRow)); } else if (event.ShiftDown()) { ChangeCurrentRow(current); unsigned int lineFrom = oldCurrentRow, lineTo = current; if ( lineTo < lineFrom ) { lineTo = lineFrom; lineFrom = m_currentRow; } SelectRows(lineFrom, lineTo, true); SendSelectionChangedEvent(GetItemByRow(m_selection[0]) ); } else // !ctrl, !shift { // test in the enclosing if should make it impossible wxFAIL_MSG( wxT("how did we get here?") ); } } if (m_currentRow != oldCurrentRow) RefreshRow( oldCurrentRow ); wxDataViewColumn *oldCurrentCol = m_currentCol; // Update selection here... m_currentCol = col; m_currentColSetByKeyboard = false; // This flag is used to decide whether we should start editing the item // label. We do it if the user clicks twice (but not double clicks, // i.e. simulateClick is false) on the same item but not if the click // was used for something else already, e.g. selecting the item (so it // must have been already selected) or giving the focus to the control // (so it must have had focus already). m_lastOnSame = !simulateClick && ((col == oldCurrentCol) && (current == oldCurrentRow)) && oldWasSelected && HasFocus(); // Call ActivateCell() after everything else as under GTK+ if ( IsCellEditableInMode(item, col, wxDATAVIEW_CELL_ACTIVATABLE) ) { // notify cell about click cell->PrepareForItem(model, item, col->GetModelColumn()); wxRect cell_rect( xpos + itemOffset, GetLineStart( current ), col->GetWidth() - itemOffset, GetLineHeight( current ) ); // Report position relative to the cell's custom area, i.e. // not the entire space as given by the control but the one // used by the renderer after calculation of alignment etc. // // Notice that this results in negative coordinates when clicking // in the upper left corner of a centre-aligned cell which doesn't // fill its column entirely so this is somewhat surprising, but we // do it like this for compatibility with the native GTK+ version, // see #12270. // adjust the rectangle ourselves to account for the alignment const int align = cell->GetEffectiveAlignment(); wxRect rectItem = cell_rect; const wxSize size = cell->GetSize(); if ( size.x >= 0 && size.x < cell_rect.width ) { if ( align & wxALIGN_CENTER_HORIZONTAL ) rectItem.x += (cell_rect.width - size.x)/2; else if ( align & wxALIGN_RIGHT ) rectItem.x += cell_rect.width - size.x; // else: wxALIGN_LEFT is the default } if ( size.y >= 0 && size.y < cell_rect.height ) { if ( align & wxALIGN_CENTER_VERTICAL ) rectItem.y += (cell_rect.height - size.y)/2; else if ( align & wxALIGN_BOTTOM ) rectItem.y += cell_rect.height - size.y; // else: wxALIGN_TOP is the default } wxMouseEvent event2(event); event2.m_x -= rectItem.x; event2.m_y -= rectItem.y; m_owner->CalcUnscrolledPosition(event2.m_x, event2.m_y, &event2.m_x, &event2.m_y); /* ignore ret */ cell->WXActivateCell ( cell_rect, model, item, col->GetModelColumn(), &event2 ); } } } void wxDataViewMainWindow::OnSetFocus( wxFocusEvent &event ) { m_hasFocus = true; if (HasCurrentRow()) Refresh(); event.Skip(); } void wxDataViewMainWindow::OnKillFocus( wxFocusEvent &event ) { m_hasFocus = false; if (HasCurrentRow()) Refresh(); event.Skip(); } void wxDataViewMainWindow::OnColumnsCountChanged() { int editableCount = 0; const unsigned cols = GetOwner()->GetColumnCount(); for ( unsigned i = 0; i < cols; i++ ) { wxDataViewColumn *c = GetOwner()->GetColumnAt(i); if ( c->IsHidden() ) continue; if ( c->GetRenderer()->GetMode() != wxDATAVIEW_CELL_INERT ) editableCount++; } m_useCellFocus = (editableCount > 0); UpdateDisplay(); } void wxDataViewMainWindow::UpdateColumnSizes() { int colsCount = GetOwner()->GetColumnCount(); if ( !colsCount ) return; wxDataViewCtrl *owner = GetOwner(); int fullWinWidth = GetSize().x; wxDataViewColumn *lastCol = owner->GetColumn(colsCount - 1); int colswidth = GetEndOfLastCol(); int lastColX = colswidth - lastCol->GetWidth(); if ( lastColX < fullWinWidth ) { int desiredWidth = wxMax(fullWinWidth - lastColX, lastCol->GetMinWidth()); lastCol->SetWidth(desiredWidth); // All columns fit on screen, so we don't need horizontal scrolling. // To prevent flickering scrollbar when resizing the window to be // narrower, force-set the virtual width to 0 here. It will eventually // be corrected at idle time. SetVirtualSize(0, m_virtualSize.y); RefreshRect(wxRect(lastColX, 0, fullWinWidth - lastColX, GetSize().y)); } else { // else: don't bother, the columns won't fit anyway SetVirtualSize(colswidth, m_virtualSize.y); } } //----------------------------------------------------------------------------- // wxDataViewCtrl //----------------------------------------------------------------------------- WX_DEFINE_LIST(wxDataViewColumnList) IMPLEMENT_DYNAMIC_CLASS(wxDataViewCtrl, wxDataViewCtrlBase) BEGIN_EVENT_TABLE(wxDataViewCtrl, wxDataViewCtrlBase) EVT_SIZE(wxDataViewCtrl::OnSize) END_EVENT_TABLE() wxDataViewCtrl::~wxDataViewCtrl() { if (m_notifier) GetModel()->RemoveNotifier( m_notifier ); m_cols.Clear(); m_colsBestWidths.clear(); } void wxDataViewCtrl::Init() { m_cols.DeleteContents(true); m_notifier = NULL; // No sorting column at start m_sortingColumnIdx = wxNOT_FOUND; m_headerArea = NULL; m_clientArea = NULL; m_colsDirty = false; } bool wxDataViewCtrl::Create(wxWindow *parent, wxWindowID id, const wxPoint& pos, const wxSize& size, long style, const wxValidator& validator, const wxString& name) { // if ( (style & wxBORDER_MASK) == 0) // style |= wxBORDER_SUNKEN; Init(); if (!wxControl::Create( parent, id, pos, size, style | wxScrolledWindowStyle, validator, name)) return false; SetInitialSize(size); #ifdef __WXMAC__ MacSetClipChildren( true ); #endif m_clientArea = new wxDataViewMainWindow( this, wxID_ANY ); // We use the cursor keys for moving the selection, not scrolling, so call // this method to ensure wxScrollHelperEvtHandler doesn't catch all // keyboard events forwarded to us from wxListMainWindow. DisableKeyboardScrolling(); if (HasFlag(wxDV_NO_HEADER)) m_headerArea = NULL; else m_headerArea = new wxDataViewHeaderWindow(this); SetTargetWindow( m_clientArea ); wxBoxSizer *sizer = new wxBoxSizer( wxVERTICAL ); if (m_headerArea) sizer->Add( m_headerArea, 0, wxGROW ); sizer->Add( m_clientArea, 1, wxGROW ); SetSizer( sizer ); return true; } wxBorder wxDataViewCtrl::GetDefaultBorder() const { return wxBORDER_THEME; } #ifdef __WXMSW__ WXLRESULT wxDataViewCtrl::MSWWindowProc(WXUINT nMsg, WXWPARAM wParam, WXLPARAM lParam) { WXLRESULT rc = wxDataViewCtrlBase::MSWWindowProc(nMsg, wParam, lParam); #ifndef __WXWINCE__ // we need to process arrows ourselves for scrolling if ( nMsg == WM_GETDLGCODE ) { rc |= DLGC_WANTARROWS; } #endif return rc; } #endif wxSize wxDataViewCtrl::GetSizeAvailableForScrollTarget(const wxSize& size) { wxSize newsize = size; if (!HasFlag(wxDV_NO_HEADER) && (m_headerArea)) newsize.y -= m_headerArea->GetSize().y; return newsize; } void wxDataViewCtrl::OnSize( wxSizeEvent &WXUNUSED(event) ) { if ( m_clientArea && GetColumnCount() ) m_clientArea->UpdateColumnSizes(); // We need to override OnSize so that our scrolled // window a) does call Layout() to use sizers for // positioning the controls but b) does not query // the sizer for their size and use that for setting // the scrollable area as set that ourselves by // calling SetScrollbar() further down. Layout(); AdjustScrollbars(); // We must redraw the headers if their height changed. Normally this // shouldn't happen as the control shouldn't let itself be resized beneath // its minimal height but avoid the display artefacts that appear if it // does happen, e.g. because there is really not enough vertical space. if ( !HasFlag(wxDV_NO_HEADER) && m_headerArea && m_headerArea->GetSize().y <= m_headerArea->GetBestSize(). y ) { m_headerArea->Refresh(); } } void wxDataViewCtrl::SetFocus() { if (m_clientArea) m_clientArea->SetFocus(); } bool wxDataViewCtrl::SetFont(const wxFont & font) { if (!wxControl::SetFont(font)) return false; if (m_headerArea) m_headerArea->SetFont(font); if (m_clientArea) { m_clientArea->SetFont(font); m_clientArea->SetRowHeight(m_clientArea->GetDefaultRowHeight()); } if (m_headerArea || m_clientArea) { InvalidateColBestWidths(); Layout(); } return true; } bool wxDataViewCtrl::AssociateModel( wxDataViewModel *model ) { if (!wxDataViewCtrlBase::AssociateModel( model )) return false; if (model) { m_notifier = new wxGenericDataViewModelNotifier( m_clientArea ); model->AddNotifier( m_notifier ); } else if (m_notifier) { m_notifier->Cleared(); m_notifier = NULL; } m_clientArea->DestroyTree(); if (model) { m_clientArea->BuildTree(model); } m_clientArea->UpdateDisplay(); return true; } #if wxUSE_DRAG_AND_DROP bool wxDataViewCtrl::EnableDragSource( const wxDataFormat &format ) { return m_clientArea->EnableDragSource( format ); } bool wxDataViewCtrl::EnableDropTarget( const wxDataFormat &format ) { return m_clientArea->EnableDropTarget( format ); } #endif // wxUSE_DRAG_AND_DROP bool wxDataViewCtrl::AppendColumn( wxDataViewColumn *col ) { if (!wxDataViewCtrlBase::AppendColumn(col)) return false; m_cols.Append( col ); m_colsBestWidths.push_back(CachedColWidthInfo()); OnColumnsCountChanged(); return true; } bool wxDataViewCtrl::PrependColumn( wxDataViewColumn *col ) { if (!wxDataViewCtrlBase::PrependColumn(col)) return false; m_cols.Insert( col ); m_colsBestWidths.insert(m_colsBestWidths.begin(), CachedColWidthInfo()); OnColumnsCountChanged(); return true; } bool wxDataViewCtrl::InsertColumn( unsigned int pos, wxDataViewColumn *col ) { if (!wxDataViewCtrlBase::InsertColumn(pos,col)) return false; m_cols.Insert( pos, col ); m_colsBestWidths.insert(m_colsBestWidths.begin() + pos, CachedColWidthInfo()); OnColumnsCountChanged(); return true; } void wxDataViewCtrl::OnColumnChange(unsigned int idx) { if ( m_headerArea ) m_headerArea->UpdateColumn(idx); m_clientArea->UpdateDisplay(); } void wxDataViewCtrl::OnColumnsCountChanged() { if (m_headerArea) m_headerArea->SetColumnCount(GetColumnCount()); m_clientArea->OnColumnsCountChanged(); } void wxDataViewCtrl::DoSetExpanderColumn() { wxDataViewColumn* column = GetExpanderColumn(); if ( column ) { int index = GetColumnIndex(column); if ( index != wxNOT_FOUND ) InvalidateColBestWidth(index); } m_clientArea->UpdateDisplay(); } void wxDataViewCtrl::DoSetIndent() { m_clientArea->UpdateDisplay(); } unsigned int wxDataViewCtrl::GetColumnCount() const { return m_cols.GetCount(); } bool wxDataViewCtrl::SetRowHeight( int lineHeight ) { if ( !m_clientArea ) return false; m_clientArea->SetRowHeight(lineHeight); return true; } wxDataViewColumn* wxDataViewCtrl::GetColumn( unsigned int idx ) const { return m_cols[idx]; } wxDataViewColumn *wxDataViewCtrl::GetColumnAt(unsigned int pos) const { // columns can't be reordered if there is no header window which allows // to do this const unsigned idx = m_headerArea ? m_headerArea->GetColumnsOrder()[pos] : pos; return GetColumn(idx); } int wxDataViewCtrl::GetColumnIndex(const wxDataViewColumn *column) const { const unsigned count = m_cols.size(); for ( unsigned n = 0; n < count; n++ ) { if ( m_cols[n] == column ) return n; } return wxNOT_FOUND; } int wxDataViewCtrl::GetModelColumnIndex( unsigned int model_column ) const { const int count = GetColumnCount(); for ( int index = 0; index < count; index++ ) { wxDataViewColumn* column = GetColumn(index); if ( column->GetModelColumn() == model_column ) return index; } return wxNOT_FOUND; } unsigned int wxDataViewCtrl::GetBestColumnWidth(int idx) const { if ( m_colsBestWidths[idx].width != 0 ) return m_colsBestWidths[idx].width; const int count = m_clientArea->GetRowCount(); wxDataViewColumn *column = GetColumn(idx); wxDataViewRenderer *renderer = const_cast(column->GetRenderer()); class MaxWidthCalculator { public: MaxWidthCalculator(const wxDataViewCtrl *dvc, wxDataViewMainWindow *clientArea, wxDataViewRenderer *renderer, const wxDataViewModel *model, unsigned int model_column, int expanderSize) : m_width(0), m_dvc(dvc), m_clientArea(clientArea), m_renderer(renderer), m_model(model), m_model_column(model_column), m_expanderSize(expanderSize) { int index = dvc->GetModelColumnIndex( model_column ); wxDataViewColumn* column = index == wxNOT_FOUND ? NULL : dvc->GetColumn(index); m_isExpanderCol = !clientArea->IsList() && (column == 0 || GetExpanderColumnOrFirstOne(const_cast(dvc)) == column ); } void UpdateWithWidth(int width) { m_width = wxMax(m_width, width); } void UpdateWithRow(int row) { int indent = 0; wxDataViewItem item; if ( m_isExpanderCol ) { wxDataViewTreeNode *node = m_clientArea->GetTreeNodeByRow(row); item = node->GetItem(); indent = m_dvc->GetIndent() * node->GetIndentLevel() + m_expanderSize; } else { item = m_clientArea->GetItemByRow(row); } m_renderer->PrepareForItem(m_model, item, m_model_column); m_width = wxMax(m_width, m_renderer->GetSize().x + indent); } int GetMaxWidth() const { return m_width; } private: int m_width; const wxDataViewCtrl *m_dvc; wxDataViewMainWindow *m_clientArea; wxDataViewRenderer *m_renderer; const wxDataViewModel *m_model; unsigned m_model_column; bool m_isExpanderCol; int m_expanderSize; }; MaxWidthCalculator calculator(this, m_clientArea, renderer, GetModel(), column->GetModelColumn(), m_clientArea->GetRowHeight()); calculator.UpdateWithWidth(column->GetMinWidth()); if ( m_headerArea ) calculator.UpdateWithWidth(m_headerArea->GetColumnTitleWidth(*column)); // The code below deserves some explanation. For very large controls, we // simply can't afford to calculate sizes for all items, it takes too // long. So the best we can do is to check the first and the last N/2 // items in the control for some sufficiently large N and calculate best // sizes from that. That can result in the calculated best width being too // small for some outliers, but it's better to get slightly imperfect // result than to wait several seconds after every update. To avoid highly // visible miscalculations, we also include all currently visible items // no matter what. Finally, the value of N is determined dynamically by // measuring how much time we spent on the determining item widths so far. #if wxUSE_STOPWATCH int top_part_end = count; static const long CALC_TIMEOUT = 20/*ms*/; // don't call wxStopWatch::Time() too often static const unsigned CALC_CHECK_FREQ = 100; wxStopWatch timer; #else // use some hard-coded limit, that's the best we can do without timer int top_part_end = wxMin(500, count); #endif // wxUSE_STOPWATCH/!wxUSE_STOPWATCH int row = 0; for ( row = 0; row < top_part_end; row++ ) { #if wxUSE_STOPWATCH if ( row % CALC_CHECK_FREQ == CALC_CHECK_FREQ-1 && timer.Time() > CALC_TIMEOUT ) break; #endif // wxUSE_STOPWATCH calculator.UpdateWithRow(row); } // row is the first unmeasured item now; that's our value of N/2 if ( row < count ) { top_part_end = row; // add bottom N/2 items now: const int bottom_part_start = wxMax(row, count - row); for ( row = bottom_part_start; row < count; row++ ) { calculator.UpdateWithRow(row); } // finally, include currently visible items in the calculation: const wxPoint origin = CalcUnscrolledPosition(wxPoint(0, 0)); int first_visible = m_clientArea->GetLineAt(origin.y); int last_visible = m_clientArea->GetLineAt(origin.y + GetClientSize().y); first_visible = wxMax(first_visible, top_part_end); last_visible = wxMin(bottom_part_start, last_visible); for ( row = first_visible; row < last_visible; row++ ) { calculator.UpdateWithRow(row); } wxLogTrace("dataview", "determined best size from %d top, %d bottom plus %d more visible items out of %d total", top_part_end, count - bottom_part_start, wxMax(0, last_visible - first_visible), count); } int max_width = calculator.GetMaxWidth(); if ( max_width > 0 ) max_width += 2 * PADDING_RIGHTLEFT; const_cast(this)->m_colsBestWidths[idx].width = max_width; return max_width; } void wxDataViewCtrl::ColumnMoved(wxDataViewColumn * WXUNUSED(col), unsigned int WXUNUSED(new_pos)) { // do _not_ reorder m_cols elements here, they should always be in the // order in which columns were added, we only display the columns in // different order m_clientArea->UpdateDisplay(); } bool wxDataViewCtrl::DeleteColumn( wxDataViewColumn *column ) { wxDataViewColumnList::compatibility_iterator ret = m_cols.Find( column ); if (!ret) return false; m_colsBestWidths.erase(m_colsBestWidths.begin() + GetColumnIndex(column)); m_cols.Erase(ret); if ( m_clientArea->GetCurrentColumn() == column ) m_clientArea->ClearCurrentColumn(); OnColumnsCountChanged(); return true; } bool wxDataViewCtrl::ClearColumns() { SetExpanderColumn(NULL); m_cols.Clear(); m_colsBestWidths.clear(); m_clientArea->ClearCurrentColumn(); OnColumnsCountChanged(); return true; } void wxDataViewCtrl::InvalidateColBestWidth(int idx) { m_colsBestWidths[idx].width = 0; m_colsBestWidths[idx].dirty = true; m_colsDirty = true; } void wxDataViewCtrl::InvalidateColBestWidths() { // mark all columns as dirty: m_colsBestWidths.clear(); m_colsBestWidths.resize(m_cols.size()); m_colsDirty = true; } void wxDataViewCtrl::UpdateColWidths() { m_colsDirty = false; if ( !m_headerArea ) return; const unsigned len = m_colsBestWidths.size(); for ( unsigned i = 0; i < len; i++ ) { // Note that we have to have an explicit 'dirty' flag here instead of // checking if the width==0, as is done in GetBestColumnWidth(). // // Testing width==0 wouldn't work correctly if some code called // GetWidth() after col. width invalidation but before // wxDataViewCtrl::UpdateColWidths() was called at idle time. This // would result in the header's column width getting out of sync with // the control itself. if ( m_colsBestWidths[i].dirty ) { m_headerArea->UpdateColumn(i); m_colsBestWidths[i].dirty = false; } } } void wxDataViewCtrl::OnInternalIdle() { wxDataViewCtrlBase::OnInternalIdle(); if ( m_colsDirty ) UpdateColWidths(); } int wxDataViewCtrl::GetColumnPosition( const wxDataViewColumn *column ) const { unsigned int len = GetColumnCount(); for ( unsigned int i = 0; i < len; i++ ) { wxDataViewColumn * col = GetColumnAt(i); if (column==col) return i; } return wxNOT_FOUND; } wxDataViewColumn *wxDataViewCtrl::GetSortingColumn() const { return m_sortingColumnIdx == wxNOT_FOUND ? NULL : GetColumn(m_sortingColumnIdx); } wxDataViewItem wxDataViewCtrl::DoGetCurrentItem() const { return GetItemByRow(m_clientArea->GetCurrentRow()); } void wxDataViewCtrl::DoSetCurrentItem(const wxDataViewItem& item) { const int row = m_clientArea->GetRowByItem(item); const unsigned oldCurrent = m_clientArea->GetCurrentRow(); if ( static_cast(row) != oldCurrent ) { m_clientArea->ChangeCurrentRow(row); m_clientArea->RefreshRow(oldCurrent); m_clientArea->RefreshRow(row); } } wxDataViewColumn *wxDataViewCtrl::GetCurrentColumn() const { return m_clientArea->GetCurrentColumn(); } int wxDataViewCtrl::GetSelectedItemsCount() const { return m_clientArea->GetSelections().size(); } int wxDataViewCtrl::GetSelections( wxDataViewItemArray & sel ) const { sel.Empty(); const wxDataViewSelection& selections = m_clientArea->GetSelections(); const size_t len = selections.size(); for ( size_t i = 0; i < len; i++ ) { wxDataViewItem item = m_clientArea->GetItemByRow(selections[i]); if ( item.IsOk() ) { sel.Add(item); } else { wxFAIL_MSG( "invalid item in selection - bad internal state" ); } } return sel.size(); } void wxDataViewCtrl::SetSelections( const wxDataViewItemArray & sel ) { wxDataViewSelection selection(wxDataViewSelectionCmp); wxDataViewItem last_parent; int len = sel.GetCount(); for( int i = 0; i < len; i ++ ) { wxDataViewItem item = sel[i]; wxDataViewItem parent = GetModel()->GetParent( item ); if (parent) { if (parent != last_parent) ExpandAncestors(item); } last_parent = parent; int row = m_clientArea->GetRowByItem( item ); if( row >= 0 ) selection.Add( static_cast(row) ); } m_clientArea->SetSelections( selection ); } void wxDataViewCtrl::Select( const wxDataViewItem & item ) { ExpandAncestors( item ); int row = m_clientArea->GetRowByItem( item ); if( row >= 0 ) { // Unselect all rows before select another in the single select mode if (m_clientArea->IsSingleSel()) m_clientArea->SelectAllRows(false); m_clientArea->SelectRow(row, true); // Also set focus to the selected item m_clientArea->ChangeCurrentRow( row ); } } void wxDataViewCtrl::Unselect( const wxDataViewItem & item ) { int row = m_clientArea->GetRowByItem( item ); if( row >= 0 ) m_clientArea->SelectRow(row, false); } bool wxDataViewCtrl::IsSelected( const wxDataViewItem & item ) const { int row = m_clientArea->GetRowByItem( item ); if( row >= 0 ) { return m_clientArea->IsRowSelected(row); } return false; } void wxDataViewCtrl::SetAlternateRowColour(const wxColour& colour) { m_alternateRowColour = colour; } void wxDataViewCtrl::SelectAll() { m_clientArea->SelectAllRows(true); } void wxDataViewCtrl::UnselectAll() { m_clientArea->SelectAllRows(false); } void wxDataViewCtrl::EnsureVisible( int row, int column ) { if( row < 0 ) row = 0; if( row > (int) m_clientArea->GetRowCount() ) row = m_clientArea->GetRowCount(); int first = m_clientArea->GetFirstVisibleRow(); int last = m_clientArea->GetLastVisibleRow(); if( row < first ) m_clientArea->ScrollTo( row, column ); else if( row > last ) m_clientArea->ScrollTo( row - last + first, column ); else m_clientArea->ScrollTo( first, column ); } void wxDataViewCtrl::EnsureVisible( const wxDataViewItem & item, const wxDataViewColumn * column ) { ExpandAncestors( item ); m_clientArea->RecalculateDisplay(); int row = m_clientArea->GetRowByItem(item); if( row >= 0 ) { if( column == NULL ) EnsureVisible(row, -1); else EnsureVisible( row, GetColumnIndex(column) ); } } void wxDataViewCtrl::HitTest( const wxPoint & point, wxDataViewItem & item, wxDataViewColumn* &column ) const { m_clientArea->HitTest(point, item, column); } wxRect wxDataViewCtrl::GetItemRect( const wxDataViewItem & item, const wxDataViewColumn* column ) const { return m_clientArea->GetItemRect(item, column); } wxDataViewItem wxDataViewCtrl::GetItemByRow( unsigned int row ) const { return m_clientArea->GetItemByRow( row ); } int wxDataViewCtrl::GetRowByItem( const wxDataViewItem & item ) const { return m_clientArea->GetRowByItem( item ); } void wxDataViewCtrl::Expand( const wxDataViewItem & item ) { ExpandAncestors( item ); int row = m_clientArea->GetRowByItem( item ); if (row != -1) { m_clientArea->Expand(row); InvalidateColBestWidths(); } } void wxDataViewCtrl::Collapse( const wxDataViewItem & item ) { int row = m_clientArea->GetRowByItem( item ); if (row != -1) { m_clientArea->Collapse(row); InvalidateColBestWidths(); } } bool wxDataViewCtrl::IsExpanded( const wxDataViewItem & item ) const { int row = m_clientArea->GetRowByItem( item ); if (row != -1) return m_clientArea->IsExpanded(row); return false; } void wxDataViewCtrl::EditItem(const wxDataViewItem& item, const wxDataViewColumn *column) { wxCHECK_RET( item.IsOk(), "invalid item" ); wxCHECK_RET( column, "no column provided" ); m_clientArea->StartEditing(item, column); } #endif // !wxUSE_GENERICDATAVIEWCTRL #endif // wxUSE_DATAVIEWCTRL