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