pdf/ev-poppler.cc

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