avcodec/anm: Don't unnecessarily use context variables

Reviewed-by: Peter Ross <pross@xvid.org>
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@gmail.com>
This commit is contained in:
Andreas Rheinhardt 2020-05-30 06:14:12 +02:00
parent 45c3502266
commit a886785018

View File

@ -31,13 +31,12 @@
typedef struct AnmContext {
AVFrame *frame;
int palette[AVPALETTE_COUNT];
GetByteContext gb;
int x; ///< x coordinate position
} AnmContext;
static av_cold int decode_init(AVCodecContext *avctx)
{
AnmContext *s = avctx->priv_data;
GetByteContext gb;
int i;
if (avctx->extradata_size < 16 * 8 + 4 * 256)
@ -49,10 +48,10 @@ static av_cold int decode_init(AVCodecContext *avctx)
if (!s->frame)
return AVERROR(ENOMEM);
bytestream2_init(&s->gb, avctx->extradata, avctx->extradata_size);
bytestream2_skipu(&s->gb, 16 * 8);
bytestream2_init(&gb, avctx->extradata, avctx->extradata_size);
bytestream2_skipu(&gb, 16 * 8);
for (i = 0; i < 256; i++)
s->palette[i] = (0xFFU << 24) | bytestream2_get_le32u(&s->gb);
s->palette[i] = (0xFFU << 24) | bytestream2_get_le32u(&gb);
return 0;
}
@ -115,7 +114,8 @@ static int decode_frame(AVCodecContext *avctx,
AnmContext *s = avctx->priv_data;
const int buf_size = avpkt->size;
uint8_t *dst, *dst_end;
int count, ret;
GetByteContext gb;
int count, ret, x = 0;
if (buf_size < 7)
return AVERROR_INVALIDDATA;
@ -125,37 +125,36 @@ static int decode_frame(AVCodecContext *avctx,
dst = s->frame->data[0];
dst_end = s->frame->data[0] + s->frame->linesize[0]*avctx->height;
bytestream2_init(&s->gb, avpkt->data, buf_size);
bytestream2_init(&gb, avpkt->data, buf_size);
if (bytestream2_get_byte(&s->gb) != 0x42) {
if (bytestream2_get_byte(&gb) != 0x42) {
avpriv_request_sample(avctx, "Unknown record type");
return AVERROR_INVALIDDATA;
}
if (bytestream2_get_byte(&s->gb)) {
if (bytestream2_get_byte(&gb)) {
avpriv_request_sample(avctx, "Padding bytes");
return AVERROR_PATCHWELCOME;
}
bytestream2_skip(&s->gb, 2);
bytestream2_skip(&gb, 2);
s->x = 0;
do {
/* if statements are ordered by probability */
#define OP(gb, pixel, count) \
op(&dst, dst_end, (gb), (pixel), (count), &s->x, avctx->width, s->frame->linesize[0])
op(&dst, dst_end, (gb), (pixel), (count), &x, avctx->width, s->frame->linesize[0])
int type = bytestream2_get_byte(&s->gb);
int type = bytestream2_get_byte(&gb);
count = type & 0x7F;
type >>= 7;
if (count) {
if (OP(type ? NULL : &s->gb, -1, count)) break;
if (OP(type ? NULL : &gb, -1, count)) break;
} else if (!type) {
int pixel;
count = bytestream2_get_byte(&s->gb); /* count==0 gives nop */
pixel = bytestream2_get_byte(&s->gb);
count = bytestream2_get_byte(&gb); /* count==0 gives nop */
pixel = bytestream2_get_byte(&gb);
if (OP(NULL, pixel, count)) break;
} else {
int pixel;
type = bytestream2_get_le16(&s->gb);
type = bytestream2_get_le16(&gb);
count = type & 0x3FFF;
type >>= 14;
if (!count) {
@ -167,11 +166,11 @@ static int decode_frame(AVCodecContext *avctx,
}
continue;
}
pixel = type == 3 ? bytestream2_get_byte(&s->gb) : -1;
pixel = type == 3 ? bytestream2_get_byte(&gb) : -1;
if (type == 1) count += 0x4000;
if (OP(type == 2 ? &s->gb : NULL, pixel, count)) break;
if (OP(type == 2 ? &gb : NULL, pixel, count)) break;
}
} while (bytestream2_get_bytes_left(&s->gb) > 0);
} while (bytestream2_get_bytes_left(&gb) > 0);
memcpy(s->frame->data[1], s->palette, AVPALETTE_SIZE);