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