eb787dee82c52ca3795cb5a145170710e6d1cd85
[dcpomatic.git] / src / lib / transcoder.cc
1 /*
2     Copyright (C) 2012-2015 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 <boost/signals2.hpp>
40 #include <boost/foreach.hpp>
41 #include <iostream>
42
43 using std::string;
44 using std::cout;
45 using std::list;
46 using boost::shared_ptr;
47 using boost::weak_ptr;
48 using boost::dynamic_pointer_cast;
49
50 /** Construct a transcoder.
51  *  @param f Film that we are transcoding.
52  *  @param j Job that this transcoder is being used in.
53  */
54 Transcoder::Transcoder (shared_ptr<const Film> film, shared_ptr<Job> j)
55         : _film (film)
56         , _player (new Player (film, film->playlist ()))
57         , _writer (new Writer (film, j))
58         , _encoder (new Encoder (film, _writer))
59         , _finishing (false)
60 {
61
62 }
63
64 void
65 Transcoder::go ()
66 {
67         _writer->start ();
68         _encoder->begin ();
69
70         DCPTime const frame = DCPTime::from_frames (1, _film->video_frame_rate ());
71         DCPTime const length = _film->length ();
72
73         int burnt_subtitles = 0;
74         int non_burnt_subtitles = 0;
75         BOOST_FOREACH (shared_ptr<const Content> c, _film->content ()) {
76                 if (c->subtitle && c->subtitle->use()) {
77                         if (c->subtitle->burn()) {
78                                 ++burnt_subtitles;
79                         } else {
80                                 ++non_burnt_subtitles;
81                         }
82                 }
83         }
84
85         if (non_burnt_subtitles) {
86                 _writer->write (_player->get_subtitle_fonts ());
87         }
88
89         for (DCPTime t; t < length; t += frame) {
90                 _encoder->encode (_player->get_video (t, true));
91                 _writer->write (_player->get_audio (t, frame, true));
92
93                 if (non_burnt_subtitles) {
94                         _writer->write (_player->get_subtitles (t, frame, true, false, true));
95                 }
96         }
97
98         BOOST_FOREACH (ReferencedReelAsset i, _player->get_reel_assets ()) {
99                 _writer->write (i);
100         }
101
102         _finishing = true;
103         _encoder->end ();
104         _writer->finish ();
105 }
106
107 float
108 Transcoder::current_encoding_rate () const
109 {
110         return _encoder->current_encoding_rate ();
111 }
112
113 int
114 Transcoder::video_frames_out () const
115 {
116         return _encoder->video_frames_out ();
117 }