8e2605f053ac731f73262e0bb9bc39372c1bf837
[libav.git] / libavcodec / txd.c
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 Libav.
8 *
9 * Libav 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 * Libav 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 Libav; 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 "avcodec.h"
27 #include "bytestream.h"
28 #include "internal.h"
29 #include "s3tc.h"
30
31 typedef struct TXDContext {
32 AVFrame picture;
33 } TXDContext;
34
35 static av_cold int txd_init(AVCodecContext *avctx) {
36 TXDContext *s = avctx->priv_data;
37
38 avcodec_get_frame_defaults(&s->picture);
39 avctx->coded_frame = &s->picture;
40
41 return 0;
42 }
43
44 static int txd_decode_frame(AVCodecContext *avctx, void *data, int *data_size,
45 AVPacket *avpkt) {
46 TXDContext * const s = avctx->priv_data;
47 GetByteContext gb;
48 AVFrame *picture = data;
49 AVFrame * const p = &s->picture;
50 unsigned int version, w, h, d3d_format, depth, stride, flags;
51 unsigned int y, v;
52 uint8_t *ptr;
53 uint32_t *pal;
54
55 bytestream2_init(&gb, avpkt->data, avpkt->size);
56 version = bytestream2_get_le32(&gb);
57 bytestream2_skip(&gb, 72);
58 d3d_format = bytestream2_get_le32(&gb);
59 w = bytestream2_get_le16(&gb);
60 h = bytestream2_get_le16(&gb);
61 depth = bytestream2_get_byte(&gb);
62 bytestream2_skip(&gb, 2);
63 flags = bytestream2_get_byte(&gb);
64
65 if (version < 8 || version > 9) {
66 av_log(avctx, AV_LOG_ERROR, "texture data version %i is unsupported\n",
67 version);
68 return -1;
69 }
70
71 if (depth == 8) {
72 avctx->pix_fmt = AV_PIX_FMT_PAL8;
73 } else if (depth == 16 || depth == 32) {
74 avctx->pix_fmt = AV_PIX_FMT_RGB32;
75 } else {
76 av_log(avctx, AV_LOG_ERROR, "depth of %i is unsupported\n", depth);
77 return -1;
78 }
79
80 if (p->data[0])
81 avctx->release_buffer(avctx, p);
82
83 if (av_image_check_size(w, h, 0, avctx))
84 return -1;
85 if (w != avctx->width || h != avctx->height)
86 avcodec_set_dimensions(avctx, w, h);
87 if (ff_get_buffer(avctx, p) < 0) {
88 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
89 return -1;
90 }
91
92 p->pict_type = AV_PICTURE_TYPE_I;
93
94 ptr = p->data[0];
95 stride = p->linesize[0];
96
97 if (depth == 8) {
98 pal = (uint32_t *) p->data[1];
99 for (y = 0; y < 256; y++) {
100 v = bytestream2_get_be32(&gb);
101 pal[y] = (v >> 8) + (v << 24);
102 }
103 bytestream2_skip(&gb, 4);
104 for (y=0; y<h; y++) {
105 bytestream2_get_buffer(&gb, ptr, w);
106 ptr += stride;
107 }
108 } else if (depth == 16) {
109 bytestream2_skip(&gb, 4);
110 switch (d3d_format) {
111 case 0:
112 if (!(flags & 1))
113 goto unsupported;
114 case FF_S3TC_DXT1:
115 ff_decode_dxt1(&gb, ptr, w, h, stride);
116 break;
117 case FF_S3TC_DXT3:
118 ff_decode_dxt3(&gb, ptr, w, h, stride);
119 break;
120 default:
121 goto unsupported;
122 }
123 } else if (depth == 32) {
124 switch (d3d_format) {
125 case 0x15:
126 case 0x16:
127 for (y=0; y<h; y++) {
128 bytestream2_get_buffer(&gb, ptr, w * 4);
129 ptr += stride;
130 }
131 break;
132 default:
133 goto unsupported;
134 }
135 }
136
137 *picture = s->picture;
138 *data_size = sizeof(AVPicture);
139
140 return avpkt->size;
141
142 unsupported:
143 av_log(avctx, AV_LOG_ERROR, "unsupported d3d format (%08x)\n", d3d_format);
144 return -1;
145 }
146
147 static av_cold int txd_end(AVCodecContext *avctx) {
148 TXDContext *s = avctx->priv_data;
149
150 if (s->picture.data[0])
151 avctx->release_buffer(avctx, &s->picture);
152
153 return 0;
154 }
155
156 AVCodec ff_txd_decoder = {
157 .name = "txd",
158 .type = AVMEDIA_TYPE_VIDEO,
159 .id = AV_CODEC_ID_TXD,
160 .priv_data_size = sizeof(TXDContext),
161 .init = txd_init,
162 .close = txd_end,
163 .decode = txd_decode_frame,
164 .capabilities = CODEC_CAP_DR1,
165 .long_name = NULL_IF_CONFIG_SMALL("Renderware TXD (TeXture Dictionary) image"),
166 };