X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=libavcodec%2Featgv.c;h=3894f2beccd8c5bee45d5cf5aaf74d474b37561e;hb=5b145290df2998a9836a93eb925289c6c8b63af0;hp=3bc650665cffe66ad9567094be54fb16f39001f3;hpb=9b8d11a76ae7bca8bbb58abb822138f8b42c776c;p=ffmpeg diff --git a/libavcodec/eatgv.c b/libavcodec/eatgv.c index 3bc650665cf..3894f2beccd 100644 --- a/libavcodec/eatgv.c +++ b/libavcodec/eatgv.c @@ -28,13 +28,14 @@ * http://wiki.multimedia.cx/index.php?title=Electronic_Arts_TGV */ -#include "avcodec.h" -#define BITSTREAM_READER_LE -#include "get_bits.h" -#include "internal.h" #include "libavutil/imgutils.h" #include "libavutil/mem.h" +#define BITSTREAM_READER_LE +#include "avcodec.h" +#include "bitstream.h" +#include "internal.h" + #define EA_PREAMBLE_SIZE 8 #define kVGT_TAG MKTAG('k', 'V', 'G', 'T') @@ -55,7 +56,7 @@ static av_cold int tgv_decode_init(AVCodecContext *avctx) { TgvContext *s = avctx->priv_data; s->avctx = avctx; - avctx->time_base = (AVRational){1, 15}; + avctx->framerate = (AVRational){ 15, 1 }; avctx->pix_fmt = AV_PIX_FMT_PAL8; s->last_frame = av_frame_alloc(); @@ -152,7 +153,7 @@ static int tgv_decode_inter(TgvContext *s, AVFrame *frame, int num_blocks_packed; int vector_bits; int i,j,x,y; - GetBitContext gb; + BitstreamContext bc; int mvbits; const uint8_t *blocks_raw; @@ -165,7 +166,7 @@ static int tgv_decode_inter(TgvContext *s, AVFrame *frame, vector_bits = AV_RL16(&buf[6]); buf += 12; - if (vector_bits > MIN_CACHE_BITS || !vector_bits) { + if (vector_bits > 32 || !vector_bits) { av_log(s->avctx, AV_LOG_ERROR, "Invalid value for motion vector bits: %d\n", vector_bits); return AVERROR_INVALIDDATA; @@ -173,7 +174,9 @@ static int tgv_decode_inter(TgvContext *s, AVFrame *frame, /* allocate codebook buffers as necessary */ if (num_mvs > s->num_mvs) { - s->mv_codebook = av_realloc(s->mv_codebook, num_mvs*2*sizeof(int)); + int err = av_reallocp(&s->mv_codebook, num_mvs * 2 * sizeof(int)); + if (err < 0) + return err; s->num_mvs = num_mvs; } @@ -192,10 +195,10 @@ static int tgv_decode_inter(TgvContext *s, AVFrame *frame, if (buf + (mvbits >> 3) + 16 * num_blocks_raw + 8 * num_blocks_packed > buf_end) return AVERROR_INVALIDDATA; - init_get_bits(&gb, buf, mvbits); + bitstream_init(&bc, buf, mvbits); for (i = 0; i < num_mvs; i++) { - s->mv_codebook[i][0] = get_sbits(&gb, 10); - s->mv_codebook[i][1] = get_sbits(&gb, 10); + s->mv_codebook[i][0] = bitstream_read_signed(&bc, 10); + s->mv_codebook[i][1] = bitstream_read_signed(&bc, 10); } buf += mvbits >> 3; @@ -204,25 +207,25 @@ static int tgv_decode_inter(TgvContext *s, AVFrame *frame, buf += num_blocks_raw * 16; /* read compressed blocks */ - init_get_bits(&gb, buf, (buf_end - buf) << 3); + bitstream_init8(&bc, buf, buf_end - buf); for (i = 0; i < num_blocks_packed; i++) { int tmp[4]; for (j = 0; j < 4; j++) - tmp[j] = get_bits(&gb, 8); + tmp[j] = bitstream_read(&bc, 8); for (j = 0; j < 16; j++) - s->block_codebook[i][15-j] = tmp[get_bits(&gb, 2)]; + s->block_codebook[i][15-j] = tmp[bitstream_read(&bc, 2)]; } - if (get_bits_left(&gb) < vector_bits * + if (bitstream_bits_left(&bc) < vector_bits * (s->avctx->height / 4) * (s->avctx->width / 4)) return AVERROR_INVALIDDATA; /* read vectors and build frame */ for (y = 0; y < s->avctx->height / 4; y++) for (x = 0; x < s->avctx->width / 4; x++) { - unsigned int vector = get_bits(&gb, vector_bits); + unsigned int vector = bitstream_read(&bc, vector_bits); const uint8_t *src; - int src_stride; + ptrdiff_t src_stride; if (vector < num_mvs) { int mx = x * 4 + s->mv_codebook[vector][0]; @@ -355,5 +358,5 @@ AVCodec ff_eatgv_decoder = { .init = tgv_decode_init, .close = tgv_decode_end, .decode = tgv_decode_frame, - .capabilities = CODEC_CAP_DR1, + .capabilities = AV_CODEC_CAP_DR1, };