+#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 <commctrl.h> "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"
+#endif
+
+#include "wx/stockitem.h"
+#include "wx/calctrl.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"
+
+//-----------------------------------------------------------------------------
+// classes
+//-----------------------------------------------------------------------------
+
+class wxDataViewCtrl;
+
+static const int SCROLL_UNIT_X = 15;
+
+// the cell padding on the left/right
+static const int PADDING_RIGHTLEFT = 3;
+
+// the cell padding on the top/bottom
+static const int PADDING_TOPBOTTOM = 1;
+
+// the expander space margin
+static const int EXPANDER_MARGIN = 4;
+
+//-----------------------------------------------------------------------------
+// wxDataViewHeaderWindow
+//-----------------------------------------------------------------------------
+
+#define USE_NATIVE_HEADER_WINDOW 1
+
+// NB: for some reason, this class must be dllexport'ed or we get warnings from
+// MSVC in DLL build
+class WXDLLIMPEXP_ADV wxDataViewHeaderWindowBase : public wxControl
+{
+public:
+ wxDataViewHeaderWindowBase()
+ { m_owner = NULL; }
+
+ bool Create(wxDataViewCtrl *parent, wxWindowID id,
+ const wxPoint &pos, const wxSize &size,
+ const wxString &name)
+ {
+ return wxWindow::Create(parent, id, pos, size, wxNO_BORDER, name);
+ }
+
+ void SetOwner( wxDataViewCtrl* owner ) { m_owner = owner; }
+ wxDataViewCtrl *GetOwner() { return m_owner; }
+
+ // called on column addition/removal
+ virtual void UpdateDisplay() { /* by default, do nothing */ }
+
+ // returns the n-th column
+ virtual wxDataViewColumn *GetColumn(unsigned int n)
+ {
+ wxASSERT(m_owner);
+ wxDataViewColumn *ret = m_owner->GetColumn(n);
+ wxASSERT(ret);
+
+ return ret;
+ }
+
+protected:
+ wxDataViewCtrl *m_owner;
+
+ // sends an event generated from the n-th wxDataViewColumn
+ void SendEvent(wxEventType type, unsigned int n);
+};
+
+// on wxMSW the header window (only that part however) can be made native!
+#if defined(__WXMSW__) && USE_NATIVE_HEADER_WINDOW
+
+#define COLUMN_WIDTH_OFFSET 2
+#define wxDataViewHeaderWindowMSW wxDataViewHeaderWindow
+
+class wxDataViewHeaderWindowMSW : public wxDataViewHeaderWindowBase
+{
+public:
+
+ wxDataViewHeaderWindowMSW( wxDataViewCtrl *parent,
+ wxWindowID id,
+ const wxPoint &pos = wxDefaultPosition,
+ const wxSize &size = wxDefaultSize,
+ const wxString &name = wxT("wxdataviewctrlheaderwindow") )
+ {
+ Create(parent, id, pos, size, name);
+ }
+
+ bool Create(wxDataViewCtrl *parent, wxWindowID id,
+ const wxPoint &pos, const wxSize &size,
+ const wxString &name);
+
+ ~wxDataViewHeaderWindowMSW();
+
+ // called when any column setting is changed and/or changed
+ // the column count
+ virtual void UpdateDisplay();
+
+ // called when the main window gets scrolled
+ virtual void ScrollWindow(int dx, int dy, const wxRect *rect = NULL);
+
+protected:
+ virtual bool MSWOnNotify(int idCtrl, WXLPARAM lParam, WXLPARAM *result);
+ virtual void DoSetSize(int x, int y, int width, int height, int sizeFlags);
+
+ unsigned int GetColumnIdxFromHeader(NMHEADER *nmHDR);
+
+ wxDataViewColumn *GetColumnFromHeader(NMHEADER *nmHDR)
+ { return GetColumn(GetColumnIdxFromHeader(nmHDR)); }
+
+private:
+ DECLARE_DYNAMIC_CLASS(wxDataViewHeaderWindowMSW)
+};
+
+#else // !defined(__WXMSW__)
+
+#define HEADER_WINDOW_HEIGHT 25
+#define HEADER_HORIZ_BORDER 5
+#define HEADER_VERT_BORDER 3
+#define wxGenericDataViewHeaderWindow wxDataViewHeaderWindow
+
+class wxGenericDataViewHeaderWindow : public wxDataViewHeaderWindowBase
+{
+public:
+ wxGenericDataViewHeaderWindow( wxDataViewCtrl *parent,
+ wxWindowID id,
+ const wxPoint &pos = wxDefaultPosition,
+ const wxSize &size = wxDefaultSize,
+ const wxString &name = wxT("wxdataviewctrlheaderwindow") )
+ {
+ Init();
+ Create(parent, id, pos, size, name);
+ }
+
+ bool Create(wxDataViewCtrl *parent, wxWindowID id,
+ const wxPoint &pos, const wxSize &size,
+ const wxString &name);
+
+ ~wxGenericDataViewHeaderWindow()
+ {
+ delete m_resizeCursor;
+ }
+
+ virtual void UpdateDisplay() { Refresh(); }
+
+ // event handlers:
+
+ void OnPaint( wxPaintEvent &event );
+ void OnMouse( wxMouseEvent &event );
+ void OnSetFocus( wxFocusEvent &event );
+
+
+protected:
+
+ // vars used for column resizing:
+
+ wxCursor *m_resizeCursor;
+ const wxCursor *m_currentCursor;
+ bool m_isDragging;
+
+ bool m_dirty; // needs refresh?
+ int m_column; // index of the column being resized
+ int m_currentX; // divider line position in logical (unscrolled) coords
+ int m_minX; // minimal position beyond which the divider line
+ // can't be dragged in logical coords
+
+ // the pen used to draw the current column width drag line
+ // when resizing the columsn
+ wxPen m_penCurrent;
+
+
+ // internal utilities:
+
+ void Init()
+ {
+ m_currentCursor = (wxCursor *) NULL;
+ m_resizeCursor = new wxCursor( wxCURSOR_SIZEWE );
+
+ m_isDragging = false;
+ m_dirty = false;
+
+ m_column = wxNOT_FOUND;
+ m_currentX = 0;
+ m_minX = 0;
+
+ wxColour col = wxSystemSettings::GetColour(wxSYS_COLOUR_3DLIGHT);
+ m_penCurrent = wxPen(col, 1, wxSOLID);
+ }
+
+ void DrawCurrent();
+ void AdjustDC(wxDC& dc);
+
+private:
+ DECLARE_DYNAMIC_CLASS(wxGenericDataViewHeaderWindow)
+ DECLARE_EVENT_TABLE()
+};
+
+#endif // defined(__WXMSW__)
+
+//-----------------------------------------------------------------------------
+// wxDataViewRenameTimer
+//-----------------------------------------------------------------------------
+
+class wxDataViewRenameTimer: public wxTimer
+{
+private:
+ wxDataViewMainWindow *m_owner;
+
+public:
+ wxDataViewRenameTimer( wxDataViewMainWindow *owner );
+ void Notify();
+};
+
+//-----------------------------------------------------------------------------
+// wxDataViewTreeNode
+//-----------------------------------------------------------------------------
+class wxDataViewTreeNode;
+WX_DEFINE_SORTED_ARRAY( wxDataViewTreeNode *, wxDataViewTreeNodes );
+WX_DEFINE_SORTED_ARRAY( void* , wxDataViewTreeLeaves);
+
+int LINKAGEMODE wxGenericTreeModelNodeCmp( wxDataViewTreeNode * node1, wxDataViewTreeNode * node2);
+int LINKAGEMODE wxGenericTreeModelItemCmp( void * id1, void * id2);
+
+class wxDataViewTreeNode
+{
+public:
+ wxDataViewTreeNode( wxDataViewTreeNode * parent )
+ :leaves( wxGenericTreeModelItemCmp ),
+ nodes(wxGenericTreeModelNodeCmp)
+ { this->parent = parent;
+ if( parent == NULL )
+ open = true;
+ else
+ open = false;
+ hasChildren = false;
+ subTreeCount = 0;
+ }
+ //I don't know what I need to do in the destructure
+ ~wxDataViewTreeNode()
+ {
+
+ }
+
+ wxDataViewTreeNode * GetParent() { return parent; }
+ void SetParent( wxDataViewTreeNode * parent ) { this->parent = parent; }
+ wxDataViewTreeNodes & GetNodes() { return nodes; }
+ wxDataViewTreeLeaves & GetChildren() { return leaves; }
+
+ void AddNode( wxDataViewTreeNode * node )
+ {
+ nodes.Add( node );
+ leaves.Add( node->GetItem().GetID() );
+ }
+ void AddLeaf( void * leaf ) { leaves.Add( leaf ); }
+
+ wxDataViewItem & GetItem() { return item; }
+ void SetItem( const wxDataViewItem & item ) { this->item = item; }
+
+ unsigned int GetChildrenNumber() { return leaves.GetCount(); }
+ unsigned int GetNodeNumber() { return nodes.GetCount(); }
+ int GetIndentLevel()
+ {
+ int ret = 0 ;
+ wxDataViewTreeNode * node = this;
+ while( node->GetParent()->GetParent() != NULL )
+ {
+ node = node->GetParent();
+ ret ++;
+ }
+ return ret;
+ }
+
+ bool IsOpen()
+ {
+ return open ;
+ }
+
+ void ToggleOpen()
+ {
+ int len = nodes.GetCount();
+ int sum = 0;
+ for ( int i = 0 ;i < len ; i ++)
+ sum += nodes[i]->GetSubTreeCount();
+
+ sum += leaves.GetCount();
+ if( open )
+ {
+ ChangeSubTreeCount(-sum);
+ open = !open;
+ }
+ else
+ {
+ open = !open;
+ ChangeSubTreeCount(sum);
+ }
+ }
+ bool HasChildren() { return hasChildren; }
+ void SetHasChildren( bool has ){ hasChildren = has; }
+
+ void SetSubTreeCount( int num ) { subTreeCount = num; }
+ int GetSubTreeCount() { return subTreeCount; }
+ void ChangeSubTreeCount( int num )
+ {
+ if( !open )
+ return ;
+ subTreeCount += num;
+ if( parent )
+ parent->ChangeSubTreeCount(num);
+ }
+
+ void Resort()
+ {
+ wxDataViewTreeNodes nds = nodes;
+ wxDataViewTreeLeaves lvs = leaves;
+ nodes.Empty();
+ leaves.Empty();
+
+ int len = nds.GetCount();
+ if(len > 0)
+ {
+ int i;
+ for(i = 0; i < len; i ++)
+ nodes.Add(nds[i]);
+ for(i = 0; i < len; i ++)
+ nodes[i]->Resort();
+ }
+
+ len = lvs.GetCount();
+ if(len > 0)
+ {
+ for(int i = 0; i < len; i++)
+ leaves.Add(lvs[i]);
+ }
+ }
+
+private:
+ wxDataViewTreeNode * parent;
+ wxDataViewTreeNodes nodes;
+ wxDataViewTreeLeaves leaves;
+ wxDataViewItem item;
+ bool open;
+ bool hasChildren;
+ int subTreeCount;
+};
+
+//Below is the compare stuff
+//For the generic implements, both the leaf nodes and the nodes are sorted for fast search when needed
+static wxDataViewModel * g_model;
+
+int LINKAGEMODE wxGenericTreeModelNodeCmp( wxDataViewTreeNode * node1, wxDataViewTreeNode * node2)
+{
+ return g_model->Compare( node1->GetItem(), node2->GetItem() );
+}
+
+int LINKAGEMODE wxGenericTreeModelItemCmp( void * id1, void * id2)
+{
+ return g_model->Compare( id1, id2 );
+}
+
+
+
+//-----------------------------------------------------------------------------
+// wxDataViewMainWindow
+//-----------------------------------------------------------------------------
+
+WX_DEFINE_SORTED_USER_EXPORTED_ARRAY_SIZE_T(unsigned int, wxDataViewSelection,
+ WXDLLIMPEXP_ADV);
+WX_DECLARE_LIST(wxDataViewItem, ItemList);
+WX_DEFINE_LIST(ItemList);
+
+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();
+
+ // 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 col );
+ bool Cleared();
+ void Resort()
+ { g_model = GetOwner()->GetModel(); m_root->Resort(); UpdateDisplay(); }
+
+ void SetOwner( wxDataViewCtrl* owner ) { m_owner = owner; }
+ wxDataViewCtrl *GetOwner() { return m_owner; }
+ const wxDataViewCtrl *GetOwner() const { return m_owner; }
+
+ void OnPaint( wxPaintEvent &event );
+ void OnArrowChar(unsigned int newCurrent, const wxKeyEvent& event);
+ void OnChar( 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 );
+
+ bool HasCurrentRow() { return m_currentRow != (unsigned int)-1; }
+ void ChangeCurrentRow( unsigned int row );
+
+ 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() ;
+
+ wxDataViewItem GetSelection();
+
+ 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 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);
+ }
+
+ //void EnsureVisible( unsigned int row );
+ wxRect GetLineRect( unsigned int row ) const;
+
+ //Some useful functions for row and item mapping
+ wxDataViewItem GetItemByRow( unsigned int row );
+ unsigned int GetRowByItem( const wxDataViewItem & item );
+
+ //Methods for building the mapping tree
+ void BuildTree( wxDataViewModel * model );
+ void DestroyTree();
+private:
+ wxDataViewTreeNode * GetTreeNodeByRow( unsigned int row );
+ //We did not need this temporarily
+ //wxDataViewTreeNode * GetTreeNodeByItem( const wxDataViewItem & item );
+
+ int RecalculateCount() ;
+
+ void OnExpanding( unsigned int row );
+ void OnCollapsing( unsigned int row );
+
+ wxDataViewTreeNode * FindNode( const wxDataViewItem & item );
+
+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;
+
+ int m_dragCount;
+ wxPoint m_dragStart;
+
+ // 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;
+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) :
+ wxDataViewRendererBase( varianttype, mode, align )
+{
+ m_dc = NULL;
+ m_align = align;
+ m_mode = mode;
+}
+
+wxDataViewRenderer::~wxDataViewRenderer()
+{
+ if (m_dc)
+ 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;
+}
+
+// ---------------------------------------------------------
+// wxDataViewCustomRenderer
+// ---------------------------------------------------------
+
+IMPLEMENT_ABSTRACT_CLASS(wxDataViewCustomRenderer, wxDataViewRenderer)
+
+wxDataViewCustomRenderer::wxDataViewCustomRenderer( const wxString &varianttype,
+ wxDataViewCellMode mode, int align ) :
+ wxDataViewRenderer( varianttype, mode, align )
+{
+}
+
+// ---------------------------------------------------------
+// wxDataViewTextRenderer
+// ---------------------------------------------------------
+
+IMPLEMENT_CLASS(wxDataViewTextRenderer, wxDataViewCustomRenderer)
+
+wxDataViewTextRenderer::wxDataViewTextRenderer( const wxString &varianttype,
+ wxDataViewCellMode mode, int align ) :
+ wxDataViewCustomRenderer( 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()
+{
+ return true;
+}
+
+wxControl* wxDataViewTextRenderer::CreateEditorCtrl( wxWindow *parent,
+ wxRect labelRect, const wxVariant &value )
+{
+ return new wxTextCtrl( parent, wxID_ANY, value,
+ wxPoint(labelRect.x,labelRect.y),
+ wxSize(labelRect.width,labelRect.height) );
+}
+
+bool wxDataViewTextRenderer::GetValueFromEditorCtrl( wxControl *editor, wxVariant &value )
+{
+ wxTextCtrl *text = (wxTextCtrl*) editor;
+ value = text->GetValue();
+ return true;
+}
+
+bool wxDataViewTextRenderer::Render( wxRect cell, wxDC *dc, int state )
+{
+ wxDataViewCtrl *view = GetOwner()->GetOwner();
+ wxColour col = (state & wxDATAVIEW_CELL_SELECTED) ?
+ wxSystemSettings::GetColour(wxSYS_COLOUR_HIGHLIGHTTEXT) :
+ view->GetForegroundColour();
+
+ dc->SetTextForeground(col);
+ dc->DrawText( m_text, cell.x, cell.y );
+
+ return true;
+}
+
+wxSize wxDataViewTextRenderer::GetSize() const
+{
+ const wxDataViewCtrl *view = GetView();
+ if (!m_text.empty())
+ {
+ int x,y;
+ view->GetTextExtent( m_text, &x, &y );
+ return wxSize( x, y );
+ }
+ return wxSize(80,20);
+}
+
+// ---------------------------------------------------------
+// wxDataViewBitmapRenderer
+// ---------------------------------------------------------
+
+IMPLEMENT_CLASS(wxDataViewBitmapRenderer, wxDataViewCustomRenderer)
+
+wxDataViewBitmapRenderer::wxDataViewBitmapRenderer( const wxString &varianttype,
+ wxDataViewCellMode mode, int align ) :
+ wxDataViewCustomRenderer( 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.Ok())
+ dc->DrawBitmap( m_bitmap, cell.x, cell.y );
+ else if (m_icon.Ok())
+ dc->DrawIcon( m_icon, cell.x, cell.y );
+
+ return true;
+}
+
+wxSize wxDataViewBitmapRenderer::GetSize() const
+{
+ if (m_bitmap.Ok())
+ return wxSize( m_bitmap.GetWidth(), m_bitmap.GetHeight() );
+ else if (m_icon.Ok())
+ return wxSize( m_icon.GetWidth(), m_icon.GetHeight() );
+
+ return wxSize(16,16);
+}
+
+// ---------------------------------------------------------
+// wxDataViewToggleRenderer
+// ---------------------------------------------------------
+
+IMPLEMENT_ABSTRACT_CLASS(wxDataViewToggleRenderer, wxDataViewCustomRenderer)
+
+wxDataViewToggleRenderer::wxDataViewToggleRenderer( const wxString &varianttype,
+ wxDataViewCellMode mode, int align ) :
+ wxDataViewCustomRenderer( 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) )
+{
+ // User wxRenderer here
+
+ wxRect rect;
+ rect.x = cell.x + cell.width/2 - 10;
+ rect.width = 20;
+ rect.y = cell.y + cell.height/2 - 10;
+ rect.height = 20;
+
+ int flags = 0;
+ if (m_toggle)
+ flags |= wxCONTROL_CHECKED;
+ if (GetMode() != wxDATAVIEW_CELL_ACTIVATABLE)
+ flags |= wxCONTROL_DISABLED;
+
+ wxRendererNative::Get().DrawCheckBox(
+ GetOwner()->GetOwner(),
+ *dc,
+ rect,
+ flags );
+
+ return true;
+}
+
+bool wxDataViewToggleRenderer::Activate( wxRect WXUNUSED(cell),
+ wxDataViewModel *model,
+ const wxDataViewItem & item, unsigned int col)
+{
+ bool value = !m_toggle;
+ wxVariant variant = value;
+ model->SetValue( variant, item, col);
+ model->ValueChanged( item, col );
+ return true;
+}
+
+wxSize wxDataViewToggleRenderer::GetSize() const
+{
+ return wxSize(20,20);
+}
+
+// ---------------------------------------------------------
+// wxDataViewProgressRenderer
+// ---------------------------------------------------------
+
+IMPLEMENT_ABSTRACT_CLASS(wxDataViewProgressRenderer, wxDataViewCustomRenderer)
+
+wxDataViewProgressRenderer::wxDataViewProgressRenderer( const wxString &label,
+ const wxString &varianttype, wxDataViewCellMode mode, int align ) :
+ wxDataViewCustomRenderer( varianttype, mode, align )
+{
+ m_label = label;
+ m_value = 0;
+}
+
+wxDataViewProgressRenderer::~wxDataViewProgressRenderer()
+{
+}
+
+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 cell, wxDC *dc, int WXUNUSED(state) )
+{
+ double pct = (double)m_value / 100.0;
+ wxRect bar = cell;
+ bar.width = (int)(cell.width * pct);
+ dc->SetPen( *wxTRANSPARENT_PEN );
+ dc->SetBrush( *wxBLUE_BRUSH );
+ dc->DrawRectangle( bar );
+
+ dc->SetBrush( *wxTRANSPARENT_BRUSH );
+ dc->SetPen( *wxBLACK_PEN );
+ dc->DrawRectangle( cell );
+
+ return true;
+}
+
+wxSize wxDataViewProgressRenderer::GetSize() const
+{
+ return wxSize(40,12);
+}
+
+// ---------------------------------------------------------
+// wxDataViewDateRenderer
+// ---------------------------------------------------------
+
+#define wxUSE_DATE_RENDERER_POPUP (wxUSE_CALENDARCTRL && wxUSE_POPUPWIN)
+
+#if wxUSE_DATE_RENDERER_POPUP
+
+class wxDataViewDateRendererPopupTransient: public wxPopupTransientWindow
+{
+public:
+ wxDataViewDateRendererPopupTransient( wxWindow* parent, wxDateTime *value,
+ wxDataViewModel *model, const wxDataViewItem & item, unsigned int col) :
+ wxPopupTransientWindow( parent, wxBORDER_SIMPLE ),
+ m_item( item )
+ {
+ m_model = model;
+ m_col = col;
+ m_cal = new wxCalendarCtrl( this, wxID_ANY, *value );
+ wxBoxSizer *sizer = new wxBoxSizer( wxHORIZONTAL );
+ sizer->Add( m_cal, 1, wxGROW );
+ SetSizer( sizer );
+ sizer->Fit( this );
+ }
+
+ void OnCalendar( wxCalendarEvent &event );
+
+ wxCalendarCtrl *m_cal;
+ wxDataViewModel *m_model;
+ unsigned int m_col;
+ const wxDataViewItem & m_item;
+
+protected:
+ virtual void OnDismiss()
+ {
+ }
+
+private:
+ DECLARE_EVENT_TABLE()
+};
+
+BEGIN_EVENT_TABLE(wxDataViewDateRendererPopupTransient,wxPopupTransientWindow)
+ EVT_CALENDAR( wxID_ANY, wxDataViewDateRendererPopupTransient::OnCalendar )
+END_EVENT_TABLE()
+
+void wxDataViewDateRendererPopupTransient::OnCalendar( wxCalendarEvent &event )
+{
+ wxDateTime date = event.GetDate();
+ wxVariant value = date;
+ m_model->SetValue( value, m_item, m_col );
+ m_model->ValueChanged( m_item, m_col );
+ DismissAndNotify();
+}
+
+#endif // wxUSE_DATE_RENDERER_POPUP
+
+IMPLEMENT_ABSTRACT_CLASS(wxDataViewDateRenderer, wxDataViewCustomRenderer)
+
+wxDataViewDateRenderer::wxDataViewDateRenderer( const wxString &varianttype,
+ wxDataViewCellMode mode, int align ) :
+ wxDataViewCustomRenderer( varianttype, mode, align )
+{
+}
+
+bool wxDataViewDateRenderer::SetValue( const wxVariant &value )
+{
+ m_date = value.GetDateTime();
+
+ return true;
+}
+
+bool wxDataViewDateRenderer::GetValue( wxVariant &value ) const
+{
+ value = m_date;
+ return true;
+}
+
+bool wxDataViewDateRenderer::Render( wxRect cell, wxDC *dc, int WXUNUSED(state) )
+{
+ dc->SetFont( GetOwner()->GetOwner()->GetFont() );
+ wxString tmp = m_date.FormatDate();
+ dc->DrawText( tmp, cell.x, cell.y );
+
+ return true;
+}
+
+wxSize wxDataViewDateRenderer::GetSize() const
+{
+ const wxDataViewCtrl* view = GetView();
+ wxString tmp = m_date.FormatDate();
+ wxCoord x,y,d;
+ view->GetTextExtent( tmp, &x, &y, &d );
+ return wxSize(x,y+d);
+}
+
+bool wxDataViewDateRenderer::Activate( wxRect WXUNUSED(cell), wxDataViewModel *model,
+ const wxDataViewItem & item, unsigned int col )
+{
+ wxVariant variant;
+ model->GetValue( variant, item, col );
+ wxDateTime value = variant.GetDateTime();
+
+#if wxUSE_DATE_RENDERER_POPUP
+ wxDataViewDateRendererPopupTransient *popup = new wxDataViewDateRendererPopupTransient(
+ GetOwner()->GetOwner()->GetParent(), &value, model, item, col);
+ wxPoint pos = wxGetMousePosition();
+ popup->Move( pos );
+ popup->Layout();
+ popup->Popup( popup->m_cal );
+#else // !wxUSE_DATE_RENDERER_POPUP
+ wxMessageBox(value.Format());
+#endif // wxUSE_DATE_RENDERER_POPUP/!wxUSE_DATE_RENDERER_POPUP
+ return true;
+}
+
+// ---------------------------------------------------------
+// wxDataViewColumn
+// ---------------------------------------------------------
+
+IMPLEMENT_ABSTRACT_CLASS(wxDataViewColumn, wxDataViewColumnBase)
+
+wxDataViewColumn::wxDataViewColumn( const wxString &title, wxDataViewRenderer *cell,
+ unsigned int model_column,
+ int width, wxAlignment align, int flags ) :
+ wxDataViewColumnBase( title, cell, model_column, width, align, flags )
+{
+ SetAlignment(align);
+ SetTitle(title);
+ SetFlags(flags);
+
+ Init(width < 0 ? wxDVC_DEFAULT_WIDTH : width);
+}
+
+wxDataViewColumn::wxDataViewColumn( const wxBitmap &bitmap, wxDataViewRenderer *cell,
+ unsigned int model_column,
+ int width, wxAlignment align, int flags ) :
+ wxDataViewColumnBase( bitmap, cell, model_column, width, align, flags )
+{
+ SetAlignment(align);
+ SetFlags(flags);
+
+ Init(width < 0 ? wxDVC_TOGGLE_DEFAULT_WIDTH : width);
+}
+
+wxDataViewColumn::~wxDataViewColumn()
+{
+}
+
+void wxDataViewColumn::Init( int width )
+{
+ m_width = width;
+ m_minWidth = wxDVC_DEFAULT_MINWIDTH;
+ m_ascending = true;
+}
+
+void wxDataViewColumn::SetResizeable( bool resizeable )
+{
+ if (resizeable)
+ m_flags |= wxDATAVIEW_COL_RESIZABLE;
+ else
+ m_flags &= ~wxDATAVIEW_COL_RESIZABLE;
+}
+
+void wxDataViewColumn::SetHidden( bool hidden )
+{
+ if (hidden)
+ m_flags |= wxDATAVIEW_COL_HIDDEN;
+ else
+ m_flags &= ~wxDATAVIEW_COL_HIDDEN;
+
+ // tell our owner to e.g. update its scrollbars:
+ if (GetOwner())
+ GetOwner()->OnColumnChange();
+}
+
+void wxDataViewColumn::SetSortable( bool sortable )
+{
+ if (sortable)
+ m_flags |= wxDATAVIEW_COL_SORTABLE;
+ else
+ m_flags &= ~wxDATAVIEW_COL_SORTABLE;
+
+ // Update header button
+ if (GetOwner())
+ GetOwner()->OnColumnChange();
+}
+
+void wxDataViewColumn::SetSortOrder( bool ascending )
+{
+ m_ascending = ascending;
+
+ // Update header button
+ if (GetOwner())
+ GetOwner()->OnColumnChange();
+}
+
+bool wxDataViewColumn::IsSortOrderAscending() const
+{
+ return m_ascending;
+}
+
+void wxDataViewColumn::SetInternalWidth( int width )
+{
+ m_width = width;
+
+ // the scrollbars of the wxDataViewCtrl needs to be recalculated!
+ if (m_owner && m_owner->m_clientArea)
+ m_owner->m_clientArea->RecalculateDisplay();
+}
+
+void wxDataViewColumn::SetWidth( int width )
+{
+ m_owner->m_headerArea->UpdateDisplay();
+
+ SetInternalWidth(width);
+}
+
+
+//-----------------------------------------------------------------------------
+// wxDataViewHeaderWindowBase
+//-----------------------------------------------------------------------------
+
+void wxDataViewHeaderWindowBase::SendEvent(wxEventType type, unsigned int n)
+{
+ wxWindow *parent = GetParent();
+ wxDataViewEvent le(type, parent->GetId());
+
+ le.SetEventObject(parent);
+ le.SetColumn(n);
+ le.SetDataViewColumn(GetColumn(n));
+ le.SetModel(GetOwner()->GetModel());
+
+ // for events created by wxDataViewHeaderWindow the
+ // row / value fields are not valid
+
+ parent->GetEventHandler()->ProcessEvent(le);
+}
+
+#if defined(__WXMSW__) && USE_NATIVE_HEADER_WINDOW
+
+// implemented in msw/listctrl.cpp:
+int WXDLLIMPEXP_CORE wxMSWGetColumnClicked(NMHDR *nmhdr, POINT *ptClick);
+
+IMPLEMENT_ABSTRACT_CLASS(wxDataViewHeaderWindowMSW, wxWindow)
+
+bool wxDataViewHeaderWindowMSW::Create( wxDataViewCtrl *parent, wxWindowID id,
+ const wxPoint &pos, const wxSize &size,
+ const wxString &name )
+{
+ m_owner = parent;
+
+ if ( !CreateControl(parent, id, pos, size, 0, wxDefaultValidator, name) )
+ return false;
+
+ int x = pos.x == wxDefaultCoord ? 0 : pos.x,
+ y = pos.y == wxDefaultCoord ? 0 : pos.y,
+ w = size.x == wxDefaultCoord ? 1 : size.x,
+ h = size.y == wxDefaultCoord ? 22 : size.y;
+
+ // create the native WC_HEADER window:
+ WXHWND hwndParent = (HWND)parent->GetHandle();
+ WXDWORD msStyle = WS_CHILD | HDS_BUTTONS | HDS_HORZ | HDS_HOTTRACK | HDS_FULLDRAG;
+ m_hWnd = CreateWindowEx(0,
+ WC_HEADER,
+ (LPCTSTR) NULL,
+ msStyle,
+ x, y, w, h,
+ (HWND)hwndParent,
+ (HMENU)-1,
+ wxGetInstance(),
+ (LPVOID) NULL);
+ if (m_hWnd == NULL)
+ {
+ wxLogLastError(_T("CreateWindowEx"));
+ return false;
+ }
+
+ // we need to subclass the m_hWnd to force wxWindow::HandleNotify
+ // to call wxDataViewHeaderWindow::MSWOnNotify
+ SubclassWin(m_hWnd);
+
+ // the following is required to get the default win's font for
+ // header windows and must be done befor sending the HDM_LAYOUT msg
+ SetFont(GetFont());
+
+ RECT rcParent;
+ HDLAYOUT hdl;
+ WINDOWPOS wp;
+
+ // Retrieve the bounding rectangle of the parent window's
+ // client area, and then request size and position values
+ // from the header control.
+ ::GetClientRect((HWND)hwndParent, &rcParent);
+
+ hdl.prc = &rcParent;
+ hdl.pwpos = ℘
+ if (!SendMessage((HWND)m_hWnd, HDM_LAYOUT, 0, (LPARAM) &hdl))
+ {
+ wxLogLastError(_T("SendMessage"));
+ return false;
+ }
+
+ // Set the size, position, and visibility of the header control.
+ SetWindowPos((HWND)m_hWnd,
+ wp.hwndInsertAfter,
+ wp.x, wp.y,
+ wp.cx, wp.cy,
+ wp.flags | SWP_SHOWWINDOW);
+
+ // set our size hints: wxDataViewCtrl will put this wxWindow inside
+ // a wxBoxSizer and in order to avoid super-big header windows,
+ // we need to set our height as fixed
+ SetMinSize(wxSize(-1, wp.cy));
+ SetMaxSize(wxSize(-1, wp.cy));
+
+ return true;
+}
+
+wxDataViewHeaderWindowMSW::~wxDataViewHeaderWindow()
+{
+ UnsubclassWin();
+}
+
+void wxDataViewHeaderWindowMSW::UpdateDisplay()
+{
+ // remove old columns
+ for (int j=0, max=Header_GetItemCount((HWND)m_hWnd); j < max; j++)
+ Header_DeleteItem((HWND)m_hWnd, 0);
+
+ // add the updated array of columns to the header control
+ unsigned int cols = GetOwner()->GetColumnCount();
+ unsigned int added = 0;
+ wxDataViewModel * model = GetOwner()->GetModel();
+ for (unsigned int i = 0; i < cols; i++)
+ {
+ wxDataViewColumn *col = GetColumn( i );
+ if (col->IsHidden())
+ continue; // don't add it!
+
+ HDITEM hdi;
+ hdi.mask = HDI_TEXT | HDI_FORMAT | HDI_WIDTH;
+ hdi.pszText = (wxChar *) col->GetTitle().wx_str();
+ hdi.cxy = col->GetWidth();
+ hdi.cchTextMax = sizeof(hdi.pszText)/sizeof(hdi.pszText[0]);
+ hdi.fmt = HDF_LEFT | HDF_STRING;
+
+ // lParam is reserved for application's use:
+ // we store there the column index to use it later in MSWOnNotify
+ // (since columns may have been hidden)
+ hdi.lParam = (LPARAM)i;
+
+ // the native wxMSW implementation of the header window
+ // draws the column separator COLUMN_WIDTH_OFFSET pixels
+ // on the right: to correct this effect we make the column
+ // exactly COLUMN_WIDTH_OFFSET wider (for the first column):
+ if (i == 0)
+ hdi.cxy += COLUMN_WIDTH_OFFSET;
+
+ switch (col->GetAlignment())
+ {
+ case wxALIGN_LEFT:
+ hdi.fmt |= HDF_LEFT;
+ break;
+ case wxALIGN_CENTER:
+ case wxALIGN_CENTER_HORIZONTAL:
+ hdi.fmt |= HDF_CENTER;
+ break;
+ case wxALIGN_RIGHT:
+ hdi.fmt |= HDF_RIGHT;
+ break;
+
+ default:
+ // such alignment is not allowed for the column header!
+ wxFAIL;
+ }
+
+ SendMessage((HWND)m_hWnd, HDM_INSERTITEM,
+ (WPARAM)added, (LPARAM)&hdi);
+ added++;
+ }
+}
+
+unsigned int wxDataViewHeaderWindowMSW::GetColumnIdxFromHeader(NMHEADER *nmHDR)
+{
+ unsigned int idx;
+
+ // NOTE: we don't just return nmHDR->iItem because when there are
+ // hidden columns, nmHDR->iItem may be different from
+ // nmHDR->pitem->lParam
+
+ if (nmHDR->pitem && nmHDR->pitem->mask & HDI_LPARAM)
+ {
+ idx = (unsigned int)nmHDR->pitem->lParam;
+ return idx;
+ }
+
+ HDITEM item;
+ item.mask = HDI_LPARAM;
+ Header_GetItem((HWND)m_hWnd, nmHDR->iItem, &item);
+
+ return (unsigned int)item.lParam;
+}
+
+bool wxDataViewHeaderWindowMSW::MSWOnNotify(int idCtrl, WXLPARAM lParam, WXLPARAM *result)
+{
+ NMHDR *nmhdr = (NMHDR *)lParam;
+
+ // is it a message from the header?
+ if ( nmhdr->hwndFrom != (HWND)m_hWnd )
+ return wxWindow::MSWOnNotify(idCtrl, lParam, result);
+
+ NMHEADER *nmHDR = (NMHEADER *)nmhdr;
+ switch ( nmhdr->code )
+ {
+ case HDN_BEGINTRACK:
+ // user has started to resize a column:
+ // do we need to veto it?
+ if (!GetColumn(nmHDR->iItem)->IsResizeable())
+ {
+ // veto it!
+ *result = TRUE;
+ }
+ break;
+
+ case HDN_BEGINDRAG:
+ // user has started to reorder a column
+ break;
+
+ case HDN_ITEMCHANGING:
+ if (nmHDR->pitem != NULL &&
+ (nmHDR->pitem->mask & HDI_WIDTH) != 0)
+ {
+ int minWidth = GetColumnFromHeader(nmHDR)->GetMinWidth();
+ if (nmHDR->pitem->cxy < minWidth)
+ {
+ // do not allow the user to resize this column under
+ // its minimal width:
+ *result = TRUE;
+ }
+ }
+ break;
+
+ case HDN_ITEMCHANGED: // user is resizing a column
+ case HDN_ENDTRACK: // user has finished resizing a column
+ case HDN_ENDDRAG: // user has finished reordering a column
+
+ // update the width of the modified column:
+ if (nmHDR->pitem != NULL &&
+ (nmHDR->pitem->mask & HDI_WIDTH) != 0)
+ {
+ unsigned int idx = GetColumnIdxFromHeader(nmHDR);
+ unsigned int w = nmHDR->pitem->cxy;
+ wxDataViewColumn *col = GetColumn(idx);
+
+ // see UpdateDisplay() for more info about COLUMN_WIDTH_OFFSET
+ if (idx == 0 && w > COLUMN_WIDTH_OFFSET)
+ w -= COLUMN_WIDTH_OFFSET;
+
+ if (w >= (unsigned)col->GetMinWidth())
+ col->SetInternalWidth(w);
+ }
+ break;
+
+ case HDN_ITEMCLICK:
+ {
+ unsigned int idx = GetColumnIdxFromHeader(nmHDR);
+ wxDataViewModel * model = GetOwner()->GetModel();
+
+ if(nmHDR->iButton == 0)
+ {
+ wxDataViewColumn *col = GetColumn(idx);
+ if(col->IsSortable())
+ {
+ if(model && model->GetSortingColumn() == idx)
+ {
+ bool order = col->IsSortOrderAscending();
+ col->SetSortOrder(!order);
+ model->SetSortOrderAscending(!order);
+ }
+ else if(model)
+ {
+ model->SetSortingColumn(idx);
+ model->SetSortOrderAscending(true);
+ }
+ }
+ UpdateDisplay();
+ if(model)
+ model->Resort();
+ }
+
+ wxEventType evt = nmHDR->iButton == 0 ?
+ wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_CLICK :
+ wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK;
+ SendEvent(evt, idx);
+ }
+ break;
+
+ case NM_RCLICK:
+ {
+ // NOTE: for some reason (i.e. for a bug in Windows)
+ // the HDN_ITEMCLICK notification is not sent on
+ // right clicks, so we need to handle NM_RCLICK
+
+ POINT ptClick;
+ int column = wxMSWGetColumnClicked(nmhdr, &ptClick);
+ if (column != wxNOT_FOUND)
+ {
+ HDITEM item;
+ item.mask = HDI_LPARAM;
+ Header_GetItem((HWND)m_hWnd, column, &item);
+
+ // 'idx' may be different from 'column' if there are
+ // hidden columns...
+ unsigned int idx = (unsigned int)item.lParam;
+ SendEvent(wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK,
+ idx);
+ }
+ }
+ break;
+
+ case HDN_GETDISPINFOW:
+ // see wxListCtrl::MSWOnNotify for more info!
+ break;
+
+ case HDN_ITEMDBLCLICK:
+ {
+ unsigned int idx = GetColumnIdxFromHeader(nmHDR);
+ int w = GetOwner()->GetBestColumnWidth(idx);
+
+ // update the native control:
+ HDITEM hd;
+ ZeroMemory(&hd, sizeof(hd));
+ hd.mask = HDI_WIDTH;
+ hd.cxy = w;
+ Header_SetItem(GetHwnd(),
+ nmHDR->iItem, // NOTE: we don't want 'idx' here!
+ &hd);
+
+ // update the wxDataViewColumn class:
+ GetColumn(idx)->SetInternalWidth(w);
+ }
+ break;
+
+ default:
+ return wxWindow::MSWOnNotify(idCtrl, lParam, result);
+ }
+
+ return true;
+}
+
+void wxDataViewHeaderWindowMSW::ScrollWindow(int WXUNUSED(dx), int WXUNUSED(dy),
+ const wxRect *WXUNUSED(rect))
+{
+ wxSize ourSz = GetClientSize();
+ wxSize ownerSz = m_owner->GetClientSize();
+
+ // where should the (logical) origin of this window be placed?
+ int x1 = 0, y1 = 0;
+ m_owner->CalcUnscrolledPosition(0, 0, &x1, &y1);
+
+ // put this window on top of our parent and
+ SetWindowPos((HWND)m_hWnd, HWND_TOP, -x1, 0,
+ ownerSz.GetWidth() + x1, ourSz.GetHeight(),
+ SWP_SHOWWINDOW);
+}
+
+void wxDataViewHeaderWindowMSW::DoSetSize(int WXUNUSED(x), int WXUNUSED(y),
+ int WXUNUSED(w), int WXUNUSED(h),
+ int WXUNUSED(f))
+{
+ // the wxDataViewCtrl's internal wxBoxSizer will call this function when
+ // the wxDataViewCtrl window gets resized: the following dummy call
+ // to ScrollWindow() is required in order to get this header window
+ // correctly repainted when it's (horizontally) scrolled:
+
+ ScrollWindow(0, 0);
+}
+
+#else // !defined(__WXMSW__)
+
+IMPLEMENT_ABSTRACT_CLASS(wxGenericDataViewHeaderWindow, wxWindow)
+BEGIN_EVENT_TABLE(wxGenericDataViewHeaderWindow, wxWindow)
+ EVT_PAINT (wxGenericDataViewHeaderWindow::OnPaint)
+ EVT_MOUSE_EVENTS (wxGenericDataViewHeaderWindow::OnMouse)
+ EVT_SET_FOCUS (wxGenericDataViewHeaderWindow::OnSetFocus)
+END_EVENT_TABLE()
+
+bool wxGenericDataViewHeaderWindow::Create(wxDataViewCtrl *parent, wxWindowID id,
+ const wxPoint &pos, const wxSize &size,
+ const wxString &name )
+{
+ m_owner = parent;
+
+ if (!wxDataViewHeaderWindowBase::Create(parent, id, pos, size, name))
+ return false;
+
+ wxVisualAttributes attr = wxPanel::GetClassDefaultAttributes();
+ SetBackgroundStyle( wxBG_STYLE_CUSTOM );
+ SetOwnForegroundColour( attr.colFg );
+ SetOwnBackgroundColour( attr.colBg );
+ if (!m_hasFont)
+ SetOwnFont( attr.font );
+
+ // set our size hints: wxDataViewCtrl will put this wxWindow inside
+ // a wxBoxSizer and in order to avoid super-big header windows,
+ // we need to set our height as fixed
+ SetMinSize(wxSize(-1, HEADER_WINDOW_HEIGHT));
+ SetMaxSize(wxSize(-1, HEADER_WINDOW_HEIGHT));
+
+ return true;
+}
+
+void wxGenericDataViewHeaderWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
+{
+ int w, h;
+ GetClientSize( &w, &h );
+
+ wxAutoBufferedPaintDC dc( this );
+
+ dc.SetBackground(GetBackgroundColour());
+ dc.Clear();
+
+ int xpix;
+ m_owner->GetScrollPixelsPerUnit( &xpix, NULL );
+
+ int x;
+ m_owner->GetViewStart( &x, NULL );
+
+ // account for the horz scrollbar offset
+ dc.SetDeviceOrigin( -x * xpix, 0 );
+
+ dc.SetFont( GetFont() );
+
+ unsigned int cols = GetOwner()->GetColumnCount();
+ unsigned int i;
+ int xpos = 0;
+ for (i = 0; i < cols; i++)
+ {
+ wxDataViewColumn *col = GetColumn( i );
+ if (col->IsHidden())
+ continue; // skip it!
+
+ int cw = col->GetWidth();
+ int ch = h;
+
+ wxHeaderSortIconType sortArrow = wxHDR_SORT_ICON_NONE;
+ if (col->IsSortable())
+ {
+ if (col->IsSortOrderAscending())
+ sortArrow = wxHDR_SORT_ICON_UP;
+ else
+ sortArrow = wxHDR_SORT_ICON_DOWN;
+ }
+
+ wxRendererNative::Get().DrawHeaderButton
+ (
+ this,
+ dc,
+ wxRect(xpos, 0, cw, ch-1),
+ m_parent->IsEnabled() ? 0
+ : (int)wxCONTROL_DISABLED,
+ sortArrow
+ );
+
+ // align as required the column title:
+ int x = xpos;
+ wxSize titleSz = dc.GetTextExtent(col->GetTitle());
+ switch (col->GetAlignment())
+ {
+ case wxALIGN_LEFT:
+ x += HEADER_HORIZ_BORDER;
+ break;
+ case wxALIGN_CENTER:
+ case wxALIGN_CENTER_HORIZONTAL:
+ x += (cw - titleSz.GetWidth() - 2 * HEADER_HORIZ_BORDER)/2;
+ break;
+ case wxALIGN_RIGHT:
+ x += cw - titleSz.GetWidth() - HEADER_HORIZ_BORDER;
+ break;
+ }
+
+ // always center the title vertically:
+ int y = wxMax((ch - titleSz.GetHeight()) / 2, HEADER_VERT_BORDER);
+
+ dc.SetClippingRegion( xpos+HEADER_HORIZ_BORDER,
+ HEADER_VERT_BORDER,
+ wxMax(cw - 2 * HEADER_HORIZ_BORDER, 1), // width
+ wxMax(ch - 2 * HEADER_VERT_BORDER, 1)); // height
+ dc.DrawText( col->GetTitle(), x, y );
+ dc.DestroyClippingRegion();
+
+ xpos += cw;
+ }
+}
+
+void wxGenericDataViewHeaderWindow::OnSetFocus( wxFocusEvent &event )
+{
+ GetParent()->SetFocus();
+ event.Skip();
+}
+
+void wxGenericDataViewHeaderWindow::OnMouse( wxMouseEvent &event )
+{
+ // we want to work with logical coords
+ int x;
+ m_owner->CalcUnscrolledPosition(event.GetX(), 0, &x, NULL);
+ int y = event.GetY();
+
+ if (m_isDragging)
+ {
+ // we don't draw the line beyond our window,
+ // but we allow dragging it there
+ int w = 0;
+ GetClientSize( &w, NULL );
+ m_owner->CalcUnscrolledPosition(w, 0, &w, NULL);
+ w -= 6;
+
+ // erase the line if it was drawn
+ if (m_currentX < w)
+ DrawCurrent();
+
+ if (event.ButtonUp())
+ {
+ m_isDragging = false;
+ if (HasCapture())
+ ReleaseMouse();
+
+ m_dirty = true;
+
+ GetColumn(m_column)->SetWidth(m_currentX - m_minX);
+
+ Refresh();
+ GetOwner()->Refresh();
+ }
+ else
+ {
+ m_currentX = wxMax(m_minX + 7, x);
+
+ // draw in the new location
+ if (m_currentX < w) DrawCurrent();
+ }
+
+ }
+ else // not dragging
+ {
+ m_minX = 0;
+ m_column = wxNOT_FOUND;
+
+ bool hit_border = false;
+
+ // end of the current column
+ int xpos = 0;
+
+ // find the column where this event occured
+ int countCol = m_owner->GetColumnCount();
+ for (int column = 0; column < countCol; column++)
+ {
+ wxDataViewColumn *p = GetColumn(column);
+
+ if (p->IsHidden())
+ continue; // skip if not shown
+
+ xpos += p->GetWidth();
+ m_column = column;
+ if ((abs(x-xpos) < 3) && (y < 22))
+ {
+ hit_border = true;
+ break;
+ }
+
+ if (x < xpos)
+ {
+ // inside the column
+ break;
+ }
+
+ m_minX = xpos;
+ }
+
+ if (m_column == wxNOT_FOUND)
+ return;
+
+ bool resizeable = GetColumn(m_column)->IsResizeable();
+ if (event.LeftDClick() && resizeable)
+ {
+ GetColumn(m_column)->SetWidth(GetOwner()->GetBestColumnWidth(m_column));
+ Refresh();
+ }
+ else if (event.LeftDown() || event.RightUp())
+ {
+ if (hit_border && event.LeftDown() && resizeable)
+ {
+ m_isDragging = true;
+ CaptureMouse();
+ m_currentX = x;
+ DrawCurrent();
+ }
+ else // click on a column
+ {
+ wxEventType evt = event.LeftDown() ?
+ wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_CLICK :
+ wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK;
+ SendEvent(evt, m_column);
+ }
+ }
+ else if (event.Moving())
+ {
+ if (hit_border && resizeable)
+ m_currentCursor = m_resizeCursor;
+ else
+ m_currentCursor = wxSTANDARD_CURSOR;
+
+ SetCursor(*m_currentCursor);
+ }
+ }
+}
+
+void wxGenericDataViewHeaderWindow::DrawCurrent()
+{
+ int x1 = m_currentX;
+ int y1 = 0;
+ ClientToScreen (&x1, &y1);
+
+ int x2 = m_currentX-1;
+#ifdef __WXMSW__
+ ++x2; // but why ????
+#endif
+ int y2 = 0;
+ m_owner->GetClientSize( NULL, &y2 );
+ m_owner->ClientToScreen( &x2, &y2 );
+
+ wxScreenDC dc;
+ dc.SetLogicalFunction(wxINVERT);
+ dc.SetPen(m_penCurrent);
+ dc.SetBrush(*wxTRANSPARENT_BRUSH);
+ AdjustDC(dc);
+ dc.DrawLine(x1, y1, x2, y2);
+}
+
+void wxGenericDataViewHeaderWindow::AdjustDC(wxDC& dc)
+{
+ int xpix, x;
+
+ m_owner->GetScrollPixelsPerUnit( &xpix, NULL );
+ m_owner->GetViewStart( &x, NULL );
+
+ // shift the DC origin to match the position of the main window horizontal
+ // scrollbar: this allows us to always use logical coords
+ dc.SetDeviceOrigin( -x * xpix, 0 );
+}
+
+#endif // defined(__WXMSW__)
+
+//-----------------------------------------------------------------------------
+// wxDataViewRenameTimer
+//-----------------------------------------------------------------------------
+
+wxDataViewRenameTimer::wxDataViewRenameTimer( wxDataViewMainWindow *owner )
+{
+ m_owner = owner;
+}
+
+void wxDataViewRenameTimer::Notify()
+{
+ m_owner->OnRenameTimer();
+}
+
+//-----------------------------------------------------------------------------
+// wxDataViewMainWindow
+//-----------------------------------------------------------------------------
+
+//The tree building helper, declared firstly
+void BuildTreeHelper( wxDataViewModel * model, 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 (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, name ),
+ m_selection( wxDataViewSelectionCmp )
+
+{
+ SetOwner( parent );
+
+ m_lastOnSame = false;
+ m_renameTimer = new wxDataViewRenameTimer( this );
+
+ // TODO: user better initial values/nothing selected
+ m_currentCol = NULL;
+ m_currentRow = 0;
+
+ // TODO: we need to calculate this smartly
+ m_lineHeight =
+#ifdef __WXMSW__
+ 17;
+#else
+ 20;
+#endif
+ wxASSERT(m_lineHeight > 2*PADDING_TOPBOTTOM);
+
+ m_dragCount = 0;
+ m_dragStart = wxPoint(0,0);
+ m_lineLastClicked = (unsigned int) -1;
+ m_lineBeforeLastClicked = (unsigned int) -1;
+ m_lineSelectSingleOnUp = (unsigned int) -1;
+
+ m_hasFocus = false;
+
+ SetBackgroundStyle( wxBG_STYLE_CUSTOM );
+ SetBackgroundColour( *wxWHITE );
+
+ m_penRule = wxPen(GetRuleColour(), 1, wxSOLID);
+
+ //Here I compose a pen can draw black lines, maybe there are something system colour to use
+ m_penExpander = wxPen( wxColour(0,0,0), 1, wxSOLID );
+ //Some new added code to deal with the tree structure
+ m_root = new wxDataViewTreeNode( NULL );
+ m_root->SetHasChildren(true);
+
+ //Make m_count = -1 will cause the class recaculate the real displaying number of rows.
+ m_count = -1 ;
+ UpdateDisplay();
+}
+
+wxDataViewMainWindow::~wxDataViewMainWindow()
+{
+ DestroyTree();
+ delete m_renameTimer;
+}
+
+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 )
+ wxSafeYield();
+
+ int xpos = 0;
+ unsigned int cols = GetOwner()->GetColumnCount();
+ unsigned int i;
+ for (i = 0; i < cols; i++)
+ {
+ wxDataViewColumn *c = GetOwner()->GetColumn( i );
+ if (c->IsHidden())
+ continue; // skip it!
+
+ if (c == m_currentCol)
+ break;
+ xpos += c->GetWidth();
+ }
+ wxRect labelRect( xpos, m_currentRow * m_lineHeight,
+ m_currentCol->GetWidth(), m_lineHeight );
+
+ GetOwner()->CalcScrolledPosition( labelRect.x, labelRect.y,
+ &labelRect.x, &labelRect.y);
+
+ wxDataViewItem item = GetItemByRow( m_currentRow );
+ m_currentCol->GetRenderer()->StartEditing( item, labelRect );
+}
+
+//------------------------------------------------------------------
+// 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
+ // 0: Job done, stop tranverse and return
+ // 1: Ignore the current node's subtree and continue
+ // 2: Job not done, continue
+ enum { OK = 0 , IGR = 1, CONT = 2 };
+ virtual int operator() ( wxDataViewTreeNode * node ) = 0 ;
+ virtual int operator() ( void * n ) = 0;
+};
+
+#if 0
+class ItemAddJob: public DoJob
+{
+public:
+ ItemAddJob( const wxDataViewItem & parent, const wxDataViewItem & item, int * count )
+ { this->parent = parent ; this->item = item ; m_count = count; }
+ virtual ~ItemAddJob(){};
+
+ virtual int operator() ( wxDataViewTreeNode * node )
+ {
+ if( node->GetItem() == parent )
+ {
+ node->SetHasChildren( true );
+ wxDataViewTreeNode * newnode = new wxDataViewTreeNode( node );
+ newnode->SetItem(item);
+ node->AppendChild( newnode);
+ *m_count = -1;
+ return OK;
+ }
+ return CONT;
+ }
+
+private:
+ int * m_count;
+ wxDataViewItem parent, item;
+};
+#endif
+
+bool Walker( wxDataViewTreeNode * node, DoJob & func )
+{
+ if( node==NULL || !node->HasChildren())
+ return false;
+
+ wxDataViewTreeNodes nodes = node->GetNodes();
+ wxDataViewTreeLeaves leaves = node->GetChildren();
+
+ int len_nodes = nodes.GetCount();
+ int len = leaves.GetCount();
+ int i = 0, nodes_i = 0;
+ for( ; i < len ; i ++ )
+ {
+ void * n = leaves[i];
+ if( nodes_i < len_nodes && n == nodes[nodes_i]->GetItem().GetID() )
+ {
+ wxDataViewTreeNode * nd = nodes[nodes_i];
+ nodes_i++;
+
+ switch( func( nd ) )
+ {
+ case DoJob::OK :
+ return true ;
+ case DoJob::IGR:
+ continue;
+ case DoJob::CONT:
+ default:
+ ;
+ }
+
+ if( Walker( nd , func ) )
+ return true;
+
+ }
+ else
+ switch( func( n ) )
+ {
+ case DoJob::OK :
+ return true ;
+ case DoJob::IGR:
+ continue;
+ case DoJob::CONT:
+ default:
+ ;
+ }
+ }
+ return false;
+}
+
+bool wxDataViewMainWindow::ItemAdded(const wxDataViewItem & parent, const wxDataViewItem & item)
+{
+ g_model = GetOwner()->GetModel();
+
+ wxDataViewTreeNode * node;
+ node = FindNode(parent);
+
+ if( node == NULL )
+ return false;
+
+ node->SetHasChildren( true );
+
+ if( g_model->IsContainer( item ) )
+ {
+ wxDataViewTreeNode * newnode = new wxDataViewTreeNode( node );
+ newnode->SetItem(item);
+ node->AddNode( newnode);
+ }
+ else
+ node->AddLeaf( item.GetID() );
+
+ node->ChangeSubTreeCount(1);
+
+ m_count = -1;
+ UpdateDisplay();
+ return true;
+}
+
+#if 0
+class ItemDeleteJob: public DoJob
+{
+public:
+ ItemDeleteJob( const wxDataViewItem & item, int * count ) { m_item = item; m_count = count; }
+ virtual ~ItemDeleteJob(){}
+ virtual int operator() ( wxDataViewTreeNode * node )
+ {
+ if( node->GetItem() == m_item )
+ {
+ node->GetParent()->GetChildren().Remove( node );
+ delete node;
+ *m_count = -1;
+ return DoJob::OK;
+ }
+ return DoJob::CONT;
+ }
+
+private:
+ int * m_count;
+ wxDataViewItem m_item;
+};
+#endif