40 #define AMR_USE_16BIT_TABLES
67 float prediction_error[4];
81 float hpf_31_mem[2], hpf_400_mem[2];
113 for (i = 0; i < 4; i++)
151 for (i = 0; i < 9; i++)
152 isf_q[i] =
dico1_isf[ind[0]][i] * (1.0f / (1 << 15));
154 for (i = 0; i < 7; i++)
155 isf_q[i + 9] =
dico2_isf[ind[1]][i] * (1.0f / (1 << 15));
157 for (i = 0; i < 5; i++)
160 for (i = 0; i < 4; i++)
163 for (i = 0; i < 7; i++)
178 for (i = 0; i < 9; i++)
179 isf_q[i] =
dico1_isf[ind[0]][i] * (1.0f / (1 << 15));
181 for (i = 0; i < 7; i++)
182 isf_q[i + 9] =
dico2_isf[ind[1]][i] * (1.0f / (1 << 15));
184 for (i = 0; i < 3; i++)
185 isf_q[i] +=
dico21_isf[ind[2]][i] * (1.0f / (1 << 15));
187 for (i = 0; i < 3; i++)
188 isf_q[i + 3] +=
dico22_isf[ind[3]][i] * (1.0f / (1 << 15));
190 for (i = 0; i < 3; i++)
191 isf_q[i + 6] +=
dico23_isf[ind[4]][i] * (1.0f / (1 << 15));
193 for (i = 0; i < 3; i++)
194 isf_q[i + 9] +=
dico24_isf[ind[5]][i] * (1.0f / (1 << 15));
196 for (i = 0; i < 4; i++)
197 isf_q[i + 12] +=
dico25_isf[ind[6]][i] * (1.0f / (1 << 15));
215 isf_q[i] +=
isf_mean[i] * (1.0f / (1 << 15));
232 for (k = 0; k < 3; k++) {
235 isp_q[k][i] = (1.0 - c) * isp4_past[i] + c * isp_q[3][i];
251 uint8_t *base_lag_int,
int subframe)
253 if (subframe == 0 || subframe == 2) {
254 if (pitch_index < 376) {
255 *lag_int = (pitch_index + 137) >> 2;
256 *lag_frac = pitch_index - (*lag_int << 2) + 136;
257 }
else if (pitch_index < 440) {
258 *lag_int = (pitch_index + 257 - 376) >> 1;
259 *lag_frac = (pitch_index - (*lag_int << 1) + 256 - 376) << 1;
262 *lag_int = pitch_index - 280;
266 *base_lag_int = av_clip(*lag_int - 8 - (*lag_frac < 0),
272 *lag_int = (pitch_index + 1) >> 2;
273 *lag_frac = pitch_index - (*lag_int << 2);
274 *lag_int += *base_lag_int;
284 uint8_t *base_lag_int,
int subframe,
enum Mode mode)
286 if (subframe == 0 || (subframe == 2 && mode !=
MODE_6k60)) {
287 if (pitch_index < 116) {
288 *lag_int = (pitch_index + 69) >> 1;
289 *lag_frac = (pitch_index - (*lag_int << 1) + 68) << 1;
291 *lag_int = pitch_index - 24;
295 *base_lag_int = av_clip(*lag_int - 8 - (*lag_frac < 0),
298 *lag_int = (pitch_index + 1) >> 1;
299 *lag_frac = (pitch_index - (*lag_int << 1)) << 1;
300 *lag_int += *base_lag_int;
316 int pitch_lag_int, pitch_lag_frac;
329 pitch_lag_int += pitch_lag_frac > 0;
335 pitch_lag_frac + (pitch_lag_frac > 0 ? 0 : 4),
340 if (amr_subframe->
ltp) {
344 ctx->
pitch_vector[i] = 0.18 * exc[i - 1] + 0.64 * exc[i] +
346 memcpy(exc, ctx->
pitch_vector, AMRWB_SFR_SIZE *
sizeof(
float));
351 #define BIT_STR(x,lsb,len) (((x) >> (lsb)) & ((1 << (len)) - 1))
354 #define BIT_POS(x, p) (((x) >> (p)) & 1)
373 out[0] =
BIT_POS(code, m) ? -pos : pos;
381 out[0] =
BIT_POS(code, 2*m) ? -pos0 : pos0;
382 out[1] =
BIT_POS(code, 2*m) ? -pos1 : pos1;
383 out[1] = pos0 > pos1 ? -out[1] : out[1];
388 int half_2p =
BIT_POS(code, 2*m - 1) << (m - 1);
391 m - 1, off + half_2p);
397 int half_4p, subhalf_2p;
398 int b_offset = 1 << (m - 1);
400 switch (
BIT_STR(code, 4*m - 2, 2)) {
402 half_4p =
BIT_POS(code, 4*m - 3) << (m - 1);
403 subhalf_2p =
BIT_POS(code, 2*m - 3) << (m - 2);
406 m - 2, off + half_4p + subhalf_2p);
408 m - 1, off + half_4p);
414 m - 1, off + b_offset);
420 m - 1, off + b_offset);
426 m - 1, off + b_offset);
433 int half_3p =
BIT_POS(code, 5*m - 1) << (m - 1);
436 m - 1, off + half_3p);
443 int b_offset = 1 << (m - 1);
445 int half_more =
BIT_POS(code, 6*m - 5) << (m - 1);
446 int half_other = b_offset - half_more;
448 switch (
BIT_STR(code, 6*m - 4, 2)) {
451 m - 1, off + half_more);
453 m - 1, off + half_more);
457 m - 1, off + half_other);
459 m - 1, off + half_more);
463 m - 1, off + half_other);
465 m - 1, off + half_more);
471 m - 1, off + b_offset);
486 const uint16_t *pulse_lo,
const enum Mode mode)
491 int spacing = (mode ==
MODE_6k60) ? 2 : 4;
496 for (i = 0; i < 2; i++)
500 for (i = 0; i < 4; i++)
504 for (i = 0; i < 4; i++)
508 for (i = 0; i < 2; i++)
510 for (i = 2; i < 4; i++)
514 for (i = 0; i < 4; i++)
518 for (i = 0; i < 4; i++)
520 ((int) pulse_hi[i] << 14), 4, 1);
523 for (i = 0; i < 2; i++)
525 ((int) pulse_hi[i] << 10), 4, 1);
526 for (i = 2; i < 4; i++)
528 ((int) pulse_hi[i] << 14), 4, 1);
532 for (i = 0; i < 4; i++)
534 ((int) pulse_hi[i] << 11), 4, 1);
540 for (i = 0; i < 4; i++)
542 int pos = (
FFABS(sig_pos[i][j]) - 1) * spacing + i;
544 fixed_vector[pos] += sig_pos[i][j] < 0 ? -1.0 : 1.0;
557 float *fixed_gain_factor,
float *pitch_gain)
562 *pitch_gain = gains[0] * (1.0f / (1 << 14));
563 *fixed_gain_factor = gains[1] * (1.0f / (1 << 11));
580 fixed_vector[i] -= fixed_vector[i - 1] * ctx->
tilt_coef;
584 fixed_vector[i] += fixed_vector[i - ctx->
pitch_lag_int] * 0.85;
596 float *f_vector,
float f_gain)
605 return (p_ener - f_ener) / (p_ener + f_ener);
619 float *fixed_vector,
float *buf)
635 if (ir_filter_nr < 2)
640 for (i = 0; i < 6; i++)
656 if (ir_filter_nr < 2) {
688 acc += (isf[i] - isf_past[i]) * (isf[i] - isf_past[i]);
692 return FFMAX(0.0, 1.25 - acc * 0.8 * 512);
707 float voice_fac,
float stab_fac)
709 float sm_fac = 0.5 * (1 - voice_fac) * stab_fac;
715 if (fixed_gain < *prev_tr_gain) {
716 g0 =
FFMIN(*prev_tr_gain, fixed_gain + fixed_gain *
717 (6226 * (1.0f / (1 << 15))));
719 g0 =
FFMAX(*prev_tr_gain, fixed_gain *
720 (27536 * (1.0f / (1 << 15))));
724 return sm_fac * g0 + (1 - sm_fac) * fixed_gain;
736 float cpe = 0.125 * (1 + voice_fac);
737 float last = fixed_vector[0];
739 fixed_vector[0] -= cpe * fixed_vector[1];
742 float cur = fixed_vector[i];
744 fixed_vector[i] -= cpe * (last + fixed_vector[i + 1]);
748 fixed_vector[AMRWB_SFR_SIZE - 1] -= cpe * last;
762 float fixed_gain,
const float *fixed_vector,
782 energy, AMRWB_SFR_SIZE);
798 static void de_emphasis(
float *out,
float *in,
float m,
float mem[1])
802 out[0] = in[0] + m * mem[0];
805 out[i] = in[i] + out[i - 1] * m;
807 mem[0] = out[AMRWB_SFR_SIZE - 1];
822 int int_part = 0, frac_part;
825 for (j = 0; j < o_size / 5; j++) {
826 out[i] = in[int_part];
830 for (k = 1; k < 5; k++) {
863 return av_clipf((1.0 -
FFMAX(0.0, tilt)) * (1.25 - 0.25 * wsp), 0.1, 1.0);
876 const float *synth_exc,
float hb_gain)
886 energy * hb_gain * hb_gain,
898 for (i = 7; i <
LP_ORDER - 2; i++) {
899 float prod = (diff_isf[i] - mean) * (diff_isf[i - lag] - mean);
914 float diff_isf[
LP_ORDER - 2], diff_mean;
917 int i, j, i_max_corr;
919 isf[LP_ORDER_16k - 1] = isf[
LP_ORDER - 1];
923 diff_isf[i] = isf[i + 1] - isf[i];
926 for (i = 2; i < LP_ORDER - 2; i++)
927 diff_mean += diff_isf[i] * (1.0f / (LP_ORDER - 4));
931 for (i = 0; i < 3; i++) {
934 if (corr_lag[i] > corr_lag[i_max_corr])
939 for (i = LP_ORDER - 1; i < LP_ORDER_16k - 1; i++)
940 isf[i] = isf[i - 1] + isf[i - 1 - i_max_corr]
941 - isf[i - 2 - i_max_corr];
944 est = 7965 + (isf[2] - isf[3] - isf[4]) / 6.0;
945 scale = 0.5 * (
FFMIN(est, 7600) - isf[LP_ORDER - 2]) /
946 (isf[LP_ORDER_16k - 2] - isf[LP_ORDER - 2]);
948 for (i = LP_ORDER - 1, j = 0; i < LP_ORDER_16k - 1; i++, j++)
949 diff_isf[j] = scale * (isf[i] - isf[i - 1]);
952 for (i = 1; i < LP_ORDER_16k -
LP_ORDER; i++)
953 if (diff_isf[i] + diff_isf[i - 1] < 5.0) {
954 if (diff_isf[i] > diff_isf[i - 1]) {
955 diff_isf[i - 1] = 5.0 - diff_isf[i];
957 diff_isf[i] = 5.0 - diff_isf[i - 1];
960 for (i = LP_ORDER - 1, j = 0; i < LP_ORDER_16k - 1; i++, j++)
961 isf[i] = isf[i - 1] + diff_isf[j] * (1.0f / (1 << 15));
964 for (i = 0; i < LP_ORDER_16k - 1; i++)
982 for (i = 0; i <
size; i++) {
983 out[i] = lpc[i] * fac;
1000 const float *exc,
const float *isf,
const float *isf_past)
1044 memcpy(data, mem, HB_FIR_SIZE *
sizeof(
float));
1050 out[i] += data[i + j] * fir_coef[j];
1053 memcpy(mem, data + AMRWB_SFR_SIZE_16k, HB_FIR_SIZE *
sizeof(
float));
1076 int *got_frame_ptr,
AVPacket *avpkt)
1081 int buf_size = avpkt->
size;
1082 int expected_fr_size, header_size;
1085 float fixed_gain_factor;
1086 float *synth_fixed_vector;
1087 float synth_fixed_gain;
1088 float voice_fac, stab_fac;
1111 if (buf_size < expected_fr_size) {
1113 "Frame too small (%d bytes). Truncated file?\n", buf_size);
1151 for (sub = 0; sub < 4; sub++)
1154 for (sub = 0; sub < 4; sub++) {
1181 ctx->
tilt_coef = voice_fac * 0.25 + 0.25;
1192 voice_fac, stab_fac);
1211 AMRWB_SFR_SIZE_16k);
1236 sub_buf[i] = (sub_buf[i] + hb_samples[i]) * (1.0f / (1 << 15));
1249 return expected_fr_size;