mirror of https://git.ffmpeg.org/ffmpeg.git
lavfi: use avfilter_get_audio_buffer_ref_from_arrays() in avfilter_default_get_audio_buffer
This commit is contained in:
parent
e727bca392
commit
6735534f19
|
@ -61,78 +61,32 @@ AVFilterBufferRef *avfilter_default_get_audio_buffer(AVFilterLink *link, int per
|
|||
enum AVSampleFormat sample_fmt, int nb_samples,
|
||||
uint64_t channel_layout)
|
||||
{
|
||||
AVFilterBuffer *samples = av_mallocz(sizeof(AVFilterBuffer));
|
||||
AVFilterBufferRef *ref = NULL;
|
||||
int i, sample_size, chans_nb, bufsize, per_channel_size, step_size = 0;
|
||||
int planar = av_sample_fmt_is_planar(sample_fmt);
|
||||
char *buf;
|
||||
AVFilterBufferRef *samplesref = NULL;
|
||||
uint8_t **data;
|
||||
int planar = av_sample_fmt_is_planar(sample_fmt);
|
||||
int nb_channels = av_get_channel_layout_nb_channels(channel_layout);
|
||||
int planes = planar ? nb_channels : 1;
|
||||
int linesize;
|
||||
|
||||
if (!samples || !(ref = av_mallocz(sizeof(AVFilterBufferRef))))
|
||||
if (!(data = av_mallocz(sizeof(*data) * planes)))
|
||||
goto fail;
|
||||
|
||||
ref->buf = samples;
|
||||
ref->format = sample_fmt;
|
||||
|
||||
ref->audio = av_mallocz(sizeof(AVFilterBufferRefAudioProps));
|
||||
if (!ref->audio)
|
||||
if (av_samples_alloc(data, &linesize, nb_channels, nb_samples, sample_fmt, 0) < 0)
|
||||
goto fail;
|
||||
|
||||
ref->audio->channel_layout = channel_layout;
|
||||
ref->audio->nb_samples = nb_samples;
|
||||
ref->audio->planar = planar;
|
||||
|
||||
/* make sure the buffer gets read permission or it's useless for output */
|
||||
ref->perms = perms | AV_PERM_READ;
|
||||
|
||||
samples->refcount = 1;
|
||||
samples->free = ff_avfilter_default_free_buffer;
|
||||
|
||||
sample_size = av_get_bytes_per_sample(sample_fmt);
|
||||
chans_nb = av_get_channel_layout_nb_channels(channel_layout);
|
||||
|
||||
per_channel_size = nb_samples * sample_size;
|
||||
|
||||
/* Set the number of bytes to traverse to reach next sample of a particular channel:
|
||||
* For planar, this is simply the sample size.
|
||||
* For packed, this is the number of samples * sample_size.
|
||||
*/
|
||||
for (i = 0; i < chans_nb; i++)
|
||||
samples->linesize[i] = planar > 0 ? per_channel_size : sample_size;
|
||||
memset(&samples->linesize[chans_nb], 0, (8-chans_nb) * sizeof(samples->linesize[0]));
|
||||
|
||||
/* Calculate total buffer size, round to multiple of 16 to be SIMD friendly */
|
||||
bufsize = (nb_samples * chans_nb * sample_size + 15)&~15;
|
||||
buf = av_malloc(bufsize);
|
||||
if (!buf)
|
||||
samplesref = avfilter_get_audio_buffer_ref_from_arrays(data, linesize, perms,
|
||||
nb_samples, sample_fmt,
|
||||
channel_layout);
|
||||
if (!samplesref)
|
||||
goto fail;
|
||||
|
||||
/* For planar, set the start point of each channel's data within the buffer
|
||||
* For packed, set the start point of the entire buffer only
|
||||
*/
|
||||
samples->data[0] = buf;
|
||||
if (buf && planar) {
|
||||
for (i = 1; i < chans_nb; i++) {
|
||||
step_size += per_channel_size;
|
||||
samples->data[i] = buf + step_size;
|
||||
}
|
||||
} else {
|
||||
for (i = 1; i < chans_nb; i++)
|
||||
samples->data[i] = buf;
|
||||
}
|
||||
|
||||
memset(&samples->data[chans_nb], 0, (8-chans_nb) * sizeof(samples->data[0]));
|
||||
|
||||
memcpy(ref->data, samples->data, sizeof(ref->data));
|
||||
memcpy(ref->linesize, samples->linesize, sizeof(ref->linesize));
|
||||
|
||||
return ref;
|
||||
av_freep(&data);
|
||||
|
||||
fail:
|
||||
if (ref)
|
||||
av_free(ref->audio);
|
||||
av_free(ref);
|
||||
av_free(samples);
|
||||
return NULL;
|
||||
if (data)
|
||||
av_freep(&data[0]);
|
||||
av_freep(&data);
|
||||
return samplesref;
|
||||
}
|
||||
|
||||
void avfilter_default_start_frame(AVFilterLink *inlink, AVFilterBufferRef *picref)
|
||||
|
|
Loading…
Reference in New Issue