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