bf19b714f0
git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@59316 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
383 lines
11 KiB
C++
383 lines
11 KiB
C++
/////////////////////////////////////////////////////////////////////////////
|
|
// Name: mymodels.cpp
|
|
// Purpose: wxDataViewCtrl wxWidgets sample
|
|
// Author: Robert Roebling
|
|
// Modified by: Francesco Montorsi, Bo Yang
|
|
// Created: 06/01/06
|
|
// RCS-ID: $Id$
|
|
// Copyright: (c) Robert Roebling
|
|
// Licence: wxWindows license
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
|
|
|
|
// For compilers that support precompilation, includes "wx/wx.h".
|
|
#include "wx/wxprec.h"
|
|
|
|
#ifdef __BORLANDC__
|
|
#pragma hdrstop
|
|
#endif
|
|
|
|
#ifndef WX_PRECOMP
|
|
#include "wx/wx.h"
|
|
#endif
|
|
|
|
#include "wx/dataview.h"
|
|
#include "mymodels.h"
|
|
|
|
// ----------------------------------------------------------------------------
|
|
// resources
|
|
// ----------------------------------------------------------------------------
|
|
|
|
#include "null.xpm"
|
|
|
|
|
|
// ----------------------------------------------------------------------------
|
|
// MyMusicTreeModel
|
|
// ----------------------------------------------------------------------------
|
|
|
|
MyMusicTreeModel::MyMusicTreeModel()
|
|
{
|
|
m_root = new MyMusicTreeModelNode( NULL, wxT("My Music" ));
|
|
m_pop = new MyMusicTreeModelNode( m_root, wxT("Pop music") );
|
|
m_root->Append( m_pop );
|
|
m_pop->Append( new MyMusicTreeModelNode( m_pop,
|
|
wxT("You are not alone"), wxT("Michael Jackson"), 1995 ) );
|
|
m_pop->Append( new MyMusicTreeModelNode( m_pop,
|
|
wxT("Take a bow"), wxT("Madonna"), 1994 ) );
|
|
m_classical = new MyMusicTreeModelNode( m_root, wxT("Classical music") );
|
|
m_root->Append( m_classical );
|
|
m_ninth = new MyMusicTreeModelNode( m_classical,
|
|
wxT("Ninth symphony"), wxT("Ludwig van Beethoven"), 1824 );
|
|
m_classical->Append( m_ninth );
|
|
m_classical->Append( new MyMusicTreeModelNode( m_classical,
|
|
wxT("German Requiem"), wxT("Johannes Brahms"), 1868 ) );
|
|
m_classicalMusicIsKnownToControl = false;
|
|
}
|
|
|
|
wxString MyMusicTreeModel::GetTitle( const wxDataViewItem &item ) const
|
|
{
|
|
MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();
|
|
if (!node)
|
|
return wxEmptyString;
|
|
|
|
return node->m_title;
|
|
}
|
|
|
|
int MyMusicTreeModel::GetYear( const wxDataViewItem &item ) const
|
|
{
|
|
MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();
|
|
if (!node)
|
|
return 2000;
|
|
|
|
return node->m_year;
|
|
}
|
|
|
|
void MyMusicTreeModel::AddToClassical( const wxString &title, const wxString &artist, int year )
|
|
{
|
|
// add to data
|
|
MyMusicTreeModelNode *child_node =
|
|
new MyMusicTreeModelNode( m_classical, title, artist, year );
|
|
|
|
m_classical->Append( child_node );
|
|
|
|
if (m_classicalMusicIsKnownToControl)
|
|
{
|
|
// notify control
|
|
wxDataViewItem child( (void*) child_node );
|
|
wxDataViewItem parent( (void*) m_classical );
|
|
ItemAdded( parent, child );
|
|
}
|
|
}
|
|
|
|
void MyMusicTreeModel::Delete( const wxDataViewItem &item )
|
|
{
|
|
MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();
|
|
wxDataViewItem parent( node->GetParent() );
|
|
|
|
node->GetParent()->GetChildren().Remove( node );
|
|
delete node;
|
|
|
|
// notify control
|
|
ItemDeleted( parent, item );
|
|
}
|
|
|
|
int MyMusicTreeModel::Compare( const wxDataViewItem &item1, const wxDataViewItem &item2,
|
|
unsigned int column, bool ascending )
|
|
{
|
|
if (IsContainer(item1) && IsContainer(item2))
|
|
{
|
|
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
|
|
wxUIntPtr litem1 = (wxUIntPtr) item1.GetID();
|
|
wxUIntPtr litem2 = (wxUIntPtr) item2.GetID();
|
|
|
|
return litem1-litem2;
|
|
}
|
|
|
|
return wxDataViewModel::Compare( item1, item2, column, ascending );
|
|
}
|
|
|
|
void MyMusicTreeModel::GetValue( wxVariant &variant,
|
|
const wxDataViewItem &item, unsigned int col ) const
|
|
{
|
|
MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();
|
|
switch (col)
|
|
{
|
|
case 0: variant = node->m_title; break;
|
|
case 1: variant = node->m_artist; break;
|
|
case 2: variant = (long) node->m_year; break;
|
|
case 3: 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
|
|
break;
|
|
case 5:
|
|
// Make size of red square depend on year
|
|
if (GetYear(item) < 1900)
|
|
variant = (long) 35;
|
|
else
|
|
variant = (long) 25;
|
|
break;
|
|
default:
|
|
{
|
|
wxLogError( wxT("MyMusicTreeModel::GetValue: wrong column %d"), col );
|
|
|
|
// provoke a crash when mouse button down
|
|
wxMouseState state = wxGetMouseState();
|
|
if (state.ShiftDown())
|
|
{
|
|
char *crash = 0;
|
|
*crash = 0;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
bool MyMusicTreeModel::SetValue( const wxVariant &variant,
|
|
const wxDataViewItem &item, unsigned int col )
|
|
{
|
|
MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();
|
|
switch (col)
|
|
{
|
|
case 0: node->m_title = variant.GetString(); return true;
|
|
case 1: node->m_artist = variant.GetString(); return true;
|
|
case 2: node->m_year = variant.GetLong(); return true;
|
|
case 3: node->m_quality = variant.GetString(); return true;
|
|
default: wxLogError( wxT("MyMusicTreeModel::SetValue: wrong column") );
|
|
}
|
|
return false;
|
|
}
|
|
|
|
wxDataViewItem MyMusicTreeModel::GetParent( const wxDataViewItem &item ) const
|
|
{
|
|
// the invisble root node has no parent
|
|
if (!item.IsOk())
|
|
return wxDataViewItem(0);
|
|
|
|
MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();
|
|
|
|
// "MyMusic" also has no parent
|
|
if (node == m_root)
|
|
return wxDataViewItem(0);
|
|
|
|
return wxDataViewItem( (void*) node->GetParent() );
|
|
}
|
|
|
|
bool MyMusicTreeModel::IsContainer( const wxDataViewItem &item ) const
|
|
{
|
|
// the invisble root node can have children (in
|
|
// our model always "MyMusic")
|
|
if (!item.IsOk())
|
|
return true;
|
|
|
|
MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) item.GetID();
|
|
return node->IsContainer();
|
|
}
|
|
|
|
unsigned int MyMusicTreeModel::GetChildren( const wxDataViewItem &parent, wxDataViewItemArray &array ) const
|
|
{
|
|
MyMusicTreeModelNode *node = (MyMusicTreeModelNode*) parent.GetID();
|
|
if (!node)
|
|
{
|
|
array.Add( wxDataViewItem( (void*) m_root ) );
|
|
return 1;
|
|
}
|
|
|
|
if (node == m_classical)
|
|
{
|
|
MyMusicTreeModel *model = (MyMusicTreeModel*)(const MyMusicTreeModel*) this;
|
|
model->m_classicalMusicIsKnownToControl = true;
|
|
}
|
|
|
|
if (node->GetChildCount() == 0)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
unsigned int count = node->GetChildren().GetCount();
|
|
unsigned int pos;
|
|
for (pos = 0; pos < count; pos++)
|
|
{
|
|
MyMusicTreeModelNode *child = node->GetChildren().Item( pos );
|
|
array.Add( wxDataViewItem( (void*) child ) );
|
|
}
|
|
return count;
|
|
}
|
|
|
|
|
|
|
|
// ----------------------------------------------------------------------------
|
|
// MyListModel
|
|
// ----------------------------------------------------------------------------
|
|
|
|
static int my_sort_reverse( int *v1, int *v2 )
|
|
{
|
|
return *v2-*v1;
|
|
}
|
|
|
|
static int my_sort( int *v1, int *v2 )
|
|
{
|
|
return *v1-*v2;
|
|
}
|
|
|
|
MyListModel::MyListModel() :
|
|
#ifdef __WXMAC__
|
|
wxDataViewVirtualListModel( 1000 + 100 )
|
|
#else
|
|
wxDataViewVirtualListModel( 100000 + 100 )
|
|
#endif
|
|
{
|
|
#ifdef __WXMAC__
|
|
m_virtualItems = 1000;
|
|
#else
|
|
m_virtualItems = 100000;
|
|
#endif
|
|
|
|
unsigned int i;
|
|
for (i = 0; i < 100; i++)
|
|
{
|
|
wxString str;
|
|
str.Printf( wxT("row number %d"), i );
|
|
m_array.Add( str );
|
|
}
|
|
|
|
m_icon = wxIcon( null_xpm );
|
|
}
|
|
|
|
void MyListModel::Prepend( const wxString &text )
|
|
{
|
|
m_array.Insert( text, 0 );
|
|
RowPrepended();
|
|
}
|
|
|
|
void MyListModel::DeleteItem( const wxDataViewItem &item )
|
|
{
|
|
unsigned int row = GetRow( item );
|
|
if (row >= m_array.GetCount())
|
|
return;
|
|
|
|
m_array.RemoveAt( row );
|
|
RowDeleted( row );
|
|
}
|
|
|
|
void MyListModel::DeleteItems( const wxDataViewItemArray &items )
|
|
{
|
|
wxArrayInt rows;
|
|
unsigned int i;
|
|
for (i = 0; i < items.GetCount(); i++)
|
|
{
|
|
unsigned int row = GetRow( items[i] );
|
|
if (row < m_array.GetCount())
|
|
rows.Add( row );
|
|
}
|
|
|
|
// Sort in descending order so that the last
|
|
// row will be deleted first. Otherwise the
|
|
// remaining indeces would all be wrong.
|
|
rows.Sort( my_sort_reverse );
|
|
for (i = 0; i < rows.GetCount(); i++)
|
|
m_array.RemoveAt( rows[i] );
|
|
|
|
// This is just to test if wxDataViewCtrl can
|
|
// cope with removing rows not sorted in
|
|
// descending order
|
|
rows.Sort( my_sort );
|
|
RowsDeleted( rows );
|
|
}
|
|
|
|
void MyListModel::AddMany()
|
|
{
|
|
m_virtualItems += 1000;
|
|
Reset( m_array.GetCount() + m_virtualItems );
|
|
}
|
|
|
|
void MyListModel::GetValueByRow( wxVariant &variant,
|
|
unsigned int row, unsigned int col ) const
|
|
{
|
|
if (col==0)
|
|
{
|
|
if (row >= m_array.GetCount())
|
|
{
|
|
wxString str;
|
|
str.Printf(wxT("row %d"), row - m_array.GetCount() );
|
|
variant = str;
|
|
}
|
|
else
|
|
{
|
|
variant = m_array[ row ];
|
|
}
|
|
} else
|
|
if (col==1)
|
|
{
|
|
wxDataViewIconText data( wxT("test"), m_icon );
|
|
variant << data;
|
|
} else
|
|
if (col==2)
|
|
{
|
|
if (row >= m_array.GetCount())
|
|
variant = wxT("plain");
|
|
else
|
|
variant = wxT("blue");
|
|
}
|
|
}
|
|
|
|
bool MyListModel::GetAttrByRow( unsigned int row, unsigned int col, wxDataViewItemAttr &attr )
|
|
{
|
|
if (col != 2)
|
|
return false;
|
|
|
|
if (row < m_array.GetCount())
|
|
{
|
|
attr.SetColour( *wxBLUE );
|
|
attr.SetItalic( true );
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool MyListModel::SetValueByRow( const wxVariant &variant,
|
|
unsigned int row, unsigned int col )
|
|
{
|
|
if (col == 0)
|
|
{
|
|
if (row >= m_array.GetCount())
|
|
return false;
|
|
|
|
m_array[row] = variant.GetString();
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|