/src/ffmpeg/libavcodec/pcm-bluray.c
Line | Count | Source |
1 | | /* |
2 | | * LPCM codecs for PCM format found in Blu-ray PCM streams |
3 | | * Copyright (c) 2009, 2013 Christian Schmidt |
4 | | * |
5 | | * This file is part of FFmpeg. |
6 | | * |
7 | | * FFmpeg is free software; you can redistribute it and/or |
8 | | * modify it under the terms of the GNU Lesser General Public |
9 | | * License as published by the Free Software Foundation; either |
10 | | * version 2.1 of the License, or (at your option) any later version. |
11 | | * |
12 | | * FFmpeg is distributed in the hope that it will be useful, |
13 | | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 | | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
15 | | * Lesser General Public License for more details. |
16 | | * |
17 | | * You should have received a copy of the GNU Lesser General Public |
18 | | * License along with FFmpeg; if not, write to the Free Software |
19 | | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
20 | | */ |
21 | | |
22 | | /** |
23 | | * @file |
24 | | * PCM codec for Blu-ray PCM audio tracks |
25 | | */ |
26 | | |
27 | | #include "libavutil/channel_layout.h" |
28 | | #include "avcodec.h" |
29 | | #include "bytestream.h" |
30 | | #include "codec_internal.h" |
31 | | #include "decode.h" |
32 | | |
33 | | /* |
34 | | * Channel Mapping according to |
35 | | * Blu-ray Disc Read-Only Format Version 1 |
36 | | * Part 3: Audio Visual Basic Specifications |
37 | | * mono M1 X |
38 | | * stereo L R |
39 | | * 3/0 L R C X |
40 | | * 2/1 L R S X |
41 | | * 3/1 L R C S |
42 | | * 2/2 L R LS RS |
43 | | * 3/2 L R C LS RS X |
44 | | * 3/2+lfe L R C LS RS lfe |
45 | | * 3/4 L R C LS Rls Rrs RS X |
46 | | * 3/4+lfe L R C LS Rls Rrs RS lfe |
47 | | */ |
48 | | |
49 | | /** |
50 | | * Parse the header of a LPCM frame read from a Blu-ray MPEG-TS stream |
51 | | * @param avctx the codec context |
52 | | * @param header pointer to the first four bytes of the data packet |
53 | | */ |
54 | | static int pcm_bluray_parse_header(AVCodecContext *avctx, |
55 | | const uint8_t *header) |
56 | 219k | { |
57 | 219k | static const uint8_t bits_per_samples[4] = { 0, 16, 20, 24 }; |
58 | 219k | static const AVChannelLayout channel_layouts[16] = { |
59 | 219k | { 0 }, AV_CHANNEL_LAYOUT_MONO, { 0 }, |
60 | 219k | AV_CHANNEL_LAYOUT_STEREO, AV_CHANNEL_LAYOUT_SURROUND, AV_CHANNEL_LAYOUT_2_1, |
61 | 219k | AV_CHANNEL_LAYOUT_4POINT0, AV_CHANNEL_LAYOUT_2_2, AV_CHANNEL_LAYOUT_5POINT0, |
62 | 219k | AV_CHANNEL_LAYOUT_5POINT1, AV_CHANNEL_LAYOUT_7POINT0, AV_CHANNEL_LAYOUT_7POINT1, |
63 | 219k | { 0 }, { 0 }, { 0 }, { 0 }, |
64 | 219k | }; |
65 | 219k | uint8_t channel_layout = header[2] >> 4; |
66 | | |
67 | 219k | if (avctx->debug & FF_DEBUG_PICT_INFO) |
68 | 0 | ff_dlog(avctx, "pcm_bluray_parse_header: header = %02x%02x%02x%02x\n", |
69 | 219k | header[0], header[1], header[2], header[3]); |
70 | | |
71 | | /* get the sample depth and derive the sample format from it */ |
72 | 219k | avctx->bits_per_coded_sample = bits_per_samples[header[3] >> 6]; |
73 | 219k | if (!(avctx->bits_per_coded_sample == 16 || avctx->bits_per_coded_sample == 24)) { |
74 | 1.96k | av_log(avctx, AV_LOG_ERROR, "unsupported sample depth (%d)\n", avctx->bits_per_coded_sample); |
75 | 1.96k | return AVERROR_INVALIDDATA; |
76 | 1.96k | } |
77 | 217k | avctx->sample_fmt = avctx->bits_per_coded_sample == 16 ? AV_SAMPLE_FMT_S16 |
78 | 217k | : AV_SAMPLE_FMT_S32; |
79 | 217k | if (avctx->sample_fmt == AV_SAMPLE_FMT_S32) |
80 | 10.9k | avctx->bits_per_raw_sample = avctx->bits_per_coded_sample; |
81 | | |
82 | | /* get the sample rate. Not all values are used. */ |
83 | 217k | switch (header[2] & 0x0f) { |
84 | 205k | case 1: |
85 | 205k | avctx->sample_rate = 48000; |
86 | 205k | break; |
87 | 4.36k | case 4: |
88 | 4.36k | avctx->sample_rate = 96000; |
89 | 4.36k | break; |
90 | 5.45k | case 5: |
91 | 5.45k | avctx->sample_rate = 192000; |
92 | 5.45k | break; |
93 | 2.52k | default: |
94 | 2.52k | avctx->sample_rate = 0; |
95 | 2.52k | av_log(avctx, AV_LOG_ERROR, "reserved sample rate (%d)\n", |
96 | 2.52k | header[2] & 0x0f); |
97 | 2.52k | return AVERROR_INVALIDDATA; |
98 | 217k | } |
99 | | |
100 | | /* |
101 | | * get the channel number (and mapping). Not all values are used. |
102 | | * It must be noted that the number of channels in the MPEG stream can |
103 | | * differ from the actual meaningful number, e.g. mono audio still has two |
104 | | * channels, one being empty. |
105 | | */ |
106 | 214k | av_channel_layout_uninit(&avctx->ch_layout); |
107 | 214k | avctx->ch_layout = channel_layouts[channel_layout]; |
108 | 214k | if (!avctx->ch_layout.nb_channels) { |
109 | 259 | av_log(avctx, AV_LOG_ERROR, "reserved channel configuration (%d)\n", |
110 | 259 | channel_layout); |
111 | 259 | return AVERROR_INVALIDDATA; |
112 | 259 | } |
113 | | |
114 | 214k | avctx->bit_rate = FFALIGN(avctx->ch_layout.nb_channels, 2) * avctx->sample_rate * |
115 | 214k | avctx->bits_per_coded_sample; |
116 | | |
117 | 214k | if (avctx->debug & FF_DEBUG_PICT_INFO) |
118 | 0 | ff_dlog(avctx, |
119 | 214k | "pcm_bluray_parse_header: %d channels, %d bits per sample, %d Hz, %"PRId64" bit/s\n", |
120 | 214k | avctx->ch_layout.nb_channels, avctx->bits_per_coded_sample, |
121 | 214k | avctx->sample_rate, avctx->bit_rate); |
122 | 214k | return 0; |
123 | 214k | } |
124 | | |
125 | | static int pcm_bluray_decode_frame(AVCodecContext *avctx, AVFrame *frame, |
126 | | int *got_frame_ptr, AVPacket *avpkt) |
127 | 354k | { |
128 | 354k | const uint8_t *src = avpkt->data; |
129 | 354k | int buf_size = avpkt->size; |
130 | 354k | GetByteContext gb; |
131 | 354k | int num_source_channels, channel, retval; |
132 | 354k | int sample_size, samples; |
133 | 354k | int16_t *dst16; |
134 | 354k | int32_t *dst32; |
135 | | |
136 | 354k | if (buf_size < 4) { |
137 | 135k | av_log(avctx, AV_LOG_ERROR, "PCM packet too small\n"); |
138 | 135k | return AVERROR_INVALIDDATA; |
139 | 135k | } |
140 | | |
141 | 219k | if ((retval = pcm_bluray_parse_header(avctx, src))) |
142 | 4.75k | return retval; |
143 | 214k | src += 4; |
144 | 214k | buf_size -= 4; |
145 | | |
146 | 214k | bytestream2_init(&gb, src, buf_size); |
147 | | |
148 | | /* There's always an even number of channels in the source */ |
149 | 214k | num_source_channels = FFALIGN(avctx->ch_layout.nb_channels, 2); |
150 | 214k | sample_size = (num_source_channels * |
151 | 214k | (avctx->sample_fmt == AV_SAMPLE_FMT_S16 ? 16 : 24)) >> 3; |
152 | 214k | samples = buf_size / sample_size; |
153 | | |
154 | | /* get output buffer */ |
155 | 214k | frame->nb_samples = samples; |
156 | 214k | if ((retval = ff_get_buffer(avctx, frame, 0)) < 0) |
157 | 7.43k | return retval; |
158 | 207k | dst16 = (int16_t *)frame->data[0]; |
159 | 207k | dst32 = (int32_t *)frame->data[0]; |
160 | | |
161 | 207k | if (samples) { |
162 | 207k | switch (avctx->ch_layout.u.mask) { |
163 | | /* cases with same number of source and coded channels */ |
164 | 2.10k | case AV_CH_LAYOUT_STEREO: |
165 | 2.52k | case AV_CH_LAYOUT_4POINT0: |
166 | 2.86k | case AV_CH_LAYOUT_2_2: |
167 | 2.86k | samples *= num_source_channels; |
168 | 2.86k | if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) { |
169 | | #if HAVE_BIGENDIAN |
170 | | bytestream2_get_buffer(&gb, (uint8_t*)dst16, buf_size); |
171 | | #else |
172 | 760k | do { |
173 | 760k | *dst16++ = bytestream2_get_be16u(&gb); |
174 | 760k | } while (--samples); |
175 | 2.58k | #endif |
176 | 2.58k | } else { |
177 | 4.91k | do { |
178 | 4.91k | *dst32++ = bytestream2_get_be24u(&gb) << 8; |
179 | 4.91k | } while (--samples); |
180 | 278 | } |
181 | 2.86k | break; |
182 | | /* cases where number of source channels = coded channels + 1 */ |
183 | 200k | case AV_CH_LAYOUT_MONO: |
184 | 200k | case AV_CH_LAYOUT_SURROUND: |
185 | 202k | case AV_CH_LAYOUT_2_1: |
186 | 202k | case AV_CH_LAYOUT_5POINT0: |
187 | 202k | if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) { |
188 | 3.42M | do { |
189 | | #if HAVE_BIGENDIAN |
190 | | bytestream2_get_buffer(&gb, (uint8_t*)dst16, |
191 | | avctx->ch_layout.nb_channels * 2); |
192 | | dst16 += avctx->ch_layout.nb_channels; |
193 | | #else |
194 | 3.42M | channel = avctx->ch_layout.nb_channels; |
195 | 3.50M | do { |
196 | 3.50M | *dst16++ = bytestream2_get_be16u(&gb); |
197 | 3.50M | } while (--channel); |
198 | 3.42M | #endif |
199 | 3.42M | bytestream2_skip(&gb, 2); |
200 | 3.42M | } while (--samples); |
201 | 198k | } else { |
202 | 578k | do { |
203 | 578k | channel = avctx->ch_layout.nb_channels; |
204 | 610k | do { |
205 | 610k | *dst32++ = bytestream2_get_be24u(&gb) << 8; |
206 | 610k | } while (--channel); |
207 | 578k | bytestream2_skip(&gb, 3); |
208 | 578k | } while (--samples); |
209 | 4.03k | } |
210 | 202k | break; |
211 | | /* remapping: L, R, C, LBack, RBack, LF */ |
212 | 497 | case AV_CH_LAYOUT_5POINT1: |
213 | 497 | if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) { |
214 | 11.3k | do { |
215 | 11.3k | dst16[0] = bytestream2_get_be16u(&gb); |
216 | 11.3k | dst16[1] = bytestream2_get_be16u(&gb); |
217 | 11.3k | dst16[2] = bytestream2_get_be16u(&gb); |
218 | 11.3k | dst16[4] = bytestream2_get_be16u(&gb); |
219 | 11.3k | dst16[5] = bytestream2_get_be16u(&gb); |
220 | 11.3k | dst16[3] = bytestream2_get_be16u(&gb); |
221 | 11.3k | dst16 += 6; |
222 | 11.3k | } while (--samples); |
223 | 255 | } else { |
224 | 1.08k | do { |
225 | 1.08k | dst32[0] = bytestream2_get_be24u(&gb) << 8; |
226 | 1.08k | dst32[1] = bytestream2_get_be24u(&gb) << 8; |
227 | 1.08k | dst32[2] = bytestream2_get_be24u(&gb) << 8; |
228 | 1.08k | dst32[4] = bytestream2_get_be24u(&gb) << 8; |
229 | 1.08k | dst32[5] = bytestream2_get_be24u(&gb) << 8; |
230 | 1.08k | dst32[3] = bytestream2_get_be24u(&gb) << 8; |
231 | 1.08k | dst32 += 6; |
232 | 1.08k | } while (--samples); |
233 | 255 | } |
234 | 497 | break; |
235 | | /* remapping: L, R, C, LSide, LBack, RBack, RSide, <unused> */ |
236 | 896 | case AV_CH_LAYOUT_7POINT0: |
237 | 896 | if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) { |
238 | 114k | do { |
239 | 114k | dst16[0] = bytestream2_get_be16u(&gb); |
240 | 114k | dst16[1] = bytestream2_get_be16u(&gb); |
241 | 114k | dst16[2] = bytestream2_get_be16u(&gb); |
242 | 114k | dst16[5] = bytestream2_get_be16u(&gb); |
243 | 114k | dst16[3] = bytestream2_get_be16u(&gb); |
244 | 114k | dst16[4] = bytestream2_get_be16u(&gb); |
245 | 114k | dst16[6] = bytestream2_get_be16u(&gb); |
246 | 114k | dst16 += 7; |
247 | 114k | bytestream2_skip(&gb, 2); |
248 | 114k | } while (--samples); |
249 | 498 | } else { |
250 | 44.8k | do { |
251 | 44.8k | dst32[0] = bytestream2_get_be24u(&gb) << 8; |
252 | 44.8k | dst32[1] = bytestream2_get_be24u(&gb) << 8; |
253 | 44.8k | dst32[2] = bytestream2_get_be24u(&gb) << 8; |
254 | 44.8k | dst32[5] = bytestream2_get_be24u(&gb) << 8; |
255 | 44.8k | dst32[3] = bytestream2_get_be24u(&gb) << 8; |
256 | 44.8k | dst32[4] = bytestream2_get_be24u(&gb) << 8; |
257 | 44.8k | dst32[6] = bytestream2_get_be24u(&gb) << 8; |
258 | 44.8k | dst32 += 7; |
259 | 44.8k | bytestream2_skip(&gb, 3); |
260 | 44.8k | } while (--samples); |
261 | 398 | } |
262 | 896 | break; |
263 | | /* remapping: L, R, C, LSide, LBack, RBack, RSide, LF */ |
264 | 516 | case AV_CH_LAYOUT_7POINT1: |
265 | 516 | if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) { |
266 | 1.93k | do { |
267 | 1.93k | dst16[0] = bytestream2_get_be16u(&gb); |
268 | 1.93k | dst16[1] = bytestream2_get_be16u(&gb); |
269 | 1.93k | dst16[2] = bytestream2_get_be16u(&gb); |
270 | 1.93k | dst16[6] = bytestream2_get_be16u(&gb); |
271 | 1.93k | dst16[4] = bytestream2_get_be16u(&gb); |
272 | 1.93k | dst16[5] = bytestream2_get_be16u(&gb); |
273 | 1.93k | dst16[7] = bytestream2_get_be16u(&gb); |
274 | 1.93k | dst16[3] = bytestream2_get_be16u(&gb); |
275 | 1.93k | dst16 += 8; |
276 | 1.93k | } while (--samples); |
277 | 281 | } else { |
278 | 499 | do { |
279 | 499 | dst32[0] = bytestream2_get_be24u(&gb) << 8; |
280 | 499 | dst32[1] = bytestream2_get_be24u(&gb) << 8; |
281 | 499 | dst32[2] = bytestream2_get_be24u(&gb) << 8; |
282 | 499 | dst32[6] = bytestream2_get_be24u(&gb) << 8; |
283 | 499 | dst32[4] = bytestream2_get_be24u(&gb) << 8; |
284 | 499 | dst32[5] = bytestream2_get_be24u(&gb) << 8; |
285 | 499 | dst32[7] = bytestream2_get_be24u(&gb) << 8; |
286 | 499 | dst32[3] = bytestream2_get_be24u(&gb) << 8; |
287 | 499 | dst32 += 8; |
288 | 499 | } while (--samples); |
289 | 235 | } |
290 | 516 | break; |
291 | 207k | } |
292 | 207k | } |
293 | | |
294 | 207k | *got_frame_ptr = 1; |
295 | | |
296 | 207k | retval = bytestream2_tell(&gb); |
297 | 207k | if (avctx->debug & FF_DEBUG_BITSTREAM) |
298 | 0 | ff_dlog(avctx, "pcm_bluray_decode_frame: decoded %d -> %d bytes\n", |
299 | 207k | retval, buf_size); |
300 | 207k | return retval + 4; |
301 | 207k | } |
302 | | |
303 | | const FFCodec ff_pcm_bluray_decoder = { |
304 | | .p.name = "pcm_bluray", |
305 | | CODEC_LONG_NAME("PCM signed 16|20|24-bit big-endian for Blu-ray media"), |
306 | | .p.type = AVMEDIA_TYPE_AUDIO, |
307 | | .p.id = AV_CODEC_ID_PCM_BLURAY, |
308 | | FF_CODEC_DECODE_CB(pcm_bluray_decode_frame), |
309 | | .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_CHANNEL_CONF, |
310 | | }; |