1 | /** |
2 | * Simple DDS data parser for compressed 2D textures. |
3 | * |
4 | * Copyright (c) 2013-2023 Sasha Szpakowski |
5 | * |
6 | * This software is provided 'as-is', without any express or implied |
7 | * warranty. In no event will the authors be held liable for any damages |
8 | * arising from the use of this software. |
9 | * |
10 | * Permission is granted to anyone to use this software for any purpose, |
11 | * including commercial applications, and to alter it and redistribute it |
12 | * freely, subject to the following restrictions: |
13 | * |
14 | * 1. The origin of this software must not be misrepresented; you must not |
15 | * claim that you wrote the original software. If you use this software |
16 | * in a product, an acknowledgment in the product documentation would be |
17 | * appreciated but is not required. |
18 | * 2. Altered source versions must be plainly marked as such, and must not be |
19 | * misrepresented as being the original software. |
20 | * 3. This notice may not be removed or altered from any source distribution. |
21 | **/ |
22 | |
23 | #include "ddsparse.h" |
24 | #include "ddsinfo.h" |
25 | |
26 | #include <algorithm> |
27 | |
28 | namespace dds |
29 | { |
30 | |
31 | using namespace dds::dxinfo; |
32 | |
33 | // Creates a packed uint representation of a FourCC code. |
34 | #define MakeFourCC(a, b, c, d) ((uint32_t) (((d)<<24) | ((c)<<16) | ((b)<<8) | (a))) |
35 | |
36 | #define ISBITMASK(r,g,b,a) (ddpf.rBitMask == r && ddpf.gBitMask == g && ddpf.bBitMask == b && ddpf.aBitMask == a) |
37 | |
38 | // Function adapted from DirectXTex: |
39 | // https://github.com/microsoft/DirectXTex/blob/master/DDSTextureLoader/DDSTextureLoader.cpp#L623 |
40 | static DXGIFormat getDXGIFormat(const DDSPixelFormat& ddpf) |
41 | { |
42 | if (ddpf.flags & DDPF_RGB) |
43 | { |
44 | // Note that sRGB formats are written using the "DX10" extended header |
45 | |
46 | switch (ddpf.rgbBitCount) |
47 | { |
48 | case 32: |
49 | if (ISBITMASK(0x000000ff, 0x0000ff00, 0x00ff0000, 0xff000000)) |
50 | return DXGI_FORMAT_R8G8B8A8_UNORM; |
51 | |
52 | if (ISBITMASK(0x00ff0000, 0x0000ff00, 0x000000ff, 0xff000000)) |
53 | return DXGI_FORMAT_B8G8R8A8_UNORM; |
54 | |
55 | if (ISBITMASK(0x00ff0000, 0x0000ff00, 0x000000ff, 0x00000000)) |
56 | return DXGI_FORMAT_B8G8R8X8_UNORM; |
57 | |
58 | // No DXGI format maps to ISBITMASK(0x000000ff,0x0000ff00,0x00ff0000,0x00000000) aka D3DFMT_X8B8G8R8 |
59 | |
60 | // Note that many common DDS reader/writers (including D3DX) swap the |
61 | // the RED/BLUE masks for 10:10:10:2 formats. We assume |
62 | // below that the 'backwards' header mask is being used since it is most |
63 | // likely written by D3DX. The more robust solution is to use the 'DX10' |
64 | // header extension and specify the DXGI_FORMAT_R10G10B10A2_UNORM format directly |
65 | |
66 | // For 'correct' writers, this should be 0x000003ff,0x000ffc00,0x3ff00000 for RGB data |
67 | if (ISBITMASK(0x3ff00000, 0x000ffc00, 0x000003ff, 0xc0000000)) |
68 | return DXGI_FORMAT_R10G10B10A2_UNORM; |
69 | |
70 | // No DXGI format maps to ISBITMASK(0x000003ff,0x000ffc00,0x3ff00000,0xc0000000) aka D3DFMT_A2R10G10B10 |
71 | |
72 | if (ISBITMASK(0x0000ffff, 0xffff0000, 0x00000000, 0x00000000)) |
73 | return DXGI_FORMAT_R16G16_UNORM; |
74 | |
75 | if (ISBITMASK(0xffffffff, 0x00000000, 0x00000000, 0x00000000)) |
76 | // Only 32-bit color channel format in D3D9 was R32F |
77 | return DXGI_FORMAT_R32_FLOAT; // D3DX writes this out as a FourCC of 114 |
78 | break; |
79 | |
80 | case 24: |
81 | // No 24bpp DXGI formats aka D3DFMT_R8G8B8 |
82 | break; |
83 | |
84 | case 16: |
85 | if (ISBITMASK(0x7c00, 0x03e0, 0x001f, 0x8000)) |
86 | return DXGI_FORMAT_B5G5R5A1_UNORM; |
87 | |
88 | if (ISBITMASK(0xf800, 0x07e0, 0x001f, 0x0000)) |
89 | return DXGI_FORMAT_B5G6R5_UNORM; |
90 | |
91 | // No DXGI format maps to ISBITMASK(0x7c00,0x03e0,0x001f,0x0000) aka D3DFMT_X1R5G5B5 |
92 | |
93 | // No DXGI format maps to ISBITMASK(0x0f00,0x00f0,0x000f,0x0000) aka D3DFMT_X4R4G4B4 |
94 | |
95 | // No 3:3:2, 3:3:2:8, or paletted DXGI formats aka D3DFMT_A8R3G3B2, D3DFMT_R3G3B2, D3DFMT_P8, D3DFMT_A8P8, etc. |
96 | break; |
97 | } |
98 | } |
99 | else if (ddpf.flags & DDPF_LUMINANCE) |
100 | { |
101 | if (ddpf.rgbBitCount == 8) |
102 | { |
103 | if (ISBITMASK(0x000000ff, 0x00000000, 0x00000000, 0x00000000)) |
104 | return DXGI_FORMAT_R8_UNORM; // D3DX10/11 writes this out as DX10 extension |
105 | |
106 | // No DXGI format maps to ISBITMASK(0x0f,0x00,0x00,0xf0) aka D3DFMT_A4L4 |
107 | |
108 | if (ISBITMASK(0x000000ff, 0x00000000, 0x00000000, 0x0000ff00)) |
109 | return DXGI_FORMAT_R8G8_UNORM; // Some DDS writers assume the bitcount should be 8 instead of 16 |
110 | } |
111 | |
112 | if (ddpf.rgbBitCount == 16) |
113 | { |
114 | if (ISBITMASK(0x0000ffff, 0x00000000, 0x00000000, 0x00000000)) |
115 | return DXGI_FORMAT_R16_UNORM; // D3DX10/11 writes this out as DX10 extension |
116 | |
117 | if (ISBITMASK(0x000000ff, 0x00000000, 0x00000000, 0x0000ff00)) |
118 | return DXGI_FORMAT_R8G8_UNORM; // D3DX10/11 writes this out as DX10 extension |
119 | } |
120 | } |
121 | else if (ddpf.flags & DDPF_ALPHA) |
122 | { |
123 | if (ddpf.rgbBitCount == 8) |
124 | return DXGI_FORMAT_A8_UNORM; |
125 | } |
126 | else if (ddpf.flags & DDPF_BUMPDUDV) |
127 | { |
128 | if (ddpf.rgbBitCount == 16) |
129 | { |
130 | if (ISBITMASK(0x00ff, 0xff00, 0x0000, 0x0000)) |
131 | return DXGI_FORMAT_R8G8_SNORM; // D3DX10/11 writes this out as DX10 extension |
132 | } |
133 | |
134 | if (ddpf.rgbBitCount == 32) |
135 | { |
136 | if (ISBITMASK(0x000000ff, 0x0000ff00, 0x00ff0000, 0xff000000)) |
137 | return DXGI_FORMAT_R8G8B8A8_SNORM; // D3DX10/11 writes this out as DX10 extension |
138 | |
139 | if (ISBITMASK(0x0000ffff, 0xffff0000, 0x00000000, 0x00000000)) |
140 | return DXGI_FORMAT_R16G16_SNORM; // D3DX10/11 writes this out as DX10 extension |
141 | |
142 | // No DXGI format maps to ISBITMASK(0x3ff00000, 0x000ffc00, 0x000003ff, 0xc0000000) aka D3DFMT_A2W10V10U10 |
143 | } |
144 | } |
145 | else if (ddpf.flags & DDPF_FOURCC) |
146 | { |
147 | switch (ddpf.fourCC) |
148 | { |
149 | case MakeFourCC('D','X','T','1'): |
150 | return DXGI_FORMAT_BC1_UNORM; |
151 | |
152 | case MakeFourCC('D','X','T','3'): |
153 | return DXGI_FORMAT_BC2_UNORM; |
154 | |
155 | case MakeFourCC('D','X','T','5'): |
156 | return DXGI_FORMAT_BC3_UNORM; |
157 | |
158 | // While pre-multiplied alpha isn't directly supported by the DXGI formats, |
159 | // they are basically the same as these BC formats so they can be mapped |
160 | case MakeFourCC('D','X','T','2'): |
161 | return DXGI_FORMAT_BC2_UNORM; |
162 | |
163 | case MakeFourCC('D','X','T','4'): |
164 | return DXGI_FORMAT_BC3_UNORM; |
165 | |
166 | case MakeFourCC('A','T','I','1'): |
167 | return DXGI_FORMAT_BC4_UNORM; |
168 | |
169 | case MakeFourCC('B','C','4','U'): |
170 | return DXGI_FORMAT_BC4_UNORM; |
171 | |
172 | case MakeFourCC('B','C','4','S'): |
173 | return DXGI_FORMAT_BC4_SNORM; |
174 | |
175 | case MakeFourCC('A','T','I','2'): |
176 | return DXGI_FORMAT_BC5_UNORM; |
177 | |
178 | case MakeFourCC('B','C','5','U'): |
179 | return DXGI_FORMAT_BC5_UNORM; |
180 | |
181 | case MakeFourCC('B','C','5','S'): |
182 | return DXGI_FORMAT_BC5_SNORM; |
183 | |
184 | // BC6H and BC7 are written using the "DX10" extended header |
185 | |
186 | case MakeFourCC('R','G','B','G'): |
187 | return DXGI_FORMAT_R8G8_B8G8_UNORM; |
188 | |
189 | case MakeFourCC('G','R','G','B'): |
190 | return DXGI_FORMAT_G8R8_G8B8_UNORM; |
191 | |
192 | // Check for D3DFORMAT enums being set here |
193 | case 36: // D3DFMT_A16B16G16R16 |
194 | return DXGI_FORMAT_R16G16B16A16_UNORM; |
195 | |
196 | case 110: // D3DFMT_Q16W16V16U16 |
197 | return DXGI_FORMAT_R16G16B16A16_SNORM; |
198 | |
199 | case 111: // D3DFMT_R16F |
200 | return DXGI_FORMAT_R16_FLOAT; |
201 | |
202 | case 112: // D3DFMT_G16R16F |
203 | return DXGI_FORMAT_R16G16_FLOAT; |
204 | |
205 | case 113: // D3DFMT_A16B16G16R16F |
206 | return DXGI_FORMAT_R16G16B16A16_FLOAT; |
207 | |
208 | case 114: // D3DFMT_R32F |
209 | return DXGI_FORMAT_R32_FLOAT; |
210 | |
211 | case 115: // D3DFMT_G32R32F |
212 | return DXGI_FORMAT_R32G32_FLOAT; |
213 | |
214 | case 116: // D3DFMT_A32B32G32R32F |
215 | return DXGI_FORMAT_R32G32B32A32_FLOAT; |
216 | } |
217 | } |
218 | |
219 | return DXGI_FORMAT_UNKNOWN; |
220 | } |
221 | |
222 | static size_t getBitsPerPixel(DXGIFormat fmt) |
223 | { |
224 | switch (fmt) |
225 | { |
226 | case DXGI_FORMAT_R32G32B32A32_TYPELESS: |
227 | case DXGI_FORMAT_R32G32B32A32_FLOAT: |
228 | case DXGI_FORMAT_R32G32B32A32_UINT: |
229 | case DXGI_FORMAT_R32G32B32A32_SINT: |
230 | return 128; |
231 | |
232 | case DXGI_FORMAT_R32G32B32_TYPELESS: |
233 | case DXGI_FORMAT_R32G32B32_FLOAT: |
234 | case DXGI_FORMAT_R32G32B32_UINT: |
235 | case DXGI_FORMAT_R32G32B32_SINT: |
236 | return 96; |
237 | |
238 | case DXGI_FORMAT_R16G16B16A16_TYPELESS: |
239 | case DXGI_FORMAT_R16G16B16A16_FLOAT: |
240 | case DXGI_FORMAT_R16G16B16A16_UNORM: |
241 | case DXGI_FORMAT_R16G16B16A16_UINT: |
242 | case DXGI_FORMAT_R16G16B16A16_SNORM: |
243 | case DXGI_FORMAT_R16G16B16A16_SINT: |
244 | case DXGI_FORMAT_R32G32_TYPELESS: |
245 | case DXGI_FORMAT_R32G32_FLOAT: |
246 | case DXGI_FORMAT_R32G32_UINT: |
247 | case DXGI_FORMAT_R32G32_SINT: |
248 | case DXGI_FORMAT_R32G8X24_TYPELESS: |
249 | case DXGI_FORMAT_D32_FLOAT_S8X24_UINT: |
250 | case DXGI_FORMAT_R32_FLOAT_X8X24_TYPELESS: |
251 | case DXGI_FORMAT_X32_TYPELESS_G8X24_UINT: |
252 | return 64; |
253 | |
254 | case DXGI_FORMAT_R10G10B10A2_TYPELESS: |
255 | case DXGI_FORMAT_R10G10B10A2_UNORM: |
256 | case DXGI_FORMAT_R10G10B10A2_UINT: |
257 | case DXGI_FORMAT_R11G11B10_FLOAT: |
258 | case DXGI_FORMAT_R8G8B8A8_TYPELESS: |
259 | case DXGI_FORMAT_R8G8B8A8_UNORM: |
260 | case DXGI_FORMAT_R8G8B8A8_UNORM_SRGB: |
261 | case DXGI_FORMAT_R8G8B8A8_UINT: |
262 | case DXGI_FORMAT_R8G8B8A8_SNORM: |
263 | case DXGI_FORMAT_R8G8B8A8_SINT: |
264 | case DXGI_FORMAT_R16G16_TYPELESS: |
265 | case DXGI_FORMAT_R16G16_FLOAT: |
266 | case DXGI_FORMAT_R16G16_UNORM: |
267 | case DXGI_FORMAT_R16G16_UINT: |
268 | case DXGI_FORMAT_R16G16_SNORM: |
269 | case DXGI_FORMAT_R16G16_SINT: |
270 | case DXGI_FORMAT_R32_TYPELESS: |
271 | case DXGI_FORMAT_D32_FLOAT: |
272 | case DXGI_FORMAT_R32_FLOAT: |
273 | case DXGI_FORMAT_R32_UINT: |
274 | case DXGI_FORMAT_R32_SINT: |
275 | case DXGI_FORMAT_R24G8_TYPELESS: |
276 | case DXGI_FORMAT_D24_UNORM_S8_UINT: |
277 | case DXGI_FORMAT_R24_UNORM_X8_TYPELESS: |
278 | case DXGI_FORMAT_X24_TYPELESS_G8_UINT: |
279 | case DXGI_FORMAT_R9G9B9E5_SHAREDEXP: |
280 | case DXGI_FORMAT_R8G8_B8G8_UNORM: |
281 | case DXGI_FORMAT_G8R8_G8B8_UNORM: |
282 | case DXGI_FORMAT_B8G8R8A8_UNORM: |
283 | case DXGI_FORMAT_B8G8R8X8_UNORM: |
284 | case DXGI_FORMAT_R10G10B10_XR_BIAS_A2_UNORM: |
285 | case DXGI_FORMAT_B8G8R8A8_TYPELESS: |
286 | case DXGI_FORMAT_B8G8R8A8_UNORM_SRGB: |
287 | case DXGI_FORMAT_B8G8R8X8_TYPELESS: |
288 | case DXGI_FORMAT_B8G8R8X8_UNORM_SRGB: |
289 | return 32; |
290 | |
291 | case DXGI_FORMAT_R8G8_TYPELESS: |
292 | case DXGI_FORMAT_R8G8_UNORM: |
293 | case DXGI_FORMAT_R8G8_UINT: |
294 | case DXGI_FORMAT_R8G8_SNORM: |
295 | case DXGI_FORMAT_R8G8_SINT: |
296 | case DXGI_FORMAT_R16_TYPELESS: |
297 | case DXGI_FORMAT_R16_FLOAT: |
298 | case DXGI_FORMAT_D16_UNORM: |
299 | case DXGI_FORMAT_R16_UNORM: |
300 | case DXGI_FORMAT_R16_UINT: |
301 | case DXGI_FORMAT_R16_SNORM: |
302 | case DXGI_FORMAT_R16_SINT: |
303 | case DXGI_FORMAT_B5G6R5_UNORM: |
304 | case DXGI_FORMAT_B5G5R5A1_UNORM: |
305 | return 16; |
306 | |
307 | case DXGI_FORMAT_R8_TYPELESS: |
308 | case DXGI_FORMAT_R8_UNORM: |
309 | case DXGI_FORMAT_R8_UINT: |
310 | case DXGI_FORMAT_R8_SNORM: |
311 | case DXGI_FORMAT_R8_SINT: |
312 | case DXGI_FORMAT_A8_UNORM: |
313 | return 8; |
314 | |
315 | case DXGI_FORMAT_R1_UNORM: |
316 | return 1; |
317 | |
318 | case DXGI_FORMAT_BC1_TYPELESS: |
319 | case DXGI_FORMAT_BC1_UNORM: |
320 | case DXGI_FORMAT_BC1_UNORM_SRGB: |
321 | case DXGI_FORMAT_BC4_TYPELESS: |
322 | case DXGI_FORMAT_BC4_UNORM: |
323 | case DXGI_FORMAT_BC4_SNORM: |
324 | return 4; |
325 | |
326 | case DXGI_FORMAT_BC2_TYPELESS: |
327 | case DXGI_FORMAT_BC2_UNORM: |
328 | case DXGI_FORMAT_BC2_UNORM_SRGB: |
329 | case DXGI_FORMAT_BC3_TYPELESS: |
330 | case DXGI_FORMAT_BC3_UNORM: |
331 | case DXGI_FORMAT_BC3_UNORM_SRGB: |
332 | case DXGI_FORMAT_BC5_TYPELESS: |
333 | case DXGI_FORMAT_BC5_UNORM: |
334 | case DXGI_FORMAT_BC5_SNORM: |
335 | case DXGI_FORMAT_BC6H_TYPELESS: |
336 | case DXGI_FORMAT_BC6H_UF16: |
337 | case DXGI_FORMAT_BC6H_SF16: |
338 | case DXGI_FORMAT_BC7_TYPELESS: |
339 | case DXGI_FORMAT_BC7_UNORM: |
340 | case DXGI_FORMAT_BC7_UNORM_SRGB: |
341 | return 8; |
342 | |
343 | default: |
344 | return 0; |
345 | } |
346 | } |
347 | |
348 | static bool isBlockCompressed(DXGIFormat fmt) |
349 | { |
350 | switch (fmt) |
351 | { |
352 | case DXGI_FORMAT_BC1_TYPELESS: |
353 | case DXGI_FORMAT_BC1_UNORM: |
354 | case DXGI_FORMAT_BC1_UNORM_SRGB: |
355 | case DXGI_FORMAT_BC4_TYPELESS: |
356 | case DXGI_FORMAT_BC4_UNORM: |
357 | case DXGI_FORMAT_BC4_SNORM: |
358 | case DXGI_FORMAT_BC2_TYPELESS: |
359 | case DXGI_FORMAT_BC2_UNORM: |
360 | case DXGI_FORMAT_BC2_UNORM_SRGB: |
361 | case DXGI_FORMAT_BC3_TYPELESS: |
362 | case DXGI_FORMAT_BC3_UNORM: |
363 | case DXGI_FORMAT_BC3_UNORM_SRGB: |
364 | case DXGI_FORMAT_BC5_TYPELESS: |
365 | case DXGI_FORMAT_BC5_UNORM: |
366 | case DXGI_FORMAT_BC5_SNORM: |
367 | case DXGI_FORMAT_BC6H_TYPELESS: |
368 | case DXGI_FORMAT_BC6H_UF16: |
369 | case DXGI_FORMAT_BC6H_SF16: |
370 | case DXGI_FORMAT_BC7_TYPELESS: |
371 | case DXGI_FORMAT_BC7_UNORM: |
372 | case DXGI_FORMAT_BC7_UNORM_SRGB: |
373 | return true; |
374 | default: |
375 | return false; |
376 | } |
377 | } |
378 | |
379 | bool isDDS(const void *data, size_t dataSize) |
380 | { |
381 | const uint8_t *readData = (const uint8_t *) data; |
382 | ptrdiff_t offset = 0; |
383 | |
384 | // Is the data large enough to hold the DDS header? |
385 | if(dataSize < sizeof(uint32_t) + sizeof(DDSHeader)) |
386 | return false; |
387 | |
388 | // All DDS files start with "DDS ". |
389 | if((*(uint32_t *) readData) != MakeFourCC('D','D','S',' ')) |
390 | return false; |
391 | |
392 | offset += sizeof(uint32_t); |
393 | |
394 | DDSHeader * = (DDSHeader *) &readData[offset]; |
395 | |
396 | // Verify header to validate DDS data. |
397 | if (header->size != sizeof(DDSHeader) || header->format.size != sizeof(DDSPixelFormat)) |
398 | return false; |
399 | |
400 | offset += sizeof(DDSHeader); |
401 | |
402 | // Check for DX10 extension. |
403 | if ((header->format.flags & DDPF_FOURCC) && (header->format.fourCC == MakeFourCC('D','X','1','0'))) |
404 | { |
405 | // Data must be big enough for both headers plus the magic value. |
406 | if (dataSize < (sizeof(uint32_t) + sizeof(DDSHeader) + sizeof(DDSHeader10))) |
407 | return false; |
408 | } |
409 | |
410 | return true; |
411 | } |
412 | |
413 | DXGIFormat getDDSPixelFormat(const void *data, size_t dataSize) |
414 | { |
415 | if (!isDDS(data, dataSize)) |
416 | return DXGI_FORMAT_UNKNOWN; |
417 | |
418 | const uint8_t *readData = (const uint8_t *) data; |
419 | ptrdiff_t offset = sizeof(uint32_t); |
420 | |
421 | DDSHeader * = (DDSHeader *) &readData[offset]; |
422 | offset += sizeof(DDSHeader); |
423 | |
424 | // Check for DX10 extension. |
425 | if ((header->format.flags & DDPF_FOURCC) && (header->format.fourCC == MakeFourCC('D','X','1','0'))) |
426 | { |
427 | DDSHeader10 * = (DDSHeader10 *) &readData[offset]; |
428 | return header10->dxgiFormat; |
429 | } |
430 | |
431 | return getDXGIFormat(header->format); |
432 | } |
433 | |
434 | bool isCompressedDDS(const void *data, size_t dataSize) |
435 | { |
436 | DXGIFormat format = getDDSPixelFormat(data, dataSize); |
437 | return format != DXGI_FORMAT_UNKNOWN && isBlockCompressed(format); |
438 | } |
439 | |
440 | Parser::Parser(const void *data, size_t dataSize) |
441 | : format(DXGI_FORMAT_UNKNOWN) |
442 | { |
443 | parseData(data, dataSize); |
444 | } |
445 | |
446 | Parser::Parser(const Parser &other) |
447 | : texData(other.texData) |
448 | , format(other.format) |
449 | { |
450 | } |
451 | |
452 | Parser::Parser() |
453 | : format(DXGI_FORMAT_UNKNOWN) |
454 | { |
455 | } |
456 | |
457 | Parser &Parser::operator = (const Parser &other) |
458 | { |
459 | texData = other.texData; |
460 | format = other.format; |
461 | |
462 | return *this; |
463 | } |
464 | |
465 | Parser::~Parser() |
466 | { |
467 | } |
468 | |
469 | DXGIFormat Parser::getFormat() const |
470 | { |
471 | return format; |
472 | } |
473 | |
474 | const Image *Parser::getImageData(size_t miplevel) const |
475 | { |
476 | if (miplevel >= texData.size()) |
477 | return 0; |
478 | |
479 | return &texData[miplevel]; |
480 | } |
481 | |
482 | size_t Parser::getMipmapCount() const |
483 | { |
484 | return texData.size(); |
485 | } |
486 | |
487 | size_t Parser::parseImageSize(DXGIFormat fmt, int width, int height) const |
488 | { |
489 | size_t bytes = 0; |
490 | size_t bytesPerBlock = 0; |
491 | |
492 | bool packed = false; |
493 | bool blockCompressed = false; |
494 | |
495 | switch (fmt) |
496 | { |
497 | case DXGI_FORMAT_BC1_TYPELESS: |
498 | case DXGI_FORMAT_BC1_UNORM: |
499 | case DXGI_FORMAT_BC1_UNORM_SRGB: |
500 | case DXGI_FORMAT_BC4_TYPELESS: |
501 | case DXGI_FORMAT_BC4_UNORM: |
502 | case DXGI_FORMAT_BC4_SNORM: |
503 | blockCompressed = true; |
504 | bytesPerBlock = 8; |
505 | break; |
506 | case DXGI_FORMAT_BC2_TYPELESS: |
507 | case DXGI_FORMAT_BC2_UNORM: |
508 | case DXGI_FORMAT_BC2_UNORM_SRGB: |
509 | case DXGI_FORMAT_BC3_TYPELESS: |
510 | case DXGI_FORMAT_BC3_UNORM: |
511 | case DXGI_FORMAT_BC3_UNORM_SRGB: |
512 | case DXGI_FORMAT_BC5_TYPELESS: |
513 | case DXGI_FORMAT_BC5_UNORM: |
514 | case DXGI_FORMAT_BC5_SNORM: |
515 | case DXGI_FORMAT_BC6H_TYPELESS: |
516 | case DXGI_FORMAT_BC6H_UF16: |
517 | case DXGI_FORMAT_BC6H_SF16: |
518 | case DXGI_FORMAT_BC7_TYPELESS: |
519 | case DXGI_FORMAT_BC7_UNORM: |
520 | case DXGI_FORMAT_BC7_UNORM_SRGB: |
521 | blockCompressed = true; |
522 | bytesPerBlock = 16; |
523 | break; |
524 | case DXGI_FORMAT_R8G8_B8G8_UNORM: |
525 | case DXGI_FORMAT_G8R8_G8B8_UNORM: |
526 | packed = true; |
527 | bytesPerBlock = 4; |
528 | break; |
529 | default: |
530 | break; |
531 | } |
532 | |
533 | if (packed) |
534 | { |
535 | size_t rowBytes = (((size_t) width + 1u) >> 1) * bytesPerBlock; |
536 | bytes = rowBytes * height; |
537 | } |
538 | else if (blockCompressed) |
539 | { |
540 | size_t numBlocksWide = width > 0 ? std::max(1, (width + 3) / 4) : 0; |
541 | size_t numBlocksHigh = height > 0 ? std::max(1, (height + 3) / 4) : 0; |
542 | bytes = numBlocksWide * bytesPerBlock * numBlocksHigh; |
543 | } |
544 | else |
545 | { |
546 | size_t bpp = getBitsPerPixel(fmt); |
547 | if (bpp == 0) |
548 | return 0; |
549 | |
550 | // Round up to the nearest byte. |
551 | size_t rowBytes = ((size_t) width * bpp + 7u) / 8u; |
552 | bytes = rowBytes * height; |
553 | } |
554 | |
555 | return bytes; |
556 | } |
557 | |
558 | bool Parser::parseTexData(const uint8_t *data, size_t dataSize, DXGIFormat fmt, int w, int h, int nb_mips) |
559 | { |
560 | size_t offset = 0; |
561 | std::vector<Image> newTexData; |
562 | |
563 | for (int i = 0; i < nb_mips; i++) |
564 | { |
565 | Image img; |
566 | img.width = w; |
567 | img.height = h; |
568 | |
569 | img.dataSize = parseImageSize(fmt, img.width, img.height); |
570 | |
571 | // Make sure the data size is valid. |
572 | if (img.dataSize == 0 || (offset + img.dataSize) > dataSize) |
573 | return false; |
574 | |
575 | // Store the memory address of the data representing this mip level. |
576 | img.data = &data[offset]; |
577 | |
578 | newTexData.push_back(img); |
579 | |
580 | // Move to the next mip level. |
581 | offset += img.dataSize; |
582 | |
583 | w = std::max(w / 2, 1); |
584 | h = std::max(h / 2, 1); |
585 | } |
586 | |
587 | texData = newTexData; |
588 | |
589 | return true; |
590 | } |
591 | |
592 | bool Parser::parseData(const void *data, size_t dataSize) |
593 | { |
594 | if (!isDDS(data, dataSize)) |
595 | return false; |
596 | |
597 | const uint8_t *readData = (const uint8_t *) data; |
598 | ptrdiff_t offset = sizeof(uint32_t); |
599 | |
600 | DDSHeader * = (DDSHeader *) &readData[offset]; |
601 | offset += sizeof(DDSHeader); |
602 | |
603 | // Check for DX10 extension. |
604 | if ((header->format.flags & DDPF_FOURCC) && (header->format.fourCC == MakeFourCC('D','X','1','0'))) |
605 | { |
606 | DDSHeader10 * = (DDSHeader10 *) &readData[offset]; |
607 | offset += sizeof(DDSHeader10); |
608 | |
609 | // We can't deal with 1D/3D textures. |
610 | switch (header10->resourceDimension) |
611 | { |
612 | case D3D10_RESOURCE_DIMENSION_TEXTURE2D: |
613 | case D3D10_RESOURCE_DIMENSION_UNKNOWN: |
614 | break; |
615 | default: |
616 | return false; |
617 | } |
618 | |
619 | // We also can't deal with texture arrays and cubemaps. |
620 | if (header10->arraySize > 1) |
621 | return false; |
622 | |
623 | format = header10->dxgiFormat; |
624 | } |
625 | else |
626 | format = getDXGIFormat(header->format); |
627 | |
628 | if (format == DXGI_FORMAT_UNKNOWN) |
629 | return false; |
630 | |
631 | int w = header->width; |
632 | int h = header->height; |
633 | |
634 | int nb_mips = std::max((int) header->mipMapCount, 1); |
635 | |
636 | return parseTexData(&readData[offset], dataSize - offset, format, w, h, nb_mips); |
637 | } |
638 | |
639 | } // dds |
640 | |