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