Index: source/libvpx/vp9/decoder/vp9_decodemv.c |
diff --git a/source/libvpx/vp9/decoder/vp9_decodemv.c b/source/libvpx/vp9/decoder/vp9_decodemv.c |
index ec32f1d304648c5ed8b3c05a26b2135d4401f536..1c2603b0a50cd59dfff3df01a9e37da674b0e1c8 100644 |
--- a/source/libvpx/vp9/decoder/vp9_decodemv.c |
+++ b/source/libvpx/vp9/decoder/vp9_decodemv.c |
@@ -27,31 +27,29 @@ 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, FRAME_COUNTS *counts, |
- vp9_reader *r, int size_group) { |
+static PREDICTION_MODE read_intra_mode_y(VP9_COMMON *cm, 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) |
- ++counts->y_mode[size_group][y_mode]; |
+ ++cm->counts.y_mode[size_group][y_mode]; |
return y_mode; |
} |
-static PREDICTION_MODE read_intra_mode_uv(VP9_COMMON *cm, FRAME_COUNTS *counts, |
- vp9_reader *r, |
+static PREDICTION_MODE read_intra_mode_uv(VP9_COMMON *cm, 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) |
- ++counts->uv_mode[y_mode][uv_mode]; |
+ ++cm->counts.uv_mode[y_mode][uv_mode]; |
return uv_mode; |
} |
-static PREDICTION_MODE read_inter_mode(VP9_COMMON *cm, FRAME_COUNTS *counts, |
- vp9_reader *r, int ctx) { |
+static PREDICTION_MODE read_inter_mode(VP9_COMMON *cm, 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) |
- ++counts->inter_mode[ctx][mode]; |
+ ++cm->counts.inter_mode[ctx][mode]; |
return NEARESTMV + mode; |
} |
@@ -61,7 +59,6 @@ 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); |
@@ -73,18 +70,17 @@ 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, &counts->tx)[tx_size]; |
+ ++get_tx_counts(max_tx_size, ctx, &cm->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, counts, max_tx_size, r); |
+ return read_selected_tx_size(cm, xd, max_tx_size, r); |
else |
return MIN(max_tx_size, tx_mode_to_biggest_tx_size[tx_mode]); |
} |
@@ -102,24 +98,7 @@ static void set_segment_id(VP9_COMMON *cm, BLOCK_SIZE bsize, |
for (y = 0; y < ymis; y++) |
for (x = 0; x < xmis; x++) |
- cm->current_frame_seg_map[mi_offset + y * cm->mi_cols + x] = segment_id; |
-} |
- |
-static void copy_segment_id(const VP9_COMMON *cm, |
- const uint8_t *last_segment_ids, |
- uint8_t *current_segment_ids, |
- BLOCK_SIZE bsize, int mi_row, int mi_col) { |
- const int mi_offset = mi_row * cm->mi_cols + mi_col; |
- const int bw = num_8x8_blocks_wide_lookup[bsize]; |
- const int bh = num_8x8_blocks_high_lookup[bsize]; |
- const int xmis = MIN(cm->mi_cols - mi_col, bw); |
- const int ymis = MIN(cm->mi_rows - mi_row, bh); |
- int x, y; |
- |
- for (y = 0; y < ymis; y++) |
- for (x = 0; x < xmis; x++) |
- current_segment_ids[mi_offset + y * cm->mi_cols + x] = last_segment_ids ? |
- last_segment_ids[mi_offset + y * cm->mi_cols + x] : 0; |
+ cm->last_frame_seg_map[mi_offset + y * cm->mi_cols + x] = segment_id; |
} |
static int read_intra_segment_id(VP9_COMMON *const cm, MACROBLOCKD *const xd, |
@@ -132,11 +111,8 @@ static int read_intra_segment_id(VP9_COMMON *const cm, MACROBLOCKD *const xd, |
if (!seg->enabled) |
return 0; // Default for disabled segmentation |
- if (!seg->update_map) { |
- copy_segment_id(cm, cm->last_frame_seg_map, cm->current_frame_seg_map, |
- bsize, mi_row, mi_col); |
+ if (!seg->update_map) |
return 0; |
- } |
segment_id = read_segment_id(r, seg); |
set_segment_id(cm, bsize, mi_row, mi_col, segment_id); |
@@ -153,14 +129,10 @@ static int read_inter_segment_id(VP9_COMMON *const cm, MACROBLOCKD *const xd, |
if (!seg->enabled) |
return 0; // Default for disabled segmentation |
- predicted_segment_id = cm->last_frame_seg_map ? |
- vp9_get_segment_id(cm, cm->last_frame_seg_map, bsize, mi_row, mi_col) : 0; |
- |
- if (!seg->update_map) { |
- copy_segment_id(cm, cm->last_frame_seg_map, cm->current_frame_seg_map, |
- bsize, mi_row, mi_col); |
+ predicted_segment_id = vp9_get_segment_id(cm, cm->last_frame_seg_map, |
+ bsize, mi_row, mi_col); |
+ if (!seg->update_map) |
return predicted_segment_id; |
- } |
if (seg->temporal_update) { |
const vp9_prob pred_prob = vp9_get_pred_prob_seg_id(seg, xd); |
@@ -175,7 +147,6 @@ 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; |
@@ -183,14 +154,13 @@ 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) |
- ++counts->skip[ctx][skip]; |
+ ++cm->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; |
@@ -200,8 +170,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, counts, mbmi->segment_id, r); |
- mbmi->tx_size = read_tx_size(cm, xd, counts, 1, r); |
+ mbmi->skip = read_skip(cm, xd, mbmi->segment_id, r); |
+ mbmi->tx_size = read_tx_size(cm, xd, 1, r); |
mbmi->ref_frame[0] = INTRA_FRAME; |
mbmi->ref_frame[1] = NONE; |
@@ -286,14 +256,13 @@ 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) |
- ++counts->comp_inter[ctx][mode]; |
+ ++cm->counts.comp_inter[ctx][mode]; |
return mode; // SINGLE_REFERENCE or COMPOUND_REFERENCE |
} else { |
return cm->reference_mode; |
@@ -302,16 +271,17 @@ 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, |
- FRAME_COUNTS *counts, vp9_reader *r, |
+ 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, counts, r); |
+ const REFERENCE_MODE mode = read_block_reference_mode(cm, xd, 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]; |
@@ -345,19 +315,17 @@ 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, |
- FRAME_COUNTS *counts, vp9_reader *r) { |
+ VP9_COMMON *const cm, MACROBLOCKD *const xd, 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) |
- ++counts->switchable_interp[ctx][type]; |
+ ++cm->counts.switchable_interp[ctx][type]; |
return type; |
} |
-static void read_intra_block_mode_info(VP9_COMMON *const cm, |
- FRAME_COUNTS *counts, MODE_INFO *mi, |
+static void read_intra_block_mode_info(VP9_COMMON *const cm, MODE_INFO *mi, |
vp9_reader *r) { |
MB_MODE_INFO *const mbmi = &mi->mbmi; |
const BLOCK_SIZE bsize = mi->mbmi.sb_type; |
@@ -369,26 +337,24 @@ static void read_intra_block_mode_info(VP9_COMMON *const cm, |
switch (bsize) { |
case BLOCK_4X4: |
for (i = 0; i < 4; ++i) |
- mi->bmi[i].as_mode = read_intra_mode_y(cm, counts, r, 0); |
+ mi->bmi[i].as_mode = read_intra_mode_y(cm, 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, counts, |
- r, 0); |
+ mi->bmi[0].as_mode = mi->bmi[2].as_mode = read_intra_mode_y(cm, r, 0); |
mi->bmi[1].as_mode = mi->bmi[3].as_mode = mbmi->mode = |
- read_intra_mode_y(cm, counts, r, 0); |
+ read_intra_mode_y(cm, r, 0); |
break; |
case BLOCK_8X4: |
- mi->bmi[0].as_mode = mi->bmi[1].as_mode = read_intra_mode_y(cm, counts, |
- r, 0); |
+ mi->bmi[0].as_mode = mi->bmi[1].as_mode = read_intra_mode_y(cm, r, 0); |
mi->bmi[2].as_mode = mi->bmi[3].as_mode = mbmi->mode = |
- read_intra_mode_y(cm, counts, r, 0); |
+ read_intra_mode_y(cm, r, 0); |
break; |
default: |
- mbmi->mode = read_intra_mode_y(cm, counts, r, size_group_lookup[bsize]); |
+ mbmi->mode = read_intra_mode_y(cm, r, size_group_lookup[bsize]); |
} |
- mbmi->uv_mode = read_intra_mode_uv(cm, counts, r, mbmi->mode); |
+ mbmi->uv_mode = read_intra_mode_uv(cm, r, mbmi->mode); |
} |
static INLINE int is_mv_valid(const MV *mv) { |
@@ -396,8 +362,7 @@ 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, FRAME_COUNTS *counts, |
- PREDICTION_MODE mode, |
+static INLINE int assign_mv(VP9_COMMON *cm, 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) { |
@@ -407,7 +372,7 @@ static INLINE int assign_mv(VP9_COMMON *cm, FRAME_COUNTS *counts, |
switch (mode) { |
case NEWMV: { |
nmv_context_counts *const mv_counts = cm->frame_parallel_decoding_mode ? |
- NULL : &counts->mv; |
+ NULL : &cm->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); |
@@ -441,7 +406,6 @@ static INLINE int assign_mv(VP9_COMMON *cm, FRAME_COUNTS *counts, |
} |
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) != |
@@ -450,31 +414,23 @@ 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) |
- ++counts->intra_inter[ctx][is_inter]; |
+ ++cm->counts.intra_inter[ctx][is_inter]; |
return is_inter; |
} |
} |
-static void fpm_sync(void *const data, int mi_row) { |
- VP9Decoder *const pbi = (VP9Decoder *)data; |
- vp9_frameworker_wait(pbi->frame_worker_owner, pbi->prev_buf, |
- mi_row << MI_BLOCK_SIZE_LOG2); |
-} |
- |
-static void read_inter_block_mode_info(VP9Decoder *const pbi, |
+static void read_inter_block_mode_info(VP9_COMMON *const cm, |
MACROBLOCKD *const xd, |
- FRAME_COUNTS *counts, |
const TileInfo *const tile, |
MODE_INFO *const mi, |
int mi_row, int mi_col, vp9_reader *r) { |
- VP9_COMMON *const cm = &pbi->common; |
MB_MODE_INFO *const mbmi = &mi->mbmi; |
const BLOCK_SIZE bsize = mbmi->sb_type; |
const int allow_hp = cm->allow_high_precision_mv; |
int_mv nearestmv[2], nearmv[2]; |
int inter_mode_ctx, ref, is_compound; |
- read_ref_frames(cm, xd, counts, r, mbmi->segment_id, mbmi->ref_frame); |
+ read_ref_frames(cm, xd, r, mbmi->segment_id, mbmi->ref_frame); |
is_compound = has_second_ref(mbmi); |
for (ref = 0; ref < 1 + is_compound; ++ref) { |
@@ -487,7 +443,7 @@ static void read_inter_block_mode_info(VP9Decoder *const pbi, |
vp9_setup_pre_planes(xd, ref, ref_buf->buf, mi_row, mi_col, |
&ref_buf->sf); |
vp9_find_mv_refs(cm, xd, tile, mi, frame, mbmi->ref_mvs[frame], |
- mi_row, mi_col, fpm_sync, (void *)pbi); |
+ mi_row, mi_col); |
} |
inter_mode_ctx = mbmi->mode_context[mbmi->ref_frame[0]]; |
@@ -501,7 +457,7 @@ static void read_inter_block_mode_info(VP9Decoder *const pbi, |
} |
} else { |
if (bsize >= BLOCK_8X8) |
- mbmi->mode = read_inter_mode(cm, counts, r, inter_mode_ctx); |
+ mbmi->mode = read_inter_mode(cm, r, inter_mode_ctx); |
} |
if (bsize < BLOCK_8X8 || mbmi->mode != ZEROMV) { |
@@ -512,7 +468,7 @@ static void read_inter_block_mode_info(VP9Decoder *const pbi, |
} |
mbmi->interp_filter = (cm->interp_filter == SWITCHABLE) |
- ? read_switchable_interp_filter(cm, xd, counts, r) |
+ ? read_switchable_interp_filter(cm, xd, r) |
: cm->interp_filter; |
if (bsize < BLOCK_8X8) { |
@@ -525,7 +481,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, counts, r, inter_mode_ctx); |
+ b_mode = read_inter_mode(cm, r, inter_mode_ctx); |
if (b_mode == NEARESTMV || b_mode == NEARMV) |
for (ref = 0; ref < 1 + is_compound; ++ref) |
@@ -533,7 +489,7 @@ static void read_inter_block_mode_info(VP9Decoder *const pbi, |
&nearest_sub8x8[ref], |
&near_sub8x8[ref]); |
- if (!assign_mv(cm, counts, b_mode, block, nearestmv, |
+ if (!assign_mv(cm, b_mode, block, nearestmv, |
nearest_sub8x8, near_sub8x8, |
is_compound, allow_hp, r)) { |
xd->corrupted |= 1; |
@@ -556,17 +512,15 @@ 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, counts, mbmi->mode, mbmi->mv, nearestmv, |
+ xd->corrupted |= !assign_mv(cm, mbmi->mode, mbmi->mv, nearestmv, |
nearestmv, nearmv, is_compound, allow_hp, r); |
} |
} |
-static void read_inter_frame_mode_info(VP9Decoder *const pbi, |
+static void read_inter_frame_mode_info(VP9_COMMON *const cm, |
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; |
MODE_INFO *const mi = xd->mi[0].src_mi; |
MB_MODE_INFO *const mbmi = &mi->mbmi; |
int inter_block; |
@@ -574,21 +528,19 @@ 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, 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); |
+ 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); |
if (inter_block) |
- read_inter_block_mode_info(pbi, xd, counts, tile, mi, mi_row, mi_col, r); |
+ read_inter_block_mode_info(cm, xd, tile, mi, mi_row, mi_col, r); |
else |
- read_intra_block_mode_info(cm, counts, mi, r); |
+ read_intra_block_mode_info(cm, mi, r); |
} |
-void vp9_read_mode_info(VP9Decoder *const pbi, MACROBLOCKD *xd, |
- FRAME_COUNTS *counts, |
+void vp9_read_mode_info(VP9_COMMON *cm, MACROBLOCKD *xd, |
const TileInfo *const tile, |
int mi_row, int mi_col, vp9_reader *r) { |
- VP9_COMMON *const cm = &pbi->common; |
MODE_INFO *const mi = xd->mi[0].src_mi; |
const int bw = num_8x8_blocks_wide_lookup[mi->mbmi.sb_type]; |
const int bh = num_8x8_blocks_high_lookup[mi->mbmi.sb_type]; |
@@ -598,9 +550,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, counts, mi_row, mi_col, r); |
+ read_intra_frame_mode_info(cm, xd, mi_row, mi_col, r); |
else |
- read_inter_frame_mode_info(pbi, xd, counts, tile, mi_row, mi_col, r); |
+ read_inter_frame_mode_info(cm, xd, tile, mi_row, mi_col, r); |
for (h = 0; h < y_mis; ++h) { |
MV_REF *const frame_mv = frame_mvs + h * cm->mi_cols; |