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                 expand = poppler_index_iter_is_open (iter);
 665                 if (action) {
 666                         gtk_tree_store_append (GTK_TREE_STORE (model), &tree_iter, parent);
 667                         link = ev_link_from_action (action);
 668                         poppler_action_free (action);
 669
 670                         gtk_tree_store_set (GTK_TREE_STORE (model), &tree_iter,
 671                                             EV_DOCUMENT_LINKS_COLUMN_MARKUP, ev_link_get_title (link),
 672                                             EV_DOCUMENT_LINKS_COLUMN_LINK, link,
 673                                             EV_DOCUMENT_LINKS_COLUMN_EXPAND, expand,
 674                                             -1);
 675                         child = poppler_index_iter_get_child (iter);
 676                         if (child)
 677                                 build_tree (pdf_document, model, &tree_iter, child);
 678                         poppler_index_iter_free (child);
 679                 }
 680         } while (poppler_index_iter_next (iter));
 681 }
 682
 683
 684 static GtkTreeModel *
 685 pdf_document_links_get_links_model (EvDocumentLinks *document_links)
 686 {
 687         PdfDocument *pdf_document = PDF_DOCUMENT (document_links);
 688         GtkTreeModel *model = NULL;
 689         PopplerIndexIter *iter;
 690
 691         g_return_val_if_fail (PDF_IS_DOCUMENT (document_links), NULL);
 692
 693         iter = poppler_index_iter_new (pdf_document->document);
 694         /* Create the model if we have items*/
 695         if (iter != NULL) {
 696                 model = (GtkTreeModel *) gtk_tree_store_new (EV_DOCUMENT_LINKS_COLUMN_NUM_COLUMNS,
 697                                                              G_TYPE_STRING,
 698                                                              G_TYPE_POINTER,
 699                                                              G_TYPE_BOOLEAN);
 700                 build_tree (pdf_document, model, NULL, iter);
 701                 poppler_index_iter_free (iter);
 702         }
 703
 704         return model;
 705 }
 706
 707 static void
 708 pdf_document_document_links_iface_init (EvDocumentLinksIface *iface)
 709 {
 710         iface->has_document_links = pdf_document_links_has_document_links;
 711         iface->get_links_model = pdf_document_links_get_links_model;
 712 }
 713
 714 static GdkPixbuf *
 715 make_thumbnail_for_size (PdfDocument *pdf_document,
 716                          gint         page,
 717                          gint         size,
 718                          gboolean     border)
 719 {
 720         PopplerPage *poppler_page;
 721         GdkPixbuf *pixbuf;
 722         int width, height;
 723         int x_offset, y_offset;
 724         double scale;
 725         gdouble unscaled_width, unscaled_height;
 726
 727         poppler_page = poppler_document_get_page (pdf_document->document, page);
 728
 729         g_return_val_if_fail (poppler_page != NULL, NULL);
 730
 731         pdf_document_thumbnails_get_dimensions (EV_DOCUMENT_THUMBNAILS (pdf_document), page, size, &width, &height);
 732         poppler_page_get_size (poppler_page, &unscaled_width, &unscaled_height);
 733         scale = width / unscaled_width;
 734
 735         if (border) {
 736                 pixbuf = ev_document_misc_get_thumbnail_frame (width, height, NULL);
 737                 x_offset = 1;
 738                 y_offset = 1;
 739         } else {
 740                 pixbuf = gdk_pixbuf_new (GDK_COLORSPACE_RGB, TRUE, 8,
 741                                          width, height);
 742                 gdk_pixbuf_fill (pixbuf, 0xffffffff);
 743                 x_offset = 0;
 744                 y_offset = 0;
 745         }
 746
 747         poppler_page_render_to_pixbuf (poppler_page, 0, 0,
 748                                        width, height,
 749                                        scale, pixbuf,
 750                                        x_offset, y_offset);
 751
 752         return pixbuf;
 753 }
 754
 755 static GdkPixbuf *
 756 pdf_document_thumbnails_get_thumbnail (EvDocumentThumbnails *document_thumbnails,
 757                                        gint                  page,
 758                                        gint                  size,
 759                                        gboolean              border)
 760 {
 761         PdfDocument *pdf_document;
 762         PopplerPage *poppler_page;
 763         GdkPixbuf *pixbuf;
 764
 765         pdf_document = PDF_DOCUMENT (document_thumbnails);
 766
 767         poppler_page = poppler_document_get_page (pdf_document->document, page);
 768         g_return_val_if_fail (poppler_page != NULL, NULL);
 769
 770         pixbuf = poppler_page_get_thumbnail (poppler_page);
 771         if (pixbuf != NULL) {
 772                 /* The document provides its own thumbnails. */
 773                 if (border) {
 774                         GdkPixbuf *real_pixbuf;
 775
 776                         real_pixbuf = ev_document_misc_get_thumbnail_frame (-1, -1, pixbuf);
 777                         g_object_unref (pixbuf);
 778                         pixbuf = real_pixbuf;
 779                 }
 780         } else {
 781                 /* There is no provided thumbnail.  We need to make one. */
 782                 pixbuf = make_thumbnail_for_size (pdf_document, page, size, border);
 783         }
 784         return pixbuf;
 785 }
 786
 787 static void
 788 pdf_document_thumbnails_get_dimensions (EvDocumentThumbnails *document_thumbnails,
 789                                         gint                  page,
 790                                         gint                  size,
 791                                         gint                 *width,
 792                                         gint                 *height)
 793 {
 794         PdfDocument *pdf_document;
 795         PopplerPage *poppler_page;
 796         gint has_thumb;
 797
 798         pdf_document = PDF_DOCUMENT (document_thumbnails);
 799         poppler_page = poppler_document_get_page (pdf_document->document, page);
 800
 801         g_return_if_fail (width != NULL);
 802         g_return_if_fail (height != NULL);
 803         g_return_if_fail (poppler_page != NULL);
 804
 805         has_thumb = poppler_page_get_thumbnail_size (poppler_page, width, height);
 806
 807         if (!has_thumb) {
 808                 double page_width, page_height;
 809
 810                 poppler_page_get_size (poppler_page, &page_width, &page_height);
 811                 if (page_width > page_height) {
 812                         *width = size;
 813                         *height = (int) (size * page_height / page_width);
 814                 } else {
 815                         *width = (int) (size * page_width / page_height);
 816                         *height = size;
 817                 }
 818         }
 819 }
 820
 821 static void
 822 pdf_document_document_thumbnails_iface_init (EvDocumentThumbnailsIface *iface)
 823 {
 824         iface->get_thumbnail = pdf_document_thumbnails_get_thumbnail;
 825         iface->get_dimensions = pdf_document_thumbnails_get_dimensions;
 826 }
 827
 828
 829 static gboolean
 830 pdf_document_search_idle_callback (void *data)
 831 {
 832         PdfDocumentSearch *search = (PdfDocumentSearch*) data;
 833         PdfDocument *pdf_document = search->document;
 834         int n_pages;
 835         GList *matches;
 836         PopplerPage *page;
 837
 838         page = poppler_document_get_page (search->document->document,
 839                                           search->search_page);
 840
 841         ev_document_doc_mutex_lock ();
 842         matches = poppler_page_find_text (page, search->text);
 843         ev_document_doc_mutex_unlock ();
 844
 845         search->pages[search->search_page] = matches;
 846         ev_document_find_changed (EV_DOCUMENT_FIND (pdf_document),
 847                                   search->search_page);
 848
 849         n_pages = pdf_document_get_n_pages (EV_DOCUMENT (search->document));
 850         search->search_page += 1;
 851         if (search->search_page == n_pages) {
 852                 /* wrap around */
 853                 search->search_page = 0;
 854         }
 855
 856         if (search->search_page != search->start_page) {
 857                 return TRUE;
 858         }
 859
 860         /* We're done. */
 861         search->idle = 0; /* will return FALSE to remove */
 862         return FALSE;
 863 }
 864
 865
 866 static PdfDocumentSearch *
 867 pdf_document_search_new (PdfDocument *pdf_document,
 868                          int          start_page,
 869                          const char  *text)
 870 {
 871         PdfDocumentSearch *search;
 872         int n_pages;
 873         int i;
 874
 875         n_pages = pdf_document_get_n_pages (EV_DOCUMENT (pdf_document));
 876
 877         search = g_new0 (PdfDocumentSearch, 1);
 878
 879         search->text = g_strdup (text);
 880         search->pages = g_new0 (GList *, n_pages);
 881         for (i = 0; i < n_pages; i++) {
 882                 search->pages[i] = NULL;
 883         }
 884
 885         search->document = pdf_document;
 886
 887         /* We add at low priority so the progress bar repaints */
 888         search->idle = g_idle_add_full (G_PRIORITY_LOW,
 889                                         pdf_document_search_idle_callback,
 890                                         search,
 891                                         NULL);
 892
 893         search->start_page = start_page;
 894         search->search_page = start_page;
 895
 896         return search;
 897 }
 898
 899 static void
 900 pdf_document_search_free (PdfDocumentSearch   *search)
 901 {
 902         PdfDocument *pdf_document = search->document;
 903         int n_pages;
 904         int i;
 905
 906         if (search->idle != 0)
 907                 g_source_remove (search->idle);
 908
 909         n_pages = pdf_document_get_n_pages (EV_DOCUMENT (pdf_document));
 910         for (i = 0; i < n_pages; i++) {
 911                 g_list_foreach (search->pages[i], (GFunc) g_free, NULL);
 912                 g_list_free (search->pages[i]);
 913         }
 914
 915         g_free (search->text);
 916 }
 917
 918 static void
 919 pdf_document_find_begin (EvDocumentFind   *document,
 920                          int               page,
 921                          const char       *search_string,
 922                          gboolean          case_sensitive)
 923 {
 924         PdfDocument *pdf_document = PDF_DOCUMENT (document);
 925
 926         /* FIXME handle case_sensitive (right now XPDF
 927          * code is always case insensitive for ASCII
 928          * and case sensitive for all other languaages)
 929          */
 930
 931         if (pdf_document->search &&
 932             strcmp (search_string, pdf_document->search->text) == 0)
 933                 return;
 934
 935         if (pdf_document->search)
 936                 pdf_document_search_free (pdf_document->search);
 937
 938         pdf_document->search = pdf_document_search_new (pdf_document,
 939                                                         page,
 940                                                         search_string);
 941 }
 942
 943 int
 944 pdf_document_find_get_n_results (EvDocumentFind *document_find, int page)
 945 {
 946         PdfDocumentSearch *search = PDF_DOCUMENT (document_find)->search;
 947
 948         if (search) {
 949                 return g_list_length (search->pages[page]);
 950         } else {
 951                 return 0;
 952         }
 953 }
 954
 955 gboolean
 956 pdf_document_find_get_result (EvDocumentFind *document_find,
 957                               int             page,
 958                               int             n_result,
 959                               EvRectangle    *rectangle)
 960 {
 961         PdfDocument *pdf_document = PDF_DOCUMENT (document_find);
 962         PdfDocumentSearch *search = pdf_document->search;
 963         PopplerPage *poppler_page;
 964         PopplerRectangle *r;
 965         double height;
 966
 967         if (search == NULL)
 968                 return FALSE;
 969
 970         r = (PopplerRectangle *) g_list_nth_data (search->pages[page],
 971                                                   n_result);
 972         if (r == NULL)
 973                 return FALSE;
 974
 975         poppler_page = poppler_document_get_page (pdf_document->document, page);
 976         poppler_page_get_size (poppler_page, NULL, &height);
 977         rectangle->x1 = r->x1;
 978         rectangle->y1 = height - r->y2;
 979         rectangle->x2 = r->x2;
 980         rectangle->y2 = height - r->y1;
 981
 982         return TRUE;
 983 }
 984
 985 int
 986 pdf_document_find_page_has_results (EvDocumentFind *document_find,
 987                                     int             page)
 988 {
 989         PdfDocumentSearch *search = PDF_DOCUMENT (document_find)->search;
 990
 991         g_return_val_if_fail (search != NULL, FALSE);
 992
 993         return search->pages[page] != NULL;
 994 }
 995
 996 double
 997 pdf_document_find_get_progress (EvDocumentFind *document_find)
 998 {
 999         PdfDocumentSearch *search;
1000         int n_pages, pages_done;
1001
1002         search = PDF_DOCUMENT (document_find)->search;
1003
1004         if (search == NULL) {
1005                 return 0;
1006         }
1007
1008         n_pages = pdf_document_get_n_pages (EV_DOCUMENT (document_find));
1009         if (search->search_page > search->start_page) {
1010                 pages_done = search->search_page - search->start_page + 1;
1011         } else if (search->search_page == search->start_page) {
1012                 pages_done = n_pages;
1013         } else {
1014                 pages_done = n_pages - search->start_page + search->search_page;
1015         }
1016
1017         return pages_done / (double) n_pages;
1018 }
1019
1020 static void
1021 pdf_document_find_cancel (EvDocumentFind *document)
1022 {
1023         PdfDocument *pdf_document = PDF_DOCUMENT (document);
1024
1025         if (pdf_document->search) {
1026                 pdf_document_search_free (pdf_document->search);
1027                 pdf_document->search = NULL;
1028         }
1029 }
1030
1031 static void
1032 pdf_document_find_iface_init (EvDocumentFindIface *iface)
1033 {
1034         iface->begin = pdf_document_find_begin;
1035         iface->get_n_results = pdf_document_find_get_n_results;
1036         iface->get_result = pdf_document_find_get_result;
1037         iface->page_has_results = pdf_document_find_page_has_results;
1038         iface->get_progress = pdf_document_find_get_progress;
1039         iface->cancel = pdf_document_find_cancel;
1040 }
1041
1042 static void
1043 pdf_document_ps_exporter_begin (EvPSExporter *exporter, const char *filename,
1044                                 int first_page, int last_page)
1045 {
1046         PdfDocument *pdf_document = PDF_DOCUMENT (exporter);
1047
1048         pdf_document->ps_file = poppler_ps_file_new (pdf_document->document, filename,
1049                                                      first_page,
1050                                                      last_page - first_page + 1);
1051 }
1052
1053 static void
1054 pdf_document_ps_exporter_do_page (EvPSExporter *exporter, int page)
1055 {
1056         PdfDocument *pdf_document = PDF_DOCUMENT (exporter);
1057         PopplerPage *poppler_page;
1058
1059         g_return_if_fail (pdf_document->ps_file != NULL);
1060
1061         poppler_page = poppler_document_get_page (pdf_document->document, page);
1062         poppler_page_render_to_ps (poppler_page, pdf_document->ps_file);
1063 }
1064
1065 static void
1066 pdf_document_ps_exporter_end (EvPSExporter *exporter)
1067 {
1068         PdfDocument *pdf_document = PDF_DOCUMENT (exporter);
1069
1070         poppler_ps_file_free (pdf_document->ps_file);
1071         pdf_document->ps_file = NULL;
1072 }
1073
1074 static void
1075 pdf_document_ps_exporter_iface_init (EvPSExporterIface *iface)
1076 {
1077         iface->begin = pdf_document_ps_exporter_begin;
1078         iface->do_page = pdf_document_ps_exporter_do_page;
1079         iface->end = pdf_document_ps_exporter_end;
1080 }
1081
1082 PdfDocument *
1083 pdf_document_new (void)
1084 {
1085         return PDF_DOCUMENT (g_object_new (PDF_TYPE_DOCUMENT, NULL));
1086 }