Fix saving RGBA images
PNG_FORMAT_RGB and PNG_COLOR_TYPE_RGB both evaluate to 2, but PNG_FORMAT_RGBA evaluates to 3 while PNG_COLOR_TYPE_RGBA evaluates to 6; the bit indicating a palette is 1 while the bit indicating alpha is 4.
This commit is contained in:
parent
99e589cc98
commit
8cf841ecc7
|
@ -68,18 +68,19 @@ bool SavePNG(const std::string& path, const u8* input, ImageByteFormat format, u
|
||||||
timer.Start();
|
timer.Start();
|
||||||
|
|
||||||
size_t byte_per_pixel;
|
size_t byte_per_pixel;
|
||||||
int png_format;
|
int color_type;
|
||||||
switch (format)
|
switch (format)
|
||||||
{
|
{
|
||||||
case ImageByteFormat::RGB:
|
case ImageByteFormat::RGB:
|
||||||
png_format = PNG_FORMAT_RGB;
|
color_type = PNG_COLOR_TYPE_RGB;
|
||||||
byte_per_pixel = 3;
|
byte_per_pixel = 3;
|
||||||
break;
|
break;
|
||||||
case ImageByteFormat::RGBA:
|
case ImageByteFormat::RGBA:
|
||||||
png_format = PNG_FORMAT_RGBA;
|
color_type = PNG_COLOR_TYPE_RGBA;
|
||||||
byte_per_pixel = 4;
|
byte_per_pixel = 4;
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
|
ASSERT_MSG(FRAMEDUMP, false, "Invalid format %d", static_cast<int>(format));
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -110,7 +111,7 @@ bool SavePNG(const std::string& path, const u8* input, ImageByteFormat format, u
|
||||||
bool success = false;
|
bool success = false;
|
||||||
if (png_ptr != nullptr && info_ptr != nullptr)
|
if (png_ptr != nullptr && info_ptr != nullptr)
|
||||||
{
|
{
|
||||||
success = SavePNG0(png_ptr, info_ptr, png_format, width, height, level, &buffer, WriteCallback,
|
success = SavePNG0(png_ptr, info_ptr, color_type, width, height, level, &buffer, WriteCallback,
|
||||||
const_cast<u8**>(rows.data()));
|
const_cast<u8**>(rows.data()));
|
||||||
}
|
}
|
||||||
png_destroy_write_struct(&png_ptr, &info_ptr);
|
png_destroy_write_struct(&png_ptr, &info_ptr);
|
||||||
|
|
|
@ -9,7 +9,7 @@
|
||||||
// The main purpose of this function is to allow specifying the compression level, which
|
// The main purpose of this function is to allow specifying the compression level, which
|
||||||
// png_image_write_to_memory does not allow. row_pointers is not modified by libpng, but also isn't
|
// png_image_write_to_memory does not allow. row_pointers is not modified by libpng, but also isn't
|
||||||
// const for some reason.
|
// const for some reason.
|
||||||
bool SavePNG0(png_structp png_ptr, png_infop info_ptr, int png_format, png_uint_32 width,
|
bool SavePNG0(png_structp png_ptr, png_infop info_ptr, int color_type, png_uint_32 width,
|
||||||
png_uint_32 height, int level, png_voidp io_ptr, png_rw_ptr write_fn,
|
png_uint_32 height, int level, png_voidp io_ptr, png_rw_ptr write_fn,
|
||||||
png_bytepp row_pointers)
|
png_bytepp row_pointers)
|
||||||
{
|
{
|
||||||
|
@ -17,7 +17,7 @@ bool SavePNG0(png_structp png_ptr, png_infop info_ptr, int png_format, png_uint_
|
||||||
return false;
|
return false;
|
||||||
|
|
||||||
png_set_compression_level(png_ptr, level);
|
png_set_compression_level(png_ptr, level);
|
||||||
png_set_IHDR(png_ptr, info_ptr, width, height, 8, png_format, PNG_INTERLACE_NONE,
|
png_set_IHDR(png_ptr, info_ptr, width, height, 8, color_type, PNG_INTERLACE_NONE,
|
||||||
PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
|
PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
|
||||||
png_set_rows(png_ptr, info_ptr, row_pointers);
|
png_set_rows(png_ptr, info_ptr, row_pointers);
|
||||||
png_set_write_fn(png_ptr, io_ptr, write_fn, NULL);
|
png_set_write_fn(png_ptr, io_ptr, write_fn, NULL);
|
||||||
|
|
|
@ -17,7 +17,7 @@ struct ErrorHandler
|
||||||
void (*StoreWarning)(struct ErrorHandler* self, const char* msg);
|
void (*StoreWarning)(struct ErrorHandler* self, const char* msg);
|
||||||
};
|
};
|
||||||
|
|
||||||
bool SavePNG0(png_structp png_ptr, png_infop info_ptr, int png_format, png_uint_32 width,
|
bool SavePNG0(png_structp png_ptr, png_infop info_ptr, int color_type, png_uint_32 width,
|
||||||
png_uint_32 height, int level, png_voidp io_ptr, png_rw_ptr write_fn,
|
png_uint_32 height, int level, png_voidp io_ptr, png_rw_ptr write_fn,
|
||||||
png_bytepp row_pointers);
|
png_bytepp row_pointers);
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue