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 strict bit-rate control is demanded */ | ||
1110 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 5232 times.
|
5232 | if (avctx->bit_rate_tolerance == 0) { |
1111 | ✗ | if (rate_bits < frame_bits) { | |
1112 | ✗ | float ratio = ((float)rate_bits) / frame_bits; | |
1113 | ✗ | s->lambda *= FFMIN(0.9f, ratio); | |
1114 | ✗ | continue; | |
1115 | } | ||
1116 | /* reset lambda when solution is found */ | ||
1117 | ✗ | s->lambda = avctx->global_quality > 0 ? avctx->global_quality : 120; | |
1118 | ✗ | break; | |
1119 | } | ||
1120 | |||
1121 | /* When using ABR, be strict (but only for increasing) */ | ||
1122 | 5232 | too_few_bits = too_few_bits - too_few_bits/8; | |
1123 | 5232 | too_many_bits = too_many_bits + too_many_bits/2; | |
1124 | |||
1125 |
2/2✓ Branch 0 taken 1427 times.
✓ Branch 1 taken 3805 times.
|
5232 | if ( its == 0 /* for steady-state Q-scale tracking */ |
1126 |
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)) |
1127 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 289 times.
|
289 | || frame_bits >= 6144 * s->channels - 3 ) |
1128 | { | ||
1129 | 4943 | float ratio = ((float)rate_bits) / frame_bits; | |
1130 | |||
1131 |
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) { |
1132 | /* | ||
1133 | * This path is for steady-state Q-scale tracking | ||
1134 | * When frame bits fall within the stable range, we still need to adjust | ||
1135 | * lambda to maintain it like so in a stable fashion (large jumps in lambda | ||
1136 | * create artifacts and should be avoided), but slowly | ||
1137 | */ | ||
1138 | 2925 | ratio = sqrtf(sqrtf(ratio)); | |
1139 | 2925 | ratio = av_clipf(ratio, 0.9f, 1.1f); | |
1140 | } else { | ||
1141 | /* Not so fast though */ | ||
1142 | 2018 | ratio = sqrtf(ratio); | |
1143 | } | ||
1144 | 4943 | s->lambda = av_clipf(s->lambda * ratio, FLT_EPSILON, 65536.f); | |
1145 | |||
1146 | /* Keep iterating if we must reduce and lambda is in the sky */ | ||
1147 |
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) { |
1148 | break; | ||
1149 | } else { | ||
1150 |
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) { |
1151 |
2/2✓ Branch 0 taken 30 times.
✓ Branch 1 taken 15 times.
|
45 | for (i = 0; i < s->chan_map[0]; i++) { |
1152 | // Must restore coeffs | ||
1153 |
2/2✓ Branch 0 taken 10 times.
✓ Branch 1 taken 20 times.
|
30 | chans = tag == TYPE_CPE ? 2 : 1; |
1154 | 30 | cpe = &s->cpe[i]; | |
1155 |
2/2✓ Branch 0 taken 40 times.
✓ Branch 1 taken 30 times.
|
70 | for (ch = 0; ch < chans; ch++) |
1156 | 40 | memcpy(cpe->ch[ch].coeffs, cpe->ch[ch].pcoeffs, sizeof(cpe->ch[ch].coeffs)); | |
1157 | } | ||
1158 | } | ||
1159 | 1427 | its++; | |
1160 | } | ||
1161 | } else { | ||
1162 | break; | ||
1163 | } | ||
1164 | } while (1); | ||
1165 | |||
1166 |
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) |
1167 | ✗ | s->coder->ltp_insert_new_frame(s); | |
1168 | |||
1169 | 3805 | put_bits(&s->pb, 3, TYPE_END); | |
1170 | 3805 | flush_put_bits(&s->pb); | |
1171 | |||
1172 | 3805 | s->last_frame_pb_count = put_bits_count(&s->pb); | |
1173 | 3805 | avpkt->size = put_bytes_output(&s->pb); | |
1174 | |||
1175 | 3805 | s->lambda_sum += s->lambda; | |
1176 | 3805 | s->lambda_count++; | |
1177 | |||
1178 | 3805 | ff_af_queue_remove(&s->afq, avctx->frame_size, &avpkt->pts, | |
1179 | &avpkt->duration); | ||
1180 | |||
1181 | 3805 | *got_packet_ptr = 1; | |
1182 | 3805 | return 0; | |
1183 | } | ||
1184 | |||
1185 | 11 | static av_cold int aac_encode_end(AVCodecContext *avctx) | |
1186 | { | ||
1187 | 11 | AACEncContext *s = avctx->priv_data; | |
1188 | |||
1189 |
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); |
1190 | |||
1191 | 11 | av_tx_uninit(&s->mdct1024); | |
1192 | 11 | av_tx_uninit(&s->mdct128); | |
1193 | 11 | ff_psy_end(&s->psy); | |
1194 | 11 | ff_lpc_end(&s->lpc); | |
1195 |
1/2✓ Branch 0 taken 11 times.
✗ Branch 1 not taken.
|
11 | if (s->psypp) |
1196 | 11 | ff_psy_preprocess_end(s->psypp); | |
1197 | 11 | av_freep(&s->buffer.samples); | |
1198 | 11 | av_freep(&s->cpe); | |
1199 | 11 | av_freep(&s->fdsp); | |
1200 | 11 | ff_af_queue_close(&s->afq); | |
1201 | 11 | return 0; | |
1202 | } | ||
1203 | |||
1204 | 11 | static av_cold int dsp_init(AVCodecContext *avctx, AACEncContext *s) | |
1205 | { | ||
1206 | 11 | int ret = 0; | |
1207 | 11 | float scale = 32768.0f; | |
1208 | |||
1209 | 11 | s->fdsp = avpriv_float_dsp_alloc(avctx->flags & AV_CODEC_FLAG_BITEXACT); | |
1210 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
|
11 | if (!s->fdsp) |
1211 | ✗ | return AVERROR(ENOMEM); | |
1212 | |||
1213 | // window init | ||
1214 | 11 | ff_aac_float_common_init(); | |
1215 | |||
1216 |
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, |
1217 | 1024, &scale, 0)) < 0) | ||
1218 | ✗ | return ret; | |
1219 |
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, |
1220 | 128, &scale, 0)) < 0) | ||
1221 | ✗ | return ret; | |
1222 | |||
1223 | 11 | return 0; | |
1224 | } | ||
1225 | |||
1226 | 11 | static av_cold int alloc_buffers(AVCodecContext *avctx, AACEncContext *s) | |
1227 | { | ||
1228 | int ch; | ||
1229 |
1/2✓ Branch 1 taken 11 times.
✗ Branch 2 not taken.
|
11 | if (!FF_ALLOCZ_TYPED_ARRAY(s->buffer.samples, s->channels * 3 * 1024) || |
1230 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
|
11 | !FF_ALLOCZ_TYPED_ARRAY(s->cpe, s->chan_map[0])) |
1231 | ✗ | return AVERROR(ENOMEM); | |
1232 | |||
1233 |
2/2✓ Branch 0 taken 25 times.
✓ Branch 1 taken 11 times.
|
36 | for(ch = 0; ch < s->channels; ch++) |
1234 | 25 | s->planar_samples[ch] = s->buffer.samples + 3 * 1024 * ch; | |
1235 | |||
1236 | 11 | return 0; | |
1237 | } | ||
1238 | |||
1239 | 11 | static av_cold int aac_encode_init(AVCodecContext *avctx) | |
1240 | { | ||
1241 | 11 | AACEncContext *s = avctx->priv_data; | |
1242 | 11 | int i, ret = 0; | |
1243 | const uint8_t *sizes[2]; | ||
1244 | uint8_t grouping[AAC_MAX_CHANNELS]; | ||
1245 | int lengths[2]; | ||
1246 | |||
1247 | /* Constants */ | ||
1248 | 11 | s->last_frame_pb_count = 0; | |
1249 | 11 | avctx->frame_size = 1024; | |
1250 | 11 | avctx->initial_padding = 1024; | |
1251 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
|
11 | s->lambda = avctx->global_quality > 0 ? avctx->global_quality : 120; |
1252 | |||
1253 | /* Channel map and unspecified bitrate guessing */ | ||
1254 | 11 | s->channels = avctx->ch_layout.nb_channels; | |
1255 | |||
1256 | 11 | s->needs_pce = 1; | |
1257 |
1/2✓ Branch 0 taken 25 times.
✗ Branch 1 not taken.
|
25 | for (i = 0; i < FF_ARRAY_ELEMS(aac_normal_chan_layouts); i++) { |
1258 |
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])) { |
1259 | 11 | s->needs_pce = s->options.pce; | |
1260 | 11 | break; | |
1261 | } | ||
1262 | } | ||
1263 | |||
1264 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
|
11 | if (s->needs_pce) { |
1265 | char buf[64]; | ||
1266 | ✗ | for (i = 0; i < FF_ARRAY_ELEMS(aac_pce_configs); i++) | |
1267 | ✗ | if (!av_channel_layout_compare(&avctx->ch_layout, &aac_pce_configs[i].layout)) | |
1268 | ✗ | break; | |
1269 | ✗ | av_channel_layout_describe(&avctx->ch_layout, buf, sizeof(buf)); | |
1270 | ✗ | if (i == FF_ARRAY_ELEMS(aac_pce_configs)) { | |
1271 | ✗ | av_log(avctx, AV_LOG_ERROR, "Unsupported channel layout \"%s\"\n", buf); | |
1272 | ✗ | return AVERROR(EINVAL); | |
1273 | } | ||
1274 | ✗ | av_log(avctx, AV_LOG_INFO, "Using a PCE to encode channel layout \"%s\"\n", buf); | |
1275 | ✗ | s->pce = aac_pce_configs[i]; | |
1276 | ✗ | s->reorder_map = s->pce.reorder_map; | |
1277 | ✗ | s->chan_map = s->pce.config_map; | |
1278 | } else { | ||
1279 | 11 | s->reorder_map = aac_chan_maps[s->channels - 1]; | |
1280 | 11 | s->chan_map = aac_chan_configs[s->channels - 1]; | |
1281 | } | ||
1282 | |||
1283 |
2/2✓ Branch 0 taken 3 times.
✓ Branch 1 taken 8 times.
|
11 | if (!avctx->bit_rate) { |
1284 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 3 times.
|
9 | for (i = 1; i <= s->chan_map[0]; i++) { |
1285 |
2/2✓ Branch 0 taken 3 times.
✓ Branch 1 taken 3 times.
|
9 | avctx->bit_rate += s->chan_map[i] == TYPE_CPE ? 128000 : /* Pair */ |
1286 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 2 times.
|
3 | s->chan_map[i] == TYPE_LFE ? 16000 : /* LFE */ |
1287 | 69000 ; /* SCE */ | ||
1288 | } | ||
1289 | } | ||
1290 | |||
1291 | /* Samplerate */ | ||
1292 |
1/2✓ Branch 0 taken 54 times.
✗ Branch 1 not taken.
|
54 | for (i = 0; i < 16; i++) |
1293 |
2/2✓ Branch 0 taken 11 times.
✓ Branch 1 taken 43 times.
|
54 | if (avctx->sample_rate == ff_mpeg4audio_sample_rates[i]) |
1294 | 11 | break; | |
1295 | 11 | s->samplerate_index = i; | |
1296 |
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 || |
1297 | s->samplerate_index >= ff_aac_swb_size_1024_len || | ||
1298 | s->samplerate_index >= ff_aac_swb_size_128_len, | ||
1299 | "Unsupported sample rate %d\n", avctx->sample_rate); | ||
1300 | |||
1301 | /* Bitrate limiting */ | ||
1302 |
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, |
1303 | "Too many bits %f > %d per frame requested, clamping to max\n", | ||
1304 | 1024.0 * avctx->bit_rate / avctx->sample_rate, | ||
1305 | 6144 * s->channels); | ||
1306 |
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, |
1307 | avctx->bit_rate); | ||
1308 | |||
1309 | /* Profile and option setting */ | ||
1310 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 10 times.
|
11 | avctx->profile = avctx->profile == AV_PROFILE_UNKNOWN ? AV_PROFILE_AAC_LOW : |
1311 | avctx->profile; | ||
1312 |
1/2✓ Branch 0 taken 21 times.
✗ Branch 1 not taken.
|
21 | for (i = 0; i < FF_ARRAY_ELEMS(aacenc_profiles); i++) |
1313 |
2/2✓ Branch 0 taken 11 times.
✓ Branch 1 taken 10 times.
|
21 | if (avctx->profile == aacenc_profiles[i]) |
1314 | 11 | break; | |
1315 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
|
11 | if (avctx->profile == AV_PROFILE_MPEG2_AAC_LOW) { |
1316 | ✗ | avctx->profile = AV_PROFILE_AAC_LOW; | |
1317 | ✗ | ERROR_IF(s->options.pred, | |
1318 | "Main prediction unavailable in the \"mpeg2_aac_low\" profile\n"); | ||
1319 | ✗ | ERROR_IF(s->options.ltp, | |
1320 | "LTP prediction unavailable in the \"mpeg2_aac_low\" profile\n"); | ||
1321 | ✗ | WARN_IF(s->options.pns, | |
1322 | "PNS unavailable in the \"mpeg2_aac_low\" profile, turning off\n"); | ||
1323 | ✗ | s->options.pns = 0; | |
1324 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
|
11 | } else if (avctx->profile == AV_PROFILE_AAC_LTP) { |
1325 | ✗ | s->options.ltp = 1; | |
1326 | ✗ | ERROR_IF(s->options.pred, | |
1327 | "Main prediction unavailable in the \"aac_ltp\" profile\n"); | ||
1328 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 10 times.
|
11 | } else if (avctx->profile == AV_PROFILE_AAC_MAIN) { |
1329 | 1 | s->options.pred = 1; | |
1330 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | ERROR_IF(s->options.ltp, |
1331 | "LTP prediction unavailable in the \"aac_main\" profile\n"); | ||
1332 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
|
10 | } else if (s->options.ltp) { |
1333 | ✗ | avctx->profile = AV_PROFILE_AAC_LTP; | |
1334 | ✗ | WARN_IF(1, | |
1335 | "Chainging profile to \"aac_ltp\"\n"); | ||
1336 | ✗ | ERROR_IF(s->options.pred, | |
1337 | "Main prediction unavailable in the \"aac_ltp\" profile\n"); | ||
1338 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
|
10 | } else if (s->options.pred) { |
1339 | ✗ | avctx->profile = AV_PROFILE_AAC_MAIN; | |
1340 | ✗ | WARN_IF(1, | |
1341 | "Chainging profile to \"aac_main\"\n"); | ||
1342 | ✗ | ERROR_IF(s->options.ltp, | |
1343 | "LTP prediction unavailable in the \"aac_main\" profile\n"); | ||
1344 | } | ||
1345 | 11 | s->profile = avctx->profile; | |
1346 | |||
1347 | /* Coder limitations */ | ||
1348 | 11 | s->coder = &ff_aac_coders[s->options.coder]; | |
1349 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
|
11 | if (s->options.coder == AAC_CODER_ANMR) { |
1350 | ✗ | ERROR_IF(avctx->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL, | |
1351 | "The ANMR coder is considered experimental, add -strict -2 to enable!\n"); | ||
1352 | ✗ | s->options.intensity_stereo = 0; | |
1353 | ✗ | s->options.pns = 0; | |
1354 | } | ||
1355 |
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, |
1356 | "The LPT profile requires experimental compliance, add -strict -2 to enable!\n"); | ||
1357 | |||
1358 | /* M/S introduces horrible artifacts with multichannel files, this is temporary */ | ||
1359 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 10 times.
|
11 | if (s->channels > 3) |
1360 | 1 | s->options.mid_side = 0; | |
1361 | |||
1362 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
|
11 | if ((ret = dsp_init(avctx, s)) < 0) |
1363 | ✗ | return ret; | |
1364 | |||
1365 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
|
11 | if ((ret = alloc_buffers(avctx, s)) < 0) |
1366 | ✗ | return ret; | |
1367 | |||
1368 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
|
11 | if ((ret = put_audio_specific_config(avctx))) |
1369 | ✗ | return ret; | |
1370 | |||
1371 | 11 | sizes[0] = ff_aac_swb_size_1024[s->samplerate_index]; | |
1372 | 11 | sizes[1] = ff_aac_swb_size_128[s->samplerate_index]; | |
1373 | 11 | lengths[0] = ff_aac_num_swb_1024[s->samplerate_index]; | |
1374 | 11 | lengths[1] = ff_aac_num_swb_128[s->samplerate_index]; | |
1375 |
2/2✓ Branch 0 taken 14 times.
✓ Branch 1 taken 11 times.
|
25 | for (i = 0; i < s->chan_map[0]; i++) |
1376 | 14 | grouping[i] = s->chan_map[i + 1] == TYPE_CPE; | |
1377 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
|
11 | if ((ret = ff_psy_init(&s->psy, avctx, 2, sizes, lengths, |
1378 | 11 | s->chan_map[0], grouping)) < 0) | |
1379 | ✗ | return ret; | |
1380 | 11 | s->psypp = ff_psy_preprocess_init(avctx); | |
1381 | 11 | ff_lpc_init(&s->lpc, 2*avctx->frame_size, TNS_MAX_ORDER, FF_LPC_TYPE_LEVINSON); | |
1382 | 11 | s->random_state = 0x1f2e3d4c; | |
1383 | |||
1384 | 11 | s->abs_pow34 = abs_pow34_v; | |
1385 | 11 | s->quant_bands = quantize_bands; | |
1386 | |||
1387 | #if ARCH_X86 | ||
1388 | 11 | ff_aac_dsp_init_x86(s); | |
1389 | #endif | ||
1390 | |||
1391 | #if HAVE_MIPSDSP | ||
1392 | ff_aac_coder_init_mips(s); | ||
1393 | #endif | ||
1394 | |||
1395 | 11 | ff_af_queue_init(avctx, &s->afq); | |
1396 | 11 | ff_aac_tableinit(); | |
1397 | |||
1398 | 11 | return 0; | |
1399 | } | ||
1400 | |||
1401 | #define AACENC_FLAGS AV_OPT_FLAG_ENCODING_PARAM | AV_OPT_FLAG_AUDIO_PARAM | ||
1402 | static const AVOption aacenc_options[] = { | ||
1403 | {"aac_coder", "Coding algorithm", offsetof(AACEncContext, options.coder), AV_OPT_TYPE_INT, {.i64 = AAC_CODER_TWOLOOP}, 0, AAC_CODER_NB-1, AACENC_FLAGS, "coder"}, | ||
1404 | {"anmr", "ANMR method", 0, AV_OPT_TYPE_CONST, {.i64 = AAC_CODER_ANMR}, INT_MIN, INT_MAX, AACENC_FLAGS, "coder"}, | ||
1405 | {"twoloop", "Two loop searching method", 0, AV_OPT_TYPE_CONST, {.i64 = AAC_CODER_TWOLOOP}, INT_MIN, INT_MAX, AACENC_FLAGS, "coder"}, | ||
1406 | {"fast", "Default fast search", 0, AV_OPT_TYPE_CONST, {.i64 = AAC_CODER_FAST}, INT_MIN, INT_MAX, AACENC_FLAGS, "coder"}, | ||
1407 | {"aac_ms", "Force M/S stereo coding", offsetof(AACEncContext, options.mid_side), AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1, AACENC_FLAGS}, | ||
1408 | {"aac_is", "Intensity stereo coding", offsetof(AACEncContext, options.intensity_stereo), AV_OPT_TYPE_BOOL, {.i64 = 1}, -1, 1, AACENC_FLAGS}, | ||
1409 | {"aac_pns", "Perceptual noise substitution", offsetof(AACEncContext, options.pns), AV_OPT_TYPE_BOOL, {.i64 = 1}, -1, 1, AACENC_FLAGS}, | ||
1410 | {"aac_tns", "Temporal noise shaping", offsetof(AACEncContext, options.tns), AV_OPT_TYPE_BOOL, {.i64 = 1}, -1, 1, AACENC_FLAGS}, | ||
1411 | {"aac_ltp", "Long term prediction", offsetof(AACEncContext, options.ltp), AV_OPT_TYPE_BOOL, {.i64 = 0}, -1, 1, AACENC_FLAGS}, | ||
1412 | {"aac_pred", "AAC-Main prediction", offsetof(AACEncContext, options.pred), AV_OPT_TYPE_BOOL, {.i64 = 0}, -1, 1, AACENC_FLAGS}, | ||
1413 | {"aac_pce", "Forces the use of PCEs", offsetof(AACEncContext, options.pce), AV_OPT_TYPE_BOOL, {.i64 = 0}, -1, 1, AACENC_FLAGS}, | ||
1414 | FF_AAC_PROFILE_OPTS | ||
1415 | {NULL} | ||
1416 | }; | ||
1417 | |||
1418 | static const AVClass aacenc_class = { | ||
1419 | .class_name = "AAC encoder", | ||
1420 | .item_name = av_default_item_name, | ||
1421 | .option = aacenc_options, | ||
1422 | .version = LIBAVUTIL_VERSION_INT, | ||
1423 | }; | ||
1424 | |||
1425 | static const FFCodecDefault aac_encode_defaults[] = { | ||
1426 | { "b", "0" }, | ||
1427 | { NULL } | ||
1428 | }; | ||
1429 | |||
1430 | const FFCodec ff_aac_encoder = { | ||
1431 | .p.name = "aac", | ||
1432 | CODEC_LONG_NAME("AAC (Advanced Audio Coding)"), | ||
1433 | .p.type = AVMEDIA_TYPE_AUDIO, | ||
1434 | .p.id = AV_CODEC_ID_AAC, | ||
1435 | .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_DELAY | | ||
1436 | AV_CODEC_CAP_SMALL_LAST_FRAME, | ||
1437 | .priv_data_size = sizeof(AACEncContext), | ||
1438 | .init = aac_encode_init, | ||
1439 | FF_CODEC_ENCODE_CB(aac_encode_frame), | ||
1440 | .close = aac_encode_end, | ||
1441 | .defaults = aac_encode_defaults, | ||
1442 | .p.supported_samplerates = ff_mpeg4audio_sample_rates, | ||
1443 | .caps_internal = FF_CODEC_CAP_INIT_CLEANUP, | ||
1444 | .p.sample_fmts = (const enum AVSampleFormat[]){ AV_SAMPLE_FMT_FLTP, | ||
1445 | AV_SAMPLE_FMT_NONE }, | ||
1446 | .p.priv_class = &aacenc_class, | ||
1447 | }; | ||
1448 |