mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2025-04-12 19:14:16 +00:00
imagecapturebin: Add another test case
Adds a test case to check if a sequence of buffers with different caps can be pushed to imagecapturebin and saved correctly
This commit is contained in:
Thiago Santos
parent
72ef18d535
commit
7ed49bda4f
2 changed files with 167 additions and 7 deletions
tests/check
|
|
@ -210,6 +210,16 @@ elements_camerabin2_LDADD = \
|
|||
$(GST_LIBS) $(GST_BASE_LIBS) $(GST_PLUGINS_BASE_LIBS) \
|
||||
$(GST_CHECK_LIBS)
|
||||
elements_camerabin2_SOURCES = elements/camerabin2.c
|
||||
|
||||
elements_imagecapturebin_CFLAGS = \
|
||||
-I$(top_builddir)/gst-libs \
|
||||
$(GST_CFLAGS) $(GST_BASE_CFLAGS) $(GST_PLUGINS_BASE_CFLAGS) \
|
||||
$(GST_CHECK_CFLAGS) \
|
||||
-DGST_USE_UNSTABLE_API
|
||||
elements_imagecapturebin_LDADD = \
|
||||
$(GST_LIBS) $(GST_BASE_LIBS) $(GST_PLUGINS_BASE_LIBS) \
|
||||
$(GST_CHECK_LIBS) -lgstapp-@GST_MAJORMINOR@
|
||||
elements_imagecapturebin_SOURCES = elements/imagecapturebin.c
|
||||
endif
|
||||
|
||||
elements_jifmux_CFLAGS = $(GST_CHECK_CFLAGS) $(GST_OPTION_CFLAGS) $(AM_CFLAGS) $(EXIF_CFLAGS) $(GST_PLUGINS_BASE_CFLAGS)
|
||||
|
|
|
|||
|
|
@ -23,6 +23,8 @@
|
|||
|
||||
#include <stdio.h>
|
||||
#include <gst/check/gstcheck.h>
|
||||
#include <gst/app/gstappsink.h>
|
||||
#include <gst/app/gstappsrc.h>
|
||||
|
||||
#define N_BUFFERS 3
|
||||
|
||||
|
|
@ -35,14 +37,14 @@ typedef struct
|
|||
|
||||
static void
|
||||
gstimagecapturebin_init_test_context (GstImageCaptureBinTestContext * ctx,
|
||||
gint num_buffers)
|
||||
const gchar * src, gint num_buffers)
|
||||
{
|
||||
fail_unless (ctx != NULL);
|
||||
|
||||
ctx->pipe = gst_pipeline_new ("pipeline");
|
||||
fail_unless (ctx->pipe != NULL);
|
||||
ctx->src = gst_element_factory_make ("videotestsrc", "src");
|
||||
fail_unless (ctx->src != NULL, "Failed to create videotestsrc element");
|
||||
ctx->src = gst_element_factory_make (src, "src");
|
||||
fail_unless (ctx->src != NULL, "Failed to create src element");
|
||||
ctx->icbin = gst_element_factory_make ("imagecapturebin", "icbin");
|
||||
fail_unless (ctx->icbin != NULL, "Failed to create imagecapturebin element");
|
||||
|
||||
|
|
@ -51,7 +53,7 @@ gstimagecapturebin_init_test_context (GstImageCaptureBinTestContext * ctx,
|
|||
|
||||
fail_unless (gst_bin_add (GST_BIN (ctx->pipe), ctx->src));
|
||||
fail_unless (gst_bin_add (GST_BIN (ctx->pipe), ctx->icbin));
|
||||
fail_unless (gst_element_link (ctx->src, ctx->icbin));
|
||||
fail_unless (gst_element_link_many (ctx->src, ctx->icbin, NULL));
|
||||
}
|
||||
|
||||
static void
|
||||
|
|
@ -69,6 +71,70 @@ make_test_file_name (void)
|
|||
"imagecapturebintest_%%d.cap", g_get_tmp_dir ());
|
||||
}
|
||||
|
||||
static gboolean
|
||||
get_file_info (const gchar * filename, gint * width, gint * height)
|
||||
{
|
||||
GstElement *playbin = gst_element_factory_make ("playbin2", NULL);
|
||||
GstElement *fakesink = gst_element_factory_make ("fakesink", NULL);
|
||||
GstState state = GST_STATE_NULL;
|
||||
GstPad *pad;
|
||||
GstCaps *caps;
|
||||
gchar *uri = g_strdup_printf ("file://%s", filename);
|
||||
|
||||
g_object_set (playbin, "video-sink", fakesink, NULL);
|
||||
g_object_set (playbin, "uri", uri, NULL);
|
||||
g_free (uri);
|
||||
|
||||
gst_element_set_state (playbin, GST_STATE_PAUSED);
|
||||
|
||||
gst_element_get_state (playbin, &state, NULL, GST_SECOND * 5);
|
||||
|
||||
fail_unless (state == GST_STATE_PAUSED);
|
||||
|
||||
g_signal_emit_by_name (playbin, "get-video-pad", 0, &pad, NULL);
|
||||
caps = gst_pad_get_negotiated_caps (pad);
|
||||
fail_unless (gst_structure_get_int (gst_caps_get_structure (caps, 0), "width",
|
||||
width));
|
||||
fail_unless (gst_structure_get_int (gst_caps_get_structure (caps, 0),
|
||||
"height", height));
|
||||
|
||||
gst_object_unref (pad);
|
||||
gst_element_set_state (playbin, GST_STATE_NULL);
|
||||
gst_object_unref (playbin);
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
static GstBuffer *
|
||||
create_video_buffer (GstCaps * caps)
|
||||
{
|
||||
GstElement *pipeline;
|
||||
GstElement *cf;
|
||||
GstElement *sink;
|
||||
GstBuffer *buffer;
|
||||
|
||||
pipeline =
|
||||
gst_parse_launch
|
||||
("videotestsrc num-buffers=1 ! capsfilter name=cf ! appsink name=sink",
|
||||
NULL);
|
||||
g_assert (pipeline != NULL);
|
||||
|
||||
cf = gst_bin_get_by_name (GST_BIN (pipeline), "cf");
|
||||
sink = gst_bin_get_by_name (GST_BIN (pipeline), "sink");
|
||||
|
||||
g_object_set (G_OBJECT (cf), "caps", caps, NULL);
|
||||
|
||||
gst_element_set_state (pipeline, GST_STATE_PLAYING);
|
||||
|
||||
buffer = gst_app_sink_pull_buffer (GST_APP_SINK (sink));
|
||||
|
||||
gst_element_set_state (pipeline, GST_STATE_NULL);
|
||||
gst_object_unref (pipeline);
|
||||
gst_object_unref (sink);
|
||||
gst_object_unref (cf);
|
||||
return buffer;
|
||||
}
|
||||
|
||||
|
||||
GST_START_TEST (test_simple_capture)
|
||||
{
|
||||
GstImageCaptureBinTestContext ctx;
|
||||
|
|
@ -77,7 +143,7 @@ GST_START_TEST (test_simple_capture)
|
|||
gchar *test_file_name;
|
||||
gint i;
|
||||
|
||||
gstimagecapturebin_init_test_context (&ctx, N_BUFFERS);
|
||||
gstimagecapturebin_init_test_context (&ctx, "videotestsrc", N_BUFFERS);
|
||||
bus = gst_element_get_bus (ctx.pipe);
|
||||
|
||||
test_file_name = make_test_file_name ();
|
||||
|
|
@ -119,6 +185,89 @@ GST_START_TEST (test_simple_capture)
|
|||
|
||||
GST_END_TEST;
|
||||
|
||||
|
||||
GST_START_TEST (test_multiple_captures_different_caps)
|
||||
{
|
||||
GstImageCaptureBinTestContext ctx;
|
||||
GstBus *bus;
|
||||
GstMessage *msg;
|
||||
gchar *test_file_name;
|
||||
gint i;
|
||||
gint widths[] = { 100, 300, 200 };
|
||||
gint heights[] = { 300, 200, 100 };
|
||||
GstPad *pad;
|
||||
|
||||
gstimagecapturebin_init_test_context (&ctx, "appsrc", N_BUFFERS);
|
||||
bus = gst_element_get_bus (ctx.pipe);
|
||||
|
||||
test_file_name = make_test_file_name ();
|
||||
g_object_set (ctx.icbin, "location", test_file_name, NULL);
|
||||
fail_if (gst_element_set_state (ctx.pipe, GST_STATE_PLAYING) ==
|
||||
GST_STATE_CHANGE_FAILURE);
|
||||
|
||||
/* push data directly because set_caps and buffer pushes on appsrc
|
||||
* are not serialized into the flow, so we can't guarantee the buffers
|
||||
* have the caps we want on them when pushed */
|
||||
pad = gst_element_get_static_pad (ctx.src, "src");
|
||||
|
||||
/* push the buffers */
|
||||
for (i = 0; i < N_BUFFERS; i++) {
|
||||
GstCaps *caps;
|
||||
GstBuffer *buf;
|
||||
|
||||
caps = gst_caps_new_simple ("video/x-raw-yuv", "width", G_TYPE_INT,
|
||||
widths[i], "height", G_TYPE_INT, heights[i], "framerate",
|
||||
GST_TYPE_FRACTION, 1, 1, "format", GST_TYPE_FOURCC,
|
||||
GST_MAKE_FOURCC ('I', '4', '2', '0'), NULL);
|
||||
|
||||
buf = create_video_buffer (caps);
|
||||
fail_if (buf == NULL);
|
||||
|
||||
fail_unless (gst_pad_push (pad, buf) == GST_FLOW_OK);
|
||||
gst_caps_unref (caps);
|
||||
}
|
||||
gst_app_src_end_of_stream (GST_APP_SRC (ctx.src));
|
||||
gst_object_unref (pad);
|
||||
|
||||
msg = gst_bus_timed_pop_filtered (bus, GST_SECOND * 10,
|
||||
GST_MESSAGE_EOS | GST_MESSAGE_ERROR);
|
||||
fail_unless (msg != NULL);
|
||||
fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
|
||||
gst_message_unref (msg);
|
||||
|
||||
/* check there are N_BUFFERS files */
|
||||
for (i = 0; i < N_BUFFERS; i++) {
|
||||
gchar *filename;
|
||||
FILE *f;
|
||||
gint width = 0, height = 0;
|
||||
|
||||
filename = g_strdup_printf (test_file_name, i);
|
||||
|
||||
fail_unless (g_file_test (filename, G_FILE_TEST_EXISTS));
|
||||
fail_unless (g_file_test (filename, G_FILE_TEST_IS_REGULAR));
|
||||
fail_if (g_file_test (filename, G_FILE_TEST_IS_SYMLINK));
|
||||
|
||||
/* check the file isn't empty */
|
||||
f = fopen (filename, "r");
|
||||
fseek (f, 0, SEEK_END);
|
||||
fail_unless (ftell (f) > 0);
|
||||
fclose (f);
|
||||
|
||||
/* get the file info */
|
||||
fail_unless (get_file_info (filename, &width, &height));
|
||||
fail_unless (width == widths[i]);
|
||||
fail_unless (height == heights[i]);
|
||||
|
||||
g_free (filename);
|
||||
}
|
||||
|
||||
gstimagecapturebin_unset_test_context (&ctx);
|
||||
gst_object_unref (bus);
|
||||
g_free (test_file_name);
|
||||
}
|
||||
|
||||
GST_END_TEST;
|
||||
|
||||
GST_START_TEST (test_setting_encoder)
|
||||
{
|
||||
GstImageCaptureBinTestContext ctx;
|
||||
|
|
@ -128,7 +277,7 @@ GST_START_TEST (test_setting_encoder)
|
|||
gchar *test_file_name;
|
||||
gint i;
|
||||
|
||||
gstimagecapturebin_init_test_context (&ctx, N_BUFFERS);
|
||||
gstimagecapturebin_init_test_context (&ctx, "videotestsrc", N_BUFFERS);
|
||||
bus = gst_element_get_bus (ctx.pipe);
|
||||
|
||||
test_file_name = make_test_file_name ();
|
||||
|
|
@ -181,7 +330,7 @@ GST_START_TEST (test_setting_muxer)
|
|||
gchar *test_file_name;
|
||||
gint i;
|
||||
|
||||
gstimagecapturebin_init_test_context (&ctx, N_BUFFERS);
|
||||
gstimagecapturebin_init_test_context (&ctx, "videotestsrc", N_BUFFERS);
|
||||
bus = gst_element_get_bus (ctx.pipe);
|
||||
|
||||
test_file_name = make_test_file_name ();
|
||||
|
|
@ -236,6 +385,7 @@ imagecapturebin_suite (void)
|
|||
|
||||
suite_add_tcase (s, tc_chain);
|
||||
tcase_add_test (tc_chain, test_simple_capture);
|
||||
tcase_add_test (tc_chain, test_multiple_captures_different_caps);
|
||||
tcase_add_test (tc_chain, test_setting_encoder);
|
||||
tcase_add_test (tc_chain, test_setting_muxer);
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue