1233d2500723e5594f3e7c70896ffeeef32b9c950ywan/* 2233d2500723e5594f3e7c70896ffeeef32b9c950ywan * Copyright (c) 2010 The WebM project authors. All Rights Reserved. 3233d2500723e5594f3e7c70896ffeeef32b9c950ywan * 4233d2500723e5594f3e7c70896ffeeef32b9c950ywan * Use of this source code is governed by a BSD-style license 5233d2500723e5594f3e7c70896ffeeef32b9c950ywan * that can be found in the LICENSE file in the root of the source 6233d2500723e5594f3e7c70896ffeeef32b9c950ywan * tree. An additional intellectual property rights grant can be found 7233d2500723e5594f3e7c70896ffeeef32b9c950ywan * in the file PATENTS. All contributing project authors may 8233d2500723e5594f3e7c70896ffeeef32b9c950ywan * be found in the AUTHORS file in the root of the source tree. 9233d2500723e5594f3e7c70896ffeeef32b9c950ywan */ 10233d2500723e5594f3e7c70896ffeeef32b9c950ywan 11233d2500723e5594f3e7c70896ffeeef32b9c950ywan#define USE_PREBUILT_TABLES 12233d2500723e5594f3e7c70896ffeeef32b9c950ywan 13233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include "entropymode.h" 14233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include "entropy.h" 15233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include "vpx_mem/vpx_mem.h" 16233d2500723e5594f3e7c70896ffeeef32b9c950ywan 17233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include "vp8_entropymodedata.h" 18233d2500723e5594f3e7c70896ffeeef32b9c950ywan 19233d2500723e5594f3e7c70896ffeeef32b9c950ywanint vp8_mv_cont(const int_mv *l, const int_mv *a) 20233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 21233d2500723e5594f3e7c70896ffeeef32b9c950ywan int lez = (l->as_int == 0); 22233d2500723e5594f3e7c70896ffeeef32b9c950ywan int aez = (a->as_int == 0); 23233d2500723e5594f3e7c70896ffeeef32b9c950ywan int lea = (l->as_int == a->as_int); 24233d2500723e5594f3e7c70896ffeeef32b9c950ywan 25233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (lea && lez) 26233d2500723e5594f3e7c70896ffeeef32b9c950ywan return SUBMVREF_LEFT_ABOVE_ZED; 27233d2500723e5594f3e7c70896ffeeef32b9c950ywan 28233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (lea) 29233d2500723e5594f3e7c70896ffeeef32b9c950ywan return SUBMVREF_LEFT_ABOVE_SAME; 30233d2500723e5594f3e7c70896ffeeef32b9c950ywan 31233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (aez) 32233d2500723e5594f3e7c70896ffeeef32b9c950ywan return SUBMVREF_ABOVE_ZED; 33233d2500723e5594f3e7c70896ffeeef32b9c950ywan 34233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (lez) 35233d2500723e5594f3e7c70896ffeeef32b9c950ywan return SUBMVREF_LEFT_ZED; 36233d2500723e5594f3e7c70896ffeeef32b9c950ywan 37233d2500723e5594f3e7c70896ffeeef32b9c950ywan return SUBMVREF_NORMAL; 38233d2500723e5594f3e7c70896ffeeef32b9c950ywan} 39233d2500723e5594f3e7c70896ffeeef32b9c950ywan 40233d2500723e5594f3e7c70896ffeeef32b9c950ywanstatic const vp8_prob sub_mv_ref_prob [VP8_SUBMVREFS-1] = { 180, 162, 25}; 41233d2500723e5594f3e7c70896ffeeef32b9c950ywan 42233d2500723e5594f3e7c70896ffeeef32b9c950ywanconst vp8_prob vp8_sub_mv_ref_prob2 [SUBMVREF_COUNT][VP8_SUBMVREFS-1] = 43233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 44233d2500723e5594f3e7c70896ffeeef32b9c950ywan { 147, 136, 18 }, 45233d2500723e5594f3e7c70896ffeeef32b9c950ywan { 106, 145, 1 }, 46233d2500723e5594f3e7c70896ffeeef32b9c950ywan { 179, 121, 1 }, 47233d2500723e5594f3e7c70896ffeeef32b9c950ywan { 223, 1 , 34 }, 48233d2500723e5594f3e7c70896ffeeef32b9c950ywan { 208, 1 , 1 } 49233d2500723e5594f3e7c70896ffeeef32b9c950ywan}; 50233d2500723e5594f3e7c70896ffeeef32b9c950ywan 51233d2500723e5594f3e7c70896ffeeef32b9c950ywan 52233d2500723e5594f3e7c70896ffeeef32b9c950ywan 53233d2500723e5594f3e7c70896ffeeef32b9c950ywanconst vp8_mbsplit vp8_mbsplits [VP8_NUMMBSPLITS] = 54233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 55233d2500723e5594f3e7c70896ffeeef32b9c950ywan { 56233d2500723e5594f3e7c70896ffeeef32b9c950ywan 0, 0, 0, 0, 57233d2500723e5594f3e7c70896ffeeef32b9c950ywan 0, 0, 0, 0, 58233d2500723e5594f3e7c70896ffeeef32b9c950ywan 1, 1, 1, 1, 59233d2500723e5594f3e7c70896ffeeef32b9c950ywan 1, 1, 1, 1, 60233d2500723e5594f3e7c70896ffeeef32b9c950ywan }, 61233d2500723e5594f3e7c70896ffeeef32b9c950ywan { 62233d2500723e5594f3e7c70896ffeeef32b9c950ywan 0, 0, 1, 1, 63233d2500723e5594f3e7c70896ffeeef32b9c950ywan 0, 0, 1, 1, 64233d2500723e5594f3e7c70896ffeeef32b9c950ywan 0, 0, 1, 1, 65233d2500723e5594f3e7c70896ffeeef32b9c950ywan 0, 0, 1, 1, 66233d2500723e5594f3e7c70896ffeeef32b9c950ywan }, 67233d2500723e5594f3e7c70896ffeeef32b9c950ywan { 68233d2500723e5594f3e7c70896ffeeef32b9c950ywan 0, 0, 1, 1, 69233d2500723e5594f3e7c70896ffeeef32b9c950ywan 0, 0, 1, 1, 70233d2500723e5594f3e7c70896ffeeef32b9c950ywan 2, 2, 3, 3, 71233d2500723e5594f3e7c70896ffeeef32b9c950ywan 2, 2, 3, 3, 72233d2500723e5594f3e7c70896ffeeef32b9c950ywan }, 73233d2500723e5594f3e7c70896ffeeef32b9c950ywan { 74233d2500723e5594f3e7c70896ffeeef32b9c950ywan 0, 1, 2, 3, 75233d2500723e5594f3e7c70896ffeeef32b9c950ywan 4, 5, 6, 7, 76233d2500723e5594f3e7c70896ffeeef32b9c950ywan 8, 9, 10, 11, 77233d2500723e5594f3e7c70896ffeeef32b9c950ywan 12, 13, 14, 15, 78233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 79233d2500723e5594f3e7c70896ffeeef32b9c950ywan}; 80233d2500723e5594f3e7c70896ffeeef32b9c950ywan 81233d2500723e5594f3e7c70896ffeeef32b9c950ywanconst int vp8_mbsplit_count [VP8_NUMMBSPLITS] = { 2, 2, 4, 16}; 82233d2500723e5594f3e7c70896ffeeef32b9c950ywan 83233d2500723e5594f3e7c70896ffeeef32b9c950ywanconst vp8_prob vp8_mbsplit_probs [VP8_NUMMBSPLITS-1] = { 110, 111, 150}; 84233d2500723e5594f3e7c70896ffeeef32b9c950ywan 85233d2500723e5594f3e7c70896ffeeef32b9c950ywan 86233d2500723e5594f3e7c70896ffeeef32b9c950ywan/* Array indices are identical to previously-existing INTRAMODECONTEXTNODES. */ 87233d2500723e5594f3e7c70896ffeeef32b9c950ywan 88233d2500723e5594f3e7c70896ffeeef32b9c950ywanconst vp8_tree_index vp8_bmode_tree[18] = /* INTRAMODECONTEXTNODE value */ 89233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 90233d2500723e5594f3e7c70896ffeeef32b9c950ywan -B_DC_PRED, 2, /* 0 = DC_NODE */ 91233d2500723e5594f3e7c70896ffeeef32b9c950ywan -B_TM_PRED, 4, /* 1 = TM_NODE */ 92233d2500723e5594f3e7c70896ffeeef32b9c950ywan -B_VE_PRED, 6, /* 2 = VE_NODE */ 93233d2500723e5594f3e7c70896ffeeef32b9c950ywan 8, 12, /* 3 = COM_NODE */ 94233d2500723e5594f3e7c70896ffeeef32b9c950ywan -B_HE_PRED, 10, /* 4 = HE_NODE */ 95233d2500723e5594f3e7c70896ffeeef32b9c950ywan -B_RD_PRED, -B_VR_PRED, /* 5 = RD_NODE */ 96233d2500723e5594f3e7c70896ffeeef32b9c950ywan -B_LD_PRED, 14, /* 6 = LD_NODE */ 97233d2500723e5594f3e7c70896ffeeef32b9c950ywan -B_VL_PRED, 16, /* 7 = VL_NODE */ 98233d2500723e5594f3e7c70896ffeeef32b9c950ywan -B_HD_PRED, -B_HU_PRED /* 8 = HD_NODE */ 99233d2500723e5594f3e7c70896ffeeef32b9c950ywan}; 100233d2500723e5594f3e7c70896ffeeef32b9c950ywan 101233d2500723e5594f3e7c70896ffeeef32b9c950ywan/* Again, these trees use the same probability indices as their 102233d2500723e5594f3e7c70896ffeeef32b9c950ywan explicitly-programmed predecessors. */ 103233d2500723e5594f3e7c70896ffeeef32b9c950ywan 104233d2500723e5594f3e7c70896ffeeef32b9c950ywanconst vp8_tree_index vp8_ymode_tree[8] = 105233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 106233d2500723e5594f3e7c70896ffeeef32b9c950ywan -DC_PRED, 2, 107233d2500723e5594f3e7c70896ffeeef32b9c950ywan 4, 6, 108233d2500723e5594f3e7c70896ffeeef32b9c950ywan -V_PRED, -H_PRED, 109233d2500723e5594f3e7c70896ffeeef32b9c950ywan -TM_PRED, -B_PRED 110233d2500723e5594f3e7c70896ffeeef32b9c950ywan}; 111233d2500723e5594f3e7c70896ffeeef32b9c950ywan 112233d2500723e5594f3e7c70896ffeeef32b9c950ywanconst vp8_tree_index vp8_kf_ymode_tree[8] = 113233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 114233d2500723e5594f3e7c70896ffeeef32b9c950ywan -B_PRED, 2, 115233d2500723e5594f3e7c70896ffeeef32b9c950ywan 4, 6, 116233d2500723e5594f3e7c70896ffeeef32b9c950ywan -DC_PRED, -V_PRED, 117233d2500723e5594f3e7c70896ffeeef32b9c950ywan -H_PRED, -TM_PRED 118233d2500723e5594f3e7c70896ffeeef32b9c950ywan}; 119233d2500723e5594f3e7c70896ffeeef32b9c950ywan 120233d2500723e5594f3e7c70896ffeeef32b9c950ywanconst vp8_tree_index vp8_uv_mode_tree[6] = 121233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 122233d2500723e5594f3e7c70896ffeeef32b9c950ywan -DC_PRED, 2, 123233d2500723e5594f3e7c70896ffeeef32b9c950ywan -V_PRED, 4, 124233d2500723e5594f3e7c70896ffeeef32b9c950ywan -H_PRED, -TM_PRED 125233d2500723e5594f3e7c70896ffeeef32b9c950ywan}; 126233d2500723e5594f3e7c70896ffeeef32b9c950ywan 127233d2500723e5594f3e7c70896ffeeef32b9c950ywanconst vp8_tree_index vp8_mbsplit_tree[6] = 128233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 129233d2500723e5594f3e7c70896ffeeef32b9c950ywan -3, 2, 130233d2500723e5594f3e7c70896ffeeef32b9c950ywan -2, 4, 131233d2500723e5594f3e7c70896ffeeef32b9c950ywan -0, -1 132233d2500723e5594f3e7c70896ffeeef32b9c950ywan}; 133233d2500723e5594f3e7c70896ffeeef32b9c950ywan 134233d2500723e5594f3e7c70896ffeeef32b9c950ywanconst vp8_tree_index vp8_mv_ref_tree[8] = 135233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 136233d2500723e5594f3e7c70896ffeeef32b9c950ywan -ZEROMV, 2, 137233d2500723e5594f3e7c70896ffeeef32b9c950ywan -NEARESTMV, 4, 138233d2500723e5594f3e7c70896ffeeef32b9c950ywan -NEARMV, 6, 139233d2500723e5594f3e7c70896ffeeef32b9c950ywan -NEWMV, -SPLITMV 140233d2500723e5594f3e7c70896ffeeef32b9c950ywan}; 141233d2500723e5594f3e7c70896ffeeef32b9c950ywan 142233d2500723e5594f3e7c70896ffeeef32b9c950ywanconst vp8_tree_index vp8_sub_mv_ref_tree[6] = 143233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 144233d2500723e5594f3e7c70896ffeeef32b9c950ywan -LEFT4X4, 2, 145233d2500723e5594f3e7c70896ffeeef32b9c950ywan -ABOVE4X4, 4, 146233d2500723e5594f3e7c70896ffeeef32b9c950ywan -ZERO4X4, -NEW4X4 147233d2500723e5594f3e7c70896ffeeef32b9c950ywan}; 148233d2500723e5594f3e7c70896ffeeef32b9c950ywan 149233d2500723e5594f3e7c70896ffeeef32b9c950ywanconst vp8_tree_index vp8_small_mvtree [14] = 150233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 151233d2500723e5594f3e7c70896ffeeef32b9c950ywan 2, 8, 152233d2500723e5594f3e7c70896ffeeef32b9c950ywan 4, 6, 153233d2500723e5594f3e7c70896ffeeef32b9c950ywan -0, -1, 154233d2500723e5594f3e7c70896ffeeef32b9c950ywan -2, -3, 155233d2500723e5594f3e7c70896ffeeef32b9c950ywan 10, 12, 156233d2500723e5594f3e7c70896ffeeef32b9c950ywan -4, -5, 157233d2500723e5594f3e7c70896ffeeef32b9c950ywan -6, -7 158233d2500723e5594f3e7c70896ffeeef32b9c950ywan}; 159233d2500723e5594f3e7c70896ffeeef32b9c950ywan 160233d2500723e5594f3e7c70896ffeeef32b9c950ywanvoid vp8_init_mbmode_probs(VP8_COMMON *x) 161233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 162233d2500723e5594f3e7c70896ffeeef32b9c950ywan vpx_memcpy(x->fc.ymode_prob, vp8_ymode_prob, sizeof(vp8_ymode_prob)); 163233d2500723e5594f3e7c70896ffeeef32b9c950ywan vpx_memcpy(x->fc.uv_mode_prob, vp8_uv_mode_prob, sizeof(vp8_uv_mode_prob)); 164233d2500723e5594f3e7c70896ffeeef32b9c950ywan vpx_memcpy(x->fc.sub_mv_ref_prob, sub_mv_ref_prob, sizeof(sub_mv_ref_prob)); 165233d2500723e5594f3e7c70896ffeeef32b9c950ywan} 166233d2500723e5594f3e7c70896ffeeef32b9c950ywan 167233d2500723e5594f3e7c70896ffeeef32b9c950ywanvoid vp8_default_bmode_probs(vp8_prob p [VP8_BINTRAMODES-1]) 168233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 169233d2500723e5594f3e7c70896ffeeef32b9c950ywan vpx_memcpy(p, vp8_bmode_prob, sizeof(vp8_bmode_prob)); 170233d2500723e5594f3e7c70896ffeeef32b9c950ywan} 171233d2500723e5594f3e7c70896ffeeef32b9c950ywan 172