Replace a shared_ptr with a weak_ptr.
[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, weak_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         {
74                 shared_ptr<Job> job = _job.lock ();
75                 DCPOMATIC_ASSERT (job);
76                 job->sub (_("Encoding picture and sound"));
77         }
78
79         DCPTime const frame = DCPTime::from_frames (1, _film->video_frame_rate ());
80         DCPTime const length = _film->length ();
81
82         int burnt_subtitles = 0;
83         int non_burnt_subtitles = 0;
84         BOOST_FOREACH (shared_ptr<const Content> c, _film->content ()) {
85                 if (c->subtitle && c->subtitle->use()) {
86                         if (c->subtitle->burn()) {
87                                 ++burnt_subtitles;
88                         } else {
89                                 ++non_burnt_subtitles;
90                         }
91                 }
92         }
93
94         if (non_burnt_subtitles) {
95                 _writer->write (_player->get_subtitle_fonts ());
96         }
97
98         for (DCPTime t; t < length; t += frame) {
99                 _encoder->encode (_player->get_video (t, true));
100                 _writer->write (_player->get_audio (t, frame, true));
101
102                 if (non_burnt_subtitles) {
103                         _writer->write (_player->get_subtitles (t, frame, true, false, true));
104                 }
105
106                 {
107                         shared_ptr<Job> job = _job.lock ();
108                         DCPOMATIC_ASSERT (job);
109                         job->set_progress (float(t.get()) / length.get());
110                 }
111         }
112
113         BOOST_FOREACH (ReferencedReelAsset i, _player->get_reel_assets ()) {
114                 _writer->write (i);
115         }
116
117         _finishing = true;
118         _encoder->end ();
119         _writer->finish ();
120 }
121
122 float
123 Transcoder::current_encoding_rate () const
124 {
125         return _encoder->current_encoding_rate ();
126 }
127
128 int
129 Transcoder::video_frames_out () const
130 {
131         return _encoder->video_frames_out ();
132 }