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