yading@10: /* yading@10: * Autodesk RLE Decoder yading@10: * Copyright (C) 2005 the ffmpeg project yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: /** yading@10: * @file yading@10: * Autodesk RLE Video Decoder by Konstantin Shishkov yading@10: */ yading@10: yading@10: #include yading@10: #include yading@10: #include yading@10: yading@10: #include "avcodec.h" yading@10: #include "internal.h" yading@10: #include "msrledec.h" yading@10: yading@10: typedef struct AascContext { yading@10: AVCodecContext *avctx; yading@10: GetByteContext gb; yading@10: AVFrame *frame; yading@10: yading@10: uint32_t palette[AVPALETTE_COUNT]; yading@10: int palette_size; yading@10: } AascContext; yading@10: yading@10: static av_cold int aasc_decode_init(AVCodecContext *avctx) yading@10: { yading@10: AascContext *s = avctx->priv_data; yading@10: uint8_t *ptr; yading@10: int i; yading@10: yading@10: s->avctx = avctx; yading@10: switch (avctx->bits_per_coded_sample) { yading@10: case 8: yading@10: avctx->pix_fmt = AV_PIX_FMT_PAL8; yading@10: yading@10: ptr = avctx->extradata; yading@10: s->palette_size = FFMIN(avctx->extradata_size, AVPALETTE_SIZE); yading@10: for (i = 0; i < s->palette_size / 4; i++) { yading@10: s->palette[i] = 0xFFU << 24 | AV_RL32(ptr); yading@10: ptr += 4; yading@10: } yading@10: break; yading@10: case 16: yading@10: avctx->pix_fmt = AV_PIX_FMT_RGB555LE; yading@10: break; yading@10: case 24: yading@10: avctx->pix_fmt = AV_PIX_FMT_BGR24; yading@10: break; yading@10: default: yading@10: av_log(avctx, AV_LOG_ERROR, "Unsupported bit depth: %d\n", avctx->bits_per_coded_sample); yading@10: return -1; yading@10: } yading@10: yading@10: s->frame = av_frame_alloc(); yading@10: if (!s->frame) yading@10: return AVERROR(ENOMEM); yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static int aasc_decode_frame(AVCodecContext *avctx, yading@10: void *data, int *got_frame, yading@10: AVPacket *avpkt) yading@10: { yading@10: const uint8_t *buf = avpkt->data; yading@10: int buf_size = avpkt->size; yading@10: AascContext *s = avctx->priv_data; yading@10: int compr, i, stride, psize, ret; yading@10: yading@10: if (buf_size < 4) { yading@10: av_log(avctx, AV_LOG_ERROR, "frame too short\n"); yading@10: return AVERROR_INVALIDDATA; yading@10: } yading@10: yading@10: if ((ret = ff_reget_buffer(avctx, s->frame)) < 0) yading@10: return ret; yading@10: yading@10: compr = AV_RL32(buf); yading@10: buf += 4; yading@10: buf_size -= 4; yading@10: psize = avctx->bits_per_coded_sample / 8; yading@10: switch (avctx->codec_tag) { yading@10: case MKTAG('A', 'A', 'S', '4'): yading@10: bytestream2_init(&s->gb, buf - 4, buf_size + 4); yading@10: ff_msrle_decode(avctx, (AVPicture*)s->frame, 8, &s->gb); yading@10: break; yading@10: case MKTAG('A', 'A', 'S', 'C'): yading@10: switch (compr) { yading@10: case 0: yading@10: stride = (avctx->width * psize + psize) & ~psize; yading@10: for (i = avctx->height - 1; i >= 0; i--) { yading@10: if (avctx->width * psize > buf_size) { yading@10: av_log(avctx, AV_LOG_ERROR, "Next line is beyond buffer bounds\n"); yading@10: break; yading@10: } yading@10: memcpy(s->frame->data[0] + i * s->frame->linesize[0], buf, avctx->width * psize); yading@10: buf += stride; yading@10: buf_size -= stride; yading@10: } yading@10: break; yading@10: case 1: yading@10: bytestream2_init(&s->gb, buf, buf_size); yading@10: ff_msrle_decode(avctx, (AVPicture*)s->frame, 8, &s->gb); yading@10: break; yading@10: default: yading@10: av_log(avctx, AV_LOG_ERROR, "Unknown compression type %d\n", compr); yading@10: return AVERROR_INVALIDDATA; yading@10: } yading@10: break; yading@10: default: yading@10: av_log(avctx, AV_LOG_ERROR, "Unknown FourCC: %X\n", avctx->codec_tag); yading@10: return -1; yading@10: } yading@10: yading@10: if (avctx->pix_fmt == AV_PIX_FMT_PAL8) yading@10: memcpy(s->frame->data[1], s->palette, s->palette_size); yading@10: yading@10: *got_frame = 1; yading@10: if ((ret = av_frame_ref(data, s->frame)) < 0) yading@10: return ret; yading@10: yading@10: /* report that the buffer was completely consumed */ yading@10: return buf_size; yading@10: } yading@10: yading@10: static av_cold int aasc_decode_end(AVCodecContext *avctx) yading@10: { yading@10: AascContext *s = avctx->priv_data; yading@10: yading@10: av_frame_free(&s->frame); yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: AVCodec ff_aasc_decoder = { yading@10: .name = "aasc", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: .id = AV_CODEC_ID_AASC, yading@10: .priv_data_size = sizeof(AascContext), yading@10: .init = aasc_decode_init, yading@10: .close = aasc_decode_end, yading@10: .decode = aasc_decode_frame, yading@10: .capabilities = CODEC_CAP_DR1, yading@10: .long_name = NULL_IF_CONFIG_SMALL("Autodesk RLE"), yading@10: };