nsrpopplerdocument.cpp

/*
 * Copyright (C) 2011-2015 Alexander Saprykin <xelfium@gmail.com>
 *
 * 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.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301  USA.
 */

#include "nsrpopplerdocument.h"

#include <QtGui>
#include <QtCore>

#include "ErrorCodes.h"

int NSRPopplerDocument::_refcount = 0;

QMutex mutex;

NSRPopplerDocument::NSRPopplerDocument(const QString& file, QObject *parent) :
	NSRAbstractDocument (file, parent),
	_doc (NULL),
	_catalog (NULL),
	_page (NULL),
	_dev (NULL),
	_dpix (72),
	_dpiy (72),
	_readyForLoad (false),
	_cachedPageSize (QSize (0, 0)),
	_cachedMinZoom (25),
	_cachedMaxZoom (100)
{
	mutex.lock();
	if (_refcount == 0)
		globalParams = new GlobalParams ();
	++_refcount;
	mutex.unlock();

	if (!QFile::exists(file))
		return;

	createInternalDoc();
}

NSRPopplerDocument::~NSRPopplerDocument()
{
	if (_readyForLoad)
		_dev->startPage(0, NULL);

	if (_dev != NULL)
		delete _dev;

	if (_doc != NULL)
		delete _doc;

#ifdef Q_OS_SYMBIAN
	if (_iBitmap != NULL) {
		_iBitmap->Reset();
		delete _iBitmap;
	}
#endif

	mutex.lock();
	--_refcount;
	if (_refcount == 0)
		delete globalParams;
	mutex.unlock();
}


int NSRPopplerDocument::getNumberOfPages() const
{
	if (_doc == NULL)
		return 0;

	return _doc->getNumPages();
}

bool NSRPopplerDocument::isValid() const
{
	return (_doc != NULL && _doc->isOk());
}

void NSRPopplerDocument::renderPage(int page)
{
	double dpix, dpiy;

	if (_doc == NULL || page > getNumberOfPages() || page < 1)
		return;

	_page = _catalog->getPage(page);

	if (isTextOnly()) {
		PDFRectangle	*rect;
		GooString	*text;
		TextOutputDev	*dev;

		dev = new TextOutputDev (0, gFalse, gFalse, gFalse);

		_doc->displayPageSlice(dev, _page->getNum(), 72, 72, 0, gFalse, gTrue, gFalse, -1, -1, -1, -1);

		rect = _page->getCropBox();
		text = dev->getText(rect->x1, rect->y1, rect->x2, rect->y2);
		_text = processText(QString::fromUtf8(text->getCString()));

		delete text;
		delete dev;
		_readyForLoad = true;

		return;
	}

	if (isZoomToWidth()) {
		double wZoom = ((double) getScreenWidth() / (double) _page->getCropWidth() * 100.0);
		setZoomSilent((int) wZoom);
	}

	if (getZoom() > getMaxZoom())
		setZoomSilent (getMaxZoom());
	else if (getZoom() < getMinZoom())
		setZoomSilent (getMinZoom());

	if (_readyForLoad)
		_dev->startPage(0, NULL);

	dpix = _dpix * getZoom() / 100.0;
	dpiy = _dpiy * getZoom() / 100.0;

	_page->display(_dev, dpix, dpiy, getRotation(), gFalse, gFalse, gTrue, _catalog);

	_readyForLoad = true;
}

int NSRPopplerDocument::getMaxZoom()
{
	if (_page == NULL)
		return 0;

	if (QSize (_page->getCropWidth(), _page->getCropHeight()) == _cachedPageSize)
		return _cachedMaxZoom;

	/* Each pixel needs 4 bytes (RGBA) of memory */
	double pageSize = _page->getCropWidth() * _page->getCropHeight() * 4;
	_cachedMaxZoom = (int) (sqrt (NSR_DOCUMENT_MAX_HEAP * 72 * 72 / pageSize ) / 72 * 100 + 0.5);
	_cachedPageSize = QSize (_page->getCropWidth(), _page->getCropHeight());

	return _cachedMaxZoom;
}

int NSRPopplerDocument::getMinZoom()
{
	if (_page == NULL)
		return 0;

	if (QSize (_page->getCropWidth(), _page->getCropHeight()) == _cachedPageSize)
		return _cachedMinZoom;

	/* Each pixel needs 4 bytes (RGBA) of memory */
	double pageSize = _page->getCropWidth() * _page->getCropHeight() * 4;

	if (pageSize > NSR_DOCUMENT_MAX_HEAP)
		_cachedMinZoom = getMaxZoom();
	else
		_cachedMinZoom = (getMaxZoom() / 10) > 25 ? 25 : getMaxZoom() / 10;

	_cachedPageSize = QSize (_page->getCropWidth(), _page->getCropHeight());

	return _cachedMinZoom;
}

QPixmap NSRPopplerDocument::getCurrentPage()
{
	if (!_readyForLoad)
		return QPixmap();

	SplashBitmap *bitmap = _dev->getBitmap();
	int bw = bitmap->getWidth();
	int bh = bitmap->getHeight();

	char *dataPtr = (char *) _dev->getBitmap()->getDataPtr();

	int rowBytes = bw * 3;
	while (rowBytes % 4)
		rowBytes += 1;

#ifdef Q_OS_SYMBIAN
	TSize size (bw, bh);

	if (size != _iBitmap->SizeInPixels()) {
		_iBitmap->Reset();
		_iBitmap->Create(size, EColor16MU);
	}

	_iBitmap->BeginDataAccess();
	char *addr = (char *) _iBitmap->DataAddress();
	TInt stride = _iBitmap->DataStride();

	for (int i = 0; i < bh; ++i) {
		char *inAddr = dataPtr + i * rowBytes;

		for (int j = 0; j < bw; ++j) {
			if (isInvertedColors()) {
				addr[j * 4 + 0] = 255 - inAddr[j * 3 + 2];
				addr[j * 4 + 1] = 255 - inAddr[j * 3 + 1];
				addr[j * 4 + 2] = 255 - inAddr[j * 3 + 0];
			} else {
				addr[j * 4 + 0] = inAddr[j * 3 + 2];
				addr[j * 4 + 1] = inAddr[j * 3 + 1];
				addr[j * 4 + 2] = inAddr[j * 3 + 0];
			}

		}

		addr += stride;
	}

	_iBitmap->EndDataAccess();
#else
	QSize size (bw, bh);
	QImage img (size, QImage::Format_RGB888);

	char *addr = (char *) img.bits ();
	int stride = img.bytesPerLine ();

	for (int i = 0; i < bh; ++i) {
		char *inAddr = dataPtr + i * rowBytes;

		for (int j = 0; j < bw; ++j) {
			if (isInvertedColors()) {
				addr[j * 3 + 2] = 255 - inAddr[j * 3 + 2];
				addr[j * 3 + 1] = 255 - inAddr[j * 3 + 1];
				addr[j * 3 + 0] = 255 - inAddr[j * 3 + 0];
			} else {
				addr[j * 3 + 2] = inAddr[j * 3 + 2];
				addr[j * 3 + 1] = inAddr[j * 3 + 1];
				addr[j * 3 + 0] = inAddr[j * 3 + 0];
			}

		}

		addr += stride;
	}
#endif

	_dev->startPage (0, NULL);
	_readyForLoad = false;

#ifdef Q_OS_SYMBIAN
	return QPixmap::fromSymbianCFbsBitmap(_iBitmap);
#else
	return QPixmap::fromImage (img);
#endif
}

QString NSRPopplerDocument::getText()
{
	if (!_readyForLoad)
		return NSRAbstractDocument::getText();

	_readyForLoad = false;

	if (_text.isEmpty())
		return NSRAbstractDocument::getText();
	else
		return _text;
}

void NSRPopplerDocument::setPassword(const QString &passwd)
{
	if (_doc != NULL)
		return;

	createInternalDoc(passwd);
}

void NSRPopplerDocument::createInternalDoc(QString passwd)
{
	SplashColor	bgColor;
	GooString	*fileName;
	GooString	*passwdStr;

	if (_doc != NULL)
		return;

	bgColor[0] = 255;
	bgColor[1] = 255;
	bgColor[2] = 255;
	_dev = new SplashOutputDev(splashModeRGB8, 4, gFalse, bgColor);

	fileName = new GooString (getDocumentPath().toUtf8().data());

	if (!passwd.isEmpty())
		passwdStr = new GooString (passwd.toUtf8().data());
	else
		passwdStr = NULL;

	_doc = new PDFDoc (fileName, passwdStr);

	if (!_doc->isOk()) {
		if (_doc->getErrorCode() == errEncrypted)
			setLastError (NSR_DOCUMENT_ERROR_PASSWD);
		else
			setLastError (NSR_DOCUMENT_ERROR_UNKNOWN);

		delete _doc;
		_doc = NULL;

#ifdef Q_OS_SYMBIAN
		_iBitmap = NULL;
#endif
		return;
	}

#ifdef Q_OS_SYMBIAN
	_iBitmap = new CFbsBitmap ();
#endif
	_catalog = _doc->getCatalog();
	_page = _catalog->getPage(1);
	_dev->startDoc(_doc->getXRef());
}