#include "config.h"
#include "ImageSource.h"
#include "BMPImageDecoder.h"
#include "GIFImageDecoder.h"
#include "ICOImageDecoder.h"
#include "JPEGImageDecoder.h"
#include "NotImplemented.h"
#include "PNGImageDecoder.h"
#include "SharedBuffer.h"
#include "XBMImageDecoder.h"
#include <wx/defs.h>
#include <wx/bitmap.h>
#if USE(WXGC)
#include <wx/graphics.h>
#endif
#include <wx/image.h>
#include <wx/rawbmp.h>
namespace WebCore {
ImageDecoder* createDecoder(const SharedBuffer& data)
{
int length = data.size();
if (length < 4)
return 0;
const unsigned char* uContents = (const unsigned char*)data.data();
const char* contents = data.data();
if (strncmp(contents, "GIF8", 4) == 0)
return new GIFImageDecoder();
if (uContents[0]==0x89 &&
uContents[1]==0x50 &&
uContents[2]==0x4E &&
uContents[3]==0x47)
return new PNGImageDecoder();
if (uContents[0]==0xFF &&
uContents[1]==0xD8 &&
uContents[2]==0xFF)
return new JPEGImageDecoder();
if (strncmp(contents, "BM", 2) == 0)
return new BMPImageDecoder();
if (!memcmp(contents, "\000\000\001\000", 4) ||
!memcmp(contents, "\000\000\002\000", 4))
return new ICOImageDecoder();
if (length >= 8 && strncmp(contents, "#define ", 8) == 0)
return new XBMImageDecoder();
return 0;
}
ImageSource::ImageSource()
: m_decoder(0)
{}
ImageSource::~ImageSource()
{
clear(true);
}
bool ImageSource::initialized() const
{
return m_decoder;
}
void ImageSource::setData(SharedBuffer* data, bool allDataReceived)
{
m_decoder = createDecoder(*data);
if (!m_decoder)
return;
m_decoder->setData(data, allDataReceived);
}
bool ImageSource::isSizeAvailable()
{
if (!m_decoder)
return false;
return m_decoder->isSizeAvailable();
}
IntSize ImageSource::size() const
{
if (!m_decoder)
return IntSize();
return m_decoder->size();
}
IntSize ImageSource::frameSizeAtIndex(size_t) const
{
return size();
}
int ImageSource::repetitionCount()
{
if (!m_decoder)
return cAnimationNone;
return m_decoder->repetitionCount();
}
String ImageSource::filenameExtension() const
{
notImplemented();
return String();
}
size_t ImageSource::frameCount() const
{
return m_decoder ? m_decoder->frameCount() : 0;
}
bool ImageSource::frameIsCompleteAtIndex(size_t index)
{
return (m_decoder && m_decoder->frameBufferAtIndex(index) != 0);
}
void ImageSource::clear(bool destroyAll, size_t clearBeforeFrame, SharedBuffer* data, bool allDataReceived)
{
if (!destroyAll) {
if (m_decoder)
m_decoder->clearFrameBufferCache(clearBeforeFrame);
return;
}
delete m_decoder;
m_decoder = 0;
if (data)
setData(data, allDataReceived);
}
NativeImagePtr ImageSource::createFrameAtIndex(size_t index)
{
if (!m_decoder)
return 0;
RGBA32Buffer* buffer = m_decoder->frameBufferAtIndex(index);
if (!buffer || buffer->status() == RGBA32Buffer::FrameEmpty)
return 0;
IntRect imageRect = buffer->rect();
unsigned char* bytes = (unsigned char*)buffer->bytes().data();
long colorSize = buffer->bytes().size();
typedef wxPixelData<wxBitmap, wxAlphaPixelFormat> PixelData;
int width = size().width();
int height = size().height();
wxBitmap* bmp = new wxBitmap(width, height, 32);
PixelData data(*bmp);
int rowCounter = 0;
long pixelCounter = 0;
PixelData::Iterator p(data);
PixelData::Iterator rowStart = p;
for (long i = 0; i < buffer->bytes().size()*4; i+=4) {
p.Red() = bytes[i+2];
p.Green() = bytes[i+1];
p.Blue() = bytes[i+0];
p.Alpha() = bytes[i+3];
p++;
pixelCounter++;
if ( (pixelCounter % width ) == 0 ) {
rowCounter++;
p = rowStart;
p.MoveTo(data, 0, rowCounter);
}
}
#if !wxCHECK_VERSION(2,9,0)
bmp->UseAlpha();
#endif
ASSERT(bmp->IsOk());
#if USE(WXGC)
wxGraphicsBitmap* bitmap = new wxGraphicsBitmap(wxGraphicsRenderer::GetDefaultRenderer()->CreateBitmap(*bmp));
delete bmp;
return bitmap;
#else
return bmp;
#endif
}
float ImageSource::frameDurationAtIndex(size_t index)
{
if (!m_decoder)
return 0;
RGBA32Buffer* buffer = m_decoder->frameBufferAtIndex(index);
if (!buffer || buffer->status() == RGBA32Buffer::FrameEmpty)
return 0;
float duration = buffer->duration() / 1000.0f;
if (duration < 0.051f)
return 0.100f;
return duration;
}
bool ImageSource::frameHasAlphaAtIndex(size_t index)
{
if (!m_decoder || !m_decoder->supportsAlpha())
return false;
RGBA32Buffer* buffer = m_decoder->frameBufferAtIndex(index);
if (!buffer || buffer->status() == RGBA32Buffer::FrameEmpty)
return false;
return buffer->hasAlpha();
}
}