]> git.saurik.com Git - wxWidgets.git/blobdiff - src/generic/datavgen.cpp
Some code renaming to avoid confusion. Test editable column in sample.
[wxWidgets.git] / src / generic / datavgen.cpp
index c9a8a6d3d23b2e8c8c9d8bb59329a3d54e58d5c1..3ca5d64b8708f53de0c6d234f83621b911181145 100644 (file)
@@ -33,6 +33,7 @@
     #include "wx/timer.h"
     #include "wx/settings.h"
     #include "wx/msgdlg.h"
+    #include "wx/dcscreen.h"
 #endif
 
 #include "wx/stockitem.h"
@@ -56,6 +57,8 @@ 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
@@ -63,14 +66,16 @@ static const int PADDING_TOPBOTTOM = 1;
 
 #define USE_NATIVE_HEADER_WINDOW    1
 
-class wxDataViewHeaderWindowBase : public wxControl
+// 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 wxPoint &pos, const wxSize &size,
                 const wxString &name)
     {
         return wxWindow::Create(parent, id, pos, size, wxNO_BORDER, name);
@@ -86,10 +91,10 @@ public:
     virtual wxDataViewColumn *GetColumn(unsigned int n)
     {
         wxASSERT(m_owner);
-        wxDataViewColumn *ret = m_owner->GetColumn(n); 
+        wxDataViewColumn *ret = m_owner->GetColumn(n);
         wxASSERT(ret);
 
-        return ret; 
+        return ret;
     }
 
 protected:
@@ -114,12 +119,12 @@ public:
                                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 wxPoint &pos, const wxSize &size,
                 const wxString &name);
 
     ~wxDataViewHeaderWindowMSW();
@@ -159,13 +164,13 @@ public:
                                     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 wxPoint &pos, const wxSize &size,
                 const wxString &name);
 
     ~wxGenericDataViewHeaderWindow()
@@ -173,6 +178,8 @@ public:
         delete m_resizeCursor;
     }
 
+    virtual void UpdateDisplay() { Refresh(); }
+    
     // event handlers:
 
     void OnPaint( wxPaintEvent &event );
@@ -191,7 +198,7 @@ protected:
     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 
+    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
@@ -242,49 +249,68 @@ public:
 };
 
 //-----------------------------------------------------------------------------
-// wxDataViewTextCtrlWrapper: wraps a wxTextCtrl for inline editing
+// wxDataViewTreeNode
 //-----------------------------------------------------------------------------
+class wxDataViewTreeNode; 
+WX_DEFINE_ARRAY_PTR( wxDataViewTreeNode *, wxDataViewTreeNodes );
 
-class wxDataViewTextCtrlWrapper : public wxEvtHandler
+class wxDataViewTreeNode
 {
 public:
-    // NB: text must be a valid object but not Create()d yet
-    wxDataViewTextCtrlWrapper( wxDataViewMainWindow *owner,
-                               wxTextCtrl *text,
-                               wxDataViewListModel *model,
-                               unsigned int col, unsigned int row,
-                               wxRect cellLabel );
-
-    wxTextCtrl *GetText() const { return m_text; }
-
-    void AcceptChangesAndFinish();
-
-protected:
-    void OnChar( wxKeyEvent &event );
-    void OnKeyUp( wxKeyEvent &event );
-    void OnKillFocus( wxFocusEvent &event );
-
-    bool AcceptChanges();
-    void Finish();
+    wxDataViewTreeNode( wxDataViewTreeNode * parent )
+       { this->parent = parent; 
+          if( parent == NULL )
+              open = true;
+          else
+               open = false;
+          hasChildren = false;
+       }
+    //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 &  GetChildren() { return children; }
+    void SetChildren( wxDataViewTreeNodes  children ) { this->children = children; }
+
+    wxDataViewTreeNode * GetChild( unsigned int n ) { return children.Item( n ); }
+    void InsertChild( wxDataViewTreeNode * child, unsigned int n) { children.Insert( child, n); }
+    void AppendChild( wxDataViewTreeNode * child ) { children.Add( child ); }
+
+    wxDataViewItem & GetItem() { return item; }
+    void SetItem( wxDataViewItem & item ) { this->item = item; }
+
+    unsigned int GetChildrenNumber() { return children.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(){ open = !open; }
+    bool HasChildren() { return hasChildren; }
+    void SetHasChildren( bool has ){ hasChildren = has; }
 private:
-    wxDataViewMainWindow   *m_owner;
-    wxTextCtrl             *m_text;
-    wxString                m_startValue;
-    wxDataViewListModel    *m_model;
-    unsigned int                  m_col;
-    unsigned int                  m_row;
-    bool                    m_finished;
-    bool                    m_aboutToFinish;
-
-    DECLARE_EVENT_TABLE()
+    wxDataViewTreeNode * parent;
+    wxDataViewTreeNodes  children; 
+    wxDataViewItem  item; 
+    bool open;
+    bool hasChildren;
 };
 
 //-----------------------------------------------------------------------------
 // wxDataViewMainWindow
 //-----------------------------------------------------------------------------
 
-WX_DEFINE_SORTED_USER_EXPORTED_ARRAY_SIZE_T(unsigned int, wxDataViewSelection, 
+WX_DEFINE_SORTED_USER_EXPORTED_ARRAY_SIZE_T(unsigned int, wxDataViewSelection,
                                             WXDLLIMPEXP_ADV);
 
 class wxDataViewMainWindow: public wxWindow
@@ -297,14 +323,11 @@ public:
                             const wxString &name = wxT("wxdataviewctrlmainwindow") );
     virtual ~wxDataViewMainWindow();
 
-    // notifications from wxDataViewListModel
-    bool RowAppended();
-    bool RowPrepended();
-    bool RowInserted( unsigned int before );
-    bool RowDeleted( unsigned int row );
-    bool RowChanged( unsigned int row );
-    bool ValueChanged( unsigned int col, unsigned int row );
-    bool RowsReordered( unsigned int *new_order );
+    // notifications from wxDataViewModel
+    bool ItemAdded( const wxDataViewItem &parent, const wxDataViewItem &item );
+    bool ItemDeleted( const wxDataViewItem &item );
+    bool ItemChanged( const wxDataViewItem &item );
+    bool ValueChanged( const wxDataViewItem &item, unsigned int col );
     bool Cleared();
 
     void SetOwner( wxDataViewCtrl* owner ) { m_owner = owner; }
@@ -323,21 +346,23 @@ public:
     void OnInternalIdle();
 
     void OnRenameTimer();
-    void FinishEditing( wxTextCtrl *text );
 
     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 IsSingleSel() const { return !GetParent()->HasFlag(wxDV_MULTIPLE); }
     bool IsEmpty() { return GetRowCount() == 0; }
 
     int GetCountPerPage() const;
     int GetEndOfLastCol() const;
     unsigned int GetFirstVisibleRow() const;
-    unsigned int GetLastVisibleRow() const;
-    unsigned int GetRowCount() 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 );
@@ -359,17 +384,33 @@ public:
     //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 );
+
 private:
     wxDataViewCtrl             *m_owner;
     int                         m_lineHeight;
     bool                        m_dirty;
 
     wxDataViewColumn           *m_currentCol;
-    unsigned int                      m_currentRow;
+    unsigned int                m_currentRow;
     wxDataViewSelection         m_selection;
 
     wxDataViewRenameTimer      *m_renameTimer;
-    wxDataViewTextCtrlWrapper  *m_textctrlWrapper;
     bool                        m_lastOnSame;
 
     bool                        m_hasFocus;
@@ -385,35 +426,35 @@ private:
     // 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()
 };
 
 // ---------------------------------------------------------
-// wxGenericDataViewListModelNotifier
+// wxGenericDataViewModelNotifier
 // ---------------------------------------------------------
 
-class wxGenericDataViewListModelNotifier: public wxDataViewListModelNotifier
+class wxGenericDataViewModelNotifier: public wxDataViewModelNotifier
 {
 public:
-    wxGenericDataViewListModelNotifier( wxDataViewMainWindow *mainWindow )
+    wxGenericDataViewModelNotifier( wxDataViewMainWindow *mainWindow )
         { m_mainWindow = mainWindow; }
 
-    virtual bool RowAppended()
-        { return m_mainWindow->RowAppended(); }
-    virtual bool RowPrepended()
-        { return m_mainWindow->RowPrepended(); }
-    virtual bool RowInserted( unsigned int before )
-        { return m_mainWindow->RowInserted( before ); }
-    virtual bool RowDeleted( unsigned int row )
-        { return m_mainWindow->RowDeleted( row ); }
-    virtual bool RowChanged( unsigned int row )
-        { return m_mainWindow->RowChanged( row ); }
-    virtual bool ValueChanged( unsigned int col, unsigned int row )
-        { return m_mainWindow->ValueChanged( col, row ); }
-    virtual bool RowsReordered( unsigned int *new_order )
-        { return m_mainWindow->RowsReordered( new_order ); }
+    virtual bool ItemAdded( const wxDataViewItem & parent, const wxDataViewItem & item )
+        { return m_mainWindow->ItemAdded( parent , item ); }
+    virtual bool ItemDeleted( const wxDataViewItem & item )
+        { return m_mainWindow->ItemDeleted( 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(); }
 
@@ -426,7 +467,7 @@ public:
 
 IMPLEMENT_ABSTRACT_CLASS(wxDataViewRenderer, wxDataViewRendererBase)
 
-wxDataViewRenderer::wxDataViewRenderer( const wxString &varianttype, 
+wxDataViewRenderer::wxDataViewRenderer( const wxString &varianttype,
                                         wxDataViewCellMode mode,
                                         int align) :
     wxDataViewRendererBase( varianttype, mode, align )
@@ -456,7 +497,6 @@ wxDC *wxDataViewRenderer::GetDC()
     return m_dc;
 }
 
-
 // ---------------------------------------------------------
 // wxDataViewCustomRenderer
 // ---------------------------------------------------------
@@ -475,7 +515,7 @@ wxDataViewCustomRenderer::wxDataViewCustomRenderer( const wxString &varianttype,
 
 IMPLEMENT_CLASS(wxDataViewTextRenderer, wxDataViewCustomRenderer)
 
-wxDataViewTextRenderer::wxDataViewTextRenderer( const wxString &varianttype, 
+wxDataViewTextRenderer::wxDataViewTextRenderer( const wxString &varianttype,
                                                 wxDataViewCellMode mode, int align ) :
     wxDataViewCustomRenderer( varianttype, mode, align )
 {
@@ -493,6 +533,26 @@ 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();
@@ -524,7 +584,7 @@ wxSize wxDataViewTextRenderer::GetSize() const
 
 IMPLEMENT_CLASS(wxDataViewBitmapRenderer, wxDataViewCustomRenderer)
 
-wxDataViewBitmapRenderer::wxDataViewBitmapRenderer( const wxString &varianttype, 
+wxDataViewBitmapRenderer::wxDataViewBitmapRenderer( const wxString &varianttype,
                                                     wxDataViewCellMode mode, int align ) :
     wxDataViewCustomRenderer( varianttype, mode, align )
 {
@@ -615,14 +675,14 @@ bool wxDataViewToggleRenderer::Render( wxRect cell, wxDC *dc, int WXUNUSED(state
     return true;
 }
 
-bool wxDataViewToggleRenderer::Activate( wxRect WXUNUSED(cell), 
-                                         wxDataViewListModel *model, 
-                                         unsigned int col, unsigned int row )
+bool wxDataViewToggleRenderer::Activate( wxRect WXUNUSED(cell),
+                                         wxDataViewModel *model,
+                                         const wxDataViewItem & item, unsigned int col)
 {
     bool value = !m_toggle;
     wxVariant variant = value;
-    model->SetValue( variant, col, row );
-    model->ValueChanged( col, row );
+    model->SetValue( variant, item, col);
+    model->ValueChanged( item, col );
     return true;
 }
 
@@ -698,12 +758,12 @@ class wxDataViewDateRendererPopupTransient: public wxPopupTransientWindow
 {
 public:
     wxDataViewDateRendererPopupTransient( wxWindow* parent, wxDateTime *value,
-        wxDataViewListModel *model, unsigned int col, unsigned int row ) :
-        wxPopupTransientWindow( parent, wxBORDER_SIMPLE )
+        wxDataViewModel *model, const wxDataViewItem & item, unsigned int col) :
+        wxPopupTransientWindow( parent, wxBORDER_SIMPLE ),
+        m_item( item )
     {
         m_model = model;
         m_col = col;
-        m_row = row;
         m_cal = new wxCalendarCtrl( this, wxID_ANY, *value );
         wxBoxSizer *sizer = new wxBoxSizer( wxHORIZONTAL );
         sizer->Add( m_cal, 1, wxGROW );
@@ -714,9 +774,9 @@ public:
     void OnCalendar( wxCalendarEvent &event );
 
     wxCalendarCtrl      *m_cal;
-    wxDataViewListModel *m_model;
+    wxDataViewModel *m_model;
     unsigned int               m_col;
-    unsigned int               m_row;
+    const wxDataViewItem &   m_item;
 
 protected:
     virtual void OnDismiss()
@@ -735,8 +795,8 @@ void wxDataViewDateRendererPopupTransient::OnCalendar( wxCalendarEvent &event )
 {
     wxDateTime date = event.GetDate();
     wxVariant value = date;
-    m_model->SetValue( value, m_col, m_row );
-    m_model->ValueChanged( m_col, m_row );
+    m_model->SetValue( value, m_item, m_col );
+    m_model->ValueChanged( m_item, m_col );
     DismissAndNotify();
 }
 
@@ -781,16 +841,16 @@ wxSize wxDataViewDateRenderer::GetSize() const
     return wxSize(x,y+d);
 }
 
-bool wxDataViewDateRenderer::Activate( wxRect WXUNUSED(cell), wxDataViewListModel *model, 
-                                       unsigned int col, unsigned int row )
+bool wxDataViewDateRenderer::Activate( wxRect WXUNUSED(cell), wxDataViewModel *model,
+                                       const wxDataViewItem & item, unsigned int col )
 {
     wxVariant variant;
-    model->GetValue( variant, col, row );
+    model->GetValue( variant, item, col );
     wxDateTime value = variant.GetDateTime();
 
 #if wxUSE_DATE_RENDERER_POPUP
     wxDataViewDateRendererPopupTransient *popup = new wxDataViewDateRendererPopupTransient(
-        GetOwner()->GetOwner()->GetParent(), &value, model, col, row );
+        GetOwner()->GetOwner()->GetParent(), &value, model, item, col);
     wxPoint pos = wxGetMousePosition();
     popup->Move( pos );
     popup->Layout();
@@ -807,7 +867,7 @@ bool wxDataViewDateRenderer::Activate( wxRect WXUNUSED(cell), wxDataViewListMode
 
 IMPLEMENT_ABSTRACT_CLASS(wxDataViewColumn, wxDataViewColumnBase)
 
-wxDataViewColumn::wxDataViewColumn( const wxString &title, wxDataViewRenderer *cell, 
+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 )
@@ -819,7 +879,7 @@ wxDataViewColumn::wxDataViewColumn( const wxString &title, wxDataViewRenderer *c
     Init(width < 0 ? wxDVC_DEFAULT_WIDTH : width);
 }
 
-wxDataViewColumn::wxDataViewColumn( const wxBitmap &bitmap, wxDataViewRenderer *cell, 
+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 )
@@ -838,6 +898,7 @@ void wxDataViewColumn::Init( int width )
 {
     m_width = width;
     m_minWidth = wxDVC_DEFAULT_MINWIDTH;
+    m_ascending = true;
 }
 
 void wxDataViewColumn::SetResizeable( bool resizeable )
@@ -866,17 +927,24 @@ void wxDataViewColumn::SetSortable( bool sortable )
         m_flags |= wxDATAVIEW_COL_SORTABLE;
     else
         m_flags &= ~wxDATAVIEW_COL_SORTABLE;
+        
+    // Update header button
+    if (GetOwner())
+        GetOwner()->OnColumnChange();
 }
 
-void wxDataViewColumn::SetSortOrder( bool WXUNUSED(ascending) )
+void wxDataViewColumn::SetSortOrder( bool ascending )
 {
-    // TODO
+    m_ascending = ascending;
+    
+    // Update header button
+    if (GetOwner())
+        GetOwner()->OnColumnChange();
 }
 
 bool wxDataViewColumn::IsSortOrderAscending() const
 {
-    // TODO
-    return true;
+    return m_ascending;
 }
 
 void wxDataViewColumn::SetInternalWidth( int width )
@@ -918,17 +986,13 @@ void wxDataViewHeaderWindowBase::SendEvent(wxEventType type, unsigned int n)
 
 #if defined(__WXMSW__) && USE_NATIVE_HEADER_WINDOW
 
-// implemented in msw/window.cpp:
-void wxAssociateWinWithHandle(HWND hWnd, wxWindowMSW *win);
-void wxRemoveHandleAssociation(wxWindowMSW *win);
-
 // implemented in msw/listctrl.cpp:
-unsigned int wxMSWGetColumnClicked(NMHDR *nmhdr, POINT *ptClick);
+int WXDLLIMPEXP_CORE wxMSWGetColumnClicked(NMHDR *nmhdr, POINT *ptClick);
 
-IMPLEMENT_ABSTRACT_CLASS(wxDataViewHeaderWindowMSW, wxWindow);
+IMPLEMENT_ABSTRACT_CLASS(wxDataViewHeaderWindowMSW, wxWindow)
 
 bool wxDataViewHeaderWindowMSW::Create( wxDataViewCtrl *parent, wxWindowID id,
-                                        const wxPoint &pos, const wxSize &size, 
+                                        const wxPoint &pos, const wxSize &size,
                                         const wxString &name )
 {
     m_owner = parent;
@@ -944,26 +1008,26 @@ bool wxDataViewHeaderWindowMSW::Create( wxDataViewCtrl *parent, wxWindowID id,
     // 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, 
+    m_hWnd = CreateWindowEx(0,
+                            WC_HEADER,
+                            (LPCTSTR) NULL,
                             msStyle,
                             x, y, w, h,
-                            (HWND)hwndParent, 
-                            (HMENU)-1, 
-                            wxGetInstance(), 
+                            (HWND)hwndParent,
+                            (HMENU)-1,
+                            wxGetInstance(),
                             (LPVOID) NULL);
-    if (m_hWnd == NULL) 
+    if (m_hWnd == NULL)
     {
         wxLogLastError(_T("CreateWindowEx"));
         return false;
     }
 
-    // we need to do the association to force wxWindow::HandleNotify
+    // we need to subclass the m_hWnd to force wxWindow::HandleNotify
     // to call wxDataViewHeaderWindow::MSWOnNotify
-    wxAssociateWinWithHandle((HWND)m_hWnd, this);
+    SubclassWin(m_hWnd);
 
-    // the following is required to get the default win's font for 
+    // 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());
 
@@ -971,24 +1035,24 @@ bool wxDataViewHeaderWindowMSW::Create( wxDataViewCtrl *parent, wxWindowID id,
     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 = &wp; 
-    if (!SendMessage((HWND)m_hWnd, HDM_LAYOUT, 0, (LPARAM) &hdl)) 
+    // 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 = &wp;
+    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, 
+
+    // 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
@@ -1002,15 +1066,15 @@ bool wxDataViewHeaderWindowMSW::Create( wxDataViewCtrl *parent, wxWindowID id,
 
 wxDataViewHeaderWindowMSW::~wxDataViewHeaderWindow()
 {
-    wxRemoveHandleAssociation(this);
+    UnsubclassWin();
 }
 
 void wxDataViewHeaderWindowMSW::UpdateDisplay()
 {
     // remove old columns
-    for (int i=0, max=Header_GetItemCount((HWND)m_hWnd); i < max; i++)
+    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;
@@ -1022,7 +1086,7 @@ void wxDataViewHeaderWindowMSW::UpdateDisplay()
 
         HDITEM hdi;
         hdi.mask = HDI_TEXT | HDI_FORMAT | HDI_WIDTH;
-        hdi.pszText = (wxChar *) col->GetTitle().c_str();
+        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;
@@ -1033,7 +1097,7 @@ void wxDataViewHeaderWindowMSW::UpdateDisplay()
         hdi.lParam = (LPARAM)i;
 
         // the native wxMSW implementation of the header window
-        // draws the column separator COLUMN_WIDTH_OFFSET pixels 
+        // 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)
@@ -1051,10 +1115,14 @@ void wxDataViewHeaderWindowMSW::UpdateDisplay()
         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); 
+
+        SendMessage((HWND)m_hWnd, HDM_INSERTITEM,
+                    (WPARAM)added, (LPARAM)&hdi);
         added++;
     }
 }
@@ -1106,13 +1174,13 @@ bool wxDataViewHeaderWindowMSW::MSWOnNotify(int idCtrl, WXLPARAM lParam, WXLPARA
             break;
 
         case HDN_ITEMCHANGING:
-            if (nmHDR->pitem != NULL && 
+            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 
+                    // do not allow the user to resize this column under
                     // its minimal width:
                     *result = TRUE;
                 }
@@ -1124,7 +1192,7 @@ bool wxDataViewHeaderWindowMSW::MSWOnNotify(int idCtrl, WXLPARAM lParam, WXLPARA
         case HDN_ENDDRAG:       // user has finished reordering a column
 
             // update the width of the modified column:
-            if (nmHDR->pitem != NULL && 
+            if (nmHDR->pitem != NULL &&
                 (nmHDR->pitem->mask & HDI_WIDTH) != 0)
             {
                 unsigned int idx = GetColumnIdxFromHeader(nmHDR);
@@ -1143,7 +1211,7 @@ bool wxDataViewHeaderWindowMSW::MSWOnNotify(int idCtrl, WXLPARAM lParam, WXLPARA
         case HDN_ITEMCLICK:
             {
                 unsigned int idx = GetColumnIdxFromHeader(nmHDR);
-                wxEventType evt = nmHDR->iButton == 0 ? 
+                wxEventType evt = nmHDR->iButton == 0 ?
                         wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_CLICK :
                         wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK;
                 SendEvent(evt, idx);
@@ -1187,7 +1255,7 @@ bool wxDataViewHeaderWindowMSW::MSWOnNotify(int idCtrl, WXLPARAM lParam, WXLPARA
                 ZeroMemory(&hd, sizeof(hd));
                 hd.mask = HDI_WIDTH;
                 hd.cxy = w;
-                Header_SetItem(GetHwnd(), 
+                Header_SetItem(GetHwnd(),
                                nmHDR->iItem,  // NOTE: we don't want 'idx' here!
                                &hd);
 
@@ -1203,7 +1271,7 @@ bool wxDataViewHeaderWindowMSW::MSWOnNotify(int idCtrl, WXLPARAM lParam, WXLPARA
     return true;
 }
 
-void wxDataViewHeaderWindowMSW::ScrollWindow(int WXUNUSED(dx), int WXUNUSED(dy), 
+void wxDataViewHeaderWindowMSW::ScrollWindow(int WXUNUSED(dx), int WXUNUSED(dy),
                                              const wxRect *WXUNUSED(rect))
 {
     wxSize ourSz = GetClientSize();
@@ -1213,14 +1281,14 @@ void wxDataViewHeaderWindowMSW::ScrollWindow(int WXUNUSED(dx), int WXUNUSED(dy),
     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(), 
+    // 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), 
+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
@@ -1241,8 +1309,8 @@ BEGIN_EVENT_TABLE(wxGenericDataViewHeaderWindow, wxWindow)
 END_EVENT_TABLE()
 
 bool wxGenericDataViewHeaderWindow::Create(wxDataViewCtrl *parent, wxWindowID id,
-                                           const wxPoint &pos, const wxSize &size, 
-                                           const wxString &name ) 
+                                           const wxPoint &pos, const wxSize &size,
+                                           const wxString &name )
 {
     m_owner = parent;
 
@@ -1298,13 +1366,23 @@ void wxGenericDataViewHeaderWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
         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
+                                                          : (int)wxCONTROL_DISABLED,
+                                    sortArrow
                                 );
 
         // align as required the column title:
@@ -1327,7 +1405,7 @@ void wxGenericDataViewHeaderWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
         // always center the title vertically:
         int y = wxMax((ch - titleSz.GetHeight()) / 2, HEADER_VERT_BORDER);
 
-        dc.SetClippingRegion( xpos+HEADER_HORIZ_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
@@ -1353,7 +1431,7 @@ void wxGenericDataViewHeaderWindow::OnMouse( wxMouseEvent &event )
 
     if (m_isDragging)
     {
-        // we don't draw the line beyond our window, 
+        // we don't draw the line beyond our window,
         // but we allow dragging it there
         int w = 0;
         GetClientSize( &w, NULL );
@@ -1361,13 +1439,13 @@ void wxGenericDataViewHeaderWindow::OnMouse( wxMouseEvent &event )
         w -= 6;
 
         // erase the line if it was drawn
-        if (m_currentX < w) 
+        if (m_currentX < w)
             DrawCurrent();
 
-        if (event.ButtonUp()) 
+        if (event.ButtonUp())
         {
             m_isDragging = false;
-            if (HasCapture()) 
+            if (HasCapture())
                 ReleaseMouse();
 
             m_dirty = true;
@@ -1398,22 +1476,22 @@ void wxGenericDataViewHeaderWindow::OnMouse( wxMouseEvent &event )
 
         // find the column where this event occured
         int countCol = m_owner->GetColumnCount();
-        for (int column = 0; column < countCol; column++) 
+        for (int column = 0; column < countCol; column++)
         {
             wxDataViewColumn *p = GetColumn(column);
 
-            if (p->IsHidden()) 
+            if (p->IsHidden())
                 continue;   // skip if not shown
 
             xpos += p->GetWidth();
             m_column = column;
-            if ((abs(x-xpos) < 3) && (y < 22)) 
+            if ((abs(x-xpos) < 3) && (y < 22))
             {
                 hit_border = true;
                 break;
             }
 
-            if (x < xpos) 
+            if (x < xpos)
             {
                 // inside the column
                 break;
@@ -1433,7 +1511,7 @@ void wxGenericDataViewHeaderWindow::OnMouse( wxMouseEvent &event )
         }
         else if (event.LeftDown() || event.RightUp())
         {
-            if (hit_border && event.LeftDown() && resizeable) 
+            if (hit_border && event.LeftDown() && resizeable)
             {
                 m_isDragging = true;
                 CaptureMouse();
@@ -1442,7 +1520,7 @@ void wxGenericDataViewHeaderWindow::OnMouse( wxMouseEvent &event )
             }
             else    // click on a column
             {
-                wxEventType evt = event.LeftDown() ? 
+                wxEventType evt = event.LeftDown() ?
                         wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_CLICK :
                         wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK;
                 SendEvent(evt, m_column);
@@ -1450,7 +1528,7 @@ void wxGenericDataViewHeaderWindow::OnMouse( wxMouseEvent &event )
         }
         else if (event.Moving())
         {
-            if (hit_border && resizeable) 
+            if (hit_border && resizeable)
                 m_currentCursor = m_resizeCursor;
             else
                 m_currentCursor = wxSTANDARD_CURSOR;
@@ -1510,153 +1588,13 @@ void wxDataViewRenameTimer::Notify()
     m_owner->OnRenameTimer();
 }
 
-//-----------------------------------------------------------------------------
-// wxDataViewTextCtrlWrapper: wraps a wxTextCtrl for inline editing
-//-----------------------------------------------------------------------------
-
-BEGIN_EVENT_TABLE(wxDataViewTextCtrlWrapper, wxEvtHandler)
-    EVT_CHAR           (wxDataViewTextCtrlWrapper::OnChar)
-    EVT_KEY_UP         (wxDataViewTextCtrlWrapper::OnKeyUp)
-    EVT_KILL_FOCUS     (wxDataViewTextCtrlWrapper::OnKillFocus)
-END_EVENT_TABLE()
-
-wxDataViewTextCtrlWrapper::wxDataViewTextCtrlWrapper(
-                        wxDataViewMainWindow *owner,
-                        wxTextCtrl *text,
-                        wxDataViewListModel *model,
-                        unsigned int col, unsigned int row,
-                        wxRect rectLabel )
-{
-    m_owner = owner;
-    m_model = model;
-    m_row = row;
-    m_col = col;
-    m_text = text;
-
-    m_finished = false;
-    m_aboutToFinish = false;
-
-    wxVariant value;
-    model->GetValue( value, col, row );
-    m_startValue = value.GetString();
-
-    m_owner->GetOwner()->CalcScrolledPosition(
-        rectLabel.x, rectLabel.y, &rectLabel.x, &rectLabel.y );
-
-    m_text->Create( owner, wxID_ANY, m_startValue,
-                    wxPoint(rectLabel.x-2,rectLabel.y-2),
-                    wxSize(rectLabel.width+7,rectLabel.height+4) );
-    m_text->SetFocus();
-
-    m_text->PushEventHandler(this);
-}
-
-void wxDataViewTextCtrlWrapper::AcceptChangesAndFinish()
-{
-    m_aboutToFinish = true;
-
-    // Notify the owner about the changes
-    AcceptChanges();
-
-    // Even if vetoed, close the control (consistent with MSW)
-    Finish();
-}
-
-void wxDataViewTextCtrlWrapper::OnChar( wxKeyEvent &event )
-{
-    switch ( event.m_keyCode )
-    {
-        case WXK_RETURN:
-            AcceptChangesAndFinish();
-            break;
-
-        case WXK_ESCAPE:
-            // m_owner->OnRenameCancelled( m_itemEdited );
-            Finish();
-            break;
-
-        default:
-            event.Skip();
-    }
-}
-
-void wxDataViewTextCtrlWrapper::OnKeyUp( wxKeyEvent &event )
-{
-    if (m_finished)
-    {
-        event.Skip();
-        return;
-    }
-
-    // auto-grow the textctrl
-    wxSize parentSize = m_owner->GetSize();
-    wxPoint myPos = m_text->GetPosition();
-    wxSize mySize = m_text->GetSize();
-    int sx, sy;
-    m_text->GetTextExtent(m_text->GetValue() + _T("MM"), &sx, &sy);
-    if (myPos.x + sx > parentSize.x)
-        sx = parentSize.x - myPos.x;
-    if (mySize.x > sx)
-        sx = mySize.x;
-    m_text->SetSize(sx, wxDefaultCoord);
-
-    event.Skip();
-}
-
-void wxDataViewTextCtrlWrapper::OnKillFocus( wxFocusEvent &event )
-{
-    if ( !m_finished && !m_aboutToFinish )
-    {
-        AcceptChanges();
-        //if ( !AcceptChanges() )
-        //    m_owner->OnRenameCancelled( m_itemEdited );
-
-        Finish();
-    }
-
-    // We must let the native text control handle focus
-    event.Skip();
-}
-
-bool wxDataViewTextCtrlWrapper::AcceptChanges()
-{
-    const wxString value = m_text->GetValue();
-
-    if ( value == m_startValue )
-        // nothing changed, always accept
-        return true;
-
-//    if ( !m_owner->OnRenameAccept(m_itemEdited, value) )
-        // vetoed by the user
-//        return false;
-
-    // accepted, do rename the item
-    wxVariant variant;
-    variant = value;
-    m_model->SetValue( variant, m_col, m_row );
-    m_model->ValueChanged( m_col, m_row );
-
-    return true;
-}
-
-void wxDataViewTextCtrlWrapper::Finish()
-{
-    if ( !m_finished )
-    {
-        m_finished = true;
-
-        m_text->RemoveEventHandler(this);
-        m_owner->FinishEditing(m_text);
-
-        // delete later
-        wxPendingDelete.Append( this );
-    }
-}
-
 //-----------------------------------------------------------------------------
 // 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;
@@ -1685,14 +1623,13 @@ wxDataViewMainWindow::wxDataViewMainWindow( wxDataViewCtrl *parent, wxWindowID i
 
     m_lastOnSame = false;
     m_renameTimer = new wxDataViewRenameTimer( this );
-    m_textctrlWrapper = NULL;
 
     // TODO: user better initial values/nothing selected
     m_currentCol = NULL;
     m_currentRow = 0;
 
     // TODO: we need to calculate this smartly
-    m_lineHeight = 
+    m_lineHeight =
 #ifdef __WXMSW__
         17;
 #else
@@ -1713,11 +1650,20 @@ wxDataViewMainWindow::wxDataViewMainWindow( wxDataViewCtrl *parent, wxWindowID i
 
     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;
 }
 
@@ -1728,7 +1674,6 @@ void wxDataViewMainWindow::OnRenameTimer()
     if ( m_dirty )
         wxSafeYield();
 
-
     int xpos = 0;
     unsigned int cols = GetOwner()->GetColumnCount();
     unsigned int i;
@@ -1742,71 +1687,153 @@ void wxDataViewMainWindow::OnRenameTimer()
             break;
         xpos += c->GetWidth();
     }
-    wxRect labelRect( xpos, m_currentRow * m_lineHeight, 
+    wxRect labelRect( xpos, m_currentRow * m_lineHeight,
                       m_currentCol->GetWidth(), m_lineHeight );
 
-    wxClassInfo *textControlClass = CLASSINFO(wxTextCtrl);
+    GetOwner()->CalcScrolledPosition( labelRect.x, labelRect.y,
+                                     &labelRect.x, &labelRect.y);
 
-    wxTextCtrl * const text = (wxTextCtrl *)textControlClass->CreateObject();
-    m_textctrlWrapper = new wxDataViewTextCtrlWrapper(this, text, GetOwner()->GetModel(),
-        m_currentCol->GetModelColumn(), m_currentRow, labelRect );
+    wxDataViewItem item = GetItemByRow( m_currentRow );
+    m_currentCol->GetRenderer()->StartEditing( item, labelRect );
 }
 
-void wxDataViewMainWindow::FinishEditing( wxTextCtrl *text )
+//------------------------------------------------------------------
+// Helper class for do operation on the tree node
+//------------------------------------------------------------------
+class DoJob
 {
-    delete text;
-    m_textctrlWrapper = NULL;
-    SetFocus();
-  //  SetFocusIgnoringChildren();
-}
+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 ;
+};
 
-bool wxDataViewMainWindow::RowAppended()
+class ItemAddJob: public DoJob
 {
-    return false;
-}
+public:
+    ItemAddJob( const wxDataViewItem & parent, const wxDataViewItem & item, int * count )
+       { this->parent = parent ; this->item = item ; m_count = count; }
+    virtual ~ItemAddJob(){};
 
-bool wxDataViewMainWindow::RowPrepended()
+    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;
+};
+
+bool Walker( wxDataViewTreeNode * node, DoJob & func )
 {
+    if( node==NULL ||  !node->HasChildren())
+        return false;
+
+    wxDataViewTreeNodes nodes = node->GetChildren();
+    int len = node->GetChildrenNumber();
+    int i = 0 ;
+    for( ; i < len ; i ++ )
+    {
+        wxDataViewTreeNode * n = nodes[i];
+        switch( func( n ) )
+        {
+            case DoJob::OK :
+                return true ;
+            case DoJob::IGR:
+                continue;
+            case DoJob::CONT:
+            default:
+                ;
+        }
+        
+        if( Walker( n , func ) )
+            return true;
+    }
     return false;
 }
 
-bool wxDataViewMainWindow::RowInserted( unsigned int WXUNUSED(before) )
+bool wxDataViewMainWindow::ItemAdded(const wxDataViewItem & parent, const wxDataViewItem & item)
 {
-    return false;
+    ItemAddJob job( parent, item, &m_count); 
+    Walker( m_root , job);
+    UpdateDisplay();
+    return true;
 }
 
-bool wxDataViewMainWindow::RowDeleted( unsigned int WXUNUSED(row) )
+class ItemDeleteJob: public DoJob
 {
-    return false;
+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;
+};
+
+bool wxDataViewMainWindow::ItemDeleted(const wxDataViewItem & item)
+{
+    ItemDeleteJob job( item, &m_count);
+    Walker( m_root, job);
+    UpdateDisplay();
+    return true;
 }
 
-bool wxDataViewMainWindow::RowChanged( unsigned int WXUNUSED(row) )
+bool wxDataViewMainWindow::ItemChanged(const wxDataViewItem & item)
 {
-    return false;
+    unsigned int row = GetRowByItem(item);
+    RefreshRow( row );
+    return true;
 }
 
-bool wxDataViewMainWindow::ValueChanged( unsigned int WXUNUSED(col), unsigned int row )
+bool wxDataViewMainWindow::ValueChanged( const wxDataViewItem & item, unsigned int WXUNUSED(col) )
 {
     // NOTE: to be valid, we cannot use e.g. INT_MAX - 1
-#define MAX_VIRTUAL_WIDTH       100000
+/*#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;
-}
-
-bool wxDataViewMainWindow::RowsReordered( unsigned int *WXUNUSED(new_order) )
-{
-    Refresh();
-
+*/
+    unsigned int row = GetRowByItem(item);
+    RefreshRow( row );
     return true;
 }
 
 bool wxDataViewMainWindow::Cleared()
 {
-    return false;
+    DestroyTree();
+    UpdateDisplay();
+    return true;
 }
 
 void wxDataViewMainWindow::UpdateDisplay()
@@ -1827,7 +1854,7 @@ void wxDataViewMainWindow::OnInternalIdle()
 
 void wxDataViewMainWindow::RecalculateDisplay()
 {
-    wxDataViewListModel *model = GetOwner()->GetModel();
+    wxDataViewModel *model = GetOwner()->GetModel();
     if (!model)
     {
         Refresh();
@@ -1835,7 +1862,7 @@ void wxDataViewMainWindow::RecalculateDisplay()
     }
 
     int width = GetEndOfLastCol();
-    int height = model->GetRowCount() * m_lineHeight;
+    int height = GetRowCount() * m_lineHeight;
 
     SetVirtualSize( width, height );
     GetOwner()->SetScrollRate( 10, m_lineHeight );
@@ -1853,7 +1880,7 @@ void wxDataViewMainWindow::ScrollWindow( int dx, int dy, const wxRect *rect )
 
 void wxDataViewMainWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
 {
-    wxDataViewListModel *model = GetOwner()->GetModel();
+    wxDataViewModel *model = GetOwner()->GetModel();
     wxAutoBufferedPaintDC dc( this );
 
     // prepare the DC
@@ -1867,9 +1894,9 @@ void wxDataViewMainWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
 
     // compute which items needs to be redrawn
     unsigned int item_start = wxMax( 0, (update.y / m_lineHeight) );
-    unsigned int item_count = 
+    unsigned int item_count =
         wxMin( (int)(((update.y + update.height) / m_lineHeight) - item_start + 1),
-               (int)(model->GetRowCount() - item_start) );
+               (int)(GetRowCount( )- item_start) );
     unsigned int item_last = item_start + item_count;
 
     // compute which columns needs to be redrawn
@@ -1936,7 +1963,7 @@ void wxDataViewMainWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
         }
 
         // Draw last vertical rule
-        dc.DrawLine(x, item_start * m_lineHeight, 
+        dc.DrawLine(x, item_start * m_lineHeight,
                     x, item_last * m_lineHeight);
     }
 
@@ -1946,7 +1973,7 @@ void wxDataViewMainWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
         bool selected = m_selection.Index( item ) != wxNOT_FOUND;
         if (selected || item == m_currentRow)
         {
-            int flags = selected ? wxCONTROL_SELECTED : 0;
+            int flags = selected ? (int)wxCONTROL_SELECTED : 0;
             if (item == m_currentRow)
                 flags |= wxCONTROL_CURRENT;
             if (m_hasFocus)
@@ -1980,15 +2007,55 @@ void wxDataViewMainWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
         {
             // get the cell value and set it into the renderer
             wxVariant value;
-            model->GetValue( value, col->GetModelColumn(), item );
+            wxDataViewTreeNode * node = GetTreeNodeByRow(item);
+            if( node == NULL )
+                continue;
+                
+            wxDataViewItem dataitem = node->GetItem();
+            model->GetValue( value, dataitem, col->GetModelColumn());
             cell->SetValue( value );
 
             // update the y offset
             cell_rect.y = item * m_lineHeight;
 
-            // cannot be bigger than allocated space
+            //Draw the expander here. Please notice that I use const number for all pixel data. When the final API are determined
+            //I will change this to the data member of the class wxDataViewCtrl
+            int indent = node->GetIndentLevel();
+            if( col->GetModelColumn() == GetOwner()->GetExpanderColumn() )
+            {
+                //Calculate the indent first
+                indent = cell_rect.x + GetOwner()->GetIndent() * indent;
+
+                int expander_width = m_lineHeight - 2*EXPANDER_MARGIN;
+                // change the cell_rect.x to the appropriate pos
+                int  expander_x = indent + EXPANDER_MARGIN , expander_y = cell_rect.y + EXPANDER_MARGIN ;
+                indent = indent + m_lineHeight ;  //try to use the m_lineHeight as the expander space
+                dc.SetPen( m_penExpander );
+                dc.SetBrush( wxNullBrush );
+                if( node->HasChildren() )
+                {
+                    //dc.DrawRoundedRectangle( expander_x,expander_y,expander_width,expander_width, 1.0);
+                    //dc.DrawLine( expander_x + 2 , expander_y + expander_width/2, expander_x + expander_width - 2, expander_y + expander_width/2 );
+                    wxRect rect( expander_x , expander_y, expander_width, expander_width);
+                    if( node->IsOpen() )
+                        wxRendererNative::Get().DrawTreeItemButton( this, dc, rect, wxCONTROL_EXPANDED );
+                    else
+                        wxRendererNative::Get().DrawTreeItemButton( this, dc, rect );
+                }
+                else
+                {
+                 // I am wandering whether we should draw dot lines between tree nodes
+                }
+
+                 //force the expander column to left-center align
+                 cell->SetAlignment( wxALIGN_CENTER_VERTICAL );
+            }
+            
+
+            // cannot be bigger than allocated space 
             wxSize size = cell->GetSize();
-            size.x = wxMin( size.x + 2*PADDING_RIGHTLEFT, cell_rect.width );
+            // Because of the tree structure indent, here we should minus the width of the cell for drawing
+            size.x = wxMin( size.x + 2*PADDING_RIGHTLEFT, cell_rect.width - indent );
             size.y = wxMin( size.y + 2*PADDING_TOPBOTTOM, cell_rect.height );
 
             wxRect item_rect(cell_rect.GetTopLeft(), size);
@@ -2016,6 +2083,9 @@ void wxDataViewMainWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
             item_rect.width = size.x - 2 * PADDING_RIGHTLEFT;
             item_rect.height = size.y - 2 * PADDING_TOPBOTTOM;
 
+            //Here we add the tree indent
+            item_rect.x += indent;
+                  
             int state = 0;
             if (m_selection.Index(item) != wxNOT_FOUND)
                 state |= wxDATAVIEW_CELL_SELECTED;
@@ -2048,7 +2118,7 @@ int wxDataViewMainWindow::GetEndOfLastCol() const
     unsigned int i;
     for (i = 0; i < GetOwner()->GetColumnCount(); i++)
     {
-        const wxDataViewColumn *c = 
+        const wxDataViewColumn *c =
             wx_const_cast(wxDataViewCtrl*, GetOwner())->GetColumn( i );
 
         if (!c->IsHidden())
@@ -2066,18 +2136,27 @@ unsigned int wxDataViewMainWindow::GetFirstVisibleRow() const
     return y / m_lineHeight;
 }
 
-unsigned int wxDataViewMainWindow::GetLastVisibleRow() const
+unsigned int wxDataViewMainWindow::GetLastVisibleRow() 
 {
     wxSize client_size = GetClientSize();
-    m_owner->CalcUnscrolledPosition( client_size.x, client_size.y, 
+    m_owner->CalcUnscrolledPosition( client_size.x, client_size.y,
                                      &client_size.x, &client_size.y );
 
     return wxMin( GetRowCount()-1, ((unsigned)client_size.y/m_lineHeight)+1 );
 }
 
-unsigned int wxDataViewMainWindow::GetRowCount() const
+unsigned int wxDataViewMainWindow::GetRowCount() 
 {
-    return wx_const_cast(wxDataViewCtrl*, GetOwner())->GetModel()->GetRowCount();
+    if ( m_count == -1 )
+    {
+        m_count = RecalculateCount();
+        int width, height;
+        GetVirtualSize( &width, &height );
+        height = m_count * m_lineHeight;
+
+        SetVirtualSize( width, height );
+    }
+    return m_count;
 }
 
 void wxDataViewMainWindow::ChangeCurrentRow( unsigned int row )
@@ -2290,6 +2369,227 @@ wxRect wxDataViewMainWindow::GetLineRect( unsigned int row ) const
     return rect;
 }
 
+class RowToItemJob: public DoJob
+{
+public:
+    RowToItemJob( unsigned int row , int current ) { this->row = row; this->current = current ;}
+    virtual ~RowToItemJob(){};
+
+    virtual int operator() ( wxDataViewTreeNode * node )
+       {
+           if( current == static_cast<int>(row))
+           {
+               ret = node->GetItem() ;
+               return DoJob::OK;
+           }
+           current ++;
+           if ( node->IsOpen())
+               return DoJob::CONT;
+           else
+               return DoJob::IGR;
+       }
+
+    wxDataViewItem GetResult(){ return ret; }
+private:
+    unsigned int row;
+    int current ;
+    wxDataViewItem ret;
+};
+
+wxDataViewItem wxDataViewMainWindow::GetItemByRow(unsigned int row)
+{
+    RowToItemJob job( row, 0 );
+    Walker( m_root , job );
+    return job.GetResult();
+}
+
+class RowToTreeNodeJob: public DoJob
+{
+public:
+    RowToTreeNodeJob( unsigned int row , int current ) { this->row = row; this->current = current ; ret = NULL ; }
+    virtual ~RowToTreeNodeJob(){};
+
+    virtual int operator() ( wxDataViewTreeNode * node )
+    {
+        if( current == static_cast<int>(row))
+        {
+            ret = node ;
+            return DoJob::OK;
+        }
+        current ++;
+        if ( node->IsOpen())
+            return DoJob::CONT;
+        else
+            return DoJob::IGR;
+    }
+
+    wxDataViewTreeNode * GetResult(){ return ret; }
+private:
+    unsigned int row;
+    int current ;
+    wxDataViewTreeNode * ret;
+};
+
+
+wxDataViewTreeNode * wxDataViewMainWindow::GetTreeNodeByRow(unsigned int row)
+{
+    RowToTreeNodeJob job( row , 0 );
+    Walker( m_root , job );
+    return job.GetResult();
+}
+
+class CountJob : public DoJob
+{
+public:
+    CountJob(){ count = 0 ; }
+    virtual ~CountJob(){};
+       
+    virtual int operator () (  wxDataViewTreeNode * node )
+    {
+         count ++;
+         if ( node->IsOpen())
+             return DoJob::CONT;
+         else
+             return DoJob::IGR;
+    }
+
+    unsigned int GetResult()
+    {
+        return count ;
+    }
+private:
+    unsigned int count;
+};
+
+void wxDataViewMainWindow::OnExpanding( unsigned int row )
+{
+    wxDataViewTreeNode * node = GetTreeNodeByRow(row);
+    if( node != NULL )
+    {
+        if( node->HasChildren())
+            if( !node->IsOpen())
+            {
+               node->ToggleOpen();
+               //Here I build the children of current node
+               if( node->GetChildrenNumber() == 0 )
+                   BuildTreeHelper(GetOwner()->GetModel(), node->GetItem(), node);
+               m_count = -1;
+               Refresh();
+            }
+    }
+}
+
+void wxDataViewMainWindow::OnCollapsing(unsigned int row)
+{
+    wxDataViewTreeNode * node = GetTreeNodeByRow(row);
+    if( node != NULL )
+    {
+        if( node->HasChildren() && node->IsOpen() )
+        {
+            node->ToggleOpen();
+            m_count = -1;
+            Refresh();
+            //RefreshRows(row,GetLastVisibleRow());
+         }
+         else
+         {
+             node = node->GetParent();
+             if( node != NULL )
+             {
+                 int  parent = GetRowByItem( node->GetItem()) ;
+                 SelectRow( row, false);
+                 SelectRow(parent , true );
+                 ChangeCurrentRow( parent );
+             }
+         }
+    }
+}
+
+int wxDataViewMainWindow::RecalculateCount() 
+{
+    CountJob job;
+    Walker( m_root, job );
+    return job.GetResult();
+}
+
+class ItemToRowJob : public DoJob
+{
+public:
+    ItemToRowJob(const wxDataViewItem & item){ this->item = item ; ret = 0 ; }
+    virtual ~ItemToRowJob(){};
+
+    virtual int operator() ( wxDataViewTreeNode * node)
+    {
+         ret ++;
+         if( node->GetItem() == item )
+             return DoJob::OK;
+
+         if( node->IsOpen())
+             return DoJob::CONT;
+         else
+             return DoJob::IGR;
+    }
+
+    //the row number is begin from zero
+    int GetResult(){ return ret -1 ; }
+private:
+    wxDataViewItem item;
+    int ret;
+};
+
+unsigned int wxDataViewMainWindow::GetRowByItem(const wxDataViewItem & item)
+{
+    ItemToRowJob job( item );
+    Walker(m_root , job );
+    return job.GetResult();
+}
+
+void BuildTreeHelper( wxDataViewModel * model,  wxDataViewItem & item, wxDataViewTreeNode * node)
+{
+    if( !model->HasChildren( item ) )
+        return ;
+    
+    wxDataViewItem i = model->GetFirstChild( item );
+    while( i.IsOk() )
+    {
+        wxDataViewTreeNode * n = new wxDataViewTreeNode( node );
+        n->SetItem(i);
+        n->SetHasChildren( model->HasChildren( i )) ;
+        node->AppendChild(n);
+        //BuildTreeHelper( model, i, n) ;        
+        i = model->GetNextSibling( i );
+    }
+}
+
+void wxDataViewMainWindow::BuildTree(wxDataViewModel * model)
+{
+    //First we define a invalid item to fetch the top-level elements
+    wxDataViewItem item;
+    BuildTreeHelper( model, item, m_root);
+    m_count = -1 ;
+}
+
+void DestroyTreeHelper( wxDataViewTreeNode * node )
+{
+    if( node->HasChildren() )
+    {
+        int len = node->GetChildrenNumber();
+        int i = 0 ;
+        wxDataViewTreeNodes nodes = node->GetChildren();
+        for( ; i < len; i ++ )
+        {
+            DestroyTreeHelper(nodes[i]);
+        }
+    }
+    delete node;
+}
+
+void wxDataViewMainWindow::DestroyTree()
+{
+    DestroyTreeHelper(m_root);
+    m_count = 0 ;
+}
+
 void wxDataViewMainWindow::OnChar( wxKeyEvent &event )
 {
     if (event.GetKeyCode() == WXK_TAB)
@@ -2325,7 +2625,13 @@ void wxDataViewMainWindow::OnChar( wxKeyEvent &event )
             if ( m_currentRow < GetRowCount() - 1 )
                 OnArrowChar( m_currentRow + 1, event );
             break;
-
+        //Add the process for tree expanding/collapsing
+        case WXK_LEFT:
+            OnCollapsing(m_currentRow);
+            break;
+        case WXK_RIGHT:
+            OnExpanding( m_currentRow);
+            break;
         case WXK_END:
             if (!IsEmpty())
                 OnArrowChar( GetRowCount() - 1, event );
@@ -2407,7 +2713,7 @@ void wxDataViewMainWindow::OnMouse( wxMouseEvent &event )
         return;
     }
 
-    wxDataViewListModel *model = GetOwner()->GetModel();
+    wxDataViewModel *model = GetOwner()->GetModel();
 
     if (event.Dragging())
     {
@@ -2449,12 +2755,13 @@ void wxDataViewMainWindow::OnMouse( wxMouseEvent &event )
         {
             if (cell->GetMode() == wxDATAVIEW_CELL_ACTIVATABLE)
             {
+                wxDataViewItem item = GetItemByRow(current);
                 wxVariant value;
-                model->GetValue( value, col->GetModelColumn(), current );
+                model->GetValue( value, item, col->GetModelColumn() );
                 cell->SetValue( value );
-                wxRect cell_rect( xpos, current * m_lineHeight, 
+                wxRect cell_rect( xpos, current * m_lineHeight,
                                   col->GetWidth(), m_lineHeight );
-                cell->Activate( cell_rect, model, col->GetModelColumn(), current );
+                cell->Activate( cell_rect, model, item, col->GetModelColumn() );
             }
             return;
         }
@@ -2475,7 +2782,26 @@ void wxDataViewMainWindow::OnMouse( wxMouseEvent &event )
             SelectRow( m_lineSelectSingleOnUp, true );
         }
 
-        if (m_lastOnSame)
+        //Process the event of user clicking the expander
+        bool expander = false;
+        wxDataViewTreeNode * node = GetTreeNodeByRow(current);
+        if( node!=NULL && node->HasChildren() )
+        {
+            int indent = node->GetIndentLevel();
+            indent = GetOwner()->GetIndent()*indent;
+            wxRect rect( xpos + indent + EXPANDER_MARGIN, current * m_lineHeight + EXPANDER_MARGIN, m_lineHeight-2*EXPANDER_MARGIN,m_lineHeight-2*EXPANDER_MARGIN);
+            if( rect.Contains( x, y) )
+            {
+                expander = true;
+                if( node->IsOpen() )
+                    OnCollapsing(current);
+                else
+                    OnExpanding( current );
+            }
+        }
+
+        //If the user click the expander, we do not do editing even if the column with expander are editable
+        if (m_lastOnSame && !expander )
         {
             if ((col == m_currentCol) && (current == m_currentRow) &&
                 (cell->GetMode() == wxDATAVIEW_CELL_EDITABLE) )
@@ -2588,7 +2914,7 @@ void wxDataViewMainWindow::OnMouse( wxMouseEvent &event )
         // Update selection here...
         m_currentCol = col;
 
-        m_lastOnSame = !forceClick && ((col == oldCurrentCol) && 
+        m_lastOnSame = !forceClick && ((col == oldCurrentCol) &&
                         (current == oldCurrentRow)) && oldWasSelected;
     }
 }
@@ -2613,6 +2939,13 @@ void wxDataViewMainWindow::OnKillFocus( wxFocusEvent &event )
     event.Skip();
 }
 
+wxDataViewItem wxDataViewMainWindow::GetSelection()
+{
+    if( m_selection.GetCount() != 1 )
+        return wxDataViewItem();
+    return GetItemByRow( m_selection.Item( 0 ) );
+}
+
 //-----------------------------------------------------------------------------
 // wxDataViewCtrl
 //-----------------------------------------------------------------------------
@@ -2638,7 +2971,7 @@ bool wxDataViewCtrl::Create(wxWindow *parent, wxWindowID id,
            const wxPoint& pos, const wxSize& size,
            long style, const wxValidator& validator )
 {
-    if (!wxControl::Create( parent, id, pos, size, 
+    if (!wxControl::Create( parent, id, pos, size,
                             style | wxScrolledWindowStyle|wxSUNKEN_BORDER, validator))
         return false;
 
@@ -2699,15 +3032,17 @@ void wxDataViewCtrl::OnSize( wxSizeEvent &WXUNUSED(event) )
     AdjustScrollbars();
 }
 
-bool wxDataViewCtrl::AssociateModel( wxDataViewListModel *model )
+bool wxDataViewCtrl::AssociateModel( wxDataViewModel *model )
 {
     if (!wxDataViewCtrlBase::AssociateModel( model ))
         return false;
 
-    m_notifier = new wxGenericDataViewListModelNotifier( m_clientArea );
+    m_notifier = new wxGenericDataViewModelNotifier( m_clientArea );
 
     model->AddNotifier( m_notifier );
 
+    m_clientArea->BuildTree(model);
+
     m_clientArea->UpdateDisplay();
 
     return true;
@@ -2730,6 +3065,22 @@ void wxDataViewCtrl::OnColumnChange()
     m_clientArea->UpdateDisplay();
 }
 
+void wxDataViewCtrl::DoSetExpanderColumn()
+{
+    m_clientArea->UpdateDisplay();
+}
+
+void wxDataViewCtrl::DoSetIndent()
+{
+    m_clientArea->UpdateDisplay();
+}
+
+wxDataViewItem wxDataViewCtrl::GetSelection() 
+{
+    return m_clientArea->GetSelection();
+}
+
+/********************************************************************
 void wxDataViewCtrl::SetSelection( int row )
 {
     m_clientArea->SelectRow(row, true);
@@ -2770,7 +3121,7 @@ int wxDataViewCtrl::GetSelections(wxArrayInt& WXUNUSED(aSelections) ) const
 
     return 0;
 }
-
+*********************************************************************/
 #endif
     // !wxUSE_GENERICDATAVIEWCTRL