aresample: add code to flush the internal swr buffer.
Inspired-by code from af_resample.c written by Anton Khirnov Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
parent
b3e1b95afa
commit
847943bc51
@ -36,6 +36,7 @@
|
|||||||
typedef struct {
|
typedef struct {
|
||||||
double ratio;
|
double ratio;
|
||||||
struct SwrContext *swr;
|
struct SwrContext *swr;
|
||||||
|
int64_t next_pts;
|
||||||
} AResampleContext;
|
} AResampleContext;
|
||||||
|
|
||||||
static av_cold int init(AVFilterContext *ctx, const char *args, void *opaque)
|
static av_cold int init(AVFilterContext *ctx, const char *args, void *opaque)
|
||||||
@ -44,6 +45,7 @@ static av_cold int init(AVFilterContext *ctx, const char *args, void *opaque)
|
|||||||
int ret = 0;
|
int ret = 0;
|
||||||
char *argd = av_strdup(args);
|
char *argd = av_strdup(args);
|
||||||
|
|
||||||
|
aresample->next_pts = AV_NOPTS_VALUE;
|
||||||
aresample->swr = swr_alloc();
|
aresample->swr = swr_alloc();
|
||||||
if (!aresample->swr)
|
if (!aresample->swr)
|
||||||
return AVERROR(ENOMEM);
|
return AVERROR(ENOMEM);
|
||||||
@ -176,15 +178,54 @@ static void filter_samples(AVFilterLink *inlink, AVFilterBufferRef *insamplesref
|
|||||||
}
|
}
|
||||||
|
|
||||||
avfilter_copy_buffer_ref_props(outsamplesref, insamplesref);
|
avfilter_copy_buffer_ref_props(outsamplesref, insamplesref);
|
||||||
|
|
||||||
outsamplesref->audio->sample_rate = outlink->sample_rate;
|
outsamplesref->audio->sample_rate = outlink->sample_rate;
|
||||||
outsamplesref->audio->nb_samples = n_out;
|
outsamplesref->audio->nb_samples = n_out;
|
||||||
outsamplesref->pts = insamplesref->pts == AV_NOPTS_VALUE ? AV_NOPTS_VALUE :
|
|
||||||
av_rescale_q(insamplesref->pts, inlink->time_base, outlink->time_base);
|
if(insamplesref->pts != AV_NOPTS_VALUE) {
|
||||||
|
aresample->next_pts = insamplesref->pts;
|
||||||
|
outsamplesref->pts = av_rescale_q(insamplesref->pts, inlink->time_base, outlink->time_base);
|
||||||
|
} else{
|
||||||
|
outsamplesref->pts = AV_NOPTS_VALUE; //aresample->next_pts;
|
||||||
|
}
|
||||||
|
if(aresample->next_pts != AV_NOPTS_VALUE)
|
||||||
|
aresample->next_pts += av_rescale_q(n_out, (AVRational){1 ,outlink->sample_rate}, outlink->time_base);
|
||||||
|
|
||||||
ff_filter_samples(outlink, outsamplesref);
|
ff_filter_samples(outlink, outsamplesref);
|
||||||
avfilter_unref_buffer(insamplesref);
|
avfilter_unref_buffer(insamplesref);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static int request_frame(AVFilterLink *outlink)
|
||||||
|
{
|
||||||
|
AVFilterContext *ctx = outlink->src;
|
||||||
|
AResampleContext *aresample = ctx->priv;
|
||||||
|
int ret = avfilter_request_frame(ctx->inputs[0]);
|
||||||
|
|
||||||
|
if (ret == AVERROR_EOF) {
|
||||||
|
AVFilterBufferRef *outsamplesref;
|
||||||
|
int n_out = 4096;
|
||||||
|
|
||||||
|
outsamplesref = ff_get_audio_buffer(outlink, AV_PERM_WRITE, n_out);
|
||||||
|
if (!outsamplesref)
|
||||||
|
return AVERROR(ENOMEM);
|
||||||
|
n_out = swr_convert(aresample->swr, outsamplesref->data, n_out, 0, 0);
|
||||||
|
if (n_out <= 0) {
|
||||||
|
avfilter_unref_buffer(outsamplesref);
|
||||||
|
return (n_out == 0) ? AVERROR_EOF : n_out;
|
||||||
|
}
|
||||||
|
|
||||||
|
outsamplesref->audio->sample_rate = outlink->sample_rate;
|
||||||
|
outsamplesref->audio->nb_samples = n_out;
|
||||||
|
outsamplesref->pts = aresample->next_pts;
|
||||||
|
if(aresample->next_pts != AV_NOPTS_VALUE)
|
||||||
|
aresample->next_pts += av_rescale_q(n_out, (AVRational){1 ,outlink->sample_rate}, outlink->time_base);
|
||||||
|
|
||||||
|
ff_filter_samples(outlink, outsamplesref);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
AVFilter avfilter_af_aresample = {
|
AVFilter avfilter_af_aresample = {
|
||||||
.name = "aresample",
|
.name = "aresample",
|
||||||
.description = NULL_IF_CONFIG_SMALL("Resample audio data."),
|
.description = NULL_IF_CONFIG_SMALL("Resample audio data."),
|
||||||
@ -200,6 +241,7 @@ AVFilter avfilter_af_aresample = {
|
|||||||
{ .name = NULL}},
|
{ .name = NULL}},
|
||||||
.outputs = (const AVFilterPad[]) {{ .name = "default",
|
.outputs = (const AVFilterPad[]) {{ .name = "default",
|
||||||
.config_props = config_output,
|
.config_props = config_output,
|
||||||
|
.request_frame = request_frame,
|
||||||
.type = AVMEDIA_TYPE_AUDIO, },
|
.type = AVMEDIA_TYPE_AUDIO, },
|
||||||
{ .name = NULL}},
|
{ .name = NULL}},
|
||||||
};
|
};
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
afd309546b14cff772f3f28ee650452f *./tests/data/acodec/g723_1.tco
|
93fcff0367883ca8e75b3063c527a2ce *./tests/data/acodec/g723_1.tco
|
||||||
4800 ./tests/data/acodec/g723_1.tco
|
4824 ./tests/data/acodec/g723_1.tco
|
||||||
99030194774ea673817a56f52a04843d *./tests/data/g723_1.acodec.out.wav
|
9f28820dc27cf207a15b2048789853cd *./tests/data/g723_1.acodec.out.wav
|
||||||
stddev: 8503.56 PSNR: 17.74 MAXDIFF:26473 bytes: 96000/ 1058400
|
stddev: 8502.50 PSNR: 17.74 MAXDIFF:26473 bytes: 96480/ 1058400
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
3fe1e3c0feeb3963685e07c75d136ed0 *./tests/data/acodec/roqaudio.roq
|
c8ff13cf7ebece23af76502f5785202e *./tests/data/acodec/roqaudio.roq
|
||||||
265992 ./tests/data/acodec/roqaudio.roq
|
265992 ./tests/data/acodec/roqaudio.roq
|
||||||
f27d1906e28e80f0955b75cc4ffe3601 *./tests/data/roqaudio.acodec.out.wav
|
709fd60aea880c73b375094ab5307c77 *./tests/data/roqaudio.acodec.out.wav
|
||||||
stddev: 4610.92 PSNR: 23.05 MAXDIFF:43883 bytes: 1058336/ 1058400
|
stddev: 4610.71 PSNR: 23.05 MAXDIFF:43883 bytes: 1058400/ 1058400
|
||||||
|
Loading…
x
Reference in New Issue
Block a user