diff options
| author | Paul B Mahol <onemda@gmail.com> | 2020-11-25 13:36:57 +0100 |
|---|---|---|
| committer | Paul B Mahol <onemda@gmail.com> | 2020-11-25 13:41:52 +0100 |
| commit | 5b1ccd748a7c5317172981151a36711772455368 (patch) | |
| tree | 736bcba92bc8950f17d9d43ad7ec9bc6e0cda983 | |
| parent | 3de3d2f5e24985bfd53b9b85122079a98816e259 (diff) | |
| download | ffmpeg-5b1ccd748a7c5317172981151a36711772455368.tar.gz | |
avfilter/af_arnndn: add timeline support
| -rw-r--r-- | libavfilter/af_arnndn.c | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/libavfilter/af_arnndn.c b/libavfilter/af_arnndn.c index ba25710c46..00a509477c 100644 --- a/libavfilter/af_arnndn.c +++ b/libavfilter/af_arnndn.c @@ -1338,7 +1338,8 @@ static void compute_rnn(AudioRNNContext *s, RNNState *rnn, float *gains, float * compute_dense(rnn->model->denoise_output, gains, rnn->denoise_gru_state); } -static float rnnoise_channel(AudioRNNContext *s, DenoiseState *st, float *out, const float *in) +static float rnnoise_channel(AudioRNNContext *s, DenoiseState *st, float *out, const float *in, + int disabled) { AVComplexFloat X[FREQ_SIZE]; AVComplexFloat P[WINDOW_SIZE]; @@ -1356,7 +1357,7 @@ static float rnnoise_channel(AudioRNNContext *s, DenoiseState *st, float *out, c biquad(x, st->mem_hp_x, in, b_hp, a_hp, FRAME_SIZE); silence = compute_frame_features(s, st, X, P, Ex, Ep, Exp, features, x); - if (!silence) { + if (!silence && !disabled) { compute_rnn(s, &st->rnn, g, &vad_prob, features); pitch_filter(X, P, Ex, Ep, Exp, g); for (int i = 0; i < NB_BANDS; i++) { @@ -1395,7 +1396,8 @@ static int rnnoise_channels(AVFilterContext *ctx, void *arg, int jobnr, int nb_j for (int ch = start; ch < end; ch++) { rnnoise_channel(s, &s->st[ch], (float *)out->extended_data[ch], - (const float *)in->extended_data[ch]); + (const float *)in->extended_data[ch], + ctx->is_disabled); } return 0; @@ -1540,5 +1542,6 @@ AVFilter ff_af_arnndn = { .uninit = uninit, .inputs = inputs, .outputs = outputs, - .flags = AVFILTER_FLAG_SLICE_THREADS, + .flags = AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL | + AVFILTER_FLAG_SLICE_THREADS, }; |