typefind: add a new method that also uses the file extension

Add a method to perform get_range typefinding that also uses the
uri/location extension as an extra hint. It will first try to call the
typefind functions of the factories that handle the given extension. The result
is that in the common case, we only call one typefind function, which speeds up
the typefinding a lot.
This commit is contained in:
Wim Taymans 2010-01-12 17:34:39 +01:00
parent 068307b70e
commit 8fff434835
4 changed files with 93 additions and 8 deletions

View file

@ -675,6 +675,7 @@ gst_type_find_helper_for_buffer
gst_type_find_helper_for_extension gst_type_find_helper_for_extension
GstTypeFindHelperGetRangeFunction GstTypeFindHelperGetRangeFunction
gst_type_find_helper_get_range gst_type_find_helper_get_range
gst_type_find_helper_get_range_ext
<SUBSECTION Private> <SUBSECTION Private>
</SECTION> </SECTION>

View file

@ -215,11 +215,12 @@ helper_find_get_length (gpointer data)
} }
/** /**
* gst_type_find_helper_get_range: * gst_type_find_helper_get_range_ext:
* @obj: A #GstObject that will be passed as first argument to @func * @obj: A #GstObject that will be passed as first argument to @func
* @func: A generic #GstTypeFindHelperGetRangeFunction that will be used * @func: A generic #GstTypeFindHelperGetRangeFunction that will be used
* to access data at random offsets when doing the typefinding * to access data at random offsets when doing the typefinding
* @size: The length in bytes * @size: The length in bytes
* @extension: extenstion of the media
* @prob: location to store the probability of the found caps, or #NULL * @prob: location to store the probability of the found caps, or #NULL
* *
* Utility function to do pull-based typefinding. Unlike gst_type_find_helper() * Utility function to do pull-based typefinding. Unlike gst_type_find_helper()
@ -231,20 +232,26 @@ helper_find_get_length (gpointer data)
* callback can then call the upstream peer pad with offsets adjusted for the * callback can then call the upstream peer pad with offsets adjusted for the
* tag size, for example). * tag size, for example).
* *
* When @extension is not NULL, this function will first try the typefind
* functions for the given extension, which might speed up the typefinding
* in many cases.
*
* Returns: The #GstCaps corresponding to the data stream. * Returns: The #GstCaps corresponding to the data stream.
* Returns #NULL if no #GstCaps matches the data stream. * Returns #NULL if no #GstCaps matches the data stream.
*
* Since: 0.10.26
*/ */
GstCaps * GstCaps *
gst_type_find_helper_get_range (GstObject * obj, gst_type_find_helper_get_range_ext (GstObject * obj,
GstTypeFindHelperGetRangeFunction func, guint64 size, GstTypeFindHelperGetRangeFunction func, guint64 size,
GstTypeFindProbability * prob) const gchar * extension, GstTypeFindProbability * prob)
{ {
GstTypeFindHelper helper; GstTypeFindHelper helper;
GstTypeFind find; GstTypeFind find;
GSList *walk; GSList *walk;
GList *l, *type_list; GList *l, *type_list;
GstCaps *result = NULL; GstCaps *result = NULL;
gint pos = 0;
g_return_val_if_fail (GST_IS_OBJECT (obj), NULL); g_return_val_if_fail (GST_IS_OBJECT (obj), NULL);
g_return_val_if_fail (func != NULL, NULL); g_return_val_if_fail (func != NULL, NULL);
@ -269,6 +276,48 @@ gst_type_find_helper_get_range (GstObject * obj,
type_list = gst_type_find_factory_get_list (); type_list = gst_type_find_factory_get_list ();
/* move the typefinders for the extension first in the list. The idea is that
* when one of them returns MAX we don't need to search further as there is a
* very high chance we got the right type. */
if (extension) {
GList *next;
GST_LOG_OBJECT (obj, "sorting typefind for extension %s to head",
extension);
for (l = type_list; l; l = next) {
GstTypeFindFactory *factory;
gint i;
gchar **ext;
next = l->next;
factory = GST_TYPE_FIND_FACTORY (l->data);
ext = gst_type_find_factory_get_extensions (factory);
if (ext == NULL)
continue;
GST_LOG_OBJECT (obj, "testing factory %s for extension %s",
GST_PLUGIN_FEATURE_NAME (factory), extension);
for (i = 0; ext[i]; i++) {
if (strcmp (ext[i], extension) == 0) {
/* found extension, move in front */
GST_LOG_OBJECT (obj, "moving typefind for extension %s to head",
extension);
/* remove entry from list */
type_list = g_list_delete_link (type_list, l);
/* insert at the position */
type_list = g_list_insert (type_list, factory, pos);
/* next element will be inserted after this one */
pos++;
break;
}
}
}
}
for (l = type_list; l; l = l->next) { for (l = type_list; l; l = l->next) {
helper.factory = GST_TYPE_FIND_FACTORY (l->data); helper.factory = GST_TYPE_FIND_FACTORY (l->data);
gst_type_find_factory_call_function (helper.factory, &find); gst_type_find_factory_call_function (helper.factory, &find);
@ -293,6 +342,34 @@ gst_type_find_helper_get_range (GstObject * obj,
return result; return result;
} }
/**
* gst_type_find_helper_get_range:
* @obj: A #GstObject that will be passed as first argument to @func
* @func: A generic #GstTypeFindHelperGetRangeFunction that will be used
* to access data at random offsets when doing the typefinding
* @size: The length in bytes
* @prob: location to store the probability of the found caps, or #NULL
*
* Utility function to do pull-based typefinding. Unlike gst_type_find_helper()
* however, this function will use the specified function @func to obtain the
* data needed by the typefind functions, rather than operating on a given
* source pad. This is useful mostly for elements like tag demuxers which
* strip off data at the beginning and/or end of a file and want to typefind
* the stripped data stream before adding their own source pad (the specified
* callback can then call the upstream peer pad with offsets adjusted for the
* tag size, for example).
*
* Returns: The #GstCaps corresponding to the data stream.
* Returns #NULL if no #GstCaps matches the data stream.
*/
GstCaps *
gst_type_find_helper_get_range (GstObject * obj,
GstTypeFindHelperGetRangeFunction func, guint64 size,
GstTypeFindProbability * prob)
{
return gst_type_find_helper_get_range_ext (obj, func, size, NULL, prob);
}
/** /**
* gst_type_find_helper: * gst_type_find_helper:
* @src: A source #GstPad * @src: A source #GstPad
@ -496,15 +573,15 @@ gst_type_find_helper_for_extension (GstObject * obj, const gchar * extension)
factory = GST_TYPE_FIND_FACTORY (l->data); factory = GST_TYPE_FIND_FACTORY (l->data);
/* we only want to check those factories without a function */
if (factory->function != NULL)
continue;
/* get the extension that this typefind factory can handle */ /* get the extension that this typefind factory can handle */
ext = gst_type_find_factory_get_extensions (factory); ext = gst_type_find_factory_get_extensions (factory);
if (ext == NULL) if (ext == NULL)
continue; continue;
/* we only want to check those factories without a function */
if (factory->function != NULL)
continue;
/* there are extension, see if one of them matches the requested /* there are extension, see if one of them matches the requested
* extension */ * extension */
for (i = 0; ext[i]; i++) { for (i = 0; ext[i]; i++) {

View file

@ -64,6 +64,12 @@ GstCaps * gst_type_find_helper_get_range (GstObject * obj
guint64 size, guint64 size,
GstTypeFindProbability *prob); GstTypeFindProbability *prob);
GstCaps * gst_type_find_helper_get_range_ext (GstObject * obj,
GstTypeFindHelperGetRangeFunction func,
guint64 size,
const gchar *extension,
GstTypeFindProbability *prob);
G_END_DECLS G_END_DECLS
#endif /* __GST_TYPEFINDHELPER_H__ */ #endif /* __GST_TYPEFINDHELPER_H__ */

View file

@ -220,3 +220,4 @@ EXPORTS
gst_type_find_helper_for_buffer gst_type_find_helper_for_buffer
gst_type_find_helper_for_extension gst_type_find_helper_for_extension
gst_type_find_helper_get_range gst_type_find_helper_get_range
gst_type_find_helper_get_range_ext