PIKApp/plug-ins/file-psd/psd-thumb-load.c

321 lines
9.1 KiB
C

/* PIKA - Photo and Image Kooker Application
* a rebranding of The GNU Image Manipulation Program (created with heckimp)
* A derived work which may be trivial. However, any changes may be (C)2023 by Aldercone Studio
*
* Original copyright, applying to most contents (license remains unchanged):
* Copyright (C) 1995 Spencer Kimball and Peter Mattis
*
* PIKA PSD Plug-in
* Copyright 2007 by John Marshall
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
#include "config.h"
#include <string.h>
#include <errno.h>
#include <glib/gstdio.h>
#include <libpika/pika.h>
#include "psd.h"
#include "psd-util.h"
#include "psd-image-res-load.h"
#include "psd-thumb-load.h"
#include "libpika/stdplugins-intl.h"
/* Local function prototypes */
static gint read_header_block (PSDimage *img_a,
GInputStream *input,
GError **error);
static gint read_color_mode_block (PSDimage *img_a,
GInputStream *input,
GError **error);
static gint read_image_resource_block (PSDimage *img_a,
GInputStream *input,
GError **error);
static PikaImage * create_pika_image (PSDimage *img_a,
GFile *file);
static gint add_image_resources (PikaImage *image,
PSDimage *img_a,
GFile *file,
GInputStream *input,
GError **error);
/* Main file load function */
PikaImage *
load_thumbnail_image (GFile *file,
gint *width,
gint *height,
GError **load_error)
{
GInputStream *input;
PSDimage img_a;
PikaImage *image = NULL;
GError *error = NULL;
/* ----- Open PSD file ----- */
pika_progress_init_printf (_("Opening thumbnail for '%s'"),
pika_file_get_utf8_name (file));
IFDBG(1) g_debug ("Open file %s", pika_file_get_utf8_name (file));
input = G_INPUT_STREAM (g_file_read (file, NULL, &error));
if (! input)
{
if (! error)
g_set_error (load_error, G_FILE_ERROR, g_file_error_from_errno (errno),
_("Could not open '%s' for reading: %s"),
pika_file_get_utf8_name (file), g_strerror (errno));
else
{
g_set_error (load_error, G_FILE_ERROR, G_FILE_ERROR_FAILED,
_("Could not open '%s' for reading: %s"),
pika_file_get_utf8_name (file), error->message);
g_error_free (error);
}
return NULL;
}
/* ----- Read the PSD file Header block ----- */
IFDBG(2) g_debug ("Read header block");
if (read_header_block (&img_a, input, &error) < 0)
goto load_error;
pika_progress_update (0.2);
/* ----- Read the PSD file Color Mode block ----- */
IFDBG(2) g_debug ("Read color mode block");
if (read_color_mode_block (&img_a, input, &error) < 0)
goto load_error;
pika_progress_update (0.4);
/* ----- Read the PSD file Image Resource block ----- */
IFDBG(2) g_debug ("Read image resource block");
if (read_image_resource_block (&img_a, input, &error) < 0)
goto load_error;
pika_progress_update (0.6);
/* ----- Create PIKA image ----- */
IFDBG(2) g_debug ("Create PIKA image");
image = create_pika_image (&img_a, file);
if (! image)
goto load_error;
/* ----- Add image resources ----- */
IFDBG(2) g_debug ("Add image resources");
if (add_image_resources (image, &img_a, file, input, &error) < 1)
goto load_error;
pika_progress_update (1.0);
pika_image_clean_all (image);
pika_image_undo_enable (image);
g_object_unref (input);
*width = img_a.columns;
*height = img_a.rows;
return image;
/* ----- Process load errors ----- */
load_error:
if (error)
{
g_set_error (load_error, G_FILE_ERROR, G_FILE_ERROR_FAILED,
_("Error loading PSD file: %s"), error->message);
g_error_free (error);
}
/* Delete partially loaded image */
if (image)
pika_image_delete (image);
/* Close file if Open */
g_object_unref (input);
return NULL;
}
/* Local functions */
static gint
read_header_block (PSDimage *img_a,
GInputStream *input,
GError **error)
{
guint16 version;
gchar sig[4];
gchar buf[6];
if (psd_read (input, sig, 4, error) < 4 ||
psd_read (input, &version, 2, error) < 2 ||
psd_read (input, buf, 6, error) < 6 ||
psd_read (input, &img_a->channels, 2, error) < 2 ||
psd_read (input, &img_a->rows, 4, error) < 4 ||
psd_read (input, &img_a->columns, 4, error) < 4 ||
psd_read (input, &img_a->bps, 2, error) < 2 ||
psd_read (input, &img_a->color_mode, 2, error) < 2)
{
psd_set_error (error);
return -1;
}
version = GUINT16_FROM_BE (version);
img_a->channels = GUINT16_FROM_BE (img_a->channels);
img_a->rows = GUINT32_FROM_BE (img_a->rows);
img_a->columns = GUINT32_FROM_BE (img_a->columns);
img_a->bps = GUINT16_FROM_BE (img_a->bps);
img_a->color_mode = GUINT16_FROM_BE (img_a->color_mode);
IFDBG(1) g_debug ("\n\n\tSig: %.4s\n\tVer: %d\n\tChannels: "
"%d\n\tSize: %dx%d\n\tBPS: %d\n\tMode: %d\n",
sig, version, img_a->channels,
img_a->columns, img_a->rows,
img_a->bps, img_a->color_mode);
if (memcmp (sig, "8BPS", 4) != 0)
return -1;
if (version != 1)
return -1;
if (img_a->channels > MAX_CHANNELS)
return -1;
if (img_a->rows < 1 || img_a->rows > PIKA_MAX_IMAGE_SIZE)
return -1;
if (img_a->columns < 1 || img_a->columns > PIKA_MAX_IMAGE_SIZE)
return -1;
return 0;
}
static gint
read_color_mode_block (PSDimage *img_a,
GInputStream *input,
GError **error)
{
guint64 block_len;
guint64 block_start;
guint64 block_end;
if (psd_read (input, &block_len, 4, error) < 4)
{
psd_set_error (error);
return -1;
}
block_len = GUINT32_FROM_BE (block_len);
block_start = g_seekable_tell (G_SEEKABLE (input));
block_end = block_start + block_len;
if (! psd_seek (input, block_end, G_SEEK_SET, error))
{
psd_set_error (error);
return -1;
}
return 0;
}
static gint
read_image_resource_block (PSDimage *img_a,
GInputStream *input,
GError **error)
{
guint64 block_len;
guint64 block_end;
if (psd_read (input, &block_len, 4, error) < 4)
{
psd_set_error (error);
return -1;
}
img_a->image_res_len = GUINT32_FROM_BE (block_len);
IFDBG(1) g_debug ("Image resource block size = %d", (int)img_a->image_res_len);
img_a->image_res_start = g_seekable_tell (G_SEEKABLE (input));
block_end = img_a->image_res_start + img_a->image_res_len;
if (! psd_seek (input, block_end, G_SEEK_SET, error))
{
psd_set_error (error);
return -1;
}
return 0;
}
static PikaImage *
create_pika_image (PSDimage *img_a,
GFile *file)
{
PikaImage *image = NULL;
img_a->base_type = PIKA_RGB;
/* Create pika image */
IFDBG(2) g_debug ("Create image");
image = pika_image_new (img_a->columns, img_a->rows, img_a->base_type);
pika_image_undo_disable (image);
return image;
}
static gint
add_image_resources (PikaImage *image,
PSDimage *img_a,
GFile *file,
GInputStream *input,
GError **error)
{
PSDimageres res_a;
gint status;
if (! psd_seek (input, img_a->image_res_start, G_SEEK_SET, error))
{
psd_set_error (error);
return -1;
}
while (g_seekable_tell (G_SEEKABLE (input)) < img_a->image_res_start + img_a->image_res_len)
{
if (get_image_resource_header (&res_a, input, error) < 0)
return -1;
if (res_a.data_start + res_a.data_len >
img_a->image_res_start + img_a->image_res_len)
return 0;
status = load_thumbnail_resource (&res_a, image, file, input, error);
/* Error */
if (status < 0)
return -1;
/* Thumbnail Loaded */
if (status > 0)
return 1;
}
return 0;
}