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, ImageSource::AlphaOption alphaOption, ImageSource::GammaAndColorProfileOption gammaAndColorProfileOption)
{
if (data.size() < 4)
return 0;
return new ImageDecoderQt(alphaOption, gammaAndColorProfileOption);
}
ImageDecoderQt::ImageDecoderQt(ImageSource::AlphaOption alphaOption, ImageSource::GammaAndColorProfileOption gammaAndColorProfileOption)
: ImageDecoder(alphaOption, gammaAndColorProfileOption)
, 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 = adoptPtr(new QBuffer);
m_buffer->setData(imageData);
m_buffer->open(QIODevice::ReadOnly | QIODevice::Unbuffered);
m_reader = adoptPtr(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);
for (size_t i = 0; i < m_frameBufferCache.size(); ++i)
m_frameBufferCache[i].setPremultiplyAlpha(m_premultiplyAlpha);
}
} else {
m_frameBufferCache.resize(1);
m_frameBufferCache[0].setPremultiplyAlpha(m_premultiplyAlpha);
}
}
return m_frameBufferCache.size();
}
int ImageDecoderQt::repetitionCount() const
{
if (m_reader && m_reader->supportsAnimation())
m_repetitionCount = m_reader->loopCount();
return m_repetitionCount;
}
String ImageDecoderQt::filenameExtension() const
{
return String(m_format.constData(), m_format.length());
};
ImageFrame* ImageDecoderQt::frameBufferAtIndex(size_t index)
{
size_t count = m_frameBufferCache.size();
if (!failed() && !count) {
internalDecodeSize();
count = frameCount();
}
if (index >= count)
return 0;
ImageFrame& frame = m_frameBufferCache[index];
if (frame.status() != ImageFrame::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() != ImageFrame::FrameComplete)
return;
}
clearPointers();
}
bool ImageDecoderQt::internalHandleCurrentImage(size_t frameIndex)
{
QPixmap pixmap = QPixmap::fromImageReader(m_reader.get());
if (pixmap.isNull()) {
frameCount();
repetitionCount();
clearPointers();
return false;
}
ImageFrame* const buffer = &m_frameBufferCache[frameIndex];
buffer->setOriginalFrameRect(m_reader->currentImageRect());
buffer->setStatus(ImageFrame::FrameComplete);
buffer->setDuration(m_reader->nextImageDelay());
buffer->setPixmap(pixmap);
return true;
}
void ImageDecoderQt::forceLoadEverything()
{
int imageCount = 0;
do {
m_frameBufferCache.resize(++imageCount);
} while (internalHandleCurrentImage(imageCount - 1));
m_frameBufferCache.resize(imageCount - 1);
for (size_t i = 0; i < m_frameBufferCache.size(); ++i)
m_frameBufferCache[i].setPremultiplyAlpha(m_premultiplyAlpha);
if (imageCount == 1)
setFailed();
}
void ImageDecoderQt::clearPointers()
{
m_reader.clear();
m_buffer.clear();
}
}