From 8b9904b6d03fab84779ece20f5efc6978e5a1f15 Mon Sep 17 00:00:00 2001 From: Wim Taymans <wtaymans@redhat.com> Date: Wed, 11 Jun 2014 17:29:33 +0200 Subject: [PATCH] shell-recorder-src: allow sending EOS to our source When we send EOS to our source, make it queue the special item to cause EOS after all buffers are pushed. --- src/shell-recorder-src.c | 32 +++++++++++++++++++++++++++----- 1 file changed, 27 insertions(+), 5 deletions(-) diff --git a/src/shell-recorder-src.c b/src/shell-recorder-src.c index b26d5ca3f..9af7aab8a 100644 --- a/src/shell-recorder-src.c +++ b/src/shell-recorder-src.c @@ -35,6 +35,7 @@ enum { /* Special marker value once the source is closed */ #define RECORDER_QUEUE_END ((GstBuffer *)1) +#define shell_recorder_src_parent_class parent_class G_DEFINE_TYPE(ShellRecorderSrc, shell_recorder_src, GST_TYPE_PUSH_SRC); static void @@ -93,6 +94,26 @@ shell_recorder_src_negotiate (GstBaseSrc * base_src) return result; } +static gboolean +shell_recorder_src_send_event (GstElement * element, GstEvent * event) +{ + ShellRecorderSrc *src = SHELL_RECORDER_SRC (element); + gboolean res; + + switch (GST_EVENT_TYPE (event)) { + case GST_EVENT_EOS: + shell_recorder_src_close (src); + gst_event_unref (event); + res = TRUE; + break; + default: + res = GST_CALL_PARENT_WITH_DEFAULT (GST_ELEMENT_CLASS, send_event, (element, + event), FALSE); + break; + } + return res; +} + /* The create() virtual function is responsible for returning the next buffer. * We just pop buffers off of the queue and block if necessary. */ @@ -224,11 +245,6 @@ shell_recorder_src_class_init (ShellRecorderSrcClass *klass) object_class->set_property = shell_recorder_src_set_property; object_class->get_property = shell_recorder_src_get_property; - base_src_class->negotiate = shell_recorder_src_negotiate; - - push_src_class->create = shell_recorder_src_create; - - g_object_class_install_property (object_class, PROP_CAPS, g_param_spec_boxed ("caps", @@ -251,6 +267,12 @@ shell_recorder_src_class_init (ShellRecorderSrcClass *klass) "Generic/Src", "Feed screen capture data to a pipeline", "Owen Taylor <otaylor@redhat.com>"); + + element_class->send_event = shell_recorder_src_send_event; + + base_src_class->negotiate = shell_recorder_src_negotiate; + + push_src_class->create = shell_recorder_src_create; } /**