Rename video/audio/subtitle part methods.

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

diff --git a/src/lib/transcoder.cc b/src/lib/transcoder.cc
index 959fac8572f55d5593fdd04b0f878bb99200c54f..ba004d634f4b6c7d2e933ce080e82b9dbfb10df3 100644 (file)
--- a/src/lib/transcoder.cc
+++ b/src/lib/transcoder.cc
@@ -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
@@ -24,107 +24,93 @@
  *  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 "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);
-
-       if (f->audio_stream()) {
-               shared_ptr<AudioStream> st = f->audio_stream();
-               _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()));
-       }
-
-       /* Set up the decoder to use the film's set streams */
-       _decoders.video->set_subtitle_stream (f->subtitle_stream ());
-       if (_decoders.audio) {
-               _decoders.audio->set_audio_stream (f->audio_stream ());
-       }
 
-       if (_matcher) {
-               _decoders.video->connect_video (_matcher);
-               _matcher->connect_video (_encoder);
-       } else {
-               _decoders.video->connect_video (_encoder);
-       }
-       
-       if (_matcher && _delay_line && _decoders.audio) {
-               _decoders.audio->connect_audio (_delay_line);
-               _delay_line->connect_audio (_matcher);
-               _matcher->connect_audio (_gain);
-               _gain->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 ();
 
-                       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;
-                       }
+       DCPTime const frame = DCPTime::from_frames (1, _film->video_frame_rate ());
+       DCPTime const length = _film->length ();
 
-                       if (done[0] && done[1]) {
-                               break;
+       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 {
+                               ++non_burnt_subtitles;
                        }
                }
-               
-       } catch (...) {
-               _encoder->process_end ();
-               throw;
        }
-       
-       if (_delay_line) {
-               _delay_line->process_end ();
+
+       if (non_burnt_subtitles) {
+               _writer->write (_player->get_subtitle_fonts ());
        }
-       if (_matcher) {
-               _matcher->process_end ();
+
+       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));
+               }
        }
-       if (_gain) {
-               _gain->process_end ();
+
+       BOOST_FOREACH (ReferencedReelAsset i, _player->get_reel_assets ()) {
+               _writer->write (i);
        }
-       _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 ();
 }