1 |
|
|
/* |
2 |
|
|
* Renderware TeXture Dictionary (.txd) image decoder |
3 |
|
|
* Copyright (c) 2007 Ivo van Poorten |
4 |
|
|
* |
5 |
|
|
* See also: http://wiki.multimedia.cx/index.php?title=TXD |
6 |
|
|
* |
7 |
|
|
* This file is part of FFmpeg. |
8 |
|
|
* |
9 |
|
|
* FFmpeg is free software; you can redistribute it and/or |
10 |
|
|
* modify it under the terms of the GNU Lesser General Public |
11 |
|
|
* License as published by the Free Software Foundation; either |
12 |
|
|
* version 2.1 of the License, or (at your option) any later version. |
13 |
|
|
* |
14 |
|
|
* FFmpeg is distributed in the hope that it will be useful, |
15 |
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
16 |
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
17 |
|
|
* Lesser General Public License for more details. |
18 |
|
|
* |
19 |
|
|
* You should have received a copy of the GNU Lesser General Public |
20 |
|
|
* License along with FFmpeg; if not, write to the Free Software |
21 |
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
22 |
|
|
*/ |
23 |
|
|
|
24 |
|
|
#include "libavutil/intreadwrite.h" |
25 |
|
|
#include "libavutil/imgutils.h" |
26 |
|
|
#include "bytestream.h" |
27 |
|
|
#include "avcodec.h" |
28 |
|
|
#include "internal.h" |
29 |
|
|
#include "texturedsp.h" |
30 |
|
|
|
31 |
|
|
#define TXD_DXT1 0x31545844 |
32 |
|
|
#define TXD_DXT3 0x33545844 |
33 |
|
|
|
34 |
|
16 |
static int txd_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, |
35 |
|
|
AVPacket *avpkt) { |
36 |
|
|
GetByteContext gb; |
37 |
|
|
TextureDSPContext dxtc; |
38 |
|
16 |
AVFrame * const p = data; |
39 |
|
|
unsigned int version, w, h, d3d_format, depth, stride, flags; |
40 |
|
|
unsigned int y, v; |
41 |
|
|
uint8_t *ptr; |
42 |
|
|
uint32_t *pal; |
43 |
|
|
int i, j; |
44 |
|
|
int ret; |
45 |
|
|
|
46 |
✗✓ |
16 |
if (avpkt->size < 88) |
47 |
|
|
return AVERROR_INVALIDDATA; |
48 |
|
|
|
49 |
|
16 |
ff_texturedsp_init(&dxtc); |
50 |
|
|
|
51 |
|
16 |
bytestream2_init(&gb, avpkt->data, avpkt->size); |
52 |
|
16 |
version = bytestream2_get_le32(&gb); |
53 |
|
16 |
bytestream2_skip(&gb, 72); |
54 |
|
16 |
d3d_format = bytestream2_get_le32(&gb); |
55 |
|
16 |
w = bytestream2_get_le16(&gb); |
56 |
|
16 |
h = bytestream2_get_le16(&gb); |
57 |
|
16 |
depth = bytestream2_get_byte(&gb); |
58 |
|
16 |
bytestream2_skip(&gb, 2); |
59 |
|
16 |
flags = bytestream2_get_byte(&gb); |
60 |
|
|
|
61 |
✓✗✗✓
|
16 |
if (version < 8 || version > 9) { |
62 |
|
|
avpriv_report_missing_feature(avctx, "Texture data version %u", version); |
63 |
|
|
return AVERROR_PATCHWELCOME; |
64 |
|
|
} |
65 |
|
|
|
66 |
✓✓ |
16 |
if (depth == 8) { |
67 |
|
2 |
avctx->pix_fmt = AV_PIX_FMT_PAL8; |
68 |
✗✓✗✗
|
14 |
} else if (depth == 16 || depth == 32) { |
69 |
|
14 |
avctx->pix_fmt = AV_PIX_FMT_RGBA; |
70 |
|
|
} else { |
71 |
|
|
avpriv_report_missing_feature(avctx, "Color depth of %u", depth); |
72 |
|
|
return AVERROR_PATCHWELCOME; |
73 |
|
|
} |
74 |
|
|
|
75 |
✗✓ |
16 |
if ((ret = ff_set_dimensions(avctx, w, h)) < 0) |
76 |
|
|
return ret; |
77 |
|
|
|
78 |
|
16 |
avctx->coded_width = FFALIGN(w, 4); |
79 |
|
16 |
avctx->coded_height = FFALIGN(h, 4); |
80 |
|
|
|
81 |
✗✓ |
16 |
if ((ret = ff_get_buffer(avctx, p, 0)) < 0) |
82 |
|
|
return ret; |
83 |
|
|
|
84 |
|
16 |
p->pict_type = AV_PICTURE_TYPE_I; |
85 |
|
|
|
86 |
|
16 |
ptr = p->data[0]; |
87 |
|
16 |
stride = p->linesize[0]; |
88 |
|
|
|
89 |
✓✓ |
16 |
if (depth == 8) { |
90 |
|
2 |
pal = (uint32_t *) p->data[1]; |
91 |
✓✓ |
514 |
for (y = 0; y < 256; y++) { |
92 |
|
512 |
v = bytestream2_get_be32(&gb); |
93 |
|
512 |
pal[y] = (v >> 8) + (v << 24); |
94 |
|
|
} |
95 |
✗✓ |
2 |
if (bytestream2_get_bytes_left(&gb) < w * h) |
96 |
|
|
return AVERROR_INVALIDDATA; |
97 |
|
2 |
bytestream2_skip(&gb, 4); |
98 |
✓✓ |
1026 |
for (y=0; y<h; y++) { |
99 |
|
1024 |
bytestream2_get_buffer(&gb, ptr, w); |
100 |
|
1024 |
ptr += stride; |
101 |
|
|
} |
102 |
✓✗ |
14 |
} else if (depth == 16) { |
103 |
|
14 |
bytestream2_skip(&gb, 4); |
104 |
✓✗✓✗
|
14 |
switch (d3d_format) { |
105 |
|
12 |
case 0: |
106 |
✗✓ |
12 |
if (!(flags & 1)) |
107 |
|
|
goto unsupported; |
108 |
|
|
case TXD_DXT1: |
109 |
✗✓ |
12 |
if (bytestream2_get_bytes_left(&gb) < AV_CEIL_RSHIFT(w, 2) * AV_CEIL_RSHIFT(h, 2) * 8) |
110 |
|
|
return AVERROR_INVALIDDATA; |
111 |
✓✓ |
204 |
for (j = 0; j < avctx->height; j += 4) { |
112 |
✓✓ |
3264 |
for (i = 0; i < avctx->width; i += 4) { |
113 |
|
3072 |
uint8_t *p = ptr + i * 4 + j * stride; |
114 |
|
3072 |
int step = dxtc.dxt1_block(p, stride, gb.buffer); |
115 |
|
3072 |
bytestream2_skip(&gb, step); |
116 |
|
|
} |
117 |
|
|
} |
118 |
|
12 |
break; |
119 |
|
2 |
case TXD_DXT3: |
120 |
✗✓ |
2 |
if (bytestream2_get_bytes_left(&gb) < AV_CEIL_RSHIFT(w, 2) * AV_CEIL_RSHIFT(h, 2) * 16) |
121 |
|
|
return AVERROR_INVALIDDATA; |
122 |
✓✓ |
128 |
for (j = 0; j < avctx->height; j += 4) { |
123 |
✓✓ |
12348 |
for (i = 0; i < avctx->width; i += 4) { |
124 |
|
12222 |
uint8_t *p = ptr + i * 4 + j * stride; |
125 |
|
12222 |
int step = dxtc.dxt3_block(p, stride, gb.buffer); |
126 |
|
12222 |
bytestream2_skip(&gb, step); |
127 |
|
|
} |
128 |
|
|
} |
129 |
|
2 |
break; |
130 |
|
|
default: |
131 |
|
|
goto unsupported; |
132 |
|
|
} |
133 |
|
|
} else if (depth == 32) { |
134 |
|
|
switch (d3d_format) { |
135 |
|
|
case 0x15: |
136 |
|
|
case 0x16: |
137 |
|
|
if (bytestream2_get_bytes_left(&gb) < h * w * 4) |
138 |
|
|
return AVERROR_INVALIDDATA; |
139 |
|
|
for (y=0; y<h; y++) { |
140 |
|
|
bytestream2_get_buffer(&gb, ptr, w * 4); |
141 |
|
|
ptr += stride; |
142 |
|
|
} |
143 |
|
|
break; |
144 |
|
|
default: |
145 |
|
|
goto unsupported; |
146 |
|
|
} |
147 |
|
|
} |
148 |
|
|
|
149 |
|
16 |
*got_frame = 1; |
150 |
|
|
|
151 |
|
16 |
return avpkt->size; |
152 |
|
|
|
153 |
|
|
unsupported: |
154 |
|
|
avpriv_report_missing_feature(avctx, "d3d format (%08x)", d3d_format); |
155 |
|
|
return AVERROR_PATCHWELCOME; |
156 |
|
|
} |
157 |
|
|
|
158 |
|
|
AVCodec ff_txd_decoder = { |
159 |
|
|
.name = "txd", |
160 |
|
|
.long_name = NULL_IF_CONFIG_SMALL("Renderware TXD (TeXture Dictionary) image"), |
161 |
|
|
.type = AVMEDIA_TYPE_VIDEO, |
162 |
|
|
.id = AV_CODEC_ID_TXD, |
163 |
|
|
.decode = txd_decode_frame, |
164 |
|
|
.capabilities = AV_CODEC_CAP_DR1, |
165 |
|
|
}; |