Rename video/audio/subtitle part methods.
[dcpomatic.git] / src / lib / transcoder.cc
index 4d3f29e83010b13ef39100f1f5f6c9cc3bb327c4..ba004d634f4b6c7d2e933ce080e82b9dbfb10df3 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
 
     This program is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
  *  as a parameter to the constructor.
  */
 
-#include <iostream>
-#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 "trimmer.h"
+#include "player.h"
+#include "job.h"
+#include "writer.h"
+#include "compose.hpp"
+#include "referenced_reel_asset.h"
+#include "subtitle_content.h"
+#include <boost/signals2.hpp>
+#include <boost/foreach.hpp>
+#include <iostream>
 
 using std::string;
+using std::cout;
+using std::list;
 using boost::shared_ptr;
+using boost::weak_ptr;
 using boost::dynamic_pointer_cast;
 
-/** Construct a transcoder using a Decoder that we create and a supplied Encoder.
+/** Construct a transcoder.
  *  @param f Film that we are transcoding.
- *  @param o Decode options.
- *  @param j Job that we are running under, or 0.
- *  @param e Encoder to use.
+ *  @param j Job that this transcoder is being used in.
  */
-Transcoder::Transcoder (shared_ptr<Film> f, DecodeOptions o, Job* j, shared_ptr<Encoder> e)
-       : _job (j)
-       , _encoder (e)
-       , _decoders (decoder_factory (f, o))
+Transcoder::Transcoder (shared_ptr<const Film> film, shared_ptr<Job> j)
+       : _film (film)
+       , _player (new Player (film, film->playlist ()))
+       , _writer (new Writer (film, j))
+       , _encoder (new Encoder (film, _writer))
+       , _finishing (false)
 {
-       assert (_encoder);
-
-       shared_ptr<AudioStream> st = f->audio_stream();
-       if (st) {
-               _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 */
-       _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);
-       
-       _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
- *  has no more data to present.
- */
 void
 Transcoder::go ()
 {
-       _encoder->process_begin ();
-       try {
-               bool done[2] = { false, false };
-               
-               while (1) {
-                       if (!done[0]) {
-                               done[0] = _decoders.video->pass ();
-                               if (_job) {
-                                       _decoders.video->set_progress (_job);
-                               }
-                       }
+       _writer->start ();
+       _encoder->begin ();
+
+       DCPTime const frame = DCPTime::from_frames (1, _film->video_frame_rate ());
+       DCPTime const length = _film->length ();
 
-                       if (!done[1] && _decoders.audio && dynamic_pointer_cast<Decoder> (_decoders.audio) != dynamic_pointer_cast<Decoder> (_decoders.video)) {
-                               done[1] = _decoders.audio->pass ();
+       int burnt_subtitles = 0;
+       int non_burnt_subtitles = 0;
+       BOOST_FOREACH (shared_ptr<const Content> c, _film->content ()) {
+               if (c->subtitle && c->subtitle->use()) {
+                       if (c->subtitle->burn()) {
+                               ++burnt_subtitles;
                        } else {
-                               done[1] = true;
+                               ++non_burnt_subtitles;
                        }
+               }
+       }
 
-                       if (done[0] && done[1]) {
-                               break;
-                       }
+       if (non_burnt_subtitles) {
+               _writer->write (_player->get_subtitle_fonts ());
+       }
+
+       for (DCPTime t; t < length; t += frame) {
+               _encoder->encode (_player->get_video (t, true));
+               _writer->write (_player->get_audio (t, frame, true));
+
+               if (non_burnt_subtitles) {
+                       _writer->write (_player->get_subtitles (t, frame, true, false, true));
                }
-               
-       } catch (...) {
-               _encoder->process_end ();
-               throw;
        }
-       
-       _delay_line->process_end ();
-       if (_matcher) {
-               _matcher->process_end ();
+
+       BOOST_FOREACH (ReferencedReelAsset i, _player->get_reel_assets ()) {
+               _writer->write (i);
        }
-       _gain->process_end ();
-       _encoder->process_end ();
+
+       _finishing = true;
+       _encoder->end ();
+       _writer->finish ();
+}
+
+float
+Transcoder::current_encoding_rate () const
+{
+       return _encoder->current_encoding_rate ();
+}
+
+int
+Transcoder::video_frames_out () const
+{
+       return _encoder->video_frames_out ();
 }