RepoMirrors
/
ffmpeg
mirror of https://git.ffmpeg.org/ffmpeg.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

avfilter/af_aiir: improve response calculation with zp coefficients

This commit is contained in:
Paul B Mahol 2020-05-30 09:47:12 +02:00
parent e2e8121eaa
commit 3fc7b01c52
1 changed files with 23 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
libavfilter/af_aiir.c
View File

@ -824,9 +824,14 @@ static void draw_line(AVFrame *out, int x0, int y0, int x1, int y1, uint32_t col
} }
} }
static double distance(double x0, double x1, double y0, double y1)
{
return hypot(x0 - x1, y0 - y1);
}
static void get_response(int channel, int format, double w, static void get_response(int channel, int format, double w,
const double *b, const double *a, const double *b, const double *a,
int nb_b, int nb_a, double *r, double *i) int nb_b, int nb_a, double *magnitude, double *phase)
{ {
double realz, realp; double realz, realp;
double imagz, imagp; double imagz, imagp;
@ -849,39 +854,26 @@ static void get_response(int channel, int format, double w,
div = realp * realp + imagp * imagp; div = realp * realp + imagp * imagp;
real = (realz * realp + imagz * imagp) / div; real = (realz * realp + imagz * imagp) / div;
imag = (imagz * realp - imagp * realz) / div; imag = (imagz * realp - imagp * realz) / div;
*magnitude = hypot(real, imag);
*phase = atan2(imag, real);
} else { } else {
real = 1; double p = 1., z = 1.;
imag = 0; double acc = 0.;
for (int x = 0; x < nb_a; x++) { for (int x = 0; x < nb_a; x++) {
double ore, oim, re, im; z *= distance(cos(w), a[2 * x], sin(w), a[2 * x + 1]);
acc += atan2(sin(w) - a[2 * x + 1], cos(w) - a[2 * x]);
re = cos(w) - a[2 * x];
im = sin(w) - a[2 * x + 1];
ore = real;
oim = imag;
real = ore * re - oim * im;
imag = ore * im + oim * re;
} }
for (int x = 0; x < nb_b; x++) { for (int x = 0; x < nb_b; x++) {
double ore, oim, re, im; p *= distance(cos(w), b[2 * x], sin(w), b[2 * x + 1]);
acc -= atan2(sin(w) - b[2 * x + 1], cos(w) - b[2 * x]);
re = cos(w) - b[2 * x];
im = sin(w) - b[2 * x + 1];
ore = real;
oim = imag;
div = re * re + im * im;
real = (ore * re + oim * im) / div;
imag = (oim * re - ore * im) / div;
}
} }
*r = real; *magnitude = z / p;
*i = imag; *phase = acc;
}
} }
static void draw_response(AVFilterContext *ctx, AVFrame *out, int sample_rate) static void draw_response(AVFilterContext *ctx, AVFrame *out, int sample_rate)
@ -909,12 +901,12 @@ static void draw_response(AVFilterContext *ctx, AVFrame *out, int sample_rate)
const int nb_b = s->iir[ch].nb_ab[0]; const int nb_b = s->iir[ch].nb_ab[0];
const int nb_a = s->iir[ch].nb_ab[1]; const int nb_a = s->iir[ch].nb_ab[1];
double w = i * M_PI / (s->w - 1); double w = i * M_PI / (s->w - 1);
double real, imag; double m, p;
get_response(ch, s->format, w, b, a, nb_b, nb_a, &real, &imag); get_response(ch, s->format, w, b, a, nb_b, nb_a, &m, &p);
mag[i] = s->iir[ch].g * hypot(real, imag); mag[i] = s->iir[ch].g * m;
phase[i] = atan2(imag, real); phase[i] = p;
min = fmin(min, mag[i]); min = fmin(min, mag[i]);
max = fmax(max, mag[i]); max = fmax(max, mag[i]);
} }