plasma-framework/svg.cpp
Davide Bettio 90999c00d6 Fixed findInCache bug. Now findInCache is faster, thanks to bibr.
svn path=/trunk/KDE/kdebase/workspace/libs/plasma/; revision=796500
2008-04-13 17:27:02 +00:00

436 lines
12 KiB
C++

/*
* Copyright 2006-2007 Aaron Seigo <aseigo@kde.org>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU Library General Public License as
* published by the Free Software Foundation; either version 2, 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 Library 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 "svg.h"
#include <QDir>
#include <QMatrix>
#include <QPainter>
#include <QPixmapCache>
#include <QSharedData>
#include <KDebug>
#include <KSharedPtr>
#include <KSvgRenderer>
#include <KColorScheme>
#include <KIconEffect>
#include <KGlobalSettings>
#include "theme.h"
namespace Plasma
{
class SharedSvgRenderer : public KSvgRenderer, public QSharedData
{
public:
typedef KSharedPtr<SharedSvgRenderer> Ptr;
SharedSvgRenderer(QObject *parent = 0)
: KSvgRenderer(parent)
{}
SharedSvgRenderer(const QString &filename, QObject *parent = 0)
: KSvgRenderer(filename, parent)
{}
SharedSvgRenderer(const QByteArray &contents, QObject *parent = 0)
: KSvgRenderer(contents, parent)
{}
~SharedSvgRenderer()
{
//kDebug() << "leaving this world for a better one.";
}
};
class Svg::Private
{
public:
Private(const QString& imagePath, Svg *q)
: renderer(0),
contentType(Svg::SingleImage),
themed(false)
{
setImagePath(imagePath, q);
}
~Private()
{
eraseRenderer();
}
void setImagePath(const QString &imagePath, Svg *q)
{
if (themed) {
disconnect(Plasma::Theme::self(), SIGNAL(changed()), q, SLOT(themeChanged()));
disconnect(KGlobalSettings::self(), SIGNAL(kdisplayPaletteChanged()), q, SLOT(colorsChanged()));
}
themed = !QDir::isAbsolutePath(imagePath);
path = themePath = QString();
if (themed) {
themePath = imagePath;
connect(Plasma::Theme::self(), SIGNAL(changed()), q, SLOT(themeChanged()));
// check if svg wants colorscheme applied
createRenderer();
applyColors = renderer->elementExists("hint-apply-color-scheme");
if (applyColors && !Theme::self()->colors()) {
connect(KGlobalSettings::self(), SIGNAL(kdisplayPaletteChanged()), q, SLOT(colorsChanged()));
}
} else {
path = imagePath;
if (!QFile::exists(path)) {
kDebug() << "Plasma::Svg: file '" << path << "' does not exist!";
}
}
}
void removeFromCache() {
if (ids.isEmpty()) {
return;
}
foreach (const QString &id, ids) {
QPixmapCache::remove(id);
}
ids.clear();
}
void findInCache(QPixmap& p, const QString& elementId, const QPainter *itemPainter, const QSizeF &size = QSizeF())
{
QSizeF localSize;
if (elementId.isEmpty() || contentType == Svg::ImageSet) {
localSize = size;
} else {
localSize = elementSize(elementId);
}
if (localSize.isEmpty())
return;
QTransform transform = itemPainter->worldTransform();
if (size.isValid()) {
transform.scale(size.width() / localSize.width(),
size.height() / localSize.height());
}
createRenderer();
QLineF v1 = transform.map(QLineF(0, 0, 1, 0));
QLineF v2 = transform.map(QLineF(0, 0, 0, 1));
QString id = QString::fromLatin1("%7_%6_%5_%4_%3_%2_%1").arg(size.width())
.arg(size.height())
.arg((int) v1.dx())
.arg((int) v1.dy())
.arg((int) v2.dx())
.arg((int) v2.dy())
.arg(path);
if (!elementId.isEmpty()) {
id.append(elementId);
}
//kDebug() << "id is " << id;
if (!ids.contains(id)) {
ids.append(id);
}
if (QPixmapCache::find(id, p)) {
//kDebug() << "found cached version of " << id;
return;
} else {
//kDebug() << "didn't find cached version of " << id << ", so re-rendering";
}
QRect deviceRect = transform.mapRect(QRectF(QPointF(), localSize)).toRect();
if (deviceRect.isEmpty())
return;
p = QPixmap(deviceRect.size());
p.fill(Qt::transparent);
QPainter renderPainter(&p);
QPointF tl = deviceRect.topLeft();
QTransform xform = QTransform().translate(-tl.x(), -tl.y());
xform = itemPainter->worldTransform() * xform;
renderPainter.setRenderHints(renderPainter.renderHints(), false);
renderPainter.setRenderHints(itemPainter->renderHints(), true);
renderPainter.setWorldTransform(xform, true);
if (elementId.isEmpty()) {
renderer->render(&renderPainter);
} else {
renderer->render(&renderPainter, elementId);
}
renderPainter.end();
// Apply current color scheme if the svg asks for it
if (applyColors) {
QImage itmp = p.toImage();
KIconEffect::colorize(itmp, Theme::self()->backgroundColor(), 1.0);
p = p.fromImage(itmp);
}
if (!QPixmapCache::insert( id, p )) {
//kDebug() << "pixmap cache is too small for inserting" << id << "of size" << s;
}
}
void createRenderer()
{
if (renderer) {
return;
}
if (themed && path.isEmpty()) {
path = Plasma::Theme::self()->image(themePath);
}
QHash<QString, SharedSvgRenderer::Ptr>::const_iterator it = renderers.find(path);
if (it != renderers.end()) {
//kDebug() << "gots us an existing one!";
renderer = it.value();
} else {
renderer = new SharedSvgRenderer(path);
renderers[path] = renderer;
}
size = renderer->defaultSize();
}
void eraseRenderer()
{
if (renderer.count() == 2) {
// this and the cache reference it; and boy is this not thread safe ;)
renderers.erase(renderers.find(themePath));
}
renderer = 0;
}
QSize elementSize(const QString& elementId)
{
createRenderer();
QSizeF elementSize = renderer->boundsOnElement(elementId).size();
QSizeF naturalSize = renderer->defaultSize();
qreal dx = size.width() / naturalSize.width();
qreal dy = size.height() / naturalSize.height();
elementSize.scale(elementSize.width() * dx, elementSize.height() * dy, Qt::IgnoreAspectRatio);
return elementSize.toSize();
}
QRect elementRect(const QString& elementId)
{
createRenderer();
QRectF elementRect = renderer->boundsOnElement(elementId);
QSizeF naturalSize = renderer->defaultSize();
qreal dx = size.width() / naturalSize.width();
qreal dy = size.height() / naturalSize.height();
return QRect(elementRect.x() * dx, elementRect.y() * dy,
elementRect.width() * dx, elementRect.height() * dy);
}
QMatrix matrixForElement(const QString& elementId)
{
createRenderer();
return renderer->matrixForElement(elementId);
}
static QHash<QString, SharedSvgRenderer::Ptr> renderers;
SharedSvgRenderer::Ptr renderer;
QString themePath;
QString path;
QList<QString> ids;
QSizeF size;
bool themed;
bool applyColors;
Svg::ContentType contentType;
};
QHash<QString, SharedSvgRenderer::Ptr> Svg::Private::renderers;
Svg::Svg(const QString& imagePath, QObject* parent)
: QObject(parent),
d(new Private(imagePath, this))
{
}
Svg::~Svg()
{
delete d;
}
void Svg::paint(QPainter* painter, const QPointF& point, const QString& elementID)
{
QPixmap pix;
d->findInCache(pix, elementID, painter);
if (pix.isNull())
return;
painter->save();
QTransform x = painter->worldTransform();
painter->setWorldTransform(QTransform(), false);
//kDebug() << "pix size is " << pix.size();
painter->drawPixmap(x.map(point), pix);
painter->restore();
}
void Svg::paint(QPainter* painter, int x, int y, const QString& elementID)
{
paint(painter, QPointF(x, y), elementID);
}
void Svg::paint(QPainter* painter, const QRectF& rect, const QString& elementID)
{
QPixmap pix;
d->findInCache(pix, elementID, painter, rect.size());
if (pix.isNull())
return;
painter->save();
QTransform x = painter->worldTransform();
painter->setWorldTransform(QTransform());
painter->drawPixmap(x.map(rect.topLeft()) , pix);
painter->restore();
}
void Svg::resize( int width, int height )
{
resize( QSize( width, height ) );
}
void Svg::resize( const QSizeF& size )
{
d->createRenderer();
d->size = size;
}
void Svg::resize()
{
d->createRenderer();
d->size = d->renderer->defaultSize();
}
QSize Svg::elementSize(const QString& elementId) const
{
return d->elementSize(elementId);
}
QRect Svg::elementRect(const QString& elementId) const
{
return d->elementRect(elementId);
}
bool Svg::elementExists(const QString& elementId) const
{
d->createRenderer();
return d->renderer->elementExists(elementId);
}
QString Svg::elementAtPoint(const QPoint &point) const
{
d->createRenderer();
QSizeF naturalSize = d->renderer->defaultSize();
qreal dx = d->size.width() / naturalSize.width();
qreal dy = d->size.height() / naturalSize.height();
//kDebug() << point << "is really" << QPoint(point.x() *dx, naturalSize.height() - point.y() * dy);
return QString(); // d->renderer->elementAtPoint(QPoint(point.x() *dx, naturalSize.height() - point.y() * dy));
}
QMatrix Svg::matrixForElement(const QString& elementId) const
{
d->createRenderer();
return d->renderer->matrixForElement(elementId);
}
bool Svg::isValid() const
{
d->createRenderer();
return d->renderer->isValid();
}
QSize Svg::size() const
{
return d->size.toSize();
}
void Svg::setContentType(ContentType contentType)
{
d->contentType = contentType;
}
Svg::ContentType Svg::contentType()
{
return d->contentType;
}
void Svg::setFile(const QString &svgFilePath)
{
d->setImagePath(svgFilePath, this);
d->eraseRenderer();
}
QString Svg::file() const
{
return d->themed ? d->themePath : d->path;
}
void Svg::themeChanged()
{
if (!d->themed) {
return;
}
QString newPath = Theme::self()->image(d->themePath);
if (d->path == newPath) {
return;
}
d->removeFromCache();
d->path = newPath;
//delete d->renderer; we're a KSharedPtr
d->eraseRenderer();
emit repaintNeeded();
}
void Svg::colorsChanged()
{
if (!d->applyColors) {
return;
}
d->removeFromCache();
d->eraseRenderer();
emit repaintNeeded();
}
} // Plasma namespace
#include "svg.moc"