txd.c 4.83 KB
Newer Older
Ivo van Poorten's avatar
Ivo van Poorten committed
1 2 3 4
/*
 * Renderware TeXture Dictionary (.txd) image decoder
 * Copyright (c) 2007 Ivo van Poorten
 *
5 6
 * See also: http://wiki.multimedia.cx/index.php?title=TXD
 *
7
 * This file is part of Libav.
Ivo van Poorten's avatar
Ivo van Poorten committed
8
 *
9
 * Libav is free software; you can redistribute it and/or
Ivo van Poorten's avatar
Ivo van Poorten committed
10 11 12 13
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
14
 * Libav is distributed in the hope that it will be useful,
Ivo van Poorten's avatar
Ivo van Poorten committed
15 16 17 18 19
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
20
 * License along with Libav; if not, write to the Free Software
Ivo van Poorten's avatar
Ivo van Poorten committed
21 22 23
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

24
#include "libavutil/intreadwrite.h"
25
#include "libavutil/imgutils.h"
Ivo van Poorten's avatar
Ivo van Poorten committed
26
#include "avcodec.h"
27
#include "bytestream.h"
28
#include "internal.h"
29 30 31 32
#include "texturedsp.h"

#define TXD_DXT1 0x31545844
#define TXD_DXT3 0x33545844
Ivo van Poorten's avatar
Ivo van Poorten committed
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;
Ivo van Poorten's avatar
Ivo van Poorten committed
40
    unsigned int y, v;
Michael Niedermayer's avatar
Michael Niedermayer committed
41 42
    uint8_t *ptr;
    uint32_t *pal;
43
    int i, j;
44
    int ret;
Ivo van Poorten's avatar
Ivo van Poorten committed
45

46 47
    ff_texturedsp_init(&dxtc);

48 49 50 51 52 53 54
    bytestream2_init(&gb, avpkt->data, avpkt->size);
    version         = bytestream2_get_le32(&gb);
    bytestream2_skip(&gb, 72);
    d3d_format      = bytestream2_get_le32(&gb);
    w               = bytestream2_get_le16(&gb);
    h               = bytestream2_get_le16(&gb);
    depth           = bytestream2_get_byte(&gb);
55
    bytestream2_skip(&gb, 2);
56
    flags           = bytestream2_get_byte(&gb);
Ivo van Poorten's avatar
Ivo van Poorten committed
57 58 59 60

    if (version < 8 || version > 9) {
        av_log(avctx, AV_LOG_ERROR, "texture data version %i is unsupported\n",
                                                                    version);
61
        return AVERROR_PATCHWELCOME;
Ivo van Poorten's avatar
Ivo van Poorten committed
62 63 64
    }

    if (depth == 8) {
65
        avctx->pix_fmt = AV_PIX_FMT_PAL8;
66
    } else if (depth == 16 || depth == 32) {
67
        avctx->pix_fmt = AV_PIX_FMT_RGBA;
68
    } else {
Ivo van Poorten's avatar
Ivo van Poorten committed
69
        av_log(avctx, AV_LOG_ERROR, "depth of %i is unsupported\n", depth);
70
        return AVERROR_PATCHWELCOME;
Ivo van Poorten's avatar
Ivo van Poorten committed
71 72
    }

73
    if ((ret = ff_set_dimensions(avctx, w, h)) < 0)
74
        return ret;
75

76 77 78
    avctx->coded_width  = FFALIGN(w, 4);
    avctx->coded_height = FFALIGN(h, 4);

79
    if ((ret = ff_get_buffer(avctx, p, 0)) < 0) {
Ivo van Poorten's avatar
Ivo van Poorten committed
80
        av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
81
        return ret;
Ivo van Poorten's avatar
Ivo van Poorten committed
82 83
    }

84
    p->pict_type = AV_PICTURE_TYPE_I;
Ivo van Poorten's avatar
Ivo van Poorten committed
85 86 87 88 89 90

    ptr    = p->data[0];
    stride = p->linesize[0];

    if (depth == 8) {
        pal = (uint32_t *) p->data[1];
91 92 93
        for (y = 0; y < 256; y++) {
            v = bytestream2_get_be32(&gb);
            pal[y] = (v >> 8) + (v << 24);
Ivo van Poorten's avatar
Ivo van Poorten committed
94
        }
95
        bytestream2_skip(&gb, 4);
Ivo van Poorten's avatar
Ivo van Poorten committed
96
        for (y=0; y<h; y++) {
97
            bytestream2_get_buffer(&gb, ptr, w);
Ivo van Poorten's avatar
Ivo van Poorten committed
98 99 100
            ptr += stride;
        }
    } else if (depth == 16) {
101
        bytestream2_skip(&gb, 4);
Ivo van Poorten's avatar
Ivo van Poorten committed
102 103
        switch (d3d_format) {
        case 0:
104 105
            if (!(flags & 1))
                goto unsupported;
106 107 108 109 110 111 112 113
        case TXD_DXT1:
            for (j = 0; j < avctx->height; j += 4) {
                for (i = 0; i < avctx->width; i += 4) {
                    uint8_t *p = ptr + i * 4 + j * stride;
                    int step = dxtc.dxt1_block(p, stride, gb.buffer);
                    bytestream2_skip(&gb, step);
                }
            }
Ivo van Poorten's avatar
Ivo van Poorten committed
114
            break;
115 116 117 118 119 120 121 122
        case TXD_DXT3:
            for (j = 0; j < avctx->height; j += 4) {
                for (i = 0; i < avctx->width; i += 4) {
                    uint8_t *p = ptr + i * 4 + j * stride;
                    int step = dxtc.dxt3_block(p, stride, gb.buffer);
                    bytestream2_skip(&gb, step);
                }
            }
Ivo van Poorten's avatar
Ivo van Poorten committed
123 124 125 126 127 128 129 130 131
            break;
        default:
            goto unsupported;
        }
    } else if (depth == 32) {
        switch (d3d_format) {
        case 0x15:
        case 0x16:
            for (y=0; y<h; y++) {
132
                bytestream2_get_buffer(&gb, ptr, w * 4);
Ivo van Poorten's avatar
Ivo van Poorten committed
133 134 135 136 137 138 139 140
                ptr += stride;
            }
            break;
        default:
            goto unsupported;
        }
    }

141
    *got_frame = 1;
Ivo van Poorten's avatar
Ivo van Poorten committed
142

143
    return avpkt->size;
Ivo van Poorten's avatar
Ivo van Poorten committed
144 145 146

unsupported:
    av_log(avctx, AV_LOG_ERROR, "unsupported d3d format (%08x)\n", d3d_format);
147
    return AVERROR_PATCHWELCOME;
Ivo van Poorten's avatar
Ivo van Poorten committed
148 149
}

150
AVCodec ff_txd_decoder = {
151
    .name           = "txd",
152
    .long_name      = NULL_IF_CONFIG_SMALL("Renderware TXD (TeXture Dictionary) image"),
153
    .type           = AVMEDIA_TYPE_VIDEO,
154
    .id             = AV_CODEC_ID_TXD,
155
    .decode         = txd_decode_frame,
156
    .capabilities   = AV_CODEC_CAP_DR1,
Ivo van Poorten's avatar
Ivo van Poorten committed
157
};