FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/aacenc.c
Date: 2023-03-22 23:59:29
Exec Total Coverage
Lines: 554 654 84.7%
Functions: 24 26 92.3%
Branches: 354 484 73.1%
Line Branch Exec Source
1 /*
2 * AAC encoder
3 * Copyright (C) 2008 Konstantin Shishkov
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 * AAC encoder
25 */
26
27 /***********************************
28 * TODOs:
29 * add sane pulse detection
30 ***********************************/
31 #include <float.h>
32
33 #include "libavutil/channel_layout.h"
34 #include "libavutil/libm.h"
35 #include "libavutil/float_dsp.h"
36 #include "libavutil/opt.h"
37 #include "avcodec.h"
38 #include "codec_internal.h"
39 #include "encode.h"
40 #include "put_bits.h"
41 #include "mpeg4audio.h"
42 #include "sinewin.h"
43 #include "profiles.h"
44 #include "version.h"
45
46 #include "aac.h"
47 #include "aactab.h"
48 #include "aacenc.h"
49 #include "aacenctab.h"
50 #include "aacenc_utils.h"
51
52 #include "psymodel.h"
53
54 /**
55 * List of PCE (Program Configuration Element) for the channel layouts listed
56 * in channel_layout.h
57 *
58 * For those wishing in the future to add other layouts:
59 *
60 * - num_ele: number of elements in each group of front, side, back, lfe channels
61 * (an element is of type SCE (single channel), CPE (channel pair) for
62 * the first 3 groups; and is LFE for LFE group).
63 *
64 * - pairing: 0 for an SCE element or 1 for a CPE; does not apply to LFE group
65 *
66 * - index: there are three independent indices for SCE, CPE and LFE;
67 * they are incremented irrespective of the group to which the element belongs;
68 * they are not reset when going from one group to another
69 *
70 * Example: for 7.0 channel layout,
71 * .pairing = { { 1, 0 }, { 1 }, { 1 }, }, (3 CPE and 1 SCE in front group)
72 * .index = { { 0, 0 }, { 1 }, { 2 }, },
73 * (index is 0 for the single SCE but goes from 0 to 2 for the CPEs)
74 *
75 * The index order impacts the channel ordering. But is otherwise arbitrary
76 * (the sequence could have been 2, 0, 1 instead of 0, 1, 2).
77 *
78 * Spec allows for discontinuous indices, e.g. if one has a total of two SCE,
79 * SCE.0 SCE.15 is OK per spec; BUT it won't be decoded by our AAC decoder
80 * which at this time requires that indices fully cover some range starting
81 * from 0 (SCE.1 SCE.0 is OK but not SCE.0 SCE.15).
82 *
83 * - config_map: total number of elements and their types. Beware, the way the
84 * types are ordered impacts the final channel ordering.
85 *
86 * - reorder_map: reorders the channels.
87 *
88 */
89 static const AACPCEInfo aac_pce_configs[] = {
90 {
91 .layout = AV_CHANNEL_LAYOUT_MONO,
92 .num_ele = { 1, 0, 0, 0 },
93 .pairing = { { 0 }, },
94 .index = { { 0 }, },
95 .config_map = { 1, TYPE_SCE, },
96 .reorder_map = { 0 },
97 },
98 {
99 .layout = AV_CHANNEL_LAYOUT_STEREO,
100 .num_ele = { 1, 0, 0, 0 },
101 .pairing = { { 1 }, },
102 .index = { { 0 }, },
103 .config_map = { 1, TYPE_CPE, },
104 .reorder_map = { 0, 1 },
105 },
106 {
107 .layout = AV_CHANNEL_LAYOUT_2POINT1,
108 .num_ele = { 1, 0, 0, 1 },
109 .pairing = { { 1 }, },
110 .index = { { 0 },{ 0 },{ 0 },{ 0 } },
111 .config_map = { 2, TYPE_CPE, TYPE_LFE },
112 .reorder_map = { 0, 1, 2 },
113 },
114 {
115 .layout = AV_CHANNEL_LAYOUT_2_1,
116 .num_ele = { 1, 0, 1, 0 },
117 .pairing = { { 1 },{ 0 },{ 0 } },
118 .index = { { 0 },{ 0 },{ 0 }, },
119 .config_map = { 2, TYPE_CPE, TYPE_SCE },
120 .reorder_map = { 0, 1, 2 },
121 },
122 {
123 .layout = AV_CHANNEL_LAYOUT_SURROUND,
124 .num_ele = { 2, 0, 0, 0 },
125 .pairing = { { 1, 0 }, },
126 .index = { { 0, 0 }, },
127 .config_map = { 2, TYPE_CPE, TYPE_SCE, },
128 .reorder_map = { 0, 1, 2 },
129 },
130 {
131 .layout = AV_CHANNEL_LAYOUT_3POINT1,
132 .num_ele = { 2, 0, 0, 1 },
133 .pairing = { { 1, 0 }, },
134 .index = { { 0, 0 }, { 0 }, { 0 }, { 0 }, },
135 .config_map = { 3, TYPE_CPE, TYPE_SCE, TYPE_LFE },
136 .reorder_map = { 0, 1, 2, 3 },
137 },
138 {
139 .layout = AV_CHANNEL_LAYOUT_4POINT0,
140 .num_ele = { 2, 0, 1, 0 },
141 .pairing = { { 1, 0 }, { 0 }, { 0 }, },
142 .index = { { 0, 0 }, { 0 }, { 1 } },
143 .config_map = { 3, TYPE_CPE, TYPE_SCE, TYPE_SCE },
144 .reorder_map = { 0, 1, 2, 3 },
145 },
146 {
147 .layout = AV_CHANNEL_LAYOUT_4POINT1,
148 .num_ele = { 2, 1, 1, 0 },
149 .pairing = { { 1, 0 }, { 0 }, { 0 }, },
150 .index = { { 0, 0 }, { 1 }, { 2 }, { 0 } },
151 .config_map = { 4, TYPE_CPE, TYPE_SCE, TYPE_SCE, TYPE_SCE },
152 .reorder_map = { 0, 1, 2, 3, 4 },
153 },
154 {
155 .layout = AV_CHANNEL_LAYOUT_2_2,
156 .num_ele = { 1, 1, 0, 0 },
157 .pairing = { { 1 }, { 1 }, },
158 .index = { { 0 }, { 1 }, },
159 .config_map = { 2, TYPE_CPE, TYPE_CPE },
160 .reorder_map = { 0, 1, 2, 3 },
161 },
162 {
163 .layout = AV_CHANNEL_LAYOUT_QUAD,
164 .num_ele = { 1, 0, 1, 0 },
165 .pairing = { { 1 }, { 0 }, { 1 }, },
166 .index = { { 0 }, { 0 }, { 1 } },
167 .config_map = { 2, TYPE_CPE, TYPE_CPE },
168 .reorder_map = { 0, 1, 2, 3 },
169 },
170 {
171 .layout = AV_CHANNEL_LAYOUT_5POINT0,
172 .num_ele = { 2, 1, 0, 0 },
173 .pairing = { { 1, 0 }, { 1 }, },
174 .index = { { 0, 0 }, { 1 } },
175 .config_map = { 3, TYPE_CPE, TYPE_SCE, TYPE_CPE },
176 .reorder_map = { 0, 1, 2, 3, 4 },
177 },
178 {
179 .layout = AV_CHANNEL_LAYOUT_5POINT1,
180 .num_ele = { 2, 1, 1, 0 },
181 .pairing = { { 1, 0 }, { 0 }, { 1 }, },
182 .index = { { 0, 0 }, { 1 }, { 1 } },
183 .config_map = { 4, TYPE_CPE, TYPE_SCE, TYPE_SCE, TYPE_CPE },
184 .reorder_map = { 0, 1, 2, 3, 4, 5 },
185 },
186 {
187 .layout = AV_CHANNEL_LAYOUT_5POINT0_BACK,
188 .num_ele = { 2, 0, 1, 0 },
189 .pairing = { { 1, 0 }, { 0 }, { 1 } },
190 .index = { { 0, 0 }, { 0 }, { 1 } },
191 .config_map = { 3, TYPE_CPE, TYPE_SCE, TYPE_CPE },
192 .reorder_map = { 0, 1, 2, 3, 4 },
193 },
194 {
195 .layout = AV_CHANNEL_LAYOUT_5POINT1_BACK,
196 .num_ele = { 2, 1, 1, 0 },
197 .pairing = { { 1, 0 }, { 0 }, { 1 }, },
198 .index = { { 0, 0 }, { 1 }, { 1 } },
199 .config_map = { 4, TYPE_CPE, TYPE_SCE, TYPE_SCE, TYPE_CPE },
200 .reorder_map = { 0, 1, 2, 3, 4, 5 },
201 },
202 {
203 .layout = AV_CHANNEL_LAYOUT_6POINT0,
204 .num_ele = { 2, 1, 1, 0 },
205 .pairing = { { 1, 0 }, { 1 }, { 0 }, },
206 .index = { { 0, 0 }, { 1 }, { 1 } },
207 .config_map = { 4, TYPE_CPE, TYPE_SCE, TYPE_CPE, TYPE_SCE },
208 .reorder_map = { 0, 1, 2, 3, 4, 5 },
209 },
210 {
211 .layout = AV_CHANNEL_LAYOUT_6POINT0_FRONT,
212 .num_ele = { 2, 1, 0, 0 },
213 .pairing = { { 1, 1 }, { 1 } },
214 .index = { { 1, 0 }, { 2 }, },
215 .config_map = { 3, TYPE_CPE, TYPE_CPE, TYPE_CPE, },
216 .reorder_map = { 0, 1, 2, 3, 4, 5 },
217 },
218 {
219 .layout = AV_CHANNEL_LAYOUT_HEXAGONAL,
220 .num_ele = { 2, 0, 2, 0 },
221 .pairing = { { 1, 0 },{ 0 },{ 1, 0 }, },
222 .index = { { 0, 0 },{ 0 },{ 1, 1 } },
223 .config_map = { 4, TYPE_CPE, TYPE_SCE, TYPE_CPE, TYPE_SCE, },
224 .reorder_map = { 0, 1, 2, 3, 4, 5 },
225 },
226 {
227 .layout = AV_CHANNEL_LAYOUT_6POINT1,
228 .num_ele = { 2, 1, 2, 0 },
229 .pairing = { { 1, 0 },{ 0 },{ 1, 0 }, },
230 .index = { { 0, 0 },{ 1 },{ 1, 2 } },
231 .config_map = { 5, TYPE_CPE, TYPE_SCE, TYPE_SCE, TYPE_CPE, TYPE_SCE },
232 .reorder_map = { 0, 1, 2, 3, 4, 5, 6 },
233 },
234 {
235 .layout = AV_CHANNEL_LAYOUT_6POINT1_BACK,
236 .num_ele = { 2, 1, 2, 0 },
237 .pairing = { { 1, 0 }, { 0 }, { 1, 0 }, },
238 .index = { { 0, 0 }, { 1 }, { 1, 2 } },
239 .config_map = { 5, TYPE_CPE, TYPE_SCE, TYPE_SCE, TYPE_CPE, TYPE_SCE },
240 .reorder_map = { 0, 1, 2, 3, 4, 5, 6 },
241 },
242 {
243 .layout = AV_CHANNEL_LAYOUT_6POINT1_FRONT,
244 .num_ele = { 2, 1, 2, 0 },
245 .pairing = { { 1, 0 }, { 0 }, { 1, 0 }, },
246 .index = { { 0, 0 }, { 1 }, { 1, 2 } },
247 .config_map = { 5, TYPE_CPE, TYPE_SCE, TYPE_SCE, TYPE_CPE, TYPE_SCE },
248 .reorder_map = { 0, 1, 2, 3, 4, 5, 6 },
249 },
250 {
251 .layout = AV_CHANNEL_LAYOUT_7POINT0,
252 .num_ele = { 2, 1, 1, 0 },
253 .pairing = { { 1, 0 }, { 1 }, { 1 }, },
254 .index = { { 0, 0 }, { 1 }, { 2 }, },
255 .config_map = { 4, TYPE_CPE, TYPE_SCE, TYPE_CPE, TYPE_CPE },
256 .reorder_map = { 0, 1, 2, 3, 4, 5, 6 },
257 },
258 {
259 .layout = AV_CHANNEL_LAYOUT_7POINT0_FRONT,
260 .num_ele = { 2, 1, 1, 0 },
261 .pairing = { { 1, 0 }, { 1 }, { 1 }, },
262 .index = { { 0, 0 }, { 1 }, { 2 }, },
263 .config_map = { 4, TYPE_CPE, TYPE_SCE, TYPE_CPE, TYPE_CPE },
264 .reorder_map = { 0, 1, 2, 3, 4, 5, 6 },
265 },
266 {
267 .layout = AV_CHANNEL_LAYOUT_7POINT1,
268 .num_ele = { 2, 1, 2, 0 },
269 .pairing = { { 1, 0 }, { 0 }, { 1, 1 }, },
270 .index = { { 0, 0 }, { 1 }, { 1, 2 }, { 0 } },
271 .config_map = { 5, TYPE_CPE, TYPE_SCE, TYPE_SCE, TYPE_CPE, TYPE_CPE },
272 .reorder_map = { 0, 1, 2, 3, 4, 5, 6, 7 },
273 },
274 {
275 .layout = AV_CHANNEL_LAYOUT_7POINT1_WIDE,
276 .num_ele = { 2, 1, 2, 0 },
277 .pairing = { { 1, 0 }, { 0 },{ 1, 1 }, },
278 .index = { { 0, 0 }, { 1 }, { 1, 2 }, { 0 } },
279 .config_map = { 5, TYPE_CPE, TYPE_SCE, TYPE_SCE, TYPE_CPE, TYPE_CPE },
280 .reorder_map = { 0, 1, 2, 3, 4, 5, 6, 7 },
281 },
282 {
283 .layout = AV_CHANNEL_LAYOUT_7POINT1_WIDE_BACK,
284 .num_ele = { 2, 1, 2, 0 },
285 .pairing = { { 1, 0 }, { 0 }, { 1, 1 }, },
286 .index = { { 0, 0 }, { 1 }, { 1, 2 }, { 0 } },
287 .config_map = { 5, TYPE_CPE, TYPE_SCE, TYPE_SCE, TYPE_CPE, TYPE_CPE },
288 .reorder_map = { 0, 1, 2, 3, 4, 5, 6, 7 },
289 },
290 {
291 .layout = AV_CHANNEL_LAYOUT_OCTAGONAL,
292 .num_ele = { 2, 1, 2, 0 },
293 .pairing = { { 1, 0 }, { 1 }, { 1, 0 }, },
294 .index = { { 0, 0 }, { 1 }, { 2, 1 } },
295 .config_map = { 5, TYPE_CPE, TYPE_SCE, TYPE_CPE, TYPE_CPE, TYPE_SCE },
296 .reorder_map = { 0, 1, 2, 3, 4, 5, 6, 7 },
297 },
298 { /* Meant for order 2/mixed ambisonics */
299 .layout = { .order = AV_CHANNEL_ORDER_NATIVE, .nb_channels = 9,
300 .u.mask = AV_CH_LAYOUT_OCTAGONAL | AV_CH_TOP_CENTER },
301 .num_ele = { 2, 2, 2, 0 },
302 .pairing = { { 1, 0 }, { 1, 0 }, { 1, 0 }, },
303 .index = { { 0, 0 }, { 1, 1 }, { 2, 2 } },
304 .config_map = { 6, TYPE_CPE, TYPE_SCE, TYPE_CPE, TYPE_SCE, TYPE_CPE, TYPE_SCE },
305 .reorder_map = { 0, 1, 2, 3, 4, 5, 6, 7, 8 },
306 },
307 { /* Meant for order 2/mixed ambisonics */
308 .layout = { .order = AV_CHANNEL_ORDER_NATIVE, .nb_channels = 10,
309 .u.mask = AV_CH_LAYOUT_6POINT0_FRONT | AV_CH_BACK_CENTER |
310 AV_CH_BACK_LEFT | AV_CH_BACK_RIGHT | AV_CH_TOP_CENTER },
311 .num_ele = { 2, 2, 2, 0 },
312 .pairing = { { 1, 1 }, { 1, 0 }, { 1, 0 }, },
313 .index = { { 0, 1 }, { 2, 0 }, { 3, 1 } },
314 .config_map = { 6, TYPE_CPE, TYPE_CPE, TYPE_CPE, TYPE_SCE, TYPE_CPE, TYPE_SCE },
315 .reorder_map = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 },
316 },
317 {
318 .layout = AV_CHANNEL_LAYOUT_HEXADECAGONAL,
319 .num_ele = { 4, 2, 4, 0 },
320 .pairing = { { 1, 0, 1, 0 }, { 1, 1 }, { 1, 0, 1, 0 }, },
321 .index = { { 0, 0, 1, 1 }, { 2, 3 }, { 4, 2, 5, 3 } },
322 .config_map = { 10, TYPE_CPE, TYPE_SCE, TYPE_CPE, TYPE_SCE, TYPE_CPE, TYPE_CPE, TYPE_CPE, TYPE_SCE, TYPE_CPE, TYPE_SCE },
323 .reorder_map = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 },
324 },
325 };
326
327 ✗ static void put_pce(PutBitContext *pb, AVCodecContext *avctx)
328 {
329 int i, j;
330 ✗ AACEncContext *s = avctx->priv_data;
331 ✗ AACPCEInfo *pce = &s->pce;
332 ✗ const int bitexact = avctx->flags & AV_CODEC_FLAG_BITEXACT;
333 ✗ const char *aux_data = bitexact ? "Lavc" : LIBAVCODEC_IDENT;
334
335 ✗ put_bits(pb, 4, 0);
336
337 ✗ put_bits(pb, 2, avctx->profile);
338 ✗ put_bits(pb, 4, s->samplerate_index);
339
340 ✗ put_bits(pb, 4, pce->num_ele[0]); /* Front */
341 ✗ put_bits(pb, 4, pce->num_ele[1]); /* Side */
342 ✗ put_bits(pb, 4, pce->num_ele[2]); /* Back */
343 ✗ put_bits(pb, 2, pce->num_ele[3]); /* LFE */
344 ✗ put_bits(pb, 3, 0); /* Assoc data */
345 ✗ put_bits(pb, 4, 0); /* CCs */
346
347 ✗ put_bits(pb, 1, 0); /* Stereo mixdown */
348 ✗ put_bits(pb, 1, 0); /* Mono mixdown */
349 ✗ put_bits(pb, 1, 0); /* Something else */
350
351 ✗ for (i = 0; i < 4; i++) {
352 ✗ for (j = 0; j < pce->num_ele[i]; j++) {
353 ✗ if (i < 3)
354 ✗ put_bits(pb, 1, pce->pairing[i][j]);
355 ✗ put_bits(pb, 4, pce->index[i][j]);
356 }
357 }
358
359 ✗ align_put_bits(pb);
360 ✗ put_bits(pb, 8, strlen(aux_data));
361 ✗ ff_put_string(pb, aux_data, 0);
362 }
363
364 /**
365 * Make AAC audio config object.
366 * @see 1.6.2.1 "Syntax - AudioSpecificConfig"
367 */
368 11 static int put_audio_specific_config(AVCodecContext *avctx)
369 {
370 PutBitContext pb;
371 11 AACEncContext *s = avctx->priv_data;
372 11 int channels = (!s->needs_pce)*(s->channels - (s->channels == 8 ? 1 : 0));
373 11 const int max_size = 32;
374
375 11 avctx->extradata = av_mallocz(max_size);
376
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
 11 if (!avctx->extradata)
377 ✗ return AVERROR(ENOMEM);
378
379 11 init_put_bits(&pb, avctx->extradata, max_size);
380 11 put_bits(&pb, 5, s->profile+1); //profile
381 11 put_bits(&pb, 4, s->samplerate_index); //sample rate index
382 11 put_bits(&pb, 4, channels);
383 //GASpecificConfig
384 11 put_bits(&pb, 1, 0); //frame length - 1024 samples
385 11 put_bits(&pb, 1, 0); //does not depend on core coder
386 11 put_bits(&pb, 1, 0); //is not extension
387
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
 11 if (s->needs_pce)
388 ✗ put_pce(&pb, avctx);
389
390 //Explicitly Mark SBR absent
391 11 put_bits(&pb, 11, 0x2b7); //sync extension
392 11 put_bits(&pb, 5, AOT_SBR);
393 11 put_bits(&pb, 1, 0);
394 11 flush_put_bits(&pb);
395 11 avctx->extradata_size = put_bytes_output(&pb);
396
397 11 return 0;
398 }
399
400 10135 void ff_quantize_band_cost_cache_init(struct AACEncContext *s)
401 {
402 10135 ++s->quantize_band_cost_cache_generation;
403
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10135 times.
 10135 if (s->quantize_band_cost_cache_generation == 0) {
404 ✗ memset(s->quantize_band_cost_cache, 0, sizeof(s->quantize_band_cost_cache));
405 ✗ s->quantize_band_cost_cache_generation = 1;
406 }
407 10135 }
408
409 #define WINDOW_FUNC(type) \
410 static void apply_ ##type ##_window(AVFloatDSPContext *fdsp, \
411 SingleChannelElement *sce, \
412 const float *audio)
413
414 6899 WINDOW_FUNC(only_long)
415 {
416
2/2
✓ Branch 0 taken 6851 times.
✓ Branch 1 taken 48 times.
 6899 const float *lwindow = sce->ics.use_kb_window[0] ? ff_aac_kbd_long_1024 : ff_sine_1024;
417
2/2
✓ Branch 0 taken 6840 times.
✓ Branch 1 taken 59 times.
 6899 const float *pwindow = sce->ics.use_kb_window[1] ? ff_aac_kbd_long_1024 : ff_sine_1024;
418 6899 float *out = sce->ret_buf;
419
420 6899 fdsp->vector_fmul (out, audio, lwindow, 1024);
421 6899 fdsp->vector_fmul_reverse(out + 1024, audio + 1024, pwindow, 1024);
422 6899 }
423
424 119 WINDOW_FUNC(long_start)
425 {
426
2/2
✓ Branch 0 taken 106 times.
✓ Branch 1 taken 13 times.
 119 const float *lwindow = sce->ics.use_kb_window[1] ? ff_aac_kbd_long_1024 : ff_sine_1024;
427
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 119 times.
 119 const float *swindow = sce->ics.use_kb_window[0] ? ff_aac_kbd_short_128 : ff_sine_128;
428 119 float *out = sce->ret_buf;
429
430 119 fdsp->vector_fmul(out, audio, lwindow, 1024);
431 119 memcpy(out + 1024, audio + 1024, sizeof(out[0]) * 448);
432 119 fdsp->vector_fmul_reverse(out + 1024 + 448, audio + 1024 + 448, swindow, 128);
433 119 memset(out + 1024 + 576, 0, sizeof(out[0]) * 448);
434 119 }
435
436 102 WINDOW_FUNC(long_stop)
437 {
438
1/2
✓ Branch 0 taken 102 times.
✗ Branch 1 not taken.
 102 const float *lwindow = sce->ics.use_kb_window[0] ? ff_aac_kbd_long_1024 : ff_sine_1024;
439
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 102 times.
 102 const float *swindow = sce->ics.use_kb_window[1] ? ff_aac_kbd_short_128 : ff_sine_128;
440 102 float *out = sce->ret_buf;
441
442 102 memset(out, 0, sizeof(out[0]) * 448);
443 102 fdsp->vector_fmul(out + 448, audio + 448, swindow, 128);
444 102 memcpy(out + 576, audio + 576, sizeof(out[0]) * 448);
445 102 fdsp->vector_fmul_reverse(out + 1024, audio + 1024, lwindow, 1024);
446 102 }
447
448 164 WINDOW_FUNC(eight_short)
449 {
450
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 164 times.
 164 const float *swindow = sce->ics.use_kb_window[0] ? ff_aac_kbd_short_128 : ff_sine_128;
451
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 164 times.
 164 const float *pwindow = sce->ics.use_kb_window[1] ? ff_aac_kbd_short_128 : ff_sine_128;
452 164 const float *in = audio + 448;
453 164 float *out = sce->ret_buf;
454 int w;
455
456
2/2
✓ Branch 0 taken 1312 times.
✓ Branch 1 taken 164 times.
 1476 for (w = 0; w < 8; w++) {
457
2/2
✓ Branch 0 taken 1148 times.
✓ Branch 1 taken 164 times.
 1312 fdsp->vector_fmul (out, in, w ? pwindow : swindow, 128);
458 1312 out += 128;
459 1312 in += 128;
460 1312 fdsp->vector_fmul_reverse(out, in, swindow, 128);
461 1312 out += 128;
462 }
463 164 }
464
465 static void (*const apply_window[4])(AVFloatDSPContext *fdsp,
466 SingleChannelElement *sce,
467 const float *audio) = {
468 [ONLY_LONG_SEQUENCE] = apply_only_long_window,
469 [LONG_START_SEQUENCE] = apply_long_start_window,
470 [EIGHT_SHORT_SEQUENCE] = apply_eight_short_window,
471 [LONG_STOP_SEQUENCE] = apply_long_stop_window
472 };
473
474 7284 static void apply_window_and_mdct(AACEncContext *s, SingleChannelElement *sce,
475 float *audio)
476 {
477 int i;
478 7284 float *output = sce->ret_buf;
479
480 7284 apply_window[sce->ics.window_sequence[0]](s->fdsp, sce, audio);
481
482
2/2
✓ Branch 0 taken 7120 times.
✓ Branch 1 taken 164 times.
 7284 if (sce->ics.window_sequence[0] != EIGHT_SHORT_SEQUENCE)
483 7120 s->mdct1024_fn(s->mdct1024, sce->coeffs, output, sizeof(float));
484 else
485
2/2
✓ Branch 0 taken 1312 times.
✓ Branch 1 taken 164 times.
 1476 for (i = 0; i < 1024; i += 128)
486 1312 s->mdct128_fn(s->mdct128, &sce->coeffs[i], output + i*2, sizeof(float));
487 7284 memcpy(audio, audio + 1024, sizeof(audio[0]) * 1024);
488 7284 memcpy(sce->pcoeffs, sce->coeffs, sizeof(sce->pcoeffs));
489 7284 }
490
491 /**
492 * Encode ics_info element.
493 * @see Table 4.6 (syntax of ics_info)
494 */
495 5693 static void put_ics_info(AACEncContext *s, IndividualChannelStream *info)
496 {
497 int w;
498
499 5693 put_bits(&s->pb, 1, 0); // ics_reserved bit
500 5693 put_bits(&s->pb, 2, info->window_sequence[0]);
501 5693 put_bits(&s->pb, 1, info->use_kb_window[0]);
502
2/2
✓ Branch 0 taken 5513 times.
✓ Branch 1 taken 180 times.
 5693 if (info->window_sequence[0] != EIGHT_SHORT_SEQUENCE) {
503 5513 put_bits(&s->pb, 6, info->max_sfb);
504 5513 put_bits(&s->pb, 1, !!info->predictor_present);
505 } else {
506 180 put_bits(&s->pb, 4, info->max_sfb);
507
2/2
✓ Branch 0 taken 1260 times.
✓ Branch 1 taken 180 times.
 1440 for (w = 1; w < 8; w++)
508 1260 put_bits(&s->pb, 1, !info->group_len[w]);
509 }
510 5693 }
511
512 /**
513 * Encode MS data.
514 * @see 4.6.8.1 "Joint Coding - M/S Stereo"
515 */
516 4538 static void encode_ms_info(PutBitContext *pb, ChannelElement *cpe)
517 {
518 int i, w;
519
520 4538 put_bits(pb, 2, cpe->ms_mode);
521
2/2
✓ Branch 0 taken 402 times.
✓ Branch 1 taken 4136 times.
 4538 if (cpe->ms_mode == 1)
522
2/2
✓ Branch 0 taken 421 times.
✓ Branch 1 taken 402 times.
 823 for (w = 0; w < cpe->ch[0].ics.num_windows; w += cpe->ch[0].ics.group_len[w])
523
2/2
✓ Branch 0 taken 19160 times.
✓ Branch 1 taken 421 times.
 19581 for (i = 0; i < cpe->ch[0].ics.max_sfb; i++)
524 19160 put_bits(pb, 1, cpe->ms_mask[w*16 + i]);
525 4538 }
526
527 /**
528 * Produce integer coefficients from scalefactors provided by the model.
529 */
530 5448 static void adjust_frame_information(ChannelElement *cpe, int chans)
531 {
532 int i, w, w2, g, ch;
533 int maxsfb, cmaxsfb;
534
535
2/2
✓ Branch 0 taken 10231 times.
✓ Branch 1 taken 5448 times.
 15679 for (ch = 0; ch < chans; ch++) {
536 10231 IndividualChannelStream *ics = &cpe->ch[ch].ics;
537 10231 maxsfb = 0;
538 10231 cpe->ch[ch].pulse.num_pulse = 0;
539
2/2
✓ Branch 0 taken 10883 times.
✓ Branch 1 taken 10231 times.
 21114 for (w = 0; w < ics->num_windows; w += ics->group_len[w]) {
540
2/2
✓ Branch 0 taken 12002 times.
✓ Branch 1 taken 10883 times.
 22885 for (w2 = 0; w2 < ics->group_len[w]; w2++) {
541
4/4
✓ Branch 0 taken 29181 times.
✓ Branch 1 taken 764 times.
✓ Branch 2 taken 17943 times.
✓ Branch 3 taken 11238 times.
 29945 for (cmaxsfb = ics->num_swb; cmaxsfb > 0 && cpe->ch[ch].zeroes[w*16+cmaxsfb-1]; cmaxsfb--)
542 ;
543 12002 maxsfb = FFMAX(maxsfb, cmaxsfb);
544 }
545 }
546 10231 ics->max_sfb = maxsfb;
547
548 //adjust zero bands for window groups
549
2/2
✓ Branch 0 taken 10883 times.
✓ Branch 1 taken 10231 times.
 21114 for (w = 0; w < ics->num_windows; w += ics->group_len[w]) {
550
2/2
✓ Branch 0 taken 487300 times.
✓ Branch 1 taken 10883 times.
 498183 for (g = 0; g < ics->max_sfb; g++) {
551 487300 i = 1;
552
2/2
✓ Branch 0 taken 487520 times.
✓ Branch 1 taken 2955 times.
 490475 for (w2 = w; w2 < w + ics->group_len[w]; w2++) {
553
2/2
✓ Branch 0 taken 484345 times.
✓ Branch 1 taken 3175 times.
 487520 if (!cpe->ch[ch].zeroes[w2*16 + g]) {
554 484345 i = 0;
555 484345 break;
556 }
557 }
558 487300 cpe->ch[ch].zeroes[w*16 + g] = i;
559 }
560 }
561 }
562
563
4/4
✓ Branch 0 taken 4783 times.
✓ Branch 1 taken 665 times.
✓ Branch 2 taken 4538 times.
✓ Branch 3 taken 245 times.
 5448 if (chans > 1 && cpe->common_window) {
564 4538 IndividualChannelStream *ics0 = &cpe->ch[0].ics;
565 4538 IndividualChannelStream *ics1 = &cpe->ch[1].ics;
566 4538 int msc = 0;
567 4538 ics0->max_sfb = FFMAX(ics0->max_sfb, ics1->max_sfb);
568 4538 ics1->max_sfb = ics0->max_sfb;
569
2/2
✓ Branch 0 taken 5049 times.
✓ Branch 1 taken 4538 times.
 9587 for (w = 0; w < ics0->num_windows*16; w += 16)
570
2/2
✓ Branch 0 taken 220504 times.
✓ Branch 1 taken 5049 times.
 225553 for (i = 0; i < ics0->max_sfb; i++)
571
2/2
✓ Branch 0 taken 24406 times.
✓ Branch 1 taken 196098 times.
 220504 if (cpe->ms_mask[w+i])
572 24406 msc++;
573
3/4
✓ Branch 0 taken 789 times.
✓ Branch 1 taken 3749 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 789 times.
 4538 if (msc == 0 || ics0->max_sfb == 0)
574 3749 cpe->ms_mode = 0;
575 else
576
2/2
✓ Branch 0 taken 402 times.
✓ Branch 1 taken 387 times.
 789 cpe->ms_mode = msc < ics0->max_sfb * ics0->num_windows ? 1 : 2;
577 }
578 5448 }
579
580 1334 static void apply_intensity_stereo(ChannelElement *cpe)
581 {
582 int w, w2, g, i;
583 1334 IndividualChannelStream *ics = &cpe->ch[0].ics;
584
2/2
✓ Branch 0 taken 707 times.
✓ Branch 1 taken 627 times.
 1334 if (!cpe->common_window)
585 707 return;
586
2/2
✓ Branch 0 taken 661 times.
✓ Branch 1 taken 627 times.
 1288 for (w = 0; w < ics->num_windows; w += ics->group_len[w]) {
587
2/2
✓ Branch 0 taken 725 times.
✓ Branch 1 taken 661 times.
 1386 for (w2 = 0; w2 < ics->group_len[w]; w2++) {
588 725 int start = (w+w2) * 128;
589
2/2
✓ Branch 0 taken 31605 times.
✓ Branch 1 taken 725 times.
 32330 for (g = 0; g < ics->num_swb; g++) {
590 31605 int p = -1 + 2 * (cpe->ch[1].band_type[w*16+g] - 14);
591 31605 float scale = cpe->ch[0].is_ener[w*16+g];
592
2/2
✓ Branch 0 taken 22930 times.
✓ Branch 1 taken 8675 times.
 31605 if (!cpe->is_mask[w*16 + g]) {
593 22930 start += ics->swb_sizes[g];
594 22930 continue;
595 }
596
2/2
✓ Branch 0 taken 2383 times.
✓ Branch 1 taken 6292 times.
 8675 if (cpe->ms_mask[w*16 + g])
597 2383 p *= -1;
598
2/2
✓ Branch 0 taken 292136 times.
✓ Branch 1 taken 8675 times.
 300811 for (i = 0; i < ics->swb_sizes[g]; i++) {
599 292136 float sum = (cpe->ch[0].coeffs[start+i] + p*cpe->ch[1].coeffs[start+i])*scale;
600 292136 cpe->ch[0].coeffs[start+i] = sum;
601 292136 cpe->ch[1].coeffs[start+i] = 0.0f;
602 }
603 8675 start += ics->swb_sizes[g];
604 }
605 }
606 }
607 }
608
609 1050 static void apply_mid_side_stereo(ChannelElement *cpe)
610 {
611 int w, w2, g, i;
612 1050 IndividualChannelStream *ics = &cpe->ch[0].ics;
613
2/2
✓ Branch 0 taken 548 times.
✓ Branch 1 taken 502 times.
 1050 if (!cpe->common_window)
614 548 return;
615
2/2
✓ Branch 0 taken 536 times.
✓ Branch 1 taken 502 times.
 1038 for (w = 0; w < ics->num_windows; w += ics->group_len[w]) {
616
2/2
✓ Branch 0 taken 600 times.
✓ Branch 1 taken 536 times.
 1136 for (w2 = 0; w2 < ics->group_len[w]; w2++) {
617 600 int start = (w+w2) * 128;
618
2/2
✓ Branch 0 taken 25480 times.
✓ Branch 1 taken 600 times.
 26080 for (g = 0; g < ics->num_swb; g++) {
619 /* ms_mask can be used for other purposes in PNS and I/S,
620 * so must not apply M/S if any band uses either, even if
621 * ms_mask is set.
622 */
623
3/4
✓ Branch 0 taken 22915 times.
✓ Branch 1 taken 2565 times.
✓ Branch 2 taken 22915 times.
✗ Branch 3 not taken.
 25480 if (!cpe->ms_mask[w*16 + g] || cpe->is_mask[w*16 + g]
624
1/2
✓ Branch 0 taken 22915 times.
✗ Branch 1 not taken.
 22915 || cpe->ch[0].band_type[w*16 + g] >= NOISE_BT
625
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 22915 times.
 22915 || cpe->ch[1].band_type[w*16 + g] >= NOISE_BT) {
626 2565 start += ics->swb_sizes[g];
627 2565 continue;
628 }
629
2/2
✓ Branch 0 taken 461484 times.
✓ Branch 1 taken 22915 times.
 484399 for (i = 0; i < ics->swb_sizes[g]; i++) {
630 461484 float L = (cpe->ch[0].coeffs[start+i] + cpe->ch[1].coeffs[start+i]) * 0.5f;
631 461484 float R = L - cpe->ch[1].coeffs[start+i];
632 461484 cpe->ch[0].coeffs[start+i] = L;
633 461484 cpe->ch[1].coeffs[start+i] = R;
634 }
635 22915 start += ics->swb_sizes[g];
636 }
637 }
638 }
639 }
640
641 /**
642 * Encode scalefactor band coding type.
643 */
644 10231 static void encode_band_info(AACEncContext *s, SingleChannelElement *sce)
645 {
646 int w;
647
648
1/2
✓ Branch 0 taken 10231 times.
✗ Branch 1 not taken.
 10231 if (s->coder->set_special_band_scalefactors)
649 10231 s->coder->set_special_band_scalefactors(s, sce);
650
651
2/2
✓ Branch 0 taken 10883 times.
✓ Branch 1 taken 10231 times.
 21114 for (w = 0; w < sce->ics.num_windows; w += sce->ics.group_len[w])
652 10883 s->coder->encode_window_bands_info(s, sce, w, sce->ics.group_len[w], s->lambda);
653 10231 }
654
655 /**
656 * Encode scalefactors.
657 */
658 10231 static void encode_scale_factors(AVCodecContext *avctx, AACEncContext *s,
659 SingleChannelElement *sce)
660 {
661 10231 int diff, off_sf = sce->sf_idx[0], off_pns = sce->sf_idx[0] - NOISE_OFFSET;
662 10231 int off_is = 0, noise_flag = 1;
663 int i, w;
664
665
2/2
✓ Branch 0 taken 10883 times.
✓ Branch 1 taken 10231 times.
 21114 for (w = 0; w < sce->ics.num_windows; w += sce->ics.group_len[w]) {
666
2/2
✓ Branch 0 taken 487305 times.
✓ Branch 1 taken 10883 times.
 498188 for (i = 0; i < sce->ics.max_sfb; i++) {
667
2/2
✓ Branch 0 taken 476374 times.
✓ Branch 1 taken 10931 times.
 487305 if (!sce->zeroes[w*16 + i]) {
668
2/2
✓ Branch 0 taken 6375 times.
✓ Branch 1 taken 469999 times.
 476374 if (sce->band_type[w*16 + i] == NOISE_BT) {
669 6375 diff = sce->sf_idx[w*16 + i] - off_pns;
670 6375 off_pns = sce->sf_idx[w*16 + i];
671
2/2
✓ Branch 0 taken 1122 times.
✓ Branch 1 taken 5253 times.
 6375 if (noise_flag-- > 0) {
672 1122 put_bits(&s->pb, NOISE_PRE_BITS, diff + NOISE_PRE);
673 1122 continue;
674 }
675
2/2
✓ Branch 0 taken 463673 times.
✓ Branch 1 taken 6326 times.
 469999 } else if (sce->band_type[w*16 + i] == INTENSITY_BT ||
676
2/2
✓ Branch 0 taken 2238 times.
✓ Branch 1 taken 461435 times.
 463673 sce->band_type[w*16 + i] == INTENSITY_BT2) {
677 8564 diff = sce->sf_idx[w*16 + i] - off_is;
678 8564 off_is = sce->sf_idx[w*16 + i];
679 } else {
680 461435 diff = sce->sf_idx[w*16 + i] - off_sf;
681 461435 off_sf = sce->sf_idx[w*16 + i];
682 }
683 475252 diff += SCALE_DIFF_ZERO;
684
2/4
✓ Branch 0 taken 475252 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 475252 times.
 475252 av_assert0(diff >= 0 && diff <= 120);
685 475252 put_bits(&s->pb, ff_aac_scalefactor_bits[diff], ff_aac_scalefactor_code[diff]);
686 }
687 }
688 }
689 10231 }
690
691 /**
692 * Encode pulse data.
693 */
694 10231 static void encode_pulses(AACEncContext *s, Pulse *pulse)
695 {
696 int i;
697
698 10231 put_bits(&s->pb, 1, !!pulse->num_pulse);
699
1/2
✓ Branch 0 taken 10231 times.
✗ Branch 1 not taken.
 10231 if (!pulse->num_pulse)
700 10231 return;
701
702 ✗ put_bits(&s->pb, 2, pulse->num_pulse - 1);
703 ✗ put_bits(&s->pb, 6, pulse->start);
704 ✗ for (i = 0; i < pulse->num_pulse; i++) {
705 ✗ put_bits(&s->pb, 5, pulse->pos[i]);
706 ✗ put_bits(&s->pb, 4, pulse->amp[i]);
707 }
708 }
709
710 /**
711 * Encode spectral coefficients processed by psychoacoustic model.
712 */
713 10231 static void encode_spectral_coeffs(AACEncContext *s, SingleChannelElement *sce)
714 {
715 int start, i, w, w2;
716
717
2/2
✓ Branch 0 taken 10883 times.
✓ Branch 1 taken 10231 times.
 21114 for (w = 0; w < sce->ics.num_windows; w += sce->ics.group_len[w]) {
718 10883 start = 0;
719
2/2
✓ Branch 0 taken 487305 times.
✓ Branch 1 taken 10883 times.
 498188 for (i = 0; i < sce->ics.max_sfb; i++) {
720
2/2
✓ Branch 0 taken 10931 times.
✓ Branch 1 taken 476374 times.
 487305 if (sce->zeroes[w*16 + i]) {
721 10931 start += sce->ics.swb_sizes[i];
722 10931 continue;
723 }
724
2/2
✓ Branch 0 taken 484380 times.
✓ Branch 1 taken 476374 times.
 960754 for (w2 = w; w2 < w + sce->ics.group_len[w]; w2++) {
725 484380 s->coder->quantize_and_encode_band(s, &s->pb,
726 484380 &sce->coeffs[start + w2*128],
727 484380 NULL, sce->ics.swb_sizes[i],
728 484380 sce->sf_idx[w*16 + i],
729 484380 sce->band_type[w*16 + i],
730 s->lambda,
731 484380 sce->ics.window_clipping[w]);
732 }
733 476374 start += sce->ics.swb_sizes[i];
734 }
735 }
736 10231 }
737
738 /**
739 * Downscale spectral coefficients for near-clipping windows to avoid artifacts
740 */
741 7284 static void avoid_clipping(AACEncContext *s, SingleChannelElement *sce)
742 {
743 int start, i, j, w;
744
745
2/2
✓ Branch 0 taken 168 times.
✓ Branch 1 taken 7116 times.
 7284 if (sce->ics.clip_avoidance_factor < 1.0f) {
746
2/2
✓ Branch 0 taken 224 times.
✓ Branch 1 taken 168 times.
 392 for (w = 0; w < sce->ics.num_windows; w++) {
747 224 start = 0;
748
2/2
✓ Branch 0 taken 8466 times.
✓ Branch 1 taken 224 times.
 8690 for (i = 0; i < sce->ics.max_sfb; i++) {
749 8466 float *swb_coeffs = &sce->coeffs[start + w*128];
750
2/2
✓ Branch 0 taken 160192 times.
✓ Branch 1 taken 8466 times.
 168658 for (j = 0; j < sce->ics.swb_sizes[i]; j++)
751 160192 swb_coeffs[j] *= sce->ics.clip_avoidance_factor;
752 8466 start += sce->ics.swb_sizes[i];
753 }
754 }
755 }
756 7284 }
757
758 /**
759 * Encode one channel of audio data.
760 */
761 10231 static int encode_individual_channel(AVCodecContext *avctx, AACEncContext *s,
762 SingleChannelElement *sce,
763 int common_window)
764 {
765 10231 put_bits(&s->pb, 8, sce->sf_idx[0]);
766
2/2
✓ Branch 0 taken 1155 times.
✓ Branch 1 taken 9076 times.
 10231 if (!common_window) {
767 1155 put_ics_info(s, &sce->ics);
768
1/2
✓ Branch 0 taken 1155 times.
✗ Branch 1 not taken.
 1155 if (s->coder->encode_main_pred)
769 1155 s->coder->encode_main_pred(s, sce);
770
1/2
✓ Branch 0 taken 1155 times.
✗ Branch 1 not taken.
 1155 if (s->coder->encode_ltp_info)
771 1155 s->coder->encode_ltp_info(s, sce, 0);
772 }
773 10231 encode_band_info(s, sce);
774 10231 encode_scale_factors(avctx, s, sce);
775 10231 encode_pulses(s, &sce->pulse);
776 10231 put_bits(&s->pb, 1, !!sce->tns.present);
777
1/2
✓ Branch 0 taken 10231 times.
✗ Branch 1 not taken.
 10231 if (s->coder->encode_tns_info)
778 10231 s->coder->encode_tns_info(s, sce);
779 10231 put_bits(&s->pb, 1, 0); //ssr
780 10231 encode_spectral_coeffs(s, sce);
781 10231 return 0;
782 }
783
784 /**
785 * Write some auxiliary information about the created AAC file.
786 */
787 ✗ static void put_bitstream_info(AACEncContext *s, const char *name)
788 {
789 int i, namelen, padbits;
790
791 ✗ namelen = strlen(name) + 2;
792 ✗ put_bits(&s->pb, 3, TYPE_FIL);
793 ✗ put_bits(&s->pb, 4, FFMIN(namelen, 15));
794 ✗ if (namelen >= 15)
795 ✗ put_bits(&s->pb, 8, namelen - 14);
796 ✗ put_bits(&s->pb, 4, 0); //extension type - filler
797 ✗ padbits = -put_bits_count(&s->pb) & 7;
798 ✗ align_put_bits(&s->pb);
799 ✗ for (i = 0; i < namelen - 2; i++)
800 ✗ put_bits(&s->pb, 8, name[i]);
801 ✗ put_bits(&s->pb, 12 - padbits, 0);
802 }
803
804 /*
805 * Copy input samples.
806 * Channels are reordered from libavcodec's default order to AAC order.
807 */
808 3816 static void copy_input_samples(AACEncContext *s, const AVFrame *frame)
809 {
810 int ch;
811
2/2
✓ Branch 0 taken 3794 times.
✓ Branch 1 taken 22 times.
 3816 int end = 2048 + (frame ? frame->nb_samples : 0);
812 3816 const uint8_t *channel_map = s->reorder_map;
813
814 /* copy and remap input samples */
815
2/2
✓ Branch 0 taken 7309 times.
✓ Branch 1 taken 3816 times.
 11125 for (ch = 0; ch < s->channels; ch++) {
816 /* copy last 1024 samples of previous frame to the start of the current frame */
817 7309 memcpy(&s->planar_samples[ch][1024], &s->planar_samples[ch][2048], 1024 * sizeof(s->planar_samples[0][0]));
818
819 /* copy new samples and zero any remaining samples */
820
2/2
✓ Branch 0 taken 7259 times.
✓ Branch 1 taken 50 times.
 7309 if (frame) {
821 7259 memcpy(&s->planar_samples[ch][2048],
822 7259 frame->extended_data[channel_map[ch]],
823 7259 frame->nb_samples * sizeof(s->planar_samples[0][0]));
824 }
825 7309 memset(&s->planar_samples[ch][end], 0,
826 7309 (3072 - end) * sizeof(s->planar_samples[0][0]));
827 }
828 3816 }
829
830 3827 static int aac_encode_frame(AVCodecContext *avctx, AVPacket *avpkt,
831 const AVFrame *frame, int *got_packet_ptr)
832 {
833 3827 AACEncContext *s = avctx->priv_data;
834 3827 float **samples = s->planar_samples, *samples2, *la, *overlap;
835 ChannelElement *cpe;
836 SingleChannelElement *sce;
837 IndividualChannelStream *ics;
838 int i, its, ch, w, chans, tag, start_ch, ret, frame_bits;
839 int target_bits, rate_bits, too_many_bits, too_few_bits;
840 3827 int ms_mode = 0, is_mode = 0, tns_mode = 0, pred_mode = 0;
841 int chan_el_counter[4];
842 FFPsyWindowInfo windows[AAC_MAX_CHANNELS];
843
844 /* add current frame to queue */
845
2/2
✓ Branch 0 taken 3794 times.
✓ Branch 1 taken 33 times.
 3827 if (frame) {
846
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3794 times.
 3794 if ((ret = ff_af_queue_add(&s->afq, frame)) < 0)
847 ✗ return ret;
848 } else {
849
3/6
✓ Branch 0 taken 22 times.
✓ Branch 1 taken 11 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 22 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 33 if (!s->afq.remaining_samples || (!s->afq.frame_alloc && !s->afq.frame_count))
850 11 return 0;
851 }
852
853 3816 copy_input_samples(s, frame);
854
1/2
✓ Branch 0 taken 3816 times.
✗ Branch 1 not taken.
 3816 if (s->psypp)
855 3816 ff_psy_preprocess(s->psypp, s->planar_samples, s->channels);
856
857
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 3805 times.
 3816 if (!avctx->frame_num)
858 11 return 0;
859
860 3805 start_ch = 0;
861
2/2
✓ Branch 0 taken 3949 times.
✓ Branch 1 taken 3805 times.
 7754 for (i = 0; i < s->chan_map[0]; i++) {
862 3949 FFPsyWindowInfo* wi = windows + start_ch;
863 3949 tag = s->chan_map[i+1];
864
2/2
✓ Branch 0 taken 3335 times.
✓ Branch 1 taken 614 times.
 3949 chans = tag == TYPE_CPE ? 2 : 1;
865 3949 cpe = &s->cpe[i];
866
2/2
✓ Branch 0 taken 7284 times.
✓ Branch 1 taken 3949 times.
 11233 for (ch = 0; ch < chans; ch++) {
867 int k;
868 float clip_avoidance_factor;
869 7284 sce = &cpe->ch[ch];
870 7284 ics = &sce->ics;
871 7284 s->cur_channel = start_ch + ch;
872 7284 overlap = &samples[s->cur_channel][0];
873 7284 samples2 = overlap + 1024;
874 7284 la = samples2 + (448+64);
875
2/2
✓ Branch 0 taken 50 times.
✓ Branch 1 taken 7234 times.
 7284 if (!frame)
876 50 la = NULL;
877
2/2
✓ Branch 0 taken 48 times.
✓ Branch 1 taken 7236 times.
 7284 if (tag == TYPE_LFE) {
878 48 wi[ch].window_type[0] = wi[ch].window_type[1] = ONLY_LONG_SEQUENCE;
879 48 wi[ch].window_shape = 0;
880 48 wi[ch].num_windows = 1;
881 48 wi[ch].grouping[0] = 1;
882 48 wi[ch].clipping[0] = 0;
883
884 /* Only the lowest 12 coefficients are used in a LFE channel.
885 * The expression below results in only the bottom 8 coefficients
886 * being used for 11.025kHz to 16kHz sample rates.
887 */
888
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 48 times.
 48 ics->num_swb = s->samplerate_index >= 8 ? 1 : 3;
889 } else {
890 7236 wi[ch] = s->psy.model->window(&s->psy, samples2, la, s->cur_channel,
891 7236 ics->window_sequence[0]);
892 }
893 7284 ics->window_sequence[1] = ics->window_sequence[0];
894 7284 ics->window_sequence[0] = wi[ch].window_type[0];
895 7284 ics->use_kb_window[1] = ics->use_kb_window[0];
896 7284 ics->use_kb_window[0] = wi[ch].window_shape;
897 7284 ics->num_windows = wi[ch].num_windows;
898
2/2
✓ Branch 0 taken 164 times.
✓ Branch 1 taken 7120 times.
 7284 ics->swb_sizes = s->psy.bands [ics->num_windows == 8];
899
4/4
✓ Branch 0 taken 48 times.
✓ Branch 1 taken 7236 times.
✓ Branch 2 taken 164 times.
✓ Branch 3 taken 7072 times.
 7284 ics->num_swb = tag == TYPE_LFE ? ics->num_swb : s->psy.num_bands[ics->num_windows == 8];
900 7284 ics->max_sfb = FFMIN(ics->max_sfb, ics->num_swb);
901 14568 ics->swb_offset = wi[ch].window_type[0] == EIGHT_SHORT_SEQUENCE ?
902
2/2
✓ Branch 0 taken 164 times.
✓ Branch 1 taken 7120 times.
 7284 ff_swb_offset_128 [s->samplerate_index]:
903 7120 ff_swb_offset_1024[s->samplerate_index];
904 14568 ics->tns_max_bands = wi[ch].window_type[0] == EIGHT_SHORT_SEQUENCE ?
905
2/2
✓ Branch 0 taken 164 times.
✓ Branch 1 taken 7120 times.
 7284 ff_tns_max_bands_128 [s->samplerate_index]:
906 7120 ff_tns_max_bands_1024[s->samplerate_index];
907
908
2/2
✓ Branch 0 taken 8432 times.
✓ Branch 1 taken 7284 times.
 15716 for (w = 0; w < ics->num_windows; w++)
909 8432 ics->group_len[w] = wi[ch].grouping[w];
910
911 /* Calculate input sample maximums and evaluate clipping risk */
912 7284 clip_avoidance_factor = 0.0f;
913
2/2
✓ Branch 0 taken 8432 times.
✓ Branch 1 taken 7284 times.
 15716 for (w = 0; w < ics->num_windows; w++) {
914 8432 const float *wbuf = overlap + w * 128;
915 8432 const int wlen = 2048 / ics->num_windows;
916 8432 float max = 0;
917 int j;
918 /* mdct input is 2 * output */
919
2/2
✓ Branch 0 taken 14917632 times.
✓ Branch 1 taken 8432 times.
 14926064 for (j = 0; j < wlen; j++)
920
2/2
✓ Branch 0 taken 14599329 times.
✓ Branch 1 taken 318303 times.
 14917632 max = FFMAX(max, fabsf(wbuf[j]));
921 8432 wi[ch].clipping[w] = max;
922 }
923
2/2
✓ Branch 0 taken 8432 times.
✓ Branch 1 taken 7284 times.
 15716 for (w = 0; w < ics->num_windows; w++) {
924
2/2
✓ Branch 0 taken 176 times.
✓ Branch 1 taken 8256 times.
 8432 if (wi[ch].clipping[w] > CLIP_AVOIDANCE_FACTOR) {
925 176 ics->window_clipping[w] = 1;
926
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 176 times.
 176 clip_avoidance_factor = FFMAX(clip_avoidance_factor, wi[ch].clipping[w]);
927 } else {
928 8256 ics->window_clipping[w] = 0;
929 }
930 }
931
2/2
✓ Branch 0 taken 168 times.
✓ Branch 1 taken 7116 times.
 7284 if (clip_avoidance_factor > CLIP_AVOIDANCE_FACTOR) {
932 168 ics->clip_avoidance_factor = CLIP_AVOIDANCE_FACTOR / clip_avoidance_factor;
933 } else {
934 7116 ics->clip_avoidance_factor = 1.0f;
935 }
936
937 7284 apply_window_and_mdct(s, sce, overlap);
938
939
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 7284 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 7284 if (s->options.ltp && s->coder->update_ltp) {
940 ✗ s->coder->update_ltp(s, sce);
941 ✗ apply_window[sce->ics.window_sequence[0]](s->fdsp, sce, &sce->ltp_state[0]);
942 ✗ s->mdct1024_fn(s->mdct1024, sce->lcoeffs, sce->ret_buf, sizeof(float));
943 }
944
945
2/2
✓ Branch 0 taken 7458816 times.
✓ Branch 1 taken 7284 times.
 7466100 for (k = 0; k < 1024; k++) {
946
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7458816 times.
 7458816 if (!(fabs(cpe->ch[ch].coeffs[k]) < 1E16)) { // Ensure headroom for energy calculation
947 ✗ av_log(avctx, AV_LOG_ERROR, "Input contains (near) NaN/+-Inf\n");
948 ✗ return AVERROR(EINVAL);
949 }
950 }
951 7284 avoid_clipping(s, sce);
952 }
953 3949 start_ch += chans;
954 }
955
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3805 times.
 3805 if ((ret = ff_alloc_packet(avctx, avpkt, 8192 * s->channels)) < 0)
956 ✗ return ret;
957 3805 frame_bits = its = 0;
958 do {
959 5232 init_put_bits(&s->pb, avpkt->data, avpkt->size);
960
961
3/4
✓ Branch 0 taken 35 times.
✓ Branch 1 taken 5197 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 35 times.
 5232 if ((avctx->frame_num & 0xFF)==1 && !(avctx->flags & AV_CODEC_FLAG_BITEXACT))
962 ✗ put_bitstream_info(s, LIBAVCODEC_IDENT);
963 5232 start_ch = 0;
964 5232 target_bits = 0;
965 5232 memset(chan_el_counter, 0, sizeof(chan_el_counter));
966
2/2
✓ Branch 0 taken 5448 times.
✓ Branch 1 taken 5232 times.
 10680 for (i = 0; i < s->chan_map[0]; i++) {
967 5448 FFPsyWindowInfo* wi = windows + start_ch;
968 const float *coeffs[2];
969 5448 tag = s->chan_map[i+1];
970
2/2
✓ Branch 0 taken 4783 times.
✓ Branch 1 taken 665 times.
 5448 chans = tag == TYPE_CPE ? 2 : 1;
971 5448 cpe = &s->cpe[i];
972 5448 cpe->common_window = 0;
973 5448 memset(cpe->is_mask, 0, sizeof(cpe->is_mask));
974 5448 memset(cpe->ms_mask, 0, sizeof(cpe->ms_mask));
975 5448 put_bits(&s->pb, 3, tag);
976 5448 put_bits(&s->pb, 4, chan_el_counter[tag]++);
977
2/2
✓ Branch 0 taken 10231 times.
✓ Branch 1 taken 5448 times.
 15679 for (ch = 0; ch < chans; ch++) {
978 10231 sce = &cpe->ch[ch];
979 10231 coeffs[ch] = sce->coeffs;
980 10231 sce->ics.predictor_present = 0;
981 10231 sce->ics.ltp.present = 0;
982 10231 memset(sce->ics.ltp.used, 0, sizeof(sce->ics.ltp.used));
983 10231 memset(sce->ics.prediction_used, 0, sizeof(sce->ics.prediction_used));
984 10231 memset(&sce->tns, 0, sizeof(TemporalNoiseShaping));
985
2/2
✓ Branch 0 taken 1309568 times.
✓ Branch 1 taken 10231 times.
 1319799 for (w = 0; w < 128; w++)
986
2/2
✓ Branch 0 taken 14871 times.
✓ Branch 1 taken 1294697 times.
 1309568 if (sce->band_type[w] > RESERVED_BT)
987 14871 sce->band_type[w] = 0;
988 }
989 5448 s->psy.bitres.alloc = -1;
990 5448 s->psy.bitres.bits = s->last_frame_pb_count / s->channels;
991 5448 s->psy.model->analyze(&s->psy, start_ch, coeffs, wi);
992
1/2
✓ Branch 0 taken 5448 times.
✗ Branch 1 not taken.
 5448 if (s->psy.bitres.alloc > 0) {
993 /* Lambda unused here on purpose, we need to take psy's unscaled allocation */
994 10896 target_bits += s->psy.bitres.alloc
995
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5448 times.
 5448 * (s->lambda / (avctx->global_quality ? avctx->global_quality : 120));
996 5448 s->psy.bitres.alloc /= chans;
997 }
998 5448 s->cur_type = tag;
999
2/2
✓ Branch 0 taken 10231 times.
✓ Branch 1 taken 5448 times.
 15679 for (ch = 0; ch < chans; ch++) {
1000 10231 s->cur_channel = start_ch + ch;
1001
3/4
✓ Branch 0 taken 2003 times.
✓ Branch 1 taken 8228 times.
✓ Branch 2 taken 2003 times.
✗ Branch 3 not taken.
 10231 if (s->options.pns && s->coder->mark_pns)
1002 2003 s->coder->mark_pns(s, avctx, &cpe->ch[ch]);
1003 10231 s->coder->search_for_quantizers(avctx, s, &cpe->ch[ch], s->lambda);
1004 }
1005
2/2
✓ Branch 0 taken 4783 times.
✓ Branch 1 taken 665 times.
 5448 if (chans > 1
1006
2/2
✓ Branch 0 taken 4547 times.
✓ Branch 1 taken 236 times.
 4783 && wi[0].window_type[0] == wi[1].window_type[0]
1007
1/2
✓ Branch 0 taken 4547 times.
✗ Branch 1 not taken.
 4547 && wi[0].window_shape == wi[1].window_shape) {
1008
1009 4547 cpe->common_window = 1;
1010
2/2
✓ Branch 0 taken 5058 times.
✓ Branch 1 taken 4538 times.
 9596 for (w = 0; w < wi[0].num_windows; w++) {
1011
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 5049 times.
 5058 if (wi[0].grouping[w] != wi[1].grouping[w]) {
1012 9 cpe->common_window = 0;
1013 9 break;
1014 }
1015 }
1016 }
1017
2/2
✓ Branch 0 taken 10231 times.
✓ Branch 1 taken 5448 times.
 15679 for (ch = 0; ch < chans; ch++) { /* TNS and PNS */
1018 10231 sce = &cpe->ch[ch];
1019 10231 s->cur_channel = start_ch + ch;
1020
3/4
✓ Branch 0 taken 2003 times.
✓ Branch 1 taken 8228 times.
✓ Branch 2 taken 2003 times.
✗ Branch 3 not taken.
 10231 if (s->options.tns && s->coder->search_for_tns)
1021 2003 s->coder->search_for_tns(s, sce);
1022
3/4
✓ Branch 0 taken 2003 times.
✓ Branch 1 taken 8228 times.
✓ Branch 2 taken 2003 times.
✗ Branch 3 not taken.
 10231 if (s->options.tns && s->coder->apply_tns_filt)
1023 2003 s->coder->apply_tns_filt(s, sce);
1024
2/2
✓ Branch 0 taken 40 times.
✓ Branch 1 taken 10191 times.
 10231 if (sce->tns.present)
1025 40 tns_mode = 1;
1026
3/4
✓ Branch 0 taken 2003 times.
✓ Branch 1 taken 8228 times.
✓ Branch 2 taken 2003 times.
✗ Branch 3 not taken.
 10231 if (s->options.pns && s->coder->search_for_pns)
1027 2003 s->coder->search_for_pns(s, avctx, sce);
1028 }
1029 5448 s->cur_channel = start_ch;
1030
2/2
✓ Branch 0 taken 1334 times.
✓ Branch 1 taken 4114 times.
 5448 if (s->options.intensity_stereo) { /* Intensity Stereo */
1031
1/2
✓ Branch 0 taken 1334 times.
✗ Branch 1 not taken.
 1334 if (s->coder->search_for_is)
1032 1334 s->coder->search_for_is(s, avctx, cpe);
1033
2/2
✓ Branch 0 taken 506 times.
✓ Branch 1 taken 828 times.
 1334 if (cpe->is_mode) is_mode = 1;
1034 1334 apply_intensity_stereo(cpe);
1035 }
1036
2/2
✓ Branch 0 taken 416 times.
✓ Branch 1 taken 5032 times.
 5448 if (s->options.pred) { /* Prediction */
1037
2/2
✓ Branch 0 taken 832 times.
✓ Branch 1 taken 416 times.
 1248 for (ch = 0; ch < chans; ch++) {
1038 832 sce = &cpe->ch[ch];
1039 832 s->cur_channel = start_ch + ch;
1040
2/4
✓ Branch 0 taken 832 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 832 times.
✗ Branch 3 not taken.
 832 if (s->options.pred && s->coder->search_for_pred)
1041 832 s->coder->search_for_pred(s, sce);
1042
2/2
✓ Branch 0 taken 440 times.
✓ Branch 1 taken 392 times.
 832 if (cpe->ch[ch].ics.predictor_present) pred_mode = 1;
1043 }
1044
1/2
✓ Branch 0 taken 416 times.
✗ Branch 1 not taken.
 416 if (s->coder->adjust_common_pred)
1045 416 s->coder->adjust_common_pred(s, cpe);
1046
2/2
✓ Branch 0 taken 832 times.
✓ Branch 1 taken 416 times.
 1248 for (ch = 0; ch < chans; ch++) {
1047 832 sce = &cpe->ch[ch];
1048 832 s->cur_channel = start_ch + ch;
1049
2/4
✓ Branch 0 taken 832 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 832 times.
✗ Branch 3 not taken.
 832 if (s->options.pred && s->coder->apply_main_pred)
1050 832 s->coder->apply_main_pred(s, sce);
1051 }
1052 416 s->cur_channel = start_ch;
1053 }
1054
2/2
✓ Branch 0 taken 1050 times.
✓ Branch 1 taken 4398 times.
 5448 if (s->options.mid_side) { /* Mid/Side stereo */
1055
3/4
✓ Branch 0 taken 630 times.
✓ Branch 1 taken 420 times.
✓ Branch 2 taken 630 times.
✗ Branch 3 not taken.
 1050 if (s->options.mid_side == -1 && s->coder->search_for_ms)
1056 630 s->coder->search_for_ms(s, cpe);
1057
2/2
✓ Branch 0 taken 393 times.
✓ Branch 1 taken 27 times.
 420 else if (cpe->common_window)
1058 393 memset(cpe->ms_mask, 1, sizeof(cpe->ms_mask));
1059 1050 apply_mid_side_stereo(cpe);
1060 }
1061 5448 adjust_frame_information(cpe, chans);
1062
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5448 times.
 5448 if (s->options.ltp) { /* LTP */
1063 ✗ for (ch = 0; ch < chans; ch++) {
1064 ✗ sce = &cpe->ch[ch];
1065 ✗ s->cur_channel = start_ch + ch;
1066 ✗ if (s->coder->search_for_ltp)
1067 ✗ s->coder->search_for_ltp(s, sce, cpe->common_window);
1068 ✗ if (sce->ics.ltp.present) pred_mode = 1;
1069 }
1070 ✗ s->cur_channel = start_ch;
1071 ✗ if (s->coder->adjust_common_ltp)
1072 ✗ s->coder->adjust_common_ltp(s, cpe);
1073 }
1074
2/2
✓ Branch 0 taken 4783 times.
✓ Branch 1 taken 665 times.
 5448 if (chans == 2) {
1075 4783 put_bits(&s->pb, 1, cpe->common_window);
1076
2/2
✓ Branch 0 taken 4538 times.
✓ Branch 1 taken 245 times.
 4783 if (cpe->common_window) {
1077 4538 put_ics_info(s, &cpe->ch[0].ics);
1078
1/2
✓ Branch 0 taken 4538 times.
✗ Branch 1 not taken.
 4538 if (s->coder->encode_main_pred)
1079 4538 s->coder->encode_main_pred(s, &cpe->ch[0]);
1080
1/2
✓ Branch 0 taken 4538 times.
✗ Branch 1 not taken.
 4538 if (s->coder->encode_ltp_info)
1081 4538 s->coder->encode_ltp_info(s, &cpe->ch[0], 1);
1082 4538 encode_ms_info(&s->pb, cpe);
1083
2/2
✓ Branch 0 taken 789 times.
✓ Branch 1 taken 3749 times.
 4538 if (cpe->ms_mode) ms_mode = 1;
1084 }
1085 }
1086
2/2
✓ Branch 0 taken 10231 times.
✓ Branch 1 taken 5448 times.
 15679 for (ch = 0; ch < chans; ch++) {
1087 10231 s->cur_channel = start_ch + ch;
1088 10231 encode_individual_channel(avctx, s, &cpe->ch[ch], cpe->common_window);
1089 }
1090 5448 start_ch += chans;
1091 }
1092
1093
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5232 times.
 5232 if (avctx->flags & AV_CODEC_FLAG_QSCALE) {
1094 /* When using a constant Q-scale, don't mess with lambda */
1095 ✗ break;
1096 }
1097
1098 /* rate control stuff
1099 * allow between the nominal bitrate, and what psy's bit reservoir says to target
1100 * but drift towards the nominal bitrate always
1101 */
1102 5232 frame_bits = put_bits_count(&s->pb);
1103 5232 rate_bits = avctx->bit_rate * 1024 / avctx->sample_rate;
1104
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5232 times.
 5232 rate_bits = FFMIN(rate_bits, 6144 * s->channels - 3);
1105 5232 too_many_bits = FFMAX(target_bits, rate_bits);
1106
2/2
✓ Branch 0 taken 2074 times.
✓ Branch 1 taken 3158 times.
 5232 too_many_bits = FFMIN(too_many_bits, 6144 * s->channels - 3);
1107 5232 too_few_bits = FFMIN(FFMAX(rate_bits - rate_bits/4, target_bits), too_many_bits);
1108
1109 /* When using ABR, be strict (but only for increasing) */
1110 5232 too_few_bits = too_few_bits - too_few_bits/8;
1111 5232 too_many_bits = too_many_bits + too_many_bits/2;
1112
1113
2/2
✓ Branch 0 taken 1427 times.
✓ Branch 1 taken 3805 times.
 5232 if ( its == 0 /* for steady-state Q-scale tracking */
1114
5/6
✓ Branch 0 taken 1146 times.
✓ Branch 1 taken 281 times.
✓ Branch 2 taken 8 times.
✓ Branch 3 taken 1138 times.
✓ Branch 4 taken 8 times.
✗ Branch 5 not taken.
 1427 || (its < 5 && (frame_bits < too_few_bits || frame_bits > too_many_bits))
1115
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 289 times.
 289 || frame_bits >= 6144 * s->channels - 3 )
1116 {
1117 4943 float ratio = ((float)rate_bits) / frame_bits;
1118
1119
4/4
✓ Branch 0 taken 2926 times.
✓ Branch 1 taken 2017 times.
✓ Branch 2 taken 2925 times.
✓ Branch 3 taken 1 times.
 4943 if (frame_bits >= too_few_bits && frame_bits <= too_many_bits) {
1120 /*
1121 * This path is for steady-state Q-scale tracking
1122 * When frame bits fall within the stable range, we still need to adjust
1123 * lambda to maintain it like so in a stable fashion (large jumps in lambda
1124 * create artifacts and should be avoided), but slowly
1125 */
1126 2925 ratio = sqrtf(sqrtf(ratio));
1127 2925 ratio = av_clipf(ratio, 0.9f, 1.1f);
1128 } else {
1129 /* Not so fast though */
1130 2018 ratio = sqrtf(ratio);
1131 }
1132 4943 s->lambda = av_clipf(s->lambda * ratio, FLT_EPSILON, 65536.f);
1133
1134 /* Keep iterating if we must reduce and lambda is in the sky */
1135
4/4
✓ Branch 0 taken 4923 times.
✓ Branch 1 taken 20 times.
✓ Branch 2 taken 1407 times.
✓ Branch 3 taken 3516 times.
 4943 if (ratio > 0.9f && ratio < 1.1f) {
1136 break;
1137 } else {
1138
6/8
✓ Branch 0 taken 1416 times.
✓ Branch 1 taken 11 times.
✓ Branch 2 taken 1412 times.
✓ Branch 3 taken 4 times.
✓ Branch 4 taken 1412 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 1412 times.
 1427 if (is_mode || ms_mode || tns_mode || pred_mode) {
1139
2/2
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 15 times.
 45 for (i = 0; i < s->chan_map[0]; i++) {
1140 // Must restore coeffs
1141
2/2
✓ Branch 0 taken 10 times.
✓ Branch 1 taken 20 times.
 30 chans = tag == TYPE_CPE ? 2 : 1;
1142 30 cpe = &s->cpe[i];
1143
2/2
✓ Branch 0 taken 40 times.
✓ Branch 1 taken 30 times.
 70 for (ch = 0; ch < chans; ch++)
1144 40 memcpy(cpe->ch[ch].coeffs, cpe->ch[ch].pcoeffs, sizeof(cpe->ch[ch].coeffs));
1145 }
1146 }
1147 1427 its++;
1148 }
1149 } else {
1150 break;
1151 }
1152 } while (1);
1153
1154
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 3805 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 3805 if (s->options.ltp && s->coder->ltp_insert_new_frame)
1155 ✗ s->coder->ltp_insert_new_frame(s);
1156
1157 3805 put_bits(&s->pb, 3, TYPE_END);
1158 3805 flush_put_bits(&s->pb);
1159
1160 3805 s->last_frame_pb_count = put_bits_count(&s->pb);
1161 3805 avpkt->size = put_bytes_output(&s->pb);
1162
1163 3805 s->lambda_sum += s->lambda;
1164 3805 s->lambda_count++;
1165
1166 3805 ff_af_queue_remove(&s->afq, avctx->frame_size, &avpkt->pts,
1167 &avpkt->duration);
1168
1169 3805 *got_packet_ptr = 1;
1170 3805 return 0;
1171 }
1172
1173 11 static av_cold int aac_encode_end(AVCodecContext *avctx)
1174 {
1175 11 AACEncContext *s = avctx->priv_data;
1176
1177
1/2
✓ Branch 0 taken 11 times.
✗ Branch 1 not taken.
 11 av_log(avctx, AV_LOG_INFO, "Qavg: %.3f\n", s->lambda_count ? s->lambda_sum / s->lambda_count : NAN);
1178
1179 11 av_tx_uninit(&s->mdct1024);
1180 11 av_tx_uninit(&s->mdct128);
1181 11 ff_psy_end(&s->psy);
1182 11 ff_lpc_end(&s->lpc);
1183
1/2
✓ Branch 0 taken 11 times.
✗ Branch 1 not taken.
 11 if (s->psypp)
1184 11 ff_psy_preprocess_end(s->psypp);
1185 11 av_freep(&s->buffer.samples);
1186 11 av_freep(&s->cpe);
1187 11 av_freep(&s->fdsp);
1188 11 ff_af_queue_close(&s->afq);
1189 11 return 0;
1190 }
1191
1192 11 static av_cold int dsp_init(AVCodecContext *avctx, AACEncContext *s)
1193 {
1194 11 int ret = 0;
1195 11 float scale = 32768.0f;
1196
1197 11 s->fdsp = avpriv_float_dsp_alloc(avctx->flags & AV_CODEC_FLAG_BITEXACT);
1198
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
 11 if (!s->fdsp)
1199 ✗ return AVERROR(ENOMEM);
1200
1201 // window init
1202 11 ff_aac_float_common_init();
1203
1204
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
 11 if ((ret = av_tx_init(&s->mdct1024, &s->mdct1024_fn, AV_TX_FLOAT_MDCT, 0,
1205 1024, &scale, 0)) < 0)
1206 ✗ return ret;
1207
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
 11 if ((ret = av_tx_init(&s->mdct128, &s->mdct128_fn, AV_TX_FLOAT_MDCT, 0,
1208 128, &scale, 0)) < 0)
1209 ✗ return ret;
1210
1211 11 return 0;
1212 }
1213
1214 11 static av_cold int alloc_buffers(AVCodecContext *avctx, AACEncContext *s)
1215 {
1216 int ch;
1217
1/2
✓ Branch 1 taken 11 times.
✗ Branch 2 not taken.
 11 if (!FF_ALLOCZ_TYPED_ARRAY(s->buffer.samples, s->channels * 3 * 1024) ||
1218
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
 11 !FF_ALLOCZ_TYPED_ARRAY(s->cpe, s->chan_map[0]))
1219 ✗ return AVERROR(ENOMEM);
1220
1221
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 11 times.
 36 for(ch = 0; ch < s->channels; ch++)
1222 25 s->planar_samples[ch] = s->buffer.samples + 3 * 1024 * ch;
1223
1224 11 return 0;
1225 }
1226
1227 11 static av_cold int aac_encode_init(AVCodecContext *avctx)
1228 {
1229 11 AACEncContext *s = avctx->priv_data;
1230 11 int i, ret = 0;
1231 const uint8_t *sizes[2];
1232 uint8_t grouping[AAC_MAX_CHANNELS];
1233 int lengths[2];
1234
1235 /* Constants */
1236 11 s->last_frame_pb_count = 0;
1237 11 avctx->frame_size = 1024;
1238 11 avctx->initial_padding = 1024;
1239
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
 11 s->lambda = avctx->global_quality > 0 ? avctx->global_quality : 120;
1240
1241 /* Channel map and unspecified bitrate guessing */
1242 11 s->channels = avctx->ch_layout.nb_channels;
1243
1244 11 s->needs_pce = 1;
1245
1/2
✓ Branch 0 taken 25 times.
✗ Branch 1 not taken.
 25 for (i = 0; i < FF_ARRAY_ELEMS(aac_normal_chan_layouts); i++) {
1246
2/2
✓ Branch 1 taken 11 times.
✓ Branch 2 taken 14 times.
 25 if (!av_channel_layout_compare(&avctx->ch_layout, &aac_normal_chan_layouts[i])) {
1247 11 s->needs_pce = s->options.pce;
1248 11 break;
1249 }
1250 }
1251
1252
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
 11 if (s->needs_pce) {
1253 char buf[64];
1254 ✗ for (i = 0; i < FF_ARRAY_ELEMS(aac_pce_configs); i++)
1255 ✗ if (!av_channel_layout_compare(&avctx->ch_layout, &aac_pce_configs[i].layout))
1256 ✗ break;
1257 ✗ av_channel_layout_describe(&avctx->ch_layout, buf, sizeof(buf));
1258 ✗ if (i == FF_ARRAY_ELEMS(aac_pce_configs)) {
1259 ✗ av_log(avctx, AV_LOG_ERROR, "Unsupported channel layout \"%s\"\n", buf);
1260 ✗ return AVERROR(EINVAL);
1261 }
1262 ✗ av_log(avctx, AV_LOG_INFO, "Using a PCE to encode channel layout \"%s\"\n", buf);
1263 ✗ s->pce = aac_pce_configs[i];
1264 ✗ s->reorder_map = s->pce.reorder_map;
1265 ✗ s->chan_map = s->pce.config_map;
1266 } else {
1267 11 s->reorder_map = aac_chan_maps[s->channels - 1];
1268 11 s->chan_map = aac_chan_configs[s->channels - 1];
1269 }
1270
1271
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 8 times.
 11 if (!avctx->bit_rate) {
1272
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 3 times.
 9 for (i = 1; i <= s->chan_map[0]; i++) {
1273
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 3 times.
 9 avctx->bit_rate += s->chan_map[i] == TYPE_CPE ? 128000 : /* Pair */
1274
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 2 times.
 3 s->chan_map[i] == TYPE_LFE ? 16000 : /* LFE */
1275 69000 ; /* SCE */
1276 }
1277 }
1278
1279 /* Samplerate */
1280
1/2
✓ Branch 0 taken 54 times.
✗ Branch 1 not taken.
 54 for (i = 0; i < 16; i++)
1281
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 43 times.
 54 if (avctx->sample_rate == ff_mpeg4audio_sample_rates[i])
1282 11 break;
1283 11 s->samplerate_index = i;
1284
3/6
✓ Branch 0 taken 11 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 11 times.
 11 ERROR_IF(s->samplerate_index == 16 ||
1285 s->samplerate_index >= ff_aac_swb_size_1024_len ||
1286 s->samplerate_index >= ff_aac_swb_size_128_len,
1287 "Unsupported sample rate %d\n", avctx->sample_rate);
1288
1289 /* Bitrate limiting */
1290
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
 11 WARN_IF(1024.0 * avctx->bit_rate / avctx->sample_rate > 6144 * s->channels,
1291 "Too many bits %f > %d per frame requested, clamping to max\n",
1292 1024.0 * avctx->bit_rate / avctx->sample_rate,
1293 6144 * s->channels);
1294
1/2
✓ Branch 0 taken 11 times.
✗ Branch 1 not taken.
 11 avctx->bit_rate = (int64_t)FFMIN(6144 * s->channels / 1024.0 * avctx->sample_rate,
1295 avctx->bit_rate);
1296
1297 /* Profile and option setting */
1298
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 10 times.
 11 avctx->profile = avctx->profile == FF_PROFILE_UNKNOWN ? FF_PROFILE_AAC_LOW :
1299 avctx->profile;
1300
1/2
✓ Branch 0 taken 21 times.
✗ Branch 1 not taken.
 21 for (i = 0; i < FF_ARRAY_ELEMS(aacenc_profiles); i++)
1301
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 10 times.
 21 if (avctx->profile == aacenc_profiles[i])
1302 11 break;
1303
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
 11 if (avctx->profile == FF_PROFILE_MPEG2_AAC_LOW) {
1304 ✗ avctx->profile = FF_PROFILE_AAC_LOW;
1305 ✗ ERROR_IF(s->options.pred,
1306 "Main prediction unavailable in the \"mpeg2_aac_low\" profile\n");
1307 ✗ ERROR_IF(s->options.ltp,
1308 "LTP prediction unavailable in the \"mpeg2_aac_low\" profile\n");
1309 ✗ WARN_IF(s->options.pns,
1310 "PNS unavailable in the \"mpeg2_aac_low\" profile, turning off\n");
1311 ✗ s->options.pns = 0;
1312
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
 11 } else if (avctx->profile == FF_PROFILE_AAC_LTP) {
1313 ✗ s->options.ltp = 1;
1314 ✗ ERROR_IF(s->options.pred,
1315 "Main prediction unavailable in the \"aac_ltp\" profile\n");
1316
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 10 times.
 11 } else if (avctx->profile == FF_PROFILE_AAC_MAIN) {
1317 1 s->options.pred = 1;
1318
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 ERROR_IF(s->options.ltp,
1319 "LTP prediction unavailable in the \"aac_main\" profile\n");
1320
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
 10 } else if (s->options.ltp) {
1321 ✗ avctx->profile = FF_PROFILE_AAC_LTP;
1322 ✗ WARN_IF(1,
1323 "Chainging profile to \"aac_ltp\"\n");
1324 ✗ ERROR_IF(s->options.pred,
1325 "Main prediction unavailable in the \"aac_ltp\" profile\n");
1326
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
 10 } else if (s->options.pred) {
1327 ✗ avctx->profile = FF_PROFILE_AAC_MAIN;
1328 ✗ WARN_IF(1,
1329 "Chainging profile to \"aac_main\"\n");
1330 ✗ ERROR_IF(s->options.ltp,
1331 "LTP prediction unavailable in the \"aac_main\" profile\n");
1332 }
1333 11 s->profile = avctx->profile;
1334
1335 /* Coder limitations */
1336 11 s->coder = &ff_aac_coders[s->options.coder];
1337
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
 11 if (s->options.coder == AAC_CODER_ANMR) {
1338 ✗ ERROR_IF(avctx->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL,
1339 "The ANMR coder is considered experimental, add -strict -2 to enable!\n");
1340 ✗ s->options.intensity_stereo = 0;
1341 ✗ s->options.pns = 0;
1342 }
1343
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 11 ERROR_IF(s->options.ltp && avctx->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL,
1344 "The LPT profile requires experimental compliance, add -strict -2 to enable!\n");
1345
1346 /* M/S introduces horrible artifacts with multichannel files, this is temporary */
1347
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 10 times.
 11 if (s->channels > 3)
1348 1 s->options.mid_side = 0;
1349
1350
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
 11 if ((ret = dsp_init(avctx, s)) < 0)
1351 ✗ return ret;
1352
1353
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
 11 if ((ret = alloc_buffers(avctx, s)) < 0)
1354 ✗ return ret;
1355
1356
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
 11 if ((ret = put_audio_specific_config(avctx)))
1357 ✗ return ret;
1358
1359 11 sizes[0] = ff_aac_swb_size_1024[s->samplerate_index];
1360 11 sizes[1] = ff_aac_swb_size_128[s->samplerate_index];
1361 11 lengths[0] = ff_aac_num_swb_1024[s->samplerate_index];
1362 11 lengths[1] = ff_aac_num_swb_128[s->samplerate_index];
1363
2/2
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 11 times.
 25 for (i = 0; i < s->chan_map[0]; i++)
1364 14 grouping[i] = s->chan_map[i + 1] == TYPE_CPE;
1365
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
 11 if ((ret = ff_psy_init(&s->psy, avctx, 2, sizes, lengths,
1366 11 s->chan_map[0], grouping)) < 0)
1367 ✗ return ret;
1368 11 s->psypp = ff_psy_preprocess_init(avctx);
1369 11 ff_lpc_init(&s->lpc, 2*avctx->frame_size, TNS_MAX_ORDER, FF_LPC_TYPE_LEVINSON);
1370 11 s->random_state = 0x1f2e3d4c;
1371
1372 11 s->abs_pow34 = abs_pow34_v;
1373 11 s->quant_bands = quantize_bands;
1374
1375 #if ARCH_X86
1376 11 ff_aac_dsp_init_x86(s);
1377 #endif
1378
1379 #if HAVE_MIPSDSP
1380 ff_aac_coder_init_mips(s);
1381 #endif
1382
1383 11 ff_af_queue_init(avctx, &s->afq);
1384 11 ff_aac_tableinit();
1385
1386 11 return 0;
1387 }
1388
1389 #define AACENC_FLAGS AV_OPT_FLAG_ENCODING_PARAM | AV_OPT_FLAG_AUDIO_PARAM
1390 static const AVOption aacenc_options[] = {
1391 {"aac_coder", "Coding algorithm", offsetof(AACEncContext, options.coder), AV_OPT_TYPE_INT, {.i64 = AAC_CODER_TWOLOOP}, 0, AAC_CODER_NB-1, AACENC_FLAGS, "coder"},
1392 {"anmr", "ANMR method", 0, AV_OPT_TYPE_CONST, {.i64 = AAC_CODER_ANMR}, INT_MIN, INT_MAX, AACENC_FLAGS, "coder"},
1393 {"twoloop", "Two loop searching method", 0, AV_OPT_TYPE_CONST, {.i64 = AAC_CODER_TWOLOOP}, INT_MIN, INT_MAX, AACENC_FLAGS, "coder"},
1394 {"fast", "Default fast search", 0, AV_OPT_TYPE_CONST, {.i64 = AAC_CODER_FAST}, INT_MIN, INT_MAX, AACENC_FLAGS, "coder"},
1395 {"aac_ms", "Force M/S stereo coding", offsetof(AACEncContext, options.mid_side), AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1, AACENC_FLAGS},
1396 {"aac_is", "Intensity stereo coding", offsetof(AACEncContext, options.intensity_stereo), AV_OPT_TYPE_BOOL, {.i64 = 1}, -1, 1, AACENC_FLAGS},
1397 {"aac_pns", "Perceptual noise substitution", offsetof(AACEncContext, options.pns), AV_OPT_TYPE_BOOL, {.i64 = 1}, -1, 1, AACENC_FLAGS},
1398 {"aac_tns", "Temporal noise shaping", offsetof(AACEncContext, options.tns), AV_OPT_TYPE_BOOL, {.i64 = 1}, -1, 1, AACENC_FLAGS},
1399 {"aac_ltp", "Long term prediction", offsetof(AACEncContext, options.ltp), AV_OPT_TYPE_BOOL, {.i64 = 0}, -1, 1, AACENC_FLAGS},
1400 {"aac_pred", "AAC-Main prediction", offsetof(AACEncContext, options.pred), AV_OPT_TYPE_BOOL, {.i64 = 0}, -1, 1, AACENC_FLAGS},
1401 {"aac_pce", "Forces the use of PCEs", offsetof(AACEncContext, options.pce), AV_OPT_TYPE_BOOL, {.i64 = 0}, -1, 1, AACENC_FLAGS},
1402 FF_AAC_PROFILE_OPTS
1403 {NULL}
1404 };
1405
1406 static const AVClass aacenc_class = {
1407 .class_name = "AAC encoder",
1408 .item_name = av_default_item_name,
1409 .option = aacenc_options,
1410 .version = LIBAVUTIL_VERSION_INT,
1411 };
1412
1413 static const FFCodecDefault aac_encode_defaults[] = {
1414 { "b", "0" },
1415 { NULL }
1416 };
1417
1418 const FFCodec ff_aac_encoder = {
1419 .p.name = "aac",
1420 CODEC_LONG_NAME("AAC (Advanced Audio Coding)"),
1421 .p.type = AVMEDIA_TYPE_AUDIO,
1422 .p.id = AV_CODEC_ID_AAC,
1423 .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_DELAY |
1424 AV_CODEC_CAP_SMALL_LAST_FRAME,
1425 .priv_data_size = sizeof(AACEncContext),
1426 .init = aac_encode_init,
1427 FF_CODEC_ENCODE_CB(aac_encode_frame),
1428 .close = aac_encode_end,
1429 .defaults = aac_encode_defaults,
1430 .p.supported_samplerates = ff_mpeg4audio_sample_rates,
1431 .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
1432 .p.sample_fmts = (const enum AVSampleFormat[]){ AV_SAMPLE_FMT_FLTP,
1433 AV_SAMPLE_FMT_NONE },
1434 .p.priv_class = &aacenc_class,
1435 };
1436