Merge master; fix crash on new film.
[dcpomatic.git] / src / lib / transcoder.cc
index 32725f77caaca32e5d07b47fe066de4c639f6b50..2e33931bd50872cc5edbb16c274af19e336a5bb1 100644 (file)
 #include <boost/signals2.hpp>
 #include "transcoder.h"
 #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 "player.h"
+#include "trimmer.h"
 
 using std::string;
 using boost::shared_ptr;
@@ -43,71 +43,68 @@ 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 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, shared_ptr<Job> j)
        : _job (j)
-       , _encoder (e)
-       , _decoders (decoder_factory (f, o, j))
+       , _player (f->player ())
+       , _encoder (new Encoder (f))
 {
-       assert (_encoder);
+       _matcher.reset (new Matcher (f->log(), f->audio_frame_rate(), f->video_frame_rate()));
+       _delay_line.reset (new DelayLine (f->log(), f->audio_delay() * f->audio_frame_rate() / 1000));
+       _gain.reset (new Gain (f->log(), f->audio_gain()));
 
-       if (f->audio_stream()) {
-               AudioStream st = f->audio_stream().get();
-               _matcher.reset (new Matcher (f->log(), st.sample_rate(), f->frames_per_second()));
-               _delay_line.reset (new DelayLine (f->log(), st.channels(), f->audio_delay() * st.sample_rate() / 1000));
-               _gain.reset (new Gain (f->log(), f->audio_gain()));
+       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->content_length(),
+                               f->audio_frame_rate(), f->video_frame_rate(), f->dcp_frame_rate()
+                               ));
+       
+       if (!f->with_subtitles ()) {
+               _player->disable_subtitles ();
        }
 
-       /* 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 (_matcher) {
-               _decoders.first->connect_video (_matcher);
-               _matcher->connect_video (_encoder);
-       } else {
-               _decoders.first->connect_video (_encoder);
-       }
+       _player->connect_video (_delay_line);
+       _delay_line->connect_video (_matcher);
+       _matcher->connect_video (_trimmer);
+       _trimmer->connect_video (_encoder);
        
-       if (_matcher && _delay_line) {
-               _decoders.second->connect_audio (_delay_line);
-               _delay_line->connect_audio (_matcher);
-               _matcher->connect_audio (_gain);
-               _gain->connect_audio (_encoder);
-       }
+       _player->connect_audio (_delay_line);
+       _delay_line->connect_audio (_matcher);
+       _matcher->connect_audio (_gain);
+       _gain->connect_audio (_trimmer);
+       _trimmer->connect_audio (_encoder);
 }
 
-/** Run the decoder, passing its output to the encoder, until the decoder
- *  has no more data to present.
- */
 void
 Transcoder::go ()
 {
        _encoder->process_begin ();
-       try {
-               while (1) {
-                       bool const v = _decoders.first->pass ();
-
-                       bool a = false;
-                       if (dynamic_pointer_cast<Decoder> (_decoders.second) != dynamic_pointer_cast<Decoder> (_decoders.first)) {
-                               a = _decoders.second->pass ();
-                       }
-
-                       if (v && a) {
-                               break;
-                       }
+       while (1) {
+               if (_player->pass ()) {
+                       break;
                }
-               
-       } catch (...) {
-               /* process_end() is important as the decoder may have worker
-                  threads that need to be cleaned up.
-               */
-               _encoder->process_end ();
-               throw;
+               _player->set_progress (_job);
        }
 
+       _delay_line->process_end ();
+       if (_matcher) {
+               _matcher->process_end ();
+       }
+       _gain->process_end ();
        _encoder->process_end ();
 }
+
+float
+Transcoder::current_encoding_rate () const
+{
+       return _encoder->current_encoding_rate ();
+}
+
+int
+Transcoder::video_frames_out () const
+{
+       return _encoder->video_frames_out ();
+}