ImageDecoderQt.cpp [plain text]
#include "config.h"
#include "ImageDecoderQt.h"
#include <QtCore/QByteArray>
#include <QtCore/QBuffer>
#include <QtGui/QImageReader>
#include <qdebug.h>
namespace WebCore {
ImageDecoder* ImageDecoder::create(const SharedBuffer& data)
{
if (data.size() < 4)
return 0;
return new ImageDecoderQt;
}
ImageDecoderQt::ImageDecoderQt()
: m_repetitionCount(cAnimationNone)
{
}
ImageDecoderQt::~ImageDecoderQt()
{
}
void ImageDecoderQt::setData(SharedBuffer* data, bool allDataReceived)
{
if (failed())
return;
if (!allDataReceived)
return;
ImageDecoder::setData(data, allDataReceived);
ASSERT(!m_buffer);
ASSERT(!m_reader);
QByteArray imageData = QByteArray::fromRawData(m_data->data(), m_data->size());
m_buffer.set(new QBuffer);
m_buffer->setData(imageData);
m_buffer->open(QIODevice::ReadOnly | QIODevice::Unbuffered);
m_reader.set(new QImageReader(m_buffer.get(), m_format));
m_reader->setQuality(49);
m_format = m_reader->format();
}
bool ImageDecoderQt::isSizeAvailable()
{
if (!ImageDecoder::isSizeAvailable() && m_reader)
internalDecodeSize();
return ImageDecoder::isSizeAvailable();
}
size_t ImageDecoderQt::frameCount()
{
if (m_frameBufferCache.isEmpty() && m_reader) {
if (m_reader->supportsAnimation()) {
int imageCount = m_reader->imageCount();
if (!imageCount)
forceLoadEverything();
else
m_frameBufferCache.resize(imageCount);
} else
m_frameBufferCache.resize(1);
}
return m_frameBufferCache.size();
}
int ImageDecoderQt::repetitionCount() const
{
if (m_reader && m_reader->supportsAnimation()) {
m_repetitionCount = m_reader->loopCount();
if (m_repetitionCount == -1)
m_repetitionCount = 0;
else if (m_repetitionCount == 0)
m_repetitionCount = cAnimationLoopOnce;
else
++m_repetitionCount;
}
return m_repetitionCount;
}
String ImageDecoderQt::filenameExtension() const
{
return String(m_format.constData(), m_format.length());
};
RGBA32Buffer* ImageDecoderQt::frameBufferAtIndex(size_t index)
{
size_t count = m_frameBufferCache.size();
if (!failed() && !count) {
internalDecodeSize();
count = frameCount();
}
if (index >= count)
return 0;
RGBA32Buffer& frame = m_frameBufferCache[index];
if (frame.status() != RGBA32Buffer::FrameComplete && m_reader)
internalReadImage(index);
return &frame;
}
void ImageDecoderQt::clearFrameBufferCache(size_t )
{
}
void ImageDecoderQt::internalDecodeSize()
{
ASSERT(m_reader);
QSize size = m_reader->size();
if (size.isEmpty()) {
setFailed();
return clearPointers();
}
setSize(size.width(), size.height());
}
void ImageDecoderQt::internalReadImage(size_t frameIndex)
{
ASSERT(m_reader);
if (m_reader->supportsAnimation())
m_reader->jumpToImage(frameIndex);
else if (frameIndex) {
setFailed();
return clearPointers();
}
if (!internalHandleCurrentImage(frameIndex))
setFailed();
for (int i = 0; i < m_frameBufferCache.size(); ++i) {
if (m_frameBufferCache[i].status() != RGBA32Buffer::FrameComplete)
return;
}
clearPointers();
}
bool ImageDecoderQt::internalHandleCurrentImage(size_t frameIndex)
{
QImage img;
if (!m_reader->read(&img)) {
frameCount();
repetitionCount();
clearPointers();
return false;
}
QSize imageSize = img.size();
RGBA32Buffer* const buffer = &m_frameBufferCache[frameIndex];
buffer->setRect(m_reader->currentImageRect());
buffer->setStatus(RGBA32Buffer::FrameComplete);
buffer->setDuration(m_reader->nextImageDelay());
buffer->setDecodedImage(img);
return true;
}
void ImageDecoderQt::forceLoadEverything()
{
int imageCount = 0;
do {
m_frameBufferCache.resize(++imageCount);
} while (internalHandleCurrentImage(imageCount - 1));
m_frameBufferCache.resize(imageCount - 1);
if (imageCount == 1)
setFailed();
}
void ImageDecoderQt::clearPointers()
{
m_reader.clear();
m_buffer.clear();
}
}