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.16 by schorsch, Thu Mar 10 01:49:00 2016 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 + #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)
# 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 than 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  
# Line 104 | Line 146 | 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 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   }
# Line 239 | Line 269 | BMPisGrayscale(const BMPHeader *hdr)
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   }
# 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 295 | Line 409 | BMPseekScanline(int y, BMPReader *br)
409  
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   }
# Line 399 | Line 548 | BMPmappedHeader(int xr, int yr, int infolen, int ncolo
548                  n = 8;
549          else
550                  return NULL;
551 +        /* XXX VC warns about 32 bit shift coerced to 64 bit */
552          hdr = (BMPHeader *)malloc(sizeof(BMPHeader) +
553                                          sizeof(RGBquad)*(1<<n) -
554                                          sizeof(hdr->palette) +
# Line 409 | Line 559 | BMPmappedHeader(int xr, int yr, int infolen, int ncolo
559          hdr->height = yr;
560          hdr->yIsDown = 0;                       /* default to upwards order */
561          hdr->bpp = n;
562 <        hdr->compr = BI_UNCOMPR;
562 >        hdr->compr = BI_UNCOMPR;                /* compression needs seek */
563          hdr->hRes = hdr->vRes = 2835;           /* default to 72 ppi */
564          hdr->nColors = ncolors;
565          hdr->impColors = 0;                     /* says all colors important */
566          hdr->infoSiz = infolen;
567 +        /* XXX VC warns about 32 bit shift coerced to 64 bit */
568          memset((void *)hdr->palette, 0, sizeof(RGBquad)*(1<<n) + infolen);
569          for (n = ncolors; n--; )
570                  hdr->palette[n].r = hdr->palette[n].g =
# Line 423 | Line 574 | BMPmappedHeader(int xr, int yr, int infolen, int ncolo
574  
575                                          /* put byte to writer */
576   #define wrbyte(c,bw)    ( (*(bw)->cput)(c,(bw)->c_data), \
577 <                                ++((bw)->fpos) > (bw)->flen ? \
577 >                                ++(bw)->fpos > (bw)->flen ? \
578                                          ((bw)->flen = (bw)->fpos) : \
579                                          (bw)->fpos )
580  
# Line 442 | Line 593 | wrint32(int32 i, BMPWriter *bw)
593          wrbyte(i& 0xff, bw);
594          wrbyte(i>>8 & 0xff, bw);
595          wrbyte(i>>16 & 0xff, bw);
596 <        wrbyte(i>>24  & 0xff, bw);
596 >        wrbyte(i>>24 & 0xff, bw);
597   }
598  
599   /* write 16-bit unsigned integer in littlendian order */
# Line 459 | Line 610 | wrseek(uint32 pos, BMPWriter *bw)
610   {
611          if (pos == bw->fpos)
612                  return BIR_OK;
613 <        if (bw->seek == NULL) {
614 <                if (pos < bw->fpos)
464 <                        return BIR_SEEKERR;
465 <                while (bw->fpos < pos)
466 <                        wrbyte(0, bw);
467 <                return BIR_OK;
468 <        }
613 >        if (bw->seek == NULL)
614 >                return BIR_SEEKERR;
615          if ((*bw->seek)(pos, bw->c_data) != 0)
616                  return BIR_SEEKERR;
617          bw->fpos = pos;
# Line 482 | Line 628 | BMPopenWriter(void (*cput)(int, void *), int (*seek)(u
628          BMPWriter       *bw;
629          uint32          hdrSiz, palSiz, scanSiz, bmSiz;
630                                                  /* check arguments */
631 <        if (cput == NULL || hdr == NULL)
631 >        if (cput == NULL)
632                  return NULL;
633 <        if (hdr->width <= 0 || hdr->height <= 0)
633 >        if (!BMPheaderOK(hdr))
634                  return NULL;
635 <        if (hdr->compr != BI_UNCOMPR && hdr->compr != BI_RLE8)
635 >        if ((hdr->bpp == 16) | (hdr->compr == BI_RLE4))
636                  return NULL;                    /* unsupported */
637 <        if (hdr->bpp > 8 && hdr->compr != BI_UNCOMPR)
637 > /* no seek means we may have the wrong file length, but most app's don't care
638 >        if (seek == NULL && ((hdr->compr == BI_RLE8) | (hdr->compr == BI_RLE4)))
639                  return NULL;
640 <        palSiz = BMPpalLen(hdr);
494 <        if (hdr->nColors > palSiz)
495 <                return NULL;
496 <        if (hdr->compr != BI_UNCOMPR && seek == NULL)
497 <                return NULL;
640 > */
641                                                  /* compute sizes */
642          hdrSiz = 2 + 6*4 + 2*2 + 6*4;
643 <        palSiz *= sizeof(RGBquad);
643 >        if (hdr->compr == BI_BITFIELDS)
644 >                hdrSiz += sizeof(uint32)*3;
645 >        palSiz = sizeof(RGBquad)*hdr->nColors;
646          scanSiz = getScanSiz(hdr);
647          bmSiz = hdr->height*scanSiz;            /* wrong if compressed */
648                                                  /* initialize writer */
# Line 547 | Line 692 | BMPopenWriter(void (*cput)(int, void *), int (*seek)(u
692   #endif
693          return bw;
694   }
695 +
696 + /* find position of next run of 5 or more identical bytes, or 255 if none */
697 + static int
698 + findNextRun(const int8 *bp, int len)
699 + {
700 +        int     pos, cnt;
701 +                                                /* look for run */
702 +        for (pos = 0; (len > 0) & (pos < 255); pos++, bp++, len--) {
703 +                if (len < 5)                    /* no hope left? */
704 +                        continue;
705 +                cnt = 1;                        /* else let's try it */
706 +                while (bp[cnt] == bp[0])
707 +                        if (++cnt >= 5)
708 +                                return pos;     /* long enough */
709 +        }
710 +        return pos;                             /* didn't find any */
711 + }
712                                  
713   /* write the current scanline */
714   int
715   BMPwriteScanline(BMPWriter *bw)
716   {
717 +        const int8      *sp;
718 +        int             n;
719 +
720          if (bw->yscan >= bw->hdr->height)
721                  return BIR_EOF;
722                                                  /* writing uncompressed? */
723 <        if (bw->hdr->compr == BI_UNCOMPR) {
723 >        if (bw->hdr->compr == BI_UNCOMPR || bw->hdr->compr == BI_BITFIELDS) {
724                  uint32  scanSiz = getScanSiz(bw->hdr);
725                  uint32  slpos = bw->fbmp + bw->yscan*scanSiz;
726                  if (wrseek(slpos, bw) != BIR_OK)
# Line 564 | Line 729 | BMPwriteScanline(BMPWriter *bw)
729                  bw->yscan++;
730                  return BIR_OK;
731          }
732 <                                                /* else write compressed */
733 < /* XXX need to do actual compression */
734 <        return BIR_UNSUPPORTED;
735 <                                                /* write file length at end */
732 >        /*
733 >         * RLE8 Encoding
734 >         *
735 >         * See the notes in BMPreadScanline() on this encoding.  Needless
736 >         * to say, we avoid the nuttier aspects of this specification.
737 >         * We also assume that every scanline ends in a line break
738 >         * (0x0000) except for the last, which ends in a bitmap break
739 >         * (0x0001).  We don't support RLE4 at all; it's too awkward.
740 >         */
741 >        sp = (const int8 *)bw->scanline;
742 >        n = bw->hdr->width;
743 >        while (n > 0) {
744 >                int     cnt, val;
745 >                cnt = findNextRun(sp, n);       /* 0-255 < n */
746 >                if (cnt >= 3) {                 /* output absolute */
747 >                        int     skipOdd = cnt & 1;
748 >                        wrbyte(0, bw);
749 >                        wrbyte(cnt, bw);
750 >                        n -= cnt;
751 >                        while (cnt--)
752 >                                wrbyte(*sp++, bw);
753 >                        if (skipOdd)
754 >                                wrbyte(0, bw);
755 >                }
756 >                if (n <= 0)                     /* was that it? */
757 >                        break;
758 >                val = *sp;                      /* output run */
759 >                for (cnt = 1; --n && cnt < 255; cnt++)
760 >                        if (*++sp != val)
761 >                                break;
762 >                wrbyte(cnt, bw);
763 >                wrbyte(val, bw);
764 >        }
765 >        bw->yscan++;                            /* write line break or EOD */
766          if (bw->yscan == bw->hdr->height) {
767 <                if (bw->seek == NULL || (*bw->seek)(2, bw->c_data) != 0)
768 <                        return BIR_SEEKERR;
769 <                bw->fpos = 2;
770 <                wrint32(bw->flen, bw);
767 >                wrbyte(0, bw); wrbyte(1, bw);   /* end of bitmap marker */
768 >                if (wrseek(2, bw) != BIR_OK)
769 >                        return BIR_OK;          /* no one may care */
770 >                wrint32(bw->flen, bw);          /* correct file length */
771 >                if (wrseek(34, bw) != BIR_OK)
772 >                        return BIR_OK;
773 >                wrint32(bw->flen-bw->fbmp, bw); /* correct bitmap length */
774 >        } else {
775 >                wrbyte(0, bw); wrbyte(0, bw);   /* end of line marker */
776          }
777          return BIR_OK;
778   }

Diff Legend

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