OLD | NEW |
1 /* Copyright (C)2004 Landmark Graphics Corporation | 1 /* |
2 * Copyright (C)2005 Sun Microsystems, Inc. | 2 * Copyright (C)2011 D. R. Commander. All Rights Reserved. |
3 * | 3 * |
4 * This library is free software and may be redistributed and/or modified under | 4 * Redistribution and use in source and binary forms, with or without |
5 * the terms of the wxWindows Library License, Version 3.1 or (at your option) | 5 * modification, are permitted provided that the following conditions are met: |
6 * any later version. The full license is in the LICENSE.txt file included | |
7 * with this distribution. | |
8 * | 6 * |
9 * This library is distributed in the hope that it will be useful, | 7 * - Redistributions of source code must retain the above copyright notice, |
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of | 8 * this list of conditions and the following disclaimer. |
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 9 * - Redistributions in binary form must reproduce the above copyright notice, |
12 * wxWindows Library License for more details. | 10 * this list of conditions and the following disclaimer in the documentation |
13 */ | 11 * and/or other materials provided with the distribution. |
14 | 12 * - Neither the name of the libjpeg-turbo Project nor the names of its |
15 #include <fcntl.h> | 13 * contributors may be used to endorse or promote products derived from this |
16 #include <sys/types.h> | 14 * software without specific prior written permission. |
17 #include <sys/stat.h> | 15 * |
18 #include <errno.h> | 16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS", |
19 #include <stdlib.h> | 17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| 18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
| 19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE |
| 20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
| 21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
| 22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
| 23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
| 24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
| 25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
| 26 * POSSIBILITY OF SUCH DAMAGE. |
| 27 */ |
| 28 |
20 #include <stdio.h> | 29 #include <stdio.h> |
21 #include <string.h> | 30 #include <string.h> |
22 #ifdef _WIN32 | 31 #include <setjmp.h> |
23 #include <io.h> | 32 #include <errno.h> |
24 #else | 33 #include "cdjpeg.h" |
25 #include <unistd.h> | 34 #include <jpeglib.h> |
26 #endif | 35 #include <jpegint.h> |
27 #include "./rrutil.h" | 36 #include "tjutil.h" |
28 #include "./bmp.h" | 37 #include "bmp.h" |
29 | 38 |
30 #ifndef BI_BITFIELDS | 39 |
31 #define BI_BITFIELDS 3L | 40 /* This duplicates the functionality of the VirtualGL bitmap library using |
32 #endif | 41 the components from cjpeg and djpeg */ |
33 #ifndef BI_RGB | 42 |
34 #define BI_RGB 0L | 43 |
35 #endif | 44 /* Error handling (based on example in example.c) */ |
36 | 45 |
37 #define BMPHDRSIZE 54 | 46 static char errStr[JMSG_LENGTH_MAX]="No error"; |
38 typedef struct _bmphdr | 47 |
39 { | 48 struct my_error_mgr |
40 » unsigned short bfType; | 49 { |
41 » unsigned int bfSize; | 50 » struct jpeg_error_mgr pub; |
42 » unsigned short bfReserved1, bfReserved2; | 51 » jmp_buf setjmp_buffer; |
43 » unsigned int bfOffBits; | 52 }; |
44 | 53 typedef struct my_error_mgr *my_error_ptr; |
45 » unsigned int biSize; | 54 |
46 » int biWidth, biHeight; | 55 static void my_error_exit(j_common_ptr cinfo) |
47 » unsigned short biPlanes, biBitCount; | 56 { |
48 » unsigned int biCompression, biSizeImage; | 57 » my_error_ptr myerr=(my_error_ptr)cinfo->err; |
49 » int biXPelsPerMeter, biYPelsPerMeter; | 58 » (*cinfo->err->output_message)(cinfo); |
50 » unsigned int biClrUsed, biClrImportant; | 59 » longjmp(myerr->setjmp_buffer, 1); |
51 } bmphdr; | 60 } |
52 | 61 |
53 static const char *__bmperr="No error"; | 62 /* Based on output_message() in jerror.c */ |
54 | 63 |
55 static const int ps[BMPPIXELFORMATS]={3, 4, 3, 4, 4, 4}; | 64 static void my_output_message(j_common_ptr cinfo) |
56 static const int roffset[BMPPIXELFORMATS]={0, 0, 2, 2, 3, 1}; | 65 { |
57 static const int goffset[BMPPIXELFORMATS]={1, 1, 1, 1, 2, 2}; | 66 » (*cinfo->err->format_message)(cinfo, errStr); |
58 static const int boffset[BMPPIXELFORMATS]={2, 2, 0, 0, 1, 3}; | 67 } |
59 | 68 |
60 #define _throw(m) {__bmperr=m; retcode=-1; goto finally;} | 69 #define _throw(m) {snprintf(errStr, JMSG_LENGTH_MAX, "%s", m); \ |
61 #define _unix(f) {if((f)==-1) _throw(strerror(errno));} | 70 » retval=-1; goto bailout;} |
62 #define _catch(f) {if((f)==-1) {retcode=-1; goto finally;}} | 71 #define _throwunix(m) {snprintf(errStr, JMSG_LENGTH_MAX, "%s\n%s", m, \ |
63 | 72 » strerror(errno)); retval=-1; goto bailout;} |
64 #define readme(fd, addr, size) \ | 73 |
65 » if((bytesread=read(fd, addr, (size)))==-1) _throw(strerror(errno)); \ | 74 |
66 » if(bytesread!=(size)) _throw("Read error"); | 75 static void pixelconvert(unsigned char *srcbuf, int srcpf, int srcbottomup, |
67 | 76 » unsigned char *dstbuf, int dstpf, int dstbottomup, int w, int h) |
68 void pixelconvert(unsigned char *srcbuf, enum BMPPIXELFORMAT srcformat, | 77 { |
69 » int srcpitch, unsigned char *dstbuf, enum BMPPIXELFORMAT dstformat, int
dstpitch, | 78 » unsigned char *srcptr=srcbuf, *srcptr2; |
70 » int w, int h, int flip) | 79 » int srcps=tjPixelSize[srcpf]; |
71 { | 80 » int srcstride=srcbottomup? -w*srcps:w*srcps; |
72 » unsigned char *srcptr, *srcptr0, *dstptr, *dstptr0; | 81 » unsigned char *dstptr=dstbuf, *dstptr2; |
73 » int i, j; | 82 » int dstps=tjPixelSize[dstpf]; |
74 | 83 » int dststride=dstbottomup? -w*dstps:w*dstps; |
75 » srcptr=flip? &srcbuf[srcpitch*(h-1)]:srcbuf; | 84 » int row, col; |
76 » for(j=0, dstptr=dstbuf; j<h; j++, | 85 |
77 » » srcptr+=flip? -srcpitch:srcpitch, dstptr+=dstpitch) | 86 » if(srcbottomup) srcptr=&srcbuf[w*srcps*(h-1)]; |
78 » { | 87 » if(dstbottomup) dstptr=&dstbuf[w*dstps*(h-1)]; |
79 » » for(i=0, srcptr0=srcptr, dstptr0=dstptr; i<w; i++, | 88 » for(row=0; row<h; row++, srcptr+=srcstride, dstptr+=dststride) |
80 » » » srcptr0+=ps[srcformat], dstptr0+=ps[dstformat]) | 89 » { |
| 90 » » for(col=0, srcptr2=srcptr, dstptr2=dstptr; col<w; col++, srcptr2
+=srcps, |
| 91 » » » dstptr2+=dstps) |
81 { | 92 { |
82 » » » dstptr0[roffset[dstformat]]=srcptr0[roffset[srcformat]]; | 93 » » » dstptr2[tjRedOffset[dstpf]]=srcptr2[tjRedOffset[srcpf]]; |
83 » » » dstptr0[goffset[dstformat]]=srcptr0[goffset[srcformat]]; | 94 » » » dstptr2[tjGreenOffset[dstpf]]=srcptr2[tjGreenOffset[srcp
f]]; |
84 » » » dstptr0[boffset[dstformat]]=srcptr0[boffset[srcformat]]; | 95 » » » dstptr2[tjBlueOffset[dstpf]]=srcptr2[tjBlueOffset[srcpf]
]; |
85 } | 96 } |
86 } | 97 } |
87 } | 98 } |
88 | 99 |
89 int loadppm(int *fd, unsigned char **buf, int *w, int *h, | 100 |
90 » enum BMPPIXELFORMAT f, int align, int dstbottomup, int ascii) | 101 int loadbmp(char *filename, unsigned char **buf, int *w, int *h, |
91 { | 102 » int dstpf, int bottomup) |
92 » FILE *fs=NULL; int retcode=0, scalefactor, dstpitch; | 103 { |
93 » unsigned char *tempbuf=NULL; char temps[255], temps2[255]; | 104 » int retval=0, dstps, srcpf, tempc; |
94 » int numread=0, totalread=0, pixel[3], i, j; | 105 » struct jpeg_compress_struct cinfo; |
95 | 106 » struct my_error_mgr jerr; |
96 » if((fs=fdopen(*fd, "r"))==NULL) _throw(strerror(errno)); | 107 » cjpeg_source_ptr src; |
97 | 108 » FILE *file=NULL; |
98 » do | 109 |
99 » { | 110 » if(!filename || !buf || !w || !h || dstpf<0 || dstpf>=TJ_NUMPF) |
100 » » if(!fgets(temps, 255, fs)) _throw("Read error"); | 111 » » _throw("loadbmp(): Invalid argument"); |
101 » » if(strlen(temps)==0 || temps[0]=='\n') continue; | 112 |
102 » » if(sscanf(temps, "%s", temps2)==1 && temps2[1]=='#') continue; | 113 » if((file=fopen(filename, "rb"))==NULL) |
103 » » switch(totalread) | 114 » » _throwunix("loadbmp(): Cannot open input file"); |
| 115 |
| 116 » cinfo.err=jpeg_std_error(&jerr.pub); |
| 117 » jerr.pub.error_exit=my_error_exit; |
| 118 » jerr.pub.output_message=my_output_message; |
| 119 |
| 120 » if(setjmp(jerr.setjmp_buffer)) |
| 121 » { |
| 122 » » /* If we get here, the JPEG code has signaled an error. */ |
| 123 » » retval=-1; goto bailout; |
| 124 » } |
| 125 |
| 126 » jpeg_create_compress(&cinfo); |
| 127 » if((tempc=getc(file))<0 || ungetc(tempc, file)==EOF) |
| 128 » » _throwunix("loadbmp(): Could not read input file") |
| 129 » else if(tempc==EOF) _throw("loadbmp(): Input file contains no data"); |
| 130 |
| 131 » if(tempc=='B') |
| 132 » { |
| 133 » » if((src=jinit_read_bmp(&cinfo))==NULL) |
| 134 » » » _throw("loadbmp(): Could not initialize bitmap loader"); |
| 135 » } |
| 136 » else if(tempc=='P') |
| 137 » { |
| 138 » » if((src=jinit_read_ppm(&cinfo))==NULL) |
| 139 » » » _throw("loadbmp(): Could not initialize bitmap loader"); |
| 140 » } |
| 141 » else _throw("loadbmp(): Unsupported file type"); |
| 142 |
| 143 » src->input_file=file; |
| 144 » (*src->start_input)(&cinfo, src); |
| 145 » (*cinfo.mem->realize_virt_arrays)((j_common_ptr)&cinfo); |
| 146 |
| 147 » *w=cinfo.image_width; *h=cinfo.image_height; |
| 148 |
| 149 » if(cinfo.input_components==1 && cinfo.in_color_space==JCS_RGB) |
| 150 » » srcpf=TJPF_GRAY; |
| 151 » else srcpf=TJPF_RGB; |
| 152 |
| 153 » dstps=tjPixelSize[dstpf]; |
| 154 » if((*buf=(unsigned char *)malloc((*w)*(*h)*dstps))==NULL) |
| 155 » » _throw("loadbmp(): Memory allocation failure"); |
| 156 |
| 157 » while(cinfo.next_scanline<cinfo.image_height) |
| 158 » { |
| 159 » » int i, nlines=(*src->get_pixel_rows)(&cinfo, src); |
| 160 » » for(i=0; i<nlines; i++) |
104 { | 161 { |
105 » » » case 0: | 162 » » » unsigned char *outbuf; int row; |
106 » » » » if((numread=sscanf(temps, "%d %d %d", w, h, &sca
lefactor))==EOF) | 163 » » » row=cinfo.next_scanline+i; |
107 » » » » » _throw("Read error"); | 164 » » » if(bottomup) outbuf=&(*buf)[((*h)-row-1)*(*w)*dstps]; |
108 » » » » break; | 165 » » » else outbuf=&(*buf)[row*(*w)*dstps]; |
109 » » » case 1: | 166 » » » pixelconvert(src->buffer[i], srcpf, 0, outbuf, dstpf, bo
ttomup, *w, |
110 » » » » if((numread=sscanf(temps, "%d %d", h, &scalefact
or))==EOF) | 167 » » » » nlines); |
111 » » » » » _throw("Read error"); | |
112 » » » » break; | |
113 » » » case 2: | |
114 » » » » if((numread=sscanf(temps, "%d", &scalefactor))==
EOF) | |
115 » » » » » _throw("Read error"); | |
116 » » » » break; | |
117 } | 168 } |
118 » » totalread+=numread; | 169 » » cinfo.next_scanline+=nlines; |
119 » } while(totalread<3); | 170 } |
120 » if((*w)<1 || (*h)<1 || scalefactor<1) _throw("Corrupt PPM header"); | 171 |
121 | 172 » (*src->finish_input)(&cinfo, src); |
122 » dstpitch=(((*w)*ps[f])+(align-1))&(~(align-1)); | 173 |
123 » if((*buf=(unsigned char *)malloc(dstpitch*(*h)))==NULL) | 174 » bailout: |
124 » » _throw("Memory allocation error"); | 175 » jpeg_destroy_compress(&cinfo); |
125 » if(ascii) | 176 » if(file) fclose(file); |
126 » { | 177 » if(retval<0 && buf && *buf) {free(*buf); *buf=NULL;} |
127 » » for(j=0; j<*h; j++) | 178 » return retval; |
| 179 } |
| 180 |
| 181 |
| 182 int savebmp(char *filename, unsigned char *buf, int w, int h, int srcpf, |
| 183 » int bottomup) |
| 184 { |
| 185 » int retval=0, srcps, dstpf; |
| 186 » struct jpeg_decompress_struct dinfo; |
| 187 » struct my_error_mgr jerr; |
| 188 » djpeg_dest_ptr dst; |
| 189 » FILE *file=NULL; |
| 190 » char *ptr=NULL; |
| 191 |
| 192 » if(!filename || !buf || w<1 || h<1 || srcpf<0 || srcpf>=TJ_NUMPF) |
| 193 » » _throw("savebmp(): Invalid argument"); |
| 194 |
| 195 » if((file=fopen(filename, "wb"))==NULL) |
| 196 » » _throwunix("savebmp(): Cannot open output file"); |
| 197 |
| 198 » dinfo.err=jpeg_std_error(&jerr.pub); |
| 199 » jerr.pub.error_exit=my_error_exit; |
| 200 » jerr.pub.output_message=my_output_message; |
| 201 |
| 202 » if(setjmp(jerr.setjmp_buffer)) |
| 203 » { |
| 204 » » /* If we get here, the JPEG code has signaled an error. */ |
| 205 » » retval=-1; goto bailout; |
| 206 » } |
| 207 |
| 208 » jpeg_create_decompress(&dinfo); |
| 209 » if(srcpf==TJPF_GRAY) |
| 210 » { |
| 211 » » dinfo.out_color_components=dinfo.output_components=1; |
| 212 » » dinfo.out_color_space=JCS_GRAYSCALE; |
| 213 » } |
| 214 » else |
| 215 » { |
| 216 » » dinfo.out_color_components=dinfo.output_components=3; |
| 217 » » dinfo.out_color_space=JCS_RGB; |
| 218 » } |
| 219 » dinfo.image_width=w; dinfo.image_height=h; |
| 220 » dinfo.global_state=DSTATE_READY; |
| 221 » dinfo.scale_num=dinfo.scale_denom=1; |
| 222 |
| 223 » ptr=strrchr(filename, '.'); |
| 224 » if(ptr && !strcasecmp(ptr, ".bmp")) |
| 225 » { |
| 226 » » if((dst=jinit_write_bmp(&dinfo, 0))==NULL) |
| 227 » » » _throw("savebmp(): Could not initialize bitmap writer"); |
| 228 » } |
| 229 » else |
| 230 » { |
| 231 » » if((dst=jinit_write_ppm(&dinfo))==NULL) |
| 232 » » » _throw("savebmp(): Could not initialize PPM writer"); |
| 233 » } |
| 234 |
| 235 dst->output_file=file; |
| 236 » (*dst->start_output)(&dinfo, dst); |
| 237 » (*dinfo.mem->realize_virt_arrays)((j_common_ptr)&dinfo); |
| 238 |
| 239 » if(srcpf==TJPF_GRAY) dstpf=srcpf; |
| 240 » else dstpf=TJPF_RGB; |
| 241 » srcps=tjPixelSize[srcpf]; |
| 242 |
| 243 » while(dinfo.output_scanline<dinfo.output_height) |
| 244 » { |
| 245 » » int i, nlines=dst->buffer_height; |
| 246 » » for(i=0; i<nlines; i++) |
128 { | 247 { |
129 » » » for(i=0; i<*w; i++) | 248 » » » unsigned char *inbuf; int row; |
130 » » » { | 249 » » » row=dinfo.output_scanline+i; |
131 » » » » if(fscanf(fs, "%d%d%d", &pixel[0], &pixel[1], &p
ixel[2])!=3) | 250 » » » if(bottomup) inbuf=&buf[(h-row-1)*w*srcps]; |
132 » » » » » _throw("Read error"); | 251 » » » else inbuf=&buf[row*w*srcps]; |
133 » » » » (*buf)[j*dstpitch+i*ps[f]+roffset[f]]=(unsigned
char)(pixel[0]*255/scalefactor); | 252 » » » pixelconvert(inbuf, srcpf, bottomup, dst->buffer[i], dst
pf, 0, w, |
134 » » » » (*buf)[j*dstpitch+i*ps[f]+goffset[f]]=(unsigned
char)(pixel[1]*255/scalefactor); | 253 » » » » nlines); |
135 » » » » (*buf)[j*dstpitch+i*ps[f]+boffset[f]]=(unsigned
char)(pixel[2]*255/scalefactor); | |
136 » » » } | |
137 } | 254 } |
138 » } | 255 » » (*dst->put_pixel_rows)(&dinfo, dst, nlines); |
139 » else | 256 » » dinfo.output_scanline+=nlines; |
140 » { | 257 } |
141 » » if(scalefactor!=255) | 258 |
142 » » » _throw("Binary PPMs must have 8-bit components"); | 259 » (*dst->finish_output)(&dinfo, dst); |
143 » » if((tempbuf=(unsigned char *)malloc((*w)*(*h)*3))==NULL) | 260 |
144 » » » _throw("Memory allocation error"); | 261 » bailout: |
145 » » if(fread(tempbuf, (*w)*(*h)*3, 1, fs)!=1) _throw("Read error"); | 262 » jpeg_destroy_decompress(&dinfo); |
146 » » pixelconvert(tempbuf, BMP_RGB, (*w)*3, *buf, f, dstpitch, *w, *h
, dstbottomup); | 263 » if(file) fclose(file); |
147 » } | 264 » return retval; |
148 | |
149 » finally: | |
150 » if(fs) {fclose(fs); *fd=-1;} | |
151 » if(tempbuf) free(tempbuf); | |
152 » return retcode; | |
153 } | |
154 | |
155 | |
156 int loadbmp(char *filename, unsigned char **buf, int *w, int *h, | |
157 » enum BMPPIXELFORMAT f, int align, int dstbottomup) | |
158 { | |
159 » int fd=-1, bytesread, srcpitch, srcbottomup=1, srcps, dstpitch, | |
160 » » retcode=0; | |
161 » unsigned char *tempbuf=NULL; | |
162 » bmphdr bh; int flags=O_RDONLY; | |
163 | |
164 » dstbottomup=dstbottomup? 1:0; | |
165 » #ifdef _WIN32 | |
166 » flags|=O_BINARY; | |
167 » #endif | |
168 » if(!filename || !buf || !w || !h || f<0 || f>BMPPIXELFORMATS-1 || align<
1) | |
169 » » _throw("invalid argument to loadbmp()"); | |
170 » if((align&(align-1))!=0) | |
171 » » _throw("Alignment must be a power of 2"); | |
172 » _unix(fd=open(filename, flags)); | |
173 | |
174 » readme(fd, &bh.bfType, sizeof(unsigned short)); | |
175 » if(!littleendian())» bh.bfType=byteswap16(bh.bfType); | |
176 | |
177 » if(bh.bfType==0x3650) | |
178 » { | |
179 » » _catch(loadppm(&fd, buf, w, h, f, align, dstbottomup, 0)); | |
180 » » goto finally; | |
181 » } | |
182 » if(bh.bfType==0x3350) | |
183 » { | |
184 » » _catch(loadppm(&fd, buf, w, h, f, align, dstbottomup, 1)); | |
185 » » goto finally; | |
186 » } | |
187 | |
188 » readme(fd, &bh.bfSize, sizeof(unsigned int)); | |
189 » readme(fd, &bh.bfReserved1, sizeof(unsigned short)); | |
190 » readme(fd, &bh.bfReserved2, sizeof(unsigned short)); | |
191 » readme(fd, &bh.bfOffBits, sizeof(unsigned int)); | |
192 » readme(fd, &bh.biSize, sizeof(unsigned int)); | |
193 » readme(fd, &bh.biWidth, sizeof(int)); | |
194 » readme(fd, &bh.biHeight, sizeof(int)); | |
195 » readme(fd, &bh.biPlanes, sizeof(unsigned short)); | |
196 » readme(fd, &bh.biBitCount, sizeof(unsigned short)); | |
197 » readme(fd, &bh.biCompression, sizeof(unsigned int)); | |
198 » readme(fd, &bh.biSizeImage, sizeof(unsigned int)); | |
199 » readme(fd, &bh.biXPelsPerMeter, sizeof(int)); | |
200 » readme(fd, &bh.biYPelsPerMeter, sizeof(int)); | |
201 » readme(fd, &bh.biClrUsed, sizeof(unsigned int)); | |
202 » readme(fd, &bh.biClrImportant, sizeof(unsigned int)); | |
203 | |
204 » if(!littleendian()) | |
205 » { | |
206 » » bh.bfSize=byteswap(bh.bfSize); | |
207 » » bh.bfOffBits=byteswap(bh.bfOffBits); | |
208 » » bh.biSize=byteswap(bh.biSize); | |
209 » » bh.biWidth=byteswap(bh.biWidth); | |
210 » » bh.biHeight=byteswap(bh.biHeight); | |
211 » » bh.biPlanes=byteswap16(bh.biPlanes); | |
212 » » bh.biBitCount=byteswap16(bh.biBitCount); | |
213 » » bh.biCompression=byteswap(bh.biCompression); | |
214 » » bh.biSizeImage=byteswap(bh.biSizeImage); | |
215 » » bh.biXPelsPerMeter=byteswap(bh.biXPelsPerMeter); | |
216 » » bh.biYPelsPerMeter=byteswap(bh.biYPelsPerMeter); | |
217 » » bh.biClrUsed=byteswap(bh.biClrUsed); | |
218 » » bh.biClrImportant=byteswap(bh.biClrImportant); | |
219 » } | |
220 | |
221 » if(bh.bfType!=0x4d42 || bh.bfOffBits<BMPHDRSIZE | |
222 » || bh.biWidth<1 || bh.biHeight==0) | |
223 » » _throw("Corrupt bitmap header"); | |
224 » if((bh.biBitCount!=24 && bh.biBitCount!=32) || bh.biCompression!=BI_RGB) | |
225 » » _throw("Only uncompessed RGB bitmaps are supported"); | |
226 | |
227 » *w=bh.biWidth; *h=bh.biHeight; srcps=bh.biBitCount/8; | |
228 » if(*h<0) {*h=-(*h); srcbottomup=0;} | |
229 » srcpitch=(((*w)*srcps)+3)&(~3); | |
230 » dstpitch=(((*w)*ps[f])+(align-1))&(~(align-1)); | |
231 | |
232 » if(srcpitch*(*h)+bh.bfOffBits!=bh.bfSize) _throw("Corrupt bitmap header"
); | |
233 » if((tempbuf=(unsigned char *)malloc(srcpitch*(*h)))==NULL | |
234 » || (*buf=(unsigned char *)malloc(dstpitch*(*h)))==NULL) | |
235 » » _throw("Memory allocation error"); | |
236 » if(lseek(fd, (long)bh.bfOffBits, SEEK_SET)!=(long)bh.bfOffBits) | |
237 » » _throw(strerror(errno)); | |
238 » _unix(bytesread=read(fd, tempbuf, srcpitch*(*h))); | |
239 » if(bytesread!=srcpitch*(*h)) _throw("Read error"); | |
240 | |
241 » pixelconvert(tempbuf, BMP_BGR, srcpitch, *buf, f, dstpitch, *w, *h, | |
242 » » srcbottomup!=dstbottomup); | |
243 | |
244 » finally: | |
245 » if(tempbuf) free(tempbuf); | |
246 » if(fd!=-1) close(fd); | |
247 » return retcode; | |
248 } | |
249 | |
250 #define writeme(fd, addr, size) \ | |
251 » if((byteswritten=write(fd, addr, (size)))==-1) _throw(strerror(errno));
\ | |
252 » if(byteswritten!=(size)) _throw("Write error"); | |
253 | |
254 int saveppm(char *filename, unsigned char *buf, int w, int h, | |
255 » enum BMPPIXELFORMAT f, int srcpitch, int srcbottomup) | |
256 { | |
257 » FILE *fs=NULL; int retcode=0; | |
258 » unsigned char *tempbuf=NULL; | |
259 | |
260 » if((fs=fopen(filename, "wb"))==NULL) _throw(strerror(errno)); | |
261 » if(fprintf(fs, "P6\n")<1) _throw("Write error"); | |
262 » if(fprintf(fs, "%d %d\n", w, h)<1) _throw("Write error"); | |
263 » if(fprintf(fs, "255\n")<1) _throw("Write error"); | |
264 | |
265 » if((tempbuf=(unsigned char *)malloc(w*h*3))==NULL) | |
266 » » _throw("Memory allocation error"); | |
267 | |
268 » pixelconvert(buf, f, srcpitch, tempbuf, BMP_RGB, w*3, w, h, | |
269 » » srcbottomup); | |
270 | |
271 » if((fwrite(tempbuf, w*h*3, 1, fs))!=1) _throw("Write error"); | |
272 | |
273 » finally: | |
274 » if(tempbuf) free(tempbuf); | |
275 » if(fs) fclose(fs); | |
276 » return retcode; | |
277 } | |
278 | |
279 int savebmp(char *filename, unsigned char *buf, int w, int h, | |
280 » enum BMPPIXELFORMAT f, int srcpitch, int srcbottomup) | |
281 { | |
282 » int fd=-1, byteswritten, dstpitch, retcode=0; | |
283 » int flags=O_RDWR|O_CREAT|O_TRUNC; | |
284 » unsigned char *tempbuf=NULL; char *temp; | |
285 » bmphdr bh; int mode; | |
286 | |
287 » #ifdef _WIN32 | |
288 » flags|=O_BINARY; mode=_S_IREAD|_S_IWRITE; | |
289 » #else | |
290 » mode=S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH; | |
291 » #endif | |
292 » if(!filename || !buf || w<1 || h<1 || f<0 || f>BMPPIXELFORMATS-1 || srcp
itch<0) | |
293 » » _throw("bad argument to savebmp()"); | |
294 | |
295 » if(srcpitch==0) srcpitch=w*ps[f]; | |
296 | |
297 » if((temp=strrchr(filename, '.'))!=NULL) | |
298 » { | |
299 » » if(!stricmp(temp, ".ppm")) | |
300 » » » return saveppm(filename, buf, w, h, f, srcpitch, srcbott
omup); | |
301 » } | |
302 | |
303 » _unix(fd=open(filename, flags, mode)); | |
304 » dstpitch=((w*3)+3)&(~3); | |
305 | |
306 » bh.bfType=0x4d42; | |
307 » bh.bfSize=BMPHDRSIZE+dstpitch*h; | |
308 » bh.bfReserved1=0; bh.bfReserved2=0; | |
309 » bh.bfOffBits=BMPHDRSIZE; | |
310 » bh.biSize=40; | |
311 » bh.biWidth=w; bh.biHeight=h; | |
312 » bh.biPlanes=0; bh.biBitCount=24; | |
313 » bh.biCompression=BI_RGB; bh.biSizeImage=0; | |
314 » bh.biXPelsPerMeter=0; bh.biYPelsPerMeter=0; | |
315 » bh.biClrUsed=0; bh.biClrImportant=0; | |
316 | |
317 » if(!littleendian()) | |
318 » { | |
319 » » bh.bfType=byteswap16(bh.bfType); | |
320 » » bh.bfSize=byteswap(bh.bfSize); | |
321 » » bh.bfOffBits=byteswap(bh.bfOffBits); | |
322 » » bh.biSize=byteswap(bh.biSize); | |
323 » » bh.biWidth=byteswap(bh.biWidth); | |
324 » » bh.biHeight=byteswap(bh.biHeight); | |
325 » » bh.biPlanes=byteswap16(bh.biPlanes); | |
326 » » bh.biBitCount=byteswap16(bh.biBitCount); | |
327 » » bh.biCompression=byteswap(bh.biCompression); | |
328 » » bh.biSizeImage=byteswap(bh.biSizeImage); | |
329 » » bh.biXPelsPerMeter=byteswap(bh.biXPelsPerMeter); | |
330 » » bh.biYPelsPerMeter=byteswap(bh.biYPelsPerMeter); | |
331 » » bh.biClrUsed=byteswap(bh.biClrUsed); | |
332 » » bh.biClrImportant=byteswap(bh.biClrImportant); | |
333 » } | |
334 | |
335 » writeme(fd, &bh.bfType, sizeof(unsigned short)); | |
336 » writeme(fd, &bh.bfSize, sizeof(unsigned int)); | |
337 » writeme(fd, &bh.bfReserved1, sizeof(unsigned short)); | |
338 » writeme(fd, &bh.bfReserved2, sizeof(unsigned short)); | |
339 » writeme(fd, &bh.bfOffBits, sizeof(unsigned int)); | |
340 » writeme(fd, &bh.biSize, sizeof(unsigned int)); | |
341 » writeme(fd, &bh.biWidth, sizeof(int)); | |
342 » writeme(fd, &bh.biHeight, sizeof(int)); | |
343 » writeme(fd, &bh.biPlanes, sizeof(unsigned short)); | |
344 » writeme(fd, &bh.biBitCount, sizeof(unsigned short)); | |
345 » writeme(fd, &bh.biCompression, sizeof(unsigned int)); | |
346 » writeme(fd, &bh.biSizeImage, sizeof(unsigned int)); | |
347 » writeme(fd, &bh.biXPelsPerMeter, sizeof(int)); | |
348 » writeme(fd, &bh.biYPelsPerMeter, sizeof(int)); | |
349 » writeme(fd, &bh.biClrUsed, sizeof(unsigned int)); | |
350 » writeme(fd, &bh.biClrImportant, sizeof(unsigned int)); | |
351 | |
352 » if((tempbuf=(unsigned char *)malloc(dstpitch*h))==NULL) | |
353 » » _throw("Memory allocation error"); | |
354 | |
355 » pixelconvert(buf, f, srcpitch, tempbuf, BMP_BGR, dstpitch, w, h, | |
356 » » !srcbottomup); | |
357 | |
358 » if((byteswritten=write(fd, tempbuf, dstpitch*h))!=dstpitch*h) | |
359 » » _throw(strerror(errno)); | |
360 | |
361 » finally: | |
362 » if(tempbuf) free(tempbuf); | |
363 » if(fd!=-1) close(fd); | |
364 » return retcode; | |
365 } | 265 } |
366 | 266 |
367 const char *bmpgeterr(void) | 267 const char *bmpgeterr(void) |
368 { | 268 { |
369 » return __bmperr; | 269 » return errStr; |
370 } | 270 } |
OLD | NEW |