1 /* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8; c-indent-level: 8 -*- */
3 * Copyright (C) 2005, Teemu Tervo <teemu.tervo@gmx.net>
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2, or (at your option)
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
22 #include <glib/gi18n.h>
23 #include <libgnomevfs/gnome-vfs-mime-utils.h>
24 #include <libgnomevfs/gnome-vfs-xfer.h>
26 #include "comics-document.h"
27 #include "ev-document-misc.h"
28 #include "ev-document-thumbnails.h"
30 struct _ComicsDocumentClass
32 GObjectClass parent_class;
35 struct _ComicsDocument
37 GObject parent_instance;
42 char *extract_command;
45 typedef struct _ComicsDocumentClass ComicsDocumentClass;
47 static void comics_document_document_iface_init (EvDocumentIface *iface);
48 static void comics_document_document_thumbnails_iface_init (EvDocumentThumbnailsIface *iface);
50 static GSList* get_supported_image_extensions (void);
51 static void get_page_size_area_prepared_cb (GdkPixbufLoader *loader,
53 static void render_pixbuf_size_prepared_cb (GdkPixbufLoader *loader,
57 static char** extract_argv (EvDocument *document,
61 G_DEFINE_TYPE_WITH_CODE (
62 ComicsDocument, comics_document, G_TYPE_OBJECT,
64 G_IMPLEMENT_INTERFACE (EV_TYPE_DOCUMENT,
65 comics_document_document_iface_init);
66 G_IMPLEMENT_INTERFACE (EV_TYPE_DOCUMENT_THUMBNAILS,
67 comics_document_document_thumbnails_iface_init);
71 comics_document_load (EvDocument *document,
75 ComicsDocument *comics_document = COMICS_DOCUMENT (document);
76 GSList *supported_extensions;
77 gchar *list_files_command = NULL, *stdout, *quoted_file, *mime_type;
82 comics_document->archive = g_filename_from_uri (uri, NULL, error);
83 g_return_val_if_fail (comics_document->archive != NULL, FALSE);
85 quoted_file = g_shell_quote (comics_document->archive);
86 mime_type = gnome_vfs_get_mime_type (uri);
88 /* FIXME, use proper cbr/cbz mime types once they're
89 * included in shared-mime-info */
90 if (!strcmp (mime_type, "application/x-cbr")) {
91 comics_document->extract_command =
92 g_strdup ("unrar p -c- -ierr");
94 g_strdup_printf ("unrar vb -c- -- %s", quoted_file);
95 } else if (!strcmp (mime_type, "application/x-cbz")) {
96 comics_document->extract_command =
97 g_strdup ("unzip -p -C");
99 g_strdup_printf ("zipinfo -1 -- %s", quoted_file);
102 g_free (quoted_file);
104 /* Get list of files in archive */
105 success = g_spawn_command_line_sync (list_files_command,
106 &stdout, NULL, &retval, error);
107 g_free (list_files_command);
112 } else if (retval != 0) {
115 EV_DOCUMENT_ERROR_INVALID,
116 _("File corrupted."));
121 cbr_files = g_strsplit (stdout, "\n", 0);
122 supported_extensions = get_supported_image_extensions ();
123 for (i = 0; cbr_files[i] != NULL; i++) {
124 gchar *suffix = g_strrstr (cbr_files[i], ".");
127 suffix = g_ascii_strdown (suffix + 1, -1);
129 if (g_slist_find_custom (supported_extensions, suffix,
130 (GCompareFunc) strcmp) != NULL) {
131 comics_document->page_names =
132 g_slist_insert_sorted (
133 comics_document->page_names,
134 g_strdup (g_strstrip (cbr_files[i])),
135 (GCompareFunc) strcmp);
136 comics_document->n_pages++;
144 g_strfreev (cbr_files);
145 g_slist_foreach (supported_extensions, (GFunc) g_free, NULL);
146 g_slist_free (supported_extensions);
148 if (comics_document->n_pages == 0) {
151 EV_DOCUMENT_ERROR_INVALID,
152 _("No images found in archive %s"),
162 comics_document_save (EvDocument *document,
166 ComicsDocument *comics_document = COMICS_DOCUMENT (document);
167 GnomeVFSURI *source, *target;
170 source = gnome_vfs_uri_new (comics_document->archive);
171 target = gnome_vfs_uri_new (uri);
173 ret = gnome_vfs_xfer_uri (source, target,
174 GNOME_VFS_XFER_DEFAULT,
175 GNOME_VFS_XFER_ERROR_MODE_ABORT,
176 GNOME_VFS_XFER_OVERWRITE_MODE_REPLACE,
179 gnome_vfs_uri_unref (source);
180 gnome_vfs_uri_unref (target);
182 return ret == GNOME_VFS_OK;
186 comics_document_get_n_pages (EvDocument *document)
188 return COMICS_DOCUMENT (document)->n_pages;
192 comics_document_get_page_size (EvDocument *document,
197 GdkPixbufLoader *loader;
200 gboolean success, got_size = FALSE;
204 argv = extract_argv (document, page);
205 success = g_spawn_async_with_pipes (NULL, argv, NULL,
206 G_SPAWN_SEARCH_PATH | G_SPAWN_STDERR_TO_DEV_NULL,
209 NULL, &outpipe, NULL, NULL);
211 g_return_if_fail (success == TRUE);
213 loader = gdk_pixbuf_loader_new ();
214 g_signal_connect (loader, "area-prepared",
215 G_CALLBACK (get_page_size_area_prepared_cb),
218 while (outpipe >= 0) {
219 gssize bytes = read (outpipe, buf, 1024);
222 gdk_pixbuf_loader_write (loader, buf, bytes, NULL);
223 if (bytes <= 0 || got_size) {
226 gdk_pixbuf_loader_close (loader, NULL);
230 if (gdk_pixbuf_loader_get_pixbuf (loader)) {
231 GdkPixbuf *pixbuf = gdk_pixbuf_loader_get_pixbuf (loader);
233 *width = gdk_pixbuf_get_width (pixbuf);
235 *height = gdk_pixbuf_get_height (pixbuf);
238 g_spawn_close_pid (child_pid);
239 g_object_unref (loader);
243 get_page_size_area_prepared_cb (GdkPixbufLoader *loader,
246 gboolean *got_size = data;
251 comics_document_render_pixbuf (EvDocument *document,
254 GdkPixbufLoader *loader;
255 GdkPixbuf *rotated_pixbuf;
262 argv = extract_argv (document, rc->page);
263 success = g_spawn_async_with_pipes (NULL, argv, NULL,
265 | G_SPAWN_STDERR_TO_DEV_NULL,
268 NULL, &outpipe, NULL, NULL);
270 g_return_val_if_fail (success == TRUE, NULL);
272 loader = gdk_pixbuf_loader_new ();
273 g_signal_connect (loader, "size-prepared",
274 G_CALLBACK (render_pixbuf_size_prepared_cb), &rc->scale);
276 while (outpipe >= 0) {
277 gssize bytes = read (outpipe, buf, 4096);
280 gdk_pixbuf_loader_write (loader, buf, bytes, NULL);
281 } else if (bytes <= 0) {
283 gdk_pixbuf_loader_close (loader, NULL);
288 rotated_pixbuf = gdk_pixbuf_rotate_simple (gdk_pixbuf_loader_get_pixbuf (loader),
290 g_spawn_close_pid (child_pid);
291 g_object_unref (loader);
292 return rotated_pixbuf;
296 render_pixbuf_size_prepared_cb (GdkPixbufLoader *loader,
301 double *scale = data;
302 int w = width * (*scale);
303 int h = height * (*scale);
305 gdk_pixbuf_loader_set_size (loader, w, h);
309 comics_document_finalize (GObject *object)
311 ComicsDocument *comics_document = COMICS_DOCUMENT (object);
313 if (comics_document->archive)
314 g_free (comics_document->archive);
316 if (comics_document->page_names) {
317 g_slist_foreach (comics_document->page_names,
318 (GFunc) g_free, NULL);
319 g_slist_free (comics_document->page_names);
322 if (comics_document->extract_command)
323 g_free (comics_document->extract_command);
325 G_OBJECT_CLASS (comics_document_parent_class)->finalize (object);
329 comics_document_class_init (ComicsDocumentClass *klass)
331 GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
332 gobject_class->finalize = comics_document_finalize;
336 comics_document_can_get_text (EvDocument *document)
341 static EvDocumentInfo *
342 comics_document_get_info (EvDocument *document)
344 EvDocumentInfo *info;
345 info = g_new0 (EvDocumentInfo, 1);
350 comics_document_document_iface_init (EvDocumentIface *iface)
352 iface->load = comics_document_load;
353 iface->save = comics_document_save;
354 iface->can_get_text = comics_document_can_get_text;
355 iface->get_n_pages = comics_document_get_n_pages;
356 iface->get_page_size = comics_document_get_page_size;
357 iface->render_pixbuf = comics_document_render_pixbuf;
358 iface->get_info = comics_document_get_info;
362 comics_document_init (ComicsDocument *comics_document)
364 comics_document->archive = NULL;
365 comics_document->page_names = NULL;
366 comics_document->extract_command = NULL;
367 comics_document->n_pages = 0;
370 /* Returns a list of file extensions supported by gdk-pixbuf */
372 get_supported_image_extensions()
374 GSList *extensions = NULL;
375 GSList *formats = gdk_pixbuf_get_formats ();
378 for (l = formats; l != NULL; l = l->next) {
380 gchar **ext = gdk_pixbuf_format_get_extensions (l->data);
382 for (i = 0; ext[i] != NULL; i++) {
383 extensions = g_slist_append (extensions,
390 g_slist_free (formats);
395 comics_document_thumbnails_get_geometry (EvDocumentThumbnails *document,
397 gint suggested_width,
400 gdouble *scale_factor)
402 gdouble orig_width, orig_height, scale;
404 comics_document_get_page_size (EV_DOCUMENT (document), page,
405 &orig_width, &orig_height);
406 scale = suggested_width / orig_width;
409 *width = suggested_width;
411 *height = orig_height * scale;
413 *scale_factor = scale;
417 comics_document_thumbnails_get_thumbnail (EvDocumentThumbnails *document,
423 GdkPixbuf *thumbnail, *framed;
424 gint thumb_width, thumb_height;
428 comics_document_thumbnails_get_geometry (document, page, size,
429 &thumb_width, &thumb_height,
432 rc = ev_render_context_new (rotation, page, scale);
433 thumbnail = comics_document_render_pixbuf (EV_DOCUMENT (document),
435 g_object_unref (G_OBJECT (rc));
438 GdkPixbuf *tmp_pixbuf = thumbnail;
439 thumbnail = ev_document_misc_get_thumbnail_frame (-1, -1, 0, tmp_pixbuf);
440 g_object_unref (tmp_pixbuf);
447 comics_document_thumbnails_get_dimensions (EvDocumentThumbnails *document,
449 gint suggested_width,
453 comics_document_thumbnails_get_geometry (document, page,
455 width, height, NULL);
459 comics_document_document_thumbnails_iface_init (EvDocumentThumbnailsIface *iface)
461 iface->get_thumbnail = comics_document_thumbnails_get_thumbnail;
462 iface->get_dimensions = comics_document_thumbnails_get_dimensions;
466 extract_argv (EvDocument *document, gint page)
468 ComicsDocument *comics_document = COMICS_DOCUMENT (document);
470 char *command_line, *quoted_archive, *quoted_filename;
472 quoted_archive = g_shell_quote (comics_document->archive);
473 quoted_filename = g_shell_quote (
474 g_slist_nth_data (comics_document->page_names, page));
476 command_line = g_strdup_printf ("%s -- %s %s",
477 comics_document->extract_command,
480 g_shell_parse_argv (command_line, NULL, &argv, NULL);
482 g_free (command_line);
483 g_free (quoted_archive);
484 g_free (quoted_filename);