mirror of
https://github.com/thorvg/thorvg.git
synced 2025-06-13 19:44:28 +00:00
3d88d7eefc
Added error string printing on jpg image loading failure. The error message help developer find the corrupted jpg file. Error message is not printed for open from char* data as there the loaders are tried on by one.
153 lines
4.1 KiB
C++
153 lines
4.1 KiB
C++
/*
|
|
* Copyright (c) 2021 Samsung Electronics Co., Ltd. All rights reserved.
|
|
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
|
|
* The above copyright notice and this permission notice shall be included in all
|
|
* copies or substantial portions of the Software.
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
* SOFTWARE.
|
|
*/
|
|
|
|
#include <memory.h>
|
|
#include <turbojpeg.h>
|
|
#include "tvgLoader.h"
|
|
#include "tvgJpgLoader.h"
|
|
|
|
/************************************************************************/
|
|
/* Internal Class Implementation */
|
|
/************************************************************************/
|
|
|
|
void JpgLoader::clear()
|
|
{
|
|
if (freeData) free(data);
|
|
data = nullptr;
|
|
size = 0;
|
|
freeData = false;
|
|
}
|
|
|
|
/************************************************************************/
|
|
/* External Class Implementation */
|
|
/************************************************************************/
|
|
|
|
JpgLoader::JpgLoader()
|
|
{
|
|
jpegDecompressor = tjInitDecompress();
|
|
}
|
|
|
|
|
|
JpgLoader::~JpgLoader()
|
|
{
|
|
if (freeData) free(data);
|
|
tjDestroy(jpegDecompressor);
|
|
|
|
//This image is shared with raster engine.
|
|
tjFree(image);
|
|
}
|
|
|
|
|
|
bool JpgLoader::open(const string& path)
|
|
{
|
|
clear();
|
|
|
|
auto jpegFile = fopen(path.c_str(), "rb");
|
|
if (!jpegFile) return false;
|
|
|
|
auto ret = false;
|
|
|
|
//determine size
|
|
if (fseek(jpegFile, 0, SEEK_END) < 0) goto finalize;
|
|
if (((size = ftell(jpegFile)) < 1)) goto finalize;
|
|
if (fseek(jpegFile, 0, SEEK_SET)) goto finalize;
|
|
|
|
data = (unsigned char *) malloc(size);
|
|
if (!data) goto finalize;
|
|
|
|
if (fread(data, size, 1, jpegFile) < 1) goto failure;
|
|
|
|
int width, height, subSample, colorSpace;
|
|
if (tjDecompressHeader3(jpegDecompressor, data, size, &width, &height, &subSample, &colorSpace) < 0) {
|
|
TVGERR("JPG LOADER", "%s", tjGetErrorStr());
|
|
goto failure;
|
|
}
|
|
|
|
w = static_cast<float>(width);
|
|
h = static_cast<float>(height);
|
|
ret = true;
|
|
freeData = true;
|
|
|
|
goto finalize;
|
|
|
|
failure:
|
|
clear();
|
|
|
|
finalize:
|
|
fclose(jpegFile);
|
|
return ret;
|
|
}
|
|
|
|
|
|
bool JpgLoader::open(const char* data, uint32_t size, bool copy)
|
|
{
|
|
clear();
|
|
|
|
int width, height, subSample, colorSpace;
|
|
if (tjDecompressHeader3(jpegDecompressor, (unsigned char *) data, size, &width, &height, &subSample, &colorSpace) < 0) return false;
|
|
|
|
if (copy) {
|
|
this->data = (unsigned char *) malloc(size);
|
|
if (!this->data) return false;
|
|
memcpy((unsigned char *)this->data, data, size);
|
|
freeData = true;
|
|
} else {
|
|
this->data = (unsigned char *) data;
|
|
}
|
|
|
|
w = static_cast<float>(width);
|
|
h = static_cast<float>(height);
|
|
this->size = size;
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool JpgLoader::read()
|
|
{
|
|
if (image) tjFree(image);
|
|
image = (unsigned char *)tjAlloc(static_cast<int>(w) * static_cast<int>(h) * tjPixelSize[TJPF_BGRX]);
|
|
if (!image) return false;
|
|
|
|
//decompress jpg image
|
|
if (tjDecompress2(jpegDecompressor, data, size, image, static_cast<int>(w), 0, static_cast<int>(h), TJPF_BGRX, 0) < 0) {
|
|
TVGERR("JPG LOADER", "%s", tjGetErrorStr());
|
|
tjFree(image);
|
|
image = nullptr;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool JpgLoader::close()
|
|
{
|
|
clear();
|
|
return true;
|
|
}
|
|
|
|
|
|
const uint32_t* JpgLoader::pixels()
|
|
{
|
|
return (const uint32_t*) image;
|
|
}
|