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 (m_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(QBuffer::ReadOnly);
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 == 0)
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 = qMax(0, m_reader->loopCount());
return m_repetitionCount;
}
String ImageDecoderQt::filenameExtension() const
{
return String(m_format.constData(), m_format.length());
};
RGBA32Buffer* ImageDecoderQt::frameBufferAtIndex(size_t index)
{
int count = m_frameBufferCache.size();
if (!m_failed && count == 0) {
internalDecodeSize();
count = frameCount();
}
if (index >= static_cast<size_t>(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())
return failRead();
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 != 0)
return failRead();
internalHandleCurrentImage(frameIndex);
for (int i = 0; i < m_frameBufferCache.size(); ++i)
if (m_frameBufferCache[i].status() != RGBA32Buffer::FrameComplete)
return;
m_reader.clear();
m_buffer.clear();
}
void ImageDecoderQt::internalHandleCurrentImage(size_t frameIndex)
{
QImage img;
if (!m_reader->read(&img))
return failRead();
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);
}
void ImageDecoderQt::forceLoadEverything()
{
int imageCount = 0;
do {
m_frameBufferCache.resize(++imageCount);
internalHandleCurrentImage(imageCount - 1);
} while(!m_failed);
m_frameBufferCache.resize(imageCount - 1);
m_failed = imageCount == 1;
}
void ImageDecoderQt::failRead()
{
setFailed();
m_reader.clear();
m_buffer.clear();
}
}