virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-04-08 16:54:03 +02:00
Code Issues Releases Activity

avfilter/af_anlmdn: add support for using writable frames

Browse Source
This commit is contained in:
Paul B Mahol 2022-03-10 10:33:26 +01:00
parent 0a005b1207
commit 34715fe4a2
1 changed files with 23 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
libavfilter/af_anlmdn.c
View File

@ -50,6 +50,7 @@ typedef struct AudioNLMeansContext {
int N; int N;
int H; int H;
AVFrame *in;
AVFrame *cache; AVFrame *cache;
AVFrame *window; AVFrame *window;
@ -201,6 +202,8 @@ static int filter_channel(AVFilterContext *ctx, void *arg, int ch, int nb_jobs)
AVFrame *out = arg; AVFrame *out = arg;
const int S = s->S; const int S = s->S;
const int K = s->K; const int K = s->K;
const int N = s->N;
const int H = s->H;
const int om = s->om; const int om = s->om;
const float *f = (const float *)(s->window->extended_data[ch]) + K; const float *f = (const float *)(s->window->extended_data[ch]) + K;
float *cache = (float *)s->cache->extended_data[ch]; float *cache = (float *)s->cache->extended_data[ch];
@ -209,8 +212,15 @@ static int filter_channel(AVFilterContext *ctx, void *arg, int ch, int nb_jobs)
const float *const weight_lut = s->weight_lut; const float *const weight_lut = s->weight_lut;
const float pdiff_lut_scale = s->pdiff_lut_scale; const float pdiff_lut_scale = s->pdiff_lut_scale;
const float smooth = fminf(s->m, WEIGHT_LUT_SIZE / pdiff_lut_scale); const float smooth = fminf(s->m, WEIGHT_LUT_SIZE / pdiff_lut_scale);
const int offset = N - H;
float *src = (float *)s->window->extended_data[ch];
const AVFrame *const in = s->in;
for (int i = S; i < s->H + S; i++) { memmove(src, &src[H], offset * sizeof(float));
memcpy(&src[offset], in->extended_data[ch], in->nb_samples * sizeof(float));
memset(&src[offset + in->nb_samples], 0, (H - in->nb_samples) * sizeof(float));
for (int i = S; i < H + S; i++) {
float P = 0.f, Q = 0.f; float P = 0.f, Q = 0.f;
int v = 0; int v = 0;
@ -260,25 +270,25 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
AVFilterContext *ctx = inlink->dst; AVFilterContext *ctx = inlink->dst;
AVFilterLink *outlink = ctx->outputs[0]; AVFilterLink *outlink = ctx->outputs[0];
AudioNLMeansContext *s = ctx->priv; AudioNLMeansContext *s = ctx->priv;
const int offset = s->N - s->H;
AVFrame *out; AVFrame *out;
out = ff_get_audio_buffer(outlink, in->nb_samples); if (av_frame_is_writable(in)) {
if (!out) out = in;
return AVERROR(ENOMEM); } else {
out = ff_get_audio_buffer(outlink, in->nb_samples);
if (!out) {
av_frame_free(&in);
return AVERROR(ENOMEM);
}
for (int ch = 0; ch < in->channels; ch++) { out->pts = in->pts;
float *src = (float *)s->window->extended_data[ch];
memmove(src, &src[s->H], offset * sizeof(float));
memcpy(&src[offset], in->extended_data[ch], in->nb_samples * sizeof(float));
memset(&src[offset + in->nb_samples], 0, (s->H - in->nb_samples) * sizeof(float));
} }
s->in = in;
ff_filter_execute(ctx, filter_channel, out, NULL, inlink->channels); ff_filter_execute(ctx, filter_channel, out, NULL, inlink->channels);
out->pts = in->pts; if (out != in)
av_frame_free(&in); av_frame_free(&in);
return ff_filter_frame(outlink, out); return ff_filter_frame(outlink, out);
} }