mupdf-document.cpp

Go to the documentation of this file.

#include "mupdf-document.h"
#include "mupdf-document_p.h"
#include "mupdf-page.h"
#include "mupdf-page_p.h"
#include "mupdf-outline.h"
#include "mupdf-outline_p.h"
extern "C" {
#include <mupdf/fitz.h>
}
#include <QString>
#include <QDateTime>

namespace MuPDF
{

Document * loadDocument(const QString &filePath)
{
    DocumentPrivate *documentp;
    Document *document;

    // Create DocumentPrivate
    documentp = new DocumentPrivate(filePath);
    if (!documentp)
        return NULL;
    else if (!documentp->context || !documentp->document) {
        delete documentp;
        return NULL;
    }

    // Create Document
    document = new Document(documentp);
    return document;
}

Document * loadDocument(const QByteArray &bytes)
{
    DocumentPrivate *documentp;
    Document *document;

    // Create DocumentPrivate
    documentp = new DocumentPrivate((unsigned char *)bytes.data(), bytes.length());
    if (!documentp)
        return NULL;
    else if (!documentp->context || !documentp->document) {
        delete documentp;
        return NULL;
    }

    // Create Document
    document = new Document(documentp);
    return document;
}

DocumentPrivate::DocumentPrivate(const QString &filePath)
    : context(NULL), document(NULL)
    , transparent(false)
    , b(-1), g(-1), r(-1), a(-1)
{
    // create context
    context = fz_new_context(NULL, NULL, FZ_STORE_UNLIMITED);
    if (!context)
        return;

    // register the default file types
    fz_register_document_handlers(context);

    // open document
    fz_try(context)
    {
        document = fz_open_document(context, filePath.toUtf8().data());
    }
    fz_catch(context)
    {
        deleteData();
    }
}

DocumentPrivate::DocumentPrivate(unsigned char *bytes, int len)
    : context(NULL), document(NULL)
    , transparent(false)
    , b(-1), g(-1), r(-1), a(-1)
{
    // create context
    context = fz_new_context(NULL, NULL, FZ_STORE_UNLIMITED);
    if (!context)
        return;

    // register the default file types
    fz_register_document_handlers(context);

    // open document
    fz_try(context)
    {
        // access to the bytes stream
        fz_stream* stream = fz_open_memory(context, bytes, len);

        document = fz_open_document_with_stream(context, ".pdf", stream);
    }
    fz_catch(context)
    {
        deleteData();
    }
}

Document::~Document()
{
    delete d;
    d = NULL;
}

bool Document::needsPassword() const
{
    return fz_needs_password(d->document);
}

bool Document::authPassword(const QString &password)
{
    return fz_authenticate_password(d->document,
            password.toLocal8Bit().data());
}

int Document::numPages() const
{
    int ret;
    fz_try(d->context)
    {
        ret = fz_count_pages(d->document);
    }
    fz_catch(d->context)
    {
        ret = -1;
    }
    return ret;
}

Page * Document::page(int index) const
{
    PagePrivate *pagep;
    Page *page;

    // Create PagePrivate
    pagep = new PagePrivate(d, index);
    if (!pagep)
        return NULL;
    else if (!pagep->page) {
        delete pagep;
        return NULL;
    }

    // Create Page
    page = new Page(pagep);
    if (page)
        d->pages << pagep;
    return page;
}

Outline * Document::outline() const
{
    fz_outline *o;
    OutlinePrivate *outlinep;
    
    o = fz_load_outline(d->document);
    if (o) {
        outlinep = new OutlinePrivate(d, o);
        d->outlines << outlinep;
        return new Outline(outlinep);
    }

    return NULL;
}

QString Document::pdfVersion() const
{
    pdf_document *xref = (pdf_document *)d->document;
    return QString::number(xref->version / 10.0f);
}

QString Document::title() const
{
    return d->info("Title");
}

QString Document::author() const
{
    return d->info("Author");
}

QString Document::subject() const
{
    return d->info("Subject");
}

QString Document::keywords() const
{
    return d->info("Keywords");
}

QString Document::creator() const
{
    return d->info("Creator");
}

QString Document::producer() const
{
    return d->info("Producer");
}

QDateTime Document::creationDate() const
{
    QString str = d->info("CreationDate");
    if (str.isEmpty()) {
        return QDateTime();
    }
    // see pdf_reference_1.7.pdf 2.8.3 Dates
    return QDateTime::fromString(str.left(16),
            "'D:'yyyyMMddHHmmss");
}

QDateTime Document::modDate() const
{
    QString str = d->info("ModDate");
    if (str.isEmpty()) {
        return QDateTime();
    }
    // see pdf_reference_1.7.pdf 2.8.3 Dates
    return QDateTime::fromString(str.left(16),
            "'D:'yyyyMMddHHmmss");
}

void Document::setTransparentRendering(bool enable)
{
    d->transparent = enable;
}

void Document::setBackgroundColor(int r, int g, int b, int a)
{
    d->r = r;
    d->g = g;
    d->b = b;
    d->a = a;
}

DocumentPrivate::~DocumentPrivate()
{
    foreach (OutlinePrivate *outlinep, outlines) {
        outlinep->deleteData();
    }
    foreach (PagePrivate *pagep, pages) {
        pagep->deleteData();
    }

    deleteData();
}

} // end namespace MuPDF