Make a couple virtuals protected so they can be overridden.

[wxWidgets.git] / samples / dataview / mymodels.cpp

diff --git a/samples/dataview/mymodels.cpp b/samples/dataview/mymodels.cpp
index 13cdd6704f5f361c9f5e11a482644a1121b49145..a3cb11119813bf39b0a076c624199e5b3f95c6c1 100644 (file)
--- a/samples/dataview/mymodels.cpp
+++ b/samples/dataview/mymodels.cpp
@@ -6,7 +6,7 @@
 // Created:     06/01/06
 // RCS-ID:      $Id$
 // Copyright:   (c) Robert Roebling
 // Created:     06/01/06
 // RCS-ID:      $Id$
 // Copyright:   (c) Robert Roebling

-// Licence:     wxWindows license
+// Licence:     wxWindows licence

 /////////////////////////////////////////////////////////////////////////////
 
 
 /////////////////////////////////////////////////////////////////////////////
 
 


@@ -203,18 +203,13 @@ void MyMusicTreeModel::GetValue( wxVariant &variant,
         variant = node->m_quality;
         break;
     case 4:
         variant = node->m_quality;
         break;
     case 4:

-        // wxMac doesn't conceal the popularity progress renderer, return 0 for containers
-        if (IsContainer(item))
-            variant = (long) 0;
-        else
-            variant = (long) 80;  // all music is very 80% popular
+        variant = 80L;  // all music is very 80% popular

         break;
     case 5:
         break;
     case 5:

-        // Make size of red square depend on year

         if (GetYear(item) < 1900)
         if (GetYear(item) < 1900)

-            variant = (long) 35;
+            variant = "old";

         else
         else

-            variant = (long) 25;
+            variant = "new";

         break;
 
     default:
         break;
 
     default:


@@ -249,6 +244,17 @@ bool MyMusicTreeModel::SetValue( const wxVariant &variant,
     return false;
 }
 
     return false;
 }
 

+bool MyMusicTreeModel::IsEnabled( const wxDataViewItem &item,
+                                  unsigned int col ) const
+{
+    wxASSERT(item.IsOk());
+
+    MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();
+
+    // disable Beethoven's ratings, his pieces can only be good
+    return !(col == 3 && node->m_artist.EndsWith("Beethoven"));
+}
+

 wxDataViewItem MyMusicTreeModel::GetParent( const wxDataViewItem &item ) const
 {
     // the invisible root node has no parent
 wxDataViewItem MyMusicTreeModel::GetParent( const wxDataViewItem &item ) const
 {
     // the invisible root node has no parent


@@ -322,43 +328,42 @@ static int my_sort( int *v1, int *v2 )
    return *v1-*v2;
 }
 
    return *v1-*v2;
 }
 

-#ifdef __WXMAC__
-    #define INITIAL_NUMBER_OF_ITEMS 100
-#else
-    #define INITIAL_NUMBER_OF_ITEMS 100000
-#endif
+#define INITIAL_NUMBER_OF_ITEMS 10000

 
 MyListModel::MyListModel() :
         wxDataViewVirtualListModel( INITIAL_NUMBER_OF_ITEMS )
 {
 
 MyListModel::MyListModel() :
         wxDataViewVirtualListModel( INITIAL_NUMBER_OF_ITEMS )
 {

-    m_virtualItems = INITIAL_NUMBER_OF_ITEMS;
-

     // the first 100 items are really stored in this model;
     // the first 100 items are really stored in this model;

-    // all the others are synthetized on request
-    for (unsigned int i = 0; i < 100; i++)
+    // all the others are synthesized on request
+    static const unsigned NUMBER_REAL_ITEMS = 100;
+
+    m_textColValues.reserve(NUMBER_REAL_ITEMS);
+    m_textColValues.push_back("first row with long label to test ellipsization");
+    for (unsigned int i = 1; i < NUMBER_REAL_ITEMS; i++)

     {
     {

-        wxString str;
-        str.Printf( "real row %d", i );
-        m_array.Add( str );
+        m_textColValues.push_back(wxString::Format("real row %d", i));

     }
 
     }
 

+    m_iconColValues.assign(NUMBER_REAL_ITEMS, "test");
+

     m_icon[0] = wxIcon( null_xpm );
     m_icon[1] = wxIcon( wx_small_xpm );
 }
 
 void MyListModel::Prepend( const wxString &text )
 {
     m_icon[0] = wxIcon( null_xpm );
     m_icon[1] = wxIcon( wx_small_xpm );
 }
 
 void MyListModel::Prepend( const wxString &text )
 {

-    m_array.Insert( text, 0 );
+    m_textColValues.Insert( text, 0 );

     RowPrepended();
 }
 
 void MyListModel::DeleteItem( const wxDataViewItem &item )
 {
     unsigned int row = GetRow( item );
     RowPrepended();
 }
 
 void MyListModel::DeleteItem( const wxDataViewItem &item )
 {
     unsigned int row = GetRow( item );

-    if (row >= m_array.GetCount())
+
+    if (row >= m_textColValues.GetCount())

         return;
 
         return;
 

-    m_array.RemoveAt( row );
+    m_textColValues.RemoveAt( row );

     RowDeleted( row );
 }
 
     RowDeleted( row );
 }
 


@@ -369,7 +374,7 @@ void MyListModel::DeleteItems( const wxDataViewItemArray &items )
     for (i = 0; i < items.GetCount(); i++)
     {
         unsigned int row = GetRow( items[i] );
     for (i = 0; i < items.GetCount(); i++)
     {
         unsigned int row = GetRow( items[i] );

-        if (row < m_array.GetCount())
+        if (row < m_textColValues.GetCount())

             rows.Add( row );
     }
 
             rows.Add( row );
     }
 


@@ -377,7 +382,7 @@ void MyListModel::DeleteItems( const wxDataViewItemArray &items )
     {
         // none of the selected items were in the range of the items
         // which we store... for simplicity, don't allow removing them
     {
         // none of the selected items were in the range of the items
         // which we store... for simplicity, don't allow removing them

-        wxLogError( "Cannot remove rows with an index greater than %d", m_array.GetCount() );
+        wxLogError( "Cannot remove rows with an index greater than %d", m_textColValues.GetCount() );

         return;
     }
 
         return;
     }
 


@@ -386,7 +391,7 @@ void MyListModel::DeleteItems( const wxDataViewItemArray &items )
     // remaining indeces would all be wrong.
     rows.Sort( my_sort_reverse );
     for (i = 0; i < rows.GetCount(); i++)
     // remaining indeces would all be wrong.
     rows.Sort( my_sort_reverse );
     for (i = 0; i < rows.GetCount(); i++)

-        m_array.RemoveAt( rows[i] );
+        m_textColValues.RemoveAt( rows[i] );

 
     // This is just to test if wxDataViewCtrl can
     // cope with removing rows not sorted in
 
     // This is just to test if wxDataViewCtrl can
     // cope with removing rows not sorted in


@@ -397,64 +402,96 @@ void MyListModel::DeleteItems( const wxDataViewItemArray &items )
 
 void MyListModel::AddMany()
 {
 
 void MyListModel::AddMany()
 {

-    m_virtualItems += 1000;
-    Reset( m_array.GetCount() + m_virtualItems );
+    Reset( GetCount()+1000 );

 }
 
 void MyListModel::GetValueByRow( wxVariant &variant,
                                  unsigned int row, unsigned int col ) const
 {
 }
 
 void MyListModel::GetValueByRow( wxVariant &variant,
                                  unsigned int row, unsigned int col ) const
 {

-    if (col==0)
+    switch ( col )

     {
     {

-        if (row >= m_array.GetCount())
-            variant = wxString::Format( "virtual row %d", row );
-        else
-            variant = m_array[ row ];
-    }
-    else if (col==1)
-    {
-        wxDataViewIconText data( "test", m_icon[ row%2 ] );
-        variant << data;
-    }
-    else if (col==2)
-    {
-        static const char *labels[5] =
-        {
-            "blue", "green", "red", "bold cyan", "default",
-        };
+        case Col_EditableText:
+            if (row >= m_textColValues.GetCount())
+                variant = wxString::Format( "virtual row %d", row );
+            else
+                variant = m_textColValues[ row ];
+            break;
+
+        case Col_IconText:
+            {
+                wxString text;
+                if ( row >= m_iconColValues.GetCount() )
+                    text = "virtual icon";
+                else
+                    text = m_iconColValues[row];
+
+                variant << wxDataViewIconText(text, m_icon[row % 2]);
+            }
+            break;
+
+        case Col_TextWithAttr:
+            {
+                static const char *labels[5] =
+                {
+                    "blue", "green", "red", "bold cyan", "default",
+                };
+
+                variant = labels[row % 5];
+            }
+            break;

 
 

-        variant = labels[row % 5];
+        case Col_Custom:
+            variant = wxString::Format("%d", row % 100);
+            break;
+
+        case Col_Max:
+            wxFAIL_MSG( "invalid column" );

     }
 }
 
 bool MyListModel::GetAttrByRow( unsigned int row, unsigned int col,
     }
 }
 
 bool MyListModel::GetAttrByRow( unsigned int row, unsigned int col,

-                                wxDataViewItemAttr &attr )
+                                wxDataViewItemAttr &attr ) const

 {
 {

-    if (col != 2)
-        return false;
-
-    // do what the labels defined above hint at
-    switch ( row % 5 )
+    switch ( col )

     {
     {

-        case 0:
-            attr.SetColour(*wxBLUE);
-            break;
-
-        case 1:
-            attr.SetColour(*wxGREEN);
-            break;
+        case Col_EditableText:
+            return false;

 
 

-        case 2:
-            attr.SetColour(*wxRED);
+        case Col_IconText:
+            if ( !(row % 2) )
+                return false;
+            attr.SetColour(*wxLIGHT_GREY);

             break;
 
             break;
 

-        case 3:
-            attr.SetColour(*wxCYAN);
-            attr.SetBold(true);
+        case Col_TextWithAttr:
+        case Col_Custom:
+            // do what the labels defined in GetValueByRow() hint at
+            switch ( row % 5 )
+            {
+                case 0:
+                    attr.SetColour(*wxBLUE);
+                    break;
+
+                case 1:
+                    attr.SetColour(*wxGREEN);
+                    break;
+
+                case 2:
+                    attr.SetColour(*wxRED);
+                    break;
+
+                case 3:
+                    attr.SetColour(*wxCYAN);
+                    attr.SetBold(true);
+                    break;
+
+                case 4:
+                    return false;
+            }

             break;
 
             break;
 

-        case 4:
-            return false;
+        case Col_Max:
+            wxFAIL_MSG( "invalid column" );

     }
 
     return true;
     }
 
     return true;


@@ -463,19 +500,50 @@ bool MyListModel::GetAttrByRow( unsigned int row, unsigned int col,
 bool MyListModel::SetValueByRow( const wxVariant &variant,
                                  unsigned int row, unsigned int col )
 {
 bool MyListModel::SetValueByRow( const wxVariant &variant,
                                  unsigned int row, unsigned int col )
 {

-    if (col == 0)
+    switch ( col )

     {
     {

-        if (row >= m_array.GetCount())
-        {
-            // the item is not in the range of the items
-            // which we store... for simplicity, don't allow editing it
-            wxLogError( "Cannot edit rows with an index greater than %d", m_array.GetCount() );
-            return false;
-        }
+        case Col_EditableText:
+        case Col_IconText:
+            if (row >= m_textColValues.GetCount())
+            {
+                // the item is not in the range of the items
+                // which we store... for simplicity, don't allow editing it
+                wxLogError( "Cannot edit rows with an index greater than %d",
+                            m_textColValues.GetCount() );
+                return false;
+            }
+
+            if ( col == Col_EditableText )
+            {
+                m_textColValues[row] = variant.GetString();
+            }
+            else // col == Col_IconText
+            {
+                wxDataViewIconText iconText;
+                iconText << variant;
+                m_iconColValues[row] = iconText.GetText();
+            }
+            return true;

 
 

-        m_array[row] = variant.GetString();
-        return true;
+        case Col_TextWithAttr:
+        case Col_Custom:
+            wxLogError("Cannot edit the column %d", col);
+            break;
+
+        case Col_Max:
+            wxFAIL_MSG( "invalid column" );

     }
 
     return false;
 }
     }
 
     return false;
 }

+
+
+// ----------------------------------------------------------------------------
+// MyListStoreDerivedModel
+// ----------------------------------------------------------------------------
+
+bool MyListStoreDerivedModel::IsEnabledByRow(unsigned int row, unsigned int col) const
+{
+    // disabled the last two checkboxes
+    return !(col == 0 && 8 <= row && row <= 9);
+}