| Index: source/libvpx/vp9/common/vp9_pred_common.c
|
| ===================================================================
|
| --- source/libvpx/vp9/common/vp9_pred_common.c (revision 223100)
|
| +++ source/libvpx/vp9/common/vp9_pred_common.c (working copy)
|
| @@ -18,28 +18,28 @@
|
|
|
| // 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->in_image;
|
| - const int above_in_image = xd->up_available && above_mbmi->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_mv_pred = left_in_image ? is_inter_mode(left_mi->mbmi.mode)
|
| + : 0;
|
| const int left_interp = left_in_image && left_mv_pred
|
| - ? left_mbmi->interp_filter
|
| + ? left_mi->mbmi.interp_filter
|
| : SWITCHABLE_FILTERS;
|
|
|
| // above
|
| - const int above_mv_pred = is_inter_mode(above_mbmi->mode);
|
| + 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_mbmi->interp_filter
|
| + ? above_mi->mbmi.interp_filter
|
| : SWITCHABLE_FILTERS;
|
|
|
| -
|
| if (left_interp == above_interp)
|
| return left_interp;
|
| else if (left_interp == SWITCHABLE_FILTERS &&
|
| @@ -53,13 +53,14 @@
|
| }
|
| // 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->in_image;
|
| - const int above_in_image = xd->up_available && above_mbmi->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.
|
| @@ -80,11 +81,12 @@
|
| 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->in_image;
|
| - const int above_in_image = xd->up_available && above_mbmi->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.
|
| @@ -124,14 +126,14 @@
|
| 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->in_image;
|
| - const int above_in_image = xd->up_available && above_mbmi->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[-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.
|
| @@ -204,14 +206,14 @@
|
| }
|
| 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->in_image;
|
| - const int above_in_image = xd->up_available && above_mbmi->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;
|
| // Note:
|
| // The mode info data structure has a one element border above and to the
|
| // left of the entries correpsonding to real macroblocks.
|
| @@ -270,13 +272,14 @@
|
|
|
| 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->in_image;
|
| - const int above_in_image = xd->up_available && above_mbmi->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;
|
|
|
| // Note:
|
| // The mode info data structure has a one element border above and to the
|
| @@ -358,22 +361,23 @@
|
| // 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->in_image;
|
| - const int above_in_image = xd->up_available && above_mbmi->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->skip_coeff ? max_tx_size
|
| - : above_mbmi->txfm_size;
|
| + : above_mbmi->tx_size;
|
|
|
| if (left_in_image)
|
| left_context = left_mbmi->skip_coeff ? max_tx_size
|
| - : left_mbmi->txfm_size;
|
| + : left_mbmi->tx_size;
|
|
|
| if (!left_in_image)
|
| left_context = above_context;
|
| @@ -384,32 +388,13 @@
|
| return above_context + left_context > max_tx_size;
|
| }
|
|
|
| -void vp9_set_pred_flag_seg_id(VP9_COMMON *cm, BLOCK_SIZE 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 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.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,
|
|
|