FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/pcm-dvdenc.c
Date: 2025-06-01 09:29:47
Exec Total Coverage
Lines: 86 93 92.5%
Functions: 2 2 100.0%
Branches: 26 32 81.2%

Line Branch Exec Source
1 /*
2 * LPCM codecs for PCM formats found in Video DVD streams
3 * Copyright (c) 2018 Paul B Mahol
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 #include "libavutil/avassert.h"
23 #include "libavutil/channel_layout.h"
24 #include "avcodec.h"
25 #include "bytestream.h"
26 #include "codec_internal.h"
27 #include "encode.h"
28
29 typedef struct PCMDVDContext {
30 uint8_t header[3]; // Header added to every frame
31 int block_size; // Size of a block of samples in bytes
32 int samples_per_block; // Number of samples per channel per block
33 int groups_per_block; // Number of 20/24-bit sample groups per block
34 } PCMDVDContext;
35
36 10 static av_cold int pcm_dvd_encode_init(AVCodecContext *avctx)
37 {
38 10 PCMDVDContext *s = avctx->priv_data;
39 int quant, freq, frame_size;
40
41
2/3
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
10 switch (avctx->sample_rate) {
42 8 case 48000:
43 8 freq = 0;
44 8 break;
45 2 case 96000:
46 2 freq = 1;
47 2 break;
48 default:
49 av_unreachable("Already checked via CODEC_SAMPLERATES");
50 }
51
52
2/3
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
10 switch (avctx->sample_fmt) {
53 6 case AV_SAMPLE_FMT_S16:
54 6 avctx->bits_per_coded_sample = 16;
55 6 quant = 0;
56 6 break;
57 4 case AV_SAMPLE_FMT_S32:
58 4 avctx->bits_per_coded_sample = 24;
59 4 quant = 2;
60 4 break;
61 default:
62 av_unreachable("Already checked via CODEC_SAMPLEFMTS");
63 }
64
65 10 avctx->bits_per_coded_sample = 16 + quant * 4;
66 10 avctx->block_align = avctx->ch_layout.nb_channels * avctx->bits_per_coded_sample / 8;
67 10 avctx->bit_rate = avctx->block_align * 8LL * avctx->sample_rate;
68
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
10 if (avctx->bit_rate > 9800000) {
69 av_log(avctx, AV_LOG_ERROR, "Too big bitrate: reduce sample rate, bitdepth or channels.\n");
70 return AVERROR(EINVAL);
71 }
72
73
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 4 times.
10 if (avctx->sample_fmt == AV_SAMPLE_FMT_S16) {
74 6 s->samples_per_block = 1;
75 6 s->block_size = avctx->ch_layout.nb_channels * 2;
76 6 frame_size = 2008 / s->block_size;
77 } else {
78
3/3
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 1 times.
4 switch (avctx->ch_layout.nb_channels) {
79 2 case 1:
80 case 2:
81 case 4:
82 /* one group has all the samples needed */
83 2 s->block_size = 4 * avctx->bits_per_coded_sample / 8;
84 2 s->samples_per_block = 4 / avctx->ch_layout.nb_channels;
85 2 s->groups_per_block = 1;
86 2 break;
87 1 case 8:
88 /* two groups have all the samples needed */
89 1 s->block_size = 8 * avctx->bits_per_coded_sample / 8;
90 1 s->samples_per_block = 1;
91 1 s->groups_per_block = 2;
92 1 break;
93 1 default:
94 /* need avctx->ch_layout.nb_channels groups */
95 1 s->block_size = 4 * avctx->ch_layout.nb_channels *
96 1 avctx->bits_per_coded_sample / 8;
97 1 s->samples_per_block = 4;
98 1 s->groups_per_block = avctx->ch_layout.nb_channels;
99 1 break;
100 }
101
102 4 frame_size = FFALIGN(2008 / s->block_size, s->samples_per_block);
103 }
104
105 10 s->header[0] = 0x0c;
106 10 s->header[1] = (quant << 6) | (freq << 4) | (avctx->ch_layout.nb_channels - 1);
107 10 s->header[2] = 0x80;
108
109
1/2
✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
10 if (!avctx->frame_size)
110 10 avctx->frame_size = frame_size;
111
112 10 return 0;
113 }
114
115 15338 static int pcm_dvd_encode_frame(AVCodecContext *avctx, AVPacket *avpkt,
116 const AVFrame *frame, int *got_packet_ptr)
117 {
118 15338 PCMDVDContext *s = avctx->priv_data;
119 15338 int samples = frame->nb_samples * avctx->ch_layout.nb_channels;
120 15338 int64_t pkt_size = (int64_t)(frame->nb_samples / s->samples_per_block) * s->block_size + 3;
121 15338 int blocks = (pkt_size - 3) / s->block_size;
122 const int16_t *src16;
123 const int32_t *src32;
124 PutByteContext pb;
125 int ret;
126
127
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 15338 times.
15338 if ((ret = ff_get_encode_buffer(avctx, avpkt, pkt_size, 0)) < 0)
128 return ret;
129
130 15338 memcpy(avpkt->data, s->header, 3);
131
132 15338 src16 = (const int16_t *)frame->data[0];
133 15338 src32 = (const int32_t *)frame->data[0];
134
135 15338 bytestream2_init_writer(&pb, avpkt->data + 3, avpkt->size - 3);
136
137
2/3
✓ Branch 0 taken 2198 times.
✓ Branch 1 taken 13140 times.
✗ Branch 2 not taken.
15338 switch (avctx->sample_fmt) {
138 2202820 case AV_SAMPLE_FMT_S16:
139 do {
140 2202820 bytestream2_put_be16(&pb, *src16++);
141
2/2
✓ Branch 0 taken 2200622 times.
✓ Branch 1 taken 2198 times.
2202820 } while (--samples);
142 2198 break;
143 13140 case AV_SAMPLE_FMT_S32:
144
2/2
✓ Branch 0 taken 206 times.
✓ Branch 1 taken 12934 times.
13140 if (avctx->ch_layout.nb_channels == 1) {
145 do {
146
2/2
✓ Branch 0 taken 17280 times.
✓ Branch 1 taken 8640 times.
25920 for (int i = 2; i; i--) {
147 17280 bytestream2_put_be16(&pb, src32[0] >> 16);
148 17280 bytestream2_put_be16(&pb, src32[1] >> 16);
149 17280 bytestream2_put_byte(&pb, (uint8_t)((*src32++) >> 8));
150 17280 bytestream2_put_byte(&pb, (uint8_t)((*src32++) >> 8));
151 }
152
2/2
✓ Branch 0 taken 8434 times.
✓ Branch 1 taken 206 times.
8640 } while (--blocks);
153 } else {
154 do {
155
2/2
✓ Branch 0 taken 886839 times.
✓ Branch 1 taken 711219 times.
1598058 for (int i = s->groups_per_block; i; i--) {
156 886839 bytestream2_put_be16(&pb, src32[0] >> 16);
157 886839 bytestream2_put_be16(&pb, src32[1] >> 16);
158 886839 bytestream2_put_be16(&pb, src32[2] >> 16);
159 886839 bytestream2_put_be16(&pb, src32[3] >> 16);
160 886839 bytestream2_put_byte(&pb, (uint8_t)((*src32++) >> 8));
161 886839 bytestream2_put_byte(&pb, (uint8_t)((*src32++) >> 8));
162 886839 bytestream2_put_byte(&pb, (uint8_t)((*src32++) >> 8));
163 886839 bytestream2_put_byte(&pb, (uint8_t)((*src32++) >> 8));
164 }
165
2/2
✓ Branch 0 taken 698285 times.
✓ Branch 1 taken 12934 times.
711219 } while (--blocks);
166 }
167 13140 break;
168 }
169
170 15338 *got_packet_ptr = 1;
171
172 15338 return 0;
173 }
174
175 const FFCodec ff_pcm_dvd_encoder = {
176 .p.name = "pcm_dvd",
177 CODEC_LONG_NAME("PCM signed 16|20|24-bit big-endian for DVD media"),
178 .p.type = AVMEDIA_TYPE_AUDIO,
179 .p.id = AV_CODEC_ID_PCM_DVD,
180 .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_SMALL_LAST_FRAME |
181 AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE,
182 .priv_data_size = sizeof(PCMDVDContext),
183 .init = pcm_dvd_encode_init,
184 FF_CODEC_ENCODE_CB(pcm_dvd_encode_frame),
185 CODEC_SAMPLERATES(48000, 96000),
186 CODEC_CH_LAYOUTS(AV_CHANNEL_LAYOUT_MONO, AV_CHANNEL_LAYOUT_STEREO,
187 AV_CHANNEL_LAYOUT_5POINT1, AV_CHANNEL_LAYOUT_7POINT1),
188 CODEC_SAMPLEFMTS(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_S32),
189 };
190