dungeon_crawler

annotate prototype/imago2/file_jpeg.c @ 67:2560a7ab0243

find changesets by author, revision, files, or words in the commit message
internalized libanim, libimago2, and libpsys
author John Tsiombikas <nuclear@member.fsf.org>
date Sun, 07 Oct 2012 02:04:00 +0300
parents
children
rev   line source
nuclear@67 1 /*
nuclear@67 2 libimago - a multi-format image file input/output library.
nuclear@67 3 Copyright (C) 2010 John Tsiombikas <nuclear@member.fsf.org>
nuclear@67 4
nuclear@67 5 This program is free software: you can redistribute it and/or modify
nuclear@67 6 it under the terms of the GNU Lesser General Public License as published
nuclear@67 7 by the Free Software Foundation, either version 3 of the License, or
nuclear@67 8 (at your option) any later version.
nuclear@67 9
nuclear@67 10 This program is distributed in the hope that it will be useful,
nuclear@67 11 but WITHOUT ANY WARRANTY; without even the implied warranty of
nuclear@67 12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
nuclear@67 13 GNU Lesser General Public License for more details.
nuclear@67 14
nuclear@67 15 You should have received a copy of the GNU Lesser General Public License
nuclear@67 16 along with this program. If not, see <http://www.gnu.org/licenses/>.
nuclear@67 17 */
nuclear@67 18
nuclear@67 19 /* -- JPEG module -- */
nuclear@67 20
nuclear@67 21 #include <stdio.h>
nuclear@67 22 #include <stdlib.h>
nuclear@67 23 #include <string.h>
nuclear@67 24
nuclear@67 25 #ifdef WIN32
nuclear@67 26 #include <windows.h>
nuclear@67 27 #define HAVE_BOOLEAN
nuclear@67 28 #endif
nuclear@67 29
nuclear@67 30 #include <jpeglib.h>
nuclear@67 31 #include "imago2.h"
nuclear@67 32 #include "ftype_module.h"
nuclear@67 33
nuclear@67 34 #define INPUT_BUF_SIZE 512
nuclear@67 35 #define OUTPUT_BUF_SIZE 512
nuclear@67 36
nuclear@67 37 /* data source manager: adapted from jdatasrc.c */
nuclear@67 38 struct src_mgr {
nuclear@67 39 struct jpeg_source_mgr pub;
nuclear@67 40
nuclear@67 41 struct img_io *io;
nuclear@67 42 unsigned char buffer[INPUT_BUF_SIZE];
nuclear@67 43 int start_of_file;
nuclear@67 44 };
nuclear@67 45
nuclear@67 46 /* data destination manager: adapted from jdatadst.c */
nuclear@67 47 struct dst_mgr {
nuclear@67 48 struct jpeg_destination_mgr pub;
nuclear@67 49
nuclear@67 50 struct img_io *io;
nuclear@67 51 unsigned char buffer[OUTPUT_BUF_SIZE];
nuclear@67 52 };
nuclear@67 53
nuclear@67 54 static int check(struct img_io *io);
nuclear@67 55 static int read(struct img_pixmap *img, struct img_io *io);
nuclear@67 56 static int write(struct img_pixmap *img, struct img_io *io);
nuclear@67 57
nuclear@67 58 /* read source functions */
nuclear@67 59 static void init_source(j_decompress_ptr jd);
nuclear@67 60 static int fill_input_buffer(j_decompress_ptr jd);
nuclear@67 61 static void skip_input_data(j_decompress_ptr jd, long num_bytes);
nuclear@67 62 static void term_source(j_decompress_ptr jd);
nuclear@67 63
nuclear@67 64 /* write destination functions */
nuclear@67 65 static void init_destination(j_compress_ptr jc);
nuclear@67 66 static int empty_output_buffer(j_compress_ptr jc);
nuclear@67 67 static void term_destination(j_compress_ptr jc);
nuclear@67 68
nuclear@67 69 int img_register_jpeg(void)
nuclear@67 70 {
nuclear@67 71 static struct ftype_module mod = {".jpg", check, read, write};
nuclear@67 72 return img_register_module(&mod);
nuclear@67 73 }
nuclear@67 74
nuclear@67 75
nuclear@67 76 static int check(struct img_io *io)
nuclear@67 77 {
nuclear@67 78 unsigned char sig[10];
nuclear@67 79
nuclear@67 80 long pos = io->seek(0, SEEK_CUR, io->uptr);
nuclear@67 81
nuclear@67 82 if(io->read(sig, 10, io->uptr) < 10) {
nuclear@67 83 io->seek(pos, SEEK_SET, io->uptr);
nuclear@67 84 return -1;
nuclear@67 85 }
nuclear@67 86
nuclear@67 87 if(memcmp(sig, "\xff\xd8\xff\xe0", 4) != 0 || memcmp(sig + 6, "JFIF", 4) != 0) {
nuclear@67 88 io->seek(pos, SEEK_SET, io->uptr);
nuclear@67 89 return -1;
nuclear@67 90 }
nuclear@67 91 io->seek(pos, SEEK_SET, io->uptr);
nuclear@67 92 return 0;
nuclear@67 93 }
nuclear@67 94
nuclear@67 95 static int read(struct img_pixmap *img, struct img_io *io)
nuclear@67 96 {
nuclear@67 97 int i, nlines = 0;
nuclear@67 98 struct jpeg_decompress_struct cinfo;
nuclear@67 99 struct jpeg_error_mgr jerr;
nuclear@67 100 struct src_mgr src;
nuclear@67 101 unsigned char **scanlines;
nuclear@67 102
nuclear@67 103 io->seek(0, SEEK_CUR, io->uptr);
nuclear@67 104
nuclear@67 105 cinfo.err = jpeg_std_error(&jerr); /* XXX change... */
nuclear@67 106 jpeg_create_decompress(&cinfo);
nuclear@67 107
nuclear@67 108 src.pub.init_source = init_source;
nuclear@67 109 src.pub.fill_input_buffer = fill_input_buffer;
nuclear@67 110 src.pub.skip_input_data = skip_input_data;
nuclear@67 111 src.pub.resync_to_restart = jpeg_resync_to_restart;
nuclear@67 112 src.pub.term_source = term_source;
nuclear@67 113 src.pub.next_input_byte = 0;
nuclear@67 114 src.pub.bytes_in_buffer = 0;
nuclear@67 115 src.io = io;
nuclear@67 116 cinfo.src = (struct jpeg_source_mgr*)&src;
nuclear@67 117
nuclear@67 118 jpeg_read_header(&cinfo, 1);
nuclear@67 119 cinfo.out_color_space = JCS_RGB;
nuclear@67 120
nuclear@67 121 if(img_set_pixels(img, cinfo.image_width, cinfo.image_height, IMG_FMT_RGB24, 0) == -1) {
nuclear@67 122 jpeg_destroy_decompress(&cinfo);
nuclear@67 123 return -1;
nuclear@67 124 }
nuclear@67 125
nuclear@67 126 if(!(scanlines = malloc(img->height * sizeof *scanlines))) {
nuclear@67 127 jpeg_destroy_decompress(&cinfo);
nuclear@67 128 return -1;
nuclear@67 129 }
nuclear@67 130 scanlines[0] = img->pixels;
nuclear@67 131 for(i=1; i<img->height; i++) {
nuclear@67 132 scanlines[i] = scanlines[i - 1] + img->width * img->pixelsz;
nuclear@67 133 }
nuclear@67 134
nuclear@67 135 jpeg_start_decompress(&cinfo);
nuclear@67 136 while(nlines < img->height) {
nuclear@67 137 int res = jpeg_read_scanlines(&cinfo, scanlines + nlines, img->height - nlines);
nuclear@67 138 nlines += res;
nuclear@67 139 }
nuclear@67 140 jpeg_finish_decompress(&cinfo);
nuclear@67 141 jpeg_destroy_decompress(&cinfo);
nuclear@67 142
nuclear@67 143 free(scanlines);
nuclear@67 144 return 0;
nuclear@67 145 }
nuclear@67 146
nuclear@67 147 static int write(struct img_pixmap *img, struct img_io *io)
nuclear@67 148 {
nuclear@67 149 int i, nlines = 0;
nuclear@67 150 struct jpeg_compress_struct cinfo;
nuclear@67 151 struct jpeg_error_mgr jerr;
nuclear@67 152 struct dst_mgr dest;
nuclear@67 153 struct img_pixmap tmpimg;
nuclear@67 154 unsigned char **scanlines;
nuclear@67 155
nuclear@67 156 img_init(&tmpimg);
nuclear@67 157
nuclear@67 158 if(img->fmt != IMG_FMT_RGB24) {
nuclear@67 159 if(img_copy(&tmpimg, img) == -1) {
nuclear@67 160 return -1;
nuclear@67 161 }
nuclear@67 162 if(img_convert(&tmpimg, IMG_FMT_RGB24) == -1) {
nuclear@67 163 img_destroy(&tmpimg);
nuclear@67 164 return -1;
nuclear@67 165 }
nuclear@67 166 img = &tmpimg;
nuclear@67 167 }
nuclear@67 168
nuclear@67 169 if(!(scanlines = malloc(img->height * sizeof *scanlines))) {
nuclear@67 170 img_destroy(&tmpimg);
nuclear@67 171 return -1;
nuclear@67 172 }
nuclear@67 173 scanlines[0] = img->pixels;
nuclear@67 174 for(i=1; i<img->height; i++) {
nuclear@67 175 scanlines[i] = scanlines[i - 1] + img->width * img->pixelsz;
nuclear@67 176 }
nuclear@67 177
nuclear@67 178 cinfo.err = jpeg_std_error(&jerr); /* XXX */
nuclear@67 179 jpeg_create_compress(&cinfo);
nuclear@67 180
nuclear@67 181 dest.pub.init_destination = init_destination;
nuclear@67 182 dest.pub.empty_output_buffer = empty_output_buffer;
nuclear@67 183 dest.pub.term_destination = term_destination;
nuclear@67 184 dest.io = io;
nuclear@67 185 cinfo.dest = (struct jpeg_destination_mgr*)&dest;
nuclear@67 186
nuclear@67 187 cinfo.image_width = img->width;
nuclear@67 188 cinfo.image_height = img->height;
nuclear@67 189 cinfo.input_components = 3;
nuclear@67 190 cinfo.in_color_space = JCS_RGB;
nuclear@67 191
nuclear@67 192 jpeg_set_defaults(&cinfo);
nuclear@67 193
nuclear@67 194 jpeg_start_compress(&cinfo, 1);
nuclear@67 195 while(nlines < img->height) {
nuclear@67 196 int res = jpeg_write_scanlines(&cinfo, scanlines + nlines, img->height - nlines);
nuclear@67 197 nlines += res;
nuclear@67 198 }
nuclear@67 199 jpeg_finish_compress(&cinfo);
nuclear@67 200 jpeg_destroy_compress(&cinfo);
nuclear@67 201
nuclear@67 202 free(scanlines);
nuclear@67 203 img_destroy(&tmpimg);
nuclear@67 204 return 0;
nuclear@67 205 }
nuclear@67 206
nuclear@67 207 /* -- read source functions --
nuclear@67 208 * the following functions are adapted from jdatasrc.c in jpeglib
nuclear@67 209 */
nuclear@67 210 static void init_source(j_decompress_ptr jd)
nuclear@67 211 {
nuclear@67 212 struct src_mgr *src = (struct src_mgr*)jd->src;
nuclear@67 213 src->start_of_file = 1;
nuclear@67 214 }
nuclear@67 215
nuclear@67 216 static int fill_input_buffer(j_decompress_ptr jd)
nuclear@67 217 {
nuclear@67 218 struct src_mgr *src = (struct src_mgr*)jd->src;
nuclear@67 219 size_t nbytes;
nuclear@67 220
nuclear@67 221 nbytes = src->io->read(src->buffer, INPUT_BUF_SIZE, src->io->uptr);
nuclear@67 222
nuclear@67 223 if(nbytes <= 0) {
nuclear@67 224 if(src->start_of_file) {
nuclear@67 225 return 0;
nuclear@67 226 }
nuclear@67 227 /* insert a fake EOI marker */
nuclear@67 228 src->buffer[0] = 0xff;
nuclear@67 229 src->buffer[1] = JPEG_EOI;
nuclear@67 230 nbytes = 2;
nuclear@67 231 }
nuclear@67 232
nuclear@67 233 src->pub.next_input_byte = src->buffer;
nuclear@67 234 src->pub.bytes_in_buffer = nbytes;
nuclear@67 235 src->start_of_file = 0;
nuclear@67 236 return 1;
nuclear@67 237 }
nuclear@67 238
nuclear@67 239 static void skip_input_data(j_decompress_ptr jd, long num_bytes)
nuclear@67 240 {
nuclear@67 241 struct src_mgr *src = (struct src_mgr*)jd->src;
nuclear@67 242
nuclear@67 243 if(num_bytes > 0) {
nuclear@67 244 while(num_bytes > (long)src->pub.bytes_in_buffer) {
nuclear@67 245 num_bytes -= (long)src->pub.bytes_in_buffer;
nuclear@67 246 fill_input_buffer(jd);
nuclear@67 247 }
nuclear@67 248 src->pub.next_input_byte += (size_t)num_bytes;
nuclear@67 249 src->pub.bytes_in_buffer -= (size_t)num_bytes;
nuclear@67 250 }
nuclear@67 251 }
nuclear@67 252
nuclear@67 253 static void term_source(j_decompress_ptr jd)
nuclear@67 254 {
nuclear@67 255 /* nothing to see here, move along */
nuclear@67 256 }
nuclear@67 257
nuclear@67 258
nuclear@67 259 /* -- write destination functions --
nuclear@67 260 * the following functions are adapted from jdatadst.c in jpeglib
nuclear@67 261 */
nuclear@67 262 static void init_destination(j_compress_ptr jc)
nuclear@67 263 {
nuclear@67 264 struct dst_mgr *dest = (struct dst_mgr*)jc->dest;
nuclear@67 265
nuclear@67 266 dest->pub.next_output_byte = dest->buffer;
nuclear@67 267 dest->pub.free_in_buffer = OUTPUT_BUF_SIZE;
nuclear@67 268 }
nuclear@67 269
nuclear@67 270 static int empty_output_buffer(j_compress_ptr jc)
nuclear@67 271 {
nuclear@67 272 struct dst_mgr *dest = (struct dst_mgr*)jc->dest;
nuclear@67 273
nuclear@67 274 if(dest->io->write(dest->buffer, OUTPUT_BUF_SIZE, dest->io->uptr) != OUTPUT_BUF_SIZE) {
nuclear@67 275 return 0;
nuclear@67 276 }
nuclear@67 277
nuclear@67 278 dest->pub.next_output_byte = dest->buffer;
nuclear@67 279 dest->pub.free_in_buffer = OUTPUT_BUF_SIZE;
nuclear@67 280 return 1;
nuclear@67 281 }
nuclear@67 282
nuclear@67 283 static void term_destination(j_compress_ptr jc)
nuclear@67 284 {
nuclear@67 285 struct dst_mgr *dest = (struct dst_mgr*)jc->dest;
nuclear@67 286 size_t datacount = OUTPUT_BUF_SIZE - dest->pub.free_in_buffer;
nuclear@67 287
nuclear@67 288 /* write any remaining data in the buffer */
nuclear@67 289 if(datacount > 0) {
nuclear@67 290 dest->io->write(dest->buffer, datacount, dest->io->uptr);
nuclear@67 291 }
nuclear@67 292 /* XXX flush? ... */
nuclear@67 293 }