| @@ -79,6 +79,7 @@ typedef struct SVQ3Context { | |||
| H264Picture *next_pic; | |||
| H264Picture *last_pic; | |||
| GetBitContext gb; | |||
| GetBitContext gb_slice; | |||
| uint8_t *slice_buf; | |||
| int slice_size; | |||
| int halfpel_flag; | |||
| @@ -527,8 +528,8 @@ static inline int svq3_mc_dir(SVQ3Context *s, int size, int mode, | |||
| if (mode == PREDICT_MODE) { | |||
| dx = dy = 0; | |||
| } else { | |||
| dy = svq3_get_se_golomb(&h->gb); | |||
| dx = svq3_get_se_golomb(&h->gb); | |||
| dy = svq3_get_se_golomb(&s->gb_slice); | |||
| dx = svq3_get_se_golomb(&s->gb_slice); | |||
| if (dx == INVALID_VLC || dy == INVALID_VLC) { | |||
| av_log(h->avctx, AV_LOG_ERROR, "invalid MV vlc\n"); | |||
| @@ -736,10 +737,10 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type) | |||
| mb_type = MB_TYPE_16x16; | |||
| } | |||
| } else if (mb_type < 8) { /* INTER */ | |||
| if (s->thirdpel_flag && s->halfpel_flag == !get_bits1(&h->gb)) | |||
| if (s->thirdpel_flag && s->halfpel_flag == !get_bits1(&s->gb_slice)) | |||
| mode = THIRDPEL_MODE; | |||
| else if (s->halfpel_flag && | |||
| s->thirdpel_flag == !get_bits1(&h->gb)) | |||
| s->thirdpel_flag == !get_bits1(&s->gb_slice)) | |||
| mode = HALFPEL_MODE; | |||
| else | |||
| mode = FULLPEL_MODE; | |||
| @@ -841,7 +842,7 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type) | |||
| /* decode prediction codes for luma blocks */ | |||
| for (i = 0; i < 16; i += 2) { | |||
| vlc = svq3_get_ue_golomb(&h->gb); | |||
| vlc = svq3_get_ue_golomb(&s->gb_slice); | |||
| if (vlc >= 25) { | |||
| av_log(h->avctx, AV_LOG_ERROR, | |||
| @@ -919,7 +920,7 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type) | |||
| if (!IS_INTRA16x16(mb_type) && | |||
| (!IS_SKIP(mb_type) || s->pict_type == AV_PICTURE_TYPE_B)) { | |||
| if ((vlc = svq3_get_ue_golomb(&h->gb)) >= 48) { | |||
| if ((vlc = svq3_get_ue_golomb(&s->gb_slice)) >= 48) { | |||
| av_log(h->avctx, AV_LOG_ERROR, "cbp_vlc=%"PRIu32"\n", vlc); | |||
| return -1; | |||
| } | |||
| @@ -929,7 +930,7 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type) | |||
| } | |||
| if (IS_INTRA16x16(mb_type) || | |||
| (s->pict_type != AV_PICTURE_TYPE_I && s->adaptive_quant && cbp)) { | |||
| s->qscale += svq3_get_se_golomb(&h->gb); | |||
| s->qscale += svq3_get_se_golomb(&s->gb_slice); | |||
| if (s->qscale > 31u) { | |||
| av_log(h->avctx, AV_LOG_ERROR, "qscale:%d\n", s->qscale); | |||
| @@ -939,7 +940,7 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type) | |||
| if (IS_INTRA16x16(mb_type)) { | |||
| AV_ZERO128(s->mb_luma_dc[0] + 0); | |||
| AV_ZERO128(s->mb_luma_dc[0] + 8); | |||
| if (svq3_decode_block(&h->gb, s->mb_luma_dc[0], 0, 1)) { | |||
| if (svq3_decode_block(&s->gb_slice, s->mb_luma_dc[0], 0, 1)) { | |||
| av_log(h->avctx, AV_LOG_ERROR, | |||
| "error while decoding intra luma dc\n"); | |||
| return -1; | |||
| @@ -958,7 +959,7 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type) | |||
| : (4 * i + j); | |||
| s->non_zero_count_cache[scan8[k]] = 1; | |||
| if (svq3_decode_block(&h->gb, &s->mb[16 * k], index, type)) { | |||
| if (svq3_decode_block(&s->gb_slice, &s->mb[16 * k], index, type)) { | |||
| av_log(h->avctx, AV_LOG_ERROR, | |||
| "error while decoding block\n"); | |||
| return -1; | |||
| @@ -968,7 +969,7 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type) | |||
| if ((cbp & 0x30)) { | |||
| for (i = 1; i < 3; ++i) | |||
| if (svq3_decode_block(&h->gb, &s->mb[16 * 16 * i], 0, 3)) { | |||
| if (svq3_decode_block(&s->gb_slice, &s->mb[16 * 16 * i], 0, 3)) { | |||
| av_log(h->avctx, AV_LOG_ERROR, | |||
| "error while decoding chroma dc block\n"); | |||
| return -1; | |||
| @@ -980,7 +981,7 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type) | |||
| k = 16 * i + j; | |||
| s->non_zero_count_cache[scan8[k]] = 1; | |||
| if (svq3_decode_block(&h->gb, &s->mb[16 * k], 1, 1)) { | |||
| if (svq3_decode_block(&s->gb_slice, &s->mb[16 * k], 1, 1)) { | |||
| av_log(h->avctx, AV_LOG_ERROR, | |||
| "error while decoding chroma ac block\n"); | |||
| return -1; | |||
| @@ -1036,11 +1037,11 @@ static int svq3_decode_slice_header(AVCodecContext *avctx) | |||
| memcpy(s->slice_buf, s->gb.buffer + s->gb.index / 8, slice_bytes); | |||
| init_get_bits(&h->gb, s->slice_buf, slice_bits); | |||
| init_get_bits(&s->gb_slice, s->slice_buf, slice_bits); | |||
| if (s->watermark_key) { | |||
| uint32_t header = AV_RL32(&h->gb.buffer[1]); | |||
| AV_WL32(&h->gb.buffer[1], header ^ s->watermark_key); | |||
| uint32_t header = AV_RL32(&s->gb_slice.buffer[1]); | |||
| AV_WL32(&s->gb_slice.buffer[1], header ^ s->watermark_key); | |||
| } | |||
| if (length > 0) { | |||
| memcpy(s->slice_buf, &s->slice_buf[slice_length], length - 1); | |||
| @@ -1048,7 +1049,7 @@ static int svq3_decode_slice_header(AVCodecContext *avctx) | |||
| skip_bits_long(&s->gb, slice_bytes * 8); | |||
| } | |||
| if ((slice_id = svq3_get_ue_golomb(&h->gb)) >= 3) { | |||
| if ((slice_id = svq3_get_ue_golomb(&s->gb_slice)) >= 3) { | |||
| av_log(h->avctx, AV_LOG_ERROR, "illegal slice type %u \n", slice_id); | |||
| return -1; | |||
| } | |||
| @@ -1057,26 +1058,26 @@ static int svq3_decode_slice_header(AVCodecContext *avctx) | |||
| if ((header & 0x9F) == 2) { | |||
| i = (s->mb_num < 64) ? 6 : (1 + av_log2(s->mb_num - 1)); | |||
| get_bits(&h->gb, i); | |||
| get_bits(&s->gb_slice, i); | |||
| } else { | |||
| skip_bits1(&h->gb); | |||
| skip_bits1(&s->gb_slice); | |||
| } | |||
| s->slice_num = get_bits(&h->gb, 8); | |||
| s->qscale = get_bits(&h->gb, 5); | |||
| s->adaptive_quant = get_bits1(&h->gb); | |||
| s->slice_num = get_bits(&s->gb_slice, 8); | |||
| s->qscale = get_bits(&s->gb_slice, 5); | |||
| s->adaptive_quant = get_bits1(&s->gb_slice); | |||
| /* unknown fields */ | |||
| skip_bits1(&h->gb); | |||
| skip_bits1(&s->gb_slice); | |||
| if (s->unknown_flag) | |||
| skip_bits1(&h->gb); | |||
| skip_bits1(&s->gb_slice); | |||
| skip_bits1(&h->gb); | |||
| skip_bits(&h->gb, 2); | |||
| skip_bits1(&s->gb_slice); | |||
| skip_bits(&s->gb_slice, 2); | |||
| while (get_bits1(&h->gb)) | |||
| skip_bits(&h->gb, 8); | |||
| while (get_bits1(&s->gb_slice)) | |||
| skip_bits(&s->gb_slice, 8); | |||
| /* reset intra predictors and invalidate motion vector references */ | |||
| if (s->mb_x > 0) { | |||
| @@ -1508,9 +1509,9 @@ static int svq3_decode_frame(AVCodecContext *avctx, void *data, | |||
| unsigned mb_type; | |||
| s->mb_xy = s->mb_x + s->mb_y * s->mb_stride; | |||
| if ((get_bits_left(&h->gb)) <= 7) { | |||
| if (((get_bits_count(&h->gb) & 7) == 0 || | |||
| show_bits(&h->gb, get_bits_left(&h->gb) & 7) == 0)) { | |||
| if ((get_bits_left(&s->gb_slice)) <= 7) { | |||
| if (((get_bits_count(&s->gb_slice) & 7) == 0 || | |||
| show_bits(&s->gb_slice, get_bits_left(&s->gb_slice) & 7) == 0)) { | |||
| if (svq3_decode_slice_header(avctx)) | |||
| return -1; | |||
| @@ -1518,7 +1519,7 @@ static int svq3_decode_frame(AVCodecContext *avctx, void *data, | |||
| /* TODO: support s->mb_skip_run */ | |||
| } | |||
| mb_type = svq3_get_ue_golomb(&h->gb); | |||
| mb_type = svq3_get_ue_golomb(&s->gb_slice); | |||
| if (s->pict_type == AV_PICTURE_TYPE_I) | |||
| mb_type += 8; | |||