CPLErr WEBPDataset::Uncompress()
{
if (bHasBeenUncompressed)
return eUncompressErrRet;
bHasBeenUncompressed = TRUE;
eUncompressErrRet = CE_Failure;
// To avoid excessive memory allocation attempts
// Normally WebP images are no larger than 16383x16383*4 ~= 1 GB
if( nRasterXSize > INT_MAX / (nRasterYSize * nBands) )
{
CPLError(CE_Failure, CPLE_NotSupported,
"Too large image");
return CE_Failure;
}
pabyUncompressed = reinterpret_cast<GByte*>(
VSIMalloc3(nRasterXSize, nRasterYSize, nBands ) );
if (pabyUncompressed == nullptr)
return CE_Failure;
VSIFSeekL(fpImage, 0, SEEK_END);
vsi_l_offset nSizeLarge = VSIFTellL(fpImage);
if( nSizeLarge != static_cast<vsi_l_offset>( static_cast<uint32_t>( nSizeLarge ) ) )
return CE_Failure;
VSIFSeekL(fpImage, 0, SEEK_SET);
uint32_t nSize = static_cast<uint32_t>( nSizeLarge );
uint8_t* pabyCompressed = reinterpret_cast<uint8_t*>( VSIMalloc(nSize) );
if (pabyCompressed == nullptr)
return CE_Failure;
VSIFReadL(pabyCompressed, 1, nSize, fpImage);
uint8_t* pRet;
if (nBands == 4)
pRet = WebPDecodeRGBAInto(
pabyCompressed,
static_cast<uint32_t>( nSize ),
static_cast<uint8_t*>( pabyUncompressed),
nRasterXSize * nRasterYSize * nBands,
nRasterXSize * nBands );
else
pRet = WebPDecodeRGBInto(
pabyCompressed,
static_cast<uint32_t>( nSize ),
static_cast<uint8_t*>( pabyUncompressed ),
nRasterXSize * nRasterYSize * nBands,
nRasterXSize * nBands );
VSIFree(pabyCompressed);
if (pRet == nullptr)
{
CPLError(CE_Failure, CPLE_AppDefined,
"WebPDecodeRGBInto() failed");
return CE_Failure;
}
eUncompressErrRet = CE_None;
return CE_None;
}
CPLErr WEBPDataset::Uncompress()
{
if (bHasBeenUncompressed)
return CE_None;
bHasBeenUncompressed = TRUE;
VSIFSeekL(fpImage, 0, SEEK_END);
vsi_l_offset nSize = VSIFTellL(fpImage);
if (nSize != (vsi_l_offset)(uint32_t)nSize)
return CE_Failure;
VSIFSeekL(fpImage, 0, SEEK_SET);
uint8_t* pabyCompressed = (uint8_t*)VSIMalloc(nSize);
if (pabyCompressed == NULL)
return CE_Failure;
VSIFReadL(pabyCompressed, 1, nSize, fpImage);
uint8_t* pRet = WebPDecodeRGBInto(pabyCompressed, (uint32_t)nSize,
(uint8_t*)pabyUncompressed, nRasterXSize * nRasterYSize * 3,
nRasterXSize * 3);
VSIFree(pabyCompressed);
if (pRet == NULL)
{
CPLError(CE_Failure, CPLE_AppDefined,
"WebPDecodeRGBInto() failed");
return CE_Failure;
}
return CE_None;
}
Error webp_load_image_from_buffer(Image *p_image, const uint8_t *p_buffer, int p_buffer_len) {
ERR_FAIL_NULL_V(p_image, ERR_INVALID_PARAMETER);
WebPBitstreamFeatures features;
if (WebPGetFeatures(p_buffer, p_buffer_len, &features) != VP8_STATUS_OK) {
// ERR_EXPLAIN("Error decoding WEBP image");
ERR_FAIL_V(ERR_FILE_CORRUPT);
}
PoolVector<uint8_t> dst_image;
int datasize = features.width * features.height * (features.has_alpha ? 4 : 3);
dst_image.resize(datasize);
PoolVector<uint8_t>::Write dst_w = dst_image.write();
bool errdec = false;
if (features.has_alpha) {
errdec = WebPDecodeRGBAInto(p_buffer, p_buffer_len, dst_w.ptr(), datasize, 4 * features.width) == NULL;
} else {
errdec = WebPDecodeRGBInto(p_buffer, p_buffer_len, dst_w.ptr(), datasize, 3 * features.width) == NULL;
}
dst_w = PoolVector<uint8_t>::Write();
//ERR_EXPLAIN("Error decoding webp!");
ERR_FAIL_COND_V(errdec, ERR_FILE_CORRUPT);
p_image->create(features.width, features.height, 0, features.has_alpha ? Image::FORMAT_RGBA8 : Image::FORMAT_RGB8, dst_image);
return OK;
}
CPLErr WEBPDataset::Uncompress()
{
if (bHasBeenUncompressed)
return eUncompressErrRet;
bHasBeenUncompressed = TRUE;
eUncompressErrRet = CE_Failure;
pabyUncompressed = reinterpret_cast<GByte*>(
VSIMalloc3(nRasterXSize, nRasterYSize, nBands ) );
if (pabyUncompressed == NULL)
return CE_Failure;
VSIFSeekL(fpImage, 0, SEEK_END);
vsi_l_offset nSizeLarge = VSIFTellL(fpImage);
if( nSizeLarge != static_cast<vsi_l_offset>( static_cast<uint32_t>( nSizeLarge ) ) )
return CE_Failure;
VSIFSeekL(fpImage, 0, SEEK_SET);
uint32_t nSize = static_cast<uint32_t>( nSizeLarge );
uint8_t* pabyCompressed = reinterpret_cast<uint8_t*>( VSIMalloc(nSize) );
if (pabyCompressed == NULL)
return CE_Failure;
VSIFReadL(pabyCompressed, 1, nSize, fpImage);
uint8_t* pRet;
if (nBands == 4)
pRet = WebPDecodeRGBAInto(
pabyCompressed,
static_cast<uint32_t>( nSize ),
static_cast<uint8_t*>( pabyUncompressed),
nRasterXSize * nRasterYSize * nBands,
nRasterXSize * nBands );
else
pRet = WebPDecodeRGBInto(
pabyCompressed,
static_cast<uint32_t>( nSize ),
static_cast<uint8_t*>( pabyUncompressed ),
nRasterXSize * nRasterYSize * nBands,
nRasterXSize * nBands );
VSIFree(pabyCompressed);
if (pRet == NULL)
{
CPLError(CE_Failure, CPLE_AppDefined,
"WebPDecodeRGBInto() failed");
return CE_Failure;
}
eUncompressErrRet = CE_None;
return CE_None;
}
static Ref<Image> _webp_lossy_unpack(const PoolVector<uint8_t> &p_buffer) {
int size = p_buffer.size() - 4;
ERR_FAIL_COND_V(size <= 0, Ref<Image>());
PoolVector<uint8_t>::Read r = p_buffer.read();
ERR_FAIL_COND_V(r[0] != 'W' || r[1] != 'E' || r[2] != 'B' || r[3] != 'P', Ref<Image>());
WebPBitstreamFeatures features;
if (WebPGetFeatures(&r[4], size, &features) != VP8_STATUS_OK) {
ERR_EXPLAIN("Error unpacking WEBP image:");
ERR_FAIL_V(Ref<Image>());
}
/*
print_line("width: "+itos(features.width));
print_line("height: "+itos(features.height));
print_line("alpha: "+itos(features.has_alpha));
*/
PoolVector<uint8_t> dst_image;
int datasize = features.width * features.height * (features.has_alpha ? 4 : 3);
dst_image.resize(datasize);
PoolVector<uint8_t>::Write dst_w = dst_image.write();
bool errdec = false;
if (features.has_alpha) {
errdec = WebPDecodeRGBAInto(&r[4], size, dst_w.ptr(), datasize, 4 * features.width) == NULL;
} else {
errdec = WebPDecodeRGBInto(&r[4], size, dst_w.ptr(), datasize, 3 * features.width) == NULL;
}
//ERR_EXPLAIN("Error decoding webp! - "+p_file);
ERR_FAIL_COND_V(errdec, Ref<Image>());
dst_w = PoolVector<uint8_t>::Write();
Ref<Image> img = memnew(Image(features.width, features.height, 0, features.has_alpha ? Image::FORMAT_RGBA8 : Image::FORMAT_RGB8, dst_image));
return img;
}
bool WEBPImageDecoder::decode(bool onlySize)
{
// Minimum number of bytes needed to ensure one can parse size information.
static const size_t sizeOfHeader = 30;
// Number of bytes per pixel.
static const int bytesPerPixel = 3;
if (failed())
return false;
const size_t dataSize = m_data->size();
if (dataSize < sizeOfHeader)
return true;
int width, height;
const uint8_t* dataBytes = reinterpret_cast<const uint8_t*>(m_data->data());
if (!WebPGetInfo(dataBytes, dataSize, &width, &height))
return setFailed();
if (!ImageDecoder::isSizeAvailable() && !setSize(width, height))
return setFailed();
if (onlySize)
return true;
bool allDataReceived = isAllDataReceived();
int stride = width * bytesPerPixel;
ASSERT(!m_frameBufferCache.isEmpty());
ImageFrame& buffer = m_frameBufferCache[0];
if (buffer.status() == ImageFrame::FrameEmpty) {
ASSERT(width == size().width());
ASSERT(height == size().height());
if (!buffer.setSize(width, height))
return setFailed();
buffer.setStatus(allDataReceived ? ImageFrame::FrameComplete : ImageFrame::FramePartial);
// FIXME: We currently hard code false below because libwebp doesn't support alpha yet.
buffer.setHasAlpha(false);
buffer.setOriginalFrameRect(IntRect(IntPoint(), size()));
m_rgbOutput.resize(height * stride);
}
int newLastVisibleRow = 0; // Last completed row.
if (allDataReceived) {
if (!WebPDecodeRGBInto(dataBytes, dataSize, m_rgbOutput.data(), m_rgbOutput.size(), stride))
return setFailed();
newLastVisibleRow = height;
} else {
if (!m_decoder) {
m_decoder = WebPINewRGB(MODE_RGB, m_rgbOutput.data(), m_rgbOutput.size(), stride);
if (!m_decoder)
return setFailed();
}
const VP8StatusCode status = WebPIUpdate(m_decoder, dataBytes, dataSize);
if (status != VP8_STATUS_OK && status != VP8_STATUS_SUSPENDED)
return setFailed();
if (!WebPIDecGetRGB(m_decoder, &newLastVisibleRow, 0, 0, 0))
return setFailed();
ASSERT(newLastVisibleRow >= 0);
ASSERT(newLastVisibleRow <= height);
}
// FIXME: remove this data copy.
for (int y = m_lastVisibleRow; y < newLastVisibleRow; ++y) {
const uint8_t* const src = &m_rgbOutput[y * stride];
for (int x = 0; x < width; ++x)
buffer.setRGBA(x, y, src[bytesPerPixel * x + 0], src[bytesPerPixel * x + 1], src[bytesPerPixel * x + 2], 0xff);
}
m_lastVisibleRow = newLastVisibleRow;
if (m_lastVisibleRow == height)
buffer.setStatus(ImageFrame::FrameComplete);
return m_lastVisibleRow == height;
}
int ImageFormatWEBP::load(const char *name, int& width, int& height, int& format) {
printf("ImageFormatWEBP::load\n");
// Open File
FILE* file = fopen(name,"rb");
if (file == 0) {
printf("ImageFormatWEBP::load(): can't open file '%s'\n",name);
return 0;
}
// init width / height / format value to 0
width = 0;
height = 0;
format = 0;
// size of file
fseek(file,0,SEEK_END);
const long data_size = ftell(file);
fseek(file,0,SEEK_SET);
unsigned char * img_data = new unsigned char[data_size];
fread(img_data,data_size,1,file);
fclose(file);
// Get header info
int webpinfo = WebPGetInfo(img_data, data_size, &width, &height);
if (webpinfo == 0) {
printf("ImageFormatWEBP::load(): wrong header in \"%s\" file\n",name);
delete img_data;
return 0;
}
// Get betstream
WebPBitstreamFeatures features;
VP8StatusCode status = WebPGetFeatures(img_data, data_size, &features);
if (status != VP8_STATUS_OK) {
printf("ImageFormatWEBP::load(): bad bitstream in \"%s\" file, error:\"%d\"\n",name,status);
delete img_data;
return 0;
}
format = features.has_alpha ? 4 : 3;
int data_decoded_size = width * height * format;
// raw data image container
unsigned char * img_data_decoded = new unsigned char[data_decoded_size];
memset(img_data_decoded,0x00,data_decoded_size);
// decode image inside container (RGB and RGBA)
if (format == 3 ) {
if ( WebPDecodeRGBInto(img_data, data_size, img_data_decoded, data_decoded_size, width * format) == NULL ) {
printf("ImageFormatWEBP::load(): can't decode RGB \"%s\" file\n",name);
delete img_data;
delete img_data_decoded;
return 0;
}
} else {
if ( WebPDecodeRGBAInto(img_data, data_size, img_data_decoded, data_decoded_size, width * format) == NULL ) {
printf("ImageFormatWEBP::load(): can't decode RGBA \"%s\" file\n",name);
delete img_data;
delete img_data_decoded;
return 0;
}
}
// create temporary file for glue code javascript
FILE* raw = fopen(kTempararyName,"wb");
if (raw) {
fwrite(img_data_decoded,data_decoded_size ,1 , raw);
fclose(raw);
}
delete img_data;
delete img_data_decoded;
return 1;
}
