mirror of https://github.com/OpenRCT2/OpenRCT2.git
Use FileStream instead of SDL2
This commit is contained in:
parent
a21cbae82a
commit
4680aeab9b
|
@ -15,6 +15,8 @@
|
||||||
#pragma endregion
|
#pragma endregion
|
||||||
|
|
||||||
#include <png.h>
|
#include <png.h>
|
||||||
|
#include "core/Exception.hpp"
|
||||||
|
#include "core/FileStream.hpp"
|
||||||
#include "core/Guard.hpp"
|
#include "core/Guard.hpp"
|
||||||
#include "core/Memory.hpp"
|
#include "core/Memory.hpp"
|
||||||
|
|
||||||
|
@ -49,76 +51,84 @@ namespace Imaging
|
||||||
}
|
}
|
||||||
|
|
||||||
// Open PNG file
|
// Open PNG file
|
||||||
SDL_RWops * fp = SDL_RWFromFile(path, "rb");
|
try
|
||||||
if (fp == nullptr)
|
|
||||||
{
|
{
|
||||||
return false;
|
auto fs = FileStream(path, FILE_MODE_OPEN);
|
||||||
}
|
|
||||||
|
|
||||||
// Set error handling
|
// Set error handling
|
||||||
if (setjmp(png_jmpbuf(png_ptr)))
|
if (setjmp(png_jmpbuf(png_ptr)))
|
||||||
{
|
|
||||||
png_destroy_read_struct(&png_ptr, &info_ptr, nullptr);
|
|
||||||
SDL_RWclose(fp);
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Setup png reading
|
|
||||||
png_set_read_fn(png_ptr, fp, PngReadData);
|
|
||||||
png_set_sig_bytes(png_ptr, sig_read);
|
|
||||||
|
|
||||||
// To simplify the reading process, convert 4-16 bit data to 24-32 bit data
|
|
||||||
png_read_png(png_ptr, info_ptr, PNG_TRANSFORM_STRIP_16 | PNG_TRANSFORM_PACKING | PNG_TRANSFORM_EXPAND, nullptr);
|
|
||||||
|
|
||||||
// Read header
|
|
||||||
png_uint_32 pngWidth, pngHeight;
|
|
||||||
int bitDepth, colourType, interlaceType;
|
|
||||||
png_get_IHDR(png_ptr, info_ptr, &pngWidth, &pngHeight, &bitDepth, &colourType, &interlaceType, nullptr, nullptr);
|
|
||||||
|
|
||||||
// Read pixels as 32bpp RGBA data
|
|
||||||
png_size_t rowBytes = png_get_rowbytes(png_ptr, info_ptr);
|
|
||||||
png_bytepp rowPointers = png_get_rows(png_ptr, info_ptr);
|
|
||||||
uint8 * pngPixels = Memory::Allocate<uint8>(pngWidth * pngHeight * 4);
|
|
||||||
uint8 * dst = pngPixels;
|
|
||||||
if (colourType == PNG_COLOR_TYPE_RGB)
|
|
||||||
{
|
|
||||||
// 24-bit PNG (no alpha)
|
|
||||||
Guard::Assert(rowBytes == pngWidth * 3, GUARD_LINE);
|
|
||||||
for (png_uint_32 i = 0; i < pngHeight; i++)
|
|
||||||
{
|
{
|
||||||
uint8 * src = rowPointers[i];
|
png_destroy_read_struct(&png_ptr, &info_ptr, nullptr);
|
||||||
for (png_uint_32 x = 0; x < pngWidth; x++)
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Setup png reading
|
||||||
|
png_set_read_fn(png_ptr, &fs, PngReadData);
|
||||||
|
png_set_sig_bytes(png_ptr, sig_read);
|
||||||
|
|
||||||
|
// To simplify the reading process, convert 4-16 bit data to 24-32 bit data
|
||||||
|
png_read_png(png_ptr, info_ptr, PNG_TRANSFORM_STRIP_16 | PNG_TRANSFORM_PACKING | PNG_TRANSFORM_EXPAND, nullptr);
|
||||||
|
|
||||||
|
// Read header
|
||||||
|
png_uint_32 pngWidth, pngHeight;
|
||||||
|
int bitDepth, colourType, interlaceType;
|
||||||
|
png_get_IHDR(png_ptr, info_ptr, &pngWidth, &pngHeight, &bitDepth, &colourType, &interlaceType, nullptr, nullptr);
|
||||||
|
|
||||||
|
// Read pixels as 32bpp RGBA data
|
||||||
|
png_size_t rowBytes = png_get_rowbytes(png_ptr, info_ptr);
|
||||||
|
png_bytepp rowPointers = png_get_rows(png_ptr, info_ptr);
|
||||||
|
uint8 * pngPixels = Memory::Allocate<uint8>(pngWidth * pngHeight * 4);
|
||||||
|
uint8 * dst = pngPixels;
|
||||||
|
if (colourType == PNG_COLOR_TYPE_RGB)
|
||||||
|
{
|
||||||
|
// 24-bit PNG (no alpha)
|
||||||
|
Guard::Assert(rowBytes == pngWidth * 3, GUARD_LINE);
|
||||||
|
for (png_uint_32 i = 0; i < pngHeight; i++)
|
||||||
{
|
{
|
||||||
*dst++ = *src++;
|
uint8 * src = rowPointers[i];
|
||||||
*dst++ = *src++;
|
for (png_uint_32 x = 0; x < pngWidth; x++)
|
||||||
*dst++ = *src++;
|
{
|
||||||
*dst++ = 255;
|
*dst++ = *src++;
|
||||||
|
*dst++ = *src++;
|
||||||
|
*dst++ = *src++;
|
||||||
|
*dst++ = 255;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else
|
else
|
||||||
{
|
|
||||||
// 32-bit PNG (with alpha)
|
|
||||||
Guard::Assert(rowBytes == pngWidth * 4, GUARD_LINE);
|
|
||||||
for (png_uint_32 i = 0; i < pngHeight; i++)
|
|
||||||
{
|
{
|
||||||
Memory::Copy(dst, rowPointers[i], rowBytes);
|
// 32-bit PNG (with alpha)
|
||||||
dst += rowBytes;
|
Guard::Assert(rowBytes == pngWidth * 4, GUARD_LINE);
|
||||||
|
for (png_uint_32 i = 0; i < pngHeight; i++)
|
||||||
|
{
|
||||||
|
Memory::Copy(dst, rowPointers[i], rowBytes);
|
||||||
|
dst += rowBytes;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Close the PNG
|
||||||
|
png_destroy_read_struct(&png_ptr, &info_ptr, nullptr);
|
||||||
|
|
||||||
|
// Return the output data
|
||||||
|
*pixels = (uint8*)pngPixels;
|
||||||
|
if (width != nullptr) *width = pngWidth;
|
||||||
|
if (height != nullptr) *height = pngHeight;
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
catch (Exception)
|
||||||
|
{
|
||||||
|
*pixels = nullptr;
|
||||||
|
if (width != nullptr) *width = 0;
|
||||||
|
if (height != nullptr) *height = 0;
|
||||||
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Close the PNG
|
|
||||||
png_destroy_read_struct(&png_ptr, &info_ptr, nullptr);
|
|
||||||
SDL_RWclose(fp);
|
|
||||||
|
|
||||||
// Return the output data
|
|
||||||
*pixels = (uint8*)pngPixels;
|
|
||||||
if (width != nullptr) *width = pngWidth;
|
|
||||||
if (height != nullptr) *height = pngHeight;
|
|
||||||
return true;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
bool PngWrite(const rct_drawpixelinfo * dpi, const rct_palette * palette, const utf8 * path)
|
bool PngWrite(const rct_drawpixelinfo * dpi, const rct_palette * palette, const utf8 * path)
|
||||||
{
|
{
|
||||||
|
bool result = false;
|
||||||
|
|
||||||
// Get image size
|
// Get image size
|
||||||
int stride = dpi->width + dpi->pitch;
|
int stride = dpi->width + dpi->pitch;
|
||||||
|
|
||||||
|
@ -147,53 +157,52 @@ namespace Imaging
|
||||||
|
|
||||||
png_set_PLTE(png_ptr, info_ptr, png_palette, PNG_MAX_PALETTE_LENGTH);
|
png_set_PLTE(png_ptr, info_ptr, png_palette, PNG_MAX_PALETTE_LENGTH);
|
||||||
|
|
||||||
// Open file for writing
|
try
|
||||||
SDL_RWops * file = SDL_RWFromFile(path, "wb");
|
|
||||||
if (file == nullptr)
|
|
||||||
{
|
{
|
||||||
png_free(png_ptr, png_palette);
|
// Open file for writing
|
||||||
png_destroy_write_struct(&png_ptr, (png_infopp)nullptr);
|
auto fs = FileStream(path, FILE_MODE_WRITE);
|
||||||
return false;
|
png_set_write_fn(png_ptr, &fs, PngWriteData, PngFlush);
|
||||||
}
|
|
||||||
png_set_write_fn(png_ptr, file, PngWriteData, PngFlush);
|
|
||||||
|
|
||||||
// Set error handler
|
// Set error handler
|
||||||
if (setjmp(png_jmpbuf(png_ptr)))
|
if (setjmp(png_jmpbuf(png_ptr)))
|
||||||
|
{
|
||||||
|
throw Exception("PNG ERROR");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Write header
|
||||||
|
png_set_IHDR(
|
||||||
|
png_ptr, info_ptr, dpi->width, dpi->height, 8,
|
||||||
|
PNG_COLOR_TYPE_PALETTE, PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT
|
||||||
|
);
|
||||||
|
png_byte transparentIndex = 0;
|
||||||
|
png_set_tRNS(png_ptr, info_ptr, &transparentIndex, 1, nullptr);
|
||||||
|
png_write_info(png_ptr, info_ptr);
|
||||||
|
|
||||||
|
// Write pixels
|
||||||
|
uint8 * bits = dpi->bits;
|
||||||
|
for (int y = 0; y < dpi->height; y++)
|
||||||
|
{
|
||||||
|
png_write_row(png_ptr, (png_byte *)bits);
|
||||||
|
bits += stride;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Finish
|
||||||
|
png_write_end(png_ptr, nullptr);
|
||||||
|
result = true;
|
||||||
|
}
|
||||||
|
catch (Exception)
|
||||||
{
|
{
|
||||||
png_free(png_ptr, png_palette);
|
|
||||||
png_destroy_write_struct(&png_ptr, &info_ptr);
|
|
||||||
SDL_RWclose(file);
|
|
||||||
return false;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Write header
|
|
||||||
png_set_IHDR(
|
|
||||||
png_ptr, info_ptr, dpi->width, dpi->height, 8,
|
|
||||||
PNG_COLOR_TYPE_PALETTE, PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT
|
|
||||||
);
|
|
||||||
png_byte transparentIndex = 0;
|
|
||||||
png_set_tRNS(png_ptr, info_ptr, &transparentIndex, 1, nullptr);
|
|
||||||
png_write_info(png_ptr, info_ptr);
|
|
||||||
|
|
||||||
// Write pixels
|
|
||||||
uint8 *bits = dpi->bits;
|
|
||||||
for (int y = 0; y < dpi->height; y++)
|
|
||||||
{
|
|
||||||
png_write_row(png_ptr, (png_byte *)bits);
|
|
||||||
bits += stride;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Finish
|
|
||||||
png_write_end(png_ptr, nullptr);
|
|
||||||
SDL_RWclose(file);
|
|
||||||
|
|
||||||
png_free(png_ptr, png_palette);
|
png_free(png_ptr, png_palette);
|
||||||
png_destroy_write_struct(&png_ptr, &info_ptr);
|
png_destroy_write_struct(&png_ptr, (png_infopp)nullptr);
|
||||||
return true;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool PngWrite32bpp(sint32 width, sint32 height, const void * pixels, const utf8 * path)
|
bool PngWrite32bpp(sint32 width, sint32 height, const void * pixels, const utf8 * path)
|
||||||
{
|
{
|
||||||
|
bool result = false;
|
||||||
|
|
||||||
// Setup PNG
|
// Setup PNG
|
||||||
png_structp png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, nullptr, PngError, PngWarning);
|
png_structp png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, nullptr, PngError, PngWarning);
|
||||||
if (png_ptr == nullptr)
|
if (png_ptr == nullptr)
|
||||||
|
@ -208,56 +217,55 @@ namespace Imaging
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Open file for writing
|
try
|
||||||
SDL_RWops * file = SDL_RWFromFile(path, "wb");
|
|
||||||
if (file == nullptr)
|
|
||||||
{
|
{
|
||||||
png_destroy_write_struct(&png_ptr, (png_infopp)nullptr);
|
// Open file for writing
|
||||||
return false;
|
auto fs = FileStream(path, FILE_MODE_WRITE);
|
||||||
}
|
png_set_write_fn(png_ptr, &fs, PngWriteData, PngFlush);
|
||||||
png_set_write_fn(png_ptr, file, PngWriteData, PngFlush);
|
|
||||||
|
|
||||||
// Set error handler
|
// Set error handler
|
||||||
if (setjmp(png_jmpbuf(png_ptr)))
|
if (setjmp(png_jmpbuf(png_ptr)))
|
||||||
|
{
|
||||||
|
throw Exception("PNG ERROR");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Write header
|
||||||
|
png_set_IHDR(
|
||||||
|
png_ptr, info_ptr, width, height, 8,
|
||||||
|
PNG_COLOR_TYPE_RGB_ALPHA, PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT
|
||||||
|
);
|
||||||
|
png_write_info(png_ptr, info_ptr);
|
||||||
|
|
||||||
|
// Write pixels
|
||||||
|
uint8 * bits = (uint8 *)pixels;
|
||||||
|
for (int y = 0; y < height; y++)
|
||||||
|
{
|
||||||
|
png_write_row(png_ptr, (png_byte *)bits);
|
||||||
|
bits += width * 4;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Finish
|
||||||
|
png_write_end(png_ptr, nullptr);
|
||||||
|
result = true;
|
||||||
|
}
|
||||||
|
catch (Exception)
|
||||||
{
|
{
|
||||||
png_destroy_write_struct(&png_ptr, &info_ptr);
|
|
||||||
SDL_RWclose(file);
|
|
||||||
return false;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Write header
|
|
||||||
png_set_IHDR(
|
|
||||||
png_ptr, info_ptr, width, height, 8,
|
|
||||||
PNG_COLOR_TYPE_RGB_ALPHA, PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT
|
|
||||||
);
|
|
||||||
png_write_info(png_ptr, info_ptr);
|
|
||||||
|
|
||||||
// Write pixels
|
|
||||||
uint8 * bits = (uint8 *)pixels;
|
|
||||||
for (int y = 0; y < height; y++)
|
|
||||||
{
|
|
||||||
png_write_row(png_ptr, (png_byte *)bits);
|
|
||||||
bits += width * 4;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Finish
|
|
||||||
png_write_end(png_ptr, nullptr);
|
|
||||||
SDL_RWclose(file);
|
|
||||||
|
|
||||||
png_destroy_write_struct(&png_ptr, &info_ptr);
|
png_destroy_write_struct(&png_ptr, &info_ptr);
|
||||||
return true;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void PngReadData(png_structp png_ptr, png_bytep data, png_size_t length)
|
static void PngReadData(png_structp png_ptr, png_bytep data, png_size_t length)
|
||||||
{
|
{
|
||||||
SDL_RWops * file = (SDL_RWops *)png_get_io_ptr(png_ptr);
|
auto * fs = static_cast<FileStream *>(png_get_io_ptr(png_ptr));
|
||||||
SDL_RWread(file, data, length, 1);
|
fs->Read(data, length);
|
||||||
}
|
}
|
||||||
|
|
||||||
static void PngWriteData(png_structp png_ptr, png_bytep data, png_size_t length)
|
static void PngWriteData(png_structp png_ptr, png_bytep data, png_size_t length)
|
||||||
{
|
{
|
||||||
SDL_RWops * file = (SDL_RWops *)png_get_io_ptr(png_ptr);
|
auto * fs = static_cast<FileStream *>(png_get_io_ptr(png_ptr));
|
||||||
SDL_RWwrite(file, data, length, 1);
|
fs->Write(data, length);
|
||||||
}
|
}
|
||||||
|
|
||||||
static void PngFlush(png_structp png_ptr)
|
static void PngFlush(png_structp png_ptr)
|
||||||
|
|
Loading…
Reference in New Issue