Add search_ctrl_height()
[dcpomatic.git] / src / wx / screens_panel.cc
1 /*
2     Copyright (C) 2015-2022 Carl Hetherington <cth@carlh.net>
3
4     This file is part of DCP-o-matic.
5
6     DCP-o-matic is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10
11     DCP-o-matic is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
18
19 */
20
21
22 #include "cinema_dialog.h"
23 #include "dcpomatic_button.h"
24 #include "screen_dialog.h"
25 #include "screens_panel.h"
26 #include "wx_util.h"
27 #include "lib/cinema.h"
28 #include "lib/config.h"
29 #include "lib/screen.h"
30
31
32 using std::cout;
33 using std::make_pair;
34 using std::make_shared;
35 using std::map;
36 using std::pair;
37 using std::shared_ptr;
38 using std::string;
39 using std::vector;
40 using boost::optional;
41 using namespace dcpomatic;
42
43
44 ScreensPanel::ScreensPanel (wxWindow* parent)
45         : wxPanel (parent, wxID_ANY)
46         , _ignore_selection_change (false)
47 {
48         auto sizer = new wxBoxSizer (wxVERTICAL);
49
50         _search = new wxSearchCtrl (this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize(200, search_ctrl_height()));
51 #ifndef __WXGTK3__
52         /* The cancel button seems to be strangely broken in GTK3; clicking on it twice sometimes works */
53         _search->ShowCancelButton (true);
54 #endif
55         sizer->Add (_search, 0, wxBOTTOM, DCPOMATIC_SIZER_GAP);
56
57         auto targets = new wxBoxSizer (wxHORIZONTAL);
58         _targets = new TreeListCtrl (this);
59         _targets->AppendColumn (wxT("foo"));
60         targets->Add (_targets, 1, wxEXPAND | wxRIGHT, DCPOMATIC_SIZER_GAP);
61
62         add_cinemas ();
63
64         auto target_buttons = new wxBoxSizer (wxVERTICAL);
65
66         _add_cinema = new Button (this, _("Add Cinema..."));
67         target_buttons->Add (_add_cinema, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
68         _edit_cinema = new Button (this, _("Edit Cinema..."));
69         target_buttons->Add (_edit_cinema, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
70         _remove_cinema = new Button (this, _("Remove Cinema"));
71         target_buttons->Add (_remove_cinema, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
72         _add_screen = new Button (this, _("Add Screen..."));
73         target_buttons->Add (_add_screen, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
74         _edit_screen = new Button (this, _("Edit Screen..."));
75         target_buttons->Add (_edit_screen, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
76         _remove_screen = new Button (this, _("Remove Screen"));
77         target_buttons->Add (_remove_screen, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
78
79         targets->Add (target_buttons, 0, 0);
80
81         sizer->Add (targets, 1, wxEXPAND);
82
83         _search->Bind        (wxEVT_TEXT, boost::bind (&ScreensPanel::search_changed, this));
84         _targets->Bind       (wxEVT_TREELIST_SELECTION_CHANGED, &ScreensPanel::selection_changed_shim, this);
85         _targets->Bind       (wxEVT_TREELIST_ITEM_CHECKED, &ScreensPanel::checkbox_changed, this);
86
87         _add_cinema->Bind    (wxEVT_BUTTON, boost::bind (&ScreensPanel::add_cinema_clicked, this));
88         _edit_cinema->Bind   (wxEVT_BUTTON, boost::bind (&ScreensPanel::edit_cinema_clicked, this));
89         _remove_cinema->Bind (wxEVT_BUTTON, boost::bind (&ScreensPanel::remove_cinema_clicked, this));
90
91         _add_screen->Bind    (wxEVT_BUTTON, boost::bind (&ScreensPanel::add_screen_clicked, this));
92         _edit_screen->Bind   (wxEVT_BUTTON, boost::bind (&ScreensPanel::edit_screen_clicked, this));
93         _remove_screen->Bind (wxEVT_BUTTON, boost::bind (&ScreensPanel::remove_screen_clicked, this));
94
95         SetSizer (sizer);
96 }
97
98
99 ScreensPanel::~ScreensPanel ()
100 {
101         _targets->Unbind (wxEVT_TREELIST_SELECTION_CHANGED, &ScreensPanel::selection_changed_shim, this);
102         _targets->Unbind (wxEVT_TREELIST_ITEM_CHECKED, &ScreensPanel::checkbox_changed, this);
103 }
104
105
106 void
107 ScreensPanel::setup_sensitivity ()
108 {
109         bool const sc = _selected_cinemas.size() == 1;
110         bool const ss = _selected_screens.size() == 1;
111
112         _edit_cinema->Enable (sc);
113         _remove_cinema->Enable (_selected_cinemas.size() >= 1);
114
115         _add_screen->Enable (sc);
116         _edit_screen->Enable (ss);
117         _remove_screen->Enable (_selected_screens.size() >= 1);
118 }
119
120
121 optional<wxTreeListItem>
122 ScreensPanel::add_cinema (shared_ptr<Cinema> c)
123 {
124         auto search = wx_to_std (_search->GetValue ());
125         transform (search.begin(), search.end(), search.begin(), ::tolower);
126
127         if (!search.empty ()) {
128                 auto name = c->name;
129                 transform (name.begin(), name.end(), name.begin(), ::tolower);
130                 if (name.find (search) == string::npos) {
131                         return {};
132                 }
133         }
134
135         auto id = _targets->AppendItem(_targets->GetRootItem(), std_to_wx(c->name));
136
137         _cinemas[id] = c;
138
139         for (auto i: c->screens()) {
140                 add_screen (c, i);
141         }
142
143         _targets->SetSortColumn (0);
144
145         return id;
146 }
147
148
149 optional<wxTreeListItem>
150 ScreensPanel::add_screen (shared_ptr<Cinema> c, shared_ptr<Screen> s)
151 {
152         auto i = _cinemas.begin();
153         while (i != _cinemas.end() && i->second != c) {
154                 ++i;
155         }
156
157         if (i == _cinemas.end()) {
158                 return {};
159         }
160
161         _screens[_targets->AppendItem (i->first, std_to_wx (s->name))] = s;
162         return i->first;
163 }
164
165
166 void
167 ScreensPanel::add_cinema_clicked ()
168 {
169         auto d = new CinemaDialog (GetParent(), _("Add Cinema"));
170         if (d->ShowModal () == wxID_OK) {
171                 auto c = make_shared<Cinema>(d->name(), d->emails(), d->notes(), d->utc_offset_hour(), d->utc_offset_minute());
172                 Config::instance()->add_cinema (c);
173                 auto id = add_cinema (c);
174                 if (id) {
175                         _targets->UnselectAll ();
176                         _targets->Select (*id);
177                 }
178         }
179
180         d->Destroy ();
181 }
182
183
184 void
185 ScreensPanel::edit_cinema_clicked ()
186 {
187         if (_selected_cinemas.size() != 1) {
188                 return;
189         }
190
191         auto c = *_selected_cinemas.begin();
192
193         auto d = new CinemaDialog (
194                 GetParent(), _("Edit cinema"), c.second->name, c.second->emails, c.second->notes, c.second->utc_offset_hour(), c.second->utc_offset_minute()
195                 );
196
197         if (d->ShowModal() == wxID_OK) {
198                 c.second->name = d->name ();
199                 c.second->emails = d->emails ();
200                 c.second->notes = d->notes ();
201                 c.second->set_utc_offset_hour (d->utc_offset_hour ());
202                 c.second->set_utc_offset_minute (d->utc_offset_minute ());
203                 _targets->SetItemText (c.first, std_to_wx (d->name()));
204                 Config::instance()->changed (Config::CINEMAS);
205         }
206
207         d->Destroy ();
208 }
209
210
211 void
212 ScreensPanel::remove_cinema_clicked ()
213 {
214         if (_selected_cinemas.size() == 1) {
215                 if (!confirm_dialog(this, wxString::Format(_("Are you sure you want to remove the cinema '%s'?"), std_to_wx(_selected_cinemas.begin()->second->name)))) {
216                         return;
217                 }
218         } else {
219                 if (!confirm_dialog(this, wxString::Format(_("Are you sure you want to remove %d cinemas?"), int(_selected_cinemas.size())))) {
220                         return;
221                 }
222         }
223
224         for (auto const& i: _selected_cinemas) {
225                 Config::instance()->remove_cinema (i.second);
226                 _targets->DeleteItem (i.first);
227         }
228
229         selection_changed ();
230 }
231
232
233 void
234 ScreensPanel::add_screen_clicked ()
235 {
236         if (_selected_cinemas.size() != 1) {
237                 return;
238         }
239
240         auto c = _selected_cinemas.begin()->second;
241
242         auto d = new ScreenDialog (GetParent(), _("Add Screen"));
243         if (d->ShowModal () != wxID_OK) {
244                 d->Destroy ();
245                 return;
246         }
247
248         for (auto i: c->screens ()) {
249                 if (i->name == d->name()) {
250                         error_dialog (
251                                 GetParent(),
252                                 wxString::Format (
253                                         _("You cannot add a screen called '%s' as the cinema already has a screen with this name."),
254                                         std_to_wx(d->name()).data()
255                                         )
256                                 );
257                         return;
258                 }
259         }
260
261         auto s = std::make_shared<Screen>(d->name(), d->notes(), d->recipient(), d->recipient_file(), d->trusted_devices());
262         c->add_screen (s);
263         auto id = add_screen (c, s);
264         if (id) {
265                 _targets->Expand (id.get ());
266         }
267
268         Config::instance()->changed (Config::CINEMAS);
269
270         d->Destroy ();
271 }
272
273
274 void
275 ScreensPanel::edit_screen_clicked ()
276 {
277         if (_selected_screens.size() != 1) {
278                 return;
279         }
280
281         auto s = *_selected_screens.begin();
282
283         auto d = new ScreenDialog (GetParent(), _("Edit screen"), s.second->name, s.second->notes, s.second->recipient, s.second->recipient_file, s.second->trusted_devices);
284         if (d->ShowModal() != wxID_OK) {
285                 d->Destroy ();
286                 return;
287         }
288
289         auto c = s.second->cinema;
290         for (auto i: c->screens ()) {
291                 if (i != s.second && i->name == d->name()) {
292                         error_dialog (
293                                 GetParent(),
294                                 wxString::Format (
295                                         _("You cannot change this screen's name to '%s' as the cinema already has a screen with this name."),
296                                         std_to_wx(d->name()).data()
297                                         )
298                                 );
299                         return;
300                 }
301         }
302
303         s.second->name = d->name ();
304         s.second->notes = d->notes ();
305         s.second->recipient = d->recipient ();
306         s.second->recipient_file = d->recipient_file ();
307         s.second->trusted_devices = d->trusted_devices ();
308         _targets->SetItemText (s.first, std_to_wx (d->name()));
309         Config::instance()->changed (Config::CINEMAS);
310
311         d->Destroy ();
312 }
313
314
315 void
316 ScreensPanel::remove_screen_clicked ()
317 {
318         if (_selected_screens.size() == 1) {
319                 if (!confirm_dialog(this, wxString::Format(_("Are you sure you want to remove the screen '%s'?"), std_to_wx(_selected_screens.begin()->second->name)))) {
320                         return;
321                 }
322         } else {
323                 if (!confirm_dialog(this, wxString::Format(_("Are you sure you want to remove %d screens?"), int(_selected_screens.size())))) {
324                         return;
325                 }
326         }
327
328         for (auto const& i: _selected_screens) {
329                 auto j = _cinemas.begin ();
330                 while (j != _cinemas.end ()) {
331                         auto sc = j->second->screens ();
332                         if (find (sc.begin(), sc.end(), i.second) != sc.end ()) {
333                                 break;
334                         }
335
336                         ++j;
337                 }
338
339                 if (j == _cinemas.end()) {
340                         continue;
341                 }
342
343                 j->second->remove_screen (i.second);
344                 _targets->DeleteItem (i.first);
345         }
346
347         Config::instance()->changed (Config::CINEMAS);
348 }
349
350
351 vector<shared_ptr<Screen>>
352 ScreensPanel::screens () const
353 {
354         vector<shared_ptr<Screen>> output;
355
356         for (auto item = _targets->GetFirstItem(); item.IsOk(); item = _targets->GetNextItem(item)) {
357                 if (_targets->GetCheckedState(item) == wxCHK_CHECKED) {
358                         auto screen_iter = _screens.find(item);
359                         if (screen_iter != _screens.end()) {
360                                 output.push_back (screen_iter->second);
361                         }
362                 }
363         }
364
365         return output;
366 }
367
368
369 void
370 ScreensPanel::selection_changed_shim (wxTreeListEvent &)
371 {
372         selection_changed ();
373 }
374
375
376 void
377 ScreensPanel::selection_changed ()
378 {
379         if (_ignore_selection_change) {
380                 return;
381         }
382
383         wxTreeListItems s;
384         _targets->GetSelections (s);
385
386         _selected_cinemas.clear ();
387         _selected_screens.clear ();
388
389         for (size_t i = 0; i < s.size(); ++i) {
390                 auto j = _cinemas.find (s[i]);
391                 if (j != _cinemas.end ()) {
392                         _selected_cinemas[j->first] = j->second;
393                 }
394                 auto k = _screens.find (s[i]);
395                 if (k != _screens.end ()) {
396                         _selected_screens[k->first] = k->second;
397                 }
398         }
399
400         setup_sensitivity ();
401 }
402
403
404 void
405 ScreensPanel::add_cinemas ()
406 {
407         for (auto i: Config::instance()->cinemas()) {
408                 add_cinema (i);
409         }
410 }
411
412
413 void
414 ScreensPanel::search_changed ()
415 {
416         _targets->DeleteAllItems ();
417         _cinemas.clear ();
418         _screens.clear ();
419
420         add_cinemas ();
421
422         _ignore_selection_change = true;
423
424         for (auto const& i: _selected_cinemas) {
425                 /* The wxTreeListItems will now be different, so we must search by cinema */
426                 auto j = _cinemas.begin ();
427                 while (j != _cinemas.end() && j->second != i.second) {
428                         ++j;
429                 }
430
431                 if (j != _cinemas.end()) {
432                         _targets->Select (j->first);
433                 }
434         }
435
436         for (auto const& i: _selected_screens) {
437                 auto j = _screens.begin ();
438                 while (j != _screens.end() && j->second != i.second) {
439                         ++j;
440                 }
441
442                 if (j != _screens.end()) {
443                         _targets->Select (j->first);
444                 }
445         }
446
447         _ignore_selection_change = false;
448 }
449
450
451 void
452 ScreensPanel::checkbox_changed (wxTreeListEvent& ev)
453 {
454         if (_cinemas.find(ev.GetItem()) != _cinemas.end()) {
455                 /* Cinema: check/uncheck all children */
456                 auto const checked = _targets->GetCheckedState(ev.GetItem());
457                 for (auto child = _targets->GetFirstChild(ev.GetItem()); child.IsOk(); child = _targets->GetNextSibling(child)) {
458                         _targets->CheckItem(child, checked);
459                 }
460         } else {
461                 /* Screen: set cinema to checked/unchecked/3state */
462                 auto parent = _targets->GetItemParent(ev.GetItem());
463                 DCPOMATIC_ASSERT (parent.IsOk());
464                 int checked = 0;
465                 int unchecked = 0;
466                 for (auto child = _targets->GetFirstChild(parent); child.IsOk(); child = _targets->GetNextSibling(child)) {
467                         if (_targets->GetCheckedState(child) == wxCHK_CHECKED) {
468                             ++checked;
469                         } else {
470                             ++unchecked;
471                         }
472                 }
473                 if (checked == 0) {
474                         _targets->CheckItem(parent, wxCHK_UNCHECKED);
475                 } else if (unchecked == 0) {
476                         _targets->CheckItem(parent, wxCHK_CHECKED);
477                 } else {
478                         _targets->CheckItem(parent, wxCHK_UNDETERMINED);
479                 }
480         }
481
482         ScreensChanged ();
483 }
484
485
486
487 wxIMPLEMENT_DYNAMIC_CLASS (TreeListCtrl, wxTreeListCtrl);
488
489
490 int
491 TreeListCtrl::OnCompareItems (wxTreeListItem const& a, wxTreeListItem const& b)
492 {
493         return strcoll (wx_to_std(GetItemText(a)).c_str(), wx_to_std(GetItemText(b)).c_str());
494 }
495
496