Created fork of mafw-gst-renderer with subtitles support

[mafwsubrenderer] / libmafw-gst-renderer / mafw-gst-renderer-utils.c

diff --git a/libmafw-gst-renderer/mafw-gst-renderer-utils.c b/libmafw-gst-renderer/mafw-gst-renderer-utils.c
index 0169862..42dae31 100644 (file)
--- a/libmafw-gst-renderer/mafw-gst-renderer-utils.c
+++ b/libmafw-gst-renderer/mafw-gst-renderer-utils.c
@@ -28,6 +28,9 @@
 
 #include <glib.h>
 
+#include <string.h>
+#include <gio/gio.h>
+
 #include "mafw-gst-renderer-utils.h"
 
 #undef  G_LOG_DOMAIN
@@ -102,4 +105,144 @@ gboolean uri_is_stream(const gchar *uri)
        }
 }
 
+/*
+ * Imported from totem-uri.c
+ * Copyright (C) 2004 Bastien Nocera
+ */
+
+/* List from xine-lib's demux_sputext.c */
+static const char subtitle_ext[][4] = {
+       "sub",
+       "srt",
+       "smi",
+       "ssa",
+       "ass",
+       "asc"
+};
+
+static inline gboolean
+uri_exists (const char *uri)
+{
+       GFile *file = g_file_new_for_uri (uri);
+       if (file != NULL) {
+               if (g_file_query_exists (file, NULL)) {
+                       g_object_unref (file);
+                       return TRUE;
+               }
+               g_object_unref (file);
+       }
+       return FALSE;
+}
+
+static char *
+uri_get_subtitle_for_uri (const char *uri)
+{
+       char *subtitle;
+       guint len, i;
+       gint suffix;
+
+       /* Find the filename suffix delimiter */
+       len = strlen (uri);
+       for (suffix = len - 1; suffix > 0; suffix--) {
+               if (uri[suffix] == G_DIR_SEPARATOR ||
+                   (uri[suffix] == '/')) {
+                       /* This filename has no extension; we'll need to 
+                        * add one */
+                       suffix = len;
+                       break;
+               }
+               if (uri[suffix] == '.') {
+                       /* Found our extension marker */
+                       break;
+               }
+       }
+       if (suffix < 0)
+               return NULL;
+
+       /* Generate a subtitle string with room at the end to store the
+        * 3 character extensions for which we want to search */
+       subtitle = g_malloc0 (suffix + 4 + 1);
+       g_return_val_if_fail (subtitle != NULL, NULL);
+       g_strlcpy (subtitle, uri, suffix + 4 + 1);
+       g_strlcpy (subtitle + suffix, ".???", 5);
+
+       /* Search for any files with one of our known subtitle extensions */
+       for (i = 0; i < G_N_ELEMENTS (subtitle_ext) ; i++) {
+               char *subtitle_ext_upper;
+               memcpy (subtitle + suffix + 1, subtitle_ext[i], 3);
+
+               if (uri_exists (subtitle))
+                       return subtitle;
+
+               /* Check with upper-cased extension */
+               subtitle_ext_upper = g_ascii_strup (subtitle_ext[i], -1);
+               memcpy (subtitle + suffix + 1, subtitle_ext_upper, 3);
+               g_free (subtitle_ext_upper);
+
+               if (uri_exists (subtitle))
+                       return subtitle;
+       }
+       g_free (subtitle);
+       return NULL;
+}
+
+static char *
+uri_get_subtitle_in_subdir (GFile *file, const char *subdir)
+{
+       char *filename, *subtitle, *full_path_str;
+       GFile *parent, *full_path, *directory;
+
+       /* Get the sibling directory @subdir of the file @file */
+       parent = g_file_get_parent (file);
+       directory = g_file_get_child (parent, subdir);
+       g_object_unref (parent);
+
+       /* Get the file of the same name as @file in the @subdir directory */
+       filename = g_file_get_basename (file);
+       full_path = g_file_get_child (directory, filename);
+       g_object_unref (directory);
+       g_free (filename);
+
+       /* Get the subtitles from that URI */
+       full_path_str = g_file_get_uri (full_path);
+       g_object_unref (full_path);
+       subtitle = uri_get_subtitle_for_uri (full_path_str);
+       g_free (full_path_str);
+
+       return subtitle;
+}
+
+char *
+uri_get_subtitle_uri (const char *uri)
+{
+       GFile *file;
+       char *subtitle;
+
+       if (g_str_has_prefix (uri, "http") != FALSE)
+               return NULL;
+
+       /* Has the user specified a subtitle file manually? */
+       if (strstr (uri, "#subtitle:") != NULL)
+               return NULL;
+
+       /* Does the file exist? */
+       file = g_file_new_for_uri (uri);
+       if (g_file_query_exists (file, NULL) != TRUE) {
+               g_object_unref (file);
+               return NULL;
+       }
+
+       /* Try in the current directory */
+       subtitle = uri_get_subtitle_for_uri (uri);
+       if (subtitle != NULL) {
+               g_object_unref (file);
+               return subtitle;
+       }
+
+       subtitle = uri_get_subtitle_in_subdir (file, "subtitles");
+       g_object_unref (file);
+
+       return subtitle;
+}
+
 /* vi: set noexpandtab ts=8 sw=8 cino=t0,(0: */