This is a pretty lame commit but here it is:
[blender.git] / source / blender / imbuf / intern / bmp.c
1 /**
2  * $Id$
3  *
4  * ***** BEGIN GPL/BL DUAL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version. The Blender
10  * Foundation also sells licenses for use in proprietary software under
11  * the Blender License.  See http://www.blender.org/BL/ for information
12  * about this.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software Foundation,
21  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
22  *
23  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
24  * All rights reserved.
25  *
26  * The Original Code is: all of this file.
27  *
28  * Contributor(s): none yet.
29  *
30  * ***** END GPL/BL DUAL LICENSE BLOCK *****
31  */
32
33 #ifdef _WIN32
34 #include "BLI_winstuff.h"
35 #endif
36 #include "BLI_blenlib.h"
37
38 #include "imbuf.h"
39 #include "imbuf_patch.h"
40
41 #include "IMB_imbuf_types.h"
42 #include "IMB_imbuf.h"
43 #include "IMB_allocimbuf.h"
44 #include "IMB_cmap.h"
45 #include "IMB_bmp.h"
46
47 /* some code copied from article on microsoft.com, copied
48   here for enhanced BMP support in the future
49   http://www.microsoft.com/msj/defaultframe.asp?page=/msj/0197/mfcp1/mfcp1.htm&nav=/msj/0197/newnav.htm
50 */
51
52 typedef struct BMPINFOHEADER{
53         unsigned int    biSize;
54         unsigned int    biWidth;
55         unsigned int    biHeight;
56         unsigned short  biPlanes;
57         unsigned short  biBitCount;
58         unsigned int    biCompression;
59         unsigned int    biSizeImage;
60         unsigned int    biXPelsPerMeter;
61         unsigned int    biYPelsPerMeter;
62         unsigned int    biClrUsed;
63         unsigned int    biClrImportant;
64 } BMPINFOHEADER;
65
66 typedef struct BMPHEADER {
67         unsigned short biType;
68         unsigned int biSize;
69         unsigned short biRes1;
70         unsigned short biRes2;
71         unsigned int biOffBits;
72 } BMPHEADER;
73
74 #define BMP_FILEHEADER_SIZE 14
75
76 static int checkbmp(unsigned char *mem)
77 {
78         int ret_val = 0;
79         BMPINFOHEADER bmi;
80         unsigned int u;
81
82         if (mem) {
83                 if ((mem[0] == 'B') && (mem[1] == 'M')) {
84                         /* skip fileheader */
85                         mem += BMP_FILEHEADER_SIZE;
86                 } else {
87                 }
88
89                 /* for systems where an int needs to be 4 bytes aligned */
90                 memcpy(&bmi, mem, sizeof(bmi));
91
92                 u = LITTLE_LONG(bmi.biSize);
93                 /* we only support uncompressed 24 or 32 bits images for now */
94                 if (u >= sizeof(BMPINFOHEADER)) {
95                         if ((bmi.biCompression == 0) && (bmi.biClrUsed == 0)) {
96                                 u = LITTLE_SHORT(bmi.biBitCount);
97                                 if (u >= 16) {
98                                         ret_val = 1;
99                                 }
100                         }
101                 }
102         }
103
104         return(ret_val);
105 }
106
107 int imb_is_a_bmp(void *buf) {
108         
109         return checkbmp(buf);
110 }
111
112 struct ImBuf *imb_bmp_decode(unsigned char *mem, int size, int flags)
113 {
114         struct ImBuf *ibuf = 0;
115         BMPINFOHEADER bmi;
116         int x, y, depth, skip, i;
117         unsigned char *bmp, *rect;
118         unsigned short col;
119
120         if (checkbmp(mem) == 0) return(0);
121
122         if ((mem[0] == 'B') && (mem[1] == 'M')) {
123                 /* skip fileheader */
124                 mem += BMP_FILEHEADER_SIZE;
125         }
126
127         /* for systems where an int needs to be 4 bytes aligned */
128         memcpy(&bmi, mem, sizeof(bmi));
129
130         skip = LITTLE_LONG(bmi.biSize);
131         x = LITTLE_LONG(bmi.biWidth);
132         y = LITTLE_LONG(bmi.biHeight);
133         depth = LITTLE_SHORT(bmi.biBitCount);
134
135         /* printf("skip: %d, x: %d y: %d, depth: %d (%x)\n", skip, x, y, 
136                 depth, bmi.biBitCount); */
137         /* printf("skip: %d, x: %d y: %d, depth: %d (%x)\n", skip, x, y, 
138                 depth, bmi.biBitCount); */
139         if (flags & IB_test) {
140                 ibuf = IMB_allocImBuf(x, y, depth, 0, 0);
141         } else {
142                 ibuf = IMB_allocImBuf(x, y, depth, IB_rect, 0);
143                 bmp = mem + skip;
144                 rect = (unsigned char *) ibuf->rect;
145
146                 if (depth == 16) {
147                         for (i = x * y; i > 0; i--) {
148                                 col = bmp[0] + (bmp[1] << 8);
149                                 rect[0] = ((col >> 10) & 0x1f) << 3;
150                                 rect[1] = ((col >>  5) & 0x1f) << 3;
151                                 rect[2] = ((col >>  0) & 0x1f) << 3;
152                                 
153                                 rect[3] = 255;
154                                 rect += 4; bmp += 2;
155                         }
156
157                 } else if (depth == 24) {
158                         for (i = x * y; i > 0; i--) {
159                                 rect[0] = bmp[2];
160                                 rect[1] = bmp[1];
161                                 rect[2] = bmp[0];
162                                 
163                                 rect[3] = 255;
164                                 rect += 4; bmp += 3;
165                         }
166                 } else if (depth == 32) {
167                         for (i = x * y; i > 0; i--) {
168                                 rect[0] = bmp[0];
169                                 rect[1] = bmp[1];
170                                 rect[2] = bmp[2];
171                                 rect[3] = bmp[3];
172                                 rect += 4; bmp += 4;
173                         }
174                 }
175         }
176
177         if (ibuf) {
178                 ibuf->ftype = BMP;
179         }
180         
181         return(ibuf);
182 }
183
184 /* Couple of helper functions for writing our data */
185 int putIntLSB(unsigned int ui,FILE *ofile) { 
186         putc((ui>>0)&0xFF,ofile); 
187         putc((ui>>8)&0xFF,ofile); 
188         putc((ui>>16)&0xFF,ofile); 
189         return putc((ui>>24)&0xFF,ofile); 
190 }
191
192 int putShortLSB(unsigned short us,FILE *ofile) { 
193         putc((us>>0)&0xFF,ofile); 
194         return putc((us>>8)&0xFF,ofile); 
195
196
197 /* Found write info at http://users.ece.gatech.edu/~slabaugh/personal/c/bitmapUnix.c */
198 short imb_savebmp(struct ImBuf *ibuf, char *name, int flags) {
199
200         BMPINFOHEADER infoheader;
201         int bytesize, extrabytes, x, y, t, ptr;
202         uchar *data;
203         FILE *ofile;
204
205         extrabytes = (4 - ibuf->x*3 % 4) % 4;
206         bytesize = (ibuf->x * 3 + extrabytes) * ibuf->y;
207
208         data = (uchar *) ibuf->rect;
209         ofile = fopen(name,"wb");
210
211         putShortLSB(19778,ofile); /* "BM" */
212         putIntLSB(0,ofile); /* This can be 0 for BI_RGB bitmaps */
213         putShortLSB(0,ofile); /* Res1 */
214         putShortLSB(0,ofile); /* Res2 */
215         putIntLSB(BMP_FILEHEADER_SIZE + sizeof(infoheader),ofile); 
216
217         putIntLSB(sizeof(infoheader),ofile);
218         putIntLSB(ibuf->x,ofile);
219         putIntLSB(ibuf->y,ofile);
220         putShortLSB(1,ofile);
221         putShortLSB(24,ofile);
222         putIntLSB(0,ofile);
223         putIntLSB(bytesize + BMP_FILEHEADER_SIZE + sizeof(infoheader),ofile);
224         putIntLSB(0,ofile);
225         putIntLSB(0,ofile);
226         putIntLSB(0,ofile);
227         putIntLSB(0,ofile);
228
229         /* Need to write out padded image data in bgr format */
230         for (y=0;y<ibuf->y;y++) {
231                 for (x=0;x<ibuf->x;x++) {
232                         ptr=(x + y * ibuf->x) * 4;
233                         if (putc(data[ptr+2],ofile) == EOF) return 0;
234                         if (putc(data[ptr+1],ofile) == EOF) return 0;
235                         if (putc(data[ptr],ofile) == EOF) return 0;
236                 }
237                 /* add padding here */
238                 for (t=0;t<extrabytes;t++) if (putc(0,ofile) == EOF) return 0;
239         }
240         if (ofile) {
241                 fflush(ofile);
242                 fclose(ofile);
243         }
244         return 1;
245 }