sgirledec.c 4.28 KB
Newer Older
Peter Ross's avatar
Peter Ross committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
/*
 * SGI RLE 8-bit decoder
 * Copyright (c) 2012 Peter Ross
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * 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.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * 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
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

/**
 * @file
 * SGI RLE 8-bit decoder
 */

#include "libavutil/intreadwrite.h"
#include "avcodec.h"
#include "bytestream.h"
30 31 32 33 34
#include "internal.h"

typedef struct SGIRLEContext {
    AVFrame *frame;
} SGIRLEContext;
Peter Ross's avatar
Peter Ross committed
35 36 37

static av_cold int sgirle_decode_init(AVCodecContext *avctx)
{
38
    SGIRLEContext *s = avctx->priv_data;
Peter Ross's avatar
Peter Ross committed
39
    avctx->pix_fmt = AV_PIX_FMT_BGR8;
40 41
    s->frame = av_frame_alloc();
    if (!s->frame)
Peter Ross's avatar
Peter Ross committed
42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
        return AVERROR(ENOMEM);
    return 0;
}

/**
 * Convert SGI RGB332 pixel into PIX_FMT_BGR8
 * SGI RGB332 is packed RGB 3:3:2, 8bpp, (msb)3R 2B 3G(lsb)
 */
#define RGB332_TO_BGR8(x) (((x << 3) & 0xC0) | ((x << 3) & 0x38) | ((x >> 5) & 7))

static av_always_inline void memcpy_rgb332_to_bgr8(uint8_t *dst, const uint8_t *src, int size)
{
    int i;
    for (i = 0; i < size; i++)
        dst[i] = RGB332_TO_BGR8(src[i]);
}

/**
 * @param[out] dst Destination buffer
 * @param[in] src Source buffer
 * @param src_size Source buffer size (bytes)
 * @param width Width of destination buffer (pixels)
 * @param height Height of destination buffer (pixels)
 * @param linesize Line size of destination buffer (bytes)
 * @return <0 on error
 */
static int decode_sgirle8(AVCodecContext *avctx, uint8_t *dst, const uint8_t *src, int src_size, int width, int height, int linesize)
{
    const uint8_t *src_end = src + src_size;
    int x = 0, y = 0;

#define INC_XY(n) \
    x += n; \
    if (x >= width) { \
        y++; \
        if (y >= height) \
            return 0; \
        x = 0; \
    }

    while (src_end - src >= 2) {
        uint8_t v = *src++;
        if (v > 0 && v < 0xC0) {
            do {
                int length = FFMIN(v, width - x);
                memset(dst + y*linesize + x, RGB332_TO_BGR8(*src), length);
                INC_XY(length);
                v   -= length;
            } while (v > 0);
            src++;
        } else if (v >= 0xC1) {
            v -= 0xC0;
            do {
                int length = FFMIN3(v, width - x, src_end - src);
                if (src_end - src < length)
                    break;
                memcpy_rgb332_to_bgr8(dst + y*linesize + x, src, length);
                INC_XY(length);
                src += length;
                v   -= length;
            } while (v > 0);
        } else {
            av_log_ask_for_sample(avctx, "unknown opcode\n");
            return AVERROR_PATCHWELCOME;
        }
    }
    return 0;
}

static int sgirle_decode_frame(AVCodecContext *avctx,
                            void *data, int *got_frame,
                            AVPacket *avpkt)
{
115
    SGIRLEContext *s = avctx->priv_data;
Peter Ross's avatar
Peter Ross committed
116 117
    int ret;

118
    if ((ret = ff_reget_buffer(avctx, s->frame)) < 0)
Peter Ross's avatar
Peter Ross committed
119 120
        return ret;

121
    ret = decode_sgirle8(avctx, s->frame->data[0], avpkt->data, avpkt->size, avctx->width, avctx->height, s->frame->linesize[0]);
Peter Ross's avatar
Peter Ross committed
122 123 124 125
    if (ret < 0)
        return ret;

    *got_frame      = 1;
126 127
    if ((ret = av_frame_ref(data, s->frame)) < 0)
        return ret;
Peter Ross's avatar
Peter Ross committed
128 129 130 131 132 133

    return avpkt->size;
}

static av_cold int sgirle_decode_end(AVCodecContext *avctx)
{
134 135 136 137
    SGIRLEContext *s = avctx->priv_data;

    av_frame_free(&s->frame);

Peter Ross's avatar
Peter Ross committed
138 139 140 141 142 143 144
    return 0;
}

AVCodec ff_sgirle_decoder = {
    .name           = "sgirle",
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = AV_CODEC_ID_SGIRLE,
145
    .priv_data_size = sizeof(SGIRLEContext),
Peter Ross's avatar
Peter Ross committed
146 147 148 149 150 151
    .init           = sgirle_decode_init,
    .close          = sgirle_decode_end,
    .decode         = sgirle_decode_frame,
    .capabilities   = CODEC_CAP_DR1,
    .long_name      = NULL_IF_CONFIG_SMALL("SGI RLE 8-bit"),
};