1 /*
2 * rdtarga.c
3 *
4 * This file was part of the Independent JPEG Group's software:
5 * Copyright (C) 1991-1996, Thomas G. Lane.
6 * Modified 2017 by Guido Vollbeding.
7 * libjpeg-turbo Modifications:
8 * Copyright (C) 2018, D. R. Commander.
9 * For conditions of distribution and use, see the accompanying README.ijg
10 * file.
11 *
12 * This file contains routines to read input images in Targa format.
13 *
14 * These routines may need modification for non-Unix environments or
15 * specialized applications. As they stand, they assume input from
16 * an ordinary stdio stream. They further assume that reading begins
17 * at the start of the file; start_input may need work if the
18 * user interface has already read some data (e.g., to determine that
19 * the file is indeed Targa format).
20 *
21 * Based on code contributed by Lee Daniel Crocker.
22 */
23
24 #include "cdjpeg.h" /* Common decls for cjpeg/djpeg applications */
25
26 #ifdef TARGA_SUPPORTED
27
28
29 /* Macros to deal with unsigned chars as efficiently as compiler allows */
30
31 #ifdef HAVE_UNSIGNED_CHAR
32 typedef unsigned char U_CHAR;
33 #define UCH(x) ((int)(x))
34 #else /* !HAVE_UNSIGNED_CHAR */
35 #ifdef __CHAR_UNSIGNED__
36 typedef char U_CHAR;
37 #define UCH(x) ((int)(x))
38 #else
39 typedef char U_CHAR;
40 #define UCH(x) ((int)(x) & 0xFF)
41 #endif
42 #endif /* HAVE_UNSIGNED_CHAR */
43
44
45 #define ReadOK(file, buffer, len) \
46 (JFREAD(file, buffer, len) == ((size_t)(len)))
47
48
49 /* Private version of data source object */
50
51 typedef struct _tga_source_struct *tga_source_ptr;
52
53 typedef struct _tga_source_struct {
54 struct cjpeg_source_struct pub; /* public fields */
55
56 j_compress_ptr cinfo; /* back link saves passing separate parm */
57
58 JSAMPARRAY colormap; /* Targa colormap (converted to my format) */
59
60 jvirt_sarray_ptr whole_image; /* Needed if funny input row order */
61 JDIMENSION current_row; /* Current logical row number to read */
62
63 /* Pointer to routine to extract next Targa pixel from input file */
64 void (*read_pixel) (tga_source_ptr sinfo);
65
66 /* Result of read_pixel is delivered here: */
67 U_CHAR tga_pixel[4];
68
69 int pixel_size; /* Bytes per Targa pixel (1 to 4) */
70 int cmap_length; /* colormap length */
71
72 /* State info for reading RLE-coded pixels; both counts must be init to 0 */
73 int block_count; /* # of pixels remaining in RLE block */
74 int dup_pixel_count; /* # of times to duplicate previous pixel */
75
76 /* This saves the correct pixel-row-expansion method for preload_image */
77 JDIMENSION (*get_pixel_rows) (j_compress_ptr cinfo, cjpeg_source_ptr sinfo);
78 } tga_source_struct;
79
80
81 /* For expanding 5-bit pixel values to 8-bit with best rounding */
82
83 static const UINT8 c5to8bits[32] = {
84 0, 8, 16, 25, 33, 41, 49, 58,
85 66, 74, 82, 90, 99, 107, 115, 123,
86 132, 140, 148, 156, 165, 173, 181, 189,
87 197, 206, 214, 222, 230, 239, 247, 255
88 };
89
90
91
92 LOCAL(int)
read_byte(tga_source_ptr sinfo)93 read_byte(tga_source_ptr sinfo)
94 /* Read next byte from Targa file */
95 {
96 register FILE *infile = sinfo->pub.input_file;
97 register int c;
98
99 if ((c = getc(infile)) == EOF)
100 ERREXIT(sinfo->cinfo, JERR_INPUT_EOF);
101 return c;
102 }
103
104
105 LOCAL(void)
read_colormap(tga_source_ptr sinfo,int cmaplen,int mapentrysize)106 read_colormap(tga_source_ptr sinfo, int cmaplen, int mapentrysize)
107 /* Read the colormap from a Targa file */
108 {
109 int i;
110
111 /* Presently only handles 24-bit BGR format */
112 if (mapentrysize != 24)
113 ERREXIT(sinfo->cinfo, JERR_TGA_BADCMAP);
114
115 for (i = 0; i < cmaplen; i++) {
116 sinfo->colormap[2][i] = (JSAMPLE)read_byte(sinfo);
117 sinfo->colormap[1][i] = (JSAMPLE)read_byte(sinfo);
118 sinfo->colormap[0][i] = (JSAMPLE)read_byte(sinfo);
119 }
120 }
121
122
123 /*
124 * read_pixel methods: get a single pixel from Targa file into tga_pixel[]
125 */
126
127 METHODDEF(void)
read_non_rle_pixel(tga_source_ptr sinfo)128 read_non_rle_pixel(tga_source_ptr sinfo)
129 /* Read one Targa pixel from the input file; no RLE expansion */
130 {
131 register int i;
132
133 for (i = 0; i < sinfo->pixel_size; i++) {
134 sinfo->tga_pixel[i] = (U_CHAR)read_byte(sinfo);
135 }
136 }
137
138
139 METHODDEF(void)
read_rle_pixel(tga_source_ptr sinfo)140 read_rle_pixel(tga_source_ptr sinfo)
141 /* Read one Targa pixel from the input file, expanding RLE data as needed */
142 {
143 register int i;
144
145 /* Duplicate previously read pixel? */
146 if (sinfo->dup_pixel_count > 0) {
147 sinfo->dup_pixel_count--;
148 return;
149 }
150
151 /* Time to read RLE block header? */
152 if (--sinfo->block_count < 0) { /* decrement pixels remaining in block */
153 i = read_byte(sinfo);
154 if (i & 0x80) { /* Start of duplicate-pixel block? */
155 sinfo->dup_pixel_count = i & 0x7F; /* number of dups after this one */
156 sinfo->block_count = 0; /* then read new block header */
157 } else {
158 sinfo->block_count = i & 0x7F; /* number of pixels after this one */
159 }
160 }
161
162 /* Read next pixel */
163 for (i = 0; i < sinfo->pixel_size; i++) {
164 sinfo->tga_pixel[i] = (U_CHAR)read_byte(sinfo);
165 }
166 }
167
168
169 /*
170 * Read one row of pixels.
171 *
172 * We provide several different versions depending on input file format.
173 */
174
175
176 METHODDEF(JDIMENSION)
get_8bit_gray_row(j_compress_ptr cinfo,cjpeg_source_ptr sinfo)177 get_8bit_gray_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
178 /* This version is for reading 8-bit grayscale pixels */
179 {
180 tga_source_ptr source = (tga_source_ptr)sinfo;
181 register JSAMPROW ptr;
182 register JDIMENSION col;
183
184 ptr = source->pub.buffer[0];
185 for (col = cinfo->image_width; col > 0; col--) {
186 (*source->read_pixel) (source); /* Load next pixel into tga_pixel */
187 *ptr++ = (JSAMPLE)UCH(source->tga_pixel[0]);
188 }
189 return 1;
190 }
191
192 METHODDEF(JDIMENSION)
get_8bit_row(j_compress_ptr cinfo,cjpeg_source_ptr sinfo)193 get_8bit_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
194 /* This version is for reading 8-bit colormap indexes */
195 {
196 tga_source_ptr source = (tga_source_ptr)sinfo;
197 register int t;
198 register JSAMPROW ptr;
199 register JDIMENSION col;
200 register JSAMPARRAY colormap = source->colormap;
201 int cmaplen = source->cmap_length;
202
203 ptr = source->pub.buffer[0];
204 for (col = cinfo->image_width; col > 0; col--) {
205 (*source->read_pixel) (source); /* Load next pixel into tga_pixel */
206 t = UCH(source->tga_pixel[0]);
207 if (t >= cmaplen)
208 ERREXIT(cinfo, JERR_TGA_BADPARMS);
209 *ptr++ = colormap[0][t];
210 *ptr++ = colormap[1][t];
211 *ptr++ = colormap[2][t];
212 }
213 return 1;
214 }
215
216 METHODDEF(JDIMENSION)
get_16bit_row(j_compress_ptr cinfo,cjpeg_source_ptr sinfo)217 get_16bit_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
218 /* This version is for reading 16-bit pixels */
219 {
220 tga_source_ptr source = (tga_source_ptr)sinfo;
221 register int t;
222 register JSAMPROW ptr;
223 register JDIMENSION col;
224
225 ptr = source->pub.buffer[0];
226 for (col = cinfo->image_width; col > 0; col--) {
227 (*source->read_pixel) (source); /* Load next pixel into tga_pixel */
228 t = UCH(source->tga_pixel[0]);
229 t += UCH(source->tga_pixel[1]) << 8;
230 /* We expand 5 bit data to 8 bit sample width.
231 * The format of the 16-bit (LSB first) input word is
232 * xRRRRRGGGGGBBBBB
233 */
234 ptr[2] = (JSAMPLE)c5to8bits[t & 0x1F];
235 t >>= 5;
236 ptr[1] = (JSAMPLE)c5to8bits[t & 0x1F];
237 t >>= 5;
238 ptr[0] = (JSAMPLE)c5to8bits[t & 0x1F];
239 ptr += 3;
240 }
241 return 1;
242 }
243
244 METHODDEF(JDIMENSION)
get_24bit_row(j_compress_ptr cinfo,cjpeg_source_ptr sinfo)245 get_24bit_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
246 /* This version is for reading 24-bit pixels */
247 {
248 tga_source_ptr source = (tga_source_ptr)sinfo;
249 register JSAMPROW ptr;
250 register JDIMENSION col;
251
252 ptr = source->pub.buffer[0];
253 for (col = cinfo->image_width; col > 0; col--) {
254 (*source->read_pixel) (source); /* Load next pixel into tga_pixel */
255 *ptr++ = (JSAMPLE)UCH(source->tga_pixel[2]); /* change BGR to RGB order */
256 *ptr++ = (JSAMPLE)UCH(source->tga_pixel[1]);
257 *ptr++ = (JSAMPLE)UCH(source->tga_pixel[0]);
258 }
259 return 1;
260 }
261
262 /*
263 * Targa also defines a 32-bit pixel format with order B,G,R,A.
264 * We presently ignore the attribute byte, so the code for reading
265 * these pixels is identical to the 24-bit routine above.
266 * This works because the actual pixel length is only known to read_pixel.
267 */
268
269 #define get_32bit_row get_24bit_row
270
271
272 /*
273 * This method is for re-reading the input data in standard top-down
274 * row order. The entire image has already been read into whole_image
275 * with proper conversion of pixel format, but it's in a funny row order.
276 */
277
278 METHODDEF(JDIMENSION)
get_memory_row(j_compress_ptr cinfo,cjpeg_source_ptr sinfo)279 get_memory_row(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
280 {
281 tga_source_ptr source = (tga_source_ptr)sinfo;
282 JDIMENSION source_row;
283
284 /* Compute row of source that maps to current_row of normal order */
285 /* For now, assume image is bottom-up and not interlaced. */
286 /* NEEDS WORK to support interlaced images! */
287 source_row = cinfo->image_height - source->current_row - 1;
288
289 /* Fetch that row from virtual array */
290 source->pub.buffer = (*cinfo->mem->access_virt_sarray)
291 ((j_common_ptr)cinfo, source->whole_image,
292 source_row, (JDIMENSION)1, FALSE);
293
294 source->current_row++;
295 return 1;
296 }
297
298
299 /*
300 * This method loads the image into whole_image during the first call on
301 * get_pixel_rows. The get_pixel_rows pointer is then adjusted to call
302 * get_memory_row on subsequent calls.
303 */
304
305 METHODDEF(JDIMENSION)
preload_image(j_compress_ptr cinfo,cjpeg_source_ptr sinfo)306 preload_image(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
307 {
308 tga_source_ptr source = (tga_source_ptr)sinfo;
309 JDIMENSION row;
310 cd_progress_ptr progress = (cd_progress_ptr)cinfo->progress;
311
312 /* Read the data into a virtual array in input-file row order. */
313 for (row = 0; row < cinfo->image_height; row++) {
314 if (progress != NULL) {
315 progress->pub.pass_counter = (long)row;
316 progress->pub.pass_limit = (long)cinfo->image_height;
317 (*progress->pub.progress_monitor) ((j_common_ptr)cinfo);
318 }
319 source->pub.buffer = (*cinfo->mem->access_virt_sarray)
320 ((j_common_ptr)cinfo, source->whole_image, row, (JDIMENSION)1, TRUE);
321 (*source->get_pixel_rows) (cinfo, sinfo);
322 }
323 if (progress != NULL)
324 progress->completed_extra_passes++;
325
326 /* Set up to read from the virtual array in unscrambled order */
327 source->pub.get_pixel_rows = get_memory_row;
328 source->current_row = 0;
329 /* And read the first row */
330 return get_memory_row(cinfo, sinfo);
331 }
332
333
334 /*
335 * Read the file header; return image size and component count.
336 */
337
338 METHODDEF(void)
start_input_tga(j_compress_ptr cinfo,cjpeg_source_ptr sinfo)339 start_input_tga(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
340 {
341 tga_source_ptr source = (tga_source_ptr)sinfo;
342 U_CHAR targaheader[18];
343 int idlen, cmaptype, subtype, flags, interlace_type, components;
344 unsigned int width, height, maplen;
345 boolean is_bottom_up;
346
347 #define GET_2B(offset) ((unsigned int)UCH(targaheader[offset]) + \
348 (((unsigned int)UCH(targaheader[offset + 1])) << 8))
349
350 if (!ReadOK(source->pub.input_file, targaheader, 18))
351 ERREXIT(cinfo, JERR_INPUT_EOF);
352
353 /* Pretend "15-bit" pixels are 16-bit --- we ignore attribute bit anyway */
354 if (targaheader[16] == 15)
355 targaheader[16] = 16;
356
357 idlen = UCH(targaheader[0]);
358 cmaptype = UCH(targaheader[1]);
359 subtype = UCH(targaheader[2]);
360 maplen = GET_2B(5);
361 width = GET_2B(12);
362 height = GET_2B(14);
363 source->pixel_size = UCH(targaheader[16]) >> 3;
364 flags = UCH(targaheader[17]); /* Image Descriptor byte */
365
366 is_bottom_up = ((flags & 0x20) == 0); /* bit 5 set => top-down */
367 interlace_type = flags >> 6; /* bits 6/7 are interlace code */
368
369 if (cmaptype > 1 || /* cmaptype must be 0 or 1 */
370 source->pixel_size < 1 || source->pixel_size > 4 ||
371 (UCH(targaheader[16]) & 7) != 0 || /* bits/pixel must be multiple of 8 */
372 interlace_type != 0 || /* currently don't allow interlaced image */
373 width == 0 || height == 0) /* image width/height must be non-zero */
374 ERREXIT(cinfo, JERR_TGA_BADPARMS);
375
376 if (subtype > 8) {
377 /* It's an RLE-coded file */
378 source->read_pixel = read_rle_pixel;
379 source->block_count = source->dup_pixel_count = 0;
380 subtype -= 8;
381 } else {
382 /* Non-RLE file */
383 source->read_pixel = read_non_rle_pixel;
384 }
385
386 /* Now should have subtype 1, 2, or 3 */
387 components = 3; /* until proven different */
388 cinfo->in_color_space = JCS_RGB;
389
390 switch (subtype) {
391 case 1: /* Colormapped image */
392 if (source->pixel_size == 1 && cmaptype == 1)
393 source->get_pixel_rows = get_8bit_row;
394 else
395 ERREXIT(cinfo, JERR_TGA_BADPARMS);
396 TRACEMS2(cinfo, 1, JTRC_TGA_MAPPED, width, height);
397 break;
398 case 2: /* RGB image */
399 switch (source->pixel_size) {
400 case 2:
401 source->get_pixel_rows = get_16bit_row;
402 break;
403 case 3:
404 source->get_pixel_rows = get_24bit_row;
405 break;
406 case 4:
407 source->get_pixel_rows = get_32bit_row;
408 break;
409 default:
410 ERREXIT(cinfo, JERR_TGA_BADPARMS);
411 break;
412 }
413 TRACEMS2(cinfo, 1, JTRC_TGA, width, height);
414 break;
415 case 3: /* Grayscale image */
416 components = 1;
417 cinfo->in_color_space = JCS_GRAYSCALE;
418 if (source->pixel_size == 1)
419 source->get_pixel_rows = get_8bit_gray_row;
420 else
421 ERREXIT(cinfo, JERR_TGA_BADPARMS);
422 TRACEMS2(cinfo, 1, JTRC_TGA_GRAY, width, height);
423 break;
424 default:
425 ERREXIT(cinfo, JERR_TGA_BADPARMS);
426 break;
427 }
428
429 if (is_bottom_up) {
430 /* Create a virtual array to buffer the upside-down image. */
431 source->whole_image = (*cinfo->mem->request_virt_sarray)
432 ((j_common_ptr)cinfo, JPOOL_IMAGE, FALSE,
433 (JDIMENSION)width * components, (JDIMENSION)height, (JDIMENSION)1);
434 if (cinfo->progress != NULL) {
435 cd_progress_ptr progress = (cd_progress_ptr)cinfo->progress;
436 progress->total_extra_passes++; /* count file input as separate pass */
437 }
438 /* source->pub.buffer will point to the virtual array. */
439 source->pub.buffer_height = 1; /* in case anyone looks at it */
440 source->pub.get_pixel_rows = preload_image;
441 } else {
442 /* Don't need a virtual array, but do need a one-row input buffer. */
443 source->whole_image = NULL;
444 source->pub.buffer = (*cinfo->mem->alloc_sarray)
445 ((j_common_ptr)cinfo, JPOOL_IMAGE,
446 (JDIMENSION)width * components, (JDIMENSION)1);
447 source->pub.buffer_height = 1;
448 source->pub.get_pixel_rows = source->get_pixel_rows;
449 }
450
451 while (idlen--) /* Throw away ID field */
452 (void)read_byte(source);
453
454 if (maplen > 0) {
455 if (maplen > 256 || GET_2B(3) != 0)
456 ERREXIT(cinfo, JERR_TGA_BADCMAP);
457 /* Allocate space to store the colormap */
458 source->colormap = (*cinfo->mem->alloc_sarray)
459 ((j_common_ptr)cinfo, JPOOL_IMAGE, (JDIMENSION)maplen, (JDIMENSION)3);
460 source->cmap_length = (int)maplen;
461 /* and read it from the file */
462 read_colormap(source, (int)maplen, UCH(targaheader[7]));
463 } else {
464 if (cmaptype) /* but you promised a cmap! */
465 ERREXIT(cinfo, JERR_TGA_BADPARMS);
466 source->colormap = NULL;
467 source->cmap_length = 0;
468 }
469
470 cinfo->input_components = components;
471 cinfo->data_precision = 8;
472 cinfo->image_width = width;
473 cinfo->image_height = height;
474 }
475
476
477 /*
478 * Finish up at the end of the file.
479 */
480
481 METHODDEF(void)
finish_input_tga(j_compress_ptr cinfo,cjpeg_source_ptr sinfo)482 finish_input_tga(j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
483 {
484 /* no work */
485 }
486
487
488 /*
489 * The module selection routine for Targa format input.
490 */
491
492 GLOBAL(cjpeg_source_ptr)
jinit_read_targa(j_compress_ptr cinfo)493 jinit_read_targa(j_compress_ptr cinfo)
494 {
495 tga_source_ptr source;
496
497 /* Create module interface object */
498 source = (tga_source_ptr)
499 (*cinfo->mem->alloc_small) ((j_common_ptr)cinfo, JPOOL_IMAGE,
500 sizeof(tga_source_struct));
501 source->cinfo = cinfo; /* make back link for subroutines */
502 /* Fill in method ptrs, except get_pixel_rows which start_input sets */
503 source->pub.start_input = start_input_tga;
504 source->pub.finish_input = finish_input_tga;
505
506 return (cjpeg_source_ptr)source;
507 }
508
509 #endif /* TARGA_SUPPORTED */
510