Directory: | ../../../ffmpeg/ |
---|---|
File: | src/libavcodec/dds.c |
Date: | 2022-07-07 01:21:54 |
Exec | Total | Coverage | |
---|---|---|---|
Lines: | 325 | 418 | 77.8% |
Branches: | 168 | 267 | 62.9% |
Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * DirectDraw Surface image decoder | ||
3 | * Copyright (C) 2015 Vittorio Giovara <vittorio.giovara@gmail.com> | ||
4 | * | ||
5 | * This file is part of FFmpeg. | ||
6 | * | ||
7 | * FFmpeg is free software; you can redistribute it and/or | ||
8 | * modify it under the terms of the GNU Lesser General Public | ||
9 | * License as published by the Free Software Foundation; either | ||
10 | * version 2.1 of the License, or (at your option) any later version. | ||
11 | * | ||
12 | * FFmpeg is distributed in the hope that it will be useful, | ||
13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
15 | * Lesser General Public License for more details. | ||
16 | * | ||
17 | * You should have received a copy of the GNU Lesser General Public | ||
18 | * License along with FFmpeg; if not, write to the Free Software | ||
19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
20 | */ | ||
21 | |||
22 | /** | ||
23 | * @file | ||
24 | * DDS decoder | ||
25 | * | ||
26 | * https://msdn.microsoft.com/en-us/library/bb943982%28v=vs.85%29.aspx | ||
27 | */ | ||
28 | |||
29 | #include <stdint.h> | ||
30 | |||
31 | #include "libavutil/libm.h" | ||
32 | #include "libavutil/imgutils.h" | ||
33 | |||
34 | #include "avcodec.h" | ||
35 | #include "bytestream.h" | ||
36 | #include "codec_internal.h" | ||
37 | #include "internal.h" | ||
38 | #include "texturedsp.h" | ||
39 | #include "thread.h" | ||
40 | |||
41 | #define DDPF_FOURCC (1 << 2) | ||
42 | #define DDPF_PALETTE (1 << 5) | ||
43 | #define DDPF_NORMALMAP (1U << 31) | ||
44 | |||
45 | enum DDSPostProc { | ||
46 | DDS_NONE = 0, | ||
47 | DDS_ALPHA_EXP, | ||
48 | DDS_NORMAL_MAP, | ||
49 | DDS_RAW_YCOCG, | ||
50 | DDS_SWAP_ALPHA, | ||
51 | DDS_SWIZZLE_A2XY, | ||
52 | DDS_SWIZZLE_RBXG, | ||
53 | DDS_SWIZZLE_RGXB, | ||
54 | DDS_SWIZZLE_RXBG, | ||
55 | DDS_SWIZZLE_RXGB, | ||
56 | DDS_SWIZZLE_XGBR, | ||
57 | DDS_SWIZZLE_XRBG, | ||
58 | DDS_SWIZZLE_XGXR, | ||
59 | }; | ||
60 | |||
61 | enum DDSDXGIFormat { | ||
62 | DXGI_FORMAT_R16G16B16A16_TYPELESS = 9, | ||
63 | DXGI_FORMAT_R16G16B16A16_FLOAT = 10, | ||
64 | DXGI_FORMAT_R16G16B16A16_UNORM = 11, | ||
65 | DXGI_FORMAT_R16G16B16A16_UINT = 12, | ||
66 | DXGI_FORMAT_R16G16B16A16_SNORM = 13, | ||
67 | DXGI_FORMAT_R16G16B16A16_SINT = 14, | ||
68 | |||
69 | DXGI_FORMAT_R8G8B8A8_TYPELESS = 27, | ||
70 | DXGI_FORMAT_R8G8B8A8_UNORM = 28, | ||
71 | DXGI_FORMAT_R8G8B8A8_UNORM_SRGB = 29, | ||
72 | DXGI_FORMAT_R8G8B8A8_UINT = 30, | ||
73 | DXGI_FORMAT_R8G8B8A8_SNORM = 31, | ||
74 | DXGI_FORMAT_R8G8B8A8_SINT = 32, | ||
75 | |||
76 | DXGI_FORMAT_BC1_TYPELESS = 70, | ||
77 | DXGI_FORMAT_BC1_UNORM = 71, | ||
78 | DXGI_FORMAT_BC1_UNORM_SRGB = 72, | ||
79 | DXGI_FORMAT_BC2_TYPELESS = 73, | ||
80 | DXGI_FORMAT_BC2_UNORM = 74, | ||
81 | DXGI_FORMAT_BC2_UNORM_SRGB = 75, | ||
82 | DXGI_FORMAT_BC3_TYPELESS = 76, | ||
83 | DXGI_FORMAT_BC3_UNORM = 77, | ||
84 | DXGI_FORMAT_BC3_UNORM_SRGB = 78, | ||
85 | DXGI_FORMAT_BC4_TYPELESS = 79, | ||
86 | DXGI_FORMAT_BC4_UNORM = 80, | ||
87 | DXGI_FORMAT_BC4_SNORM = 81, | ||
88 | DXGI_FORMAT_BC5_TYPELESS = 82, | ||
89 | DXGI_FORMAT_BC5_UNORM = 83, | ||
90 | DXGI_FORMAT_BC5_SNORM = 84, | ||
91 | DXGI_FORMAT_B5G6R5_UNORM = 85, | ||
92 | DXGI_FORMAT_B8G8R8A8_UNORM = 87, | ||
93 | DXGI_FORMAT_B8G8R8X8_UNORM = 88, | ||
94 | DXGI_FORMAT_B8G8R8A8_TYPELESS = 90, | ||
95 | DXGI_FORMAT_B8G8R8A8_UNORM_SRGB = 91, | ||
96 | DXGI_FORMAT_B8G8R8X8_TYPELESS = 92, | ||
97 | DXGI_FORMAT_B8G8R8X8_UNORM_SRGB = 93, | ||
98 | }; | ||
99 | |||
100 | typedef struct DDSContext { | ||
101 | TextureDSPContext texdsp; | ||
102 | GetByteContext gbc; | ||
103 | |||
104 | int compressed; | ||
105 | int paletted; | ||
106 | int bpp; | ||
107 | enum DDSPostProc postproc; | ||
108 | |||
109 | TextureDSPThreadContext dec; | ||
110 | } DDSContext; | ||
111 | |||
112 | 96 | static int parse_pixel_format(AVCodecContext *avctx) | |
113 | { | ||
114 | 96 | DDSContext *ctx = avctx->priv_data; | |
115 | 96 | GetByteContext *gbc = &ctx->gbc; | |
116 | uint32_t flags, fourcc, gimp_tag; | ||
117 | enum DDSDXGIFormat dxgi; | ||
118 | int size, bpp, r, g, b, a; | ||
119 | int alpha_exponent, ycocg_classic, ycocg_scaled, normal_map, array; | ||
120 | |||
121 | /* Alternative DDS implementations use reserved1 as custom header. */ | ||
122 | 96 | bytestream2_skip(gbc, 4 * 3); | |
123 | 96 | gimp_tag = bytestream2_get_le32(gbc); | |
124 | 96 | alpha_exponent = gimp_tag == MKTAG('A', 'E', 'X', 'P'); | |
125 | 96 | ycocg_classic = gimp_tag == MKTAG('Y', 'C', 'G', '1'); | |
126 | 96 | ycocg_scaled = gimp_tag == MKTAG('Y', 'C', 'G', '2'); | |
127 | 96 | bytestream2_skip(gbc, 4 * 7); | |
128 | |||
129 | /* Now the real DDPF starts. */ | ||
130 | 96 | size = bytestream2_get_le32(gbc); | |
131 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 96 times.
|
96 | if (size != 32) { |
132 | ✗ | av_log(avctx, AV_LOG_ERROR, "Invalid pixel format header %d.\n", size); | |
133 | ✗ | return AVERROR_INVALIDDATA; | |
134 | } | ||
135 | 96 | flags = bytestream2_get_le32(gbc); | |
136 | 96 | ctx->compressed = flags & DDPF_FOURCC; | |
137 | 96 | ctx->paletted = flags & DDPF_PALETTE; | |
138 | 96 | normal_map = flags & DDPF_NORMALMAP; | |
139 | 96 | fourcc = bytestream2_get_le32(gbc); | |
140 | |||
141 |
3/4✓ Branch 0 taken 68 times.
✓ Branch 1 taken 28 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 68 times.
|
96 | if (ctx->compressed && ctx->paletted) { |
142 | ✗ | av_log(avctx, AV_LOG_WARNING, | |
143 | "Disabling invalid palette flag for compressed dds.\n"); | ||
144 | ✗ | ctx->paletted = 0; | |
145 | } | ||
146 | |||
147 | 96 | bpp = ctx->bpp = bytestream2_get_le32(gbc); // rgbbitcount | |
148 | 96 | r = bytestream2_get_le32(gbc); // rbitmask | |
149 | 96 | g = bytestream2_get_le32(gbc); // gbitmask | |
150 | 96 | b = bytestream2_get_le32(gbc); // bbitmask | |
151 | 96 | a = bytestream2_get_le32(gbc); // abitmask | |
152 | |||
153 | 96 | bytestream2_skip(gbc, 4); // caps | |
154 | 96 | bytestream2_skip(gbc, 4); // caps2 | |
155 | 96 | bytestream2_skip(gbc, 4); // caps3 | |
156 | 96 | bytestream2_skip(gbc, 4); // caps4 | |
157 | 96 | bytestream2_skip(gbc, 4); // reserved2 | |
158 | |||
159 | 96 | av_log(avctx, AV_LOG_VERBOSE, "fourcc %s bpp %d " | |
160 | 96 | "r 0x%x g 0x%x b 0x%x a 0x%x\n", av_fourcc2str(fourcc), bpp, r, g, b, a); | |
161 |
2/2✓ Branch 0 taken 10 times.
✓ Branch 1 taken 86 times.
|
96 | if (gimp_tag) |
162 | 10 | av_log(avctx, AV_LOG_VERBOSE, "and GIMP-DDS tag %s\n", av_fourcc2str(gimp_tag)); | |
163 | |||
164 |
2/2✓ Branch 0 taken 68 times.
✓ Branch 1 taken 28 times.
|
96 | if (ctx->compressed) |
165 | 68 | avctx->pix_fmt = AV_PIX_FMT_RGBA; | |
166 | |||
167 |
2/2✓ Branch 0 taken 68 times.
✓ Branch 1 taken 28 times.
|
96 | if (ctx->compressed) { |
168 | 68 | ctx->dec.raw_ratio = 16; | |
169 |
15/17✓ Branch 0 taken 6 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 2 times.
✓ Branch 4 taken 24 times.
✓ Branch 5 taken 2 times.
✓ Branch 6 taken 2 times.
✓ Branch 7 taken 2 times.
✓ Branch 8 taken 4 times.
✗ Branch 9 not taken.
✓ Branch 10 taken 2 times.
✓ Branch 11 taken 2 times.
✓ Branch 12 taken 2 times.
✓ Branch 13 taken 2 times.
✓ Branch 14 taken 2 times.
✓ Branch 15 taken 12 times.
✗ Branch 16 not taken.
|
68 | switch (fourcc) { |
170 | 6 | case MKTAG('D', 'X', 'T', '1'): | |
171 | 6 | ctx->dec.tex_ratio = 8; | |
172 | 6 | ctx->dec.tex_funct = ctx->texdsp.dxt1a_block; | |
173 | 68 | break; | |
174 | 2 | case MKTAG('D', 'X', 'T', '2'): | |
175 | 2 | ctx->dec.tex_ratio = 16; | |
176 | 2 | ctx->dec.tex_funct = ctx->texdsp.dxt2_block; | |
177 | 2 | break; | |
178 | 2 | case MKTAG('D', 'X', 'T', '3'): | |
179 | 2 | ctx->dec.tex_ratio = 16; | |
180 | 2 | ctx->dec.tex_funct = ctx->texdsp.dxt3_block; | |
181 | 2 | break; | |
182 | 2 | case MKTAG('D', 'X', 'T', '4'): | |
183 | 2 | ctx->dec.tex_ratio = 16; | |
184 | 2 | ctx->dec.tex_funct = ctx->texdsp.dxt4_block; | |
185 | 2 | break; | |
186 | 24 | case MKTAG('D', 'X', 'T', '5'): | |
187 | 24 | ctx->dec.tex_ratio = 16; | |
188 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 22 times.
|
24 | if (ycocg_scaled) |
189 | 2 | ctx->dec.tex_funct = ctx->texdsp.dxt5ys_block; | |
190 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 20 times.
|
22 | else if (ycocg_classic) |
191 | 2 | ctx->dec.tex_funct = ctx->texdsp.dxt5y_block; | |
192 | else | ||
193 | 20 | ctx->dec.tex_funct = ctx->texdsp.dxt5_block; | |
194 | 24 | break; | |
195 | 2 | case MKTAG('R', 'X', 'G', 'B'): | |
196 | 2 | ctx->dec.tex_ratio = 16; | |
197 | 2 | ctx->dec.tex_funct = ctx->texdsp.dxt5_block; | |
198 | /* This format may be considered as a normal map, | ||
199 | * but it is handled differently in a separate postproc. */ | ||
200 | 2 | ctx->postproc = DDS_SWIZZLE_RXGB; | |
201 | 2 | normal_map = 0; | |
202 | 2 | break; | |
203 | 2 | case MKTAG('A', 'T', 'I', '1'): | |
204 | case MKTAG('B', 'C', '4', 'U'): | ||
205 | 2 | ctx->dec.tex_ratio = 8; | |
206 | 2 | ctx->dec.tex_funct = ctx->texdsp.rgtc1u_block; | |
207 | 2 | break; | |
208 | 2 | case MKTAG('B', 'C', '4', 'S'): | |
209 | 2 | ctx->dec.tex_ratio = 8; | |
210 | 2 | ctx->dec.tex_funct = ctx->texdsp.rgtc1s_block; | |
211 | 2 | break; | |
212 | 4 | case MKTAG('A', 'T', 'I', '2'): | |
213 | /* RGT2 variant with swapped R and G (3Dc)*/ | ||
214 | 4 | ctx->dec.tex_ratio = 16; | |
215 | 4 | ctx->dec.tex_funct = ctx->texdsp.dxn3dc_block; | |
216 | 4 | break; | |
217 | ✗ | case MKTAG('B', 'C', '5', 'U'): | |
218 | ✗ | ctx->dec.tex_ratio = 16; | |
219 | ✗ | ctx->dec.tex_funct = ctx->texdsp.rgtc2u_block; | |
220 | ✗ | break; | |
221 | 2 | case MKTAG('B', 'C', '5', 'S'): | |
222 | 2 | ctx->dec.tex_ratio = 16; | |
223 | 2 | ctx->dec.tex_funct = ctx->texdsp.rgtc2s_block; | |
224 | 2 | break; | |
225 | 2 | case MKTAG('U', 'Y', 'V', 'Y'): | |
226 | 2 | ctx->compressed = 0; | |
227 | 2 | avctx->pix_fmt = AV_PIX_FMT_UYVY422; | |
228 | 2 | break; | |
229 | 2 | case MKTAG('Y', 'U', 'Y', '2'): | |
230 | 2 | ctx->compressed = 0; | |
231 | 2 | avctx->pix_fmt = AV_PIX_FMT_YUYV422; | |
232 | 2 | break; | |
233 | 2 | case MKTAG('P', '8', ' ', ' '): | |
234 | /* ATI Palette8, same as normal palette */ | ||
235 | 2 | ctx->compressed = 0; | |
236 | 2 | ctx->paletted = 1; | |
237 | 2 | avctx->pix_fmt = AV_PIX_FMT_PAL8; | |
238 | 2 | break; | |
239 | 2 | case MKTAG('G', '1', ' ', ' '): | |
240 | 2 | ctx->compressed = 0; | |
241 | 2 | avctx->pix_fmt = AV_PIX_FMT_MONOBLACK; | |
242 | 2 | break; | |
243 | 12 | case MKTAG('D', 'X', '1', '0'): | |
244 | /* DirectX 10 extra header */ | ||
245 | 12 | dxgi = bytestream2_get_le32(gbc); | |
246 | 12 | bytestream2_skip(gbc, 4); // resourceDimension | |
247 | 12 | bytestream2_skip(gbc, 4); // miscFlag | |
248 | 12 | array = bytestream2_get_le32(gbc); | |
249 | 12 | bytestream2_skip(gbc, 4); // miscFlag2 | |
250 | |||
251 |
1/2✓ Branch 0 taken 12 times.
✗ Branch 1 not taken.
|
12 | if (array != 0) |
252 | 12 | av_log(avctx, AV_LOG_VERBOSE, | |
253 | "Found array of size %d (ignored).\n", array); | ||
254 | |||
255 | /* Only BC[1-5] are actually compressed. */ | ||
256 |
2/4✓ Branch 0 taken 12 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 12 times.
✗ Branch 3 not taken.
|
12 | ctx->compressed = (dxgi >= 70) && (dxgi <= 84); |
257 | |||
258 | 12 | av_log(avctx, AV_LOG_VERBOSE, "DXGI format %d.\n", dxgi); | |
259 | switch (dxgi) { | ||
260 | /* RGB types. */ | ||
261 | ✗ | case DXGI_FORMAT_R16G16B16A16_TYPELESS: | |
262 | case DXGI_FORMAT_R16G16B16A16_FLOAT: | ||
263 | case DXGI_FORMAT_R16G16B16A16_UNORM: | ||
264 | case DXGI_FORMAT_R16G16B16A16_UINT: | ||
265 | case DXGI_FORMAT_R16G16B16A16_SNORM: | ||
266 | case DXGI_FORMAT_R16G16B16A16_SINT: | ||
267 | ✗ | avctx->pix_fmt = AV_PIX_FMT_BGRA64; | |
268 | ✗ | break; | |
269 | ✗ | case DXGI_FORMAT_R8G8B8A8_UNORM_SRGB: | |
270 | ✗ | avctx->colorspace = AVCOL_SPC_RGB; | |
271 | ✗ | case DXGI_FORMAT_R8G8B8A8_TYPELESS: | |
272 | case DXGI_FORMAT_R8G8B8A8_UNORM: | ||
273 | case DXGI_FORMAT_R8G8B8A8_UINT: | ||
274 | case DXGI_FORMAT_R8G8B8A8_SNORM: | ||
275 | case DXGI_FORMAT_R8G8B8A8_SINT: | ||
276 | ✗ | avctx->pix_fmt = AV_PIX_FMT_BGRA; | |
277 | ✗ | break; | |
278 | ✗ | case DXGI_FORMAT_B8G8R8A8_UNORM_SRGB: | |
279 | ✗ | avctx->colorspace = AVCOL_SPC_RGB; | |
280 | ✗ | case DXGI_FORMAT_B8G8R8A8_TYPELESS: | |
281 | case DXGI_FORMAT_B8G8R8A8_UNORM: | ||
282 | ✗ | avctx->pix_fmt = AV_PIX_FMT_RGBA; | |
283 | ✗ | break; | |
284 | ✗ | case DXGI_FORMAT_B8G8R8X8_UNORM_SRGB: | |
285 | ✗ | avctx->colorspace = AVCOL_SPC_RGB; | |
286 | ✗ | case DXGI_FORMAT_B8G8R8X8_TYPELESS: | |
287 | case DXGI_FORMAT_B8G8R8X8_UNORM: | ||
288 | ✗ | avctx->pix_fmt = AV_PIX_FMT_RGBA; // opaque | |
289 | ✗ | break; | |
290 | ✗ | case DXGI_FORMAT_B5G6R5_UNORM: | |
291 | ✗ | avctx->pix_fmt = AV_PIX_FMT_RGB565LE; | |
292 | ✗ | break; | |
293 | /* Texture types. */ | ||
294 | ✗ | case DXGI_FORMAT_BC1_UNORM_SRGB: | |
295 | ✗ | avctx->colorspace = AVCOL_SPC_RGB; | |
296 | 4 | case DXGI_FORMAT_BC1_TYPELESS: | |
297 | case DXGI_FORMAT_BC1_UNORM: | ||
298 | 4 | ctx->dec.tex_ratio = 8; | |
299 | 4 | ctx->dec.tex_funct = ctx->texdsp.dxt1a_block; | |
300 | 4 | break; | |
301 | ✗ | case DXGI_FORMAT_BC2_UNORM_SRGB: | |
302 | ✗ | avctx->colorspace = AVCOL_SPC_RGB; | |
303 | 2 | case DXGI_FORMAT_BC2_TYPELESS: | |
304 | case DXGI_FORMAT_BC2_UNORM: | ||
305 | 2 | ctx->dec.tex_ratio = 16; | |
306 | 2 | ctx->dec.tex_funct = ctx->texdsp.dxt3_block; | |
307 | 2 | break; | |
308 | ✗ | case DXGI_FORMAT_BC3_UNORM_SRGB: | |
309 | ✗ | avctx->colorspace = AVCOL_SPC_RGB; | |
310 | 2 | case DXGI_FORMAT_BC3_TYPELESS: | |
311 | case DXGI_FORMAT_BC3_UNORM: | ||
312 | 2 | ctx->dec.tex_ratio = 16; | |
313 | 2 | ctx->dec.tex_funct = ctx->texdsp.dxt5_block; | |
314 | 2 | break; | |
315 | 2 | case DXGI_FORMAT_BC4_TYPELESS: | |
316 | case DXGI_FORMAT_BC4_UNORM: | ||
317 | 2 | ctx->dec.tex_ratio = 8; | |
318 | 2 | ctx->dec.tex_funct = ctx->texdsp.rgtc1u_block; | |
319 | 2 | break; | |
320 | ✗ | case DXGI_FORMAT_BC4_SNORM: | |
321 | ✗ | ctx->dec.tex_ratio = 8; | |
322 | ✗ | ctx->dec.tex_funct = ctx->texdsp.rgtc1s_block; | |
323 | ✗ | break; | |
324 | 2 | case DXGI_FORMAT_BC5_TYPELESS: | |
325 | case DXGI_FORMAT_BC5_UNORM: | ||
326 | 2 | ctx->dec.tex_ratio = 16; | |
327 | 2 | ctx->dec.tex_funct = ctx->texdsp.rgtc2u_block; | |
328 | 2 | break; | |
329 | ✗ | case DXGI_FORMAT_BC5_SNORM: | |
330 | ✗ | ctx->dec.tex_ratio = 16; | |
331 | ✗ | ctx->dec.tex_funct = ctx->texdsp.rgtc2s_block; | |
332 | ✗ | break; | |
333 | ✗ | default: | |
334 | ✗ | av_log(avctx, AV_LOG_ERROR, | |
335 | "Unsupported DXGI format %d.\n", dxgi); | ||
336 | ✗ | return AVERROR_INVALIDDATA; | |
337 | } | ||
338 | 12 | break; | |
339 | ✗ | default: | |
340 | ✗ | av_log(avctx, AV_LOG_ERROR, "Unsupported %s fourcc.\n", av_fourcc2str(fourcc)); | |
341 | ✗ | return AVERROR_INVALIDDATA; | |
342 | } | ||
343 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 26 times.
|
28 | } else if (ctx->paletted) { |
344 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2 | if (bpp == 8) { |
345 | 2 | avctx->pix_fmt = AV_PIX_FMT_PAL8; | |
346 | } else { | ||
347 | ✗ | av_log(avctx, AV_LOG_ERROR, "Unsupported palette bpp %d.\n", bpp); | |
348 | ✗ | return AVERROR_INVALIDDATA; | |
349 | } | ||
350 | } else { | ||
351 | /* 4 bpp */ | ||
352 |
1/10✗ Branch 0 not taken.
✓ Branch 1 taken 26 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✗ Branch 9 not taken.
|
26 | if (bpp == 4 && r == 0 && g == 0 && b == 0 && a == 0) |
353 | ✗ | avctx->pix_fmt = AV_PIX_FMT_PAL8; | |
354 | /* 8 bpp */ | ||
355 |
7/10✓ Branch 0 taken 4 times.
✓ Branch 1 taken 22 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 2 times.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 2 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 2 times.
✗ Branch 9 not taken.
|
26 | else if (bpp == 8 && r == 0xff && g == 0 && b == 0 && a == 0) |
356 | 2 | avctx->pix_fmt = AV_PIX_FMT_GRAY8; | |
357 |
6/10✓ Branch 0 taken 2 times.
✓ Branch 1 taken 22 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 2 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 2 times.
✗ Branch 9 not taken.
|
24 | else if (bpp == 8 && r == 0 && g == 0 && b == 0 && a == 0xff) |
358 | 2 | avctx->pix_fmt = AV_PIX_FMT_GRAY8; | |
359 | /* 16 bpp */ | ||
360 |
7/10✓ Branch 0 taken 8 times.
✓ Branch 1 taken 14 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 6 times.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 2 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 2 times.
✗ Branch 9 not taken.
|
22 | else if (bpp == 16 && r == 0xff && g == 0 && b == 0 && a == 0xff00) |
361 | 2 | avctx->pix_fmt = AV_PIX_FMT_YA8; | |
362 |
3/10✓ Branch 0 taken 6 times.
✓ Branch 1 taken 14 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 6 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✗ Branch 9 not taken.
|
20 | else if (bpp == 16 && r == 0xff00 && g == 0 && b == 0 && a == 0xff) { |
363 | ✗ | avctx->pix_fmt = AV_PIX_FMT_YA8; | |
364 | ✗ | ctx->postproc = DDS_SWAP_ALPHA; | |
365 | } | ||
366 |
3/10✓ Branch 0 taken 6 times.
✓ Branch 1 taken 14 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 6 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✗ Branch 9 not taken.
|
20 | else if (bpp == 16 && r == 0xffff && g == 0 && b == 0 && a == 0) |
367 | ✗ | avctx->pix_fmt = AV_PIX_FMT_GRAY16LE; | |
368 |
8/10✓ Branch 0 taken 6 times.
✓ Branch 1 taken 14 times.
✓ Branch 2 taken 4 times.
✓ Branch 3 taken 2 times.
✓ Branch 4 taken 4 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 4 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 2 times.
✓ Branch 9 taken 2 times.
|
20 | else if (bpp == 16 && r == 0x7c00 && g == 0x3e0 && b == 0x1f && a == 0) |
369 | 2 | avctx->pix_fmt = AV_PIX_FMT_RGB555LE; | |
370 |
7/10✓ Branch 0 taken 4 times.
✓ Branch 1 taken 14 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 2 times.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 2 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 2 times.
✗ Branch 9 not taken.
|
18 | else if (bpp == 16 && r == 0x7c00 && g == 0x3e0 && b == 0x1f && a == 0x8000) |
371 | 2 | avctx->pix_fmt = AV_PIX_FMT_RGB555LE; // alpha ignored | |
372 |
6/10✓ Branch 0 taken 2 times.
✓ Branch 1 taken 14 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 2 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 2 times.
✗ Branch 9 not taken.
|
16 | else if (bpp == 16 && r == 0xf800 && g == 0x7e0 && b == 0x1f && a == 0) |
373 | 2 | avctx->pix_fmt = AV_PIX_FMT_RGB565LE; | |
374 | /* 24 bpp */ | ||
375 |
6/10✓ Branch 0 taken 2 times.
✓ Branch 1 taken 12 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 2 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 2 times.
✗ Branch 9 not taken.
|
14 | else if (bpp == 24 && r == 0xff0000 && g == 0xff00 && b == 0xff && a == 0) |
376 | 2 | avctx->pix_fmt = AV_PIX_FMT_BGR24; | |
377 | /* 32 bpp */ | ||
378 |
7/10✓ Branch 0 taken 12 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 10 times.
✓ Branch 3 taken 2 times.
✓ Branch 4 taken 10 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 10 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 2 times.
✓ Branch 9 taken 8 times.
|
12 | else if (bpp == 32 && r == 0xff0000 && g == 0xff00 && b == 0xff && a == 0) |
379 | 2 | avctx->pix_fmt = AV_PIX_FMT_BGR0; // opaque | |
380 |
6/10✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 8 times.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 2 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 2 times.
✗ Branch 9 not taken.
|
10 | else if (bpp == 32 && r == 0xff && g == 0xff00 && b == 0xff0000 && a == 0) |
381 | 2 | avctx->pix_fmt = AV_PIX_FMT_RGB0; // opaque | |
382 |
5/10✓ Branch 0 taken 8 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 8 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 8 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 8 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 8 times.
✗ Branch 9 not taken.
|
8 | else if (bpp == 32 && r == 0xff0000 && g == 0xff00 && b == 0xff && a == 0xff000000) |
383 | 8 | avctx->pix_fmt = AV_PIX_FMT_BGRA; | |
384 | ✗ | else if (bpp == 32 && r == 0xff && g == 0xff00 && b == 0xff0000 && a == 0xff000000) | |
385 | ✗ | avctx->pix_fmt = AV_PIX_FMT_RGBA; | |
386 | /* give up */ | ||
387 | else { | ||
388 | ✗ | av_log(avctx, AV_LOG_ERROR, "Unknown pixel format " | |
389 | "[bpp %d r 0x%x g 0x%x b 0x%x a 0x%x].\n", bpp, r, g, b, a); | ||
390 | ✗ | return AVERROR_INVALIDDATA; | |
391 | } | ||
392 | } | ||
393 | |||
394 | /* Set any remaining post-proc that should happen before frame is ready. */ | ||
395 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 92 times.
|
96 | if (alpha_exponent) |
396 | 4 | ctx->postproc = DDS_ALPHA_EXP; | |
397 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 88 times.
|
92 | else if (normal_map) |
398 | 4 | ctx->postproc = DDS_NORMAL_MAP; | |
399 |
4/4✓ Branch 0 taken 4 times.
✓ Branch 1 taken 84 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 2 times.
|
88 | else if (ycocg_classic && !ctx->compressed) |
400 | 2 | ctx->postproc = DDS_RAW_YCOCG; | |
401 | |||
402 | /* ATI/NVidia variants sometimes add swizzling in bpp. */ | ||
403 |
9/9✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 2 times.
✓ Branch 4 taken 2 times.
✓ Branch 5 taken 2 times.
✓ Branch 6 taken 2 times.
✓ Branch 7 taken 2 times.
✓ Branch 8 taken 80 times.
|
96 | switch (bpp) { |
404 | 2 | case MKTAG('A', '2', 'X', 'Y'): | |
405 | 2 | ctx->postproc = DDS_SWIZZLE_A2XY; | |
406 | 2 | break; | |
407 | 2 | case MKTAG('x', 'G', 'B', 'R'): | |
408 | 2 | ctx->postproc = DDS_SWIZZLE_XGBR; | |
409 | 2 | break; | |
410 | 2 | case MKTAG('x', 'R', 'B', 'G'): | |
411 | 2 | ctx->postproc = DDS_SWIZZLE_XRBG; | |
412 | 2 | break; | |
413 | 2 | case MKTAG('R', 'B', 'x', 'G'): | |
414 | 2 | ctx->postproc = DDS_SWIZZLE_RBXG; | |
415 | 2 | break; | |
416 | 2 | case MKTAG('R', 'G', 'x', 'B'): | |
417 | 2 | ctx->postproc = DDS_SWIZZLE_RGXB; | |
418 | 2 | break; | |
419 | 2 | case MKTAG('R', 'x', 'B', 'G'): | |
420 | 2 | ctx->postproc = DDS_SWIZZLE_RXBG; | |
421 | 2 | break; | |
422 | 2 | case MKTAG('x', 'G', 'x', 'R'): | |
423 | 2 | ctx->postproc = DDS_SWIZZLE_XGXR; | |
424 | 2 | break; | |
425 | 2 | case MKTAG('A', '2', 'D', '5'): | |
426 | 2 | ctx->postproc = DDS_NORMAL_MAP; | |
427 | 2 | break; | |
428 | } | ||
429 | |||
430 | 96 | return 0; | |
431 | } | ||
432 | |||
433 | 26 | static void do_swizzle(AVFrame *frame, int x, int y) | |
434 | { | ||
435 | int i; | ||
436 |
2/2✓ Branch 0 taken 106496 times.
✓ Branch 1 taken 26 times.
|
106522 | for (i = 0; i < frame->linesize[0] * frame->height; i += 4) { |
437 | 106496 | uint8_t *src = frame->data[0] + i; | |
438 | 106496 | FFSWAP(uint8_t, src[x], src[y]); | |
439 | } | ||
440 | 26 | } | |
441 | |||
442 | 28 | static void run_postproc(AVCodecContext *avctx, AVFrame *frame) | |
443 | { | ||
444 | 28 | DDSContext *ctx = avctx->priv_data; | |
445 | int i, x_off; | ||
446 | |||
447 |
11/13✓ Branch 0 taken 4 times.
✓ Branch 1 taken 6 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 2 times.
✓ Branch 5 taken 2 times.
✓ Branch 6 taken 2 times.
✓ Branch 7 taken 2 times.
✓ Branch 8 taken 2 times.
✓ Branch 9 taken 2 times.
✓ Branch 10 taken 2 times.
✓ Branch 11 taken 2 times.
✗ Branch 12 not taken.
|
28 | switch (ctx->postproc) { |
448 | 4 | case DDS_ALPHA_EXP: | |
449 | /* Alpha-exponential mode divides each channel by the maximum | ||
450 | * R, G or B value, and stores the multiplying factor in the | ||
451 | * alpha channel. */ | ||
452 | 4 | av_log(avctx, AV_LOG_DEBUG, "Post-processing alpha exponent.\n"); | |
453 | |||
454 |
2/2✓ Branch 0 taken 16384 times.
✓ Branch 1 taken 4 times.
|
16388 | for (i = 0; i < frame->linesize[0] * frame->height; i += 4) { |
455 | 16384 | uint8_t *src = frame->data[0] + i; | |
456 | 16384 | int r = src[0]; | |
457 | 16384 | int g = src[1]; | |
458 | 16384 | int b = src[2]; | |
459 | 16384 | int a = src[3]; | |
460 | |||
461 | 16384 | src[0] = r * a / 255; | |
462 | 16384 | src[1] = g * a / 255; | |
463 | 16384 | src[2] = b * a / 255; | |
464 | 16384 | src[3] = 255; | |
465 | } | ||
466 | 4 | break; | |
467 | 6 | case DDS_NORMAL_MAP: | |
468 | /* Normal maps work in the XYZ color space and they encode | ||
469 | * X in R or in A, depending on the texture type, Y in G and | ||
470 | * derive Z with a square root of the distance. | ||
471 | * | ||
472 | * http://www.realtimecollisiondetection.net/blog/?p=28 */ | ||
473 | 6 | av_log(avctx, AV_LOG_DEBUG, "Post-processing normal map.\n"); | |
474 | |||
475 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 4 times.
|
6 | x_off = ctx->dec.tex_ratio == 8 ? 0 : 3; |
476 |
2/2✓ Branch 0 taken 24576 times.
✓ Branch 1 taken 6 times.
|
24582 | for (i = 0; i < frame->linesize[0] * frame->height; i += 4) { |
477 | 24576 | uint8_t *src = frame->data[0] + i; | |
478 | 24576 | int x = src[x_off]; | |
479 | 24576 | int y = src[1]; | |
480 | 24576 | int z = 127; | |
481 | |||
482 | 24576 | int d = (255 * 255 - x * x - y * y) / 2; | |
483 |
2/2✓ Branch 0 taken 20740 times.
✓ Branch 1 taken 3836 times.
|
24576 | if (d > 0) |
484 | 20740 | z = lrint(sqrtf(d)); | |
485 | |||
486 | 24576 | src[0] = x; | |
487 | 24576 | src[1] = y; | |
488 | 24576 | src[2] = z; | |
489 | 24576 | src[3] = 255; | |
490 | } | ||
491 | 6 | break; | |
492 | 2 | case DDS_RAW_YCOCG: | |
493 | /* Data is Y-Co-Cg-A and not RGBA, but they are represented | ||
494 | * with the same masks in the DDPF header. */ | ||
495 | 2 | av_log(avctx, AV_LOG_DEBUG, "Post-processing raw YCoCg.\n"); | |
496 | |||
497 |
2/2✓ Branch 0 taken 8192 times.
✓ Branch 1 taken 2 times.
|
8194 | for (i = 0; i < frame->linesize[0] * frame->height; i += 4) { |
498 | 8192 | uint8_t *src = frame->data[0] + i; | |
499 | 8192 | int a = src[0]; | |
500 | 8192 | int cg = src[1] - 128; | |
501 | 8192 | int co = src[2] - 128; | |
502 | 8192 | int y = src[3]; | |
503 | |||
504 | 8192 | src[0] = av_clip_uint8(y + co - cg); | |
505 | 8192 | src[1] = av_clip_uint8(y + cg); | |
506 | 8192 | src[2] = av_clip_uint8(y - co - cg); | |
507 | 8192 | src[3] = a; | |
508 | } | ||
509 | 2 | break; | |
510 | ✗ | case DDS_SWAP_ALPHA: | |
511 | /* Alpha and Luma are stored swapped. */ | ||
512 | ✗ | av_log(avctx, AV_LOG_DEBUG, "Post-processing swapped Luma/Alpha.\n"); | |
513 | |||
514 | ✗ | for (i = 0; i < frame->linesize[0] * frame->height; i += 2) { | |
515 | ✗ | uint8_t *src = frame->data[0] + i; | |
516 | ✗ | FFSWAP(uint8_t, src[0], src[1]); | |
517 | } | ||
518 | ✗ | break; | |
519 | 2 | case DDS_SWIZZLE_A2XY: | |
520 | /* Swap R and G, often used to restore a standard RGTC2. */ | ||
521 | 2 | av_log(avctx, AV_LOG_DEBUG, "Post-processing A2XY swizzle.\n"); | |
522 | 2 | do_swizzle(frame, 0, 1); | |
523 | 2 | break; | |
524 | 2 | case DDS_SWIZZLE_RBXG: | |
525 | /* Swap G and A, then B and new A (G). */ | ||
526 | 2 | av_log(avctx, AV_LOG_DEBUG, "Post-processing RBXG swizzle.\n"); | |
527 | 2 | do_swizzle(frame, 1, 3); | |
528 | 2 | do_swizzle(frame, 2, 3); | |
529 | 2 | break; | |
530 | 2 | case DDS_SWIZZLE_RGXB: | |
531 | /* Swap B and A. */ | ||
532 | 2 | av_log(avctx, AV_LOG_DEBUG, "Post-processing RGXB swizzle.\n"); | |
533 | 2 | do_swizzle(frame, 2, 3); | |
534 | 2 | break; | |
535 | 2 | case DDS_SWIZZLE_RXBG: | |
536 | /* Swap G and A. */ | ||
537 | 2 | av_log(avctx, AV_LOG_DEBUG, "Post-processing RXBG swizzle.\n"); | |
538 | 2 | do_swizzle(frame, 1, 3); | |
539 | 2 | break; | |
540 | 2 | case DDS_SWIZZLE_RXGB: | |
541 | /* Swap R and A (misleading name). */ | ||
542 | 2 | av_log(avctx, AV_LOG_DEBUG, "Post-processing RXGB swizzle.\n"); | |
543 | 2 | do_swizzle(frame, 0, 3); | |
544 | 2 | break; | |
545 | 2 | case DDS_SWIZZLE_XGBR: | |
546 | /* Swap B and A, then R and new A (B). */ | ||
547 | 2 | av_log(avctx, AV_LOG_DEBUG, "Post-processing XGBR swizzle.\n"); | |
548 | 2 | do_swizzle(frame, 2, 3); | |
549 | 2 | do_swizzle(frame, 0, 3); | |
550 | 2 | break; | |
551 | 2 | case DDS_SWIZZLE_XGXR: | |
552 | /* Swap G and A, then R and new A (G), then new R (G) and new G (A). | ||
553 | * This variant does not store any B component. */ | ||
554 | 2 | av_log(avctx, AV_LOG_DEBUG, "Post-processing XGXR swizzle.\n"); | |
555 | 2 | do_swizzle(frame, 1, 3); | |
556 | 2 | do_swizzle(frame, 0, 3); | |
557 | 2 | do_swizzle(frame, 0, 1); | |
558 | 2 | break; | |
559 | 2 | case DDS_SWIZZLE_XRBG: | |
560 | /* Swap G and A, then R and new A (G). */ | ||
561 | 2 | av_log(avctx, AV_LOG_DEBUG, "Post-processing XRBG swizzle.\n"); | |
562 | 2 | do_swizzle(frame, 1, 3); | |
563 | 2 | do_swizzle(frame, 0, 3); | |
564 | 2 | break; | |
565 | } | ||
566 | 28 | } | |
567 | |||
568 | 96 | static int dds_decode(AVCodecContext *avctx, AVFrame *frame, | |
569 | int *got_frame, AVPacket *avpkt) | ||
570 | { | ||
571 | 96 | DDSContext *ctx = avctx->priv_data; | |
572 | 96 | GetByteContext *gbc = &ctx->gbc; | |
573 | int mipmap; | ||
574 | int ret; | ||
575 | int width, height; | ||
576 | |||
577 | 96 | ff_texturedsp_init(&ctx->texdsp); | |
578 | 96 | bytestream2_init(gbc, avpkt->data, avpkt->size); | |
579 | |||
580 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 96 times.
|
96 | if (bytestream2_get_bytes_left(gbc) < 128) { |
581 | ✗ | av_log(avctx, AV_LOG_ERROR, "Frame is too small (%d).\n", | |
582 | bytestream2_get_bytes_left(gbc)); | ||
583 | ✗ | return AVERROR_INVALIDDATA; | |
584 | } | ||
585 | |||
586 |
2/4✓ Branch 1 taken 96 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 96 times.
|
192 | if (bytestream2_get_le32(gbc) != MKTAG('D', 'D', 'S', ' ') || |
587 | 96 | bytestream2_get_le32(gbc) != 124) { // header size | |
588 | ✗ | av_log(avctx, AV_LOG_ERROR, "Invalid DDS header.\n"); | |
589 | ✗ | return AVERROR_INVALIDDATA; | |
590 | } | ||
591 | |||
592 | 96 | bytestream2_skip(gbc, 4); // flags | |
593 | |||
594 | 96 | height = bytestream2_get_le32(gbc); | |
595 | 96 | width = bytestream2_get_le32(gbc); | |
596 | 96 | ret = ff_set_dimensions(avctx, width, height); | |
597 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 96 times.
|
96 | if (ret < 0) { |
598 | ✗ | av_log(avctx, AV_LOG_ERROR, "Invalid image size %dx%d.\n", | |
599 | avctx->width, avctx->height); | ||
600 | ✗ | return ret; | |
601 | } | ||
602 | |||
603 | /* Since codec is based on 4x4 blocks, size is aligned to 4. */ | ||
604 | 96 | avctx->coded_width = FFALIGN(avctx->width, TEXTURE_BLOCK_W); | |
605 | 96 | avctx->coded_height = FFALIGN(avctx->height, TEXTURE_BLOCK_H); | |
606 | |||
607 | 96 | bytestream2_skip(gbc, 4); // pitch | |
608 | 96 | bytestream2_skip(gbc, 4); // depth | |
609 | 96 | mipmap = bytestream2_get_le32(gbc); | |
610 |
2/2✓ Branch 0 taken 56 times.
✓ Branch 1 taken 40 times.
|
96 | if (mipmap != 0) |
611 | 56 | av_log(avctx, AV_LOG_VERBOSE, "Found %d mipmaps (ignored).\n", mipmap); | |
612 | |||
613 | /* Extract pixel format information, considering additional elements | ||
614 | * in reserved1 and reserved2. */ | ||
615 | 96 | ret = parse_pixel_format(avctx); | |
616 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 96 times.
|
96 | if (ret < 0) |
617 | ✗ | return ret; | |
618 | |||
619 | 96 | ret = ff_get_buffer(avctx, frame, 0); | |
620 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 96 times.
|
96 | if (ret < 0) |
621 | ✗ | return ret; | |
622 | |||
623 |
2/2✓ Branch 0 taken 60 times.
✓ Branch 1 taken 36 times.
|
96 | if (ctx->compressed) { |
624 | 60 | int size = (avctx->coded_height / TEXTURE_BLOCK_H) * | |
625 | 60 | (avctx->coded_width / TEXTURE_BLOCK_W) * ctx->dec.tex_ratio; | |
626 | 60 | ctx->dec.slice_count = av_clip(avctx->thread_count, 1, | |
627 | 60 | avctx->coded_height / TEXTURE_BLOCK_H); | |
628 | |||
629 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 60 times.
|
60 | if (bytestream2_get_bytes_left(gbc) < size) { |
630 | ✗ | av_log(avctx, AV_LOG_ERROR, | |
631 | "Compressed Buffer is too small (%d < %d).\n", | ||
632 | bytestream2_get_bytes_left(gbc), size); | ||
633 | ✗ | return AVERROR_INVALIDDATA; | |
634 | } | ||
635 | |||
636 | /* Use the decompress function on the texture, one block per thread. */ | ||
637 | 60 | ctx->dec.tex_data.in = gbc->buffer; | |
638 | 60 | ctx->dec.frame_data.out = frame->data[0]; | |
639 | 60 | ctx->dec.stride = frame->linesize[0]; | |
640 | 60 | avctx->execute2(avctx, ff_texturedsp_decompress_thread, &ctx->dec, NULL, ctx->dec.slice_count); | |
641 |
3/6✓ Branch 0 taken 32 times.
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 32 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
36 | } else if (!ctx->paletted && ctx->bpp == 4 && avctx->pix_fmt == AV_PIX_FMT_PAL8) { |
642 | ✗ | uint8_t *dst = frame->data[0]; | |
643 | int x, y, i; | ||
644 | |||
645 | /* Use the first 64 bytes as palette, then copy the rest. */ | ||
646 | ✗ | bytestream2_get_buffer(gbc, frame->data[1], 16 * 4); | |
647 | ✗ | for (i = 0; i < 16; i++) { | |
648 | ✗ | AV_WN32(frame->data[1] + i*4, | |
649 | (frame->data[1][2+i*4]<<0)+ | ||
650 | (frame->data[1][1+i*4]<<8)+ | ||
651 | (frame->data[1][0+i*4]<<16)+ | ||
652 | ((unsigned)frame->data[1][3+i*4]<<24) | ||
653 | ); | ||
654 | } | ||
655 | ✗ | frame->palette_has_changed = 1; | |
656 | |||
657 | ✗ | if (bytestream2_get_bytes_left(gbc) < frame->height * frame->width / 2) { | |
658 | ✗ | av_log(avctx, AV_LOG_ERROR, "Buffer is too small (%d < %d).\n", | |
659 | ✗ | bytestream2_get_bytes_left(gbc), frame->height * frame->width / 2); | |
660 | ✗ | return AVERROR_INVALIDDATA; | |
661 | } | ||
662 | |||
663 | ✗ | for (y = 0; y < frame->height; y++) { | |
664 | ✗ | for (x = 0; x < frame->width; x += 2) { | |
665 | ✗ | uint8_t val = bytestream2_get_byte(gbc); | |
666 | ✗ | dst[x ] = val & 0xF; | |
667 | ✗ | dst[x + 1] = val >> 4; | |
668 | } | ||
669 | ✗ | dst += frame->linesize[0]; | |
670 | } | ||
671 | } else { | ||
672 | 36 | int linesize = av_image_get_linesize(avctx->pix_fmt, frame->width, 0); | |
673 | |||
674 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 32 times.
|
36 | if (ctx->paletted) { |
675 | int i; | ||
676 | /* Use the first 1024 bytes as palette, then copy the rest. */ | ||
677 | 4 | bytestream2_get_buffer(gbc, frame->data[1], 256 * 4); | |
678 |
2/2✓ Branch 0 taken 1024 times.
✓ Branch 1 taken 4 times.
|
1028 | for (i = 0; i < 256; i++) |
679 | 1024 | AV_WN32(frame->data[1] + i*4, | |
680 | (frame->data[1][2+i*4]<<0)+ | ||
681 | (frame->data[1][1+i*4]<<8)+ | ||
682 | (frame->data[1][0+i*4]<<16)+ | ||
683 | ((unsigned)frame->data[1][3+i*4]<<24) | ||
684 | ); | ||
685 | |||
686 | 4 | frame->palette_has_changed = 1; | |
687 | } | ||
688 | |||
689 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 36 times.
|
36 | if (bytestream2_get_bytes_left(gbc) < frame->height * linesize) { |
690 | ✗ | av_log(avctx, AV_LOG_ERROR, "Buffer is too small (%d < %d).\n", | |
691 | ✗ | bytestream2_get_bytes_left(gbc), frame->height * linesize); | |
692 | ✗ | return AVERROR_INVALIDDATA; | |
693 | } | ||
694 | |||
695 | 36 | av_image_copy_plane(frame->data[0], frame->linesize[0], | |
696 | gbc->buffer, linesize, | ||
697 | linesize, frame->height); | ||
698 | } | ||
699 | |||
700 | /* Run any post processing here if needed. */ | ||
701 |
2/2✓ Branch 0 taken 28 times.
✓ Branch 1 taken 68 times.
|
96 | if (ctx->postproc != DDS_NONE) |
702 | 28 | run_postproc(avctx, frame); | |
703 | |||
704 | /* Frame is ready to be output. */ | ||
705 | 96 | frame->pict_type = AV_PICTURE_TYPE_I; | |
706 | 96 | frame->key_frame = 1; | |
707 | 96 | *got_frame = 1; | |
708 | |||
709 | 96 | return avpkt->size; | |
710 | } | ||
711 | |||
712 | const FFCodec ff_dds_decoder = { | ||
713 | .p.name = "dds", | ||
714 | .p.long_name = NULL_IF_CONFIG_SMALL("DirectDraw Surface image decoder"), | ||
715 | .p.type = AVMEDIA_TYPE_VIDEO, | ||
716 | .p.id = AV_CODEC_ID_DDS, | ||
717 | FF_CODEC_DECODE_CB(dds_decode), | ||
718 | .priv_data_size = sizeof(DDSContext), | ||
719 | .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_SLICE_THREADS, | ||
720 | .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE | ||
721 | }; | ||
722 |