summaryrefslogtreecommitdiffstats
path: root/deconstructive/main.cpp
blob: 3bed79aba5da710ffd45849991415fa415278c58 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
/****************************************************************************
**
** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).
** Contact: Qt Software Information (qt-info@nokia.com)
**
** This file is part of the Graphics Dojo project on Qt Labs.
**
** This file may be used under the terms of the GNU General Public
** License version 2.0 or 3.0 as published by the Free Software Foundation
** and appearing in the file LICENSE.GPL included in the packaging of
** this file.  Please review the following information to ensure GNU
** General Public Licensing requirements will be met:
** http://www.fsf.org/licensing/licenses/info/GPLv2.html and
** http://www.gnu.org/copyleft/gpl.html.
**
** If you are unsure which license is appropriate for your use, please
** contact the sales department at qt-sales@nokia.com.
**
** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
**
****************************************************************************/

#include <QtGui/QApplication>
#include <QWebView>
#include <QWebElement>
#include <QWebPage>
#include <QWebFrame>
#include <QNetworkReply>
#include <QNetworkAccessManager>
#include <QTimer>
#include <QBuffer>
#include <QDebug>
#include <QFile>
#include <QImage>
namespace {
    class RendererNetworkReply : public QNetworkReply
    {
        Q_OBJECT
        QBuffer buffer;
        public:
            RendererNetworkReply(QWebFrame* frame,const QNetworkRequest & r)
                   : QNetworkReply(frame)
            {
                setRequest(r);
                QString selector = r.url().toString(QUrl::RemoveScheme|QUrl::RemoveQuery);
                selector = QUrl::fromPercentEncoding(selector.toAscii());
                setOperation(QNetworkAccessManager::GetOperation);
                setHeader(QNetworkRequest::ContentTypeHeader,QVariant("image/png"));
                open(ReadOnly|Unbuffered);
                setUrl(r.url());
                QWebElement element = frame->findFirstElement(selector);
                QImage img(element.geometry().size(),QImage::Format_ARGB32);
                img.fill(0);
                {
                    QPainter painter(&img);
                    element.render(&painter);
                }
                buffer.open(QIODevice::WriteOnly);
                img.save(&buffer,"png");
                buffer.close();
                buffer.open(QIODevice::ReadOnly);
                QTimer::singleShot(0,this,SIGNAL(readyRead()));
                QTimer::singleShot(0,this,SIGNAL(finished()));
            }

            virtual qint64 readData(char* data, qint64 maxSize)
            {
                return buffer.read(data,maxSize);
            }

            virtual qint64 bytesAvailable() const
            {
                return buffer.bytesAvailable();
            }

            void setData(const QByteArray & data)
            {
                buffer.setData(data);
                buffer.open(QIODevice::ReadOnly);
            }


            void abort()
            {
            }
    };

    class RendererNetworkAccessManager : public QNetworkAccessManager
    {
        QWebPage* webPage;
        public:
        RendererNetworkAccessManager(QWebPage* page) : QNetworkAccessManager(page),webPage(page)
        {
        }
        QNetworkReply* createRequest(Operation op, const QNetworkRequest & req, QIODevice* outgoingData)
        {
            if (req.url().scheme() == "render" && op == GetOperation) {
                return new RendererNetworkReply(webPage->currentFrame(),req);
            } else
                return QNetworkAccessManager::createRequest(op,req,outgoingData);
        }
    };

};

int main(int argc, char *argv[])
{
    QApplication a(argc, argv);
    QWebSettings::globalSettings()->setAttribute(QWebSettings::DeveloperExtrasEnabled,true);
    QWebView w;
    w.page()->setNetworkAccessManager(new RendererNetworkAccessManager(w.page()));
    QFile f("example.html");
    f.open(QIODevice::ReadOnly);
    w.setHtml(f.readAll());
    f.close();
    w.show();
    return a.exec();
}


#include <main.moc>