Some tidying up. Do encode progress in the writer to improve progress bar movement...
[dcpomatic.git] / src / lib / player.cc
index c66d091cf66fb27f82ed58235ec162aef36917be..95036cfe0ae18f1ca24ffebea6a8f2a899eb61c6 100644 (file)
@@ -30,6 +30,7 @@
 
 using std::list;
 using std::cout;
+using std::vector;
 using boost::shared_ptr;
 using boost::weak_ptr;
 using boost::dynamic_pointer_cast;
@@ -41,7 +42,6 @@ Player::Player (shared_ptr<const Film> f, shared_ptr<const Playlist> p)
        , _audio (true)
        , _subtitles (true)
        , _have_valid_decoders (false)
-       , _video_sync (true)
 {
        _playlist->Changed.connect (bind (&Player::playlist_changed, this));
        _playlist->ContentChanged.connect (bind (&Player::content_changed, this, _1, _2));
@@ -75,73 +75,83 @@ Player::pass ()
        
        bool done = true;
        
-       if (_video_decoder != _video_decoders.end ()) {
-               if ((*_video_decoder)->pass ()) {
+       if (_video && _video_decoder < _video_decoders.size ()) {
+
+               /* Run video decoder; this may also produce audio */
+
+               if (_video_decoders[_video_decoder]->pass ()) {
                        _video_decoder++;
                }
                
-               if (_video_decoder != _video_decoders.end ()) {
+               if (_video_decoder < _video_decoders.size ()) {
+                       done = false;
+               }
+               
+       }
+
+       if (!_video && _audio && _playlist->audio_from() == Playlist::AUDIO_FFMPEG && _sequential_audio_decoder < _audio_decoders.size ()) {
+
+               /* We're not producing video, so we may need to run FFmpeg content to get the audio */
+               
+               if (_audio_decoders[_sequential_audio_decoder]->pass ()) {
+                       _sequential_audio_decoder++;
+               }
+               
+               if (_sequential_audio_decoder < _audio_decoders.size ()) {
                        done = false;
                }
+               
        }
 
-       if (_playlist->audio_from() == Playlist::AUDIO_SNDFILE) {
-               for (list<shared_ptr<SndfileDecoder> >::iterator i = _sndfile_decoders.begin(); i != _sndfile_decoders.end(); ++i) {
+       if (_audio && _playlist->audio_from() == Playlist::AUDIO_SNDFILE) {
+               
+               /* We're getting audio from SndfileContent */
+               
+               for (vector<shared_ptr<AudioDecoder> >::iterator i = _audio_decoders.begin(); i != _audio_decoders.end(); ++i) {
                        if (!(*i)->pass ()) {
                                done = false;
                        }
                }
 
-               Audio (_sndfile_buffers);
-               _sndfile_buffers.reset ();
+               Audio (_audio_buffers, _audio_time.get());
+               _audio_buffers.reset ();
+               _audio_time = boost::none;
        }
 
        return done;
 }
 
 void
-Player::set_progress (shared_ptr<Job> job)
-{
-       /* Assume progress can be divined from how far through the video we are */
-
-       if (_video_decoder == _video_decoders.end() || !_playlist->video_length()) {
-               return;
-       }
-       
-       ContentVideoFrame p = 0;
-       list<shared_ptr<VideoDecoder> >::iterator i = _video_decoders.begin ();
-       while (i != _video_decoders.end() && i != _video_decoder) {
-               p += (*i)->video_length ();
-       }
-
-       job->set_progress (float ((*_video_decoder)->video_frame ()) / _playlist->video_length ());
-}
-
-void
-Player::process_video (shared_ptr<Image> i, bool same, shared_ptr<Subtitle> s)
+Player::process_video (shared_ptr<const Image> i, bool same, shared_ptr<Subtitle> s, double t)
 {
-       Video (i, same, s);
+       Video (i, same, s, _video_start[_video_decoder] + t);
 }
 
 void
-Player::process_audio (weak_ptr<const AudioContent> c, shared_ptr<AudioBuffers> b)
+Player::process_audio (weak_ptr<const AudioContent> c, shared_ptr<const AudioBuffers> b, double t)
 {
-       if (_playlist->audio_from() == Playlist::AUDIO_SNDFILE) {
-               AudioMapping mapping = _film->audio_mapping ();
-               if (!_sndfile_buffers) {
-                       _sndfile_buffers.reset (new AudioBuffers (mapping.dcp_channels(), b->frames ()));
-                       _sndfile_buffers->make_silent ();
+       AudioMapping mapping = _film->audio_mapping ();
+       if (!_audio_buffers) {
+               _audio_buffers.reset (new AudioBuffers (mapping.dcp_channels(), b->frames ()));
+               _audio_buffers->make_silent ();
+               _audio_time = t;
+               if (_playlist->audio_from() == Playlist::AUDIO_FFMPEG) {
+                       _audio_time = _audio_time.get() + _audio_start[_sequential_audio_decoder];
                }
+       }
 
-               for (int i = 0; i < b->channels(); ++i) {
-                       list<libdcp::Channel> dcp = mapping.content_to_dcp (AudioMapping::Channel (c, i));
-                       for (list<libdcp::Channel>::iterator j = dcp.begin(); j != dcp.end(); ++j) {
-                               _sndfile_buffers->accumulate (b, i, static_cast<int> (*j));
-                       }
+       for (int i = 0; i < b->channels(); ++i) {
+               list<libdcp::Channel> dcp = mapping.content_to_dcp (AudioMapping::Channel (c, i));
+               for (list<libdcp::Channel>::iterator j = dcp.begin(); j != dcp.end(); ++j) {
+                       _audio_buffers->accumulate (b, i, static_cast<int> (*j));
                }
+       }
 
-       } else {
-               Audio (b);
+       if (_playlist->audio_from() == Playlist::AUDIO_FFMPEG) {
+               /* We can just emit this audio now as it will all be here */
+               Audio (_audio_buffers, t);
+               _audio_buffers.reset ();
+               _audio_time = boost::none;
        }
 }
 
@@ -154,19 +164,23 @@ Player::seek (double t)
                _have_valid_decoders = true;
        }
 
+       if (_video_decoders.empty ()) {
+               return true;
+       }
+
        /* Find the decoder that contains this position */
-       _video_decoder = _video_decoders.begin ();
-       while (_video_decoder != _video_decoders.end ()) {
-               double const this_length = double ((*_video_decoder)->video_length()) / _film->video_frame_rate ();
-               if (t < this_length) {
+       _video_decoder = 0;
+       while (1) {
+               ++_video_decoder;
+               if (_video_decoder >= _video_decoders.size () || t < _video_start[_video_decoder]) {
+                       --_video_decoder;
+                       t -= _video_start[_video_decoder];
                        break;
                }
-               t -= this_length;
-               ++_video_decoder;
        }
-       
-       if (_video_decoder != _video_decoders.end()) {
-               (*_video_decoder)->seek (t);
+
+       if (_video_decoder < _video_decoders.size()) {
+               _video_decoders[_video_decoder]->seek (t);
        } else {
                return true;
        }
@@ -176,18 +190,44 @@ Player::seek (double t)
        return false;
 }
 
+
+void
+Player::seek_back ()
+{
+       /* XXX */
+}
+
+void
+Player::seek_forward ()
+{
+       /* XXX */
+}
+
+
 void
 Player::setup_decoders ()
 {
+       vector<shared_ptr<VideoDecoder> > old_video_decoders = _video_decoders;
+
        _video_decoders.clear ();
-       _video_decoder = _video_decoders.end ();
-       _sndfile_decoders.clear ();
-       
-       if (_video) {
+       _video_decoder = 0;
+       _audio_decoders.clear ();
+       _sequential_audio_decoder = 0;
+
+       _video_start.clear();
+       _audio_start.clear();
+
+       double video_so_far = 0;
+       double audio_so_far = 0;
+
+       for (int l = 0; l < _playlist->loop(); ++l) {
                list<shared_ptr<const VideoContent> > vc = _playlist->video ();
                for (list<shared_ptr<const VideoContent> >::iterator i = vc.begin(); i != vc.end(); ++i) {
-
-                       shared_ptr<VideoDecoder> d;
+                       
+                       shared_ptr<const VideoContent> video_content;
+                       shared_ptr<const AudioContent> audio_content;
+                       shared_ptr<VideoDecoder> video_decoder;
+                       shared_ptr<AudioDecoder> audio_decoder;
                        
                        /* XXX: into content? */
                        
@@ -197,55 +237,76 @@ Player::setup_decoders ()
                                        new FFmpegDecoder (
                                                _film, fc, _video,
                                                _audio && _playlist->audio_from() == Playlist::AUDIO_FFMPEG,
-                                               _subtitles && _film->with_subtitles(),
-                                               _video_sync
+                                               _subtitles
                                                )
                                        );
+                               
+                               video_content = fc;
+                               audio_content = fc;
+                               video_decoder = fd;
+                               audio_decoder = fd;
 
-                               if (_playlist->audio_from() == Playlist::AUDIO_FFMPEG) {
-                                       fd->Audio.connect (bind (&Player::process_audio, this, fc, _1));
-                               }
-
-                               d = fd;
+                               video_decoder->connect_video (shared_from_this ());
                        }
-
+                       
                        shared_ptr<const ImageMagickContent> ic = dynamic_pointer_cast<const ImageMagickContent> (*i);
                        if (ic) {
-                               d.reset (new ImageMagickDecoder (_film, ic));
-                       }
+                               video_content = ic;
+
+                               /* See if we can re-use an old ImageMagickDecoder */
+                               for (vector<shared_ptr<VideoDecoder> >::const_iterator i = old_video_decoders.begin(); i != old_video_decoders.end(); ++i) {
+                                       shared_ptr<ImageMagickDecoder> imd = dynamic_pointer_cast<ImageMagickDecoder> (*i);
+                                       if (imd && imd->content() == ic) {
+                                               video_decoder = *i;
+                                       }
+                               }
 
-                       d->connect_video (shared_from_this ());
-                       _video_decoders.push_back (d);
+                               if (!video_decoder) {
+                                       video_decoder.reset (new ImageMagickDecoder (_film, ic));
+                                       video_decoder->connect_video (shared_from_this ());
+                               }
+                       }
+                       
+                       _video_decoders.push_back (video_decoder);
+                       _video_start.push_back (video_so_far);
+                       video_so_far += video_content->video_length() / video_content->video_frame_rate();
+                       
+                       if (audio_decoder && _playlist->audio_from() == Playlist::AUDIO_FFMPEG) {
+                               audio_decoder->Audio.connect (bind (&Player::process_audio, this, audio_content, _1, _2));
+                               _audio_decoders.push_back (audio_decoder);
+                               _audio_start.push_back (audio_so_far);
+                               audio_so_far += double(audio_content->audio_length()) / audio_content->audio_frame_rate();
+                       }
                }
-
-               _video_decoder = _video_decoders.begin ();
-       }
-
-       if (_audio && _playlist->audio_from() == Playlist::AUDIO_SNDFILE) {
-               list<shared_ptr<const SndfileContent> > sc = _playlist->sndfile ();
-               for (list<shared_ptr<const SndfileContent> >::iterator i = sc.begin(); i != sc.end(); ++i) {
-                       shared_ptr<SndfileDecoder> d (new SndfileDecoder (_film, *i));
-                       _sndfile_decoders.push_back (d);
-                       d->Audio.connect (bind (&Player::process_audio, this, *i, _1));
+               
+               _video_decoder = 0;
+               _sequential_audio_decoder = 0;
+               
+               if (_playlist->audio_from() == Playlist::AUDIO_SNDFILE) {
+                       
+                       list<shared_ptr<const AudioContent> > ac = _playlist->audio ();
+                       for (list<shared_ptr<const AudioContent> >::iterator i = ac.begin(); i != ac.end(); ++i) {
+                               
+                               shared_ptr<const SndfileContent> sc = dynamic_pointer_cast<const SndfileContent> (*i);
+                               assert (sc);
+                               
+                               shared_ptr<AudioDecoder> d (new SndfileDecoder (_film, sc));
+                               d->Audio.connect (bind (&Player::process_audio, this, sc, _1, _2));
+                               _audio_decoders.push_back (d);
+                               _audio_start.push_back (audio_so_far);
+                       }
                }
        }
 }
 
-void
-Player::disable_video_sync ()
-{
-       _video_sync = false;
-}
-
 double
 Player::last_video_time () const
 {
-       double t = 0;
-       for (list<shared_ptr<VideoDecoder> >::const_iterator i = _video_decoders.begin(); i != _video_decoder; ++i) {
-               t += (*i)->video_length() / (*i)->video_frame_rate ();
+       if (_video_decoder >= _video_decoders.size ()) {
+               return 0;
        }
-
-       return t + (*_video_decoder)->last_content_time ();
+       
+       return _video_start[_video_decoder] + _video_decoders[_video_decoder]->last_content_time ();
 }
 
 void
@@ -257,10 +318,7 @@ Player::content_changed (weak_ptr<Content> w, int p)
        }
 
        if (p == VideoContentProperty::VIDEO_LENGTH) {
-               if (dynamic_pointer_cast<FFmpegContent> (c)) {
-                       /* FFmpeg content length changes are serious; we need new decoders */
-                       _have_valid_decoders = false;
-               }
+               _have_valid_decoders = false;
        }
 }