]> git.jsancho.org Git - lugaru.git/blob - Dependencies/libjpeg/rdbmp.c
Slightly squash pack stuff
[lugaru.git] / Dependencies / libjpeg / rdbmp.c
1 /*
2  * rdbmp.c
3  *
4  * Copyright (C) 1994-1996, Thomas G. Lane.
5  * Modified 2009-2010 by Guido Vollbeding.
6  * This file is part of the Independent JPEG Group's software.
7  * For conditions of distribution and use, see the accompanying README file.
8  *
9  * This file contains routines to read input images in Microsoft "BMP"
10  * format (MS Windows 3.x, OS/2 1.x, and OS/2 2.x flavors).
11  * Currently, only 8-bit and 24-bit images are supported, not 1-bit or
12  * 4-bit (feeding such low-depth images into JPEG would be silly anyway).
13  * Also, we don't support RLE-compressed files.
14  *
15  * These routines may need modification for non-Unix environments or
16  * specialized applications.  As they stand, they assume input from
17  * an ordinary stdio stream.  They further assume that reading begins
18  * at the start of the file; start_input may need work if the
19  * user interface has already read some data (e.g., to determine that
20  * the file is indeed BMP format).
21  *
22  * This code contributed by James Arthur Boucher.
23  */
24
25 #include "cdjpeg.h"             /* Common decls for cjpeg/djpeg applications */
26
27 #ifdef BMP_SUPPORTED
28
29
30 /* Macros to deal with unsigned chars as efficiently as compiler allows */
31
32 #ifdef HAVE_UNSIGNED_CHAR
33 typedef unsigned char U_CHAR;
34 #define UCH(x)  ((int) (x))
35 #else /* !HAVE_UNSIGNED_CHAR */
36 #ifdef CHAR_IS_UNSIGNED
37 typedef char U_CHAR;
38 #define UCH(x)  ((int) (x))
39 #else
40 typedef char U_CHAR;
41 #define UCH(x)  ((int) (x) & 0xFF)
42 #endif
43 #endif /* HAVE_UNSIGNED_CHAR */
44
45
46 #define ReadOK(file,buffer,len) (JFREAD(file,buffer,len) == ((size_t) (len)))
47
48
49 /* Private version of data source object */
50
51 typedef struct _bmp_source_struct * bmp_source_ptr;
52
53 typedef struct _bmp_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;          /* BMP colormap (converted to my format) */
59
60   jvirt_sarray_ptr whole_image; /* Needed to reverse row order */
61   JDIMENSION source_row;        /* Current source row number */
62   JDIMENSION row_width;         /* Physical width of scanlines in file */
63
64   int bits_per_pixel;           /* remembers 8- or 24-bit format */
65 } bmp_source_struct;
66
67
68 LOCAL(int)
69 read_byte (bmp_source_ptr sinfo)
70 /* Read next byte from BMP file */
71 {
72   register FILE *infile = sinfo->pub.input_file;
73   register int c;
74
75   if ((c = getc(infile)) == EOF)
76     ERREXIT(sinfo->cinfo, JERR_INPUT_EOF);
77   return c;
78 }
79
80
81 LOCAL(void)
82 read_colormap (bmp_source_ptr sinfo, int cmaplen, int mapentrysize)
83 /* Read the colormap from a BMP file */
84 {
85   int i;
86
87   switch (mapentrysize) {
88   case 3:
89     /* BGR format (occurs in OS/2 files) */
90     for (i = 0; i < cmaplen; i++) {
91       sinfo->colormap[2][i] = (JSAMPLE) read_byte(sinfo);
92       sinfo->colormap[1][i] = (JSAMPLE) read_byte(sinfo);
93       sinfo->colormap[0][i] = (JSAMPLE) read_byte(sinfo);
94     }
95     break;
96   case 4:
97     /* BGR0 format (occurs in MS Windows files) */
98     for (i = 0; i < cmaplen; i++) {
99       sinfo->colormap[2][i] = (JSAMPLE) read_byte(sinfo);
100       sinfo->colormap[1][i] = (JSAMPLE) read_byte(sinfo);
101       sinfo->colormap[0][i] = (JSAMPLE) read_byte(sinfo);
102       (void) read_byte(sinfo);
103     }
104     break;
105   default:
106     ERREXIT(sinfo->cinfo, JERR_BMP_BADCMAP);
107     break;
108   }
109 }
110
111
112 /*
113  * Read one row of pixels.
114  * The image has been read into the whole_image array, but is otherwise
115  * unprocessed.  We must read it out in top-to-bottom row order, and if
116  * it is an 8-bit image, we must expand colormapped pixels to 24bit format.
117  */
118
119 METHODDEF(JDIMENSION)
120 get_8bit_row (j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
121 /* This version is for reading 8-bit colormap indexes */
122 {
123   bmp_source_ptr source = (bmp_source_ptr) sinfo;
124   register JSAMPARRAY colormap = source->colormap;
125   JSAMPARRAY image_ptr;
126   register int t;
127   register JSAMPROW inptr, outptr;
128   register JDIMENSION col;
129
130   /* Fetch next row from virtual array */
131   source->source_row--;
132   image_ptr = (*cinfo->mem->access_virt_sarray)
133     ((j_common_ptr) cinfo, source->whole_image,
134      source->source_row, (JDIMENSION) 1, FALSE);
135
136   /* Expand the colormap indexes to real data */
137   inptr = image_ptr[0];
138   outptr = source->pub.buffer[0];
139   for (col = cinfo->image_width; col > 0; col--) {
140     t = GETJSAMPLE(*inptr++);
141     *outptr++ = colormap[0][t]; /* can omit GETJSAMPLE() safely */
142     *outptr++ = colormap[1][t];
143     *outptr++ = colormap[2][t];
144   }
145
146   return 1;
147 }
148
149
150 METHODDEF(JDIMENSION)
151 get_24bit_row (j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
152 /* This version is for reading 24-bit pixels */
153 {
154   bmp_source_ptr source = (bmp_source_ptr) sinfo;
155   JSAMPARRAY image_ptr;
156   register JSAMPROW inptr, outptr;
157   register JDIMENSION col;
158
159   /* Fetch next row from virtual array */
160   source->source_row--;
161   image_ptr = (*cinfo->mem->access_virt_sarray)
162     ((j_common_ptr) cinfo, source->whole_image,
163      source->source_row, (JDIMENSION) 1, FALSE);
164
165   /* Transfer data.  Note source values are in BGR order
166    * (even though Microsoft's own documents say the opposite).
167    */
168   inptr = image_ptr[0];
169   outptr = source->pub.buffer[0];
170   for (col = cinfo->image_width; col > 0; col--) {
171     outptr[2] = *inptr++;       /* can omit GETJSAMPLE() safely */
172     outptr[1] = *inptr++;
173     outptr[0] = *inptr++;
174     outptr += 3;
175   }
176
177   return 1;
178 }
179
180
181 METHODDEF(JDIMENSION)
182 get_32bit_row (j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
183 /* This version is for reading 32-bit pixels */
184 {
185   bmp_source_ptr source = (bmp_source_ptr) sinfo;
186   JSAMPARRAY image_ptr;
187   register JSAMPROW inptr, outptr;
188   register JDIMENSION col;
189
190   /* Fetch next row from virtual array */
191   source->source_row--;
192   image_ptr = (*cinfo->mem->access_virt_sarray)
193     ((j_common_ptr) cinfo, source->whole_image,
194      source->source_row, (JDIMENSION) 1, FALSE);
195   /* Transfer data.  Note source values are in BGR order
196    * (even though Microsoft's own documents say the opposite).
197    */
198   inptr = image_ptr[0];
199   outptr = source->pub.buffer[0];
200   for (col = cinfo->image_width; col > 0; col--) {
201     outptr[2] = *inptr++;       /* can omit GETJSAMPLE() safely */
202     outptr[1] = *inptr++;
203     outptr[0] = *inptr++;
204     inptr++;                    /* skip the 4th byte (Alpha channel) */
205     outptr += 3;
206   }
207
208   return 1;
209 }
210
211
212 /*
213  * This method loads the image into whole_image during the first call on
214  * get_pixel_rows.  The get_pixel_rows pointer is then adjusted to call
215  * get_8bit_row, get_24bit_row, or get_32bit_row on subsequent calls.
216  */
217
218 METHODDEF(JDIMENSION)
219 preload_image (j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
220 {
221   bmp_source_ptr source = (bmp_source_ptr) sinfo;
222   register FILE *infile = source->pub.input_file;
223   register int c;
224   register JSAMPROW out_ptr;
225   JSAMPARRAY image_ptr;
226   JDIMENSION row, col;
227   cd_progress_ptr progress = (cd_progress_ptr) cinfo->progress;
228
229   /* Read the data into a virtual array in input-file row order. */
230   for (row = 0; row < cinfo->image_height; row++) {
231     if (progress != NULL) {
232       progress->pub.pass_counter = (long) row;
233       progress->pub.pass_limit = (long) cinfo->image_height;
234       (*progress->pub.progress_monitor) ((j_common_ptr) cinfo);
235     }
236     image_ptr = (*cinfo->mem->access_virt_sarray)
237       ((j_common_ptr) cinfo, source->whole_image,
238        row, (JDIMENSION) 1, TRUE);
239     out_ptr = image_ptr[0];
240     for (col = source->row_width; col > 0; col--) {
241       /* inline copy of read_byte() for speed */
242       if ((c = getc(infile)) == EOF)
243         ERREXIT(cinfo, JERR_INPUT_EOF);
244       *out_ptr++ = (JSAMPLE) c;
245     }
246   }
247   if (progress != NULL)
248     progress->completed_extra_passes++;
249
250   /* Set up to read from the virtual array in top-to-bottom order */
251   switch (source->bits_per_pixel) {
252   case 8:
253     source->pub.get_pixel_rows = get_8bit_row;
254     break;
255   case 24:
256     source->pub.get_pixel_rows = get_24bit_row;
257     break;
258   case 32:
259     source->pub.get_pixel_rows = get_32bit_row;
260     break;
261   default:
262     ERREXIT(cinfo, JERR_BMP_BADDEPTH);
263   }
264   source->source_row = cinfo->image_height;
265
266   /* And read the first row */
267   return (*source->pub.get_pixel_rows) (cinfo, sinfo);
268 }
269
270
271 /*
272  * Read the file header; return image size and component count.
273  */
274
275 METHODDEF(void)
276 start_input_bmp (j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
277 {
278   bmp_source_ptr source = (bmp_source_ptr) sinfo;
279   U_CHAR bmpfileheader[14];
280   U_CHAR bmpinfoheader[64];
281 #define GET_2B(array,offset)  ((unsigned int) UCH(array[offset]) + \
282                                (((unsigned int) UCH(array[offset+1])) << 8))
283 #define GET_4B(array,offset)  ((INT32) UCH(array[offset]) + \
284                                (((INT32) UCH(array[offset+1])) << 8) + \
285                                (((INT32) UCH(array[offset+2])) << 16) + \
286                                (((INT32) UCH(array[offset+3])) << 24))
287   INT32 bfOffBits;
288   INT32 headerSize;
289   INT32 biWidth;
290   INT32 biHeight;
291   unsigned int biPlanes;
292   INT32 biCompression;
293   INT32 biXPelsPerMeter,biYPelsPerMeter;
294   INT32 biClrUsed = 0;
295   int mapentrysize = 0;         /* 0 indicates no colormap */
296   INT32 bPad;
297   JDIMENSION row_width;
298
299   /* Read and verify the bitmap file header */
300   if (! ReadOK(source->pub.input_file, bmpfileheader, 14))
301     ERREXIT(cinfo, JERR_INPUT_EOF);
302   if (GET_2B(bmpfileheader,0) != 0x4D42) /* 'BM' */
303     ERREXIT(cinfo, JERR_BMP_NOT);
304   bfOffBits = (INT32) GET_4B(bmpfileheader,10);
305   /* We ignore the remaining fileheader fields */
306
307   /* The infoheader might be 12 bytes (OS/2 1.x), 40 bytes (Windows),
308    * or 64 bytes (OS/2 2.x).  Check the first 4 bytes to find out which.
309    */
310   if (! ReadOK(source->pub.input_file, bmpinfoheader, 4))
311     ERREXIT(cinfo, JERR_INPUT_EOF);
312   headerSize = (INT32) GET_4B(bmpinfoheader,0);
313   if (headerSize < 12 || headerSize > 64)
314     ERREXIT(cinfo, JERR_BMP_BADHEADER);
315   if (! ReadOK(source->pub.input_file, bmpinfoheader+4, headerSize-4))
316     ERREXIT(cinfo, JERR_INPUT_EOF);
317
318   switch ((int) headerSize) {
319   case 12:
320     /* Decode OS/2 1.x header (Microsoft calls this a BITMAPCOREHEADER) */
321     biWidth = (INT32) GET_2B(bmpinfoheader,4);
322     biHeight = (INT32) GET_2B(bmpinfoheader,6);
323     biPlanes = GET_2B(bmpinfoheader,8);
324     source->bits_per_pixel = (int) GET_2B(bmpinfoheader,10);
325
326     switch (source->bits_per_pixel) {
327     case 8:                     /* colormapped image */
328       mapentrysize = 3;         /* OS/2 uses RGBTRIPLE colormap */
329       TRACEMS2(cinfo, 1, JTRC_BMP_OS2_MAPPED, (int) biWidth, (int) biHeight);
330       break;
331     case 24:                    /* RGB image */
332       TRACEMS2(cinfo, 1, JTRC_BMP_OS2, (int) biWidth, (int) biHeight);
333       break;
334     default:
335       ERREXIT(cinfo, JERR_BMP_BADDEPTH);
336       break;
337     }
338     break;
339   case 40:
340   case 64:
341     /* Decode Windows 3.x header (Microsoft calls this a BITMAPINFOHEADER) */
342     /* or OS/2 2.x header, which has additional fields that we ignore */
343     biWidth = GET_4B(bmpinfoheader,4);
344     biHeight = GET_4B(bmpinfoheader,8);
345     biPlanes = GET_2B(bmpinfoheader,12);
346     source->bits_per_pixel = (int) GET_2B(bmpinfoheader,14);
347     biCompression = GET_4B(bmpinfoheader,16);
348     biXPelsPerMeter = GET_4B(bmpinfoheader,24);
349     biYPelsPerMeter = GET_4B(bmpinfoheader,28);
350     biClrUsed = GET_4B(bmpinfoheader,32);
351     /* biSizeImage, biClrImportant fields are ignored */
352
353     switch (source->bits_per_pixel) {
354     case 8:                     /* colormapped image */
355       mapentrysize = 4;         /* Windows uses RGBQUAD colormap */
356       TRACEMS2(cinfo, 1, JTRC_BMP_MAPPED, (int) biWidth, (int) biHeight);
357       break;
358     case 24:                    /* RGB image */
359       TRACEMS2(cinfo, 1, JTRC_BMP, (int) biWidth, (int) biHeight);
360       break;
361     case 32:                    /* RGB image + Alpha channel */
362       TRACEMS2(cinfo, 1, JTRC_BMP, (int) biWidth, (int) biHeight);
363       break;
364     default:
365       ERREXIT(cinfo, JERR_BMP_BADDEPTH);
366       break;
367     }
368     if (biCompression != 0)
369       ERREXIT(cinfo, JERR_BMP_COMPRESSED);
370
371     if (biXPelsPerMeter > 0 && biYPelsPerMeter > 0) {
372       /* Set JFIF density parameters from the BMP data */
373       cinfo->X_density = (UINT16) (biXPelsPerMeter/100); /* 100 cm per meter */
374       cinfo->Y_density = (UINT16) (biYPelsPerMeter/100);
375       cinfo->density_unit = 2;  /* dots/cm */
376     }
377     break;
378   default:
379     ERREXIT(cinfo, JERR_BMP_BADHEADER);
380     return;
381   }
382
383   if (biWidth <= 0 || biHeight <= 0)
384     ERREXIT(cinfo, JERR_BMP_EMPTY);
385   if (biPlanes != 1)
386     ERREXIT(cinfo, JERR_BMP_BADPLANES);
387
388   /* Compute distance to bitmap data --- will adjust for colormap below */
389   bPad = bfOffBits - (headerSize + 14);
390
391   /* Read the colormap, if any */
392   if (mapentrysize > 0) {
393     if (biClrUsed <= 0)
394       biClrUsed = 256;          /* assume it's 256 */
395     else if (biClrUsed > 256)
396       ERREXIT(cinfo, JERR_BMP_BADCMAP);
397     /* Allocate space to store the colormap */
398     source->colormap = (*cinfo->mem->alloc_sarray)
399       ((j_common_ptr) cinfo, JPOOL_IMAGE,
400        (JDIMENSION) biClrUsed, (JDIMENSION) 3);
401     /* and read it from the file */
402     read_colormap(source, (int) biClrUsed, mapentrysize);
403     /* account for size of colormap */
404     bPad -= biClrUsed * mapentrysize;
405   }
406
407   /* Skip any remaining pad bytes */
408   if (bPad < 0)                 /* incorrect bfOffBits value? */
409     ERREXIT(cinfo, JERR_BMP_BADHEADER);
410   while (--bPad >= 0) {
411     (void) read_byte(source);
412   }
413
414   /* Compute row width in file, including padding to 4-byte boundary */
415   if (source->bits_per_pixel == 24)
416     row_width = (JDIMENSION) (biWidth * 3);
417   else if (source->bits_per_pixel == 32)
418     row_width = (JDIMENSION) (biWidth * 4);
419   else
420     row_width = (JDIMENSION) biWidth;
421   while ((row_width & 3) != 0) row_width++;
422   source->row_width = row_width;
423
424   /* Allocate space for inversion array, prepare for preload pass */
425   source->whole_image = (*cinfo->mem->request_virt_sarray)
426     ((j_common_ptr) cinfo, JPOOL_IMAGE, FALSE,
427      row_width, (JDIMENSION) biHeight, (JDIMENSION) 1);
428   source->pub.get_pixel_rows = preload_image;
429   if (cinfo->progress != NULL) {
430     cd_progress_ptr progress = (cd_progress_ptr) cinfo->progress;
431     progress->total_extra_passes++; /* count file input as separate pass */
432   }
433
434   /* Allocate one-row buffer for returned data */
435   source->pub.buffer = (*cinfo->mem->alloc_sarray)
436     ((j_common_ptr) cinfo, JPOOL_IMAGE,
437      (JDIMENSION) (biWidth * 3), (JDIMENSION) 1);
438   source->pub.buffer_height = 1;
439
440   cinfo->in_color_space = JCS_RGB;
441   cinfo->input_components = 3;
442   cinfo->data_precision = 8;
443   cinfo->image_width = (JDIMENSION) biWidth;
444   cinfo->image_height = (JDIMENSION) biHeight;
445 }
446
447
448 /*
449  * Finish up at the end of the file.
450  */
451
452 METHODDEF(void)
453 finish_input_bmp (j_compress_ptr cinfo, cjpeg_source_ptr sinfo)
454 {
455   /* no work */
456 }
457
458
459 /*
460  * The module selection routine for BMP format input.
461  */
462
463 GLOBAL(cjpeg_source_ptr)
464 jinit_read_bmp (j_compress_ptr cinfo)
465 {
466   bmp_source_ptr source;
467
468   /* Create module interface object */
469   source = (bmp_source_ptr)
470       (*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_IMAGE,
471                                   SIZEOF(bmp_source_struct));
472   source->cinfo = cinfo;        /* make back link for subroutines */
473   /* Fill in method ptrs, except get_pixel_rows which start_input sets */
474   source->pub.start_input = start_input_bmp;
475   source->pub.finish_input = finish_input_bmp;
476
477   return (cjpeg_source_ptr) source;
478 }
479
480 #endif /* BMP_SUPPORTED */