1dc0185ba59e229e778c7b2426c39883a2ae544e
[dcpomatic.git] / src / wx / video_panel.cc
1 /*
2     Copyright (C) 2012-2018 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 #include "filter_dialog.h"
22 #include "video_panel.h"
23 #include "wx_util.h"
24 #include "content_colour_conversion_dialog.h"
25 #include "content_widget.h"
26 #include "content_panel.h"
27 #include "static_text.h"
28 #include "check_box.h"
29 #include "dcpomatic_button.h"
30 #include "lib/filter.h"
31 #include "lib/ffmpeg_content.h"
32 #include "lib/colour_conversion.h"
33 #include "lib/config.h"
34 #include "lib/util.h"
35 #include "lib/ratio.h"
36 #include "lib/frame_rate_change.h"
37 #include "lib/dcp_content.h"
38 #include "lib/video_content.h"
39 #include <wx/spinctrl.h>
40 #include <boost/foreach.hpp>
41 #include <set>
42 #include <iostream>
43
44 using std::vector;
45 using std::string;
46 using std::pair;
47 using std::cout;
48 using std::list;
49 using std::set;
50 using boost::shared_ptr;
51 using boost::dynamic_pointer_cast;
52 using boost::bind;
53 using boost::optional;
54
55 static VideoContentScale
56 index_to_scale (int n)
57 {
58         vector<VideoContentScale> scales = VideoContentScale::all ();
59         DCPOMATIC_ASSERT (n >= 0);
60         DCPOMATIC_ASSERT (n < int (scales.size ()));
61         return scales[n];
62 }
63
64 static int
65 scale_to_index (VideoContentScale scale)
66 {
67         vector<VideoContentScale> scales = VideoContentScale::all ();
68         for (size_t i = 0; i < scales.size(); ++i) {
69                 if (scales[i] == scale) {
70                         return i;
71                 }
72         }
73
74         DCPOMATIC_ASSERT (false);
75 }
76
77 VideoPanel::VideoPanel (ContentPanel* p)
78         : ContentSubPanel (p, _("Video"))
79 {
80         _reference = new CheckBox (this, _("Use this DCP's video as OV and make VF"));
81         _reference_note = new StaticText (this, wxT(""));
82         _reference_note->Wrap (200);
83         wxFont font = _reference_note->GetFont();
84         font.SetStyle(wxFONTSTYLE_ITALIC);
85         font.SetPointSize(font.GetPointSize() - 1);
86         _reference_note->SetFont(font);
87
88         _type_label = create_label (this, _("Type"), true);
89         _frame_type = new ContentChoice<VideoContent, VideoFrameType> (
90                 this,
91                 new wxChoice (this, wxID_ANY),
92                 VideoContentProperty::FRAME_TYPE,
93                 &Content::video,
94                 boost::mem_fn (&VideoContent::frame_type),
95                 boost::mem_fn (&VideoContent::set_frame_type),
96                 &caster<int, VideoFrameType>,
97                 &caster<VideoFrameType, int>
98                 );
99
100         _left_crop_label = create_label (this, _("Left crop"), true);
101         _left_crop = new ContentSpinCtrl<VideoContent> (
102                 this,
103                 new wxSpinCtrl (this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize (64, -1)),
104                 VideoContentProperty::CROP,
105                 &Content::video,
106                 boost::mem_fn (&VideoContent::left_crop),
107                 boost::mem_fn (&VideoContent::set_left_crop)
108                 );
109
110         _right_crop_label = create_label (this, _("Right crop"), true);
111         _right_crop = new ContentSpinCtrl<VideoContent> (
112                 this,
113                 new wxSpinCtrl (this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize (64, -1)),
114                 VideoContentProperty::CROP,
115                 &Content::video,
116                 boost::mem_fn (&VideoContent::right_crop),
117                 boost::mem_fn (&VideoContent::set_right_crop)
118                 );
119
120         _top_crop_label = create_label (this, _("Top crop"), true);
121         _top_crop = new ContentSpinCtrl<VideoContent> (
122                 this,
123                 new wxSpinCtrl (this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize (64, -1)),
124                 VideoContentProperty::CROP,
125                 &Content::video,
126                 boost::mem_fn (&VideoContent::top_crop),
127                 boost::mem_fn (&VideoContent::set_top_crop)
128                 );
129
130         _bottom_crop_label = create_label (this, _("Bottom crop"), true);
131         _bottom_crop = new ContentSpinCtrl<VideoContent> (
132                 this,
133                 new wxSpinCtrl (this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize (64, -1)),
134                 VideoContentProperty::CROP,
135                 &Content::video,
136                 boost::mem_fn (&VideoContent::bottom_crop),
137                 boost::mem_fn (&VideoContent::set_bottom_crop)
138                 );
139
140         _fade_in_label = create_label (this, _("Fade in"), true);
141         _fade_in = new Timecode<ContentTime> (this);
142
143         _fade_out_label = create_label (this, _("Fade out"), true);
144         _fade_out = new Timecode<ContentTime> (this);
145
146         _scale_to_label = create_label (this, _("Scale to"), true);
147         _scale = new ContentChoice<VideoContent, VideoContentScale> (
148                 this,
149                 new wxChoice (this, wxID_ANY),
150                 VideoContentProperty::SCALE,
151                 &Content::video,
152                 boost::mem_fn (&VideoContent::scale),
153                 boost::mem_fn (&VideoContent::set_scale),
154                 &index_to_scale,
155                 &scale_to_index
156                 );
157
158         wxClientDC dc (this);
159         wxSize size = dc.GetTextExtent (wxT ("A quite long name"));
160         size.SetHeight (-1);
161
162         _filters_label = create_label (this, _("Filters"), true);
163         _filters = new StaticText (this, _("None"), wxDefaultPosition, size);
164         _filters_button = new Button (this, _("Edit..."));
165
166         _colour_conversion_label = create_label (this, _("Colour conversion"), true);
167         _colour_conversion = new wxChoice (this, wxID_ANY, wxDefaultPosition, size);
168         _colour_conversion->Append (_("None"));
169         BOOST_FOREACH (PresetColourConversion const & i, PresetColourConversion::all()) {
170                 _colour_conversion->Append (std_to_wx (i.name));
171         }
172
173         /// TRANSLATORS: translate the word "Custom" here; do not include the "Colour|" prefix
174         _colour_conversion->Append (S_("Colour|Custom"));
175         _edit_colour_conversion_button = new Button (this, _("Edit..."));
176
177         _description = new StaticText (this, wxT ("\n \n \n \n \n"), wxDefaultPosition, wxDefaultSize);
178         _description->SetFont(font);
179
180         _left_crop->wrapped()->SetRange (0, 4096);
181         _top_crop->wrapped()->SetRange (0, 4096);
182         _right_crop->wrapped()->SetRange (0, 4096);
183         _bottom_crop->wrapped()->SetRange (0, 4096);
184
185         _scale->wrapped()->Clear ();
186         BOOST_FOREACH (VideoContentScale const & i, VideoContentScale::all ()) {
187                 _scale->wrapped()->Append (std_to_wx (i.name ()));
188         }
189
190         _frame_type->wrapped()->Append (_("2D"));
191         _frame_type->wrapped()->Append (_("3D"));
192         _frame_type->wrapped()->Append (_("3D left/right"));
193         _frame_type->wrapped()->Append (_("3D top/bottom"));
194         _frame_type->wrapped()->Append (_("3D alternate"));
195         _frame_type->wrapped()->Append (_("3D left only"));
196         _frame_type->wrapped()->Append (_("3D right only"));
197
198         content_selection_changed ();
199
200         _fade_in->Changed.connect (boost::bind (&VideoPanel::fade_in_changed, this));
201         _fade_out->Changed.connect (boost::bind (&VideoPanel::fade_out_changed, this));
202
203         _reference->Bind                     (wxEVT_CHECKBOX, boost::bind (&VideoPanel::reference_clicked, this));
204         _filters_button->Bind                (wxEVT_BUTTON,   boost::bind (&VideoPanel::edit_filters_clicked, this));
205         _colour_conversion->Bind             (wxEVT_CHOICE,   boost::bind (&VideoPanel::colour_conversion_changed, this));
206         _edit_colour_conversion_button->Bind (wxEVT_BUTTON,   boost::bind (&VideoPanel::edit_colour_conversion_clicked, this));
207
208         add_to_grid ();
209 }
210
211 void
212 VideoPanel::add_to_grid ()
213 {
214         bool const full = Config::instance()->interface_complexity() == Config::INTERFACE_FULL;
215
216         int r = 0;
217
218         _reference->Show (full);
219         _reference_note->Show (full);
220
221         if (full) {
222                 wxBoxSizer* reference_sizer = new wxBoxSizer (wxVERTICAL);
223                 reference_sizer->Add (_reference, 0);
224                 reference_sizer->Add (_reference_note, 0);
225                 _grid->Add (reference_sizer, wxGBPosition(r, 0), wxGBSpan(1, 3));
226                 ++r;
227         }
228
229         add_label_to_sizer (_grid, _type_label, true, wxGBPosition(r, 0));
230         _frame_type->add (_grid, wxGBPosition(r, 1), wxGBSpan(1, 2));
231         ++r;
232
233         int cr = 0;
234         wxGridBagSizer* crop = new wxGridBagSizer (DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP);
235         add_label_to_sizer (crop, _left_crop_label, true, wxGBPosition (cr, 0));
236         _left_crop->add (crop, wxGBPosition (cr, 1));
237         add_label_to_sizer (crop, _right_crop_label, true, wxGBPosition (cr, 2));
238         _right_crop->add (crop, wxGBPosition (cr, 3));
239         ++cr;
240         add_label_to_sizer (crop, _top_crop_label, true, wxGBPosition (cr, 0));
241         _top_crop->add (crop, wxGBPosition (cr, 1));
242         add_label_to_sizer (crop, _bottom_crop_label, true, wxGBPosition (cr, 2));
243         _bottom_crop->add (crop, wxGBPosition (cr, 3));
244         _grid->Add (crop, wxGBPosition (r, 0), wxGBSpan (2, 4));
245         r += 2;
246
247         _scale_to_label->Show (full);
248         _scale->show (full);
249         _filters_label->Show (full);
250         _filters->Show (full);
251         _filters_button->Show (full);
252         _colour_conversion_label->Show (full);
253         _colour_conversion->Show (full);
254         _edit_colour_conversion_button->Show (full);
255
256         add_label_to_sizer (_grid, _fade_in_label, true, wxGBPosition (r, 0));
257         _grid->Add (_fade_in, wxGBPosition (r, 1), wxGBSpan (1, 3));
258         ++r;
259
260         add_label_to_sizer (_grid, _fade_out_label, true, wxGBPosition (r, 0));
261         _grid->Add (_fade_out, wxGBPosition (r, 1), wxGBSpan (1, 3));
262         ++r;
263
264         if (full) {
265                 add_label_to_sizer (_grid, _scale_to_label, true, wxGBPosition (r, 0));
266                 _scale->add (_grid, wxGBPosition (r, 1), wxGBSpan (1, 2));
267                 ++r;
268
269                 add_label_to_sizer (_grid, _filters_label, true, wxGBPosition (r, 0));
270                 {
271                         wxSizer* s = new wxBoxSizer (wxHORIZONTAL);
272                         s->Add (_filters, 1, wxALIGN_CENTER_VERTICAL | wxTOP | wxBOTTOM | wxRIGHT, 6);
273                         s->Add (_filters_button, 0, wxALIGN_CENTER_VERTICAL);
274                         _grid->Add (s, wxGBPosition (r, 1), wxDefaultSpan, wxALIGN_CENTER_VERTICAL);
275                 }
276                 ++r;
277
278                 add_label_to_sizer (_grid, _colour_conversion_label, true, wxGBPosition(r, 0));
279                 {
280                         wxSizer* s = new wxBoxSizer (wxHORIZONTAL);
281                         s->Add (_colour_conversion, 1, wxALIGN_CENTER_VERTICAL | wxTOP | wxBOTTOM | wxRIGHT, 6);
282                         s->Add (_edit_colour_conversion_button, 0, wxALIGN_CENTER_VERTICAL);
283                         _grid->Add (s, wxGBPosition (r, 1), wxDefaultSpan, wxALIGN_CENTER_VERTICAL);
284                 }
285                 ++r;
286         }
287
288         _grid->Add (_description, wxGBPosition (r, 0), wxGBSpan (1, 4), wxEXPAND | wxALIGN_CENTER_VERTICAL, 6);
289         ++r;
290 }
291
292
293 void
294 VideoPanel::film_changed (Film::Property property)
295 {
296         switch (property) {
297         case Film::VIDEO_FRAME_RATE:
298         case Film::CONTAINER:
299         case Film::RESOLUTION:
300                 setup_description ();
301                 setup_sensitivity ();
302                 break;
303         case Film::REEL_TYPE:
304         case Film::INTEROP:
305                 setup_sensitivity ();
306                 break;
307         default:
308                 break;
309         }
310 }
311
312 void
313 VideoPanel::film_content_changed (int property)
314 {
315         ContentList vc = _parent->selected_video ();
316         shared_ptr<Content> vcs;
317         shared_ptr<FFmpegContent> fcs;
318         if (!vc.empty ()) {
319                 vcs = vc.front ();
320                 fcs = dynamic_pointer_cast<FFmpegContent> (vcs);
321         }
322
323         if (property == ContentProperty::VIDEO_FRAME_RATE ||
324             property == VideoContentProperty::FRAME_TYPE ||
325             property == VideoContentProperty::CROP ||
326             property == VideoContentProperty::SCALE) {
327                 setup_description ();
328         } else if (property == VideoContentProperty::COLOUR_CONVERSION) {
329                 if (vcs && vcs->video->colour_conversion ()) {
330                         optional<size_t> preset = vcs->video->colour_conversion().get().preset ();
331                         vector<PresetColourConversion> cc = PresetColourConversion::all ();
332                         if (preset) {
333                                 checked_set (_colour_conversion, preset.get() + 1);
334                         } else {
335                                 checked_set (_colour_conversion, cc.size() + 1);
336                         }
337                 } else {
338                         checked_set (_colour_conversion, 0);
339                 }
340
341                 setup_sensitivity ();
342
343         } else if (property == FFmpegContentProperty::FILTERS) {
344                 if (fcs) {
345                         string p = Filter::ffmpeg_string (fcs->filters ());
346                         if (p.empty ()) {
347                                 checked_set (_filters, _("None"));
348                         } else {
349                                 if (p.length() > 25) {
350                                         p = p.substr (0, 25) + "...";
351                                 }
352                                 checked_set (_filters, p);
353                         }
354                 }
355         } else if (property == VideoContentProperty::FADE_IN) {
356                 set<Frame> check;
357                 BOOST_FOREACH (shared_ptr<const Content> i, vc) {
358                         check.insert (i->video->fade_in ());
359                 }
360
361                 if (check.size() == 1) {
362                         _fade_in->set (
363                                 ContentTime::from_frames (vc.front()->video->fade_in(), vc.front()->active_video_frame_rate(_parent->film())),
364                                 vc.front()->active_video_frame_rate(_parent->film())
365                                 );
366                 } else {
367                         _fade_in->clear ();
368                 }
369         } else if (property == VideoContentProperty::FADE_OUT) {
370                 set<Frame> check;
371                 BOOST_FOREACH (shared_ptr<const Content> i, vc) {
372                         check.insert (i->video->fade_out ());
373                 }
374
375                 if (check.size() == 1) {
376                         _fade_out->set (
377                                 ContentTime::from_frames (vc.front()->video->fade_out(), vc.front()->active_video_frame_rate(_parent->film())),
378                                 vc.front()->active_video_frame_rate(_parent->film())
379                                 );
380                 } else {
381                         _fade_out->clear ();
382                 }
383         } else if (property == DCPContentProperty::REFERENCE_VIDEO) {
384                 if (vc.size() == 1) {
385                         shared_ptr<DCPContent> dcp = dynamic_pointer_cast<DCPContent> (vc.front ());
386                         checked_set (_reference, dcp ? dcp->reference_video () : false);
387                 } else {
388                         checked_set (_reference, false);
389                 }
390
391                 setup_sensitivity ();
392         }
393 }
394
395 /** Called when the `Edit filters' button has been clicked */
396 void
397 VideoPanel::edit_filters_clicked ()
398 {
399         FFmpegContentList c = _parent->selected_ffmpeg ();
400         if (c.size() != 1) {
401                 return;
402         }
403
404         FilterDialog* d = new FilterDialog (this, c.front()->filters());
405         d->ActiveChanged.connect (bind (&FFmpegContent::set_filters, c.front(), _1));
406         d->ShowModal ();
407         d->Destroy ();
408 }
409
410 void
411 VideoPanel::setup_description ()
412 {
413         ContentList vc = _parent->selected_video ();
414         if (vc.empty ()) {
415                 checked_set (_description, wxT (""));
416                 return;
417         } else if (vc.size() > 1) {
418                 checked_set (_description, _("Multiple content selected"));
419                 return;
420         }
421
422         string d = vc.front()->video->processing_description (_parent->film());
423         size_t lines = count (d.begin(), d.end(), '\n');
424
425         for (int i = lines; i < 6; ++i) {
426                 d += "\n ";
427         }
428
429         checked_set (_description, d);
430         _sizer->Layout ();
431 }
432
433 void
434 VideoPanel::colour_conversion_changed ()
435 {
436         ContentList vc = _parent->selected_video ();
437         if (vc.size() != 1) {
438                 return;
439         }
440
441         int const s = _colour_conversion->GetSelection ();
442         vector<PresetColourConversion> all = PresetColourConversion::all ();
443
444         if (s == 0) {
445                 vc.front()->video->unset_colour_conversion ();
446         } else if (s == int (all.size() + 1)) {
447                 edit_colour_conversion_clicked ();
448         } else {
449                 vc.front()->video->set_colour_conversion (all[s - 1].conversion);
450         }
451 }
452
453 void
454 VideoPanel::edit_colour_conversion_clicked ()
455 {
456         ContentList vc = _parent->selected_video ();
457         if (vc.size() != 1) {
458                 return;
459         }
460
461         ContentColourConversionDialog* d = new ContentColourConversionDialog (this, vc.front()->video->yuv ());
462         d->set (vc.front()->video->colour_conversion().get_value_or (PresetColourConversion::all().front().conversion));
463         if (d->ShowModal() == wxID_OK) {
464                 vc.front()->video->set_colour_conversion (d->get ());
465         } else {
466                 /* Reset the colour conversion choice */
467                 film_content_changed (VideoContentProperty::COLOUR_CONVERSION);
468         }
469         d->Destroy ();
470 }
471
472 void
473 VideoPanel::content_selection_changed ()
474 {
475         ContentList video_sel = _parent->selected_video ();
476
477         _frame_type->set_content (video_sel);
478         _left_crop->set_content (video_sel);
479         _right_crop->set_content (video_sel);
480         _top_crop->set_content (video_sel);
481         _bottom_crop->set_content (video_sel);
482         _scale->set_content (video_sel);
483
484         film_content_changed (ContentProperty::VIDEO_FRAME_RATE);
485         film_content_changed (VideoContentProperty::CROP);
486         film_content_changed (VideoContentProperty::COLOUR_CONVERSION);
487         film_content_changed (VideoContentProperty::FADE_IN);
488         film_content_changed (VideoContentProperty::FADE_OUT);
489         film_content_changed (FFmpegContentProperty::FILTERS);
490         film_content_changed (DCPContentProperty::REFERENCE_VIDEO);
491
492         setup_sensitivity ();
493 }
494
495 void
496 VideoPanel::setup_sensitivity ()
497 {
498         ContentList sel = _parent->selected ();
499
500         shared_ptr<DCPContent> dcp;
501         if (sel.size() == 1) {
502                 dcp = dynamic_pointer_cast<DCPContent> (sel.front ());
503         }
504
505         string why_not;
506         bool const can_reference = dcp && dcp->can_reference_video (_parent->film(), why_not);
507         setup_refer_button (_reference, _reference_note, dcp, can_reference, why_not);
508
509         if (_reference->GetValue ()) {
510                 _frame_type->wrapped()->Enable (false);
511                 _left_crop->wrapped()->Enable (false);
512                 _right_crop->wrapped()->Enable (false);
513                 _top_crop->wrapped()->Enable (false);
514                 _bottom_crop->wrapped()->Enable (false);
515                 _fade_in->Enable (false);
516                 _fade_out->Enable (false);
517                 _scale->wrapped()->Enable (false);
518                 _description->Enable (false);
519                 _filters->Enable (false);
520                 _filters_button->Enable (false);
521                 _colour_conversion->Enable (false);
522         } else {
523                 ContentList video_sel = _parent->selected_video ();
524                 FFmpegContentList ffmpeg_sel = _parent->selected_ffmpeg ();
525                 bool const single = video_sel.size() == 1;
526
527                 _frame_type->wrapped()->Enable (true);
528                 _left_crop->wrapped()->Enable (true);
529                 _right_crop->wrapped()->Enable (true);
530                 _top_crop->wrapped()->Enable (true);
531                 _bottom_crop->wrapped()->Enable (true);
532                 _fade_in->Enable (!video_sel.empty ());
533                 _fade_out->Enable (!video_sel.empty ());
534                 _scale->wrapped()->Enable (true);
535                 _description->Enable (true);
536                 _filters->Enable (true);
537                 _filters_button->Enable (single && !ffmpeg_sel.empty ());
538                 _colour_conversion->Enable (single && !video_sel.empty ());
539         }
540
541         ContentList vc = _parent->selected_video ();
542         shared_ptr<Content> vcs;
543         if (!vc.empty ()) {
544                 vcs = vc.front ();
545         }
546
547         if (vcs && vcs->video->colour_conversion ()) {
548                 _edit_colour_conversion_button->Enable (!vcs->video->colour_conversion().get().preset());
549         } else {
550                 _edit_colour_conversion_button->Enable (false);
551         }
552 }
553
554 void
555 VideoPanel::fade_in_changed ()
556 {
557         BOOST_FOREACH (shared_ptr<Content> i, _parent->selected_video ()) {
558                 double const vfr = i->active_video_frame_rate (_parent->film());
559                 i->video->set_fade_in (_fade_in->get(vfr).frames_round(vfr));
560         }
561 }
562
563 void
564 VideoPanel::fade_out_changed ()
565 {
566         BOOST_FOREACH (shared_ptr<Content> i, _parent->selected_video ()) {
567                 double const vfr = i->active_video_frame_rate (_parent->film());
568                 i->video->set_fade_out (_fade_out->get(vfr).frames_round(vfr));
569         }
570 }
571
572 void
573 VideoPanel::reference_clicked ()
574 {
575         ContentList c = _parent->selected ();
576         if (c.size() != 1) {
577                 return;
578         }
579
580         shared_ptr<DCPContent> d = dynamic_pointer_cast<DCPContent> (c.front ());
581         if (!d) {
582                 return;
583         }
584
585         d->set_reference_video (_reference->GetValue ());
586 }