shithub: libvpx

Info  •  Files  •  Log  •  Branches

Eliminated frame_type check in get_partition_probs()

Moved the frame_type check to the tile level and stored
the prob ptr in MACROBLOCKD.

Change-Id: I10b5a4abd58213dc7610e3ade1a1583c01526842

--- a/vp9/common/vp9_blockd.h

+++ b/vp9/common/vp9_blockd.h

@@ -156,6 +156,8 @@

   int up_available;

   int left_available;

+  const vp9_prob (*partition_probs)[PARTITION_TYPES - 1];

+

   /* Distance of MB away from frame edges */

   int mb_to_left_edge;

   int mb_to_right_edge;

--- a/vp9/common/vp9_onyxc_int.h

+++ b/vp9/common/vp9_onyxc_int.h

@@ -335,6 +335,18 @@

   return ALIGN_POWER_OF_TWO(n_mis, MI_BLOCK_SIZE_LOG2);

 }

+static INLINE int frame_is_intra_only(const VP9_COMMON *const cm) {

+  return cm->frame_type == KEY_FRAME || cm->intra_only;

+}

+

+static INLINE void set_partition_probs(const VP9_COMMON *const cm,

+                                       MACROBLOCKD *const xd) {

+  xd->partition_probs =

+      frame_is_intra_only(cm) ?

+          &vp9_kf_partition_probs[0] :

+          (const vp9_prob (*)[PARTITION_TYPES - 1])cm->fc->partition_prob;

+}

+

 static INLINE void init_macroblockd(VP9_COMMON *cm, MACROBLOCKD *xd) {

   int i;

@@ -355,16 +367,13 @@

   xd->above_seg_context = cm->above_seg_context;

   xd->mi_stride = cm->mi_stride;

   xd->error_info = &cm->error;

-}

-static INLINE int frame_is_intra_only(const VP9_COMMON *const cm) {

-  return cm->frame_type == KEY_FRAME || cm->intra_only;

+  set_partition_probs(cm, xd);

 }

-static INLINE const vp9_prob* get_partition_probs(const VP9_COMMON *cm,

+static INLINE const vp9_prob* get_partition_probs(const MACROBLOCKD *xd,

                                                   int ctx) {

-  return frame_is_intra_only(cm) ? vp9_kf_partition_probs[ctx]

-                                 : cm->fc->partition_prob[ctx];

+  return xd->partition_probs[ctx];

 }

 static INLINE void set_skip_context(MACROBLOCKD *xd, int mi_row, int mi_col) {

--- a/vp9/decoder/vp9_decodeframe.c

+++ b/vp9/decoder/vp9_decodeframe.c

@@ -425,7 +425,7 @@

                                      int mi_row, int mi_col, BLOCK_SIZE bsize,

                                      vp9_reader *r) {

   const int ctx = partition_plane_context(xd, mi_row, mi_col, bsize);

-  const vp9_prob *const probs = get_partition_probs(cm, ctx);

+  const vp9_prob *const probs = get_partition_probs(xd, ctx);

   const int has_rows = (mi_row + hbs) < cm->mi_rows;

   const int has_cols = (mi_col + hbs) < cm->mi_cols;

   FRAME_COUNTS *counts = xd->counts;

--- a/vp9/encoder/vp9_bitstream.c

+++ b/vp9/encoder/vp9_bitstream.c

@@ -403,7 +403,7 @@

                             int hbs, int mi_row, int mi_col,

                             PARTITION_TYPE p, BLOCK_SIZE bsize, vp9_writer *w) {

   const int ctx = partition_plane_context(xd, mi_row, mi_col, bsize);

-  const vp9_prob *const probs = get_partition_probs(cm, ctx);

+  const vp9_prob *const probs = xd->partition_probs[ctx];

   const int has_rows = (mi_row + hbs) < cm->mi_rows;

   const int has_cols = (mi_col + hbs) < cm->mi_cols;

@@ -481,8 +481,11 @@

 static void write_modes(VP9_COMP *cpi,

                         const TileInfo *const tile, vp9_writer *w,

                         TOKENEXTRA **tok, const TOKENEXTRA *const tok_end) {

+  const VP9_COMMON *const cm = &cpi->common;

   MACROBLOCKD *const xd = &cpi->td.mb.e_mbd;

   int mi_row, mi_col;

+

+  set_partition_probs(cm, xd);

   for (mi_row = tile->mi_row_start; mi_row < tile->mi_row_end;

        mi_row += MI_BLOCK_SIZE) {

--- a/vp9/encoder/vp9_rd.c

+++ b/vp9/encoder/vp9_rd.c

@@ -265,6 +265,7 @@

 void vp9_initialize_rd_consts(VP9_COMP *cpi) {

   VP9_COMMON *const cm = &cpi->common;

   MACROBLOCK *const x = &cpi->td.mb;

+  MACROBLOCKD *const xd = &cpi->td.mb.e_mbd;

   RD_OPT *const rd = &cpi->rd;

   int i;

@@ -280,6 +281,7 @@

                        cm->frame_type != KEY_FRAME) ? 0 : 1;

   set_block_thresholds(cm, rd);

+  set_partition_probs(cm, xd);

   if (!cpi->sf.use_nonrd_pick_mode || cm->frame_type == KEY_FRAME)

     fill_token_costs(x->token_costs, cm->fc->coef_probs);

@@ -287,7 +289,7 @@

   if (cpi->sf.partition_search_type != VAR_BASED_PARTITION ||

       cm->frame_type == KEY_FRAME) {

     for (i = 0; i < PARTITION_CONTEXTS; ++i)

-      vp9_cost_tokens(cpi->partition_cost[i], get_partition_probs(cm, i),

+      vp9_cost_tokens(cpi->partition_cost[i], get_partition_probs(xd, i),

                       vp9_partition_tree);

   }