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