5 * Copyright (C) <2008> Mark Nauwelaerts <mnauw@users.sf.net>
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Library General Public
9 * License as published by the Free Software Foundation; either
10 * version 2 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Library General Public License for more details.
17 * You should have received a copy of the GNU Library General Public
18 * License along with this library; if not, write to the
19 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
20 * Boston, MA 02111-1307, USA.
31 #include <gst/check/gstcheck.h>
32 #include <gst/pbutils/encoding-profile.h>
34 /* For ease of programming we use globals to keep refs for our floating
35 * src and sink pads we create; otherwise we always have to do get_pad,
36 * get_peer, and then remove references in every test function */
37 static GstPad *mysrcpad, *mysinkpad;
39 #define AUDIO_CAPS_STRING "audio/mpeg, " \
40 "mpegversion = (int) 1, " \
42 "channels = (int) 2, " \
44 #define VIDEO_CAPS_STRING "video/mpeg, " \
45 "mpegversion = (int) 4, " \
46 "width = (int) 384, " \
47 "height = (int) 288, " \
48 "framerate = (fraction) 25/1"
50 static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
53 GST_STATIC_CAPS ("video/quicktime"));
54 static GstStaticPadTemplate srcvideotemplate = GST_STATIC_PAD_TEMPLATE ("src",
57 GST_STATIC_CAPS (VIDEO_CAPS_STRING));
59 static GstStaticPadTemplate srcaudiotemplate = GST_STATIC_PAD_TEMPLATE ("src",
62 GST_STATIC_CAPS (AUDIO_CAPS_STRING));
65 /* setup and teardown needs some special handling for muxer */
67 setup_src_pad (GstElement * element,
68 GstStaticPadTemplate * template, GstCaps * caps, const gchar * sinkname)
70 GstPad *srcpad, *sinkpad;
72 GST_DEBUG_OBJECT (element, "setting up sending pad");
74 srcpad = gst_pad_new_from_static_template (template, "src");
75 fail_if (srcpad == NULL, "Could not create a srcpad");
76 ASSERT_OBJECT_REFCOUNT (srcpad, "srcpad", 1);
78 if (!(sinkpad = gst_element_get_static_pad (element, sinkname)))
79 sinkpad = gst_element_get_request_pad (element, sinkname);
80 fail_if (sinkpad == NULL, "Could not get sink pad from %s",
81 GST_ELEMENT_NAME (element));
82 /* references are owned by: 1) us, 2) qtmux, 3) collect pads */
83 ASSERT_OBJECT_REFCOUNT (sinkpad, "sinkpad", 3);
85 fail_unless (gst_pad_set_caps (srcpad, caps));
86 fail_unless (gst_pad_link (srcpad, sinkpad) == GST_PAD_LINK_OK,
87 "Could not link source and %s sink pads", GST_ELEMENT_NAME (element));
88 gst_object_unref (sinkpad); /* because we got it higher up */
90 /* references are owned by: 1) qtmux, 2) collect pads */
91 ASSERT_OBJECT_REFCOUNT (sinkpad, "sinkpad", 2);
97 teardown_src_pad (GstPad * srcpad)
101 /* clean up floating src pad */
102 sinkpad = gst_pad_get_peer (srcpad);
103 fail_if (sinkpad == NULL);
104 /* pad refs held by 1) qtmux 2) collectpads and 3) us (through _get_peer) */
105 ASSERT_OBJECT_REFCOUNT (sinkpad, "sinkpad", 3);
107 gst_pad_unlink (srcpad, sinkpad);
109 /* after unlinking, pad refs still held by
110 * 1) qtmux and 2) collectpads and 3) us (through _get_peer) */
111 ASSERT_OBJECT_REFCOUNT (sinkpad, "sinkpad", 3);
112 gst_object_unref (sinkpad);
113 /* one more ref is held by element itself */
115 /* pad refs held by creator */
116 ASSERT_OBJECT_REFCOUNT (srcpad, "srcpad", 1);
117 gst_object_unref (srcpad);
121 setup_qtmux (GstStaticPadTemplate * srctemplate, const gchar * sinkname)
125 GST_DEBUG ("setup_qtmux");
126 qtmux = gst_check_setup_element ("qtmux");
127 mysrcpad = setup_src_pad (qtmux, srctemplate, NULL, sinkname);
128 mysinkpad = gst_check_setup_sink_pad (qtmux, &sinktemplate, NULL);
129 gst_pad_set_active (mysrcpad, TRUE);
130 gst_pad_set_active (mysinkpad, TRUE);
136 cleanup_qtmux (GstElement * qtmux, const gchar * sinkname)
138 GST_DEBUG ("cleanup_qtmux");
139 gst_element_set_state (qtmux, GST_STATE_NULL);
141 gst_pad_set_active (mysrcpad, FALSE);
142 gst_pad_set_active (mysinkpad, FALSE);
143 teardown_src_pad (mysrcpad);
144 gst_check_teardown_sink_pad (qtmux);
145 gst_check_teardown_element (qtmux);
149 check_qtmux_pad (GstStaticPadTemplate * srctemplate, const gchar * sinkname,
153 GstBuffer *inbuffer, *outbuffer;
157 guint8 data0[12] = "\000\000\000\024ftypqt ";
158 guint8 data1[8] = "\000\000\000\001mdat";
159 guint8 data2[4] = "moov";
161 qtmux = setup_qtmux (srctemplate, sinkname);
162 g_object_set (qtmux, "dts-method", dts_method, NULL);
163 fail_unless (gst_element_set_state (qtmux,
164 GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
165 "could not set to playing");
167 inbuffer = gst_buffer_new_and_alloc (1);
168 caps = gst_caps_copy (gst_pad_get_pad_template_caps (mysrcpad));
169 gst_buffer_set_caps (inbuffer, caps);
170 gst_caps_unref (caps);
171 GST_BUFFER_TIMESTAMP (inbuffer) = 0;
172 GST_BUFFER_DURATION (inbuffer) = 40 * GST_MSECOND;
173 ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
174 fail_unless (gst_pad_push (mysrcpad, inbuffer) == GST_FLOW_OK);
176 /* send eos to have moov written */
177 fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_eos ()) == TRUE);
179 num_buffers = g_list_length (buffers);
180 /* at least expect ftyp, mdat header, buffer chunk and moov */
181 fail_unless (num_buffers >= 4);
183 for (i = 0; i < num_buffers; ++i) {
184 outbuffer = GST_BUFFER (buffers->data);
185 fail_if (outbuffer == NULL);
186 buffers = g_list_remove (buffers, outbuffer);
192 guint8 *data = GST_BUFFER_DATA (outbuffer);
194 fail_unless (GST_BUFFER_SIZE (outbuffer) >= 20);
195 fail_unless (memcmp (data, data0, sizeof (data0)) == 0);
196 fail_unless (memcmp (data + 16, data0 + 8, 4) == 0);
199 case 1: /* mdat header */
200 fail_unless (GST_BUFFER_SIZE (outbuffer) == 16);
201 fail_unless (memcmp (GST_BUFFER_DATA (outbuffer), data1, sizeof (data1))
204 case 2: /* buffer we put in */
205 fail_unless (GST_BUFFER_SIZE (outbuffer) == 1);
208 fail_unless (GST_BUFFER_SIZE (outbuffer) > 8);
209 fail_unless (memcmp (GST_BUFFER_DATA (outbuffer) + 4, data2,
210 sizeof (data2)) == 0);
216 ASSERT_BUFFER_REFCOUNT (outbuffer, "outbuffer", 1);
217 gst_buffer_unref (outbuffer);
221 g_list_free (buffers);
224 cleanup_qtmux (qtmux, sinkname);
228 check_qtmux_pad_fragmented (GstStaticPadTemplate * srctemplate,
229 const gchar * sinkname, guint32 dts_method, gboolean streamable)
232 GstBuffer *inbuffer, *outbuffer;
236 guint8 data0[12] = "\000\000\000\024ftypqt ";
237 guint8 data1[4] = "mdat";
238 guint8 data2[4] = "moov";
239 guint8 data3[4] = "moof";
240 guint8 data4[4] = "mfra";
242 qtmux = setup_qtmux (srctemplate, sinkname);
243 g_object_set (qtmux, "dts-method", dts_method, NULL);
244 g_object_set (qtmux, "fragment-duration", 2000, NULL);
245 g_object_set (qtmux, "streamable", streamable, NULL);
246 fail_unless (gst_element_set_state (qtmux,
247 GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
248 "could not set to playing");
250 inbuffer = gst_buffer_new_and_alloc (1);
251 caps = gst_caps_copy (gst_pad_get_pad_template_caps (mysrcpad));
252 gst_buffer_set_caps (inbuffer, caps);
253 gst_caps_unref (caps);
254 GST_BUFFER_TIMESTAMP (inbuffer) = 0;
255 GST_BUFFER_DURATION (inbuffer) = 40 * GST_MSECOND;
256 ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
257 fail_unless (gst_pad_push (mysrcpad, inbuffer) == GST_FLOW_OK);
259 /* send eos to have all written */
260 fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_eos ()) == TRUE);
262 num_buffers = g_list_length (buffers);
263 /* at least expect ftyp, moov, moof, mdat header, buffer chunk
264 * and optionally mfra */
265 fail_unless (num_buffers >= 5);
267 for (i = 0; i < num_buffers; ++i) {
268 outbuffer = GST_BUFFER (buffers->data);
269 fail_if (outbuffer == NULL);
270 buffers = g_list_remove (buffers, outbuffer);
276 guint8 *data = GST_BUFFER_DATA (outbuffer);
278 fail_unless (GST_BUFFER_SIZE (outbuffer) >= 20);
279 fail_unless (memcmp (data, data0, sizeof (data0)) == 0);
280 fail_unless (memcmp (data + 16, data0 + 8, 4) == 0);
284 fail_unless (GST_BUFFER_SIZE (outbuffer) > 8);
285 fail_unless (memcmp (GST_BUFFER_DATA (outbuffer) + 4, data2,
286 sizeof (data2)) == 0);
289 fail_unless (GST_BUFFER_SIZE (outbuffer) > 8);
290 fail_unless (memcmp (GST_BUFFER_DATA (outbuffer) + 4, data3,
291 sizeof (data3)) == 0);
293 case 3: /* mdat header */
294 fail_unless (GST_BUFFER_SIZE (outbuffer) == 8);
295 fail_unless (memcmp (GST_BUFFER_DATA (outbuffer) + 4, data1,
296 sizeof (data1)) == 0);
298 case 4: /* buffer we put in */
299 fail_unless (GST_BUFFER_SIZE (outbuffer) == 1);
302 fail_unless (GST_BUFFER_SIZE (outbuffer) > 8);
303 fail_unless (memcmp (GST_BUFFER_DATA (outbuffer) + 4, data4,
304 sizeof (data4)) == 0);
310 ASSERT_BUFFER_REFCOUNT (outbuffer, "outbuffer", 1);
311 gst_buffer_unref (outbuffer);
315 g_list_free (buffers);
318 cleanup_qtmux (qtmux, sinkname);
323 GST_START_TEST (test_video_pad_dd)
325 check_qtmux_pad (&srcvideotemplate, "video_%d", 0);
330 GST_START_TEST (test_audio_pad_dd)
332 check_qtmux_pad (&srcaudiotemplate, "audio_%d", 0);
338 GST_START_TEST (test_video_pad_frag_dd)
340 check_qtmux_pad_fragmented (&srcvideotemplate, "video_%d", 0, FALSE);
345 GST_START_TEST (test_audio_pad_frag_dd)
347 check_qtmux_pad_fragmented (&srcaudiotemplate, "audio_%d", 0, FALSE);
353 GST_START_TEST (test_video_pad_frag_dd_streamable)
355 check_qtmux_pad_fragmented (&srcvideotemplate, "video_%d", 0, TRUE);
361 GST_START_TEST (test_audio_pad_frag_dd_streamable)
363 check_qtmux_pad_fragmented (&srcaudiotemplate, "audio_%d", 0, TRUE);
368 /* dts-method reorder */
370 GST_START_TEST (test_video_pad_reorder)
372 check_qtmux_pad (&srcvideotemplate, "video_%d", 1);
377 GST_START_TEST (test_audio_pad_reorder)
379 check_qtmux_pad (&srcaudiotemplate, "audio_%d", 1);
385 GST_START_TEST (test_video_pad_frag_reorder)
387 check_qtmux_pad_fragmented (&srcvideotemplate, "video_%d", 1, FALSE);
392 GST_START_TEST (test_audio_pad_frag_reorder)
394 check_qtmux_pad_fragmented (&srcaudiotemplate, "audio_%d", 1, FALSE);
400 GST_START_TEST (test_video_pad_frag_reorder_streamable)
402 check_qtmux_pad_fragmented (&srcvideotemplate, "video_%d", 1, TRUE);
408 GST_START_TEST (test_audio_pad_frag_reorder_streamable)
410 check_qtmux_pad_fragmented (&srcaudiotemplate, "audio_%d", 1, TRUE);
417 GST_START_TEST (test_video_pad_asc)
419 check_qtmux_pad (&srcvideotemplate, "video_%d", 2);
424 GST_START_TEST (test_audio_pad_asc)
426 check_qtmux_pad (&srcaudiotemplate, "audio_%d", 2);
432 GST_START_TEST (test_video_pad_frag_asc)
434 check_qtmux_pad_fragmented (&srcvideotemplate, "video_%d", 2, FALSE);
439 GST_START_TEST (test_audio_pad_frag_asc)
441 check_qtmux_pad_fragmented (&srcaudiotemplate, "audio_%d", 2, FALSE);
447 GST_START_TEST (test_video_pad_frag_asc_streamable)
449 check_qtmux_pad_fragmented (&srcvideotemplate, "video_%d", 2, TRUE);
455 GST_START_TEST (test_audio_pad_frag_asc_streamable)
457 check_qtmux_pad_fragmented (&srcaudiotemplate, "audio_%d", 2, TRUE);
462 GST_START_TEST (test_reuse)
464 GstElement *qtmux = setup_qtmux (&srcvideotemplate, "video_%d");
468 gst_element_set_state (qtmux, GST_STATE_PLAYING);
469 gst_element_set_state (qtmux, GST_STATE_NULL);
470 gst_element_set_state (qtmux, GST_STATE_PLAYING);
471 gst_pad_set_active (mysrcpad, TRUE);
472 gst_pad_set_active (mysinkpad, TRUE);
474 inbuffer = gst_buffer_new_and_alloc (1);
475 fail_unless (inbuffer != NULL);
476 caps = gst_caps_copy (gst_pad_get_pad_template_caps (mysrcpad));
477 gst_buffer_set_caps (inbuffer, caps);
478 gst_caps_unref (caps);
479 GST_BUFFER_TIMESTAMP (inbuffer) = 0;
480 GST_BUFFER_DURATION (inbuffer) = 40 * GST_MSECOND;
481 ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
482 fail_unless (gst_pad_push (mysrcpad, inbuffer) == GST_FLOW_OK);
484 /* send eos to have all written */
485 fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_eos ()) == TRUE);
487 cleanup_qtmux (qtmux, "video_%d");
492 static GstEncodingContainerProfile *
493 create_qtmux_profile (const gchar * variant)
495 GstEncodingContainerProfile *cprof;
498 if (variant == NULL) {
499 caps = gst_caps_new_simple ("video/quicktime", NULL);
501 caps = gst_caps_new_simple ("video/quicktime",
502 "variant", G_TYPE_STRING, variant, NULL);
505 cprof = gst_encoding_container_profile_new ("Name", "blah", caps, NULL);
506 gst_caps_unref (caps);
508 caps = gst_caps_new_simple ("audio/x-raw-int", "width", G_TYPE_INT, 16,
509 "depth", G_TYPE_INT, 16, "endianness", G_TYPE_INT, 4321,
510 "channels", G_TYPE_INT, 2, "rate", G_TYPE_INT, 44100,
511 "signed", G_TYPE_BOOLEAN, TRUE, NULL);
512 gst_encoding_container_profile_add_profile (cprof,
513 GST_ENCODING_PROFILE (gst_encoding_audio_profile_new (caps, NULL, NULL,
515 gst_caps_unref (caps);
520 GST_START_TEST (test_encodebin)
522 GstEncodingContainerProfile *cprof;
526 enc = gst_element_factory_make ("encodebin", NULL);
530 /* Make sure encodebin finds a muxer for a profile with a variant field .. */
531 cprof = create_qtmux_profile ("apple");
532 g_object_set (enc, "profile", cprof, NULL);
533 gst_encoding_profile_unref (cprof);
535 /* should have created a pad after setting the profile */
536 pad = gst_element_get_static_pad (enc, "audio_0");
537 fail_unless (pad != NULL);
538 gst_object_unref (pad);
539 gst_object_unref (enc);
541 /* ... and for a profile without a variant field */
542 enc = gst_element_factory_make ("encodebin", NULL);
543 cprof = create_qtmux_profile (NULL);
544 g_object_set (enc, "profile", cprof, NULL);
545 gst_encoding_profile_unref (cprof);
547 /* should have created a pad after setting the profile */
548 pad = gst_element_get_static_pad (enc, "audio_0");
549 fail_unless (pad != NULL);
550 gst_object_unref (pad);
551 gst_object_unref (enc);
559 Suite *s = suite_create ("qtmux");
560 TCase *tc_chain = tcase_create ("general");
562 suite_add_tcase (s, tc_chain);
563 tcase_add_test (tc_chain, test_video_pad_dd);
564 tcase_add_test (tc_chain, test_audio_pad_dd);
565 tcase_add_test (tc_chain, test_video_pad_frag_dd);
566 tcase_add_test (tc_chain, test_audio_pad_frag_dd);
567 tcase_add_test (tc_chain, test_video_pad_frag_dd_streamable);
568 tcase_add_test (tc_chain, test_audio_pad_frag_dd_streamable);
570 tcase_add_test (tc_chain, test_video_pad_reorder);
571 tcase_add_test (tc_chain, test_audio_pad_reorder);
572 tcase_add_test (tc_chain, test_video_pad_frag_reorder);
573 tcase_add_test (tc_chain, test_audio_pad_frag_reorder);
574 tcase_add_test (tc_chain, test_video_pad_frag_reorder_streamable);
575 tcase_add_test (tc_chain, test_audio_pad_frag_reorder_streamable);
577 tcase_add_test (tc_chain, test_video_pad_asc);
578 tcase_add_test (tc_chain, test_audio_pad_asc);
579 tcase_add_test (tc_chain, test_video_pad_frag_asc);
580 tcase_add_test (tc_chain, test_audio_pad_frag_asc);
581 tcase_add_test (tc_chain, test_video_pad_frag_asc_streamable);
582 tcase_add_test (tc_chain, test_audio_pad_frag_asc_streamable);
584 tcase_add_test (tc_chain, test_reuse);
585 tcase_add_test (tc_chain, test_encodebin);
590 GST_CHECK_MAIN (qtmux)