qdrw.c 4.78 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/common.h"
28
#include "libavutil/intreadwrite.h"
29 30 31 32 33 34 35
#include "avcodec.h"

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

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

51 52 53 54 55 56 57 58
    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;
    }
59
    p->pict_type= AV_PICTURE_TYPE_I;
60 61 62
    p->key_frame= 1;

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

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

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

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

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

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

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

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

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

145 146 147
    return buf_size;
}

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

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

    return 0;
}

156 157 158 159 160 161 162 163 164 165
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;
}

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