+
+ // 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 method for wxLog
+
+ wxString GetTitle( const wxDataViewItem &item )
+ {
+ MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
+ if (!node)
+ return wxEmptyString;
+
+ return node->m_title;
+ }
+
+ // helper methods to change the model
+
+ void AddToClassical( const wxString &title, const wxString &artist, int year )
+ {
+ // add to data
+ MyMusicModelNode *child_node =
+ new MyMusicModelNode( 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 Delete( const wxDataViewItem &item )
+ {
+ MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
+ wxDataViewItem parent( node->GetParent() );
+
+ node->GetParent()->GetChildren().Remove( node );
+ delete node;
+
+ // notify control
+ ItemDeleted( parent, item );
+ }
+
+ // override sorting to always sort branches ascendingly
+
+ int 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
+ unsigned long litem1 = (unsigned long) item1.GetID();
+ unsigned long litem2 = (unsigned long) item2.GetID();
+
+ return litem1-litem2;
+ }
+
+ return wxDataViewModel::Compare( item1, item2, column, ascending );
+ }
+
+ // implementation of base class virtuals to define model
+
+ virtual unsigned int GetColumnCount() const
+ {
+ return 3;
+ }
+
+ virtual wxString GetColumnType( unsigned int col ) const
+ {
+ if (col == 2)
+ return "long";
+
+ return "string";
+ }
+
+ 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 = (long) node->m_year; break;
+ default:
+ {
+ wxLogError( "MyMusicModel::GetValue: wrong column" );
+
+ // provoke a crash when mouse button down
+ wxMouseState state = wxGetMouseState();
+ if (state.ShiftDown())
+ {
+ char *crash = 0;
+ *crash = 0;
+ }
+ }
+ }
+ }
+
+ virtual bool SetValue( const wxVariant &variant,
+ const wxDataViewItem &item, unsigned int col )
+ {
+ MyMusicModelNode *node = (MyMusicModelNode*) 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;
+ default: wxLogError( "MyMusicModel::SetValue: wrong column" );