IMG_jpg.c
author Sam Lantinga <slouken@libsdl.org>
Sat, 04 Feb 2006 23:39:03 +0000
changeset 121 1bf9c0c87374
parent 118 c5e736a47ad2
child 123 1f7908a9cabb
permissions -rw-r--r--
Updated copyright information
slouken@0
     1
/*
slouken@53
     2
    SDL_image:  An example image loading library for use with SDL
slouken@121
     3
    Copyright (C) 1997-2006 Sam Lantinga
slouken@0
     4
slouken@0
     5
    This library is free software; you can redistribute it and/or
slouken@121
     6
    modify it under the terms of the GNU Lesser General Public
slouken@0
     7
    License as published by the Free Software Foundation; either
slouken@121
     8
    version 2.1 of the License, or (at your option) any later version.
slouken@0
     9
slouken@0
    10
    This library is distributed in the hope that it will be useful,
slouken@0
    11
    but WITHOUT ANY WARRANTY; without even the implied warranty of
slouken@0
    12
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
slouken@121
    13
    Lesser General Public License for more details.
slouken@0
    14
slouken@121
    15
    You should have received a copy of the GNU Lesser General Public
slouken@121
    16
    License along with this library; if not, write to the Free Software
slouken@121
    17
    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
slouken@0
    18
slouken@0
    19
    Sam Lantinga
slouken@53
    20
    slouken@libsdl.org
slouken@0
    21
*/
slouken@0
    22
slouken@0
    23
/* This is a JPEG image file loading framework */
slouken@0
    24
slouken@0
    25
#include <stdio.h>
slouken@7
    26
#include <string.h>
slouken@61
    27
#include <setjmp.h>
slouken@0
    28
slouken@0
    29
#include "SDL_image.h"
slouken@0
    30
slouken@0
    31
#ifdef LOAD_JPG
slouken@0
    32
slouken@0
    33
#include <jpeglib.h>
slouken@0
    34
slouken@0
    35
/* Define this for fast loading and not as good image quality */
slouken@0
    36
/*#define FAST_JPEG*/
slouken@0
    37
slouken@0
    38
/* See if an image is contained in a data source */
slouken@0
    39
int IMG_isJPG(SDL_RWops *src)
slouken@0
    40
{
slouken@117
    41
	int start;
slouken@0
    42
	int is_JPG;
slouken@0
    43
	Uint8 magic[4];
slouken@0
    44
slouken@117
    45
	start = SDL_RWtell(src);
slouken@0
    46
	is_JPG = 0;
slouken@0
    47
	if ( SDL_RWread(src, magic, 2, 1) ) {
slouken@0
    48
		if ( (magic[0] == 0xFF) && (magic[1] == 0xD8) ) {
slouken@0
    49
			SDL_RWread(src, magic, 4, 1);
icculus@109
    50
icculus@109
    51
			/* skip IPTC info block if it exists... */
icculus@109
    52
			if ( (magic[0] == 0xFF) && (magic[1] == 0xED) ) {
icculus@109
    53
				SDL_RWseek(src, (((Uint16)magic[2] << 8) |magic[3]) + 2, SEEK_CUR);
icculus@109
    54
			}
icculus@109
    55
slouken@0
    56
			SDL_RWread(src, magic, 4, 1);
slouken@71
    57
			if ( memcmp((char *)magic, "JFIF", 4) == 0 ||
slouken@108
    58
			     memcmp((char *)magic, "Exif", 4) == 0 ||
slouken@108
    59
			     memcmp((char *)magic, "VVL", 3) == 0 ) {
slouken@0
    60
				is_JPG = 1;
slouken@0
    61
			}
slouken@0
    62
		}
slouken@0
    63
	}
slouken@117
    64
	SDL_RWseek(src, start, SEEK_SET);
slouken@0
    65
	return(is_JPG);
slouken@0
    66
}
slouken@0
    67
slouken@0
    68
#define INPUT_BUFFER_SIZE	4096
slouken@0
    69
typedef struct {
slouken@0
    70
	struct jpeg_source_mgr pub;
slouken@0
    71
slouken@0
    72
	SDL_RWops *ctx;
slouken@0
    73
	Uint8 buffer[INPUT_BUFFER_SIZE];
slouken@0
    74
} my_source_mgr;
slouken@0
    75
slouken@0
    76
/*
slouken@0
    77
 * Initialize source --- called by jpeg_read_header
slouken@0
    78
 * before any data is actually read.
slouken@0
    79
 */
slouken@27
    80
static void init_source (j_decompress_ptr cinfo)
slouken@0
    81
{
slouken@0
    82
	/* We don't actually need to do anything */
slouken@0
    83
	return;
slouken@0
    84
}
slouken@0
    85
slouken@0
    86
/*
slouken@0
    87
 * Fill the input buffer --- called whenever buffer is emptied.
slouken@0
    88
 */
slouken@27
    89
static int fill_input_buffer (j_decompress_ptr cinfo)
slouken@0
    90
{
slouken@0
    91
	my_source_mgr * src = (my_source_mgr *) cinfo->src;
slouken@0
    92
	int nbytes;
slouken@0
    93
slouken@0
    94
	nbytes = SDL_RWread(src->ctx, src->buffer, 1, INPUT_BUFFER_SIZE);
slouken@0
    95
	if (nbytes <= 0) {
slouken@0
    96
		/* Insert a fake EOI marker */
slouken@0
    97
		src->buffer[0] = (Uint8) 0xFF;
slouken@0
    98
		src->buffer[1] = (Uint8) JPEG_EOI;
slouken@0
    99
		nbytes = 2;
slouken@0
   100
	}
slouken@0
   101
	src->pub.next_input_byte = src->buffer;
slouken@0
   102
	src->pub.bytes_in_buffer = nbytes;
slouken@0
   103
slouken@0
   104
	return TRUE;
slouken@0
   105
}
slouken@0
   106
slouken@0
   107
slouken@0
   108
/*
slouken@0
   109
 * Skip data --- used to skip over a potentially large amount of
slouken@0
   110
 * uninteresting data (such as an APPn marker).
slouken@0
   111
 *
slouken@0
   112
 * Writers of suspendable-input applications must note that skip_input_data
slouken@0
   113
 * is not granted the right to give a suspension return.  If the skip extends
slouken@0
   114
 * beyond the data currently in the buffer, the buffer can be marked empty so
slouken@0
   115
 * that the next read will cause a fill_input_buffer call that can suspend.
slouken@0
   116
 * Arranging for additional bytes to be discarded before reloading the input
slouken@0
   117
 * buffer is the application writer's problem.
slouken@0
   118
 */
slouken@27
   119
static void skip_input_data (j_decompress_ptr cinfo, long num_bytes)
slouken@0
   120
{
slouken@0
   121
	my_source_mgr * src = (my_source_mgr *) cinfo->src;
slouken@0
   122
slouken@0
   123
	/* Just a dumb implementation for now.	Could use fseek() except
slouken@0
   124
	 * it doesn't work on pipes.  Not clear that being smart is worth
slouken@0
   125
	 * any trouble anyway --- large skips are infrequent.
slouken@0
   126
	 */
slouken@0
   127
	if (num_bytes > 0) {
slouken@0
   128
		while (num_bytes > (long) src->pub.bytes_in_buffer) {
slouken@0
   129
			num_bytes -= (long) src->pub.bytes_in_buffer;
slouken@0
   130
			(void) src->pub.fill_input_buffer(cinfo);
slouken@0
   131
			/* note we assume that fill_input_buffer will never
slouken@0
   132
			 * return FALSE, so suspension need not be handled.
slouken@0
   133
			 */
slouken@0
   134
		}
slouken@0
   135
		src->pub.next_input_byte += (size_t) num_bytes;
slouken@0
   136
		src->pub.bytes_in_buffer -= (size_t) num_bytes;
slouken@0
   137
	}
slouken@0
   138
}
slouken@0
   139
slouken@0
   140
/*
slouken@0
   141
 * Terminate source --- called by jpeg_finish_decompress
slouken@0
   142
 * after all data has been read.
slouken@0
   143
 */
slouken@27
   144
static void term_source (j_decompress_ptr cinfo)
slouken@0
   145
{
slouken@0
   146
	/* We don't actually need to do anything */
slouken@0
   147
	return;
slouken@0
   148
}
slouken@0
   149
slouken@0
   150
/*
slouken@0
   151
 * Prepare for input from a stdio stream.
slouken@0
   152
 * The caller must have already opened the stream, and is responsible
slouken@0
   153
 * for closing it after finishing decompression.
slouken@0
   154
 */
slouken@27
   155
static void jpeg_SDL_RW_src (j_decompress_ptr cinfo, SDL_RWops *ctx)
slouken@0
   156
{
slouken@0
   157
  my_source_mgr *src;
slouken@0
   158
slouken@0
   159
  /* The source object and input buffer are made permanent so that a series
slouken@0
   160
   * of JPEG images can be read from the same file by calling jpeg_stdio_src
slouken@0
   161
   * only before the first one.  (If we discarded the buffer at the end of
slouken@0
   162
   * one image, we'd likely lose the start of the next one.)
slouken@0
   163
   * This makes it unsafe to use this manager and a different source
slouken@0
   164
   * manager serially with the same JPEG object.  Caveat programmer.
slouken@0
   165
   */
slouken@0
   166
  if (cinfo->src == NULL) {	/* first time for this JPEG object? */
slouken@0
   167
    cinfo->src = (struct jpeg_source_mgr *)
slouken@0
   168
      (*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_PERMANENT,
slouken@0
   169
				  sizeof(my_source_mgr));
slouken@0
   170
    src = (my_source_mgr *) cinfo->src;
slouken@0
   171
  }
slouken@0
   172
slouken@0
   173
  src = (my_source_mgr *) cinfo->src;
slouken@27
   174
  src->pub.init_source = init_source;
slouken@27
   175
  src->pub.fill_input_buffer = fill_input_buffer;
slouken@27
   176
  src->pub.skip_input_data = skip_input_data;
slouken@0
   177
  src->pub.resync_to_restart = jpeg_resync_to_restart; /* use default method */
slouken@27
   178
  src->pub.term_source = term_source;
slouken@0
   179
  src->ctx = ctx;
slouken@0
   180
  src->pub.bytes_in_buffer = 0; /* forces fill_input_buffer on first read */
slouken@0
   181
  src->pub.next_input_byte = NULL; /* until buffer loaded */
slouken@0
   182
}
slouken@0
   183
slouken@61
   184
struct my_error_mgr {
slouken@61
   185
	struct jpeg_error_mgr errmgr;
slouken@61
   186
	jmp_buf escape;
slouken@61
   187
};
slouken@61
   188
slouken@61
   189
static void my_error_exit(j_common_ptr cinfo)
slouken@61
   190
{
slouken@61
   191
	struct my_error_mgr *err = (struct my_error_mgr *)cinfo->err;
slouken@61
   192
	longjmp(err->escape, 1);
slouken@61
   193
}
slouken@61
   194
slouken@61
   195
static void output_no_message(j_common_ptr cinfo)
slouken@61
   196
{
slouken@61
   197
	/* do nothing */
slouken@61
   198
}
slouken@61
   199
slouken@0
   200
/* Load a JPEG type image from an SDL datasource */
slouken@0
   201
SDL_Surface *IMG_LoadJPG_RW(SDL_RWops *src)
slouken@0
   202
{
slouken@118
   203
	int start;
slouken@0
   204
	struct jpeg_decompress_struct cinfo;
slouken@0
   205
	JSAMPROW rowptr[1];
slouken@61
   206
	SDL_Surface *volatile surface = NULL;
slouken@61
   207
	struct my_error_mgr jerr;
slouken@0
   208
slouken@98
   209
	if ( !src ) {
slouken@98
   210
		/* The error message has been set in SDL_RWFromFile */
slouken@98
   211
		return NULL;
slouken@98
   212
	}
slouken@118
   213
	start = SDL_RWtell(src);
slouken@98
   214
slouken@0
   215
	/* Create a decompression structure and load the JPEG header */
slouken@61
   216
	cinfo.err = jpeg_std_error(&jerr.errmgr);
slouken@61
   217
	jerr.errmgr.error_exit = my_error_exit;
slouken@61
   218
	jerr.errmgr.output_message = output_no_message;
slouken@61
   219
	if(setjmp(jerr.escape)) {
slouken@61
   220
		/* If we get here, libjpeg found an error */
slouken@61
   221
		jpeg_destroy_decompress(&cinfo);
slouken@118
   222
		if ( surface != NULL ) {
slouken@118
   223
			SDL_FreeSurface(surface);
slouken@118
   224
		}
slouken@118
   225
		SDL_RWseek(src, start, SEEK_SET);
slouken@61
   226
		IMG_SetError("JPEG loading error");
slouken@61
   227
		return NULL;
slouken@61
   228
	}
slouken@61
   229
slouken@0
   230
	jpeg_create_decompress(&cinfo);
slouken@0
   231
	jpeg_SDL_RW_src(&cinfo, src);
slouken@0
   232
	jpeg_read_header(&cinfo, TRUE);
slouken@0
   233
slouken@0
   234
	/* Set 24-bit RGB output */
slouken@0
   235
	cinfo.out_color_space = JCS_RGB;
slouken@0
   236
	cinfo.quantize_colors = FALSE;
slouken@0
   237
#ifdef FAST_JPEG
slouken@0
   238
	cinfo.scale_num   = 1;
slouken@0
   239
	cinfo.scale_denom = 1;
slouken@0
   240
	cinfo.dct_method = JDCT_FASTEST;
slouken@0
   241
	cinfo.do_fancy_upsampling = FALSE;
slouken@0
   242
#endif
slouken@0
   243
	jpeg_calc_output_dimensions(&cinfo);
slouken@0
   244
slouken@0
   245
	/* Allocate an output surface to hold the image */
slouken@0
   246
	surface = SDL_AllocSurface(SDL_SWSURFACE,
slouken@0
   247
	               cinfo.output_width, cinfo.output_height, 24,
slouken@0
   248
#if SDL_BYTEORDER == SDL_LIL_ENDIAN
slouken@0
   249
	                           0x0000FF, 0x00FF00, 0xFF0000,
slouken@0
   250
#else
slouken@0
   251
	                           0xFF0000, 0x00FF00, 0x0000FF,
slouken@0
   252
#endif
slouken@0
   253
				   0);
slouken@0
   254
	if ( surface == NULL ) {
slouken@118
   255
		jpeg_destroy_decompress(&cinfo);
slouken@118
   256
		SDL_RWseek(src, start, SEEK_SET);
slouken@0
   257
		IMG_SetError("Out of memory");
slouken@118
   258
		return NULL;
slouken@0
   259
	}
slouken@0
   260
slouken@0
   261
	/* Decompress the image */
slouken@0
   262
	jpeg_start_decompress(&cinfo);
slouken@0
   263
	while ( cinfo.output_scanline < cinfo.output_height ) {
slouken@0
   264
		rowptr[0] = (JSAMPROW)(Uint8 *)surface->pixels +
slouken@0
   265
		                    cinfo.output_scanline * surface->pitch;
slouken@0
   266
		jpeg_read_scanlines(&cinfo, rowptr, (JDIMENSION) 1);
slouken@0
   267
	}
slouken@0
   268
	jpeg_finish_decompress(&cinfo);
slouken@118
   269
	jpeg_destroy_decompress(&cinfo);
slouken@0
   270
slouken@0
   271
	return(surface);
slouken@0
   272
}
slouken@0
   273
slouken@0
   274
#else
slouken@0
   275
slouken@0
   276
/* See if an image is contained in a data source */
slouken@0
   277
int IMG_isJPG(SDL_RWops *src)
slouken@0
   278
{
slouken@0
   279
	return(0);
slouken@0
   280
}
slouken@0
   281
slouken@0
   282
/* Load a JPEG type image from an SDL datasource */
slouken@0
   283
SDL_Surface *IMG_LoadJPG_RW(SDL_RWops *src)
slouken@0
   284
{
slouken@0
   285
	return(NULL);
slouken@0
   286
}
slouken@0
   287
slouken@0
   288
#endif /* LOAD_JPG */