diff --git a/tests/check/Makefile.am b/tests/check/Makefile.am index 1611bb8f23..b7a9857a8c 100644 --- a/tests/check/Makefile.am +++ b/tests/check/Makefile.am @@ -117,6 +117,7 @@ check_PROGRAMS = \ libs/tag \ libs/video \ libs/videodecoder \ + libs/videoencoder \ libs/xmpwriter \ $(cxx_checks) \ $(check_orc) \ @@ -453,6 +454,16 @@ libs_videodecoder_LDADD = \ $(GST_BASE_LIBS) \ $(LDADD) +libs_videoencoder_CFLAGS = \ + $(GST_PLUGINS_BASE_CFLAGS) \ + $(GST_BASE_CFLAGS) \ + $(AM_CFLAGS) + +libs_videoencoder_LDADD = \ + $(top_builddir)/gst-libs/gst/video/libgstvideo-@GST_API_VERSION@.la \ + $(GST_BASE_LIBS) \ + $(LDADD) + elements_multisocketsink_CFLAGS = $(GIO_CFLAGS) $(AM_CFLAGS) elements_multisocketsink_LDADD = $(GIO_LIBS) $(LDADD) diff --git a/tests/check/libs/.gitignore b/tests/check/libs/.gitignore index 18aeed69ca..71edb75b93 100644 --- a/tests/check/libs/.gitignore +++ b/tests/check/libs/.gitignore @@ -18,4 +18,5 @@ tag utils video videodecoder +videoencoder xmpwriter diff --git a/tests/check/libs/videoencoder.c b/tests/check/libs/videoencoder.c new file mode 100644 index 0000000000..6a60e926f0 --- /dev/null +++ b/tests/check/libs/videoencoder.c @@ -0,0 +1,290 @@ +/* GStreamer + * + * Copyright (C) 2014 Samsung Electronics. All rights reserved. + * Author: Thiago Santos + * + * 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. + */ + +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif +#include +#include +#include +#include + +static GstPad *mysrcpad, *mysinkpad; +static GstElement *dec; +static GList *events = NULL; + +#define TEST_VIDEO_WIDTH 640 +#define TEST_VIDEO_HEIGHT 480 +#define TEST_VIDEO_FPS_N 30 +#define TEST_VIDEO_FPS_D 1 + +#define GST_VIDEO_ENCODER_TESTER_TYPE gst_video_encoder_tester_get_type() +static GType gst_video_encoder_tester_get_type (void); + +typedef struct _GstVideoEncoderTester GstVideoEncoderTester; +typedef struct _GstVideoEncoderTesterClass GstVideoEncoderTesterClass; + +struct _GstVideoEncoderTester +{ + GstVideoEncoder parent; +}; + +struct _GstVideoEncoderTesterClass +{ + GstVideoEncoderClass parent_class; +}; + +G_DEFINE_TYPE (GstVideoEncoderTester, gst_video_encoder_tester, + GST_TYPE_VIDEO_ENCODER); + +static gboolean +gst_video_encoder_tester_start (GstVideoEncoder * dec) +{ + return TRUE; +} + +static gboolean +gst_video_encoder_tester_stop (GstVideoEncoder * dec) +{ + return TRUE; +} + +static gboolean +gst_video_encoder_tester_set_format (GstVideoEncoder * dec, + GstVideoCodecState * state) +{ + GstVideoCodecState *res = gst_video_encoder_set_output_state (dec, + gst_caps_new_simple ("video/x-test-custom", "width", G_TYPE_INT, + 480, "height", G_TYPE_INT, 360, NULL), + NULL); + + gst_video_codec_state_unref (res); + return TRUE; +} + +static GstFlowReturn +gst_video_encoder_tester_handle_frame (GstVideoEncoder * dec, + GstVideoCodecFrame * frame) +{ + guint8 *data; + GstMapInfo map; + guint64 input_num; + + gst_buffer_map (frame->input_buffer, &map, GST_MAP_READ); + input_num = *((guint64 *) map.data); + gst_buffer_unmap (frame->input_buffer, &map); + + data = g_malloc (sizeof (guint64)); + *(guint64 *) data = input_num; + + frame->output_buffer = gst_buffer_new_wrapped (data, sizeof (guint64)); + frame->pts = GST_BUFFER_PTS (frame->input_buffer); + frame->duration = GST_BUFFER_DURATION (frame->input_buffer); + + return gst_video_encoder_finish_frame (dec, frame); +} + +static void +gst_video_encoder_tester_class_init (GstVideoEncoderTesterClass * klass) +{ + GstElementClass *element_class = GST_ELEMENT_CLASS (klass); + GstVideoEncoderClass *videoencoder_class = GST_VIDEO_ENCODER_CLASS (klass); + + static GstStaticPadTemplate sink_templ = GST_STATIC_PAD_TEMPLATE ("sink", + GST_PAD_SINK, GST_PAD_ALWAYS, + GST_STATIC_CAPS ("video/x-raw")); + + static GstStaticPadTemplate src_templ = GST_STATIC_PAD_TEMPLATE ("src", + GST_PAD_SRC, GST_PAD_ALWAYS, + GST_STATIC_CAPS ("video/x-test-custom")); + + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&sink_templ)); + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&src_templ)); + + gst_element_class_set_metadata (element_class, + "VideoEncoderTester", "Encoder/Video", "yep", "me"); + + videoencoder_class->start = gst_video_encoder_tester_start; + videoencoder_class->stop = gst_video_encoder_tester_stop; + videoencoder_class->handle_frame = gst_video_encoder_tester_handle_frame; + videoencoder_class->set_format = gst_video_encoder_tester_set_format; +} + +static void +gst_video_encoder_tester_init (GstVideoEncoderTester * tester) +{ +} + +static gboolean +_mysinkpad_event (GstPad * pad, GstObject * parent, GstEvent * event) +{ + events = g_list_append (events, event); + return TRUE; +} + +static void +setup_videoencodertester (void) +{ + GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink", + GST_PAD_SINK, + GST_PAD_ALWAYS, + GST_STATIC_CAPS ("video/x-test-custom") + ); + GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_STATIC_CAPS ("video/x-raw") + ); + + dec = g_object_new (GST_VIDEO_ENCODER_TESTER_TYPE, NULL); + mysrcpad = gst_check_setup_src_pad (dec, &srctemplate); + mysinkpad = gst_check_setup_sink_pad (dec, &sinktemplate); + + gst_pad_set_event_function (mysinkpad, _mysinkpad_event); +} + +static void +cleanup_videoencodertest (void) +{ + gst_pad_set_active (mysrcpad, FALSE); + gst_pad_set_active (mysinkpad, FALSE); + gst_check_teardown_src_pad (dec); + gst_check_teardown_sink_pad (dec); + gst_check_teardown_element (dec); + + g_list_free_full (events, (GDestroyNotify) gst_event_unref); + events = NULL; +} + +static GstBuffer * +create_test_buffer (guint64 num) +{ + GstBuffer *buffer; + guint64 *data = g_malloc (sizeof (guint64)); + + *data = num; + + buffer = gst_buffer_new_wrapped (data, sizeof (guint64)); + + GST_BUFFER_PTS (buffer) = + gst_util_uint64_scale_round (num, GST_SECOND * TEST_VIDEO_FPS_D, + TEST_VIDEO_FPS_N); + GST_BUFFER_DURATION (buffer) = + gst_util_uint64_scale_round (GST_SECOND, TEST_VIDEO_FPS_D, + TEST_VIDEO_FPS_N); + + return buffer; +} + +static void +send_startup_events (void) +{ + GstCaps *caps; + + fail_unless (gst_pad_push_event (mysrcpad, + gst_event_new_stream_start ("randomvalue"))); + + /* push caps */ + caps = + gst_caps_new_simple ("video/x-raw", "width", G_TYPE_INT, + TEST_VIDEO_WIDTH, "height", G_TYPE_INT, TEST_VIDEO_HEIGHT, "framerate", + GST_TYPE_FRACTION, TEST_VIDEO_FPS_N, TEST_VIDEO_FPS_D, + "format", G_TYPE_STRING, "GRAY8", NULL); + fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_caps (caps))); + gst_caps_unref (caps); +} + +#define NUM_BUFFERS 100 +GST_START_TEST (videoencoder_playback) +{ + GstSegment segment; + GstBuffer *buffer; + guint64 i; + GList *iter; + + setup_videoencodertester (); + + gst_pad_set_active (mysrcpad, TRUE); + gst_element_set_state (dec, GST_STATE_PLAYING); + gst_pad_set_active (mysinkpad, TRUE); + + send_startup_events (); + + /* push a new segment */ + gst_segment_init (&segment, GST_FORMAT_TIME); + fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_segment (&segment))); + + /* push buffers, the data is actually a number so we can track them */ + for (i = 0; i < NUM_BUFFERS; i++) { + buffer = create_test_buffer (i); + + fail_unless (gst_pad_push (mysrcpad, buffer) == GST_FLOW_OK); + } + + fail_unless (gst_pad_push_event (mysrcpad, gst_event_new_eos ())); + + /* check that all buffers were received by our source pad */ + fail_unless (g_list_length (buffers) == NUM_BUFFERS); + i = 0; + for (iter = buffers; iter; iter = g_list_next (iter)) { + GstMapInfo map; + guint64 num; + + buffer = iter->data; + + gst_buffer_map (buffer, &map, GST_MAP_READ); + + num = *(guint64 *) map.data; + fail_unless (i == num); + fail_unless (GST_BUFFER_PTS (buffer) == gst_util_uint64_scale_round (i, + GST_SECOND * TEST_VIDEO_FPS_D, TEST_VIDEO_FPS_N)); + fail_unless (GST_BUFFER_DURATION (buffer) == + gst_util_uint64_scale_round (GST_SECOND, TEST_VIDEO_FPS_D, + TEST_VIDEO_FPS_N)); + + gst_buffer_unmap (buffer, &map); + i++; + } + + g_list_free_full (buffers, (GDestroyNotify) gst_buffer_unref); + buffers = NULL; + + cleanup_videoencodertest (); +} + +GST_END_TEST; + + +static Suite * +gst_videoencoder_suite (void) +{ + Suite *s = suite_create ("GstVideoEncoder"); + TCase *tc = tcase_create ("general"); + + suite_add_tcase (s, tc); + tcase_add_test (tc, videoencoder_playback); + + return s; +} + +GST_CHECK_MAIN (gst_videoencoder);