| @@ -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, | |||