src/image-load-ffmpegthumbnailer.cc

   1 /*
   2  * Copyright (C) 2004 John Ellis
   3  * Copyright (C) 2008 - 2016 The Geeqie Team
   4  *
   5  * Author: Tomasz Golinski <tomaszg@math.uwb.edu.pl>
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License along
  18  * with this program; if not, write to the Free Software Foundation, Inc.,
  19  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  20  */
  21
  22 #include "main.h"
  23 #include "image-load.h"
  24 #include "image-load-ffmpegthumbnailer.h"
  25
  26 #include "filedata.h"
  27
  28 #ifdef HAVE_FFMPEGTHUMBNAILER
  29 #include <libffmpegthumbnailer/videothumbnailerc.h>
  30
  31 struct ImageLoaderFT {
  32         ImageLoaderBackendCbAreaUpdated area_updated_cb;
  33         ImageLoaderBackendCbSize size_cb;
  34         ImageLoaderBackendCbAreaPrepared area_prepared_cb;
  35
  36         video_thumbnailer *vt;
  37
  38         gpointer data;
  39
  40         GdkPixbuf *pixbuf;
  41         guint requested_width;
  42         guint requested_height;
  43
  44 };
  45
  46 #if HAVE_FFMPEGTHUMBNAILER_RGB
  47 static void image_loader_ft_log_cb(ThumbnailerLogLevel log_level, const char* msg)
  48 {
  49         if (log_level == ThumbnailerLogLevelError)
  50                 log_printf("ImageLoaderFFmpegthumbnailer: %s",msg);
  51         else
  52                 DEBUG_1("ImageLoaderFFmpegthumbnailer: %s",msg);
  53 }
  54 #endif
  55
  56 void image_loader_ft_destroy_image_data(guchar *, gpointer data)
  57 {
  58         auto image = static_cast<image_data *>(data);
  59
  60         video_thumbnailer_destroy_image_data (image);
  61 }
  62
  63 static gchar* image_loader_ft_get_format_name(gpointer)
  64 {
  65         return g_strdup("ffmpeg");
  66 }
  67
  68 static gchar** image_loader_ft_get_format_mime_types(gpointer)
  69 {
  70         static const gchar *mime[] = {"video/mp4", nullptr};
  71         return g_strdupv(const_cast<gchar **>(mime));
  72 }
  73
  74 static gpointer image_loader_ft_new(ImageLoaderBackendCbAreaUpdated area_updated_cb, ImageLoaderBackendCbSize size_cb, ImageLoaderBackendCbAreaPrepared area_prepared_cb, gpointer data)
  75 {
  76         auto loader = g_new0(ImageLoaderFT, 1);
  77
  78         loader->area_updated_cb = area_updated_cb;
  79         loader->size_cb = size_cb;
  80         loader->area_prepared_cb = area_prepared_cb;
  81         loader->data = data;
  82
  83         loader->vt = video_thumbnailer_create();
  84         loader->vt->overlay_film_strip = 1;
  85         loader->vt->maintain_aspect_ratio = 1;
  86 #if HAVE_FFMPEGTHUMBNAILER_RGB
  87         video_thumbnailer_set_log_callback(loader->vt, image_loader_ft_log_cb);
  88 #endif
  89
  90         return loader;
  91 }
  92
  93 static void image_loader_ft_set_size(gpointer loader, int width, int height)
  94 {
  95         auto lft = static_cast<ImageLoaderFT *>(loader);
  96         lft->requested_width = width;
  97         lft->requested_height = height;
  98         DEBUG_1("TG: setting size, w=%d, h=%d", width, height);
  99 }
 100
 101 static gboolean image_loader_ft_load (gpointer loader, const guchar *, gsize, GError **)
 102 {
 103         auto lft = static_cast<ImageLoaderFT *>(loader);
 104         auto il = static_cast<ImageLoader *>(lft->data);
 105
 106         image_data *image = video_thumbnailer_create_image_data();
 107
 108 #ifdef HAVE_FFMPEGTHUMBNAILER_WH
 109         video_thumbnailer_set_size(lft->vt, lft->requested_width, lft->requested_height);
 110 #else
 111         lft->vt->thumbnail_size = MAX(lft->requested_width,lft->requested_height);
 112 #endif
 113
 114 #ifdef HAVE_FFMPEGTHUMBNAILER_METADATA
 115         lft->vt->prefer_embedded_metadata = options->thumbnails.use_ft_metadata ? 1 : 0;
 116 #endif
 117
 118 #if HAVE_FFMPEGTHUMBNAILER_RGB
 119         lft->vt->thumbnail_image_type = Rgb;
 120 #else
 121         lft->vt->thumbnail_image_type = Png;
 122 #endif
 123
 124         video_thumbnailer_generate_thumbnail_to_buffer (lft->vt, il->fd->path, image);
 125
 126 #if HAVE_FFMPEGTHUMBNAILER_RGB
 127         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);
 128         lft->size_cb(loader, image->image_data_width, image->image_data_height, lft->data);
 129         lft->area_updated_cb(loader, 0, 0, image->image_data_width, image->image_data_height, lft->data);
 130 #else
 131         GInputStream *image_stream;
 132         image_stream = g_memory_input_stream_new_from_data (image->image_data_ptr, image->image_data_size, nullptr);
 133
 134         if (image_stream == nullptr)
 135         {
 136         video_thumbnailer_destroy_image_data (image);
 137         DEBUG_1("FFmpegthumbnailer: cannot open stream for %s", il->fd->path);
 138         return FALSE;
 139     }
 140
 141         lft->pixbuf  = gdk_pixbuf_new_from_stream (image_stream, nullptr, nullptr);
 142         lft->size_cb(loader, gdk_pixbuf_get_width(lft->pixbuf), gdk_pixbuf_get_height(lft->pixbuf), lft->data);
 143         g_object_unref (image_stream);
 144         video_thumbnailer_destroy_image_data (image);
 145 #endif
 146
 147         if (!lft->pixbuf)
 148                 {
 149                 DEBUG_1("FFmpegthumbnailer: no frame generated for %s", il->fd->path);
 150                 return FALSE;
 151                 }
 152
 153 /** See comment in image_loader_area_prepared_cb
 154  * Geeqie uses area_prepared signal to fill pixbuf with background color.
 155  * We can't do it here as pixbuf already contains the data
 156  * lft->area_prepared_cb(loader, lft->data);
 157  */
 158         lft->area_updated_cb(loader, 0, 0, gdk_pixbuf_get_width(lft->pixbuf), gdk_pixbuf_get_height(lft->pixbuf), lft->data);
 159
 160         return TRUE;
 161 }
 162
 163 static GdkPixbuf* image_loader_ft_get_pixbuf(gpointer loader)
 164 {
 165         auto lft = static_cast<ImageLoaderFT *>(loader);
 166         return lft->pixbuf;
 167 }
 168
 169 static void image_loader_ft_abort(gpointer)
 170 {
 171 }
 172
 173 static gboolean image_loader_ft_close(gpointer, GError **)
 174 {
 175         return TRUE;
 176 }
 177
 178 static void image_loader_ft_free(gpointer loader)
 179 {
 180         auto lft = static_cast<ImageLoaderFT *>(loader);
 181         if (lft->pixbuf) g_object_unref(lft->pixbuf);
 182         video_thumbnailer_destroy (lft->vt);
 183
 184         g_free(lft);
 185 }
 186
 187 void image_loader_backend_set_ft(ImageLoaderBackend *funcs)
 188 {
 189         funcs->loader_new = image_loader_ft_new;
 190         funcs->set_size = image_loader_ft_set_size;
 191         funcs->load = image_loader_ft_load;
 192         funcs->write = nullptr;
 193         funcs->get_pixbuf = image_loader_ft_get_pixbuf;
 194         funcs->close = image_loader_ft_close;
 195         funcs->abort = image_loader_ft_abort;
 196         funcs->free = image_loader_ft_free;
 197
 198         funcs->get_format_name = image_loader_ft_get_format_name;
 199         funcs->get_format_mime_types = image_loader_ft_get_format_mime_types;
 200 }
 201
 202 #endif
 203 /* vim: set shiftwidth=8 softtabstop=0 cindent cinoptions={1s: */