diff --git a/vp9/decoder/vp9_decodemv.c b/vp9/decoder/vp9_decodemv.c index 55c67374abd2dee2ed937d813882b23f2f5db682..6c3aad5f484802dc754f285fcf84bf51fb094a8f 100644 --- a/vp9/decoder/vp9_decodemv.c +++ b/vp9/decoder/vp9_decodemv.c @@ -968,7 +968,12 @@ static void read_mb_modes_mv(VP9D_COMP *pbi, MODE_INFO *mi, MB_MODE_INFO *mbmi, // required for left and above block mv mv0->as_int = 0; - if (mbmi->sb_type > BLOCK_SIZE_MB16X16) { +#if CONFIG_SB8X8 + if (mbmi->sb_type > BLOCK_SIZE_SB8X8) +#else + if (mbmi->sb_type > BLOCK_SIZE_MB16X16) +#endif + { mbmi->mode = read_sb_ymode(r, cm->fc.sb_ymode_prob); cm->fc.sb_ymode_counts[mbmi->mode]++; } else { @@ -986,7 +991,7 @@ static void read_mb_modes_mv(VP9D_COMP *pbi, MODE_INFO *mi, MB_MODE_INFO *mbmi, if (m == B_CONTEXT_PRED) m -= CONTEXT_PRED_REPLACEMENTS; #endif cm->fc.bmode_counts[m]++; - } while (++j < 16); + } while (++j < (16 >> (2 * CONFIG_SB8X8))); } #if !CONFIG_SB8X8 diff --git a/vp9/encoder/vp9_bitstream.c b/vp9/encoder/vp9_bitstream.c index f8e273364f8a2b37befed37aee5d9edf90badee6..197a4993940bbc97f1bff9f39cba2dbada5f5aac 100644 --- a/vp9/encoder/vp9_bitstream.c +++ b/vp9/encoder/vp9_bitstream.c @@ -722,7 +722,11 @@ static void pack_inter_mode_mvs(VP9_COMP *cpi, MODE_INFO *m, active_section = 6; #endif +#if CONFIG_SB8X8 + if (m->mbmi.sb_type > BLOCK_SIZE_SB8X8) +#else if (m->mbmi.sb_type > BLOCK_SIZE_MB16X16) +#endif write_sb_ymode(bc, mode, pc->fc.sb_ymode_prob); else write_ymode(bc, mode, pc->fc.ymode_prob); diff --git a/vp9/encoder/vp9_encodeframe.c b/vp9/encoder/vp9_encodeframe.c index 0642c31bdc5785d72fdaa5d50e940240bcbe95a8..9e77f8125c4fb637a703663560d1c2b46e7b41eb 100644 --- a/vp9/encoder/vp9_encodeframe.c +++ b/vp9/encoder/vp9_encodeframe.c @@ -2073,7 +2073,12 @@ static void sum_intra_stats(VP9_COMP *cpi, MACROBLOCK *x) { #endif #endif - if (xd->mode_info_context->mbmi.sb_type > BLOCK_SIZE_MB16X16) { +#if CONFIG_SB8X8 + if (xd->mode_info_context->mbmi.sb_type > BLOCK_SIZE_SB8X8) +#else + if (xd->mode_info_context->mbmi.sb_type > BLOCK_SIZE_MB16X16) +#endif + { ++cpi->sb_ymode_count[m]; } else { ++cpi->ymode_count[m];