Fix play button behaviour.
[dcpomatic.git] / src / wx / film_viewer.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 /** @file  src/film_viewer.cc
22  *  @brief A wx widget to view a preview of a Film.
23  */
24
25 #include "film_viewer.h"
26 #include "playhead_to_timecode_dialog.h"
27 #include "playhead_to_frame_dialog.h"
28 #include "wx_util.h"
29 #include "closed_captions_dialog.h"
30 #include "lib/film.h"
31 #include "lib/ratio.h"
32 #include "lib/util.h"
33 #include "lib/job_manager.h"
34 #include "lib/image.h"
35 #include "lib/exceptions.h"
36 #include "lib/examine_content_job.h"
37 #include "lib/filter.h"
38 #include "lib/player.h"
39 #include "lib/player_video.h"
40 #include "lib/video_content.h"
41 #include "lib/video_decoder.h"
42 #include "lib/timer.h"
43 #include "lib/butler.h"
44 #include "lib/log.h"
45 #include "lib/config.h"
46 extern "C" {
47 #include <libavutil/pixfmt.h>
48 }
49 #include <dcp/exceptions.h>
50 #include <wx/tglbtn.h>
51 #include <iostream>
52 #include <iomanip>
53
54 using std::string;
55 using std::pair;
56 using std::min;
57 using std::max;
58 using std::cout;
59 using std::list;
60 using std::bad_alloc;
61 using std::make_pair;
62 using std::exception;
63 using boost::shared_ptr;
64 using boost::dynamic_pointer_cast;
65 using boost::weak_ptr;
66 using boost::optional;
67 using dcp::Size;
68
69 static
70 int
71 rtaudio_callback (void* out, void *, unsigned int frames, double, RtAudioStreamStatus, void* data)
72 {
73         return reinterpret_cast<FilmViewer*>(data)->audio_callback (out, frames);
74 }
75
76 FilmViewer::FilmViewer (wxWindow* p)
77         : _panel (new wxPanel (p))
78         , _coalesce_player_changes (false)
79         , _audio (DCPOMATIC_RTAUDIO_API)
80         , _audio_channels (0)
81         , _audio_block_size (1024)
82         , _playing (false)
83         , _latency_history_count (0)
84         , _dropped (0)
85         , _closed_captions_dialog (new ClosedCaptionsDialog(p))
86         , _outline_content (false)
87         , _eyes (EYES_LEFT)
88 {
89 #ifndef __WXOSX__
90         _panel->SetDoubleBuffered (true);
91 #endif
92
93         _panel->SetBackgroundStyle (wxBG_STYLE_PAINT);
94
95         _panel->Bind            (wxEVT_PAINT,               boost::bind (&FilmViewer::paint_panel,     this));
96         _panel->Bind            (wxEVT_SIZE,                boost::bind (&FilmViewer::panel_sized,     this, _1));
97         _timer.Bind             (wxEVT_TIMER,               boost::bind (&FilmViewer::timer,           this));
98
99         set_film (shared_ptr<Film> ());
100
101         _config_changed_connection = Config::instance()->Changed.connect (bind (&FilmViewer::config_changed, this, _1));
102         config_changed (Config::SOUND_OUTPUT);
103 }
104
105 FilmViewer::~FilmViewer ()
106 {
107         stop ();
108 }
109
110 void
111 FilmViewer::set_film (shared_ptr<Film> film)
112 {
113         if (_film == film) {
114                 return;
115         }
116
117         _film = film;
118
119         _frame.reset ();
120         _closed_captions_dialog->clear ();
121
122         if (!_film) {
123                 _player.reset ();
124                 recreate_butler ();
125                 _frame.reset ();
126                 refresh_panel ();
127                 return;
128         }
129
130         try {
131                 _player.reset (new Player (_film, _film->playlist ()));
132                 _player->set_fast ();
133                 if (_dcp_decode_reduction) {
134                         _player->set_dcp_decode_reduction (_dcp_decode_reduction);
135                 }
136         } catch (bad_alloc) {
137                 error_dialog (_panel, _("There is not enough free memory to do that."));
138                 _film.reset ();
139                 return;
140         }
141
142         _player->set_always_burn_open_subtitles ();
143         _player->set_play_referenced ();
144
145         _film->Change.connect (boost::bind (&FilmViewer::film_change, this, _1, _2));
146         _player->Change.connect (boost::bind (&FilmViewer::player_change, this, _1, _2, _3));
147
148         /* Keep about 1 second's worth of history samples */
149         _latency_history_count = _film->audio_frame_rate() / _audio_block_size;
150
151         recreate_butler ();
152
153         calculate_sizes ();
154         slow_refresh ();
155 }
156
157 void
158 FilmViewer::recreate_butler ()
159 {
160         bool const was_running = stop ();
161         _butler.reset ();
162
163         if (!_film) {
164                 return;
165         }
166
167         AudioMapping map = AudioMapping (_film->audio_channels(), _audio_channels);
168
169         if (_audio_channels != 2 || _film->audio_channels() < 3) {
170                 for (int i = 0; i < min (_film->audio_channels(), _audio_channels); ++i) {
171                         map.set (i, i, 1);
172                 }
173         } else {
174                 /* Special case: stereo output, at least 3 channel input.
175                    Map so that Lt = L(-3dB) + Ls(-3dB) + C(-6dB) + Lfe(-10dB)
176                                Rt = R(-3dB) + Rs(-3dB) + C(-6dB) + Lfe(-10dB)
177                 */
178                 map.set (dcp::LEFT,   0, 1 / sqrt(2)); // L -> Lt
179                 map.set (dcp::RIGHT,  1, 1 / sqrt(2)); // R -> Rt
180                 map.set (dcp::CENTRE, 0, 1 / 2.0); // C -> Lt
181                 map.set (dcp::CENTRE, 1, 1 / 2.0); // C -> Rt
182                 map.set (dcp::LFE,    0, 1 / sqrt(10)); // Lfe -> Lt
183                 map.set (dcp::LFE,    1, 1 / sqrt(10)); // Lfe -> Rt
184                 map.set (dcp::LS,     0, 1 / sqrt(2)); // Ls -> Lt
185                 map.set (dcp::RS,     1, 1 / sqrt(2)); // Rs -> Rt
186         }
187
188         _butler.reset (new Butler (_player, _film->log(), map, _audio_channels));
189         if (!Config::instance()->sound() && !_audio.isStreamOpen()) {
190                 _butler->disable_audio ();
191         }
192
193         _closed_captions_dialog->set_butler (_butler);
194
195         if (was_running) {
196                 start ();
197         }
198 }
199
200 void
201 FilmViewer::refresh_panel ()
202 {
203         _panel->Refresh ();
204         _panel->Update ();
205 }
206
207 void
208 FilmViewer::get ()
209 {
210         DCPOMATIC_ASSERT (_butler);
211
212         do {
213                 Butler::Error e;
214                 _player_video = _butler->get_video (&e);
215                 if (!_player_video.first && e == Butler::AGAIN) {
216                         signal_manager->when_idle (boost::bind(&FilmViewer::get, this));
217                         return;
218                 }
219         } while (
220                 _player_video.first &&
221                 _film->three_d() &&
222                 (_eyes != _player_video.first->eyes())
223                 );
224
225         _butler->rethrow ();
226
227         display_player_video ();
228 }
229
230 void
231 FilmViewer::display_player_video ()
232 {
233         if (!_player_video.first) {
234                 _frame.reset ();
235                 refresh_panel ();
236                 return;
237         }
238
239         if (_playing && (time() - _player_video.second) > one_video_frame()) {
240                 /* Too late; just drop this frame before we try to get its image (which will be the time-consuming
241                    part if this frame is J2K).
242                 */
243                 _video_position = _player_video.second;
244                 ++_dropped;
245                 return;
246         }
247
248         /* In an ideal world, what we would do here is:
249          *
250          * 1. convert to XYZ exactly as we do in the DCP creation path.
251          * 2. convert back to RGB for the preview display, compensating
252          *    for the monitor etc. etc.
253          *
254          * but this is inefficient if the source is RGB.  Since we don't
255          * (currently) care too much about the precise accuracy of the preview's
256          * colour mapping (and we care more about its speed) we try to short-
257          * circuit this "ideal" situation in some cases.
258          *
259          * The content's specified colour conversion indicates the colourspace
260          * which the content is in (according to the user).
261          *
262          * PlayerVideo::image (bound to PlayerVideo::always_rgb) will take the source
263          * image and convert it (from whatever the user has said it is) to RGB.
264          */
265
266         _frame = _player_video.first->image (
267                 bind (&Log::dcp_log, _film->log().get(), _1, _2),
268                 bind (&PlayerVideo::always_rgb, _1),
269                 false, true
270                 );
271
272         ImageChanged (_player_video.first);
273
274         _video_position = _player_video.second;
275         _inter_position = _player_video.first->inter_position ();
276         _inter_size = _player_video.first->inter_size ();
277
278         refresh_panel ();
279
280         _closed_captions_dialog->update (time());
281 }
282
283 void
284 FilmViewer::timer ()
285 {
286         if (!_film || !_playing) {
287                 return;
288         }
289
290         get ();
291         PositionChanged ();
292         DCPTime const next = _video_position + one_video_frame();
293
294         if (next >= _film->length()) {
295                 stop ();
296         }
297
298         _timer.Start (max ((next.seconds() - time().seconds()) * 1000, 1.0), wxTIMER_ONE_SHOT);
299
300         if (_butler) {
301                 _butler->rethrow ();
302         }
303 }
304
305 void
306 FilmViewer::paint_panel ()
307 {
308         wxPaintDC dc (_panel);
309
310         if (!_frame || !_film || !_out_size.width || !_out_size.height || _out_size != _frame->size()) {
311                 dc.Clear ();
312                 return;
313         }
314
315         wxImage frame (_out_size.width, _out_size.height, _frame->data()[0], true);
316         wxBitmap frame_bitmap (frame);
317         dc.DrawBitmap (frame_bitmap, 0, 0);
318
319         if (_out_size.width < _panel_size.width) {
320                 wxPen p (_panel->GetParent()->GetBackgroundColour());
321                 wxBrush b (_panel->GetParent()->GetBackgroundColour());
322                 dc.SetPen (p);
323                 dc.SetBrush (b);
324                 dc.DrawRectangle (_out_size.width, 0, _panel_size.width - _out_size.width, _panel_size.height);
325         }
326
327         if (_out_size.height < _panel_size.height) {
328                 wxPen p (_panel->GetParent()->GetBackgroundColour());
329                 wxBrush b (_panel->GetParent()->GetBackgroundColour());
330                 dc.SetPen (p);
331                 dc.SetBrush (b);
332                 dc.DrawRectangle (0, _out_size.height, _panel_size.width, _panel_size.height - _out_size.height);
333         }
334
335         if (_outline_content) {
336                 wxPen p (wxColour (255, 0, 0), 2);
337                 dc.SetPen (p);
338                 dc.SetBrush (*wxTRANSPARENT_BRUSH);
339                 dc.DrawRectangle (_inter_position.x, _inter_position.y, _inter_size.width, _inter_size.height);
340         }
341 }
342
343 void
344 FilmViewer::set_outline_content (bool o)
345 {
346         _outline_content = o;
347         refresh_panel ();
348 }
349
350 void
351 FilmViewer::set_eyes (Eyes e)
352 {
353         _eyes = e;
354         slow_refresh ();
355 }
356
357 void
358 FilmViewer::panel_sized (wxSizeEvent& ev)
359 {
360         _panel_size.width = ev.GetSize().GetWidth();
361         _panel_size.height = ev.GetSize().GetHeight();
362
363         calculate_sizes ();
364         if (!quick_refresh()) {
365                 slow_refresh ();
366         }
367         PositionChanged ();
368 }
369
370 void
371 FilmViewer::calculate_sizes ()
372 {
373         if (!_film || !_player) {
374                 return;
375         }
376
377         Ratio const * container = _film->container ();
378
379         float const panel_ratio = _panel_size.ratio ();
380         float const film_ratio = container ? container->ratio () : 1.78;
381
382         if (panel_ratio < film_ratio) {
383                 /* panel is less widscreen than the film; clamp width */
384                 _out_size.width = _panel_size.width;
385                 _out_size.height = lrintf (_out_size.width / film_ratio);
386         } else {
387                 /* panel is more widescreen than the film; clamp height */
388                 _out_size.height = _panel_size.height;
389                 _out_size.width = lrintf (_out_size.height * film_ratio);
390         }
391
392         /* Catch silly values */
393         _out_size.width = max (64, _out_size.width);
394         _out_size.height = max (64, _out_size.height);
395
396         _player->set_video_container_size (_out_size);
397 }
398
399 void
400 FilmViewer::start ()
401 {
402         if (!_film) {
403                 return;
404         }
405
406         if (_audio.isStreamOpen()) {
407                 _audio.setStreamTime (_video_position.seconds());
408                 _audio.startStream ();
409         }
410
411         _playing = true;
412         _dropped = 0;
413         timer ();
414         Started ();
415 }
416
417 bool
418 FilmViewer::stop ()
419 {
420         if (_audio.isStreamRunning()) {
421                 /* stop stream and discard any remaining queued samples */
422                 _audio.abortStream ();
423         }
424
425         if (!_playing) {
426                 return false;
427         }
428
429         _playing = false;
430         Stopped ();
431         return true;
432 }
433
434 void
435 FilmViewer::go_to (DCPTime t)
436 {
437         if (t < DCPTime ()) {
438                 t = DCPTime ();
439         }
440
441         if (t >= _film->length ()) {
442                 t = _film->length ();
443         }
444
445         seek (t, true);
446         PositionChanged ();
447 }
448
449 void
450 FilmViewer::player_change (ChangeType type, int property, bool frequent)
451 {
452         if (type != CHANGE_TYPE_DONE || frequent) {
453                 return;
454         }
455
456         if (_coalesce_player_changes) {
457                 _pending_player_changes.push_back (property);
458                 return;
459         }
460
461         calculate_sizes ();
462         bool refreshed = false;
463         if (
464                 property == VideoContentProperty::CROP ||
465                 property == VideoContentProperty::SCALE ||
466                 property == VideoContentProperty::FADE_IN ||
467                 property == VideoContentProperty::FADE_OUT ||
468                 property == VideoContentProperty::COLOUR_CONVERSION ||
469                 property == PlayerProperty::VIDEO_CONTAINER_SIZE ||
470                 property == PlayerProperty::FILM_CONTAINER
471                 ) {
472                 refreshed = quick_refresh ();
473         }
474
475         if (!refreshed) {
476                 slow_refresh ();
477         }
478         PositionChanged ();
479 }
480
481 void
482 FilmViewer::film_change (ChangeType type, Film::Property p)
483 {
484         if (type == CHANGE_TYPE_DONE && p == Film::AUDIO_CHANNELS) {
485                 recreate_butler ();
486         }
487 }
488
489 /** Re-get the current frame slowly by seeking */
490 void
491 FilmViewer::slow_refresh ()
492 {
493         seek (_video_position, true);
494 }
495
496 /** Try to re-get the current frame quickly by resetting the metadata
497  *  in the PlayerVideo that we used last time.
498  *  @return true if this was possible, false if not.
499  */
500 bool
501 FilmViewer::quick_refresh ()
502 {
503         if (!_player_video.first) {
504                 return false;
505         }
506
507         if (!_player_video.first->reset_metadata (_player->video_container_size(), _film->frame_size())) {
508                 return false;
509         }
510
511         display_player_video ();
512         return true;
513 }
514
515 void
516 FilmViewer::set_position (DCPTime p)
517 {
518         _video_position = p;
519         seek (p, true);
520         PositionChanged ();
521 }
522
523 void
524 FilmViewer::set_position (shared_ptr<Content> content, ContentTime t)
525 {
526         optional<DCPTime> dt = _player->content_time_to_dcp (content, t);
527         if (dt) {
528                 set_position (*dt);
529         }
530 }
531
532 void
533 FilmViewer::set_coalesce_player_changes (bool c)
534 {
535         _coalesce_player_changes = c;
536
537         if (!c) {
538                 BOOST_FOREACH (int i, _pending_player_changes) {
539                         player_change (CHANGE_TYPE_DONE, i, false);
540                 }
541                 _pending_player_changes.clear ();
542         }
543 }
544
545 void
546 FilmViewer::seek (DCPTime t, bool accurate)
547 {
548         if (!_butler) {
549                 return;
550         }
551
552         bool const was_running = stop ();
553
554         _closed_captions_dialog->clear ();
555         _butler->seek (t, accurate);
556         get ();
557
558         if (was_running) {
559                 start ();
560         }
561 }
562
563 void
564 FilmViewer::config_changed (Config::Property p)
565 {
566         if (p != Config::SOUND && p != Config::SOUND_OUTPUT) {
567                 return;
568         }
569
570         if (_audio.isStreamOpen ()) {
571                 _audio.closeStream ();
572         }
573
574         if (Config::instance()->sound() && _audio.getDeviceCount() > 0) {
575                 unsigned int st = 0;
576                 if (Config::instance()->sound_output()) {
577                         while (st < _audio.getDeviceCount()) {
578                                 if (_audio.getDeviceInfo(st).name == Config::instance()->sound_output().get()) {
579                                         break;
580                                 }
581                                 ++st;
582                         }
583                         if (st == _audio.getDeviceCount()) {
584                                 st = _audio.getDefaultOutputDevice();
585                         }
586                 } else {
587                         st = _audio.getDefaultOutputDevice();
588                 }
589
590                 _audio_channels = _audio.getDeviceInfo(st).outputChannels;
591
592                 RtAudio::StreamParameters sp;
593                 sp.deviceId = st;
594                 sp.nChannels = _audio_channels;
595                 sp.firstChannel = 0;
596                 try {
597                         _audio.openStream (&sp, 0, RTAUDIO_FLOAT32, 48000, &_audio_block_size, &rtaudio_callback, this);
598 #ifdef DCPOMATIC_USE_RTERROR
599                 } catch (RtError& e) {
600 #else
601                 } catch (RtAudioError& e) {
602 #endif
603                         error_dialog (
604                                 _panel,
605                                 _("Could not set up audio output.  There will be no audio during the preview."), std_to_wx(e.what())
606                                 );
607                 }
608                 recreate_butler ();
609
610         } else {
611                 _audio_channels = 0;
612                 recreate_butler ();
613         }
614 }
615
616 DCPTime
617 FilmViewer::uncorrected_time () const
618 {
619         if (_audio.isStreamRunning ()) {
620                 return DCPTime::from_seconds (const_cast<RtAudio*>(&_audio)->getStreamTime());
621         }
622
623         return _video_position;
624 }
625
626 DCPTime
627 FilmViewer::time () const
628 {
629         if (_audio.isStreamRunning ()) {
630                 return DCPTime::from_seconds (const_cast<RtAudio*>(&_audio)->getStreamTime ()) -
631                         DCPTime::from_frames (average_latency(), _film->audio_frame_rate());
632         }
633
634         return _video_position;
635 }
636
637 int
638 FilmViewer::audio_callback (void* out_p, unsigned int frames)
639 {
640         while (true) {
641                 optional<DCPTime> t = _butler->get_audio (reinterpret_cast<float*> (out_p), frames);
642                 if (!t || DCPTime(uncorrected_time() - *t) < one_video_frame()) {
643                         /* There was an underrun or this audio is on time; carry on */
644                         break;
645                 }
646                 /* The audio we just got was (very) late; drop it and get some more. */
647         }
648
649         boost::mutex::scoped_lock lm (_latency_history_mutex, boost::try_to_lock);
650         if (lm) {
651                 _latency_history.push_back (_audio.getStreamLatency ());
652                 if (_latency_history.size() > static_cast<size_t> (_latency_history_count)) {
653                         _latency_history.pop_front ();
654                 }
655         }
656
657         return 0;
658 }
659
660 Frame
661 FilmViewer::average_latency () const
662 {
663         boost::mutex::scoped_lock lm (_latency_history_mutex);
664         if (_latency_history.empty()) {
665                 return 0;
666         }
667
668         Frame total = 0;
669         BOOST_FOREACH (Frame i, _latency_history) {
670                 total += i;
671         }
672
673         return total / _latency_history.size();
674 }
675
676 void
677 FilmViewer::set_dcp_decode_reduction (optional<int> reduction)
678 {
679         _dcp_decode_reduction = reduction;
680         if (_player) {
681                 _player->set_dcp_decode_reduction (reduction);
682         }
683 }
684
685 optional<int>
686 FilmViewer::dcp_decode_reduction () const
687 {
688         return _dcp_decode_reduction;
689 }
690
691 DCPTime
692 FilmViewer::one_video_frame () const
693 {
694         return DCPTime::from_frames (1, _film->video_frame_rate());
695 }
696
697 void
698 FilmViewer::show_closed_captions ()
699 {
700         _closed_captions_dialog->Show();
701 }
702
703 void
704 FilmViewer::move (DCPTime by)
705 {
706         if (!_film) {
707                 return;
708         }
709
710         go_to (_video_position + by);
711 }