pdf/ev-poppler.cc

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8; c-indent-level: 8 -*- */
   2 /* pdfdocument.h: Implementation of EvDocument for PDF
   3  * Copyright (C) 2004, Red Hat, Inc.
   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 <math.h>
  21 #include <string.h>
  22 #include <gtk/gtk.h>
  23 #include <poppler.h>
  24 #include <poppler-document.h>
  25 #include <poppler-page.h>
  26 #include <glib/gi18n.h>
  27
  28 #include "ev-poppler.h"
  29 #include "ev-ps-exporter.h"
  30 #include "ev-document-find.h"
  31 #include "ev-document-misc.h"
  32 #include "ev-document-links.h"
  33 #include "ev-document-fonts.h"
  34 #include "ev-document-security.h"
  35 #include "ev-document-thumbnails.h"
  36
  37 typedef struct {
  38         PdfDocument *document;
  39         char *text;
  40         GList **pages;
  41         guint idle;
  42         int start_page;
  43         int search_page;
  44 } PdfDocumentSearch;
  45
  46 struct _PdfDocumentClass
  47 {
  48         GObjectClass parent_class;
  49 };
  50
  51 struct _PdfDocument
  52 {
  53         GObject parent_instance;
  54
  55         PopplerDocument *document;
  56         PopplerPSFile *ps_file;
  57         gchar *password;
  58
  59         PdfDocumentSearch *search;
  60 };
  61
  62 static void pdf_document_document_iface_init            (EvDocumentIface           *iface);
  63 static void pdf_document_security_iface_init            (EvDocumentSecurityIface   *iface);
  64 static void pdf_document_document_thumbnails_iface_init (EvDocumentThumbnailsIface *iface);
  65 static void pdf_document_document_links_iface_init      (EvDocumentLinksIface      *iface);
  66 static void pdf_document_document_fonts_iface_init      (EvDocumentFontsIface      *iface);
  67 static void pdf_document_find_iface_init                (EvDocumentFindIface       *iface);
  68 static void pdf_document_ps_exporter_iface_init         (EvPSExporterIface         *iface);
  69 static void pdf_document_thumbnails_get_dimensions      (EvDocumentThumbnails      *document_thumbnails,
  70                                                          gint                       page,
  71                                                          gint                       size,
  72                                                          gint                      *width,
  73                                                          gint                      *height);
  74 static EvLink * ev_link_from_action (PopplerAction *action);
  75
  76
  77 G_DEFINE_TYPE_WITH_CODE (PdfDocument, pdf_document, G_TYPE_OBJECT,
  78                          {
  79                                  G_IMPLEMENT_INTERFACE (EV_TYPE_DOCUMENT,
  80                                                         pdf_document_document_iface_init);
  81                                  G_IMPLEMENT_INTERFACE (EV_TYPE_DOCUMENT_SECURITY,
  82                                                         pdf_document_security_iface_init);
  83                                  G_IMPLEMENT_INTERFACE (EV_TYPE_DOCUMENT_THUMBNAILS,
  84                                                         pdf_document_document_thumbnails_iface_init);
  85                                  G_IMPLEMENT_INTERFACE (EV_TYPE_DOCUMENT_LINKS,
  86                                                         pdf_document_document_links_iface_init);
  87                                  G_IMPLEMENT_INTERFACE (EV_TYPE_DOCUMENT_FONTS,
  88                                                         pdf_document_document_fonts_iface_init);
  89                                  G_IMPLEMENT_INTERFACE (EV_TYPE_DOCUMENT_FIND,
  90                                                         pdf_document_find_iface_init);
  91                                  G_IMPLEMENT_INTERFACE (EV_TYPE_PS_EXPORTER,
  92                                                         pdf_document_ps_exporter_iface_init);
  93                          });
  94
  95 static void
  96 pdf_document_dispose (GObject *object)
  97 {
  98         PdfDocument *pdf_document = PDF_DOCUMENT(object);
  99
 100         if (pdf_document->document) {
 101                 g_object_unref (pdf_document->document);
 102         }
 103 }
 104
 105 static void
 106 pdf_document_class_init (PdfDocumentClass *klass)
 107 {
 108         GObjectClass *g_object_class = G_OBJECT_CLASS (klass);
 109
 110         g_object_class->dispose = pdf_document_dispose;
 111 }
 112
 113 static void
 114 pdf_document_init (PdfDocument *pdf_document)
 115 {
 116         pdf_document->password = NULL;
 117 }
 118
 119 static void
 120 convert_error (GError  *poppler_error,
 121                GError **error)
 122 {
 123         if (poppler_error == NULL)
 124                 return;
 125
 126         if (poppler_error->domain == POPPLER_ERROR) {
 127                 /* convert poppler errors into EvDocument errors */
 128                 gint code = EV_DOCUMENT_ERROR_INVALID;
 129                 if (poppler_error->code == POPPLER_ERROR_INVALID)
 130                         code = EV_DOCUMENT_ERROR_INVALID;
 131                 else if (poppler_error->code == POPPLER_ERROR_ENCRYPTED)
 132                         code = EV_DOCUMENT_ERROR_ENCRYPTED;
 133
 134
 135                 g_set_error (error,
 136                              EV_DOCUMENT_ERROR,
 137                              code,
 138                              poppler_error->message,
 139                              NULL);
 140         } else {
 141                 g_propagate_error (error, poppler_error);
 142         }
 143 }
 144
 145
 146 /* EvDocument */
 147 static gboolean
 148 pdf_document_save (EvDocument  *document,
 149                    const char  *uri,
 150                    GError     **error)
 151 {
 152         gboolean retval;
 153         GError *poppler_error = NULL;
 154
 155         retval = poppler_document_save (PDF_DOCUMENT (document)->document,
 156                                         uri,
 157                                         &poppler_error);
 158         if (! retval)
 159                 convert_error (poppler_error, error);
 160
 161         return retval;
 162 }
 163
 164 static gboolean
 165 pdf_document_load (EvDocument   *document,
 166                    const char   *uri,
 167                    GError      **error)
 168 {
 169         GError *poppler_error = NULL;
 170         PdfDocument *pdf_document = PDF_DOCUMENT (document);
 171
 172         pdf_document->document =
 173                 poppler_document_new_from_file (uri, pdf_document->password, &poppler_error);
 174
 175         if (pdf_document->document == NULL) {
 176                 convert_error (poppler_error, error);
 177                 return FALSE;
 178         }
 179
 180         return TRUE;
 181 }
 182
 183 static int
 184 pdf_document_get_n_pages (EvDocument *document)
 185 {
 186         return poppler_document_get_n_pages (PDF_DOCUMENT (document)->document);
 187 }
 188
 189 static void
 190 pdf_document_get_page_size (EvDocument   *document,
 191                             int           page,
 192                             double       *width,
 193                             double       *height)
 194 {
 195         PopplerPage *poppler_page;
 196
 197         poppler_page = poppler_document_get_page (PDF_DOCUMENT (document)->document,
 198                                                   page);
 199
 200         poppler_page_get_size (poppler_page, width, height);
 201 }
 202
 203 static char *
 204 pdf_document_get_page_label (EvDocument *document,
 205                              int         page)
 206 {
 207         PopplerPage *poppler_page;
 208         char *label = NULL;
 209
 210         poppler_page = poppler_document_get_page (PDF_DOCUMENT (document)->document,
 211                                                   page);
 212
 213         g_object_get (G_OBJECT (poppler_page),
 214                       "label", &label,
 215                       NULL);
 216
 217         return label;
 218 }
 219
 220 static GList *
 221 pdf_document_get_links (EvDocument *document,
 222                         int         page)
 223 {
 224         PdfDocument *pdf_document;
 225         PopplerPage *poppler_page;
 226         GList *retval = NULL;
 227         GList *mapping_list;
 228         GList *list;
 229         double height;
 230
 231         pdf_document = PDF_DOCUMENT (document);
 232         poppler_page = poppler_document_get_page (pdf_document->document,
 233                                                   page);
 234         mapping_list = poppler_page_get_link_mapping (poppler_page);
 235         poppler_page_get_size (poppler_page, NULL, &height);
 236
 237         for (list = mapping_list; list; list = list->next) {
 238                 PopplerLinkMapping *link_mapping;
 239                 EvLinkMapping *ev_link_mapping;
 240
 241                 link_mapping = (PopplerLinkMapping *)list->data;
 242                 ev_link_mapping = g_new (EvLinkMapping, 1);
 243                 ev_link_mapping->link = ev_link_from_action (link_mapping->action);
 244                 ev_link_mapping->x1 = link_mapping->area.x1;
 245                 ev_link_mapping->x2 = link_mapping->area.x2;
 246                 /* Invert this for X-style coordinates */
 247                 ev_link_mapping->y1 = height - link_mapping->area.y2;
 248                 ev_link_mapping->y2 = height - link_mapping->area.y1;
 249
 250                 retval = g_list_prepend (retval, ev_link_mapping);
 251         }
 252
 253         poppler_page_free_link_mapping (mapping_list);
 254
 255         return g_list_reverse (retval);
 256 }
 257
 258
 259 static GdkPixbuf *
 260 pdf_document_render_pixbuf (EvDocument   *document,
 261                             int           page,
 262                             double        scale)
 263 {
 264         PdfDocument *pdf_document;
 265         PopplerPage *poppler_page;
 266         GdkPixbuf *pixbuf;
 267         double width_points, height_points;
 268         gint width, height;
 269
 270         pdf_document = PDF_DOCUMENT (document);
 271         poppler_page = poppler_document_get_page (pdf_document->document,
 272                                                   page);
 273
 274         poppler_page_get_size (poppler_page, &width_points, &height_points);
 275         width = (int) ((width_points * scale) + 0.5);
 276         height = (int) ((height_points * scale) + 0.5);
 277
 278         pixbuf = gdk_pixbuf_new (GDK_COLORSPACE_RGB,
 279                                  FALSE, 8,
 280                                  width, height);
 281
 282         poppler_page_render_to_pixbuf (poppler_page,
 283                                        0, 0,
 284                                        width, height,
 285                                        scale,
 286                                        pixbuf,
 287                                        0, 0);
 288
 289         return pixbuf;
 290 }
 291
 292 /* EvDocumentSecurity */
 293
 294 static gboolean
 295 pdf_document_has_document_security (EvDocumentSecurity *document_security)
 296 {
 297         /* FIXME: do we really need to have this? */
 298         return FALSE;
 299 }
 300
 301 static void
 302 pdf_document_set_password (EvDocumentSecurity *document_security,
 303                            const char         *password)
 304 {
 305         PdfDocument *document = PDF_DOCUMENT (document_security);
 306
 307         if (document->password)
 308                 g_free (document->password);
 309
 310         document->password = g_strdup (password);
 311 }
 312
 313 static gboolean
 314 pdf_document_can_get_text (EvDocument *document)
 315 {
 316         return TRUE;
 317 }
 318
 319 static EvDocumentInfo *
 320 pdf_document_get_info (EvDocument *document)
 321 {
 322         EvDocumentInfo *info;
 323         PopplerPageLayout layout;
 324         PopplerPageMode mode;
 325         PopplerViewerPreferences view_prefs;
 326         PopplerPermissions permissions;
 327
 328         info = g_new0 (EvDocumentInfo, 1);
 329
 330         info->fields_mask = EV_DOCUMENT_INFO_TITLE |
 331                             EV_DOCUMENT_INFO_FORMAT |
 332                             EV_DOCUMENT_INFO_AUTHOR |
 333                             EV_DOCUMENT_INFO_SUBJECT |
 334                             EV_DOCUMENT_INFO_KEYWORDS |
 335                             EV_DOCUMENT_INFO_LAYOUT |
 336                             EV_DOCUMENT_INFO_START_MODE |
 337                             EV_DOCUMENT_INFO_PERMISSIONS |
 338                             EV_DOCUMENT_INFO_UI_HINTS |
 339                             EV_DOCUMENT_INFO_CREATOR |
 340                             EV_DOCUMENT_INFO_PRODUCER |
 341                             EV_DOCUMENT_INFO_CREATION_DATE |
 342                             EV_DOCUMENT_INFO_MOD_DATE |
 343                             EV_DOCUMENT_INFO_LINEARIZED |
 344                             EV_DOCUMENT_INFO_N_PAGES |
 345                             EV_DOCUMENT_INFO_SECURITY;
 346
 347
 348         g_object_get (PDF_DOCUMENT (document)->document,
 349                       "title", &(info->title),
 350                       "format", &(info->format),
 351                       "author", &(info->author),
 352                       "subject", &(info->subject),
 353                       "keywords", &(info->keywords),
 354                       "page-mode", &mode,
 355                       "page-layout", &layout,
 356                       "viewer-preferences", &view_prefs,
 357                       "permissions", &permissions,
 358                       "creator", &(info->creator),
 359                       "producer", &(info->producer),
 360                       "creation-date", &(info->creation_date),
 361                       "mod-date", &(info->modified_date),
 362                       "linearized", &(info->linearized),
 363                       NULL);
 364
 365         switch (layout) {
 366                 case POPPLER_PAGE_LAYOUT_SINGLE_PAGE:
 367                         info->layout = EV_DOCUMENT_LAYOUT_SINGLE_PAGE;
 368                         break;
 369                 case POPPLER_PAGE_LAYOUT_ONE_COLUMN:
 370                         info->layout = EV_DOCUMENT_LAYOUT_ONE_COLUMN;
 371                         break;
 372                 case POPPLER_PAGE_LAYOUT_TWO_COLUMN_LEFT:
 373                         info->layout = EV_DOCUMENT_LAYOUT_TWO_COLUMN_LEFT;
 374                         break;
 375                 case POPPLER_PAGE_LAYOUT_TWO_COLUMN_RIGHT:
 376                         info->layout = EV_DOCUMENT_LAYOUT_TWO_COLUMN_RIGHT;
 377                 case POPPLER_PAGE_LAYOUT_TWO_PAGE_LEFT:
 378                         info->layout = EV_DOCUMENT_LAYOUT_TWO_PAGE_LEFT;
 379                         break;
 380                 case POPPLER_PAGE_LAYOUT_TWO_PAGE_RIGHT:
 381                         info->layout = EV_DOCUMENT_LAYOUT_TWO_PAGE_RIGHT;
 382                         break;
 383                 default:
 384                         break;
 385         }
 386
 387         switch (mode) {
 388                 case POPPLER_PAGE_MODE_NONE:
 389                         info->mode = EV_DOCUMENT_MODE_NONE;
 390                         break;
 391                 case POPPLER_PAGE_MODE_USE_THUMBS:
 392                         info->mode = EV_DOCUMENT_MODE_USE_THUMBS;
 393                         break;
 394                 case POPPLER_PAGE_MODE_USE_OC:
 395                         info->mode = EV_DOCUMENT_MODE_USE_OC;
 396                         break;
 397                 case POPPLER_PAGE_MODE_FULL_SCREEN:
 398                         info->mode = EV_DOCUMENT_MODE_FULL_SCREEN;
 399                         break;
 400                 case POPPLER_PAGE_MODE_USE_ATTACHMENTS:
 401                         info->mode = EV_DOCUMENT_MODE_USE_ATTACHMENTS;
 402                 default:
 403                         break;
 404         }
 405
 406         info->ui_hints = 0;
 407         if (view_prefs & POPPLER_VIEWER_PREFERENCES_HIDE_TOOLBAR) {
 408                 info->ui_hints |= EV_DOCUMENT_UI_HINT_HIDE_TOOLBAR;
 409         }
 410         if (view_prefs & POPPLER_VIEWER_PREFERENCES_HIDE_MENUBAR) {
 411                 info->ui_hints |= EV_DOCUMENT_UI_HINT_HIDE_MENUBAR;
 412         }
 413         if (view_prefs & POPPLER_VIEWER_PREFERENCES_HIDE_WINDOWUI) {
 414                 info->ui_hints |= EV_DOCUMENT_UI_HINT_HIDE_WINDOWUI;
 415         }
 416         if (view_prefs & POPPLER_VIEWER_PREFERENCES_FIT_WINDOW) {
 417                 info->ui_hints |= EV_DOCUMENT_UI_HINT_FIT_WINDOW;
 418         }
 419         if (view_prefs & POPPLER_VIEWER_PREFERENCES_CENTER_WINDOW) {
 420                 info->ui_hints |= EV_DOCUMENT_UI_HINT_CENTER_WINDOW;
 421         }
 422         if (view_prefs & POPPLER_VIEWER_PREFERENCES_DISPLAY_DOC_TITLE) {
 423                 info->ui_hints |= EV_DOCUMENT_UI_HINT_DISPLAY_DOC_TITLE;
 424         }
 425         if (view_prefs & POPPLER_VIEWER_PREFERENCES_DIRECTION_RTL) {
 426                 info->ui_hints |=  EV_DOCUMENT_UI_HINT_DIRECTION_RTL;
 427         }
 428
 429         info->permissions = 0;
 430         if (permissions & POPPLER_PERMISSIONS_OK_TO_PRINT) {
 431                 info->permissions |= EV_DOCUMENT_PERMISSIONS_OK_TO_PRINT;
 432         }
 433         if (permissions & POPPLER_PERMISSIONS_OK_TO_MODIFY) {
 434                 info->permissions |= EV_DOCUMENT_PERMISSIONS_OK_TO_MODIFY;
 435         }
 436         if (permissions & POPPLER_PERMISSIONS_OK_TO_COPY) {
 437                 info->permissions |= EV_DOCUMENT_PERMISSIONS_OK_TO_COPY;
 438         }
 439         if (permissions & POPPLER_PERMISSIONS_OK_TO_ADD_NOTES) {
 440                 info->permissions |= EV_DOCUMENT_PERMISSIONS_OK_TO_ADD_NOTES;
 441         }
 442
 443         info->n_pages = ev_document_get_n_pages (document);
 444
 445         if (ev_document_security_has_document_security (EV_DOCUMENT_SECURITY (document))) {
 446                 /* translators: this is the document security state */
 447                 info->security = g_strdup (_("Yes"));
 448         } else {
 449                 /* translators: this is the document security state */
 450                 info->security = g_strdup (_("No"));
 451         }
 452
 453         return info;
 454 }
 455
 456 static char *
 457 pdf_document_get_text (EvDocument *document, int page, EvRectangle *rect)
 458 {
 459         PdfDocument *pdf_document = PDF_DOCUMENT (document);
 460         PopplerPage *poppler_page;
 461         PopplerRectangle r;
 462         double height;
 463
 464         poppler_page = poppler_document_get_page (pdf_document->document, page);
 465         g_return_val_if_fail (poppler_page != NULL, NULL);
 466
 467         poppler_page_get_size (poppler_page, NULL, &height);
 468         r.x1 = rect->x1;
 469         r.y1 = height - rect->y2;
 470         r.x2 = rect->x2;
 471         r.y2 = height - rect->y1;
 472
 473         return poppler_page_get_text (poppler_page, &r);
 474 }
 475
 476 static void
 477 pdf_document_document_iface_init (EvDocumentIface *iface)
 478 {
 479         iface->save = pdf_document_save;
 480         iface->load = pdf_document_load;
 481         iface->get_n_pages = pdf_document_get_n_pages;
 482         iface->get_page_size = pdf_document_get_page_size;
 483         iface->get_page_label = pdf_document_get_page_label;
 484         iface->get_links = pdf_document_get_links;
 485         iface->render_pixbuf = pdf_document_render_pixbuf;
 486         iface->get_text = pdf_document_get_text;
 487         iface->can_get_text = pdf_document_can_get_text;
 488         iface->get_info = pdf_document_get_info;
 489 };
 490
 491 static void
 492 pdf_document_security_iface_init (EvDocumentSecurityIface *iface)
 493 {
 494         iface->has_document_security = pdf_document_has_document_security;
 495         iface->set_password = pdf_document_set_password;
 496 }
 497
 498 static void
 499 build_fonts_list (PdfDocument      *pdf_document,
 500                   GtkTreeModel     *model,
 501                   GtkTreeIter      *parent,
 502                   PopplerFontsIter *iter)
 503 {
 504 #ifdef POPPLER_FONT_INFO
 505         do {
 506                 GtkTreeIter list_iter;
 507                 PopplerIndexIter *child;
 508                 const char *name;
 509
 510                 name = poppler_fonts_iter_get_name (iter);
 511                 gtk_list_store_append (GTK_LIST_STORE (model), &list_iter);
 512                 gtk_list_store_set (GTK_LIST_STORE (model), &list_iter,
 513                                     EV_DOCUMENT_FONTS_COLUMN_NAME, name,
 514                                     -1);
 515         } while (poppler_fonts_iter_next (iter));
 516 #endif
 517 }
 518
 519 static GtkTreeModel *
 520 pdf_document_fonts_get_fonts_model (EvDocumentFonts *document_fonts)
 521 {
 522         PdfDocument *pdf_document = PDF_DOCUMENT (document_fonts);
 523         GtkTreeModel *model = NULL;
 524         PopplerFontsIter *iter;
 525
 526         g_return_val_if_fail (PDF_IS_DOCUMENT (document_fonts), NULL);
 527
 528         iter = poppler_fonts_iter_new (pdf_document->document);
 529         /* Create the model iff we have items*/
 530         if (iter != NULL) {
 531                 model = (GtkTreeModel *) gtk_list_store_new (EV_DOCUMENT_FONTS_COLUMN_NUM_COLUMNS,
 532                                                              G_TYPE_STRING);
 533                 build_fonts_list (pdf_document, model, NULL, iter);
 534                 poppler_fonts_iter_free (iter);
 535         }
 536
 537         return model;
 538 }
 539
 540 static void
 541 pdf_document_document_fonts_iface_init (EvDocumentFontsIface *iface)
 542 {
 543         iface->get_fonts_model = pdf_document_fonts_get_fonts_model;
 544 }
 545
 546 static gboolean
 547 pdf_document_links_has_document_links (EvDocumentLinks *document_links)
 548 {
 549         PdfDocument *pdf_document = PDF_DOCUMENT (document_links);
 550         PopplerIndexIter *iter;
 551
 552         g_return_val_if_fail (PDF_IS_DOCUMENT (document_links), FALSE);
 553
 554         iter = poppler_index_iter_new (pdf_document->document);
 555         if (iter == NULL)
 556                 return FALSE;
 557         poppler_index_iter_free (iter);
 558
 559         return TRUE;
 560 }
 561
 562 static EvLink *
 563 ev_link_from_action (PopplerAction *action)
 564 {
 565         EvLink *link;
 566         const char *title;
 567
 568         title = action->any.title;
 569
 570         if (action->type == POPPLER_ACTION_GOTO_DEST) {
 571                 link = ev_link_new_page (title, action->goto_dest.dest->page_num - 1);
 572         } else if (action->type == POPPLER_ACTION_URI) {
 573                 link = ev_link_new_external (title, action->uri.uri);
 574         } else {
 575                 link = ev_link_new_title (title);
 576         }
 577
 578         return link;
 579 }
 580
 581 static void
 582 build_tree (PdfDocument      *pdf_document,
 583             GtkTreeModel     *model,
 584             GtkTreeIter      *parent,
 585             PopplerIndexIter *iter)
 586 {
 587
 588         do {
 589                 GtkTreeIter tree_iter;
 590                 PopplerIndexIter *child;
 591                 PopplerAction *action;
 592                 EvLink *link;
 593
 594                 action = poppler_index_iter_get_action (iter);
 595                 if (action) {
 596                         gtk_tree_store_append (GTK_TREE_STORE (model), &tree_iter, parent);
 597                         link = ev_link_from_action (action);
 598                         poppler_action_free (action);
 599
 600                         gtk_tree_store_set (GTK_TREE_STORE (model), &tree_iter,
 601                                             EV_DOCUMENT_LINKS_COLUMN_MARKUP, ev_link_get_title (link),
 602                                             EV_DOCUMENT_LINKS_COLUMN_LINK, link,
 603                                             -1);
 604                         child = poppler_index_iter_get_child (iter);
 605                         if (child)
 606                                 build_tree (pdf_document, model, &tree_iter, child);
 607                         poppler_index_iter_free (child);
 608                 }
 609         } while (poppler_index_iter_next (iter));
 610 }
 611
 612
 613 static GtkTreeModel *
 614 pdf_document_links_get_links_model (EvDocumentLinks *document_links)
 615 {
 616         PdfDocument *pdf_document = PDF_DOCUMENT (document_links);
 617         GtkTreeModel *model = NULL;
 618         PopplerIndexIter *iter;
 619
 620         g_return_val_if_fail (PDF_IS_DOCUMENT (document_links), NULL);
 621
 622         iter = poppler_index_iter_new (pdf_document->document);
 623         /* Create the model iff we have items*/
 624         if (iter != NULL) {
 625                 model = (GtkTreeModel *) gtk_tree_store_new (EV_DOCUMENT_LINKS_COLUMN_NUM_COLUMNS,
 626                                                              G_TYPE_STRING,
 627                                                              G_TYPE_POINTER);
 628                 build_tree (pdf_document, model, NULL, iter);
 629                 poppler_index_iter_free (iter);
 630         }
 631
 632
 633         return model;
 634 }
 635
 636 static void
 637 pdf_document_document_links_iface_init (EvDocumentLinksIface *iface)
 638 {
 639         iface->has_document_links = pdf_document_links_has_document_links;
 640         iface->get_links_model = pdf_document_links_get_links_model;
 641 }
 642
 643 static GdkPixbuf *
 644 make_thumbnail_for_size (PdfDocument *pdf_document,
 645                          gint         page,
 646                          gint         size,
 647                          gboolean     border)
 648 {
 649         PopplerPage *poppler_page;
 650         GdkPixbuf *pixbuf;
 651         int width, height;
 652         int x_offset, y_offset;
 653         double scale;
 654         gdouble unscaled_width, unscaled_height;
 655
 656         poppler_page = poppler_document_get_page (pdf_document->document, page);
 657
 658         g_return_val_if_fail (poppler_page != NULL, NULL);
 659
 660         pdf_document_thumbnails_get_dimensions (EV_DOCUMENT_THUMBNAILS (pdf_document), page, size, &width, &height);
 661         poppler_page_get_size (poppler_page, &unscaled_width, &unscaled_height);
 662         scale = width / unscaled_width;
 663
 664         if (border) {
 665                 pixbuf = ev_document_misc_get_thumbnail_frame (width, height, NULL);
 666                 x_offset = 1;
 667                 y_offset = 1;
 668         } else {
 669                 pixbuf = gdk_pixbuf_new (GDK_COLORSPACE_RGB, TRUE, 8,
 670                                          width, height);
 671                 gdk_pixbuf_fill (pixbuf, 0xffffffff);
 672                 x_offset = 0;
 673                 y_offset = 0;
 674         }
 675
 676         poppler_page_render_to_pixbuf (poppler_page, 0, 0,
 677                                        width, height,
 678                                        scale, pixbuf,
 679                                        x_offset, y_offset);
 680
 681         return pixbuf;
 682 }
 683
 684 static GdkPixbuf *
 685 pdf_document_thumbnails_get_thumbnail (EvDocumentThumbnails *document_thumbnails,
 686                                        gint                  page,
 687                                        gint                  size,
 688                                        gboolean              border)
 689 {
 690         PdfDocument *pdf_document;
 691         PopplerPage *poppler_page;
 692         GdkPixbuf *pixbuf;
 693
 694         pdf_document = PDF_DOCUMENT (document_thumbnails);
 695
 696         poppler_page = poppler_document_get_page (pdf_document->document, page);
 697         g_return_val_if_fail (poppler_page != NULL, NULL);
 698
 699         pixbuf = poppler_page_get_thumbnail (poppler_page);
 700         if (pixbuf != NULL) {
 701                 /* The document provides its own thumbnails. */
 702                 if (border) {
 703                         GdkPixbuf *real_pixbuf;
 704
 705                         real_pixbuf = ev_document_misc_get_thumbnail_frame (-1, -1, pixbuf);
 706                         g_object_unref (pixbuf);
 707                         pixbuf = real_pixbuf;
 708                 }
 709         } else {
 710                 /* There is no provided thumbnail.  We need to make one. */
 711                 pixbuf = make_thumbnail_for_size (pdf_document, page, size, border);
 712         }
 713         return pixbuf;
 714 }
 715
 716 static void
 717 pdf_document_thumbnails_get_dimensions (EvDocumentThumbnails *document_thumbnails,
 718                                         gint                  page,
 719                                         gint                  size,
 720                                         gint                 *width,
 721                                         gint                 *height)
 722 {
 723         PdfDocument *pdf_document;
 724         PopplerPage *poppler_page;
 725         gint has_thumb;
 726
 727         pdf_document = PDF_DOCUMENT (document_thumbnails);
 728         poppler_page = poppler_document_get_page (pdf_document->document, page);
 729
 730         g_return_if_fail (width != NULL);
 731         g_return_if_fail (height != NULL);
 732         g_return_if_fail (poppler_page != NULL);
 733
 734         has_thumb = poppler_page_get_thumbnail_size (poppler_page, width, height);
 735
 736         if (!has_thumb) {
 737                 double page_width, page_height;
 738
 739                 poppler_page_get_size (poppler_page, &page_width, &page_height);
 740                 if (page_width > page_height) {
 741                         *width = size;
 742                         *height = (int) (size * page_height / page_width);
 743                 } else {
 744                         *width = (int) (size * page_width / page_height);
 745                         *height = size;
 746                 }
 747         }
 748 }
 749
 750 static void
 751 pdf_document_document_thumbnails_iface_init (EvDocumentThumbnailsIface *iface)
 752 {
 753         iface->get_thumbnail = pdf_document_thumbnails_get_thumbnail;
 754         iface->get_dimensions = pdf_document_thumbnails_get_dimensions;
 755 }
 756
 757
 758 static gboolean
 759 pdf_document_search_idle_callback (void *data)
 760 {
 761         PdfDocumentSearch *search = (PdfDocumentSearch*) data;
 762         PdfDocument *pdf_document = search->document;
 763         int n_pages;
 764         GList *matches;
 765         PopplerPage *page;
 766
 767         page = poppler_document_get_page (search->document->document,
 768                                           search->search_page);
 769
 770         ev_document_doc_mutex_lock ();
 771         matches = poppler_page_find_text (page, search->text);
 772         ev_document_doc_mutex_unlock ();
 773
 774         search->pages[search->search_page] = matches;
 775         ev_document_find_changed (EV_DOCUMENT_FIND (pdf_document),
 776                                   search->search_page);
 777
 778         n_pages = pdf_document_get_n_pages (EV_DOCUMENT (search->document));
 779         search->search_page += 1;
 780         if (search->search_page == n_pages) {
 781                 /* wrap around */
 782                 search->search_page = 0;
 783         }
 784
 785         if (search->search_page != search->start_page) {
 786                 return TRUE;
 787         }
 788
 789         /* We're done. */
 790         search->idle = 0; /* will return FALSE to remove */
 791         return FALSE;
 792 }
 793
 794
 795 static PdfDocumentSearch *
 796 pdf_document_search_new (PdfDocument *pdf_document,
 797                          int          start_page,
 798                          const char  *text)
 799 {
 800         PdfDocumentSearch *search;
 801         int n_pages;
 802         int i;
 803
 804         n_pages = pdf_document_get_n_pages (EV_DOCUMENT (pdf_document));
 805
 806         search = g_new0 (PdfDocumentSearch, 1);
 807
 808         search->text = g_strdup (text);
 809         search->pages = g_new0 (GList *, n_pages);
 810         for (i = 0; i < n_pages; i++) {
 811                 search->pages[i] = NULL;
 812         }
 813
 814         search->document = pdf_document;
 815
 816         /* We add at low priority so the progress bar repaints */
 817         search->idle = g_idle_add_full (G_PRIORITY_LOW,
 818                                         pdf_document_search_idle_callback,
 819                                         search,
 820                                         NULL);
 821
 822         search->start_page = start_page;
 823         search->search_page = start_page;
 824
 825         return search;
 826 }
 827
 828 static void
 829 pdf_document_search_free (PdfDocumentSearch   *search)
 830 {
 831         PdfDocument *pdf_document = search->document;
 832         int n_pages;
 833         int i;
 834
 835         if (search->idle != 0)
 836                 g_source_remove (search->idle);
 837
 838         n_pages = pdf_document_get_n_pages (EV_DOCUMENT (pdf_document));
 839         for (i = 0; i < n_pages; i++) {
 840                 g_list_foreach (search->pages[i], (GFunc) g_free, NULL);
 841                 g_list_free (search->pages[i]);
 842         }
 843
 844         g_free (search->text);
 845 }
 846
 847 static void
 848 pdf_document_find_begin (EvDocumentFind   *document,
 849                          int               page,
 850                          const char       *search_string,
 851                          gboolean          case_sensitive)
 852 {
 853         PdfDocument *pdf_document = PDF_DOCUMENT (document);
 854
 855         /* FIXME handle case_sensitive (right now XPDF
 856          * code is always case insensitive for ASCII
 857          * and case sensitive for all other languaages)
 858          */
 859
 860         if (pdf_document->search &&
 861             strcmp (search_string, pdf_document->search->text) == 0)
 862                 return;
 863
 864         if (pdf_document->search)
 865                 pdf_document_search_free (pdf_document->search);
 866
 867         pdf_document->search = pdf_document_search_new (pdf_document,
 868                                                         page,
 869                                                         search_string);
 870 }
 871
 872 int
 873 pdf_document_find_get_n_results (EvDocumentFind *document_find, int page)
 874 {
 875         PdfDocumentSearch *search = PDF_DOCUMENT (document_find)->search;
 876
 877         if (search) {
 878                 return g_list_length (search->pages[page]);
 879         } else {
 880                 return 0;
 881         }
 882 }
 883
 884 gboolean
 885 pdf_document_find_get_result (EvDocumentFind *document_find,
 886                               int             page,
 887                               int             n_result,
 888                               EvRectangle    *rectangle)
 889 {
 890         PdfDocument *pdf_document = PDF_DOCUMENT (document_find);
 891         PdfDocumentSearch *search = pdf_document->search;
 892         PopplerPage *poppler_page;
 893         PopplerRectangle *r;
 894         double height;
 895
 896         if (search == NULL)
 897                 return FALSE;
 898
 899         r = (PopplerRectangle *) g_list_nth_data (search->pages[page],
 900                                                   n_result);
 901         if (r == NULL)
 902                 return FALSE;
 903
 904         poppler_page = poppler_document_get_page (pdf_document->document, page);
 905         poppler_page_get_size (poppler_page, NULL, &height);
 906         rectangle->x1 = r->x1;
 907         rectangle->y1 = height - r->y2;
 908         rectangle->x2 = r->x2;
 909         rectangle->y2 = height - r->y1;
 910
 911         return TRUE;
 912 }
 913
 914 int
 915 pdf_document_find_page_has_results (EvDocumentFind *document_find,
 916                                     int             page)
 917 {
 918         PdfDocumentSearch *search = PDF_DOCUMENT (document_find)->search;
 919
 920         g_return_val_if_fail (search != NULL, FALSE);
 921
 922         return search->pages[page] != NULL;
 923 }
 924
 925 double
 926 pdf_document_find_get_progress (EvDocumentFind *document_find)
 927 {
 928         PdfDocumentSearch *search;
 929         int n_pages, pages_done;
 930
 931         search = PDF_DOCUMENT (document_find)->search;
 932
 933         if (search == NULL) {
 934                 return 0;
 935         }
 936
 937         n_pages = pdf_document_get_n_pages (EV_DOCUMENT (document_find));
 938         if (search->search_page > search->start_page) {
 939                 pages_done = search->search_page - search->start_page + 1;
 940         } else if (search->search_page == search->start_page) {
 941                 pages_done = n_pages;
 942         } else {
 943                 pages_done = n_pages - search->start_page + search->search_page;
 944         }
 945
 946         return pages_done / (double) n_pages;
 947 }
 948
 949 static void
 950 pdf_document_find_cancel (EvDocumentFind *document)
 951 {
 952         PdfDocument *pdf_document = PDF_DOCUMENT (document);
 953
 954         if (pdf_document->search) {
 955                 pdf_document_search_free (pdf_document->search);
 956                 pdf_document->search = NULL;
 957         }
 958 }
 959
 960 static void
 961 pdf_document_find_iface_init (EvDocumentFindIface *iface)
 962 {
 963         iface->begin = pdf_document_find_begin;
 964         iface->get_n_results = pdf_document_find_get_n_results;
 965         iface->get_result = pdf_document_find_get_result;
 966         iface->page_has_results = pdf_document_find_page_has_results;
 967         iface->get_progress = pdf_document_find_get_progress;
 968         iface->cancel = pdf_document_find_cancel;
 969 }
 970
 971 static void
 972 pdf_document_ps_exporter_begin (EvPSExporter *exporter, const char *filename,
 973                                 int first_page, int last_page)
 974 {
 975         PdfDocument *pdf_document = PDF_DOCUMENT (exporter);
 976
 977         pdf_document->ps_file = poppler_ps_file_new (pdf_document->document, filename,
 978                                                      first_page,
 979                                                      last_page - first_page + 1);
 980 }
 981
 982 static void
 983 pdf_document_ps_exporter_do_page (EvPSExporter *exporter, int page)
 984 {
 985         PdfDocument *pdf_document = PDF_DOCUMENT (exporter);
 986         PopplerPage *poppler_page;
 987
 988         g_return_if_fail (pdf_document->ps_file != NULL);
 989
 990         poppler_page = poppler_document_get_page (pdf_document->document, page);
 991         poppler_page_render_to_ps (poppler_page, pdf_document->ps_file);
 992 }
 993
 994 static void
 995 pdf_document_ps_exporter_end (EvPSExporter *exporter)
 996 {
 997         PdfDocument *pdf_document = PDF_DOCUMENT (exporter);
 998
 999         poppler_ps_file_free (pdf_document->ps_file);
1000         pdf_document->ps_file = NULL;
1001 }
1002
1003 static void
1004 pdf_document_ps_exporter_iface_init (EvPSExporterIface *iface)
1005 {
1006         iface->begin = pdf_document_ps_exporter_begin;
1007         iface->do_page = pdf_document_ps_exporter_do_page;
1008         iface->end = pdf_document_ps_exporter_end;
1009 }
1010
1011 PdfDocument *
1012 pdf_document_new (void)
1013 {
1014         return PDF_DOCUMENT (g_object_new (PDF_TYPE_DOCUMENT, NULL));
1015 }