Expand cinema when adding a screen in the KDM window.
[dcpomatic.git] / src / wx / kdm_dialog.cc
1 /*
2     Copyright (C) 2012-2015 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 "kdm_dialog.h"
21 #include "cinema_dialog.h"
22 #include "screen_dialog.h"
23 #include "wx_util.h"
24 #include "lib/cinema.h"
25 #include "lib/config.h"
26 #include "lib/film.h"
27 #include <libcxml/cxml.h>
28 #ifdef DCPOMATIC_USE_OWN_DIR_PICKER
29 #include "dir_picker_ctrl.h"
30 #else
31 #include <wx/filepicker.h>
32 #endif
33 #include <wx/treectrl.h>
34 #include <wx/datectrl.h>
35 #include <wx/timectrl.h>
36 #include <wx/stdpaths.h>
37 #include <wx/listctrl.h>
38
39 using std::string;
40 using std::map;
41 using std::list;
42 using std::pair;
43 using std::cout;
44 using std::vector;
45 using std::make_pair;
46 using boost::shared_ptr;
47
48 KDMDialog::KDMDialog (wxWindow* parent, boost::shared_ptr<const Film> film)
49         : wxDialog (parent, wxID_ANY, _("Make KDMs"))
50 {
51         /* Main sizer */
52         wxBoxSizer* vertical = new wxBoxSizer (wxVERTICAL);
53
54         /* Font for sub-headings */
55         wxFont subheading_font (*wxNORMAL_FONT);
56         subheading_font.SetWeight (wxFONTWEIGHT_BOLD);
57
58
59         /* Sub-heading: Screens */
60         wxStaticText* h = new wxStaticText (this, wxID_ANY, _("Screens"));
61         h->SetFont (subheading_font);
62         vertical->Add (h, 0, wxALIGN_CENTER_VERTICAL);
63
64         wxBoxSizer* targets = new wxBoxSizer (wxHORIZONTAL);
65         _targets = new wxTreeCtrl (this, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTR_HIDE_ROOT | wxTR_MULTIPLE | wxTR_HAS_BUTTONS | wxTR_LINES_AT_ROOT);
66         targets->Add (_targets, 1, wxEXPAND | wxTOP | wxRIGHT, DCPOMATIC_SIZER_GAP);
67
68         _root = _targets->AddRoot ("Foo");
69
70         list<shared_ptr<Cinema> > c = Config::instance()->cinemas ();
71         for (list<shared_ptr<Cinema> >::iterator i = c.begin(); i != c.end(); ++i) {
72                 add_cinema (*i);
73         }
74
75         _targets->ExpandAll ();
76
77         wxBoxSizer* target_buttons = new wxBoxSizer (wxVERTICAL);
78
79         _add_cinema = new wxButton (this, wxID_ANY, _("Add Cinema..."));
80         target_buttons->Add (_add_cinema, 1, wxEXPAND | wxALL, DCPOMATIC_BUTTON_STACK_GAP);
81         _edit_cinema = new wxButton (this, wxID_ANY, _("Edit Cinema..."));
82         target_buttons->Add (_edit_cinema, 1, wxEXPAND | wxALL, DCPOMATIC_BUTTON_STACK_GAP);
83         _remove_cinema = new wxButton (this, wxID_ANY, _("Remove Cinema"));
84         target_buttons->Add (_remove_cinema, 1, wxEXPAND | wxALL, DCPOMATIC_BUTTON_STACK_GAP);
85
86         _add_screen = new wxButton (this, wxID_ANY, _("Add Screen..."));
87         target_buttons->Add (_add_screen, 1, wxEXPAND | wxALL, DCPOMATIC_BUTTON_STACK_GAP);
88         _edit_screen = new wxButton (this, wxID_ANY, _("Edit Screen..."));
89         target_buttons->Add (_edit_screen, 1, wxEXPAND | wxALL, DCPOMATIC_BUTTON_STACK_GAP);
90         _remove_screen = new wxButton (this, wxID_ANY, _("Remove Screen"));
91         target_buttons->Add (_remove_screen, 1, wxEXPAND | wxALL, DCPOMATIC_BUTTON_STACK_GAP);
92
93         targets->Add (target_buttons, 0, 0);
94
95         vertical->Add (targets, 1, wxEXPAND);
96
97
98         /* Sub-heading: Timing */
99         h = new wxStaticText (this, wxID_ANY, S_("KDM|Timing"));
100         h->SetFont (subheading_font);
101         vertical->Add (h, 0, wxALIGN_CENTER_VERTICAL | wxTOP, DCPOMATIC_SIZER_Y_GAP * 2);
102
103         wxFlexGridSizer* table = new wxFlexGridSizer (3, DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP);
104         add_label_to_sizer (table, this, _("From"), true);
105         wxDateTime from;
106         from.SetToCurrent ();
107         _from_date = new wxDatePickerCtrl (this, wxID_ANY, from);
108         table->Add (_from_date, 1, wxEXPAND);
109         _from_time = new wxTimePickerCtrl (this, wxID_ANY, from);
110         table->Add (_from_time, 1, wxEXPAND);
111
112         add_label_to_sizer (table, this, _("Until"), true);
113         wxDateTime to = from;
114         /* 1 week from now */
115         to.Add (wxDateSpan (0, 0, 1, 0));
116         _until_date = new wxDatePickerCtrl (this, wxID_ANY, to);
117         table->Add (_until_date, 1, wxEXPAND);
118         _until_time = new wxTimePickerCtrl (this, wxID_ANY, to);
119         table->Add (_until_time, 1, wxEXPAND);
120
121         vertical->Add (table, 0, wxEXPAND | wxTOP, DCPOMATIC_SIZER_GAP);
122
123
124         /* Sub-heading: CPL */
125         h = new wxStaticText (this, wxID_ANY, _("CPL"));
126         h->SetFont (subheading_font);
127         vertical->Add (h, 0, wxALIGN_CENTER_VERTICAL | wxTOP, DCPOMATIC_SIZER_Y_GAP * 2);
128
129         /* CPL choice */
130         wxBoxSizer* s = new wxBoxSizer (wxHORIZONTAL);
131         add_label_to_sizer (s, this, _("CPL"), true);
132         _cpl = new wxChoice (this, wxID_ANY);
133         s->Add (_cpl, 1, wxEXPAND);
134         _cpl_browse = new wxButton (this, wxID_ANY, _("Browse..."));
135         s->Add (_cpl_browse, 0);
136         vertical->Add (s, 0, wxEXPAND | wxTOP, DCPOMATIC_SIZER_GAP + 2);
137
138         /* CPL details */
139         table = new wxFlexGridSizer (2, DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP);
140         add_label_to_sizer (table, this, _("DCP directory"), true);
141         _dcp_directory = new wxStaticText (this, wxID_ANY, "");
142         table->Add (_dcp_directory);
143         add_label_to_sizer (table, this, _("CPL ID"), true);
144         _cpl_id = new wxStaticText (this, wxID_ANY, "");
145         table->Add (_cpl_id);
146         add_label_to_sizer (table, this, _("CPL annotation text"), true);
147         _cpl_annotation_text = new wxStaticText (this, wxID_ANY, "");
148         table->Add (_cpl_annotation_text);
149         vertical->Add (table, 0, wxEXPAND | wxTOP, DCPOMATIC_SIZER_GAP + 2);
150
151         _cpls = film->cpls ();
152         update_cpl_choice ();
153
154
155         /* Sub-heading: Output */
156         h = new wxStaticText (this, wxID_ANY, _("Output"));
157         h->SetFont (subheading_font);
158         vertical->Add (h, 0, wxALIGN_CENTER_VERTICAL | wxTOP, DCPOMATIC_SIZER_Y_GAP * 2);
159
160         table = new wxFlexGridSizer (2, DCPOMATIC_SIZER_X_GAP, 0);
161
162         add_label_to_sizer (table, this, _("KDM type"), true);
163         _type = new wxChoice (this, wxID_ANY);
164         _type->Append ("Modified Transitional 1", ((void *) dcp::MODIFIED_TRANSITIONAL_1));
165         if (!film->interop ()) {
166                 _type->Append ("DCI Any", ((void *) dcp::DCI_ANY));
167                 _type->Append ("DCI Specific", ((void *) dcp::DCI_SPECIFIC));
168         }
169         table->Add (_type, 1, wxEXPAND);
170         _type->SetSelection (0);
171
172         _write_to = new wxRadioButton (this, wxID_ANY, _("Write to"));
173         table->Add (_write_to, 1, wxEXPAND);
174
175 #ifdef DCPOMATIC_USE_OWN_DIR_PICKER
176         _folder = new DirPickerCtrl (this);
177 #else
178         _folder = new wxDirPickerCtrl (this, wxID_ANY, wxEmptyString, wxDirSelectorPromptStr, wxDefaultPosition, wxSize (300, -1));
179 #endif
180
181         _folder->SetPath (wxStandardPaths::Get().GetDocumentsDir());
182
183         table->Add (_folder, 1, wxEXPAND);
184
185         _email = new wxRadioButton (this, wxID_ANY, _("Send by email"));
186         table->Add (_email, 1, wxEXPAND);
187         table->AddSpacer (0);
188
189         vertical->Add (table, 0, wxEXPAND | wxTOP, DCPOMATIC_SIZER_GAP);
190
191         /* Make an overall sizer to get a nice border, and put some buttons in */
192
193         wxBoxSizer* overall_sizer = new wxBoxSizer (wxVERTICAL);
194         overall_sizer->Add (vertical, 0, wxEXPAND | wxTOP | wxLEFT | wxRIGHT, DCPOMATIC_DIALOG_BORDER);
195
196         wxSizer* buttons = CreateSeparatedButtonSizer (wxOK | wxCANCEL);
197         if (buttons) {
198                 overall_sizer->Add (buttons, 0, wxEXPAND | wxALL, DCPOMATIC_SIZER_Y_GAP);
199         }
200
201         _write_to->SetValue (true);
202
203         /* Bind */
204
205         _targets->Bind       (wxEVT_COMMAND_TREE_SEL_CHANGED, boost::bind (&KDMDialog::setup_sensitivity, this));
206
207         _add_cinema->Bind    (wxEVT_COMMAND_BUTTON_CLICKED, boost::bind (&KDMDialog::add_cinema_clicked, this));
208         _edit_cinema->Bind   (wxEVT_COMMAND_BUTTON_CLICKED, boost::bind (&KDMDialog::edit_cinema_clicked, this));
209         _remove_cinema->Bind (wxEVT_COMMAND_BUTTON_CLICKED, boost::bind (&KDMDialog::remove_cinema_clicked, this));
210
211         _add_screen->Bind    (wxEVT_COMMAND_BUTTON_CLICKED, boost::bind (&KDMDialog::add_screen_clicked, this));
212         _edit_screen->Bind   (wxEVT_COMMAND_BUTTON_CLICKED, boost::bind (&KDMDialog::edit_screen_clicked, this));
213         _remove_screen->Bind (wxEVT_COMMAND_BUTTON_CLICKED, boost::bind (&KDMDialog::remove_screen_clicked, this));
214
215         _cpl->Bind           (wxEVT_COMMAND_CHOICE_SELECTED, boost::bind (&KDMDialog::update_cpl_summary, this));
216         _cpl_browse->Bind    (wxEVT_COMMAND_BUTTON_CLICKED,  boost::bind (&KDMDialog::cpl_browse_clicked, this));
217
218         _write_to->Bind      (wxEVT_COMMAND_RADIOBUTTON_SELECTED, boost::bind (&KDMDialog::setup_sensitivity, this));
219         _email->Bind         (wxEVT_COMMAND_RADIOBUTTON_SELECTED, boost::bind (&KDMDialog::setup_sensitivity, this));
220
221         setup_sensitivity ();
222
223         SetSizer (overall_sizer);
224         overall_sizer->Layout ();
225         overall_sizer->SetSizeHints (this);
226 }
227
228 list<pair<wxTreeItemId, shared_ptr<Cinema> > >
229 KDMDialog::selected_cinemas () const
230 {
231         wxArrayTreeItemIds s;
232         _targets->GetSelections (s);
233
234         list<pair<wxTreeItemId, shared_ptr<Cinema> > > c;
235         for (size_t i = 0; i < s.GetCount(); ++i) {
236                 map<wxTreeItemId, shared_ptr<Cinema> >::const_iterator j = _cinemas.find (s[i]);
237                 if (j != _cinemas.end ()) {
238                         c.push_back (make_pair (j->first, j->second));
239                 }
240         }
241
242         return c;
243 }
244
245 list<pair<wxTreeItemId, shared_ptr<Screen> > >
246 KDMDialog::selected_screens () const
247 {
248         wxArrayTreeItemIds s;
249         _targets->GetSelections (s);
250
251         list<pair<wxTreeItemId, shared_ptr<Screen> > > c;
252         for (size_t i = 0; i < s.GetCount(); ++i) {
253                 map<wxTreeItemId, shared_ptr<Screen> >::const_iterator j = _screens.find (s[i]);
254                 if (j != _screens.end ()) {
255                         c.push_back (make_pair (j->first, j->second));
256                 }
257         }
258
259         return c;
260 }
261
262 void
263 KDMDialog::setup_sensitivity ()
264 {
265         bool const sc = selected_cinemas().size() == 1;
266         bool const ss = selected_screens().size() == 1;
267         bool const sd = _cpl->GetSelection() != -1;
268
269         _edit_cinema->Enable (sc);
270         _remove_cinema->Enable (sc);
271
272         _add_screen->Enable (sc);
273         _edit_screen->Enable (ss);
274         _remove_screen->Enable (ss);
275
276         wxButton* ok = dynamic_cast<wxButton *> (FindWindowById (wxID_OK, this));
277         if (ok) {
278                 ok->Enable ((selected_cinemas().size() > 0 || selected_screens().size() > 0) && sd);
279         }
280
281         _folder->Enable (_write_to->GetValue ());
282 }
283
284 void
285 KDMDialog::add_cinema (shared_ptr<Cinema> c)
286 {
287         _cinemas[_targets->AppendItem (_root, std_to_wx (c->name))] = c;
288
289         list<shared_ptr<Screen> > sc = c->screens ();
290         for (list<shared_ptr<Screen> >::iterator i = sc.begin(); i != sc.end(); ++i) {
291                 add_screen (c, *i);
292         }
293 }
294
295 void
296 KDMDialog::add_screen (shared_ptr<Cinema> c, shared_ptr<Screen> s)
297 {
298         map<wxTreeItemId, shared_ptr<Cinema> >::const_iterator i = _cinemas.begin();
299         while (i != _cinemas.end() && i->second != c) {
300                 ++i;
301         }
302
303         if (i == _cinemas.end()) {
304                 return;
305         }
306
307         _screens[_targets->AppendItem (i->first, std_to_wx (s->name))] = s;
308         _targets->Expand (i->first);
309 }
310
311 void
312 KDMDialog::add_cinema_clicked ()
313 {
314         CinemaDialog* d = new CinemaDialog (this, "Add Cinema");
315         if (d->ShowModal () == wxID_OK) {
316                 shared_ptr<Cinema> c (new Cinema (d->name(), d->email()));
317                 Config::instance()->add_cinema (c);
318                 add_cinema (c);
319         }
320
321         d->Destroy ();
322 }
323
324 void
325 KDMDialog::edit_cinema_clicked ()
326 {
327         if (selected_cinemas().size() != 1) {
328                 return;
329         }
330
331         pair<wxTreeItemId, shared_ptr<Cinema> > c = selected_cinemas().front();
332
333         CinemaDialog* d = new CinemaDialog (this, "Edit cinema", c.second->name, c.second->email);
334         if (d->ShowModal () == wxID_OK) {
335                 c.second->name = d->name ();
336                 c.second->email = d->email ();
337                 _targets->SetItemText (c.first, std_to_wx (d->name()));
338                 Config::instance()->changed ();
339         }
340
341         d->Destroy ();
342 }
343
344 void
345 KDMDialog::remove_cinema_clicked ()
346 {
347         if (selected_cinemas().size() != 1) {
348                 return;
349         }
350
351         pair<wxTreeItemId, shared_ptr<Cinema> > c = selected_cinemas().front();
352
353         Config::instance()->remove_cinema (c.second);
354         _targets->Delete (c.first);
355 }
356
357 void
358 KDMDialog::add_screen_clicked ()
359 {
360         if (selected_cinemas().size() != 1) {
361                 return;
362         }
363
364         shared_ptr<Cinema> c = selected_cinemas().front().second;
365
366         ScreenDialog* d = new ScreenDialog (this, "Add Screen");
367         if (d->ShowModal () != wxID_OK) {
368                 return;
369         }
370
371         shared_ptr<Screen> s (new Screen (d->name(), d->certificate()));
372         c->add_screen (s);
373         add_screen (c, s);
374
375         Config::instance()->changed ();
376
377         d->Destroy ();
378 }
379
380 void
381 KDMDialog::edit_screen_clicked ()
382 {
383         if (selected_screens().size() != 1) {
384                 return;
385         }
386
387         pair<wxTreeItemId, shared_ptr<Screen> > s = selected_screens().front();
388
389         ScreenDialog* d = new ScreenDialog (this, "Edit screen", s.second->name, s.second->certificate);
390         if (d->ShowModal () == wxID_OK) {
391                 s.second->name = d->name ();
392                 s.second->certificate = d->certificate ();
393                 _targets->SetItemText (s.first, std_to_wx (d->name()));
394                 Config::instance()->changed ();
395         }
396
397         d->Destroy ();
398 }
399
400 void
401 KDMDialog::remove_screen_clicked ()
402 {
403         if (selected_screens().size() != 1) {
404                 return;
405         }
406
407         pair<wxTreeItemId, shared_ptr<Screen> > s = selected_screens().front();
408
409         map<wxTreeItemId, shared_ptr<Cinema> >::iterator i = _cinemas.begin ();
410         while (i != _cinemas.end ()) {
411                 list<shared_ptr<Screen> > sc = i->second->screens ();
412                 if (find (sc.begin(), sc.end(), s.second) != sc.end ()) {
413                         break;
414                 }
415         }
416
417         if (i == _cinemas.end()) {
418                 return;
419         }
420
421         i->second->remove_screen (s.second);
422         _targets->Delete (s.first);
423
424         Config::instance()->changed ();
425 }
426
427 list<shared_ptr<Screen> >
428 KDMDialog::screens () const
429 {
430         list<shared_ptr<Screen> > s;
431
432         list<pair<wxTreeItemId, shared_ptr<Cinema> > > cinemas = selected_cinemas ();
433         for (list<pair<wxTreeItemId, shared_ptr<Cinema> > >::iterator i = cinemas.begin(); i != cinemas.end(); ++i) {
434                 list<shared_ptr<Screen> > sc = i->second->screens ();
435                 for (list<shared_ptr<Screen> >::const_iterator j = sc.begin(); j != sc.end(); ++j) {
436                         s.push_back (*j);
437                 }
438         }
439
440         list<pair<wxTreeItemId, shared_ptr<Screen> > > screens = selected_screens ();
441         for (list<pair<wxTreeItemId, shared_ptr<Screen> > >::iterator i = screens.begin(); i != screens.end(); ++i) {
442                 s.push_back (i->second);
443         }
444
445         s.sort ();
446         s.unique ();
447
448         return s;
449 }
450
451 boost::posix_time::ptime
452 KDMDialog::from () const
453 {
454         return posix_time (_from_date, _from_time);
455 }
456
457 boost::posix_time::ptime
458 KDMDialog::posix_time (wxDatePickerCtrl* date_picker, wxTimePickerCtrl* time_picker)
459 {
460         wxDateTime const date = date_picker->GetValue ();
461         wxDateTime const time = time_picker->GetValue ();
462         return boost::posix_time::ptime (
463                 boost::gregorian::date (date.GetYear(), date.GetMonth() + 1, date.GetDay()),
464                 boost::posix_time::time_duration (time.GetHour(), time.GetMinute(), time.GetSecond())
465                 );
466 }
467
468 boost::posix_time::ptime
469 KDMDialog::until () const
470 {
471         return posix_time (_until_date, _until_time);
472 }
473
474 boost::filesystem::path
475 KDMDialog::cpl () const
476 {
477         int const item = _cpl->GetSelection ();
478         DCPOMATIC_ASSERT (item >= 0);
479         return _cpls[item].cpl_file;
480 }
481
482 boost::filesystem::path
483 KDMDialog::directory () const
484 {
485         return wx_to_std (_folder->GetPath ());
486 }
487
488 bool
489 KDMDialog::write_to () const
490 {
491         return _write_to->GetValue ();
492 }
493
494 dcp::Formulation
495 KDMDialog::formulation () const
496 {
497         return (dcp::Formulation) reinterpret_cast<intptr_t> (_type->GetClientData (_type->GetSelection()));
498 }
499
500 void
501 KDMDialog::update_cpl_choice ()
502 {
503         _cpl->Clear ();
504
505         for (vector<CPLSummary>::const_iterator i = _cpls.begin(); i != _cpls.end(); ++i) {
506                 _cpl->Append (std_to_wx (i->cpl_id));
507
508                 if (_cpls.size() > 0) {
509                         _cpl->SetSelection (0);
510                 }
511         }
512
513         update_cpl_summary ();
514 }
515
516 void
517 KDMDialog::update_cpl_summary ()
518 {
519         int const n = _cpl->GetSelection();
520         if (n == wxNOT_FOUND) {
521                 return;
522         }
523
524         _dcp_directory->SetLabel (std_to_wx (_cpls[n].dcp_directory));
525         _cpl_id->SetLabel (std_to_wx (_cpls[n].cpl_id));
526         _cpl_annotation_text->SetLabel (std_to_wx (_cpls[n].cpl_annotation_text));
527 }
528
529 void
530 KDMDialog::cpl_browse_clicked ()
531 {
532         wxFileDialog* d = new wxFileDialog (this, _("Select CPL XML file"), wxEmptyString, wxEmptyString, "*.xml");
533         if (d->ShowModal() == wxID_CANCEL) {
534                 d->Destroy ();
535                 return;
536         }
537
538         boost::filesystem::path cpl_file (wx_to_std (d->GetPath ()));
539         boost::filesystem::path dcp_dir = cpl_file.parent_path ();
540
541         d->Destroy ();
542
543         /* XXX: hack alert */
544         cxml::Document cpl_document ("CompositionPlaylist");
545         cpl_document.read_file (cpl_file);
546
547         try {
548                 _cpls.push_back (
549                         CPLSummary (
550                                 dcp_dir.filename().string(),
551                                 cpl_document.string_child("Id").substr (9),
552                                 cpl_document.string_child ("ContentTitleText"),
553                                 cpl_file
554                                 )
555                         );
556         } catch (cxml::Error) {
557                 error_dialog (this, _("This is not a valid CPL file"));
558                 return;
559         }
560
561         update_cpl_choice ();
562         _cpl->SetSelection (_cpls.size() - 1);
563         update_cpl_summary ();
564 }