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