vp8/common/entropymode.c

   1 /*
   2  *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
   3  *
   4  *  Use of this source code is governed by a BSD-style license
   5  *  that can be found in the LICENSE file in the root of the source
   6  *  tree. An additional intellectual property rights grant can be found
   7  *  in the file PATENTS.  All contributing project authors may
   8  *  be found in the AUTHORS file in the root of the source tree.
   9  */
  10
  11
  12 #include "entropymode.h"
  13 #include "entropy.h"
  14 #include "vpx_mem/vpx_mem.h"
  15
  16 static const unsigned int kf_y_mode_cts[VP8_YMODES] = { 1607, 915, 812, 811, 5455};
  17 static const unsigned int y_mode_cts  [VP8_YMODES] = { 8080, 1908, 1582, 1007, 5874};
  18
  19 static const unsigned int uv_mode_cts  [VP8_UV_MODES] = { 59483, 13605, 16492, 4230};
  20 static const unsigned int kf_uv_mode_cts[VP8_UV_MODES] = { 5319, 1904, 1703, 674};
  21
  22 static const unsigned int bmode_cts[VP8_BINTRAMODES] =
  23 {
  24     43891, 17694, 10036, 3920, 3363, 2546, 5119, 3221, 2471, 1723
  25 };
  26
  27 typedef enum
  28 {
  29     SUBMVREF_NORMAL,
  30     SUBMVREF_LEFT_ZED,
  31     SUBMVREF_ABOVE_ZED,
  32     SUBMVREF_LEFT_ABOVE_SAME,
  33     SUBMVREF_LEFT_ABOVE_ZED
  34 } sumvfref_t;
  35
  36 int vp8_mv_cont(const MV *l, const MV *a)
  37 {
  38     int lez = (l->row == 0 && l->col == 0);
  39     int aez = (a->row == 0 && a->col == 0);
  40     int lea = (l->row == a->row && l->col == a->col);
  41
  42     if (lea && lez)
  43         return SUBMVREF_LEFT_ABOVE_ZED;
  44
  45     if (lea)
  46         return SUBMVREF_LEFT_ABOVE_SAME;
  47
  48     if (aez)
  49         return SUBMVREF_ABOVE_ZED;
  50
  51     if (lez)
  52         return SUBMVREF_LEFT_ZED;
  53
  54     return SUBMVREF_NORMAL;
  55 }
  56
  57 static const vp8_prob sub_mv_ref_prob [VP8_SUBMVREFS-1] = { 180, 162, 25};
  58
  59 const vp8_prob vp8_sub_mv_ref_prob2 [SUBMVREF_COUNT][VP8_SUBMVREFS-1] =
  60 {
  61     { 147, 136, 18 },
  62     { 106, 145, 1  },
  63     { 179, 121, 1  },
  64     { 223, 1  , 34 },
  65     { 208, 1  , 1  }
  66 };
  67
  68
  69
  70 vp8_mbsplit vp8_mbsplits [VP8_NUMMBSPLITS] =
  71 {
  72     {
  73         0,  0,  0,  0,
  74         0,  0,  0,  0,
  75         1,  1,  1,  1,
  76         1,  1,  1,  1,
  77     },
  78     {
  79         0,  0,  1,  1,
  80         0,  0,  1,  1,
  81         0,  0,  1,  1,
  82         0,  0,  1,  1,
  83     },
  84     {
  85         0,  0,  1,  1,
  86         0,  0,  1,  1,
  87         2,  2,  3,  3,
  88         2,  2,  3,  3,
  89     },
  90     {
  91         0,  1,  2,  3,
  92         4,  5,  6,  7,
  93         8,  9,  10, 11,
  94         12, 13, 14, 15,
  95     },
  96 };
  97
  98 const int vp8_mbsplit_count [VP8_NUMMBSPLITS] = { 2, 2, 4, 16};
  99
 100 const vp8_prob vp8_mbsplit_probs [VP8_NUMMBSPLITS-1] = { 110, 111, 150};
 101
 102
 103 /* Array indices are identical to previously-existing INTRAMODECONTEXTNODES. */
 104
 105 const vp8_tree_index vp8_bmode_tree[18] =     /* INTRAMODECONTEXTNODE value */
 106 {
 107     -B_DC_PRED, 2,                             /* 0 = DC_NODE */
 108     -B_TM_PRED, 4,                            /* 1 = TM_NODE */
 109     -B_VE_PRED, 6,                           /* 2 = VE_NODE */
 110     8, 12,                                  /* 3 = COM_NODE */
 111     -B_HE_PRED, 10,                        /* 4 = HE_NODE */
 112     -B_RD_PRED, -B_VR_PRED,               /* 5 = RD_NODE */
 113     -B_LD_PRED, 14,                        /* 6 = LD_NODE */
 114     -B_VL_PRED, 16,                      /* 7 = VL_NODE */
 115     -B_HD_PRED, -B_HU_PRED             /* 8 = HD_NODE */
 116 };
 117
 118 /* Again, these trees use the same probability indices as their
 119    explicitly-programmed predecessors. */
 120
 121 const vp8_tree_index vp8_ymode_tree[8] =
 122 {
 123     -DC_PRED, 2,
 124     4, 6,
 125     -V_PRED, -H_PRED,
 126     -TM_PRED, -B_PRED
 127 };
 128
 129 const vp8_tree_index vp8_kf_ymode_tree[8] =
 130 {
 131     -B_PRED, 2,
 132     4, 6,
 133     -DC_PRED, -V_PRED,
 134     -H_PRED, -TM_PRED
 135 };
 136
 137 const vp8_tree_index vp8_uv_mode_tree[6] =
 138 {
 139     -DC_PRED, 2,
 140     -V_PRED, 4,
 141     -H_PRED, -TM_PRED
 142 };
 143
 144 const vp8_tree_index vp8_mbsplit_tree[6] =
 145 {
 146     -3, 2,
 147     -2, 4,
 148     -0, -1
 149 };
 150
 151 const vp8_tree_index vp8_mv_ref_tree[8] =
 152 {
 153     -ZEROMV, 2,
 154     -NEARESTMV, 4,
 155     -NEARMV, 6,
 156     -NEWMV, -SPLITMV
 157 };
 158
 159 const vp8_tree_index vp8_sub_mv_ref_tree[6] =
 160 {
 161     -LEFT4X4, 2,
 162     -ABOVE4X4, 4,
 163     -ZERO4X4, -NEW4X4
 164 };
 165
 166
 167 struct vp8_token_struct vp8_bmode_encodings   [VP8_BINTRAMODES];
 168 struct vp8_token_struct vp8_ymode_encodings   [VP8_YMODES];
 169 struct vp8_token_struct vp8_kf_ymode_encodings [VP8_YMODES];
 170 struct vp8_token_struct vp8_uv_mode_encodings  [VP8_UV_MODES];
 171 struct vp8_token_struct vp8_mbsplit_encodings [VP8_NUMMBSPLITS];
 172
 173 struct vp8_token_struct vp8_mv_ref_encoding_array    [VP8_MVREFS];
 174 struct vp8_token_struct vp8_sub_mv_ref_encoding_array [VP8_SUBMVREFS];
 175
 176
 177 const vp8_tree_index vp8_small_mvtree [14] =
 178 {
 179     2, 8,
 180     4, 6,
 181     -0, -1,
 182     -2, -3,
 183     10, 12,
 184     -4, -5,
 185     -6, -7
 186 };
 187
 188 struct vp8_token_struct vp8_small_mvencodings [8];
 189
 190 void vp8_init_mbmode_probs(VP8_COMMON *x)
 191 {
 192     unsigned int bct [VP8_YMODES] [2];      /* num Ymodes > num UV modes */
 193
 194     vp8_tree_probs_from_distribution(
 195         VP8_YMODES, vp8_ymode_encodings, vp8_ymode_tree,
 196         x->fc.ymode_prob, bct, y_mode_cts,
 197         256, 1
 198     );
 199     vp8_tree_probs_from_distribution(
 200         VP8_YMODES, vp8_kf_ymode_encodings, vp8_kf_ymode_tree,
 201         x->kf_ymode_prob, bct, kf_y_mode_cts,
 202         256, 1
 203     );
 204     vp8_tree_probs_from_distribution(
 205         VP8_UV_MODES, vp8_uv_mode_encodings, vp8_uv_mode_tree,
 206         x->fc.uv_mode_prob, bct, uv_mode_cts,
 207         256, 1
 208     );
 209     vp8_tree_probs_from_distribution(
 210         VP8_UV_MODES, vp8_uv_mode_encodings, vp8_uv_mode_tree,
 211         x->kf_uv_mode_prob, bct, kf_uv_mode_cts,
 212         256, 1
 213     );
 214     vpx_memcpy(x->fc.sub_mv_ref_prob, sub_mv_ref_prob, sizeof(sub_mv_ref_prob));
 215 }
 216
 217
 218 static void intra_bmode_probs_from_distribution(
 219     vp8_prob p [VP8_BINTRAMODES-1],
 220     unsigned int branch_ct [VP8_BINTRAMODES-1] [2],
 221     const unsigned int events [VP8_BINTRAMODES]
 222 )
 223 {
 224     vp8_tree_probs_from_distribution(
 225         VP8_BINTRAMODES, vp8_bmode_encodings, vp8_bmode_tree,
 226         p, branch_ct, events,
 227         256, 1
 228     );
 229 }
 230
 231 void vp8_default_bmode_probs(vp8_prob p [VP8_BINTRAMODES-1])
 232 {
 233     unsigned int branch_ct [VP8_BINTRAMODES-1] [2];
 234     intra_bmode_probs_from_distribution(p, branch_ct, bmode_cts);
 235 }
 236
 237 void vp8_kf_default_bmode_probs(vp8_prob p [VP8_BINTRAMODES] [VP8_BINTRAMODES] [VP8_BINTRAMODES-1])
 238 {
 239     unsigned int branch_ct [VP8_BINTRAMODES-1] [2];
 240
 241     int i = 0;
 242
 243     do
 244     {
 245         int j = 0;
 246
 247         do
 248         {
 249             intra_bmode_probs_from_distribution(
 250                 p[i][j], branch_ct, vp8_kf_default_bmode_counts[i][j]);
 251
 252         }
 253         while (++j < VP8_BINTRAMODES);
 254     }
 255     while (++i < VP8_BINTRAMODES);
 256 }
 257
 258
 259 void vp8_entropy_mode_init()
 260 {
 261     vp8_tokens_from_tree(vp8_bmode_encodings,   vp8_bmode_tree);
 262     vp8_tokens_from_tree(vp8_ymode_encodings,   vp8_ymode_tree);
 263     vp8_tokens_from_tree(vp8_kf_ymode_encodings, vp8_kf_ymode_tree);
 264     vp8_tokens_from_tree(vp8_uv_mode_encodings,  vp8_uv_mode_tree);
 265     vp8_tokens_from_tree(vp8_mbsplit_encodings, vp8_mbsplit_tree);
 266
 267     vp8_tokens_from_tree_offset(vp8_mv_ref_encoding_array,
 268                                 vp8_mv_ref_tree, NEARESTMV);
 269     vp8_tokens_from_tree_offset(vp8_sub_mv_ref_encoding_array,
 270                                 vp8_sub_mv_ref_tree, LEFT4X4);
 271
 272     vp8_tokens_from_tree(vp8_small_mvencodings, vp8_small_mvtree);
 273 }