samples/dataview/dataview.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        dataview.cpp
   3 // Purpose:     DataVewCtrl  wxWidgets sample
   4 // Author:      Robert Roebling
   5 // Modified by:
   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
  25 #ifndef __WXMSW__
  26     #include "../sample.xpm"
  27 #endif
  28
  29 #include "null.xpm"
  30
  31 #include "wx/dataview.h"
  32
  33 // -------------------------------------
  34 // MyTextModel
  35 // -------------------------------------
  36
  37 WX_DECLARE_LIST(wxDateTime,wxArrayDate);
  38 #include <wx/listimpl.cpp>
  39 WX_DEFINE_LIST(wxArrayDate)
  40
  41 class MyTextModel: public wxDataViewListModel
  42 {
  43 public:
  44     MyTextModel()
  45     {
  46         size_t i;
  47         for (i = 0; i < 1000; i++)
  48             m_list.Add( wxT("Test") );
  49         for (i = 0; i < 500; i++)
  50             { m_bools.Add( 0 ); m_bools.Add( 1 ); }
  51         for (i = 0; i < 500; i++)
  52             { m_colours.Add( wxT("red") ); m_colours.Add( wxT("green") ); }
  53         for (i = 0; i < 1000; i++)
  54             { m_progress.Add( i/10 ); }
  55         for (i = 0; i < 1000; i++)
  56             {
  57                 wxDateTime *date = new wxDateTime( wxDateTime::Now() );
  58                 m_dates.Append( date );
  59             }
  60     }
  61
  62     virtual size_t GetNumberOfRows()
  63         { return 1000; }
  64     virtual size_t GetNumberOfCols()
  65         { return 7; }
  66
  67     // as reported by wxVariant
  68     virtual wxString GetColType( size_t col )
  69         {
  70             if (col == 6)
  71                 return wxT("datetime");
  72
  73             if (col == 5)
  74                 return wxT("long");
  75
  76             if (col == 3)
  77                 return wxT("bool");
  78
  79             return wxT("string");
  80         }
  81
  82     virtual void GetValue( wxVariant &variant, size_t col, size_t row )
  83         {
  84             if (col == 6)
  85             {
  86                 variant = (wxDateTime) *m_dates[row];
  87             } else
  88             if (col == 5)
  89             {
  90                 variant = (long) m_progress[row];
  91             } else
  92             if (col == 4)
  93             {
  94                 variant = m_colours[row];
  95             } else
  96             if (col == 3)
  97             {
  98                 variant = (bool) m_bools[row];
  99             } else
 100             if (col == 2)
 101             {
 102                 variant = m_list[row];
 103             }
 104             else
 105             {
 106                 wxString tmp;
 107                 tmp.Printf( wxT("item(%d;%d)"), (int)row, (int)col );
 108                 variant = tmp;
 109             }
 110         }
 111     virtual bool SetValue( wxVariant &value, size_t col, size_t row )
 112         {
 113             if (col == 6)
 114             {
 115                 *m_dates[row] = value.GetDateTime();
 116             } else
 117             if (col == 3)
 118             {
 119                 m_bools[row] = (int) value.GetBool();
 120             } else
 121             if (col == 2)
 122             {
 123                 m_list[row] = value.GetString();
 124             }
 125             return true;
 126         }
 127
 128     wxArrayString m_list;
 129     wxArrayInt    m_bools;
 130     wxArrayString m_colours;
 131     wxArrayInt    m_progress;
 132     wxArrayDate   m_dates;
 133 };
 134
 135 // -------------------------------------
 136 // MyCustomCell
 137 // -------------------------------------
 138
 139 class MyCustomCell: public wxDataViewCustomCell
 140 {
 141 public:
 142     MyCustomCell() :
 143         wxDataViewCustomCell( wxT("string"), wxDATAVIEW_CELL_ACTIVATABLE )
 144     {
 145         m_colour = wxT("black");
 146     }
 147     bool SetValue( const wxVariant &value )
 148     {
 149         m_colour = value.GetString();
 150         return true;
 151     }
 152     bool Render( wxRect rect, wxDC *dc, int WXUNUSED(state) )
 153     {
 154         dc->SetPen( *wxBLACK_PEN );
 155         if (m_colour == wxT("red"))
 156             dc->SetBrush( *wxRED_BRUSH );
 157         else if (m_colour == wxT("green"))
 158             dc->SetBrush( *wxGREEN_BRUSH );
 159         else
 160             dc->SetBrush( *wxBLACK_BRUSH );
 161         dc->DrawRectangle( rect );
 162         return true;
 163     }
 164     wxSize GetSize()
 165     {
 166         return wxSize(20,8);
 167     }
 168     bool Activate( wxRect WXUNUSED(rect),
 169                    wxDataViewListModel *WXUNUSED(model),
 170                    size_t WXUNUSED(col),
 171                    size_t WXUNUSED(row) )
 172     {
 173         return false;
 174     }
 175
 176 private:
 177     wxString m_colour;
 178 };
 179
 180 // -------------------------------------
 181 // MyUnsortedTextModel
 182 // -------------------------------------
 183
 184 class MyUnsortedTextModel: public wxDataViewListModel
 185 {
 186 public:
 187     MyUnsortedTextModel()
 188     {
 189         m_list.Add( wxT("This") );
 190         m_list.Add( wxT("is") );
 191         m_list.Add( wxT("an") );
 192         m_list.Add( wxT("unsorted") );
 193         m_list.Add( wxT("list") );
 194         m_list.Add( wxT("of") );
 195         m_list.Add( wxT("words.") );
 196
 197         m_bitmap = wxBitmap( null_xpm );
 198     }
 199
 200     virtual size_t GetNumberOfRows() { return m_list.GetCount(); }
 201     virtual size_t GetNumberOfCols() { return 2; }
 202     virtual wxString GetColType( size_t WXUNUSED(col) ) { return wxT("string"); }
 203     virtual void GetValue( wxVariant &variant, size_t col, size_t row )
 204     {
 205         if (col == 0)
 206         {
 207             variant = m_list[row];
 208             return;
 209         }
 210         if ((col == 2) || (col == 3))
 211         {
 212             variant << m_bitmap;
 213             return;
 214         }
 215         wxString tmp;
 216         tmp.Printf( wxT("item(%d;%d)"), (int)row, (int)col );
 217         variant = tmp;
 218     }
 219     virtual bool SetValue( wxVariant &variant, size_t col, size_t row )
 220     {
 221         if (col == 0)
 222         {
 223             m_list[row] = variant.GetString();
 224             return true;
 225         }
 226         return false;
 227
 228     }
 229
 230     void AppendRow( const wxString &text )
 231     {
 232         m_list.Add( text );
 233         RowAppended();
 234     }
 235
 236     void PrependRow( const wxString &text )
 237     {
 238         m_list.Insert( text, 0 );
 239         RowPrepended();
 240     }
 241
 242     void InsertRowAt1( const wxString &text )
 243     {
 244         m_list.Insert( text, 1 );
 245         RowInserted( 1 );
 246     }
 247
 248     void DeleteRow( size_t index )
 249     {
 250         m_list.RemoveAt( index );
 251         RowDeleted( index );
 252     }
 253
 254     wxArrayString m_list;
 255     wxBitmap m_bitmap;
 256 };
 257
 258 // -------------------------------------
 259 // MyApp
 260 // -------------------------------------
 261
 262 class MyApp: public wxApp
 263 {
 264 public:
 265     bool OnInit(void);
 266 };
 267
 268 // -------------------------------------
 269 // MyFrame
 270 // -------------------------------------
 271
 272 class MyFrame: public wxFrame
 273 {
 274 public:
 275     MyFrame(wxFrame *frame, wxChar *title, int x, int y, int w, int h);
 276
 277 public:
 278     void OnQuit(wxCommandEvent& event);
 279     void OnAbout(wxCommandEvent& event);
 280
 281 private:
 282     wxDataViewCtrl* dataview_left;
 283     wxDataViewCtrl* dataview_right;
 284 };
 285
 286 // -------------------------------------
 287 // MySortingFrame
 288 // -------------------------------------
 289
 290 enum my_events
 291 {
 292     ID_APPEND_ROW_LEFT = 1000,
 293     ID_PREPEND_ROW_LEFT,
 294     ID_INSERT_ROW_LEFT,
 295     ID_DELETE_ROW_LEFT,
 296     ID_EDIT_ROW_LEFT,
 297
 298     ID_APPEND_ROW_RIGHT,
 299     ID_PREPEND_ROW_RIGHT,
 300     ID_INSERT_ROW_RIGHT,
 301     ID_DELETE_ROW_RIGHT,
 302     ID_EDIT_ROW_RIGHT
 303 };
 304
 305 class MySortingFrame: public wxFrame
 306 {
 307 public:
 308     MySortingFrame(wxFrame *frame, wxChar *title, int x, int y, int w, int h);
 309
 310 public:
 311     void OnQuit(wxCommandEvent& event);
 312     void OnAbout(wxCommandEvent& event);
 313
 314     void OnAppendRowLeft(wxCommandEvent& event);
 315     void OnPrependRowLeft(wxCommandEvent& event);
 316     void OnInsertRowLeft(wxCommandEvent& event);
 317     void OnDeleteRowLeft(wxCommandEvent& event);
 318     void OnEditRowLeft(wxCommandEvent& event);
 319
 320     void OnAppendRowRight(wxCommandEvent& event);
 321     void OnPrependRowRight(wxCommandEvent& event);
 322     void OnInsertRowRight(wxCommandEvent& event);
 323     void OnDeleteRowRight(wxCommandEvent& event);
 324     void OnEditRowRight(wxCommandEvent& event);
 325
 326 private:
 327     wxDataViewCtrl* dataview_left;
 328     wxDataViewCtrl* dataview_right;
 329
 330     MyUnsortedTextModel *m_unsorted_model;
 331
 332     DECLARE_EVENT_TABLE()
 333 };
 334
 335 // -------------------------------------
 336 // MyApp
 337 // -------------------------------------
 338
 339 #define DYNAMIC_QUIT   wxID_EXIT
 340 #define DYNAMIC_ABOUT  wxID_ABOUT
 341
 342 IMPLEMENT_APP  (MyApp)
 343
 344 bool MyApp::OnInit(void)
 345 {
 346     MyFrame *frame = new MyFrame(NULL, wxT("wxDataViewCtrl feature test"), 10, 10, 800, 340);
 347     frame->Show(true);
 348
 349     MySortingFrame *frame2 = new MySortingFrame(NULL, wxT("wxDataViewCtrl sorting test"), 10, 350, 600, 300);
 350     frame2->Show(true);
 351
 352     SetTopWindow(frame);
 353
 354     return true;
 355 }
 356
 357 // -------------------------------------
 358 // MyFrame
 359 // -------------------------------------
 360
 361 MyFrame::MyFrame(wxFrame *frame, wxChar *title, int x, int y, int w, int h):
 362   wxFrame(frame, wxID_ANY, title, wxPoint(x, y), wxSize(w, h))
 363 {
 364     SetIcon(wxICON(sample));
 365
 366     wxMenu *file_menu = new wxMenu;
 367
 368     file_menu->Append(DYNAMIC_ABOUT, _T("&About"));
 369     file_menu->Append(DYNAMIC_QUIT, _T("E&xit"));
 370     wxMenuBar *menu_bar = new wxMenuBar;
 371     menu_bar->Append(file_menu, _T("&File"));
 372     SetMenuBar(menu_bar);
 373
 374     // You used to have to do some casting for param 4, but now there are type-safe handlers
 375     Connect( DYNAMIC_QUIT,  wxID_ANY,
 376                     wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(MyFrame::OnQuit) );
 377     Connect( DYNAMIC_ABOUT, wxID_ANY,
 378                     wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(MyFrame::OnAbout) );
 379
 380     CreateStatusBar();
 381
 382     wxPanel *panel = new wxPanel( this, wxID_ANY );
 383
 384
 385     // Left wxDataViewCtrl
 386     dataview_left = new wxDataViewCtrl( panel, wxID_ANY );
 387
 388     MyTextModel *model = new MyTextModel;
 389     dataview_left->AssociateModel( model );
 390
 391     dataview_left->AppendTextColumn( wxT("first"), 0 );
 392     dataview_left->AppendTextColumn( wxT("second"), 1 );
 393
 394     wxDataViewTextCell *text_cell = new wxDataViewTextCell( wxT("string"), wxDATAVIEW_CELL_EDITABLE );
 395     wxDataViewColumn *column = new wxDataViewColumn( wxT("editable"), text_cell, 2 );
 396     dataview_left->AppendColumn( column );
 397
 398     dataview_left->AppendToggleColumn( wxT("fourth"), 3 );
 399
 400     MyCustomCell *custom_cell = new MyCustomCell;
 401     column = new wxDataViewColumn( wxT("custom"), custom_cell, 4 );
 402     dataview_left->AppendColumn( column );
 403
 404     dataview_left->AppendProgressColumn( wxT("progress"), 5 );
 405
 406     dataview_left->AppendDateColumn( wxT("date"), 6 );
 407
 408     // Right wxDataViewCtrl using the same model
 409     dataview_right = new wxDataViewCtrl( panel, wxID_ANY );
 410     dataview_right->AssociateModel( model );
 411
 412     text_cell = new wxDataViewTextCell( wxT("string"), wxDATAVIEW_CELL_EDITABLE );
 413     column = new wxDataViewColumn( wxT("editable"), text_cell, 2 );
 414     dataview_right->AppendColumn( column );
 415     dataview_right->AppendTextColumn( wxT("first"), 0 );
 416     dataview_right->AppendTextColumn( wxT("second"), 1 );
 417     wxDataViewToggleCell *toggle_cell = new wxDataViewToggleCell( wxT("bool"), wxDATAVIEW_CELL_ACTIVATABLE );
 418     column = new wxDataViewColumn( wxT("bool"), toggle_cell, 3, 30 );
 419     dataview_right->AppendColumn( column );
 420
 421     dataview_right->AppendDateColumn( wxT("date"), 6 );
 422
 423     // layout dataview controls.
 424
 425     wxBoxSizer *sizer = new wxBoxSizer( wxHORIZONTAL );
 426     sizer->Add( dataview_left, 3, wxGROW );
 427     sizer->Add(10,10);
 428     sizer->Add( dataview_right, 2, wxGROW );
 429     panel->SetSizer( sizer );
 430 }
 431
 432 void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event) )
 433 {
 434     Close(true);
 435 }
 436
 437 void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event) )
 438 {
 439     wxMessageDialog dialog(this, _T("This demonstrates the dataview control handling"),
 440         _T("About DataView"), wxOK);
 441
 442     dialog.ShowModal();
 443 }
 444
 445 // -------------------------------------
 446 // MySortingFrame
 447 // -------------------------------------
 448
 449 BEGIN_EVENT_TABLE(MySortingFrame,wxFrame)
 450     EVT_BUTTON( ID_APPEND_ROW_LEFT, MySortingFrame::OnAppendRowLeft )
 451     EVT_BUTTON( ID_PREPEND_ROW_LEFT, MySortingFrame::OnPrependRowLeft )
 452     EVT_BUTTON( ID_INSERT_ROW_LEFT, MySortingFrame::OnInsertRowLeft )
 453     EVT_BUTTON( ID_DELETE_ROW_LEFT, MySortingFrame::OnDeleteRowLeft )
 454 END_EVENT_TABLE()
 455
 456 MySortingFrame::MySortingFrame(wxFrame *frame, wxChar *title, int x, int y, int w, int h):
 457   wxFrame(frame, wxID_ANY, title, wxPoint(x, y), wxSize(w, h))
 458 {
 459     SetIcon(wxICON(sample));
 460
 461     wxMenu *file_menu = new wxMenu;
 462
 463     file_menu->Append(DYNAMIC_ABOUT, _T("&About"));
 464     file_menu->Append(DYNAMIC_QUIT, _T("E&xit"));
 465     wxMenuBar *menu_bar = new wxMenuBar;
 466     menu_bar->Append(file_menu, _T("&File"));
 467     SetMenuBar(menu_bar);
 468
 469     // You used to have to do some casting for param 4, but now there are type-safe handlers
 470     Connect( DYNAMIC_QUIT,  wxID_ANY,
 471                     wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(MySortingFrame::OnQuit) );
 472     Connect( DYNAMIC_ABOUT, wxID_ANY,
 473                     wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(MySortingFrame::OnAbout) );
 474
 475     CreateStatusBar();
 476
 477
 478     // Left wxDataViewCtrl
 479     dataview_left = new wxDataViewCtrl( this, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxDV_MULTIPLE );
 480
 481     m_unsorted_model = new MyUnsortedTextModel;
 482     dataview_left->AssociateModel( m_unsorted_model );
 483     wxDataViewTextCell *text_cell = new wxDataViewTextCell( wxT("string"), wxDATAVIEW_CELL_EDITABLE );
 484     wxDataViewColumn *column = new wxDataViewColumn( wxT("editable"), text_cell, 0 );
 485     dataview_left->AppendColumn( column );
 486     dataview_left->AppendTextColumn( wxT("second"), 1 );
 487     dataview_left->AppendColumn( new wxDataViewColumn( wxT("icon"), new wxDataViewBitmapCell, 2, 25 ) );
 488     dataview_left->AppendColumn( new wxDataViewColumn( wxT("icon"), new wxDataViewBitmapCell, 3, 25 ) );
 489
 490     // Right wxDataViewCtrl using the sorting model
 491     dataview_right = new wxDataViewCtrl( this, wxID_ANY );
 492
 493     wxDataViewSortedListModel *sorted_model =
 494         new wxDataViewSortedListModel( m_unsorted_model );
 495     dataview_right->AssociateModel( sorted_model );
 496     text_cell = new wxDataViewTextCell( wxT("string"), wxDATAVIEW_CELL_EDITABLE );
 497     column = new wxDataViewColumn( wxT("editable"), text_cell, 0, -1, wxDATAVIEW_COL_SORTABLE|wxDATAVIEW_COL_RESIZABLE );
 498     dataview_right->AppendColumn( column );
 499
 500     dataview_right->AppendTextColumn( wxT("second"), 1 );
 501
 502     // layout dataview controls.
 503
 504     wxBoxSizer *top_sizer = new wxBoxSizer( wxHORIZONTAL );
 505     top_sizer->Add( dataview_left, 1, wxGROW );
 506     top_sizer->Add(10,10);
 507     top_sizer->Add( dataview_right, 1, wxGROW );
 508
 509     wxBoxSizer *button_sizer = new wxBoxSizer( wxHORIZONTAL );
 510     button_sizer->Add( 10, 10, 1 );
 511     wxFlexGridSizer *left_sizer = new wxFlexGridSizer( 2 );
 512     left_sizer->Add( new wxButton( this, ID_APPEND_ROW_LEFT, wxT("Append") ), 0, wxALL, 5 );
 513     left_sizer->Add( new wxButton( this, ID_PREPEND_ROW_LEFT, wxT("Prepend") ), 0, wxALL, 5 );
 514     left_sizer->Add( new wxButton( this, ID_INSERT_ROW_LEFT, wxT("Insert") ), 0, wxALL, 5 );
 515     left_sizer->Add( new wxButton( this, ID_DELETE_ROW_LEFT, wxT("Delete second") ), 0, wxALL, 5 );
 516     left_sizer->Add( new wxButton( this, ID_EDIT_ROW_LEFT, wxT("Edit") ), 0, wxALL, 5 );
 517     button_sizer->Add( left_sizer );
 518     button_sizer->Add( 10, 10, 2 );
 519     wxFlexGridSizer *right_sizer = new wxFlexGridSizer( 2 );
 520     right_sizer->Add( new wxButton( this, ID_APPEND_ROW_RIGHT, wxT("Append") ), 0, wxALL, 5 );
 521     right_sizer->Add( new wxButton( this, ID_PREPEND_ROW_RIGHT, wxT("Prepend") ), 0, wxALL, 5 );
 522     right_sizer->Add( new wxButton( this, ID_INSERT_ROW_RIGHT, wxT("Insert") ), 0, wxALL, 5 );
 523     right_sizer->Add( new wxButton( this, ID_DELETE_ROW_RIGHT, wxT("Delete second") ), 0, wxALL, 5 );
 524     right_sizer->Add( new wxButton( this, ID_EDIT_ROW_RIGHT, wxT("Edit") ), 0, wxALL, 5 );
 525     button_sizer->Add( right_sizer );
 526     button_sizer->Add( 10, 10, 1 );
 527
 528     wxBoxSizer *main_sizer = new wxBoxSizer( wxVERTICAL );
 529     main_sizer->Add( top_sizer, 1, wxGROW );
 530     main_sizer->Add( button_sizer, 0, wxGROW );
 531
 532     SetSizer( main_sizer );
 533 }
 534
 535 void MySortingFrame::OnQuit(wxCommandEvent& WXUNUSED(event) )
 536 {
 537     Close(true);
 538 }
 539
 540 void MySortingFrame::OnAbout(wxCommandEvent& WXUNUSED(event) )
 541 {
 542     wxMessageDialog dialog(this, _T("This demonstrates the dataview control sorting"),
 543         _T("About DataView"), wxOK);
 544
 545     dialog.ShowModal();
 546 }
 547
 548 void MySortingFrame::OnAppendRowLeft(wxCommandEvent& WXUNUSED(event))
 549 {
 550     wxTextEntryDialog dialog( this, wxT("Enter text to append") );
 551     if (dialog.ShowModal() == wxID_OK)
 552     {
 553         wxString value = dialog.GetValue();
 554         if (!value.empty())
 555             m_unsorted_model->AppendRow( value );
 556     }
 557 }
 558
 559 void MySortingFrame::OnPrependRowLeft(wxCommandEvent& WXUNUSED(event))
 560 {
 561     wxTextEntryDialog dialog( this, wxT("Enter text to prepend") );
 562     if (dialog.ShowModal() == wxID_OK)
 563     {
 564         wxString value = dialog.GetValue();
 565         if (!value.empty())
 566             m_unsorted_model->PrependRow( value );
 567     }
 568 }
 569
 570 void MySortingFrame::OnInsertRowLeft(wxCommandEvent& WXUNUSED(event))
 571 {
 572     wxTextEntryDialog dialog( this, wxT("Enter text to insert before second") );
 573     if (dialog.ShowModal() == wxID_OK)
 574     {
 575         wxString value = dialog.GetValue();
 576         if (!value.empty())
 577             m_unsorted_model->InsertRowAt1( value );
 578     }
 579 }
 580
 581 void MySortingFrame::OnDeleteRowLeft(wxCommandEvent& WXUNUSED(event))
 582 {
 583     m_unsorted_model->DeleteRow( 1 );
 584 }
 585
 586 void MySortingFrame::OnEditRowLeft(wxCommandEvent& WXUNUSED(event))
 587 {
 588 }
 589
 590 void MySortingFrame::OnAppendRowRight(wxCommandEvent& WXUNUSED(event))
 591 {
 592 }
 593
 594 void MySortingFrame::OnPrependRowRight(wxCommandEvent& WXUNUSED(event))
 595 {
 596 }
 597
 598 void MySortingFrame::OnInsertRowRight(wxCommandEvent& WXUNUSED(event))
 599 {
 600 }
 601
 602 void MySortingFrame::OnDeleteRowRight(wxCommandEvent& WXUNUSED(event))
 603 {
 604 }
 605
 606 void MySortingFrame::OnEditRowRight(wxCommandEvent& WXUNUSED(event))
 607 {
 608 }
 609