mirror of
https://gitlab.gnome.org/GNOME/gtk.git
synced 2024-12-26 13:41:07 +00:00
27fbdcbecb
We need them for mask-only textures. For tiffs, we convert the formats to RGBA (the idea that tiff can save everything needs to be buried I guess) as tiffs can't do alpha-only.
461 lines
11 KiB
C
461 lines
11 KiB
C
/* GDK - The GIMP Drawing Kit
|
|
* Copyright (C) 2021 Red Hat, Inc.
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include "gdkpngprivate.h"
|
|
|
|
#include <glib/gi18n-lib.h>
|
|
#include "gdkmemoryformatprivate.h"
|
|
#include "gdkmemorytexture.h"
|
|
#include "gdkprofilerprivate.h"
|
|
#include "gdktexturedownloaderprivate.h"
|
|
#include "gsk/gl/fp16private.h"
|
|
#include <png.h>
|
|
#include <stdio.h>
|
|
|
|
/* The main difference between the png load/save code here and
|
|
* gdk-pixbuf is that we can support loading 16-bit data in the
|
|
* future, and we can extract gamma and colorspace information
|
|
* to produce linear, color-corrected data.
|
|
*/
|
|
|
|
/* {{{ Callbacks */
|
|
|
|
/* No sigsetjmp on Windows */
|
|
#ifndef HAVE_SIGSETJMP
|
|
#define sigjmp_buf jmp_buf
|
|
#define sigsetjmp(jb, x) setjmp(jb)
|
|
#define siglongjmp longjmp
|
|
#endif
|
|
|
|
typedef struct
|
|
{
|
|
guchar *data;
|
|
gsize size;
|
|
gsize position;
|
|
} png_io;
|
|
|
|
|
|
static void
|
|
png_read_func (png_structp png,
|
|
png_bytep data,
|
|
png_size_t size)
|
|
{
|
|
png_io *io;
|
|
|
|
io = png_get_io_ptr (png);
|
|
|
|
if (io->position + size > io->size)
|
|
png_error (png, "Read past EOF");
|
|
|
|
memcpy (data, io->data + io->position, size);
|
|
io->position += size;
|
|
}
|
|
|
|
static void
|
|
png_write_func (png_structp png,
|
|
png_bytep data,
|
|
png_size_t size)
|
|
{
|
|
png_io *io;
|
|
|
|
io = png_get_io_ptr (png);
|
|
|
|
if (io->position > io->size ||
|
|
io->size - io->position < size)
|
|
{
|
|
io->size = io->position + size;
|
|
io->data = g_realloc (io->data, io->size);
|
|
}
|
|
|
|
memcpy (io->data + io->position, data, size);
|
|
io->position += size;
|
|
}
|
|
|
|
static void
|
|
png_flush_func (png_structp png)
|
|
{
|
|
}
|
|
|
|
static png_voidp
|
|
png_malloc_callback (png_structp o,
|
|
png_size_t size)
|
|
{
|
|
return g_try_malloc (size);
|
|
}
|
|
|
|
static void
|
|
png_free_callback (png_structp o,
|
|
png_voidp x)
|
|
{
|
|
g_free (x);
|
|
}
|
|
|
|
G_GNUC_NORETURN static void
|
|
png_simple_error_callback (png_structp png,
|
|
png_const_charp error_msg)
|
|
{
|
|
GError **error = png_get_error_ptr (png);
|
|
|
|
if (error && !*error)
|
|
g_set_error (error,
|
|
GDK_TEXTURE_ERROR, GDK_TEXTURE_ERROR_CORRUPT_IMAGE,
|
|
_("Error reading png (%s)"), error_msg);
|
|
|
|
longjmp (png_jmpbuf (png), 1);
|
|
}
|
|
|
|
static void
|
|
png_simple_warning_callback (png_structp png,
|
|
png_const_charp error_msg)
|
|
{
|
|
}
|
|
|
|
/* }}} */
|
|
/* {{{ Public API */
|
|
|
|
GdkTexture *
|
|
gdk_load_png (GBytes *bytes,
|
|
GError **error)
|
|
{
|
|
png_io io;
|
|
png_struct *png = NULL;
|
|
png_info *info;
|
|
guint width, height;
|
|
int depth, color_type;
|
|
int interlace, stride;
|
|
GdkMemoryFormat format;
|
|
guchar *buffer = NULL;
|
|
guchar **row_pointers = NULL;
|
|
GBytes *out_bytes;
|
|
GdkTexture *texture;
|
|
int bpp;
|
|
G_GNUC_UNUSED gint64 before = GDK_PROFILER_CURRENT_TIME;
|
|
|
|
io.data = (guchar *)g_bytes_get_data (bytes, &io.size);
|
|
io.position = 0;
|
|
|
|
png = png_create_read_struct_2 (PNG_LIBPNG_VER_STRING,
|
|
error,
|
|
png_simple_error_callback,
|
|
png_simple_warning_callback,
|
|
NULL,
|
|
png_malloc_callback,
|
|
png_free_callback);
|
|
if (png == NULL)
|
|
g_error ("Out of memory");
|
|
|
|
info = png_create_info_struct (png);
|
|
if (info == NULL)
|
|
g_error ("Out of memory");
|
|
|
|
png_set_read_fn (png, &io, png_read_func);
|
|
|
|
if (sigsetjmp (png_jmpbuf (png), 1))
|
|
{
|
|
g_free (buffer);
|
|
g_free (row_pointers);
|
|
png_destroy_read_struct (&png, &info, NULL);
|
|
return NULL;
|
|
}
|
|
|
|
png_read_info (png, info);
|
|
|
|
png_get_IHDR (png, info,
|
|
&width, &height, &depth,
|
|
&color_type, &interlace, NULL, NULL);
|
|
|
|
if (color_type == PNG_COLOR_TYPE_PALETTE)
|
|
png_set_palette_to_rgb (png);
|
|
|
|
if (color_type == PNG_COLOR_TYPE_GRAY)
|
|
png_set_expand_gray_1_2_4_to_8 (png);
|
|
|
|
if (png_get_valid (png, info, PNG_INFO_tRNS))
|
|
png_set_tRNS_to_alpha (png);
|
|
|
|
if (depth < 8)
|
|
png_set_packing (png);
|
|
|
|
if (interlace != PNG_INTERLACE_NONE)
|
|
png_set_interlace_handling (png);
|
|
|
|
#if G_BYTE_ORDER == G_LITTLE_ENDIAN
|
|
png_set_swap (png);
|
|
#endif
|
|
|
|
png_read_update_info (png, info);
|
|
png_get_IHDR (png, info,
|
|
&width, &height, &depth,
|
|
&color_type, &interlace, NULL, NULL);
|
|
if (depth != 8 && depth != 16)
|
|
{
|
|
png_destroy_read_struct (&png, &info, NULL);
|
|
g_set_error (error,
|
|
GDK_TEXTURE_ERROR, GDK_TEXTURE_ERROR_UNSUPPORTED_CONTENT,
|
|
_("Unsupported depth %u in png image"), depth);
|
|
return NULL;
|
|
}
|
|
|
|
switch (color_type)
|
|
{
|
|
case PNG_COLOR_TYPE_RGB_ALPHA:
|
|
if (depth == 8)
|
|
{
|
|
format = GDK_MEMORY_R8G8B8A8;
|
|
}
|
|
else
|
|
{
|
|
format = GDK_MEMORY_R16G16B16A16;
|
|
}
|
|
break;
|
|
case PNG_COLOR_TYPE_RGB:
|
|
if (depth == 8)
|
|
{
|
|
format = GDK_MEMORY_R8G8B8;
|
|
}
|
|
else if (depth == 16)
|
|
{
|
|
format = GDK_MEMORY_R16G16B16;
|
|
}
|
|
break;
|
|
case PNG_COLOR_TYPE_GRAY:
|
|
if (depth == 8)
|
|
{
|
|
format = GDK_MEMORY_G8;
|
|
}
|
|
else if (depth == 16)
|
|
{
|
|
format = GDK_MEMORY_G16;
|
|
}
|
|
break;
|
|
case PNG_COLOR_TYPE_GRAY_ALPHA:
|
|
if (depth == 8)
|
|
{
|
|
format = GDK_MEMORY_G8A8;
|
|
}
|
|
else if (depth == 16)
|
|
{
|
|
format = GDK_MEMORY_G16A16;
|
|
}
|
|
break;
|
|
default:
|
|
png_destroy_read_struct (&png, &info, NULL);
|
|
g_set_error (error,
|
|
GDK_TEXTURE_ERROR, GDK_TEXTURE_ERROR_UNSUPPORTED_CONTENT,
|
|
_("Unsupported color type %u in png image"), color_type);
|
|
return NULL;
|
|
}
|
|
|
|
bpp = gdk_memory_format_bytes_per_pixel (format);
|
|
stride = width * bpp;
|
|
if (stride % 8)
|
|
stride += 8 - stride % 8;
|
|
|
|
buffer = g_try_malloc_n (height, stride);
|
|
row_pointers = g_try_malloc_n (height, sizeof (char *));
|
|
|
|
if (!buffer || !row_pointers)
|
|
{
|
|
g_free (buffer);
|
|
g_free (row_pointers);
|
|
png_destroy_read_struct (&png, &info, NULL);
|
|
g_set_error (error,
|
|
GDK_TEXTURE_ERROR, GDK_TEXTURE_ERROR_TOO_LARGE,
|
|
_("Not enough memory for image size %ux%u"), width, height);
|
|
return NULL;
|
|
}
|
|
|
|
for (int i = 0; i < height; i++)
|
|
row_pointers[i] = &buffer[i * stride];
|
|
|
|
png_read_image (png, row_pointers);
|
|
png_read_end (png, info);
|
|
|
|
out_bytes = g_bytes_new_take (buffer, height * stride);
|
|
texture = gdk_memory_texture_new (width, height, format, out_bytes, stride);
|
|
g_bytes_unref (out_bytes);
|
|
|
|
g_free (row_pointers);
|
|
png_destroy_read_struct (&png, &info, NULL);
|
|
|
|
if (GDK_PROFILER_IS_RUNNING)
|
|
{
|
|
gint64 end = GDK_PROFILER_CURRENT_TIME;
|
|
if (end - before > 500000)
|
|
gdk_profiler_add_mark (before, end - before, "png load", NULL);
|
|
}
|
|
|
|
return texture;
|
|
}
|
|
|
|
GBytes *
|
|
gdk_save_png (GdkTexture *texture)
|
|
{
|
|
png_struct *png = NULL;
|
|
png_info *info;
|
|
png_io io = { NULL, 0, 0 };
|
|
int width, height;
|
|
int y;
|
|
GdkMemoryFormat format;
|
|
GdkTextureDownloader downloader;
|
|
GBytes *bytes;
|
|
gsize stride;
|
|
const guchar *data;
|
|
int png_format;
|
|
int depth;
|
|
|
|
width = gdk_texture_get_width (texture);
|
|
height = gdk_texture_get_height (texture);
|
|
format = gdk_texture_get_format (texture);
|
|
|
|
switch (format)
|
|
{
|
|
case GDK_MEMORY_B8G8R8A8_PREMULTIPLIED:
|
|
case GDK_MEMORY_A8R8G8B8_PREMULTIPLIED:
|
|
case GDK_MEMORY_R8G8B8A8_PREMULTIPLIED:
|
|
case GDK_MEMORY_B8G8R8A8:
|
|
case GDK_MEMORY_A8R8G8B8:
|
|
case GDK_MEMORY_R8G8B8A8:
|
|
case GDK_MEMORY_A8B8G8R8:
|
|
format = GDK_MEMORY_R8G8B8A8;
|
|
png_format = PNG_COLOR_TYPE_RGB_ALPHA;
|
|
depth = 8;
|
|
break;
|
|
|
|
case GDK_MEMORY_R8G8B8:
|
|
case GDK_MEMORY_B8G8R8:
|
|
format = GDK_MEMORY_R8G8B8;
|
|
png_format = PNG_COLOR_TYPE_RGB;
|
|
depth = 8;
|
|
break;
|
|
|
|
case GDK_MEMORY_R16G16B16A16:
|
|
case GDK_MEMORY_R16G16B16A16_PREMULTIPLIED:
|
|
case GDK_MEMORY_R16G16B16A16_FLOAT:
|
|
case GDK_MEMORY_R16G16B16A16_FLOAT_PREMULTIPLIED:
|
|
case GDK_MEMORY_R32G32B32A32_FLOAT:
|
|
case GDK_MEMORY_R32G32B32A32_FLOAT_PREMULTIPLIED:
|
|
format = GDK_MEMORY_R16G16B16A16;
|
|
png_format = PNG_COLOR_TYPE_RGB_ALPHA;
|
|
depth = 16;
|
|
break;
|
|
|
|
case GDK_MEMORY_R16G16B16:
|
|
case GDK_MEMORY_R16G16B16_FLOAT:
|
|
case GDK_MEMORY_R32G32B32_FLOAT:
|
|
format = GDK_MEMORY_R16G16B16;
|
|
png_format = PNG_COLOR_TYPE_RGB;
|
|
depth = 16;
|
|
break;
|
|
|
|
case GDK_MEMORY_G8:
|
|
format = GDK_MEMORY_G8;
|
|
png_format = PNG_COLOR_TYPE_GRAY;
|
|
depth = 8;
|
|
break;
|
|
|
|
case GDK_MEMORY_G8A8_PREMULTIPLIED:
|
|
case GDK_MEMORY_G8A8:
|
|
case GDK_MEMORY_A8:
|
|
format = GDK_MEMORY_G8A8;
|
|
png_format = PNG_COLOR_TYPE_GRAY_ALPHA;
|
|
depth = 8;
|
|
break;
|
|
|
|
case GDK_MEMORY_G16:
|
|
format = GDK_MEMORY_G16;
|
|
png_format = PNG_COLOR_TYPE_GRAY;
|
|
depth = 16;
|
|
break;
|
|
|
|
case GDK_MEMORY_G16A16_PREMULTIPLIED:
|
|
case GDK_MEMORY_G16A16:
|
|
case GDK_MEMORY_A16:
|
|
case GDK_MEMORY_A16_FLOAT:
|
|
case GDK_MEMORY_A32_FLOAT:
|
|
format = GDK_MEMORY_G16A16;
|
|
png_format = PNG_COLOR_TYPE_GRAY_ALPHA;
|
|
depth = 16;
|
|
break;
|
|
|
|
case GDK_MEMORY_N_FORMATS:
|
|
default:
|
|
g_assert_not_reached ();
|
|
}
|
|
|
|
png = png_create_write_struct_2 (PNG_LIBPNG_VER_STRING, NULL,
|
|
png_simple_error_callback,
|
|
png_simple_warning_callback,
|
|
NULL,
|
|
png_malloc_callback,
|
|
png_free_callback);
|
|
if (!png)
|
|
return NULL;
|
|
|
|
info = png_create_info_struct (png);
|
|
if (!info)
|
|
{
|
|
png_destroy_read_struct (&png, NULL, NULL);
|
|
return NULL;
|
|
}
|
|
|
|
gdk_texture_downloader_init (&downloader, texture);
|
|
gdk_texture_downloader_set_format (&downloader, format);
|
|
bytes = gdk_texture_downloader_download_bytes (&downloader, &stride);
|
|
gdk_texture_downloader_finish (&downloader);
|
|
data = g_bytes_get_data (bytes, NULL);
|
|
|
|
if (sigsetjmp (png_jmpbuf (png), 1))
|
|
{
|
|
g_bytes_unref (bytes);
|
|
g_free (io.data);
|
|
png_destroy_read_struct (&png, &info, NULL);
|
|
return NULL;
|
|
}
|
|
|
|
png_set_write_fn (png, &io, png_write_func, png_flush_func);
|
|
|
|
png_set_IHDR (png, info, width, height, depth,
|
|
png_format,
|
|
PNG_INTERLACE_NONE,
|
|
PNG_COMPRESSION_TYPE_DEFAULT,
|
|
PNG_FILTER_TYPE_DEFAULT);
|
|
|
|
png_write_info (png, info);
|
|
|
|
#if G_BYTE_ORDER == G_LITTLE_ENDIAN
|
|
png_set_swap (png);
|
|
#endif
|
|
|
|
for (y = 0; y < height; y++)
|
|
png_write_row (png, data + y * stride);
|
|
|
|
png_write_end (png, info);
|
|
|
|
png_destroy_write_struct (&png, &info);
|
|
|
|
g_bytes_unref (bytes);
|
|
|
|
return g_bytes_new_take (io.data, io.size);
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* vim:set foldmethod=marker expandtab: */
|