Coverage Report

Created: 2026-09-14 08:00

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
232k
{
57
232k
    static const uint8_t bits_per_samples[4] = { 0, 16, 20, 24 };
58
232k
    static const AVChannelLayout channel_layouts[16] = {
59
232k
        { 0 },                     AV_CHANNEL_LAYOUT_MONO,     { 0 },
60
232k
        AV_CHANNEL_LAYOUT_STEREO,  AV_CHANNEL_LAYOUT_SURROUND, AV_CHANNEL_LAYOUT_2_1,
61
232k
        AV_CHANNEL_LAYOUT_4POINT0, AV_CHANNEL_LAYOUT_2_2,      AV_CHANNEL_LAYOUT_5POINT0,
62
232k
        AV_CHANNEL_LAYOUT_5POINT1, AV_CHANNEL_LAYOUT_7POINT0,  AV_CHANNEL_LAYOUT_7POINT1,
63
232k
        { 0 }, { 0 }, { 0 }, { 0 },
64
232k
    };
65
232k
    uint8_t channel_layout = header[2] >> 4;
66
67
232k
    if (avctx->debug & FF_DEBUG_PICT_INFO)
68
0
        ff_dlog(avctx, "pcm_bluray_parse_header: header = %02x%02x%02x%02x\n",
69
232k
                header[0], header[1], header[2], header[3]);
70
71
    /* get the sample depth and derive the sample format from it */
72
232k
    avctx->bits_per_coded_sample = bits_per_samples[header[3] >> 6];
73
232k
    if (!(avctx->bits_per_coded_sample == 16 || avctx->bits_per_coded_sample == 24)) {
74
2.14k
        av_log(avctx, AV_LOG_ERROR, "unsupported sample depth (%d)\n", avctx->bits_per_coded_sample);
75
2.14k
        return AVERROR_INVALIDDATA;
76
2.14k
    }
77
230k
    avctx->sample_fmt = avctx->bits_per_coded_sample == 16 ? AV_SAMPLE_FMT_S16
78
230k
                                                           : AV_SAMPLE_FMT_S32;
79
230k
    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
230k
    switch (header[2] & 0x0f) {
84
216k
    case 1:
85
216k
        avctx->sample_rate = 48000;
86
216k
        break;
87
5.77k
    case 4:
88
5.77k
        avctx->sample_rate = 96000;
89
5.77k
        break;
90
5.85k
    case 5:
91
5.85k
        avctx->sample_rate = 192000;
92
5.85k
        break;
93
2.33k
    default:
94
2.33k
        avctx->sample_rate = 0;
95
2.33k
        av_log(avctx, AV_LOG_ERROR, "reserved sample rate (%d)\n",
96
2.33k
               header[2] & 0x0f);
97
2.33k
        return AVERROR_INVALIDDATA;
98
230k
    }
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
227k
    av_channel_layout_uninit(&avctx->ch_layout);
107
227k
    avctx->ch_layout = channel_layouts[channel_layout];
108
227k
    if (!avctx->ch_layout.nb_channels) {
109
272
        av_log(avctx, AV_LOG_ERROR, "reserved channel configuration (%d)\n",
110
272
               channel_layout);
111
272
        return AVERROR_INVALIDDATA;
112
272
    }
113
114
227k
    avctx->bit_rate = FFALIGN(avctx->ch_layout.nb_channels, 2) * avctx->sample_rate *
115
227k
                      avctx->bits_per_coded_sample;
116
117
227k
    if (avctx->debug & FF_DEBUG_PICT_INFO)
118
0
        ff_dlog(avctx,
119
227k
                "pcm_bluray_parse_header: %d channels, %d bits per sample, %d Hz, %"PRId64" bit/s\n",
120
227k
                avctx->ch_layout.nb_channels, avctx->bits_per_coded_sample,
121
227k
                avctx->sample_rate, avctx->bit_rate);
122
227k
    return 0;
123
227k
}
124
125
static int pcm_bluray_decode_frame(AVCodecContext *avctx, AVFrame *frame,
126
                                   int *got_frame_ptr, AVPacket *avpkt)
127
370k
{
128
370k
    const uint8_t *src = avpkt->data;
129
370k
    int buf_size = avpkt->size;
130
370k
    GetByteContext gb;
131
370k
    int num_source_channels, channel, retval;
132
370k
    int sample_size, samples;
133
370k
    int16_t *dst16;
134
370k
    int32_t *dst32;
135
136
370k
    if (buf_size < 4) {
137
138k
        av_log(avctx, AV_LOG_ERROR, "PCM packet too small\n");
138
138k
        return AVERROR_INVALIDDATA;
139
138k
    }
140
141
232k
    if ((retval = pcm_bluray_parse_header(avctx, src)))
142
4.74k
        return retval;
143
227k
    src += 4;
144
227k
    buf_size -= 4;
145
146
227k
    bytestream2_init(&gb, src, buf_size);
147
148
    /* There's always an even number of channels in the source */
149
227k
    num_source_channels = FFALIGN(avctx->ch_layout.nb_channels, 2);
150
227k
    sample_size = (num_source_channels *
151
227k
                   (avctx->sample_fmt == AV_SAMPLE_FMT_S16 ? 16 : 24)) >> 3;
152
227k
    samples = buf_size / sample_size;
153
154
    /* get output buffer */
155
227k
    frame->nb_samples = samples;
156
227k
    if ((retval = ff_get_buffer(avctx, frame, 0)) < 0)
157
8.91k
        return retval;
158
218k
    dst16 = (int16_t *)frame->data[0];
159
218k
    dst32 = (int32_t *)frame->data[0];
160
161
218k
    if (samples) {
162
218k
        switch (avctx->ch_layout.u.mask) {
163
            /* cases with same number of source and coded channels */
164
2.26k
        case AV_CH_LAYOUT_STEREO:
165
2.78k
        case AV_CH_LAYOUT_4POINT0:
166
3.16k
        case AV_CH_LAYOUT_2_2:
167
3.16k
            samples *= num_source_channels;
168
3.16k
            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
1.72M
                do {
173
1.72M
                    *dst16++ = bytestream2_get_be16u(&gb);
174
1.72M
                } while (--samples);
175
2.81k
#endif
176
2.81k
            } else {
177
4.68k
                do {
178
4.68k
                    *dst32++ = bytestream2_get_be24u(&gb) << 8;
179
4.68k
                } while (--samples);
180
349
            }
181
3.16k
            break;
182
        /* cases where number of source channels = coded channels + 1 */
183
211k
        case AV_CH_LAYOUT_MONO:
184
211k
        case AV_CH_LAYOUT_SURROUND:
185
213k
        case AV_CH_LAYOUT_2_1:
186
213k
        case AV_CH_LAYOUT_5POINT0:
187
213k
            if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
188
3.49M
                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.49M
                    channel = avctx->ch_layout.nb_channels;
195
3.68M
                    do {
196
3.68M
                        *dst16++ = bytestream2_get_be16u(&gb);
197
3.68M
                    } while (--channel);
198
3.49M
#endif
199
3.49M
                    bytestream2_skip(&gb, 2);
200
3.49M
                } while (--samples);
201
210k
            } else {
202
476k
                do {
203
476k
                    channel = avctx->ch_layout.nb_channels;
204
620k
                    do {
205
620k
                        *dst32++ = bytestream2_get_be24u(&gb) << 8;
206
620k
                    } while (--channel);
207
476k
                    bytestream2_skip(&gb, 3);
208
476k
                } while (--samples);
209
3.15k
            }
210
213k
            break;
211
            /* remapping: L, R, C, LBack, RBack, LF */
212
511
        case AV_CH_LAYOUT_5POINT1:
213
511
            if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
214
565
                do {
215
565
                    dst16[0] = bytestream2_get_be16u(&gb);
216
565
                    dst16[1] = bytestream2_get_be16u(&gb);
217
565
                    dst16[2] = bytestream2_get_be16u(&gb);
218
565
                    dst16[4] = bytestream2_get_be16u(&gb);
219
565
                    dst16[5] = bytestream2_get_be16u(&gb);
220
565
                    dst16[3] = bytestream2_get_be16u(&gb);
221
565
                    dst16 += 6;
222
565
                } while (--samples);
223
269
            } else {
224
19.2k
                do {
225
19.2k
                    dst32[0] = bytestream2_get_be24u(&gb) << 8;
226
19.2k
                    dst32[1] = bytestream2_get_be24u(&gb) << 8;
227
19.2k
                    dst32[2] = bytestream2_get_be24u(&gb) << 8;
228
19.2k
                    dst32[4] = bytestream2_get_be24u(&gb) << 8;
229
19.2k
                    dst32[5] = bytestream2_get_be24u(&gb) << 8;
230
19.2k
                    dst32[3] = bytestream2_get_be24u(&gb) << 8;
231
19.2k
                    dst32 += 6;
232
19.2k
                } while (--samples);
233
242
            }
234
511
            break;
235
            /* remapping: L, R, C, LSide, LBack, RBack, RSide, <unused> */
236
957
        case AV_CH_LAYOUT_7POINT0:
237
957
            if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
238
91.4k
                do {
239
91.4k
                    dst16[0] = bytestream2_get_be16u(&gb);
240
91.4k
                    dst16[1] = bytestream2_get_be16u(&gb);
241
91.4k
                    dst16[2] = bytestream2_get_be16u(&gb);
242
91.4k
                    dst16[5] = bytestream2_get_be16u(&gb);
243
91.4k
                    dst16[3] = bytestream2_get_be16u(&gb);
244
91.4k
                    dst16[4] = bytestream2_get_be16u(&gb);
245
91.4k
                    dst16[6] = bytestream2_get_be16u(&gb);
246
91.4k
                    dst16 += 7;
247
91.4k
                    bytestream2_skip(&gb, 2);
248
91.4k
                } while (--samples);
249
551
            } else {
250
28.7k
                do {
251
28.7k
                    dst32[0] = bytestream2_get_be24u(&gb) << 8;
252
28.7k
                    dst32[1] = bytestream2_get_be24u(&gb) << 8;
253
28.7k
                    dst32[2] = bytestream2_get_be24u(&gb) << 8;
254
28.7k
                    dst32[5] = bytestream2_get_be24u(&gb) << 8;
255
28.7k
                    dst32[3] = bytestream2_get_be24u(&gb) << 8;
256
28.7k
                    dst32[4] = bytestream2_get_be24u(&gb) << 8;
257
28.7k
                    dst32[6] = bytestream2_get_be24u(&gb) << 8;
258
28.7k
                    dst32 += 7;
259
28.7k
                    bytestream2_skip(&gb, 3);
260
28.7k
                } while (--samples);
261
551
            }
262
957
            break;
263
            /* remapping: L, R, C, LSide, LBack, RBack, RSide, LF */
264
488
        case AV_CH_LAYOUT_7POINT1:
265
488
            if (AV_SAMPLE_FMT_S16 == avctx->sample_fmt) {
266
1.39k
                do {
267
1.39k
                    dst16[0] = bytestream2_get_be16u(&gb);
268
1.39k
                    dst16[1] = bytestream2_get_be16u(&gb);
269
1.39k
                    dst16[2] = bytestream2_get_be16u(&gb);
270
1.39k
                    dst16[6] = bytestream2_get_be16u(&gb);
271
1.39k
                    dst16[4] = bytestream2_get_be16u(&gb);
272
1.39k
                    dst16[5] = bytestream2_get_be16u(&gb);
273
1.39k
                    dst16[7] = bytestream2_get_be16u(&gb);
274
1.39k
                    dst16[3] = bytestream2_get_be16u(&gb);
275
1.39k
                    dst16 += 8;
276
1.39k
                } while (--samples);
277
250
            } else {
278
572
                do {
279
572
                    dst32[0] = bytestream2_get_be24u(&gb) << 8;
280
572
                    dst32[1] = bytestream2_get_be24u(&gb) << 8;
281
572
                    dst32[2] = bytestream2_get_be24u(&gb) << 8;
282
572
                    dst32[6] = bytestream2_get_be24u(&gb) << 8;
283
572
                    dst32[4] = bytestream2_get_be24u(&gb) << 8;
284
572
                    dst32[5] = bytestream2_get_be24u(&gb) << 8;
285
572
                    dst32[7] = bytestream2_get_be24u(&gb) << 8;
286
572
                    dst32[3] = bytestream2_get_be24u(&gb) << 8;
287
572
                    dst32 += 8;
288
572
                } while (--samples);
289
238
            }
290
488
            break;
291
218k
        }
292
218k
    }
293
294
218k
    *got_frame_ptr = 1;
295
296
218k
    retval = bytestream2_tell(&gb);
297
218k
    if (avctx->debug & FF_DEBUG_BITSTREAM)
298
0
        ff_dlog(avctx, "pcm_bluray_decode_frame: decoded %d -> %d bytes\n",
299
218k
                retval, buf_size);
300
218k
    return retval + 4;
301
218k
}
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
};