audio-convert: improve converter API

Improve the converter API to allow for an max input and output number of
samples and return the number of consumed/produced samples.
This commit is contained in:
Wim Taymans 2015-12-09 17:16:26 +01:00
parent 872f40d7d9
commit 5e55968546
3 changed files with 47 additions and 33 deletions

View File

@ -374,36 +374,48 @@ gst_audio_converter_free (GstAudioConverter * convert)
* gst_audio_converter_samples: * gst_audio_converter_samples:
* @convert: a #GstAudioConverter * @convert: a #GstAudioConverter
* @flags: extra #GstAudioConverterFlags * @flags: extra #GstAudioConverterFlags
* @src: source samples * @in: input samples
* @dst: output samples
* @in_samples: number of input samples * @in_samples: number of input samples
* @out: output samples
* @out_samples: number of output samples * @out_samples: number of output samples
* @in_consumed: number of input samples consumed
* @out_produced: number of output samples produced
* *
* Perform the conversion @src to @dst using @convert. * Perform the conversion with @in_samples in @in to @out_samples in @out
* using @convert.
*
* The actual number of samples used from @in is returned in @in_consumed and
* can be less than @in_samples. The actual number of samples produced is
* returned in @out_produced and can be less than @out_samples.
* *
* Returns: %TRUE is the conversion could be performed. * Returns: %TRUE is the conversion could be performed.
*/ */
gboolean gboolean
gst_audio_converter_samples (GstAudioConverter * convert, gst_audio_converter_samples (GstAudioConverter * convert,
GstAudioConverterFlags flags, gpointer src, gpointer dst, GstAudioConverterFlags flags, gpointer in, gsize in_samples,
gsize in_samples, gsize * out_samples) gpointer out, gsize out_samples, gsize * in_consumed, gsize * out_produced)
{ {
guint size; guint size;
gpointer outbuf, tmpbuf, tmpbuf2; gpointer outbuf, tmpbuf, tmpbuf2;
g_return_val_if_fail (convert != NULL, FALSE); g_return_val_if_fail (convert != NULL, FALSE);
g_return_val_if_fail (src != NULL, FALSE); g_return_val_if_fail (in != NULL, FALSE);
g_return_val_if_fail (dst != NULL, FALSE); g_return_val_if_fail (out != NULL, FALSE);
g_return_val_if_fail (out_samples != NULL, FALSE); g_return_val_if_fail (in_consumed != NULL, FALSE);
g_return_val_if_fail (out_produced != NULL, FALSE);
in_samples = MIN (in_samples, out_samples);
if (in_samples == 0) { if (in_samples == 0) {
*out_samples = 0; *in_consumed = 0;
*out_produced = 0;
return TRUE; return TRUE;
} }
if (convert->passthrough) { if (convert->passthrough) {
memcpy (dst, src, in_samples * convert->in.bpf); memcpy (out, in, in_samples * convert->in.bpf);
*out_samples = in_samples; *out_produced = in_samples;
*in_consumed = in_samples;
return TRUE; return TRUE;
} }
@ -423,68 +435,69 @@ gst_audio_converter_samples (GstAudioConverter * convert,
if (!convert->in_default) { if (!convert->in_default) {
if (!convert->convert_in && convert->mix_passthrough if (!convert->convert_in && convert->mix_passthrough
&& !convert->convert_out && !convert->quant && convert->out_default) && !convert->convert_out && !convert->quant && convert->out_default)
outbuf = dst; outbuf = out;
else else
outbuf = tmpbuf; outbuf = tmpbuf;
convert->in.finfo->unpack_func (convert->in.finfo, convert->in.finfo->unpack_func (convert->in.finfo,
GST_AUDIO_PACK_FLAG_TRUNCATE_RANGE, outbuf, src, GST_AUDIO_PACK_FLAG_TRUNCATE_RANGE, outbuf, in,
in_samples * convert->in.channels); in_samples * convert->in.channels);
src = outbuf; in = outbuf;
} }
/* 2. optionally convert for mixing */ /* 2. optionally convert for mixing */
if (convert->convert_in) { if (convert->convert_in) {
if (convert->mix_passthrough && !convert->convert_out && !convert->quant if (convert->mix_passthrough && !convert->convert_out && !convert->quant
&& convert->out_default) && convert->out_default)
outbuf = dst; outbuf = out;
else if (src == tmpbuf) else if (in == tmpbuf)
outbuf = tmpbuf2; outbuf = tmpbuf2;
else else
outbuf = tmpbuf; outbuf = tmpbuf;
convert->convert_in (outbuf, src, in_samples * convert->in.channels); convert->convert_in (outbuf, in, in_samples * convert->in.channels);
src = outbuf; in = outbuf;
} }
/* step 3, channel mix if not passthrough */ /* step 3, channel mix if not passthrough */
if (!convert->mix_passthrough) { if (!convert->mix_passthrough) {
if (!convert->convert_out && !convert->quant && convert->out_default) if (!convert->convert_out && !convert->quant && convert->out_default)
outbuf = dst; outbuf = out;
else else
outbuf = tmpbuf; outbuf = tmpbuf;
gst_audio_channel_mix_samples (convert->mix, src, outbuf, in_samples); gst_audio_channel_mix_samples (convert->mix, in, outbuf, in_samples);
src = outbuf; in = outbuf;
} }
/* step 4, optional convert F64 -> S32 for quantize */ /* step 4, optional convert F64 -> S32 for quantize */
if (convert->convert_out) { if (convert->convert_out) {
if (!convert->quant && convert->out_default) if (!convert->quant && convert->out_default)
outbuf = dst; outbuf = out;
else else
outbuf = tmpbuf; outbuf = tmpbuf;
convert->convert_out (outbuf, src, in_samples * convert->out.channels); convert->convert_out (outbuf, in, in_samples * convert->out.channels);
src = outbuf; in = outbuf;
} }
/* step 5, optional quantize */ /* step 5, optional quantize */
if (convert->quant) { if (convert->quant) {
if (convert->out_default) if (convert->out_default)
outbuf = dst; outbuf = out;
else else
outbuf = tmpbuf; outbuf = tmpbuf;
gst_audio_quantize_samples (convert->quant, outbuf, src, in_samples); gst_audio_quantize_samples (convert->quant, outbuf, in, in_samples);
src = outbuf; in = outbuf;
} }
/* step 6, pack */ /* step 6, pack */
if (!convert->out_default) { if (!convert->out_default) {
convert->out.finfo->pack_func (convert->out.finfo, 0, src, dst, convert->out.finfo->pack_func (convert->out.finfo, 0, in, out,
in_samples * convert->out.channels); in_samples * convert->out.channels);
} }
*out_samples = in_samples; *out_produced = in_samples;
*in_consumed = in_samples;
return TRUE; return TRUE;
} }

View File

@ -81,7 +81,8 @@ const GstStructure * gst_audio_converter_get_config (GstAudioConverter * con
gboolean gst_audio_converter_samples (GstAudioConverter * convert, gboolean gst_audio_converter_samples (GstAudioConverter * convert,
GstAudioConverterFlags flags, GstAudioConverterFlags flags,
gpointer src, gpointer dst, gpointer in, gsize in_samples,
gsize in_samples, gsize *out_samples); gpointer out, gsize out_samples,
gsize *in_consumed, gsize *out_produced);
#endif /* __GST_AUDIO_CONVERTER_H__ */ #endif /* __GST_AUDIO_CONVERTER_H__ */

View File

@ -701,7 +701,7 @@ gst_audio_convert_transform (GstBaseTransform * base, GstBuffer * inbuf,
gint insize, outsize; gint insize, outsize;
gboolean inbuf_writable; gboolean inbuf_writable;
GstAudioConverterFlags flags; GstAudioConverterFlags flags;
gsize samples, out; gsize samples, consumed, produced;
/* get amount of samples to convert. */ /* get amount of samples to convert. */
samples = gst_buffer_get_size (inbuf) / this->in_info.bpf; samples = gst_buffer_get_size (inbuf) / this->in_info.bpf;
@ -736,7 +736,7 @@ gst_audio_convert_transform (GstBaseTransform * base, GstBuffer * inbuf,
if (!GST_BUFFER_FLAG_IS_SET (inbuf, GST_BUFFER_FLAG_GAP)) { if (!GST_BUFFER_FLAG_IS_SET (inbuf, GST_BUFFER_FLAG_GAP)) {
if (!gst_audio_converter_samples (this->convert, flags, srcmap.data, if (!gst_audio_converter_samples (this->convert, flags, srcmap.data,
dstmap.data, samples, &out)) samples, dstmap.data, samples, &consumed, &produced))
goto convert_error; goto convert_error;
} else { } else {
/* Create silence buffer */ /* Create silence buffer */