Use specific PICK_MODE_CONTEXT for supertx.

Change-Id: I402d10dd666ebc4a06ce4472810a7e22f2e056ff
This commit is contained in:
Geza Lore
2016-01-12 13:06:42 +00:00
parent d738ce2b0a
commit e4663e6da1
3 changed files with 13 additions and 4 deletions

View File

@@ -81,6 +81,9 @@ static void alloc_tree_contexts(VP10_COMMON *cm, PC_TREE *tree,
alloc_mode_context(cm, num_4x4_blk, &tree->none);
alloc_mode_context(cm, num_4x4_blk/2, &tree->horizontal[0]);
alloc_mode_context(cm, num_4x4_blk/2, &tree->vertical[0]);
#ifdef CONFIG_SUPERTX
alloc_mode_context(cm, num_4x4_blk, &tree->super_tx);
#endif
if (num_4x4_blk > 4) {
alloc_mode_context(cm, num_4x4_blk/2, &tree->horizontal[1]);
@@ -97,6 +100,9 @@ static void free_tree_contexts(PC_TREE *tree) {
free_mode_context(&tree->horizontal[1]);
free_mode_context(&tree->vertical[0]);
free_mode_context(&tree->vertical[1]);
#ifdef CONFIG_SUPERTX
free_mode_context(&tree->super_tx);
#endif
}
// This function sets up a tree of contexts such that at each square