Pad silence for things that already have at least some audio.

[dcpomatic.git] / src / lib / transcoder.cc

diff --git a/src/lib/transcoder.cc b/src/lib/transcoder.cc
index ac908768e34a14d2c20eede1390756fd06a14438..a202d440c0c736cd63d0c24521a2e1ab28984f73 100644 (file)
--- a/src/lib/transcoder.cc
+++ b/src/lib/transcoder.cc
@@ -30,29 +30,70 @@
 #include "encoder.h"
 #include "decoder_factory.h"
 #include "film.h"
+#include "matcher.h"
+#include "delay_line.h"
+#include "options.h"
+#include "gain.h"
+#include "video_decoder.h"
+#include "audio_decoder.h"
+#include "trimmer.h"
 
 using std::string;
 using boost::shared_ptr;
+using boost::dynamic_pointer_cast;
 
 /** Construct a transcoder using a Decoder that we create and a supplied Encoder.
  *  @param f Film that we are transcoding.
- *  @param o Options.
+ *  @param o Decode options.
  *  @param j Job that we are running under, or 0.
  *  @param e Encoder to use.
  */
-Transcoder::Transcoder (shared_ptr<Film> f, shared_ptr<const Options> o, Job* j, shared_ptr<Encoder> e)
+Transcoder::Transcoder (shared_ptr<Film> f, DecodeOptions o, Job* j, shared_ptr<Encoder> e)
        : _job (j)
        , _encoder (e)
-       , _decoder (decoder_factory (f, o, j))
+       , _decoders (decoder_factory (f, o))
 {
        assert (_encoder);
 
+       shared_ptr<AudioStream> st = f->audio_stream();
+       if (st && st->sample_rate ()) {
+               _matcher.reset (new Matcher (f->log(), st->sample_rate(), f->source_frame_rate()));
+       }
+       _delay_line.reset (new DelayLine (f->log(), f->audio_delay() / 1000.0f));
+       _gain.reset (new Gain (f->log(), f->audio_gain()));
+
+       int const sr = st ? st->sample_rate() : 0;
+       int const trim_start = f->trim_type() == Film::ENCODE ? f->trim_start() : 0;
+       int const trim_end = f->trim_type() == Film::ENCODE ? f->trim_end() : 0;
+       _trimmer.reset (new Trimmer (
+                               f->log(), trim_start, trim_end, f->length().get_value_or(0),
+                               sr, f->source_frame_rate(), f->dcp_frame_rate()
+                               ));
+
        /* Set up the decoder to use the film's set streams */
-       _decoder->set_audio_stream (f->audio_stream ());
-       _decoder->set_subtitle_stream (f->subtitle_stream ());
+       _decoders.video->set_subtitle_stream (f->subtitle_stream ());
+       if (f->audio_stream ()) {
+               _decoders.audio->set_audio_stream (f->audio_stream ());
+       }
+
+       _decoders.video->connect_video (_delay_line);
+       if (_matcher) {
+               _delay_line->connect_video (_matcher);
+               _matcher->connect_video (_trimmer);
+       } else {
+               _delay_line->connect_video (_trimmer);
+       }
+       _trimmer->connect_video (_encoder);
        
-       _decoder->Video.connect (bind (&Encoder::process_video, e, _1, _2, _3));
-       _decoder->Audio.connect (bind (&Encoder::process_audio, e, _1, _2));
+       _decoders.audio->connect_audio (_delay_line);
+       if (_matcher) {
+               _delay_line->connect_audio (_matcher);
+               _matcher->connect_audio (_gain);
+       } else {
+               _delay_line->connect_audio (_gain);
+       }
+       _gain->connect_audio (_trimmer);
+       _trimmer->connect_audio (_encoder);
 }
 
 /** Run the decoder, passing its output to the encoder, until the decoder
@@ -62,15 +103,32 @@ void
 Transcoder::go ()
 {
        _encoder->process_begin ();
-       try {
-               _decoder->go ();
-       } catch (...) {
-               /* process_end() is important as the decoder may have worker
-                  threads that need to be cleaned up.
-               */
-               _encoder->process_end ();
-               throw;
-       }
 
+       bool done[2] = { false, false };
+       
+       while (1) {
+               if (!done[0]) {
+                       done[0] = _decoders.video->pass ();
+                       if (_job) {
+                               _decoders.video->set_progress (_job);
+                       }
+               }
+               
+               if (!done[1] && _decoders.audio && dynamic_pointer_cast<Decoder> (_decoders.audio) != dynamic_pointer_cast<Decoder> (_decoders.video)) {
+                       done[1] = _decoders.audio->pass ();
+               } else {
+                       done[1] = true;
+               }
+               
+               if (done[0] && done[1]) {
+                       break;
+               }
+       }
+       
+       _delay_line->process_end ();
+       if (_matcher) {
+               _matcher->process_end ();
+       }
+       _gain->process_end ();
        _encoder->process_end ();
 }