Merge master.
[dcpomatic.git] / src / lib / audio_decoder.cc
1 /*
2     Copyright (C) 2012-2014 Carl Hetherington <cth@carlh.net>
3
4     This program is free software; you can redistribute it and/or modify
5     it under the terms of the GNU General Public License as published by
6     the Free Software Foundation; either version 2 of the License, or
7     (at your option) any later version.
8
9     This program is distributed in the hope that it will be useful,
10     but WITHOUT ANY WARRANTY; without even the implied warranty of
11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12     GNU General Public License for more details.
13
14     You should have received a copy of the GNU General Public License
15     along with this program; if not, write to the Free Software
16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17
18 */
19
20 #include "audio_decoder.h"
21 #include "audio_buffers.h"
22 #include "exceptions.h"
23 #include "log.h"
24 #include "resampler.h"
25 #include "util.h"
26 #include "film.h"
27
28 #include "i18n.h"
29
30 using std::stringstream;
31 using std::list;
32 using std::pair;
33 using std::cout;
34 using std::min;
35 using boost::optional;
36 using boost::shared_ptr;
37
38 AudioDecoder::AudioDecoder (shared_ptr<const AudioContent> content)
39         : _audio_content (content)
40 {
41         if (content->output_audio_frame_rate() != content->content_audio_frame_rate() && content->audio_channels ()) {
42                 _resampler.reset (new Resampler (content->content_audio_frame_rate(), content->output_audio_frame_rate(), content->audio_channels ()));
43         }
44
45         reset_decoded_audio ();
46 }
47
48 void
49 AudioDecoder::reset_decoded_audio ()
50 {
51         _decoded_audio = ContentAudio (shared_ptr<AudioBuffers> (new AudioBuffers (_audio_content->audio_channels(), 0)), 0);
52 }
53
54 shared_ptr<ContentAudio>
55 AudioDecoder::get_audio (AudioFrame frame, AudioFrame length, bool accurate)
56 {
57         shared_ptr<ContentAudio> dec;
58
59         AudioFrame const end = frame + length - 1;
60                 
61         if (frame < _decoded_audio.frame || end > (_decoded_audio.frame + length * 4)) {
62                 /* Either we have no decoded data, or what we do have is a long way from what we want: seek */
63                 seek (ContentTime::from_frames (frame, _audio_content->content_audio_frame_rate()), accurate);
64         }
65
66         AudioFrame decoded_offset = 0;
67         
68         /* Now enough pass() calls will either:
69          *  (a) give us what we want, or
70          *  (b) hit the end of the decoder.
71          *
72          * If we are being accurate, we want the right frames,
73          * otherwise any frames will do.
74          */
75         if (accurate) {
76                 while (!pass() && _decoded_audio.audio->frames() < length) {}
77                 /* Use decoded_offset of 0, as we don't really care what frames we return */
78         } else {
79                 while (!pass() && (_decoded_audio.frame > frame || (_decoded_audio.frame + _decoded_audio.audio->frames()) < end)) {}
80                 decoded_offset = frame - _decoded_audio.frame;
81         }
82
83         AudioFrame const amount_left = _decoded_audio.audio->frames() - decoded_offset;
84         
85         AudioFrame const to_return = min (amount_left, length);
86         shared_ptr<AudioBuffers> out (new AudioBuffers (_decoded_audio.audio->channels(), to_return));
87         out->copy_from (_decoded_audio.audio.get(), to_return, decoded_offset, 0);
88         
89         /* Clean up decoded */
90         _decoded_audio.audio->move (decoded_offset + to_return, 0, amount_left - to_return);
91         _decoded_audio.audio->set_frames (amount_left - to_return);
92
93         return shared_ptr<ContentAudio> (new ContentAudio (out, frame));
94 }
95
96 /** Called by subclasses when audio data is ready.
97  *
98  *  Audio timestamping is made hard by many factors, but perhaps the most entertaining is resampling.
99  *  We have to assume that we are feeding continuous data into the resampler, and so we get continuous
100  *  data out.  Hence we do the timestamping here, post-resampler, just by counting samples.
101  *
102  *  The time is passed in here so that after a seek we can set up our _audio_position.  The
103  *  time is ignored once this has been done.
104  */
105 void
106 AudioDecoder::audio (shared_ptr<const AudioBuffers> data, ContentTime time)
107 {
108         if (_resampler) {
109                 data = _resampler->run (data);
110         }
111
112         if (!_audio_position) {
113                 _audio_position = time.frames (_audio_content->output_audio_frame_rate ());
114         }
115
116         assert (_audio_position.get() >= (_decoded_audio.frame + _decoded_audio.audio->frames()));
117
118         /* Resize _decoded_audio to fit the new data */
119         int const new_size = _audio_position.get() + data->frames() - _decoded_audio.frame;
120         _decoded_audio.audio->ensure_size (new_size);
121         _decoded_audio.audio->set_frames (new_size);
122
123         /* Copy new data in */
124         _decoded_audio.audio->copy_from (data.get(), data->frames(), 0, _audio_position.get() - _decoded_audio.frame);
125         _audio_position = _audio_position.get() + data->frames ();
126 }
127
128 /* XXX: called? */
129 void
130 AudioDecoder::flush ()
131 {
132         if (!_resampler) {
133                 return;
134         }
135
136         /*
137         shared_ptr<const AudioBuffers> b = _resampler->flush ();
138         if (b) {
139                 _pending.push_back (shared_ptr<DecodedAudio> (new DecodedAudio (b, _audio_position.get ())));
140                 _audio_position = _audio_position.get() + b->frames ();
141         }
142         */
143 }
144
145 void
146 AudioDecoder::seek (ContentTime, bool)
147 {
148         _audio_position.reset ();
149         reset_decoded_audio ();
150 }