src/wx/editable_list.h

   1 /*
   2     Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #include <wx/wx.h>
  21 #include <wx/listctrl.h>
  22
  23 template<class T, class S>
  24 class EditableList : public wxPanel
  25 {
  26 public:
  27         EditableList (
  28                 wxWindow* parent,
  29                 std::vector<std::string> columns,
  30                 boost::function<std::vector<T> ()> get,
  31                 boost::function<void (std::vector<T>)> set,
  32                 boost::function<std::string (T, int)> column,
  33                 int height = 100
  34                 )
  35                 : wxPanel (parent)
  36                 , _get (get)
  37                 , _set (set)
  38                 , _columns (columns.size ())
  39                 , _column (column)
  40         {
  41                 wxBoxSizer* s = new wxBoxSizer (wxVERTICAL);
  42                 SetSizer (s);
  43
  44                 wxFlexGridSizer* table = new wxFlexGridSizer (2, DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP);
  45                 table->AddGrowableCol (0, 1);
  46                 s->Add (table, 1, wxEXPAND);
  47
  48                 _list = new wxListCtrl (this, wxID_ANY, wxDefaultPosition, wxSize (columns.size() * 200, height), wxLC_REPORT | wxLC_SINGLE_SEL);
  49
  50                 for (size_t i = 0; i < columns.size(); ++i) {
  51                         wxListItem ip;
  52                         ip.SetId (i);
  53                         ip.SetText (std_to_wx (columns[i]));
  54                         ip.SetWidth (200);
  55                         _list->InsertColumn (i, ip);
  56                 }
  57
  58                 table->Add (_list, 1, wxEXPAND | wxALL);
  59
  60                 {
  61                         wxSizer* s = new wxBoxSizer (wxVERTICAL);
  62                         _add = new wxButton (this, wxID_ANY, _("Add..."));
  63                         s->Add (_add, 0, wxTOP | wxBOTTOM, 2);
  64                         _edit = new wxButton (this, wxID_ANY, _("Edit..."));
  65                         s->Add (_edit, 0, wxTOP | wxBOTTOM, 2);
  66                         _remove = new wxButton (this, wxID_ANY, _("Remove"));
  67                         s->Add (_remove, 0, wxTOP | wxBOTTOM, 2);
  68                         table->Add (s, 0);
  69                 }
  70
  71                 std::vector<T> current = _get ();
  72                 for (typename std::vector<T>::iterator i = current.begin (); i != current.end(); ++i) {
  73                         add_to_control (*i);
  74                 }
  75
  76                 _add->Bind (wxEVT_COMMAND_BUTTON_CLICKED, boost::bind (&EditableList::add_clicked, this));
  77                 _edit->Bind (wxEVT_COMMAND_BUTTON_CLICKED, boost::bind (&EditableList::edit_clicked, this));
  78                 _remove->Bind (wxEVT_COMMAND_BUTTON_CLICKED, boost::bind (&EditableList::remove_clicked, this));
  79
  80                 _list->Bind (wxEVT_COMMAND_LIST_ITEM_SELECTED, boost::bind (&EditableList::selection_changed, this));
  81                 _list->Bind (wxEVT_COMMAND_LIST_ITEM_DESELECTED, boost::bind (&EditableList::selection_changed, this));
  82                 _list->Bind (wxEVT_SIZE, boost::bind (&EditableList::resized, this, _1));
  83                 selection_changed ();
  84
  85         }
  86
  87 private:
  88
  89         void add_to_control (T item)
  90         {
  91                 wxListItem list_item;
  92                 int const n = _list->GetItemCount ();
  93                 list_item.SetId (n);
  94                 _list->InsertItem (list_item);
  95
  96                 for (int i = 0; i < _columns; ++i) {
  97                         _list->SetItem (n, i, std_to_wx (_column (item, i)));
  98                 }
  99         }
 100
 101         void selection_changed ()
 102         {
 103                 int const i = _list->GetNextItem (-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
 104                 _edit->Enable (i >= 0);
 105                 _remove->Enable (i >= 0);
 106         }
 107
 108         void add_clicked ()
 109         {
 110                 T new_item;
 111                 S* dialog = new S (this);
 112                 dialog->set (new_item);
 113                 dialog->ShowModal ();
 114
 115                 add_to_control (dialog->get ());
 116
 117                 std::vector<T> all = _get ();
 118                 all.push_back (dialog->get ());
 119                 _set (all);
 120
 121                 dialog->Destroy ();
 122         }
 123
 124         void edit_clicked ()
 125         {
 126                 int item = _list->GetNextItem (-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
 127                 if (item == -1) {
 128                         return;
 129                 }
 130
 131                 std::vector<T> all = _get ();
 132                 assert (item >= 0 && item < int (all.size ()));
 133
 134                 S* dialog = new S (this);
 135                 dialog->set (all[item]);
 136                 dialog->ShowModal ();
 137                 all[item] = dialog->get ();
 138                 dialog->Destroy ();
 139
 140                 for (int i = 0; i < _columns; ++i) {
 141                         _list->SetItem (item, i, std_to_wx (_column (all[item], i)));
 142                 }
 143
 144                 _set (all);
 145         }
 146
 147         void remove_clicked ()
 148         {
 149                 int i = _list->GetNextItem (-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
 150                 if (i == -1) {
 151                         return;
 152                 }
 153
 154                 _list->DeleteItem (i);
 155                 std::vector<T> all = _get ();
 156                 all.erase (all.begin() + i);
 157                 _set (all);
 158
 159                 selection_changed ();
 160         }
 161
 162         void resized (wxSizeEvent& ev)
 163         {
 164                 int const w = GetSize().GetWidth() / _columns;
 165                 for (int i = 0; i < _columns; ++i) {
 166                         _list->SetColumnWidth (i, w);
 167                 }
 168                 ev.Skip ();
 169         }
 170
 171         boost::function <std::vector<T> ()> _get;
 172         boost::function <void (std::vector<T>)> _set;
 173         int _columns;
 174         boost::function<std::string (T, int)> _column;
 175
 176         wxButton* _add;
 177         wxButton* _edit;
 178         wxButton* _remove;
 179         wxListCtrl* _list;
 180 };