]> git.saurik.com Git - wxWidgets.git/blobdiff - samples/dataview/dataview.cpp
extracted MSW-specific setup.h options to their own wx/msw/setup_inc.h file which...
[wxWidgets.git] / samples / dataview / dataview.cpp
index f199c5870aed39f25f2bd2a73dc9b1fde5231915..bd5594373002a3f5de6e8afdf537fa67285c8ff7 100644 (file)
@@ -1,8 +1,8 @@
 /////////////////////////////////////////////////////////////////////////////
 // Name:        dataview.cpp
-// Purpose:     DataVewCtrl  wxWidgets sample
+// Purpose:     wxDataViewCtrl wxWidgets sample
 // Author:      Robert Roebling
-// Modified by:
+// Modified by: Francesco Montorsi
 // Created:     06/01/06
 // RCS-ID:      $Id$
 // Copyright:   (c) Robert Roebling
 #endif
 
 #ifndef WX_PRECOMP
-#include "wx/wx.h"
+    #include "wx/wx.h"
 #endif
 
+#include "wx/datetime.h"
+#include "wx/splitter.h"
+#include "wx/aboutdlg.h"
+#include "wx/choicdlg.h"
+#include "wx/numdlg.h"
+#include "wx/dataview.h"
+#include "wx/spinctrl.h"
+
 #ifndef __WXMSW__
-#include "mondrian.xpm"
+    #include "../sample.xpm"
 #endif
 
-#include "wx/dataview.h"
+#include "null.xpm"
+
+
+#define DEFAULT_ALIGN                   wxALIGN_LEFT
+#define DATAVIEW_DEFAULT_STYLE          (wxDV_MULTIPLE|wxDV_HORIZ_RULES|wxDV_VERT_RULES)
+
+
 
 // -------------------------------------
-// MyTextModel
+// MyMusicModel
 // -------------------------------------
 
-class MyTextModel: public wxDataViewListModel
+/*
+Implement this data model
+            Title               Artist               Year
+-------------------------------------------------------------
+1: My Music:
+    2:  Pop music
+        3:  You are not alone   Michael Jackson      1995
+        4:  Take a bow          Madonna              1994
+    5:  Classical music
+        6:  Ninth Symphony      Ludwig v. Beethoven  1824
+        7:  German Requiem      Johannes Brahms      1868
+*/
+
+
+
+class MyMusicModelNode;
+WX_DEFINE_ARRAY_PTR( MyMusicModelNode*, MyMusicModelNodes );
+
+class MyMusicModelNode
 {
 public:
-    MyTextModel() 
-    {
-        size_t i;
-        for (i = 0; i < 1000; i++)
-            m_list.Add( wxT("Test") );
-        for (i = 0; i < 500; i++)
-            { m_bools.Add( 0 ); m_bools.Add( 1 ); }
-        for (i = 0; i < 500; i++)
-            { m_colours.Add( wxT("red") ); m_colours.Add( wxT("green") ); }
-        for (i = 0; i < 1000; i++)
-            { m_progress.Add( i/10 ); }
+    MyMusicModelNode( MyMusicModelNode* parent, 
+                      const wxString &title, const wxString &artist, const wxString &year )
+    { 
+        m_parent = parent; 
+        m_title = title;
+        m_artist = artist;
+        m_year = year;
+        m_isContainer = false;
     }
     
-    virtual size_t GetNumberOfRows() 
-        { return 1000; }
-    virtual size_t GetNumberOfCols()
-        { return 6; }
-        
-    // as reported by wxVariant
-    virtual wxString GetColType( size_t col )
+    MyMusicModelNode( MyMusicModelNode* parent,
+                      const wxString &branch )
+    { 
+        m_parent = parent; 
+        m_title = branch;
+        m_isContainer = true;
+    }
+    
+    ~MyMusicModelNode()
+    { 
+        size_t count = m_children.GetCount();
+        size_t i;
+        for (i = 0; i < count; i++)
         {
-            if (col == 5)
-                return wxT("long");
-                
-            if (col == 3)
-                return wxT("bool");
-                 
-            return wxT("string"); 
+            MyMusicModelNode *child = m_children[i];
+            delete child;
         }
+    }
+
+    bool IsContainer()                                    { return m_isContainer; }
+
+    MyMusicModelNode* GetParent()                         { return m_parent; }
+    MyMusicModelNodes &GetChildren()                      { return m_children; }
+    MyMusicModelNode* GetNthChild( unsigned int n )       { return m_children.Item( n ); }
+    void Insert( MyMusicModelNode* child, unsigned int n) { m_children.Insert( child, n); }
+    void Append( MyMusicModelNode* child )                { m_children.Add( child ); }
+    unsigned int GetChildCount()                          { return m_children.GetCount(); }
+
+public:
+    wxString            m_title;
+    wxString            m_artist;
+    wxString            m_year;
+    
+private:
+    MyMusicModelNode   *m_parent;
+    MyMusicModelNodes   m_children; 
+    bool                m_isContainer;
+};
+
+class MyMusicModel: public wxDataViewModel
+{
+public:
+
+    // constructor
+
+    MyMusicModel() 
+    {
+        m_root = new MyMusicModelNode( NULL, "My Music" );
+        m_pop = new MyMusicModelNode( m_root, "Pop music" );
+        m_root->Append( m_pop );
+        m_pop->Append( new MyMusicModelNode( m_pop, 
+            "You are not alone", "Michael Jackson", "1995" ) );
+        m_pop->Append( new MyMusicModelNode( m_pop, 
+            "Take a bow", "Madonna", "1994" ) );
+        m_classical = new MyMusicModelNode( m_root, "Classical music" );
+        m_root->Append( m_classical );
+        m_classical->Append( new MyMusicModelNode( m_classical, 
+            "Ninth symphony", "Ludwig van Beethoven", "1824" ) );
+        m_classical->Append( new MyMusicModelNode( m_classical, 
+            "German Requiem", "Johannes Brahms", "1868" ) );
+        m_classicalMusicIsKnownToControl = false;
+    }
+    
+    // helper methods to change the model
+
+    void AddToClassical( const wxString &title, const wxString &artist, const wxString &year )
+    {
+        // add to data
+        MyMusicModelNode *child_node = 
+            new MyMusicModelNode( m_classical, title, artist, year );
+        m_classical->Append( child_node );
         
-    virtual wxVariant GetValue( size_t col, size_t row )
+        if (m_classicalMusicIsKnownToControl)
         {
-            if (col == 5)
-            {
-                return (long) m_progress[row];
-            } else
-            if (col == 4)
-            {
-                return m_colours[row];
-            } else
-            if (col == 3)
-            {
-                return (bool) m_bools[row];
-            } else 
-            if (col == 2)
-            {
-                return m_list[row];
-            }
-            else
-            {
-                wxString tmp; 
-                tmp.Printf( wxT("item(%d;%d)"), (int)row, (int)col ); 
-                return tmp;
-            }
+            // notify control
+            wxDataViewItem child( (void*) child_node );
+            wxDataViewItem parent( (void*) m_classical );
+            ItemAdded( parent, child );
         }
-    virtual bool SetValue( wxVariant &value, size_t col, size_t row )
+    }
+
+    void Delete( const wxDataViewItem &item )
+    {
+        MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
+        node->GetParent()->GetChildren().Remove( node );
+        delete node;
+        
+        // notify control
+        ItemDeleted( item );
+    }
+    
+    // override sorting to always sort branches ascendingly
+    
+    int Compare( const wxDataViewItem &item1, const wxDataViewItem &item2 )
+    {
+        if (IsContainer(item1) && IsContainer(item2))
         {
-            if (col == 3)
-            {
-                m_bools[row] = (int) value.GetBool();
-            } else
-            if (col == 2)
-            {
-                m_list[row] = value.GetString();
-            }
-            return true;
+            wxVariant value1,value2;
+            GetValue( value1, item1, 0 );
+            GetValue( value2, item2, 0 );
+
+            wxString str1 = value1.GetString();
+            wxString str2 = value2.GetString();
+            int res = str1.Cmp( str2 );
+            if (res) return res;
+            
+            // items must be different
+            unsigned long litem1 = (unsigned long) item1.GetID();    
+            unsigned long litem2 = (unsigned long) item2.GetID();    
+
+            return litem1-litem2;
         }
+        
+        return wxDataViewModel::Compare( item1, item2 );
+    }
+
+    // implementation of base class virtuals to define model
     
-    wxArrayString m_list;
-    wxArrayInt    m_bools;
-    wxArrayString m_colours;
-    wxArrayInt    m_progress;
-};
+    virtual unsigned int GetColumnCount() const
+    {
+        return 3;
+    }
 
-// -------------------------------------
-// MyCustomCell
-// -------------------------------------
+    virtual wxString GetColumnType( unsigned int col ) const
+    {
+        return "string";
+    }
 
-class MyCustomCell: public wxDataViewCustomCell
-{
-public:
-    MyCustomCell() :
-        wxDataViewCustomCell( wxT("string"), wxDATAVIEW_CELL_INERT ) 
-    { 
-        m_colour = wxT("black"); 
+    virtual void GetValue( wxVariant &variant, 
+                           const wxDataViewItem &item, unsigned int col ) const
+    {
+        MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
+        switch (col)
+        {
+            case 0: variant = node->m_title; break;
+            case 1: variant = node->m_artist; break;
+            case 2: variant = node->m_year; break;
+            default: wxLogError( "MyMusicModel::GetValue: wrong column" );
+        }
     }
-    bool SetValue( const wxVariant &value )
+
+    virtual bool SetValue( const wxVariant &variant, 
+                           const wxDataViewItem &item, unsigned int col )
     {
-        m_colour = value.GetString();
-        return true;
+        MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
+        switch (col)
+        {
+            case 0: node->m_title = variant.GetString(); break;
+            case 1: node->m_artist  = variant.GetString(); break;
+            case 2: node->m_year  = variant.GetString(); break;
+            default: wxLogError( "MyMusicModel::SetValue: wrong column" );
+        }
     }
-    bool Render( wxRect rect, wxDC *dc, int state )
+
+    virtual wxDataViewItem GetParent( const wxDataViewItem &item ) const
     {
-        dc->SetPen( *wxBLACK_PEN );
-        if (m_colour == wxT("red"))
-            dc->SetBrush( *wxRED_BRUSH );
-        else if (m_colour == wxT("green"))
-            dc->SetBrush( *wxGREEN_BRUSH );
-        else 
-            dc->SetBrush( *wxBLACK_BRUSH );
-        dc->DrawRectangle( rect );
-        return true;
+        // the invisble root node has no parent
+        if (!item.IsOk())
+            return wxDataViewItem(0);
+            
+        MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
+        
+        // "MyMusic" also has no parent
+        if (node == m_root)
+            return wxDataViewItem(0);
+            
+        return wxDataViewItem( (void*) node->GetParent() );
     }
-    wxSize GetSize()
+
+    virtual bool IsContainer( const wxDataViewItem &item ) const
+    {
+        // the invisble root node can have children (in
+        // our model always "MyMusic")
+        if (!item.IsOk())
+            return true;
+    
+        MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
+        return node->IsContainer();
+    }
+    
+    virtual wxDataViewItem GetFirstChild( const wxDataViewItem &parent ) const
     {
-        return wxSize(20,8);
+        MyMusicModelNode *node = (MyMusicModelNode*) parent.GetID();
+        if (!node)
+            return wxDataViewItem( (void*) m_root );
+        
+        if (node->GetChildCount() == 0)
+            return wxDataViewItem( 0 );
+           
+        if (node == m_classical)
+        {
+            MyMusicModel *model = (MyMusicModel*)(const MyMusicModel*) this;
+            model->m_classicalMusicIsKnownToControl = true;
+        }
+        
+        MyMusicModelNode *first_child = node->GetChildren().Item( 0 );
+        return wxDataViewItem( (void*) first_child );
     }
+    
+    virtual wxDataViewItem GetNextSibling( const wxDataViewItem &item ) const
+    {
+        MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
+        
+        // "MyMusic" has no siblings in our model
+        if (node == m_root)
+            return wxDataViewItem(0);
 
+        MyMusicModelNode *parent = node->GetParent();
+        int pos = parent->GetChildren().Index( node );
+        
+        // Something went wrong
+        if (pos == wxNOT_FOUND)
+            return wxDataViewItem(0);
+            
+        // No more children
+        if (pos == parent->GetChildCount()-1)
+            return wxDataViewItem(0);
+            
+        node = parent->GetChildren().Item( pos+1 );
+        return wxDataViewItem( (void*) node );
+    } 
+    
 private:
-    wxString m_colour;    
+    MyMusicModelNode*   m_root;
+    MyMusicModelNode*   m_pop;
+    MyMusicModelNode*   m_classical;
+    bool                m_classicalMusicIsKnownToControl;
 };
 
 // -------------------------------------
@@ -153,13 +313,14 @@ class MyApp: public wxApp
 {
 public:
     bool OnInit(void);
+    int OnExit();
 };
 
 // -------------------------------------
 // MyFrame
 // -------------------------------------
 
-class MyFrame: public wxFrame
+class MyFrame : public wxFrame
 {
 public:
     MyFrame(wxFrame *frame, wxChar *title, int x, int y, int w, int h);
@@ -167,101 +328,112 @@ public:
 public:
     void OnQuit(wxCommandEvent& event);
     void OnAbout(wxCommandEvent& event);
-    
+    void OnAdd(wxCommandEvent& event);
+    void OnDelete(wxCommandEvent& event);
+
 private:
-    wxDataViewCtrl* dataview_left;
-    wxDataViewCtrl* dataview_right;
+    wxDataViewCtrl* m_dataview;
+    wxTextCtrl    * m_log;
+    wxObjectDataPtr<MyMusicModel> m_model;
+
+private:
+    DECLARE_EVENT_TABLE()
 };
 
 // -------------------------------------
 // MyApp
 // -------------------------------------
 
-#define DYNAMIC_QUIT   wxID_EXIT
-#define DYNAMIC_ABOUT  wxID_ABOUT
-
-IMPLEMENT_APP  (MyApp)
+IMPLEMENT_APP(MyApp)
 
 bool MyApp::OnInit(void)
 {
-    MyFrame *frame = new MyFrame(NULL, _T("Dynamic wxWidgets App"), 50, 50, 600, 340);
+    if ( !wxApp::OnInit() )
+        return false;
 
+    // build the first frame
+    MyFrame *frame = 
+        new MyFrame(NULL, wxT("wxDataViewCtrl feature test"), 10, 10, 700, 440);
     frame->Show(true);
 
     SetTopWindow(frame);
-
     return true;
 }
 
+int MyApp::OnExit()
+{
+    return 0;
+}
+
+
 // -------------------------------------
 // MyFrame
 // -------------------------------------
 
+enum
+{
+    // file menu
+    ID_ABOUT = wxID_ABOUT,
+    ID_EXIT = wxID_EXIT,
+    
+    ID_ADD = 100,
+    ID_DELETE = 101,
+};
+
+BEGIN_EVENT_TABLE(MyFrame, wxFrame)
+    EVT_MENU( ID_ABOUT, MyFrame::OnAbout )
+    EVT_MENU( ID_EXIT, MyFrame::OnQuit )
+    EVT_BUTTON( ID_ADD, MyFrame::OnAdd )
+    EVT_BUTTON( ID_DELETE, MyFrame::OnDelete )
+END_EVENT_TABLE()
+
 MyFrame::MyFrame(wxFrame *frame, wxChar *title, int x, int y, int w, int h):
   wxFrame(frame, wxID_ANY, title, wxPoint(x, y), wxSize(w, h))
 {
-#ifdef __WXMSW__
-    SetIcon(wxIcon(_T("mondrian")));
-#else
-    SetIcon(wxIcon(mondrian_xpm));
-#endif
+    SetIcon(wxICON(sample));
+
+    // build the menus:
 
     wxMenu *file_menu = new wxMenu;
+    file_menu->Append(ID_ABOUT, "&About");
+    file_menu->AppendSeparator();
+    file_menu->Append(ID_EXIT, "E&xit");
 
-    file_menu->Append(DYNAMIC_ABOUT, _T("&About"));
-    file_menu->Append(DYNAMIC_QUIT, _T("E&xit"));
     wxMenuBar *menu_bar = new wxMenuBar;
-    menu_bar->Append(file_menu, _T("&File"));
+    menu_bar->Append(file_menu, "&File");
+
     SetMenuBar(menu_bar);
+    CreateStatusBar();
 
-    // You used to have to do some casting for param 4, but now there are type-safe handlers
-    Connect( DYNAMIC_QUIT,  wxID_ANY,
-                    wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(MyFrame::OnQuit) );
-    Connect( DYNAMIC_ABOUT, wxID_ANY,
-                    wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(MyFrame::OnAbout) );
+    wxBoxSizer *main_sizer = new wxBoxSizer( wxVERTICAL );
 
-    CreateStatusBar();
-    
-    
-    // Left wxDataViewCtrl
-    dataview_left = new wxDataViewCtrl( this, -1 );
-    
-    MyTextModel *model = new MyTextModel;
-    dataview_left->AssociateModel( model );
-    
-    dataview_left->AppendTextColumn( wxT("first"), 0 );
-    dataview_left->AppendTextColumn( wxT("second"), 1 );
-    
-    wxDataViewTextCell *text_cell = new wxDataViewTextCell( wxT("string"), wxDATAVIEW_CELL_EDITABLE );
-    wxDataViewColumn *column = new wxDataViewColumn( wxT("editable"), text_cell, 2 );
-    dataview_left->AppendColumn( column );
+    m_dataview = new wxDataViewCtrl( this, wxID_ANY, wxDefaultPosition,
+                                     wxDefaultSize );
+
+    m_model = new MyMusicModel;
+    m_dataview->AssociateModel( m_model.get() );
+
+    m_dataview->AppendTextColumn( "Title", 0, wxDATAVIEW_CELL_INERT, 200, 
+                                     DEFAULT_ALIGN, wxDATAVIEW_COL_SORTABLE );
+    m_dataview->AppendTextColumn( "Artist", 1, wxDATAVIEW_CELL_EDITABLE, 200,
+                                     DEFAULT_ALIGN, wxDATAVIEW_COL_SORTABLE );
+    m_dataview->AppendTextColumn( "Year", 2, wxDATAVIEW_CELL_INERT, 50,
+                                     DEFAULT_ALIGN );
+
+    main_sizer->Add( m_dataview, 2, wxGROW );
     
-    dataview_left->AppendToggleColumn( wxT("fourth"), 3 );
+    wxBoxSizer *button_sizer = new wxBoxSizer( wxHORIZONTAL );
     
-    MyCustomCell *custom_cell = new MyCustomCell;
-    column = new wxDataViewColumn( wxT("custom"), custom_cell, 4 );
-    dataview_left->AppendColumn( column );
+    button_sizer->Add( new wxButton( this, ID_ADD, "Add Mozart"), 0, wxALL, 10 );
+    button_sizer->Add( new wxButton( this, ID_DELETE, "Delete selected"), 0, wxALL, 10 );
     
-    dataview_left->AppendProgressColumn( wxT("progress"), 5 );
+    main_sizer->Add( button_sizer, 0, 0, 0 );
     
-    // Right wxDataViewCtrl using the same model
-    dataview_right = new wxDataViewCtrl( this, -1 );
-    dataview_right->AssociateModel( model );
+    m_log = new wxTextCtrl( this, -1, "", wxDefaultPosition, wxDefaultSize, wxTE_MULTILINE );
     
-    text_cell = new wxDataViewTextCell( wxT("string"), wxDATAVIEW_CELL_EDITABLE );
-    column = new wxDataViewColumn( wxT("editable"), text_cell, 2 );
-    dataview_right->AppendColumn( column );
-    dataview_right->AppendTextColumn( wxT("first"), 0 );
-    dataview_right->AppendTextColumn( wxT("second"), 1 );
-    wxDataViewToggleCell *toggle_cell = new wxDataViewToggleCell( wxT("bool"), wxDATAVIEW_CELL_EDITABLE );
-    column = new wxDataViewColumn( wxT("bool"), toggle_cell, 3 );
-    dataview_right->AppendColumn( column );
+    main_sizer->Add( m_log, 1, wxGROW );
     
-    wxBoxSizer *sizer = new wxBoxSizer( wxHORIZONTAL );
-    sizer->Add( dataview_left, 1, wxGROW );
-    sizer->Add(10,10);
-    sizer->Add( dataview_right, 1, wxGROW );
-    SetSizer( sizer );
+    SetSizer( main_sizer );
 }
 
 void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event) )
@@ -269,12 +441,25 @@ void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event) )
     Close(true);
 }
 
-void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event) )
+void MyFrame::OnDelete(wxCommandEvent& WXUNUSED(event) )
 {
-    wxMessageDialog dialog(this, _T("This demonstrates the dataview control handling"),
-        _T("About DataView"), wxOK);
+    wxDataViewItem item = m_dataview->GetSelection();
+    if (item.IsOk())
+        m_model->Delete( item );
+}
 
-    dialog.ShowModal();
+void MyFrame::OnAdd(wxCommandEvent& WXUNUSED(event) )
+{
+    m_model->AddToClassical( "Kleine Nachtmusik", "Wolfgang Mozart", "1787" );
 }
 
+void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event) )
+{
+    wxAboutDialogInfo info;
+    info.SetName(_("DataView sample"));
+    info.SetDescription(_("This sample demonstrates the dataview control handling"));
+    info.SetCopyright(_T("(C) 2007 Robert Roebling"));
+
+    wxAboutBox(info);
+}