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 "s3tc.h" 30 31static int txd_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, 32 AVPacket *avpkt) { 33 GetByteContext gb; 34 AVFrame * const p = data; 35 unsigned int version, w, h, d3d_format, depth, stride, flags; 36 unsigned int y, v; 37 uint8_t *ptr; 38 uint32_t *pal; 39 int ret; 40 41 bytestream2_init(&gb, avpkt->data, avpkt->size); 42 version = bytestream2_get_le32(&gb); 43 bytestream2_skip(&gb, 72); 44 d3d_format = bytestream2_get_le32(&gb); 45 w = bytestream2_get_le16(&gb); 46 h = bytestream2_get_le16(&gb); 47 depth = bytestream2_get_byte(&gb); 48 bytestream2_skip(&gb, 2); 49 flags = bytestream2_get_byte(&gb); 50 51 if (version < 8 || version > 9) { 52 av_log(avctx, AV_LOG_ERROR, "texture data version %i is unsupported\n", 53 version); 54 return AVERROR_PATCHWELCOME; 55 } 56 57 if (depth == 8) { 58 avctx->pix_fmt = AV_PIX_FMT_PAL8; 59 } else if (depth == 16 || depth == 32) { 60 avctx->pix_fmt = AV_PIX_FMT_RGB32; 61 } else { 62 av_log(avctx, AV_LOG_ERROR, "depth of %i is unsupported\n", depth); 63 return AVERROR_PATCHWELCOME; 64 } 65 66 if ((ret = ff_set_dimensions(avctx, w, h)) < 0) 67 return ret; 68 69 if ((ret = ff_get_buffer(avctx, p, 0)) < 0) 70 return ret; 71 72 p->pict_type = AV_PICTURE_TYPE_I; 73 74 ptr = p->data[0]; 75 stride = p->linesize[0]; 76 77 if (depth == 8) { 78 pal = (uint32_t *) p->data[1]; 79 for (y = 0; y < 256; y++) { 80 v = bytestream2_get_be32(&gb); 81 pal[y] = (v >> 8) + (v << 24); 82 } 83 if (bytestream2_get_bytes_left(&gb) < w * h) 84 return AVERROR_INVALIDDATA; 85 bytestream2_skip(&gb, 4); 86 for (y=0; y<h; y++) { 87 bytestream2_get_buffer(&gb, ptr, w); 88 ptr += stride; 89 } 90 } else if (depth == 16) { 91 bytestream2_skip(&gb, 4); 92 switch (d3d_format) { 93 case 0: 94 if (!(flags & 1)) 95 goto unsupported; 96 case FF_S3TC_DXT1: 97 if (bytestream2_get_bytes_left(&gb) < (w/4) * (h/4) * 8) 98 return AVERROR_INVALIDDATA; 99 ff_decode_dxt1(&gb, ptr, w, h, stride); 100 break; 101 case FF_S3TC_DXT3: 102 if (bytestream2_get_bytes_left(&gb) < (w/4) * (h/4) * 16) 103 return AVERROR_INVALIDDATA; 104 ff_decode_dxt3(&gb, ptr, w, h, stride); 105 break; 106 default: 107 goto unsupported; 108 } 109 } else if (depth == 32) { 110 switch (d3d_format) { 111 case 0x15: 112 case 0x16: 113 if (bytestream2_get_bytes_left(&gb) < h * w * 4) 114 return AVERROR_INVALIDDATA; 115 for (y=0; y<h; y++) { 116 bytestream2_get_buffer(&gb, ptr, w * 4); 117 ptr += stride; 118 } 119 break; 120 default: 121 goto unsupported; 122 } 123 } 124 125 *got_frame = 1; 126 127 return avpkt->size; 128 129unsupported: 130 av_log(avctx, AV_LOG_ERROR, "unsupported d3d format (%08x)\n", d3d_format); 131 return AVERROR_PATCHWELCOME; 132} 133 134AVCodec ff_txd_decoder = { 135 .name = "txd", 136 .long_name = NULL_IF_CONFIG_SMALL("Renderware TXD (TeXture Dictionary) image"), 137 .type = AVMEDIA_TYPE_VIDEO, 138 .id = AV_CODEC_ID_TXD, 139 .decode = txd_decode_frame, 140 .capabilities = CODEC_CAP_DR1, 141}; 142