Fix #541: Showing existing, or maybe generating thumbnails for MP4 and WEBM
[geeqie.git] / src / image_load_ffmpegthumbnailer.c
diff --git a/src/image_load_ffmpegthumbnailer.c b/src/image_load_ffmpegthumbnailer.c
new file mode 100644 (file)
index 0000000..fb3c2bb
--- /dev/null
@@ -0,0 +1,201 @@
+/*
+ * Copyright (C) 2004 John Ellis
+ * Copyright (C) 2008 - 2016 The Geeqie Team
+ *
+ * Author: Tomasz Golinski <tomaszg@math.uwb.edu.pl>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#include "main.h"
+#include "image-load.h"
+#include "image_load_ffmpegthumbnailer.h"
+
+#ifdef HAVE_FFMPEGTHUMBNAILER
+#include <libffmpegthumbnailer/videothumbnailerc.h>
+
+typedef struct _ImageLoaderFT ImageLoaderFT;
+struct _ImageLoaderFT {
+       ImageLoaderBackendCbAreaUpdated area_updated_cb;
+       ImageLoaderBackendCbSize size_cb;
+       ImageLoaderBackendCbAreaPrepared area_prepared_cb;
+
+       video_thumbnailer *vt;
+
+       gpointer data;
+
+       GdkPixbuf *pixbuf;
+       guint requested_width;
+       guint requested_height;
+
+};
+
+static void image_loader_ft_log_cb(ThumbnailerLogLevel log_level, const char* msg)
+{
+       if (log_level == ThumbnailerLogLevelError)
+               log_printf("ImageLoaderFFmpegthumbnailer: %s",msg);
+       else
+               DEBUG_1("ImageLoaderFFmpegthumbnailer: %s",msg);
+}
+
+void image_loader_ft_destroy_image_data(guchar *pixels, gpointer data)
+{
+       image_data *image = (image_data *) data;
+
+       video_thumbnailer_destroy_image_data (image);
+}
+
+static gchar* image_loader_ft_get_format_name(gpointer loader)
+{
+       return g_strdup("ffmpeg");
+}
+
+static gchar** image_loader_ft_get_format_mime_types(gpointer loader)
+{
+       static gchar *mime[] = {"video/mp4", NULL};
+       return g_strdupv(mime);}
+
+static gpointer image_loader_ft_new(ImageLoaderBackendCbAreaUpdated area_updated_cb, ImageLoaderBackendCbSize size_cb, ImageLoaderBackendCbAreaPrepared area_prepared_cb, gpointer data)
+{
+       ImageLoaderFT *loader = g_new0(ImageLoaderFT, 1);
+
+       loader->area_updated_cb = area_updated_cb;
+       loader->size_cb = size_cb;
+       loader->area_prepared_cb = area_prepared_cb;
+       loader->data = data;
+
+       loader->vt = video_thumbnailer_create();
+       loader->vt->overlay_film_strip = 1;
+       loader->vt->maintain_aspect_ratio = 1;
+#if HAVE_FFMPEGTHUMBNAILER_RGB
+       video_thumbnailer_set_log_callback(loader->vt, image_loader_ft_log_cb);
+#endif
+
+       return (gpointer) loader;
+}
+
+static void image_loader_ft_set_size(gpointer loader, int width, int height)
+{
+       ImageLoaderFT *lft = (ImageLoaderFT *) loader;
+       lft->requested_width = width;
+       lft->requested_height = height;
+       DEBUG_1("TG: setting size, w=%d, h=%d", width, height);
+}
+
+// static gboolean image_loader_ft_loadfromdisk(gpointer loader, const gchar *path, GError **error)
+static gboolean image_loader_ft_load (gpointer loader, const guchar *buf, gsize count, GError **error)
+{
+       ImageLoaderFT *lft = (ImageLoaderFT *) loader;
+       ImageLoader *il = lft->data;
+
+       image_data *image = video_thumbnailer_create_image_data();
+
+#ifdef HAVE_FFMPEGTHUMBNAILER_WH
+//     DEBUG_1("TG: FT requested size w=%d:h=%d for %s", lft->requested_width > 0, lft->requested_height, il->fd->path);
+       video_thumbnailer_set_size(lft->vt, lft->requested_width, lft->requested_height);
+#else
+       lft->vt->thumbnail_size = MAX(lft->requested_width,lft->requested_width);
+#endif
+
+#ifdef HAVE_FFMPEGTHUMBNAILER_METADATA
+       lft->vt->prefer_embedded_metadata = options->thumbnails.use_ft_metadata ? 1 : 0;
+#endif
+
+#if HAVE_FFMPEGTHUMBNAILER_RGB
+       lft->vt->thumbnail_image_type = Rgb;
+#else
+       lft->vt->thumbnail_image_type = Png;
+#endif
+
+       video_thumbnailer_generate_thumbnail_to_buffer (lft->vt, il->fd->path, image);
+
+#if HAVE_FFMPEGTHUMBNAILER_RGB
+       lft->pixbuf  = gdk_pixbuf_new_from_data (image->image_data_ptr, GDK_COLORSPACE_RGB, FALSE, 8, image->image_data_width, image->image_data_height,  image->image_data_width*3, image_loader_ft_destroy_image_data, image);
+       lft->size_cb(loader, image->image_data_width, image->image_data_height, lft->data);
+       lft->area_updated_cb(loader, 0, 0, image->image_data_width, image->image_data_height, lft->data);
+#else
+       GInputStream *image_stream;
+       image_stream = g_memory_input_stream_new_from_data (image->image_data_ptr, image->image_data_size, NULL);
+
+       if (image_stream == NULL)
+       {
+       video_thumbnailer_destroy_image_data (image);
+       DEBUG_1("FFmpegthumbnailer: cannot open stream for %s", il->fd->path);
+       return FALSE;
+    }
+
+       lft->pixbuf  = gdk_pixbuf_new_from_stream (image_stream, NULL, NULL);
+       lft->size_cb(loader, gdk_pixbuf_get_width(lft->pixbuf), gdk_pixbuf_get_height(lft->pixbuf), lft->data);
+       g_object_unref (image_stream);
+       video_thumbnailer_destroy_image_data (image);
+#endif
+
+       if (!lft->pixbuf)
+               {
+               DEBUG_1("FFmpegthumbnailer: no frame generated for %s", il->fd->path);
+               return FALSE;
+               }
+
+/* See comment in image_loader_area_prepared_cb
+ * Geeqie uses area_prepared signal to fill pixbuf with background color.
+ * We can't do it here as pixbuf already contains the data */
+//     lft->area_prepared_cb(loader, lft->data);
+
+       lft->area_updated_cb(loader, 0, 0, gdk_pixbuf_get_width(lft->pixbuf), gdk_pixbuf_get_height(lft->pixbuf), lft->data);
+
+       return TRUE;
+}
+
+static GdkPixbuf* image_loader_ft_get_pixbuf(gpointer loader)
+{
+       ImageLoaderFT *lft = (ImageLoaderFT *) loader;
+       return lft->pixbuf;
+}
+
+static void image_loader_ft_abort(gpointer loader)
+{
+}
+
+static gboolean image_loader_ft_close(gpointer loader, GError **error)
+{
+       return TRUE;
+}
+
+static void image_loader_ft_free(gpointer loader)
+{
+       ImageLoaderFT *lft = (ImageLoaderFT *) loader;
+       if (lft->pixbuf) g_object_unref(lft->pixbuf);
+       video_thumbnailer_destroy (lft->vt);
+
+       g_free(lft);
+}
+
+void image_loader_backend_set_ft(ImageLoaderBackend *funcs)
+{
+       funcs->loader_new = image_loader_ft_new;
+       funcs->set_size = image_loader_ft_set_size;
+       funcs->load = image_loader_ft_load;
+       funcs->write = NULL;
+       funcs->get_pixbuf = image_loader_ft_get_pixbuf;
+       funcs->close = image_loader_ft_close;
+       funcs->abort = image_loader_ft_abort;
+       funcs->free = image_loader_ft_free;
+
+       funcs->get_format_name = image_loader_ft_get_format_name;
+       funcs->get_format_mime_types = image_loader_ft_get_format_mime_types;
+}
+
+#endif
+/* vim: set shiftwidth=8 softtabstop=0 cindent cinoptions={1s: */