mirror of
https://bitbucket.org/smil3y/kde-extraapps.git
synced 2025-02-24 10:52:53 +00:00
133 lines
3.3 KiB
C++
133 lines
3.3 KiB
C++
/***************************************************************************
|
|
* Copyright (C) 2008 by Ely Levy <elylevy@cs.huji.ac.il> *
|
|
* *
|
|
* This program is free software; you can redistribute it and/or modify *
|
|
* it under the terms of the GNU General Public License as published by *
|
|
* the Free Software Foundation; either version 2 of the License, or *
|
|
* (at your option) any later version. *
|
|
***************************************************************************/
|
|
|
|
#include "epubdocument.h"
|
|
#include <QTemporaryFile>
|
|
#include <QDir>
|
|
|
|
#include <KDebug>
|
|
|
|
#include <QRegExp>
|
|
|
|
using namespace Epub;
|
|
|
|
namespace {
|
|
|
|
QString resourceUrl(const KUrl &baseUrl, const QString &u)
|
|
{
|
|
KUrl newUrl(KUrl(baseUrl.directory(KUrl::AppendTrailingSlash)), u);
|
|
QString newDir = newUrl.toLocalFile();
|
|
newDir.remove(0, 1);
|
|
return newDir;
|
|
}
|
|
|
|
}
|
|
|
|
EpubDocument::EpubDocument(const QString &fileName) : QTextDocument(),
|
|
padding(20)
|
|
{
|
|
mEpub = epub_open(qPrintable(fileName), 3);
|
|
|
|
setPageSize(QSizeF(600, 800));
|
|
}
|
|
|
|
bool EpubDocument::isValid()
|
|
{
|
|
return (mEpub?true:false);
|
|
}
|
|
|
|
EpubDocument::~EpubDocument() {
|
|
|
|
if (mEpub)
|
|
epub_close(mEpub);
|
|
|
|
epub_cleanup();
|
|
}
|
|
|
|
struct epub *EpubDocument::getEpub()
|
|
{
|
|
return mEpub;
|
|
}
|
|
|
|
void EpubDocument::setCurrentSubDocument(const QString &doc)
|
|
{
|
|
mCurrentSubDocument = KUrl::fromPath("/" + doc);
|
|
}
|
|
|
|
int EpubDocument::maxContentHeight() const
|
|
{
|
|
return pageSize().height() - (2 * padding);
|
|
}
|
|
|
|
int EpubDocument::maxContentWidth() const
|
|
{
|
|
return pageSize().width() - (2 * padding);
|
|
}
|
|
|
|
void EpubDocument::checkCSS(QString &css)
|
|
{
|
|
// remove paragraph line-heights
|
|
css.remove(QRegExp("line-height\\s*:\\s*[\\w\\.]*;"));
|
|
}
|
|
|
|
QVariant EpubDocument::loadResource(int type, const QUrl &name)
|
|
{
|
|
int size;
|
|
char *data;
|
|
|
|
// Get the data from the epub file
|
|
size = epub_get_data(mEpub, resourceUrl(mCurrentSubDocument, name.toString()).toUtf8(), &data);
|
|
|
|
QVariant resource;
|
|
|
|
if (data) {
|
|
switch(type) {
|
|
case QTextDocument::ImageResource:{
|
|
#ifndef QT_KATIE
|
|
QImage img = QImage::fromData((unsigned char *)data, size);
|
|
#else
|
|
QImage img = QImage::fromData(data, size);
|
|
#endif
|
|
const int maxHeight = maxContentHeight();
|
|
const int maxWidth = maxContentWidth();
|
|
if(img.height() > maxHeight)
|
|
img = img.scaledToHeight(maxHeight);
|
|
if(img.width() > maxWidth)
|
|
img = img.scaledToWidth(maxWidth);
|
|
resource.setValue(img);
|
|
break;
|
|
}
|
|
case QTextDocument::StyleSheetResource: {
|
|
QString css = QString::fromUtf8(data, size);
|
|
checkCSS(css);
|
|
resource.setValue(css);
|
|
break;
|
|
}
|
|
case EpubDocument::MovieResource:
|
|
case EpubDocument::AudioResource: {
|
|
QTemporaryFile *tmp = new QTemporaryFile(QString("%1/okrXXXXXX").arg(QDir::tempPath()),this);
|
|
if(!tmp->open()) kWarning() << "EPUB : error creating temporary video file";
|
|
if(tmp->write(data,size) == -1) kWarning() << "EPUB : error writing data" << tmp->errorString();
|
|
tmp->flush();
|
|
resource.setValue(tmp->fileName());
|
|
break;
|
|
}
|
|
default:
|
|
resource.setValue(QString::fromUtf8(data, size));
|
|
break;
|
|
}
|
|
|
|
free(data);
|
|
}
|
|
|
|
// add to cache
|
|
addResource(type, name, resource);
|
|
|
|
return resource;
|
|
}
|