Some tidying up. Do encode progress in the writer to improve progress bar movement...
[dcpomatic.git] / src / lib / player.cc
index 7a149d734ca61fff7f173c548f9548586a8893b5..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;
@@ -74,35 +75,39 @@ Player::pass ()
        
        bool done = true;
        
-       if (_video_decoder != _video_decoders.end ()) {
+       if (_video && _video_decoder < _video_decoders.size ()) {
 
                /* Run video decoder; this may also produce audio */
-               
-               if ((*_video_decoder)->pass ()) {
+
+               if (_video_decoders[_video_decoder]->pass ()) {
                        _video_decoder++;
                }
                
-               if (_video_decoder != _video_decoders.end ()) {
+               if (_video_decoder < _video_decoders.size ()) {
                        done = false;
                }
                
-       } else if (!_video && _playlist->audio_from() == Playlist::AUDIO_FFMPEG && _sequential_audio_decoder != _audio_decoders.end ()) {
+       }
+
+       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 ((*_sequential_audio_decoder)->pass ()) {
+               if (_audio_decoders[_sequential_audio_decoder]->pass ()) {
                        _sequential_audio_decoder++;
                }
                
-               if (_sequential_audio_decoder != _audio_decoders.end ()) {
+               if (_sequential_audio_decoder < _audio_decoders.size ()) {
                        done = false;
                }
                
-       } else if (_playlist->audio_from() == Playlist::AUDIO_SNDFILE) {
+       }
 
+       if (_audio && _playlist->audio_from() == Playlist::AUDIO_SNDFILE) {
+               
                /* We're getting audio from SndfileContent */
                
-               for (list<shared_ptr<AudioDecoder> >::iterator i = _audio_decoders.begin(); i != _audio_decoders.end(); ++i) {
+               for (vector<shared_ptr<AudioDecoder> >::iterator i = _audio_decoders.begin(); i != _audio_decoders.end(); ++i) {
                        if (!(*i)->pass ()) {
                                done = false;
                        }
@@ -117,39 +122,22 @@ Player::pass ()
 }
 
 void
-Player::set_progress (shared_ptr<Job> job)
+Player::process_video (shared_ptr<const Image> i, bool same, shared_ptr<Subtitle> s, double t)
 {
-       /* 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 ());
+       Video (i, same, s, _video_start[_video_decoder] + t);
 }
 
 void
-Player::process_video (shared_ptr<Image> i, bool same, shared_ptr<Subtitle> s, double t)
+Player::process_audio (weak_ptr<const AudioContent> c, shared_ptr<const AudioBuffers> b, double t)
 {
-       /* XXX: this time will need mangling to add on the offset of the start of the content */
-       Video (i, same, s, t);
-}
-
-void
-Player::process_audio (weak_ptr<const AudioContent> c, shared_ptr<AudioBuffers> b, double t)
-{
-       /* XXX: this time will need mangling to add on the offset of the start of the content */
        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) {
@@ -176,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;
        }
@@ -215,15 +207,27 @@ Player::seek_forward ()
 void
 Player::setup_decoders ()
 {
+       vector<shared_ptr<VideoDecoder> > old_video_decoders = _video_decoders;
+
        _video_decoders.clear ();
-       _video_decoder = _video_decoders.end ();
+       _video_decoder = 0;
        _audio_decoders.clear ();
-       
-       if (_video) {
+       _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? */
                        
@@ -236,64 +240,61 @@ Player::setup_decoders ()
                                                _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, _2));
-                               }
-
-                               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));
-                       }
-
-                       d->connect_video (shared_from_this ());
-                       _video_decoders.push_back (d);
-               }
-
-               _video_decoder = _video_decoders.begin ();
-       }
-
-       if (_playlist->audio_from() == Playlist::AUDIO_FFMPEG && !_video) {
-
-               /* If we're getting audio from FFmpegContent but not the video, we need a set
-                  of decoders for the audio.
-               */
-               
-               list<shared_ptr<const AudioContent> > ac = _playlist->audio ();
-               for (list<shared_ptr<const AudioContent> >::iterator i = ac.begin(); i != ac.end(); ++i) {
+                               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;
+                                       }
+                               }
 
-                       shared_ptr<const FFmpegContent> fc = dynamic_pointer_cast<const FFmpegContent> (*i);
-                       assert (fc);
+                               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();
                        
-                       shared_ptr<AudioDecoder> d (
-                               new FFmpegDecoder (
-                                       _film, fc, _video,
-                                       _audio && _playlist->audio_from() == Playlist::AUDIO_FFMPEG,
-                                       _subtitles
-                                       )
-                               );
-
-                       d->Audio.connect (bind (&Player::process_audio, this, fc, _1, _2));
-                       _audio_decoders.push_back (d);
+                       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();
+                       }
                }
-
-               _sequential_audio_decoder = _audio_decoders.begin ();
-       }
-
-       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);
+               _video_decoder = 0;
+               _sequential_audio_decoder = 0;
+               
+               if (_playlist->audio_from() == Playlist::AUDIO_SNDFILE) {
                        
-                       shared_ptr<AudioDecoder> d (new SndfileDecoder (_film, sc));
-                       d->Audio.connect (bind (&Player::process_audio, this, sc, _1, _2));
-                       _audio_decoders.push_back (d);
+                       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);
+                       }
                }
        }
 }
@@ -301,12 +302,11 @@ Player::setup_decoders ()
 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
@@ -318,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;
        }
 }