ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/common/bmpfile.c
(Generate patch)

Comparing ray/src/common/bmpfile.c (file contents):
Revision 2.1 by greg, Fri Mar 26 03:11:50 2004 UTC vs.
Revision 2.20 by greg, Wed Jan 23 19:13:55 2019 UTC

# Line 5 | Line 5 | static const char RCSid[] = "$Id$";
5   *  Windows and OS/2 BMP file support
6   */
7  
8 – /* XXX reading and writing of compressed files currently unsupported */
9 –
8   #include <stdio.h>
9   #include <stdlib.h>
10   #include <string.h>
11   #include "bmpfile.h"
12  
13 < /* get corresponding error message */
13 > #ifdef getc_unlocked            /* avoid horrendous overhead of flockfile */
14 > #undef getc
15 > #undef putc
16 > #define getc    getc_unlocked
17 > #define putc    putc_unlocked
18 > #endif
19 >
20 > /* Get corresponding error message */
21   const char *
22   BMPerrorMessage(int ec)
23   {
# Line 25 | Line 30 | BMPerrorMessage(int ec)
30                  return "Truncated BMP image";
31          case BIR_UNSUPPORTED:
32                  return "Unsupported BMP feature";
33 +        case BIR_RLERROR:
34 +                return "BMP runlength encoding error";
35          case BIR_SEEKERR:
36                  return "BMP seek error";
37          }
38          return "Unknown BMP error";
39   }
40  
41 < /* compute uncompressed scanline size */
42 < static uint32
43 < getScanSiz(BMPHeader *hdr)
41 > /* Check that header is sensible */
42 > static int
43 > BMPheaderOK(const BMPHeader *hdr)
44   {
45 <        uint32  scanSiz;
46 <                                                /* bytes per scanline */
47 <        scanSiz = (hdr->bpp*hdr->width + 7) >> 3;
48 <        if (scanSiz & 3)                        /* 32-bit word boundary */
49 <                scanSiz += 4 - (scanSiz & 3);
50 <
51 <        return scanSiz;
45 >        if (!hdr)
46 >                return 0;
47 >        if ((hdr->width <= 0) | (hdr->height <= 0))
48 >                return 0;
49 >        switch (hdr->bpp) {             /* check compression */
50 >        case 1:
51 >        case 24:
52 >                if (hdr->compr != BI_UNCOMPR)
53 >                        return 0;
54 >                break;
55 >        case 16:
56 >        case 32:
57 >                if ((hdr->compr != BI_UNCOMPR) & (hdr->compr != BI_BITFIELDS))
58 >                        return 0;
59 >                break;
60 >        case 4:
61 >                if ((hdr->compr != BI_UNCOMPR) & (hdr->compr != BI_RLE4))
62 >                        return 0;
63 >                break;
64 >        case 8:
65 >                if ((hdr->compr != BI_UNCOMPR) & (hdr->compr != BI_RLE8))
66 >                        return 0;
67 >                break;
68 >        default:
69 >                return 0;
70 >        }
71 >        if (hdr->compr == BI_BITFIELDS && (BMPbitField(hdr)[0] &
72 >                                BMPbitField(hdr)[1] & BMPbitField(hdr)[2]))
73 >                return 0;
74 >        if (hdr->bpp > 8) {
75 >                if (hdr->nColors != 0)
76 >                        return 0;
77 >        } else {
78 >                if ((hdr->nColors < 0) | (hdr->nColors > 1<<hdr->bpp))
79 >                        return 0;
80 >                if ((hdr->impColors < 0) | (hdr->impColors > hdr->nColors))
81 >                        return 0;
82 >        }
83 >        return 1;
84   }
85  
86 +                                        /* compute uncompressed scan size */
87 + #define getScanSiz(h)   ( ((((h)->bpp*(h)->width+7) >>3) + 3) & ~03 )
88 +
89                                          /* get next byte from reader */
90   #define rdbyte(c,br)    ((br)->fpos += (c=(*(br)->cget)((br)->c_data))!=EOF, c)
91  
92 < /* read n bytes */
92 > /* Read n bytes */
93   static int
94   rdbytes(char *bp, uint32 n, BMPReader *br)
95   {
# Line 61 | Line 103 | rdbytes(char *bp, uint32 n, BMPReader *br)
103          return BIR_OK;
104   }
105  
106 < /* read 32-bit integer in littlendian order */
106 > /* Read 32-bit integer in littlendian order */
107   static int32
108   rdint32(BMPReader *br)
109   {
# Line 75 | Line 117 | rdint32(BMPReader *br)
117          return i;                       /* -1 on EOF */
118   }
119  
120 < /* read 16-bit unsigned integer in littlendian order */
120 > /* Read 16-bit unsigned integer in littlendian order */
121   static int
122   rduint16(BMPReader *br)
123   {
# Line 87 | Line 129 | rduint16(BMPReader *br)
129          return i;                       /* -1 on EOF */
130   }
131  
132 < /* seek on reader or return 0 (BIR_OK) on success */
132 > /* Seek on reader or return 0 (BIR_OK) on success */
133   static int
134   rdseek(uint32 pos, BMPReader *br)
135   {
# Line 99 | Line 141 | rdseek(uint32 pos, BMPReader *br)
141          return BIR_OK;
142   }
143  
144 < /* open BMP stream for reading and get first scanline */
144 > /* Open BMP stream for reading and get first scanline */
145   BMPReader *
146   BMPopenReader(int (*cget)(void *), int (*seek)(uint32, void *), void *c_data)
147   {
148          BMPReader       *br;
149 <        uint32          bmPos, hdrSiz;
150 <        int             palLen;
149 >        uint32          bmPos, hdrSiz, palSiz;
150 >        int             magic[2];               /* check magic number */
151  
152          if (cget == NULL)
153                  return NULL;
112 –        int     magic[2];               /* check magic number */
154          magic[0] = (*cget)(c_data);
155          if (magic[0] != 'B')
156                  return NULL;
# Line 130 | Line 171 | BMPopenReader(int (*cget)(void *), int (*seek)(uint32,
171          (void)rdint32(br);                      /* file size */
172          (void)rdint32(br);                      /* reserved word */
173          bmPos = rdint32(br);                    /* offset to bitmap */
174 <        hdrSiz = rdint32(br) + 14;              /* header size */
174 >        hdrSiz = 2 + 3*4 + rdint32(br);         /* header size */
175          if (hdrSiz < 2 + 6*4 + 2*2 + 6*4)
176                  goto err;
177          br->hdr->width = rdint32(br);           /* bitmap width */
178          br->hdr->height = rdint32(br);          /* bitmap height */
179 <        if ((br->hdr->width <= 0 | br->hdr->height == 0))
179 >        if (((br->hdr->width <= 0) | (br->hdr->height == 0)))
180                  goto err;
181          if ((br->hdr->yIsDown = br->hdr->height < 0))
182                  br->hdr->height = -br->hdr->height;
183          if (rduint16(br) != 1)                  /* number of planes */
184                  goto err;
185          br->hdr->bpp = rduint16(br);            /* bits per pixel */
145 –        if (br->hdr->bpp != 1 && br->hdr->bpp != 4 &&
146 –                        br->hdr->bpp != 8 && br->hdr->bpp != 24 &&
147 –                        br->hdr->bpp != 32)
148 –                goto err;
186          br->hdr->compr = rdint32(br);           /* compression mode */
150 –        if (br->hdr->compr != BI_UNCOMPR && br->hdr->compr != BI_RLE8)
151 –                goto err;                       /* don't support the rest */
187          (void)rdint32(br);                      /* bitmap size */
188          br->hdr->hRes = rdint32(br);            /* horizontal resolution */
189          br->hdr->vRes = rdint32(br);            /* vertical resolution */
190          br->hdr->nColors = rdint32(br);         /* # colors used */
191 +        if (!br->hdr->nColors && br->hdr->bpp <= 8)
192 +                br->hdr->nColors = 1<<br->hdr->bpp;
193          br->hdr->impColors = rdint32(br);       /* # important colors */
194          if (br->hdr->impColors < 0)
195                  goto err;                       /* catch premature EOF */
196 <        palLen = 0;
160 <        if (br->hdr->bpp <= 8) {
161 <                palLen = 1 << br->hdr->bpp;
162 <                if (br->hdr->nColors <= 0)
163 <                        br->hdr->nColors = palLen;
164 <                else if (br->hdr->nColors > palLen)
165 <                        goto err;
166 <                if (br->hdr->impColors <= 0)
167 <                        br->hdr->impColors = br->hdr->nColors;
168 <                else if (br->hdr->impColors > br->hdr->nColors)
169 <                        goto err;
170 <        } else if (br->hdr->nColors > 0 || br->hdr->compr != BI_UNCOMPR)
196 >        if (!BMPheaderOK(br->hdr))
197                  goto err;
198 +        palSiz = sizeof(RGBquad)*br->hdr->nColors;
199 +        if (br->hdr->impColors <= 0)
200 +                br->hdr->impColors = br->hdr->nColors;
201                                                  /* extend header */
202 <        if (bmPos < hdrSiz + sizeof(RGBquad)*palLen)
202 >        if (bmPos < hdrSiz + palSiz)
203                  goto err;
204 <        br->hdr->infoSiz = bmPos - (hdrSiz + sizeof(RGBquad)*palLen);
205 <        if (palLen > 0 || br->hdr->infoSiz > 0) {
204 >        br->hdr->infoSiz = bmPos - (hdrSiz + palSiz);
205 >        if (br->hdr->nColors > 0 || br->hdr->infoSiz > 0) {
206                  br->hdr = (BMPHeader *)realloc((void *)br->hdr,
207                                          sizeof(BMPHeader) +
208 <                                        sizeof(RGBquad)*palLen -
180 <                                        sizeof(br->hdr->palette) +
181 <                                        br->hdr->infoSiz);
208 >                                        palSiz + br->hdr->infoSiz);
209                  if (br->hdr == NULL)
210                          goto err;
211          }
212 <                                                /* read color palette */
213 <        if (rdbytes((char *)br->hdr->palette,
214 <                        sizeof(RGBquad)*palLen, br) != BIR_OK)
212 >                                                /* read colors or fields */
213 >        if (br->hdr->compr == BI_BITFIELDS) {
214 >                BMPbitField(br->hdr)[0] = (uint32)rdint32(br);
215 >                BMPbitField(br->hdr)[1] = (uint32)rdint32(br);
216 >                BMPbitField(br->hdr)[2] = (uint32)rdint32(br);
217 >        } else if (rdbytes((char *)br->hdr->palette, palSiz, br) != BIR_OK)
218                  goto err;
219                                                  /* read add'l information */
220          if (rdbytes(BMPinfo(br->hdr), br->hdr->infoSiz, br) != BIR_OK)
# Line 194 | Line 224 | BMPopenReader(int (*cget)(void *), int (*seek)(uint32,
224          if (br->scanline == NULL)
225                  goto err;
226          br->yscan = -1;
227 <        if (seek != NULL && br->hdr->compr != BI_UNCOMPR) {
227 >        if (seek != NULL && ((br->hdr->compr == BI_RLE8) |
228 >                                        (br->hdr->compr == BI_RLE4))) {
229                  BMPReader       *newbr = (BMPReader *)realloc((void *)br,
230                                                  sizeof(BMPReader) +
231                                                  sizeof(br->scanpos[0]) *
# Line 206 | Line 237 | BMPopenReader(int (*cget)(void *), int (*seek)(uint32,
237                                  sizeof(br->scanpos[0])*br->hdr->height);
238          }
239          br->scanpos[0] = br->fpos;
240 <        if (BMPreadScanline(br) != BIR_OK)
241 <                goto err;
211 <        return br;
240 >        if (BMPreadScanline(br) == BIR_OK)
241 >                return br;
242   err:
243          if (br->hdr != NULL)
244                  free((void *)br->hdr);
# Line 218 | Line 248 | err:
248          return NULL;
249   }
250  
251 < /* determine if image is grayscale */
251 > /* Determine if image is grayscale */
252   int
253   BMPisGrayscale(const BMPHeader *hdr)
254   {
# Line 230 | Line 260 | BMPisGrayscale(const BMPHeader *hdr)
260          if (hdr->bpp > 8)               /* assume they had a reason for it */
261                  return 0;
262          for (rgbp = hdr->palette, n = hdr->impColors; n-- > 0; rgbp++)
263 <                if ((rgbp->r != rgbp->g | rgbp->g != rgbp->b))
263 >                if ((rgbp->r != rgbp->g) | (rgbp->g != rgbp->b))
264                          return 0;
265          return 1;                       /* all colors neutral in map */
266   }
267  
268 < /* read and decode next BMP scanline */
268 > /* Read and decode next BMP scanline */
269   int
270   BMPreadScanline(BMPReader *br)
271   {
272 +        int     n;
273 +        int8    *sp;
274 +
275          if (br->yscan + 1 >= br->hdr->height)
276                  return BIR_EOF;
277          br->yscan++;                    /* prepare to read */
278 <        if (br->hdr->compr == BI_UNCOMPR)
279 <                return rdbytes((char *)br->scanline, getScanSiz(br->hdr), br);
280 < /* XXX need to perform actual decompression */
281 <        return BIR_UNSUPPORTED;
282 <        if (br->seek != NULL)
278 >        n = getScanSiz(br->hdr);        /* reading uncompressed data? */
279 >        if (br->hdr->compr == BI_UNCOMPR || br->hdr->compr == BI_BITFIELDS)
280 >                return rdbytes((char *)br->scanline, n, br);
281 >        /*
282 >         * RLE4/RLE8 Decoding
283 >         *
284 >         * Certain aspects of this scheme are completely insane, so
285 >         * we don't support them.  Fortunately, they rarely appear.
286 >         * One is the mid-file EOD (0x0001) and another is the ill-conceived
287 >         * "delta" (0x0002), which is like a "goto" statement for bitmaps.
288 >         * Whoever thought this up should be wrestled to the ground and told
289 >         * why it's impossible to support such a scheme in any reasonable way.
290 >         * Also, RLE4 mode allows runs to stop halfway through a byte,
291 >         * which is likewise uncodeable, so we don't even try.
292 >         * Finally, the scanline break is ambiguous -- we assume here that
293 >         * it is required at the end of each scanline, though I haven't
294 >         * found anywhere this is written.  Otherwise, we would read to
295 >         * the end of the scanline, assuming the next bit of data belongs
296 >         * the following scan.  If a break follows the last pixel, as it
297 >         * seems to in the files I've tested out of Photoshop, you end up
298 >         * painting every other line black.  Also, I assume any skipped
299 >         * pixels are painted with color 0, which is often black.  Nowhere
300 >         * is it specified what we should assume for missing pixels.  This
301 >         * is undoubtedly the most brain-dead format I've ever encountered.
302 >         */
303 >        sp = (int8 *)br->scanline;
304 >        n = br->hdr->width;
305 >        if (br->hdr->compr == BI_RLE4)
306 >                n = (n + 1) >> 1;
307 >        while (n > 0) {
308 >                int     skipOdd, len, val;
309 >
310 >                if (rdbyte(len, br) == EOF)
311 >                        return BIR_TRUNCATED;
312 >                if (len > 0) {          /* got a run */
313 >                        if (br->hdr->compr == BI_RLE4) {
314 >                                if (len & 1)
315 >                                        return BIR_UNSUPPORTED;
316 >                                len >>= 1;
317 >                        }
318 >                        if (len > n)
319 >                                return BIR_RLERROR;
320 >                        if (rdbyte(val, br) == EOF)
321 >                                return BIR_TRUNCATED;
322 >                        n -= len;
323 >                        while (len--)
324 >                                *sp++ = val;
325 >                        continue;
326 >                }
327 >                                        /* check for escape */
328 >                switch (rdbyte(len, br)) {
329 >                case EOF:
330 >                        return BIR_TRUNCATED;
331 >                case 0:                 /* end of line */
332 >                        while (n--)
333 >                                *sp++ = 0;
334 >                        /* leaves n == -1 as flag for test after loop */
335 >                        continue;
336 >                case 1:                 /* end of bitmap */
337 >                case 2:                 /* delta */
338 >                        return BIR_UNSUPPORTED;
339 >                }
340 >                                        /* absolute mode */
341 >                if (br->hdr->compr == BI_RLE4) {
342 >                        if (len & 1)
343 >                                return BIR_UNSUPPORTED;
344 >                        len >>= 1;
345 >                }
346 >                skipOdd = len & 1;
347 >                if (len > n)
348 >                        return BIR_RLERROR;
349 >                n -= len;
350 >                while (len--) {
351 >                        if (rdbyte(val, br) == EOF)
352 >                                return BIR_TRUNCATED;
353 >                        *sp++ = val;
354 >                }
355 >                if (skipOdd && rdbyte(val, br) == EOF)
356 >                        return BIR_TRUNCATED;
357 >        }
358 >                                        /* verify break at end of line */
359 >        if (!n && (rdbyte(n, br) != 0 || (rdbyte(n, br) != 0 &&
360 >                                (n != 1 || br->yscan != br->hdr->height-1))))
361 >                return BIR_RLERROR;
362 >        if (br->seek != NULL)           /* record next scanline position */
363                  br->scanpos[br->yscan + 1] = br->fpos;
364          return BIR_OK;
365   }
366  
367 < /* read a specific scanline */
367 > /* Read a specific scanline */
368   int
369   BMPseekScanline(int y, BMPReader *br)
370   {
# Line 270 | Line 383 | BMPseekScanline(int y, BMPReader *br)
383          if (y != br->yscan + 1 && br->seek != NULL) {
384                  int     yseek;
385                  uint32  seekp;
386 <                if (br->hdr->compr == BI_UNCOMPR) {
386 >                if (br->hdr->compr == BI_UNCOMPR ||
387 >                                        br->hdr->compr == BI_BITFIELDS) {
388                          yseek = y;
389                          seekp = br->scanpos[0] + y*getScanSiz(br->hdr);
390                  } else {
# Line 293 | Line 407 | BMPseekScanline(int y, BMPReader *br)
407          return BIR_OK;
408   }
409  
410 < /* get ith pixel from last scanline */
410 > /* Get ith pixel from last scanline */
411   RGBquad
412 < BMPdecodePixel(int i, BMPReader *br)
412 > BMPdecodePixel(int i, const BMPReader *br)
413   {
414 +        static const uint32     std16mask[3] = {0x7c00, 0x3e0, 0x1f};
415          static const RGBquad    black = {0, 0, 0, 0};
416 +        const uint32            *mask;
417 +        const uint8             *pp;
418 +        uint32                  pval, v;
419 +        RGBquad                 cval;
420          
421 <        if ((br == NULL | i < 0) || i >= br->hdr->width)
421 >        if (((br == NULL) | (i < 0)) || i >= br->hdr->width)
422                  return black;
423  
424 +        cval.padding = 0;
425 +
426          switch (br->hdr->bpp) {
427 <        case 24: {
428 <                uint8   *bgr = br->scanline + 3*i;
429 <                RGBquad cval;
430 <                cval.b = bgr[0]; cval.g = bgr[1]; cval.r = bgr[2];
431 <                cval.padding = 0;
427 >        case 24:
428 >                pp = br->scanline + 3*i;
429 >                cval.b = *pp++;
430 >                cval.g = *pp++;
431 >                cval.r = *pp;
432                  return cval;
312 –                }
433          case 32:
434 <                return ((RGBquad *)br->scanline)[i];
434 >                if (br->hdr->compr == BI_UNCOMPR)
435 >                        return ((RGBquad *)br->scanline)[i];
436 >                                                /* convert bit fields */
437 >                pp = br->scanline + 4*i;
438 >                pval = *pp++;
439 >                pval |= *pp++ << 8;
440 >                pval |= *pp++ << 16;
441 >                pval |= *pp << 24;
442 >                mask = BMPbitField(br->hdr);
443 >                v = pval & mask[0];
444 >                while (v & ~0xff) v >>= 8;
445 >                cval.r = v;
446 >                v = pval & mask[1];
447 >                while (v & ~0xff) v >>= 8;
448 >                cval.g = v;
449 >                v = pval & mask[2];
450 >                while (v & ~0xff) v >>= 8;
451 >                cval.b = v;
452 >                return cval;
453          case 8:
454                  return br->hdr->palette[br->scanline[i]];
455          case 1:
456 <                return br->hdr->palette[br->scanline[i>>3]>>(i&7) & 1];
456 >                return br->hdr->palette[br->scanline[i>>3]>>(7-(i&7)) & 1];
457          case 4:
458 <                return br->hdr->palette[br->scanline[i>>1]>>((i&1)<<2) & 0xf];
458 >                return br->hdr->palette[br->scanline[i>>1]>>(i&1?4:0) & 0xf];
459 >        case 16:
460 >                pp = br->scanline + 2*i;
461 >                pval = *pp++;
462 >                pval |= *pp++ << 8;
463 >                mask = std16mask;
464 >                if (br->hdr->compr == BI_BITFIELDS)
465 >                        mask = BMPbitField(br->hdr);
466 >                cval.r = ((pval & mask[0]) << 8) / (mask[0] + 1);
467 >                cval.g = ((pval & mask[1]) << 8) / (mask[1] + 1);
468 >                cval.b = ((pval & mask[2]) << 8) / (mask[2] + 1);
469 >                return cval;
470          }
471          return black;                           /* should never happen */
472   }
473  
474 < /* free BMP reader resources */
474 > /* Free BMP reader resources */
475   void
476   BMPfreeReader(BMPReader *br)
477   {
# Line 359 | Line 508 | stdio_fseek(uint32 pos, void *p)
508          return fseek((FILE *)p, (long)pos, 0);
509   }
510  
511 < /* allocate uncompressed (24-bit) RGB header */
511 > /* Allocate uncompressed (24-bit) RGB header */
512   BMPHeader *
513   BMPtruecolorHeader(int xr, int yr, int infolen)
514   {
# Line 382 | Line 531 | BMPtruecolorHeader(int xr, int yr, int infolen)
531          return hdr;
532   }
533  
534 < /* allocate color-mapped header (defaults to minimal grayscale) */
534 > /* Allocate color-mapped header (defaults to minimal grayscale) */
535   BMPHeader *
536   BMPmappedHeader(int xr, int yr, int infolen, int ncolors)
537   {
# Line 400 | Line 549 | BMPmappedHeader(int xr, int yr, int infolen, int ncolo
549          else
550                  return NULL;
551          hdr = (BMPHeader *)malloc(sizeof(BMPHeader) +
552 <                                        sizeof(RGBquad)*(1<<n) -
552 >                                        sizeof(RGBquad)*((size_t)1<<n) -
553                                          sizeof(hdr->palette) +
554                                          infolen);
555          if (hdr == NULL)
# Line 409 | Line 558 | BMPmappedHeader(int xr, int yr, int infolen, int ncolo
558          hdr->height = yr;
559          hdr->yIsDown = 0;                       /* default to upwards order */
560          hdr->bpp = n;
561 <        hdr->compr = BI_UNCOMPR;
561 >        hdr->compr = BI_UNCOMPR;                /* compression needs seek */
562          hdr->hRes = hdr->vRes = 2835;           /* default to 72 ppi */
563          hdr->nColors = ncolors;
564          hdr->impColors = 0;                     /* says all colors important */
565          hdr->infoSiz = infolen;
566 <        memset((void *)hdr->palette, 0, sizeof(RGBquad)*(1<<n) + infolen);
566 >        memset((void *)hdr->palette, 0, sizeof(RGBquad)*((size_t)1<<n) + infolen);
567          for (n = ncolors; n--; )
568                  hdr->palette[n].r = hdr->palette[n].g =
569                          hdr->palette[n].b = n*255/(ncolors-1);
# Line 423 | Line 572 | BMPmappedHeader(int xr, int yr, int infolen, int ncolo
572  
573                                          /* put byte to writer */
574   #define wrbyte(c,bw)    ( (*(bw)->cput)(c,(bw)->c_data), \
575 <                                ++((bw)->fpos) > (bw)->flen ? \
575 >                                ++(bw)->fpos > (bw)->flen ? \
576                                          ((bw)->flen = (bw)->fpos) : \
577                                          (bw)->fpos )
578  
579 < /* write out a string of bytes */
579 > /* Write out a string of bytes */
580   static void
581   wrbytes(char *bp, uint32 n, BMPWriter *bw)
582   {
# Line 435 | Line 584 | wrbytes(char *bp, uint32 n, BMPWriter *bw)
584                  wrbyte(*bp++, bw);
585   }
586  
587 < /* write 32-bit integer in littlendian order */
587 > /* Write 32-bit integer in littlendian order */
588   static void
589   wrint32(int32 i, BMPWriter *bw)
590   {
591          wrbyte(i& 0xff, bw);
592          wrbyte(i>>8 & 0xff, bw);
593          wrbyte(i>>16 & 0xff, bw);
594 <        wrbyte(i>>24  & 0xff, bw);
594 >        wrbyte(i>>24 & 0xff, bw);
595   }
596  
597 < /* write 16-bit unsigned integer in littlendian order */
597 > /* Write 16-bit unsigned integer in littlendian order */
598   static void
599   wruint16(uint16 ui, BMPWriter *bw)
600   {
# Line 453 | Line 602 | wruint16(uint16 ui, BMPWriter *bw)
602          wrbyte(ui>>8 & 0xff, bw);
603   }
604  
605 < /* seek to the specified file position, returning 0 (BIR_OK) on success */
605 > /* Seek to the specified file position, returning 0 (BIR_OK) on success */
606   static int
607   wrseek(uint32 pos, BMPWriter *bw)
608   {
609          if (pos == bw->fpos)
610                  return BIR_OK;
611 <        if (bw->seek == NULL) {
612 <                if (pos < bw->fpos)
464 <                        return BIR_SEEKERR;
465 <                while (bw->fpos < pos)
466 <                        wrbyte(0, bw);
467 <                return BIR_OK;
468 <        }
611 >        if (bw->seek == NULL)
612 >                return BIR_SEEKERR;
613          if ((*bw->seek)(pos, bw->c_data) != 0)
614                  return BIR_SEEKERR;
615          bw->fpos = pos;
# Line 474 | Line 618 | wrseek(uint32 pos, BMPWriter *bw)
618          return BIR_OK;
619   }
620  
621 < /* open BMP stream for writing */
621 > /* Open BMP stream for writing */
622   BMPWriter *
623   BMPopenWriter(void (*cput)(int, void *), int (*seek)(uint32, void *),
624                          void *c_data, BMPHeader *hdr)
# Line 482 | Line 626 | BMPopenWriter(void (*cput)(int, void *), int (*seek)(u
626          BMPWriter       *bw;
627          uint32          hdrSiz, palSiz, scanSiz, bmSiz;
628                                                  /* check arguments */
629 <        if (cput == NULL || hdr == NULL)
629 >        if (cput == NULL)
630                  return NULL;
631 <        if (hdr->width <= 0 || hdr->height <= 0)
631 >        if (!BMPheaderOK(hdr))
632                  return NULL;
633 <        if (hdr->compr != BI_UNCOMPR && hdr->compr != BI_RLE8)
633 >        if ((hdr->bpp == 16) | (hdr->compr == BI_RLE4))
634                  return NULL;                    /* unsupported */
635 <        if (hdr->bpp > 8 && hdr->compr != BI_UNCOMPR)
635 > /* no seek means we may have the wrong file length, but most app's don't care
636 >        if (seek == NULL && ((hdr->compr == BI_RLE8) | (hdr->compr == BI_RLE4)))
637                  return NULL;
638 <        palSiz = BMPpalLen(hdr);
494 <        if (hdr->nColors > palSiz)
495 <                return NULL;
496 <        if (hdr->compr != BI_UNCOMPR && seek == NULL)
497 <                return NULL;
638 > */
639                                                  /* compute sizes */
640          hdrSiz = 2 + 6*4 + 2*2 + 6*4;
641 <        palSiz *= sizeof(RGBquad);
641 >        if (hdr->compr == BI_BITFIELDS)
642 >                hdrSiz += sizeof(uint32)*3;
643 >        palSiz = sizeof(RGBquad)*hdr->nColors;
644          scanSiz = getScanSiz(hdr);
645          bmSiz = hdr->height*scanSiz;            /* wrong if compressed */
646                                                  /* initialize writer */
# Line 547 | Line 690 | BMPopenWriter(void (*cput)(int, void *), int (*seek)(u
690   #endif
691          return bw;
692   }
693 +
694 + /* Find position of next run of 5 or more identical bytes, or 255 if none */
695 + static int
696 + findNextRun(const int8 *bp, int len)
697 + {
698 +        int     pos, cnt;
699 +                                                /* look for run */
700 +        for (pos = 0; (len > 0) & (pos < 255); pos++, bp++, len--) {
701 +                if (len < 5)                    /* no hope left? */
702 +                        continue;
703 +                cnt = 1;                        /* else let's try it */
704 +                while (bp[cnt] == bp[0])
705 +                        if (++cnt >= 5)
706 +                                return pos;     /* long enough */
707 +        }
708 +        return pos;                             /* didn't find any */
709 + }
710                                  
711 < /* write the current scanline */
711 > /* Write the current scanline */
712   int
713   BMPwriteScanline(BMPWriter *bw)
714   {
715 +        const int8      *sp;
716 +        int             n;
717 +
718          if (bw->yscan >= bw->hdr->height)
719                  return BIR_EOF;
720                                                  /* writing uncompressed? */
721 <        if (bw->hdr->compr == BI_UNCOMPR) {
721 >        if (bw->hdr->compr == BI_UNCOMPR || bw->hdr->compr == BI_BITFIELDS) {
722                  uint32  scanSiz = getScanSiz(bw->hdr);
723                  uint32  slpos = bw->fbmp + bw->yscan*scanSiz;
724                  if (wrseek(slpos, bw) != BIR_OK)
# Line 564 | Line 727 | BMPwriteScanline(BMPWriter *bw)
727                  bw->yscan++;
728                  return BIR_OK;
729          }
730 <                                                /* else write compressed */
731 < /* XXX need to do actual compression */
732 <        return BIR_UNSUPPORTED;
733 <                                                /* write file length at end */
730 >        /*
731 >         * RLE8 Encoding
732 >         *
733 >         * See the notes in BMPreadScanline() on this encoding.  Needless
734 >         * to say, we avoid the nuttier aspects of this specification.
735 >         * We also assume that every scanline ends in a line break
736 >         * (0x0000) except for the last, which ends in a bitmap break
737 >         * (0x0001).  We don't support RLE4 at all; it's too awkward.
738 >         */
739 >        sp = (const int8 *)bw->scanline;
740 >        n = bw->hdr->width;
741 >        while (n > 0) {
742 >                int     cnt, val;
743 >                cnt = findNextRun(sp, n);       /* 0-255 < n */
744 >                if (cnt >= 3) {                 /* output absolute */
745 >                        int     skipOdd = cnt & 1;
746 >                        wrbyte(0, bw);
747 >                        wrbyte(cnt, bw);
748 >                        n -= cnt;
749 >                        while (cnt--)
750 >                                wrbyte(*sp++, bw);
751 >                        if (skipOdd)
752 >                                wrbyte(0, bw);
753 >                }
754 >                if (n <= 0)                     /* was that it? */
755 >                        break;
756 >                val = *sp++;                    /* output run */
757 >                for (cnt = 1; --n && cnt < 255; cnt++, sp++)
758 >                        if (*sp != val)
759 >                                break;
760 >                wrbyte(cnt, bw);
761 >                wrbyte(val, bw);
762 >        }
763 >        bw->yscan++;                            /* write line break or EOD */
764          if (bw->yscan == bw->hdr->height) {
765 <                if (bw->seek == NULL || (*bw->seek)(2, bw->c_data) != 0)
766 <                        return BIR_SEEKERR;
767 <                bw->fpos = 2;
768 <                wrint32(bw->flen, bw);
765 >                wrbyte(0, bw); wrbyte(1, bw);   /* end of bitmap marker */
766 >                if (wrseek(2, bw) != BIR_OK)
767 >                        return BIR_OK;          /* no one may care */
768 >                wrint32(bw->flen, bw);          /* correct file length */
769 >                if (wrseek(34, bw) != BIR_OK)
770 >                        return BIR_OK;
771 >                wrint32(bw->flen-bw->fbmp, bw); /* correct bitmap length */
772 >        } else {
773 >                wrbyte(0, bw); wrbyte(0, bw);   /* end of line marker */
774          }
775          return BIR_OK;
776   }
777  
778 < /* free BMP writer resources */
778 > /* Free BMP writer resources */
779   void
780   BMPfreeWriter(BMPWriter *bw)
781   {

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines