samples/dataview/dataview.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        dataview.cpp
   3 // Purpose:     wxDataViewCtrl wxWidgets sample
   4 // Author:      Robert Roebling
   5 // Modified by: Francesco Montorsi, Bo Yang
   6 // Created:     06/01/06
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) Robert Roebling
   9 // Licence:     wxWindows license
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 // For compilers that support precompilation, includes "wx/wx.h".
  13 #include "wx/wxprec.h"
  14
  15 #ifdef __BORLANDC__
  16 #pragma hdrstop
  17 #endif
  18
  19 #ifndef WX_PRECOMP
  20     #include "wx/wx.h"
  21 #endif
  22
  23 #include "wx/datetime.h"
  24 #include "wx/splitter.h"
  25 #include "wx/aboutdlg.h"
  26 #include "wx/choicdlg.h"
  27 #include "wx/numdlg.h"
  28 #include "wx/dataview.h"
  29 #include "wx/spinctrl.h"
  30 #include "wx/menu.h"
  31
  32 #ifndef __WXMSW__
  33     #include "../sample.xpm"
  34 #endif
  35
  36 #include "null.xpm"
  37
  38 /* XPM */
  39 static const char *small1_xpm[] = {
  40 /* columns rows colors chars-per-pixel */
  41 "16 16 6 1",
  42 ". c Black",
  43 "o c #FFFFFF",
  44 "X c #000080",
  45 "O c #FFFF00",
  46 "  c None",
  47 "+ c #FF0000",
  48 /* pixels */
  49 "                ",
  50 "                ",
  51 "                ",
  52 "    .......     ",
  53 "    .XXXXX.     ",
  54 "    .oXXXX.     ",
  55 "    .oXXX.......",
  56 ".....oXXX.OOOOO.",
  57 ".+++.XXXX.oOOOO.",
  58 ".o++......oOOOO.",
  59 ".o++++.  .oOOOO.",
  60 ".o++++.  .OOOOO.",
  61 ".+++++.  .......",
  62 ".......         ",
  63 "                ",
  64 "                "
  65 };
  66
  67
  68 #define DEFAULT_ALIGN                   wxALIGN_LEFT
  69 #define DATAVIEW_DEFAULT_STYLE          (wxDV_MULTIPLE|wxDV_HORIZ_RULES|wxDV_VERT_RULES)
  70
  71 // -------------------------------------
  72 // MyMusicModel
  73 // -------------------------------------
  74
  75 /*
  76 Implement this data model
  77             Title               Artist               Year        Judgement
  78 --------------------------------------------------------------------------
  79 1: My Music:
  80     2:  Pop music
  81         3:  You are not alone   Michael Jackson      1995        good
  82         4:  Take a bow          Madonna              1994        good
  83     5:  Classical music
  84         6:  Ninth Symphony      Ludwig v. Beethoven  1824        good
  85         7:  German Requiem      Johannes Brahms      1868        good
  86 */
  87
  88
  89
  90 class MyMusicModelNode;
  91 WX_DEFINE_ARRAY_PTR( MyMusicModelNode*, MyMusicModelNodes );
  92
  93 class MyMusicModelNode
  94 {
  95 public:
  96     MyMusicModelNode( MyMusicModelNode* parent,
  97                       const wxString &title, const wxString &artist, int year )
  98     {
  99         m_parent = parent;
 100         m_title = title;
 101         m_artist = artist;
 102         m_year = year;
 103         m_quality = "good";
 104         m_isContainer = false;
 105     }
 106
 107     MyMusicModelNode( MyMusicModelNode* parent,
 108                       const wxString &branch )
 109     {
 110         m_parent = parent;
 111         m_title = branch;
 112         m_year = -1;
 113         m_isContainer = true;
 114     }
 115
 116     ~MyMusicModelNode()
 117     {
 118         size_t count = m_children.GetCount();
 119         size_t i;
 120         for (i = 0; i < count; i++)
 121         {
 122             MyMusicModelNode *child = m_children[i];
 123             delete child;
 124         }
 125     }
 126
 127     bool IsContainer()                                    { return m_isContainer; }
 128
 129     MyMusicModelNode* GetParent()                         { return m_parent; }
 130     MyMusicModelNodes &GetChildren()                      { return m_children; }
 131     MyMusicModelNode* GetNthChild( unsigned int n )       { return m_children.Item( n ); }
 132     void Insert( MyMusicModelNode* child, unsigned int n) { m_children.Insert( child, n); }
 133     void Append( MyMusicModelNode* child )                { m_children.Add( child ); }
 134     unsigned int GetChildCount()                          { return m_children.GetCount(); }
 135
 136 public:
 137     wxString            m_title;
 138     wxString            m_artist;
 139     int                 m_year;
 140     wxString            m_quality;
 141
 142 private:
 143     MyMusicModelNode   *m_parent;
 144     MyMusicModelNodes   m_children;
 145     bool                m_isContainer;
 146 };
 147
 148
 149 class MyMusicModel: public wxDataViewModel
 150 {
 151 public:
 152
 153     // constructor
 154
 155     MyMusicModel()
 156     {
 157         m_root = new MyMusicModelNode( NULL, wxT("My Music" ));
 158         m_pop = new MyMusicModelNode( m_root, wxT("Pop music") );
 159         m_root->Append( m_pop );
 160         m_pop->Append( new MyMusicModelNode( m_pop,
 161             wxT("You are not alone"), wxT("Michael Jackson"), 1995 ) );
 162         m_pop->Append( new MyMusicModelNode( m_pop,
 163             wxT("Take a bow"), wxT("Madonna"), 1994 ) );
 164         m_classical = new MyMusicModelNode( m_root, wxT("Classical music") );
 165         m_root->Append( m_classical );
 166         m_ninth = new MyMusicModelNode( m_classical,
 167             wxT("Ninth symphony"), wxT("Ludwig van Beethoven"), 1824 );
 168         m_classical->Append( m_ninth );
 169         m_classical->Append( new MyMusicModelNode( m_classical,
 170             wxT("German Requiem"), wxT("Johannes Brahms"), 1868 ) );
 171         m_classicalMusicIsKnownToControl = false;
 172     }
 173
 174     ~MyMusicModel()
 175     {
 176         delete m_root;
 177     }
 178
 179     // helper method for wxLog
 180
 181     wxString GetTitle( const wxDataViewItem &item ) const
 182     {
 183         MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
 184         if (!node)
 185             return wxEmptyString;
 186
 187         return node->m_title;
 188     }
 189
 190     int GetYear( const wxDataViewItem &item ) const
 191     {
 192         MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
 193         if (!node)
 194             return 2000;
 195
 196         return node->m_year;
 197     }
 198
 199     // helper methods to change the model
 200
 201     void AddToClassical( const wxString &title, const wxString &artist, int year )
 202     {
 203         // add to data
 204         MyMusicModelNode *child_node =
 205             new MyMusicModelNode( m_classical, title, artist, year );
 206
 207         m_classical->Append( child_node );
 208
 209         if (m_classicalMusicIsKnownToControl)
 210         {
 211             // notify control
 212             wxDataViewItem child( (void*) child_node );
 213             wxDataViewItem parent( (void*) m_classical );
 214             ItemAdded( parent, child );
 215         }
 216     }
 217
 218     void Delete( const wxDataViewItem &item )
 219     {
 220         MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
 221         wxDataViewItem parent( node->GetParent() );
 222
 223         node->GetParent()->GetChildren().Remove( node );
 224         delete node;
 225
 226         // notify control
 227         ItemDeleted( parent, item );
 228     }
 229
 230     // override sorting to always sort branches ascendingly
 231
 232     int Compare( const wxDataViewItem &item1, const wxDataViewItem &item2,
 233                  unsigned int column, bool ascending )
 234     {
 235         if (IsContainer(item1) && IsContainer(item2))
 236         {
 237             wxVariant value1,value2;
 238             GetValue( value1, item1, 0 );
 239             GetValue( value2, item2, 0 );
 240
 241             wxString str1 = value1.GetString();
 242             wxString str2 = value2.GetString();
 243             int res = str1.Cmp( str2 );
 244             if (res) return res;
 245
 246             // items must be different
 247             wxUIntPtr litem1 = (wxUIntPtr) item1.GetID();
 248             wxUIntPtr litem2 = (wxUIntPtr) item2.GetID();
 249
 250             return litem1-litem2;
 251         }
 252
 253         return wxDataViewModel::Compare( item1, item2, column, ascending );
 254     }
 255
 256     // implementation of base class virtuals to define model
 257
 258     virtual unsigned int GetColumnCount() const
 259     {
 260         return 6;
 261     }
 262
 263     virtual wxString GetColumnType( unsigned int col ) const
 264     {
 265         if (col == 2)
 266             return wxT("long");
 267
 268         return wxT("string");
 269     }
 270
 271     virtual void GetValue( wxVariant &variant,
 272                            const wxDataViewItem &item, unsigned int col ) const
 273     {
 274         MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
 275         switch (col)
 276         {
 277             case 0: variant = node->m_title; break;
 278             case 1: variant = node->m_artist; break;
 279             case 2: variant = (long) node->m_year; break;
 280             case 3: variant = node->m_quality; break;
 281             case 4:
 282                // wxMac doesn't conceal the popularity progress renderer, return 0 for containers
 283                if (IsContainer(item))
 284                   variant = (long) 0;
 285                else
 286                   variant = (long) 80;  // all music is very 80% popular
 287                break;
 288             case 5:
 289                // Make size of red square depend on year
 290                if (GetYear(item) < 1900)
 291                   variant = (long) 35;
 292                else
 293                   variant = (long) 25;
 294                break;
 295             default:
 296             {
 297                 wxLogError( wxT("MyMusicModel::GetValue: wrong column %d"), col );
 298
 299                 // provoke a crash when mouse button down
 300                 wxMouseState state = wxGetMouseState();
 301                 if (state.ShiftDown())
 302                 {
 303                     char *crash = 0;
 304                     *crash = 0;
 305                 }
 306             }
 307         }
 308     }
 309
 310     virtual bool SetValue( const wxVariant &variant,
 311                            const wxDataViewItem &item, unsigned int col )
 312     {
 313         MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
 314         switch (col)
 315         {
 316             case 0: node->m_title = variant.GetString(); return true;
 317             case 1: node->m_artist  = variant.GetString(); return true;
 318             case 2: node->m_year  = variant.GetLong(); return true;
 319             case 3: node->m_quality  = variant.GetString(); return true;
 320             default: wxLogError( wxT("MyMusicModel::SetValue: wrong column") );
 321         }
 322         return false;
 323     }
 324
 325     virtual wxDataViewItem GetParent( const wxDataViewItem &item ) const
 326     {
 327         // the invisble root node has no parent
 328         if (!item.IsOk())
 329             return wxDataViewItem(0);
 330
 331         MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
 332
 333         // "MyMusic" also has no parent
 334         if (node == m_root)
 335             return wxDataViewItem(0);
 336
 337         return wxDataViewItem( (void*) node->GetParent() );
 338     }
 339
 340     virtual bool IsContainer( const wxDataViewItem &item ) const
 341     {
 342         // the invisble root node can have children (in
 343         // our model always "MyMusic")
 344         if (!item.IsOk())
 345             return true;
 346
 347         MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
 348         return node->IsContainer();
 349     }
 350
 351     virtual unsigned int GetChildren( const wxDataViewItem &parent, wxDataViewItemArray &array ) const
 352     {
 353         MyMusicModelNode *node = (MyMusicModelNode*) parent.GetID();
 354         if (!node)
 355         {
 356             array.Add( wxDataViewItem( (void*) m_root ) );
 357             return 1;
 358         }
 359
 360         if (node == m_classical)
 361         {
 362             MyMusicModel *model = (MyMusicModel*)(const MyMusicModel*) this;
 363             model->m_classicalMusicIsKnownToControl = true;
 364         }
 365
 366         if (node->GetChildCount() == 0)
 367         {
 368             return 0;
 369         }
 370
 371         unsigned int count = node->GetChildren().GetCount();
 372         unsigned int pos;
 373         for (pos = 0; pos < count; pos++)
 374         {
 375             MyMusicModelNode *child = node->GetChildren().Item( pos );
 376             array.Add( wxDataViewItem( (void*) child ) );
 377         }
 378         return count;
 379     }
 380
 381     wxDataViewItem GetNinthItem()
 382     {
 383        return wxDataViewItem( m_ninth );
 384     }
 385
 386 private:
 387     MyMusicModelNode*   m_root;
 388     MyMusicModelNode*   m_pop;
 389     MyMusicModelNode*   m_classical;
 390     MyMusicModelNode*   m_ninth;
 391     bool                m_classicalMusicIsKnownToControl;
 392 };
 393
 394
 395 static int my_sort_reverse( int *v1, int *v2 )
 396 {
 397    return *v2-*v1;
 398 }
 399
 400 static int my_sort( int *v1, int *v2 )
 401 {
 402    return *v1-*v2;
 403 }
 404
 405 class MyListModel: public wxDataViewVirtualListModel
 406 {
 407 public:
 408     MyListModel() :
 409 #ifdef __WXMAC__
 410         wxDataViewVirtualListModel( 1000 + 100 )
 411 #else
 412         wxDataViewVirtualListModel( 100000 + 100 )
 413 #endif
 414     {
 415 #ifdef __WXMAC__
 416         m_virtualItems = 1000;
 417 #else
 418         m_virtualItems = 100000;
 419 #endif
 420
 421         unsigned int i;
 422         for (i = 0; i < 100; i++)
 423         {
 424             wxString str;
 425             str.Printf( wxT("row number %d"), i );
 426             m_array.Add( str );
 427         }
 428
 429         m_icon = wxIcon( null_xpm );
 430     }
 431
 432     // helper methods to change the model
 433
 434     void Prepend( const wxString &text )
 435     {
 436         m_array.Insert( text, 0 );
 437         RowPrepended();
 438     }
 439
 440     void DeleteItem( const wxDataViewItem &item )
 441     {
 442         unsigned int row = GetRow( item );
 443         if (row >= m_array.GetCount())
 444            return;
 445
 446         m_array.RemoveAt( row );
 447         RowDeleted( row );
 448     }
 449
 450     void DeleteItems( const wxDataViewItemArray &items )
 451     {
 452         wxArrayInt rows;
 453         unsigned int i;
 454         for (i = 0; i < items.GetCount(); i++)
 455         {
 456             unsigned int row = GetRow( items[i] );
 457             if (row < m_array.GetCount())
 458                 rows.Add( row );
 459         }
 460
 461         // Sort in descending order so that the last
 462         // row will be deleted first. Otherwise the
 463         // remaining indeces would all be wrong.
 464         rows.Sort( my_sort_reverse );
 465         for (i = 0; i < rows.GetCount(); i++)
 466             m_array.RemoveAt( rows[i] );
 467
 468         // This is just to test if wxDataViewCtrl can
 469         // cope with removing rows not sorted in
 470         // descending order
 471         rows.Sort( my_sort );
 472         RowsDeleted( rows );
 473     }
 474
 475     void AddMany()
 476     {
 477         m_virtualItems += 1000;
 478         Reset( m_array.GetCount() + m_virtualItems );
 479     }
 480
 481     // implementation of base class virtuals to define model
 482
 483     virtual unsigned int GetColumnCount() const
 484     {
 485         return 3;
 486     }
 487
 488     virtual wxString GetColumnType( unsigned int col ) const
 489     {
 490         if (col == 1)
 491             return wxT("wxDataViewIconText");
 492
 493         return wxT("string");
 494     }
 495
 496     virtual unsigned int GetRowCount()
 497     {
 498         return m_array.GetCount();
 499     }
 500
 501     virtual void GetValueByRow( wxVariant &variant,
 502                            unsigned int row, unsigned int col ) const
 503     {
 504         if (col==0)
 505         {
 506             if (row >= m_array.GetCount())
 507             {
 508                 wxString str;
 509                 str.Printf(wxT("row %d"), row - m_array.GetCount() );
 510                 variant = str;
 511             }
 512             else
 513             {
 514                 variant = m_array[ row ];
 515             }
 516         } else
 517         if (col==1)
 518         {
 519             wxDataViewIconText data( wxT("test"), m_icon );
 520             variant << data;
 521         } else
 522         if (col==2)
 523         {
 524             if (row >= m_array.GetCount())
 525                 variant = wxT("plain");
 526             else
 527                 variant = wxT("blue");
 528         }
 529     }
 530
 531     virtual bool GetAttrByRow( unsigned int row, unsigned int col, wxDataViewItemAttr &attr )
 532     {
 533         if (col != 2)
 534             return false;
 535
 536         if (row < m_array.GetCount())
 537         {
 538             attr.SetColour( *wxBLUE );
 539             attr.SetItalic( true );
 540         }
 541
 542         return true;
 543     }
 544
 545     virtual bool SetValueByRow( const wxVariant &variant,
 546                            unsigned int row, unsigned int col )
 547     {
 548         if (col == 0)
 549         {
 550             if (row >= m_array.GetCount())
 551                return false;
 552
 553             m_array[row] = variant.GetString();
 554             return true;
 555         }
 556
 557         return false;
 558     }
 559
 560     wxArrayString    m_array;
 561     wxIcon           m_icon;
 562     int              m_virtualItems;
 563 };
 564
 565 // -------------------------------------
 566 // MyCustomRenderer
 567 // -------------------------------------
 568
 569 class MyCustomRenderer: public wxDataViewCustomRenderer
 570 {
 571 public:
 572     MyCustomRenderer( wxDataViewCellMode mode, int alignment ) :
 573        wxDataViewCustomRenderer( wxString("long"), mode, alignment )
 574        { m_height = 25; }
 575
 576     virtual bool Render( wxRect rect, wxDC *dc, int WXUNUSED(state) )
 577     {
 578         dc->SetBrush( *wxRED_BRUSH );
 579         dc->SetPen( *wxTRANSPARENT_PEN );
 580         dc->DrawRectangle( rect );
 581         return true;
 582     }
 583
 584
 585     virtual bool Activate( wxRect WXUNUSED(cell),
 586                            wxDataViewModel *WXUNUSED(model), const wxDataViewItem &WXUNUSED(item), unsigned int WXUNUSED(col) )
 587     {
 588         wxLogMessage( wxT("MyCustomRenderer Activate()") );
 589         return false;
 590     }
 591
 592     virtual bool LeftClick( wxPoint cursor, wxRect WXUNUSED(cell),
 593                            wxDataViewModel *WXUNUSED(model), const wxDataViewItem &WXUNUSED(item), unsigned int WXUNUSED(col) )
 594     {
 595         wxLogMessage( wxT("MyCustomRenderer LeftClick( %d, %d )"), cursor.x, cursor.y );
 596         return false;
 597     }
 598
 599     virtual wxSize GetSize() const
 600     {
 601         //return wxSize(60,m_height);
 602         return wxSize(60,20);
 603     }
 604
 605     virtual bool SetValue( const wxVariant &value )
 606     {
 607         m_height = value;
 608         return true;
 609     }
 610
 611     virtual bool GetValue( wxVariant &WXUNUSED(value) ) const { return true; }
 612
 613 private:
 614     long m_height;
 615 };
 616
 617 // -------------------------------------
 618 // MyApp
 619 // -------------------------------------
 620
 621 class MyApp: public wxApp
 622 {
 623 public:
 624     bool OnInit(void);
 625     int OnExit();
 626 };
 627
 628 // -------------------------------------
 629 // MyFrame
 630 // -------------------------------------
 631
 632 class MyFrame : public wxFrame
 633 {
 634 public:
 635     MyFrame(wxFrame *frame, const wxString &title, int x, int y, int w, int h);
 636     ~MyFrame();
 637
 638 public:
 639     void OnQuit(wxCommandEvent& event);
 640     void OnAbout(wxCommandEvent& event);
 641
 642     void OnAddMozart(wxCommandEvent& event);
 643     void OnDeleteMusic(wxCommandEvent& event);
 644     void OnDeleteYear(wxCommandEvent& event);
 645     void OnSelectNinth(wxCommandEvent& event);
 646
 647     void OnPrependList(wxCommandEvent& event);
 648     void OnDeleteList(wxCommandEvent& event);
 649
 650     void OnValueChanged( wxDataViewEvent &event );
 651
 652     void OnActivated( wxDataViewEvent &event );
 653     void OnExpanding( wxDataViewEvent &event );
 654     void OnExpanded( wxDataViewEvent &event );
 655     void OnCollapsing( wxDataViewEvent &event );
 656     void OnCollapsed( wxDataViewEvent &event );
 657     void OnSelectionChanged( wxDataViewEvent &event );
 658
 659     void OnEditingStarted( wxDataViewEvent &event );
 660     void OnEditingDone( wxDataViewEvent &event );
 661
 662     void OnHeaderClick( wxDataViewEvent &event );
 663     void OnHeaderRightClick( wxDataViewEvent &event );
 664     void OnSorted( wxDataViewEvent &event );
 665
 666     void OnContextMenu( wxDataViewEvent &event );
 667
 668     void OnRightClick( wxMouseEvent &event );
 669     void OnGoto( wxCommandEvent &event);
 670     void OnAddMany( wxCommandEvent &event);
 671
 672     void OnBeginDrag( wxDataViewEvent &event );
 673
 674 private:
 675     wxDataViewCtrl* m_musicCtrl;
 676     wxObjectDataPtr<MyMusicModel> m_music_model;
 677
 678     wxDataViewCtrl* m_listCtrl;
 679     wxObjectDataPtr<MyListModel> m_list_model;
 680
 681     wxDataViewColumn * m_col;
 682
 683     wxTextCtrl    * m_log;
 684     wxLog *m_logOld;
 685
 686 private:
 687     DECLARE_EVENT_TABLE()
 688 };
 689
 690 // -------------------------------------
 691 // MyApp
 692 // -------------------------------------
 693
 694 IMPLEMENT_APP(MyApp)
 695
 696 bool MyApp::OnInit(void)
 697 {
 698     if ( !wxApp::OnInit() )
 699         return false;
 700
 701     // build the first frame
 702     MyFrame *frame =
 703         new MyFrame(NULL, wxT("wxDataViewCtrl feature test"), 40, 40, 1000, 540);
 704     frame->Show(true);
 705
 706     SetTopWindow(frame);
 707     return true;
 708 }
 709
 710 int MyApp::OnExit()
 711 {
 712     return 0;
 713 }
 714
 715
 716 // -------------------------------------
 717 // MyFrame
 718 // -------------------------------------
 719
 720 enum
 721 {
 722     // file menu
 723     ID_ABOUT = wxID_ABOUT,
 724     ID_EXIT = wxID_EXIT,
 725
 726     ID_MUSIC_CTRL       = 50,
 727
 728     ID_ADD_MOZART       = 100,
 729     ID_DELETE_MUSIC     = 101,
 730     ID_DELETE_YEAR      = 102,
 731     ID_SELECT_NINTH     = 103,
 732
 733     ID_PREPEND_LIST     = 200,
 734     ID_DELETE_LIST      = 201,
 735     ID_GOTO             = 202,
 736     ID_ADD_MANY         = 203
 737 };
 738
 739 BEGIN_EVENT_TABLE(MyFrame, wxFrame)
 740     EVT_MENU( ID_ABOUT, MyFrame::OnAbout )
 741     EVT_MENU( ID_EXIT, MyFrame::OnQuit )
 742     EVT_BUTTON( ID_ADD_MOZART, MyFrame::OnAddMozart )
 743     EVT_BUTTON( ID_DELETE_MUSIC, MyFrame::OnDeleteMusic )
 744     EVT_BUTTON( ID_DELETE_YEAR, MyFrame::OnDeleteYear )
 745     EVT_BUTTON( ID_SELECT_NINTH, MyFrame::OnSelectNinth )
 746     EVT_BUTTON( ID_PREPEND_LIST, MyFrame::OnPrependList )
 747     EVT_BUTTON( ID_DELETE_LIST, MyFrame::OnDeleteList )
 748     EVT_BUTTON( ID_GOTO, MyFrame::OnGoto)
 749     EVT_BUTTON( ID_ADD_MANY, MyFrame::OnAddMany)
 750
 751     EVT_DATAVIEW_ITEM_VALUE_CHANGED( ID_MUSIC_CTRL, MyFrame::OnValueChanged )
 752
 753     EVT_DATAVIEW_ITEM_ACTIVATED(ID_MUSIC_CTRL, MyFrame::OnActivated )
 754     EVT_DATAVIEW_ITEM_EXPANDING(ID_MUSIC_CTRL, MyFrame::OnExpanding)
 755     EVT_DATAVIEW_ITEM_EXPANDED(ID_MUSIC_CTRL, MyFrame::OnExpanded)
 756     EVT_DATAVIEW_ITEM_COLLAPSING(ID_MUSIC_CTRL, MyFrame::OnCollapsing)
 757     EVT_DATAVIEW_ITEM_COLLAPSED(ID_MUSIC_CTRL, MyFrame::OnCollapsed)
 758     EVT_DATAVIEW_SELECTION_CHANGED(ID_MUSIC_CTRL, MyFrame::OnSelectionChanged)
 759
 760     EVT_DATAVIEW_ITEM_EDITING_STARTED(ID_MUSIC_CTRL, MyFrame::OnEditingStarted)
 761     EVT_DATAVIEW_ITEM_EDITING_DONE(ID_MUSIC_CTRL, MyFrame::OnEditingDone)
 762
 763     EVT_DATAVIEW_COLUMN_HEADER_CLICK(ID_MUSIC_CTRL, MyFrame::OnHeaderClick)
 764     EVT_DATAVIEW_COLUMN_HEADER_RIGHT_CLICKED(ID_MUSIC_CTRL, MyFrame::OnHeaderRightClick)
 765     EVT_DATAVIEW_COLUMN_SORTED(ID_MUSIC_CTRL, MyFrame::OnSorted)
 766
 767     EVT_DATAVIEW_ITEM_CONTEXT_MENU(ID_MUSIC_CTRL, MyFrame::OnContextMenu)
 768
 769     EVT_DATAVIEW_ITEM_BEGIN_DRAG( ID_MUSIC_CTRL, MyFrame::OnBeginDrag )
 770
 771     EVT_RIGHT_UP(MyFrame::OnRightClick)
 772 END_EVENT_TABLE()
 773
 774 MyFrame::MyFrame(wxFrame *frame, const wxString &title, int x, int y, int w, int h):
 775   wxFrame(frame, wxID_ANY, title, wxPoint(x, y), wxSize(w, h))
 776 {
 777     m_log = NULL;
 778     m_col = NULL;
 779
 780     SetIcon(wxICON(sample));
 781
 782     // build the menus:
 783
 784     wxMenu *file_menu = new wxMenu;
 785     file_menu->Append(ID_ABOUT, wxT("&About"));
 786     file_menu->AppendSeparator();
 787     file_menu->Append(ID_EXIT, wxT("E&xit"));
 788
 789     wxMenuBar *menu_bar = new wxMenuBar;
 790     menu_bar->Append(file_menu, wxT("&File"));
 791
 792     SetMenuBar(menu_bar);
 793     CreateStatusBar();
 794
 795     wxPanel *panel = new wxPanel( this, -1 );
 796
 797     wxBoxSizer *main_sizer = new wxBoxSizer( wxVERTICAL );
 798
 799     wxBoxSizer *data_sizer = new wxBoxSizer( wxHORIZONTAL );
 800
 801     // MyMusic
 802
 803     m_musicCtrl = new wxDataViewCtrl( panel, ID_MUSIC_CTRL, wxDefaultPosition,
 804                                     wxSize(400,200), wxDV_MULTIPLE|wxDV_VARIABLE_LINE_HEIGHT );
 805
 806     m_music_model = new MyMusicModel;
 807     m_musicCtrl->AssociateModel( m_music_model.get() );
 808
 809     m_musicCtrl->EnableDragSource( wxDF_TEXT );
 810
 811     wxDataViewTextRenderer *tr = new wxDataViewTextRenderer( wxT("string"), wxDATAVIEW_CELL_INERT );
 812     wxDataViewColumn *column0 = new wxDataViewColumn( wxT("title"), tr, 0, 200, wxALIGN_LEFT,
 813         wxDATAVIEW_COL_SORTABLE | wxDATAVIEW_COL_RESIZABLE );
 814     m_musicCtrl->AppendColumn( column0 );
 815 #if 0
 816     // Call this and sorting is enabled
 817     // immediatly upon start up.
 818     column0->SetAsSortKey();
 819 #endif
 820
 821     tr = new wxDataViewTextRenderer( wxT("string"), wxDATAVIEW_CELL_EDITABLE );
 822     wxDataViewColumn *column1 = new wxDataViewColumn( wxT("artist"), tr, 1, 150, wxALIGN_LEFT,
 823         wxDATAVIEW_COL_SORTABLE | wxDATAVIEW_COL_REORDERABLE | wxDATAVIEW_COL_RESIZABLE );
 824     column1->SetMinWidth(150); // this column can't be resized to be smaller
 825     m_musicCtrl->AppendColumn( column1 );
 826
 827     wxDataViewSpinRenderer *sr = new wxDataViewSpinRenderer( 0, 2010, wxDATAVIEW_CELL_EDITABLE, wxALIGN_RIGHT );
 828     wxDataViewColumn *column2 = new wxDataViewColumn( wxT("year"), sr, 2, 60, wxALIGN_LEFT,
 829         wxDATAVIEW_COL_SORTABLE | wxDATAVIEW_COL_REORDERABLE );
 830     m_musicCtrl->AppendColumn( column2 );
 831
 832     wxArrayString choices;
 833     choices.Add( "good" );
 834     choices.Add( "bad" );
 835     choices.Add( "lousy" );
 836     wxDataViewChoiceRenderer *c = new wxDataViewChoiceRenderer( choices, wxDATAVIEW_CELL_EDITABLE, wxALIGN_RIGHT );
 837     wxDataViewColumn *column3 = new wxDataViewColumn( wxT("rating"), c, 3, 100, wxALIGN_LEFT,
 838         wxDATAVIEW_COL_REORDERABLE | wxDATAVIEW_COL_RESIZABLE );
 839     m_musicCtrl->AppendColumn( column3 );
 840
 841     m_musicCtrl->AppendProgressColumn( wxT("popularity"), 4, wxDATAVIEW_CELL_INERT, 80 );
 842
 843     MyCustomRenderer *cr = new MyCustomRenderer( wxDATAVIEW_CELL_ACTIVATABLE, wxALIGN_RIGHT );
 844     wxDataViewColumn *column4 = new wxDataViewColumn( wxT("custom"), cr, 5, -1, wxALIGN_LEFT,
 845         wxDATAVIEW_COL_RESIZABLE );
 846     m_musicCtrl->AppendColumn( column4 );
 847
 848     data_sizer->Add( m_musicCtrl, 3, wxGROW );
 849
 850     // MyList
 851
 852     m_listCtrl = new wxDataViewCtrl( panel, wxID_ANY, wxDefaultPosition,
 853                                      wxSize(400,200), wxDV_MULTIPLE | wxDV_ROW_LINES);
 854
 855     m_list_model = new MyListModel;
 856     m_listCtrl->AssociateModel( m_list_model.get() );
 857
 858 #if 1
 859     m_listCtrl->AppendTextColumn    (wxT("editable string"), 0, wxDATAVIEW_CELL_EDITABLE, 120 );
 860     m_listCtrl->AppendIconTextColumn(wxIcon(small1_xpm), 1, wxDATAVIEW_CELL_INERT )->SetTitle( wxT("icon") );
 861 #else
 862     m_listCtrl->AppendTextColumn    (wxT("editable string"), 0, wxDATAVIEW_CELL_EDITABLE );
 863     m_listCtrl->AppendIconTextColumn(wxT("icon"),            1, wxDATAVIEW_CELL_INERT );
 864 #endif
 865
 866     wxDataViewTextRendererAttr *ra = new wxDataViewTextRendererAttr;
 867     wxDataViewColumn *column5 = new wxDataViewColumn(wxT("attributes"), ra, 2 );
 868     m_listCtrl->AppendColumn( column5 );
 869
 870     data_sizer->Add( m_listCtrl, 2, wxGROW );
 871
 872     main_sizer->Add( data_sizer, 2, wxGROW );
 873
 874     wxBoxSizer *button_sizer = new wxBoxSizer( wxHORIZONTAL );
 875
 876     button_sizer->Add( new wxButton( panel, ID_ADD_MOZART,  _("Add Mozart")),             0, wxALL, 10 );
 877     button_sizer->Add( new wxButton( panel, ID_DELETE_MUSIC,_("Delete selected")),        0, wxALL, 10 );
 878     button_sizer->Add( new wxButton( panel, ID_DELETE_YEAR, _("Delete \"Year\" column")), 0, wxALL, 10 );
 879     button_sizer->Add( new wxButton( panel, ID_SELECT_NINTH, _("Select Ninth")), 0, wxALL, 10 );
 880     button_sizer->Add( 10, 10, 1 );
 881     wxFlexGridSizer *grid_sizer = new wxFlexGridSizer( 2, 2 );
 882     grid_sizer->Add( new wxButton( panel, ID_PREPEND_LIST,_("Prepend")),         0, wxALL, 2 );
 883     grid_sizer->Add( new wxButton( panel, ID_DELETE_LIST, _("Delete selected")), 0, wxALL, 2 );
 884     grid_sizer->Add( new wxButton( panel, ID_GOTO,        _("Goto 50")),         0, wxALL, 2 );
 885     grid_sizer->Add( new wxButton( panel, ID_ADD_MANY,    _("Add 1000")),        0, wxALL, 2 );
 886     button_sizer->Add( grid_sizer, 0, wxALL, 10 );
 887
 888     main_sizer->Add( button_sizer, 0, wxGROW, 0 );
 889
 890     wxBoxSizer *bottom_sizer = new wxBoxSizer( wxHORIZONTAL );
 891
 892     m_log = new wxTextCtrl( panel, -1, wxString(), wxDefaultPosition, wxSize(100,200), wxTE_MULTILINE );
 893     m_logOld = wxLog::SetActiveTarget(new wxLogTextCtrl(m_log));
 894     wxLogMessage(_("This is the log window"));
 895
 896     bottom_sizer->Add( m_log, 1, wxGROW );
 897
 898     // wxDataViewListCtrl
 899
 900     wxDataViewListCtrl *listctrl = new wxDataViewListCtrl( panel, -1,
 901         wxDefaultPosition, wxSize(100,200) );
 902
 903     listctrl->AppendToggleColumn( wxT("Toggle") );
 904     listctrl->AppendTextColumn( wxT("Text") );
 905
 906     wxVector<wxVariant> data;
 907     data.push_back( true );
 908     data.push_back( "row 1" );
 909     listctrl->AppendItem( data );
 910
 911     data.clear();
 912     data.push_back( false );
 913     data.push_back( "row 3" );
 914     listctrl->AppendItem( data );
 915
 916     bottom_sizer->Add( listctrl, 1 );
 917
 918
 919     // wxDataViewTreeCtrl
 920
 921     wxDataViewTreeCtrl *treectrl2 = new wxDataViewTreeCtrl( panel, -1, wxDefaultPosition, wxSize(100,200) );
 922
 923     wxImageList *ilist = new wxImageList( 16, 16 );
 924     ilist->Add( wxIcon(small1_xpm) );
 925     treectrl2->SetImageList( ilist );
 926
 927     wxDataViewItem parent2 = treectrl2->AppendContainer( wxDataViewItem(0),wxT("Root 1"), 0 );
 928     wxDataViewItem child2 = treectrl2->AppendItem( parent2, wxT("Child 1"), 0 );
 929     child2 = treectrl2->AppendItem( parent2, wxT("Child 2"), 0 );
 930     child2 = treectrl2->AppendItem( parent2, wxT("Child 3, very long, long, long, long"), 0 );
 931
 932     bottom_sizer->Add( treectrl2, 1 );
 933
 934     // main sizer
 935
 936     main_sizer->Add( bottom_sizer, 0, wxGROW );
 937
 938     panel->SetSizer( main_sizer );
 939 }
 940
 941 MyFrame::~MyFrame()
 942 {
 943     delete wxLog::SetActiveTarget(m_logOld);
 944 }
 945
 946 void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event) )
 947 {
 948     Close(true);
 949 }
 950
 951 void MyFrame::OnAddMozart(wxCommandEvent& WXUNUSED(event) )
 952 {
 953     m_music_model->AddToClassical( wxT("Kleine Nachtmusik"), wxT("Wolfgang Mozart"), 1787 );
 954 }
 955
 956 void MyFrame::OnDeleteMusic(wxCommandEvent& WXUNUSED(event) )
 957 {
 958     wxDataViewItemArray items;
 959     int len = m_musicCtrl->GetSelections( items );
 960     for( int i = 0; i < len; i ++ )
 961         if (items[i].IsOk())
 962             m_music_model->Delete( items[i] );
 963 }
 964
 965 void MyFrame::OnDeleteYear( wxCommandEvent& WXUNUSED(event) )
 966 {
 967     m_musicCtrl->DeleteColumn( m_musicCtrl->GetColumn( 2 ) );
 968     FindWindow( ID_DELETE_YEAR )->Disable();
 969 }
 970
 971 void MyFrame::OnSelectNinth( wxCommandEvent& WXUNUSED(event) )
 972 {
 973     m_musicCtrl->Select( m_music_model->GetNinthItem() );
 974 }
 975
 976 void MyFrame::OnPrependList( wxCommandEvent& WXUNUSED(event) )
 977 {
 978     m_list_model->Prepend(wxT("Test"));
 979 }
 980
 981 void MyFrame::OnDeleteList( wxCommandEvent& WXUNUSED(event) )
 982 {
 983     wxDataViewItemArray items;
 984     int len = m_listCtrl->GetSelections( items );
 985     if (len > 0)
 986         m_list_model->DeleteItems( items );
 987 }
 988
 989 void MyFrame::OnValueChanged( wxDataViewEvent &event )
 990 {
 991     if (!m_log)
 992         return;
 993
 994     wxLogMessage( wxT("EVT_DATAVIEW_ITEM_VALUE_CHANGED, Item Id: %d;  Column: %d"), event.GetItem().GetID(), event.GetColumn() );
 995 }
 996
 997 void MyFrame::OnActivated( wxDataViewEvent &event )
 998 {
 999     if(!m_log)
1000         return;
1001
1002     wxString title = m_music_model->GetTitle( event.GetItem() );
1003     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_ITEM_ACTIVATED, Item: %s"), title );
1004
1005     if (m_musicCtrl->IsExpanded( event.GetItem() ))
1006     wxLogMessage(wxT("Item: %s is expanded"), title );
1007 }
1008
1009 void MyFrame::OnSelectionChanged( wxDataViewEvent &event )
1010 {
1011     if(!m_log)
1012         return;
1013
1014     wxString title = m_music_model->GetTitle( event.GetItem() );
1015     if (title.empty())
1016         title = wxT("None");
1017
1018     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_SELECTION_CHANGED, First selected Item: %s"), title );
1019 }
1020
1021 void MyFrame::OnExpanding( wxDataViewEvent &event )
1022 {
1023     if (!m_log)
1024         return;
1025
1026     wxString title = m_music_model->GetTitle( event.GetItem() );
1027     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_ITEM_EXPANDING, Item: %s"), title );
1028 }
1029
1030
1031 void MyFrame::OnEditingStarted( wxDataViewEvent &event )
1032 {
1033     if (!m_log)
1034         return;
1035
1036     wxString title = m_music_model->GetTitle( event.GetItem() );
1037     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_ITEM_EDITING_STARTED, Item: %s"), title );
1038 }
1039
1040 void MyFrame::OnEditingDone( wxDataViewEvent &event )
1041 {
1042     if (!m_log)
1043         return;
1044
1045     wxString title = m_music_model->GetTitle( event.GetItem() );
1046     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_ITEM_EDITING_DONE, Item: %s"), title );
1047 }
1048
1049 void MyFrame::OnExpanded( wxDataViewEvent &event )
1050 {
1051     if (!m_log)
1052         return;
1053
1054     wxString title = m_music_model->GetTitle( event.GetItem() );
1055     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_ITEM_EXPANDED, Item: %s"), title );
1056 }
1057
1058 void MyFrame::OnCollapsing( wxDataViewEvent &event )
1059 {
1060     if (!m_log)
1061         return;
1062
1063     wxString title = m_music_model->GetTitle( event.GetItem() );
1064     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_ITEM_COLLAPSING, Item: %s"), title );
1065 }
1066
1067 void MyFrame::OnCollapsed( wxDataViewEvent &event )
1068 {
1069     if (!m_log)
1070         return;
1071
1072     wxString title = m_music_model->GetTitle( event.GetItem() );
1073     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_ITEM_COLLAPSED, Item: %s"),title);
1074 }
1075
1076 void MyFrame::OnContextMenu( wxDataViewEvent &event )
1077 {
1078     if (!m_log)
1079         return;
1080
1081     wxString title = m_music_model->GetTitle( event.GetItem() );
1082     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_ITEM_CONTEXT_MENU, Item: %s"),title );
1083
1084     wxMenu menu;
1085     menu.Append( 1, wxT("entry 1") );
1086     menu.Append( 2, wxT("entry 2") );
1087     menu.Append( 3, wxT("entry 3") );
1088
1089     m_musicCtrl->PopupMenu(&menu);
1090
1091 //    wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_ITEM_CONTEXT_MENU, Item: %s Value: %s"),title.GetData(), event.GetValue().GetString());
1092 }
1093
1094 void MyFrame::OnHeaderClick( wxDataViewEvent &event )
1095 {
1096     // we need to skip the event to let the default behaviour of sorting by
1097     // this column when it is clicked to take place
1098     event.Skip();
1099
1100     if(!m_log)
1101         return;
1102
1103     int pos = m_musicCtrl->GetColumnPosition( event.GetDataViewColumn() );
1104
1105     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_CLICK, Column position: %d"), pos );
1106 }
1107
1108 void MyFrame::OnHeaderRightClick( wxDataViewEvent &event )
1109 {
1110     if(!m_log)
1111         return;
1112
1113     int pos = m_musicCtrl->GetColumnPosition( event.GetDataViewColumn() );
1114
1115     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_COLUMN_HEADER_RIGHT_CLICK, Column position: %d"), pos );
1116 }
1117
1118 void MyFrame::OnSorted( wxDataViewEvent &event )
1119 {
1120     if(!m_log)
1121         return;
1122
1123     int pos = m_musicCtrl->GetColumnPosition( event.GetDataViewColumn() );
1124
1125     wxLogMessage(wxT("wxEVT_COMMAND_DATAVIEW_COLUMN_SORTED, Column position: %d"), pos );
1126 }
1127
1128 void MyFrame::OnRightClick( wxMouseEvent &event )
1129 {
1130     if(!m_log)
1131         return;
1132
1133     wxLogMessage(wxT("wxEVT_MOUSE_RIGHT_UP, Click Point is X: %d, Y: %d"), event.GetX(), event.GetY());
1134 }
1135
1136 void MyFrame::OnGoto(wxCommandEvent& WXUNUSED(event))
1137 {
1138     wxDataViewItem item = m_list_model->GetItem( 50 );
1139     m_listCtrl->EnsureVisible(item,m_col);
1140 }
1141
1142 void MyFrame::OnAddMany(wxCommandEvent& WXUNUSED(event))
1143 {
1144     m_list_model->AddMany();
1145 }
1146
1147
1148 void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event) )
1149 {
1150     wxAboutDialogInfo info;
1151     info.SetName(_("DataView sample"));
1152     info.SetDescription(_("This sample demonstrates wxDataViewCtrl"));
1153     info.SetCopyright(_T("(C) 2007-2009 Robert Roebling"));
1154
1155     wxAboutBox(info);
1156 }
1157
1158 void MyFrame::OnBeginDrag( wxDataViewEvent &event )
1159 {
1160     wxDataViewItem item( event.GetItem() );
1161
1162     // only allow drags for item, not containers
1163     if (m_music_model->IsContainer( item ) )
1164     {
1165         event.Veto();
1166         return;
1167     }
1168
1169     MyMusicModelNode *node = (MyMusicModelNode*) item.GetID();
1170     wxTextDataObject *obj = new wxTextDataObject;
1171     obj->SetText( node->m_title );
1172     event.SetDataObject( obj );
1173 }