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

avfilter/af_crossfeed: always return same number of samples with block processing

This commit is contained in:
Paul B Mahol 2022-05-13 11:04:36 +02:00
parent 52a14b8505
commit 2c77d9150d
1 changed files with 59 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
libavfilter/af_crossfeed.c
View File

@ -40,6 +40,9 @@ typedef struct CrossfeedContext {
double w1, w2;
int64_t pts;
int nb_samples;
double *mid;
double *side[3];
} CrossfeedContext;
@ -126,7 +129,7 @@ static void filter_samples(double *dst, const double *src,
*sw2 = w2;
}
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
static int filter_frame(AVFilterLink *inlink, AVFrame *in, int eof)
{
AVFilterContext *ctx = inlink->dst;
AVFilterLink *outlink = ctx->outputs[0];
@ -140,12 +143,13 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
const double a1 = -s->a1;
const double a2 = -s->a2;
AVFrame *out;
int drop = 0;
double *dst;
if (av_frame_is_writable(in)) {
if (av_frame_is_writable(in) && s->block_samples == 0) {
out = in;
} else {
out = ff_get_audio_buffer(outlink, in->nb_samples);
out = ff_get_audio_buffer(outlink, s->block_samples > 0 ? s->block_samples : in->nb_samples);
if (!out) {
av_frame_free(&in);
return AVERROR(ENOMEM);
@ -154,6 +158,9 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
}
dst = (double *)out->data[0];
if (s->block_samples > 0 && s->pts == AV_NOPTS_VALUE)
drop = 1;
if (s->block_samples == 0) {
double w1 = s->w1;
double w2 = s->w2;
@ -177,6 +184,20 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
s->w1 = w1;
s->w2 = w2;
} else if (eof) {
const double *src = (const double *)in->data[0];
double *ssrc = s->side[1] + s->block_samples;
double *msrc = s->mid;
for (int n = 0; n < out->nb_samples; n++, src += 2, dst += 2) {
if (ctx->is_disabled) {
dst[0] = src[0];
dst[1] = src[1];
} else {
dst[0] = (msrc[n] + ssrc[n]) * level_out;
dst[1] = (msrc[n] - ssrc[n]) * level_out;
}
}
} else {
double *mdst = s->mid + s->block_samples;
double *sdst = s->side[0] + s->block_samples;
@ -209,7 +230,7 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
b0, b1, b2, a1, a2,
&w1, &w2);
reverse_samples(s->side[1], s->side[2] + s->block_samples, s->block_samples);
reverse_samples(s->side[1], s->side[2], s->block_samples * 2);
src = (const double *)in->data[0];
ssrc = s->side[1];
@ -229,9 +250,25 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
s->block_samples * sizeof(*s->side[0]));
}
if (s->block_samples > 0) {
int nb_samples = in->nb_samples;
int64_t pts = in->pts;
out->pts = s->pts;
out->nb_samples = s->nb_samples;
s->pts = pts;
s->nb_samples = nb_samples;
}
if (out != in)
av_frame_free(&in);
return ff_filter_frame(outlink, out);
if (!drop) {
return ff_filter_frame(outlink, out);
} else {
av_frame_free(&out);
ff_filter_set_ready(ctx, 10);
return 0;
}
}
static int activate(AVFilterContext *ctx)
@ -240,6 +277,8 @@ static int activate(AVFilterContext *ctx)
AVFilterLink *outlink = ctx->outputs[0];
CrossfeedContext *s = ctx->priv;
AVFrame *in = NULL;
int64_t pts;
int status;
int ret;
FF_FILTER_FORWARD_STATUS_BACK(outlink, inlink);
@ -252,14 +291,27 @@ static int activate(AVFilterContext *ctx)
if (ret < 0)
return ret;
if (ret > 0)
return filter_frame(inlink, in);
return filter_frame(inlink, in, 0);
if (s->block_samples > 0 && ff_inlink_queued_samples(inlink) >= s->block_samples) {
ff_filter_set_ready(ctx, 10);
return 0;
}
FF_FILTER_FORWARD_STATUS(inlink, outlink);
if (ff_inlink_acknowledge_status(inlink, &status, &pts)) {
if (s->block_samples > 0) {
AVFrame *in = ff_get_audio_buffer(outlink, s->block_samples);
if (!in)
return AVERROR(ENOMEM);
ret = filter_frame(inlink, in, 1);
}
ff_outlink_set_status(outlink, status, pts);
return ret;
}
FF_FILTER_FORWARD_WANTED(outlink, inlink);
return FFERROR_NOT_READY;