Merge 1.0 in.
[dcpomatic.git] / 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
22 template<class T, class S>
23 class EditableList : public wxPanel
24 {
25 public:
26         EditableList (
27                 wxWindow* parent,
28                 std::vector<std::string> columns,
29                 boost::function<std::vector<T> ()> get,
30                 boost::function<void (std::vector<T>)> set,
31                 boost::function<std::string (T, int)> column
32                 )
33                 : wxPanel (parent)
34                 , _get (get)
35                 , _set (set)
36                 , _columns (columns.size ())
37                 , _column (column)
38         {
39                 wxBoxSizer* s = new wxBoxSizer (wxVERTICAL);
40                 SetSizer (s);
41
42                 wxFlexGridSizer* table = new wxFlexGridSizer (2, DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP);
43                 table->AddGrowableCol (0, 1);
44                 s->Add (table, 1, wxALL | wxEXPAND, 8);
45
46                 _list = new wxListCtrl (this, wxID_ANY, wxDefaultPosition, wxSize (columns.size() * 200, 100), wxLC_REPORT | wxLC_SINGLE_SEL);
47
48                 for (size_t i = 0; i < columns.size(); ++i) {
49                         wxListItem ip;
50                         ip.SetId (i);
51                         ip.SetText (std_to_wx (columns[i]));
52                         ip.SetWidth (200);
53                         _list->InsertColumn (i, ip);
54                 }
55
56                 table->Add (_list, 1, wxEXPAND | wxALL);
57
58                 {
59                         wxSizer* s = new wxBoxSizer (wxVERTICAL);
60                         _add = new wxButton (this, wxID_ANY, _("Add..."));
61                         s->Add (_add, 0, wxTOP | wxBOTTOM, 2);
62                         _edit = new wxButton (this, wxID_ANY, _("Edit..."));
63                         s->Add (_edit, 0, wxTOP | wxBOTTOM, 2);
64                         _remove = new wxButton (this, wxID_ANY, _("Remove"));
65                         s->Add (_remove, 0, wxTOP | wxBOTTOM, 2);
66                         table->Add (s, 0);
67                 }
68
69                 std::vector<T> current = _get ();
70                 for (typename std::vector<T>::iterator i = current.begin (); i != current.end(); ++i) {
71                         add_to_control (*i);
72                 }
73
74                 _add->Bind (wxEVT_COMMAND_BUTTON_CLICKED, boost::bind (&EditableList::add_clicked, this));
75                 _edit->Bind (wxEVT_COMMAND_BUTTON_CLICKED, boost::bind (&EditableList::edit_clicked, this));
76                 _remove->Bind (wxEVT_COMMAND_BUTTON_CLICKED, boost::bind (&EditableList::remove_clicked, this));
77
78                 _list->Bind (wxEVT_COMMAND_LIST_ITEM_SELECTED, boost::bind (&EditableList::selection_changed, this));
79                 _list->Bind (wxEVT_COMMAND_LIST_ITEM_DESELECTED, boost::bind (&EditableList::selection_changed, this));
80                 _list->Bind (wxEVT_SIZE, boost::bind (&EditableList::resized, this, _1));
81                 selection_changed ();
82
83         }
84
85 private:        
86
87         void add_to_control (T item)
88         {
89                 wxListItem list_item;
90                 int const n = _list->GetItemCount ();
91                 list_item.SetId (n);
92                 _list->InsertItem (list_item);
93
94                 for (int i = 0; i < _columns; ++i) {
95                         _list->SetItem (n, i, std_to_wx (_column (item, i)));
96                 }
97         }
98
99         void selection_changed ()
100         {
101                 int const i = _list->GetNextItem (-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
102                 _edit->Enable (i >= 0);
103                 _remove->Enable (i >= 0);
104         }
105
106         void add_clicked ()
107         {
108                 T new_item;
109                 S* dialog = new S (this);
110                 dialog->set (new_item);
111                 dialog->ShowModal ();
112
113                 add_to_control (dialog->get ());
114                 
115                 std::vector<T> all = _get ();
116                 all.push_back (dialog->get ());
117                 _set (all);
118                 
119                 dialog->Destroy ();
120         }
121
122         void edit_clicked ()
123         {
124                 int item = _list->GetNextItem (-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
125                 if (item == -1) {
126                         return;
127                 }
128
129                 std::vector<T> all = _get ();
130                 assert (item >= 0 && item < int (all.size ()));
131
132                 S* dialog = new S (this);
133                 dialog->set (all[item]);
134                 dialog->ShowModal ();
135                 all[item] = dialog->get ();
136                 dialog->Destroy ();
137                 
138                 for (int i = 0; i < _columns; ++i) {
139                         _list->SetItem (item, i, std_to_wx (_column (all[item], i)));
140                 }
141         }
142
143         void remove_clicked ()
144         {
145                 int i = _list->GetNextItem (-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
146                 if (i == -1) {
147                         return;
148                 }
149                 
150                 _list->DeleteItem (i);
151                 std::vector<T> all = _get ();
152                 all.erase (all.begin() + i);
153                 _set (all);
154
155                 selection_changed ();
156         }
157
158         void resized (wxSizeEvent& ev)
159         {
160                 int const w = GetSize().GetWidth() / _columns;
161                 for (int i = 0; i < _columns; ++i) {
162                         _list->SetColumnWidth (i, w);
163                 }
164                 ev.Skip ();
165         }
166
167         boost::function <std::vector<T> ()> _get;
168         boost::function <void (std::vector<T>)> _set;
169         int _columns;
170         boost::function<std::string (T, int)> _column;
171
172         wxButton* _add;
173         wxButton* _edit;
174         wxButton* _remove;
175         wxListCtrl* _list;
176 };