qdrw.c 4.75 KB
Newer Older
1 2 3 4
/*
 * QuickDraw (qdrw) codec
 * Copyright (c) 2004 Konstantin Shishkov
 *
5
 * This file is part of Libav.
6
 *
7
 * Libav is free software; you can redistribute it and/or
8 9
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
10
 * version 2.1 of the License, or (at your option) any later version.
11
 *
12
 * Libav is distributed in the hope that it will be useful,
13 14 15 16 17
 * 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
18
 * License along with Libav; if not, write to the Free Software
19
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20
 */
21

22
/**
23
 * @file
24 25
 * Apple QuickDraw codec.
 */
26

27
#include "libavutil/intreadwrite.h"
28 29 30 31 32 33 34
#include "avcodec.h"

typedef struct QdrawContext{
    AVCodecContext *avctx;
    AVFrame pic;
} QdrawContext;

35
static int decode_frame(AVCodecContext *avctx,
36
                        void *data, int *data_size,
37
                        AVPacket *avpkt)
38
{
39
    const uint8_t *buf = avpkt->data;
40
    const uint8_t *buf_end = avpkt->data + avpkt->size;
41
    int buf_size = avpkt->size;
42 43 44 45 46
    QdrawContext * const a = avctx->priv_data;
    AVFrame * const p= (AVFrame*)&a->pic;
    uint8_t* outdata;
    int colors;
    int i;
47 48
    uint32_t *pal;
    int r, g, b;
49

50 51 52 53 54 55 56 57
    if(p->data[0])
        avctx->release_buffer(avctx, p);

    p->reference= 0;
    if(avctx->get_buffer(avctx, p) < 0){
        av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
        return -1;
    }
58
    p->pict_type= AV_PICTURE_TYPE_I;
59 60 61
    p->key_frame= 1;

    outdata = a->pic.data[0];
62

63 64
    if (buf_end - buf < 0x68 + 4)
        return AVERROR_INVALIDDATA;
65
    buf += 0x68; /* jump to palette */
66
    colors = AV_RB32(buf);
67
    buf += 4;
68

69 70 71 72
    if(colors < 0 || colors > 256) {
        av_log(avctx, AV_LOG_ERROR, "Error color count - %i(0x%X)\n", colors, colors);
        return -1;
    }
73 74
    if (buf_end - buf < (colors + 1) * 8)
        return AVERROR_INVALIDDATA;
75

76
    pal = (uint32_t*)p->data[1];
77
    for (i = 0; i <= colors; i++) {
78
        unsigned int idx;
79
        idx = AV_RB16(buf); /* color index */
80
        buf += 2;
81

82 83 84 85 86
        if (idx > 255) {
            av_log(avctx, AV_LOG_ERROR, "Palette index out of range: %u\n", idx);
            buf += 6;
            continue;
        }
87
        r = *buf++;
88
        buf++;
89
        g = *buf++;
90
        buf++;
91
        b = *buf++;
92
        buf++;
93
        pal[idx] = (r << 16) | (g << 8) | b;
94
    }
95
    p->palette_has_changed = 1;
96

97 98
    if (buf_end - buf < 18)
        return AVERROR_INVALIDDATA;
99 100 101
    buf += 18; /* skip unneeded data */
    for (i = 0; i < avctx->height; i++) {
        int size, left, code, pix;
Michael Niedermayer's avatar
Michael Niedermayer committed
102
        const uint8_t *next;
103 104
        uint8_t *out;
        int tsize = 0;
105

106 107
        /* decode line */
        out = outdata;
108
        size = AV_RB16(buf); /* size of packed line */
109
        buf += 2;
110 111 112
        if (buf_end - buf < size)
            return AVERROR_INVALIDDATA;

113 114 115 116 117 118
        left = size;
        next = buf + size;
        while (left > 0) {
            code = *buf++;
            if (code & 0x80 ) { /* run */
                pix = *buf++;
119
                if ((out + (257 - code)) > (outdata +  a->pic.linesize[0]))
120
                    break;
121 122
                memset(out, pix, 257 - code);
                out += 257 - code;
123 124 125
                tsize += 257 - code;
                left -= 2;
            } else { /* copy */
126
                if ((out + code) > (outdata +  a->pic.linesize[0]))
127
                    break;
128 129
                if (buf_end - buf < code + 1)
                    return AVERROR_INVALIDDATA;
130 131 132
                memcpy(out, buf, code + 1);
                out += code + 1;
                buf += code + 1;
133 134 135 136 137 138 139 140 141 142
                left -= 2 + code;
                tsize += code + 1;
            }
        }
        buf = next;
        outdata += a->pic.linesize[0];
    }

    *data_size = sizeof(AVFrame);
    *(AVFrame*)data = a->pic;
143

144 145 146
    return buf_size;
}

147
static av_cold int decode_init(AVCodecContext *avctx){
148 149
//    QdrawContext * const a = avctx->priv_data;

150
    avctx->pix_fmt= PIX_FMT_PAL8;
151 152 153 154

    return 0;
}

155 156 157 158 159 160 161 162 163 164
static av_cold int decode_end(AVCodecContext *avctx){
    QdrawContext * const a = avctx->priv_data;
    AVFrame *pic = &a->pic;

    if (pic->data[0])
        avctx->release_buffer(avctx, pic);

    return 0;
}

165
AVCodec ff_qdraw_decoder = {
166 167 168 169 170 171 172 173
    .name           = "qdraw",
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = CODEC_ID_QDRAW,
    .priv_data_size = sizeof(QdrawContext),
    .init           = decode_init,
    .close          = decode_end,
    .decode         = decode_frame,
    .capabilities   = CODEC_CAP_DR1,
174
    .long_name = NULL_IF_CONFIG_SMALL("Apple QuickDraw"),
175
};