diff options
author | hkuang <hkuang@google.com> | 2013-09-16 15:09:58 -0700 |
---|---|---|
committer | Hangyu Kuang <hkuang@google.com> | 2013-09-17 22:05:28 +0000 |
commit | 1184aebb761cbeac9124c37189a80a1a58f04b6b (patch) | |
tree | b1ce6b3d29c43ffd22eb18999c5c3bad26513a48 /libvpx/vp9/common/vp9_pred_common.c | |
parent | f3bed9137f66ef693bd406e43b17e9a1114f1e14 (diff) | |
download | android_external_libvpx-1184aebb761cbeac9124c37189a80a1a58f04b6b.tar.gz android_external_libvpx-1184aebb761cbeac9124c37189a80a1a58f04b6b.tar.bz2 android_external_libvpx-1184aebb761cbeac9124c37189a80a1a58f04b6b.zip |
Roll latest libvpx into Android.
The latest libvpx has more neon optimizations and a lot of
algorithm optimizations which make the vp9 decode much more faster.
bug:10804666
Change-Id: I75eaacea57ecc7542a780be778f0e9e157978524
(cherry picked from commit 3df0563f1b24dac6c0bd122fc922a48211269061)
Diffstat (limited to 'libvpx/vp9/common/vp9_pred_common.c')
-rw-r--r-- | libvpx/vp9/common/vp9_pred_common.c | 362 |
1 files changed, 172 insertions, 190 deletions
diff --git a/libvpx/vp9/common/vp9_pred_common.c b/libvpx/vp9/common/vp9_pred_common.c index 795962a..81fbf1f 100644 --- a/libvpx/vp9/common/vp9_pred_common.c +++ b/libvpx/vp9/common/vp9_pred_common.c @@ -18,50 +18,49 @@ // Returns a context number for the given MB prediction signal unsigned char vp9_get_pred_context_switchable_interp(const MACROBLOCKD *xd) { - const MODE_INFO *const mi = xd->mode_info_context; - const MB_MODE_INFO *const above_mbmi = &mi[-xd->mode_info_stride].mbmi; - const MB_MODE_INFO *const left_mbmi = &mi[-1].mbmi; - const int left_in_image = xd->left_available && left_mbmi->mb_in_image; - const int above_in_image = xd->up_available && above_mbmi->mb_in_image; + const MODE_INFO * const above_mi = xd->mi_8x8[-xd->mode_info_stride]; + const MODE_INFO * const left_mi = xd->mi_8x8[-1]; + const int left_in_image = xd->left_available && left_mi; + const int above_in_image = xd->up_available && above_mi; // Note: // The mode info data structure has a one element border above and to the // left of the entries correpsonding to real macroblocks. // The prediction flags in these dummy entries are initialised to 0. // left - const int left_mv_pred = is_inter_mode(left_mbmi->mode); - const int left_interp = left_in_image && left_mv_pred ? - vp9_switchable_interp_map[left_mbmi->interp_filter] : - VP9_SWITCHABLE_FILTERS; + const int left_mv_pred = left_in_image ? is_inter_mode(left_mi->mbmi.mode) + : 0; + const int left_interp = left_in_image && left_mv_pred + ? left_mi->mbmi.interp_filter + : SWITCHABLE_FILTERS; // above - const int above_mv_pred = is_inter_mode(above_mbmi->mode); - const int above_interp = above_in_image && above_mv_pred ? - vp9_switchable_interp_map[above_mbmi->interp_filter] : - VP9_SWITCHABLE_FILTERS; - - assert(left_interp != -1); - assert(above_interp != -1); + const int above_mv_pred = above_in_image ? is_inter_mode(above_mi->mbmi.mode) + : 0; + const int above_interp = above_in_image && above_mv_pred + ? above_mi->mbmi.interp_filter + : SWITCHABLE_FILTERS; if (left_interp == above_interp) return left_interp; - else if (left_interp == VP9_SWITCHABLE_FILTERS && - above_interp != VP9_SWITCHABLE_FILTERS) + else if (left_interp == SWITCHABLE_FILTERS && + above_interp != SWITCHABLE_FILTERS) return above_interp; - else if (left_interp != VP9_SWITCHABLE_FILTERS && - above_interp == VP9_SWITCHABLE_FILTERS) + else if (left_interp != SWITCHABLE_FILTERS && + above_interp == SWITCHABLE_FILTERS) return left_interp; else - return VP9_SWITCHABLE_FILTERS; + return SWITCHABLE_FILTERS; } // Returns a context number for the given MB prediction signal unsigned char vp9_get_pred_context_intra_inter(const MACROBLOCKD *xd) { - const MODE_INFO *const mi = xd->mode_info_context; - const MB_MODE_INFO *const above_mbmi = &mi[-xd->mode_info_stride].mbmi; - const MB_MODE_INFO *const left_mbmi = &mi[-1].mbmi; - const int left_in_image = xd->left_available && left_mbmi->mb_in_image; - const int above_in_image = xd->up_available && above_mbmi->mb_in_image; - const int left_intra = !is_inter_block(left_mbmi); - const int above_intra = !is_inter_block(above_mbmi); + const MODE_INFO * const above_mi = xd->mi_8x8[-xd->mode_info_stride]; + const MODE_INFO * const left_mi = xd->mi_8x8[-1]; + const MB_MODE_INFO *const above_mbmi = above_mi ? &above_mi->mbmi : 0; + const MB_MODE_INFO *const left_mbmi = left_mi ? &left_mi->mbmi : 0; + const int left_in_image = xd->left_available && left_mi; + const int above_in_image = xd->up_available && above_mi; + const int left_intra = left_in_image ? !is_inter_block(left_mbmi) : 1; + const int above_intra = above_in_image ? !is_inter_block(above_mbmi) : 1; // The mode info data structure has a one element border above and to the // left of the entries corresponding to real macroblocks. @@ -82,35 +81,35 @@ unsigned char vp9_get_pred_context_intra_inter(const MACROBLOCKD *xd) { unsigned char vp9_get_pred_context_comp_inter_inter(const VP9_COMMON *cm, const MACROBLOCKD *xd) { int pred_context; - const MODE_INFO *const mi = xd->mode_info_context; - const MB_MODE_INFO *const above_mbmi = &mi[-cm->mode_info_stride].mbmi; - const MB_MODE_INFO *const left_mbmi = &mi[-1].mbmi; - const int left_in_image = xd->left_available && left_mbmi->mb_in_image; - const int above_in_image = xd->up_available && above_mbmi->mb_in_image; + const MODE_INFO * const above_mi = xd->mi_8x8[-xd->mode_info_stride]; + const MODE_INFO * const left_mi = xd->mi_8x8[-1]; + const MB_MODE_INFO *const above_mbmi = above_mi ? &above_mi->mbmi : 0; + const MB_MODE_INFO *const left_mbmi = left_mi ? &left_mi->mbmi : 0; + const int left_in_image = xd->left_available && left_mi; + const int above_in_image = xd->up_available && above_mi; // Note: // The mode info data structure has a one element border above and to the // left of the entries correpsonding to real macroblocks. // The prediction flags in these dummy entries are initialised to 0. if (above_in_image && left_in_image) { // both edges available - if (above_mbmi->ref_frame[1] <= INTRA_FRAME && - left_mbmi->ref_frame[1] <= INTRA_FRAME) + if (!has_second_ref(above_mbmi) && !has_second_ref(left_mbmi)) // neither edge uses comp pred (0/1) pred_context = (above_mbmi->ref_frame[0] == cm->comp_fixed_ref) ^ (left_mbmi->ref_frame[0] == cm->comp_fixed_ref); - else if (above_mbmi->ref_frame[1] <= INTRA_FRAME) + else if (!has_second_ref(above_mbmi)) // one of two edges uses comp pred (2/3) pred_context = 2 + (above_mbmi->ref_frame[0] == cm->comp_fixed_ref || - above_mbmi->ref_frame[0] == INTRA_FRAME); - else if (left_mbmi->ref_frame[1] <= INTRA_FRAME) + !is_inter_block(above_mbmi)); + else if (!has_second_ref(left_mbmi)) // one of two edges uses comp pred (2/3) pred_context = 2 + (left_mbmi->ref_frame[0] == cm->comp_fixed_ref || - left_mbmi->ref_frame[0] == INTRA_FRAME); + !is_inter_block(left_mbmi)); else // both edges use comp pred (4) pred_context = 4; } else if (above_in_image || left_in_image) { // one edge available const MB_MODE_INFO *edge_mbmi = above_in_image ? above_mbmi : left_mbmi; - if (edge_mbmi->ref_frame[1] <= INTRA_FRAME) + if (!has_second_ref(edge_mbmi)) // edge does not use comp pred (0/1) pred_context = edge_mbmi->ref_frame[0] == cm->comp_fixed_ref; else @@ -127,11 +126,14 @@ unsigned char vp9_get_pred_context_comp_inter_inter(const VP9_COMMON *cm, unsigned char vp9_get_pred_context_comp_ref_p(const VP9_COMMON *cm, const MACROBLOCKD *xd) { int pred_context; - const MODE_INFO *const mi = xd->mode_info_context; - const MB_MODE_INFO *const above_mbmi = &mi[-cm->mode_info_stride].mbmi; - const MB_MODE_INFO *const left_mbmi = &mi[-1].mbmi; - const int left_in_image = xd->left_available && left_mbmi->mb_in_image; - const int above_in_image = xd->up_available && above_mbmi->mb_in_image; + const MODE_INFO * const above_mi = xd->mi_8x8[-cm->mode_info_stride]; + const MODE_INFO * const left_mi = xd->mi_8x8[-1]; + const MB_MODE_INFO *const above_mbmi = above_mi ? &above_mi->mbmi : 0; + const MB_MODE_INFO *const left_mbmi = left_mi ? &left_mi->mbmi : 0; + const int left_in_image = xd->left_available && left_mi; + const int above_in_image = xd->up_available && above_mi; + const int left_intra = left_in_image ? !is_inter_block(left_mbmi) : 1; + const int above_intra = above_in_image ? !is_inter_block(above_mbmi) : 1; // Note: // The mode info data structure has a one element border above and to the // left of the entries correpsonding to real macroblocks. @@ -140,22 +142,19 @@ unsigned char vp9_get_pred_context_comp_ref_p(const VP9_COMMON *cm, const int var_ref_idx = !fix_ref_idx; if (above_in_image && left_in_image) { // both edges available - if (above_mbmi->ref_frame[0] == INTRA_FRAME && - left_mbmi->ref_frame[0] == INTRA_FRAME) { // intra/intra (2) + if (above_intra && left_intra) { // intra/intra (2) pred_context = 2; - } else if (above_mbmi->ref_frame[0] == INTRA_FRAME || - left_mbmi->ref_frame[0] == INTRA_FRAME) { // intra/inter - const MB_MODE_INFO *edge_mbmi = above_mbmi->ref_frame[0] == INTRA_FRAME ? - left_mbmi : above_mbmi; + } else if (above_intra || left_intra) { // intra/inter + const MB_MODE_INFO *edge_mbmi = above_intra ? left_mbmi : above_mbmi; - if (edge_mbmi->ref_frame[1] <= INTRA_FRAME) // single pred (1/3) + if (!has_second_ref(edge_mbmi)) // single pred (1/3) pred_context = 1 + 2 * (edge_mbmi->ref_frame[0] != cm->comp_var_ref[1]); else // comp pred (1/3) pred_context = 1 + 2 * (edge_mbmi->ref_frame[var_ref_idx] != cm->comp_var_ref[1]); } else { // inter/inter - int l_sg = left_mbmi->ref_frame[1] <= INTRA_FRAME; - int a_sg = above_mbmi->ref_frame[1] <= INTRA_FRAME; + const int l_sg = !has_second_ref(left_mbmi); + const int a_sg = !has_second_ref(above_mbmi); MV_REFERENCE_FRAME vrfa = a_sg ? above_mbmi->ref_frame[0] : above_mbmi->ref_frame[var_ref_idx]; MV_REFERENCE_FRAME vrfl = l_sg ? left_mbmi->ref_frame[0] @@ -189,13 +188,15 @@ unsigned char vp9_get_pred_context_comp_ref_p(const VP9_COMMON *cm, } else if (above_in_image || left_in_image) { // one edge available const MB_MODE_INFO *edge_mbmi = above_in_image ? above_mbmi : left_mbmi; - if (edge_mbmi->ref_frame[0] == INTRA_FRAME) + if (!is_inter_block(edge_mbmi)) { pred_context = 2; - else if (edge_mbmi->ref_frame[1] > INTRA_FRAME) - pred_context = 4 * (edge_mbmi->ref_frame[var_ref_idx] + } else { + if (has_second_ref(edge_mbmi)) + pred_context = 4 * (edge_mbmi->ref_frame[var_ref_idx] != cm->comp_var_ref[1]); - else - pred_context = 3 * (edge_mbmi->ref_frame[0] != cm->comp_var_ref[1]); + else + pred_context = 3 * (edge_mbmi->ref_frame[0] != cm->comp_var_ref[1]); + } } else { // no edges available (2) pred_context = 2; } @@ -205,91 +206,91 @@ unsigned char vp9_get_pred_context_comp_ref_p(const VP9_COMMON *cm, } unsigned char vp9_get_pred_context_single_ref_p1(const MACROBLOCKD *xd) { int pred_context; - const MODE_INFO *const mi = xd->mode_info_context; - const MB_MODE_INFO *const above_mbmi = &mi[-xd->mode_info_stride].mbmi; - const MB_MODE_INFO *const left_mbmi = &mi[-1].mbmi; - const int left_in_image = xd->left_available && left_mbmi->mb_in_image; - const int above_in_image = xd->up_available && above_mbmi->mb_in_image; + const MODE_INFO * const above_mi = xd->mi_8x8[-xd->mode_info_stride]; + const MODE_INFO * const left_mi = xd->mi_8x8[-1]; + const MB_MODE_INFO *const above_mbmi = above_mi ? &above_mi->mbmi : 0; + const MB_MODE_INFO *const left_mbmi = left_mi ? &left_mi->mbmi : 0; + const int left_in_image = xd->left_available && left_mi; + const int above_in_image = xd->up_available && above_mi; + const int left_intra = left_in_image ? !is_inter_block(left_mbmi) : 1; + const int above_intra = above_in_image ? !is_inter_block(above_mbmi) : 1; // Note: // The mode info data structure has a one element border above and to the // left of the entries correpsonding to real macroblocks. // The prediction flags in these dummy entries are initialised to 0. if (above_in_image && left_in_image) { // both edges available - if (above_mbmi->ref_frame[0] == INTRA_FRAME && - left_mbmi->ref_frame[0] == INTRA_FRAME) { + if (above_intra && left_intra) { // intra/intra pred_context = 2; - } else if (above_mbmi->ref_frame[0] == INTRA_FRAME || - left_mbmi->ref_frame[0] == INTRA_FRAME) { - const MB_MODE_INFO *edge_mbmi = above_mbmi->ref_frame[0] == INTRA_FRAME ? - left_mbmi : above_mbmi; - - if (edge_mbmi->ref_frame[1] <= INTRA_FRAME) + } else if (above_intra || left_intra) { // intra/inter or inter/intra + const MB_MODE_INFO *edge_mbmi = above_intra ? left_mbmi : above_mbmi; + if (!has_second_ref(edge_mbmi)) pred_context = 4 * (edge_mbmi->ref_frame[0] == LAST_FRAME); else pred_context = 1 + (edge_mbmi->ref_frame[0] == LAST_FRAME || edge_mbmi->ref_frame[1] == LAST_FRAME); - } else if (above_mbmi->ref_frame[1] <= INTRA_FRAME && - left_mbmi->ref_frame[1] <= INTRA_FRAME) { - pred_context = 2 * (above_mbmi->ref_frame[0] == LAST_FRAME) + - 2 * (left_mbmi->ref_frame[0] == LAST_FRAME); - } else if (above_mbmi->ref_frame[1] > INTRA_FRAME && - left_mbmi->ref_frame[1] > INTRA_FRAME) { - pred_context = 1 + (above_mbmi->ref_frame[0] == LAST_FRAME || - above_mbmi->ref_frame[1] == LAST_FRAME || - left_mbmi->ref_frame[0] == LAST_FRAME || - left_mbmi->ref_frame[1] == LAST_FRAME); - } else { - MV_REFERENCE_FRAME rfs = above_mbmi->ref_frame[1] <= INTRA_FRAME ? - above_mbmi->ref_frame[0] : left_mbmi->ref_frame[0]; - MV_REFERENCE_FRAME crf1 = above_mbmi->ref_frame[1] > INTRA_FRAME ? - above_mbmi->ref_frame[0] : left_mbmi->ref_frame[0]; - MV_REFERENCE_FRAME crf2 = above_mbmi->ref_frame[1] > INTRA_FRAME ? - above_mbmi->ref_frame[1] : left_mbmi->ref_frame[1]; - - if (rfs == LAST_FRAME) - pred_context = 3 + (crf1 == LAST_FRAME || crf2 == LAST_FRAME); - else - pred_context = crf1 == LAST_FRAME || crf2 == LAST_FRAME; + } else { // inter/inter + if (!has_second_ref(above_mbmi) && !has_second_ref(left_mbmi)) { + pred_context = 2 * (above_mbmi->ref_frame[0] == LAST_FRAME) + + 2 * (left_mbmi->ref_frame[0] == LAST_FRAME); + } else if (has_second_ref(above_mbmi) && has_second_ref(left_mbmi)) { + pred_context = 1 + (above_mbmi->ref_frame[0] == LAST_FRAME || + above_mbmi->ref_frame[1] == LAST_FRAME || + left_mbmi->ref_frame[0] == LAST_FRAME || + left_mbmi->ref_frame[1] == LAST_FRAME); + } else { + const MV_REFERENCE_FRAME rfs = !has_second_ref(above_mbmi) ? + above_mbmi->ref_frame[0] : left_mbmi->ref_frame[0]; + const MV_REFERENCE_FRAME crf1 = has_second_ref(above_mbmi) ? + above_mbmi->ref_frame[0] : left_mbmi->ref_frame[0]; + const MV_REFERENCE_FRAME crf2 = has_second_ref(above_mbmi) ? + above_mbmi->ref_frame[1] : left_mbmi->ref_frame[1]; + + if (rfs == LAST_FRAME) + pred_context = 3 + (crf1 == LAST_FRAME || crf2 == LAST_FRAME); + else + pred_context = crf1 == LAST_FRAME || crf2 == LAST_FRAME; + } } } else if (above_in_image || left_in_image) { // one edge available const MB_MODE_INFO *edge_mbmi = above_in_image ? above_mbmi : left_mbmi; - - if (edge_mbmi->ref_frame[0] == INTRA_FRAME) + if (!is_inter_block(edge_mbmi)) { // intra pred_context = 2; - else if (edge_mbmi->ref_frame[1] <= INTRA_FRAME) - pred_context = 4 * (edge_mbmi->ref_frame[0] == LAST_FRAME); - else - pred_context = 1 + (edge_mbmi->ref_frame[0] == LAST_FRAME || - edge_mbmi->ref_frame[1] == LAST_FRAME); - } else { // no edges available (2) + } else { // inter + if (!has_second_ref(edge_mbmi)) + pred_context = 4 * (edge_mbmi->ref_frame[0] == LAST_FRAME); + else + pred_context = 1 + (edge_mbmi->ref_frame[0] == LAST_FRAME || + edge_mbmi->ref_frame[1] == LAST_FRAME); + } + } else { // no edges available pred_context = 2; } + assert(pred_context >= 0 && pred_context < REF_CONTEXTS); return pred_context; } unsigned char vp9_get_pred_context_single_ref_p2(const MACROBLOCKD *xd) { int pred_context; - const MODE_INFO *const mi = xd->mode_info_context; - const MB_MODE_INFO *const above_mbmi = &mi[-xd->mode_info_stride].mbmi; - const MB_MODE_INFO *const left_mbmi = &mi[-1].mbmi; - const int left_in_image = xd->left_available && left_mbmi->mb_in_image; - const int above_in_image = xd->up_available && above_mbmi->mb_in_image; + const MODE_INFO * const above_mi = xd->mi_8x8[-xd->mode_info_stride]; + const MODE_INFO * const left_mi = xd->mi_8x8[-1]; + const MB_MODE_INFO *const above_mbmi = above_mi ? &above_mi->mbmi : 0; + const MB_MODE_INFO *const left_mbmi = left_mi ? &left_mi->mbmi : 0; + const int left_in_image = xd->left_available && left_mi; + const int above_in_image = xd->up_available && above_mi; + const int left_intra = left_in_image ? !is_inter_block(left_mbmi) : 1; + const int above_intra = above_in_image ? !is_inter_block(above_mbmi) : 1; // Note: // The mode info data structure has a one element border above and to the // left of the entries correpsonding to real macroblocks. // The prediction flags in these dummy entries are initialised to 0. if (above_in_image && left_in_image) { // both edges available - if (above_mbmi->ref_frame[0] == INTRA_FRAME && - left_mbmi->ref_frame[0] == INTRA_FRAME) { + if (above_intra && left_intra) { // intra/intra pred_context = 2; - } else if (above_mbmi->ref_frame[0] == INTRA_FRAME || - left_mbmi->ref_frame[0] == INTRA_FRAME) { - const MB_MODE_INFO *edge_mbmi = above_mbmi->ref_frame[0] == INTRA_FRAME ? - left_mbmi : above_mbmi; - - if (edge_mbmi->ref_frame[1] <= INTRA_FRAME) { + } else if (above_intra || left_intra) { // intra/inter or inter/intra + const MB_MODE_INFO *edge_mbmi = above_intra ? left_mbmi : above_mbmi; + if (!has_second_ref(edge_mbmi)) { if (edge_mbmi->ref_frame[0] == LAST_FRAME) pred_context = 3; else @@ -298,54 +299,53 @@ unsigned char vp9_get_pred_context_single_ref_p2(const MACROBLOCKD *xd) { pred_context = 1 + 2 * (edge_mbmi->ref_frame[0] == GOLDEN_FRAME || edge_mbmi->ref_frame[1] == GOLDEN_FRAME); } - } else if (above_mbmi->ref_frame[1] <= INTRA_FRAME && - left_mbmi->ref_frame[1] <= INTRA_FRAME) { - if (above_mbmi->ref_frame[0] == LAST_FRAME && - left_mbmi->ref_frame[0] == LAST_FRAME) { - pred_context = 3; - } else if (above_mbmi->ref_frame[0] == LAST_FRAME || - left_mbmi->ref_frame[0] == LAST_FRAME) { - const MB_MODE_INFO *edge_mbmi = above_mbmi->ref_frame[0] == LAST_FRAME ? - left_mbmi : above_mbmi; + } else { // inter/inter + if (!has_second_ref(above_mbmi) && !has_second_ref(left_mbmi)) { + if (above_mbmi->ref_frame[0] == LAST_FRAME && + left_mbmi->ref_frame[0] == LAST_FRAME) { + pred_context = 3; + } else if (above_mbmi->ref_frame[0] == LAST_FRAME || + left_mbmi->ref_frame[0] == LAST_FRAME) { + const MB_MODE_INFO *edge_mbmi = + above_mbmi->ref_frame[0] == LAST_FRAME ? left_mbmi : above_mbmi; - pred_context = 4 * (edge_mbmi->ref_frame[0] == GOLDEN_FRAME); + pred_context = 4 * (edge_mbmi->ref_frame[0] == GOLDEN_FRAME); + } else { + pred_context = 2 * (above_mbmi->ref_frame[0] == GOLDEN_FRAME) + + 2 * (left_mbmi->ref_frame[0] == GOLDEN_FRAME); + } + } else if (has_second_ref(above_mbmi) && has_second_ref(left_mbmi)) { + if (above_mbmi->ref_frame[0] == left_mbmi->ref_frame[0] && + above_mbmi->ref_frame[1] == left_mbmi->ref_frame[1]) + pred_context = 3 * (above_mbmi->ref_frame[0] == GOLDEN_FRAME || + above_mbmi->ref_frame[1] == GOLDEN_FRAME || + left_mbmi->ref_frame[0] == GOLDEN_FRAME || + left_mbmi->ref_frame[1] == GOLDEN_FRAME); + else + pred_context = 2; } else { - pred_context = 2 * (above_mbmi->ref_frame[0] == GOLDEN_FRAME) + - 2 * (left_mbmi->ref_frame[0] == GOLDEN_FRAME); + const MV_REFERENCE_FRAME rfs = !has_second_ref(above_mbmi) ? + above_mbmi->ref_frame[0] : left_mbmi->ref_frame[0]; + const MV_REFERENCE_FRAME crf1 = has_second_ref(above_mbmi) ? + above_mbmi->ref_frame[0] : left_mbmi->ref_frame[0]; + const MV_REFERENCE_FRAME crf2 = has_second_ref(above_mbmi) ? + above_mbmi->ref_frame[1] : left_mbmi->ref_frame[1]; + + if (rfs == GOLDEN_FRAME) + pred_context = 3 + (crf1 == GOLDEN_FRAME || crf2 == GOLDEN_FRAME); + else if (rfs == ALTREF_FRAME) + pred_context = crf1 == GOLDEN_FRAME || crf2 == GOLDEN_FRAME; + else + pred_context = 1 + 2 * (crf1 == GOLDEN_FRAME || crf2 == GOLDEN_FRAME); } - } else if (above_mbmi->ref_frame[1] > INTRA_FRAME && - left_mbmi->ref_frame[1] > INTRA_FRAME) { - if (above_mbmi->ref_frame[0] == left_mbmi->ref_frame[0] && - above_mbmi->ref_frame[1] == left_mbmi->ref_frame[1]) - pred_context = 3 * (above_mbmi->ref_frame[0] == GOLDEN_FRAME || - above_mbmi->ref_frame[1] == GOLDEN_FRAME || - left_mbmi->ref_frame[0] == GOLDEN_FRAME || - left_mbmi->ref_frame[1] == GOLDEN_FRAME); - else - pred_context = 2; - } else { - MV_REFERENCE_FRAME rfs = above_mbmi->ref_frame[1] <= INTRA_FRAME ? - above_mbmi->ref_frame[0] : left_mbmi->ref_frame[0]; - MV_REFERENCE_FRAME crf1 = above_mbmi->ref_frame[1] > INTRA_FRAME ? - above_mbmi->ref_frame[0] : left_mbmi->ref_frame[0]; - MV_REFERENCE_FRAME crf2 = above_mbmi->ref_frame[1] > INTRA_FRAME ? - above_mbmi->ref_frame[1] : left_mbmi->ref_frame[1]; - - if (rfs == GOLDEN_FRAME) - pred_context = 3 + (crf1 == GOLDEN_FRAME || crf2 == GOLDEN_FRAME); - else if (rfs == ALTREF_FRAME) - pred_context = crf1 == GOLDEN_FRAME || crf2 == GOLDEN_FRAME; - else - pred_context = 1 + 2 * (crf1 == GOLDEN_FRAME || crf2 == GOLDEN_FRAME); } } else if (above_in_image || left_in_image) { // one edge available const MB_MODE_INFO *edge_mbmi = above_in_image ? above_mbmi : left_mbmi; - if (edge_mbmi->ref_frame[0] == INTRA_FRAME || - (edge_mbmi->ref_frame[0] == LAST_FRAME && - edge_mbmi->ref_frame[1] <= INTRA_FRAME)) + if (!is_inter_block(edge_mbmi) || + (edge_mbmi->ref_frame[0] == LAST_FRAME && !has_second_ref(edge_mbmi))) pred_context = 2; - else if (edge_mbmi->ref_frame[1] <= INTRA_FRAME) + else if (!has_second_ref(edge_mbmi)) pred_context = 4 * (edge_mbmi->ref_frame[0] == GOLDEN_FRAME); else pred_context = 3 * (edge_mbmi->ref_frame[0] == GOLDEN_FRAME || @@ -361,22 +361,23 @@ unsigned char vp9_get_pred_context_single_ref_p2(const MACROBLOCKD *xd) { // left of the entries corresponding to real blocks. // The prediction flags in these dummy entries are initialized to 0. unsigned char vp9_get_pred_context_tx_size(const MACROBLOCKD *xd) { - const MODE_INFO *const mi = xd->mode_info_context; - const MB_MODE_INFO *const above_mbmi = &mi[-xd->mode_info_stride].mbmi; - const MB_MODE_INFO *const left_mbmi = &mi[-1].mbmi; - const int left_in_image = xd->left_available && left_mbmi->mb_in_image; - const int above_in_image = xd->up_available && above_mbmi->mb_in_image; - const int max_tx_size = max_txsize_lookup[mi->mbmi.sb_type]; + const MODE_INFO * const above_mi = xd->mi_8x8[-xd->mode_info_stride]; + const MODE_INFO * const left_mi = xd->mi_8x8[-1]; + const MB_MODE_INFO *const above_mbmi = above_mi ? &above_mi->mbmi : 0; + const MB_MODE_INFO *const left_mbmi = left_mi ? &left_mi->mbmi : 0; + const int left_in_image = xd->left_available && left_mi; + const int above_in_image = xd->up_available && above_mi; + const int max_tx_size = max_txsize_lookup[xd->mi_8x8[0]->mbmi.sb_type]; int above_context = max_tx_size; int left_context = max_tx_size; if (above_in_image) - above_context = above_mbmi->mb_skip_coeff ? max_tx_size - : above_mbmi->txfm_size; + above_context = above_mbmi->skip_coeff ? max_tx_size + : above_mbmi->tx_size; if (left_in_image) - left_context = left_mbmi->mb_skip_coeff ? max_tx_size - : left_mbmi->txfm_size; + left_context = left_mbmi->skip_coeff ? max_tx_size + : left_mbmi->tx_size; if (!left_in_image) left_context = above_context; @@ -387,36 +388,17 @@ unsigned char vp9_get_pred_context_tx_size(const MACROBLOCKD *xd) { return above_context + left_context > max_tx_size; } -void vp9_set_pred_flag_seg_id(VP9_COMMON *cm, BLOCK_SIZE_TYPE bsize, - int mi_row, int mi_col, uint8_t pred_flag) { - MODE_INFO *mi = &cm->mi[mi_row * cm->mode_info_stride + mi_col]; - const int bw = 1 << mi_width_log2(bsize); - const int bh = 1 << mi_height_log2(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++) - mi[y * cm->mode_info_stride + x].mbmi.seg_id_predicted = pred_flag; +void vp9_set_pred_flag_seg_id(MACROBLOCKD *xd, uint8_t pred_flag) { + xd->this_mi->mbmi.seg_id_predicted = pred_flag; } -void vp9_set_pred_flag_mbskip(VP9_COMMON *cm, BLOCK_SIZE_TYPE bsize, - int mi_row, int mi_col, uint8_t pred_flag) { - MODE_INFO *mi = &cm->mi[mi_row * cm->mode_info_stride + mi_col]; - const int bw = 1 << mi_width_log2(bsize); - const int bh = 1 << mi_height_log2(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++) - mi[y * cm->mode_info_stride + x].mbmi.mb_skip_coeff = pred_flag; +void vp9_set_pred_flag_mbskip(MACROBLOCKD *xd, BLOCK_SIZE bsize, + uint8_t pred_flag) { + xd->this_mi->mbmi.skip_coeff = pred_flag; } int vp9_get_segment_id(VP9_COMMON *cm, const uint8_t *segment_ids, - BLOCK_SIZE_TYPE bsize, int mi_row, int mi_col) { + BLOCK_SIZE bsize, int mi_row, int mi_col) { const int mi_offset = mi_row * cm->mi_cols + mi_col; const int bw = 1 << mi_width_log2(bsize); const int bh = 1 << mi_height_log2(bsize); |