mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-15 12:56:33 +00:00
162383e6b2
Warnings passed me by on the last commits to camerabin2 tests, fixing them.
410 lines
12 KiB
C
410 lines
12 KiB
C
/* GStreamer unit test for the imagecapturebin element
|
|
* Copyright (C) 2010 Thiago Santos <thiago.sousa.santos@collabora.co.uk>
|
|
*
|
|
* 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., 59 Temple Place - Suite 330,
|
|
* Boston, MA 02111-1307, USA.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
# include "config.h"
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
#include <gst/check/gstcheck.h>
|
|
#include <gst/app/gstappsink.h>
|
|
#include <gst/app/gstappsrc.h>
|
|
|
|
#define N_BUFFERS 3
|
|
|
|
typedef struct
|
|
{
|
|
GstElement *pipe;
|
|
GstElement *src;
|
|
GstElement *icbin;
|
|
} GstImageCaptureBinTestContext;
|
|
|
|
static void
|
|
gstimagecapturebin_init_test_context (GstImageCaptureBinTestContext * ctx,
|
|
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 (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");
|
|
|
|
if (num_buffers > 0)
|
|
g_object_set (ctx->src, "num-buffers", num_buffers, NULL);
|
|
|
|
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_many (ctx->src, ctx->icbin, NULL));
|
|
}
|
|
|
|
static void
|
|
gstimagecapturebin_unset_test_context (GstImageCaptureBinTestContext * ctx)
|
|
{
|
|
gst_element_set_state (ctx->pipe, GST_STATE_NULL);
|
|
gst_object_unref (ctx->pipe);
|
|
memset (ctx, 0, sizeof (GstImageCaptureBinTestContext));
|
|
}
|
|
|
|
static gchar *
|
|
make_test_file_name (void)
|
|
{
|
|
return g_strdup_printf ("%s" G_DIR_SEPARATOR_S
|
|
"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;
|
|
GstBus *bus;
|
|
GstMessage *msg;
|
|
gchar *test_file_name;
|
|
gint i;
|
|
|
|
gstimagecapturebin_init_test_context (&ctx, "videotestsrc", 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);
|
|
|
|
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;
|
|
|
|
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);
|
|
|
|
g_free (filename);
|
|
}
|
|
|
|
gstimagecapturebin_unset_test_context (&ctx);
|
|
gst_object_unref (bus);
|
|
g_free (test_file_name);
|
|
}
|
|
|
|
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;
|
|
GstBus *bus;
|
|
GstMessage *msg;
|
|
GstElement *encoder;
|
|
gchar *test_file_name;
|
|
gint i;
|
|
|
|
gstimagecapturebin_init_test_context (&ctx, "videotestsrc", 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);
|
|
|
|
encoder = gst_element_factory_make ("jpegenc", NULL);
|
|
g_object_set (ctx.icbin, "image-encoder", encoder, NULL);
|
|
|
|
fail_if (gst_element_set_state (ctx.pipe, GST_STATE_PLAYING) ==
|
|
GST_STATE_CHANGE_FAILURE);
|
|
|
|
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);
|
|
|
|
/* check there are N_BUFFERS files */
|
|
for (i = 0; i < N_BUFFERS; i++) {
|
|
gchar *filename;
|
|
FILE *f;
|
|
|
|
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);
|
|
|
|
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_muxer)
|
|
{
|
|
GstImageCaptureBinTestContext ctx;
|
|
GstBus *bus;
|
|
GstMessage *msg;
|
|
GstElement *encoder;
|
|
gchar *test_file_name;
|
|
gint i;
|
|
|
|
gstimagecapturebin_init_test_context (&ctx, "videotestsrc", 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);
|
|
|
|
encoder = gst_element_factory_make ("pngenc", NULL);
|
|
g_object_set (ctx.icbin, "image-encoder", encoder, NULL);
|
|
|
|
encoder = gst_element_factory_make ("identity", NULL);
|
|
g_object_set (ctx.icbin, "image-muxer", encoder, NULL);
|
|
|
|
fail_if (gst_element_set_state (ctx.pipe, GST_STATE_PLAYING) ==
|
|
GST_STATE_CHANGE_FAILURE);
|
|
|
|
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);
|
|
|
|
/* check there are N_BUFFERS files */
|
|
for (i = 0; i < N_BUFFERS; i++) {
|
|
gchar *filename;
|
|
FILE *f;
|
|
|
|
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);
|
|
|
|
g_free (filename);
|
|
}
|
|
|
|
gstimagecapturebin_unset_test_context (&ctx);
|
|
gst_object_unref (bus);
|
|
g_free (test_file_name);
|
|
}
|
|
|
|
GST_END_TEST;
|
|
|
|
static Suite *
|
|
imagecapturebin_suite (void)
|
|
{
|
|
GstElementFactory *jpegenc_factory;
|
|
|
|
Suite *s = suite_create ("imagecapturebin");
|
|
TCase *tc_chain = tcase_create ("general");
|
|
|
|
jpegenc_factory = gst_element_factory_find ("jpegenc");
|
|
|
|
suite_add_tcase (s, tc_chain);
|
|
if (jpegenc_factory) {
|
|
tcase_add_test (tc_chain, test_simple_capture);
|
|
|
|
/* only adds this test if jpegenc contains the fix for its getcaps
|
|
* The fix on good: dcbba0932dc579abd6aab4460fa1a416374eda1b */
|
|
if (gst_plugin_feature_check_version ((GstPluginFeature *) jpegenc_factory,
|
|
0, 10, 27))
|
|
tcase_add_test (tc_chain, test_multiple_captures_different_caps);
|
|
else
|
|
GST_WARNING ("Skipped test that needs gst-plugins-good 0.10.27");
|
|
|
|
tcase_add_test (tc_chain, test_setting_encoder);
|
|
tcase_add_test (tc_chain, test_setting_muxer);
|
|
} else
|
|
GST_WARNING ("Skipped imagecapturebin tests because jpegenc is missing");
|
|
|
|
return s;
|
|
}
|
|
|
|
GST_CHECK_MAIN (imagecapturebin);
|