FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/aaccoder.c
Date: 2025-08-07 17:35:02
Exec Total Coverage
Lines: 446 462 96.5%
Functions: 17 18 94.4%
Branches: 332 388 85.6%
Line Branch Exec Source
1 /*
2 * AAC coefficients encoder
3 * Copyright (C) 2008-2009 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 coefficients encoder
25 */
26
27 /***********************************
28 * TODOs:
29 * speedup quantizer selection
30 * add sane pulse detection
31 ***********************************/
32
33 #include "libavutil/libm.h" // brought forward to work around cygwin header breakage
34
35 #include <float.h>
36
37 #include "libavutil/mathematics.h"
38 #include "mathops.h"
39 #include "avcodec.h"
40 #include "put_bits.h"
41 #include "aac.h"
42 #include "aacenc.h"
43 #include "aactab.h"
44 #include "aacenctab.h"
45 #include "aacenc_utils.h"
46 #include "aacenc_quantization.h"
47
48 #include "aacenc_is.h"
49 #include "aacenc_tns.h"
50
51 #include "libavcodec/aaccoder_twoloop.h"
52
53 /* Parameter of f(x) = a*(lambda/100), defines the maximum fourier spread
54 * beyond which no PNS is used (since the SFBs contain tone rather than noise) */
55 #define NOISE_SPREAD_THRESHOLD 0.9f
56
57 /* Parameter of f(x) = a*(100/lambda), defines how much PNS is allowed to
58 * replace low energy non zero bands */
59 #define NOISE_LAMBDA_REPLACE 1.948f
60
61 #include "libavcodec/aaccoder_trellis.h"
62
63 typedef float (*quantize_and_encode_band_func)(struct AACEncContext *s, PutBitContext *pb,
64 const float *in, float *quant, const float *scaled,
65 int size, int scale_idx, int cb,
66 const float lambda, const float uplim,
67 int *bits, float *energy);
68
69 /**
70 * Calculate rate distortion cost for quantizing with given codebook
71 *
72 * @return quantization distortion
73 */
74 8458971 static av_always_inline float quantize_and_encode_band_cost_template(
75 struct AACEncContext *s,
76 PutBitContext *pb, const float *in, float *out,
77 const float *scaled, int size, int scale_idx,
78 int cb, const float lambda, const float uplim,
79 int *bits, float *energy, int BT_ZERO, int BT_UNSIGNED,
80 int BT_PAIR, int BT_ESC, int BT_NOISE, int BT_STEREO,
81 const float ROUNDING)
82 {
83 8458971 const int q_idx = POW_SF2_ZERO - scale_idx + SCALE_ONE_POS - SCALE_DIV_512;
84 8458971 const float Q = ff_aac_pow2sf_tab [q_idx];
85 8458971 const float Q34 = ff_aac_pow34sf_tab[q_idx];
86 8458971 const float IQ = ff_aac_pow2sf_tab [POW_SF2_ZERO + scale_idx - SCALE_ONE_POS + SCALE_DIV_512];
87 8458971 const float CLIPPED_ESCAPE = 165140.0f*IQ;
88 8458971 float cost = 0;
89 8458971 float qenergy = 0;
90
2/2
✓ Branch 0 taken 4196191 times.
✓ Branch 1 taken 4262780 times.
 8458971 const int dim = BT_PAIR ? 2 : 4;
91 8458971 int resbits = 0;
92 int off;
93
94
6/6
✓ Branch 0 taken 6926523 times.
✓ Branch 1 taken 1532448 times.
✓ Branch 2 taken 6912506 times.
✓ Branch 3 taken 14017 times.
✓ Branch 4 taken 17312 times.
✓ Branch 5 taken 6895194 times.
 8458971 if (BT_ZERO || BT_NOISE || BT_STEREO) {
95
2/2
✓ Branch 0 taken 35194620 times.
✓ Branch 1 taken 1563777 times.
 36758397 for (int i = 0; i < size; i++)
96 35194620 cost += in[i]*in[i];
97
2/2
✓ Branch 0 taken 1540374 times.
✓ Branch 1 taken 23403 times.
 1563777 if (bits)
98 1540374 *bits = 0;
99
2/2
✓ Branch 0 taken 1512532 times.
✓ Branch 1 taken 51245 times.
 1563777 if (energy)
100 1512532 *energy = qenergy;
101
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1563777 times.
 1563777 if (out) {
102 ✗ for (int i = 0; i < size; i += dim)
103 ✗ for (int j = 0; j < dim; j++)
104 ✗ out[i+j] = 0.0f;
105 }
106 1563777 return cost * lambda;
107 }
108
2/2
✓ Branch 0 taken 429391 times.
✓ Branch 1 taken 6465803 times.
 6895194 if (!scaled) {
109 429391 s->aacdsp.abs_pow34(s->scoefs, in, size);
110 429391 scaled = s->scoefs;
111 }
112 6895194 s->aacdsp.quant_bands(s->qcoefs, in, scaled, size, !BT_UNSIGNED, aac_cb_maxval[cb], Q34, ROUNDING);
113
2/2
✓ Branch 0 taken 4270420 times.
✓ Branch 1 taken 2624774 times.
 6895194 if (BT_UNSIGNED) {
114 4270420 off = 0;
115 } else {
116 2624774 off = aac_cb_maxval[cb];
117 }
118
2/2
✓ Branch 0 taken 54787430 times.
✓ Branch 1 taken 6895194 times.
 61682624 for (int i = 0; i < size; i += dim) {
119 const float *vec;
120 54787430 int *quants = s->qcoefs + i;
121 54787430 int curidx = 0;
122 int curbits;
123 54787430 float quantized, rd = 0.0f;
124
2/2
✓ Branch 0 taken 140874108 times.
✓ Branch 1 taken 54787430 times.
 195661538 for (int j = 0; j < dim; j++) {
125 140874108 curidx *= aac_cb_range[cb];
126 140874108 curidx += quants[j] + off;
127 }
128 54787430 curbits = ff_aac_spectral_bits[cb-1][curidx];
129 54787430 vec = &ff_aac_codebook_vectors[cb-1][curidx*dim];
130
2/2
✓ Branch 0 taken 34726365 times.
✓ Branch 1 taken 20061065 times.
 54787430 if (BT_UNSIGNED) {
131
2/2
✓ Branch 0 taken 82841916 times.
✓ Branch 1 taken 34726365 times.
 117568281 for (int j = 0; j < dim; j++) {
132 82841916 float t = fabsf(in[i+j]);
133 float di;
134
4/4
✓ Branch 0 taken 14862476 times.
✓ Branch 1 taken 67979440 times.
✓ Branch 2 taken 1612815 times.
✓ Branch 3 taken 13249661 times.
 82841916 if (BT_ESC && vec[j] == 64.0f) { //FIXME: slow
135
2/2
✓ Branch 0 taken 113 times.
✓ Branch 1 taken 1612702 times.
 1612815 if (t >= CLIPPED_ESCAPE) {
136 113 quantized = CLIPPED_ESCAPE;
137 113 curbits += 21;
138 } else {
139 1612702 int c = av_clip_uintp2(quant(t, Q, ROUNDING), 13);
140 1612702 quantized = c*cbrtf(c)*IQ;
141 1612702 curbits += av_log2(c)*2 - 4 + 1;
142 }
143 } else {
144 81229101 quantized = vec[j]*IQ;
145 }
146 82841916 di = t - quantized;
147
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 82841916 times.
 82841916 if (out)
148 ✗ out[i+j] = in[i+j] >= 0 ? quantized : -quantized;
149
2/2
✓ Branch 0 taken 49428861 times.
✓ Branch 1 taken 33413055 times.
 82841916 if (vec[j] != 0.0f)
150 49428861 curbits++;
151 82841916 qenergy += quantized*quantized;
152 82841916 rd += di*di;
153 }
154 } else {
155
2/2
✓ Branch 0 taken 58032192 times.
✓ Branch 1 taken 20061065 times.
 78093257 for (int j = 0; j < dim; j++) {
156 58032192 quantized = vec[j]*IQ;
157 58032192 qenergy += quantized*quantized;
158
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 58032192 times.
 58032192 if (out)
159 ✗ out[i+j] = quantized;
160 58032192 rd += (in[i+j] - quantized)*(in[i+j] - quantized);
161 }
162 }
163 54787430 cost += rd * lambda + curbits;
164 54787430 resbits += curbits;
165
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 54787430 times.
 54787430 if (cost >= uplim)
166 ✗ return uplim;
167
2/2
✓ Branch 0 taken 3136648 times.
✓ Branch 1 taken 51650782 times.
 54787430 if (pb) {
168 3136648 put_bits(pb, ff_aac_spectral_bits[cb-1][curidx], ff_aac_spectral_codes[cb-1][curidx]);
169
2/2
✓ Branch 0 taken 1675091 times.
✓ Branch 1 taken 1461557 times.
 3136648 if (BT_UNSIGNED)
170
2/2
✓ Branch 0 taken 4382896 times.
✓ Branch 1 taken 1675091 times.
 6057987 for (int j = 0; j < dim; j++)
171
2/2
✓ Branch 0 taken 3087564 times.
✓ Branch 1 taken 1295332 times.
 4382896 if (ff_aac_codebook_vectors[cb-1][curidx*dim+j] != 0.0f)
172 3087564 put_bits(pb, 1, in[i+j] < 0.0f);
173
2/2
✓ Branch 0 taken 444802 times.
✓ Branch 1 taken 2691846 times.
 3136648 if (BT_ESC) {
174
2/2
✓ Branch 0 taken 889604 times.
✓ Branch 1 taken 444802 times.
 1334406 for (int j = 0; j < 2; j++) {
175
2/2
✓ Branch 0 taken 131584 times.
✓ Branch 1 taken 758020 times.
 889604 if (ff_aac_codebook_vectors[cb-1][curidx*2+j] == 64.0f) {
176 131584 int coef = av_clip_uintp2(quant(fabsf(in[i+j]), Q, ROUNDING), 13);
177 131584 int len = av_log2(coef);
178
179 131584 put_bits(pb, len - 4 + 1, (1 << (len - 4 + 1)) - 2);
180 131584 put_sbits(pb, len, coef);
181 }
182 }
183 }
184 }
185 }
186
187
2/2
✓ Branch 0 taken 6395175 times.
✓ Branch 1 taken 500019 times.
 6895194 if (bits)
188 6395175 *bits = resbits;
189
2/2
✓ Branch 0 taken 3187214 times.
✓ Branch 1 taken 3707980 times.
 6895194 if (energy)
190 3187214 *energy = qenergy;
191 6895194 return cost;
192 }
193
194 ✗ static inline float quantize_and_encode_band_cost_NONE(struct AACEncContext *s, PutBitContext *pb,
195 const float *in, float *quant, const float *scaled,
196 int size, int scale_idx, int cb,
197 const float lambda, const float uplim,
198 int *bits, float *energy) {
199 ✗ av_assert0(0);
200 return 0.0f;
201 }
202
203 #define QUANTIZE_AND_ENCODE_BAND_COST_FUNC(NAME, BT_ZERO, BT_UNSIGNED, BT_PAIR, BT_ESC, BT_NOISE, BT_STEREO, ROUNDING) \
204 static float quantize_and_encode_band_cost_ ## NAME( \
205 struct AACEncContext *s, \
206 PutBitContext *pb, const float *in, float *quant, \
207 const float *scaled, int size, int scale_idx, \
208 int cb, const float lambda, const float uplim, \
209 int *bits, float *energy) { \
210 return quantize_and_encode_band_cost_template( \
211 s, pb, in, quant, scaled, size, scale_idx, \
212 BT_ESC ? ESC_BT : cb, lambda, uplim, bits, energy, \
213 BT_ZERO, BT_UNSIGNED, BT_PAIR, BT_ESC, BT_NOISE, BT_STEREO, \
214 ROUNDING); \
215 }
216
217 1532448 QUANTIZE_AND_ENCODE_BAND_COST_FUNC(ZERO, 1, 0, 0, 0, 0, 0, ROUND_STANDARD)
218 1497674 QUANTIZE_AND_ENCODE_BAND_COST_FUNC(SQUAD, 0, 0, 0, 0, 0, 0, ROUND_STANDARD)
219 1201329 QUANTIZE_AND_ENCODE_BAND_COST_FUNC(UQUAD, 0, 1, 0, 0, 0, 0, ROUND_STANDARD)
220 1127100 QUANTIZE_AND_ENCODE_BAND_COST_FUNC(SPAIR, 0, 0, 1, 0, 0, 0, ROUND_STANDARD)
221 2109672 QUANTIZE_AND_ENCODE_BAND_COST_FUNC(UPAIR, 0, 1, 1, 0, 0, 0, ROUND_STANDARD)
222 954333 QUANTIZE_AND_ENCODE_BAND_COST_FUNC(ESC, 0, 1, 1, 1, 0, 0, ROUND_STANDARD)
223 5086 QUANTIZE_AND_ENCODE_BAND_COST_FUNC(ESC_RTZ, 0, 1, 1, 1, 0, 0, ROUND_TO_ZERO)
224 14017 QUANTIZE_AND_ENCODE_BAND_COST_FUNC(NOISE, 0, 0, 0, 0, 1, 0, ROUND_STANDARD)
225 17312 QUANTIZE_AND_ENCODE_BAND_COST_FUNC(STEREO,0, 0, 0, 0, 0, 1, ROUND_STANDARD)
226
227 static const quantize_and_encode_band_func quantize_and_encode_band_cost_arr[] =
228 {
229 quantize_and_encode_band_cost_ZERO,
230 quantize_and_encode_band_cost_SQUAD,
231 quantize_and_encode_band_cost_SQUAD,
232 quantize_and_encode_band_cost_UQUAD,
233 quantize_and_encode_band_cost_UQUAD,
234 quantize_and_encode_band_cost_SPAIR,
235 quantize_and_encode_band_cost_SPAIR,
236 quantize_and_encode_band_cost_UPAIR,
237 quantize_and_encode_band_cost_UPAIR,
238 quantize_and_encode_band_cost_UPAIR,
239 quantize_and_encode_band_cost_UPAIR,
240 quantize_and_encode_band_cost_ESC,
241 quantize_and_encode_band_cost_NONE, /* CB 12 doesn't exist */
242 quantize_and_encode_band_cost_NOISE,
243 quantize_and_encode_band_cost_STEREO,
244 quantize_and_encode_band_cost_STEREO,
245 };
246
247 static const quantize_and_encode_band_func quantize_and_encode_band_cost_rtz_arr[] =
248 {
249 quantize_and_encode_band_cost_ZERO,
250 quantize_and_encode_band_cost_SQUAD,
251 quantize_and_encode_band_cost_SQUAD,
252 quantize_and_encode_band_cost_UQUAD,
253 quantize_and_encode_band_cost_UQUAD,
254 quantize_and_encode_band_cost_SPAIR,
255 quantize_and_encode_band_cost_SPAIR,
256 quantize_and_encode_band_cost_UPAIR,
257 quantize_and_encode_band_cost_UPAIR,
258 quantize_and_encode_band_cost_UPAIR,
259 quantize_and_encode_band_cost_UPAIR,
260 quantize_and_encode_band_cost_ESC_RTZ,
261 quantize_and_encode_band_cost_NONE, /* CB 12 doesn't exist */
262 quantize_and_encode_band_cost_NOISE,
263 quantize_and_encode_band_cost_STEREO,
264 quantize_and_encode_band_cost_STEREO,
265 };
266
267 8014275 float ff_quantize_and_encode_band_cost(struct AACEncContext *s, PutBitContext *pb,
268 const float *in, float *quant, const float *scaled,
269 int size, int scale_idx, int cb,
270 const float lambda, const float uplim,
271 int *bits, float *energy)
272 {
273 8014275 return quantize_and_encode_band_cost_arr[cb](s, pb, in, quant, scaled, size,
274 scale_idx, cb, lambda, uplim,
275 bits, energy);
276 }
277
278 444696 static inline void quantize_and_encode_band(struct AACEncContext *s, PutBitContext *pb,
279 const float *in, float *out, int size, int scale_idx,
280 int cb, const float lambda, int rtz)
281 {
282
2/2
✓ Branch 0 taken 16086 times.
✓ Branch 1 taken 428610 times.
 444696 (rtz ? quantize_and_encode_band_cost_rtz_arr : quantize_and_encode_band_cost_arr)[cb](s, pb, in, out, NULL, size, scale_idx, cb,
283 lambda, INFINITY, NULL, NULL);
284 444696 }
285
286 /**
287 * structure used in optimal codebook search
288 */
289 typedef struct BandCodingPath {
290 int prev_idx; ///< pointer to the previous path point
291 float cost; ///< path cost
292 int run;
293 } BandCodingPath;
294
295 typedef struct TrellisPath {
296 float cost;
297 int prev;
298 } TrellisPath;
299
300 #define TRELLIS_STAGES 121
301 #define TRELLIS_STATES (SCALE_MAX_DIFF+1)
302
303 9397 static void set_special_band_scalefactors(AACEncContext *s, SingleChannelElement *sce)
304 {
305 int w, g;
306 9397 int prevscaler_n = -255, prevscaler_i = 0;
307 9397 int bands = 0;
308
309
2/2
✓ Branch 0 taken 9976 times.
✓ Branch 1 taken 9397 times.
 19373 for (w = 0; w < sce->ics.num_windows; w += sce->ics.group_len[w]) {
310
2/2
✓ Branch 0 taken 457407 times.
✓ Branch 1 taken 9976 times.
 467383 for (g = 0; g < sce->ics.num_swb; g++) {
311
2/2
✓ Branch 0 taken 13370 times.
✓ Branch 1 taken 444037 times.
 457407 if (sce->zeroes[w*16+g])
312 13370 continue;
313
4/4
✓ Branch 0 taken 437730 times.
✓ Branch 1 taken 6307 times.
✓ Branch 2 taken 2238 times.
✓ Branch 3 taken 435492 times.
 444037 if (sce->band_type[w*16+g] == INTENSITY_BT || sce->band_type[w*16+g] == INTENSITY_BT2) {
314 8545 sce->sf_idx[w*16+g] = av_clip(roundf(log2f(sce->is_ener[w*16+g])*2), -155, 100);
315 8545 bands++;
316
2/2
✓ Branch 0 taken 6614 times.
✓ Branch 1 taken 428878 times.
 435492 } else if (sce->band_type[w*16+g] == NOISE_BT) {
317 6614 sce->sf_idx[w*16+g] = av_clip(3+ceilf(log2f(sce->pns_ener[w*16+g])*2), -100, 155);
318
2/2
✓ Branch 0 taken 1161 times.
✓ Branch 1 taken 5453 times.
 6614 if (prevscaler_n == -255)
319 1161 prevscaler_n = sce->sf_idx[w*16+g];
320 6614 bands++;
321 }
322 }
323 }
324
325
2/2
✓ Branch 0 taken 7794 times.
✓ Branch 1 taken 1603 times.
 9397 if (!bands)
326 7794 return;
327
328 /* Clip the scalefactor indices */
329
2/2
✓ Branch 0 taken 1670 times.
✓ Branch 1 taken 1603 times.
 3273 for (w = 0; w < sce->ics.num_windows; w += sce->ics.group_len[w]) {
330
2/2
✓ Branch 0 taken 78680 times.
✓ Branch 1 taken 1670 times.
 80350 for (g = 0; g < sce->ics.num_swb; g++) {
331
2/2
✓ Branch 0 taken 5387 times.
✓ Branch 1 taken 73293 times.
 78680 if (sce->zeroes[w*16+g])
332 5387 continue;
333
4/4
✓ Branch 0 taken 66986 times.
✓ Branch 1 taken 6307 times.
✓ Branch 2 taken 2238 times.
✓ Branch 3 taken 64748 times.
 73293 if (sce->band_type[w*16+g] == INTENSITY_BT || sce->band_type[w*16+g] == INTENSITY_BT2) {
334 8545 sce->sf_idx[w*16+g] = prevscaler_i = av_clip(sce->sf_idx[w*16+g], prevscaler_i - SCALE_MAX_DIFF, prevscaler_i + SCALE_MAX_DIFF);
335
2/2
✓ Branch 0 taken 6614 times.
✓ Branch 1 taken 58134 times.
 64748 } else if (sce->band_type[w*16+g] == NOISE_BT) {
336 6614 sce->sf_idx[w*16+g] = prevscaler_n = av_clip(sce->sf_idx[w*16+g], prevscaler_n - SCALE_MAX_DIFF, prevscaler_n + SCALE_MAX_DIFF);
337 }
338 }
339 }
340 }
341
342 8660 static void search_for_quantizers_fast(AVCodecContext *avctx, AACEncContext *s,
343 SingleChannelElement *sce,
344 const float lambda)
345 {
346 8660 int start = 0, i, w, w2, g;
347 8660 int destbits = avctx->bit_rate * 1024.0 / avctx->sample_rate / avctx->ch_layout.nb_channels * (lambda / 120.f);
348 8660 float dists[128] = { 0 }, uplims[128] = { 0 };
349 float maxvals[128];
350 int fflag, minscaler;
351 8660 int its = 0;
352 8660 int allz = 0;
353 8660 float minthr = INFINITY;
354
355 // for values above this the decoder might end up in an endless loop
356 // due to always having more bits than what can be encoded.
357 8660 destbits = FFMIN(destbits, 5800);
358 //some heuristic to determine initial quantizers will reduce search time
359 //determine zero bands and upper limits
360
2/2
✓ Branch 0 taken 9205 times.
✓ Branch 1 taken 8660 times.
 17865 for (w = 0; w < sce->ics.num_windows; w += sce->ics.group_len[w]) {
361 9205 start = 0;
362
2/2
✓ Branch 0 taken 421238 times.
✓ Branch 1 taken 9205 times.
 430443 for (g = 0; g < sce->ics.num_swb; g++) {
363 421238 int nz = 0;
364 421238 float uplim = 0.0f;
365
2/2
✓ Branch 0 taken 434384 times.
✓ Branch 1 taken 421238 times.
 855622 for (w2 = 0; w2 < sce->ics.group_len[w]; w2++) {
366 434384 FFPsyBand *band = &s->psy.ch[s->cur_channel].psy_bands[(w+w2)*16+g];
367 434384 uplim += band->threshold;
368
3/4
✓ Branch 0 taken 421345 times.
✓ Branch 1 taken 13039 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 421345 times.
 434384 if (band->energy <= band->threshold || band->threshold == 0.0f) {
369 13039 sce->zeroes[(w+w2)*16+g] = 1;
370 13039 continue;
371 }
372 421345 nz = 1;
373 }
374 421238 uplims[w*16+g] = uplim *512;
375 421238 sce->band_type[w*16+g] = 0;
376 421238 sce->zeroes[w*16+g] = !nz;
377
2/2
✓ Branch 0 taken 413884 times.
✓ Branch 1 taken 7354 times.
 421238 if (nz)
378
2/2
✓ Branch 0 taken 92607 times.
✓ Branch 1 taken 321277 times.
 413884 minthr = FFMIN(minthr, uplim);
379 421238 allz |= nz;
380 421238 start += sce->ics.swb_sizes[g];
381 }
382 }
383
2/2
✓ Branch 0 taken 9205 times.
✓ Branch 1 taken 8660 times.
 17865 for (w = 0; w < sce->ics.num_windows; w += sce->ics.group_len[w]) {
384
2/2
✓ Branch 0 taken 421238 times.
✓ Branch 1 taken 9205 times.
 430443 for (g = 0; g < sce->ics.num_swb; g++) {
385
2/2
✓ Branch 0 taken 7354 times.
✓ Branch 1 taken 413884 times.
 421238 if (sce->zeroes[w*16+g]) {
386 7354 sce->sf_idx[w*16+g] = SCALE_ONE_POS;
387 7354 continue;
388 }
389
2/2
✓ Branch 0 taken 180176 times.
✓ Branch 1 taken 233708 times.
 413884 sce->sf_idx[w*16+g] = SCALE_ONE_POS + FFMIN(log2f(uplims[w*16+g]/minthr)*4,59);
390 }
391 }
392
393
2/2
✓ Branch 0 taken 84 times.
✓ Branch 1 taken 8576 times.
 8660 if (!allz)
394 84 return;
395 8576 s->aacdsp.abs_pow34(s->scoefs, sce->coeffs, 1024);
396 8576 ff_quantize_band_cost_cache_init(s);
397
398
2/2
✓ Branch 0 taken 8965 times.
✓ Branch 1 taken 8576 times.
 17541 for (w = 0; w < sce->ics.num_windows; w += sce->ics.group_len[w]) {
399 8965 start = w*128;
400
2/2
✓ Branch 0 taken 417668 times.
✓ Branch 1 taken 8965 times.
 426633 for (g = 0; g < sce->ics.num_swb; g++) {
401 417668 const float *scaled = s->scoefs + start;
402 417668 maxvals[w*16+g] = find_max_val(sce->ics.group_len[w], sce->ics.swb_sizes[g], scaled);
403 417668 start += sce->ics.swb_sizes[g];
404 }
405 }
406
407 //perform two-loop search
408 //outer loop - improve quality
409 do {
410 int tbits, qstep;
411 36546 minscaler = sce->sf_idx[0];
412 //inner loop - quantize spectrum to fit into given number of bits
413
2/2
✓ Branch 0 taken 27970 times.
✓ Branch 1 taken 8576 times.
 36546 qstep = its ? 1 : 32;
414 do {
415 83451 int prev = -1;
416 83451 tbits = 0;
417
2/2
✓ Branch 0 taken 89801 times.
✓ Branch 1 taken 83451 times.
 173252 for (w = 0; w < sce->ics.num_windows; w += sce->ics.group_len[w]) {
418 89801 start = w*128;
419
2/2
✓ Branch 0 taken 4051004 times.
✓ Branch 1 taken 89801 times.
 4140805 for (g = 0; g < sce->ics.num_swb; g++) {
420 4051004 const float *coefs = sce->coeffs + start;
421 4051004 const float *scaled = s->scoefs + start;
422 4051004 int bits = 0;
423 int cb;
424 4051004 float dist = 0.0f;
425
426
3/4
✓ Branch 0 taken 3995104 times.
✓ Branch 1 taken 55900 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 3995104 times.
 4051004 if (sce->zeroes[w*16+g] || sce->sf_idx[w*16+g] >= 218) {
427 55900 start += sce->ics.swb_sizes[g];
428 55900 continue;
429 }
430 3995104 minscaler = FFMIN(minscaler, sce->sf_idx[w*16+g]);
431 3995104 cb = find_min_book(maxvals[w*16+g], sce->sf_idx[w*16+g]);
432
2/2
✓ Branch 0 taken 4119984 times.
✓ Branch 1 taken 3995104 times.
 8115088 for (w2 = 0; w2 < sce->ics.group_len[w]; w2++) {
433 int b;
434 8239968 dist += quantize_band_cost_cached(s, w + w2, g,
435 4119984 coefs + w2*128,
436 4119984 scaled + w2*128,
437 4119984 sce->ics.swb_sizes[g],
438 4119984 sce->sf_idx[w*16+g],
439 cb, 1.0f, INFINITY,
440 &b, NULL, 0);
441 4119984 bits += b;
442 }
443 3995104 dists[w*16+g] = dist - bits;
444
2/2
✓ Branch 0 taken 3911653 times.
✓ Branch 1 taken 83451 times.
 3995104 if (prev != -1) {
445 3911653 bits += ff_aac_scalefactor_bits[sce->sf_idx[w*16+g] - prev + SCALE_DIFF_ZERO];
446 }
447 3995104 tbits += bits;
448 3995104 start += sce->ics.swb_sizes[g];
449 3995104 prev = sce->sf_idx[w*16+g];
450 }
451 }
452
2/2
✓ Branch 0 taken 16100 times.
✓ Branch 1 taken 67351 times.
 83451 if (tbits > destbits) {
453
2/2
✓ Branch 0 taken 2060800 times.
✓ Branch 1 taken 16100 times.
 2076900 for (i = 0; i < 128; i++)
454
1/2
✓ Branch 0 taken 2060800 times.
✗ Branch 1 not taken.
 2060800 if (sce->sf_idx[i] < 218 - qstep)
455 2060800 sce->sf_idx[i] += qstep;
456 } else {
457
2/2
✓ Branch 0 taken 8620928 times.
✓ Branch 1 taken 67351 times.
 8688279 for (i = 0; i < 128; i++)
458
2/2
✓ Branch 0 taken 3491091 times.
✓ Branch 1 taken 5129837 times.
 8620928 if (sce->sf_idx[i] > 60 - qstep)
459 3491091 sce->sf_idx[i] -= qstep;
460 }
461 83451 qstep >>= 1;
462
5/6
✓ Branch 0 taken 40571 times.
✓ Branch 1 taken 42880 times.
✓ Branch 2 taken 4025 times.
✓ Branch 3 taken 36546 times.
✓ Branch 4 taken 4025 times.
✗ Branch 5 not taken.
 83451 if (!qstep && tbits > destbits*1.02 && sce->sf_idx[0] < 217)
463 4025 qstep = 1;
464
2/2
✓ Branch 0 taken 46905 times.
✓ Branch 1 taken 36546 times.
 83451 } while (qstep);
465
466 36546 fflag = 0;
467 36546 minscaler = av_clip(minscaler, 60, 255 - SCALE_MAX_DIFF);
468
469
2/2
✓ Branch 0 taken 39014 times.
✓ Branch 1 taken 36546 times.
 75560 for (w = 0; w < sce->ics.num_windows; w += sce->ics.group_len[w]) {
470
2/2
✓ Branch 0 taken 1762191 times.
✓ Branch 1 taken 39014 times.
 1801205 for (g = 0; g < sce->ics.num_swb; g++) {
471 1762191 int prevsc = sce->sf_idx[w*16+g];
472
3/4
✓ Branch 0 taken 12605 times.
✓ Branch 1 taken 1749586 times.
✓ Branch 2 taken 12605 times.
✗ Branch 3 not taken.
 1762191 if (dists[w*16+g] > uplims[w*16+g] && sce->sf_idx[w*16+g] > 60) {
473
2/2
✓ Branch 1 taken 12000 times.
✓ Branch 2 taken 605 times.
 12605 if (find_min_book(maxvals[w*16+g], sce->sf_idx[w*16+g]-1))
474 12000 sce->sf_idx[w*16+g]--;
475 else //Try to make sure there is some energy in every band
476 605 sce->sf_idx[w*16+g]-=2;
477 }
478 1762191 sce->sf_idx[w*16+g] = av_clip(sce->sf_idx[w*16+g], minscaler, minscaler + SCALE_MAX_DIFF);
479 1762191 sce->sf_idx[w*16+g] = FFMIN(sce->sf_idx[w*16+g], 219);
480
2/2
✓ Branch 0 taken 510281 times.
✓ Branch 1 taken 1251910 times.
 1762191 if (sce->sf_idx[w*16+g] != prevsc)
481 510281 fflag = 1;
482 1762191 sce->band_type[w*16+g] = find_min_book(maxvals[w*16+g], sce->sf_idx[w*16+g]);
483 }
484 }
485 36546 its++;
486
4/4
✓ Branch 0 taken 30964 times.
✓ Branch 1 taken 5582 times.
✓ Branch 2 taken 27970 times.
✓ Branch 3 taken 2994 times.
 36546 } while (fflag && its < 10);
487 }
488
489 2001 static void search_for_pns(AACEncContext *s, AVCodecContext *avctx, SingleChannelElement *sce)
490 {
491 FFPsyBand *band;
492 int w, g, w2, i;
493 2001 int wlen = 1024 / sce->ics.num_windows;
494 int bandwidth, cutoff;
495 2001 float *PNS = &s->scoefs[0*128], *PNS34 = &s->scoefs[1*128];
496 2001 float *NOR34 = &s->scoefs[3*128];
497 uint8_t nextband[128];
498 2001 const float lambda = s->lambda;
499 2001 const float freq_mult = avctx->sample_rate*0.5f/wlen;
500 2001 const float thr_mult = NOISE_LAMBDA_REPLACE*(100.0f/lambda);
501
2/6
✗ Branch 0 not taken.
✓ Branch 1 taken 2001 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2001 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 2001 const float spread_threshold = FFMIN(0.75f, NOISE_SPREAD_THRESHOLD*FFMAX(0.5f, lambda/100.f));
502 2001 const float dist_bias = av_clipf(4.f * 120 / lambda, 0.25f, 4.0f);
503
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2001 times.
 2001 const float pns_transient_energy_r = FFMIN(0.7f, lambda / 140.f);
504
505 4002 int refbits = avctx->bit_rate * 1024.0 / avctx->sample_rate
506
1/2
✓ Branch 0 taken 2001 times.
✗ Branch 1 not taken.
 2001 / ((avctx->flags & AV_CODEC_FLAG_QSCALE) ? 2.0f : avctx->ch_layout.nb_channels)
507 2001 * (lambda / 120.f);
508
509 /** Keep this in sync with twoloop's cutoff selection */
510 2001 float rate_bandwidth_multiplier = 1.5f;
511 2001 int prev = -1000, prev_sf = -1;
512
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2001 times.
 2001 int frame_bit_rate = (avctx->flags & AV_CODEC_FLAG_QSCALE)
513 ✗ ? (refbits * rate_bandwidth_multiplier * avctx->sample_rate / 1024)
514 2001 : (avctx->bit_rate / avctx->ch_layout.nb_channels);
515
516 2001 frame_bit_rate *= 1.15f;
517
518
2/2
✓ Branch 0 taken 832 times.
✓ Branch 1 taken 1169 times.
 2001 if (avctx->cutoff > 0) {
519 832 bandwidth = avctx->cutoff;
520 } else {
521
5/10
✓ Branch 0 taken 1169 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1169 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 1169 times.
✓ Branch 6 taken 1169 times.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✓ Branch 9 taken 1169 times.
 1169 bandwidth = FFMAX(3000, AAC_CUTOFF_FROM_BITRATE(frame_bit_rate, 1, avctx->sample_rate));
522 }
523
524 2001 cutoff = bandwidth * 2 * wlen / avctx->sample_rate;
525
526 2001 memcpy(sce->band_alt, sce->band_type, sizeof(sce->band_type));
527 2001 ff_init_nextband_map(sce, nextband);
528
2/2
✓ Branch 0 taken 2129 times.
✓ Branch 1 taken 2001 times.
 4130 for (w = 0; w < sce->ics.num_windows; w += sce->ics.group_len[w]) {
529 2129 int wstart = w*128;
530
2/2
✓ Branch 0 taken 94849 times.
✓ Branch 1 taken 2129 times.
 96978 for (g = 0; g < sce->ics.num_swb; g++) {
531 int noise_sfi;
532 94849 float dist1 = 0.0f, dist2 = 0.0f, noise_amp;
533 94849 float pns_energy = 0.0f, pns_tgt_energy, energy_ratio, dist_thresh;
534 94849 float sfb_energy = 0.0f, threshold = 0.0f, spread = 2.0f;
535 94849 float min_energy = -1.0f, max_energy = 0.0f;
536 94849 const int start = wstart+sce->ics.swb_offset[g];
537 94849 const float freq = (start-wstart)*freq_mult;
538
2/2
✓ Branch 0 taken 46905 times.
✓ Branch 1 taken 47944 times.
 94849 const float freq_boost = FFMAX(0.88f*freq/NOISE_LOW_LIMIT, 1.0f);
539
4/4
✓ Branch 0 taken 48786 times.
✓ Branch 1 taken 46063 times.
✓ Branch 2 taken 6589 times.
✓ Branch 3 taken 42197 times.
 94849 if (freq < NOISE_LOW_LIMIT || (start-wstart) >= cutoff) {
540
2/2
✓ Branch 0 taken 43215 times.
✓ Branch 1 taken 9437 times.
 52652 if (!sce->zeroes[w*16+g])
541 43215 prev_sf = sce->sf_idx[w*16+g];
542 52652 continue;
543 }
544
2/2
✓ Branch 0 taken 43713 times.
✓ Branch 1 taken 42197 times.
 85910 for (w2 = 0; w2 < sce->ics.group_len[w]; w2++) {
545 43713 band = &s->psy.ch[s->cur_channel].psy_bands[(w+w2)*16+g];
546 43713 sfb_energy += band->energy;
547
2/2
✓ Branch 0 taken 42531 times.
✓ Branch 1 taken 1182 times.
 43713 spread = FFMIN(spread, band->spread);
548 43713 threshold += band->threshold;
549
2/2
✓ Branch 0 taken 42197 times.
✓ Branch 1 taken 1516 times.
 43713 if (!w2) {
550 42197 min_energy = max_energy = band->energy;
551 } else {
552
2/2
✓ Branch 0 taken 526 times.
✓ Branch 1 taken 990 times.
 1516 min_energy = FFMIN(min_energy, band->energy);
553
2/2
✓ Branch 0 taken 676 times.
✓ Branch 1 taken 840 times.
 1516 max_energy = FFMAX(max_energy, band->energy);
554 }
555 }
556
557 /* Ramps down at ~8000Hz and loosens the dist threshold */
558 42197 dist_thresh = av_clipf(2.5f*NOISE_LOW_LIMIT/freq, 0.5f, 2.5f) * dist_bias;
559
560 /* PNS is acceptable when all of these are true:
561 * 1. high spread energy (noise-like band)
562 * 2. near-threshold energy (high PE means the random nature of PNS content will be noticed)
563 * 3. on short window groups, all windows have similar energy (variations in energy would be destroyed by PNS)
564 *
565 * At this stage, point 2 is relaxed for zeroed bands near the noise threshold (hole avoidance is more important)
566 */
567
4/4
✓ Branch 0 taken 38939 times.
✓ Branch 1 taken 3258 times.
✓ Branch 3 taken 38882 times.
✓ Branch 4 taken 57 times.
 42197 if ((!sce->zeroes[w*16+g] && !ff_sfdelta_can_remove_band(sce, nextband, prev_sf, w*16+g)) ||
568
8/8
✓ Branch 0 taken 38882 times.
✓ Branch 1 taken 3258 times.
✓ Branch 2 taken 4276 times.
✓ Branch 3 taken 34606 times.
✓ Branch 4 taken 6710 times.
✓ Branch 5 taken 824 times.
✓ Branch 6 taken 40837 times.
✓ Branch 7 taken 479 times.
 42140 ((sce->zeroes[w*16+g] || !sce->band_alt[w*16+g]) && sfb_energy < threshold*sqrtf(1.0f/freq_boost)) || spread < spread_threshold ||
569
6/6
✓ Branch 0 taken 38429 times.
✓ Branch 1 taken 2408 times.
✓ Branch 2 taken 34154 times.
✓ Branch 3 taken 4275 times.
✓ Branch 4 taken 21051 times.
✓ Branch 5 taken 13103 times.
 40837 (!sce->zeroes[w*16+g] && sce->band_alt[w*16+g] && sfb_energy > threshold*thr_mult*freq_boost) ||
570
2/2
✓ Branch 0 taken 70 times.
✓ Branch 1 taken 27664 times.
 27734 min_energy < pns_transient_energy_r * max_energy ) {
571 14533 sce->pns_ener[w*16+g] = sfb_energy;
572
2/2
✓ Branch 0 taken 13683 times.
✓ Branch 1 taken 850 times.
 14533 if (!sce->zeroes[w*16+g])
573 13683 prev_sf = sce->sf_idx[w*16+g];
574 14533 continue;
575 }
576
577
2/2
✓ Branch 0 taken 26732 times.
✓ Branch 1 taken 932 times.
 27664 pns_tgt_energy = sfb_energy*FFMIN(1.0f, spread*spread);
578 27664 noise_sfi = av_clip(roundf(log2f(pns_tgt_energy)*2), -100, 155); /* Quantize */
579 27664 noise_amp = -ff_aac_pow2sf_tab[noise_sfi + POW_SF2_ZERO]; /* Dequantize */
580
2/2
✓ Branch 0 taken 15132 times.
✓ Branch 1 taken 12532 times.
 27664 if (prev != -1000) {
581 15132 int noise_sfdiff = noise_sfi - prev + SCALE_DIFF_ZERO;
582
2/4
✓ Branch 0 taken 15132 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 15132 times.
 15132 if (noise_sfdiff < 0 || noise_sfdiff > 2*SCALE_MAX_DIFF) {
583 ✗ if (!sce->zeroes[w*16+g])
584 ✗ prev_sf = sce->sf_idx[w*16+g];
585 ✗ continue;
586 }
587 }
588
2/2
✓ Branch 0 taken 27699 times.
✓ Branch 1 taken 27664 times.
 55363 for (w2 = 0; w2 < sce->ics.group_len[w]; w2++) {
589 float band_energy, scale, pns_senergy;
590 27699 const int start_c = (w+w2)*128+sce->ics.swb_offset[g];
591 27699 band = &s->psy.ch[s->cur_channel].psy_bands[(w+w2)*16+g];
592
2/2
✓ Branch 0 taken 904492 times.
✓ Branch 1 taken 27699 times.
 932191 for (i = 0; i < sce->ics.swb_sizes[g]; i++) {
593 904492 s->random_state = lcg_random(s->random_state);
594 904492 PNS[i] = s->random_state;
595 }
596 27699 band_energy = s->fdsp->scalarproduct_float(PNS, PNS, sce->ics.swb_sizes[g]);
597 27699 scale = noise_amp/sqrtf(band_energy);
598 27699 s->fdsp->vector_fmul_scalar(PNS, PNS, scale, sce->ics.swb_sizes[g]);
599 27699 pns_senergy = s->fdsp->scalarproduct_float(PNS, PNS, sce->ics.swb_sizes[g]);
600 27699 pns_energy += pns_senergy;
601 27699 s->aacdsp.abs_pow34(NOR34, &sce->coeffs[start_c], sce->ics.swb_sizes[g]);
602 27699 s->aacdsp.abs_pow34(PNS34, PNS, sce->ics.swb_sizes[g]);
603 55398 dist1 += quantize_band_cost(s, &sce->coeffs[start_c],
604 NOR34,
605 27699 sce->ics.swb_sizes[g],
606 27699 sce->sf_idx[(w+w2)*16+g],
607 27699 sce->band_alt[(w+w2)*16+g],
608 27699 lambda/band->threshold, INFINITY, NULL, NULL);
609 /* Estimate rd on average as 5 bits for SF, 4 for the CB, plus spread energy * lambda/thr */
610 27699 dist2 += band->energy/(band->spread*band->spread)*lambda*dist_thresh/band->threshold;
611 }
612
3/4
✓ Branch 0 taken 27664 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 5750 times.
✓ Branch 3 taken 21914 times.
 27664 if (g && sce->band_type[w*16+g-1] == NOISE_BT) {
613 5750 dist2 += 5;
614 } else {
615 21914 dist2 += 9;
616 }
617 27664 energy_ratio = pns_tgt_energy/pns_energy; /* Compensates for quantization error */
618 27664 sce->pns_ener[w*16+g] = energy_ratio*pns_tgt_energy;
619
8/10
✓ Branch 0 taken 25256 times.
✓ Branch 1 taken 2408 times.
✓ Branch 2 taken 21050 times.
✓ Branch 3 taken 4206 times.
✓ Branch 4 taken 20403 times.
✓ Branch 5 taken 647 times.
✓ Branch 6 taken 20403 times.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✓ Branch 9 taken 20403 times.
 27664 if (sce->zeroes[w*16+g] || !sce->band_alt[w*16+g] || (energy_ratio > 0.85f && energy_ratio < 1.25f && dist2 < dist1)) {
620 6614 sce->band_type[w*16+g] = NOISE_BT;
621 6614 sce->zeroes[w*16+g] = 0;
622 6614 prev = noise_sfi;
623 } else {
624
1/2
✓ Branch 0 taken 21050 times.
✗ Branch 1 not taken.
 21050 if (!sce->zeroes[w*16+g])
625 21050 prev_sf = sce->sf_idx[w*16+g];
626 }
627 }
628 }
629 2001 }
630
631 2001 static void mark_pns(AACEncContext *s, AVCodecContext *avctx, SingleChannelElement *sce)
632 {
633 FFPsyBand *band;
634 int w, g, w2;
635 2001 int wlen = 1024 / sce->ics.num_windows;
636 int bandwidth, cutoff;
637 2001 const float lambda = s->lambda;
638 2001 const float freq_mult = avctx->sample_rate*0.5f/wlen;
639
2/6
✗ Branch 0 not taken.
✓ Branch 1 taken 2001 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2001 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 2001 const float spread_threshold = FFMIN(0.75f, NOISE_SPREAD_THRESHOLD*FFMAX(0.5f, lambda/100.f));
640
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2001 times.
 2001 const float pns_transient_energy_r = FFMIN(0.7f, lambda / 140.f);
641
642 4002 int refbits = avctx->bit_rate * 1024.0 / avctx->sample_rate
643
1/2
✓ Branch 0 taken 2001 times.
✗ Branch 1 not taken.
 2001 / ((avctx->flags & AV_CODEC_FLAG_QSCALE) ? 2.0f : avctx->ch_layout.nb_channels)
644 2001 * (lambda / 120.f);
645
646 /** Keep this in sync with twoloop's cutoff selection */
647 2001 float rate_bandwidth_multiplier = 1.5f;
648
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2001 times.
 2001 int frame_bit_rate = (avctx->flags & AV_CODEC_FLAG_QSCALE)
649 ✗ ? (refbits * rate_bandwidth_multiplier * avctx->sample_rate / 1024)
650 2001 : (avctx->bit_rate / avctx->ch_layout.nb_channels);
651
652 2001 frame_bit_rate *= 1.15f;
653
654
2/2
✓ Branch 0 taken 832 times.
✓ Branch 1 taken 1169 times.
 2001 if (avctx->cutoff > 0) {
655 832 bandwidth = avctx->cutoff;
656 } else {
657
5/10
✓ Branch 0 taken 1169 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1169 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 1169 times.
✓ Branch 6 taken 1169 times.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✓ Branch 9 taken 1169 times.
 1169 bandwidth = FFMAX(3000, AAC_CUTOFF_FROM_BITRATE(frame_bit_rate, 1, avctx->sample_rate));
658 }
659
660 2001 cutoff = bandwidth * 2 * wlen / avctx->sample_rate;
661
662 2001 memcpy(sce->band_alt, sce->band_type, sizeof(sce->band_type));
663
2/2
✓ Branch 0 taken 2129 times.
✓ Branch 1 taken 2001 times.
 4130 for (w = 0; w < sce->ics.num_windows; w += sce->ics.group_len[w]) {
664
2/2
✓ Branch 0 taken 94849 times.
✓ Branch 1 taken 2129 times.
 96978 for (g = 0; g < sce->ics.num_swb; g++) {
665 94849 float sfb_energy = 0.0f, threshold = 0.0f, spread = 2.0f;
666 94849 float min_energy = -1.0f, max_energy = 0.0f;
667 94849 const int start = sce->ics.swb_offset[g];
668 94849 const float freq = start*freq_mult;
669
2/2
✓ Branch 0 taken 46905 times.
✓ Branch 1 taken 47944 times.
 94849 const float freq_boost = FFMAX(0.88f*freq/NOISE_LOW_LIMIT, 1.0f);
670
4/4
✓ Branch 0 taken 48786 times.
✓ Branch 1 taken 46063 times.
✓ Branch 2 taken 6589 times.
✓ Branch 3 taken 42197 times.
 94849 if (freq < NOISE_LOW_LIMIT || start >= cutoff) {
671 52652 sce->can_pns[w*16+g] = 0;
672 52652 continue;
673 }
674
2/2
✓ Branch 0 taken 43713 times.
✓ Branch 1 taken 42197 times.
 85910 for (w2 = 0; w2 < sce->ics.group_len[w]; w2++) {
675 43713 band = &s->psy.ch[s->cur_channel].psy_bands[(w+w2)*16+g];
676 43713 sfb_energy += band->energy;
677
2/2
✓ Branch 0 taken 42531 times.
✓ Branch 1 taken 1182 times.
 43713 spread = FFMIN(spread, band->spread);
678 43713 threshold += band->threshold;
679
2/2
✓ Branch 0 taken 42197 times.
✓ Branch 1 taken 1516 times.
 43713 if (!w2) {
680 42197 min_energy = max_energy = band->energy;
681 } else {
682
2/2
✓ Branch 0 taken 526 times.
✓ Branch 1 taken 990 times.
 1516 min_energy = FFMIN(min_energy, band->energy);
683
2/2
✓ Branch 0 taken 676 times.
✓ Branch 1 taken 840 times.
 1516 max_energy = FFMAX(max_energy, band->energy);
684 }
685 }
686
687 /* PNS is acceptable when all of these are true:
688 * 1. high spread energy (noise-like band)
689 * 2. near-threshold energy (high PE means the random nature of PNS content will be noticed)
690 * 3. on short window groups, all windows have similar energy (variations in energy would be destroyed by PNS)
691 */
692 42197 sce->pns_ener[w*16+g] = sfb_energy;
693
6/6
✓ Branch 0 taken 41363 times.
✓ Branch 1 taken 834 times.
✓ Branch 2 taken 40894 times.
✓ Branch 3 taken 469 times.
✓ Branch 4 taken 411 times.
✓ Branch 5 taken 40483 times.
 42197 if (sfb_energy < threshold*sqrtf(1.5f/freq_boost) || spread < spread_threshold || min_energy < pns_transient_energy_r * max_energy) {
694 1714 sce->can_pns[w*16+g] = 0;
695 } else {
696 40483 sce->can_pns[w*16+g] = 1;
697 }
698 }
699 }
700 2001 }
701
702 629 static void search_for_ms(AACEncContext *s, ChannelElement *cpe)
703 {
704 629 int start = 0, i, w, w2, g, sid_sf_boost, prev_mid, prev_side;
705 uint8_t nextband0[128], nextband1[128];
706 629 float *M = s->scoefs + 128*0, *S = s->scoefs + 128*1;
707 629 float *L34 = s->scoefs + 128*2, *R34 = s->scoefs + 128*3;
708 629 float *M34 = s->scoefs + 128*4, *S34 = s->scoefs + 128*5;
709 629 const float lambda = s->lambda;
710
2/2
✓ Branch 0 taken 95 times.
✓ Branch 1 taken 534 times.
 629 const float mslambda = FFMIN(1.0f, lambda / 120.f);
711 629 SingleChannelElement *sce0 = &cpe->ch[0];
712 629 SingleChannelElement *sce1 = &cpe->ch[1];
713
2/2
✓ Branch 0 taken 521 times.
✓ Branch 1 taken 108 times.
 629 if (!cpe->common_window)
714 521 return;
715
716 /** Scout out next nonzero bands */
717 108 ff_init_nextband_map(sce0, nextband0);
718 108 ff_init_nextband_map(sce1, nextband1);
719
720 108 prev_mid = sce0->sf_idx[0];
721 108 prev_side = sce1->sf_idx[0];
722
2/2
✓ Branch 0 taken 122 times.
✓ Branch 1 taken 108 times.
 230 for (w = 0; w < sce0->ics.num_windows; w += sce0->ics.group_len[w]) {
723 122 start = 0;
724
2/2
✓ Branch 0 taken 5313 times.
✓ Branch 1 taken 122 times.
 5435 for (g = 0; g < sce0->ics.num_swb; g++) {
725 5313 float bmax = bval2bmax(g * 17.0f / sce0->ics.num_swb) / 0.0045f;
726
2/2
✓ Branch 0 taken 4874 times.
✓ Branch 1 taken 439 times.
 5313 if (!cpe->is_mask[w*16+g])
727 4874 cpe->ms_mask[w*16+g] = 0;
728
5/6
✓ Branch 0 taken 3608 times.
✓ Branch 1 taken 1705 times.
✓ Branch 2 taken 3608 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 3169 times.
✓ Branch 5 taken 439 times.
 5313 if (!sce0->zeroes[w*16+g] && !sce1->zeroes[w*16+g] && !cpe->is_mask[w*16+g]) {
729 3169 float Mmax = 0.0f, Smax = 0.0f;
730
731 /* Must compute mid/side SF and book for the whole window group */
732
2/2
✓ Branch 0 taken 3381 times.
✓ Branch 1 taken 3169 times.
 6550 for (w2 = 0; w2 < sce0->ics.group_len[w]; w2++) {
733
2/2
✓ Branch 0 taken 66700 times.
✓ Branch 1 taken 3381 times.
 70081 for (i = 0; i < sce0->ics.swb_sizes[g]; i++) {
734 66700 M[i] = (sce0->coeffs[start+(w+w2)*128+i]
735 66700 + sce1->coeffs[start+(w+w2)*128+i]) * 0.5;
736 66700 S[i] = M[i]
737 66700 - sce1->coeffs[start+(w+w2)*128+i];
738 }
739 3381 s->aacdsp.abs_pow34(M34, M, sce0->ics.swb_sizes[g]);
740 3381 s->aacdsp.abs_pow34(S34, S, sce0->ics.swb_sizes[g]);
741
2/2
✓ Branch 0 taken 66700 times.
✓ Branch 1 taken 3381 times.
 70081 for (i = 0; i < sce0->ics.swb_sizes[g]; i++ ) {
742
2/2
✓ Branch 0 taken 50122 times.
✓ Branch 1 taken 16578 times.
 66700 Mmax = FFMAX(Mmax, M34[i]);
743
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 66700 times.
 66700 Smax = FFMAX(Smax, S34[i]);
744 }
745 }
746
747
2/2
✓ Branch 0 taken 3439 times.
✓ Branch 1 taken 90 times.
 3529 for (sid_sf_boost = 0; sid_sf_boost < 4; sid_sf_boost++) {
748 3439 float dist1 = 0.0f, dist2 = 0.0f;
749 3439 int B0 = 0, B1 = 0;
750 int minidx;
751 int mididx, sididx;
752 int midcb, sidcb;
753
754 3439 minidx = FFMIN(sce0->sf_idx[w*16+g], sce1->sf_idx[w*16+g]);
755 3439 mididx = av_clip(minidx, 0, SCALE_MAX_POS - SCALE_DIV_512);
756 3439 sididx = av_clip(minidx - sid_sf_boost * 3, 0, SCALE_MAX_POS - SCALE_DIV_512);
757
4/4
✓ Branch 0 taken 3104 times.
✓ Branch 1 taken 335 times.
✓ Branch 2 taken 3027 times.
✓ Branch 3 taken 77 times.
 3439 if (sce0->band_type[w*16+g] != NOISE_BT && sce1->band_type[w*16+g] != NOISE_BT
758
1/2
✓ Branch 1 taken 3027 times.
✗ Branch 2 not taken.
 3027 && ( !ff_sfdelta_can_replace(sce0, nextband0, prev_mid, mididx, w*16+g)
759
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3027 times.
 3027 || !ff_sfdelta_can_replace(sce1, nextband1, prev_side, sididx, w*16+g))) {
760 /* scalefactor range violation, bad stuff, will decrease quality unacceptably */
761 ✗ continue;
762 }
763
764 3439 midcb = find_min_book(Mmax, mididx);
765 3439 sidcb = find_min_book(Smax, sididx);
766
767 /* No CB can be zero */
768 3439 midcb = FFMAX(1,midcb);
769 3439 sidcb = FFMAX(1,sidcb);
770
771
2/2
✓ Branch 0 taken 3651 times.
✓ Branch 1 taken 3439 times.
 7090 for (w2 = 0; w2 < sce0->ics.group_len[w]; w2++) {
772 3651 FFPsyBand *band0 = &s->psy.ch[s->cur_channel+0].psy_bands[(w+w2)*16+g];
773 3651 FFPsyBand *band1 = &s->psy.ch[s->cur_channel+1].psy_bands[(w+w2)*16+g];
774
2/2
✓ Branch 0 taken 1416 times.
✓ Branch 1 taken 2235 times.
 3651 float minthr = FFMIN(band0->threshold, band1->threshold);
775 int b1,b2,b3,b4;
776
2/2
✓ Branch 0 taken 68860 times.
✓ Branch 1 taken 3651 times.
 72511 for (i = 0; i < sce0->ics.swb_sizes[g]; i++) {
777 68860 M[i] = (sce0->coeffs[start+(w+w2)*128+i]
778 68860 + sce1->coeffs[start+(w+w2)*128+i]) * 0.5;
779 68860 S[i] = M[i]
780 68860 - sce1->coeffs[start+(w+w2)*128+i];
781 }
782
783 3651 s->aacdsp.abs_pow34(L34, sce0->coeffs+start+(w+w2)*128, sce0->ics.swb_sizes[g]);
784 3651 s->aacdsp.abs_pow34(R34, sce1->coeffs+start+(w+w2)*128, sce0->ics.swb_sizes[g]);
785 3651 s->aacdsp.abs_pow34(M34, M, sce0->ics.swb_sizes[g]);
786 3651 s->aacdsp.abs_pow34(S34, S, sce0->ics.swb_sizes[g]);
787 7302 dist1 += quantize_band_cost(s, &sce0->coeffs[start + (w+w2)*128],
788 L34,
789 3651 sce0->ics.swb_sizes[g],
790 3651 sce0->sf_idx[w*16+g],
791 3651 sce0->band_type[w*16+g],
792 3651 lambda / (band0->threshold + FLT_MIN), INFINITY, &b1, NULL);
793 7302 dist1 += quantize_band_cost(s, &sce1->coeffs[start + (w+w2)*128],
794 R34,
795 3651 sce1->ics.swb_sizes[g],
796 3651 sce1->sf_idx[w*16+g],
797 3651 sce1->band_type[w*16+g],
798 3651 lambda / (band1->threshold + FLT_MIN), INFINITY, &b2, NULL);
799 7302 dist2 += quantize_band_cost(s, M,
800 M34,
801 3651 sce0->ics.swb_sizes[g],
802 mididx,
803 midcb,
804 3651 lambda / (minthr + FLT_MIN), INFINITY, &b3, NULL);
805 7302 dist2 += quantize_band_cost(s, S,
806 S34,
807 3651 sce1->ics.swb_sizes[g],
808 sididx,
809 sidcb,
810 3651 mslambda / (minthr * bmax + FLT_MIN), INFINITY, &b4, NULL);
811 3651 B0 += b1+b2;
812 3651 B1 += b3+b4;
813 3651 dist1 -= b1+b2;
814 3651 dist2 -= b3+b4;
815 }
816
3/4
✓ Branch 0 taken 3439 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 2667 times.
✓ Branch 3 taken 772 times.
 3439 cpe->ms_mask[w*16+g] = dist2 <= dist1 && B1 < B0;
817
2/2
✓ Branch 0 taken 2667 times.
✓ Branch 1 taken 772 times.
 3439 if (cpe->ms_mask[w*16+g]) {
818
2/4
✓ Branch 0 taken 2667 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 2667 times.
✗ Branch 3 not taken.
 2667 if (sce0->band_type[w*16+g] != NOISE_BT && sce1->band_type[w*16+g] != NOISE_BT) {
819 2667 sce0->sf_idx[w*16+g] = mididx;
820 2667 sce1->sf_idx[w*16+g] = sididx;
821 2667 sce0->band_type[w*16+g] = midcb;
822 2667 sce1->band_type[w*16+g] = sidcb;
823 ✗ } else if ((sce0->band_type[w*16+g] != NOISE_BT) ^ (sce1->band_type[w*16+g] != NOISE_BT)) {
824 /* ms_mask unneeded, and it confuses some decoders */
825 ✗ cpe->ms_mask[w*16+g] = 0;
826 }
827 2667 break;
828
2/2
✓ Branch 0 taken 412 times.
✓ Branch 1 taken 360 times.
 772 } else if (B1 > B0) {
829 /* More boost won't fix this */
830 412 break;
831 }
832 }
833 }
834
4/4
✓ Branch 0 taken 3608 times.
✓ Branch 1 taken 1705 times.
✓ Branch 2 taken 3273 times.
✓ Branch 3 taken 335 times.
 5313 if (!sce0->zeroes[w*16+g] && sce0->band_type[w*16+g] < RESERVED_BT)
835 3273 prev_mid = sce0->sf_idx[w*16+g];
836
6/6
✓ Branch 0 taken 3608 times.
✓ Branch 1 taken 1705 times.
✓ Branch 2 taken 3169 times.
✓ Branch 3 taken 439 times.
✓ Branch 4 taken 2801 times.
✓ Branch 5 taken 368 times.
 5313 if (!sce1->zeroes[w*16+g] && !cpe->is_mask[w*16+g] && sce1->band_type[w*16+g] < RESERVED_BT)
837 2801 prev_side = sce1->sf_idx[w*16+g];
838 5313 start += sce0->ics.swb_sizes[g];
839 }
840 }
841 }
842
843 const AACCoefficientsEncoder ff_aac_coders[AAC_CODER_NB] = {
844 [AAC_CODER_TWOLOOP] = {
845 search_for_quantizers_twoloop,
846 codebook_trellis_rate,
847 quantize_and_encode_band,
848 ff_aac_encode_tns_info,
849 ff_aac_apply_tns,
850 set_special_band_scalefactors,
851 search_for_pns,
852 mark_pns,
853 ff_aac_search_for_tns,
854 search_for_ms,
855 ff_aac_search_for_is,
856 },
857 [AAC_CODER_FAST] = {
858 search_for_quantizers_fast,
859 codebook_trellis_rate,
860 quantize_and_encode_band,
861 ff_aac_encode_tns_info,
862 ff_aac_apply_tns,
863 set_special_band_scalefactors,
864 search_for_pns,
865 mark_pns,
866 ff_aac_search_for_tns,
867 search_for_ms,
868 ff_aac_search_for_is,
869 },
870 };
871