| /* GStreamer |
| * |
| * unit test for the taglib-based id3v2mux element |
| * |
| * Copyright (C) 2006 Tim-Philipp Müller <tim centricular net> |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Library General Public |
| * License as published by the Free Software Foundation; either |
| * version 2 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Library General Public License for more details. |
| * |
| * You should have received a copy of the GNU Library General Public |
| * License along with this library; if not, write to the |
| * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, |
| * Boston, MA 02110-1301, USA. |
| */ |
| |
| #include <gst/check/gstcheck.h> |
| |
| #include <gst/gst.h> |
| #include <string.h> |
| |
| #define TEST_ARTIST "Ar T\303\255st" |
| #define TEST_TITLE "M\303\274llermilch!" |
| #define TEST_ALBUM "Boom" |
| #define TEST_DATE g_date_new_dmy(1,1,2006) |
| #define TEST_TRACK_NUMBER 7 |
| #define TEST_TRACK_COUNT 19 |
| #define TEST_VOLUME_NUMBER 2 |
| #define TEST_VOLUME_COUNT 3 |
| #define TEST_TRACK_GAIN 1.45 |
| #define TEST_ALBUM_GAIN 0.78 |
| #define TEST_TRACK_PEAK 0.83 |
| #define TEST_ALBUM_PEAK 0.18 |
| #define TEST_BPM 113.0 |
| |
| /* for dummy mp3 frame sized MP3_FRAME_SIZE bytes, |
| * start: ff fb b0 44 00 00 08 00 00 4b 00 00 00 00 00 00 */ |
| static const guint8 mp3_dummyhdr[] = { 0xff, 0xfb, 0xb0, 0x44, 0x00, 0x00, |
| 0x08, 0x00, 0x4b, 0x00, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00 |
| }; |
| |
| #define MP3_FRAME_SIZE 626 |
| |
| /* the peak and gain values are stored pretty roughly, so check that they're |
| * within 2% of the expected value. |
| */ |
| #define fail_unless_sorta_equals_float(a, b) \ |
| G_STMT_START { \ |
| double first = a; \ |
| double second = b; \ |
| fail_unless(fabs (first - second) < (0.02 * fabs (first)), \ |
| "'" #a "' (%g) is not equal to '" #b "' (%g)", first, second); \ |
| } G_STMT_END; |
| |
| |
| static GstTagList * |
| test_taglib_id3mux_create_tags (guint32 mask) |
| { |
| GstTagList *tags; |
| |
| tags = gst_tag_list_new_empty (); |
| |
| if (mask & (1 << 0)) { |
| gst_tag_list_add (tags, GST_TAG_MERGE_KEEP, |
| GST_TAG_ARTIST, TEST_ARTIST, NULL); |
| } |
| if (mask & (1 << 1)) { |
| gst_tag_list_add (tags, GST_TAG_MERGE_KEEP, |
| GST_TAG_TITLE, TEST_TITLE, NULL); |
| } |
| if (mask & (1 << 2)) { |
| gst_tag_list_add (tags, GST_TAG_MERGE_KEEP, |
| GST_TAG_ALBUM, TEST_ALBUM, NULL); |
| } |
| if (mask & (1 << 3)) { |
| GDate *date; |
| |
| date = TEST_DATE; |
| gst_tag_list_add (tags, GST_TAG_MERGE_KEEP, GST_TAG_DATE, date, NULL); |
| g_date_free (date); |
| } |
| if (mask & (1 << 4)) { |
| gst_tag_list_add (tags, GST_TAG_MERGE_KEEP, |
| GST_TAG_TRACK_NUMBER, TEST_TRACK_NUMBER, NULL); |
| } |
| if (mask & (1 << 5)) { |
| gst_tag_list_add (tags, GST_TAG_MERGE_KEEP, |
| GST_TAG_TRACK_COUNT, TEST_TRACK_COUNT, NULL); |
| } |
| if (mask & (1 << 6)) { |
| gst_tag_list_add (tags, GST_TAG_MERGE_KEEP, |
| GST_TAG_ALBUM_VOLUME_NUMBER, TEST_VOLUME_NUMBER, NULL); |
| } |
| if (mask & (1 << 7)) { |
| gst_tag_list_add (tags, GST_TAG_MERGE_KEEP, |
| GST_TAG_ALBUM_VOLUME_COUNT, TEST_VOLUME_COUNT, NULL); |
| } |
| if (mask & (1 << 8)) { |
| gst_tag_list_add (tags, GST_TAG_MERGE_KEEP, |
| GST_TAG_TRACK_GAIN, TEST_TRACK_GAIN, NULL); |
| } |
| if (mask & (1 << 9)) { |
| gst_tag_list_add (tags, GST_TAG_MERGE_KEEP, |
| GST_TAG_ALBUM_GAIN, TEST_ALBUM_GAIN, NULL); |
| } |
| if (mask & (1 << 10)) { |
| gst_tag_list_add (tags, GST_TAG_MERGE_KEEP, |
| GST_TAG_TRACK_PEAK, TEST_TRACK_PEAK, NULL); |
| } |
| if (mask & (1 << 11)) { |
| gst_tag_list_add (tags, GST_TAG_MERGE_KEEP, |
| GST_TAG_ALBUM_PEAK, TEST_ALBUM_PEAK, NULL); |
| } |
| if (mask & (1 << 12)) { |
| gst_tag_list_add (tags, GST_TAG_MERGE_KEEP, |
| GST_TAG_BEATS_PER_MINUTE, TEST_BPM, NULL); |
| } |
| if (mask & (1 << 13)) { |
| } |
| return tags; |
| } |
| |
| static gboolean |
| utf8_string_in_buf (GstBuffer * buf, const gchar * s) |
| { |
| gint i, len; |
| GstMapInfo map; |
| |
| len = strlen (s); |
| gst_buffer_map (buf, &map, GST_MAP_READ); |
| for (i = 0; i < (map.size - len); ++i) { |
| if (memcmp (map.data + i, s, len) == 0) { |
| gst_buffer_unmap (buf, &map); |
| return TRUE; |
| } |
| } |
| gst_buffer_unmap (buf, &map); |
| |
| return FALSE; |
| } |
| |
| static void |
| test_taglib_id3mux_check_tag_buffer (GstBuffer * buf, guint32 mask) |
| { |
| /* make sure our UTF-8 string hasn't been put into the tag as ISO-8859-1 */ |
| if (mask & (1 << 0)) { |
| fail_unless (utf8_string_in_buf (buf, TEST_ARTIST)); |
| } |
| /* make sure our UTF-8 string hasn't been put into the tag as ISO-8859-1 */ |
| if (mask & (1 << 1)) { |
| fail_unless (utf8_string_in_buf (buf, TEST_TITLE)); |
| } |
| /* make sure our UTF-8 string hasn't been put into the tag as ISO-8859-1 */ |
| if (mask & (1 << 2)) { |
| fail_unless (utf8_string_in_buf (buf, TEST_ALBUM)); |
| } |
| } |
| |
| static void |
| test_taglib_id3mux_check_tags (GstTagList * tags, guint32 mask) |
| { |
| if (mask & (1 << 0)) { |
| gchar *s = NULL; |
| |
| fail_unless (gst_tag_list_get_string (tags, GST_TAG_ARTIST, &s)); |
| fail_unless (g_str_equal (s, TEST_ARTIST)); |
| g_free (s); |
| } |
| if (mask & (1 << 1)) { |
| gchar *s = NULL; |
| |
| fail_unless (gst_tag_list_get_string (tags, GST_TAG_TITLE, &s)); |
| fail_unless (g_str_equal (s, TEST_TITLE)); |
| g_free (s); |
| } |
| if (mask & (1 << 2)) { |
| gchar *s = NULL; |
| |
| fail_unless (gst_tag_list_get_string (tags, GST_TAG_ALBUM, &s)); |
| fail_unless (g_str_equal (s, TEST_ALBUM)); |
| g_free (s); |
| } |
| if (mask & (1 << 3)) { |
| GDate *shouldbe, *date = NULL; |
| |
| shouldbe = TEST_DATE; |
| fail_unless (gst_tag_list_get_date (tags, GST_TAG_DATE, &date)); |
| fail_unless (g_date_compare (shouldbe, date) == 0); |
| g_date_free (shouldbe); |
| g_date_free (date); |
| } |
| if (mask & (1 << 4)) { |
| guint num; |
| |
| fail_unless (gst_tag_list_get_uint (tags, GST_TAG_TRACK_NUMBER, &num)); |
| fail_unless (num == TEST_TRACK_NUMBER); |
| } |
| if (mask & (1 << 5)) { |
| guint count; |
| |
| fail_unless (gst_tag_list_get_uint (tags, GST_TAG_TRACK_COUNT, &count)); |
| fail_unless (count == TEST_TRACK_COUNT); |
| } |
| if (mask & (1 << 6)) { |
| guint num; |
| |
| fail_unless (gst_tag_list_get_uint (tags, GST_TAG_ALBUM_VOLUME_NUMBER, |
| &num)); |
| fail_unless (num == TEST_VOLUME_NUMBER); |
| } |
| if (mask & (1 << 7)) { |
| guint count; |
| |
| fail_unless (gst_tag_list_get_uint (tags, GST_TAG_ALBUM_VOLUME_COUNT, |
| &count)); |
| fail_unless (count == TEST_VOLUME_COUNT); |
| } |
| if (mask & (1 << 8)) { |
| gdouble gain; |
| |
| fail_unless (gst_tag_list_get_double (tags, GST_TAG_TRACK_GAIN, &gain)); |
| fail_unless_sorta_equals_float (gain, TEST_TRACK_GAIN); |
| } |
| if (mask & (1 << 9)) { |
| gdouble gain; |
| |
| fail_unless (gst_tag_list_get_double (tags, GST_TAG_ALBUM_GAIN, &gain)); |
| fail_unless_sorta_equals_float (gain, TEST_ALBUM_GAIN); |
| } |
| if (mask & (1 << 10)) { |
| gdouble peak; |
| |
| fail_unless (gst_tag_list_get_double (tags, GST_TAG_TRACK_PEAK, &peak)); |
| fail_unless_sorta_equals_float (peak, TEST_TRACK_PEAK); |
| } |
| if (mask & (1 << 11)) { |
| gdouble peak; |
| |
| fail_unless (gst_tag_list_get_double (tags, GST_TAG_ALBUM_PEAK, &peak)); |
| fail_unless_sorta_equals_float (peak, TEST_ALBUM_PEAK); |
| } |
| if (mask & (1 << 12)) { |
| gdouble bpm; |
| |
| fail_unless (gst_tag_list_get_double (tags, GST_TAG_BEATS_PER_MINUTE, |
| &bpm)); |
| fail_unless_sorta_equals_float (bpm, TEST_BPM); |
| } |
| if (mask & (1 << 13)) { |
| } |
| } |
| |
| static void |
| fill_mp3_buffer (GstElement * fakesrc, GstBuffer * buf, GstPad * pad, |
| guint64 * p_offset) |
| { |
| gsize size; |
| |
| size = gst_buffer_get_size (buf); |
| |
| fail_unless (size == MP3_FRAME_SIZE); |
| |
| GST_LOG ("filling buffer with fake mp3 data, offset = %" G_GUINT64_FORMAT, |
| *p_offset); |
| |
| gst_buffer_fill (buf, 0, mp3_dummyhdr, sizeof (mp3_dummyhdr)); |
| |
| #if 0 |
| /* can't use gst_buffer_set_caps() here because the metadata isn't writable |
| * because of the extra refcounts taken by the signal emission mechanism; |
| * we know it's fine to use GST_BUFFER_CAPS() here though */ |
| GST_BUFFER_CAPS (buf) = gst_caps_new_simple ("audio/mpeg", "mpegversion", |
| G_TYPE_INT, 1, "layer", G_TYPE_INT, 3, NULL); |
| #endif |
| |
| GST_BUFFER_OFFSET (buf) = *p_offset; |
| *p_offset += size; |
| } |
| |
| static void |
| got_buffer (GstElement * fakesink, GstBuffer * buf, GstPad * pad, |
| GstBuffer ** p_buf) |
| { |
| gint64 off; |
| GstMapInfo map; |
| |
| off = GST_BUFFER_OFFSET (buf); |
| |
| gst_buffer_map (buf, &map, GST_MAP_READ); |
| |
| GST_LOG ("size=%" G_GSIZE_FORMAT ", offset=%" G_GINT64_FORMAT, map.size, off); |
| |
| fail_unless (GST_BUFFER_OFFSET_IS_VALID (buf)); |
| |
| if (*p_buf == NULL || (off + map.size) > gst_buffer_get_size (*p_buf)) { |
| GstBuffer *newbuf; |
| |
| /* not very elegant, but who cares */ |
| newbuf = gst_buffer_new_and_alloc (off + map.size); |
| if (*p_buf) { |
| GstMapInfo pmap; |
| |
| gst_buffer_map (*p_buf, &pmap, GST_MAP_READ); |
| gst_buffer_fill (newbuf, 0, pmap.data, pmap.size); |
| gst_buffer_unmap (*p_buf, &pmap); |
| } |
| gst_buffer_fill (newbuf, off, map.data, map.size); |
| |
| if (*p_buf) |
| gst_buffer_unref (*p_buf); |
| *p_buf = newbuf; |
| } else { |
| gst_buffer_fill (*p_buf, off, map.data, map.size); |
| } |
| gst_buffer_unmap (buf, &map); |
| } |
| |
| static void |
| test_taglib_id3mux_check_output_buffer (GstBuffer * buf) |
| { |
| GstMapInfo map; |
| guint off; |
| |
| gst_buffer_map (buf, &map, GST_MAP_READ); |
| g_assert (map.size % MP3_FRAME_SIZE == 0); |
| |
| for (off = 0; off < map.size; off += MP3_FRAME_SIZE) { |
| fail_unless (memcmp (map.data + off, mp3_dummyhdr, |
| sizeof (mp3_dummyhdr)) == 0); |
| } |
| gst_buffer_unmap (buf, &map); |
| } |
| |
| static void |
| identity_cb (GstElement * identity, GstBuffer * buf, GstBuffer ** p_tagbuf) |
| { |
| if (*p_tagbuf == NULL) { |
| *p_tagbuf = gst_buffer_ref (buf); |
| } |
| } |
| |
| static void |
| test_taglib_id3mux_with_tags (GstTagList * tags, guint32 mask) |
| { |
| GstMessage *msg; |
| GstTagList *tags_read = NULL; |
| GstElement *pipeline, *id3mux, *id3demux, *fakesrc, *identity, *fakesink; |
| GstBus *bus; |
| guint64 offset; |
| GstBuffer *outbuf = NULL; |
| GstBuffer *tagbuf = NULL; |
| GstStateChangeReturn state_result; |
| |
| pipeline = gst_pipeline_new ("pipeline"); |
| g_assert (pipeline != NULL); |
| |
| fakesrc = gst_element_factory_make ("fakesrc", "fakesrc"); |
| g_assert (fakesrc != NULL); |
| |
| id3mux = gst_element_factory_make ("id3v2mux", "id3v2mux"); |
| g_assert (id3mux != NULL); |
| |
| identity = gst_element_factory_make ("identity", "identity"); |
| g_assert (identity != NULL); |
| |
| id3demux = gst_element_factory_make ("id3demux", "id3demux"); |
| g_assert (id3demux != NULL); |
| |
| fakesink = gst_element_factory_make ("fakesink", "fakesink"); |
| g_assert (fakesink != NULL); |
| |
| /* set up sink */ |
| outbuf = NULL; |
| g_object_set (fakesink, "signal-handoffs", TRUE, NULL); |
| g_signal_connect (fakesink, "handoff", G_CALLBACK (got_buffer), &outbuf); |
| |
| gst_bin_add (GST_BIN (pipeline), fakesrc); |
| gst_bin_add (GST_BIN (pipeline), id3mux); |
| gst_bin_add (GST_BIN (pipeline), identity); |
| gst_bin_add (GST_BIN (pipeline), id3demux); |
| gst_bin_add (GST_BIN (pipeline), fakesink); |
| |
| gst_tag_setter_merge_tags (GST_TAG_SETTER (id3mux), tags, |
| GST_TAG_MERGE_APPEND); |
| |
| gst_element_link_many (fakesrc, id3mux, identity, id3demux, fakesink, NULL); |
| |
| /* set up source */ |
| g_object_set (fakesrc, "signal-handoffs", TRUE, "can-activate-pull", FALSE, |
| "filltype", 2, "sizetype", 2, "sizemax", MP3_FRAME_SIZE, |
| "num-buffers", 16, NULL); |
| |
| offset = 0; |
| g_signal_connect (fakesrc, "handoff", G_CALLBACK (fill_mp3_buffer), &offset); |
| |
| /* set up identity to catch tag buffer */ |
| g_signal_connect (identity, "handoff", G_CALLBACK (identity_cb), &tagbuf); |
| |
| GST_LOG ("setting and getting state ..."); |
| gst_element_set_state (pipeline, GST_STATE_PLAYING); |
| state_result = gst_element_get_state (pipeline, NULL, NULL, -1); |
| fail_unless (state_result == GST_STATE_CHANGE_SUCCESS, |
| "Unexpected result from get_state(). Expected success, got %d", |
| state_result); |
| |
| bus = gst_pipeline_get_bus (GST_PIPELINE (pipeline)); |
| |
| GST_LOG ("Waiting for tag ..."); |
| msg = |
| gst_bus_poll (bus, GST_MESSAGE_TAG | GST_MESSAGE_EOS | GST_MESSAGE_ERROR, |
| -1); |
| if (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_ERROR) { |
| GError *err; |
| gchar *dbg; |
| |
| gst_message_parse_error (msg, &err, &dbg); |
| g_printerr ("ERROR from element %s: %s\n%s\n", |
| GST_OBJECT_NAME (msg->src), err->message, GST_STR_NULL (dbg)); |
| g_error_free (err); |
| g_free (dbg); |
| } else if (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS) { |
| g_printerr ("EOS message, but were waiting for TAGS!\n"); |
| } |
| fail_unless (msg->type == GST_MESSAGE_TAG); |
| |
| gst_message_parse_tag (msg, &tags_read); |
| gst_message_unref (msg); |
| |
| GST_LOG ("Got tags: %" GST_PTR_FORMAT, tags_read); |
| test_taglib_id3mux_check_tags (tags_read, mask); |
| gst_tag_list_unref (tags_read); |
| |
| fail_unless (tagbuf != NULL); |
| test_taglib_id3mux_check_tag_buffer (tagbuf, mask); |
| gst_buffer_unref (tagbuf); |
| |
| GST_LOG ("Waiting for EOS ..."); |
| msg = gst_bus_poll (bus, GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1); |
| if (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_ERROR) { |
| GError *err; |
| gchar *dbg; |
| |
| gst_message_parse_error (msg, &err, &dbg); |
| g_printerr ("ERROR from element %s: %s\n%s\n", |
| GST_OBJECT_NAME (msg->src), err->message, GST_STR_NULL (dbg)); |
| g_error_free (err); |
| g_free (dbg); |
| } |
| fail_unless (msg->type == GST_MESSAGE_EOS); |
| gst_message_unref (msg); |
| |
| gst_object_unref (bus); |
| |
| GST_LOG ("Got EOS, shutting down ..."); |
| gst_element_set_state (pipeline, GST_STATE_NULL); |
| gst_object_unref (pipeline); |
| |
| test_taglib_id3mux_check_output_buffer (outbuf); |
| gst_buffer_unref (outbuf); |
| |
| GST_LOG ("Done"); |
| } |
| |
| GST_START_TEST (test_id3v2mux) |
| { |
| GstTagList *tags; |
| gint i; |
| |
| g_random_set_seed (247166295); |
| |
| /* internal consistency check */ |
| tags = test_taglib_id3mux_create_tags (0xFFFFFFFF); |
| test_taglib_id3mux_check_tags (tags, 0xFFFFFFFF); |
| gst_tag_list_unref (tags); |
| |
| /* now the real tests */ |
| for (i = 0; i < 50; ++i) { |
| guint32 mask; |
| |
| mask = g_random_int (); |
| GST_LOG ("tag mask = %08x (i=%d)", mask, i); |
| |
| if (mask == 0) |
| continue; |
| |
| /* create tags */ |
| tags = test_taglib_id3mux_create_tags (mask); |
| GST_LOG ("tags for mask %08x = %" GST_PTR_FORMAT, mask, tags); |
| |
| /* double-check for internal consistency */ |
| test_taglib_id3mux_check_tags (tags, mask); |
| |
| /* test with pipeline */ |
| test_taglib_id3mux_with_tags (tags, mask); |
| |
| /* free tags */ |
| gst_tag_list_unref (tags); |
| } |
| } |
| |
| GST_END_TEST; |
| |
| static Suite * |
| id3v2mux_suite (void) |
| { |
| Suite *s = suite_create ("id3v2mux"); |
| TCase *tc_chain = tcase_create ("general"); |
| |
| suite_add_tcase (s, tc_chain); |
| tcase_add_test (tc_chain, test_id3v2mux); |
| |
| return s; |
| } |
| |
| GST_CHECK_MAIN (id3v2mux); |