Coverage Report

Created: 2026-02-14 06:59

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/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
};