highperfocused/ffmpeg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

avfilter/af_afftdn: remove special handling for first and last bin

Browse Source
This commit is contained in:
Paul B Mahol
2022-03-10 15:34:45 +01:00
parent ea777333de
commit e7caa18b4a
1 changed files with 4 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
libavfilter/af_afftdn.c
View File

@@ -309,20 +309,9 @@ static void process_frame(AudioFFTDeNoiseContext *s, DeNoiseChannel *dnch,
double *prior, double *prior_band_excit, int track_noise) double *prior, double *prior_band_excit, int track_noise)
{ {
double d1, d2, d3, gain; double d1, d2, d3, gain;
int n, i1; int n = 0, i1;
d1 = fft_data[0].re * fft_data[0].re; for (int i = 0; i < s->fft_length2; i++) {
dnch->noisy_data[0] = d1;
d2 = d1 / dnch->abs_var[0];
d3 = RATIO * prior[0] + RRATIO * fmax(d2 - 1.0, 0.0);
gain = d3 / (1.0 + d3);
gain *= (gain + M_PI_4 / fmax(d2, 1.0E-6));
prior[0] = (d2 * gain);
dnch->clean_data[0] = (d1 * gain);
gain = sqrt(gain);
dnch->gain[0] = gain;
n = 0;
for (int i = 1; i < s->fft_length2; i++) {
d1 = fft_data[i].re * fft_data[i].re + fft_data[i].im * fft_data[i].im; d1 = fft_data[i].re * fft_data[i].re + fft_data[i].im * fft_data[i].im;
if (d1 > s->sample_floor) if (d1 > s->sample_floor)
n = i; n = i;
@@ -337,19 +326,7 @@ static void process_frame(AudioFFTDeNoiseContext *s, DeNoiseChannel *dnch,
gain = sqrt(gain); gain = sqrt(gain);
dnch->gain[i] = gain; dnch->gain[i] = gain;
} }
d1 = fft_data[0].im * fft_data[0].im;
if (d1 > s->sample_floor)
n = s->fft_length2;
dnch->noisy_data[s->fft_length2] = d1;
d2 = d1 / dnch->abs_var[s->fft_length2];
d3 = RATIO * prior[s->fft_length2] + RRATIO * fmax(d2 - 1.0, 0.0);
gain = d3 / (1.0 + d3);
gain *= gain + M_PI_4 / fmax(d2, 1.0E-6);
prior[s->fft_length2] = d2 * gain;
dnch->clean_data[s->fft_length2] = d1 * gain;
gain = sqrt(gain);
dnch->gain[s->fft_length2] = gain;
if (n > s->fft_length2 - 2) { if (n > s->fft_length2 - 2) {
n = s->bin_count; n = s->bin_count;
i1 = s->noise_band_count; i1 = s->noise_band_count;
@@ -472,7 +449,7 @@ static void process_frame(AudioFFTDeNoiseContext *s, DeNoiseChannel *dnch,
dnch->gain[s->fft_length2] = limit_gain(dnch->gain[s->fft_length2], s->max_gain); dnch->gain[s->fft_length2] = limit_gain(dnch->gain[s->fft_length2], s->max_gain);
} }
for (int i = 1; i < s->fft_length2; i++) { for (int i = 0; i < s->fft_length2; i++) {
if (dnch->amt[i] > dnch->abs_var[i]) { if (dnch->amt[i] > dnch->abs_var[i]) {
dnch->gain[i] = 1.0; dnch->gain[i] = 1.0;
} else if (dnch->amt[i] > dnch->min_abs_var[i]) { } else if (dnch->amt[i] > dnch->min_abs_var[i]) {
@@ -483,13 +460,7 @@ static void process_frame(AudioFFTDeNoiseContext *s, DeNoiseChannel *dnch,
} }
} }
gain = dnch->gain[0]; for (int i = 0; i < s->fft_length2; i++) {
dnch->clean_data[0] = (gain * gain * dnch->noisy_data[0]);
fft_data[0].re *= gain;
gain = dnch->gain[s->fft_length2];
dnch->clean_data[s->fft_length2] = (gain * gain * dnch->noisy_data[s->fft_length2]);
fft_data[0].im *= gain;
for (int i = 1; i < s->fft_length2; i++) {
gain = dnch->gain[i]; gain = dnch->gain[i];
dnch->clean_data[i] = (gain * gain * dnch->noisy_data[i]); dnch->clean_data[i] = (gain * gain * dnch->noisy_data[i]);
fft_data[i].re *= gain; fft_data[i].re *= gain;
@@ -911,10 +882,6 @@ static void sample_noise_block(AudioFFTDeNoiseContext *s,
k = 0; k = 0;
n = j; n = j;
edgemax = fmin(s->fft_length2, s->noise_band_edge[15]); edgemax = fmin(s->fft_length2, s->noise_band_edge[15]);
dnch->fft_out[s->fft_length2].re = dnch->fft_out[0].im;
dnch->fft_out[0].im = 0.0;
dnch->fft_out[s->fft_length2].im = 0.0;
for (int i = j; i <= edgemax; i++) { for (int i = j; i <= edgemax; i++) {
if ((i == j) && (i < edgemax)) { if ((i == j) && (i < edgemax)) {
if (j > edge) { if (j > edge) {