src/lib/transcoder.cc

   1 /*
   2     Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 /** @file  src/transcoder.cc
  21  *  @brief A class which takes a Film and some Options, then uses those to transcode the film.
  22  *
  23  *  A decoder is selected according to the content type, and the encoder can be specified
  24  *  as a parameter to the constructor.
  25  */
  26
  27 #include <iostream>
  28 #include <boost/signals2.hpp>
  29 #include "transcoder.h"
  30 #include "encoder.h"
  31 #include "decoder_factory.h"
  32 #include "film.h"
  33 #include "matcher.h"
  34 #include "delay_line.h"
  35 #include "options.h"
  36 #include "gain.h"
  37 #include "video_decoder.h"
  38 #include "audio_decoder.h"
  39 #include "trimmer.h"
  40
  41 using std::string;
  42 using boost::shared_ptr;
  43 using boost::dynamic_pointer_cast;
  44
  45 /** Construct a transcoder using a Decoder that we create and a supplied Encoder.
  46  *  @param f Film that we are transcoding.
  47  *  @param o Decode options.
  48  *  @param j Job that we are running under, or 0.
  49  *  @param e Encoder to use.
  50  */
  51 Transcoder::Transcoder (shared_ptr<Film> f, DecodeOptions o, Job* j, shared_ptr<Encoder> e)
  52         : _job (j)
  53         , _encoder (e)
  54         , _decoders (decoder_factory (f, o))
  55 {
  56         assert (_encoder);
  57
  58         shared_ptr<AudioStream> st = f->audio_stream();
  59         if (st && st->sample_rate ()) {
  60                 _matcher.reset (new Matcher (f->log(), st->sample_rate(), f->source_frame_rate()));
  61         }
  62         _delay_line.reset (new DelayLine (f->log(), f->audio_delay() / 1000.0f));
  63         _gain.reset (new Gain (f->log(), f->audio_gain()));
  64
  65         int const sr = st ? st->sample_rate() : 0;
  66         int const trim_start = f->trim_type() == Film::ENCODE ? f->trim_start() : 0;
  67         int const trim_end = f->trim_type() == Film::ENCODE ? f->trim_end() : 0;
  68         _trimmer.reset (new Trimmer (
  69                                 f->log(), trim_start, trim_end, f->length().get_value_or(0),
  70                                 sr, f->source_frame_rate(), f->dcp_frame_rate()
  71                                 ));
  72
  73         /* Set up the decoder to use the film's set streams */
  74         _decoders.video->set_subtitle_stream (f->subtitle_stream ());
  75         if (f->audio_stream ()) {
  76                 _decoders.audio->set_audio_stream (f->audio_stream ());
  77         }
  78
  79         _decoders.video->connect_video (_delay_line);
  80         if (_matcher) {
  81                 _delay_line->connect_video (_matcher);
  82                 _matcher->connect_video (_trimmer);
  83         } else {
  84                 _delay_line->connect_video (_trimmer);
  85         }
  86         _trimmer->connect_video (_encoder);
  87
  88         _decoders.audio->connect_audio (_delay_line);
  89         if (_matcher) {
  90                 _delay_line->connect_audio (_matcher);
  91                 _matcher->connect_audio (_gain);
  92         } else {
  93                 _delay_line->connect_audio (_gain);
  94         }
  95         _gain->connect_audio (_trimmer);
  96         _trimmer->connect_audio (_encoder);
  97 }
  98
  99 /** Run the decoder, passing its output to the encoder, until the decoder
 100  *  has no more data to present.
 101  */
 102 void
 103 Transcoder::go ()
 104 {
 105         _encoder->process_begin ();
 106
 107         bool done[2] = { false, false };
 108
 109         while (1) {
 110                 if (!done[0]) {
 111                         done[0] = _decoders.video->pass ();
 112                         if (_job) {
 113                                 _decoders.video->set_progress (_job);
 114                         }
 115                 }
 116
 117                 if (!done[1] && _decoders.audio && dynamic_pointer_cast<Decoder> (_decoders.audio) != dynamic_pointer_cast<Decoder> (_decoders.video)) {
 118                         done[1] = _decoders.audio->pass ();
 119                 } else {
 120                         done[1] = true;
 121                 }
 122
 123                 if (done[0] && done[1]) {
 124                         break;
 125                 }
 126         }
 127
 128         _delay_line->process_end ();
 129         if (_matcher) {
 130                 _matcher->process_end ();
 131         }
 132         _gain->process_end ();
 133         _encoder->process_end ();
 134 }