]> git.saurik.com Git - wxWidgets.git/commitdiff
all code revisited.
authorFrancesco Montorsi <f18m_cpp217828@yahoo.it>
Wed, 4 Mar 2009 22:33:18 +0000 (22:33 +0000)
committerFrancesco Montorsi <f18m_cpp217828@yahoo.it>
Wed, 4 Mar 2009 22:33:18 +0000 (22:33 +0000)
Main additions:
- now it's possible to change the DVC style dynamically
- make models more robust against deleting items, and against other allowed user actions
- comment portions of the code to make it more didactive

git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@59323 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775

samples/dataview/dataview.cpp
samples/dataview/mymodels.cpp
samples/dataview/mymodels.h

index 3b5774ba7d87b0705fddc9b52eeb38fc18db8c0d..2b0dc8efd250737f3f08ab2498edbbfa1fee99bd 100644 (file)
@@ -66,10 +66,18 @@ public:
     MyFrame(wxFrame *frame, const wxString &title, int x, int y, int w, int h);
     ~MyFrame();
 
-public:
+    void BuildDataViewCtrl(wxPanel* parent, 
+                           unsigned int nPanel, 
+                           unsigned long style = 0);
+
+public:     // event handlers
+
+    void OnStyleChange(wxCommandEvent& event);
     void OnQuit(wxCommandEvent& event);
     void OnAbout(wxCommandEvent& event);
 
+    void OnPageChanged(wxBookCtrlEvent& event);
+
     void OnAddMozart(wxCommandEvent& event);
     void OnDeleteMusic(wxCommandEvent& event);
     void OnDeleteYear(wxCommandEvent& event);
@@ -109,14 +117,12 @@ private:
 
     // the controls stored in the various tabs of the main notebook:
 
-    wxDataViewCtrl* m_myMusicModelViewCtrl;
-    wxObjectDataPtr<MyMusicTreeModel> m_music_model;
+    wxDataViewCtrl* m_ctrl[4];
 
-    wxDataViewCtrl* m_myListModelViewCtrl;
-    wxObjectDataPtr<MyListModel> m_list_model;
+    // the models associated with the first two DVC:
 
-    wxDataViewListCtrl* m_listctrl;
-    wxDataViewTreeCtrl* m_treectrl;
+    wxObjectDataPtr<MyMusicTreeModel> m_music_model;
+    wxObjectDataPtr<MyListModel> m_list_model;
 
     // other data:
 
@@ -145,17 +151,16 @@ public:
     {
         dc->SetBrush( *wxRED_BRUSH );
         dc->SetPen( *wxTRANSPARENT_PEN );
-        dc->DrawRectangle( rect );
+        dc->DrawRectangle( rect.Deflate(2) );
         return true;
     }
 
-
     virtual bool Activate( wxRect WXUNUSED(cell),
                            wxDataViewModel *WXUNUSED(model), 
                            const wxDataViewItem &WXUNUSED(item), 
                            unsigned int WXUNUSED(col) )
     {
-        wxLogMessage( wxT("MyCustomRenderer Activate()") );
+        wxLogMessage( "MyCustomRenderer Activate()" );
         return false;
     }
 
@@ -164,7 +169,7 @@ public:
                            const wxDataViewItem &WXUNUSED(item), 
                            unsigned int WXUNUSED(col) )
     {
-        wxLogMessage( wxT("MyCustomRenderer LeftClick( %d, %d )"), cursor.x, cursor.y );
+        wxLogMessage( "MyCustomRenderer LeftClick( %d, %d )", cursor.x, cursor.y );
         return false;
     }
 
@@ -217,10 +222,23 @@ bool MyApp::OnInit()
 
 enum
 {
+    ID_STYLE_MENU = wxID_HIGHEST+1,
+
     // file menu
-    ID_ABOUT = wxID_ABOUT,
+    //ID_SINGLE,        wxDV_SINGLE==0 so it's always present
+    ID_MULTIPLE,
+    ID_ROW_LINES,
+    ID_HORIZ_RULES,
+    ID_VERT_RULES,
+
     ID_EXIT = wxID_EXIT,
 
+    // about menu
+    ID_ABOUT = wxID_ABOUT,
+
+
+    // control IDs
+
     ID_MUSIC_CTRL       = 50,
 
     ID_ADD_MOZART       = 100,
@@ -235,8 +253,12 @@ enum
 };
 
 BEGIN_EVENT_TABLE(MyFrame, wxFrame)
-    EVT_MENU( ID_ABOUT, MyFrame::OnAbout )
+    EVT_MENU_RANGE( ID_MULTIPLE, ID_VERT_RULES, MyFrame::OnStyleChange )
     EVT_MENU( ID_EXIT, MyFrame::OnQuit )
+    EVT_MENU( ID_ABOUT, MyFrame::OnAbout )
+
+    EVT_NOTEBOOK_PAGE_CHANGED( wxID_ANY, MyFrame::OnPageChanged )
+
     EVT_BUTTON( ID_ADD_MOZART, MyFrame::OnAddMozart )
     EVT_BUTTON( ID_DELETE_MUSIC, MyFrame::OnDeleteMusic )
     EVT_BUTTON( ID_DELETE_YEAR, MyFrame::OnDeleteYear )
@@ -277,19 +299,35 @@ MyFrame::MyFrame(wxFrame *frame, const wxString &title, int x, int y, int w, int
     m_log = NULL;
     m_col = NULL;
 
+    m_ctrl[0] = NULL;
+    m_ctrl[1] = NULL;
+    m_ctrl[2] = NULL;
+    m_ctrl[3] = NULL;
+
     SetIcon(wxICON(sample));
 
 
     // build the menus
     // ----------------
 
+    wxMenu *style_menu = new wxMenu;
+    //style_menu->AppendCheckItem(ID_SINGLE, wxT("Single selection"));
+    style_menu->AppendCheckItem(ID_MULTIPLE, wxT("Multiple selection"));
+    style_menu->AppendCheckItem(ID_ROW_LINES, wxT("Alternating colours"));
+    style_menu->AppendCheckItem(ID_HORIZ_RULES, wxT("Display horizontal rules"));
+    style_menu->AppendCheckItem(ID_VERT_RULES, wxT("Display vertical rules"));
+
     wxMenu *file_menu = new wxMenu;
-    file_menu->Append(ID_ABOUT, wxT("&About"));
+    file_menu->Append(ID_STYLE_MENU, wxT("&Style"), style_menu);
     file_menu->AppendSeparator();
     file_menu->Append(ID_EXIT, wxT("E&xit"));
 
+    wxMenu *about_menu = new wxMenu;
+    about_menu->Append(ID_ABOUT, wxT("&About"));
+
     wxMenuBar *menu_bar = new wxMenuBar;
     menu_bar->Append(file_menu, wxT("&File"));
+    menu_bar->Append(about_menu, wxT("&About"));
 
     SetMenuBar(menu_bar);
     CreateStatusBar();
@@ -301,75 +339,8 @@ MyFrame::MyFrame(wxFrame *frame, const wxString &title, int x, int y, int w, int
     m_notebook = new wxNotebook( this, wxID_ANY );
 
     wxPanel *firstPanel = new wxPanel( m_notebook, wxID_ANY );
-    m_myMusicModelViewCtrl = 
-        new wxDataViewCtrl( firstPanel, ID_MUSIC_CTRL, wxDefaultPosition,
-                            wxDefaultSize, wxDV_MULTIPLE|wxDV_VARIABLE_LINE_HEIGHT );
-
-    m_music_model = new MyMusicTreeModel;
-    m_myMusicModelViewCtrl->AssociateModel( m_music_model.get() );
-
-    m_myMusicModelViewCtrl->EnableDragSource( wxDF_UNICODETEXT );
-    m_myMusicModelViewCtrl->EnableDropTarget( wxDF_UNICODETEXT );
-
-    // column 0 of the view control:
-
-    wxDataViewTextRenderer *tr = 
-        new wxDataViewTextRenderer( wxT("string"), wxDATAVIEW_CELL_INERT );
-    wxDataViewColumn *column0 = 
-        new wxDataViewColumn( wxT("title"), tr, 0, 200, wxALIGN_LEFT,
-                              wxDATAVIEW_COL_SORTABLE | wxDATAVIEW_COL_RESIZABLE );
-    m_myMusicModelViewCtrl->AppendColumn( column0 );
-#if 0
-    // Call this and sorting is enabled
-    // immediatly upon start up.
-    column0->SetAsSortKey();
-#endif
-
-    // column 1 of the view control:
 
-    tr = new wxDataViewTextRenderer( wxT("string"), wxDATAVIEW_CELL_EDITABLE );
-    wxDataViewColumn *column1 = 
-        new wxDataViewColumn( wxT("artist"), tr, 1, 150, wxALIGN_LEFT,
-                              wxDATAVIEW_COL_SORTABLE | wxDATAVIEW_COL_REORDERABLE | 
-                              wxDATAVIEW_COL_RESIZABLE );
-    column1->SetMinWidth(150); // this column can't be resized to be smaller
-    m_myMusicModelViewCtrl->AppendColumn( column1 );
-
-    // column 2 of the view control:
-
-    wxDataViewSpinRenderer *sr = 
-        new wxDataViewSpinRenderer( 0, 2010, wxDATAVIEW_CELL_EDITABLE, wxALIGN_RIGHT );
-    wxDataViewColumn *column2 = 
-        new wxDataViewColumn( wxT("year"), sr, 2, 60, wxALIGN_LEFT,
-                              wxDATAVIEW_COL_SORTABLE | wxDATAVIEW_COL_REORDERABLE );
-    m_myMusicModelViewCtrl->AppendColumn( column2 );
-
-    // column 3 of the view control:
-
-    wxArrayString choices;
-    choices.Add( "good" );
-    choices.Add( "bad" );
-    choices.Add( "lousy" );
-    wxDataViewChoiceRenderer *c = 
-        new wxDataViewChoiceRenderer( choices, wxDATAVIEW_CELL_EDITABLE, wxALIGN_RIGHT );
-    wxDataViewColumn *column3 = 
-        new wxDataViewColumn( wxT("rating"), c, 3, 100, wxALIGN_LEFT,
-                              wxDATAVIEW_COL_REORDERABLE | wxDATAVIEW_COL_RESIZABLE );
-    m_myMusicModelViewCtrl->AppendColumn( column3 );
-
-    // column 4 of the view control:
-
-    m_myMusicModelViewCtrl->AppendProgressColumn( wxT("popularity"), 4, wxDATAVIEW_CELL_INERT, 80 );
-
-    // column 5 of the view control:
-
-    MyCustomRenderer *cr = new MyCustomRenderer( wxDATAVIEW_CELL_ACTIVATABLE, wxALIGN_RIGHT );
-    wxDataViewColumn *column5 = 
-        new wxDataViewColumn( wxT("custom"), cr, 5, -1, wxALIGN_LEFT,
-                              wxDATAVIEW_COL_RESIZABLE );
-    m_myMusicModelViewCtrl->AppendColumn( column5 );
-
-    // complete this page:
+    BuildDataViewCtrl(firstPanel, 0);    // sets m_ctrl[0]
 
     wxBoxSizer *button_sizer = new wxBoxSizer( wxHORIZONTAL );
     button_sizer->Add( new wxButton( firstPanel, ID_ADD_MOZART,  _("Add Mozart")),             0, wxALL, 10 );
@@ -378,8 +349,11 @@ MyFrame::MyFrame(wxFrame *frame, const wxString &title, int x, int y, int w, int
     button_sizer->Add( new wxButton( firstPanel, ID_SELECT_NINTH,_("Select Ninth")),           0, wxALL, 10 );
 
     wxSizer *firstPanelSz = new wxBoxSizer( wxVERTICAL );
-    m_myMusicModelViewCtrl->SetMinSize(wxSize(-1, 200));
-    firstPanelSz->Add(m_myMusicModelViewCtrl, 1, wxGROW|wxALL, 5);
+    m_ctrl[0]->SetMinSize(wxSize(-1, 200));
+    firstPanelSz->Add(m_ctrl[0], 1, wxGROW|wxALL, 5);
+    firstPanelSz->Add(
+        new wxStaticText(firstPanel, wxID_ANY, wxT("Most of the cells above are editable!")), 
+        0, wxGROW|wxALL, 5);
     firstPanelSz->Add(button_sizer);
     firstPanel->SetSizerAndFit(firstPanelSz);
 
@@ -389,24 +363,7 @@ MyFrame::MyFrame(wxFrame *frame, const wxString &title, int x, int y, int w, int
 
     wxPanel *secondPanel = new wxPanel( m_notebook, wxID_ANY );
 
-    m_myListModelViewCtrl = new wxDataViewCtrl( secondPanel, wxID_ANY, wxDefaultPosition,
-                                                wxDefaultSize, wxDV_MULTIPLE | wxDV_ROW_LINES);
-
-    m_list_model = new MyListModel;
-    m_myListModelViewCtrl->AssociateModel( m_list_model.get() );
-
-    // the various columns
-#if 1
-    m_myListModelViewCtrl->AppendTextColumn    (wxT("editable string"), 0, wxDATAVIEW_CELL_EDITABLE, 120 );
-    m_myListModelViewCtrl->AppendIconTextColumn(wxIcon(wx_small_xpm), 1, wxDATAVIEW_CELL_INERT )->SetTitle( wxT("icon") );
-#else
-    m_myListModelViewCtrl->AppendTextColumn    (wxT("editable string"), 0, wxDATAVIEW_CELL_EDITABLE );
-    m_myListModelViewCtrl->AppendIconTextColumn(wxT("icon"),            1, wxDATAVIEW_CELL_INERT );
-#endif
-    m_myListModelViewCtrl->AppendColumn(
-        new wxDataViewColumn(wxT("attributes"), new wxDataViewTextRendererAttr, 2 ));
-
-    // complete this page:
+    BuildDataViewCtrl(secondPanel, 1);    // sets m_ctrl[1]
 
     wxBoxSizer *button_sizer2 = new wxBoxSizer( wxHORIZONTAL );
     button_sizer2->Add( new wxButton( secondPanel, ID_PREPEND_LIST,_("Prepend")),         0, wxALL, 10 );
@@ -415,35 +372,20 @@ MyFrame::MyFrame(wxFrame *frame, const wxString &title, int x, int y, int w, int
     button_sizer2->Add( new wxButton( secondPanel, ID_ADD_MANY,    _("Add 1000")),        0, wxALL, 10 );
 
     wxSizer *secondPanelSz = new wxBoxSizer( wxVERTICAL );
-    secondPanelSz->Add(m_myListModelViewCtrl, 1, wxGROW|wxALL, 5);
+    secondPanelSz->Add(m_ctrl[1], 1, wxGROW|wxALL, 5);
     secondPanelSz->Add(button_sizer2);
     secondPanel->SetSizerAndFit(secondPanelSz);
 
 
-
     // third page of the notebook
     // ---------------------------
 
     wxPanel *thirdPanel = new wxPanel( m_notebook, wxID_ANY );
 
-    m_listctrl = new wxDataViewListCtrl( thirdPanel, wxID_ANY );
-    m_listctrl->AppendToggleColumn( wxT("Toggle") );
-    m_listctrl->AppendTextColumn( wxT("Text") );
-
-    wxVector<wxVariant> data;
-    data.push_back( true );
-    data.push_back( "row 1" );
-    m_listctrl->AppendItem( data );
-
-    data.clear();
-    data.push_back( false );
-    data.push_back( "row 3" );
-    m_listctrl->AppendItem( data );
-
-    // complete this page:
+    BuildDataViewCtrl(thirdPanel, 2);    // sets m_ctrl[2]
 
     wxSizer *thirdPanelSz = new wxBoxSizer( wxVERTICAL );
-    thirdPanelSz->Add(m_listctrl, 1, wxGROW|wxALL, 5);
+    thirdPanelSz->Add(m_ctrl[2], 1, wxGROW|wxALL, 5);
     thirdPanel->SetSizerAndFit(thirdPanelSz);
 
 
@@ -452,21 +394,10 @@ MyFrame::MyFrame(wxFrame *frame, const wxString &title, int x, int y, int w, int
     
     wxPanel *fourthPanel = new wxPanel( m_notebook, wxID_ANY );
 
-    m_treectrl = new wxDataViewTreeCtrl( fourthPanel, wxID_ANY );
-
-    wxImageList *ilist = new wxImageList( 16, 16 );
-    ilist->Add( wxIcon(wx_small_xpm) );
-    m_treectrl->SetImageList( ilist );
-
-    wxDataViewItem parent2 = m_treectrl->AppendContainer( wxDataViewItem(0),wxT("Root 1"), 0 );
-    m_treectrl->AppendItem( parent2, wxT("Child 1"), 0 );
-    m_treectrl->AppendItem( parent2, wxT("Child 2"), 0 );
-    m_treectrl->AppendItem( parent2, wxT("Child 3, very long, long, long, long"), 0 );
-
-    // complete this page:
+    BuildDataViewCtrl(fourthPanel, 3);    // sets m_ctrl[3]
 
     wxSizer *fourthPanelSz = new wxBoxSizer( wxVERTICAL );
-    fourthPanelSz->Add(m_treectrl, 1, wxGROW|wxALL, 5);
+    fourthPanelSz->Add(m_ctrl[3], 1, wxGROW|wxALL, 5);
     fourthPanel->SetSizerAndFit(fourthPanelSz);
 
 
@@ -498,20 +429,322 @@ MyFrame::~MyFrame()
     delete wxLog::SetActiveTarget(m_logOld);
 }
 
-void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event) )
+void MyFrame::BuildDataViewCtrl(wxPanel* parent, unsigned int nPanel, unsigned long style)
+{
+    switch (nPanel)
+    {
+    case 0:
+        {
+            wxASSERT(!m_ctrl[0] && !m_music_model);
+            m_ctrl[0] = 
+                new wxDataViewCtrl( parent, ID_MUSIC_CTRL, wxDefaultPosition,
+                                    wxDefaultSize, style );
+
+            m_music_model = new MyMusicTreeModel;
+            m_ctrl[0]->AssociateModel( m_music_model.get() );
+
+            m_ctrl[0]->EnableDragSource( wxDF_UNICODETEXT );
+            m_ctrl[0]->EnableDropTarget( wxDF_UNICODETEXT );
+
+            // column 0 of the view control:
+
+            wxDataViewTextRenderer *tr = 
+                new wxDataViewTextRenderer( wxT("string"), wxDATAVIEW_CELL_INERT );
+            wxDataViewColumn *column0 = 
+                new wxDataViewColumn( wxT("title"), tr, 0, 200, wxALIGN_LEFT,
+                                      wxDATAVIEW_COL_SORTABLE | wxDATAVIEW_COL_RESIZABLE );
+            m_ctrl[0]->AppendColumn( column0 );
+#if 0
+            // Call this and sorting is enabled
+            // immediatly upon start up.
+            column0->SetAsSortKey();
+#endif
+
+            // column 1 of the view control:
+
+            tr = new wxDataViewTextRenderer( wxT("string"), wxDATAVIEW_CELL_EDITABLE );
+            wxDataViewColumn *column1 = 
+                new wxDataViewColumn( wxT("artist"), tr, 1, 150, wxALIGN_LEFT,
+                                      wxDATAVIEW_COL_SORTABLE | wxDATAVIEW_COL_REORDERABLE | 
+                                      wxDATAVIEW_COL_RESIZABLE );
+            column1->SetMinWidth(150); // this column can't be resized to be smaller
+            m_ctrl[0]->AppendColumn( column1 );
+
+            // column 2 of the view control:
+
+            wxDataViewSpinRenderer *sr = 
+                new wxDataViewSpinRenderer( 0, 2010, wxDATAVIEW_CELL_EDITABLE, wxALIGN_RIGHT );
+            wxDataViewColumn *column2 = 
+                new wxDataViewColumn( wxT("year"), sr, 2, 60, wxALIGN_LEFT,
+                                      wxDATAVIEW_COL_SORTABLE | wxDATAVIEW_COL_REORDERABLE );
+            m_ctrl[0]->AppendColumn( column2 );
+
+            // column 3 of the view control:
+
+            wxArrayString choices;
+            choices.Add( "good" );
+            choices.Add( "bad" );
+            choices.Add( "lousy" );
+            wxDataViewChoiceRenderer *c = 
+                new wxDataViewChoiceRenderer( choices, wxDATAVIEW_CELL_EDITABLE, wxALIGN_RIGHT );
+            wxDataViewColumn *column3 = 
+                new wxDataViewColumn( wxT("rating"), c, 3, 100, wxALIGN_LEFT,
+                                      wxDATAVIEW_COL_REORDERABLE | wxDATAVIEW_COL_RESIZABLE );
+            m_ctrl[0]->AppendColumn( column3 );
+
+            // column 4 of the view control:
+
+            m_ctrl[0]->AppendProgressColumn( wxT("popularity"), 4, wxDATAVIEW_CELL_INERT, 80 );
+
+            // column 5 of the view control:
+
+            MyCustomRenderer *cr = new MyCustomRenderer( wxDATAVIEW_CELL_ACTIVATABLE, wxALIGN_RIGHT );
+            wxDataViewColumn *column5 = 
+                new wxDataViewColumn( wxT("custom"), cr, 5, -1, wxALIGN_LEFT,
+                                      wxDATAVIEW_COL_RESIZABLE );
+            m_ctrl[0]->AppendColumn( column5 );
+            
+
+            // select initially the ninth symphony:
+            m_ctrl[0]->Select(m_music_model->GetNinthItem());
+        }
+        break;
+
+    case 1:
+        {
+            wxASSERT(!m_ctrl[1] && !m_list_model);
+            m_ctrl[1] = new wxDataViewCtrl( parent, wxID_ANY, wxDefaultPosition,
+                                            wxDefaultSize, style );
+
+            m_list_model = new MyListModel;
+            m_ctrl[1]->AssociateModel( m_list_model.get() );
+
+            // the various columns
+#if 1
+            m_ctrl[1]->AppendTextColumn    (wxT("editable string"), 0, wxDATAVIEW_CELL_EDITABLE, 120 );
+            m_ctrl[1]->AppendIconTextColumn(wxIcon(wx_small_xpm),   1, wxDATAVIEW_CELL_INERT )->SetTitle( wxT("icon") );
+#else
+            m_ctrl[1]->AppendTextColumn    (wxT("editable string"), 0, wxDATAVIEW_CELL_EDITABLE );
+            m_ctrl[1]->AppendIconTextColumn(wxT("icon"),            1, wxDATAVIEW_CELL_INERT );
+#endif
+            m_ctrl[1]->AppendColumn(
+                new wxDataViewColumn(wxT("attributes"), new wxDataViewTextRendererAttr, 2 ));
+        }
+        break;
+
+    case 2:
+        {
+            wxASSERT(!m_ctrl[2]);
+            wxDataViewListCtrl* lc = 
+                new wxDataViewListCtrl( parent, wxID_ANY, wxDefaultPosition,
+                                        wxDefaultSize, style );
+            m_ctrl[2] = lc;
+
+            lc->AppendToggleColumn( wxT("Toggle") );
+            lc->AppendTextColumn( wxT("Text") );
+            lc->AppendProgressColumn( wxT("Progress") );
+
+            wxVector<wxVariant> data;
+            for (unsigned int i=0; i<10; i++)
+            {
+                data.clear();
+                data.push_back( (i%3) == 0 );
+                data.push_back( wxString::Format("row %d", i) );
+                data.push_back( long(5*i) );
+
+                lc->AppendItem( data );
+            }
+        }
+        break;
+
+    case 3:
+        {
+            wxASSERT(!m_ctrl[3]);
+            wxDataViewTreeCtrl* tc = 
+                new wxDataViewTreeCtrl( parent, wxID_ANY, wxDefaultPosition,
+                                        wxDefaultSize, style );
+            m_ctrl[3] = tc;
+
+            wxImageList *ilist = new wxImageList( 16, 16 );
+            ilist->Add( wxIcon(wx_small_xpm) );
+            tc->SetImageList( ilist );
+
+            wxDataViewItem parent = 
+                tc->AppendContainer( wxDataViewItem(0), wxT("The Root"), 0 );
+            tc->AppendItem( parent, wxT("Child 1"), 0 );
+            tc->AppendItem( parent, wxT("Child 2"), 0 );
+            tc->AppendItem( parent, wxT("Child 3, very long, long, long, long"), 0 );
+
+            wxDataViewItem cont =
+                tc->AppendContainer( parent, wxT("Container child"), 0 );
+            tc->AppendItem( cont, wxT("Child 4"), 0 );
+            tc->AppendItem( cont, wxT("Child 5"), 0 );
+
+            tc->Expand(cont);
+        } 
+        break;
+    }
+}
+
+
+// ----------------------------------------------------------------------------
+// MyFrame - generic event handlers
+// ----------------------------------------------------------------------------
+
+void MyFrame::OnPageChanged(wxBookCtrlEvent& WXUNUSED(event) )
+{
+    unsigned int nPanel = m_notebook->GetSelection();
+
+    GetMenuBar()->FindItem(ID_STYLE_MENU)->SetItemLabel(
+                wxString::Format("Style of panel #%d", nPanel+1));
+
+    for (unsigned int id = ID_MULTIPLE; id <= ID_VERT_RULES; id++)
+    {
+        unsigned long style = 0;
+        switch (id)
+        {
+        /*case ID_SINGLE:
+            style = wxDV_SINGLE;
+            break;*/
+        case ID_MULTIPLE:
+            style = wxDV_MULTIPLE;
+            break;
+        case ID_ROW_LINES:
+            style = wxDV_ROW_LINES;
+            break;
+        case ID_HORIZ_RULES:
+            style = wxDV_HORIZ_RULES;
+            break;
+        case ID_VERT_RULES:
+            style = wxDV_VERT_RULES;
+            break;
+        default:
+            wxFAIL;
+        }
+
+        GetMenuBar()->FindItem(id)->Check( m_ctrl[nPanel]->HasFlag(style) );
+    }
+}
+
+void MyFrame::OnStyleChange(wxCommandEvent& WXUNUSED(event) )
+{
+    unsigned int nPanel = m_notebook->GetSelection();
+
+    // build the style
+    unsigned long style = 0;
+    /*if (GetMenuBar()->FindItem(ID_SINGLE)->IsChecked())
+        style |= wxDV_SINGLE;*/
+    if (GetMenuBar()->FindItem(ID_MULTIPLE)->IsChecked())
+        style |= wxDV_MULTIPLE;
+    if (GetMenuBar()->FindItem(ID_ROW_LINES)->IsChecked())
+        style |= wxDV_ROW_LINES;
+    if (GetMenuBar()->FindItem(ID_HORIZ_RULES)->IsChecked())
+        style |= wxDV_HORIZ_RULES;
+    if (GetMenuBar()->FindItem(ID_VERT_RULES)->IsChecked())
+        style |= wxDV_VERT_RULES;
+
+    wxSizer* sz = m_ctrl[nPanel]->GetContainingSizer();
+    wxASSERT(sz);
+
+    sz->Detach(m_ctrl[nPanel]);
+    wxDELETE(m_ctrl[nPanel]);
+    m_ctrl[nPanel] = NULL;
+
+    if (nPanel == 0)
+        m_music_model.reset(NULL);
+    else if (nPanel == 1)
+        m_list_model.reset(NULL);
+
+    // rebuild the DVC for the selected panel:
+    BuildDataViewCtrl((wxPanel*)m_notebook->GetPage(nPanel), nPanel, style);
+
+    sz->Prepend(m_ctrl[nPanel], 1, wxGROW|wxALL, 5);
+    sz->Layout();
+}
+
+void MyFrame::OnQuit( wxCommandEvent& WXUNUSED(event) )
 {
     Close(true);
 }
 
-void MyFrame::OnAddMozart(wxCommandEvent& WXUNUSED(event) )
+void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event) )
+{
+    wxAboutDialogInfo info;
+    info.SetName(_("DataView sample"));
+    info.SetDescription(_("This sample demonstrates wxDataViewCtrl"));
+    info.SetCopyright(_T("(C) 2007-2009 Robert Roebling"));
+    info.AddDeveloper("Robert Roebling");
+    info.AddDeveloper("Francesco Montorsi");
+
+    wxAboutBox(info);
+}
+
+
+// ----------------------------------------------------------------------------
+// MyFrame - event handlers for the first page
+// ----------------------------------------------------------------------------
+
+void MyFrame::OnBeginDrag( wxDataViewEvent &event )
+{
+    wxDataViewItem item( event.GetItem() );
+
+    // only allow drags for item, not containers
+    if (m_music_model->IsContainer( item ) )
+    {
+        event.Veto();
+        return;
+    }
+
+    MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();
+    wxTextDataObject *obj = new wxTextDataObject;
+    obj->SetText( node->m_title );
+    event.SetDataObject( obj );
+}
+
+void MyFrame::OnDropPossible( wxDataViewEvent &event )
+{
+    wxDataViewItem item( event.GetItem() );
+
+    // only allow drags for item, not containers
+    if (m_music_model->IsContainer( item ) )
+        event.Veto();
+
+    if (event.GetDataFormat() != wxDF_UNICODETEXT)
+        event.Veto();
+}
+
+void MyFrame::OnDrop( wxDataViewEvent &event )
+{
+    wxDataViewItem item( event.GetItem() );
+
+    // only allow drops for item, not containers
+    if (m_music_model->IsContainer( item ) )
+    {
+        event.Veto();
+        return;
+    }
+
+    if (event.GetDataFormat() != wxDF_UNICODETEXT)
+    {
+        event.Veto();
+        return;
+    }
+
+    wxTextDataObject obj;
+    obj.SetData( wxDF_UNICODETEXT, event.GetDataSize(), event.GetDataBuffer() );
+
+    wxLogMessage(wxT("Text dropped: %s"), obj.GetText() );
+}
+
+void MyFrame::OnAddMozart( wxCommandEvent& WXUNUSED(event) )
 {
     m_music_model->AddToClassical( wxT("Kleine Nachtmusik"), wxT("Wolfgang Mozart"), 1787 );
 }
 
-void MyFrame::OnDeleteMusic(wxCommandEvent& WXUNUSED(event) )
+void MyFrame::OnDeleteMusic( wxCommandEvent& WXUNUSED(event) )
 {
     wxDataViewItemArray items;
-    int len = m_myMusicModelViewCtrl->GetSelections( items );
+    int len = m_ctrl[0]->GetSelections( items );
     for( int i = 0; i < len; i ++ )
         if (items[i].IsOk())
             m_music_model->Delete( items[i] );
@@ -519,26 +752,13 @@ void MyFrame::OnDeleteMusic(wxCommandEvent& WXUNUSED(event) )
 
 void MyFrame::OnDeleteYear( wxCommandEvent& WXUNUSED(event) )
 {
-    m_myMusicModelViewCtrl->DeleteColumn( m_myMusicModelViewCtrl->GetColumn( 2 ) );
+    m_ctrl[0]->DeleteColumn( m_ctrl[0]->GetColumn( 2 ) );
     FindWindow( ID_DELETE_YEAR )->Disable();
 }
 
 void MyFrame::OnSelectNinth( wxCommandEvent& WXUNUSED(event) )
 {
-    m_myMusicModelViewCtrl->Select( m_music_model->GetNinthItem() );
-}
-
-void MyFrame::OnPrependList( wxCommandEvent& WXUNUSED(event) )
-{
-    m_list_model->Prepend(wxT("Test"));
-}
-
-void MyFrame::OnDeleteList( wxCommandEvent& WXUNUSED(event) )
-{
-    wxDataViewItemArray items;
-    int len = m_myListModelViewCtrl->GetSelections( items );
-    if (len > 0)
-        m_list_model->DeleteItems( items );
+    m_ctrl[0]->Select( m_music_model->GetNinthItem() );
 }
 
 void MyFrame::OnValueChanged( wxDataViewEvent &event )
@@ -558,7 +778,7 @@ void MyFrame::OnActivated( wxDataViewEvent &event )
     wxString title = m_music_model->GetTitle( event.GetItem() );
     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_ITEM_ACTIVATED, Item: %s"), title );
 
-    if (m_myMusicModelViewCtrl->IsExpanded( event.GetItem() ))
+    if (m_ctrl[0]->IsExpanded( event.GetItem() ))
         wxLogMessage(wxT("Item: %s is expanded"), title );
 }
 
@@ -642,10 +862,10 @@ void MyFrame::OnContextMenu( wxDataViewEvent &event )
     menu.Append( 2, wxT("entry 2") );
     menu.Append( 3, wxT("entry 3") );
 
-    m_myMusicModelViewCtrl->PopupMenu(&menu);
+    m_ctrl[0]->PopupMenu(&menu);
 
-//    wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_ITEM_CONTEXT_MENU, Item: %s Value: %s"),
-//                 title.GetData(), event.GetValue().GetString());
+    wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_ITEM_CONTEXT_MENU, Item: %s Value: %s"),
+                 title.GetData(), event.GetValue().GetString());
 }
 
 void MyFrame::OnHeaderClick( wxDataViewEvent &event )
@@ -654,10 +874,10 @@ void MyFrame::OnHeaderClick( wxDataViewEvent &event )
     // this column when it is clicked to take place
     event.Skip();
 
-    if(!m_log)
+    if (!m_log)
         return;
 
-    int pos = m_myMusicModelViewCtrl->GetColumnPosition( event.GetDataViewColumn() );
+    int pos = m_ctrl[0]->GetColumnPosition( event.GetDataViewColumn() );
 
     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_CLICK, Column position: %d"), pos );
 }
@@ -667,7 +887,7 @@ void MyFrame::OnHeaderRightClick( wxDataViewEvent &event )
     if(!m_log)
         return;
 
-    int pos = m_myMusicModelViewCtrl->GetColumnPosition( event.GetDataViewColumn() );
+    int pos = m_ctrl[0]->GetColumnPosition( event.GetDataViewColumn() );
 
     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK, Column position: %d"), pos );
 }
@@ -677,7 +897,7 @@ void MyFrame::OnSorted( wxDataViewEvent &event )
     if(!m_log)
         return;
 
-    int pos = m_myMusicModelViewCtrl->GetColumnPosition( event.GetDataViewColumn() );
+    int pos = m_ctrl[0]->GetColumnPosition( event.GetDataViewColumn() );
 
     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_COLUMN_SORTED, Column position: %d"), pos );
 }
@@ -691,79 +911,32 @@ void MyFrame::OnRightClick( wxMouseEvent &event )
                  event.GetX(), event.GetY());
 }
 
-void MyFrame::OnGoto(wxCommandEvent& WXUNUSED(event))
-{
-    wxDataViewItem item = m_list_model->GetItem( 50 );
-    m_myListModelViewCtrl->EnsureVisible(item,m_col);
-}
-
-void MyFrame::OnAddMany(wxCommandEvent& WXUNUSED(event))
-{
-    m_list_model->AddMany();
-}
 
+// ----------------------------------------------------------------------------
+// MyFrame - event handlers for the second page
+// ----------------------------------------------------------------------------
 
-void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event) )
+void MyFrame::OnPrependList( wxCommandEvent& WXUNUSED(event) )
 {
-    wxAboutDialogInfo info;
-    info.SetName(_("DataView sample"));
-    info.SetDescription(_("This sample demonstrates wxDataViewCtrl"));
-    info.SetCopyright(_T("(C) 2007-2009 Robert Roebling"));
-    info.AddDeveloper("Robert Roebling");
-    info.AddDeveloper("Francesco Montorsi");
-
-    wxAboutBox(info);
+    m_list_model->Prepend(wxT("Test"));
 }
 
-void MyFrame::OnBeginDrag( wxDataViewEvent &event )
+void MyFrame::OnDeleteList( wxCommandEvent& WXUNUSED(event) )
 {
-    wxDataViewItem item( event.GetItem() );
-
-    // only allow drags for item, not containers
-    if (m_music_model->IsContainer( item ) )
-    {
-        event.Veto();
-        return;
-    }
-
-    MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();
-    wxTextDataObject *obj = new wxTextDataObject;
-    obj->SetText( node->m_title );
-    event.SetDataObject( obj );
+    wxDataViewItemArray items;
+    int len = m_ctrl[1]->GetSelections( items );
+    if (len > 0)
+        m_list_model->DeleteItems( items );
 }
 
-void MyFrame::OnDropPossible( wxDataViewEvent &event )
+void MyFrame::OnGoto(wxCommandEvent& WXUNUSED(event))
 {
-    wxDataViewItem item( event.GetItem() );
-
-    // only allow drags for item, not containers
-    if (m_music_model->IsContainer( item ) )
-        event.Veto();
-
-    if (event.GetDataFormat() != wxDF_UNICODETEXT)
-        event.Veto();
+    wxDataViewItem item = m_list_model->GetItem( 50 );
+    m_ctrl[1]->EnsureVisible(item,m_col);
 }
 
-void MyFrame::OnDrop( wxDataViewEvent &event )
+void MyFrame::OnAddMany(wxCommandEvent& WXUNUSED(event))
 {
-    wxDataViewItem item( event.GetItem() );
-
-    // only allow drops for item, not containers
-    if (m_music_model->IsContainer( item ) )
-    {
-        event.Veto();
-        return;
-    }
-
-    if (event.GetDataFormat() != wxDF_UNICODETEXT)
-    {
-        event.Veto();
-        return;
-    }
-
-    wxTextDataObject obj;
-    obj.SetData( wxDF_TEXT, event.GetDataSize(), event.GetDataBuffer() );
-
-    wxLogMessage(wxT("Text dropped: %s"), obj.GetText() );
+    m_list_model->AddMany();
 }
 
index 75d687bdf1de24ae868de3b445df853f2b11370e..282e4e8a8736817f8089932eabfdeadaca489fd8 100644 (file)
@@ -29,6 +29,7 @@
 // ----------------------------------------------------------------------------\r
 \r
 #include "null.xpm"\r
+#include "wx_small.xpm"\r
 \r
 \r
 // ----------------------------------------------------------------------------\r
 MyMusicTreeModel::MyMusicTreeModel()\r
 {\r
     m_root = new MyMusicTreeModelNode( NULL, wxT("My Music" ));\r
+\r
+    // setup pop music\r
     m_pop = new MyMusicTreeModelNode( m_root, wxT("Pop music") );\r
+    m_pop->Append( \r
+        new MyMusicTreeModelNode( m_pop, wxT("You are not alone"), wxT("Michael Jackson"), 1995 ) );\r
+    m_pop->Append( \r
+        new MyMusicTreeModelNode( m_pop, wxT("Take a bow"), wxT("Madonna"), 1994 ) );\r
     m_root->Append( m_pop );\r
-    m_pop->Append( new MyMusicTreeModelNode( m_pop,\r
-        wxT("You are not alone"), wxT("Michael Jackson"), 1995 ) );\r
-    m_pop->Append( new MyMusicTreeModelNode( m_pop,\r
-        wxT("Take a bow"), wxT("Madonna"), 1994 ) );\r
+\r
+    // setup classical music\r
     m_classical = new MyMusicTreeModelNode( m_root, wxT("Classical music") );\r
-    m_root->Append( m_classical );\r
-    m_ninth = new MyMusicTreeModelNode( m_classical,\r
-        wxT("Ninth symphony"), wxT("Ludwig van Beethoven"), 1824 );\r
+    m_ninth = new MyMusicTreeModelNode( m_classical, wxT("Ninth symphony"), \r
+                                        wxT("Ludwig van Beethoven"), 1824 );\r
     m_classical->Append( m_ninth );\r
-    m_classical->Append( new MyMusicTreeModelNode( m_classical,\r
-        wxT("German Requiem"), wxT("Johannes Brahms"), 1868 ) );\r
+    m_classical->Append( new MyMusicTreeModelNode( m_classical, wxT("German Requiem"), \r
+                                                   wxT("Johannes Brahms"), 1868 ) );\r
+    m_root->Append( m_classical );\r
+\r
     m_classicalMusicIsKnownToControl = false;\r
 }\r
 \r
 wxString MyMusicTreeModel::GetTitle( const wxDataViewItem &item ) const\r
 {\r
     MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();\r
-    if (!node)\r
+    if (!node)      // happens if item.IsOk()==false\r
         return wxEmptyString;\r
 \r
     return node->m_title;\r
@@ -66,18 +72,27 @@ wxString MyMusicTreeModel::GetTitle( const wxDataViewItem &item ) const
 int MyMusicTreeModel::GetYear( const wxDataViewItem &item ) const\r
 {\r
     MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();\r
-    if (!node)\r
+    if (!node)      // happens if item.IsOk()==false\r
         return 2000;\r
 \r
     return node->m_year;\r
 }\r
 \r
-void MyMusicTreeModel::AddToClassical( const wxString &title, const wxString &artist, int year )\r
+void MyMusicTreeModel::AddToClassical( const wxString &title, const wxString &artist, \r
+                                       unsigned int year )\r
 {\r
-    // add to data\r
+    if (!m_classical)\r
+    {\r
+        wxASSERT(m_root);\r
+\r
+        // it was removed: restore it\r
+        m_classical = new MyMusicTreeModelNode( m_root, wxT("Classical music") );\r
+        m_root->Append( m_classical );\r
+    }\r
+\r
+    // add to the classical music node a new node:\r
     MyMusicTreeModelNode *child_node =\r
         new MyMusicTreeModelNode( m_classical, title, artist, year );\r
-\r
     m_classical->Append( child_node );\r
 \r
     if (m_classicalMusicIsKnownToControl)\r
@@ -92,9 +107,33 @@ void MyMusicTreeModel::AddToClassical( const wxString &title, const wxString &ar
 void MyMusicTreeModel::Delete( const wxDataViewItem &item )\r
 {\r
     MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();\r
+    if (!node)      // happens if item.IsOk()==false\r
+        return;\r
+\r
     wxDataViewItem parent( node->GetParent() );\r
+    if (!parent.IsOk())\r
+    {\r
+        wxASSERT(node == m_root);\r
 \r
+        // don't make the control completely empty: \r
+        wxLogError("Cannot remove the root item!");\r
+        return;\r
+    }\r
+\r
+    // is the node one of those we keep stored in special pointers?\r
+    if (node == m_pop)\r
+        m_pop = NULL;\r
+    else if (node == m_classical)\r
+        m_classical = NULL;\r
+    else if (node == m_ninth)\r
+        m_ninth = NULL;\r
+\r
+    // first remove the node from the parent's array of children;\r
+    // NOTE: MyMusicTreeModelNodePtrArray is only an array of _pointers_\r
+    //       thus removing the node from it doesn't result in freeing it\r
     node->GetParent()->GetChildren().Remove( node );\r
+\r
+    // free the node\r
     delete node;\r
 \r
     // notify control\r
@@ -104,9 +143,12 @@ void MyMusicTreeModel::Delete( const wxDataViewItem &item )
 int MyMusicTreeModel::Compare( const wxDataViewItem &item1, const wxDataViewItem &item2,\r
                                unsigned int column, bool ascending )\r
 {\r
+    wxASSERT(item1.IsOk() && item2.IsOk());\r
+        // should never happen\r
+\r
     if (IsContainer(item1) && IsContainer(item2))\r
     {\r
-        wxVariant value1,value2;\r
+        wxVariant value1, value2;\r
         GetValue( value1, item1, 0 );\r
         GetValue( value2, item2, 0 );\r
 \r
@@ -128,60 +170,73 @@ int MyMusicTreeModel::Compare( const wxDataViewItem &item1, const wxDataViewItem
 void MyMusicTreeModel::GetValue( wxVariant &variant,\r
                                  const wxDataViewItem &item, unsigned int col ) const\r
 {\r
+    wxASSERT(item.IsOk());\r
+\r
     MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();\r
     switch (col)\r
     {\r
-        case 0: variant = node->m_title; break;\r
-        case 1: variant = node->m_artist; break;\r
-        case 2: variant = (long) node->m_year; break;\r
-        case 3: variant = node->m_quality; break;\r
-        case 4:\r
-            // wxMac doesn't conceal the popularity progress renderer, return 0 for containers\r
-            if (IsContainer(item))\r
-                variant = (long) 0;\r
-            else\r
-                variant = (long) 80;  // all music is very 80% popular\r
-            break;\r
-        case 5:\r
-            // Make size of red square depend on year\r
-            if (GetYear(item) < 1900)\r
-                variant = (long) 35;\r
-            else\r
-                variant = (long) 25;\r
-            break;\r
-        default:\r
-        {\r
-            wxLogError( wxT("MyMusicTreeModel::GetValue: wrong column %d"), col );\r
-\r
-            // provoke a crash when mouse button down\r
-            wxMouseState state = wxGetMouseState();\r
-            if (state.ShiftDown())\r
-            {\r
-                char *crash = 0;\r
-                *crash = 0;\r
-            }\r
-        }\r
+    case 0:\r
+        variant = node->m_title; \r
+        break;\r
+    case 1: \r
+        variant = node->m_artist; \r
+        break;\r
+    case 2: \r
+        variant = (long) node->m_year; \r
+        break;\r
+    case 3: \r
+        variant = node->m_quality; \r
+        break;\r
+    case 4:\r
+        // wxMac doesn't conceal the popularity progress renderer, return 0 for containers\r
+        if (IsContainer(item))\r
+            variant = (long) 0;\r
+        else\r
+            variant = (long) 80;  // all music is very 80% popular\r
+        break;\r
+    case 5:\r
+        // Make size of red square depend on year\r
+        if (GetYear(item) < 1900)\r
+            variant = (long) 35;\r
+        else\r
+            variant = (long) 25;\r
+        break;\r
+\r
+    default:\r
+        wxLogError( wxT("MyMusicTreeModel::GetValue: wrong column %d"), col );\r
     }\r
 }\r
 \r
 bool MyMusicTreeModel::SetValue( const wxVariant &variant,\r
                                  const wxDataViewItem &item, unsigned int col )\r
 {\r
+    wxASSERT(item.IsOk());\r
+\r
     MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();\r
     switch (col)\r
     {\r
-        case 0: node->m_title = variant.GetString(); return true;\r
-        case 1: node->m_artist  = variant.GetString(); return true;\r
-        case 2: node->m_year  = variant.GetLong(); return true;\r
-        case 3: node->m_quality  = variant.GetString(); return true;\r
-        default: wxLogError( wxT("MyMusicTreeModel::SetValue: wrong column") );\r
+        case 0: \r
+            node->m_title = variant.GetString(); \r
+            return true;\r
+        case 1: \r
+            node->m_artist = variant.GetString(); \r
+            return true;\r
+        case 2: \r
+            node->m_year = variant.GetLong(); \r
+            return true;\r
+        case 3: \r
+            node->m_quality = variant.GetString(); \r
+            return true;\r
+\r
+        default: \r
+            wxLogError( wxT("MyMusicTreeModel::SetValue: wrong column") );\r
     }\r
     return false;\r
 }\r
 \r
 wxDataViewItem MyMusicTreeModel::GetParent( const wxDataViewItem &item ) const\r
 {\r
-    // the invisble root node has no parent\r
+    // the invisible root node has no parent\r
     if (!item.IsOk())\r
         return wxDataViewItem(0);\r
 \r
@@ -196,8 +251,8 @@ wxDataViewItem MyMusicTreeModel::GetParent( const wxDataViewItem &item ) const
 \r
 bool MyMusicTreeModel::IsContainer( const wxDataViewItem &item ) const\r
 {\r
-    // the invisble root node can have children (in\r
-    // our model always "MyMusic")\r
+    // the invisble root node can have children\r
+    // (in our model always "MyMusic")\r
     if (!item.IsOk())\r
         return true;\r
 \r
@@ -205,7 +260,8 @@ bool MyMusicTreeModel::IsContainer( const wxDataViewItem &item ) const
     return node->IsContainer();\r
 }\r
 \r
-unsigned int MyMusicTreeModel::GetChildren( const wxDataViewItem &parent, wxDataViewItemArray &array ) const\r
+unsigned int MyMusicTreeModel::GetChildren( const wxDataViewItem &parent, \r
+                                            wxDataViewItemArray &array ) const\r
 {\r
     MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) parent.GetID();\r
     if (!node)\r
@@ -226,12 +282,12 @@ unsigned int MyMusicTreeModel::GetChildren( const wxDataViewItem &parent, wxData
     }\r
 \r
     unsigned int count = node->GetChildren().GetCount();\r
-    unsigned int pos;\r
-    for (pos = 0; pos < count; pos++)\r
+    for (unsigned int pos = 0; pos < count; pos++)\r
     {\r
         MyMusicTreeModelNode *child = node->GetChildren().Item( pos );\r
         array.Add( wxDataViewItem( (void*) child ) );\r
     }\r
+\r
     return count;\r
 }\r
 \r
@@ -251,28 +307,28 @@ static int my_sort( int *v1, int *v2 )
    return *v1-*v2;\r
 }\r
 \r
-MyListModel::MyListModel() :\r
 #ifdef __WXMAC__\r
-        wxDataViewVirtualListModel( 1000 + 100 )\r
+    #define INITIAL_NUMBER_OF_ITEMS 100\r
 #else\r
-        wxDataViewVirtualListModel( 100000 + 100 )\r
+    #define INITIAL_NUMBER_OF_ITEMS 100000\r
 #endif\r
+\r
+MyListModel::MyListModel() :\r
+        wxDataViewVirtualListModel( INITIAL_NUMBER_OF_ITEMS )\r
 {\r
-#ifdef __WXMAC__\r
-    m_virtualItems = 1000;\r
-#else\r
-    m_virtualItems = 100000;\r
-#endif\r
+    m_virtualItems = INITIAL_NUMBER_OF_ITEMS;\r
 \r
-    unsigned int i;\r
-    for (i = 0; i < 100; i++)\r
+    // the first 100 items are really stored in this model;\r
+    // all the others are synthetized on request \r
+    for (unsigned int i = 0; i < 100; i++)\r
     {\r
         wxString str;\r
-        str.Printf( wxT("row number %d"), i );\r
+        str.Printf( wxT("real row %d"), i );\r
         m_array.Add( str );\r
     }\r
 \r
-    m_icon = wxIcon( null_xpm );\r
+    m_icon[0] = wxIcon( null_xpm );\r
+    m_icon[1] = wxIcon( wx_small_xpm );\r
 }\r
 \r
 void MyListModel::Prepend( const wxString &text )\r
@@ -294,14 +350,21 @@ void MyListModel::DeleteItem( const wxDataViewItem &item )
 void MyListModel::DeleteItems( const wxDataViewItemArray &items )\r
 {\r
     wxArrayInt rows;\r
-    unsigned int i;\r
-    for (i = 0; i < items.GetCount(); i++)\r
+    for (unsigned int i = 0; i < items.GetCount(); i++)\r
     {\r
         unsigned int row = GetRow( items[i] );\r
         if (row < m_array.GetCount())\r
             rows.Add( row );\r
     }\r
 \r
+    if (rows.GetCount() == 0)\r
+    {\r
+        // none of the selected items were in the range of the items\r
+        // which we store... for simplicity, don't allow removing them\r
+        wxLogError("Cannot remove rows with an index greater than %d", m_array.GetCount());\r
+        return;\r
+    }\r
+\r
     // Sort in descending order so that the last\r
     // row will be deleted first. Otherwise the\r
     // remaining indeces would all be wrong.\r
@@ -328,39 +391,35 @@ void MyListModel::GetValueByRow( wxVariant &variant,
     if (col==0)\r
     {\r
         if (row >= m_array.GetCount())\r
-        {\r
-            wxString str;\r
-            str.Printf(wxT("row %d"), row - m_array.GetCount() );\r
-            variant = str;\r
-        }\r
+            variant = wxString::Format( wxT("virtual row %d"), row );\r
         else\r
-        {\r
             variant = m_array[ row ];\r
-        }\r
-    } else\r
-    if (col==1)\r
+    } \r
+    else if (col==1)\r
     {\r
-        wxDataViewIconText data( wxT("test"), m_icon );\r
+        wxDataViewIconText data( wxT("test"), m_icon[ row%2 ] );\r
         variant << data;\r
-    } else\r
-    if (col==2)\r
+    } \r
+    else if (col==2)\r
     {\r
         if (row >= m_array.GetCount())\r
             variant = wxT("plain");\r
         else\r
-            variant = wxT("blue");\r
+            variant = wxT("blue/green/red");\r
     }\r
 }\r
 \r
-bool MyListModel::GetAttrByRow( unsigned int row, unsigned int col, wxDataViewItemAttr &attr )\r
+bool MyListModel::GetAttrByRow( unsigned int row, unsigned int col, \r
+                                wxDataViewItemAttr &attr )\r
 {\r
     if (col != 2)\r
         return false;\r
 \r
     if (row < m_array.GetCount())\r
     {\r
-        attr.SetColour( *wxBLUE );\r
-        attr.SetItalic( true );\r
+        attr.SetColour( (row%2) == 0 ? *wxBLUE :\r
+                            ((row%3) == 0 ? *wxGREEN : *wxRED) );\r
+        attr.SetItalic( (row%2) == 5 );\r
     }\r
 \r
     return true;\r
@@ -372,7 +431,12 @@ bool MyListModel::SetValueByRow( const wxVariant &variant,
     if (col == 0)\r
     {\r
         if (row >= m_array.GetCount())\r
+        {\r
+            // the item is not in the range of the items\r
+            // which we store... for simplicity, don't allow editing it\r
+            wxLogError("Cannot edit rows with an index greater than %d", m_array.GetCount());\r
             return false;\r
+        }\r
 \r
         m_array[row] = variant.GetString();\r
         return true;\r
index 2591c86e71b400b0744dafea497ab65220d7bb98..c55e8b2184a6bd1507dcdc6a7f7346f028192267 100644 (file)
 \r
 \r
 // ----------------------------------------------------------------------------\r
-// MyMusicTreeModel\r
+// MyMusicTreeModelNode: a node inside MyMusicTreeModel\r
 // ----------------------------------------------------------------------------\r
 \r
-/*\r
-Implement this data model\r
-            Title               Artist               Year        Judgement\r
---------------------------------------------------------------------------\r
-1: My Music:\r
-    2:  Pop music\r
-        3:  You are not alone   Michael Jackson      1995        good\r
-        4:  Take a bow          Madonna              1994        good\r
-    5:  Classical music\r
-        6:  Ninth Symphony      Ludwig v. Beethoven  1824        good\r
-        7:  German Requiem      Johannes Brahms      1868        good\r
-*/\r
-\r
 class MyMusicTreeModelNode;\r
-WX_DEFINE_ARRAY_PTR( MyMusicTreeModelNode*, MyMusicTreeModelNodes );\r
+WX_DEFINE_ARRAY_PTR( MyMusicTreeModelNode*, MyMusicTreeModelNodePtrArray );\r
 \r
 class MyMusicTreeModelNode\r
 {\r
 public:\r
     MyMusicTreeModelNode( MyMusicTreeModelNode* parent,\r
-                      const wxString &title, const wxString &artist, int year )\r
+                          const wxString &title, const wxString &artist, \r
+                          unsigned int year )\r
     {\r
         m_parent = parent;\r
+\r
         m_title = title;\r
         m_artist = artist;\r
         m_year = year;\r
         m_quality = "good";\r
-        m_isContainer = false;\r
     }\r
 \r
     MyMusicTreeModelNode( MyMusicTreeModelNode* parent,\r
-                      const wxString &branch )\r
+                          const wxString &branch )\r
     {\r
         m_parent = parent;\r
+\r
         m_title = branch;\r
         m_year = -1;\r
-        m_isContainer = true;\r
     }\r
 \r
     ~MyMusicTreeModelNode()\r
     {\r
+        // free all our children nodes\r
         size_t count = m_children.GetCount();\r
-        size_t i;\r
-        for (i = 0; i < count; i++)\r
+        for (size_t i = 0; i < count; i++)\r
         {\r
             MyMusicTreeModelNode *child = m_children[i];\r
             delete child;\r
         }\r
     }\r
 \r
-    bool IsContainer()                                        { return m_isContainer; }\r
-\r
-    MyMusicTreeModelNode* GetParent()                         { return m_parent; }\r
-    MyMusicTreeModelNodes &GetChildren()                      { return m_children; }\r
-    MyMusicTreeModelNode* GetNthChild( unsigned int n )       { return m_children.Item( n ); }\r
-    void Insert( MyMusicTreeModelNode* child, unsigned int n) { m_children.Insert( child, n); }\r
-    void Append( MyMusicTreeModelNode* child )                { m_children.Add( child ); }\r
-    unsigned int GetChildCount()                              { return m_children.GetCount(); }\r
-\r
-public:\r
-    wxString            m_title;\r
-    wxString            m_artist;\r
-    int                 m_year;\r
-    wxString            m_quality;\r
+    bool IsContainer() const\r
+        { return m_children.GetCount()>0; }\r
+\r
+    MyMusicTreeModelNode* GetParent()\r
+        { return m_parent; }\r
+    MyMusicTreeModelNodePtrArray& GetChildren()\r
+        { return m_children; }\r
+    MyMusicTreeModelNode* GetNthChild( unsigned int n )\r
+        { return m_children.Item( n ); }\r
+    void Insert( MyMusicTreeModelNode* child, unsigned int n)\r
+        { m_children.Insert( child, n); }\r
+    void Append( MyMusicTreeModelNode* child )\r
+        { m_children.Add( child ); }\r
+    unsigned int GetChildCount() const\r
+        { return m_children.GetCount(); }\r
+\r
+public:     // public to avoid getters/setters\r
+    wxString                m_title;\r
+    wxString                m_artist;\r
+    int                     m_year;\r
+    wxString                m_quality;\r
 \r
 private:\r
-    MyMusicTreeModelNode   *m_parent;\r
-    MyMusicTreeModelNodes   m_children;\r
-    bool                    m_isContainer;\r
+    MyMusicTreeModelNode          *m_parent;\r
+    MyMusicTreeModelNodePtrArray   m_children;\r
 };\r
 \r
+\r
+// ----------------------------------------------------------------------------\r
+// MyMusicTreeModel\r
+// ----------------------------------------------------------------------------\r
+\r
+/*\r
+Implement this data model\r
+            Title               Artist               Year        Judgement\r
+--------------------------------------------------------------------------\r
+1: My Music:\r
+    2:  Pop music\r
+        3:  You are not alone   Michael Jackson      1995        good\r
+        4:  Take a bow          Madonna              1994        good\r
+    5:  Classical music\r
+        6:  Ninth Symphony      Ludwig v. Beethoven  1824        good\r
+        7:  German Requiem      Johannes Brahms      1868        good\r
+*/\r
+\r
 class MyMusicTreeModel: public wxDataViewModel\r
 {\r
 public:\r
@@ -101,9 +113,15 @@ public:
 \r
     // helper methods to change the model\r
 \r
-    void AddToClassical( const wxString &title, const wxString &artist, int year );\r
+    void AddToClassical( const wxString &title, const wxString &artist, \r
+                         unsigned int year );\r
     void Delete( const wxDataViewItem &item );\r
 \r
+    wxDataViewItem GetNinthItem() const\r
+    {\r
+       return wxDataViewItem( m_ninth );\r
+    }\r
+\r
     // override sorting to always sort branches ascendingly\r
 \r
     int Compare( const wxDataViewItem &item1, const wxDataViewItem &item2,\r
@@ -134,16 +152,15 @@ public:
     virtual unsigned int GetChildren( const wxDataViewItem &parent, \r
                                       wxDataViewItemArray &array ) const;\r
 \r
-    wxDataViewItem GetNinthItem() const\r
-    {\r
-       return wxDataViewItem( m_ninth );\r
-    }\r
-\r
 private:\r
     MyMusicTreeModelNode*   m_root;\r
+\r
+    // pointers to some "special" nodes of the tree:\r
     MyMusicTreeModelNode*   m_pop;\r
     MyMusicTreeModelNode*   m_classical;\r
     MyMusicTreeModelNode*   m_ninth;\r
+\r
+    // ??\r
     bool                    m_classicalMusicIsKnownToControl;\r
 };\r
 \r
@@ -193,7 +210,7 @@ public:
 \r
 private:\r
     wxArrayString    m_array;\r
-    wxIcon           m_icon;\r
+    wxIcon           m_icon[2];\r
     int              m_virtualItems;\r
 };\r
 \r