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 "mondrian.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 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 rect,
 167                    wxDataViewListModel *model, size_t col, size_t row )
 168     {
 169         return false;
 170     }
 171
 172 private:
 173     wxString m_colour;
 174 };
 175
 176 // -------------------------------------
 177 // MyUnsortedTextModel
 178 // -------------------------------------
 179
 180 class MyUnsortedTextModel: public wxDataViewListModel
 181 {
 182 public:
 183     MyUnsortedTextModel()
 184     {
 185         m_list.Add( wxT("This") );
 186         m_list.Add( wxT("is") );
 187         m_list.Add( wxT("an") );
 188         m_list.Add( wxT("unsorted") );
 189         m_list.Add( wxT("list") );
 190         m_list.Add( wxT("of") );
 191         m_list.Add( wxT("words.") );
 192     }
 193
 194     virtual size_t GetNumberOfRows() { return m_list.GetCount(); }
 195     virtual size_t GetNumberOfCols() { return 2; }
 196     virtual wxString GetColType( size_t col ) { return wxT("string"); }
 197     virtual void GetValue( wxVariant &variant, size_t col, size_t row )
 198     {
 199         if (col == 0)
 200         {
 201             variant = m_list[row];
 202             return;
 203         }
 204         wxString tmp;
 205         tmp.Printf( wxT("item(%d;%d)"), (int)row, (int)col );
 206         variant = tmp;
 207     }
 208     virtual bool SetValue( wxVariant &variant, size_t col, size_t row )
 209     {
 210         if (col == 0)
 211         {
 212             m_list[row] = variant.GetString();
 213             return true;
 214         }
 215         return false;
 216
 217     }
 218
 219     wxArrayString m_list;
 220 };
 221
 222 // -------------------------------------
 223 // MyApp
 224 // -------------------------------------
 225
 226 class MyApp: public wxApp
 227 {
 228 public:
 229     bool OnInit(void);
 230 };
 231
 232 // -------------------------------------
 233 // MyFrame
 234 // -------------------------------------
 235
 236 class MyFrame: public wxFrame
 237 {
 238 public:
 239     MyFrame(wxFrame *frame, wxChar *title, int x, int y, int w, int h);
 240
 241 public:
 242     void OnQuit(wxCommandEvent& event);
 243     void OnAbout(wxCommandEvent& event);
 244
 245 private:
 246     wxDataViewCtrl* dataview_left;
 247     wxDataViewCtrl* dataview_right;
 248 };
 249
 250 // -------------------------------------
 251 // MySortingFrame
 252 // -------------------------------------
 253
 254 enum my_events
 255 {
 256     ID_APPEND_ROW_LEFT = 1000,
 257     ID_PREPEND_ROW_LEFT,
 258     ID_INSERT_ROW_LEFT,
 259     ID_DELETE_ROW_LEFT,
 260     ID_EDIT_ROW_LEFT,
 261
 262     ID_APPEND_ROW_RIGHT,
 263     ID_PREPEND_ROW_RIGHT,
 264     ID_INSERT_ROW_RIGHT,
 265     ID_DELETE_ROW_RIGHT,
 266     ID_EDIT_ROW_RIGHT
 267 };
 268
 269 class MySortingFrame: public wxFrame
 270 {
 271 public:
 272     MySortingFrame(wxFrame *frame, wxChar *title, int x, int y, int w, int h);
 273
 274 public:
 275     void OnQuit(wxCommandEvent& event);
 276     void OnAbout(wxCommandEvent& event);
 277
 278     void OnAppendRowLeft(wxCommandEvent& event);
 279     void OnPrependRowLeft(wxCommandEvent& event);
 280     void OnInsertRowLeft(wxCommandEvent& event);
 281     void OnDeleteRowLeft(wxCommandEvent& event);
 282     void OnEditRowLeft(wxCommandEvent& event);
 283
 284     void OnAppendRowRight(wxCommandEvent& event);
 285     void OnPrependRowRight(wxCommandEvent& event);
 286     void OnInsertRowRight(wxCommandEvent& event);
 287     void OnDeleteRowRight(wxCommandEvent& event);
 288     void OnEditRowRight(wxCommandEvent& event);
 289
 290 private:
 291     wxDataViewCtrl* dataview_left;
 292     wxDataViewCtrl* dataview_right;
 293
 294     DECLARE_EVENT_TABLE()
 295 };
 296
 297 // -------------------------------------
 298 // MyApp
 299 // -------------------------------------
 300
 301 #define DYNAMIC_QUIT   wxID_EXIT
 302 #define DYNAMIC_ABOUT  wxID_ABOUT
 303
 304 IMPLEMENT_APP  (MyApp)
 305
 306 bool MyApp::OnInit(void)
 307 {
 308     MyFrame *frame = new MyFrame(NULL, wxT("wxDataViewCtrl feature test"), 10, 10, 800, 340);
 309     frame->Show(true);
 310
 311     MySortingFrame *frame2 = new MySortingFrame(NULL, wxT("wxDataViewCtrl sorting test"), 10, 350, 600, 300);
 312     frame2->Show(true);
 313
 314     SetTopWindow(frame);
 315
 316     return true;
 317 }
 318
 319 // -------------------------------------
 320 // MyFrame
 321 // -------------------------------------
 322
 323 MyFrame::MyFrame(wxFrame *frame, wxChar *title, int x, int y, int w, int h):
 324   wxFrame(frame, wxID_ANY, title, wxPoint(x, y), wxSize(w, h))
 325 {
 326 #ifdef __WXMSW__
 327     SetIcon(wxIcon(_T("mondrian")));
 328 #else
 329     SetIcon(wxIcon(mondrian_xpm));
 330 #endif
 331
 332     wxMenu *file_menu = new wxMenu;
 333
 334     file_menu->Append(DYNAMIC_ABOUT, _T("&About"));
 335     file_menu->Append(DYNAMIC_QUIT, _T("E&xit"));
 336     wxMenuBar *menu_bar = new wxMenuBar;
 337     menu_bar->Append(file_menu, _T("&File"));
 338     SetMenuBar(menu_bar);
 339
 340     // You used to have to do some casting for param 4, but now there are type-safe handlers
 341     Connect( DYNAMIC_QUIT,  wxID_ANY,
 342                     wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(MyFrame::OnQuit) );
 343     Connect( DYNAMIC_ABOUT, wxID_ANY,
 344                     wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(MyFrame::OnAbout) );
 345
 346     CreateStatusBar();
 347
 348
 349     // Left wxDataViewCtrl
 350     dataview_left = new wxDataViewCtrl( this, -1 );
 351
 352     MyTextModel *model = new MyTextModel;
 353     dataview_left->AssociateModel( model );
 354
 355     dataview_left->AppendTextColumn( wxT("first"), 0 );
 356     dataview_left->AppendTextColumn( wxT("second"), 1 );
 357
 358     wxDataViewTextCell *text_cell = new wxDataViewTextCell( wxT("string"), wxDATAVIEW_CELL_EDITABLE );
 359     wxDataViewColumn *column = new wxDataViewColumn( wxT("editable"), text_cell, 2 );
 360     dataview_left->AppendColumn( column );
 361
 362     dataview_left->AppendToggleColumn( wxT("fourth"), 3 );
 363
 364     MyCustomCell *custom_cell = new MyCustomCell;
 365     column = new wxDataViewColumn( wxT("custom"), custom_cell, 4 );
 366     dataview_left->AppendColumn( column );
 367
 368     dataview_left->AppendProgressColumn( wxT("progress"), 5 );
 369
 370     dataview_left->AppendDateColumn( wxT("date"), 6 );
 371
 372     // Right wxDataViewCtrl using the same model
 373     dataview_right = new wxDataViewCtrl( this, -1 );
 374     dataview_right->AssociateModel( model );
 375
 376     text_cell = new wxDataViewTextCell( wxT("string"), wxDATAVIEW_CELL_EDITABLE );
 377     column = new wxDataViewColumn( wxT("editable"), text_cell, 2 );
 378     dataview_right->AppendColumn( column );
 379     dataview_right->AppendTextColumn( wxT("first"), 0 );
 380     dataview_right->AppendTextColumn( wxT("second"), 1 );
 381     wxDataViewToggleCell *toggle_cell = new wxDataViewToggleCell( wxT("bool"), wxDATAVIEW_CELL_ACTIVATABLE );
 382     column = new wxDataViewColumn( wxT("bool"), toggle_cell, 3 );
 383     dataview_right->AppendColumn( column );
 384
 385     dataview_right->AppendDateColumn( wxT("date"), 6 );
 386
 387     // layout dataview controls.
 388
 389     wxBoxSizer *sizer = new wxBoxSizer( wxHORIZONTAL );
 390     sizer->Add( dataview_left, 3, wxGROW );
 391     sizer->Add(10,10);
 392     sizer->Add( dataview_right, 2, wxGROW );
 393     SetSizer( sizer );
 394 }
 395
 396 void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event) )
 397 {
 398     Close(true);
 399 }
 400
 401 void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event) )
 402 {
 403     wxMessageDialog dialog(this, _T("This demonstrates the dataview control handling"),
 404         _T("About DataView"), wxOK);
 405
 406     dialog.ShowModal();
 407 }
 408
 409 // -------------------------------------
 410 // MySortingFrame
 411 // -------------------------------------
 412
 413 BEGIN_EVENT_TABLE(MySortingFrame,wxFrame)
 414     EVT_BUTTON( ID_APPEND_ROW_LEFT, MySortingFrame::OnAppendRowLeft )
 415 END_EVENT_TABLE()
 416
 417 MySortingFrame::MySortingFrame(wxFrame *frame, wxChar *title, int x, int y, int w, int h):
 418   wxFrame(frame, wxID_ANY, title, wxPoint(x, y), wxSize(w, h))
 419 {
 420 #ifdef __WXMSW__
 421     SetIcon(wxIcon(_T("mondrian")));
 422 #else
 423     SetIcon(wxIcon(mondrian_xpm));
 424 #endif
 425
 426     wxMenu *file_menu = new wxMenu;
 427
 428     file_menu->Append(DYNAMIC_ABOUT, _T("&About"));
 429     file_menu->Append(DYNAMIC_QUIT, _T("E&xit"));
 430     wxMenuBar *menu_bar = new wxMenuBar;
 431     menu_bar->Append(file_menu, _T("&File"));
 432     SetMenuBar(menu_bar);
 433
 434     // You used to have to do some casting for param 4, but now there are type-safe handlers
 435     Connect( DYNAMIC_QUIT,  wxID_ANY,
 436                     wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(MySortingFrame::OnQuit) );
 437     Connect( DYNAMIC_ABOUT, wxID_ANY,
 438                     wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(MySortingFrame::OnAbout) );
 439
 440     CreateStatusBar();
 441
 442
 443     // Left wxDataViewCtrl
 444     dataview_left = new wxDataViewCtrl( this, -1 );
 445
 446     MyUnsortedTextModel *model = new MyUnsortedTextModel;
 447     dataview_left->AssociateModel( model );
 448     wxDataViewTextCell *text_cell = new wxDataViewTextCell( wxT("string"), wxDATAVIEW_CELL_EDITABLE );
 449     wxDataViewColumn *column = new wxDataViewColumn( wxT("editable"), text_cell, 0 );
 450     dataview_left->AppendColumn( column );
 451     dataview_left->AppendTextColumn( wxT("second"), 1 );
 452
 453     // Right wxDataViewCtrl using the sorting model
 454     dataview_right = new wxDataViewCtrl( this, -1 );
 455     wxDataViewSortedListModel *sorted_model =
 456         new wxDataViewSortedListModel( model );
 457     dataview_right->AssociateModel( sorted_model );
 458     text_cell = new wxDataViewTextCell( wxT("string"), wxDATAVIEW_CELL_EDITABLE );
 459     column = new wxDataViewColumn( wxT("editable"), text_cell, 0 );
 460     dataview_right->AppendColumn( column );
 461     dataview_right->AppendTextColumn( wxT("second"), 1 );
 462
 463     // layout dataview controls.
 464
 465     wxBoxSizer *top_sizer = new wxBoxSizer( wxHORIZONTAL );
 466     top_sizer->Add( dataview_left, 1, wxGROW );
 467     top_sizer->Add(10,10);
 468     top_sizer->Add( dataview_right, 1, wxGROW );
 469
 470     wxBoxSizer *button_sizer = new wxBoxSizer( wxHORIZONTAL );
 471     button_sizer->Add( 10, 10, 1 );
 472     wxFlexGridSizer *left_sizer = new wxFlexGridSizer( 2 );
 473     left_sizer->Add( new wxButton( this, ID_APPEND_ROW_LEFT, wxT("Append") ), 0, wxALL, 5 );
 474     left_sizer->Add( new wxButton( this, ID_PREPEND_ROW_LEFT, wxT("Prepend") ), 0, wxALL, 5 );
 475     left_sizer->Add( new wxButton( this, ID_INSERT_ROW_LEFT, wxT("Insert") ), 0, wxALL, 5 );
 476     left_sizer->Add( new wxButton( this, ID_DELETE_ROW_LEFT, wxT("Delete") ), 0, wxALL, 5 );
 477     left_sizer->Add( new wxButton( this, ID_EDIT_ROW_LEFT, wxT("Edit") ), 0, wxALL, 5 );
 478     button_sizer->Add( left_sizer );
 479     button_sizer->Add( 10, 10, 2 );
 480     wxFlexGridSizer *right_sizer = new wxFlexGridSizer( 2 );
 481     right_sizer->Add( new wxButton( this, ID_APPEND_ROW_RIGHT, wxT("Append") ), 0, wxALL, 5 );
 482     right_sizer->Add( new wxButton( this, ID_PREPEND_ROW_RIGHT, wxT("Prepend") ), 0, wxALL, 5 );
 483     right_sizer->Add( new wxButton( this, ID_INSERT_ROW_RIGHT, wxT("Insert") ), 0, wxALL, 5 );
 484     right_sizer->Add( new wxButton( this, ID_DELETE_ROW_RIGHT, wxT("Delete") ), 0, wxALL, 5 );
 485     right_sizer->Add( new wxButton( this, ID_EDIT_ROW_RIGHT, wxT("Edit") ), 0, wxALL, 5 );
 486     button_sizer->Add( right_sizer );
 487     button_sizer->Add( 10, 10, 1 );
 488
 489     wxBoxSizer *main_sizer = new wxBoxSizer( wxVERTICAL );
 490     main_sizer->Add( top_sizer, 1, wxGROW );
 491     main_sizer->Add( button_sizer, 0, wxGROW );
 492
 493     SetSizer( main_sizer );
 494 }
 495
 496 void MySortingFrame::OnQuit(wxCommandEvent& WXUNUSED(event) )
 497 {
 498     Close(true);
 499 }
 500
 501 void MySortingFrame::OnAbout(wxCommandEvent& WXUNUSED(event) )
 502 {
 503     wxMessageDialog dialog(this, _T("This demonstrates the dataview control sorting"),
 504         _T("About DataView"), wxOK);
 505
 506     dialog.ShowModal();
 507 }
 508
 509 void MySortingFrame::OnAppendRowLeft(wxCommandEvent& event)
 510 {
 511 }
 512
 513 void MySortingFrame::OnPrependRowLeft(wxCommandEvent& event)
 514 {
 515 }
 516
 517 void MySortingFrame::OnInsertRowLeft(wxCommandEvent& event)
 518 {
 519 }
 520
 521 void MySortingFrame::OnDeleteRowLeft(wxCommandEvent& event)
 522 {
 523 }
 524
 525 void MySortingFrame::OnEditRowLeft(wxCommandEvent& event)
 526 {
 527 }
 528
 529 void MySortingFrame::OnAppendRowRight(wxCommandEvent& event)
 530 {
 531 }
 532
 533 void MySortingFrame::OnPrependRowRight(wxCommandEvent& event)
 534 {
 535 }
 536
 537 void MySortingFrame::OnInsertRowRight(wxCommandEvent& event)
 538 {
 539 }
 540
 541 void MySortingFrame::OnDeleteRowRight(wxCommandEvent& event)
 542 {
 543 }
 544
 545 void MySortingFrame::OnEditRowRight(wxCommandEvent& event)
 546 {
 547 }
 548