|
|
|
@ -91,6 +91,8 @@ typedef struct SVQ3Context { |
|
|
|
|
int v_edge_pos; |
|
|
|
|
int last_frame_output; |
|
|
|
|
|
|
|
|
|
enum AVPictureType pict_type; |
|
|
|
|
|
|
|
|
|
int mb_x, mb_y; |
|
|
|
|
int mb_xy; |
|
|
|
|
int mb_width, mb_height; |
|
|
|
@ -705,12 +707,12 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type) |
|
|
|
|
s->topright_samples_available = 0xFFFF; |
|
|
|
|
|
|
|
|
|
if (mb_type == 0) { /* SKIP */ |
|
|
|
|
if (h->pict_type == AV_PICTURE_TYPE_P || |
|
|
|
|
if (s->pict_type == AV_PICTURE_TYPE_P || |
|
|
|
|
s->next_pic->mb_type[mb_xy] == -1) { |
|
|
|
|
svq3_mc_dir_part(s, 16 * s->mb_x, 16 * s->mb_y, 16, 16, |
|
|
|
|
0, 0, 0, 0, 0, 0); |
|
|
|
|
|
|
|
|
|
if (h->pict_type == AV_PICTURE_TYPE_B) |
|
|
|
|
if (s->pict_type == AV_PICTURE_TYPE_B) |
|
|
|
|
svq3_mc_dir_part(s, 16 * s->mb_x, 16 * s->mb_y, 16, 16, |
|
|
|
|
0, 0, 0, 0, 1, 1); |
|
|
|
|
|
|
|
|
@ -777,12 +779,12 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type) |
|
|
|
|
memset(&s->ref_cache[m][scan8[0] - 1 * 8 - 1], |
|
|
|
|
PART_NOT_AVAILABLE, 8); |
|
|
|
|
|
|
|
|
|
if (h->pict_type != AV_PICTURE_TYPE_B) |
|
|
|
|
if (s->pict_type != AV_PICTURE_TYPE_B) |
|
|
|
|
break; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/* decode motion vector(s) and form prediction(s) */ |
|
|
|
|
if (h->pict_type == AV_PICTURE_TYPE_P) { |
|
|
|
|
if (s->pict_type == AV_PICTURE_TYPE_P) { |
|
|
|
|
if (svq3_mc_dir(s, mb_type - 1, mode, 0, 0) < 0) |
|
|
|
|
return -1; |
|
|
|
|
} else { /* AV_PICTURE_TYPE_B */ |
|
|
|
@ -889,11 +891,11 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type) |
|
|
|
|
mb_type = MB_TYPE_INTRA16x16; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (!IS_INTER(mb_type) && h->pict_type != AV_PICTURE_TYPE_I) { |
|
|
|
|
if (!IS_INTER(mb_type) && s->pict_type != AV_PICTURE_TYPE_I) { |
|
|
|
|
for (i = 0; i < 4; i++) |
|
|
|
|
memset(h->cur_pic.motion_val[0][b_xy + i * s->b_stride], |
|
|
|
|
0, 4 * 2 * sizeof(int16_t)); |
|
|
|
|
if (h->pict_type == AV_PICTURE_TYPE_B) { |
|
|
|
|
if (s->pict_type == AV_PICTURE_TYPE_B) { |
|
|
|
|
for (i = 0; i < 4; i++) |
|
|
|
|
memset(h->cur_pic.motion_val[1][b_xy + i * s->b_stride], |
|
|
|
|
0, 4 * 2 * sizeof(int16_t)); |
|
|
|
@ -902,12 +904,12 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type) |
|
|
|
|
if (!IS_INTRA4x4(mb_type)) { |
|
|
|
|
memset(s->intra4x4_pred_mode + s->mb2br_xy[mb_xy], DC_PRED, 8); |
|
|
|
|
} |
|
|
|
|
if (!IS_SKIP(mb_type) || h->pict_type == AV_PICTURE_TYPE_B) { |
|
|
|
|
if (!IS_SKIP(mb_type) || s->pict_type == AV_PICTURE_TYPE_B) { |
|
|
|
|
memset(sl->non_zero_count_cache + 8, 0, 14 * 8 * sizeof(uint8_t)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (!IS_INTRA16x16(mb_type) && |
|
|
|
|
(!IS_SKIP(mb_type) || h->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) { |
|
|
|
|
av_log(h->avctx, AV_LOG_ERROR, "cbp_vlc=%"PRIu32"\n", vlc); |
|
|
|
|
return -1; |
|
|
|
@ -917,7 +919,7 @@ static int svq3_decode_mb(SVQ3Context *s, unsigned int mb_type) |
|
|
|
|
: ff_h264_golomb_to_inter_cbp[vlc]; |
|
|
|
|
} |
|
|
|
|
if (IS_INTRA16x16(mb_type) || |
|
|
|
|
(h->pict_type != AV_PICTURE_TYPE_I && s->adaptive_quant && cbp)) { |
|
|
|
|
(s->pict_type != AV_PICTURE_TYPE_I && s->adaptive_quant && cbp)) { |
|
|
|
|
sl->qscale += svq3_get_se_golomb(&h->gb); |
|
|
|
|
|
|
|
|
|
if (sl->qscale > 31u) { |
|
|
|
@ -1043,7 +1045,7 @@ static int svq3_decode_slice_header(AVCodecContext *avctx) |
|
|
|
|
return -1; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
sl->slice_type = ff_h264_golomb_to_pict_type[slice_id]; |
|
|
|
|
s->pict_type = ff_h264_golomb_to_pict_type[slice_id]; |
|
|
|
|
|
|
|
|
|
if ((header & 0x9F) == 2) { |
|
|
|
|
i = (s->mb_num < 64) ? 6 : (1 + av_log2(s->mb_num - 1)); |
|
|
|
@ -1354,7 +1356,7 @@ static int get_buffer(AVCodecContext *avctx, H264Picture *pic) |
|
|
|
|
pic->ref_index[i] = pic->ref_index_buf[i]->data; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
pic->reference = !(h->pict_type == AV_PICTURE_TYPE_B); |
|
|
|
|
pic->reference = !(s->pict_type == AV_PICTURE_TYPE_B); |
|
|
|
|
|
|
|
|
|
ret = ff_get_buffer(avctx, pic->f, |
|
|
|
|
pic->reference ? AV_GET_BUFFER_FLAG_REF : 0); |
|
|
|
@ -1407,16 +1409,14 @@ static int svq3_decode_frame(AVCodecContext *avctx, void *data, |
|
|
|
|
if (svq3_decode_slice_header(avctx)) |
|
|
|
|
return -1; |
|
|
|
|
|
|
|
|
|
h->pict_type = sl->slice_type; |
|
|
|
|
|
|
|
|
|
if (h->pict_type != AV_PICTURE_TYPE_B) |
|
|
|
|
if (s->pict_type != AV_PICTURE_TYPE_B) |
|
|
|
|
FFSWAP(H264Picture*, s->next_pic, s->last_pic); |
|
|
|
|
|
|
|
|
|
av_frame_unref(s->cur_pic->f); |
|
|
|
|
|
|
|
|
|
/* for skipping the frame */ |
|
|
|
|
s->cur_pic->f->pict_type = h->pict_type; |
|
|
|
|
s->cur_pic->f->key_frame = (h->pict_type == AV_PICTURE_TYPE_I); |
|
|
|
|
s->cur_pic->f->pict_type = s->pict_type; |
|
|
|
|
s->cur_pic->f->key_frame = (s->pict_type == AV_PICTURE_TYPE_I); |
|
|
|
|
|
|
|
|
|
ret = get_buffer(avctx, s->cur_pic); |
|
|
|
|
if (ret < 0) |
|
|
|
@ -1436,7 +1436,7 @@ static int svq3_decode_frame(AVCodecContext *avctx, void *data, |
|
|
|
|
h->block_offset[48 + 32 + i] = (4 * ((scan8[i] - scan8[0]) & 7)) + 8 * sl->uvlinesize * ((scan8[i] - scan8[0]) >> 3); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (h->pict_type != AV_PICTURE_TYPE_I) { |
|
|
|
|
if (s->pict_type != AV_PICTURE_TYPE_I) { |
|
|
|
|
if (!s->last_pic->f->data[0]) { |
|
|
|
|
av_log(avctx, AV_LOG_ERROR, "Missing reference frame.\n"); |
|
|
|
|
ret = get_buffer(avctx, s->last_pic); |
|
|
|
@ -1449,7 +1449,7 @@ static int svq3_decode_frame(AVCodecContext *avctx, void *data, |
|
|
|
|
s->last_pic->f->linesize[2]); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (h->pict_type == AV_PICTURE_TYPE_B && !s->next_pic->f->data[0]) { |
|
|
|
|
if (s->pict_type == AV_PICTURE_TYPE_B && !s->next_pic->f->data[0]) { |
|
|
|
|
av_log(avctx, AV_LOG_ERROR, "Missing reference frame.\n"); |
|
|
|
|
ret = get_buffer(avctx, s->next_pic); |
|
|
|
|
if (ret < 0) |
|
|
|
@ -1465,23 +1465,23 @@ static int svq3_decode_frame(AVCodecContext *avctx, void *data, |
|
|
|
|
if (avctx->debug & FF_DEBUG_PICT_INFO) |
|
|
|
|
av_log(h->avctx, AV_LOG_DEBUG, |
|
|
|
|
"%c hpel:%d, tpel:%d aqp:%d qp:%d, slice_num:%02X\n", |
|
|
|
|
av_get_picture_type_char(h->pict_type), |
|
|
|
|
av_get_picture_type_char(s->pict_type), |
|
|
|
|
s->halfpel_flag, s->thirdpel_flag, |
|
|
|
|
s->adaptive_quant, h->slice_ctx[0].qscale, sl->slice_num); |
|
|
|
|
|
|
|
|
|
if (avctx->skip_frame >= AVDISCARD_NONREF && h->pict_type == AV_PICTURE_TYPE_B || |
|
|
|
|
avctx->skip_frame >= AVDISCARD_NONKEY && h->pict_type != AV_PICTURE_TYPE_I || |
|
|
|
|
if (avctx->skip_frame >= AVDISCARD_NONREF && s->pict_type == AV_PICTURE_TYPE_B || |
|
|
|
|
avctx->skip_frame >= AVDISCARD_NONKEY && s->pict_type != AV_PICTURE_TYPE_I || |
|
|
|
|
avctx->skip_frame >= AVDISCARD_ALL) |
|
|
|
|
return 0; |
|
|
|
|
|
|
|
|
|
if (s->next_p_frame_damaged) { |
|
|
|
|
if (h->pict_type == AV_PICTURE_TYPE_B) |
|
|
|
|
if (s->pict_type == AV_PICTURE_TYPE_B) |
|
|
|
|
return 0; |
|
|
|
|
else |
|
|
|
|
s->next_p_frame_damaged = 0; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (h->pict_type == AV_PICTURE_TYPE_B) { |
|
|
|
|
if (s->pict_type == AV_PICTURE_TYPE_B) { |
|
|
|
|
h->frame_num_offset = sl->slice_num - h->prev_frame_num; |
|
|
|
|
|
|
|
|
|
if (h->frame_num_offset < 0) |
|
|
|
@ -1528,9 +1528,9 @@ static int svq3_decode_frame(AVCodecContext *avctx, void *data, |
|
|
|
|
|
|
|
|
|
mb_type = svq3_get_ue_golomb(&h->gb); |
|
|
|
|
|
|
|
|
|
if (h->pict_type == AV_PICTURE_TYPE_I) |
|
|
|
|
if (s->pict_type == AV_PICTURE_TYPE_I) |
|
|
|
|
mb_type += 8; |
|
|
|
|
else if (h->pict_type == AV_PICTURE_TYPE_B && mb_type >= 4) |
|
|
|
|
else if (s->pict_type == AV_PICTURE_TYPE_B && mb_type >= 4) |
|
|
|
|
mb_type += 4; |
|
|
|
|
if (mb_type > 33 || svq3_decode_mb(s, mb_type)) { |
|
|
|
|
av_log(h->avctx, AV_LOG_ERROR, |
|
|
|
@ -1541,9 +1541,9 @@ static int svq3_decode_frame(AVCodecContext *avctx, void *data, |
|
|
|
|
if (mb_type != 0) |
|
|
|
|
hl_decode_mb(s, h, &h->slice_ctx[0]); |
|
|
|
|
|
|
|
|
|
if (h->pict_type != AV_PICTURE_TYPE_B && !h->low_delay) |
|
|
|
|
if (s->pict_type != AV_PICTURE_TYPE_B && !h->low_delay) |
|
|
|
|
h->cur_pic.mb_type[s->mb_x + s->mb_y * s->mb_stride] = |
|
|
|
|
(h->pict_type == AV_PICTURE_TYPE_P && mb_type < 8) ? (mb_type - 1) : -1; |
|
|
|
|
(s->pict_type == AV_PICTURE_TYPE_P && mb_type < 8) ? (mb_type - 1) : -1; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
ff_draw_horiz_band(avctx, s->cur_pic->f, |
|
|
|
@ -1552,7 +1552,7 @@ static int svq3_decode_frame(AVCodecContext *avctx, void *data, |
|
|
|
|
h->low_delay); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (h->pict_type == AV_PICTURE_TYPE_B || h->low_delay) |
|
|
|
|
if (s->pict_type == AV_PICTURE_TYPE_B || h->low_delay) |
|
|
|
|
ret = av_frame_ref(data, s->cur_pic->f); |
|
|
|
|
else if (s->last_pic->f->data[0]) |
|
|
|
|
ret = av_frame_ref(data, s->last_pic->f); |
|
|
|
@ -1563,7 +1563,7 @@ static int svq3_decode_frame(AVCodecContext *avctx, void *data, |
|
|
|
|
if (s->last_pic->f->data[0] || h->low_delay) |
|
|
|
|
*got_frame = 1; |
|
|
|
|
|
|
|
|
|
if (h->pict_type != AV_PICTURE_TYPE_B) { |
|
|
|
|
if (s->pict_type != AV_PICTURE_TYPE_B) { |
|
|
|
|
FFSWAP(H264Picture*, s->cur_pic, s->next_pic); |
|
|
|
|
} else { |
|
|
|
|
av_frame_unref(s->cur_pic->f); |
|
|
|
|