mirror of
https://github.com/shadps4-emu/shadPS4.git
synced 2025-05-29 14:53:18 +00:00
8c9d7d1a08
* intial try to include libpng * fixing libpng cmake * cleanup structs and error codes * building libpng , destroying pkg ;/ * fixed pkg with zlib_comp mode * attemp to fix ci * rewrote png encoder with libpng * small corrections * clang fix * clang-fix? * take alpha value from decode parameters * more cleanup * fix stride calculation * libpng: avoid unnecessary allocation in decoding * libpng: interlaced support * libpng: lowered log level * revert wrong merge --------- Co-authored-by: Vinicius Rangel <me@viniciusrangel.dev>
282 lines
No EOL
10 KiB
C++
282 lines
No EOL
10 KiB
C++
// SPDX-FileCopyrightText: Copyright 2024 shadPS4 Emulator Project
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
#include <png.h>
|
|
#include "common/assert.h"
|
|
#include "common/logging/log.h"
|
|
#include "core/libraries/error_codes.h"
|
|
#include "core/libraries/libs.h"
|
|
#include "pngdec.h"
|
|
|
|
namespace Libraries::PngDec {
|
|
|
|
struct PngHandler {
|
|
png_structp png_ptr;
|
|
png_infop info_ptr;
|
|
};
|
|
|
|
static inline OrbisPngDecColorSpace MapPngColor(int color) {
|
|
switch (color) {
|
|
case PNG_COLOR_TYPE_GRAY:
|
|
return OrbisPngDecColorSpace::ORBIS_PNG_DEC_COLOR_SPACE_GRAYSCALE;
|
|
|
|
case PNG_COLOR_TYPE_GRAY_ALPHA:
|
|
return OrbisPngDecColorSpace::ORBIS_PNG_DEC_COLOR_SPACE_GRAYSCALE_ALPHA;
|
|
|
|
case PNG_COLOR_TYPE_PALETTE:
|
|
return OrbisPngDecColorSpace::ORBIS_PNG_DEC_COLOR_SPACE_CLUT;
|
|
|
|
case PNG_COLOR_TYPE_RGB:
|
|
return OrbisPngDecColorSpace::ORBIS_PNG_DEC_COLOR_SPACE_RGB;
|
|
|
|
case PNG_COLOR_TYPE_RGB_ALPHA:
|
|
return OrbisPngDecColorSpace::ORBIS_PNG_DEC_COLOR_SPACE_RGBA;
|
|
}
|
|
|
|
UNREACHABLE_MSG("unknown png color type");
|
|
}
|
|
|
|
void PngDecError(png_structp png_ptr, png_const_charp error_message) {
|
|
LOG_ERROR(Lib_Png, "PNG error {}", error_message);
|
|
}
|
|
|
|
void PngDecWarning(png_structp png_ptr, png_const_charp error_message) {
|
|
LOG_ERROR(Lib_Png, "PNG warning {}", error_message);
|
|
}
|
|
|
|
s32 PS4_SYSV_ABI scePngDecCreate(const OrbisPngDecCreateParam* param, void* memoryAddress,
|
|
u32 memorySize, OrbisPngDecHandle* handle) {
|
|
if (param == nullptr || param->attribute > 1) {
|
|
LOG_ERROR(Lib_Png, "Invalid param!");
|
|
return ORBIS_PNG_DEC_ERROR_INVALID_PARAM;
|
|
}
|
|
if (memoryAddress == nullptr) {
|
|
LOG_ERROR(Lib_Png, "Invalid memory address!");
|
|
return ORBIS_PNG_DEC_ERROR_INVALID_ADDR;
|
|
}
|
|
if (param->maxImageWidth - 1 > 1000000) {
|
|
LOG_ERROR(Lib_Png, "Invalid size! width = {}", param->maxImageWidth);
|
|
return ORBIS_PNG_DEC_ERROR_INVALID_SIZE;
|
|
}
|
|
auto pngh = (PngHandler*)memoryAddress;
|
|
|
|
pngh->png_ptr =
|
|
png_create_read_struct(PNG_LIBPNG_VER_STRING, nullptr, PngDecError, PngDecWarning);
|
|
|
|
if (pngh->png_ptr == nullptr)
|
|
return ORBIS_PNG_DEC_ERROR_FATAL;
|
|
|
|
pngh->info_ptr = png_create_info_struct(pngh->png_ptr);
|
|
if (pngh->info_ptr == nullptr) {
|
|
png_destroy_read_struct(&pngh->png_ptr, nullptr, nullptr);
|
|
return false;
|
|
}
|
|
|
|
*handle = pngh;
|
|
return ORBIS_OK;
|
|
}
|
|
|
|
s32 PS4_SYSV_ABI scePngDecDecode(OrbisPngDecHandle handle, const OrbisPngDecDecodeParam* param,
|
|
OrbisPngDecImageInfo* imageInfo) {
|
|
if (handle == nullptr) {
|
|
LOG_ERROR(Lib_Png, "invalid handle!");
|
|
return ORBIS_PNG_DEC_ERROR_INVALID_HANDLE;
|
|
}
|
|
if (param == nullptr) {
|
|
LOG_ERROR(Lib_Png, "Invalid param!");
|
|
return ORBIS_PNG_DEC_ERROR_INVALID_PARAM;
|
|
}
|
|
if (param->pngMemAddr == nullptr || param->pngMemAddr == nullptr) {
|
|
LOG_ERROR(Lib_Png, "invalid image address!");
|
|
return ORBIS_PNG_DEC_ERROR_INVALID_ADDR;
|
|
}
|
|
LOG_TRACE(Lib_Png,
|
|
"pngMemSize = {} , imageMemSize = {} , pixelFormat = {} , alphaValue = {} , "
|
|
"imagePitch = {}",
|
|
param->pngMemSize, param->imageMemSize, param->pixelFormat, param->alphaValue,
|
|
param->imagePitch);
|
|
|
|
auto pngh = (PngHandler*)handle;
|
|
|
|
struct pngstruct {
|
|
const u8* data;
|
|
size_t size;
|
|
u64 offset;
|
|
} pngdata = {
|
|
.data = (const u8*)param->pngMemAddr,
|
|
.size = param->pngMemSize,
|
|
.offset = 0,
|
|
};
|
|
|
|
// Read png from memory
|
|
png_set_read_fn(pngh->png_ptr, (void*)&pngdata,
|
|
[](png_structp ps, png_bytep data, png_size_t len) {
|
|
if (len == 0)
|
|
return;
|
|
auto pngdata = (pngstruct*)png_get_io_ptr(ps);
|
|
::memcpy(data, pngdata->data + pngdata->offset, len);
|
|
pngdata->offset += len;
|
|
});
|
|
|
|
u32 width, height;
|
|
int color_type, bit_depth;
|
|
png_read_info(pngh->png_ptr, pngh->info_ptr);
|
|
|
|
width = png_get_image_width(pngh->png_ptr, pngh->info_ptr);
|
|
height = png_get_image_height(pngh->png_ptr, pngh->info_ptr);
|
|
color_type = MapPngColor(png_get_color_type(pngh->png_ptr, pngh->info_ptr));
|
|
bit_depth = png_get_bit_depth(pngh->png_ptr, pngh->info_ptr);
|
|
|
|
if (imageInfo != nullptr) {
|
|
imageInfo->bitDepth = bit_depth;
|
|
imageInfo->imageWidth = width;
|
|
imageInfo->imageHeight = height;
|
|
imageInfo->colorSpace = color_type;
|
|
imageInfo->imageFlag = 0;
|
|
if (png_get_interlace_type(pngh->png_ptr, pngh->info_ptr) == 1) {
|
|
imageInfo->imageFlag |= OrbisPngDecImageFlag::ORBIS_PNG_DEC_IMAGE_FLAG_ADAM7_INTERLACE;
|
|
}
|
|
if (png_get_valid(pngh->png_ptr, pngh->info_ptr, PNG_INFO_tRNS)) {
|
|
|
|
imageInfo->imageFlag |= ORBIS_PNG_DEC_IMAGE_FLAG_TRNS_CHUNK_EXIST;
|
|
}
|
|
}
|
|
|
|
if (bit_depth == 16)
|
|
png_set_strip_16(pngh->png_ptr);
|
|
if (color_type == PNG_COLOR_TYPE_PALETTE)
|
|
png_set_palette_to_rgb(pngh->png_ptr);
|
|
if (color_type == PNG_COLOR_TYPE_GRAY && bit_depth < 8)
|
|
png_set_expand_gray_1_2_4_to_8(pngh->png_ptr);
|
|
if (png_get_valid(pngh->png_ptr, pngh->info_ptr, PNG_INFO_tRNS))
|
|
png_set_tRNS_to_alpha(pngh->png_ptr);
|
|
if (color_type == PNG_COLOR_TYPE_GRAY || color_type == PNG_COLOR_TYPE_GRAY_ALPHA)
|
|
png_set_gray_to_rgb(pngh->png_ptr);
|
|
if (param->pixelFormat == OrbisPngDecPixelFormat::ORBIS_PNG_DEC_PIXEL_FORMAT_B8G8R8A8)
|
|
png_set_bgr(pngh->png_ptr);
|
|
if (color_type == PNG_COLOR_TYPE_RGB || color_type == PNG_COLOR_TYPE_GRAY ||
|
|
color_type == PNG_COLOR_TYPE_PALETTE)
|
|
png_set_add_alpha(pngh->png_ptr, param->alphaValue, PNG_FILLER_AFTER);
|
|
|
|
int pass = png_set_interlace_handling(pngh->png_ptr);
|
|
png_read_update_info(pngh->png_ptr, pngh->info_ptr);
|
|
|
|
auto const numChannels = png_get_channels(pngh->png_ptr, pngh->info_ptr);
|
|
auto horizontal_bytes = numChannels * width;
|
|
|
|
int stride = param->imagePitch > 0 ? param->imagePitch : horizontal_bytes;
|
|
|
|
for (int j = 0; j < pass; j++) { // interlaced
|
|
auto ptr = (png_bytep)param->imageMemAddr;
|
|
for (int y = 0; y < height; y++) {
|
|
png_read_row(pngh->png_ptr, ptr, nullptr);
|
|
ptr += stride;
|
|
}
|
|
}
|
|
|
|
return (width > 32767 || height > 32767) ? 0 : (width << 16) | height;
|
|
}
|
|
|
|
s32 PS4_SYSV_ABI scePngDecDecodeWithInputControl() {
|
|
LOG_ERROR(Lib_Png, "(STUBBED)called");
|
|
return ORBIS_OK;
|
|
}
|
|
|
|
s32 PS4_SYSV_ABI scePngDecDelete(OrbisPngDecHandle handle) {
|
|
auto pngh = *(PngHandler**)handle;
|
|
png_destroy_read_struct(&pngh->png_ptr, &pngh->info_ptr, nullptr);
|
|
return ORBIS_OK;
|
|
}
|
|
|
|
s32 PS4_SYSV_ABI scePngDecParseHeader(const OrbisPngDecParseParam* param,
|
|
OrbisPngDecImageInfo* imageInfo) {
|
|
if (param == nullptr) {
|
|
LOG_ERROR(Lib_Png, "Invalid param!");
|
|
return ORBIS_PNG_DEC_ERROR_INVALID_PARAM;
|
|
}
|
|
|
|
u8 header[8];
|
|
memcpy(header, param->pngMemAddr, 8);
|
|
// Check if the header indicates a valid PNG file
|
|
if (png_sig_cmp(header, 0, 8)) {
|
|
LOG_ERROR(Lib_Png, "Memory doesn't contain a valid png file");
|
|
return ORBIS_PNG_DEC_ERROR_INVALID_DATA;
|
|
}
|
|
// Create a libpng structure, also pass our custom error/warning functions
|
|
auto png_ptr =
|
|
png_create_read_struct(PNG_LIBPNG_VER_STRING, nullptr, PngDecError, PngDecWarning);
|
|
|
|
// Create a libpng info structure
|
|
auto info_ptr = png_create_info_struct(png_ptr);
|
|
|
|
struct pngstruct {
|
|
const u8* data;
|
|
size_t size;
|
|
u64 offset;
|
|
} pngdata = {
|
|
.data = (const u8*)param->pngMemAddr,
|
|
.size = param->pngMemSize,
|
|
.offset = 0,
|
|
};
|
|
|
|
png_set_read_fn(png_ptr, (void*)&pngdata, [](png_structp ps, png_bytep data, png_size_t len) {
|
|
auto pngdata = (pngstruct*)png_get_io_ptr(ps);
|
|
::memcpy(data, pngdata->data + pngdata->offset, len);
|
|
pngdata->offset += len;
|
|
});
|
|
|
|
// Now call png_read_info with our pngPtr as image handle, and infoPtr to receive the file
|
|
// info.
|
|
png_read_info(png_ptr, info_ptr);
|
|
|
|
imageInfo->imageWidth = png_get_image_width(png_ptr, info_ptr);
|
|
imageInfo->imageHeight = png_get_image_height(png_ptr, info_ptr);
|
|
imageInfo->colorSpace = MapPngColor(png_get_color_type(png_ptr, info_ptr));
|
|
imageInfo->bitDepth = png_get_bit_depth(png_ptr, info_ptr);
|
|
imageInfo->imageFlag = 0;
|
|
if (png_get_interlace_type(png_ptr, info_ptr) == 1) {
|
|
imageInfo->imageFlag |= OrbisPngDecImageFlag::ORBIS_PNG_DEC_IMAGE_FLAG_ADAM7_INTERLACE;
|
|
}
|
|
if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS)) {
|
|
|
|
imageInfo->imageFlag |= ORBIS_PNG_DEC_IMAGE_FLAG_TRNS_CHUNK_EXIST;
|
|
}
|
|
|
|
png_destroy_read_struct(&png_ptr, &info_ptr, nullptr);
|
|
LOG_TRACE(
|
|
Lib_Png,
|
|
"imageWidth = {} , imageHeight = {} , colorSpace = {} , bitDepth = {} , imageFlag = {}",
|
|
imageInfo->imageWidth, imageInfo->imageHeight, imageInfo->colorSpace, imageInfo->bitDepth,
|
|
imageInfo->imageFlag);
|
|
return ORBIS_OK;
|
|
}
|
|
|
|
s32 PS4_SYSV_ABI scePngDecQueryMemorySize(const OrbisPngDecCreateParam* param) {
|
|
if (param == nullptr) {
|
|
LOG_ERROR(Lib_Png, "Invalid param!");
|
|
return ORBIS_PNG_DEC_ERROR_INVALID_PARAM;
|
|
}
|
|
if (param->attribute > 1) {
|
|
LOG_ERROR(Lib_Png, "Invalid attribute! attribute = {}", param->attribute);
|
|
return ORBIS_PNG_DEC_ERROR_INVALID_ADDR;
|
|
}
|
|
if (param->maxImageWidth - 1 > 1000000) {
|
|
LOG_ERROR(Lib_Png, "Invalid size! width = {}", param->maxImageWidth);
|
|
return ORBIS_PNG_DEC_ERROR_INVALID_SIZE;
|
|
}
|
|
return sizeof(PngHandler);
|
|
}
|
|
|
|
void RegisterlibScePngDec(Core::Loader::SymbolsResolver* sym) {
|
|
LIB_FUNCTION("m0uW+8pFyaw", "libScePngDec", 1, "libScePngDec", 1, 1, scePngDecCreate);
|
|
LIB_FUNCTION("WC216DD3El4", "libScePngDec", 1, "libScePngDec", 1, 1, scePngDecDecode);
|
|
LIB_FUNCTION("cJ--1xAbj-I", "libScePngDec", 1, "libScePngDec", 1, 1,
|
|
scePngDecDecodeWithInputControl);
|
|
LIB_FUNCTION("QbD+eENEwo8", "libScePngDec", 1, "libScePngDec", 1, 1, scePngDecDelete);
|
|
LIB_FUNCTION("U6h4e5JRPaQ", "libScePngDec", 1, "libScePngDec", 1, 1, scePngDecParseHeader);
|
|
LIB_FUNCTION("-6srIGbLTIU", "libScePngDec", 1, "libScePngDec", 1, 1, scePngDecQueryMemorySize);
|
|
LIB_FUNCTION("cJ--1xAbj-I", "libScePngDec_jvm", 1, "libScePngDec", 1, 1,
|
|
scePngDecDecodeWithInputControl);
|
|
};
|
|
|
|
} // namespace Libraries::PngDec
|