diff -r 000000000000 -r 1918ee327afb tools/qvfb/qanimationwriter.cpp
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tools/qvfb/qanimationwriter.cpp	Mon Jan 11 14:00:40 2010 +0000
@@ -0,0 +1,451 @@
+/****************************************************************************
+**
+** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).
+** All rights reserved.
+** Contact: Nokia Corporation (qt-info@nokia.com)
+**
+** This file is part of the tools applications of the Qt Toolkit.
+**
+** $QT_BEGIN_LICENSE:LGPL$
+** No Commercial Usage
+** This file contains pre-release code and may not be distributed.
+** You may use this file in accordance with the terms and conditions
+** contained in the Technology Preview License Agreement accompanying
+** this package.
+**
+** GNU Lesser General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU Lesser
+** General Public License version 2.1 as published by the Free Software
+** Foundation and appearing in the file LICENSE.LGPL included in the
+** packaging of this file.  Please review the following information to
+** ensure the GNU Lesser General Public License version 2.1 requirements
+** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
+**
+** In addition, as a special exception, Nokia gives you certain additional
+** rights.  These rights are described in the Nokia Qt LGPL Exception
+** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
+**
+** If you have questions regarding the use of this file, please contact
+** Nokia at qt-info@nokia.com.
+**
+**
+**
+**
+**
+**
+**
+**
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+
+#include "qanimationwriter.h"
+#include <QFile>
+#include <QString>
+#include <QPainter>
+#include <png.h>
+#include <limits.h>
+#include <netinet/in.h> // for htonl
+
+#ifdef QT_LINUXBASE
+#  include <arpa/inet.h> // for htonl (LSB only)
+#endif
+
+QT_BEGIN_NAMESPACE
+
+class QAnimationWriterData
+{
+public:
+    QAnimationWriterData(QIODevice* d) : framerate(1000), dev(d) {}
+    void setFrameRate(int d) { framerate = d; }
+    virtual ~QAnimationWriterData() { }
+    virtual void setImage(const QImage& src)=0;
+    virtual bool canCompose() const { return false; }
+    virtual void composeImage(const QImage&, const QPoint& ) {}
+
+protected:
+    int framerate;
+    QIODevice* dev;
+};
+
+
+class QAnimationWriterMNG : public QAnimationWriterData {
+    bool first;
+    png_structp png_ptr;
+    png_infop info_ptr;
+public:
+    QAnimationWriterMNG(QIODevice* d) : QAnimationWriterData(d)
+    {
+        first = true;
+        begin_png();
+    }
+
+    ~QAnimationWriterMNG()
+    {
+        if (first) {
+            // Eh? Not images.
+            QImage dummy(1,1,QImage::Format_RGB32);
+            setImage(dummy);
+        }
+        writeMEND();
+        end_png();
+    }
+
+    void begin_png()
+    {
+        png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING,0,0,0);
+        info_ptr = png_create_info_struct(png_ptr);
+        png_set_compression_level(png_ptr,9);
+        png_set_write_fn(png_ptr, (void*)this, write, 0);
+    }
+
+    void end_png()
+    {
+        png_destroy_write_struct(&png_ptr, &info_ptr);
+    }
+
+    static void write( png_structp png_ptr, png_bytep data, png_size_t length)
+    {
+        QAnimationWriterMNG* that = (QAnimationWriterMNG*)png_get_io_ptr(png_ptr);
+        /*uint nw =*/ that->dev->write((const char*)data,length);
+    }
+
+    void writePNG(const QImage& image)
+    {
+#ifndef QT_LINUXBASE
+        // LSB disallows accessing the info_ptr directly. LSB's png_set_IHDR sets
+        // the channels anyways, so just comment it out for LSB usage
+        info_ptr->channels = 4;
+#endif
+        png_set_sig_bytes(png_ptr, 8); // Pretend we already wrote the sig
+        png_set_IHDR(png_ptr, info_ptr, image.width(), image.height(),
+                     8, image.hasAlphaChannel()
+                     ? PNG_COLOR_TYPE_RGB_ALPHA : PNG_COLOR_TYPE_RGB,
+                     0, 0, 0);
+        png_write_info(png_ptr, info_ptr);
+        if (!image.hasAlphaChannel())
+            png_set_filler(png_ptr, 0,
+                           QSysInfo::ByteOrder == QSysInfo::BigEndian ?
+                           PNG_FILLER_BEFORE : PNG_FILLER_AFTER);
+        //if ( QImage::systemByteOrder() == QImage::BigEndian ) {
+        //png_set_swap_alpha(png_ptr);
+        //}
+        if (QSysInfo::ByteOrder == QSysInfo::LittleEndian) {
+            png_set_bgr(png_ptr);
+        }
+
+        int height = image.height();
+        png_bytep *row_pointers = new png_bytep[height];
+        for (int i = 0; i < height; ++i)
+            row_pointers[i] = (png_bytep)image.scanLine(i);
+        png_write_image(png_ptr, row_pointers);
+        delete [] row_pointers;
+        png_write_end(png_ptr, info_ptr);
+        end_png();
+        begin_png();
+    }
+
+    void writeMHDR(const QSize& size, int framerate)
+    {
+        dev->write("\212MNG\r\n\032\n", 8);
+
+        struct {
+            int width;
+            int height;
+            int framerate;
+            int a,b,c;
+            int profile;
+        } chunk;
+        chunk.width = htonl(size.width());
+        chunk.height = htonl(size.height());
+        chunk.framerate = htonl(framerate);
+        chunk.a=0;
+        chunk.b=0;
+        chunk.c=0;
+        chunk.profile = htonl(0x00000003);
+
+        png_write_chunk(png_ptr, (png_byte*)"MHDR", (png_byte*)&chunk, sizeof(chunk));
+    }
+
+    void writeMEND()
+    {
+        png_write_chunk(png_ptr, (png_byte*)"MEND", 0, 0);
+    }
+
+    void writeDEFI(const QPoint& offset, const QSize& /*size*/)
+    {
+        struct {
+            ushort o;
+            uchar s;
+            uchar concrete;
+            int x,y;
+            int lc,rc,tc,bc;
+        } chunk;
+        chunk.o=0;
+        chunk.s=0;
+        chunk.concrete=1;
+        chunk.x=htonl(offset.x());
+        chunk.y=htonl(offset.y());
+        chunk.lc=0;
+        chunk.rc=0;
+        chunk.tc=htonl(INT_MAX);
+        chunk.bc=htonl(INT_MAX);
+
+        png_write_chunk(png_ptr, (png_byte*)"DEFI", (png_byte*)&chunk, sizeof(chunk));
+    }
+
+    void writeFRAM(const QSize& size)
+    {
+        struct {
+            uchar mode;
+            uchar n;
+            uchar nu;
+            uchar d;
+            uchar t;
+            uchar clip;
+            uchar s;
+            uchar deltatype;
+            uint left;
+            uint right;
+            uint top;
+            uint bottom;
+        } chunk;
+        chunk.mode=1;
+        chunk.n='a';
+        chunk.nu=0;
+        chunk.d=0;
+        chunk.clip=1;
+        chunk.t=0;
+        chunk.s=0;
+        chunk.deltatype=0;
+        chunk.left=0;
+        chunk.right=htonl(size.width());
+        chunk.top=0;
+        chunk.bottom=htonl(size.height());
+
+        png_write_chunk(png_ptr, (png_byte*)"FRAM", (png_byte*)&chunk, sizeof(chunk));
+    }
+
+    void writeMOVE(const QPoint& offset)
+    {
+        struct {
+            uchar filler[3];
+            uchar z[5];
+            int x,y;
+        } chunk;
+        memset(chunk.z,0,5);
+        chunk.x=htonl(offset.x());
+        chunk.y=htonl(offset.y());
+
+        png_write_chunk(png_ptr, (png_byte*)"MOVE", ((png_byte*)&chunk)+3, sizeof(chunk)-3);
+    }
+
+    void setImage(const QImage& src)
+    {
+        if (first) {
+            first = false;
+            writeMHDR(src.size(),framerate);
+        }
+        composeImage(src,QPoint(0,0));
+    }
+
+    bool canCompose() const { return true; }
+
+    void composeImage(const QImage& src, const QPoint& offset)
+    {
+        writeMOVE(offset);
+        //writeFRAM(src.size());
+        writePNG(src);
+    }
+};
+
+QAnimationWriter::QAnimationWriter(const QString& filename, const char* format)
+{
+    if (qstrncmp(format, "MNG", 4)) {
+	qWarning("Format \"%s\" not supported, only MNG", format);
+	dev = 0;
+	d = 0;
+    } else {
+	QFile *f = new QFile(filename);
+	f->open(QIODevice::WriteOnly);
+	dev = f;
+	d = new QAnimationWriterMNG(dev);
+    }
+}
+
+bool QAnimationWriter::okay() const
+{
+    if (!dev)
+        return false;
+    QFile *file = qobject_cast<QFile*>(dev);
+    Q_ASSERT(file);
+    return (file->error() == QFile::NoError);
+}
+
+QAnimationWriter::~QAnimationWriter()
+{
+    delete d;
+    delete dev;
+}
+
+void QAnimationWriter::setFrameRate(int r)
+{
+    if (d)
+        d->setFrameRate(r);
+}
+
+void QAnimationWriter::appendFrame(const QImage& frm, const QPoint& offset)
+{
+    if (!dev)
+        return;
+
+    const QImage frame = frm.convertToFormat(QImage::Format_RGB32);
+    const int alignx = 1;
+    if (prev.isNull() || !d->canCompose()) {
+        d->setImage(frame);
+    } else {
+        bool done;
+        int minx, maxx, miny, maxy;
+        int w = frame.width();
+        int h = frame.height();
+
+        const quint32 *framePtr = reinterpret_cast<const quint32*>(frame.bits());
+        const quint32 *prevPtr = reinterpret_cast<const quint32*>(prev.bits());
+        const int frameStride = frame.bytesPerLine() / sizeof(quint32);
+        const int prevStride = prev.bytesPerLine() / sizeof(quint32);
+
+        // Find left edge of change
+        done = false;
+        for (minx = 0; minx < w && !done; ++minx) {
+            const quint32 *p1 = framePtr + minx;
+            const quint32 *p2 = prevPtr + minx + offset.x();
+            for (int y = 0; y < h; ++y) {
+                if (*p1 != *p2) {
+                    done = true;
+                    break;
+                }
+                p1 += frameStride;
+                p2 += prevStride;
+            }
+        }
+        --minx;
+
+        // Find right edge of change
+        done = false;
+        for (maxx = w-1; maxx >= 0 && !done; --maxx) {
+            const quint32 *p1 = framePtr + maxx;
+            const quint32 *p2 = prevPtr + maxx + offset.x();
+            for (int y = 0; y < h; ++y) {
+                if (*p1 != *p2) {
+                    done = true;
+                    break;
+                }
+                p1 += frameStride;
+                p2 += prevStride;
+            }
+        }
+        ++maxx;
+
+        // Find top edge of change
+        done = false;
+        for (miny = 0; miny < h && !done; ++miny) {
+            const quint32 *p1 = framePtr + miny * frameStride;
+            const quint32 *p2 = prevPtr + miny * prevStride + offset.x();
+            for (int x = 0; x < w; ++x) {
+                if (*p1 != *p2) {
+                    done = true;
+                    break;
+                }
+                ++p1;
+                ++p2;
+            }
+        }
+        --miny;
+
+        // Find right edge of change
+        done = false;
+        for (maxy = h-1; maxy >= 0 && !done; --maxy) {
+            const quint32 *p1 = framePtr + maxy * frameStride;
+            const quint32 *p2 = prevPtr + maxy * prevStride + offset.x();
+            for (int x = 0; x < w; ++x) {
+                if (*p1 != *p2) {
+                    done = true;
+                    break;
+                }
+                ++p1;
+                ++p2;
+            }
+        }
+        ++maxy;
+
+        if (minx > maxx)
+            minx = maxx = 0;
+        if (miny > maxy)
+            miny = maxy = 0;
+
+        if (alignx > 1) {
+            minx -= minx % alignx;
+            maxx = maxx - maxx % alignx + alignx - 1;
+        }
+
+        int dw = maxx - minx + 1;
+        int dh = maxy - miny + 1;
+
+        QImage diff(dw, dh, QImage::Format_ARGB32);
+
+        int x, y;
+        for (y = 0; y < dh; ++y) {
+            QRgb* li = (QRgb*)frame.scanLine(y+miny)+minx;
+            QRgb* lp = (QRgb*)prev.scanLine(y+miny+offset.y())+minx+offset.x();
+            QRgb* ld = (QRgb*)diff.scanLine(y);
+            if (alignx) {
+                for (x = 0; x < dw; x += alignx) {
+                    int i;
+                    for (i = 0; i < alignx; ++i) {
+                        if (li[x+i] != lp[x+i])
+                            break;
+                    }
+                    if (i == alignx) {
+                        // All the same
+                        for (i = 0; i < alignx; ++i)
+                            ld[x+i] = qRgba(0,0,0,0);
+                    } else {
+                        // Some different
+                        for (i = 0; i < alignx; ++i)
+                            ld[x+i] = 0xff000000 | li[x+i];
+                    }
+                }
+            } else {
+                for (x = 0; x < dw; ++x) {
+                    if (li[x] != lp[x])
+                        ld[x] = 0xff000000 | li[x];
+                    else
+                        ld[x] = qRgba(0,0,0,0);
+                }
+            }
+        }
+
+        d->composeImage(diff, QPoint(minx, miny) + offset);
+    }
+    if (prev.isNull() || (prev.size() == frame.size() && offset == QPoint(0,0))) {
+        prev = frame;
+    } else {
+        QPainter p(&prev);
+        p.drawImage(offset.x(), offset.y(), frame, 0, 0,
+                    frame.width(), frame.height());
+    }
+}
+
+void QAnimationWriter::appendFrame(const QImage& frm)
+{
+    appendFrame(frm, QPoint(0,0));
+}
+
+void QAnimationWriter::appendBlankFrame()
+{
+    QImage i(1,1,QImage::Format_ARGB32);
+    i.fill(0);
+    d->composeImage(i, QPoint(0,0));
+}
+
+QT_END_NAMESPACE