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