src/lib/transcoder.cc

   1 /*
   2     Copyright (C) 2012-2016 Carl Hetherington <cth@carlh.net>
   3
   4     This file is part of DCP-o-matic.
   5
   6     DCP-o-matic is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     DCP-o-matic is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 */
  20
  21 /** @file  src/transcoder.cc
  22  *  @brief A class which takes a Film and some Options, then uses those to transcode the film.
  23  *
  24  *  A decoder is selected according to the content type, and the encoder can be specified
  25  *  as a parameter to the constructor.
  26  */
  27
  28 #include "transcoder.h"
  29 #include "encoder.h"
  30 #include "film.h"
  31 #include "video_decoder.h"
  32 #include "audio_decoder.h"
  33 #include "player.h"
  34 #include "job.h"
  35 #include "writer.h"
  36 #include "compose.hpp"
  37 #include "referenced_reel_asset.h"
  38 #include "subtitle_content.h"
  39 #include "player_video.h"
  40 #include <boost/signals2.hpp>
  41 #include <boost/foreach.hpp>
  42 #include <iostream>
  43
  44 #include "i18n.h"
  45
  46 using std::string;
  47 using std::cout;
  48 using std::list;
  49 using boost::shared_ptr;
  50 using boost::weak_ptr;
  51 using boost::dynamic_pointer_cast;
  52
  53 /** Construct a transcoder.
  54  *  @param f Film that we are transcoding.
  55  *  @param j Job that this transcoder is being used in.
  56  */
  57 Transcoder::Transcoder (shared_ptr<const Film> film, weak_ptr<Job> j)
  58         : _film (film)
  59         , _job (j)
  60         , _player (new Player (film, film->playlist ()))
  61         , _writer (new Writer (film, j))
  62         , _encoder (new Encoder (film, _writer))
  63         , _finishing (false)
  64 {
  65
  66 }
  67
  68 void
  69 Transcoder::go ()
  70 {
  71         _writer->start ();
  72         _encoder->begin ();
  73
  74         {
  75                 shared_ptr<Job> job = _job.lock ();
  76                 DCPOMATIC_ASSERT (job);
  77                 job->sub (_("Encoding picture and sound"));
  78         }
  79
  80         DCPTime const frame = DCPTime::from_frames (1, _film->video_frame_rate ());
  81         DCPTime const length = _film->length ();
  82
  83         int burnt_subtitles = 0;
  84         int non_burnt_subtitles = 0;
  85         BOOST_FOREACH (shared_ptr<const Content> c, _film->content ()) {
  86                 if (c->subtitle && c->subtitle->use()) {
  87                         if (c->subtitle->burn()) {
  88                                 ++burnt_subtitles;
  89                         } else {
  90                                 ++non_burnt_subtitles;
  91                         }
  92                 }
  93         }
  94
  95         if (non_burnt_subtitles) {
  96                 _writer->write (_player->get_subtitle_fonts ());
  97         }
  98
  99         for (DCPTime t; t < length; t += frame) {
 100
 101                 BOOST_FOREACH (shared_ptr<PlayerVideo> i, _player->get_video (t, true)) {
 102                         if (!_film->three_d()) {
 103                                 /* 2D DCP */
 104                                 if (i->eyes() == EYES_RIGHT) {
 105                                         /* Discard right-eye images */
 106                                         continue;
 107                                 } else if (i->eyes() == EYES_LEFT) {
 108                                         /* Use left-eye images for both eyes */
 109                                         i->set_eyes (EYES_BOTH);
 110                                 }
 111                         }
 112
 113                         _encoder->encode (i);
 114                 }
 115
 116                 _writer->write (_player->get_audio (t, frame, true));
 117
 118                 if (non_burnt_subtitles) {
 119                         _writer->write (_player->get_subtitles (t, frame, true, false, true));
 120                 }
 121
 122                 {
 123                         shared_ptr<Job> job = _job.lock ();
 124                         DCPOMATIC_ASSERT (job);
 125                         job->set_progress (float(t.get()) / length.get());
 126                 }
 127         }
 128
 129         BOOST_FOREACH (ReferencedReelAsset i, _player->get_reel_assets ()) {
 130                 _writer->write (i);
 131         }
 132
 133         _finishing = true;
 134         _encoder->end ();
 135         _writer->finish ();
 136 }
 137
 138 float
 139 Transcoder::current_encoding_rate () const
 140 {
 141         return _encoder->current_encoding_rate ();
 142 }
 143
 144 int
 145 Transcoder::video_frames_enqueued () const
 146 {
 147         return _encoder->video_frames_enqueued ();
 148 }