~oif-team/ubuntu/natty/qt4-x11/xi2.1

« back to all changes in this revision

Viewing changes to examples/webkit/framecapture/framecapture.cpp

  • Committer: Bazaar Package Importer
  • Author(s): Alessandro Ghersi
  • Date: 2009-11-02 18:30:08 UTC
  • mfrom: (1.2.2 upstream)
  • mto: (15.2.5 experimental)
  • mto: This revision was merged to the branch mainline in revision 88.
  • Revision ID: james.westby@ubuntu.com-20091102183008-b6a4gcs128mvfb3m
Tags: upstream-4.6.0~beta1
ImportĀ upstreamĀ versionĀ 4.6.0~beta1

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
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