FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavformat/iamfdec.c
Date: 2026-09-20 02:22:12
Exec Total Coverage
Lines: 84 92 91.3%
Functions: 5 5 100.0%
Branches: 57 70 81.4%
Line Branch Exec Source
1 /*
2 * Immersive Audio Model and Formats demuxer
3 * Copyright (c) 2023 James Almer <jamrial@gmail.com>
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/intreadwrite.h"
24 #include "avformat.h"
25 #include "demux.h"
26 #include "iamf.h"
27 #include "iamf_reader.h"
28 #include "iamf_parse.h"
29 #include "internal.h"
30
31 //return < 0 if we need more data
32 7502 static int get_score(const uint8_t *buf, int buf_size, enum IAMF_OBU_Type type, int *seq)
33 {
34
2/2
✓ Branch 0 taken 146 times.
✓ Branch 1 taken 7356 times.
 7502 if (type == IAMF_OBU_IA_SEQUENCE_HEADER) {
35
3/4
✓ Branch 0 taken 146 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 124 times.
✓ Branch 3 taken 22 times.
 146 if (buf_size < 4 || AV_RB32(buf) != MKBETAG('i','a','m','f'))
36 124 return 0;
37 22 *seq = 1;
38 22 return -1;
39 }
40
2/2
✓ Branch 0 taken 5489 times.
✓ Branch 1 taken 1867 times.
 7356 if (type >= IAMF_OBU_IA_CODEC_CONFIG && type <= IAMF_OBU_IA_TEMPORAL_DELIMITER)
41
2/2
✓ Branch 0 taken 75 times.
✓ Branch 1 taken 5414 times.
 5489 return *seq ? -1 : 0;
42
3/4
✓ Branch 0 taken 1867 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1838 times.
✓ Branch 3 taken 29 times.
 1867 if (type >= IAMF_OBU_IA_AUDIO_FRAME && type <= IAMF_OBU_IA_AUDIO_FRAME_ID17)
43
2/2
✓ Branch 0 taken 22 times.
✓ Branch 1 taken 1816 times.
 1838 return *seq ? AVPROBE_SCORE_EXTENSION + 1 : 0;
44 29 return 0;
45 }
46
47 8062 static int iamf_probe(const AVProbeData *p)
48 {
49 unsigned obu_size;
50 enum IAMF_OBU_Type type;
51 8062 int seq = 0, cnt = 0, start_pos;
52 int ret;
53
54 97 while (1) {
55 8159 int size = ff_iamf_parse_obu_header(p->buf + cnt, p->buf_size - cnt,
56 &obu_size, &start_pos, &type,
57 NULL, NULL);
58
2/2
✓ Branch 0 taken 657 times.
✓ Branch 1 taken 7502 times.
 8159 if (size < 0)
59 657 return 0;
60
61 7502 ret = get_score(p->buf + cnt + start_pos,
62 7502 p->buf_size - cnt - start_pos,
63 type, &seq);
64
2/2
✓ Branch 0 taken 7405 times.
✓ Branch 1 taken 97 times.
 7502 if (ret >= 0)
65 7405 return ret;
66
67 97 cnt += FFMIN(size, p->buf_size - cnt);
68 }
69 return 0;
70 }
71
72 29 static int iamf_read_header(AVFormatContext *s)
73 {
74 29 IAMFDemuxContext *const c = s->priv_data;
75 29 IAMFContext *const iamf = &c->iamf;
76 int ret;
77
78 29 ret = ff_iamfdec_read_descriptors(iamf, s->pb, INT_MAX, s);
79
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 29 times.
 29 if (ret < 0)
80 ✗ return ret;
81
82
2/2
✓ Branch 0 taken 31 times.
✓ Branch 1 taken 29 times.
 60 for (int i = 0; i < iamf->nb_audio_elements; i++) {
83 31 IAMFAudioElement *audio_element = iamf->audio_elements[i];
84 31 AVStreamGroup *stg = avformat_stream_group_create(s, AV_STREAM_GROUP_PARAMS_IAMF_AUDIO_ELEMENT, NULL);
85
86
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
 31 if (!stg)
87 ✗ return AVERROR(ENOMEM);
88
89 31 av_iamf_audio_element_free(&stg->params.iamf_audio_element);
90 31 stg->id = audio_element->audio_element_id;
91 /* Transfer ownership */
92 31 stg->params.iamf_audio_element = audio_element->element;
93 31 audio_element->element = NULL;
94
95
2/2
✓ Branch 0 taken 186 times.
✓ Branch 1 taken 31 times.
 217 for (int j = 0; j < audio_element->nb_substreams; j++) {
96 186 IAMFSubStream *substream = &audio_element->substreams[j];
97 186 AVStream *st = avformat_new_stream(s, NULL);
98
99
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 186 times.
 186 if (!st)
100 ✗ return AVERROR(ENOMEM);
101
102 186 ret = avformat_stream_group_add_stream(stg, st);
103
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 186 times.
 186 if (ret < 0)
104 ✗ return ret;
105
106 186 ret = avcodec_parameters_copy(st->codecpar, substream->codecpar);
107
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 186 times.
 186 if (ret < 0)
108 ✗ return ret;
109
110
6/6
✓ Branch 0 taken 184 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 29 times.
✓ Branch 3 taken 155 times.
✓ Branch 4 taken 13 times.
✓ Branch 5 taken 16 times.
 186 if (!i && !j && audio_element->layers[0].substream_count == 1)
111 13 st->disposition |= AV_DISPOSITION_DEFAULT;
112
4/4
✓ Branch 0 taken 136 times.
✓ Branch 1 taken 37 times.
✓ Branch 2 taken 134 times.
✓ Branch 3 taken 2 times.
 173 else if (audio_element->nb_layers > 1 || audio_element->layers[0].substream_count > 1)
113 171 st->disposition |= AV_DISPOSITION_DEPENDENT;
114 186 st->id = substream->audio_substream_id;
115 186 avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate);
116 }
117 }
118
119
2/2
✓ Branch 0 taken 29 times.
✓ Branch 1 taken 29 times.
 58 for (int i = 0; i < iamf->nb_mix_presentations; i++) {
120 29 IAMFMixPresentation *mix_presentation = iamf->mix_presentations[i];
121 29 AVStreamGroup *stg = avformat_stream_group_create(s, AV_STREAM_GROUP_PARAMS_IAMF_MIX_PRESENTATION, NULL);
122 29 const AVIAMFMixPresentation *mix = mix_presentation->cmix;
123
124
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 29 times.
 29 if (!stg)
125 ✗ return AVERROR(ENOMEM);
126
127 29 av_iamf_mix_presentation_free(&stg->params.iamf_mix_presentation);
128 29 stg->id = mix_presentation->mix_presentation_id;
129 /* Transfer ownership */
130 29 stg->params.iamf_mix_presentation = mix_presentation->mix;
131 29 mix_presentation->mix = NULL;
132
133
2/2
✓ Branch 0 taken 31 times.
✓ Branch 1 taken 29 times.
 60 for (int j = 0; j < mix->nb_submixes; j++) {
134 31 const AVIAMFSubmix *sub_mix = mix->submixes[j];
135
136
2/2
✓ Branch 0 taken 33 times.
✓ Branch 1 taken 31 times.
 64 for (int k = 0; k < sub_mix->nb_elements; k++) {
137 33 const AVIAMFSubmixElement *submix_element = sub_mix->elements[k];
138 33 AVStreamGroup *audio_element = NULL;
139
140
1/2
✓ Branch 0 taken 35 times.
✗ Branch 1 not taken.
 35 for (int l = 0; l < s->nb_stream_groups; l++)
141
1/2
✓ Branch 0 taken 35 times.
✗ Branch 1 not taken.
 35 if (s->stream_groups[l]->type == AV_STREAM_GROUP_PARAMS_IAMF_AUDIO_ELEMENT &&
142
2/2
✓ Branch 0 taken 33 times.
✓ Branch 1 taken 2 times.
 35 s->stream_groups[l]->id == submix_element->audio_element_id) {
143 33 audio_element = s->stream_groups[l];
144 33 break;
145 }
146
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 33 times.
 33 av_assert0(audio_element);
147
148
2/2
✓ Branch 0 taken 188 times.
✓ Branch 1 taken 33 times.
 221 for (int l = 0; l < audio_element->nb_streams; l++) {
149 188 ret = avformat_stream_group_add_stream(stg, audio_element->streams[l]);
150
3/4
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 186 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
 188 if (ret < 0 && ret != AVERROR(EEXIST))
151 ✗ return ret;
152 }
153 }
154 }
155 }
156
157
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 29 times.
 29 if (!s->nb_streams)
158 ✗ return AVERROR_INVALIDDATA;
159
160 29 return 0;
161 }
162
163 5353 static int iamf_read_packet(AVFormatContext *s, AVPacket *pkt)
164 {
165 5353 IAMFDemuxContext *const c = s->priv_data;
166 int ret;
167
168 5353 ret = ff_iamf_read_packet(s, c, s->pb, INT_MAX, 0, pkt);
169
2/2
✓ Branch 0 taken 10 times.
✓ Branch 1 taken 5343 times.
 5353 if (ret < 0)
170 10 return ret;
171
172 5343 return 0;
173 }
174
175 29 static int iamf_read_close(AVFormatContext *s)
176 {
177 29 IAMFDemuxContext *const c = s->priv_data;
178
179 29 ff_iamf_read_deinit(c);
180
181 29 return 0;
182 }
183
184 const FFInputFormat ff_iamf_demuxer = {
185 .p.name = "iamf",
186 .p.long_name = NULL_IF_CONFIG_SMALL("Raw Immersive Audio Model and Formats"),
187 .p.extensions = "iamf",
188 .p.flags = AVFMT_GENERIC_INDEX | AVFMT_NO_BYTE_SEEK | AVFMT_NOTIMESTAMPS | AVFMT_SHOW_IDS,
189 .priv_data_size = sizeof(IAMFDemuxContext),
190 .flags_internal = FF_INFMT_FLAG_INIT_CLEANUP,
191 .read_probe = iamf_probe,
192 .read_header = iamf_read_header,
193 .read_packet = iamf_read_packet,
194 .read_close = iamf_read_close,
195 };
196