source: psi/trunk/libpsi/psipng/psipng.cpp@ 189

/*
 * psipng.cpp - QImageFormat for loading Psi PNG animations
 * Copyright (C) 2003  Michail Pishchagin
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

#include "psipng.h"

#include <qasyncimageio.h>

//! if _hide_doc_
class PsiPNGFormat : public QImageFormat {
private:
        int w, h;
        QImage image;
        int numFrames;
        int frame, frameW;
        int frameBytes;

        int maxLen, curLen;

        static const int framePeriod;

public:
        PsiPNGFormat()
        {
        }

        ~PsiPNGFormat()
        {
        }

        int decode(QImage &img, QImageConsumer *consumer, const uchar *buffer, int length)
        {
                if ( !length )
                        return 0;

                if ( image.isNull() ) { // cache our image for speedup
                        if ( image.loadFromData( buffer, length ) ) {
                                w = image.width();
                                h = image.height();

                                numFrames = w / h;
                                frameBytes = length / numFrames;
                                frameW = w / numFrames;
                                frame = 0;

                                consumer->setLooping( numFrames > 1 ? 0 : 1 );
                                consumer->setFramePeriod(framePeriod);
                                consumer->setSize(frameW, h);

                                maxLen = length;
                                curLen = 0;
                        }
                        else {
                                if ( !image.isNull() )
                                        qWarning("PsiPNGFormat::decode: WARNING: QImage is not loaded and is NOT null!!!");
                                return 0;
                        }
                }

                if ( frame < numFrames ) {
                        img = image.copy(frame++ * frameW, 0, frameW, h);
                        consumer->frameDone(QPoint(0, 0), QRect(0, 0, frameW, h));

                        if ( frame >= numFrames ) {
                                consumer->end();
                                frameBytes = maxLen - curLen;
                        }
                        else
                                consumer->setFramePeriod(framePeriod);
                }

                curLen += frameBytes;
                return frameBytes;
        }
};
//! \endif
const int PsiPNGFormat::framePeriod = 120;

class PsiPNGFormatType : public QImageFormatType
{
        QImageFormat *decoderFor(const uchar *buffer, int length)
        {
                if (length < 8)
                        return 0;

                if (buffer[0]==137
                 && buffer[1]=='P'
                 && buffer[2]=='N'
                 && buffer[3]=='G'
                 && buffer[4]==13
                 && buffer[5]==10
                 && buffer[6]==26
                 && buffer[7]==10)
                        return new PsiPNGFormat;

                return 0;
        }

        const char *formatName() const
        {
                return "PsiPNG";
        }
};

PsiPNGFormatType *globalPsiPngFormatTypeObject = 0;

void cleanupPsiPngIO()
{
        if ( globalPsiPngFormatTypeObject ) {
                delete globalPsiPngFormatTypeObject;
                globalPsiPngFormatTypeObject = 0;
        }
}

/*!
        Call this function to register PsiPNG animation format.

        In this format, all animation frames are stored in one .png file.
        <tt>Frame width = Frame height</tt>, and
        <tt>total .png width = tatal .png height * numFrames</tt>.
*/
void initPsiPngIO()
{
        static bool done = FALSE;
        if ( !done ) {
                done = TRUE;
                globalPsiPngFormatTypeObject = new PsiPNGFormatType;
                qAddPostRoutine( cleanupPsiPngIO );
        }
}