Color Management, Stage 2: Switch color pipeline to use OpenColorIO
[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 #include "BLI_fileops.h"
33
34 #include "imbuf.h"
35
36 #include "IMB_imbuf_types.h"
37 #include "IMB_imbuf.h"
38 #include "IMB_allocimbuf.h"
39 #include "IMB_filetype.h"
40
41 #include "IMB_colormanagement.h"
42 #include "IMB_colormanagement_intern.h"
43
44 /* some code copied from article on microsoft.com, copied
45  * here for enhanced BMP support in the future
46  * http://www.microsoft.com/msj/defaultframe.asp?page=/msj/0197/mfcp1/mfcp1.htm&nav=/msj/0197/newnav.htm
47  */
48
49 typedef struct BMPINFOHEADER {
50         unsigned int    biSize;
51         unsigned int    biWidth;
52         unsigned int    biHeight;
53         unsigned short  biPlanes;
54         unsigned short  biBitCount;
55         unsigned int    biCompression;
56         unsigned int    biSizeImage;
57         unsigned int    biXPelsPerMeter;
58         unsigned int    biYPelsPerMeter;
59         unsigned int    biClrUsed;
60         unsigned int    biClrImportant;
61 } BMPINFOHEADER;
62
63 typedef struct BMPHEADER {
64         unsigned short biType;
65         unsigned int biSize;
66         unsigned short biRes1;
67         unsigned short biRes2;
68         unsigned int biOffBits;
69 } BMPHEADER;
70
71 #define BMP_FILEHEADER_SIZE 14
72
73 static int checkbmp(unsigned char *mem)
74 {
75 #define CHECK_HEADER_FIELD(mem, field) ((mem[0] == field[0]) && (mem[1] == field[1]))
76
77         int ret_val = 0;
78         BMPINFOHEADER bmi;
79         unsigned int u;
80
81         if (mem) {
82                 if (CHECK_HEADER_FIELD(mem, "BM") ||
83                     CHECK_HEADER_FIELD(mem, "BA") ||
84                     CHECK_HEADER_FIELD(mem, "CI") ||
85                     CHECK_HEADER_FIELD(mem, "CP") ||
86                     CHECK_HEADER_FIELD(mem, "IC") ||
87                     CHECK_HEADER_FIELD(mem, "PT"))
88                 {
89                         /* skip fileheader */
90                         mem += BMP_FILEHEADER_SIZE;
91                 }
92                 else {
93                         return 0;
94                 }
95
96                 /* for systems where an int needs to be 4 bytes aligned */
97                 memcpy(&bmi, mem, sizeof(bmi));
98
99                 u = LITTLE_LONG(bmi.biSize);
100                 /* we only support uncompressed 24 or 32 bits images for now */
101                 if (u >= sizeof(BMPINFOHEADER)) {
102                         if ((bmi.biCompression == 0) && (bmi.biClrUsed == 0)) {
103                                 u = LITTLE_SHORT(bmi.biBitCount);
104                                 if (u >= 16) {
105                                         ret_val = 1;
106                                 }
107                         }
108                 }
109         }
110
111         return(ret_val);
112
113 #undef CHECK_HEADER_FIELD
114 }
115
116 int imb_is_a_bmp(unsigned char *buf)
117 {
118         return checkbmp(buf);
119 }
120
121 struct ImBuf *imb_bmp_decode(unsigned char *mem, size_t size, int flags, char colorspace[IM_MAX_SPACE])
122 {
123         struct ImBuf *ibuf = NULL;
124         BMPINFOHEADER bmi;
125         int x, y, depth, skip, i;
126         unsigned char *bmp, *rect;
127         unsigned short col;
128         
129         (void)size; /* unused */
130
131         if (checkbmp(mem) == 0) return(NULL);
132
133         colorspace_set_default_role(colorspace, IM_MAX_SPACE, COLOR_ROLE_DEFAULT_BYTE);
134
135         if ((mem[0] == 'B') && (mem[1] == 'M')) {
136                 /* skip fileheader */
137                 mem += BMP_FILEHEADER_SIZE;
138         }
139
140         /* for systems where an int needs to be 4 bytes aligned */
141         memcpy(&bmi, mem, sizeof(bmi));
142
143         skip = LITTLE_LONG(bmi.biSize);
144         x = LITTLE_LONG(bmi.biWidth);
145         y = LITTLE_LONG(bmi.biHeight);
146         depth = LITTLE_SHORT(bmi.biBitCount);
147
148 #if 0
149         printf("skip: %d, x: %d y: %d, depth: %d (%x)\n", skip, x, y,
150                depth, bmi.biBitCount);
151         printf("skip: %d, x: %d y: %d, depth: %d (%x)\n", skip, x, y,
152                depth, bmi.biBitCount);
153 #endif
154
155         if (flags & IB_test) {
156                 ibuf = IMB_allocImBuf(x, y, depth, 0);
157         }
158         else {
159                 ibuf = IMB_allocImBuf(x, y, depth, IB_rect);
160                 bmp = mem + skip;
161                 rect = (unsigned char *) ibuf->rect;
162
163                 if (depth == 16) {
164                         for (i = x * y; i > 0; i--) {
165                                 col = bmp[0] + (bmp[1] << 8);
166                                 rect[0] = ((col >> 10) & 0x1f) << 3;
167                                 rect[1] = ((col >>  5) & 0x1f) << 3;
168                                 rect[2] = ((col >>  0) & 0x1f) << 3;
169                                 
170                                 rect[3] = 255;
171                                 rect += 4; bmp += 2;
172                         }
173
174                 }
175                 else if (depth == 24) {
176                         for (i = y; i > 0; i--) {
177                                 int j;
178                                 for (j = x; j > 0; j--) {
179                                         rect[0] = bmp[2];
180                                         rect[1] = bmp[1];
181                                         rect[2] = bmp[0];
182                                         
183                                         rect[3] = 255;
184                                         rect += 4; bmp += 3;
185                                 }
186                                 /* for 24-bit images, rows are padded to multiples of 4 */
187                                 bmp += x % 4;   
188                         }
189                 }
190                 else if (depth == 32) {
191                         for (i = x * y; i > 0; i--) {
192                                 rect[0] = bmp[2];
193                                 rect[1] = bmp[1];
194                                 rect[2] = bmp[0];
195                                 rect[3] = bmp[3];
196                                 rect += 4; bmp += 4;
197                         }
198                 }
199         }
200
201         if (ibuf) {
202                 ibuf->ftype = BMP;
203         }
204         
205         return(ibuf);
206 }
207
208 /* Couple of helper functions for writing our data */
209 static int putIntLSB(unsigned int ui, FILE *ofile)
210 {
211         putc((ui >> 0) & 0xFF, ofile);
212         putc((ui >> 8) & 0xFF, ofile);
213         putc((ui >> 16) & 0xFF, ofile);
214         return putc((ui >> 24) & 0xFF, ofile);
215 }
216
217 static int putShortLSB(unsigned short us, FILE *ofile)
218 {
219         putc((us >> 0) & 0xFF, ofile);
220         return putc((us >> 8) & 0xFF, ofile);
221
222
223 /* Found write info at http://users.ece.gatech.edu/~slabaugh/personal/c/bitmapUnix.c */
224 int imb_savebmp(struct ImBuf *ibuf, const char *name, int flags)
225 {
226         BMPINFOHEADER infoheader;
227         int bytesize, extrabytes, x, y, t, ptr;
228         uchar *data;
229         FILE *ofile;
230         
231         (void)flags; /* unused */
232
233         extrabytes = (4 - ibuf->x * 3 % 4) % 4;
234         bytesize = (ibuf->x * 3 + extrabytes) * ibuf->y;
235
236         data = (uchar *) ibuf->rect;
237         ofile = BLI_fopen(name, "wb");
238         if (!ofile) return 0;
239
240         putShortLSB(19778, ofile); /* "BM" */
241         putIntLSB(0, ofile); /* This can be 0 for BI_RGB bitmaps */
242         putShortLSB(0, ofile); /* Res1 */
243         putShortLSB(0, ofile); /* Res2 */
244         putIntLSB(BMP_FILEHEADER_SIZE + sizeof(infoheader), ofile);
245
246         putIntLSB(sizeof(infoheader), ofile);
247         putIntLSB(ibuf->x, ofile);
248         putIntLSB(ibuf->y, ofile);
249         putShortLSB(1, ofile);
250         putShortLSB(24, ofile);
251         putIntLSB(0, ofile);
252         putIntLSB(bytesize + BMP_FILEHEADER_SIZE + sizeof(infoheader), ofile);
253         putIntLSB(0, ofile);
254         putIntLSB(0, ofile);
255         putIntLSB(0, ofile);
256         putIntLSB(0, ofile);
257
258         /* Need to write out padded image data in bgr format */
259         for (y = 0; y < ibuf->y; y++) {
260                 for (x = 0; x < ibuf->x; x++) {
261                         ptr = (x + y * ibuf->x) * 4;
262                         if (putc(data[ptr + 2], ofile) == EOF) return 0;
263                         if (putc(data[ptr + 1], ofile) == EOF) return 0;
264                         if (putc(data[ptr], ofile) == EOF) return 0;
265                 }
266                 /* add padding here */
267                 for (t = 0; t < extrabytes; t++) if (putc(0, ofile) == EOF) return 0;
268         }
269         if (ofile) {
270                 fflush(ofile);
271                 fclose(ofile);
272         }
273         return 1;
274 }