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 #define USE_PREBUILT_TABLES
12 
13 #include "entropymode.h"
14 #include "entropy.h"
15 #include "vpx_mem/vpx_mem.h"
16 
17 #include "vp8_entropymodedata.h"
18 
vp8_mv_cont(const int_mv * l,const int_mv * a)19 int vp8_mv_cont(const int_mv *l, const int_mv *a)
20 {
21     int lez = (l->as_int == 0);
22     int aez = (a->as_int == 0);
23     int lea = (l->as_int == a->as_int);
24 
25     if (lea && lez)
26         return SUBMVREF_LEFT_ABOVE_ZED;
27 
28     if (lea)
29         return SUBMVREF_LEFT_ABOVE_SAME;
30 
31     if (aez)
32         return SUBMVREF_ABOVE_ZED;
33 
34     if (lez)
35         return SUBMVREF_LEFT_ZED;
36 
37     return SUBMVREF_NORMAL;
38 }
39 
40 static const vp8_prob sub_mv_ref_prob [VP8_SUBMVREFS-1] = { 180, 162, 25};
41 
42 const vp8_prob vp8_sub_mv_ref_prob2 [SUBMVREF_COUNT][VP8_SUBMVREFS-1] =
43 {
44     { 147, 136, 18 },
45     { 106, 145, 1  },
46     { 179, 121, 1  },
47     { 223, 1  , 34 },
48     { 208, 1  , 1  }
49 };
50 
51 
52 
53 const vp8_mbsplit vp8_mbsplits [VP8_NUMMBSPLITS] =
54 {
55     {
56         0,  0,  0,  0,
57         0,  0,  0,  0,
58         1,  1,  1,  1,
59         1,  1,  1,  1,
60     },
61     {
62         0,  0,  1,  1,
63         0,  0,  1,  1,
64         0,  0,  1,  1,
65         0,  0,  1,  1,
66     },
67     {
68         0,  0,  1,  1,
69         0,  0,  1,  1,
70         2,  2,  3,  3,
71         2,  2,  3,  3,
72     },
73     {
74         0,  1,  2,  3,
75         4,  5,  6,  7,
76         8,  9,  10, 11,
77         12, 13, 14, 15,
78     }
79 };
80 
81 const int vp8_mbsplit_count [VP8_NUMMBSPLITS] = { 2, 2, 4, 16};
82 
83 const vp8_prob vp8_mbsplit_probs [VP8_NUMMBSPLITS-1] = { 110, 111, 150};
84 
85 
86 /* Array indices are identical to previously-existing INTRAMODECONTEXTNODES. */
87 
88 const vp8_tree_index vp8_bmode_tree[18] =     /* INTRAMODECONTEXTNODE value */
89 {
90     -B_DC_PRED, 2,                             /* 0 = DC_NODE */
91     -B_TM_PRED, 4,                            /* 1 = TM_NODE */
92     -B_VE_PRED, 6,                           /* 2 = VE_NODE */
93     8, 12,                                  /* 3 = COM_NODE */
94     -B_HE_PRED, 10,                        /* 4 = HE_NODE */
95     -B_RD_PRED, -B_VR_PRED,               /* 5 = RD_NODE */
96     -B_LD_PRED, 14,                        /* 6 = LD_NODE */
97     -B_VL_PRED, 16,                      /* 7 = VL_NODE */
98     -B_HD_PRED, -B_HU_PRED             /* 8 = HD_NODE */
99 };
100 
101 /* Again, these trees use the same probability indices as their
102    explicitly-programmed predecessors. */
103 
104 const vp8_tree_index vp8_ymode_tree[8] =
105 {
106     -DC_PRED, 2,
107     4, 6,
108     -V_PRED, -H_PRED,
109     -TM_PRED, -B_PRED
110 };
111 
112 const vp8_tree_index vp8_kf_ymode_tree[8] =
113 {
114     -B_PRED, 2,
115     4, 6,
116     -DC_PRED, -V_PRED,
117     -H_PRED, -TM_PRED
118 };
119 
120 const vp8_tree_index vp8_uv_mode_tree[6] =
121 {
122     -DC_PRED, 2,
123     -V_PRED, 4,
124     -H_PRED, -TM_PRED
125 };
126 
127 const vp8_tree_index vp8_mbsplit_tree[6] =
128 {
129     -3, 2,
130     -2, 4,
131     -0, -1
132 };
133 
134 const vp8_tree_index vp8_mv_ref_tree[8] =
135 {
136     -ZEROMV, 2,
137     -NEARESTMV, 4,
138     -NEARMV, 6,
139     -NEWMV, -SPLITMV
140 };
141 
142 const vp8_tree_index vp8_sub_mv_ref_tree[6] =
143 {
144     -LEFT4X4, 2,
145     -ABOVE4X4, 4,
146     -ZERO4X4, -NEW4X4
147 };
148 
149 const vp8_tree_index vp8_small_mvtree [14] =
150 {
151     2, 8,
152     4, 6,
153     -0, -1,
154     -2, -3,
155     10, 12,
156     -4, -5,
157     -6, -7
158 };
159 
vp8_init_mbmode_probs(VP8_COMMON * x)160 void vp8_init_mbmode_probs(VP8_COMMON *x)
161 {
162     memcpy(x->fc.ymode_prob, vp8_ymode_prob, sizeof(vp8_ymode_prob));
163     memcpy(x->fc.uv_mode_prob, vp8_uv_mode_prob, sizeof(vp8_uv_mode_prob));
164     memcpy(x->fc.sub_mv_ref_prob, sub_mv_ref_prob, sizeof(sub_mv_ref_prob));
165 }
166 
vp8_default_bmode_probs(vp8_prob p[VP8_BINTRAMODES-1])167 void vp8_default_bmode_probs(vp8_prob p [VP8_BINTRAMODES-1])
168 {
169     memcpy(p, vp8_bmode_prob, sizeof(vp8_bmode_prob));
170 }
171 
172