From: Yunqing Wang Date: Fri, 30 Jan 2015 18:14:44 +0000 (-0800) Subject: vp9_dthread: pass frame counts to decoder functions X-Git-Tag: v1.4.0~194^2 X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=85a9bc04d4ebe871076cf36c8aac8ab10a967d9c;p=libvpx vp9_dthread: pass frame counts to decoder functions The current multi-threaded tile decoder requires that the videoes are encoded with frame_parallel_decoding_mode = 1. This requirement is not necessary, and is better to be removed. This patch includes the first part of the work. Change-Id: Ic7695fb3cfe13f9022582c9f0edd2aa6e2e36d28 --- diff --git a/vp9/decoder/vp9_decodeframe.c b/vp9/decoder/vp9_decodeframe.c index e6c12f5b2..07249d092 100644 --- a/vp9/decoder/vp9_decodeframe.c +++ b/vp9/decoder/vp9_decodeframe.c @@ -299,6 +299,7 @@ static void inverse_transform_block(MACROBLOCKD* xd, int plane, int block, struct intra_args { VP9_COMMON *cm; MACROBLOCKD *xd; + FRAME_COUNTS *counts; vp9_reader *r; }; @@ -323,7 +324,7 @@ static void predict_and_reconstruct_intra_block(int plane, int block, x, y, plane); if (!mi->mbmi.skip) { - const int eob = vp9_decode_block_tokens(cm, xd, plane, block, + const int eob = vp9_decode_block_tokens(cm, xd, args->counts, plane, block, plane_bsize, x, y, tx_size, args->r); inverse_transform_block(xd, plane, block, tx_size, dst, pd->dst.stride, @@ -335,6 +336,7 @@ struct inter_args { VP9_COMMON *cm; MACROBLOCKD *xd; vp9_reader *r; + FRAME_COUNTS *counts; int *eobtotal; }; @@ -347,8 +349,8 @@ static void reconstruct_inter_block(int plane, int block, struct macroblockd_plane *const pd = &xd->plane[plane]; int x, y, eob; txfrm_block_to_raster_xy(plane_bsize, tx_size, block, &x, &y); - eob = vp9_decode_block_tokens(cm, xd, plane, block, plane_bsize, x, y, - tx_size, args->r); + eob = vp9_decode_block_tokens(cm, xd, args->counts, plane, block, plane_bsize, + x, y, tx_size, args->r); inverse_transform_block(xd, plane, block, tx_size, &pd->dst.buf[4 * y * pd->dst.stride + 4 * x], pd->dst.stride, eob); @@ -385,13 +387,14 @@ static MB_MODE_INFO *set_offsets(VP9_COMMON *const cm, MACROBLOCKD *const xd, } static void decode_block(VP9Decoder *const pbi, MACROBLOCKD *const xd, + FRAME_COUNTS *counts, const TileInfo *const tile, int mi_row, int mi_col, vp9_reader *r, BLOCK_SIZE bsize) { VP9_COMMON *const cm = &pbi->common; const int less8x8 = bsize < BLOCK_8X8; MB_MODE_INFO *mbmi = set_offsets(cm, xd, tile, bsize, mi_row, mi_col); - vp9_read_mode_info(pbi, xd, tile, mi_row, mi_col, r); + vp9_read_mode_info(pbi, xd, counts, tile, mi_row, mi_col, r); if (less8x8) bsize = BLOCK_8X8; @@ -405,7 +408,7 @@ static void decode_block(VP9Decoder *const pbi, MACROBLOCKD *const xd, } if (!is_inter_block(mbmi)) { - struct intra_args arg = { cm, xd, r }; + struct intra_args arg = { cm, xd, counts, r }; vp9_foreach_transformed_block(xd, bsize, predict_and_reconstruct_intra_block, &arg); } else { @@ -415,7 +418,7 @@ static void decode_block(VP9Decoder *const pbi, MACROBLOCKD *const xd, // Reconstruction if (!mbmi->skip) { int eobtotal = 0; - struct inter_args arg = { cm, xd, r, &eobtotal }; + struct inter_args arg = { cm, xd, r, counts, &eobtotal }; vp9_foreach_transformed_block(xd, bsize, reconstruct_inter_block, &arg); if (!less8x8 && eobtotal == 0) mbmi->skip = 1; // skip loopfilter @@ -425,7 +428,8 @@ static void decode_block(VP9Decoder *const pbi, MACROBLOCKD *const xd, xd->corrupted |= vp9_reader_has_error(r); } -static PARTITION_TYPE read_partition(VP9_COMMON *cm, MACROBLOCKD *xd, int hbs, +static PARTITION_TYPE read_partition(VP9_COMMON *cm, MACROBLOCKD *xd, + FRAME_COUNTS *counts, int hbs, int mi_row, int mi_col, BLOCK_SIZE bsize, vp9_reader *r) { const int ctx = partition_plane_context(xd, mi_row, mi_col, bsize); @@ -444,12 +448,13 @@ static PARTITION_TYPE read_partition(VP9_COMMON *cm, MACROBLOCKD *xd, int hbs, p = PARTITION_SPLIT; if (!cm->frame_parallel_decoding_mode) - ++cm->counts.partition[ctx][p]; + ++counts->partition[ctx][p]; return p; } static void decode_partition(VP9Decoder *const pbi, MACROBLOCKD *const xd, + FRAME_COUNTS *counts, const TileInfo *const tile, int mi_row, int mi_col, vp9_reader* r, BLOCK_SIZE bsize) { @@ -461,34 +466,37 @@ static void decode_partition(VP9Decoder *const pbi, MACROBLOCKD *const xd, if (mi_row >= cm->mi_rows || mi_col >= cm->mi_cols) return; - partition = read_partition(cm, xd, hbs, mi_row, mi_col, bsize, r); + partition = read_partition(cm, xd, counts, hbs, mi_row, mi_col, bsize, r); subsize = get_subsize(bsize, partition); uv_subsize = ss_size_lookup[subsize][cm->subsampling_x][cm->subsampling_y]; if (subsize >= BLOCK_8X8 && uv_subsize == BLOCK_INVALID) vpx_internal_error(xd->error_info, VPX_CODEC_CORRUPT_FRAME, "Invalid block size."); if (subsize < BLOCK_8X8) { - decode_block(pbi, xd, tile, mi_row, mi_col, r, subsize); + decode_block(pbi, xd, counts, tile, mi_row, mi_col, r, subsize); } else { switch (partition) { case PARTITION_NONE: - decode_block(pbi, xd, tile, mi_row, mi_col, r, subsize); + decode_block(pbi, xd, counts, tile, mi_row, mi_col, r, subsize); break; case PARTITION_HORZ: - decode_block(pbi, xd, tile, mi_row, mi_col, r, subsize); + decode_block(pbi, xd, counts, tile, mi_row, mi_col, r, subsize); if (mi_row + hbs < cm->mi_rows) - decode_block(pbi, xd, tile, mi_row + hbs, mi_col, r, subsize); + decode_block(pbi, xd, counts, tile, mi_row + hbs, mi_col, r, subsize); break; case PARTITION_VERT: - decode_block(pbi, xd, tile, mi_row, mi_col, r, subsize); + decode_block(pbi, xd, counts, tile, mi_row, mi_col, r, subsize); if (mi_col + hbs < cm->mi_cols) - decode_block(pbi, xd, tile, mi_row, mi_col + hbs, r, subsize); + decode_block(pbi, xd, counts, tile, mi_row, mi_col + hbs, r, subsize); break; case PARTITION_SPLIT: - decode_partition(pbi, xd, tile, mi_row, mi_col, r, subsize); - decode_partition(pbi, xd, tile, mi_row, mi_col + hbs, r, subsize); - decode_partition(pbi, xd, tile, mi_row + hbs, mi_col, r, subsize); - decode_partition(pbi, xd, tile, mi_row + hbs, mi_col + hbs, r, subsize); + decode_partition(pbi, xd, counts, tile, mi_row, mi_col, r, subsize); + decode_partition(pbi, xd, counts, tile, mi_row, mi_col + hbs, r, + subsize); + decode_partition(pbi, xd, counts, tile, mi_row + hbs, mi_col, r, + subsize); + decode_partition(pbi, xd, counts, tile, mi_row + hbs, mi_col + hbs, r, + subsize); break; default: assert(0 && "Invalid partition type"); @@ -983,8 +991,8 @@ static const uint8_t *decode_tiles(VP9Decoder *pbi, vp9_zero(tile_data->xd.left_seg_context); for (mi_col = tile.mi_col_start; mi_col < tile.mi_col_end; mi_col += MI_BLOCK_SIZE) { - decode_partition(pbi, &tile_data->xd, &tile, mi_row, mi_col, - &tile_data->bit_reader, BLOCK_64X64); + decode_partition(pbi, &tile_data->xd, &cm->counts, &tile, mi_row, + mi_col, &tile_data->bit_reader, BLOCK_64X64); } pbi->mb.corrupted |= tile_data->xd.corrupted; if (pbi->mb.corrupted) @@ -1056,8 +1064,10 @@ static int tile_worker_hook(TileWorkerData *const tile_data, vp9_zero(tile_data->xd.left_seg_context); for (mi_col = tile->mi_col_start; mi_col < tile->mi_col_end; mi_col += MI_BLOCK_SIZE) { - decode_partition(tile_data->pbi, &tile_data->xd, tile, - mi_row, mi_col, &tile_data->bit_reader, BLOCK_64X64); + decode_partition(tile_data->pbi, &tile_data->xd, + &tile_data->pbi->common.counts, + tile, mi_row, mi_col, &tile_data->bit_reader, + BLOCK_64X64); } } return !tile_data->xd.corrupted; diff --git a/vp9/decoder/vp9_decodemv.c b/vp9/decoder/vp9_decodemv.c index 40280ba5d..ec32f1d30 100644 --- a/vp9/decoder/vp9_decodemv.c +++ b/vp9/decoder/vp9_decodemv.c @@ -27,29 +27,31 @@ static PREDICTION_MODE read_intra_mode(vp9_reader *r, const vp9_prob *p) { return (PREDICTION_MODE)vp9_read_tree(r, vp9_intra_mode_tree, p); } -static PREDICTION_MODE read_intra_mode_y(VP9_COMMON *cm, vp9_reader *r, - int size_group) { +static PREDICTION_MODE read_intra_mode_y(VP9_COMMON *cm, FRAME_COUNTS *counts, + vp9_reader *r, int size_group) { const PREDICTION_MODE y_mode = read_intra_mode(r, cm->fc->y_mode_prob[size_group]); if (!cm->frame_parallel_decoding_mode) - ++cm->counts.y_mode[size_group][y_mode]; + ++counts->y_mode[size_group][y_mode]; return y_mode; } -static PREDICTION_MODE read_intra_mode_uv(VP9_COMMON *cm, vp9_reader *r, +static PREDICTION_MODE read_intra_mode_uv(VP9_COMMON *cm, FRAME_COUNTS *counts, + vp9_reader *r, PREDICTION_MODE y_mode) { const PREDICTION_MODE uv_mode = read_intra_mode(r, cm->fc->uv_mode_prob[y_mode]); if (!cm->frame_parallel_decoding_mode) - ++cm->counts.uv_mode[y_mode][uv_mode]; + ++counts->uv_mode[y_mode][uv_mode]; return uv_mode; } -static PREDICTION_MODE read_inter_mode(VP9_COMMON *cm, vp9_reader *r, int ctx) { +static PREDICTION_MODE read_inter_mode(VP9_COMMON *cm, FRAME_COUNTS *counts, + vp9_reader *r, int ctx) { const int mode = vp9_read_tree(r, vp9_inter_mode_tree, cm->fc->inter_mode_probs[ctx]); if (!cm->frame_parallel_decoding_mode) - ++cm->counts.inter_mode[ctx][mode]; + ++counts->inter_mode[ctx][mode]; return NEARESTMV + mode; } @@ -59,6 +61,7 @@ static int read_segment_id(vp9_reader *r, const struct segmentation *seg) { } static TX_SIZE read_selected_tx_size(VP9_COMMON *cm, MACROBLOCKD *xd, + FRAME_COUNTS *counts, TX_SIZE max_tx_size, vp9_reader *r) { const int ctx = vp9_get_tx_size_context(xd); const vp9_prob *tx_probs = get_tx_probs(max_tx_size, ctx, &cm->fc->tx_probs); @@ -70,17 +73,18 @@ static TX_SIZE read_selected_tx_size(VP9_COMMON *cm, MACROBLOCKD *xd, } if (!cm->frame_parallel_decoding_mode) - ++get_tx_counts(max_tx_size, ctx, &cm->counts.tx)[tx_size]; + ++get_tx_counts(max_tx_size, ctx, &counts->tx)[tx_size]; return (TX_SIZE)tx_size; } static TX_SIZE read_tx_size(VP9_COMMON *cm, MACROBLOCKD *xd, + FRAME_COUNTS *counts, int allow_select, vp9_reader *r) { TX_MODE tx_mode = cm->tx_mode; BLOCK_SIZE bsize = xd->mi[0].src_mi->mbmi.sb_type; const TX_SIZE max_tx_size = max_txsize_lookup[bsize]; if (allow_select && tx_mode == TX_MODE_SELECT && bsize >= BLOCK_8X8) - return read_selected_tx_size(cm, xd, max_tx_size, r); + return read_selected_tx_size(cm, xd, counts, max_tx_size, r); else return MIN(max_tx_size, tx_mode_to_biggest_tx_size[tx_mode]); } @@ -171,6 +175,7 @@ static int read_inter_segment_id(VP9_COMMON *const cm, MACROBLOCKD *const xd, } static int read_skip(VP9_COMMON *cm, const MACROBLOCKD *xd, + FRAME_COUNTS *counts, int segment_id, vp9_reader *r) { if (vp9_segfeature_active(&cm->seg, segment_id, SEG_LVL_SKIP)) { return 1; @@ -178,13 +183,14 @@ static int read_skip(VP9_COMMON *cm, const MACROBLOCKD *xd, const int ctx = vp9_get_skip_context(xd); const int skip = vp9_read(r, cm->fc->skip_probs[ctx]); if (!cm->frame_parallel_decoding_mode) - ++cm->counts.skip[ctx][skip]; + ++counts->skip[ctx][skip]; return skip; } } static void read_intra_frame_mode_info(VP9_COMMON *const cm, MACROBLOCKD *const xd, + FRAME_COUNTS *counts, int mi_row, int mi_col, vp9_reader *r) { MODE_INFO *const mi = xd->mi[0].src_mi; MB_MODE_INFO *const mbmi = &mi->mbmi; @@ -194,8 +200,8 @@ static void read_intra_frame_mode_info(VP9_COMMON *const cm, int i; mbmi->segment_id = read_intra_segment_id(cm, xd, mi_row, mi_col, r); - mbmi->skip = read_skip(cm, xd, mbmi->segment_id, r); - mbmi->tx_size = read_tx_size(cm, xd, 1, r); + mbmi->skip = read_skip(cm, xd, counts, mbmi->segment_id, r); + mbmi->tx_size = read_tx_size(cm, xd, counts, 1, r); mbmi->ref_frame[0] = INTRA_FRAME; mbmi->ref_frame[1] = NONE; @@ -280,13 +286,14 @@ static INLINE void read_mv(vp9_reader *r, MV *mv, const MV *ref, static REFERENCE_MODE read_block_reference_mode(VP9_COMMON *cm, const MACROBLOCKD *xd, + FRAME_COUNTS *counts, vp9_reader *r) { if (cm->reference_mode == REFERENCE_MODE_SELECT) { const int ctx = vp9_get_reference_mode_context(cm, xd); const REFERENCE_MODE mode = (REFERENCE_MODE)vp9_read(r, cm->fc->comp_inter_prob[ctx]); if (!cm->frame_parallel_decoding_mode) - ++cm->counts.comp_inter[ctx][mode]; + ++counts->comp_inter[ctx][mode]; return mode; // SINGLE_REFERENCE or COMPOUND_REFERENCE } else { return cm->reference_mode; @@ -295,17 +302,16 @@ static REFERENCE_MODE read_block_reference_mode(VP9_COMMON *cm, // Read the referncence frame static void read_ref_frames(VP9_COMMON *const cm, MACROBLOCKD *const xd, - vp9_reader *r, + FRAME_COUNTS *counts, vp9_reader *r, int segment_id, MV_REFERENCE_FRAME ref_frame[2]) { FRAME_CONTEXT *const fc = cm->fc; - FRAME_COUNTS *const counts = &cm->counts; if (vp9_segfeature_active(&cm->seg, segment_id, SEG_LVL_REF_FRAME)) { ref_frame[0] = (MV_REFERENCE_FRAME)vp9_get_segdata(&cm->seg, segment_id, SEG_LVL_REF_FRAME); ref_frame[1] = NONE; } else { - const REFERENCE_MODE mode = read_block_reference_mode(cm, xd, r); + const REFERENCE_MODE mode = read_block_reference_mode(cm, xd, counts, r); // FIXME(rbultje) I'm pretty sure this breaks segmentation ref frame coding if (mode == COMPOUND_REFERENCE) { const int idx = cm->ref_frame_sign_bias[cm->comp_fixed_ref]; @@ -339,17 +345,19 @@ static void read_ref_frames(VP9_COMMON *const cm, MACROBLOCKD *const xd, static INLINE INTERP_FILTER read_switchable_interp_filter( - VP9_COMMON *const cm, MACROBLOCKD *const xd, vp9_reader *r) { + VP9_COMMON *const cm, MACROBLOCKD *const xd, + FRAME_COUNTS *counts, vp9_reader *r) { const int ctx = vp9_get_pred_context_switchable_interp(xd); const INTERP_FILTER type = (INTERP_FILTER)vp9_read_tree(r, vp9_switchable_interp_tree, cm->fc->switchable_interp_prob[ctx]); if (!cm->frame_parallel_decoding_mode) - ++cm->counts.switchable_interp[ctx][type]; + ++counts->switchable_interp[ctx][type]; return type; } -static void read_intra_block_mode_info(VP9_COMMON *const cm, MODE_INFO *mi, +static void read_intra_block_mode_info(VP9_COMMON *const cm, + FRAME_COUNTS *counts, MODE_INFO *mi, vp9_reader *r) { MB_MODE_INFO *const mbmi = &mi->mbmi; const BLOCK_SIZE bsize = mi->mbmi.sb_type; @@ -361,24 +369,26 @@ static void read_intra_block_mode_info(VP9_COMMON *const cm, MODE_INFO *mi, switch (bsize) { case BLOCK_4X4: for (i = 0; i < 4; ++i) - mi->bmi[i].as_mode = read_intra_mode_y(cm, r, 0); + mi->bmi[i].as_mode = read_intra_mode_y(cm, counts, r, 0); mbmi->mode = mi->bmi[3].as_mode; break; case BLOCK_4X8: - mi->bmi[0].as_mode = mi->bmi[2].as_mode = read_intra_mode_y(cm, r, 0); + mi->bmi[0].as_mode = mi->bmi[2].as_mode = read_intra_mode_y(cm, counts, + r, 0); mi->bmi[1].as_mode = mi->bmi[3].as_mode = mbmi->mode = - read_intra_mode_y(cm, r, 0); + read_intra_mode_y(cm, counts, r, 0); break; case BLOCK_8X4: - mi->bmi[0].as_mode = mi->bmi[1].as_mode = read_intra_mode_y(cm, r, 0); + mi->bmi[0].as_mode = mi->bmi[1].as_mode = read_intra_mode_y(cm, counts, + r, 0); mi->bmi[2].as_mode = mi->bmi[3].as_mode = mbmi->mode = - read_intra_mode_y(cm, r, 0); + read_intra_mode_y(cm, counts, r, 0); break; default: - mbmi->mode = read_intra_mode_y(cm, r, size_group_lookup[bsize]); + mbmi->mode = read_intra_mode_y(cm, counts, r, size_group_lookup[bsize]); } - mbmi->uv_mode = read_intra_mode_uv(cm, r, mbmi->mode); + mbmi->uv_mode = read_intra_mode_uv(cm, counts, r, mbmi->mode); } static INLINE int is_mv_valid(const MV *mv) { @@ -386,7 +396,8 @@ static INLINE int is_mv_valid(const MV *mv) { mv->col > MV_LOW && mv->col < MV_UPP; } -static INLINE int assign_mv(VP9_COMMON *cm, PREDICTION_MODE mode, +static INLINE int assign_mv(VP9_COMMON *cm, FRAME_COUNTS *counts, + PREDICTION_MODE mode, int_mv mv[2], int_mv ref_mv[2], int_mv nearest_mv[2], int_mv near_mv[2], int is_compound, int allow_hp, vp9_reader *r) { @@ -396,7 +407,7 @@ static INLINE int assign_mv(VP9_COMMON *cm, PREDICTION_MODE mode, switch (mode) { case NEWMV: { nmv_context_counts *const mv_counts = cm->frame_parallel_decoding_mode ? - NULL : &cm->counts.mv; + NULL : &counts->mv; for (i = 0; i < 1 + is_compound; ++i) { read_mv(r, &mv[i].as_mv, &ref_mv[i].as_mv, &cm->fc->nmvc, mv_counts, allow_hp); @@ -430,6 +441,7 @@ static INLINE int assign_mv(VP9_COMMON *cm, PREDICTION_MODE mode, } static int read_is_inter_block(VP9_COMMON *const cm, MACROBLOCKD *const xd, + FRAME_COUNTS *counts, int segment_id, vp9_reader *r) { if (vp9_segfeature_active(&cm->seg, segment_id, SEG_LVL_REF_FRAME)) { return vp9_get_segdata(&cm->seg, segment_id, SEG_LVL_REF_FRAME) != @@ -438,7 +450,7 @@ static int read_is_inter_block(VP9_COMMON *const cm, MACROBLOCKD *const xd, const int ctx = vp9_get_intra_inter_context(xd); const int is_inter = vp9_read(r, cm->fc->intra_inter_prob[ctx]); if (!cm->frame_parallel_decoding_mode) - ++cm->counts.intra_inter[ctx][is_inter]; + ++counts->intra_inter[ctx][is_inter]; return is_inter; } } @@ -451,6 +463,7 @@ static void fpm_sync(void *const data, int mi_row) { static void read_inter_block_mode_info(VP9Decoder *const pbi, MACROBLOCKD *const xd, + FRAME_COUNTS *counts, const TileInfo *const tile, MODE_INFO *const mi, int mi_row, int mi_col, vp9_reader *r) { @@ -461,7 +474,7 @@ static void read_inter_block_mode_info(VP9Decoder *const pbi, int_mv nearestmv[2], nearmv[2]; int inter_mode_ctx, ref, is_compound; - read_ref_frames(cm, xd, r, mbmi->segment_id, mbmi->ref_frame); + read_ref_frames(cm, xd, counts, r, mbmi->segment_id, mbmi->ref_frame); is_compound = has_second_ref(mbmi); for (ref = 0; ref < 1 + is_compound; ++ref) { @@ -488,7 +501,7 @@ static void read_inter_block_mode_info(VP9Decoder *const pbi, } } else { if (bsize >= BLOCK_8X8) - mbmi->mode = read_inter_mode(cm, r, inter_mode_ctx); + mbmi->mode = read_inter_mode(cm, counts, r, inter_mode_ctx); } if (bsize < BLOCK_8X8 || mbmi->mode != ZEROMV) { @@ -499,7 +512,7 @@ static void read_inter_block_mode_info(VP9Decoder *const pbi, } mbmi->interp_filter = (cm->interp_filter == SWITCHABLE) - ? read_switchable_interp_filter(cm, xd, r) + ? read_switchable_interp_filter(cm, xd, counts, r) : cm->interp_filter; if (bsize < BLOCK_8X8) { @@ -512,7 +525,7 @@ static void read_inter_block_mode_info(VP9Decoder *const pbi, for (idx = 0; idx < 2; idx += num_4x4_w) { int_mv block[2]; const int j = idy * 2 + idx; - b_mode = read_inter_mode(cm, r, inter_mode_ctx); + b_mode = read_inter_mode(cm, counts, r, inter_mode_ctx); if (b_mode == NEARESTMV || b_mode == NEARMV) for (ref = 0; ref < 1 + is_compound; ++ref) @@ -520,7 +533,7 @@ static void read_inter_block_mode_info(VP9Decoder *const pbi, &nearest_sub8x8[ref], &near_sub8x8[ref]); - if (!assign_mv(cm, b_mode, block, nearestmv, + if (!assign_mv(cm, counts, b_mode, block, nearestmv, nearest_sub8x8, near_sub8x8, is_compound, allow_hp, r)) { xd->corrupted |= 1; @@ -543,13 +556,14 @@ static void read_inter_block_mode_info(VP9Decoder *const pbi, mbmi->mv[0].as_int = mi->bmi[3].as_mv[0].as_int; mbmi->mv[1].as_int = mi->bmi[3].as_mv[1].as_int; } else { - xd->corrupted |= !assign_mv(cm, mbmi->mode, mbmi->mv, nearestmv, + xd->corrupted |= !assign_mv(cm, counts, mbmi->mode, mbmi->mv, nearestmv, nearestmv, nearmv, is_compound, allow_hp, r); } } static void read_inter_frame_mode_info(VP9Decoder *const pbi, MACROBLOCKD *const xd, + FRAME_COUNTS *counts, const TileInfo *const tile, int mi_row, int mi_col, vp9_reader *r) { VP9_COMMON *const cm = &pbi->common; @@ -560,17 +574,18 @@ static void read_inter_frame_mode_info(VP9Decoder *const pbi, mbmi->mv[0].as_int = 0; mbmi->mv[1].as_int = 0; mbmi->segment_id = read_inter_segment_id(cm, xd, mi_row, mi_col, r); - mbmi->skip = read_skip(cm, xd, mbmi->segment_id, r); - inter_block = read_is_inter_block(cm, xd, mbmi->segment_id, r); - mbmi->tx_size = read_tx_size(cm, xd, !mbmi->skip || !inter_block, r); + mbmi->skip = read_skip(cm, xd, counts, mbmi->segment_id, r); + inter_block = read_is_inter_block(cm, xd, counts, mbmi->segment_id, r); + mbmi->tx_size = read_tx_size(cm, xd, counts, !mbmi->skip || !inter_block, r); if (inter_block) - read_inter_block_mode_info(pbi, xd, tile, mi, mi_row, mi_col, r); + read_inter_block_mode_info(pbi, xd, counts, tile, mi, mi_row, mi_col, r); else - read_intra_block_mode_info(cm, mi, r); + read_intra_block_mode_info(cm, counts, mi, r); } void vp9_read_mode_info(VP9Decoder *const pbi, MACROBLOCKD *xd, + FRAME_COUNTS *counts, const TileInfo *const tile, int mi_row, int mi_col, vp9_reader *r) { VP9_COMMON *const cm = &pbi->common; @@ -583,9 +598,9 @@ void vp9_read_mode_info(VP9Decoder *const pbi, MACROBLOCKD *xd, int w, h; if (frame_is_intra_only(cm)) - read_intra_frame_mode_info(cm, xd, mi_row, mi_col, r); + read_intra_frame_mode_info(cm, xd, counts, mi_row, mi_col, r); else - read_inter_frame_mode_info(pbi, xd, tile, mi_row, mi_col, r); + read_inter_frame_mode_info(pbi, xd, counts, tile, mi_row, mi_col, r); for (h = 0; h < y_mis; ++h) { MV_REF *const frame_mv = frame_mvs + h * cm->mi_cols; diff --git a/vp9/decoder/vp9_decodemv.h b/vp9/decoder/vp9_decodemv.h index dd97d8da0..c79dff718 100644 --- a/vp9/decoder/vp9_decodemv.h +++ b/vp9/decoder/vp9_decodemv.h @@ -21,6 +21,7 @@ extern "C" { struct TileInfo; void vp9_read_mode_info(VP9Decoder *const pbi, MACROBLOCKD *xd, + FRAME_COUNTS *counts, const struct TileInfo *const tile, int mi_row, int mi_col, vp9_reader *r); diff --git a/vp9/decoder/vp9_detokenize.c b/vp9/decoder/vp9_detokenize.c index 23d622d70..28ef7111c 100644 --- a/vp9/decoder/vp9_detokenize.c +++ b/vp9/decoder/vp9_detokenize.c @@ -56,13 +56,13 @@ static const vp9_tree_index coeff_subtree_high[TREE_SIZE(ENTROPY_TOKENS)] = { -CATEGORY5_TOKEN, -CATEGORY6_TOKEN /* 7 = CAT_FIVE */ }; -static int decode_coefs(VP9_COMMON *cm, const MACROBLOCKD *xd, PLANE_TYPE type, +static int decode_coefs(VP9_COMMON *cm, const MACROBLOCKD *xd, + FRAME_COUNTS *counts, PLANE_TYPE type, tran_low_t *dqcoeff, TX_SIZE tx_size, const int16_t *dq, int ctx, const int16_t *scan, const int16_t *nb, vp9_reader *r) { const int max_eob = 16 << (tx_size << 1); const FRAME_CONTEXT *const fc = cm->fc; - FRAME_COUNTS *const counts = &cm->counts; const int ref = is_inter_block(&xd->mi[0].src_mi->mbmi); int band, c = 0; const vp9_prob (*coef_probs)[COEFF_CONTEXTS][UNCONSTRAINED_NODES] = @@ -213,13 +213,14 @@ static int decode_coefs(VP9_COMMON *cm, const MACROBLOCKD *xd, PLANE_TYPE type, } int vp9_decode_block_tokens(VP9_COMMON *cm, MACROBLOCKD *xd, - int plane, int block, BLOCK_SIZE plane_bsize, - int x, int y, TX_SIZE tx_size, vp9_reader *r) { + FRAME_COUNTS *counts, int plane, int block, + BLOCK_SIZE plane_bsize, int x, int y, + TX_SIZE tx_size, vp9_reader *r) { struct macroblockd_plane *const pd = &xd->plane[plane]; const int ctx = get_entropy_context(tx_size, pd->above_context + x, pd->left_context + y); const scan_order *so = get_scan(xd, tx_size, pd->plane_type, block); - const int eob = decode_coefs(cm, xd, pd->plane_type, + const int eob = decode_coefs(cm, xd, counts, pd->plane_type, BLOCK_OFFSET(pd->dqcoeff, block), tx_size, pd->dequant, ctx, so->scan, so->neighbors, r); vp9_set_contexts(xd, pd, plane_bsize, tx_size, eob > 0, x, y); diff --git a/vp9/decoder/vp9_detokenize.h b/vp9/decoder/vp9_detokenize.h index 5278e97a3..30d157c0e 100644 --- a/vp9/decoder/vp9_detokenize.h +++ b/vp9/decoder/vp9_detokenize.h @@ -20,8 +20,9 @@ extern "C" { #endif int vp9_decode_block_tokens(VP9_COMMON *cm, MACROBLOCKD *xd, - int plane, int block, BLOCK_SIZE plane_bsize, - int x, int y, TX_SIZE tx_size, vp9_reader *r); + FRAME_COUNTS *counts, int plane, int block, + BLOCK_SIZE plane_bsize, int x, int y, + TX_SIZE tx_size, vp9_reader *r); #ifdef __cplusplus } // extern "C"