Try to allow users to specify that the film's header should be trusted wrt length...
authorCarl Hetherington <cth@carlh.net>
Mon, 10 Dec 2012 22:20:39 +0000 (22:20 +0000)
committerCarl Hetherington <cth@carlh.net>
Mon, 10 Dec 2012 22:20:39 +0000 (22:20 +0000)
src/lib/examine_content_job.cc
src/lib/ffmpeg_decoder.cc
src/lib/ffmpeg_decoder.h
src/lib/film.cc
src/lib/imagemagick_decoder.h
src/lib/transcoder.cc
src/lib/transcoder.h
src/lib/video_decoder.h
wscript

index 8db74801fdd634feed5e2a822c4d4f97691884b6..93333605b8e6a6983fc63fd668f420bd426fae35 100644 (file)
@@ -60,51 +60,78 @@ ExamineContentJob::name () const
 void
 ExamineContentJob::run ()
 {
-       /* Decode the content to get an accurate length */
+       float progress_remaining = 1;
 
-       /* We don't want to use any existing length here, as progress
-          will be messed up.
+       /* Set the film's length to either
+          a) a length judged by running through the content or
+          b) the length from a decoder's header.
        */
-       _film->unset_length ();
-       
-       shared_ptr<Options> o (new Options ("", "", ""));
-       o->out_size = Size (512, 512);
-       o->apply_crop = false;
-       o->decode_audio = false;
-
-       descend (0.5);
 
-       pair<shared_ptr<VideoDecoder>, shared_ptr<AudioDecoder> > decoders = decoder_factory (_film, o, this);
+       if (!_film->trust_content_header()) {
+               /* Decode the content to get an accurate length */
+               
+               /* We don't want to use any existing length here, as progress
+                  will be messed up.
+               */
+               _film->unset_length ();
+               
+               shared_ptr<Options> o (new Options ("", "", ""));
+               o->out_size = Size (512, 512);
+               o->apply_crop = false;
+               o->decode_audio = false;
+               
+               descend (0.5);
+               
+               pair<shared_ptr<VideoDecoder>, shared_ptr<AudioDecoder> > decoders = decoder_factory (_film, o, this);
+               
+               set_progress_unknown ();
+               while (!decoders.first->pass()) {
+                       /* keep going */
+               }
+               
+               _film->set_length (decoders.first->video_frame());
+               
+               _film->log()->log (String::compose ("Video length examined as %1 frames", _film->length().get()));
+               
+               ascend ();
+               
+               progress_remaining -= 0.5;
+               
+       } else {
 
-       set_progress_unknown ();
-       while (!decoders.first->pass()) {
-               /* keep going */
+               /* Get a quick decoder to get the content's length from its header.
+                  It would have been nice to just use the thumbnail transcoder's decoder,
+                  but that's a bit fiddly, and this isn't too expensive.
+               */
+               
+               shared_ptr<Options> o (new Options ("", "", ""));
+               o->out_size = Size (1024, 1024);
+               pair<shared_ptr<VideoDecoder>, shared_ptr<AudioDecoder> > d = decoder_factory (_film, o, 0);
+               _film->set_length (d.first->length());
+       
+               _film->log()->log (String::compose ("Video length obtained from header as %1 frames", _film->length().get()));
        }
 
-       _film->set_length (decoders.first->video_frame());
-
-       _film->log()->log (String::compose ("Video length is %1 frames", _film->length()));
-
-       ascend ();
-
        /* Now make thumbnails for it */
 
-       descend (0.5);
+       descend (progress_remaining);
 
        try {
-               o.reset (new Options (_film->dir ("thumbs"), ".png", ""));
+               shared_ptr<Options> o (new Options (_film->dir ("thumbs"), ".png", ""));
                o->out_size = _film->size ();
                o->apply_crop = false;
                o->decode_audio = false;
-               if (_film->length() > 0) {
-                       o->decode_video_skip = _film->length().get() / 128;
-               } else {
-                       o->decode_video_skip = 0;
-               }
+               o->decode_video_skip = _film->length().get() / 128;
                o->decode_subtitles = true;
                shared_ptr<ImageMagickEncoder> e (new ImageMagickEncoder (_film, o));
                Transcoder w (_film, o, this, e);
                w.go ();
+
+               /* Now set the film's length from the transcoder's decoder, since we
+                  went to all the trouble of going through the content.
+               */
+
+               _film->set_length (w.video_decoder()->video_frame());
                
        } catch (std::exception& e) {
 
index ef9c05fa6e14b200a77b22092d7dc9991f55fdbb..fd522a5ac374bc10116bbd916c6a27dce0f611ea 100644 (file)
@@ -629,3 +629,9 @@ FFmpegAudioStream::to_string () const
        return String::compose ("ffmpeg %1 %2 %3 %4", _id, _sample_rate, _channel_layout, _name);
 }
 
+/** @return Length (in video frames) according to our content's header */
+SourceFrame
+FFmpegDecoder::length () const
+{
+       return (double(_format_context->duration) / AV_TIME_BASE) * frames_per_second();
+}
index 87eebe1ecc4ba7b83f92fb23f096435caf5bdd2b..1771551fcf1080cb5bb7b38b5d0744d3992c3d65 100644 (file)
@@ -56,7 +56,7 @@ public:
                , _name (n)
                , _id (i)
        {}
-                 
+
        std::string to_string () const;
 
        std::string name () const {
@@ -89,6 +89,7 @@ public:
 
        float frames_per_second () const;
        Size native_size () const;
+       SourceFrame length () const;
        int time_base_numerator () const;
        int time_base_denominator () const;
        int sample_aspect_ratio_numerator () const;
index 563147f68bdf47a6dfce4d28e0dde22dbb1737ec..e2a4cbeda60efda1089f2b8e840c6a64940f5702 100644 (file)
@@ -1003,7 +1003,7 @@ Film::set_trust_content_header (bool t)
        
        signal_changed (TRUST_CONTENT_HEADER);
 
-       if (!_trust_content_header) {
+       if (!_trust_content_header) && !content().empty()) {
                /* We just said that we don't trust the content's header */
                examine_content ();
        }
index f636191f2e3bdbb6da13d799c13a496cabbd6885..de49c1b566b6ef0a36ae2bb44c0e05d77a65da6c 100644 (file)
@@ -35,6 +35,11 @@ public:
 
        Size native_size () const;
 
+       SourceFrame length () const {
+               /* We don't know */
+               return 0;
+       }
+
        int audio_channels () const {
                return 0;
        }
index 537b9b66452d6f4b8e30dbe4f824b9983166fb0b..a7e79b05f74d6338cd2c5debfe6d77381b905839 100644 (file)
@@ -64,7 +64,9 @@ Transcoder::Transcoder (shared_ptr<Film> f, shared_ptr<const Options> o, Job* j,
 
        /* Set up the decoder to use the film's set streams */
        _decoders.first->set_subtitle_stream (f->subtitle_stream ());
-       _decoders.second->set_audio_stream (f->audio_stream ());
+       if (_decoders.second) {
+               _decoders.second->set_audio_stream (f->audio_stream ());
+       }
 
        if (_matcher) {
                _decoders.first->connect_video (_matcher);
@@ -73,7 +75,7 @@ Transcoder::Transcoder (shared_ptr<Film> f, shared_ptr<const Options> o, Job* j,
                _decoders.first->connect_video (_encoder);
        }
        
-       if (_matcher && _delay_line) {
+       if (_matcher && _delay_line && _decoders.second) {
                _decoders.second->connect_audio (_delay_line);
                _delay_line->connect_audio (_matcher);
                _matcher->connect_audio (_gain);
@@ -97,7 +99,7 @@ Transcoder::go ()
                                _decoders.first->set_progress ();
                        }
 
-                       if (!done[1] && dynamic_pointer_cast<Decoder> (_decoders.second) != dynamic_pointer_cast<Decoder> (_decoders.first)) {
+                       if (!done[1] && _decoders.second && dynamic_pointer_cast<Decoder> (_decoders.second) != dynamic_pointer_cast<Decoder> (_decoders.first)) {
                                done[1] = _decoders.second->pass ();
                        } else {
                                done[1] = true;
index e3ca2bb32085b128bb0c47673bcab7b55fed804a..4a9667b3c5165b95aa9f8aeea2786b206f184b56 100644 (file)
@@ -49,6 +49,10 @@ public:
 
        void go ();
 
+       boost::shared_ptr<VideoDecoder> video_decoder () const {
+               return _decoders.first;
+       }
+
 protected:
        /** A Job that is running this Transcoder, or 0 */
        Job* _job;
index ea1899840a7252a6358c682a0a1e85167bb3c7bd..685138a58c761d05307a71adb5b5dfdb74fc9889 100644 (file)
@@ -33,6 +33,8 @@ public:
        virtual float frames_per_second () const = 0;
        /** @return native size in pixels */
        virtual Size native_size () const = 0;
+       /** @return length (in source video frames), according to our content's header */
+       virtual SourceFrame length () const = 0;
 
        virtual int time_base_numerator () const = 0;
        virtual int time_base_denominator () const = 0;
diff --git a/wscript b/wscript
index b0ca5d49ddd4c7b60f837fda00e6b52c5b4c24ee..290bf9b52496e033a232a45bb3fc5db19d5c140a 100644 (file)
--- a/wscript
+++ b/wscript
@@ -3,7 +3,7 @@ import os
 import sys
 
 APPNAME = 'dvdomatic'
-VERSION = '0.59beta5'
+VERSION = '0.60pre'
 
 def options(opt):
     opt.load('compiler_cxx')