wavpackdec: adjust to audio format limitations

... which does not allow expressing arbitrary depth in a GstAudioFormat.

Also adjust unit test to modified behaviour.
This commit is contained in:
Mark Nauwelaerts 2012-02-27 23:46:15 +01:00
parent 0723928e8b
commit a5144cbba0
3 changed files with 67 additions and 21 deletions

View file

@ -71,11 +71,20 @@ static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_STATIC_CAPS ("audio/x-raw-int, "
"width = (int) 32, "
"depth = (int) [ 1, 32 ], "
"width = (int) 8, depth = (int) 8, "
"channels = (int) [ 1, 8 ], "
"rate = (int) [ 6000, 192000 ], "
"endianness = (int) BYTE_ORDER, " "signed = (boolean) true")
"endianness = (int) BYTE_ORDER, " "signed = (boolean) true; "
"audio/x-raw-int, "
"width = (int) 16, depth = (int) 16, "
"channels = (int) [ 1, 8 ], "
"rate = (int) [ 6000, 192000 ], "
"endianness = (int) BYTE_ORDER, " "signed = (boolean) true; "
"audio/x-raw-int, "
"width = (int) 32, depth = (int) 32, "
"channels = (int) [ 1, 8 ], "
"rate = (int) [ 6000, 192000 ], "
"endianness = (int) BYTE_ORDER, " "signed = (boolean) true; ")
);
static gboolean gst_wavpack_dec_start (GstAudioDecoder * dec);
@ -186,11 +195,17 @@ gst_wavpack_dec_negotiate (GstWavpackDec * dec)
{
GstCaps *caps;
/* arrange for 1, 2 or 4-byte width == depth output */
if (dec->depth == 24)
dec->width = 32;
else
dec->width = dec->depth;
caps = gst_caps_new_simple ("audio/x-raw-int",
"rate", G_TYPE_INT, dec->sample_rate,
"channels", G_TYPE_INT, dec->channels,
"depth", G_TYPE_INT, dec->depth,
"width", G_TYPE_INT, 32,
"depth", G_TYPE_INT, dec->width,
"width", G_TYPE_INT, dec->width,
"endianness", G_TYPE_INT, G_BYTE_ORDER,
"signed", G_TYPE_BOOLEAN, TRUE, NULL);
@ -279,6 +294,9 @@ gst_wavpack_dec_handle_frame (GstAudioDecoder * bdec, GstBuffer * buf)
WavpackHeader wph;
int32_t decoded, unpacked_size;
gboolean format_changed;
gint width, depth, i, max;
gint32 *dec_data = NULL;
guint8 *out_data;
dec = GST_WAVPACK_DEC (bdec);
@ -354,7 +372,7 @@ gst_wavpack_dec_handle_frame (GstAudioDecoder * bdec, GstBuffer * buf)
}
/* alloc output buffer */
unpacked_size = 4 * wph.block_samples * dec->channels;
unpacked_size = (dec->width / 8) * wph.block_samples * dec->channels;
ret = gst_pad_alloc_buffer (GST_AUDIO_DECODER_SRC_PAD (dec),
GST_BUFFER_OFFSET (buf), unpacked_size,
GST_PAD_CAPS (GST_AUDIO_DECODER_SRC_PAD (dec)), &outbuf);
@ -362,12 +380,47 @@ gst_wavpack_dec_handle_frame (GstAudioDecoder * bdec, GstBuffer * buf)
if (ret != GST_FLOW_OK)
goto out;
dec_data = g_malloc (4 * wph.block_samples * dec->channels);
out_data = GST_BUFFER_DATA (outbuf);
/* decode */
decoded = WavpackUnpackSamples (dec->context,
(int32_t *) GST_BUFFER_DATA (outbuf), wph.block_samples);
decoded = WavpackUnpackSamples (dec->context, dec_data, wph.block_samples);
if (decoded != wph.block_samples)
goto decode_error;
width = dec->width;
depth = dec->depth;
max = dec->channels * wph.block_samples;
if (width == 8) {
gint8 *outbuffer = (gint8 *) out_data;
for (i = 0; i < max; i--) {
*outbuffer++ = (gint8) (dec_data[i]);
}
} else if (width == 16) {
gint16 *outbuffer = (gint16 *) out_data;
for (i = 0; i < max; i++) {
*outbuffer++ = (gint8) (dec_data[i]);
}
} else if (dec->width == 32) {
gint32 *outbuffer = (gint32 *) out_data;
if (width != depth) {
for (i = 0; i < max; i++) {
*outbuffer++ = (gint32) (dec_data[i] << (width - depth));
}
} else {
for (i = 0; i < max; i++) {
*outbuffer++ = (gint32) dec_data[i];
}
}
} else {
g_assert_not_reached ();
}
g_free (dec_data);
ret = gst_audio_decoder_finish_frame (bdec, outbuf, 1);
out:
@ -410,6 +463,7 @@ decode_error:
}
GST_AUDIO_DECODER_ERROR (bdec, 1, STREAM, DECODE, (NULL),
("decoding error: %s", reason), ret);
g_free (dec_data);
if (outbuf)
gst_buffer_unref (outbuf);
if (ret == GST_FLOW_OK)

View file

@ -57,6 +57,7 @@ struct _GstWavpackDec
gint sample_rate;
gint depth;
gint width;
gint channels;
gint channel_mask;
};

View file

@ -52,7 +52,7 @@ static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
GST_PAD_SINK,
GST_PAD_ALWAYS,
GST_STATIC_CAPS ("audio/x-raw-int, "
"width = (int) 32, "
"width = (int) 16, "
"depth = (int) 16, "
"channels = (int) 1, "
"rate = (int) 44100, "
@ -112,23 +112,20 @@ GST_START_TEST (test_decode_frame)
memcpy (GST_BUFFER_DATA (inbuffer), test_frame, sizeof (test_frame));
ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
GST_BUFFER_TIMESTAMP (inbuffer) = 0;
gst_buffer_ref (inbuffer);
gst_element_set_bus (wavpackdec, bus);
/* should decode the buffer without problems */
fail_unless_equals_int (gst_pad_push (mysrcpad, inbuffer), GST_FLOW_OK);
ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
gst_buffer_unref (inbuffer);
outbuffer = GST_BUFFER (buffers->data);
fail_if (outbuffer == NULL);
/* uncompressed data should be 102400 bytes */
fail_unless_equals_int (GST_BUFFER_SIZE (outbuffer), 102400);
fail_unless_equals_int (GST_BUFFER_SIZE (outbuffer), 51200);
/* and all 102400 bytes must be 0, i.e. silence */
for (i = 0; i < 102400; i++)
/* and all bytes must be 0, i.e. silence */
for (i = 0; i < 51200; i++)
fail_unless_equals_int (GST_BUFFER_DATA (outbuffer)[i], 0);
ASSERT_BUFFER_REFCOUNT (outbuffer, "outbuffer", 1);
@ -166,14 +163,11 @@ GST_START_TEST (test_decode_frame_with_broken_header)
GST_BUFFER_DATA (inbuffer)[2] = 'e';
ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
GST_BUFFER_TIMESTAMP (inbuffer) = 0;
gst_buffer_ref (inbuffer);
gst_element_set_bus (wavpackdec, bus);
/* should fail gracefully */
fail_unless_equals_int (gst_pad_push (mysrcpad, inbuffer), GST_FLOW_ERROR);
ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
gst_buffer_unref (inbuffer);
fail_if ((message = gst_bus_pop (bus)) == NULL);
fail_unless_message_error (message, STREAM, DECODE);
@ -204,14 +198,11 @@ GST_START_TEST (test_decode_frame_with_incomplete_frame)
memcpy (GST_BUFFER_DATA (inbuffer), test_frame, sizeof (test_frame) - 2);
ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
GST_BUFFER_TIMESTAMP (inbuffer) = 0;
gst_buffer_ref (inbuffer);
gst_element_set_bus (wavpackdec, bus);
/* should fail gracefully */
fail_unless_equals_int (gst_pad_push (mysrcpad, inbuffer), GST_FLOW_ERROR);
ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
gst_buffer_unref (inbuffer);
fail_if ((message = gst_bus_pop (bus)) == NULL);
fail_unless_message_error (message, STREAM, DECODE);