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 #ifdef HAVE_FFMPEGTHUMBNAILER
27 #include <libffmpegthumbnailer/videothumbnailerc.h>
28
29 typedef struct _ImageLoaderFT ImageLoaderFT;
30 struct _ImageLoaderFT {
31 ImageLoaderBackendCbAreaUpdated area_updated_cb;
32 ImageLoaderBackendCbSize size_cb;
33 ImageLoaderBackendCbAreaPrepared area_prepared_cb;
34
35 video_thumbnailer *vt;
36
37 gpointer data;
38
39 GdkPixbuf *pixbuf;
40 guint requested_width;
41 guint requested_height;
42
43 };
44
45 #if HAVE_FFMPEGTHUMBNAILER_RGB
image_loader_ft_log_cb(ThumbnailerLogLevel log_level,const char * msg)46 static void image_loader_ft_log_cb(ThumbnailerLogLevel log_level, const char* msg)
47 {
48 if (log_level == ThumbnailerLogLevelError)
49 log_printf("ImageLoaderFFmpegthumbnailer: %s",msg);
50 else
51 DEBUG_1("ImageLoaderFFmpegthumbnailer: %s",msg);
52 }
53 #endif
54
image_loader_ft_destroy_image_data(guchar * pixels,gpointer data)55 void image_loader_ft_destroy_image_data(guchar *pixels, gpointer data)
56 {
57 image_data *image = (image_data *) data;
58
59 video_thumbnailer_destroy_image_data (image);
60 }
61
image_loader_ft_get_format_name(gpointer loader)62 static gchar* image_loader_ft_get_format_name(gpointer loader)
63 {
64 return g_strdup("ffmpeg");
65 }
66
image_loader_ft_get_format_mime_types(gpointer loader)67 static gchar** image_loader_ft_get_format_mime_types(gpointer loader)
68 {
69 static gchar *mime[] = {"video/mp4", NULL};
70 return g_strdupv(mime);}
71
image_loader_ft_new(ImageLoaderBackendCbAreaUpdated area_updated_cb,ImageLoaderBackendCbSize size_cb,ImageLoaderBackendCbAreaPrepared area_prepared_cb,gpointer data)72 static gpointer image_loader_ft_new(ImageLoaderBackendCbAreaUpdated area_updated_cb, ImageLoaderBackendCbSize size_cb, ImageLoaderBackendCbAreaPrepared area_prepared_cb, gpointer data)
73 {
74 ImageLoaderFT *loader = g_new0(ImageLoaderFT, 1);
75
76 loader->area_updated_cb = area_updated_cb;
77 loader->size_cb = size_cb;
78 loader->area_prepared_cb = area_prepared_cb;
79 loader->data = data;
80
81 loader->vt = video_thumbnailer_create();
82 loader->vt->overlay_film_strip = 1;
83 loader->vt->maintain_aspect_ratio = 1;
84 #if HAVE_FFMPEGTHUMBNAILER_RGB
85 video_thumbnailer_set_log_callback(loader->vt, image_loader_ft_log_cb);
86 #endif
87
88 return (gpointer) loader;
89 }
90
image_loader_ft_set_size(gpointer loader,int width,int height)91 static void image_loader_ft_set_size(gpointer loader, int width, int height)
92 {
93 ImageLoaderFT *lft = (ImageLoaderFT *) loader;
94 lft->requested_width = width;
95 lft->requested_height = height;
96 DEBUG_1("TG: setting size, w=%d, h=%d", width, height);
97 }
98
99 // static gboolean image_loader_ft_loadfromdisk(gpointer loader, const gchar *path, GError **error)
image_loader_ft_load(gpointer loader,const guchar * buf,gsize count,GError ** error)100 static gboolean image_loader_ft_load (gpointer loader, const guchar *buf, gsize count, GError **error)
101 {
102 ImageLoaderFT *lft = (ImageLoaderFT *) loader;
103 ImageLoader *il = lft->data;
104
105 image_data *image = video_thumbnailer_create_image_data();
106
107 #ifdef HAVE_FFMPEGTHUMBNAILER_WH
108 // DEBUG_1("TG: FT requested size w=%d:h=%d for %s", lft->requested_width > 0, lft->requested_height, il->fd->path);
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_width);
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, NULL);
133
134 if (image_stream == NULL)
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, NULL, NULL);
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
image_loader_ft_get_pixbuf(gpointer loader)163 static GdkPixbuf* image_loader_ft_get_pixbuf(gpointer loader)
164 {
165 ImageLoaderFT *lft = (ImageLoaderFT *) loader;
166 return lft->pixbuf;
167 }
168
image_loader_ft_abort(gpointer loader)169 static void image_loader_ft_abort(gpointer loader)
170 {
171 }
172
image_loader_ft_close(gpointer loader,GError ** error)173 static gboolean image_loader_ft_close(gpointer loader, GError **error)
174 {
175 return TRUE;
176 }
177
image_loader_ft_free(gpointer loader)178 static void image_loader_ft_free(gpointer loader)
179 {
180 ImageLoaderFT *lft = (ImageLoaderFT *) loader;
181 if (lft->pixbuf) g_object_unref(lft->pixbuf);
182 video_thumbnailer_destroy (lft->vt);
183
184 g_free(lft);
185 }
186
image_loader_backend_set_ft(ImageLoaderBackend * funcs)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 = NULL;
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: */
204