ImageDecoderQt.cpp [plain text]
#include "config.h"
#include "ImageDecoderQt.h"
#include <QtCore/QBuffer>
#include <QtCore/QByteArray>
#include <QtGui/QImageReader>
namespace WebCore {
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());
prepareScaleDataIfNecessary();
if (m_scaled)
m_reader->setScaledSize(scaledSize());
}
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)
{
ImageFrame* const buffer = &m_frameBufferCache[frameIndex];
QSize imageSize = m_reader->scaledSize();
if (imageSize.isEmpty())
imageSize = m_reader->size();
if (!buffer->setSize(imageSize.width(), imageSize.height()))
return false;
QImage image(reinterpret_cast<uchar*>(buffer->getAddr(0, 0)), imageSize.width(), imageSize.height(), sizeof(ImageFrame::PixelData) * imageSize.width(), m_reader->imageFormat());
buffer->setDuration(m_reader->nextImageDelay());
m_reader->read(&image);
if (buffer->premultiplyAlpha()) {
if (image.format() != QImage::Format_ARGB32_Premultiplied)
image = image.convertToFormat(QImage::Format_ARGB32_Premultiplied);
} else {
if (image.format() != QImage::Format_ARGB32)
image = image.convertToFormat(QImage::Format_ARGB32);
}
if (reinterpret_cast<const uchar*>(image.constBits()) != reinterpret_cast<const uchar*>(buffer->getAddr(0, 0))) {
memcpy(buffer->getAddr(0, 0), image.constBits(), image.byteCount());
}
if (image.isNull()) {
frameCount();
repetitionCount();
clearPointers();
return false;
}
buffer->setOriginalFrameRect(image.rect());
buffer->setHasAlpha(image.hasAlphaChannel());
buffer->setStatus(ImageFrame::FrameComplete);
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();
}
PassNativeImagePtr ImageFrame::asNewNativeImage() const
{
QImage::Format format;
if (m_hasAlpha)
format = m_premultiplyAlpha ? QImage::Format_ARGB32_Premultiplied : QImage::Format_ARGB32;
else
format = QImage::Format_RGB32;
QImage img(reinterpret_cast<uchar*>(m_bytes), m_size.width(), m_size.height(), sizeof(PixelData) * m_size.width(), format);
return new QPixmap(QPixmap::fromImage(img));
}
}