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