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; }