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