Fix comparison operator for functions keys (fixes #10547).

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

/////////////////////////////////////////////////////////////////////////////\r
// Name:        mymodels.cpp\r
// Purpose:     wxDataViewCtrl wxWidgets sample\r
// Author:      Robert Roebling\r
// Modified by: Francesco Montorsi, Bo Yang\r
// Created:     06/01/06\r
// RCS-ID:      $Id$\r
// Copyright:   (c) Robert Roebling\r
// Licence:     wxWindows license\r
/////////////////////////////////////////////////////////////////////////////\r
\r
\r
// For compilers that support precompilation, includes "wx/wx.h".\r
#include "wx/wxprec.h"\r
\r
#ifdef __BORLANDC__\r
#pragma hdrstop\r
#endif\r
\r
#ifndef WX_PRECOMP\r
    #include "wx/wx.h"\r
#endif\r
\r
#include "wx/dataview.h"\r
#include "mymodels.h"\r
\r
// ----------------------------------------------------------------------------\r
// resources\r
// ----------------------------------------------------------------------------\r
\r
#include "null.xpm"\r
#include "wx_small.xpm"\r
\r
\r
// ----------------------------------------------------------------------------\r
// MyMusicTreeModel\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
\r
    // setup classical music\r
    m_classical = new MyMusicTreeModelNode( m_root, wxT("Classical music") );\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, 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)      // happens if item.IsOk()==false\r
        return wxEmptyString;\r
\r
    return node->m_title;\r
}\r
\r
int MyMusicTreeModel::GetYear( const wxDataViewItem &item ) const\r
{\r
    MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();\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, \r
                                       unsigned int year )\r
{\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
    m_classical->Append( child_node );\r
\r
    if (m_classicalMusicIsKnownToControl)\r
    {\r
        // notify control\r
        wxDataViewItem child( (void*) child_node );\r
        wxDataViewItem parent( (void*) m_classical );\r
        ItemAdded( parent, child );\r
    }\r
}\r
\r
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
    ItemDeleted( parent, item );\r
}\r
\r
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
        GetValue( value1, item1, 0 );\r
        GetValue( value2, item2, 0 );\r
\r
        wxString str1 = value1.GetString();\r
        wxString str2 = value2.GetString();\r
        int res = str1.Cmp( str2 );\r
        if (res) return res;\r
\r
        // items must be different\r
        wxUIntPtr litem1 = (wxUIntPtr) item1.GetID();\r
        wxUIntPtr litem2 = (wxUIntPtr) item2.GetID();\r
\r
        return litem1-litem2;\r
    }\r
\r
    return wxDataViewModel::Compare( item1, item2, column, ascending );\r
}\r
\r
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:\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: \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 invisible root node has no parent\r
    if (!item.IsOk())\r
        return wxDataViewItem(0);\r
\r
    MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();\r
\r
    // "MyMusic" also has no parent\r
    if (node == m_root)\r
        return wxDataViewItem(0);\r
\r
    return wxDataViewItem( (void*) node->GetParent() );\r
}\r
\r
bool MyMusicTreeModel::IsContainer( const wxDataViewItem &item ) const\r
{\r
    // the invisble root node can have children\r
    // (in our model always "MyMusic")\r
    if (!item.IsOk())\r
        return true;\r
\r
    MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();\r
    return node->IsContainer();\r
}\r
\r
unsigned int MyMusicTreeModel::GetChildren( const wxDataViewItem &parent, \r
                                            wxDataViewItemArray &array ) const\r
{\r
    MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) parent.GetID();\r
    if (!node)\r
    {\r
        array.Add( wxDataViewItem( (void*) m_root ) );\r
        return 1;\r
    }\r
\r
    if (node == m_classical)\r
    {\r
        MyMusicTreeModel *model = (MyMusicTreeModel*)(const MyMusicTreeModel*) this;\r
        model->m_classicalMusicIsKnownToControl = true;\r
    }\r
\r
    if (node->GetChildCount() == 0)\r
    {\r
        return 0;\r
    }\r
\r
    unsigned int count = node->GetChildren().GetCount();\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
\r
\r
// ----------------------------------------------------------------------------\r
// MyListModel\r
// ----------------------------------------------------------------------------\r
\r
static int my_sort_reverse( int *v1, int *v2 )\r
{\r
   return *v2-*v1;\r
}\r
\r
static int my_sort( int *v1, int *v2 )\r
{\r
   return *v1-*v2;\r
}\r
\r
#ifdef __WXMAC__\r
    #define INITIAL_NUMBER_OF_ITEMS 100\r
#else\r
    #define INITIAL_NUMBER_OF_ITEMS 100000\r
#endif\r
\r
MyListModel::MyListModel() :\r
        wxDataViewVirtualListModel( INITIAL_NUMBER_OF_ITEMS )\r
{\r
    m_virtualItems = INITIAL_NUMBER_OF_ITEMS;\r
\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("real row %d"), i );\r
        m_array.Add( str );\r
    }\r
\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
{\r
    m_array.Insert( text, 0 );\r
    RowPrepended();\r
}\r
\r
void MyListModel::DeleteItem( const wxDataViewItem &item )\r
{\r
    unsigned int row = GetRow( item );\r
    if (row >= m_array.GetCount())\r
        return;\r
\r
    m_array.RemoveAt( row );\r
    RowDeleted( row );\r
}\r
\r
void MyListModel::DeleteItems( const wxDataViewItemArray &items )\r
{\r
    wxArrayInt rows;\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
    rows.Sort( my_sort_reverse );\r
    for (unsigned int i = 0; i < rows.GetCount(); i++)\r
        m_array.RemoveAt( rows[i] );\r
\r
    // This is just to test if wxDataViewCtrl can\r
    // cope with removing rows not sorted in\r
    // descending order\r
    rows.Sort( my_sort );\r
    RowsDeleted( rows );\r
}\r
\r
void MyListModel::AddMany()\r
{\r
    m_virtualItems += 1000;\r
    Reset( m_array.GetCount() + m_virtualItems );\r
}\r
\r
void MyListModel::GetValueByRow( wxVariant &variant,\r
                                 unsigned int row, unsigned int col ) const\r
{\r
    if (col==0)\r
    {\r
        if (row >= m_array.GetCount())\r
            variant = wxString::Format( wxT("virtual row %d"), row );\r
        else\r
            variant = m_array[ row ];\r
    } \r
    else if (col==1)\r
    {\r
        wxDataViewIconText data( wxT("test"), m_icon[ row%2 ] );\r
        variant << data;\r
    } \r
    else if (col==2)\r
    {\r
        if (row >= m_array.GetCount())\r
            variant = wxT("plain");\r
        else\r
            variant = wxT("blue/green/red");\r
    }\r
}\r
\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( (row%2) == 0 ? *wxBLUE :\r
                            ((row%3) == 0 ? *wxGREEN : *wxRED) );\r
        attr.SetItalic( (row%2) == 5 );\r
    }\r
\r
    return true;\r
}\r
\r
bool MyListModel::SetValueByRow( const wxVariant &variant,\r
                                 unsigned int row, unsigned int col )\r
{\r
    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
    }\r
\r
    return false;\r
}\r