Various hacks.
[dcpomatic.git] / src / lib / audio_decoder.cc
1 /*
2     Copyright (C) 2012 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
25 #include "i18n.h"
26
27 using std::stringstream;
28 using boost::optional;
29 using boost::shared_ptr;
30
31 AudioDecoder::AudioDecoder (shared_ptr<const Film> f, shared_ptr<const AudioContent> c)
32         : Decoder (f)
33         , _next_audio (0)
34         , _audio_content (c)
35         , _output_audio_frame_rate (_audio_content->output_audio_frame_rate (f))
36 {
37         if (_audio_content->content_audio_frame_rate() != _output_audio_frame_rate) {
38
39                 stringstream s;
40                 s << String::compose ("Will resample audio from %1 to %2", _audio_content->content_audio_frame_rate(), _output_audio_frame_rate);
41                 _film->log()->log (s.str ());
42
43                 /* We will be using planar float data when we call the
44                    resampler.  As far as I can see, the audio channel
45                    layout is not necessary for our purposes; it seems
46                    only to be used get the number of channels and
47                    decide if rematrixing is needed.  It won't be, since
48                    input and output layouts are the same.
49                 */
50
51                 _swr_context = swr_alloc_set_opts (
52                         0,
53                         av_get_default_channel_layout (MAX_AUDIO_CHANNELS),
54                         AV_SAMPLE_FMT_FLTP,
55                         _output_audio_frame_rate,
56                         av_get_default_channel_layout (MAX_AUDIO_CHANNELS),
57                         AV_SAMPLE_FMT_FLTP,
58                         _audio_content->content_audio_frame_rate(),
59                         0, 0
60                         );
61                 
62                 swr_init (_swr_context);
63         } else {
64                 _swr_context = 0;
65         }
66 }
67
68 AudioDecoder::~AudioDecoder ()
69 {
70         if (_swr_context) {
71                 swr_free (&_swr_context);
72         }
73 }
74         
75
76 #if 0
77 void
78 AudioDecoder::process_end ()
79 {
80         if (_swr_context) {
81
82                 shared_ptr<AudioBuffers> out (new AudioBuffers (_film->audio_mapping().dcp_channels(), 256));
83                         
84                 while (1) {
85                         int const frames = swr_convert (_swr_context, (uint8_t **) out->data(), 256, 0, 0);
86
87                         if (frames < 0) {
88                                 throw EncodeError (_("could not run sample-rate converter"));
89                         }
90
91                         if (frames == 0) {
92                                 break;
93                         }
94
95                         out->set_frames (frames);
96                         _writer->write (out);
97                 }
98
99         }
100 }
101 #endif
102
103 void
104 AudioDecoder::emit_audio (shared_ptr<const AudioBuffers> data, Time time)
105 {
106         /* XXX: map audio to 5.1 */
107         
108         /* Maybe resample */
109         if (_swr_context) {
110
111                 /* Compute the resampled frames count and add 32 for luck */
112                 int const max_resampled_frames = ceil ((int64_t) data->frames() * _output_audio_frame_rate / _audio_content->content_audio_frame_rate()) + 32;
113
114                 shared_ptr<AudioBuffers> resampled (new AudioBuffers (MAX_AUDIO_CHANNELS, max_resampled_frames));
115
116                 /* Resample audio */
117                 int const resampled_frames = swr_convert (
118                         _swr_context, (uint8_t **) resampled->data(), max_resampled_frames, (uint8_t const **) data->data(), data->frames()
119                         );
120                 
121                 if (resampled_frames < 0) {
122                         throw EncodeError (_("could not run sample-rate converter"));
123                 }
124
125                 resampled->set_frames (resampled_frames);
126                 
127                 /* And point our variables at the resampled audio */
128                 data = resampled;
129         }
130
131         Audio (data, time);
132
133         _next_audio = time + _film->audio_frames_to_time (data->frames());
134 }
135
136