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 gboolean
 499 pdf_document_fonts_fill_model (EvDocumentFonts *document_fonts,
 500                                GtkTreeModel    *model,
 501                                int              n_pages)
 502 {
 503 #ifdef POPPLER_FONT_INFO
 504         PdfDocument *pdf_document = PDF_DOCUMENT (document_fonts);
 505         PopplerFontInfo *info;
 506         PopplerFontsIter *iter;
 507         gboolean result;
 508
 509         g_return_val_if_fail (PDF_IS_DOCUMENT (document_fonts), FALSE);
 510
 511         info = (PopplerFontInfo *)g_object_get_data (G_OBJECT (model), "font_info");
 512         if (info == NULL) {
 513                 info = poppler_font_info_new (pdf_document->document);
 514                 g_object_set_data_full (G_OBJECT (model), "font_info",
 515                                         (PopplerFontInfo *)info,
 516                                         (GDestroyNotify)poppler_font_info_free);
 517         }
 518
 519         result = poppler_font_info_scan (info, n_pages, &iter);
 520
 521         if (iter) {
 522                 do {
 523                         GtkTreeIter list_iter;
 524                         PopplerIndexIter *child;
 525                         const char *name;
 526
 527                         name = poppler_fonts_iter_get_name (iter);
 528                         gtk_list_store_append (GTK_LIST_STORE (model), &list_iter);
 529                         gtk_list_store_set (GTK_LIST_STORE (model), &list_iter,
 530                                             EV_DOCUMENT_FONTS_COLUMN_NAME, name,
 531                                             -1);
 532                 } while (poppler_fonts_iter_next (iter));
 533                 poppler_fonts_iter_free (iter);
 534         }
 535
 536         return result;
 537 #else
 538         return FALSE;
 539 #endif
 540 }
 541
 542 static void
 543 pdf_document_document_fonts_iface_init (EvDocumentFontsIface *iface)
 544 {
 545         iface->fill_model = pdf_document_fonts_fill_model;
 546 }
 547
 548 static gboolean
 549 pdf_document_links_has_document_links (EvDocumentLinks *document_links)
 550 {
 551         PdfDocument *pdf_document = PDF_DOCUMENT (document_links);
 552         PopplerIndexIter *iter;
 553
 554         g_return_val_if_fail (PDF_IS_DOCUMENT (document_links), FALSE);
 555
 556         iter = poppler_index_iter_new (pdf_document->document);
 557         if (iter == NULL)
 558                 return FALSE;
 559         poppler_index_iter_free (iter);
 560
 561         return TRUE;
 562 }
 563
 564 static EvLink *
 565 ev_link_from_action (PopplerAction *action)
 566 {
 567         EvLink *link;
 568         const char *title;
 569
 570         title = action->any.title;
 571
 572         if (action->type == POPPLER_ACTION_GOTO_DEST) {
 573                 link = ev_link_new_page (title, action->goto_dest.dest->page_num - 1);
 574         } else if (action->type == POPPLER_ACTION_URI) {
 575                 link = ev_link_new_external (title, action->uri.uri);
 576         } else {
 577                 link = ev_link_new_title (title);
 578         }
 579
 580         return link;
 581 }
 582
 583 static void
 584 build_tree (PdfDocument      *pdf_document,
 585             GtkTreeModel     *model,
 586             GtkTreeIter      *parent,
 587             PopplerIndexIter *iter)
 588 {
 589
 590         do {
 591                 GtkTreeIter tree_iter;
 592                 PopplerIndexIter *child;
 593                 PopplerAction *action;
 594                 EvLink *link;
 595
 596                 action = poppler_index_iter_get_action (iter);
 597                 if (action) {
 598                         gtk_tree_store_append (GTK_TREE_STORE (model), &tree_iter, parent);
 599                         link = ev_link_from_action (action);
 600                         poppler_action_free (action);
 601
 602                         gtk_tree_store_set (GTK_TREE_STORE (model), &tree_iter,
 603                                             EV_DOCUMENT_LINKS_COLUMN_MARKUP, ev_link_get_title (link),
 604                                             EV_DOCUMENT_LINKS_COLUMN_LINK, link,
 605                                             -1);
 606                         child = poppler_index_iter_get_child (iter);
 607                         if (child)
 608                                 build_tree (pdf_document, model, &tree_iter, child);
 609                         poppler_index_iter_free (child);
 610                 }
 611         } while (poppler_index_iter_next (iter));
 612 }
 613
 614
 615 static GtkTreeModel *
 616 pdf_document_links_get_links_model (EvDocumentLinks *document_links)
 617 {
 618         PdfDocument *pdf_document = PDF_DOCUMENT (document_links);
 619         GtkTreeModel *model = NULL;
 620         PopplerIndexIter *iter;
 621
 622         g_return_val_if_fail (PDF_IS_DOCUMENT (document_links), NULL);
 623
 624         iter = poppler_index_iter_new (pdf_document->document);
 625         /* Create the model iff we have items*/
 626         if (iter != NULL) {
 627                 model = (GtkTreeModel *) gtk_tree_store_new (EV_DOCUMENT_LINKS_COLUMN_NUM_COLUMNS,
 628                                                              G_TYPE_STRING,
 629                                                              G_TYPE_POINTER);
 630                 build_tree (pdf_document, model, NULL, iter);
 631                 poppler_index_iter_free (iter);
 632         }
 633
 634
 635         return model;
 636 }
 637
 638 static void
 639 pdf_document_document_links_iface_init (EvDocumentLinksIface *iface)
 640 {
 641         iface->has_document_links = pdf_document_links_has_document_links;
 642         iface->get_links_model = pdf_document_links_get_links_model;
 643 }
 644
 645 static GdkPixbuf *
 646 make_thumbnail_for_size (PdfDocument *pdf_document,
 647                          gint         page,
 648                          gint         size,
 649                          gboolean     border)
 650 {
 651         PopplerPage *poppler_page;
 652         GdkPixbuf *pixbuf;
 653         int width, height;
 654         int x_offset, y_offset;
 655         double scale;
 656         gdouble unscaled_width, unscaled_height;
 657
 658         poppler_page = poppler_document_get_page (pdf_document->document, page);
 659
 660         g_return_val_if_fail (poppler_page != NULL, NULL);
 661
 662         pdf_document_thumbnails_get_dimensions (EV_DOCUMENT_THUMBNAILS (pdf_document), page, size, &width, &height);
 663         poppler_page_get_size (poppler_page, &unscaled_width, &unscaled_height);
 664         scale = width / unscaled_width;
 665
 666         if (border) {
 667                 pixbuf = ev_document_misc_get_thumbnail_frame (width, height, NULL);
 668                 x_offset = 1;
 669                 y_offset = 1;
 670         } else {
 671                 pixbuf = gdk_pixbuf_new (GDK_COLORSPACE_RGB, TRUE, 8,
 672                                          width, height);
 673                 gdk_pixbuf_fill (pixbuf, 0xffffffff);
 674                 x_offset = 0;
 675                 y_offset = 0;
 676         }
 677
 678         poppler_page_render_to_pixbuf (poppler_page, 0, 0,
 679                                        width, height,
 680                                        scale, pixbuf,
 681                                        x_offset, y_offset);
 682
 683         return pixbuf;
 684 }
 685
 686 static GdkPixbuf *
 687 pdf_document_thumbnails_get_thumbnail (EvDocumentThumbnails *document_thumbnails,
 688                                        gint                  page,
 689                                        gint                  size,
 690                                        gboolean              border)
 691 {
 692         PdfDocument *pdf_document;
 693         PopplerPage *poppler_page;
 694         GdkPixbuf *pixbuf;
 695
 696         pdf_document = PDF_DOCUMENT (document_thumbnails);
 697
 698         poppler_page = poppler_document_get_page (pdf_document->document, page);
 699         g_return_val_if_fail (poppler_page != NULL, NULL);
 700
 701         pixbuf = poppler_page_get_thumbnail (poppler_page);
 702         if (pixbuf != NULL) {
 703                 /* The document provides its own thumbnails. */
 704                 if (border) {
 705                         GdkPixbuf *real_pixbuf;
 706
 707                         real_pixbuf = ev_document_misc_get_thumbnail_frame (-1, -1, pixbuf);
 708                         g_object_unref (pixbuf);
 709                         pixbuf = real_pixbuf;
 710                 }
 711         } else {
 712                 /* There is no provided thumbnail.  We need to make one. */
 713                 pixbuf = make_thumbnail_for_size (pdf_document, page, size, border);
 714         }
 715         return pixbuf;
 716 }
 717
 718 static void
 719 pdf_document_thumbnails_get_dimensions (EvDocumentThumbnails *document_thumbnails,
 720                                         gint                  page,
 721                                         gint                  size,
 722                                         gint                 *width,
 723                                         gint                 *height)
 724 {
 725         PdfDocument *pdf_document;
 726         PopplerPage *poppler_page;
 727         gint has_thumb;
 728
 729         pdf_document = PDF_DOCUMENT (document_thumbnails);
 730         poppler_page = poppler_document_get_page (pdf_document->document, page);
 731
 732         g_return_if_fail (width != NULL);
 733         g_return_if_fail (height != NULL);
 734         g_return_if_fail (poppler_page != NULL);
 735
 736         has_thumb = poppler_page_get_thumbnail_size (poppler_page, width, height);
 737
 738         if (!has_thumb) {
 739                 double page_width, page_height;
 740
 741                 poppler_page_get_size (poppler_page, &page_width, &page_height);
 742                 if (page_width > page_height) {
 743                         *width = size;
 744                         *height = (int) (size * page_height / page_width);
 745                 } else {
 746                         *width = (int) (size * page_width / page_height);
 747                         *height = size;
 748                 }
 749         }
 750 }
 751
 752 static void
 753 pdf_document_document_thumbnails_iface_init (EvDocumentThumbnailsIface *iface)
 754 {
 755         iface->get_thumbnail = pdf_document_thumbnails_get_thumbnail;
 756         iface->get_dimensions = pdf_document_thumbnails_get_dimensions;
 757 }
 758
 759
 760 static gboolean
 761 pdf_document_search_idle_callback (void *data)
 762 {
 763         PdfDocumentSearch *search = (PdfDocumentSearch*) data;
 764         PdfDocument *pdf_document = search->document;
 765         int n_pages;
 766         GList *matches;
 767         PopplerPage *page;
 768
 769         page = poppler_document_get_page (search->document->document,
 770                                           search->search_page);
 771
 772         ev_document_doc_mutex_lock ();
 773         matches = poppler_page_find_text (page, search->text);
 774         ev_document_doc_mutex_unlock ();
 775
 776         search->pages[search->search_page] = matches;
 777         ev_document_find_changed (EV_DOCUMENT_FIND (pdf_document),
 778                                   search->search_page);
 779
 780         n_pages = pdf_document_get_n_pages (EV_DOCUMENT (search->document));
 781         search->search_page += 1;
 782         if (search->search_page == n_pages) {
 783                 /* wrap around */
 784                 search->search_page = 0;
 785         }
 786
 787         if (search->search_page != search->start_page) {
 788                 return TRUE;
 789         }
 790
 791         /* We're done. */
 792         search->idle = 0; /* will return FALSE to remove */
 793         return FALSE;
 794 }
 795
 796
 797 static PdfDocumentSearch *
 798 pdf_document_search_new (PdfDocument *pdf_document,
 799                          int          start_page,
 800                          const char  *text)
 801 {
 802         PdfDocumentSearch *search;
 803         int n_pages;
 804         int i;
 805
 806         n_pages = pdf_document_get_n_pages (EV_DOCUMENT (pdf_document));
 807
 808         search = g_new0 (PdfDocumentSearch, 1);
 809
 810         search->text = g_strdup (text);
 811         search->pages = g_new0 (GList *, n_pages);
 812         for (i = 0; i < n_pages; i++) {
 813                 search->pages[i] = NULL;
 814         }
 815
 816         search->document = pdf_document;
 817
 818         /* We add at low priority so the progress bar repaints */
 819         search->idle = g_idle_add_full (G_PRIORITY_LOW,
 820                                         pdf_document_search_idle_callback,
 821                                         search,
 822                                         NULL);
 823
 824         search->start_page = start_page;
 825         search->search_page = start_page;
 826
 827         return search;
 828 }
 829
 830 static void
 831 pdf_document_search_free (PdfDocumentSearch   *search)
 832 {
 833         PdfDocument *pdf_document = search->document;
 834         int n_pages;
 835         int i;
 836
 837         if (search->idle != 0)
 838                 g_source_remove (search->idle);
 839
 840         n_pages = pdf_document_get_n_pages (EV_DOCUMENT (pdf_document));
 841         for (i = 0; i < n_pages; i++) {
 842                 g_list_foreach (search->pages[i], (GFunc) g_free, NULL);
 843                 g_list_free (search->pages[i]);
 844         }
 845
 846         g_free (search->text);
 847 }
 848
 849 static void
 850 pdf_document_find_begin (EvDocumentFind   *document,
 851                          int               page,
 852                          const char       *search_string,
 853                          gboolean          case_sensitive)
 854 {
 855         PdfDocument *pdf_document = PDF_DOCUMENT (document);
 856
 857         /* FIXME handle case_sensitive (right now XPDF
 858          * code is always case insensitive for ASCII
 859          * and case sensitive for all other languaages)
 860          */
 861
 862         if (pdf_document->search &&
 863             strcmp (search_string, pdf_document->search->text) == 0)
 864                 return;
 865
 866         if (pdf_document->search)
 867                 pdf_document_search_free (pdf_document->search);
 868
 869         pdf_document->search = pdf_document_search_new (pdf_document,
 870                                                         page,
 871                                                         search_string);
 872 }
 873
 874 int
 875 pdf_document_find_get_n_results (EvDocumentFind *document_find, int page)
 876 {
 877         PdfDocumentSearch *search = PDF_DOCUMENT (document_find)->search;
 878
 879         if (search) {
 880                 return g_list_length (search->pages[page]);
 881         } else {
 882                 return 0;
 883         }
 884 }
 885
 886 gboolean
 887 pdf_document_find_get_result (EvDocumentFind *document_find,
 888                               int             page,
 889                               int             n_result,
 890                               EvRectangle    *rectangle)
 891 {
 892         PdfDocument *pdf_document = PDF_DOCUMENT (document_find);
 893         PdfDocumentSearch *search = pdf_document->search;
 894         PopplerPage *poppler_page;
 895         PopplerRectangle *r;
 896         double height;
 897
 898         if (search == NULL)
 899                 return FALSE;
 900
 901         r = (PopplerRectangle *) g_list_nth_data (search->pages[page],
 902                                                   n_result);
 903         if (r == NULL)
 904                 return FALSE;
 905
 906         poppler_page = poppler_document_get_page (pdf_document->document, page);
 907         poppler_page_get_size (poppler_page, NULL, &height);
 908         rectangle->x1 = r->x1;
 909         rectangle->y1 = height - r->y2;
 910         rectangle->x2 = r->x2;
 911         rectangle->y2 = height - r->y1;
 912
 913         return TRUE;
 914 }
 915
 916 int
 917 pdf_document_find_page_has_results (EvDocumentFind *document_find,
 918                                     int             page)
 919 {
 920         PdfDocumentSearch *search = PDF_DOCUMENT (document_find)->search;
 921
 922         g_return_val_if_fail (search != NULL, FALSE);
 923
 924         return search->pages[page] != NULL;
 925 }
 926
 927 double
 928 pdf_document_find_get_progress (EvDocumentFind *document_find)
 929 {
 930         PdfDocumentSearch *search;
 931         int n_pages, pages_done;
 932
 933         search = PDF_DOCUMENT (document_find)->search;
 934
 935         if (search == NULL) {
 936                 return 0;
 937         }
 938
 939         n_pages = pdf_document_get_n_pages (EV_DOCUMENT (document_find));
 940         if (search->search_page > search->start_page) {
 941                 pages_done = search->search_page - search->start_page + 1;
 942         } else if (search->search_page == search->start_page) {
 943                 pages_done = n_pages;
 944         } else {
 945                 pages_done = n_pages - search->start_page + search->search_page;
 946         }
 947
 948         return pages_done / (double) n_pages;
 949 }
 950
 951 static void
 952 pdf_document_find_cancel (EvDocumentFind *document)
 953 {
 954         PdfDocument *pdf_document = PDF_DOCUMENT (document);
 955
 956         if (pdf_document->search) {
 957                 pdf_document_search_free (pdf_document->search);
 958                 pdf_document->search = NULL;
 959         }
 960 }
 961
 962 static void
 963 pdf_document_find_iface_init (EvDocumentFindIface *iface)
 964 {
 965         iface->begin = pdf_document_find_begin;
 966         iface->get_n_results = pdf_document_find_get_n_results;
 967         iface->get_result = pdf_document_find_get_result;
 968         iface->page_has_results = pdf_document_find_page_has_results;
 969         iface->get_progress = pdf_document_find_get_progress;
 970         iface->cancel = pdf_document_find_cancel;
 971 }
 972
 973 static void
 974 pdf_document_ps_exporter_begin (EvPSExporter *exporter, const char *filename,
 975                                 int first_page, int last_page)
 976 {
 977         PdfDocument *pdf_document = PDF_DOCUMENT (exporter);
 978
 979         pdf_document->ps_file = poppler_ps_file_new (pdf_document->document, filename,
 980                                                      first_page,
 981                                                      last_page - first_page + 1);
 982 }
 983
 984 static void
 985 pdf_document_ps_exporter_do_page (EvPSExporter *exporter, int page)
 986 {
 987         PdfDocument *pdf_document = PDF_DOCUMENT (exporter);
 988         PopplerPage *poppler_page;
 989
 990         g_return_if_fail (pdf_document->ps_file != NULL);
 991
 992         poppler_page = poppler_document_get_page (pdf_document->document, page);
 993         poppler_page_render_to_ps (poppler_page, pdf_document->ps_file);
 994 }
 995
 996 static void
 997 pdf_document_ps_exporter_end (EvPSExporter *exporter)
 998 {
 999         PdfDocument *pdf_document = PDF_DOCUMENT (exporter);
1000
1001         poppler_ps_file_free (pdf_document->ps_file);
1002         pdf_document->ps_file = NULL;
1003 }
1004
1005 static void
1006 pdf_document_ps_exporter_iface_init (EvPSExporterIface *iface)
1007 {
1008         iface->begin = pdf_document_ps_exporter_begin;
1009         iface->do_page = pdf_document_ps_exporter_do_page;
1010         iface->end = pdf_document_ps_exporter_end;
1011 }
1012
1013 PdfDocument *
1014 pdf_document_new (void)
1015 {
1016         return PDF_DOCUMENT (g_object_new (PDF_TYPE_DOCUMENT, NULL));
1017 }