Set AudioDecoder::fast a different way.
[dcpomatic.git] / src / lib / dcp_decoder.cc
1 /*
2     Copyright (C) 2014-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 #include "dcp_decoder.h"
22 #include "dcp_content.h"
23 #include "audio_content.h"
24 #include "video_decoder.h"
25 #include "audio_decoder.h"
26 #include "j2k_image_proxy.h"
27 #include "subtitle_decoder.h"
28 #include "image.h"
29 #include "config.h"
30 #include <dcp/dcp.h>
31 #include <dcp/decrypted_kdm.h>
32 #include <dcp/cpl.h>
33 #include <dcp/reel.h>
34 #include <dcp/mono_picture_asset.h>
35 #include <dcp/mono_picture_asset_reader.h>
36 #include <dcp/stereo_picture_asset.h>
37 #include <dcp/stereo_picture_asset_reader.h>
38 #include <dcp/reel_picture_asset.h>
39 #include <dcp/reel_sound_asset.h>
40 #include <dcp/reel_subtitle_asset.h>
41 #include <dcp/mono_picture_frame.h>
42 #include <dcp/stereo_picture_frame.h>
43 #include <dcp/sound_frame.h>
44 #include <dcp/sound_asset_reader.h>
45 #include <boost/foreach.hpp>
46 #include <iostream>
47
48 using std::list;
49 using std::cout;
50 using boost::shared_ptr;
51 using boost::dynamic_pointer_cast;
52
53 DCPDecoder::DCPDecoder (shared_ptr<const DCPContent> c, shared_ptr<Log> log)
54         : _dcp_content (c)
55 {
56         video.reset (new VideoDecoder (this, c, log));
57         audio.reset (new AudioDecoder (this, c->audio, log));
58
59         subtitle.reset (
60                 new SubtitleDecoder (
61                         this,
62                         c->subtitle,
63                         bind (&DCPDecoder::image_subtitles_during, this, _1, _2),
64                         bind (&DCPDecoder::text_subtitles_during, this, _1, _2)
65                         )
66                 );
67
68         dcp::DCP dcp (c->directory ());
69         dcp.read (false, 0, true);
70         if (c->kdm ()) {
71                 dcp.add (dcp::DecryptedKDM (c->kdm().get (), Config::instance()->decryption_chain()->key().get ()));
72         }
73         DCPOMATIC_ASSERT (dcp.cpls().size() == 1);
74         _reels = dcp.cpls().front()->reels ();
75
76         _reel = _reels.begin ();
77         _offset = 0;
78         get_readers ();
79 }
80
81 bool
82 DCPDecoder::pass (PassReason reason, bool)
83 {
84         if (_reel == _reels.end () || !_dcp_content->can_be_played ()) {
85                 return true;
86         }
87
88         double const vfr = _dcp_content->active_video_frame_rate ();
89
90         /* Frame within the (played part of the) reel that is coming up next */
91         int64_t const frame = _next.frames_round (vfr);
92
93         if ((_mono_reader || _stereo_reader) && reason != PASS_REASON_SUBTITLE) {
94                 shared_ptr<dcp::PictureAsset> asset = (*_reel)->main_picture()->asset ();
95                 int64_t const entry_point = (*_reel)->main_picture()->entry_point ();
96                 if (_mono_reader) {
97                         video->give (shared_ptr<ImageProxy> (new J2KImageProxy (_mono_reader->get_frame (entry_point + frame), asset->size())), _offset + frame);
98                 } else {
99                         video->give (
100                                 shared_ptr<ImageProxy> (new J2KImageProxy (_stereo_reader->get_frame (entry_point + frame), asset->size(), dcp::EYE_LEFT)),
101                                 _offset + frame
102                                 );
103
104                         video->give (
105                                 shared_ptr<ImageProxy> (new J2KImageProxy (_stereo_reader->get_frame (entry_point + frame), asset->size(), dcp::EYE_RIGHT)),
106                                 _offset + frame
107                                 );
108                 }
109         }
110
111         if (_sound_reader && reason != PASS_REASON_SUBTITLE) {
112                 int64_t const entry_point = (*_reel)->main_sound()->entry_point ();
113                 shared_ptr<const dcp::SoundFrame> sf = _sound_reader->get_frame (entry_point + frame);
114                 uint8_t const * from = sf->data ();
115
116                 int const channels = _dcp_content->audio->stream()->channels ();
117                 int const frames = sf->size() / (3 * channels);
118                 shared_ptr<AudioBuffers> data (new AudioBuffers (channels, frames));
119                 float** data_data = data->data();
120                 for (int i = 0; i < frames; ++i) {
121                         for (int j = 0; j < channels; ++j) {
122                                 data_data[j][i] = static_cast<int> ((from[0] << 8) | (from[1] << 16) | (from[2] << 24)) / static_cast<float> (INT_MAX - 256);
123                                 from += 3;
124                         }
125                 }
126
127                 audio->give (_dcp_content->audio->stream(), data, ContentTime::from_frames (_offset, vfr) + _next);
128         }
129
130         if ((*_reel)->main_subtitle ()) {
131                 int64_t const entry_point = (*_reel)->main_subtitle()->entry_point ();
132                 list<dcp::SubtitleString> subs = (*_reel)->main_subtitle()->asset()->subtitles_during (
133                         dcp::Time (entry_point + frame, vfr, vfr),
134                         dcp::Time (entry_point + frame + 1, vfr, vfr),
135                         true
136                         );
137
138                 if (!subs.empty ()) {
139                         /* XXX: assuming that all `subs' are at the same time; maybe this is ok */
140                         subtitle->give_text (
141                                 ContentTimePeriod (
142                                         ContentTime::from_frames (_offset - entry_point, vfr) + ContentTime::from_seconds (subs.front().in().as_seconds ()),
143                                         ContentTime::from_frames (_offset - entry_point, vfr) + ContentTime::from_seconds (subs.front().out().as_seconds ())
144                                         ),
145                                 subs
146                                 );
147                 }
148         }
149
150         _next += ContentTime::from_frames (1, vfr);
151
152         if ((*_reel)->main_picture ()) {
153                 if (_next.frames_round (vfr) >= (*_reel)->main_picture()->duration()) {
154                         next_reel ();
155                         _next = ContentTime ();
156                 }
157         }
158
159         return false;
160 }
161
162 void
163 DCPDecoder::next_reel ()
164 {
165         _offset += (*_reel)->main_picture()->duration();
166         ++_reel;
167         get_readers ();
168 }
169
170 void
171 DCPDecoder::get_readers ()
172 {
173         if (_reel == _reels.end()) {
174                 _mono_reader.reset ();
175                 _stereo_reader.reset ();
176                 _sound_reader.reset ();
177                 return;
178         }
179
180         if ((*_reel)->main_picture()) {
181                 shared_ptr<dcp::PictureAsset> asset = (*_reel)->main_picture()->asset ();
182                 shared_ptr<dcp::MonoPictureAsset> mono = dynamic_pointer_cast<dcp::MonoPictureAsset> (asset);
183                 shared_ptr<dcp::StereoPictureAsset> stereo = dynamic_pointer_cast<dcp::StereoPictureAsset> (asset);
184                 DCPOMATIC_ASSERT (mono || stereo);
185                 if (mono) {
186                         _mono_reader = mono->start_read ();
187                         _stereo_reader.reset ();
188                 } else {
189                         _stereo_reader = stereo->start_read ();
190                         _mono_reader.reset ();
191                 }
192         } else {
193                 _mono_reader.reset ();
194                 _stereo_reader.reset ();
195         }
196
197         if ((*_reel)->main_sound()) {
198                 _sound_reader = (*_reel)->main_sound()->asset()->start_read ();
199         } else {
200                 _sound_reader.reset ();
201         }
202 }
203
204 void
205 DCPDecoder::seek (ContentTime t, bool accurate)
206 {
207         video->seek (t, accurate);
208         audio->seek (t, accurate);
209         subtitle->seek (t, accurate);
210
211         _offset = 0;
212         _reel = _reels.begin ();
213         while (_reel != _reels.end() && t >= ContentTime::from_frames ((*_reel)->main_picture()->duration(), _dcp_content->active_video_frame_rate ())) {
214                 t -= ContentTime::from_frames ((*_reel)->main_picture()->duration(), _dcp_content->active_video_frame_rate ());
215                 next_reel ();
216         }
217
218         _next = t;
219 }
220
221
222 list<ContentTimePeriod>
223 DCPDecoder::image_subtitles_during (ContentTimePeriod, bool) const
224 {
225         return list<ContentTimePeriod> ();
226 }
227
228 list<ContentTimePeriod>
229 DCPDecoder::text_subtitles_during (ContentTimePeriod period, bool starting) const
230 {
231         /* XXX: inefficient */
232
233         list<ContentTimePeriod> ctp;
234         double const vfr = _dcp_content->active_video_frame_rate ();
235
236         BOOST_FOREACH (shared_ptr<dcp::Reel> r, _reels) {
237                 if (!r->main_subtitle ()) {
238                         continue;
239                 }
240
241                 int64_t const entry_point = r->main_subtitle()->entry_point ();
242
243                 list<dcp::SubtitleString> subs = r->main_subtitle()->asset()->subtitles_during (
244                         dcp::Time (period.from.seconds(), 1000) - dcp::Time (entry_point, vfr, vfr),
245                         dcp::Time (period.to.seconds(), 1000) - dcp::Time (entry_point, vfr, vfr),
246                         starting
247                         );
248
249                 BOOST_FOREACH (dcp::SubtitleString const & s, subs) {
250                         ctp.push_back (
251                                 ContentTimePeriod (
252                                         ContentTime::from_seconds (s.in().as_seconds ()),
253                                         ContentTime::from_seconds (s.out().as_seconds ())
254                                         )
255                                 );
256                 }
257         }
258
259         return ctp;
260 }