0
|
1 |
/****************************************************************************
|
|
2 |
**
|
|
3 |
** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).
|
|
4 |
** All rights reserved.
|
|
5 |
** Contact: Nokia Corporation (qt-info@nokia.com)
|
|
6 |
**
|
|
7 |
** This file is part of the examples of the Qt Toolkit.
|
|
8 |
**
|
|
9 |
** $QT_BEGIN_LICENSE:LGPL$
|
|
10 |
** No Commercial Usage
|
|
11 |
** This file contains pre-release code and may not be distributed.
|
|
12 |
** You may use this file in accordance with the terms and conditions
|
|
13 |
** contained in the Technology Preview License Agreement accompanying
|
|
14 |
** this package.
|
|
15 |
**
|
|
16 |
** GNU Lesser General Public License Usage
|
|
17 |
** Alternatively, this file may be used under the terms of the GNU Lesser
|
|
18 |
** General Public License version 2.1 as published by the Free Software
|
|
19 |
** Foundation and appearing in the file LICENSE.LGPL included in the
|
|
20 |
** packaging of this file. Please review the following information to
|
|
21 |
** ensure the GNU Lesser General Public License version 2.1 requirements
|
|
22 |
** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
|
|
23 |
**
|
|
24 |
** In addition, as a special exception, Nokia gives you certain additional
|
|
25 |
** rights. These rights are described in the Nokia Qt LGPL Exception
|
|
26 |
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
|
|
27 |
**
|
|
28 |
** If you have questions regarding the use of this file, please contact
|
|
29 |
** Nokia at qt-info@nokia.com.
|
|
30 |
**
|
|
31 |
**
|
|
32 |
**
|
|
33 |
**
|
|
34 |
**
|
|
35 |
**
|
|
36 |
**
|
|
37 |
**
|
|
38 |
** $QT_END_LICENSE$
|
|
39 |
**
|
|
40 |
****************************************************************************/
|
|
41 |
|
|
42 |
#include "framecapture.h"
|
|
43 |
|
|
44 |
#include <iostream>
|
|
45 |
#include <QtWebKit>
|
|
46 |
|
|
47 |
FrameCapture::FrameCapture(): QObject(), m_percent(0)
|
|
48 |
{
|
|
49 |
connect(&m_page, SIGNAL(loadProgress(int)), this, SLOT(printProgress(int)));
|
|
50 |
connect(&m_page, SIGNAL(loadFinished(bool)), this, SLOT(saveResult(bool)));
|
|
51 |
}
|
|
52 |
|
|
53 |
void FrameCapture::load(const QUrl &url, const QString &outputFileName)
|
|
54 |
{
|
|
55 |
std::cout << "Loading " << qPrintable(url.toString()) << std::endl;
|
|
56 |
m_percent = 0;
|
|
57 |
int index = outputFileName.lastIndexOf('.');
|
|
58 |
m_fileName = (index == -1) ? outputFileName + ".png" : outputFileName;
|
|
59 |
m_page.mainFrame()->load(url);
|
|
60 |
m_page.mainFrame()->setScrollBarPolicy(Qt::Vertical, Qt::ScrollBarAlwaysOff);
|
|
61 |
m_page.mainFrame()->setScrollBarPolicy(Qt::Horizontal, Qt::ScrollBarAlwaysOff);
|
|
62 |
}
|
|
63 |
|
|
64 |
void FrameCapture::printProgress(int percent)
|
|
65 |
{
|
|
66 |
if (m_percent >= percent)
|
|
67 |
return;
|
|
68 |
|
|
69 |
while (m_percent++ < percent)
|
|
70 |
std::cout << "#" << std::flush;
|
|
71 |
}
|
|
72 |
|
|
73 |
void FrameCapture::saveResult(bool ok)
|
|
74 |
{
|
|
75 |
std::cout << std::endl;
|
|
76 |
|
|
77 |
// crude error-checking
|
|
78 |
if (!ok) {
|
|
79 |
std::cerr << "Failed loading " << qPrintable(m_page.mainFrame()->url().toString()) << std::endl;
|
|
80 |
emit finished();
|
|
81 |
return;
|
|
82 |
}
|
|
83 |
|
|
84 |
// save each internal frame in different image files
|
|
85 |
int frameCounter = 0;
|
|
86 |
foreach(QWebFrame *frame, m_page.mainFrame()->childFrames()) {
|
|
87 |
QString fileName(m_fileName);
|
|
88 |
int index = m_fileName.lastIndexOf('.');
|
|
89 |
fileName = fileName.insert(index, "_frame" + QString::number(++frameCounter));
|
|
90 |
|
|
91 |
frame->setClipRenderToViewport(false);
|
|
92 |
|
|
93 |
QImage image(frame->contentsSize(), QImage::Format_ARGB32_Premultiplied);
|
|
94 |
image.fill(Qt::transparent);
|
|
95 |
|
|
96 |
saveFrame(frame, image, fileName);
|
|
97 |
}
|
|
98 |
|
|
99 |
// save the main frame
|
|
100 |
m_page.setViewportSize(m_page.mainFrame()->contentsSize());
|
|
101 |
QImage image(m_page.mainFrame()->contentsSize(), QImage::Format_ARGB32_Premultiplied);
|
|
102 |
image.fill(Qt::transparent);
|
|
103 |
saveFrame(m_page.mainFrame(), image, m_fileName);
|
|
104 |
|
|
105 |
emit finished();
|
|
106 |
}
|
|
107 |
|
|
108 |
void FrameCapture::saveFrame(QWebFrame *frame, QImage image, QString fileName)
|
|
109 |
{
|
|
110 |
QPainter painter(&image);
|
|
111 |
painter.setRenderHint(QPainter::Antialiasing, true);
|
|
112 |
painter.setRenderHint(QPainter::TextAntialiasing, true);
|
|
113 |
painter.setRenderHint(QPainter::SmoothPixmapTransform, true);
|
|
114 |
|
|
115 |
frame->render(&painter);
|
|
116 |
|
|
117 |
painter.end();
|
|
118 |
|
|
119 |
image.save(fileName);
|
|
120 |
}
|
|
121 |
|