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
  38 using std::string;
  39 using boost::shared_ptr;
  40
  41 /** Construct a transcoder using a Decoder that we create and a supplied Encoder.
  42  *  @param f Film that we are transcoding.
  43  *  @param o Options.
  44  *  @param j Job that we are running under, or 0.
  45  *  @param e Encoder to use.
  46  */
  47 Transcoder::Transcoder (shared_ptr<Film> f, shared_ptr<const Options> o, Job* j, shared_ptr<Encoder> e)
  48         : _job (j)
  49         , _encoder (e)
  50         , _decoder (decoder_factory (f, o, j))
  51 {
  52         assert (_encoder);
  53
  54         if (f->audio_stream()) {
  55                 AudioStream st = f->audio_stream().get();
  56                 _matcher.reset (new Matcher (f->log(), st.sample_rate(), f->frames_per_second()));
  57                 _delay_line.reset (new DelayLine (f->log(), st.channels(), f->audio_delay() * st.sample_rate() / 1000));
  58                 _gain.reset (new Gain (f->log(), f->audio_gain()));
  59         }
  60
  61         /* Set up the decoder to use the film's set streams */
  62         _decoder->set_audio_stream (f->audio_stream ());
  63         _decoder->set_subtitle_stream (f->subtitle_stream ());
  64
  65         if (_matcher) {
  66                 _decoder->connect_video (_matcher);
  67                 _matcher->connect_video (_encoder);
  68         } else {
  69                 _decoder->connect_video (_encoder);
  70         }
  71
  72         if (_matcher && _delay_line) {
  73                 _decoder->connect_audio (_delay_line);
  74                 _delay_line->connect_audio (_matcher);
  75                 _matcher->connect_audio (_gain);
  76                 _gain->connect_audio (_encoder);
  77         }
  78 }
  79
  80 /** Run the decoder, passing its output to the encoder, until the decoder
  81  *  has no more data to present.
  82  */
  83 void
  84 Transcoder::go ()
  85 {
  86         _encoder->process_begin ();
  87         try {
  88                 _decoder->go ();
  89         } catch (...) {
  90                 /* process_end() is important as the decoder may have worker
  91                    threads that need to be cleaned up.
  92                 */
  93                 _encoder->process_end ();
  94                 throw;
  95         }
  96
  97         _encoder->process_end ();
  98 }