6342143454af5d629b3472ebc349ff7d35fbe998
[geeqie.git] / src / image_load_dds.c
1 /*
2  * Copyright (C) 2018 The Geeqie Team
3  *
4  * Author: Wolfgang Lieff
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (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 along
17  * with this program; if not, write to the Free Software Foundation, Inc.,
18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19
20  * Derived from dds_reader written by:
21  * Copyright (c) 2015 Kenji Sasaki
22  * Released under the MIT license.
23  * https://github.com/npedotnet/DDSReader/blob/master/LICENSE
24
25  */
26
27 #include "main.h"
28
29 #include "image-load.h"
30 #include "image_load_dds.h"
31
32 typedef struct _ImageLoaderDDS ImageLoaderDDS;
33 struct _ImageLoaderDDS {
34         ImageLoaderBackendCbAreaUpdated area_updated_cb;
35         ImageLoaderBackendCbSize size_cb;
36         ImageLoaderBackendCbAreaPrepared area_prepared_cb;
37         gpointer data;
38         GdkPixbuf *pixbuf;
39         guint requested_width;
40         guint requested_height;
41         gboolean abort;
42 };
43
44 static void free_buffer(guchar *pixels, gpointer UNUSED(data))
45 {
46         g_free(pixels);
47 }
48
49 uint ddsGetHeight(unsigned const char * buffer) {
50         return (buffer[12] & 0xFF) | (buffer[13] & 0xFF) << 8 | (buffer[14] & 0xFF) << 16 | (buffer[15] & 0xFF) << 24;
51 }
52
53 uint ddsGetWidth(unsigned const char * buffer) {
54         return (buffer[16] & 0xFF) | (buffer[17] & 0xFF) << 8 | (buffer[18] & 0xFF) << 16 | (buffer[19] & 0xFF) << 24;
55 }
56
57 uint ddsGetMipmap(unsigned const char * buffer) {
58         return (buffer[28] & 0xFF) | (buffer[29] & 0xFF) << 8 | (buffer[30] & 0xFF) << 16 | (buffer[31] & 0xFF) << 24;
59 }
60
61 uint ddsGetPixelFormatFlags(unsigned const char * buffer) {
62         return (buffer[80] & 0xFF) | (buffer[81] & 0xFF) << 8 | (buffer[82] & 0xFF) << 16 | (buffer[83] & 0xFF) << 24;
63 }
64
65 uint ddsGetFourCC(unsigned const char * buffer) {
66         return (buffer[84] & 0xFF) << 24 | (buffer[85] & 0xFF) << 16 | (buffer[86] & 0xFF) << 8 | (buffer[87] & 0xFF);
67 }
68
69 uint ddsGetBitCount(unsigned const char * buffer) {
70         return (buffer[88] & 0xFF) | (buffer[89] & 0xFF) << 8 | (buffer[90] & 0xFF) << 16 | (buffer[91] & 0xFF) << 24;
71 }
72
73 uint ddsGetRedMask(unsigned const char * buffer) {
74         return (buffer[92] & 0xFF) | (buffer[93] & 0xFF) << 8 | (buffer[94] & 0xFF) << 16 | (buffer[95] & 0xFF) << 24;
75 }
76
77 uint ddsGetGreenMask(unsigned const char * buffer) {
78         return (buffer[96] & 0xFF) | (buffer[97] & 0xFF) << 8 | (buffer[98] & 0xFF) << 16 | (buffer[99] & 0xFF) << 24;
79 }
80
81 uint ddsGetBlueMask(unsigned const char * buffer) {
82         return (buffer[100] & 0xFF) | (buffer[101] & 0xFF) << 8 | (buffer[102] & 0xFF) << 16 | (buffer[103] & 0xFF) << 24;
83 }
84
85 uint ddsGetAlphaMask(unsigned const char * buffer) {
86         return (buffer[104] & 0xFF) | (buffer[105] & 0xFF) << 8 | (buffer[106] & 0xFF) << 16 | (buffer[107] & 0xFF) << 24;
87 }
88
89 // Image Type
90 #define DXT1 (0x44585431)
91 #define DXT2 (0x44585432)
92 #define DXT3 (0x44585433)
93 #define DXT4 (0x44585434)
94 #define DXT5 (0x44585435)
95 #define A1R5G5B5 ((1 << 16) | 2)
96 #define X1R5G5B5 ((2 << 16) | 2)
97 #define A4R4G4B4 ((3 << 16) | 2)
98 #define X4R4G4B4 ((4 << 16) | 2)
99 #define R5G6B5 ((5 << 16) | 2)
100 #define R8G8B8 ((1 << 16) | 3)
101 #define A8B8G8R8 ((1 << 16) | 4)
102 #define X8B8G8R8 ((2 << 16) | 4)
103 #define A8R8G8B8 ((3 << 16) | 4)
104 #define X8R8G8B8 ((4 << 16) | 4)
105
106 // RGBA Masks
107 static const uint A1R5G5B5_MASKS[] = { 0x7C00, 0x03E0, 0x001F, 0x8000 };
108 static const uint X1R5G5B5_MASKS[] = { 0x7C00, 0x03E0, 0x001F, 0x0000 };
109 static const uint A4R4G4B4_MASKS[] = { 0x0F00, 0x00F0, 0x000F, 0xF000 };
110 static const uint X4R4G4B4_MASKS[] = { 0x0F00, 0x00F0, 0x000F, 0x0000 };
111 static const uint R5G6B5_MASKS[] = { 0xF800, 0x07E0, 0x001F, 0x0000 };
112 static const uint R8G8B8_MASKS[] = { 0xFF0000, 0x00FF00, 0x0000FF, 0x000000 };
113 static const uint A8B8G8R8_MASKS[] = { 0x000000FF, 0x0000FF00, 0x00FF0000, 0xFF000000 };
114 static const uint X8B8G8R8_MASKS[] = { 0x000000FF, 0x0000FF00, 0x00FF0000, 0x00000000 };
115 static const uint A8R8G8B8_MASKS[] = { 0x00FF0000, 0x0000FF00, 0x000000FF, 0xFF000000 };
116 static const uint X8R8G8B8_MASKS[] = { 0x00FF0000, 0x0000FF00, 0x000000FF, 0x00000000 };
117
118 // BIT4 = 17 * index;
119 static const uint BIT5[] = { 0, 8, 16, 25, 33, 41, 49, 58, 66, 74, 82, 90, 99, 107, 115, 123, 132, 140, 148, 156, 165, 173, 181, 189, 197, 206, 214, 222, 230, 239, 247, 255 };
120 static const uint BIT6[] = { 0, 4, 8, 12, 16, 20, 24, 28, 32, 36, 40, 45, 49, 53, 57, 61, 65, 69, 73, 77, 81, 85, 89, 93, 97, 101, 105, 109, 113, 117, 121, 125, 130, 134, 138, 142, 146, 150, 154, 158, 162, 166, 170, 174, 178, 182, 186, 190, 194, 198, 202, 206, 210, 215, 219, 223, 227, 231, 235, 239, 243, 247, 251, 255 };
121
122 static uint ddsGetType(const unsigned char *buffer) {
123         uint type = 0;
124         uint flags = ddsGetPixelFormatFlags(buffer);
125         if ((flags & 0x04) != 0) {
126                 // DXT
127                 type = ddsGetFourCC(buffer);
128         }
129         else if ((flags & 0x40) != 0) {
130                 // RGB
131                 uint bitCount = ddsGetBitCount(buffer);
132                 uint redMask = ddsGetRedMask(buffer);
133                 uint greenMask = ddsGetGreenMask(buffer);
134                 uint blueMask = ddsGetBlueMask(buffer);
135                 uint alphaMask = ((flags & 0x01) != 0) ? ddsGetAlphaMask(buffer) : 0; // 0x01 alpha
136                 if (bitCount == 16) {
137                         if (redMask == A1R5G5B5_MASKS[0] && greenMask == A1R5G5B5_MASKS[1] && blueMask == A1R5G5B5_MASKS[2] && alphaMask == A1R5G5B5_MASKS[3]) {
138                                 // A1R5G5B5
139                                 type = A1R5G5B5;
140                         }
141                         else if (redMask == X1R5G5B5_MASKS[0] && greenMask == X1R5G5B5_MASKS[1] && blueMask == X1R5G5B5_MASKS[2] && alphaMask == X1R5G5B5_MASKS[3]) {
142                                 // X1R5G5B5
143                                 type = X1R5G5B5;
144                         }
145                         else if (redMask == A4R4G4B4_MASKS[0] && greenMask == A4R4G4B4_MASKS[1] && blueMask == A4R4G4B4_MASKS[2] && alphaMask == A4R4G4B4_MASKS[3]) {
146                                 // A4R4G4B4
147                                 type = A4R4G4B4;
148                         }
149                         else if (redMask == X4R4G4B4_MASKS[0] && greenMask == X4R4G4B4_MASKS[1] && blueMask == X4R4G4B4_MASKS[2] && alphaMask == X4R4G4B4_MASKS[3]) {
150                                 // X4R4G4B4
151                                 type = X4R4G4B4;
152                         }
153                         else if (redMask == R5G6B5_MASKS[0] && greenMask == R5G6B5_MASKS[1] && blueMask == R5G6B5_MASKS[2] && alphaMask == R5G6B5_MASKS[3]) {
154                                 // R5G6B5
155                                 type = R5G6B5;
156                         }
157                         else {
158                                 // Unsupported 16bit RGB image
159                         }
160                 }
161                 else if (bitCount == 24) {
162                         if (redMask == R8G8B8_MASKS[0] && greenMask == R8G8B8_MASKS[1] && blueMask == R8G8B8_MASKS[2] && alphaMask == R8G8B8_MASKS[3]) {
163                                 // R8G8B8
164                                 type = R8G8B8;
165                         }
166                         else {
167                                 // Unsupported 24bit RGB image
168                         }
169                 }
170                 else if (bitCount == 32) {
171                         if (redMask == A8B8G8R8_MASKS[0] && greenMask == A8B8G8R8_MASKS[1] && blueMask == A8B8G8R8_MASKS[2] && alphaMask == A8B8G8R8_MASKS[3]) {
172                                 // A8B8G8R8
173                                 type = A8B8G8R8;
174                         }
175                         else if (redMask == X8B8G8R8_MASKS[0] && greenMask == X8B8G8R8_MASKS[1] && blueMask == X8B8G8R8_MASKS[2] && alphaMask == X8B8G8R8_MASKS[3]) {
176                                 // X8B8G8R8
177                                 type = X8B8G8R8;
178                         }
179                         else if (redMask == A8R8G8B8_MASKS[0] && greenMask == A8R8G8B8_MASKS[1] && blueMask == A8R8G8B8_MASKS[2] && alphaMask == A8R8G8B8_MASKS[3]) {
180                                 // A8R8G8B8
181                                 type = A8R8G8B8;
182                         }
183                         else if (redMask == X8R8G8B8_MASKS[0] && greenMask == X8R8G8B8_MASKS[1] && blueMask == X8R8G8B8_MASKS[2] && alphaMask == X8R8G8B8_MASKS[3]) {
184                                 // X8R8G8B8
185                                 type = X8R8G8B8;
186                         }
187                         else {
188                                 // Unsupported 32bit RGB image
189                         }
190                 }
191         }
192         else {
193                 // YUV or LUMINANCE image
194         }
195         return type;
196 }
197
198 uint ddsGetDXTColor2_1(uint c0, uint c1, uint a) {
199         // 2*c0/3 + c1/3
200         uint r = (2 * BIT5[(c0 & 0xFC00) >> 11] + BIT5[(c1 & 0xFC00) >> 11]) / 3;
201         uint g = (2 * BIT6[(c0 & 0x07E0) >> 5] + BIT6[(c1 & 0x07E0) >> 5]) / 3;
202         uint b = (2 * BIT5[c0 & 0x001F] + BIT5[c1 & 0x001F]) / 3;
203         return (a << 24) | (r << 0) | (g << 8) | (b << 16);
204 }
205
206 uint ddsGetDXTColor1_1(uint c0, uint c1, uint a) {
207         // (c0+c1) / 2
208         uint r = (BIT5[(c0 & 0xFC00) >> 11] + BIT5[(c1 & 0xFC00) >> 11]) / 2;
209         uint g = (BIT6[(c0 & 0x07E0) >> 5] + BIT6[(c1 & 0x07E0) >> 5]) / 2;
210         uint b = (BIT5[c0 & 0x001F] + BIT5[c1 & 0x001F]) / 2;
211         return (a << 24) | (r << 0) | (g << 8) | (b << 16);
212 }
213
214 uint ddsGetDXTColor1(uint c, uint a) {
215         uint r = BIT5[(c & 0xFC00) >> 11];
216         uint g = BIT6[(c & 0x07E0) >> 5];
217         uint b = BIT5[(c & 0x001F)];
218         return (a << 24) | (r << 0) | (g << 8) | (b << 16);
219 }
220
221 uint ddsGetDXTColor(uint c0, uint c1, uint a, uint t) {
222         switch (t) {
223         case 0: return ddsGetDXTColor1(c0, a);
224         case 1: return ddsGetDXTColor1(c1, a);
225         case 2: return (c0 > c1) ? ddsGetDXTColor2_1(c0, c1, a) : ddsGetDXTColor1_1(c0, c1, a);
226         case 3: return (c0 > c1) ? ddsGetDXTColor2_1(c1, c0, a) : 0;
227         }
228         return 0;
229 }
230
231 guchar *ddsDecodeDXT1(uint width, uint height, const unsigned char *buffer) {
232         uint *pixels = g_try_malloc(4 * width*height);
233         if (pixels == NULL) return NULL;
234         uint index = 128;
235         uint w = (width + 3) / 4;
236         uint h = (height + 3) / 4;
237         for (uint i = 0; i<h; i++) {
238                 for (uint j = 0; j<w; j++) {
239                         uint c0 = (buffer[index] & 0xFF) | (buffer[index + 1] & 0xFF) << 8; index += 2;
240                         uint c1 = (buffer[index] & 0xFF) | (buffer[index + 1] & 0xFF) << 8; index += 2;
241                         for (uint k = 0; k<4; k++) {
242                                 if (4 * i + k >= height) break;
243                                 uint t0 = (buffer[index] & 0x03);
244                                 uint t1 = (buffer[index] & 0x0C) >> 2;
245                                 uint t2 = (buffer[index] & 0x30) >> 4;
246                                 uint t3 = (buffer[index++] & 0xC0) >> 6;
247                                 pixels[4 * width*i + 4 * j + width*k + 0] = ddsGetDXTColor(c0, c1, 0xFF, t0);
248                                 if (4 * j + 1 >= width) continue;
249                                 pixels[4 * width*i + 4 * j + width*k + 1] = ddsGetDXTColor(c0, c1, 0xFF, t1);
250                                 if (4 * j + 2 >= width) continue;
251                                 pixels[4 * width*i + 4 * j + width*k + 2] = ddsGetDXTColor(c0, c1, 0xFF, t2);
252                                 if (4 * j + 3 >= width) continue;
253                                 pixels[4 * width*i + 4 * j + width*k + 3] = ddsGetDXTColor(c0, c1, 0xFF, t3);
254                         }
255                 }
256         }
257         return (guchar *) pixels;
258 }
259
260 guchar *ddsDecodeDXT3(uint width, uint height, const unsigned char *buffer) {
261         uint *pixels = g_try_malloc(4 * width*height);
262         if (pixels == NULL) return NULL;
263         uint index = 128;
264         uint w = (width + 3) / 4;
265         uint h = (height + 3) / 4;
266         uint alphaTable[16];
267         for (uint i = 0; i<h; i++) {
268                 for (uint j = 0; j<w; j++) {
269                         // create alpha table(4bit to 8bit)
270                         for (uint k = 0; k<4; k++) {
271                                 uint a0 = (buffer[index++] & 0xFF);
272                                 uint a1 = (buffer[index++] & 0xFF);
273                                 // 4bit alpha to 8bit alpha
274                                 alphaTable[4 * k + 0] = 17 * ((a0 & 0xF0) >> 4);
275                                 alphaTable[4 * k + 1] = 17 * (a0 & 0x0F);
276                                 alphaTable[4 * k + 2] = 17 * ((a1 & 0xF0) >> 4);
277                                 alphaTable[4 * k + 3] = 17 * (a1 & 0x0F);
278                         }
279                         uint c0 = (buffer[index] & 0xFF) | (buffer[index + 1] & 0xFF) << 8; index += 2;
280                         uint c1 = (buffer[index] & 0xFF) | (buffer[index + 1] & 0xFF) << 8; index += 2;
281                         for (uint k = 0; k<4; k++) {
282                                 if (4 * i + k >= height) break;
283                                 uint t0 = (buffer[index] & 0x03);
284                                 uint t1 = (buffer[index] & 0x0C) >> 2;
285                                 uint t2 = (buffer[index] & 0x30) >> 4;
286                                 uint t3 = (buffer[index++] & 0xC0) >> 6;
287                                 pixels[4 * width*i + 4 * j + width*k + 0] = ddsGetDXTColor(c0, c1, alphaTable[4 * k + 0], t0);
288                                 if (4 * j + 1 >= width) continue;
289                                 pixels[4 * width*i + 4 * j + width*k + 1] = ddsGetDXTColor(c0, c1, alphaTable[4 * k + 1], t1);
290                                 if (4 * j + 2 >= width) continue;
291                                 pixels[4 * width*i + 4 * j + width*k + 2] = ddsGetDXTColor(c0, c1, alphaTable[4 * k + 2], t2);
292                                 if (4 * j + 3 >= width) continue;
293                                 pixels[4 * width*i + 4 * j + width*k + 3] = ddsGetDXTColor(c0, c1, alphaTable[4 * k + 3], t3);
294                         }
295                 }
296         }
297         return (guchar *) pixels;
298 }
299
300 guchar *ddsDecodeDXT2(uint width, uint height, const unsigned char *buffer) {
301         return ddsDecodeDXT3(width, height, buffer);
302 }
303
304 int ddsGetDXT5Alpha(uint a0, uint a1, uint t) {
305         if (a0 > a1) switch (t) {
306         case 0: return a0;
307         case 1: return a1;
308         case 2: return (6 * a0 + a1) / 7;
309         case 3: return (5 * a0 + 2 * a1) / 7;
310         case 4: return (4 * a0 + 3 * a1) / 7;
311         case 5: return (3 * a0 + 4 * a1) / 7;
312         case 6: return (2 * a0 + 5 * a1) / 7;
313         case 7: return (a0 + 6 * a1) / 7;
314         }
315         else switch (t) {
316         case 0: return a0;
317         case 1: return a1;
318         case 2: return (4 * a0 + a1) / 5;
319         case 3: return (3 * a0 + 2 * a1) / 5;
320         case 4: return (2 * a0 + 3 * a1) / 5;
321         case 5: return (a0 + 4 * a1) / 5;
322         case 6: return 0;
323         case 7: return 255;
324         }
325         return 0;
326 }
327
328 guchar *ddsDecodeDXT5(uint width, uint height, const unsigned char *buffer) {
329         uint *pixels = g_try_malloc(4 * width*height);
330         if (pixels == NULL) return NULL;
331         uint index = 128;
332         uint w = (width + 3) / 4;
333         uint h = (height + 3) / 4;
334         uint alphaTable[16];
335         for (uint i = 0; i<h; i++) {
336                 for (uint j = 0; j<w; j++) {
337                         // create alpha table
338                         uint a0 = (buffer[index++] & 0xFF);
339                         uint a1 = (buffer[index++] & 0xFF);
340                         uint b0 = (buffer[index] & 0xFF) | (buffer[index + 1] & 0xFF) << 8 | (buffer[index + 2] & 0xFF) << 16; index += 3;
341                         uint b1 = (buffer[index] & 0xFF) | (buffer[index + 1] & 0xFF) << 8 | (buffer[index + 2] & 0xFF) << 16; index += 3;
342                         alphaTable[0] = b0 & 0x07;
343                         alphaTable[1] = (b0 >> 3) & 0x07;
344                         alphaTable[2] = (b0 >> 6) & 0x07;
345                         alphaTable[3] = (b0 >> 9) & 0x07;
346                         alphaTable[4] = (b0 >> 12) & 0x07;
347                         alphaTable[5] = (b0 >> 15) & 0x07;
348                         alphaTable[6] = (b0 >> 18) & 0x07;
349                         alphaTable[7] = (b0 >> 21) & 0x07;
350                         alphaTable[8] = b1 & 0x07;
351                         alphaTable[9] = (b1 >> 3) & 0x07;
352                         alphaTable[10] = (b1 >> 6) & 0x07;
353                         alphaTable[11] = (b1 >> 9) & 0x07;
354                         alphaTable[12] = (b1 >> 12) & 0x07;
355                         alphaTable[13] = (b1 >> 15) & 0x07;
356                         alphaTable[14] = (b1 >> 18) & 0x07;
357                         alphaTable[15] = (b1 >> 21) & 0x07;
358                         uint c0 = (buffer[index] & 0xFF) | (buffer[index + 1] & 0xFF) << 8; index += 2;
359                         uint c1 = (buffer[index] & 0xFF) | (buffer[index + 1] & 0xFF) << 8; index += 2;
360                         for (uint k = 0; k<4; k++) {
361                                 if (4 * i + k >= height) break;
362                                 uint t0 = (buffer[index] & 0x03);
363                                 uint t1 = (buffer[index] & 0x0C) >> 2;
364                                 uint t2 = (buffer[index] & 0x30) >> 4;
365                                 uint t3 = (buffer[index++] & 0xC0) >> 6;
366                                 pixels[4 * width*i + 4 * j + width*k + 0] = ddsGetDXTColor(c0, c1, ddsGetDXT5Alpha(a0, a1, alphaTable[4 * k + 0]), t0);
367                                 if (4 * j + 1 >= width) continue;
368                                 pixels[4 * width*i + 4 * j + width*k + 1] = ddsGetDXTColor(c0, c1, ddsGetDXT5Alpha(a0, a1, alphaTable[4 * k + 1]), t1);
369                                 if (4 * j + 2 >= width) continue;
370                                 pixels[4 * width*i + 4 * j + width*k + 2] = ddsGetDXTColor(c0, c1, ddsGetDXT5Alpha(a0, a1, alphaTable[4 * k + 2]), t2);
371                                 if (4 * j + 3 >= width) continue;
372                                 pixels[4 * width*i + 4 * j + width*k + 3] = ddsGetDXTColor(c0, c1, ddsGetDXT5Alpha(a0, a1, alphaTable[4 * k + 3]), t3);
373                         }
374                 }
375         }
376         return (guchar *) pixels;
377 }
378
379 guchar *ddsDecodeDXT4(uint width, uint height, const unsigned char *buffer) {
380         return ddsDecodeDXT5(width, height, buffer);
381 }
382
383 guchar *ddsReadA1R5G5B5(uint width, uint height, const unsigned char *buffer) {
384         uint *pixels = g_try_malloc(4 * width*height);
385         if (pixels == NULL) return NULL;
386         uint index = 128;
387         for (uint i = 0; i<height*width; i++) {
388                 uint rgba = (buffer[index] & 0xFF) | (buffer[index + 1] & 0xFF) << 8; index += 2;
389                 uint r = BIT5[(rgba & A1R5G5B5_MASKS[0]) >> 10];
390                 uint g = BIT5[(rgba & A1R5G5B5_MASKS[1]) >> 5];
391                 uint b = BIT5[(rgba & A1R5G5B5_MASKS[2])];
392                 uint a = 255 * ((rgba & A1R5G5B5_MASKS[3]) >> 15);
393                 pixels[i] = (a << 24) | (r << 0) | (g << 8) | (b << 16);
394         }
395         return (guchar *) pixels;
396 }
397
398 guchar *ddsReadX1R5G5B5(uint width, uint height, const unsigned char *buffer) {
399         uint *pixels = g_try_malloc(4 * width*height);
400         if (pixels == NULL) return NULL;
401         uint index = 128;
402         for (uint i = 0; i<height*width; i++) {
403                 uint rgba = (buffer[index] & 0xFF) | (buffer[index + 1] & 0xFF) << 8; index += 2;
404                 uint r = BIT5[(rgba & X1R5G5B5_MASKS[0]) >> 10];
405                 uint g = BIT5[(rgba & X1R5G5B5_MASKS[1]) >> 5];
406                 uint b = BIT5[(rgba & X1R5G5B5_MASKS[2])];
407                 uint a = 255;
408                 pixels[i] = (a << 24) | (r << 0) | (g << 8) | (b << 16);
409         }
410         return (guchar *) pixels;
411 }
412
413 guchar *ddsReadA4R4G4B4(uint width, uint height, const unsigned char *buffer) {
414         uint *pixels = g_try_malloc(4 * width*height);
415         if (pixels == NULL) return NULL;
416         uint index = 128;
417         for (uint i = 0; i<height*width; i++) {
418                 uint rgba = (buffer[index] & 0xFF) | (buffer[index + 1] & 0xFF) << 8; index += 2;
419                 uint r = 17 * ((rgba & A4R4G4B4_MASKS[0]) >> 8);
420                 uint g = 17 * ((rgba & A4R4G4B4_MASKS[1]) >> 4);
421                 uint b = 17 * ((rgba & A4R4G4B4_MASKS[2]));
422                 uint a = 17 * ((rgba & A4R4G4B4_MASKS[3]) >> 12);
423                 pixels[i] = (a << 24) | (r << 0) | (g << 8) | (b << 16);
424         }
425         return (guchar *) pixels;
426 }
427
428 guchar *ddsReadX4R4G4B4(uint width, uint height, const unsigned char *buffer) {
429         uint *pixels = g_try_malloc(4 * width*height);
430         if (pixels == NULL) return NULL;
431         uint index = 128;
432         for (uint i = 0; i<height*width; i++) {
433                 uint rgba = (buffer[index] & 0xFF) | (buffer[index + 1] & 0xFF) << 8; index += 2;
434                 uint r = 17 * ((rgba & A4R4G4B4_MASKS[0]) >> 8);
435                 uint g = 17 * ((rgba & A4R4G4B4_MASKS[1]) >> 4);
436                 uint b = 17 * ((rgba & A4R4G4B4_MASKS[2]));
437                 uint a = 255;
438                 pixels[i] = (a << 24) | (r << 0) | (g << 8) | (b << 16);
439         }
440         return (guchar *) pixels;
441 }
442
443 guchar *ddsReadR5G6B5(uint width, uint height, const unsigned char *buffer) {
444         uint *pixels = g_try_malloc(4 * width*height);
445         if (pixels == NULL) return NULL;
446         uint index = 128;
447         for (uint i = 0; i<height*width; i++) {
448                 uint rgba = (buffer[index] & 0xFF) | (buffer[index + 1] & 0xFF) << 8; index += 2;
449                 uint r = BIT5[((rgba & R5G6B5_MASKS[0]) >> 11)];
450                 uint g = BIT6[((rgba & R5G6B5_MASKS[1]) >> 5)];
451                 uint b = BIT5[((rgba & R5G6B5_MASKS[2]))];
452                 uint a = 255;
453                 pixels[i] = (a << 24) | (r << 0) | (g << 8) | (b << 16);
454         }
455         return (guchar *) pixels;
456 }
457
458 guchar *ddsReadR8G8B8(uint width, uint height, const unsigned char *buffer) {
459         uint *pixels = g_try_malloc(4 * width*height);
460         if (pixels == NULL) return NULL;
461         uint index = 128;
462         for (uint i = 0; i<height*width; i++) {
463                 uint b = buffer[index++] & 0xFF;
464                 uint g = buffer[index++] & 0xFF;
465                 uint r = buffer[index++] & 0xFF;
466                 uint a = 255;
467                 pixels[i] = (a << 24) | (r << 0) | (g << 8) | (b << 16);
468         }
469         return (guchar *) pixels;
470 }
471
472 guchar *ddsReadA8B8G8R8(uint width, uint height, const unsigned char *buffer) {
473         uint *pixels = g_try_malloc(4 * width*height);
474         if (pixels == NULL) return NULL;
475         uint index = 128;
476         for (uint i = 0; i<height*width; i++) {
477                 uint r = buffer[index++] & 0xFF;
478                 uint g = buffer[index++] & 0xFF;
479                 uint b = buffer[index++] & 0xFF;
480                 uint a = buffer[index++] & 0xFF;
481                 pixels[i] = (a << 24) | (r << 0) | (g << 8) | (b << 16);
482         }
483         return (guchar *) pixels;
484 }
485
486 guchar *ddsReadX8B8G8R8(uint width, uint height, const unsigned char *buffer) {
487         uint *pixels = g_try_malloc(4 * width*height);
488         if (pixels == NULL) return NULL;
489         uint index = 128;
490         for (uint i = 0; i<height*width; i++) {
491                 uint r = buffer[index++] & 0xFF;
492                 uint g = buffer[index++] & 0xFF;
493                 uint b = buffer[index++] & 0xFF;
494                 uint a = 255; index++;
495                 pixels[i] = (a << 24) | (r << 0) | (g << 8) | (b << 16);
496         }
497         return (guchar *) pixels;
498 }
499
500 guchar *ddsReadA8R8G8B8(uint width, uint height, const unsigned char *buffer) {
501         uint *pixels = g_try_malloc(4 * width*height);
502         if (pixels == NULL) return NULL;
503         uint index = 128;
504         for (uint i = 0; i<height*width; i++) {
505                 uint b = buffer[index++] & 0xFF;
506                 uint g = buffer[index++] & 0xFF;
507                 uint r = buffer[index++] & 0xFF;
508                 uint a = buffer[index++] & 0xFF;
509                 pixels[i] = (a << 24) | (r << 0) | (g << 8) | (b << 16);
510         }
511         return (guchar *) pixels;
512 }
513
514 guchar *ddsReadX8R8G8B8(uint width, uint height, const unsigned char *buffer) {
515         uint *pixels = g_try_malloc(4 * width*height);
516         if (pixels == NULL) return NULL;
517         uint index = 128;
518         for (uint i = 0; i<height*width; i++) {
519                 uint b = buffer[index++] & 0xFF;
520                 uint g = buffer[index++] & 0xFF;
521                 uint r = buffer[index++] & 0xFF;
522                 uint a = 255; index++;
523                 pixels[i] = (a << 24) | (r << 0) | (g << 8) | (b << 16);
524         }
525         return (guchar *) pixels;
526 }
527
528 static gboolean image_loader_dds_load (gpointer loader, const guchar *buf, gsize UNUSED(count), GError **UNUSED(error))
529 {
530         ImageLoaderDDS *ld = (ImageLoaderDDS *) loader;
531         uint width = ddsGetWidth(buf);
532         uint height = ddsGetHeight(buf);
533         uint type = ddsGetType(buf);
534         if (type == 0) return FALSE;
535         {
536                 guchar *pixels = NULL;
537                 guint rowstride = width * 4;
538                 switch (type) {
539                 case DXT1: pixels = ddsDecodeDXT1(width, height, buf); break;
540                 case DXT2: pixels = ddsDecodeDXT2(width, height, buf); break;
541                 case DXT3: pixels = ddsDecodeDXT3(width, height, buf); break;
542                 case DXT4: pixels = ddsDecodeDXT4(width, height, buf); break;
543                 case DXT5: pixels = ddsDecodeDXT5(width, height, buf); break;
544                 case A1R5G5B5: pixels = ddsReadA1R5G5B5(width, height, buf); break;
545                 case X1R5G5B5: pixels = ddsReadX1R5G5B5(width, height, buf); break;
546                 case A4R4G4B4: pixels = ddsReadA4R4G4B4(width, height, buf); break;
547                 case X4R4G4B4: pixels = ddsReadX4R4G4B4(width, height, buf); break;
548                 case R5G6B5: pixels = ddsReadR5G6B5(width, height, buf); break;
549                 case R8G8B8: pixels = ddsReadR8G8B8(width, height, buf); break;
550                 case A8B8G8R8: pixels = ddsReadA8B8G8R8(width, height, buf); break;
551                 case X8B8G8R8: pixels = ddsReadX8B8G8R8(width, height, buf); break;
552                 case A8R8G8B8: pixels = ddsReadA8R8G8B8(width, height, buf); break;
553                 case X8R8G8B8: pixels = ddsReadX8R8G8B8(width, height, buf); break;
554                 }
555                 ld->pixbuf = gdk_pixbuf_new_from_data (pixels, GDK_COLORSPACE_RGB, TRUE, 8, width, height, rowstride, free_buffer, NULL);
556                 ld->area_updated_cb(loader, 0, 0, width, height, ld->data);
557                 return TRUE;
558         }
559 }
560
561 static gpointer image_loader_dds_new(ImageLoaderBackendCbAreaUpdated area_updated_cb, ImageLoaderBackendCbSize size_cb, ImageLoaderBackendCbAreaPrepared area_prepared_cb, gpointer data)
562 {
563         ImageLoaderDDS *loader = g_new0(ImageLoaderDDS, 1);
564         loader->area_updated_cb = area_updated_cb;
565         loader->size_cb = size_cb;
566         loader->area_prepared_cb = area_prepared_cb;
567         loader->data = data;
568         return (gpointer) loader;
569 }
570
571 static void image_loader_dds_set_size(gpointer loader, int width, int height)
572 {
573         ImageLoaderDDS *ld = (ImageLoaderDDS *) loader;
574         ld->requested_width = width;
575         ld->requested_height = height;
576 }
577
578 static GdkPixbuf* image_loader_dds_get_pixbuf(gpointer loader)
579 {
580         ImageLoaderDDS *ld = (ImageLoaderDDS *) loader;
581         return ld->pixbuf;
582 }
583
584 static gchar* image_loader_dds_get_format_name(gpointer UNUSED(loader))
585 {
586         return g_strdup("dds");
587 }
588 static gchar** image_loader_dds_get_format_mime_types(gpointer UNUSED(loader))
589 {
590         static gchar *mime[] = {"image/vnd-ms.dds", NULL};
591         return g_strdupv(mime);
592 }
593
594 static gboolean image_loader_dds_close(gpointer UNUSED(loader), GError **UNUSED(error))
595 {
596         return TRUE;
597 }
598
599 static void image_loader_dds_abort(gpointer loader)
600 {
601         ImageLoaderDDS *ld = (ImageLoaderDDS *) loader;
602         ld->abort = TRUE;
603 }
604
605 static void image_loader_dds_free(gpointer loader)
606 {
607         ImageLoaderDDS *ld = (ImageLoaderDDS *) loader;
608         if (ld->pixbuf) g_object_unref(ld->pixbuf);
609         g_free(ld);
610 }
611
612 void image_loader_backend_set_dds(ImageLoaderBackend *funcs)
613 {
614         funcs->loader_new = image_loader_dds_new;
615         funcs->set_size = image_loader_dds_set_size;
616         funcs->load = image_loader_dds_load;
617         funcs->write = NULL;
618         funcs->get_pixbuf = image_loader_dds_get_pixbuf;
619         funcs->close = image_loader_dds_close;
620         funcs->abort = image_loader_dds_abort;
621         funcs->free = image_loader_dds_free;
622         funcs->get_format_name = image_loader_dds_get_format_name;
623         funcs->get_format_mime_types = image_loader_dds_get_format_mime_types;
624 }
625
626 /* vim: set shiftwidth=8 softtabstop=0 cindent cinoptions={1s: */