zathura/ft/pdf-poppler/pdf.c
2011-03-18 14:10:42 +01:00

302 lines
7.2 KiB
C

/* See LICENSE file for license and copyright information */
#include <stdlib.h>
#include <poppler/glib/poppler.h>
#include "pdf.h"
#include "../../zathura.h"
void
plugin_register(zathura_document_plugin_t* plugin)
{
plugin->file_extension = "pdf";
plugin->open_function = pdf_document_open;
}
bool
pdf_document_open(zathura_document_t* document)
{
if (!document) {
goto error_out;
}
document->functions.document_free = pdf_document_free;
document->functions.document_index_generate = pdf_document_index_generate;;
document->functions.document_save_as = pdf_document_save_as;
document->functions.document_attachments_get = pdf_document_attachments_get;
document->functions.page_get = pdf_page_get;
document->functions.page_search_text = pdf_page_search_text;
document->functions.page_links_get = pdf_page_links_get;
document->functions.page_form_fields_get = pdf_page_form_fields_get;
document->functions.page_render = pdf_page_render;
document->functions.page_free = pdf_page_free;
document->data = malloc(sizeof(pdf_document_t));
if (!document->data) {
goto error_out;
}
/* format path */
GError* error = NULL;
char* file_uri = g_filename_to_uri(document->file_path, NULL, &error);
if (!file_uri) {
fprintf(stderr, "error: could not open file: %s\n", error->message);
goto error_free;
}
pdf_document_t* pdf_document = (pdf_document_t*) document->data;
pdf_document->document = poppler_document_new_from_file(file_uri, document->password, &error);
if (!pdf_document->document) {
fprintf(stderr, "error: could not open file: %s\n", error->message);
goto error_free;
}
document->number_of_pages = poppler_document_get_n_pages(pdf_document->document);
g_free(file_uri);
return true;
error_free:
if (error) {
g_error_free(error);
}
if (file_uri) {
g_free(file_uri);
}
free(document->data);
document->data = NULL;
error_out:
return false;
}
bool
pdf_document_free(zathura_document_t* document)
{
if (!document) {
return false;
}
if (document->data) {
pdf_document_t* pdf_document = (pdf_document_t*) document->data;
g_object_unref(pdf_document->document);
free(document->data);
document->data = NULL;
}
return true;
}
static void
build_index(pdf_document_t* pdf, girara_tree_node_t* root, PopplerIndexIter* iter)
{
if (!root || !iter) {
return;
}
do
{
PopplerAction* action = poppler_index_iter_get_action(iter);
if (!action) {
continue;
}
gchar* markup = g_markup_escape_text(action->any.title, -1);
zathura_index_element_t* index_element = zathura_index_element_new(markup);
if (action->type == POPPLER_ACTION_URI) {
index_element->type = ZATHURA_LINK_EXTERNAL;
index_element->target.uri = g_strdup(action->uri.uri);
} else if (action->type == POPPLER_ACTION_GOTO_DEST) {
index_element->type = ZATHURA_LINK_TO_PAGE;
if (action->goto_dest.dest->type == POPPLER_DEST_NAMED) {
PopplerDest* dest = poppler_document_find_dest(pdf->document, action->goto_dest.dest->named_dest);
if (dest) {
index_element->target.page_number = dest->page_num - 1;
poppler_dest_free(dest);
}
} else {
index_element->target.page_number = action->goto_dest.dest->page_num - 1;
}
} else {
poppler_action_free(action);
zathura_index_element_free(index_element);
continue;
}
poppler_action_free(action);
girara_tree_node_t* node = girara_node_append_data(root, index_element);
PopplerIndexIter* child = poppler_index_iter_get_child(iter);
if (child) {
build_index(pdf, node, child);
}
poppler_index_iter_free(child);
} while (poppler_index_iter_next(iter));
}
girara_tree_node_t*
pdf_document_index_generate(zathura_document_t* document)
{
if (!document || !document->data) {
return NULL;
}
pdf_document_t* pdf_document = (pdf_document_t*) document->data;
PopplerIndexIter* iter = poppler_index_iter_new(pdf_document->document);
if (!iter) {
// XXX: error message?
return NULL;
}
girara_tree_node_t* root = girara_node_new(zathura_index_element_new("ROOT"));
girara_node_set_free_function(root, (girara_free_function_t)zathura_index_element_free);
build_index(pdf_document, root, iter);
poppler_index_iter_free(iter);
return root;
}
bool
pdf_document_save_as(zathura_document_t* document, const char* path)
{
if (!document || !document->data || !path) {
return false;
}
pdf_document_t* pdf_document = (pdf_document_t*) document->data;
char* file_path = g_strdup_printf("file://%s", path);
poppler_document_save(pdf_document->document, file_path, NULL);
g_free(file_path);
return false;
}
zathura_list_t*
pdf_document_attachments_get(zathura_document_t* document)
{
return NULL;
}
zathura_page_t*
pdf_page_get(zathura_document_t* document, unsigned int page)
{
if (!document || !document->data) {
return NULL;
}
pdf_document_t* pdf_document = (pdf_document_t*) document->data;
zathura_page_t* document_page = malloc(sizeof(zathura_page_t));
if (!document_page) {
return NULL;
}
document_page->document = document;
document_page->data = poppler_document_get_page(pdf_document->document, page);
if (!document_page->data) {
free(document_page);
return NULL;
}
poppler_page_get_size(document_page->data, &(document_page->width), &(document_page->height));
return document_page;
}
bool
pdf_page_free(zathura_page_t* page)
{
if (!page) {
return false;
}
g_object_unref(page->data);
free(page);
return true;
}
zathura_list_t*
pdf_page_search_text(zathura_page_t* page, const char* text)
{
return NULL;
}
zathura_list_t*
pdf_page_links_get(zathura_page_t* page)
{
return NULL;
}
zathura_list_t*
pdf_page_form_fields_get(zathura_page_t* page)
{
return NULL;
}
zathura_image_buffer_t*
pdf_page_render(zathura_page_t* page)
{
if (!Zathura.document || !page || !page->data || !page->document) {
return NULL;
}
/* calculate sizes */
unsigned int page_width = Zathura.document->scale * page->width;
unsigned int page_height = Zathura.document->scale * page->height;
/* create pixbuf */
GdkPixbuf* pixbuf = gdk_pixbuf_new(GDK_COLORSPACE_RGB, FALSE, 8,
page_width, page_height);
if (pixbuf == NULL) {
return NULL;
}
poppler_page_render_to_pixbuf(page->data, 0, 0, page_width, page_height,
Zathura.document->scale, 90, pixbuf);
/* create image buffer */
zathura_image_buffer_t* image_buffer = zathura_image_buffer_create(page_width, page_height);
if (image_buffer == NULL) {
g_object_unref(pixbuf);
return NULL;
}
/* copy buffer */
guchar* pixels = gdk_pixbuf_get_pixels(pixbuf);
int rowstride = gdk_pixbuf_get_rowstride(pixbuf);
int n_channels = gdk_pixbuf_get_n_channels(pixbuf);
for (unsigned int y = 0; y < page_height; y++) {
for (unsigned int x = 0; x < page_width; x++) {
unsigned char *s = pixels + y * rowstride + x * n_channels;
guchar* p = image_buffer->data + y * image_buffer->width + x;
p[0] = s[0];
p[1] = s[1];
p[2] = s[2];
}
}
g_object_unref(pixbuf);
return image_buffer;
}