bool Pdf::open(const QString& path)
{
char* name = path.toAscii().data();
fz_try(ctx) {
xref = pdf_open_xref(ctx, name, 0);
pdf_load_page_tree(xref);
}
fz_catch(ctx) {
pdf_free_xref(xref);
xref = 0;
}
return true;
}
fz_outline *
pdf_load_outline(fz_context *ctx, pdf_document *doc)
{
pdf_obj *root, *obj, *first;
fz_outline *outline = NULL;
pdf_load_page_tree(ctx, doc); /* cache page tree for fast link destination lookups */
fz_try(ctx)
{
root = pdf_dict_get(ctx, pdf_trailer(ctx, doc), PDF_NAME_Root);
obj = pdf_dict_get(ctx, root, PDF_NAME_Outlines);
first = pdf_dict_get(ctx, obj, PDF_NAME_First);
if (first)
outline = pdf_load_outline_imp(ctx, doc, first);
}
fz_always(ctx)
pdf_drop_page_tree(ctx, doc);
fz_catch(ctx)
fz_rethrow(ctx);
return outline;
}
static int _pdf_doc_load(struct _pdf_doc *self, mume_stream_t *stm)
{
fz_error error;
fz_stream *fzstm;
fz_rect *mbox;
fz_obj *page_obj, *box_obj;
int i, c;
_pdf_doc_clear(self);
fzstm = fz_new_stream(stm, _pdf_stream_read, _pdf_stream_close);
mume_stream_reference(stm);
fzstm->seek = _pdf_stream_seek;
error = pdf_open_xref_with_stream(&self->xref, fzstm, NULL);
fz_close(fzstm);
if (error) {
mume_error(("Read xref failed\n", error));
return 0;
}
assert(!pdf_needs_password(self->xref));
/* Load meta information. */
error = pdf_load_page_tree(self->xref);
if (error) {
mume_error(("Cannot load page tree\n"));
return 0;
}
c = pdf_count_pages(self->xref);
self->glyph_cache = fz_new_glyph_cache();
self->pages = calloc_abort(c, sizeof(pdf_page*));
self->disps = calloc_abort(c, sizeof(fz_display_list*));
self->media_boxes = malloc_abort(c * sizeof(fz_rect));
self->page_rotates = malloc_abort(c * sizeof(int));
/* Extract each pages' media box and rotation. */
for (i = 0; i < c; ++i) {
mbox = self->media_boxes + i;
page_obj = self->xref->page_objs[i];
if (!page_obj) {
*mbox = fz_empty_rect;
continue;
}
box_obj = fz_dict_gets(page_obj, "MediaBox");
*mbox = pdf_to_rect(box_obj);
if (fz_is_empty_rect(*mbox)) {
fz_warn("Cannot find page bounds, guessing page bounds.");
mbox->x1 = 612;
mbox->y1 = 792;
}
box_obj = fz_dict_gets(page_obj, "CropBox");
if (fz_is_array(box_obj))
*mbox = fz_intersect_rect(*mbox, pdf_to_rect(box_obj));
self->page_rotates[i] = fz_to_int(
fz_dict_gets(page_obj, "Rotate"));
if (self->page_rotates[i] % 90)
self->page_rotates[i] = 0;
}
return 1;
}
static void pdfapp_open_pdf(pdfapp_t *app, char *filename, int fd)
{
fz_error error;
fz_stream *file;
char *password = "";
fz_obj *obj;
fz_obj *info;
/*
* Open PDF and load xref table
*/
file = fz_open_fd(fd);
error = pdf_open_xref_with_stream(&app->xref, file, NULL);
if (error)
pdfapp_error(app, fz_rethrow(error, "cannot open document '%s'", filename));
fz_close(file);
/*
* Handle encrypted PDF files
*/
if (pdf_needs_password(app->xref))
{
int okay = pdf_authenticate_password(app->xref, password);
while (!okay)
{
password = winpassword(app, filename);
if (!password)
exit(1);
okay = pdf_authenticate_password(app->xref, password);
if (!okay)
pdfapp_warn(app, "Invalid password.");
}
}
/*
* Load meta information
*/
app->outline = pdf_load_outline(app->xref);
app->doctitle = filename;
if (strrchr(app->doctitle, '\\'))
app->doctitle = strrchr(app->doctitle, '\\') + 1;
if (strrchr(app->doctitle, '/'))
app->doctitle = strrchr(app->doctitle, '/') + 1;
info = fz_dict_gets(app->xref->trailer, "Info");
if (info)
{
obj = fz_dict_gets(info, "Title");
if (obj)
app->doctitle = pdf_to_utf8(obj);
}
/*
* Start at first page
*/
error = pdf_load_page_tree(app->xref);
if (error)
pdfapp_error(app, fz_rethrow(error, "cannot load page tree"));
app->pagecount = pdf_count_pages(app->xref);
}
void pdfapp_open(pdfapp_t *app, char *filename)
{
#if 0
fz_error error;
fz_obj *obj;
char *password = "";
#else
fz_error error;
fz_stream *file;
char *password = "";
fz_obj *obj;
fz_obj *info;
int fd;
fd = open(filename, O_BINARY | O_RDONLY, 0666);
if (fd < 0)
fprintf(stderr, "error, file %s does not exist\n", filename);
#endif
/*
* Open PDF and load xref table
*/
app->filename = filename;
#if 0
app->xref = pdf_newxref();
error = pdf_loadxref(app->xref, filename);
if (error)
{
fz_catch(error, "trying to air");
error = pdf_repairxref(app->xref, filename);
if (error)
pdfapp_error(app, error);
}
error = pdf_decryptxref(app->xref);
if (error)
pdfapp_error(app, error);
#else
file = fz_open_fd(fd);
error = pdf_open_xref_with_stream(&app->xref, file, NULL);
if (error)
pdfapp_error(app, fz_rethrow(error, "cannot open document '%s'", filename));
fz_close(file);
#endif
/*
* Handle encrypted PDF files
*/
if (pdf_needs_password(app->xref))
{
int okay = pdf_authenticate_password(app->xref, password);
while (!okay)
{
//password = winpassword(app, filename);
if (!password)
exit(1);
okay = pdf_authenticate_password(app->xref, password);
if (!okay)
pdfapp_warn(app, "Invalid password.");
}
}
/*
* Load meta information
* TODO: move this into mupdf library
*/
#if 0
obj = fz_dictgets(app->xref->trailer, "Root");
app->xref->root = fz_resolveindirect(obj);
if (!app->xref->root)
pdfapp_error(app, fz_throw("syntaxerror: missing Root object"));
fz_keepobj(app->xref->root);
obj = fz_dictgets(app->xref->trailer, "Info");
app->xref->info = fz_resolveindirect(obj);
if (!app->xref->info)
pdfapp_warn(app, "Could not load PDF meta information.");
if (app->xref->info)
fz_keepobj(app->xref->info);
/*app->outline = pdf_loadoutline(app->xref);*/
app->doctitle = filename;
if (strrchr(app->doctitle, '\\'))
app->doctitle = strrchr(app->doctitle, '\\') + 1;
if (strrchr(app->doctitle, '/'))
app->doctitle = strrchr(app->doctitle, '/') + 1;
if (app->xref->info)
{
obj = fz_dictgets(app->xref->info, "Title");
if (obj)
{
app->doctitle = pdf_toutf8(obj);
}
}
#else
app->outline = pdf_load_outline(app->xref);
app->doctitle = filename;
if (strrchr(app->doctitle, '\\'))
app->doctitle = strrchr(app->doctitle, '\\') + 1;
if (strrchr(app->doctitle, '/'))
app->doctitle = strrchr(app->doctitle, '/') + 1;
info = fz_dict_gets(app->xref->trailer, "Info");
if (info)
{
obj = fz_dict_gets(info, "Title");
if (obj)
app->doctitle = pdf_to_utf8(obj);
}
#endif
/*
* Start at first page
*/
#if 0
app->pagecount = pdf_getpagecount(app->xref);
#else
error = pdf_load_page_tree(app->xref);
if (error)
pdfapp_error(app, fz_rethrow(error, "cannot load page tree"));
app->pagecount = pdf_count_pages(app->xref);
#endif
app->rotate = 0;
}
static void retainpages(int argc, char **argv)
{
fz_error error;
fz_obj *oldroot, *root, *pages, *kids, *countobj, *parent, *olddests;
/* Load the old page tree */
error = pdf_load_page_tree(xref);
if (error)
die(fz_error_note(ctx, error, "cannot load page tree"));
/* Keep only pages/type and (reduced) dest entries to avoid
* references to unretained pages */
oldroot = fz_dict_gets(ctx, xref->trailer, "Root");
pages = fz_dict_gets(ctx, oldroot, "Pages");
olddests = pdf_load_name_tree(xref, "Dests");
root = fz_new_dict(ctx, 2);
fz_dict_puts(ctx, root, "Type", fz_dict_gets(ctx, oldroot, "Type"));
fz_dict_puts(ctx, root, "Pages", fz_dict_gets(ctx, oldroot, "Pages"));
pdf_update_object(xref, fz_to_num(oldroot), fz_to_gen(oldroot), root);
fz_drop_obj(ctx, root);
/* Create a new kids array with only the pages we want to keep */
parent = fz_new_indirect(ctx, fz_to_num(pages), fz_to_gen(pages), xref);
kids = fz_new_array(ctx, 1);
/* Retain pages specified */
while (argc - fz_optind)
{
int page, spage, epage;
char *spec, *dash;
char *pagelist = argv[fz_optind];
spec = fz_strsep(&pagelist, ",");
while (spec)
{
dash = strchr(spec, '-');
if (dash == spec)
spage = epage = pdf_count_pages(xref);
else
spage = epage = atoi(spec);
if (dash)
{
if (strlen(dash) > 1)
epage = atoi(dash + 1);
else
epage = pdf_count_pages(xref);
}
if (spage > epage)
page = spage, spage = epage, epage = page;
if (spage < 1)
spage = 1;
if (epage > pdf_count_pages(xref))
epage = pdf_count_pages(xref);
for (page = spage; page <= epage; page++)
{
fz_obj *pageobj = xref->page_objs[page-1];
fz_obj *pageref = xref->page_refs[page-1];
fz_dict_puts(ctx, pageobj, "Parent", parent);
/* Store page object in new kids array */
fz_array_push(ctx, kids, pageref);
}
spec = fz_strsep(&pagelist, ",");
}
fz_optind++;
}
fz_drop_obj(ctx, parent);
/* Update page count and kids array */
countobj = fz_new_int(ctx, fz_array_len(ctx, kids));
fz_dict_puts(ctx, pages, "Count", countobj);
fz_drop_obj(ctx, countobj);
fz_dict_puts(ctx, pages, "Kids", kids);
fz_drop_obj(ctx, kids);
/* Also preserve the (partial) Dests name tree */
if (olddests)
{
int i;
fz_obj *names = fz_new_dict(ctx, 1);
fz_obj *dests = fz_new_dict(ctx, 1);
fz_obj *names_list = fz_new_array(ctx, 32);
for (i = 0; i < fz_dict_len(ctx, olddests); i++)
{
fz_obj *key = fz_dict_get_key(ctx, olddests, i);
fz_obj *val = fz_dict_get_val(ctx, olddests, i);
fz_obj *key_str = fz_new_string(ctx, fz_to_name(ctx, key), strlen(fz_to_name(ctx, key)));
fz_obj *dest = fz_dict_gets(ctx, val, "D");
dest = fz_array_get(ctx, dest ? dest : val, 0);
if (fz_array_contains(ctx, fz_dict_gets(ctx, pages, "Kids"), dest))
{
fz_array_push(ctx, names_list, key_str);
fz_array_push(ctx, names_list, val);
}
fz_drop_obj(ctx, key_str);
}
root = fz_dict_gets(ctx, xref->trailer, "Root");
fz_dict_puts(ctx, dests, "Names", names_list);
fz_dict_puts(ctx, names, "Dests", dests);
fz_dict_puts(ctx, root, "Names", names);
fz_drop_obj(ctx, names);
fz_drop_obj(ctx, dests);
fz_drop_obj(ctx, names_list);
fz_drop_obj(ctx, olddests);
}
}
