ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/common/bmpfile.c
Revision: 2.21
Committed: Mon Oct 4 23:04:59 2021 UTC (2 years, 7 months ago) by greg
Content type: text/plain
Branch: MAIN
CVS Tags: rad5R4, HEAD
Changes since 2.20: +5 -1 lines
Log Message:
fix: added checks for stdio open errors

File Contents

# User Rev Content
1 greg 2.1 #ifndef lint
2 greg 2.21 static const char RCSid[] = "$Id: bmpfile.c,v 2.20 2019/01/23 19:13:55 greg Exp $";
3 greg 2.1 #endif
4     /*
5     * Windows and OS/2 BMP file support
6     */
7    
8     #include <stdio.h>
9     #include <stdlib.h>
10     #include <string.h>
11     #include "bmpfile.h"
12    
13 greg 2.9 #ifdef getc_unlocked /* avoid horrendous overhead of flockfile */
14 greg 2.12 #undef getc
15     #undef putc
16 greg 2.9 #define getc getc_unlocked
17     #define putc putc_unlocked
18     #endif
19    
20 greg 2.20 /* Get corresponding error message */
21 greg 2.1 const char *
22     BMPerrorMessage(int ec)
23     {
24     switch (ec) {
25     case BIR_OK:
26     return "No error";
27     case BIR_EOF:
28     return "End of BMP image";
29     case BIR_TRUNCATED:
30     return "Truncated BMP image";
31     case BIR_UNSUPPORTED:
32     return "Unsupported BMP feature";
33 greg 2.3 case BIR_RLERROR:
34     return "BMP runlength encoding error";
35 greg 2.1 case BIR_SEEKERR:
36     return "BMP seek error";
37     }
38     return "Unknown BMP error";
39     }
40    
41 greg 2.20 /* Check that header is sensible */
42 greg 2.3 static int
43     BMPheaderOK(const BMPHeader *hdr)
44     {
45     if (!hdr)
46     return 0;
47 schorsch 2.4 if ((hdr->width <= 0) | (hdr->height <= 0))
48 greg 2.3 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 schorsch 2.4 if ((hdr->compr != BI_UNCOMPR) & (hdr->compr != BI_BITFIELDS))
58 greg 2.3 return 0;
59     break;
60     case 4:
61 schorsch 2.4 if ((hdr->compr != BI_UNCOMPR) & (hdr->compr != BI_RLE4))
62 greg 2.3 return 0;
63     break;
64     case 8:
65 schorsch 2.4 if ((hdr->compr != BI_UNCOMPR) & (hdr->compr != BI_RLE8))
66 greg 2.3 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 greg 2.13 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 greg 2.3 return 1;
84     }
85 greg 2.1
86 greg 2.3 /* compute uncompressed scan size */
87 schorsch 2.4 #define getScanSiz(h) ( ((((h)->bpp*(h)->width+7) >>3) + 3) & ~03 )
88 greg 2.1
89     /* get next byte from reader */
90     #define rdbyte(c,br) ((br)->fpos += (c=(*(br)->cget)((br)->c_data))!=EOF, c)
91    
92 greg 2.20 /* Read n bytes */
93 greg 2.1 static int
94     rdbytes(char *bp, uint32 n, BMPReader *br)
95     {
96     int c;
97    
98     while (n--) {
99     if (rdbyte(c, br) == EOF)
100     return BIR_TRUNCATED;
101     *bp++ = c;
102     }
103     return BIR_OK;
104     }
105    
106 greg 2.20 /* Read 32-bit integer in littlendian order */
107 greg 2.1 static int32
108     rdint32(BMPReader *br)
109     {
110     int32 i;
111     int c;
112    
113     i = rdbyte(c, br);
114     i |= rdbyte(c, br) << 8;
115     i |= rdbyte(c, br) << 16;
116     i |= rdbyte(c, br) << 24;
117     return i; /* -1 on EOF */
118     }
119    
120 greg 2.20 /* Read 16-bit unsigned integer in littlendian order */
121 greg 2.1 static int
122     rduint16(BMPReader *br)
123     {
124     int i;
125     int c;
126    
127     i = rdbyte(c, br);
128     i |= rdbyte(c, br) << 8;
129     return i; /* -1 on EOF */
130     }
131    
132 greg 2.20 /* Seek on reader or return 0 (BIR_OK) on success */
133 greg 2.1 static int
134     rdseek(uint32 pos, BMPReader *br)
135     {
136     if (pos == br->fpos)
137     return BIR_OK;
138     if (br->seek == NULL || (*br->seek)(pos, br->c_data) != 0)
139     return BIR_SEEKERR;
140     br->fpos = pos;
141     return BIR_OK;
142     }
143    
144 greg 2.20 /* Open BMP stream for reading and get first scanline */
145 greg 2.1 BMPReader *
146     BMPopenReader(int (*cget)(void *), int (*seek)(uint32, void *), void *c_data)
147     {
148     BMPReader *br;
149 greg 2.13 uint32 bmPos, hdrSiz, palSiz;
150     int magic[2]; /* check magic number */
151 greg 2.1
152     if (cget == NULL)
153     return NULL;
154 greg 2.21 if (cget == &stdio_getc && c_data == NULL)
155     return NULL; /* stdio error condition */
156 greg 2.1 magic[0] = (*cget)(c_data);
157     if (magic[0] != 'B')
158     return NULL;
159     magic[1] = (*cget)(c_data);
160     if (magic[1] != 'M' && magic[1] != 'A')
161     return NULL;
162     br = (BMPReader *)calloc(1, sizeof(BMPReader));
163     if (br == NULL)
164     return NULL;
165     br->cget = cget;
166     br->seek = seek;
167     br->c_data = c_data;
168     br->hdr = (BMPHeader *)malloc(sizeof(BMPHeader));
169     if (br->hdr == NULL)
170     goto err;
171     br->fpos = 2;
172     /* read & verify file header */
173     (void)rdint32(br); /* file size */
174     (void)rdint32(br); /* reserved word */
175     bmPos = rdint32(br); /* offset to bitmap */
176 greg 2.3 hdrSiz = 2 + 3*4 + rdint32(br); /* header size */
177 greg 2.1 if (hdrSiz < 2 + 6*4 + 2*2 + 6*4)
178     goto err;
179     br->hdr->width = rdint32(br); /* bitmap width */
180     br->hdr->height = rdint32(br); /* bitmap height */
181 schorsch 2.2 if (((br->hdr->width <= 0) | (br->hdr->height == 0)))
182 greg 2.1 goto err;
183     if ((br->hdr->yIsDown = br->hdr->height < 0))
184     br->hdr->height = -br->hdr->height;
185     if (rduint16(br) != 1) /* number of planes */
186     goto err;
187     br->hdr->bpp = rduint16(br); /* bits per pixel */
188     br->hdr->compr = rdint32(br); /* compression mode */
189     (void)rdint32(br); /* bitmap size */
190     br->hdr->hRes = rdint32(br); /* horizontal resolution */
191     br->hdr->vRes = rdint32(br); /* vertical resolution */
192     br->hdr->nColors = rdint32(br); /* # colors used */
193 greg 2.14 if (!br->hdr->nColors && br->hdr->bpp <= 8)
194     br->hdr->nColors = 1<<br->hdr->bpp;
195 greg 2.1 br->hdr->impColors = rdint32(br); /* # important colors */
196     if (br->hdr->impColors < 0)
197     goto err; /* catch premature EOF */
198 greg 2.3 if (!BMPheaderOK(br->hdr))
199     goto err;
200 greg 2.13 palSiz = sizeof(RGBquad)*br->hdr->nColors;
201     if (br->hdr->impColors <= 0)
202     br->hdr->impColors = br->hdr->nColors;
203 greg 2.1 /* extend header */
204 greg 2.13 if (bmPos < hdrSiz + palSiz)
205 greg 2.1 goto err;
206 greg 2.13 br->hdr->infoSiz = bmPos - (hdrSiz + palSiz);
207     if (br->hdr->nColors > 0 || br->hdr->infoSiz > 0) {
208 greg 2.1 br->hdr = (BMPHeader *)realloc((void *)br->hdr,
209     sizeof(BMPHeader) +
210 greg 2.13 palSiz + br->hdr->infoSiz);
211 greg 2.1 if (br->hdr == NULL)
212     goto err;
213     }
214 greg 2.3 /* read colors or fields */
215     if (br->hdr->compr == BI_BITFIELDS) {
216     BMPbitField(br->hdr)[0] = (uint32)rdint32(br);
217     BMPbitField(br->hdr)[1] = (uint32)rdint32(br);
218     BMPbitField(br->hdr)[2] = (uint32)rdint32(br);
219 greg 2.13 } else if (rdbytes((char *)br->hdr->palette, palSiz, br) != BIR_OK)
220 greg 2.1 goto err;
221     /* read add'l information */
222     if (rdbytes(BMPinfo(br->hdr), br->hdr->infoSiz, br) != BIR_OK)
223     goto err;
224     /* read first scanline */
225     br->scanline = (uint8 *)calloc(getScanSiz(br->hdr), sizeof(uint8));
226     if (br->scanline == NULL)
227     goto err;
228     br->yscan = -1;
229 schorsch 2.4 if (seek != NULL && ((br->hdr->compr == BI_RLE8) |
230     (br->hdr->compr == BI_RLE4))) {
231 greg 2.1 BMPReader *newbr = (BMPReader *)realloc((void *)br,
232     sizeof(BMPReader) +
233     sizeof(br->scanpos[0]) *
234     br->hdr->height);
235     if (newbr == NULL)
236     goto err;
237     br = newbr;
238     memset((void *)(br->scanpos + 1), 0,
239     sizeof(br->scanpos[0])*br->hdr->height);
240     }
241     br->scanpos[0] = br->fpos;
242 greg 2.10 if (BMPreadScanline(br) == BIR_OK)
243     return br;
244 greg 2.1 err:
245     if (br->hdr != NULL)
246     free((void *)br->hdr);
247     if (br->scanline != NULL)
248     free((void *)br->scanline);
249     free((void *)br);
250     return NULL;
251     }
252    
253 greg 2.20 /* Determine if image is grayscale */
254 greg 2.1 int
255     BMPisGrayscale(const BMPHeader *hdr)
256     {
257     const RGBquad *rgbp;
258     int n;
259    
260     if (hdr == NULL)
261     return -1;
262     if (hdr->bpp > 8) /* assume they had a reason for it */
263     return 0;
264 greg 2.14 for (rgbp = hdr->palette, n = hdr->impColors; n-- > 0; rgbp++)
265     if ((rgbp->r != rgbp->g) | (rgbp->g != rgbp->b))
266 greg 2.1 return 0;
267     return 1; /* all colors neutral in map */
268     }
269    
270 greg 2.20 /* Read and decode next BMP scanline */
271 greg 2.1 int
272     BMPreadScanline(BMPReader *br)
273     {
274 greg 2.3 int n;
275     int8 *sp;
276    
277 greg 2.1 if (br->yscan + 1 >= br->hdr->height)
278     return BIR_EOF;
279     br->yscan++; /* prepare to read */
280 greg 2.3 n = getScanSiz(br->hdr); /* reading uncompressed data? */
281     if (br->hdr->compr == BI_UNCOMPR || br->hdr->compr == BI_BITFIELDS)
282     return rdbytes((char *)br->scanline, n, br);
283     /*
284 greg 2.10 * RLE4/RLE8 Decoding
285 greg 2.3 *
286     * Certain aspects of this scheme are completely insane, so
287     * we don't support them. Fortunately, they rarely appear.
288 greg 2.6 * One is the mid-file EOD (0x0001) and another is the ill-conceived
289 greg 2.3 * "delta" (0x0002), which is like a "goto" statement for bitmaps.
290 greg 2.10 * Whoever thought this up should be wrestled to the ground and told
291     * why it's impossible to support such a scheme in any reasonable way.
292 greg 2.3 * Also, RLE4 mode allows runs to stop halfway through a byte,
293     * which is likewise uncodeable, so we don't even try.
294     * Finally, the scanline break is ambiguous -- we assume here that
295     * it is required at the end of each scanline, though I haven't
296     * found anywhere this is written. Otherwise, we would read to
297     * the end of the scanline, assuming the next bit of data belongs
298     * the following scan. If a break follows the last pixel, as it
299     * seems to in the files I've tested out of Photoshop, you end up
300 greg 2.6 * painting every other line black. Also, I assume any skipped
301 greg 2.3 * pixels are painted with color 0, which is often black. Nowhere
302     * is it specified what we should assume for missing pixels. This
303     * is undoubtedly the most brain-dead format I've ever encountered.
304     */
305 greg 2.15 sp = (int8 *)br->scanline;
306 greg 2.10 n = br->hdr->width;
307     if (br->hdr->compr == BI_RLE4)
308     n = (n + 1) >> 1;
309 greg 2.3 while (n > 0) {
310     int skipOdd, len, val;
311    
312     if (rdbyte(len, br) == EOF)
313     return BIR_TRUNCATED;
314     if (len > 0) { /* got a run */
315     if (br->hdr->compr == BI_RLE4) {
316     if (len & 1)
317     return BIR_UNSUPPORTED;
318     len >>= 1;
319     }
320     if (len > n)
321     return BIR_RLERROR;
322     if (rdbyte(val, br) == EOF)
323     return BIR_TRUNCATED;
324     n -= len;
325     while (len--)
326     *sp++ = val;
327     continue;
328     }
329 greg 2.10 /* check for escape */
330 greg 2.3 switch (rdbyte(len, br)) {
331     case EOF:
332     return BIR_TRUNCATED;
333     case 0: /* end of line */
334     while (n--)
335     *sp++ = 0;
336 greg 2.6 /* leaves n == -1 as flag for test after loop */
337 greg 2.3 continue;
338     case 1: /* end of bitmap */
339     case 2: /* delta */
340     return BIR_UNSUPPORTED;
341     }
342     /* absolute mode */
343     if (br->hdr->compr == BI_RLE4) {
344     if (len & 1)
345     return BIR_UNSUPPORTED;
346     len >>= 1;
347     }
348     skipOdd = len & 1;
349     if (len > n)
350     return BIR_RLERROR;
351     n -= len;
352     while (len--) {
353     if (rdbyte(val, br) == EOF)
354     return BIR_TRUNCATED;
355     *sp++ = val;
356     }
357     if (skipOdd && rdbyte(val, br) == EOF)
358     return BIR_TRUNCATED;
359     }
360     /* verify break at end of line */
361 greg 2.6 if (!n && (rdbyte(n, br) != 0 || (rdbyte(n, br) != 0 &&
362     (n != 1 || br->yscan != br->hdr->height-1))))
363 greg 2.3 return BIR_RLERROR;
364     if (br->seek != NULL) /* record next scanline position */
365 greg 2.1 br->scanpos[br->yscan + 1] = br->fpos;
366     return BIR_OK;
367     }
368    
369 greg 2.20 /* Read a specific scanline */
370 greg 2.1 int
371     BMPseekScanline(int y, BMPReader *br)
372     {
373     int rv;
374     /* check arguments */
375     if (br == NULL)
376     return BIR_EOF;
377     if (y < 0)
378     return BIR_SEEKERR;
379     if (y >= br->hdr->height)
380     return BIR_EOF;
381     /* already read? */
382     if (y == br->yscan)
383     return BIR_OK;
384     /* shall we seek? */
385     if (y != br->yscan + 1 && br->seek != NULL) {
386     int yseek;
387     uint32 seekp;
388 greg 2.3 if (br->hdr->compr == BI_UNCOMPR ||
389     br->hdr->compr == BI_BITFIELDS) {
390 greg 2.1 yseek = y;
391     seekp = br->scanpos[0] + y*getScanSiz(br->hdr);
392     } else {
393     yseek = br->yscan + 1;
394     while (yseek < y && br->scanpos[yseek+1] != 0)
395     ++yseek;
396     if (y < yseek && br->scanpos[yseek=y] == 0)
397     return BIR_SEEKERR;
398     seekp = br->scanpos[yseek];
399     }
400     if ((rv = rdseek(seekp, br)) != BIR_OK)
401     return rv;
402     br->yscan = yseek - 1;
403     } else if (y < br->yscan) /* else we can't back up */
404     return BIR_SEEKERR;
405     /* read until we get there */
406     while (br->yscan < y)
407     if ((rv = BMPreadScanline(br)) != BIR_OK)
408     return rv;
409     return BIR_OK;
410     }
411    
412 greg 2.20 /* Get ith pixel from last scanline */
413 greg 2.1 RGBquad
414 greg 2.3 BMPdecodePixel(int i, const BMPReader *br)
415 greg 2.1 {
416 greg 2.3 static const uint32 std16mask[3] = {0x7c00, 0x3e0, 0x1f};
417 greg 2.1 static const RGBquad black = {0, 0, 0, 0};
418 greg 2.3 const uint32 *mask;
419     const uint8 *pp;
420     uint32 pval, v;
421     RGBquad cval;
422 greg 2.1
423 schorsch 2.2 if (((br == NULL) | (i < 0)) || i >= br->hdr->width)
424 greg 2.1 return black;
425    
426 greg 2.3 cval.padding = 0;
427    
428 greg 2.1 switch (br->hdr->bpp) {
429 greg 2.3 case 24:
430     pp = br->scanline + 3*i;
431     cval.b = *pp++;
432     cval.g = *pp++;
433     cval.r = *pp;
434 greg 2.1 return cval;
435     case 32:
436 greg 2.3 if (br->hdr->compr == BI_UNCOMPR)
437     return ((RGBquad *)br->scanline)[i];
438     /* convert bit fields */
439     pp = br->scanline + 4*i;
440     pval = *pp++;
441     pval |= *pp++ << 8;
442     pval |= *pp++ << 16;
443     pval |= *pp << 24;
444     mask = BMPbitField(br->hdr);
445     v = pval & mask[0];
446     while (v & ~0xff) v >>= 8;
447     cval.r = v;
448     v = pval & mask[1];
449     while (v & ~0xff) v >>= 8;
450     cval.g = v;
451     v = pval & mask[2];
452     while (v & ~0xff) v >>= 8;
453     cval.b = v;
454     return cval;
455 greg 2.1 case 8:
456     return br->hdr->palette[br->scanline[i]];
457     case 1:
458 greg 2.18 return br->hdr->palette[br->scanline[i>>3]>>(7-(i&7)) & 1];
459 greg 2.1 case 4:
460 greg 2.3 return br->hdr->palette[br->scanline[i>>1]>>(i&1?4:0) & 0xf];
461     case 16:
462     pp = br->scanline + 2*i;
463     pval = *pp++;
464     pval |= *pp++ << 8;
465     mask = std16mask;
466     if (br->hdr->compr == BI_BITFIELDS)
467     mask = BMPbitField(br->hdr);
468     cval.r = ((pval & mask[0]) << 8) / (mask[0] + 1);
469     cval.g = ((pval & mask[1]) << 8) / (mask[1] + 1);
470     cval.b = ((pval & mask[2]) << 8) / (mask[2] + 1);
471     return cval;
472 greg 2.1 }
473     return black; /* should never happen */
474     }
475    
476 greg 2.20 /* Free BMP reader resources */
477 greg 2.1 void
478     BMPfreeReader(BMPReader *br)
479     {
480     if (br == NULL)
481     return;
482     free((void *)br->hdr);
483     free((void *)br->scanline);
484     free((void *)br);
485     }
486    
487     /* stdio getc() callback */
488     int
489     stdio_getc(void *p)
490     {
491     if (!p)
492     return EOF;
493     return getc((FILE *)p);
494     }
495    
496     /* stdio putc() callback */
497     void
498     stdio_putc(int c, void *p)
499     {
500     if (p)
501     putc(c, (FILE *)p);
502     }
503    
504     /* stdio fseek() callback */
505     int
506     stdio_fseek(uint32 pos, void *p)
507     {
508     if (!p)
509     return -1;
510     return fseek((FILE *)p, (long)pos, 0);
511     }
512    
513 greg 2.20 /* Allocate uncompressed (24-bit) RGB header */
514 greg 2.1 BMPHeader *
515     BMPtruecolorHeader(int xr, int yr, int infolen)
516     {
517     BMPHeader *hdr;
518    
519     if (xr <= 0 || yr <= 0 || infolen < 0)
520     return NULL;
521     hdr = (BMPHeader *)malloc(sizeof(BMPHeader) - sizeof(hdr->palette) +
522     infolen);
523     if (hdr == NULL)
524     return NULL;
525     hdr->width = xr;
526     hdr->height = yr;
527     hdr->yIsDown = 0; /* default to upwards order */
528     hdr->bpp = 24;
529     hdr->compr = BI_UNCOMPR;
530     hdr->hRes = hdr->vRes = 2835; /* default to 72 ppi */
531     hdr->nColors = hdr->impColors = 0;
532     hdr->infoSiz = infolen;
533     return hdr;
534     }
535    
536 greg 2.20 /* Allocate color-mapped header (defaults to minimal grayscale) */
537 greg 2.1 BMPHeader *
538     BMPmappedHeader(int xr, int yr, int infolen, int ncolors)
539     {
540     int n;
541     BMPHeader *hdr;
542    
543     if (xr <= 0 || yr <= 0 || infolen < 0 || ncolors < 2)
544     return NULL;
545     if (ncolors <= 2)
546     n = 1;
547     else if (ncolors <= 16)
548     n = 4;
549     else if (ncolors <= 256)
550     n = 8;
551     else
552     return NULL;
553     hdr = (BMPHeader *)malloc(sizeof(BMPHeader) +
554 schorsch 2.17 sizeof(RGBquad)*((size_t)1<<n) -
555 greg 2.1 sizeof(hdr->palette) +
556     infolen);
557     if (hdr == NULL)
558     return NULL;
559     hdr->width = xr;
560     hdr->height = yr;
561     hdr->yIsDown = 0; /* default to upwards order */
562     hdr->bpp = n;
563 greg 2.5 hdr->compr = BI_UNCOMPR; /* compression needs seek */
564 greg 2.1 hdr->hRes = hdr->vRes = 2835; /* default to 72 ppi */
565     hdr->nColors = ncolors;
566     hdr->impColors = 0; /* says all colors important */
567     hdr->infoSiz = infolen;
568 schorsch 2.17 memset((void *)hdr->palette, 0, sizeof(RGBquad)*((size_t)1<<n) + infolen);
569 greg 2.1 for (n = ncolors; n--; )
570     hdr->palette[n].r = hdr->palette[n].g =
571     hdr->palette[n].b = n*255/(ncolors-1);
572     return hdr;
573     }
574    
575     /* put byte to writer */
576     #define wrbyte(c,bw) ( (*(bw)->cput)(c,(bw)->c_data), \
577 greg 2.3 ++(bw)->fpos > (bw)->flen ? \
578 greg 2.1 ((bw)->flen = (bw)->fpos) : \
579     (bw)->fpos )
580    
581 greg 2.20 /* Write out a string of bytes */
582 greg 2.1 static void
583     wrbytes(char *bp, uint32 n, BMPWriter *bw)
584     {
585     while (n--)
586     wrbyte(*bp++, bw);
587     }
588    
589 greg 2.20 /* Write 32-bit integer in littlendian order */
590 greg 2.1 static void
591     wrint32(int32 i, BMPWriter *bw)
592     {
593     wrbyte(i& 0xff, bw);
594     wrbyte(i>>8 & 0xff, bw);
595     wrbyte(i>>16 & 0xff, bw);
596 greg 2.3 wrbyte(i>>24 & 0xff, bw);
597 greg 2.1 }
598    
599 greg 2.20 /* Write 16-bit unsigned integer in littlendian order */
600 greg 2.1 static void
601     wruint16(uint16 ui, BMPWriter *bw)
602     {
603     wrbyte(ui & 0xff, bw);
604     wrbyte(ui>>8 & 0xff, bw);
605     }
606    
607 greg 2.20 /* Seek to the specified file position, returning 0 (BIR_OK) on success */
608 greg 2.1 static int
609     wrseek(uint32 pos, BMPWriter *bw)
610     {
611     if (pos == bw->fpos)
612     return BIR_OK;
613 greg 2.8 if (bw->seek == NULL)
614     return BIR_SEEKERR;
615 greg 2.1 if ((*bw->seek)(pos, bw->c_data) != 0)
616     return BIR_SEEKERR;
617     bw->fpos = pos;
618     if (pos > bw->flen)
619     bw->flen = pos;
620     return BIR_OK;
621     }
622    
623 greg 2.20 /* Open BMP stream for writing */
624 greg 2.1 BMPWriter *
625     BMPopenWriter(void (*cput)(int, void *), int (*seek)(uint32, void *),
626     void *c_data, BMPHeader *hdr)
627     {
628     BMPWriter *bw;
629     uint32 hdrSiz, palSiz, scanSiz, bmSiz;
630     /* check arguments */
631 greg 2.3 if (cput == NULL)
632 greg 2.1 return NULL;
633 greg 2.21 if (cput == &stdio_putc && c_data == NULL)
634     return NULL; /* stdio error condition */
635 greg 2.3 if (!BMPheaderOK(hdr))
636 greg 2.1 return NULL;
637 schorsch 2.4 if ((hdr->bpp == 16) | (hdr->compr == BI_RLE4))
638 greg 2.1 return NULL; /* unsupported */
639 greg 2.5 /* no seek means we may have the wrong file length, but most app's don't care
640 schorsch 2.4 if (seek == NULL && ((hdr->compr == BI_RLE8) | (hdr->compr == BI_RLE4)))
641 greg 2.1 return NULL;
642 greg 2.5 */
643 greg 2.1 /* compute sizes */
644     hdrSiz = 2 + 6*4 + 2*2 + 6*4;
645 greg 2.3 if (hdr->compr == BI_BITFIELDS)
646     hdrSiz += sizeof(uint32)*3;
647 greg 2.13 palSiz = sizeof(RGBquad)*hdr->nColors;
648 greg 2.1 scanSiz = getScanSiz(hdr);
649     bmSiz = hdr->height*scanSiz; /* wrong if compressed */
650     /* initialize writer */
651     bw = (BMPWriter *)malloc(sizeof(BMPWriter));
652     if (bw == NULL)
653     return NULL;
654     bw->hdr = hdr;
655     bw->yscan = 0;
656     bw->scanline = (uint8 *)calloc(scanSiz, sizeof(uint8));
657     if (bw->scanline == NULL) {
658     free((void *)bw);
659     return NULL;
660     }
661     bw->fbmp = hdrSiz + palSiz + hdr->infoSiz;
662     bw->fpos = bw->flen = 0;
663     bw->cput = cput;
664     bw->seek = seek;
665     bw->c_data = c_data;
666     /* write out header */
667     wrbyte('B', bw); wrbyte('M', bw); /* magic number */
668     wrint32(bw->fbmp + bmSiz, bw); /* file size */
669     wrint32(0, bw); /* reserved word */
670     wrint32(bw->fbmp, bw); /* offset to bitmap */
671     wrint32(hdrSiz - bw->fpos, bw); /* info header size */
672     wrint32(hdr->width, bw); /* bitmap width */
673     if (hdr->yIsDown) /* bitmap height */
674     wrint32(-hdr->height, bw);
675     else
676     wrint32(hdr->height, bw);
677     wruint16(1, bw); /* number of planes */
678     wruint16(hdr->bpp, bw); /* bits per pixel */
679     wrint32(hdr->compr, bw); /* compression mode */
680     wrint32(bmSiz, bw); /* bitmap size */
681     wrint32(hdr->hRes, bw); /* horizontal resolution */
682     wrint32(hdr->vRes, bw); /* vertical resolution */
683     wrint32(hdr->nColors, bw); /* # colors used */
684     wrint32(hdr->impColors, bw); /* # important colors */
685     /* write out color palette */
686     wrbytes((char *)hdr->palette, palSiz, bw);
687     /* write add'l information */
688     wrbytes(BMPinfo(hdr), hdr->infoSiz, bw);
689     #ifndef NDEBUG
690     if (bw->fpos != bw->fbmp) {
691     fputs("Coding error 1 in BMPopenWriter\n", stderr);
692     exit(1);
693     }
694     #endif
695     return bw;
696     }
697 greg 2.3
698 greg 2.20 /* Find position of next run of 5 or more identical bytes, or 255 if none */
699 greg 2.3 static int
700     findNextRun(const int8 *bp, int len)
701     {
702     int pos, cnt;
703     /* look for run */
704 schorsch 2.4 for (pos = 0; (len > 0) & (pos < 255); pos++, bp++, len--) {
705 greg 2.3 if (len < 5) /* no hope left? */
706     continue;
707     cnt = 1; /* else let's try it */
708     while (bp[cnt] == bp[0])
709 greg 2.5 if (++cnt >= 5)
710     return pos; /* long enough */
711 greg 2.3 }
712     return pos; /* didn't find any */
713     }
714 greg 2.1
715 greg 2.20 /* Write the current scanline */
716 greg 2.1 int
717     BMPwriteScanline(BMPWriter *bw)
718     {
719 greg 2.3 const int8 *sp;
720     int n;
721    
722 greg 2.1 if (bw->yscan >= bw->hdr->height)
723     return BIR_EOF;
724     /* writing uncompressed? */
725 greg 2.3 if (bw->hdr->compr == BI_UNCOMPR || bw->hdr->compr == BI_BITFIELDS) {
726 greg 2.1 uint32 scanSiz = getScanSiz(bw->hdr);
727     uint32 slpos = bw->fbmp + bw->yscan*scanSiz;
728     if (wrseek(slpos, bw) != BIR_OK)
729     return BIR_SEEKERR;
730     wrbytes((char *)bw->scanline, scanSiz, bw);
731     bw->yscan++;
732     return BIR_OK;
733     }
734 greg 2.3 /*
735     * RLE8 Encoding
736     *
737     * See the notes in BMPreadScanline() on this encoding. Needless
738     * to say, we avoid the nuttier aspects of this specification.
739     * We also assume that every scanline ends in a line break
740     * (0x0000) except for the last, which ends in a bitmap break
741     * (0x0001). We don't support RLE4 at all; it's too awkward.
742     */
743 greg 2.15 sp = (const int8 *)bw->scanline;
744 greg 2.3 n = bw->hdr->width;
745     while (n > 0) {
746     int cnt, val;
747     cnt = findNextRun(sp, n); /* 0-255 < n */
748 greg 2.10 if (cnt >= 3) { /* output absolute */
749 greg 2.3 int skipOdd = cnt & 1;
750     wrbyte(0, bw);
751     wrbyte(cnt, bw);
752     n -= cnt;
753     while (cnt--)
754     wrbyte(*sp++, bw);
755     if (skipOdd)
756     wrbyte(0, bw);
757     }
758     if (n <= 0) /* was that it? */
759     break;
760 greg 2.19 val = *sp++; /* output run */
761     for (cnt = 1; --n && cnt < 255; cnt++, sp++)
762     if (*sp != val)
763 greg 2.5 break;
764 greg 2.3 wrbyte(cnt, bw);
765     wrbyte(val, bw);
766     }
767 greg 2.5 bw->yscan++; /* write line break or EOD */
768 greg 2.1 if (bw->yscan == bw->hdr->height) {
769 greg 2.3 wrbyte(0, bw); wrbyte(1, bw); /* end of bitmap marker */
770 greg 2.8 if (wrseek(2, bw) != BIR_OK)
771 greg 2.5 return BIR_OK; /* no one may care */
772 greg 2.8 wrint32(bw->flen, bw); /* correct file length */
773     if (wrseek(34, bw) != BIR_OK)
774     return BIR_OK;
775 greg 2.6 wrint32(bw->flen-bw->fbmp, bw); /* correct bitmap length */
776 greg 2.3 } else {
777     wrbyte(0, bw); wrbyte(0, bw); /* end of line marker */
778 greg 2.1 }
779     return BIR_OK;
780     }
781    
782 greg 2.20 /* Free BMP writer resources */
783 greg 2.1 void
784     BMPfreeWriter(BMPWriter *bw)
785     {
786     if (bw == NULL)
787     return;
788     free((void *)bw->hdr);
789     free((void *)bw->scanline);
790     free((void *)bw);
791     }