2 * Copyright (C) 2003 Benjamin Otte <in7y118@public.uni-hamburg.de>
4 * gsttypefindelement.c: element that detects type of stream
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Library General Public
8 * License as published by the Free Software Foundation; either
9 * version 2 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Library General Public License for more details.
16 * You should have received a copy of the GNU Library General Public
17 * License along with this library; if not, write to the
18 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
19 * Boston, MA 02110-1301, USA.
22 * SECTION:element-typefind
24 * Determines the media-type of a stream. It applies typefind functions in the
25 * order of their rank. Once the type has been detected it sets its src pad caps
26 * to the found media type.
28 * Whenever a type is found the #GstTypeFindElement::have-type signal is
29 * emitted, either from the streaming thread or the application thread
30 * (the latter may happen when typefinding is done pull-based from the
31 * state change function).
33 * Plugins can register custom typefinders by using #GstTypeFindFactory.
36 /* FIXME: need a better solution for non-seekable streams */
39 * 1) get a list of all typefind functions sorted best to worst
40 * 2) if all elements have been called with all requested data goto 8
41 * 3) call all functions once with all available data
42 * 4) if a function returns a value >= PROP_MAXIMUM goto 8 (never implemented))
43 * 5) all functions with a result > PROP_MINIMUM or functions that did not get
44 * all requested data (where peek returned NULL) stay in list
45 * 6) seek to requested offset of best function that still has open data
48 * 8) take best available result and use its caps
50 * The element has two scheduling modes:
52 * 1) chain based, it will collect buffers and run the typefind function on
53 * the buffer until something is found.
54 * 2) getrange based, it will proxy the getrange function to the sinkpad. It
55 * is assumed that the peer element is happy with whatever format we
58 * By default it tries to do pull based typefinding (this avoids joining
59 * received buffers and holding them back in store.)
61 * When the element has no connected srcpad, and the sinkpad can operate in
62 * getrange based mode, the element starts its own task to figure out the
65 * Most of the actual implementation is in libs/gst/base/gsttypefindhelper.c.
72 #include "gst/gst_private.h"
74 #include "gsttypefindelement.h"
75 #include "gst/gst-i18n-lib.h"
76 #include "gst/base/gsttypefindhelper.h"
78 #include <gst/gsttypefind.h>
79 #include <gst/gstutils.h>
80 #include <gst/gsterror.h>
82 GST_DEBUG_CATEGORY_STATIC (gst_type_find_element_debug);
83 #define GST_CAT_DEFAULT gst_type_find_element_debug
85 /* generic templates */
86 static GstStaticPadTemplate type_find_element_sink_template =
87 GST_STATIC_PAD_TEMPLATE ("sink",
92 static GstStaticPadTemplate type_find_element_src_template =
93 GST_STATIC_PAD_TEMPLATE ("src",
98 /* Require at least 2kB of data before we attempt typefinding in chain-mode.
99 * 128kB is massive overkill for the maximum, but doesn't do any harm */
100 #define TYPE_FIND_MIN_SIZE (2*1024)
101 #define TYPE_FIND_MAX_SIZE (128*1024)
103 /* TypeFind signals and args */
119 MODE_NORMAL, /* act as identity */
120 MODE_TYPEFIND, /* do typefinding */
121 MODE_ERROR /* had fatal error */
126 GST_DEBUG_CATEGORY_INIT (gst_type_find_element_debug, "typefind", \
127 GST_DEBUG_BG_YELLOW | GST_DEBUG_FG_GREEN, "type finding element");
128 #define gst_type_find_element_parent_class parent_class
129 G_DEFINE_TYPE_WITH_CODE (GstTypeFindElement, gst_type_find_element,
130 GST_TYPE_ELEMENT, _do_init);
132 static void gst_type_find_element_dispose (GObject * object);
133 static void gst_type_find_element_set_property (GObject * object,
134 guint prop_id, const GValue * value, GParamSpec * pspec);
135 static void gst_type_find_element_get_property (GObject * object,
136 guint prop_id, GValue * value, GParamSpec * pspec);
138 static gboolean gst_type_find_element_src_event (GstPad * pad,
139 GstObject * parent, GstEvent * event);
140 static gboolean gst_type_find_handle_src_query (GstPad * pad,
141 GstObject * parent, GstQuery * query);
143 static gboolean gst_type_find_element_sink_event (GstPad * pad,
144 GstObject * parent, GstEvent * event);
145 static gboolean gst_type_find_element_setcaps (GstTypeFindElement * typefind,
147 static GstFlowReturn gst_type_find_element_chain (GstPad * sinkpad,
148 GstObject * parent, GstBuffer * buffer);
149 static GstFlowReturn gst_type_find_element_getrange (GstPad * srcpad,
150 GstObject * parent, guint64 offset, guint length, GstBuffer ** buffer);
152 static GstStateChangeReturn
153 gst_type_find_element_change_state (GstElement * element,
154 GstStateChange transition);
155 static gboolean gst_type_find_element_activate_sink (GstPad * pad,
157 static gboolean gst_type_find_element_activate_sink_mode (GstPad * pad,
158 GstObject * parent, GstPadMode mode, gboolean active);
159 static gboolean gst_type_find_element_activate_src_mode (GstPad * pad,
160 GstObject * parent, GstPadMode mode, gboolean active);
162 gst_type_find_element_chain_do_typefinding (GstTypeFindElement * typefind,
163 gboolean check_avail, gboolean at_eos);
164 static void gst_type_find_element_send_cached_events (GstTypeFindElement *
167 static void gst_type_find_element_loop (GstPad * pad);
169 static guint gst_type_find_element_signals[LAST_SIGNAL] = { 0 };
172 gst_type_find_element_have_type (GstTypeFindElement * typefind,
173 guint probability, GstCaps * caps)
175 g_assert (caps != NULL);
177 GST_INFO_OBJECT (typefind, "found caps %" GST_PTR_FORMAT ", probability=%u",
180 GST_OBJECT_LOCK (typefind);
182 gst_caps_unref (typefind->caps);
183 typefind->caps = gst_caps_ref (caps);
184 GST_OBJECT_UNLOCK (typefind);
186 gst_pad_set_caps (typefind->src, caps);
190 gst_type_find_element_class_init (GstTypeFindElementClass * typefind_class)
192 GObjectClass *gobject_class = G_OBJECT_CLASS (typefind_class);
193 GstElementClass *gstelement_class = GST_ELEMENT_CLASS (typefind_class);
195 gobject_class->set_property = gst_type_find_element_set_property;
196 gobject_class->get_property = gst_type_find_element_get_property;
197 gobject_class->dispose = gst_type_find_element_dispose;
199 g_object_class_install_property (gobject_class, PROP_CAPS,
200 g_param_spec_boxed ("caps", _("caps"),
201 _("detected capabilities in stream"), GST_TYPE_CAPS,
202 G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
203 g_object_class_install_property (gobject_class, PROP_MINIMUM,
204 g_param_spec_uint ("minimum", _("minimum"),
205 "minimum probability required to accept caps", GST_TYPE_FIND_MINIMUM,
206 GST_TYPE_FIND_MAXIMUM, GST_TYPE_FIND_MINIMUM,
207 G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
208 g_object_class_install_property (gobject_class, PROP_FORCE_CAPS,
209 g_param_spec_boxed ("force-caps", _("force caps"),
210 _("force caps without doing a typefind"), GST_TYPE_CAPS,
211 G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
213 * GstTypeFindElement::have-type:
214 * @typefind: the typefind instance
215 * @probability: the probability of the type found
216 * @caps: the caps of the type found
218 * This signal gets emitted when the type and its probability has
221 gst_type_find_element_signals[HAVE_TYPE] = g_signal_new ("have-type",
222 G_TYPE_FROM_CLASS (typefind_class), G_SIGNAL_RUN_LAST,
223 G_STRUCT_OFFSET (GstTypeFindElementClass, have_type), NULL, NULL,
224 g_cclosure_marshal_generic, G_TYPE_NONE, 2,
225 G_TYPE_UINT, GST_TYPE_CAPS | G_SIGNAL_TYPE_STATIC_SCOPE);
227 typefind_class->have_type =
228 GST_DEBUG_FUNCPTR (gst_type_find_element_have_type);
230 gst_element_class_set_static_metadata (gstelement_class,
233 "Finds the media type of a stream",
234 "Benjamin Otte <in7y118@public.uni-hamburg.de>");
235 gst_element_class_add_pad_template (gstelement_class,
236 gst_static_pad_template_get (&type_find_element_src_template));
237 gst_element_class_add_pad_template (gstelement_class,
238 gst_static_pad_template_get (&type_find_element_sink_template));
240 gstelement_class->change_state =
241 GST_DEBUG_FUNCPTR (gst_type_find_element_change_state);
245 gst_type_find_element_init (GstTypeFindElement * typefind)
249 gst_pad_new_from_static_template (&type_find_element_sink_template,
252 gst_pad_set_activate_function (typefind->sink,
253 GST_DEBUG_FUNCPTR (gst_type_find_element_activate_sink));
254 gst_pad_set_activatemode_function (typefind->sink,
255 GST_DEBUG_FUNCPTR (gst_type_find_element_activate_sink_mode));
256 gst_pad_set_chain_function (typefind->sink,
257 GST_DEBUG_FUNCPTR (gst_type_find_element_chain));
258 gst_pad_set_event_function (typefind->sink,
259 GST_DEBUG_FUNCPTR (gst_type_find_element_sink_event));
260 GST_PAD_SET_PROXY_ALLOCATION (typefind->sink);
261 gst_element_add_pad (GST_ELEMENT (typefind), typefind->sink);
265 gst_pad_new_from_static_template (&type_find_element_src_template, "src");
267 gst_pad_set_activatemode_function (typefind->src,
268 GST_DEBUG_FUNCPTR (gst_type_find_element_activate_src_mode));
269 gst_pad_set_getrange_function (typefind->src,
270 GST_DEBUG_FUNCPTR (gst_type_find_element_getrange));
271 gst_pad_set_event_function (typefind->src,
272 GST_DEBUG_FUNCPTR (gst_type_find_element_src_event));
273 gst_pad_set_query_function (typefind->src,
274 GST_DEBUG_FUNCPTR (gst_type_find_handle_src_query));
275 gst_pad_use_fixed_caps (typefind->src);
276 gst_element_add_pad (GST_ELEMENT (typefind), typefind->src);
278 typefind->mode = MODE_TYPEFIND;
279 typefind->caps = NULL;
280 typefind->min_probability = 1;
282 typefind->adapter = gst_adapter_new ();
286 gst_type_find_element_dispose (GObject * object)
288 GstTypeFindElement *typefind = GST_TYPE_FIND_ELEMENT (object);
290 if (typefind->adapter) {
291 g_object_unref (typefind->adapter);
292 typefind->adapter = NULL;
295 if (typefind->force_caps) {
296 gst_caps_unref (typefind->force_caps);
297 typefind->force_caps = NULL;
300 G_OBJECT_CLASS (parent_class)->dispose (object);
304 gst_type_find_element_set_property (GObject * object, guint prop_id,
305 const GValue * value, GParamSpec * pspec)
307 GstTypeFindElement *typefind;
309 typefind = GST_TYPE_FIND_ELEMENT (object);
313 typefind->min_probability = g_value_get_uint (value);
315 case PROP_FORCE_CAPS:
316 GST_OBJECT_LOCK (typefind);
317 if (typefind->force_caps)
318 gst_caps_unref (typefind->force_caps);
319 typefind->force_caps = g_value_dup_boxed (value);
320 GST_OBJECT_UNLOCK (typefind);
323 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
329 gst_type_find_element_get_property (GObject * object, guint prop_id,
330 GValue * value, GParamSpec * pspec)
332 GstTypeFindElement *typefind;
334 typefind = GST_TYPE_FIND_ELEMENT (object);
338 GST_OBJECT_LOCK (typefind);
339 g_value_set_boxed (value, typefind->caps);
340 GST_OBJECT_UNLOCK (typefind);
343 g_value_set_uint (value, typefind->min_probability);
345 case PROP_FORCE_CAPS:
346 GST_OBJECT_LOCK (typefind);
347 g_value_set_boxed (value, typefind->force_caps);
348 GST_OBJECT_UNLOCK (typefind);
351 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
357 gst_type_find_handle_src_query (GstPad * pad, GstObject * parent,
360 GstTypeFindElement *typefind;
361 gboolean res = FALSE;
363 typefind = GST_TYPE_FIND_ELEMENT (parent);
364 GST_DEBUG_OBJECT (typefind, "Handling src query %s",
365 GST_QUERY_TYPE_NAME (query));
367 switch (GST_QUERY_TYPE (query)) {
368 case GST_QUERY_SCHEDULING:
369 /* FIXME, filter out the scheduling modes that we understand */
370 res = gst_pad_peer_query (typefind->sink, query);
374 GST_DEBUG_OBJECT (typefind,
375 "Got caps query, our caps are %" GST_PTR_FORMAT, typefind->caps);
377 /* We can hijack caps query if we typefind already */
378 if (typefind->caps) {
379 gst_query_set_caps_result (query, typefind->caps);
382 res = gst_pad_peer_query (typefind->sink, query);
386 case GST_QUERY_POSITION:
391 if (!(res = gst_pad_peer_query (typefind->sink, query)))
394 gst_query_parse_position (query, &format, &peer_pos);
396 GST_OBJECT_LOCK (typefind);
397 /* FIXME: this code assumes that there's no discont in the queue */
399 case GST_FORMAT_BYTES:
400 peer_pos -= gst_adapter_available (typefind->adapter);
406 GST_OBJECT_UNLOCK (typefind);
407 gst_query_set_position (query, format, peer_pos);
411 res = gst_pad_query_default (pad, parent, query);
419 gst_type_find_element_seek (GstTypeFindElement * typefind, GstEvent * event)
422 GstSeekType start_type, stop_type;
427 GstSegment seeksegment = { 0, };
429 gst_event_parse_seek (event, &rate, &format, &flags, &start_type, &start,
432 /* we can only seek on bytes */
433 if (format != GST_FORMAT_BYTES) {
434 GST_DEBUG_OBJECT (typefind, "Can only seek on BYTES");
438 /* copy segment, we need this because we still need the old
439 * segment when we close the current segment. */
440 memcpy (&seeksegment, &typefind->segment, sizeof (GstSegment));
442 GST_DEBUG_OBJECT (typefind, "configuring seek");
443 gst_segment_do_seek (&seeksegment, rate, format, flags,
444 start_type, start, stop_type, stop, NULL);
446 flush = ! !(flags & GST_SEEK_FLAG_FLUSH);
448 GST_DEBUG_OBJECT (typefind, "New segment %" GST_SEGMENT_FORMAT, &seeksegment);
451 GST_DEBUG_OBJECT (typefind, "Starting flush");
452 gst_pad_push_event (typefind->sink, gst_event_new_flush_start ());
453 gst_pad_push_event (typefind->src, gst_event_new_flush_start ());
455 GST_DEBUG_OBJECT (typefind, "Non-flushing seek, pausing task");
456 gst_pad_pause_task (typefind->sink);
459 /* now grab the stream lock so that streaming cannot continue, for
460 * non flushing seeks when the element is in PAUSED this could block
462 GST_DEBUG_OBJECT (typefind, "Waiting for streaming to stop");
463 GST_PAD_STREAM_LOCK (typefind->sink);
466 GST_DEBUG_OBJECT (typefind, "Stopping flush");
467 gst_pad_push_event (typefind->sink, gst_event_new_flush_stop (TRUE));
468 gst_pad_push_event (typefind->src, gst_event_new_flush_stop (TRUE));
471 /* now update the real segment info */
472 GST_DEBUG_OBJECT (typefind, "Committing new seek segment");
473 memcpy (&typefind->segment, &seeksegment, sizeof (GstSegment));
474 typefind->offset = typefind->segment.start;
476 /* notify start of new segment */
477 if (typefind->segment.flags & GST_SEGMENT_FLAG_SEGMENT) {
480 msg = gst_message_new_segment_start (GST_OBJECT (typefind),
481 GST_FORMAT_BYTES, typefind->segment.start);
482 gst_element_post_message (GST_ELEMENT (typefind), msg);
485 typefind->need_segment = TRUE;
487 /* restart our task since it might have been stopped when we did the
489 gst_pad_start_task (typefind->sink,
490 (GstTaskFunction) gst_type_find_element_loop, typefind->sink, NULL);
492 /* streaming can continue now */
493 GST_PAD_STREAM_UNLOCK (typefind->sink);
499 gst_type_find_element_src_event (GstPad * pad, GstObject * parent,
502 GstTypeFindElement *typefind = GST_TYPE_FIND_ELEMENT (parent);
505 if (typefind->mode != MODE_NORMAL) {
506 /* need to do more? */
507 gst_event_unref (event);
511 /* Only handle seeks here if driving the pipeline */
512 if (typefind->segment.format != GST_FORMAT_UNDEFINED &&
513 GST_EVENT_TYPE (event) == GST_EVENT_SEEK) {
514 result = gst_type_find_element_seek (typefind, event);
515 gst_event_unref (event);
518 return gst_pad_push_event (typefind->sink, event);
523 start_typefinding (GstTypeFindElement * typefind)
525 GST_DEBUG_OBJECT (typefind, "starting typefinding");
527 GST_OBJECT_LOCK (typefind);
529 gst_caps_replace (&typefind->caps, NULL);
530 typefind->initial_offset = GST_BUFFER_OFFSET_NONE;
531 GST_OBJECT_UNLOCK (typefind);
533 typefind->mode = MODE_TYPEFIND;
537 stop_typefinding (GstTypeFindElement * typefind)
540 gboolean push_cached_buffers;
543 GstClockTime pts, dts;
545 gst_element_get_state (GST_ELEMENT (typefind), &state, NULL, 0);
547 push_cached_buffers = (state >= GST_STATE_PAUSED && typefind->caps);
549 GST_DEBUG_OBJECT (typefind, "stopping typefinding%s",
550 push_cached_buffers ? " and pushing cached events and buffers" : "");
552 typefind->mode = MODE_NORMAL;
553 if (push_cached_buffers)
554 gst_type_find_element_send_cached_events (typefind);
556 GST_OBJECT_LOCK (typefind);
557 avail = gst_adapter_available (typefind->adapter);
561 pts = gst_adapter_prev_pts (typefind->adapter, NULL);
562 dts = gst_adapter_prev_dts (typefind->adapter, NULL);
563 buffer = gst_adapter_take_buffer (typefind->adapter, avail);
564 GST_BUFFER_PTS (buffer) = pts;
565 GST_BUFFER_DTS (buffer) = dts;
566 GST_BUFFER_OFFSET (buffer) = typefind->initial_offset;
567 GST_OBJECT_UNLOCK (typefind);
569 if (!push_cached_buffers) {
570 gst_buffer_unref (buffer);
572 GstPad *peer = gst_pad_get_peer (typefind->src);
574 /* make sure the user gets a meaningful error message in this case,
575 * which is not a core bug or bug of any kind (as the default error
576 * message emitted by gstpad.c otherwise would make you think) */
577 if (peer && GST_PAD_CHAINFUNC (peer) == NULL) {
578 GST_DEBUG_OBJECT (typefind, "upstream only supports push mode, while "
579 "downstream element only works in pull mode, erroring out");
580 GST_ELEMENT_ERROR (typefind, STREAM, FAILED,
581 ("%s cannot work in push mode. The operation is not supported "
582 "with this source element or protocol.",
583 G_OBJECT_TYPE_NAME (GST_PAD_PARENT (peer))),
584 ("Downstream pad %s:%s has no chainfunction, and the upstream "
585 "element does not support pull mode", GST_DEBUG_PAD_NAME (peer)));
586 typefind->mode = MODE_ERROR; /* make the chain function error out */
587 gst_buffer_unref (buffer);
589 gst_pad_push (typefind->src, buffer);
592 gst_object_unref (peer);
599 GST_DEBUG_OBJECT (typefind, "we have no data to typefind");
600 GST_OBJECT_UNLOCK (typefind);
606 gst_type_find_element_sink_event (GstPad * pad, GstObject * parent,
609 gboolean res = FALSE;
610 GstTypeFindElement *typefind = GST_TYPE_FIND_ELEMENT (parent);
612 GST_DEBUG_OBJECT (typefind, "got %s event in mode %d",
613 GST_EVENT_TYPE_NAME (event), typefind->mode);
615 switch (typefind->mode) {
617 switch (GST_EVENT_TYPE (event)) {
622 /* Parse and push out our caps and data */
623 gst_event_parse_caps (event, &caps);
624 res = gst_type_find_element_setcaps (typefind, caps);
626 gst_event_unref (event);
631 GST_FIXME_OBJECT (typefind,
632 "GAP events during typefinding not handled properly");
634 /* FIXME: These would need to be inserted in the stream at
635 * the right position between buffers, but we combine all
636 * buffers with a GstAdapter. Drop the GAP event for now,
637 * which will only cause an implicit GAP between buffers.
639 gst_event_unref (event);
645 GST_INFO_OBJECT (typefind, "Got EOS and no type found yet");
646 gst_type_find_element_chain_do_typefinding (typefind, FALSE, TRUE);
648 res = gst_pad_push_event (typefind->src, event);
651 case GST_EVENT_FLUSH_STOP:{
654 GST_OBJECT_LOCK (typefind);
656 for (l = typefind->cached_events; l; l = l->next) {
657 if (GST_EVENT_IS_STICKY (l->data) &&
658 GST_EVENT_TYPE (l->data) != GST_EVENT_SEGMENT &&
659 GST_EVENT_TYPE (l->data) != GST_EVENT_EOS) {
660 gst_pad_store_sticky_event (typefind->src, l->data);
662 gst_event_unref (l->data);
665 g_list_free (typefind->cached_events);
666 typefind->cached_events = NULL;
667 gst_adapter_clear (typefind->adapter);
668 GST_OBJECT_UNLOCK (typefind);
671 case GST_EVENT_FLUSH_START:
672 res = gst_pad_push_event (typefind->src, event);
675 /* Forward events that would happen before the caps event
676 * directly instead of storing them. There's no reason not
677 * to send them directly and we should only store events
678 * for later sending that would need to come after the caps
680 if (GST_EVENT_TYPE (event) < GST_EVENT_CAPS) {
681 res = gst_pad_push_event (typefind->src, event);
683 GST_DEBUG_OBJECT (typefind, "Saving %s event to send later",
684 GST_EVENT_TYPE_NAME (event));
685 GST_OBJECT_LOCK (typefind);
686 typefind->cached_events =
687 g_list_append (typefind->cached_events, event);
688 GST_OBJECT_UNLOCK (typefind);
695 res = gst_pad_push_event (typefind->src, event);
700 g_assert_not_reached ();
706 gst_type_find_element_send_cached_events (GstTypeFindElement * typefind)
708 GList *l, *cached_events;
710 GST_OBJECT_LOCK (typefind);
711 cached_events = typefind->cached_events;
712 typefind->cached_events = NULL;
713 GST_OBJECT_UNLOCK (typefind);
715 for (l = cached_events; l != NULL; l = l->next) {
716 GstEvent *event = GST_EVENT (l->data);
718 GST_DEBUG_OBJECT (typefind, "sending cached %s event",
719 GST_EVENT_TYPE_NAME (event));
720 gst_pad_push_event (typefind->src, event);
722 g_list_free (cached_events);
726 gst_type_find_element_setcaps (GstTypeFindElement * typefind, GstCaps * caps)
728 /* don't operate on ANY caps */
729 if (gst_caps_is_any (caps))
732 g_signal_emit (typefind, gst_type_find_element_signals[HAVE_TYPE], 0,
733 GST_TYPE_FIND_MAXIMUM, caps);
735 /* Shortcircuit typefinding if we get caps */
736 GST_DEBUG_OBJECT (typefind, "Skipping typefinding, using caps from "
737 "upstream: %" GST_PTR_FORMAT, caps);
739 stop_typefinding (typefind);
745 gst_type_find_get_extension (GstTypeFindElement * typefind, GstPad * pad)
752 query = gst_query_new_uri ();
754 /* try getting the caps with an uri query and from the extension */
755 if (!gst_pad_peer_query (pad, query))
756 goto peer_query_failed;
758 gst_query_parse_uri (query, &uri);
762 GST_DEBUG_OBJECT (typefind, "finding extension of %s", uri);
764 /* find the extension on the uri, this is everything after a '.' */
769 if (uri[find] == '.')
776 result = g_strdup (&uri[find + 1]);
778 GST_DEBUG_OBJECT (typefind, "found extension %s", result);
779 gst_query_unref (query);
787 GST_WARNING_OBJECT (typefind, "failed to query peer uri");
788 gst_query_unref (query);
793 GST_WARNING_OBJECT (typefind, "could not parse the peer uri");
794 gst_query_unref (query);
799 GST_WARNING_OBJECT (typefind, "could not find uri extension in %s", uri);
800 gst_query_unref (query);
807 gst_type_find_guess_by_extension (GstTypeFindElement * typefind, GstPad * pad,
808 GstTypeFindProbability * probability)
813 ext = gst_type_find_get_extension (typefind, pad);
817 caps = gst_type_find_helper_for_extension (GST_OBJECT_CAST (typefind), ext);
819 *probability = GST_TYPE_FIND_MAXIMUM;
827 gst_type_find_element_chain (GstPad * pad, GstObject * parent,
830 GstTypeFindElement *typefind;
831 GstFlowReturn res = GST_FLOW_OK;
833 typefind = GST_TYPE_FIND_ELEMENT (parent);
835 GST_LOG_OBJECT (typefind, "handling buffer in mode %d", typefind->mode);
837 switch (typefind->mode) {
839 /* we should already have called GST_ELEMENT_ERROR */
840 return GST_FLOW_ERROR;
842 /* don't take object lock as typefind->caps should not change anymore */
843 return gst_pad_push (typefind->src, buffer);
846 GST_OBJECT_LOCK (typefind);
847 if (typefind->initial_offset == GST_BUFFER_OFFSET_NONE)
848 typefind->initial_offset = GST_BUFFER_OFFSET (buffer);
849 gst_adapter_push (typefind->adapter, buffer);
850 GST_OBJECT_UNLOCK (typefind);
852 res = gst_type_find_element_chain_do_typefinding (typefind, TRUE, FALSE);
854 if (typefind->mode == MODE_ERROR)
855 res = GST_FLOW_ERROR;
860 g_assert_not_reached ();
861 return GST_FLOW_ERROR;
868 gst_type_find_element_chain_do_typefinding (GstTypeFindElement * typefind,
869 gboolean check_avail, gboolean at_eos)
871 GstTypeFindProbability probability;
872 GstCaps *caps = NULL;
875 gboolean have_min, have_max;
877 GST_OBJECT_LOCK (typefind);
878 if (typefind->force_caps) {
879 caps = gst_caps_ref (typefind->force_caps);
880 probability = GST_TYPE_FIND_MAXIMUM;
884 avail = gst_adapter_available (typefind->adapter);
887 have_min = avail >= TYPE_FIND_MIN_SIZE;
888 have_max = avail >= TYPE_FIND_MAX_SIZE;
890 have_min = avail > 0;
895 goto not_enough_data;
897 /* map all available data */
898 data = gst_adapter_map (typefind->adapter, avail);
899 caps = gst_type_find_helper_for_data (GST_OBJECT (typefind),
900 data, avail, &probability);
901 gst_adapter_unmap (typefind->adapter);
903 if (caps == NULL && have_max)
905 else if (caps == NULL)
909 if (probability < typefind->min_probability)
910 goto low_probability;
913 GST_OBJECT_UNLOCK (typefind);
915 /* probability is good enough too, so let's make it known ... emiting this
916 * signal calls our object handler which sets the caps. */
917 g_signal_emit (typefind, gst_type_find_element_signals[HAVE_TYPE], 0,
920 /* .. and send out the accumulated data */
921 stop_typefinding (typefind);
922 gst_caps_unref (caps);
928 GST_OBJECT_UNLOCK (typefind);
931 GST_ELEMENT_ERROR (typefind, STREAM, TYPE_NOT_FOUND,
932 (_("Stream contains not enough data.")), ("Can't typefind stream"));
933 return GST_FLOW_ERROR;
935 GST_DEBUG_OBJECT (typefind, "not enough data for typefinding yet "
936 "(%" G_GSIZE_FORMAT " bytes)", avail);
942 GST_OBJECT_UNLOCK (typefind);
943 GST_ELEMENT_ERROR (typefind, STREAM, TYPE_NOT_FOUND, (NULL), (NULL));
944 stop_typefinding (typefind);
945 return GST_FLOW_ERROR;
949 GST_OBJECT_UNLOCK (typefind);
952 GST_ELEMENT_ERROR (typefind, STREAM, TYPE_NOT_FOUND,
953 (_("Stream contains not enough data.")), ("Can't typefind stream"));
954 return GST_FLOW_ERROR;
956 GST_DEBUG_OBJECT (typefind,
957 "no caps found with %" G_GSIZE_FORMAT " bytes of data, "
958 "waiting for more data", avail);
964 GST_DEBUG_OBJECT (typefind, "found caps %" GST_PTR_FORMAT ", but "
965 "probability is %u which is lower than the required minimum of %u",
966 caps, probability, typefind->min_probability);
968 gst_caps_unref (caps);
973 GST_OBJECT_UNLOCK (typefind);
974 GST_DEBUG_OBJECT (typefind, "waiting for more data to try again");
980 gst_type_find_element_getrange (GstPad * srcpad, GstObject * parent,
981 guint64 offset, guint length, GstBuffer ** buffer)
983 GstTypeFindElement *typefind;
986 typefind = GST_TYPE_FIND_ELEMENT (parent);
988 ret = gst_pad_pull_range (typefind->sink, offset, length, buffer);
994 gst_type_find_element_activate_src_mode (GstPad * pad, GstObject * parent,
995 GstPadMode mode, gboolean active)
998 GstTypeFindElement *typefind;
1000 typefind = GST_TYPE_FIND_ELEMENT (parent);
1003 case GST_PAD_MODE_PULL:
1004 /* make sure our task stops pushing, we can't call _stop here because this
1005 * activation might happen from the streaming thread. */
1006 gst_pad_pause_task (typefind->sink);
1007 res = gst_pad_activate_mode (typefind->sink, mode, active);
1017 gst_type_find_element_loop (GstPad * pad)
1019 GstTypeFindElement *typefind;
1020 GstFlowReturn ret = GST_FLOW_OK;
1022 typefind = GST_TYPE_FIND_ELEMENT (GST_PAD_PARENT (pad));
1024 if (typefind->need_stream_start) {
1028 stream_id = gst_pad_create_stream_id (typefind->src,
1029 GST_ELEMENT_CAST (typefind), NULL);
1031 GST_DEBUG_OBJECT (typefind, "Pushing STREAM_START");
1032 event = gst_event_new_stream_start (stream_id);
1033 gst_event_set_group_id (event, gst_util_group_id_next ());
1034 gst_pad_push_event (typefind->src, event);
1036 typefind->need_stream_start = FALSE;
1040 if (typefind->mode == MODE_TYPEFIND) {
1041 GstPad *peer = NULL;
1042 GstCaps *found_caps = NULL;
1043 GstTypeFindProbability probability = GST_TYPE_FIND_NONE;
1045 GST_DEBUG_OBJECT (typefind, "find type in pull mode");
1047 GST_OBJECT_LOCK (typefind);
1048 if (typefind->force_caps) {
1049 found_caps = gst_caps_ref (typefind->force_caps);
1050 probability = GST_TYPE_FIND_MAXIMUM;
1052 GST_OBJECT_UNLOCK (typefind);
1055 peer = gst_pad_get_peer (pad);
1060 if (!gst_pad_query_duration (peer, GST_FORMAT_BYTES, &size)) {
1061 GST_WARNING_OBJECT (typefind, "Could not query upstream length!");
1062 gst_object_unref (peer);
1064 ret = GST_FLOW_ERROR;
1068 /* the size if 0, we cannot continue */
1070 /* keep message in sync with message in sink event handler */
1071 GST_ELEMENT_ERROR (typefind, STREAM, TYPE_NOT_FOUND,
1072 (_("Stream contains no data.")), ("Can't typefind empty stream"));
1073 gst_object_unref (peer);
1074 ret = GST_FLOW_ERROR;
1077 ext = gst_type_find_get_extension (typefind, pad);
1080 gst_type_find_helper_get_range (GST_OBJECT_CAST (peer),
1081 GST_OBJECT_PARENT (peer),
1082 (GstTypeFindHelperGetRangeFunction) (GST_PAD_GETRANGEFUNC (peer)),
1083 (guint64) size, ext, &probability);
1086 GST_DEBUG ("Found caps %" GST_PTR_FORMAT, found_caps);
1088 gst_object_unref (peer);
1092 if (!found_caps || probability < typefind->min_probability) {
1093 GST_DEBUG ("Trying to guess using extension");
1094 gst_caps_replace (&found_caps, NULL);
1096 gst_type_find_guess_by_extension (typefind, pad, &probability);
1099 if (!found_caps || probability < typefind->min_probability) {
1100 GST_ELEMENT_ERROR (typefind, STREAM, TYPE_NOT_FOUND, (NULL), (NULL));
1101 gst_caps_replace (&found_caps, NULL);
1102 ret = GST_FLOW_ERROR;
1106 GST_DEBUG ("Emiting found caps %" GST_PTR_FORMAT, found_caps);
1107 g_signal_emit (typefind, gst_type_find_element_signals[HAVE_TYPE],
1108 0, probability, found_caps);
1109 typefind->mode = MODE_NORMAL;
1110 gst_caps_unref (found_caps);
1111 } else if (typefind->mode == MODE_NORMAL) {
1112 GstBuffer *outbuf = NULL;
1114 if (typefind->need_segment) {
1115 typefind->need_segment = FALSE;
1116 gst_pad_push_event (typefind->src,
1117 gst_event_new_segment (&typefind->segment));
1120 /* Pull 4k blocks and send downstream */
1121 ret = gst_pad_pull_range (typefind->sink, typefind->offset, 4096, &outbuf);
1122 if (ret != GST_FLOW_OK)
1125 typefind->offset += gst_buffer_get_size (outbuf);
1127 ret = gst_pad_push (typefind->src, outbuf);
1128 if (ret != GST_FLOW_OK)
1132 ret = GST_FLOW_ERROR;
1140 const gchar *reason = gst_flow_get_name (ret);
1141 gboolean push_eos = FALSE;
1143 GST_LOG_OBJECT (typefind, "pausing task, reason %s", reason);
1144 gst_pad_pause_task (typefind->sink);
1146 if (ret == GST_FLOW_EOS) {
1147 /* perform EOS logic */
1149 if (typefind->segment.flags & GST_SEGMENT_FLAG_SEGMENT) {
1152 /* for segment playback we need to post when (in stream time)
1153 * we stopped, this is either stop (when set) or the duration. */
1154 if ((stop = typefind->segment.stop) == -1)
1155 stop = typefind->offset;
1157 GST_LOG_OBJECT (typefind, "Sending segment done, at end of segment");
1158 gst_element_post_message (GST_ELEMENT (typefind),
1159 gst_message_new_segment_done (GST_OBJECT (typefind),
1160 GST_FORMAT_BYTES, stop));
1161 gst_pad_push_event (typefind->src,
1162 gst_event_new_segment_done (GST_FORMAT_BYTES, stop));
1166 } else if (ret == GST_FLOW_NOT_LINKED || ret < GST_FLOW_EOS) {
1167 /* for fatal errors we post an error message */
1168 GST_ELEMENT_ERROR (typefind, STREAM, FAILED, (NULL),
1169 ("stream stopped, reason %s", reason));
1173 /* send EOS, and prevent hanging if no streams yet */
1174 GST_LOG_OBJECT (typefind, "Sending EOS, at end of stream");
1175 gst_pad_push_event (typefind->src, gst_event_new_eos ());
1182 gst_type_find_element_activate_sink_mode (GstPad * pad, GstObject * parent,
1183 GstPadMode mode, gboolean active)
1186 GstTypeFindElement *typefind;
1188 typefind = GST_TYPE_FIND_ELEMENT (parent);
1191 case GST_PAD_MODE_PULL:
1193 gst_segment_init (&typefind->segment, GST_FORMAT_BYTES);
1194 typefind->need_segment = TRUE;
1195 typefind->need_stream_start = TRUE;
1196 typefind->offset = 0;
1199 res = gst_pad_stop_task (pad);
1200 gst_segment_init (&typefind->segment, GST_FORMAT_UNDEFINED);
1203 case GST_PAD_MODE_PUSH:
1205 gst_segment_init (&typefind->segment, GST_FORMAT_UNDEFINED);
1206 start_typefinding (typefind);
1208 stop_typefinding (typefind);
1209 gst_segment_init (&typefind->segment, GST_FORMAT_UNDEFINED);
1221 gst_type_find_element_activate_sink (GstPad * pad, GstObject * parent)
1225 GstSchedulingFlags sched_flags;
1227 query = gst_query_new_scheduling ();
1229 if (!gst_pad_peer_query (pad, query)) {
1230 gst_query_unref (query);
1234 gst_query_parse_scheduling (query, &sched_flags, NULL, NULL, NULL);
1236 pull_mode = gst_query_has_scheduling_mode (query, GST_PAD_MODE_PULL)
1237 && ((sched_flags & GST_SCHEDULING_FLAG_SEEKABLE) != 0);
1239 gst_query_unref (query);
1244 if (!gst_pad_activate_mode (pad, GST_PAD_MODE_PULL, TRUE))
1247 /* only start our task if we ourselves decide to start in pull mode */
1248 return gst_pad_start_task (pad, (GstTaskFunction) gst_type_find_element_loop,
1253 return gst_pad_activate_mode (pad, GST_PAD_MODE_PUSH, TRUE);
1257 static GstStateChangeReturn
1258 gst_type_find_element_change_state (GstElement * element,
1259 GstStateChange transition)
1261 GstStateChangeReturn ret;
1262 GstTypeFindElement *typefind;
1264 typefind = GST_TYPE_FIND_ELEMENT (element);
1267 ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
1269 switch (transition) {
1270 case GST_STATE_CHANGE_PAUSED_TO_READY:
1271 case GST_STATE_CHANGE_READY_TO_NULL:
1272 GST_OBJECT_LOCK (typefind);
1273 gst_caps_replace (&typefind->caps, NULL);
1275 g_list_foreach (typefind->cached_events,
1276 (GFunc) gst_mini_object_unref, NULL);
1277 g_list_free (typefind->cached_events);
1278 typefind->cached_events = NULL;
1279 typefind->mode = MODE_TYPEFIND;
1280 GST_OBJECT_UNLOCK (typefind);