gtk2/gdk-pixbuf/io-gif.c

/* GdkPixbuf library - GIF image loader
 *
 * Copyright (C) 1999 Mark Crichton
 * Copyright (C) 1999 The Free Software Foundation
 *
 * Authors: Jonathan Blandford <jrb@redhat.com>
 *          Adapted from the gimp gif filter written by Adam Moss <adam@gimp.org>
 *          Gimp work based on earlier work by ......
 *          Permission to relicense under the LGPL obtained.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library 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
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

#include <config.h>
#include <stdio.h>
#include <glib.h>
#include <string.h>
#include "gdk-pixbuf.h"
#include "gdk-pixbuf-io.h"



#define MAXCOLORMAPSIZE  256

#define CM_RED           0
#define CM_GREEN         1
#define CM_BLUE          2

#define MAX_LWZ_BITS     12

#define INTERLACE          0x40
#define LOCALCOLORMAP      0x80
#define BitSet(byte, bit)  (((byte) & (bit)) == (bit))
#define LM_to_uint(a,b)         (((b)<<8)|(a))

#define GRAYSCALE        1
#define COLOR            2

typedef unsigned char CMap[3][MAXCOLORMAPSIZE];

typedef struct _Gif89 Gif89;
struct _Gif89
{
	int transparent;
	int delay_time;
	int input_flag;
	int disposal;
};

typedef struct _GifContext GifContext;
struct _GifContext
{
	unsigned int width;
	unsigned int height;
	CMap color_map;
	unsigned int bit_pixel;
	unsigned int color_resolution;
	unsigned int background;
	unsigned int aspect_ratio;
	int gray_scale;
	GdkPixbuf *pixbuf;
	Gif89 gif89;
	gint stage;

	/* stuff per frame.  As we only support the first one, not so
	 * relevant.  But still needed */
	int frame_len;
	int frame_height;
	int frame_interlace;

	/* Static read only */
	FILE *file;

	/* progressive read, only. */
	ModulePreparedNotifyFunc func;
	gpointer user_data;
	guchar *buf;
	guint ptr;
	guint size;

	/* our lwz context */
	int lwz_fresh;
	int lwz_code_size;
	int lwz_set_code_size;
	int lwz_max_code;
	int lwz_max_code_size;
	int lwz_firstcode;
	int lwz_oldcode;
	int lwz_clear_code;
	int lwz_end_code;
	int lwz_table[2][(1 << MAX_LWZ_BITS)];
	int lwz_stack[(1 << (MAX_LWZ_BITS)) * 2];
	int *lwz_sp;

};



static int ReadColorMap (GifContext *, int, CMap, int *);
static int DoExtension (GifContext *, int label);
static int GetDataBlock (GifContext *, unsigned char *);
static int GetCode (GifContext *, int, int);
static int LWZReadByte (GifContext *, int, int);

static void ReadImage (GifContext *context);

static int
ReadOK (GifContext *context, guchar *buffer, size_t len)
{
	if (context->file)
		return fread(buffer, len, 1, context->file) != 0;
	return 0;
}

static int
ReadColorMap (GifContext *context,
	      int   number,
	      CMap  buffer,
	      int  *format)
{
	int i;
	unsigned char rgb[3];
	int flag;

	flag = TRUE;

	for (i = 0; i < number; ++i) {
		if (!ReadOK (context, rgb, sizeof (rgb))) {
			/*g_message (_("GIF: bad colormap\n"));*/
			return TRUE;
		}

		buffer[CM_RED][i] = rgb[0];
		buffer[CM_GREEN][i] = rgb[1];
		buffer[CM_BLUE][i] = rgb[2];

		flag &= (rgb[0] == rgb[1] && rgb[1] == rgb[2]);
	}

	*format = (flag) ? GRAYSCALE : COLOR;

	return FALSE;
}

static int
DoExtension (GifContext *context, int label)
{
	static guchar buf[256];

	switch (label) {
	case 0xf9:			/* Graphic Control Extension */
		(void) GetDataBlock (context, (unsigned char *) buf);
		context->gif89.disposal = (buf[0] >> 2) & 0x7;
		context->gif89.input_flag = (buf[0] >> 1) & 0x1;
		context->gif89.delay_time = LM_to_uint (buf[1], buf[2]);
		if (context->pixbuf == NULL) {
			/* I only want to set the transparency if I haven't
			 * created the pixbuf yet. */
			if ((buf[0] & 0x1) != 0)
				context->gif89.transparent = buf[3];
			else
				context->gif89.transparent = -1;
		}
		while (GetDataBlock (context, (unsigned char *) buf) != 0)
			;
		return FALSE;
		break;
	default:
		/* Unhandled extension */
		break;
	}

	while (GetDataBlock (context, (unsigned char *) buf) != 0)
		;

	return FALSE;
}

static int ZeroDataBlock = FALSE;

static int
GetDataBlock (GifContext    *context,
	      unsigned char *buf)
{
	unsigned char count;

	if (!ReadOK (context, &count, 1)) {
		/*g_message (_("GIF: error in getting DataBlock size\n"));*/
		return -1;
	}

	ZeroDataBlock = count == 0;

	if ((count != 0) && (!ReadOK (context, buf, count))) {
		/*g_message (_("GIF: error in reading DataBlock\n"));*/
		return -1;
	}

	return count;
}

static int
GetCode (GifContext *context,
	 int   code_size,
	 int   flag)
{
	static unsigned char buf[280];
	static int curbit, lastbit, done, last_byte;
	int i, j, ret;
	unsigned char count;

	if (flag) {
		curbit = 0;
		lastbit = 0;
		done = FALSE;
		return 0;
	}

	if ((curbit + code_size) >= lastbit){
		if (done) {
			if (curbit >= lastbit) {
				/*g_message (_("GIF: ran off the end of by bits\n"));*/
				return -1;
			}
			return -1;
		}
		buf[0] = buf[last_byte - 2];
		buf[1] = buf[last_byte - 1];

		if ((count = GetDataBlock (context, &buf[2])) == 0)
			done = TRUE;

		last_byte = 2 + count;
		curbit = (curbit - lastbit) + 16;
		lastbit = (2 + count) * 8;
	}

	ret = 0;
	for (i = curbit, j = 0; j < code_size; ++i, ++j)
		ret |= ((buf[i / 8] & (1 << (i % 8))) != 0) << j;

	curbit += code_size;

	return ret;
}

static int
LWZReadByte (GifContext *context,
	     int   flag,
	     int   input_code_size)
{
	int code, incode;
	register int i;

	if (flag) {
		context->lwz_set_code_size = input_code_size;
		context->lwz_code_size = context->lwz_set_code_size + 1;
		context->lwz_clear_code = 1 << context->lwz_set_code_size;
		context->lwz_end_code = context->lwz_clear_code + 1;
		context->lwz_max_code_size = 2 * context->lwz_clear_code;
		context->lwz_max_code = context->lwz_clear_code + 2;

		GetCode (context, 0, TRUE);

		context->lwz_fresh = TRUE;

		for (i = 0; i < context->lwz_clear_code; ++i) {
			context->lwz_table[0][i] = 0;
			context->lwz_table[1][i] = i;
		}
		for (; i < (1 << MAX_LWZ_BITS); ++i)
			context->lwz_table[0][i] = context->lwz_table[1][0] = 0;

		context->lwz_sp = context->lwz_stack;

		return 0;
	} else if (context->lwz_fresh) {
		context->lwz_fresh = FALSE;
		do {
			context->lwz_firstcode = context->lwz_oldcode = GetCode (context, context->lwz_code_size, FALSE);
		} while (context->lwz_firstcode == context->lwz_clear_code);
		return context->lwz_firstcode;
	}

	if (context->lwz_sp > context->lwz_stack)
		return *--(context->lwz_sp);

	while ((code = GetCode (context, context->lwz_code_size, FALSE)) >= 0) {
		if (code == context->lwz_clear_code) {
			for (i = 0; i < context->lwz_clear_code; ++i) {
				context->lwz_table[0][i] = 0;
				context->lwz_table[1][i] = i;
			}
			for (; i < (1 << MAX_LWZ_BITS); ++i)
				context->lwz_table[0][i] = context->lwz_table[1][i] = 0;
			context->lwz_code_size = context->lwz_set_code_size + 1;
			context->lwz_max_code_size = 2 * context->lwz_clear_code;
			context->lwz_max_code = context->lwz_clear_code + 2;
			context->lwz_sp = context->lwz_stack;
			context->lwz_firstcode = context->lwz_oldcode =
				GetCode (context, context->lwz_code_size, FALSE);
			return context->lwz_firstcode;
		} else if (code == context->lwz_end_code) {
			int count;
			unsigned char buf[260];

			if (ZeroDataBlock)
				return -2;

			while ((count = GetDataBlock (context, buf)) > 0)
				;

			if (count != 0)
				/*g_print (_("GIF: missing EOD in data stream (common occurence)"));*/
			return -2;
		}

		incode = code;

		if (code >= context->lwz_max_code) {
			*(context->lwz_sp)++ = context->lwz_firstcode;
			code = context->lwz_oldcode;
		}

		while (code >= context->lwz_clear_code) {
			*(context->lwz_sp)++ = context->lwz_table[1][code];
			if (code == context->lwz_table[0][code]) {
				/*g_message (_("GIF: circular table entry BIG ERROR\n"));*/
				/*gimp_quit ();*/
				return -1;
			}
			code = context->lwz_table[0][code];
		}

		*(context->lwz_sp)++ = context->lwz_firstcode = context->lwz_table[1][code];

		if ((code = context->lwz_max_code) < (1 << MAX_LWZ_BITS)) {
			context->lwz_table[0][code] = context->lwz_oldcode;
			context->lwz_table[1][code] = context->lwz_firstcode;
			++context->lwz_max_code;
			if ((context->lwz_max_code >= context->lwz_max_code_size) &&
			    (context->lwz_max_code_size < (1 << MAX_LWZ_BITS))) {
				context->lwz_max_code_size *= 2;
				++context->lwz_code_size;
			}
		}

		context->lwz_oldcode = incode;

		if (context->lwz_sp > context->lwz_stack)
			return *--(context->lwz_sp);
	}
	return code;
}

static void
ReadImage (GifContext *context)
{
	guchar *dest, *temp;
	guchar c;
	gint xpos = 0, ypos = 0, pass = 0;
	gint v;

	/*
	**  Initialize the Compression routines
	*/
	if (!ReadOK (context, &c, 1)) {
		/*g_message (_("GIF: EOF / read error on image data\n"));*/
		return;
	}

	if (LWZReadByte (context, TRUE, c) < 0) {
		/*g_message (_("GIF: error while reading\n"));*/
		return;
	}

	context->pixbuf = gdk_pixbuf_new (ART_PIX_RGB,
					  context->gif89.transparent,
					  8,
					  context->width,
					  context->height);

	dest = gdk_pixbuf_get_pixels (context->pixbuf);
	while ((v = LWZReadByte ( context, FALSE, c)) >= 0) {
		if (context->gif89.transparent) {
			temp = dest + ypos * gdk_pixbuf_get_rowstride (context->pixbuf) + xpos * 4;
			*temp = context->color_map [0][(guchar) v];
			*(temp+1) = context->color_map [1][(guchar) v];
			*(temp+2) = context->color_map [2][(guchar) v];
			*(temp+3) = (guchar) ((v == context->gif89.transparent) ? 0 : 65535);
		} else {
			temp = dest + ypos * gdk_pixbuf_get_rowstride (context->pixbuf) + xpos * 3;
			*temp = context->color_map [0][(guchar) v];
			*(temp+1) = context->color_map [1][(guchar) v];
			*(temp+2) = context->color_map [2][(guchar) v];
		}

		xpos++;
		if (xpos == context->frame_len) {
			xpos = 0;
			if (context->frame_interlace) {
				switch (pass) {
				case 0:
				case 1:
					ypos += 8;
					break;
				case 2:
					ypos += 4;
					break;
				case 3:
					ypos += 2;
					break;
				}

				if (ypos >= context->frame_height) {
					pass++;
					switch (pass) {
					case 1:
						ypos = 4;
						break;
					case 2:
						ypos = 2;
						break;
					case 3:
						ypos = 1;
						break;
					default:
						goto fini;
					}
				}
			} else {
				ypos++;
			}
		}
		if (ypos >= context->frame_height)
			break;
	}

 fini:
	ypos = 0;

	if (LWZReadByte (context, FALSE, c) >= 0)
		g_print ("GIF: too much input data, ignoring extra...\n");
}

/* called until 1 is returned.
 * if -1 is returned, then there was an error, and we must stop loading the image
 * if 0 is returned, we need more bytes.  However, this isn't handled correctly currently,
 * so we assume that the initial 128 bytes is enough to get us started.
 * If 1 is returned, then we are ready to start loading the image. */

static gint
prepare_iter_loop (GifContext *context)
{
	unsigned char buf[16];
	unsigned char c;
	char version[4];

	if (!ReadOK (context, buf, 6)) {
		/* Unable to read magic number */
		return 0;
	}

	if (strncmp ((char *) buf, "GIF", 3) != 0) {
		/* Not a GIF file */
		return -1;
	}

	strncpy (version, (char *) buf + 3, 3);
	version[3] = '\0';

	if ((strcmp (version, "87a") != 0) && (strcmp (version, "89a") != 0)) {
		/* bad version number, not '87a' or '89a' */
		return -1;
	}

	/* read the screen descriptor */
	if (!ReadOK (context, buf, 7)) {
		/* Failed to read screen descriptor */
		return 0;
	}

	context->width = LM_to_uint (buf[0], buf[1]);
	context->height = LM_to_uint (buf[2], buf[3]);
	context->bit_pixel = 2 << (buf[4] & 0x07);
	context->color_resolution = (((buf[4] & 0x70) >> 3) + 1);
	context->background = buf[5];
	context->aspect_ratio = buf[6];
	context->pixbuf = NULL;

	if (BitSet (buf[4], LOCALCOLORMAP)) {
		/* Global Colormap */
		if (ReadColorMap (context, context->bit_pixel,
				  context->color_map,
				  &context->gray_scale)) {
			return -1;
		}
	}

	context->stage = 1;
	for (;;) {
		if (!ReadOK (context, &c, 1)) {
			return 0;
		}
		
		if (c == ';') {
			/* GIF terminator */
			/* hmm.  Not 100% sure what to do about this.  Should
			 * i try to return a blank image instead? */
			return -1;
		}

		if (c == '!') {
				/* Check the extention */
			if (!ReadOK (context, &c, 1)) {
				return 0;
			}
			DoExtension (context, c);
			continue;
		}
		if (c != ',') {
				/* Not a valid start character */
			continue;
		}

		if (!ReadOK (context, buf, 9)) {
			return 0;
		}

		/* Okay, we got all the info we need.  Lets record it */
		context->frame_len = LM_to_uint (buf[4], buf[5]);
		context->frame_height = LM_to_uint (buf[6], buf[7]);
		if (context->frame_height >context->height)
			return -1;

		context->frame_interlace = BitSet (buf[8], INTERLACE);
		if (!BitSet (buf[8], LOCALCOLORMAP)) {
			/* Does this frame have it's own colormap. */
			/* really only relevant when looking at the first frame
			 * of an animated gif. */
			/* if it does, we need to re-read in the colormap,
			 * the gray_scale, and the bit_pixel */
			context->bit_pixel = 1 << ((buf[8] & 0x07) + 1);
			if (ReadColorMap (context, context->bit_pixel, context->color_map, &context->gray_scale)) {
				return -1;
			}
		}
		return 1;
	}
	g_assert_not_reached ();
	return -1;
}

/* Shared library entry point */
GdkPixbuf *
image_load (FILE *file)
{
	GifContext *context;

	g_return_val_if_fail (file != NULL, NULL);

	context = g_new (GifContext, 1);
	context->lwz_fresh = FALSE;
	context->file = file;
	context->pixbuf = NULL;
	context->stage = 0;

	prepare_iter_loop (context);
	ReadImage (context);
	return context->pixbuf;
}

gpointer
image_begin_load (ModulePreparedNotifyFunc func, gpointer user_data)
{
	GifContext *context;

	context = g_new (GifContext, 1);
	context->func = func;
	context->user_data = user_data;
	context->lwz_fresh = FALSE;
	context->file = NULL;
	context->pixbuf = NULL;
	context->stage = 0;

	return (gpointer) context;
}

void
image_stop_load (gpointer data)
{
	GifContext *context = (GifContext *) data;

	if (context->pixbuf)
		gdk_pixbuf_unref (context->pixbuf);
	g_free (context);
}

gboolean
image_load_increment (gpointer data, guchar *buf, guint size)
{
	GifContext *context = (GifContext *) data;

	return FALSE;
}