backend/djvu/djvu-document.c

   1 /* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8; c-indent-level: 8 -*- */
   2 /*
   3  * Copyright (C) 2005, Nickolay V. Shmyrev <nshmyrev@yandex.ru>
   4  *
   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)
   8  * any later version.
   9  *
  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.
  14  *
  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.
  18  */
  19
  20 #include "djvu-document.h"
  21 #include "djvu-text.h"
  22 #include "djvu-links.h"
  23 #include "djvu-document-private.h"
  24 #include "ev-document-thumbnails.h"
  25 #include "ev-document-misc.h"
  26 #include "ev-document-find.h"
  27 #include "ev-document-links.h"
  28
  29 #include <libdjvu/ddjvuapi.h>
  30 #include <libdjvu/miniexp.h>
  31 #include <gtk/gtk.h>
  32 #include <gdk-pixbuf/gdk-pixbuf-core.h>
  33 #include <glib/gunicode.h>
  34 #include <string.h>
  35
  36 #define SCALE_FACTOR 0.2
  37
  38 enum {
  39         PROP_0,
  40         PROP_TITLE
  41 };
  42
  43 struct _DjvuDocumentClass
  44 {
  45         GObjectClass parent_class;
  46 };
  47
  48 typedef struct _DjvuDocumentClass DjvuDocumentClass;
  49
  50 static void djvu_document_document_iface_init (EvDocumentIface *iface);
  51 static void djvu_document_document_thumbnails_iface_init (EvDocumentThumbnailsIface *iface);
  52 static void djvu_document_find_iface_init (EvDocumentFindIface *iface);
  53 static void djvu_document_document_links_iface_init  (EvDocumentLinksIface *iface);
  54
  55 G_DEFINE_TYPE_WITH_CODE
  56     (DjvuDocument, djvu_document, G_TYPE_OBJECT,
  57     {
  58       G_IMPLEMENT_INTERFACE (EV_TYPE_DOCUMENT, djvu_document_document_iface_init);
  59       G_IMPLEMENT_INTERFACE (EV_TYPE_DOCUMENT_THUMBNAILS, djvu_document_document_thumbnails_iface_init)
  60       G_IMPLEMENT_INTERFACE (EV_TYPE_DOCUMENT_FIND, djvu_document_find_iface_init);
  61       G_IMPLEMENT_INTERFACE (EV_TYPE_DOCUMENT_LINKS, djvu_document_document_links_iface_init);
  62      });
  63
  64
  65 void
  66 djvu_handle_events (DjvuDocument *djvu_document, int wait)
  67 {
  68         ddjvu_context_t *ctx = djvu_document->d_context;
  69         const ddjvu_message_t *msg;
  70         if (!ctx)
  71                 return;
  72         if (wait)
  73                 msg = ddjvu_message_wait (ctx);
  74         while ((msg = ddjvu_message_peek (ctx))) {
  75                 switch (msg->m_any.tag) {
  76                         case DDJVU_ERROR:
  77                                 g_warning ("DjvuLibre error: %s",
  78                                            msg->m_error.message);
  79                                 if (msg->m_error.filename)
  80                                         g_warning ("DjvuLibre error: %s:%d",
  81                                                    msg->m_error.filename,
  82                                                    msg->m_error.lineno);
  83                         default:
  84                                 break;
  85                 }
  86                 ddjvu_message_pop (ctx);
  87         }
  88 }
  89
  90 static gboolean
  91 djvu_document_load (EvDocument  *document,
  92                     const char  *uri,
  93                     GError     **error)
  94 {
  95         DjvuDocument *djvu_document = DJVU_DOCUMENT (document);
  96         ddjvu_document_t *doc;
  97         gchar *filename;
  98
  99         /* FIXME: We could actually load uris  */
 100         filename = g_filename_from_uri (uri, NULL, error);
 101         if (!filename)
 102                 return FALSE;
 103
 104         doc = ddjvu_document_create_by_filename (djvu_document->d_context, filename, TRUE);
 105
 106         if (!doc) return FALSE;
 107
 108         if (djvu_document->d_document)
 109             ddjvu_document_release (djvu_document->d_document);
 110
 111         djvu_document->d_document = doc;
 112
 113         while (!ddjvu_document_decoding_done (djvu_document->d_document))
 114                 djvu_handle_events(djvu_document, TRUE);
 115         g_free (djvu_document->uri);
 116         djvu_document->uri = g_strdup (uri);
 117
 118         return TRUE;
 119 }
 120
 121
 122 static gboolean
 123 djvu_document_save (EvDocument  *document,
 124                     const char  *uri,
 125                     GError     **error)
 126 {
 127         DjvuDocument *djvu_document = DJVU_DOCUMENT (document);
 128
 129         return ev_xfer_uri_simple (djvu_document->uri, uri, error);
 130 }
 131
 132 int
 133 djvu_document_get_n_pages (EvDocument  *document)
 134 {
 135         DjvuDocument *djvu_document = DJVU_DOCUMENT (document);
 136
 137         g_return_val_if_fail (djvu_document->d_document, 0);
 138
 139         return ddjvu_document_get_pagenum (djvu_document->d_document);
 140 }
 141
 142 static void
 143 djvu_document_get_page_size (EvDocument   *document,
 144                              int           page,
 145                              double       *width,
 146                              double       *height)
 147 {
 148         DjvuDocument *djvu_document = DJVU_DOCUMENT (document);
 149         ddjvu_pageinfo_t info;
 150         ddjvu_status_t r;
 151
 152         g_return_if_fail (djvu_document->d_document);
 153
 154         while ((r = ddjvu_document_get_pageinfo(djvu_document->d_document, page, &info)) < DDJVU_JOB_OK)
 155                 djvu_handle_events(djvu_document, TRUE);
 156
 157         if (r >= DDJVU_JOB_FAILED)
 158                 djvu_handle_events(djvu_document, TRUE);
 159
 160         *width = info.width * SCALE_FACTOR;
 161         *height = info.height * SCALE_FACTOR;
 162 }
 163
 164 static GdkPixbuf *
 165 djvu_document_render_pixbuf (EvDocument  *document,
 166                              EvRenderContext *rc)
 167 {
 168         DjvuDocument *djvu_document = DJVU_DOCUMENT (document);
 169         GdkPixbuf *pixbuf;
 170         GdkPixbuf *rotated_pixbuf;
 171
 172         ddjvu_rect_t rrect;
 173         ddjvu_rect_t prect;
 174         ddjvu_page_t *d_page;
 175
 176         double page_width, page_height;
 177
 178         d_page = ddjvu_page_create_by_pageno (djvu_document->d_document, rc->page);
 179
 180         while (!ddjvu_page_decoding_done (d_page))
 181                 djvu_handle_events(djvu_document, TRUE);
 182
 183         page_width = ddjvu_page_get_width (d_page) * rc->scale * SCALE_FACTOR;
 184         page_height = ddjvu_page_get_height (d_page) * rc->scale * SCALE_FACTOR;
 185
 186         pixbuf = gdk_pixbuf_new (GDK_COLORSPACE_RGB, FALSE, 8, page_width, page_height);
 187
 188         prect.x = 0; prect.y = 0;
 189         prect.w = page_width; prect.h = page_height;
 190         rrect = prect;
 191
 192         ddjvu_page_render(d_page, DDJVU_RENDER_COLOR,
 193                           &prect,
 194                           &rrect,
 195                           djvu_document->d_format,
 196                           gdk_pixbuf_get_rowstride (pixbuf),
 197                           (gchar *)gdk_pixbuf_get_pixels (pixbuf));
 198
 199         rotated_pixbuf = gdk_pixbuf_rotate_simple (pixbuf, 360 - rc->rotation);
 200         g_object_unref (pixbuf);
 201
 202         return rotated_pixbuf;
 203 }
 204
 205 static void
 206 djvu_document_finalize (GObject *object)
 207 {
 208         DjvuDocument *djvu_document = DJVU_DOCUMENT (object);
 209
 210         if (djvu_document->d_document)
 211             ddjvu_document_release (djvu_document->d_document);
 212
 213         ddjvu_context_release (djvu_document->d_context);
 214         ddjvu_format_release (djvu_document->d_format);
 215         g_free (djvu_document->uri);
 216
 217         G_OBJECT_CLASS (djvu_document_parent_class)->finalize (object);
 218 }
 219
 220 static void
 221 djvu_document_class_init (DjvuDocumentClass *klass)
 222 {
 223         GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
 224
 225         gobject_class->finalize = djvu_document_finalize;
 226 }
 227
 228 static gboolean
 229 djvu_document_can_get_text (EvDocument *document)
 230 {
 231         return TRUE;
 232 }
 233
 234
 235 static char *
 236 djvu_document_get_text (EvDocument *document, int page, EvRectangle *rect)
 237 {
 238         DjvuDocument *djvu_document = DJVU_DOCUMENT (document);
 239         double width, height;
 240         EvRectangle rectangle;
 241         char* text;
 242
 243         djvu_document_get_page_size (document, page, &width, &height);
 244         rectangle.x1 = rect->x1 / SCALE_FACTOR;
 245         rectangle.y1 = (height - rect->y2) / SCALE_FACTOR;
 246         rectangle.x2 = rect->x2 / SCALE_FACTOR;
 247         rectangle.y2 = (height - rect->y1) / SCALE_FACTOR;
 248
 249         text = djvu_text_copy (djvu_document, page, &rectangle);
 250
 251         if (text == NULL)
 252                 text = g_strdup ("");
 253
 254         return text;
 255 }
 256
 257 static EvDocumentInfo *
 258 djvu_document_get_info (EvDocument *document)
 259 {
 260         EvDocumentInfo *info;
 261
 262         info = g_new0 (EvDocumentInfo, 1);
 263
 264         return info;
 265 }
 266
 267 static void
 268 djvu_document_document_iface_init (EvDocumentIface *iface)
 269 {
 270         iface->load = djvu_document_load;
 271         iface->save = djvu_document_save;
 272         iface->can_get_text = djvu_document_can_get_text;
 273         iface->get_text = djvu_document_get_text;
 274         iface->get_n_pages = djvu_document_get_n_pages;
 275         iface->get_page_size = djvu_document_get_page_size;
 276         iface->render_pixbuf = djvu_document_render_pixbuf;
 277         iface->get_info = djvu_document_get_info;
 278 }
 279
 280 static void
 281 djvu_document_thumbnails_get_dimensions (EvDocumentThumbnails *document,
 282                                            gint                  page,
 283                                            gint                  suggested_width,
 284                                            gint                  *width,
 285                                            gint                  *height)
 286 {
 287         DjvuDocument *djvu_document = DJVU_DOCUMENT (document);
 288         gdouble p_width, p_height;
 289         gdouble page_ratio;
 290
 291         djvu_document_get_page_size (EV_DOCUMENT(djvu_document), page, &p_width, &p_height);
 292
 293         page_ratio = p_height / p_width;
 294         *width = suggested_width;
 295         *height = (gint) (suggested_width * page_ratio);
 296
 297         return;
 298 }
 299
 300 static GdkPixbuf *
 301 djvu_document_thumbnails_get_thumbnail (EvDocumentThumbnails   *document,
 302                                           gint                   page,
 303                                           gint                   rotation,
 304                                           gint                   width,
 305                                           gboolean               border)
 306 {
 307         DjvuDocument *djvu_document = DJVU_DOCUMENT (document);
 308         GdkPixbuf *pixbuf, *rotated_pixbuf;
 309         gint thumb_width, thumb_height;
 310
 311         guchar *pixels;
 312
 313         g_return_val_if_fail (djvu_document->d_document, NULL);
 314
 315         djvu_document_thumbnails_get_dimensions (document, page, width, &thumb_width, &thumb_height);
 316
 317         pixbuf = gdk_pixbuf_new (GDK_COLORSPACE_RGB, FALSE, 8,
 318                                  thumb_width, thumb_height);
 319         gdk_pixbuf_fill (pixbuf, 0xffffffff);
 320         pixels = gdk_pixbuf_get_pixels (pixbuf);
 321
 322         while (ddjvu_thumbnail_status (djvu_document->d_document, page, 1) < DDJVU_JOB_OK)
 323                 djvu_handle_events(djvu_document, TRUE);
 324
 325         ddjvu_thumbnail_render (djvu_document->d_document, page,
 326                                 &thumb_width, &thumb_height,
 327                                 djvu_document->d_format,
 328                                 gdk_pixbuf_get_rowstride (pixbuf),
 329                                 (gchar *)pixels);
 330
 331         rotated_pixbuf = gdk_pixbuf_rotate_simple (pixbuf, 360 - rotation);
 332         g_object_unref (pixbuf);
 333
 334         if (border) {
 335               GdkPixbuf *tmp_pixbuf = rotated_pixbuf;
 336               rotated_pixbuf = ev_document_misc_get_thumbnail_frame (-1, -1, 0, tmp_pixbuf);
 337               g_object_unref (tmp_pixbuf);
 338         }
 339
 340         return rotated_pixbuf;
 341 }
 342
 343 static void
 344 djvu_document_document_thumbnails_iface_init (EvDocumentThumbnailsIface *iface)
 345 {
 346         iface->get_thumbnail = djvu_document_thumbnails_get_thumbnail;
 347         iface->get_dimensions = djvu_document_thumbnails_get_dimensions;
 348 }
 349
 350 static void
 351 djvu_document_init (DjvuDocument *djvu_document)
 352 {
 353         djvu_document->d_context = ddjvu_context_create ("Evince");
 354         djvu_document->d_format = ddjvu_format_create (DDJVU_FORMAT_RGB24, 0, 0);
 355         ddjvu_format_set_row_order (djvu_document->d_format,1);
 356
 357         djvu_document->d_document = NULL;
 358 }
 359
 360 static void
 361 djvu_document_find_begin (EvDocumentFind   *document,
 362                           int               page,
 363                           const char       *search_string,
 364                           gboolean          case_sensitive)
 365 {
 366         DjvuDocument *djvu_document = DJVU_DOCUMENT (document);
 367
 368         if (djvu_document->search &&
 369             strcmp (search_string, djvu_text_get_text (djvu_document->search)) == 0)
 370                 return;
 371
 372         if (djvu_document->search)
 373                 djvu_text_free (djvu_document->search);
 374
 375         djvu_document->search = djvu_text_new (djvu_document,
 376                                                           page,
 377                                                           case_sensitive,
 378                                                           search_string);
 379 }
 380
 381 static int
 382 djvu_document_find_get_n_results (EvDocumentFind *document_find, int page)
 383 {
 384         DjvuText *search = DJVU_DOCUMENT (document_find)->search;
 385
 386         if (search) {
 387                 return djvu_text_n_results (search, page);
 388         } else {
 389                 return 0;
 390         }
 391 }
 392
 393 static gboolean
 394 djvu_document_find_get_result (EvDocumentFind *document_find,
 395                                int             page,
 396                                int             n_result,
 397                                EvRectangle    *rectangle)
 398 {
 399         DjvuDocument *djvu_document = DJVU_DOCUMENT (document_find);
 400         DjvuText *search = djvu_document->search;
 401         EvRectangle *r;
 402         double width, height;
 403
 404         if (search == NULL)
 405                 return FALSE;
 406
 407         r = djvu_text_get_result (search, page, n_result);
 408         if (r == NULL)
 409                 return FALSE;
 410
 411         djvu_document_get_page_size (EV_DOCUMENT (djvu_document),
 412                 page, &width, &height);
 413         rectangle->x1 = r->x1 * SCALE_FACTOR;
 414         rectangle->y1 = height - r->y2 * SCALE_FACTOR;
 415         rectangle->x2 = r->x2 * SCALE_FACTOR;
 416         rectangle->y2 = height - r->y1 * SCALE_FACTOR;
 417
 418         return TRUE;
 419 }
 420
 421 static int
 422 djvu_document_find_page_has_results (EvDocumentFind *document_find,
 423                                     int             page)
 424 {
 425         DjvuText *search = DJVU_DOCUMENT (document_find)->search;
 426
 427         return search && djvu_text_has_results (search, page);
 428 }
 429
 430 static double
 431 djvu_document_find_get_progress (EvDocumentFind *document_find)
 432 {
 433         DjvuText *search = DJVU_DOCUMENT (document_find)->search;
 434
 435         if (search == NULL) {
 436                 return 0;
 437         }
 438
 439         return djvu_text_get_progress (search);
 440 }
 441
 442 static void
 443 djvu_document_find_cancel (EvDocumentFind *document)
 444 {
 445         DjvuDocument *djvu_document = DJVU_DOCUMENT (document);
 446
 447         if (djvu_document->search) {
 448                 djvu_text_free (djvu_document->search);
 449                 djvu_document->search = NULL;
 450         }
 451 }
 452
 453 static void
 454 djvu_document_find_iface_init (EvDocumentFindIface *iface)
 455 {
 456         iface->begin = djvu_document_find_begin;
 457         iface->get_n_results = djvu_document_find_get_n_results;
 458         iface->get_result = djvu_document_find_get_result;
 459         iface->page_has_results = djvu_document_find_page_has_results;
 460         iface->get_progress = djvu_document_find_get_progress;
 461         iface->cancel = djvu_document_find_cancel;
 462 }
 463
 464 static GList *
 465 djvu_document_links_get_links (EvDocumentLinks *document_links,
 466                                gint             page)
 467 {
 468         return djvu_links_get_links (document_links, page, SCALE_FACTOR);
 469 }
 470
 471 static void
 472 djvu_document_document_links_iface_init  (EvDocumentLinksIface *iface)
 473 {
 474         iface->has_document_links = djvu_links_has_document_links;
 475         iface->get_links_model = djvu_links_get_links_model;
 476         iface->get_links = djvu_document_links_get_links;
 477         iface->find_link_dest = djvu_links_find_link_dest;
 478 }