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