Browse Source

avfilter/af_asubboost: use transposed II form

tags/n4.4
Paul B Mahol 4 years ago
parent
commit
8750c8f45c
1 changed files with 14 additions and 14 deletions
  1. +14
    -14
      libavfilter/af_asubboost.c

+ 14
- 14
libavfilter/af_asubboost.c View File

@@ -40,7 +40,7 @@ typedef struct ASubBoostContext {
int write_pos;
int buffer_samples;

AVFrame *i, *o;
AVFrame *w;
AVFrame *buffer;
} ASubBoostContext;

@@ -104,9 +104,8 @@ static int config_input(AVFilterLink *inlink)
ASubBoostContext *s = ctx->priv;

s->buffer = ff_get_audio_buffer(inlink, inlink->sample_rate / 10);
s->i = ff_get_audio_buffer(inlink, 2);
s->o = ff_get_audio_buffer(inlink, 2);
if (!s->buffer || !s->i || !s->o)
s->w = ff_get_audio_buffer(inlink, 2);
if (!s->buffer || !s->w)
return AVERROR(ENOMEM);

return get_coeffs(ctx);
@@ -117,7 +116,12 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
AVFilterContext *ctx = inlink->dst;
AVFilterLink *outlink = ctx->outputs[0];
ASubBoostContext *s = ctx->priv;
const float wet = s->wet_gain, dry = s->dry_gain, feedback = s->feedback, decay = s->decay;
const double wet = s->wet_gain, dry = s->dry_gain, feedback = s->feedback, decay = s->decay;
const double b0 = s->b0;
const double b1 = s->b1;
const double b2 = s->b2;
const double a1 = -s->a1;
const double a2 = -s->a2;
int write_pos;
AVFrame *out;

@@ -136,18 +140,15 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
const double *src = (const double *)in->extended_data[ch];
double *dst = (double *)out->extended_data[ch];
double *buffer = (double *)s->buffer->extended_data[ch];
double *ix = (double *)s->i->extended_data[ch];
double *ox = (double *)s->o->extended_data[ch];
double *w = (double *)s->w->extended_data[ch];

write_pos = s->write_pos;
for (int n = 0; n < in->nb_samples; n++) {
double out_sample;

out_sample = src[n] * s->b0 + ix[0] * s->b1 + ix[1] * s->b2 - ox[0] * s->a1 - ox[1] * s->a2;
ix[1] = ix[0];
ix[0] = src[n];
ox[1] = ox[0];
ox[0] = out_sample;
out_sample = src[n] * b0 + w[0];
w[0] = b1 * src[n] + w[1] + a1 * out_sample;
w[1] = b2 * src[n] + a2 * out_sample;

buffer[write_pos] = buffer[write_pos] * decay + out_sample * feedback;
dst[n] = src[n] * dry + buffer[write_pos] * wet;
@@ -169,8 +170,7 @@ static av_cold void uninit(AVFilterContext *ctx)
ASubBoostContext *s = ctx->priv;

av_frame_free(&s->buffer);
av_frame_free(&s->i);
av_frame_free(&s->o);
av_frame_free(&s->w);
}

static int process_command(AVFilterContext *ctx, const char *cmd, const char *args,


Loading…
Cancel
Save