blob: 8b20475d3925dd9cba750775ec6b4c7eb7702f17
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 | static int txd_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, |
35 | AVPacket *avpkt) { |
36 | GetByteContext gb; |
37 | TextureDSPContext dxtc; |
38 | 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 | ff_texturedsp_init(&dxtc); |
47 | |
48 | bytestream2_init(&gb, avpkt->data, avpkt->size); |
49 | version = bytestream2_get_le32(&gb); |
50 | bytestream2_skip(&gb, 72); |
51 | d3d_format = bytestream2_get_le32(&gb); |
52 | w = bytestream2_get_le16(&gb); |
53 | h = bytestream2_get_le16(&gb); |
54 | depth = bytestream2_get_byte(&gb); |
55 | bytestream2_skip(&gb, 2); |
56 | flags = bytestream2_get_byte(&gb); |
57 | |
58 | if (version < 8 || version > 9) { |
59 | avpriv_report_missing_feature(avctx, "Texture data version %u", version); |
60 | return AVERROR_PATCHWELCOME; |
61 | } |
62 | |
63 | if (depth == 8) { |
64 | avctx->pix_fmt = AV_PIX_FMT_PAL8; |
65 | } else if (depth == 16 || depth == 32) { |
66 | avctx->pix_fmt = AV_PIX_FMT_RGBA; |
67 | } else { |
68 | avpriv_report_missing_feature(avctx, "Color depth of %u", depth); |
69 | return AVERROR_PATCHWELCOME; |
70 | } |
71 | |
72 | if ((ret = ff_set_dimensions(avctx, w, h)) < 0) |
73 | return ret; |
74 | |
75 | avctx->coded_width = FFALIGN(w, 4); |
76 | avctx->coded_height = FFALIGN(h, 4); |
77 | |
78 | if ((ret = ff_get_buffer(avctx, p, 0)) < 0) |
79 | return ret; |
80 | |
81 | p->pict_type = AV_PICTURE_TYPE_I; |
82 | |
83 | ptr = p->data[0]; |
84 | stride = p->linesize[0]; |
85 | |
86 | if (depth == 8) { |
87 | pal = (uint32_t *) p->data[1]; |
88 | for (y = 0; y < 256; y++) { |
89 | v = bytestream2_get_be32(&gb); |
90 | pal[y] = (v >> 8) + (v << 24); |
91 | } |
92 | if (bytestream2_get_bytes_left(&gb) < w * h) |
93 | return AVERROR_INVALIDDATA; |
94 | bytestream2_skip(&gb, 4); |
95 | for (y=0; y<h; y++) { |
96 | bytestream2_get_buffer(&gb, ptr, w); |
97 | ptr += stride; |
98 | } |
99 | } else if (depth == 16) { |
100 | bytestream2_skip(&gb, 4); |
101 | switch (d3d_format) { |
102 | case 0: |
103 | if (!(flags & 1)) |
104 | goto unsupported; |
105 | case TXD_DXT1: |
106 | if (bytestream2_get_bytes_left(&gb) < AV_CEIL_RSHIFT(w, 2) * AV_CEIL_RSHIFT(h, 2) * 8) |
107 | return AVERROR_INVALIDDATA; |
108 | for (j = 0; j < avctx->height; j += 4) { |
109 | for (i = 0; i < avctx->width; i += 4) { |
110 | uint8_t *p = ptr + i * 4 + j * stride; |
111 | int step = dxtc.dxt1_block(p, stride, gb.buffer); |
112 | bytestream2_skip(&gb, step); |
113 | } |
114 | } |
115 | break; |
116 | case TXD_DXT3: |
117 | if (bytestream2_get_bytes_left(&gb) < AV_CEIL_RSHIFT(w, 2) * AV_CEIL_RSHIFT(h, 2) * 16) |
118 | return AVERROR_INVALIDDATA; |
119 | for (j = 0; j < avctx->height; j += 4) { |
120 | for (i = 0; i < avctx->width; i += 4) { |
121 | uint8_t *p = ptr + i * 4 + j * stride; |
122 | int step = dxtc.dxt3_block(p, stride, gb.buffer); |
123 | bytestream2_skip(&gb, step); |
124 | } |
125 | } |
126 | break; |
127 | default: |
128 | goto unsupported; |
129 | } |
130 | } else if (depth == 32) { |
131 | switch (d3d_format) { |
132 | case 0x15: |
133 | case 0x16: |
134 | if (bytestream2_get_bytes_left(&gb) < h * w * 4) |
135 | return AVERROR_INVALIDDATA; |
136 | for (y=0; y<h; y++) { |
137 | bytestream2_get_buffer(&gb, ptr, w * 4); |
138 | ptr += stride; |
139 | } |
140 | break; |
141 | default: |
142 | goto unsupported; |
143 | } |
144 | } |
145 | |
146 | *got_frame = 1; |
147 | |
148 | return avpkt->size; |
149 | |
150 | unsupported: |
151 | avpriv_report_missing_feature(avctx, "d3d format (%08x)", d3d_format); |
152 | return AVERROR_PATCHWELCOME; |
153 | } |
154 | |
155 | AVCodec ff_txd_decoder = { |
156 | .name = "txd", |
157 | .long_name = NULL_IF_CONFIG_SMALL("Renderware TXD (TeXture Dictionary) image"), |
158 | .type = AVMEDIA_TYPE_VIDEO, |
159 | .id = AV_CODEC_ID_TXD, |
160 | .decode = txd_decode_frame, |
161 | .capabilities = AV_CODEC_CAP_DR1, |
162 | }; |
163 |