41 #define ENVELOPE_ADJUSTMENT_OFFSET 2
42 #define NOISE_FLOOR_OFFSET 6.0f
76 { 60, 60, 24, 24, 31, 31, 12, 12, 31, 12 };
78 #define SBR_INIT_VLC_STATIC(num, size) \
79 INIT_VLC_STATIC(&vlc_sbr[num], 9, sbr_tmp[num].table_size / sbr_tmp[num].elem_size, \
80 sbr_tmp[num].sbr_bits , 1, 1, \
81 sbr_tmp[num].sbr_codes, sbr_tmp[num].elem_size, sbr_tmp[num].elem_size, \
84 #define SBR_VLC_ROW(name) \
85 { name ## _codes, name ## _bits, sizeof(name ## _codes), sizeof(name ## _codes[0]) }
91 const void *sbr_codes, *sbr_bits;
92 const unsigned int table_size, elem_size;
118 for (n = 1; n < 320; n++)
123 for (n = 0; n < 320; n++)
142 sbr->
kx[0] = sbr->
kx[1];
163 return *(
const int16_t *)a - *(
const int16_t *)
b;
166 static inline int in_table_int16(
const int16_t *table,
int last_el, int16_t needle)
169 for (i = 0; i <= last_el; i++)
170 if (table[i] == needle)
180 static const float bands_warped[3] = { 1.32715174233856803909f,
181 1.18509277094158210129f,
182 1.11987160404675912501f };
183 const float lim_bands_per_octave_warped = bands_warped[sbr->
bs_limiter_bands - 1];
184 int16_t patch_borders[7];
187 patch_borders[0] = sbr->
kx[1];
194 memcpy(sbr->
f_tablelim + sbr->
n[0] + 1, patch_borders + 1,
195 (sbr->
num_patches - 1) *
sizeof(patch_borders[0]));
202 while (out < sbr->f_tablelim + sbr->
n_lim) {
203 if (*in >= *out * lim_bands_per_octave_warped) {
205 }
else if (*in == *out ||
245 if (bs_header_extra_1) {
259 if (bs_header_extra_2) {
279 int i,
min = array[0];
280 for (i = 1; i < nel; i++)
281 min =
FFMIN(array[i], min);
285 static void make_bands(int16_t* bands,
int start,
int stop,
int num_bands)
287 int k, previous, present;
290 base =
powf((
float)stop / start, 1.0f / num_bands);
294 for (k = 0; k < num_bands-1; k++) {
297 bands[k] = present - previous;
300 bands[num_bands-1] = stop - previous;
310 if (bs_xover_band >= n_master) {
312 "Invalid bitstream, crossover band index beyond array bounds: %d\n",
323 unsigned int temp, max_qmf_subbands;
324 unsigned int start_min, stop_min;
326 const int8_t *sbr_offset_ptr;
352 case 44100:
case 48000:
case 64000:
355 case 88200:
case 96000:
case 128000:
case 176400:
case 192000:
360 "Unsupported sample rate for SBR: %d\n", sbr->
sample_rate);
367 sbr->
k[2] = stop_min;
371 sbr->
k[2] += stop_dk[k];
373 sbr->
k[2] = 2*sbr->
k[0];
375 sbr->
k[2] = 3*sbr->
k[0];
381 sbr->
k[2] =
FFMIN(64, sbr->
k[2]);
385 max_qmf_subbands = 48;
387 max_qmf_subbands = 35;
389 max_qmf_subbands = 32;
391 if (sbr->
k[2] - sbr->
k[0] > max_qmf_subbands) {
393 "Invalid bitstream, too many QMF subbands: %d\n", sbr->
k[2] - sbr->
k[0]);
401 sbr->
n_master = ((sbr->
k[2] - sbr->
k[0] + (dk&2)) >> dk) << 1;
405 for (k = 1; k <= sbr->
n_master; k++)
408 k2diff = sbr->
k[2] - sbr->
k[0] - sbr->
n_master * dk;
417 for (k = 1; k <= sbr->
n_master; k++)
422 int two_regions, num_bands_0;
423 int vdk0_max, vdk1_min;
426 if (49 * sbr->
k[2] > 110 * sbr->
k[0]) {
428 sbr->
k[1] = 2 * sbr->
k[0];
431 sbr->
k[1] = sbr->
k[2];
434 num_bands_0 =
lrintf(half_bands *
log2f(sbr->
k[1] / (
float)sbr->
k[0])) * 2;
436 if (num_bands_0 <= 0) {
446 vdk0_max = vk0[num_bands_0];
449 for (k = 1; k <= num_bands_0; k++) {
459 float invwarp = spectrum->
bs_alter_scale ? 0.76923076923076923077f
461 int num_bands_1 =
lrintf(half_bands * invwarp *
462 log2f(sbr->
k[2] / (
float)sbr->
k[1])) * 2;
468 if (vdk1_min < vdk0_max) {
471 change =
FFMIN(vdk0_max - vk1[1], (vk1[num_bands_1] - vk1[1]) >> 1);
473 vk1[num_bands_1] -= change;
479 for (k = 1; k <= num_bands_1; k++) {
487 sbr->
n_master = num_bands_0 + num_bands_1;
491 (num_bands_0 + 1) *
sizeof(sbr->
f_master[0]));
492 memcpy(&sbr->
f_master[num_bands_0 + 1], vk1 + 1,
493 num_bands_1 *
sizeof(sbr->
f_master[0]));
511 int usb = sbr->
kx[1];
516 if (goal_sb < sbr->kx[1] + sbr->
m[1]) {
517 for (k = 0; sbr->
f_master[k] < goal_sb; k++) ;
523 for (i = k; i == k || sb > (sbr->
k[0] - 1 + msb - odd); i--) {
525 odd = (sb + sbr->
k[0]) & 1;
549 }
while (sb != sbr->
kx[1] + sbr->
m[1]);
563 sbr->
n[0] = (sbr->
n[1] + 1) >> 1;
566 (sbr->
n[1] + 1) *
sizeof(sbr->
f_master[0]));
571 if (sbr->
kx[1] + sbr->
m[1] > 64) {
573 "Stop frequency border too high: %d\n", sbr->
kx[1] + sbr->
m[1]);
576 if (sbr->
kx[1] > 32) {
582 temp = sbr->
n[1] & 1;
583 for (k = 1; k <= sbr->
n[0]; k++)
587 log2f(sbr->
k[2] / (
float)sbr->
kx[1])));
595 for (k = 1; k <= sbr->
n_q; k++) {
596 temp += (sbr->
n[0] - temp) / (sbr->
n_q + 1 - k);
615 for (i = 0; i < elements; i++) {
629 unsigned bs_pointer = 0;
631 int abs_bord_trail = 16;
632 int num_rel_lead, num_rel_trail;
633 unsigned bs_num_env_old = ch_data->
bs_num_env;
648 "Invalid bitstream, too many SBR envelopes in FIXFIX type SBR frame: %d\n",
653 ch_data->
t_env[0] = 0;
656 abs_bord_trail = (abs_bord_trail + (ch_data->
bs_num_env >> 1)) /
658 for (i = 0; i < num_rel_lead; i++)
659 ch_data->
t_env[i + 1] = ch_data->
t_env[i] + abs_bord_trail;
669 ch_data->
t_env[0] = 0;
672 for (i = 0; i < num_rel_trail; i++)
687 for (i = 0; i < num_rel_lead; i++)
699 ch_data->
bs_num_env = num_rel_lead + num_rel_trail + 1;
703 "Invalid bitstream, too many SBR envelopes in VARVAR type SBR frame: %d\n",
710 for (i = 0; i < num_rel_lead; i++)
712 for (i = 0; i < num_rel_trail; i++)
724 "Invalid bitstream, bs_pointer points to a middle noise border outside the time borders table: %d\n",
730 if (ch_data->
t_env[i-1] > ch_data->
t_env[i]) {
738 ch_data->
t_q[0] = ch_data->
t_env[0];
749 else if (bs_pointer == 1)
752 idx = bs_pointer - 1;
754 ch_data->
t_q[1] = ch_data->
t_env[idx];
757 ch_data->
e_a[0] = -(ch_data->
e_a[1] != bs_num_env_old);
758 ch_data->
e_a[1] = -1;
762 ch_data->
e_a[1] = bs_pointer - 1;
776 memcpy(dst->
t_q, src->
t_q,
sizeof(dst->
t_q));
781 dst->
e_a[1] = src->
e_a[1];
799 for (i = 0; i < sbr->
n_q; i++)
808 VLC_TYPE (*t_huff)[2], (*f_huff)[2];
811 const int odd = sbr->
n[1] & 1;
847 for (j = 0; j < sbr->
n[ch_data->
bs_freq_res[i + 1]]; j++)
850 for (j = 0; j < sbr->
n[ch_data->
bs_freq_res[i + 1]]; j++) {
855 for (j = 0; j < sbr->
n[ch_data->
bs_freq_res[i + 1]]; j++) {
856 k = j ? 2*j - odd : 0;
862 for (j = 1; j < sbr->
n[ch_data->
bs_freq_res[i + 1]]; j++)
876 VLC_TYPE (*t_huff)[2], (*f_huff)[2];
894 for (j = 0; j < sbr->
n_q; j++)
898 for (j = 1; j < sbr->
n_q; j++)
910 int bs_extension_id,
int *num_bits_left)
912 switch (bs_extension_id) {
915 av_log(ac->
avctx,
AV_LOG_ERROR,
"Parametric Stereo signaled to be not-present but was found in the bitstream.\n");
1015 "Invalid bitstream - cannot apply SBR to element type %d\n", id_aac);
1020 int num_bits_left =
get_bits(gb, 4);
1021 if (num_bits_left == 15)
1024 num_bits_left <<= 3;
1025 while (num_bits_left > 7) {
1029 if (num_bits_left < 0) {
1032 if (num_bits_left > 0)
1047 "SBR reset failed. Switching SBR to pure upsampling mode.\n");
1063 unsigned int num_sbr_bits = 0, num_align_bits;
1064 unsigned bytes_read;
1081 sbr->
kx[0] = sbr->
kx[1];
1082 sbr->
m[0] = sbr->
m[1];
1095 num_align_bits = ((cnt << 3) - 4 - num_sbr_bits) & 7;
1096 bytes_read = ((num_sbr_bits + num_align_bits + 4) >> 3);
1098 if (bytes_read > cnt) {
1100 "Expected to read %d SBR bytes actually read %d.\n", cnt, bytes_read);
1118 float fac = temp1 / (1.0f + temp2);
1124 for (k = 0; k < sbr->
n_q; k++) {
1127 float fac = temp1 / (1.0f + temp2);
1133 for (ch = 0; ch < (id_aac ==
TYPE_CPE) + 1; ch++) {
1140 for (k = 0; k < sbr->
n_q; k++)
1155 float z[320],
float W[2][32][32][2],
int buf_idx)
1158 memcpy(x , x+1024, (320-32)*
sizeof(x[0]));
1159 memcpy(x+288, in, 1024*
sizeof(x[0]));
1160 for (i = 0; i < 32; i++) {
1177 float *out,
float X[2][38][64],
1178 float mdct_buf[2][64],
1179 float *
v0,
int *v_off,
const unsigned int div)
1183 const int step = 128 >> div;
1185 for (i = 0; i < 32; i++) {
1186 if (*v_off < step) {
1187 int saved_samples = (1280 - 128) >> div;
1195 for (n = 0; n < 32; n++) {
1196 X[0][i][ n] = -X[0][i][n];
1197 X[0][i][32+n] = X[1][i][31-n];
1199 mdct->
imdct_half(mdct, mdct_buf[0], X[0][i]);
1203 mdct->
imdct_half(mdct, mdct_buf[0], X[0][i]);
1204 mdct->
imdct_half(mdct, mdct_buf[1], X[1][i]);
1207 fdsp->
vector_fmul (out, v , sbr_qmf_window , 64 >> div);
1208 dsp->
vector_fmul_add(out, v + ( 192 >> div), sbr_qmf_window + ( 64 >> div), out , 64 >> div);
1209 dsp->
vector_fmul_add(out, v + ( 256 >> div), sbr_qmf_window + (128 >> div), out , 64 >> div);
1210 dsp->
vector_fmul_add(out, v + ( 448 >> div), sbr_qmf_window + (192 >> div), out , 64 >> div);
1211 dsp->
vector_fmul_add(out, v + ( 512 >> div), sbr_qmf_window + (256 >> div), out , 64 >> div);
1212 dsp->
vector_fmul_add(out, v + ( 704 >> div), sbr_qmf_window + (320 >> div), out , 64 >> div);
1213 dsp->
vector_fmul_add(out, v + ( 768 >> div), sbr_qmf_window + (384 >> div), out , 64 >> div);
1214 dsp->
vector_fmul_add(out, v + ( 960 >> div), sbr_qmf_window + (448 >> div), out , 64 >> div);
1215 dsp->
vector_fmul_add(out, v + (1024 >> div), sbr_qmf_window + (512 >> div), out , 64 >> div);
1216 dsp->
vector_fmul_add(out, v + (1216 >> div), sbr_qmf_window + (576 >> div), out , 64 >> div);
1226 float (*alpha0)[2],
float (*alpha1)[2],
1227 const float X_low[32][40][2],
int k0)
1230 for (k = 0; k < k0; k++) {
1236 dk = phi[2][1][0] * phi[1][0][0] -
1237 (phi[1][1][0] * phi[1][1][0] + phi[1][1][1] * phi[1][1][1]) / 1.000001f;
1243 float temp_real, temp_im;
1244 temp_real = phi[0][0][0] * phi[1][1][0] -
1245 phi[0][0][1] * phi[1][1][1] -
1246 phi[0][1][0] * phi[1][0][0];
1247 temp_im = phi[0][0][0] * phi[1][1][1] +
1248 phi[0][0][1] * phi[1][1][0] -
1249 phi[0][1][1] * phi[1][0][0];
1251 alpha1[k][0] = temp_real / dk;
1252 alpha1[k][1] = temp_im / dk;
1255 if (!phi[1][0][0]) {
1259 float temp_real, temp_im;
1260 temp_real = phi[0][0][0] + alpha1[k][0] * phi[1][1][0] +
1261 alpha1[k][1] * phi[1][1][1];
1262 temp_im = phi[0][0][1] + alpha1[k][1] * phi[1][1][0] -
1263 alpha1[k][0] * phi[1][1][1];
1265 alpha0[k][0] = -temp_real / phi[1][0][0];
1266 alpha0[k][1] = -temp_im / phi[1][0][0];
1269 if (alpha1[k][0] * alpha1[k][0] + alpha1[k][1] * alpha1[k][1] >= 16.0f ||
1270 alpha0[k][0] * alpha0[k][0] + alpha0[k][1] * alpha0[k][1] >= 16.0f) {
1284 static const float bw_tab[] = { 0.0f, 0.75f, 0.9f, 0.98f };
1286 for (i = 0; i < sbr->
n_q; i++) {
1292 if (new_bw < ch_data->bw_array[i]) {
1293 new_bw = 0.75f * new_bw + 0.25f * ch_data->
bw_array[i];
1295 new_bw = 0.90625f * new_bw + 0.09375f * ch_data->
bw_array[i];
1296 ch_data->
bw_array[i] = new_bw < 0.015625f ? 0.0f : new_bw;
1302 float X_low[32][40][2],
const float W[2][32][32][2],
1306 const int t_HFGen = 8;
1308 memset(X_low, 0, 32*
sizeof(*X_low));
1309 for (k = 0; k < sbr->
kx[1]; k++) {
1310 for (i = t_HFGen; i < i_f + t_HFGen; i++) {
1311 X_low[k][i][0] = W[buf_idx][i - t_HFGen][k][0];
1312 X_low[k][i][1] = W[buf_idx][i - t_HFGen][k][1];
1315 buf_idx = 1-buf_idx;
1316 for (k = 0; k < sbr->
kx[0]; k++) {
1317 for (i = 0; i < t_HFGen; i++) {
1318 X_low[k][i][0] = W[buf_idx][i + i_f - t_HFGen][k][0];
1319 X_low[k][i][1] = W[buf_idx][i + i_f - t_HFGen][k][1];
1327 float X_high[64][40][2],
const float X_low[32][40][2],
1328 const float (*alpha0)[2],
const float (*alpha1)[2],
1329 const float bw_array[5],
const uint8_t *t_env,
1344 "ERROR : no subband found for frequency %d\n", k);
1350 alpha0[p], alpha1[p], bw_array[g],
1351 2 * t_env[0], 2 * t_env[bs_num_env]);
1354 if (k < sbr->m[1] + sbr->
kx[1])
1355 memset(X_high + k, 0, (sbr->
m[1] + sbr->
kx[1] - k) *
sizeof(*X_high));
1362 const float Y0[38][64][2],
const float Y1[38][64][2],
1363 const float X_low[32][40][2],
int ch)
1368 memset(X, 0, 2*
sizeof(*X));
1369 for (k = 0; k < sbr->
kx[0]; k++) {
1370 for (i = 0; i < i_Temp; i++) {
1375 for (; k < sbr->
kx[0] + sbr->
m[0]; k++) {
1376 for (i = 0; i < i_Temp; i++) {
1377 X[0][i][k] = Y0[i + i_f][k][0];
1378 X[1][i][k] = Y0[i + i_f][k][1];
1382 for (k = 0; k < sbr->
kx[1]; k++) {
1383 for (i = i_Temp; i < 38; i++) {
1388 for (; k < sbr->
kx[1] + sbr->
m[1]; k++) {
1389 for (i = i_Temp; i < i_f; i++) {
1390 X[0][i][k] = Y1[i][k][0];
1391 X[1][i][k] = Y1[i][k][1];
1407 const unsigned int ilim = sbr->
n[ch_data->
bs_freq_res[e + 1]];
1411 if (sbr->
kx[1] != table[0]) {
1413 "Derived frequency tables were not regenerated.\n");
1417 for (i = 0; i < ilim; i++)
1418 for (m = table[i]; m < table[i + 1]; m++)
1423 for (i = 0; i < sbr->
n_q; i++)
1424 for (m = sbr->
f_tablenoise[i]; m < sbr->f_tablenoise[i + 1]; m++)
1427 for (i = 0; i < sbr->
n[1]; i++) {
1429 const unsigned int m_midpoint =
1433 (e >= e_a[1] || (ch_data->
s_indexmapped[0][m_midpoint - sbr->
kx[1]] == 1));
1437 for (i = 0; i < ilim; i++) {
1438 int additional_sinusoid_present = 0;
1439 for (m = table[i]; m < table[i + 1]; m++) {
1441 additional_sinusoid_present = 1;
1445 memset(&sbr->
s_mapped[e][table[i] - sbr->
kx[1]], additional_sinusoid_present,
1446 (table[i + 1] - table[i]) *
sizeof(sbr->
s_mapped[e][0]));
1459 int kx1 = sbr->
kx[1];
1463 const float recip_env_size = 0.5f / (ch_data->
t_env[e + 1] - ch_data->
t_env[e]);
1467 for (m = 0; m < sbr->
m[1]; m++) {
1468 float sum = sbr->
dsp.
sum_square(X_high[m+kx1] + ilb, iub - ilb);
1469 e_curr[e][m] = sum * recip_env_size;
1476 const int env_size = 2 * (ch_data->
t_env[e + 1] - ch_data->
t_env[e]);
1481 for (p = 0; p < sbr->
n[ch_data->
bs_freq_res[e + 1]]; p++) {
1483 const int den = env_size * (table[p + 1] - table[p]);
1485 for (k = table[p]; k < table[p + 1]; k++) {
1489 for (k = table[p]; k < table[p + 1]; k++) {
1490 e_curr[e][k - kx1] = sum;
1502 SBRData *ch_data,
const int e_a[2])
1506 static const float limgain[4] = { 0.70795, 1.0, 1.41254, 10000000000 };
1509 int delta = !((e == e_a[1]) || (e == e_a[0]));
1510 for (k = 0; k < sbr->
n_lim; k++) {
1511 float gain_boost, gain_max;
1512 float sum[2] = { 0.0f, 0.0f };
1513 for (m = sbr->
f_tablelim[k] - sbr->
kx[1]; m < sbr->f_tablelim[k + 1] - sbr->
kx[1]; m++) {
1515 sbr->
q_m[e][m] = sqrtf(temp * sbr->
q_mapped[e][m]);
1519 ((1.0f + sbr->
e_curr[e][m]) *
1520 (1.0f + sbr->
q_mapped[e][m] * delta)));
1523 ((1.0f + sbr->
e_curr[e][m]) *
1527 for (m = sbr->
f_tablelim[k] - sbr->
kx[1]; m < sbr->f_tablelim[k + 1] - sbr->
kx[1]; m++) {
1529 sum[1] += sbr->
e_curr[e][m];
1531 gain_max = limgain[sbr->
bs_limiter_gains] * sqrtf((FLT_EPSILON + sum[0]) / (FLT_EPSILON + sum[1]));
1532 gain_max =
FFMIN(100000.f, gain_max);
1533 for (m = sbr->
f_tablelim[k] - sbr->
kx[1]; m < sbr->f_tablelim[k + 1] - sbr->
kx[1]; m++) {
1534 float q_m_max = sbr->
q_m[e][m] * gain_max / sbr->
gain[e][m];
1538 sum[0] = sum[1] = 0.0f;
1539 for (m = sbr->
f_tablelim[k] - sbr->
kx[1]; m < sbr->f_tablelim[k + 1] - sbr->
kx[1]; m++) {
1542 + sbr->
s_m[e][m] * sbr->
s_m[e][m]
1543 + (delta && !sbr->
s_m[e][m]) * sbr->
q_m[e][m] * sbr->
q_m[e][m];
1545 gain_boost = sqrtf((FLT_EPSILON + sum[0]) / (FLT_EPSILON + sum[1]));
1546 gain_boost =
FFMIN(1.584893192f, gain_boost);
1547 for (m = sbr->
f_tablelim[k] - sbr->
kx[1]; m < sbr->f_tablelim[k + 1] - sbr->
kx[1]; m++) {
1548 sbr->
gain[e][m] *= gain_boost;
1549 sbr->
q_m[e][m] *= gain_boost;
1550 sbr->
s_m[e][m] *= gain_boost;
1558 const float X_high[64][40][2],
1564 const int kx = sbr->
kx[1];
1565 const int m_max = sbr->
m[1];
1566 static const float h_smooth[5] = {
1573 static const int8_t phi[2][4] = {
1577 float (*g_temp)[48] = ch_data->
g_temp, (*q_temp)[48] = ch_data->
q_temp;
1582 for (i = 0; i < h_SL; i++) {
1583 memcpy(g_temp[i + 2*ch_data->
t_env[0]], sbr->
gain[0], m_max *
sizeof(sbr->
gain[0][0]));
1584 memcpy(q_temp[i + 2*ch_data->
t_env[0]], sbr->
q_m[0], m_max *
sizeof(sbr->
q_m[0][0]));
1592 for (i = 2 * ch_data->
t_env[e]; i < 2 * ch_data->t_env[e + 1]; i++) {
1593 memcpy(g_temp[h_SL + i], sbr->
gain[e], m_max *
sizeof(sbr->
gain[0][0]));
1594 memcpy(q_temp[h_SL + i], sbr->
q_m[e], m_max *
sizeof(sbr->
q_m[0][0]));
1599 for (i = 2 * ch_data->
t_env[e]; i < 2 * ch_data->t_env[e + 1]; i++) {
1600 int phi_sign = (1 - 2*(kx & 1));
1603 float *g_filt, *q_filt;
1605 if (h_SL && e != e_a[0] && e != e_a[1]) {
1606 g_filt = g_filt_tab;
1607 q_filt = q_filt_tab;
1608 for (m = 0; m < m_max; m++) {
1609 const int idx1 = i + h_SL;
1612 for (j = 0; j <= h_SL; j++) {
1613 g_filt[m] += g_temp[idx1 - j][m] * h_smooth[j];
1614 q_filt[m] += q_temp[idx1 - j][m] * h_smooth[j];
1618 g_filt = g_temp[i + h_SL];
1622 sbr->
dsp.
hf_g_filt(Y1[i] + kx, X_high + kx, g_filt, m_max,
1625 if (e != e_a[0] && e != e_a[1]) {
1630 for (m = 0; m < m_max; m++) {
1632 sbr->
s_m[e][m] * phi[0][indexsine];
1634 sbr->
s_m[e][m] * (phi[1][indexsine] * phi_sign);
1635 phi_sign = -phi_sign;
1638 indexnoise = (indexnoise + m_max) & 0x1ff;
1639 indexsine = (indexsine + 1) & 3;
1651 int nch = (id_aac ==
TYPE_CPE) ? 2 : 1;
1655 sbr->
kx[0] = sbr->
kx[1];
1656 sbr->
m[0] = sbr->
m[1];
1664 for (ch = 0; ch < nch; ch++) {
1700 memcpy(sbr->
X[1], sbr->
X[0],
sizeof(sbr->
X[0]));