@@ -45,7 +45,6 @@ typedef struct TgqContext {
int width, height;
int qtable[64];
DECLARE_ALIGNED(16, int16_t, block)[6][64];
- GetByteContext gb;
} TgqContext;
static av_cold int tgq_decode_init(AVCodecContext *avctx)
@@ -147,34 +146,35 @@ static void tgq_idct_put_mb_dconly(TgqContext *s, AVFrame *frame,
}
}
-static int tgq_decode_mb(TgqContext *s, AVFrame *frame, int mb_y, int mb_x)
+static int tgq_decode_mb(TgqContext *s, GetByteContext *gbyte,
+ AVFrame *frame, int mb_y, int mb_x)
{
int mode;
int i;
int8_t dc[6];
- mode = bytestream2_get_byte(&s->gb);
+ mode = bytestream2_get_byte(gbyte);
if (mode > 12) {
GetBitContext gb;
- int ret = init_get_bits8(&gb, s->gb.buffer, FFMIN(bytestream2_get_bytes_left(&s->gb), mode));
+ int ret = init_get_bits8(&gb, gbyte->buffer, FFMIN(bytestream2_get_bytes_left(gbyte), mode));
if (ret < 0)
return ret;
for (i = 0; i < 6; i++)
tgq_decode_block(s, s->block[i], &gb);
tgq_idct_put_mb(s, s->block, frame, mb_x, mb_y);
- bytestream2_skip(&s->gb, mode);
+ bytestream2_skip(gbyte, mode);
} else {
if (mode == 3) {
- memset(dc, bytestream2_get_byte(&s->gb), 4);
- dc[4] = bytestream2_get_byte(&s->gb);
- dc[5] = bytestream2_get_byte(&s->gb);
+ memset(dc, bytestream2_get_byte(gbyte), 4);
+ dc[4] = bytestream2_get_byte(gbyte);
+ dc[5] = bytestream2_get_byte(gbyte);
} else if (mode == 6) {
- bytestream2_get_buffer(&s->gb, dc, 6);
+ bytestream2_get_buffer(gbyte, dc, 6);
} else if (mode == 12) {
for (i = 0; i < 6; i++) {
- dc[i] = bytestream2_get_byte(&s->gb);
- bytestream2_skip(&s->gb, 1);
+ dc[i] = bytestream2_get_byte(gbyte);
+ bytestream2_skip(gbyte, 1);
}
} else {
av_log(s->avctx, AV_LOG_ERROR, "unsupported mb mode %i\n", mode);
@@ -202,6 +202,7 @@ static int tgq_decode_frame(AVCodecContext *avctx, AVFrame *frame,
const uint8_t *buf = avpkt->data;
int buf_size = avpkt->size;
TgqContext *s = avctx->priv_data;
+ GetByteContext gbyte;
int x, y, ret;
int big_endian;
@@ -210,21 +211,21 @@ static int tgq_decode_frame(AVCodecContext *avctx, AVFrame *frame,
return AVERROR_INVALIDDATA;
}
big_endian = AV_RL32(&buf[4]) > 0x000FFFFF;
- bytestream2_init(&s->gb, buf + 8, buf_size - 8);
+ bytestream2_init(&gbyte, buf + 8, buf_size - 8);
if (big_endian) {
- s->width = bytestream2_get_be16u(&s->gb);
- s->height = bytestream2_get_be16u(&s->gb);
+ s->width = bytestream2_get_be16u(&gbyte);
+ s->height = bytestream2_get_be16u(&gbyte);
} else {
- s->width = bytestream2_get_le16u(&s->gb);
- s->height = bytestream2_get_le16u(&s->gb);
+ s->width = bytestream2_get_le16u(&gbyte);
+ s->height = bytestream2_get_le16u(&gbyte);
}
ret = ff_set_dimensions(s->avctx, s->width, s->height);
if (ret < 0)
return ret;
- tgq_calculate_qtable(s, bytestream2_get_byteu(&s->gb));
- bytestream2_skip(&s->gb, 3);
+ tgq_calculate_qtable(s, bytestream2_get_byteu(&gbyte));
+ bytestream2_skipu(&gbyte, 3);
if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
return ret;
@@ -233,7 +234,7 @@ static int tgq_decode_frame(AVCodecContext *avctx, AVFrame *frame,
for (y = 0; y < FFALIGN(avctx->height, 16) >> 4; y++)
for (x = 0; x < FFALIGN(avctx->width, 16) >> 4; x++)
- if (tgq_decode_mb(s, frame, y, x) < 0)
+ if (tgq_decode_mb(s, &gbyte, frame, y, x) < 0)
return AVERROR_INVALIDDATA;
*got_frame = 1;
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com> --- libavcodec/eatgq.c | 39 ++++++++++++++++++++------------------- 1 file changed, 20 insertions(+), 19 deletions(-)