a9336328d4
This commit makes the encoding process of the recursive transform block partition support both rectangular and square transform block sizes as the starting point. If the coding block size is rectangular, it would allow the transform block size to start from the largest rectangular transform size, and recursive parse to the selected coding sizes. Change-Id: I576628b9166565bada6a918f0a1e67849dfef4cd
920 lines
33 KiB
C
920 lines
33 KiB
C
/*
|
|
* Copyright (c) 2016, Alliance for Open Media. All rights reserved
|
|
*
|
|
* This source code is subject to the terms of the BSD 2 Clause License and
|
|
* the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
|
|
* was not distributed with this source code in the LICENSE file, you can
|
|
* obtain it at www.aomedia.org/license/software. If the Alliance for Open
|
|
* Media Patent License 1.0 was not distributed with this source code in the
|
|
* PATENTS file, you can obtain it at www.aomedia.org/license/patent.
|
|
*/
|
|
|
|
#ifndef AV1_COMMON_COMMON_DATA_H_
|
|
#define AV1_COMMON_COMMON_DATA_H_
|
|
|
|
#include "av1/common/enums.h"
|
|
#include "aom/aom_integer.h"
|
|
#include "aom_dsp/aom_dsp_common.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#if CONFIG_EXT_PARTITION
|
|
#define IF_EXT_PARTITION(...) __VA_ARGS__
|
|
#else
|
|
#define IF_EXT_PARTITION(...)
|
|
#endif
|
|
|
|
// Log 2 conversion lookup tables for block width and height
|
|
static const uint8_t b_width_log2_lookup[BLOCK_SIZES] = {
|
|
0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, IF_EXT_PARTITION(4, 5, 5)
|
|
};
|
|
static const uint8_t b_height_log2_lookup[BLOCK_SIZES] = {
|
|
0, 1, 0, 1, 2, 1, 2, 3, 2, 3, 4, 3, 4, IF_EXT_PARTITION(5, 4, 5)
|
|
};
|
|
// Log 2 conversion lookup tables for modeinfo width and height
|
|
static const uint8_t mi_width_log2_lookup[BLOCK_SIZES] = {
|
|
0, 0, 0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, IF_EXT_PARTITION(3, 4, 4)
|
|
};
|
|
static const uint8_t mi_height_log2_lookup[BLOCK_SIZES] = {
|
|
0, 0, 0, 0, 1, 0, 1, 2, 1, 2, 3, 2, 3, IF_EXT_PARTITION(4, 3, 4)
|
|
};
|
|
|
|
// Width/height lookup tables in units of various block sizes
|
|
static const uint8_t block_size_wide[BLOCK_SIZES] = {
|
|
4, 4, 8, 8, 8, 16, 16, 16, 32, 32, 32, 64, 64, IF_EXT_PARTITION(64, 128, 128)
|
|
};
|
|
static const uint8_t block_size_high[BLOCK_SIZES] = {
|
|
4, 8, 4, 8, 16, 8, 16, 32, 16, 32, 64, 32, 64, IF_EXT_PARTITION(128, 64, 128)
|
|
};
|
|
static const uint8_t num_4x4_blocks_wide_lookup[BLOCK_SIZES] = {
|
|
1, 1, 2, 2, 2, 4, 4, 4, 8, 8, 8, 16, 16, IF_EXT_PARTITION(16, 32, 32)
|
|
};
|
|
static const uint8_t num_4x4_blocks_high_lookup[BLOCK_SIZES] = {
|
|
1, 2, 1, 2, 4, 2, 4, 8, 4, 8, 16, 8, 16, IF_EXT_PARTITION(32, 16, 32)
|
|
};
|
|
static const uint8_t num_8x8_blocks_wide_lookup[BLOCK_SIZES] = {
|
|
1, 1, 1, 1, 1, 2, 2, 2, 4, 4, 4, 8, 8, IF_EXT_PARTITION(8, 16, 16)
|
|
};
|
|
static const uint8_t num_8x8_blocks_high_lookup[BLOCK_SIZES] = {
|
|
1, 1, 1, 1, 2, 1, 2, 4, 2, 4, 8, 4, 8, IF_EXT_PARTITION(16, 8, 16)
|
|
};
|
|
static const uint8_t num_16x16_blocks_wide_lookup[BLOCK_SIZES] = {
|
|
1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 4, 4, IF_EXT_PARTITION(4, 8, 8)
|
|
};
|
|
static const uint8_t num_16x16_blocks_high_lookup[BLOCK_SIZES] = {
|
|
1, 1, 1, 1, 1, 1, 1, 2, 1, 2, 4, 2, 4, IF_EXT_PARTITION(8, 4, 8)
|
|
};
|
|
|
|
// AOMMIN(3, AOMMIN(b_width_log2(bsize), b_height_log2(bsize)))
|
|
static const uint8_t size_group_lookup[BLOCK_SIZES] = {
|
|
0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 3, IF_EXT_PARTITION(3, 3, 3)
|
|
};
|
|
|
|
static const uint8_t num_pels_log2_lookup[BLOCK_SIZES] = {
|
|
4, 5, 5, 6, 7, 7, 8, 9, 9, 10, 11, 11, 12, IF_EXT_PARTITION(13, 13, 14)
|
|
};
|
|
|
|
/* clang-format off */
|
|
static const PARTITION_TYPE
|
|
partition_lookup[MAX_SB_SIZE_LOG2 - 1][BLOCK_SIZES] = {
|
|
{ // 4X4 ->
|
|
// 4X4
|
|
PARTITION_NONE,
|
|
// 4X8, 8X4, 8X8
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
// 8X16, 16X8, 16X16
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
// 16X32, 32X16, 32X32
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
// 32X64, 64X32, 64X64
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
#if CONFIG_EXT_PARTITION
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
}, { // 8X8 ->
|
|
// 4X4
|
|
PARTITION_SPLIT,
|
|
// 4X8, 8X4, 8X8
|
|
PARTITION_VERT, PARTITION_HORZ, PARTITION_NONE,
|
|
// 8X16, 16X8, 16X16
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
// 16X32, 32X16, 32X32
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
// 32X64, 64X32, 64X64
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
}, { // 16X16 ->
|
|
// 4X4
|
|
PARTITION_SPLIT,
|
|
// 4X8, 8X4, 8X8
|
|
PARTITION_SPLIT, PARTITION_SPLIT, PARTITION_SPLIT,
|
|
// 8X16, 16X8, 16X16
|
|
PARTITION_VERT, PARTITION_HORZ, PARTITION_NONE,
|
|
// 16X32, 32X16, 32X32
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
// 32X64, 64X32, 64X64
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
}, { // 32X32 ->
|
|
// 4X4
|
|
PARTITION_SPLIT,
|
|
// 4X8, 8X4, 8X8
|
|
PARTITION_SPLIT, PARTITION_SPLIT, PARTITION_SPLIT,
|
|
// 8X16, 16X8, 16X16
|
|
PARTITION_SPLIT, PARTITION_SPLIT, PARTITION_SPLIT,
|
|
// 16X32, 32X16, 32X32
|
|
PARTITION_VERT, PARTITION_HORZ, PARTITION_NONE,
|
|
// 32X64, 64X32, 64X64
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
}, { // 64X64 ->
|
|
// 4X4
|
|
PARTITION_SPLIT,
|
|
// 4X8, 8X4, 8X8
|
|
PARTITION_SPLIT, PARTITION_SPLIT, PARTITION_SPLIT,
|
|
// 8X16, 16X8, 16X16
|
|
PARTITION_SPLIT, PARTITION_SPLIT, PARTITION_SPLIT,
|
|
// 16X32, 32X16, 32X32
|
|
PARTITION_SPLIT, PARTITION_SPLIT, PARTITION_SPLIT,
|
|
// 32X64, 64X32, 64X64
|
|
PARTITION_VERT, PARTITION_HORZ, PARTITION_NONE,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
PARTITION_INVALID, PARTITION_INVALID, PARTITION_INVALID,
|
|
}, { // 128x128 ->
|
|
// 4X4
|
|
PARTITION_SPLIT,
|
|
// 4X8, 8X4, 8X8
|
|
PARTITION_SPLIT, PARTITION_SPLIT, PARTITION_SPLIT,
|
|
// 8X16, 16X8, 16X16
|
|
PARTITION_SPLIT, PARTITION_SPLIT, PARTITION_SPLIT,
|
|
// 16X32, 32X16, 32X32
|
|
PARTITION_SPLIT, PARTITION_SPLIT, PARTITION_SPLIT,
|
|
// 32X64, 64X32, 64X64
|
|
PARTITION_SPLIT, PARTITION_SPLIT, PARTITION_SPLIT,
|
|
// 64x128, 128x64, 128x128
|
|
PARTITION_VERT, PARTITION_HORZ, PARTITION_NONE,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
}
|
|
};
|
|
|
|
#if CONFIG_EXT_PARTITION_TYPES
|
|
static const BLOCK_SIZE subsize_lookup[EXT_PARTITION_TYPES][BLOCK_SIZES] =
|
|
#else
|
|
static const BLOCK_SIZE subsize_lookup[PARTITION_TYPES][BLOCK_SIZES] =
|
|
#endif // CONFIG_EXT_PARTITION_TYPES
|
|
{
|
|
{ // PARTITION_NONE
|
|
// 4X4
|
|
BLOCK_4X4,
|
|
// 4X8, 8X4, 8X8
|
|
BLOCK_4X8, BLOCK_8X4, BLOCK_8X8,
|
|
// 8X16, 16X8, 16X16
|
|
BLOCK_8X16, BLOCK_16X8, BLOCK_16X16,
|
|
// 16X32, 32X16, 32X32
|
|
BLOCK_16X32, BLOCK_32X16, BLOCK_32X32,
|
|
// 32X64, 64X32, 64X64
|
|
BLOCK_32X64, BLOCK_64X32, BLOCK_64X64,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
BLOCK_64X128, BLOCK_128X64, BLOCK_128X128,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
}, { // PARTITION_HORZ
|
|
// 4X4
|
|
BLOCK_INVALID,
|
|
// 4X8, 8X4, 8X8
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_8X4,
|
|
// 8X16, 16X8, 16X16
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_16X8,
|
|
// 16X32, 32X16, 32X32
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_32X16,
|
|
// 32X64, 64X32, 64X64
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_64X32,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_128X64,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
}, { // PARTITION_VERT
|
|
// 4X4
|
|
BLOCK_INVALID,
|
|
// 4X8, 8X4, 8X8
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_4X8,
|
|
// 8X16, 16X8, 16X16
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_8X16,
|
|
// 16X32, 32X16, 32X32
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_16X32,
|
|
// 32X64, 64X32, 64X64
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_32X64,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_64X128,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
}, { // PARTITION_SPLIT
|
|
// 4X4
|
|
BLOCK_INVALID,
|
|
// 4X8, 8X4, 8X8
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_4X4,
|
|
// 8X16, 16X8, 16X16
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_8X8,
|
|
// 16X32, 32X16, 32X32
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_16X16,
|
|
// 32X64, 64X32, 64X64
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_32X32,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_64X64,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
#if CONFIG_EXT_PARTITION_TYPES
|
|
}, { // PARTITION_HORZ_A
|
|
// 4X4
|
|
BLOCK_INVALID,
|
|
// 4X8, 8X4, 8X8
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_8X4,
|
|
// 8X16, 16X8, 16X16
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_16X8,
|
|
// 16X32, 32X16, 32X32
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_32X16,
|
|
// 32X64, 64X32, 64X64
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_64X32,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_128X64,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
}, { // PARTITION_HORZ_B
|
|
// 4X4
|
|
BLOCK_INVALID,
|
|
// 4X8, 8X4, 8X8
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_8X4,
|
|
// 8X16, 16X8, 16X16
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_16X8,
|
|
// 16X32, 32X16, 32X32
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_32X16,
|
|
// 32X64, 64X32, 64X64
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_64X32,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_128X64,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
}, { // PARTITION_VERT_A
|
|
// 4X4
|
|
BLOCK_INVALID,
|
|
// 4X8, 8X4, 8X8
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_4X8,
|
|
// 8X16, 16X8, 16X16
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_8X16,
|
|
// 16X32, 32X16, 32X32
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_16X32,
|
|
// 32X64, 64X32, 64X64
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_32X64,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_64X128,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
}, { // PARTITION_VERT_B
|
|
// 4X4
|
|
BLOCK_INVALID,
|
|
// 4X8, 8X4, 8X8
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_4X8,
|
|
// 8X16, 16X8, 16X16
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_8X16,
|
|
// 16X32, 32X16, 32X32
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_16X32,
|
|
// 32X64, 64X32, 64X64
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_32X64,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
BLOCK_INVALID, BLOCK_INVALID, BLOCK_64X128,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
#endif // CONFIG_EXT_PARTITION_TYPES
|
|
}
|
|
};
|
|
|
|
static const TX_SIZE max_txsize_lookup[BLOCK_SIZES] = {
|
|
// 4X4
|
|
TX_4X4,
|
|
// 4X8, 8X4, 8X8
|
|
TX_4X4, TX_4X4, TX_8X8,
|
|
// 8X16, 16X8, 16X16
|
|
TX_8X8, TX_8X8, TX_16X16,
|
|
// 16X32, 32X16, 32X32
|
|
TX_16X16, TX_16X16, TX_32X32,
|
|
// 32X64, 64X32, 64X64
|
|
TX_32X32, TX_32X32, TX_32X32,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
TX_32X32, TX_32X32, TX_32X32,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
};
|
|
|
|
static const TX_SIZE max_txsize_rect_lookup[BLOCK_SIZES] = {
|
|
// 4X4
|
|
TX_4X4,
|
|
// 4X8, 8X4, 8X8
|
|
TX_4X8, TX_8X4, TX_8X8,
|
|
// 8X16, 16X8, 16X16
|
|
TX_8X16, TX_16X8, TX_16X16,
|
|
// 16X32, 32X16, 32X32
|
|
TX_16X32, TX_32X16, TX_32X32,
|
|
// 32X64, 64X32, 64X64
|
|
TX_32X32, TX_32X32, TX_32X32,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
TX_32X32, TX_32X32, TX_32X32,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
};
|
|
|
|
// Same as "max_txsize_lookup[bsize] - TX_8X8", invalid for bsize < 8X8
|
|
static const int32_t intra_tx_size_cat_lookup[BLOCK_SIZES] = {
|
|
// 4X4
|
|
INT32_MIN,
|
|
// 4X8, 8X4, 8X8
|
|
INT32_MIN, INT32_MIN, TX_8X8 - TX_8X8,
|
|
// 8X16, 16X8, 16X16
|
|
TX_8X8 - TX_8X8, TX_8X8 - TX_8X8, TX_16X16 - TX_8X8,
|
|
// 16X32, 32X16, 32X32
|
|
TX_16X16 - TX_8X8, TX_16X16 - TX_8X8, TX_32X32 - TX_8X8,
|
|
// 32X64, 64X32, 64X64
|
|
TX_32X32 - TX_8X8, TX_32X32 - TX_8X8, TX_32X32 - TX_8X8,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
TX_32X32 - TX_8X8, TX_32X32 - TX_8X8, TX_32X32 - TX_8X8,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
};
|
|
|
|
#if CONFIG_EXT_TX && CONFIG_RECT_TX
|
|
// Same as "max_txsize_lookup[bsize] - TX_8X8", except for rectangular
|
|
// block which may use a rectangular transform, in which case it is
|
|
// "(max_txsize_lookup[bsize] + 1) - TX_8X8", invalid for bsize < 8X8
|
|
static const int32_t inter_tx_size_cat_lookup[BLOCK_SIZES] = {
|
|
// 4X4
|
|
INT32_MIN,
|
|
// 4X8, 8X4, 8X8
|
|
INT32_MIN, INT32_MIN, TX_8X8 - TX_8X8,
|
|
// 8X16, 16X8, 16X16
|
|
TX_16X16 - TX_8X8, TX_16X16 - TX_8X8, TX_16X16 - TX_8X8,
|
|
// 16X32, 32X16, 32X32
|
|
TX_32X32 - TX_8X8, TX_32X32 - TX_8X8, TX_32X32 - TX_8X8,
|
|
// 32X64, 64X32, 64X64
|
|
TX_32X32 - TX_8X8, TX_32X32 - TX_8X8, TX_32X32 - TX_8X8,
|
|
#if CONFIG_EXT_PARTITION
|
|
// 64x128, 128x64, 128x128
|
|
TX_32X32 - TX_8X8, TX_32X32 - TX_8X8, TX_32X32 - TX_8X8,
|
|
#endif // CONFIG_EXT_PARTITION
|
|
};
|
|
#else
|
|
#define inter_tx_size_cat_lookup intra_tx_size_cat_lookup
|
|
#endif // CONFIG_EXT_TX && CONFIG_RECT_TX
|
|
|
|
/* clang-format on */
|
|
|
|
static const TX_SIZE sub_tx_size_map[TX_SIZES_ALL] = {
|
|
#if CONFIG_CB4X4
|
|
TX_2X2, // TX_2X2
|
|
#endif
|
|
TX_4X4, // TX_4X4
|
|
TX_4X4, // TX_8X8
|
|
TX_8X8, // TX_16X16
|
|
TX_16X16, // TX_32X32
|
|
TX_4X4, // TX_4X8
|
|
TX_4X4, // TX_8X4
|
|
TX_8X8, // TX_8X16
|
|
TX_8X8, // TX_16X8
|
|
TX_16X16, // TX_16X32
|
|
TX_16X16 // TX_32X16
|
|
};
|
|
|
|
static const TX_SIZE txsize_horz_map[TX_SIZES_ALL] = {
|
|
#if CONFIG_CB4X4
|
|
TX_2X2, // TX_2X2
|
|
#endif
|
|
TX_4X4, // TX_4X4
|
|
TX_8X8, // TX_8X8
|
|
TX_16X16, // TX_16X16
|
|
TX_32X32, // TX_32X32
|
|
TX_4X4, // TX_4X8
|
|
TX_8X8, // TX_8X4
|
|
TX_8X8, // TX_8X16
|
|
TX_16X16, // TX_16X8
|
|
TX_16X16, // TX_16X32
|
|
TX_32X32 // TX_32X16
|
|
};
|
|
|
|
static const TX_SIZE txsize_vert_map[TX_SIZES_ALL] = {
|
|
#if CONFIG_CB4X4
|
|
TX_2X2, // TX_2X2
|
|
#endif
|
|
TX_4X4, // TX_4X4
|
|
TX_8X8, // TX_8X8
|
|
TX_16X16, // TX_16X16
|
|
TX_32X32, // TX_32X32
|
|
TX_8X8, // TX_4X8
|
|
TX_4X4, // TX_8X4
|
|
TX_16X16, // TX_8X16
|
|
TX_8X8, // TX_16X8
|
|
TX_32X32, // TX_16X32
|
|
TX_16X16 // TX_32X16
|
|
};
|
|
|
|
// Transform block width in pixels
|
|
static const int tx_size_wide[TX_SIZES_ALL] = {
|
|
#if CONFIG_CB4X4
|
|
2,
|
|
#endif
|
|
4, 8, 16, 32, 4, 8, 8, 16, 16, 32,
|
|
};
|
|
|
|
// Transform block height in pixels
|
|
static const int tx_size_high[TX_SIZES_ALL] = {
|
|
#if CONFIG_CB4X4
|
|
2,
|
|
#endif
|
|
4, 8, 16, 32, 8, 4, 16, 8, 32, 16,
|
|
};
|
|
|
|
// Transform block width in unit
|
|
static const int tx_size_wide_unit[TX_SIZES_ALL] = {
|
|
#if CONFIG_CB4X4
|
|
1,
|
|
#endif
|
|
1, 2, 4, 8, 1, 2, 2, 4, 4, 8,
|
|
};
|
|
|
|
// Transform block height in unit
|
|
static const int tx_size_high_unit[TX_SIZES_ALL] = {
|
|
#if CONFIG_CB4X4
|
|
1,
|
|
#endif
|
|
1, 2, 4, 8, 2, 1, 4, 2, 8, 4,
|
|
};
|
|
|
|
// Transform block width in log2
|
|
static const int tx_size_wide_log2[TX_SIZES_ALL] = {
|
|
#if CONFIG_CB4X4
|
|
2,
|
|
#endif
|
|
2, 3, 4, 5, 2, 3, 3, 4, 4, 5,
|
|
};
|
|
|
|
// Transform block height in log2
|
|
static const int tx_size_high_log2[TX_SIZES_ALL] = {
|
|
#if CONFIG_CB4X4
|
|
2,
|
|
#endif
|
|
2, 3, 4, 5, 3, 2, 4, 3, 5, 4,
|
|
};
|
|
|
|
static const int tx_size_2d[TX_SIZES_ALL] = {
|
|
#if CONFIG_CB4X4
|
|
4,
|
|
#endif
|
|
16, 64, 256, 1024, 32, 32, 128, 128, 512, 512,
|
|
};
|
|
|
|
static const uint8_t tx_size_1d_log2[TX_SIZES] = { 2, 3, 4, 5 };
|
|
|
|
// TODO(jingning): Temporary table during the construction.
|
|
static const int tx_size_1d_in_unit_log2[TX_SIZES] = { 0, 1, 2, 3 };
|
|
|
|
static const BLOCK_SIZE txsize_to_bsize[TX_SIZES_ALL] = {
|
|
#if CONFIG_CB4X4
|
|
BLOCK_4X4, // TX_2X2
|
|
#endif
|
|
BLOCK_4X4, // TX_4X4
|
|
BLOCK_8X8, // TX_8X8
|
|
BLOCK_16X16, // TX_16X16
|
|
BLOCK_32X32, // TX_32X32
|
|
BLOCK_4X8, // TX_4X8
|
|
BLOCK_8X4, // TX_8X4
|
|
BLOCK_8X16, // TX_8X16
|
|
BLOCK_16X8, // TX_16X8
|
|
BLOCK_16X32, // TX_16X32
|
|
BLOCK_32X16, // TX_32X16
|
|
};
|
|
|
|
static const TX_SIZE txsize_sqr_map[TX_SIZES_ALL] = {
|
|
#if CONFIG_CB4X4
|
|
TX_2X2, // TX_2X2
|
|
#endif
|
|
TX_4X4, // TX_4X4
|
|
TX_8X8, // TX_8X8
|
|
TX_16X16, // TX_16X16
|
|
TX_32X32, // TX_32X32
|
|
TX_4X4, // TX_4X8
|
|
TX_4X4, // TX_8X4
|
|
TX_8X8, // TX_8X16
|
|
TX_8X8, // TX_16X8
|
|
TX_16X16, // TX_16X32
|
|
TX_16X16, // TX_32X16
|
|
};
|
|
|
|
static const TX_SIZE txsize_sqr_up_map[TX_SIZES_ALL] = {
|
|
#if CONFIG_CB4X4
|
|
TX_2X2, // TX_2X2
|
|
#endif
|
|
TX_4X4, // TX_4X4
|
|
TX_8X8, // TX_8X8
|
|
TX_16X16, // TX_16X16
|
|
TX_32X32, // TX_32X32
|
|
TX_8X8, // TX_4X8
|
|
TX_8X8, // TX_8X4
|
|
TX_16X16, // TX_8X16
|
|
TX_16X16, // TX_16X8
|
|
TX_32X32, // TX_16X32
|
|
TX_32X32, // TX_32X16
|
|
};
|
|
|
|
static const TX_SIZE tx_mode_to_biggest_tx_size[TX_MODES] = {
|
|
TX_4X4, // ONLY_4X4
|
|
TX_8X8, // ALLOW_8X8
|
|
TX_16X16, // ALLOW_16X16
|
|
TX_32X32, // ALLOW_32X32
|
|
TX_32X32, // TX_MODE_SELECT
|
|
};
|
|
|
|
static const BLOCK_SIZE ss_size_lookup[BLOCK_SIZES][2][2] = {
|
|
// ss_x == 0 ss_x == 0 ss_x == 1 ss_x == 1
|
|
// ss_y == 0 ss_y == 1 ss_y == 0 ss_y == 1
|
|
{ { BLOCK_4X4, BLOCK_INVALID }, { BLOCK_INVALID, BLOCK_INVALID } },
|
|
{ { BLOCK_4X8, BLOCK_4X4 }, { BLOCK_INVALID, BLOCK_INVALID } },
|
|
{ { BLOCK_8X4, BLOCK_INVALID }, { BLOCK_4X4, BLOCK_INVALID } },
|
|
{ { BLOCK_8X8, BLOCK_8X4 }, { BLOCK_4X8, BLOCK_4X4 } },
|
|
{ { BLOCK_8X16, BLOCK_8X8 }, { BLOCK_INVALID, BLOCK_4X8 } },
|
|
{ { BLOCK_16X8, BLOCK_INVALID }, { BLOCK_8X8, BLOCK_8X4 } },
|
|
{ { BLOCK_16X16, BLOCK_16X8 }, { BLOCK_8X16, BLOCK_8X8 } },
|
|
{ { BLOCK_16X32, BLOCK_16X16 }, { BLOCK_INVALID, BLOCK_8X16 } },
|
|
{ { BLOCK_32X16, BLOCK_INVALID }, { BLOCK_16X16, BLOCK_16X8 } },
|
|
{ { BLOCK_32X32, BLOCK_32X16 }, { BLOCK_16X32, BLOCK_16X16 } },
|
|
{ { BLOCK_32X64, BLOCK_32X32 }, { BLOCK_INVALID, BLOCK_16X32 } },
|
|
{ { BLOCK_64X32, BLOCK_INVALID }, { BLOCK_32X32, BLOCK_32X16 } },
|
|
{ { BLOCK_64X64, BLOCK_64X32 }, { BLOCK_32X64, BLOCK_32X32 } },
|
|
#if CONFIG_EXT_PARTITION
|
|
{ { BLOCK_64X128, BLOCK_64X64 }, { BLOCK_INVALID, BLOCK_32X64 } },
|
|
{ { BLOCK_128X64, BLOCK_INVALID }, { BLOCK_64X64, BLOCK_64X32 } },
|
|
{ { BLOCK_128X128, BLOCK_128X64 }, { BLOCK_64X128, BLOCK_64X64 } },
|
|
#endif // CONFIG_EXT_PARTITION
|
|
};
|
|
|
|
#define USE_UV_RECT_TX 1
|
|
static const TX_SIZE uv_txsize_lookup[BLOCK_SIZES][TX_SIZES_ALL][2][2] = {
|
|
// ss_x == 0 ss_x == 0 ss_x == 1 ss_x == 1
|
|
// ss_y == 0 ss_y == 1 ss_y == 0 ss_y == 1
|
|
{
|
|
// BLOCK_4X4
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
},
|
|
{
|
|
// BLOCK_4X8
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#if CONFIG_RECT_TX && USE_UV_RECT_TX
|
|
{ { TX_4X8, TX_4X4 }, { TX_4X4, TX_4X4 } }, // used
|
|
#else
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } }, // used
|
|
#endif // CONFIG_RECT_TX && USE_UV_RECT_TX
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
},
|
|
{
|
|
// BLOCK_8X4
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
#if CONFIG_RECT_TX && USE_UV_RECT_TX
|
|
{ { TX_8X4, TX_4X4 }, { TX_4X4, TX_4X4 } }, // used
|
|
#else
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } }, // used
|
|
#endif // CONFIG_RECT_TX && USE_UV_RECT_TX
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
},
|
|
{
|
|
// BLOCK_8X8
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X8, TX_4X4 }, { TX_4X8, TX_4X4 } },
|
|
{ { TX_8X4, TX_8X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X4 }, { TX_4X8, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X4 }, { TX_4X8, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X4 }, { TX_4X8, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X4 }, { TX_4X8, TX_4X4 } },
|
|
},
|
|
{
|
|
// BLOCK_8X16
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_4X8, TX_4X8 }, { TX_4X8, TX_4X8 } },
|
|
{ { TX_8X4, TX_8X4 }, { TX_4X4, TX_4X4 } },
|
|
#if CONFIG_RECT_TX && USE_UV_RECT_TX
|
|
{ { TX_8X16, TX_8X8 }, { TX_4X8, TX_4X8 } }, // used
|
|
#else
|
|
{ { TX_8X8, TX_8X8 }, { TX_4X4, TX_4X4 } }, // used
|
|
#endif // CONFIG_RECT_TX && USE_UV_RECT_TX
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
},
|
|
{
|
|
// BLOCK_16X8
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_4X4 }, { TX_8X8, TX_4X4 } },
|
|
{ { TX_8X8, TX_4X4 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_8X8, TX_4X4 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_4X8, TX_4X4 }, { TX_4X8, TX_4X4 } },
|
|
{ { TX_8X4, TX_8X4 }, { TX_8X4, TX_8X4 } },
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
#if CONFIG_RECT_TX && USE_UV_RECT_TX
|
|
{ { TX_16X8, TX_8X4 }, { TX_8X8, TX_8X4 } }, // used
|
|
#else
|
|
{ { TX_8X8, TX_4X4 }, { TX_8X8, TX_4X4 } }, // used
|
|
#endif // CONFIG_RECT_TX && USE_UV_RECT_TX
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
},
|
|
{
|
|
// BLOCK_16X16
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_16X16, TX_8X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_16X16, TX_8X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_4X8, TX_4X8 }, { TX_4X8, TX_4X8 } },
|
|
{ { TX_8X4, TX_8X4 }, { TX_8X4, TX_8X4 } },
|
|
{ { TX_8X16, TX_8X8 }, { TX_8X16, TX_8X8 } },
|
|
{ { TX_16X8, TX_16X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_16X16, TX_16X8 }, { TX_8X16, TX_8X8 } },
|
|
{ { TX_16X16, TX_16X8 }, { TX_8X16, TX_8X8 } },
|
|
},
|
|
{
|
|
// BLOCK_16X32
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_16X16, TX_16X16 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_16X16, TX_16X16 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_4X8, TX_4X8 }, { TX_4X8, TX_4X8 } },
|
|
{ { TX_8X4, TX_8X4 }, { TX_8X4, TX_8X4 } },
|
|
{ { TX_8X16, TX_8X16 }, { TX_8X16, TX_8X16 } },
|
|
{ { TX_16X8, TX_16X8 }, { TX_8X8, TX_8X8 } },
|
|
#if CONFIG_RECT_TX && USE_UV_RECT_TX
|
|
{ { TX_16X32, TX_16X16 }, { TX_8X16, TX_8X16 } }, // used
|
|
#else
|
|
{ { TX_16X16, TX_16X16 }, { TX_8X8, TX_8X8 } }, // used
|
|
#endif // CONFIG_RECT_TX && USE_UV_RECT_TX
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
},
|
|
{
|
|
// BLOCK_32X16
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_16X16, TX_8X8 }, { TX_16X16, TX_8X8 } },
|
|
{ { TX_16X16, TX_8X8 }, { TX_16X16, TX_8X8 } },
|
|
{ { TX_4X8, TX_4X8 }, { TX_4X8, TX_4X8 } },
|
|
{ { TX_8X4, TX_8X4 }, { TX_8X4, TX_8X4 } },
|
|
{ { TX_8X16, TX_8X8 }, { TX_8X16, TX_8X8 } },
|
|
{ { TX_16X8, TX_16X8 }, { TX_16X8, TX_16X8 } },
|
|
{ { TX_INVALID, TX_INVALID }, { TX_INVALID, TX_INVALID } },
|
|
#if CONFIG_RECT_TX && USE_UV_RECT_TX
|
|
{ { TX_32X16, TX_16X8 }, { TX_16X16, TX_16X8 } }, // used
|
|
#else
|
|
{ { TX_16X16, TX_8X8 }, { TX_16X16, TX_8X8 } }, // used
|
|
#endif // CONFIG_RECT_TX && USE_UV_RECT_TX
|
|
},
|
|
{
|
|
// BLOCK_32X32
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_16X16, TX_16X16 }, { TX_16X16, TX_16X16 } },
|
|
{ { TX_32X32, TX_16X16 }, { TX_16X16, TX_16X16 } },
|
|
{ { TX_4X8, TX_4X8 }, { TX_4X8, TX_4X8 } },
|
|
{ { TX_8X4, TX_8X4 }, { TX_8X4, TX_8X4 } },
|
|
{ { TX_8X16, TX_8X16 }, { TX_8X16, TX_8X16 } },
|
|
{ { TX_16X8, TX_16X8 }, { TX_16X8, TX_16X8 } },
|
|
{ { TX_16X32, TX_16X16 }, { TX_16X32, TX_16X16 } },
|
|
{ { TX_32X16, TX_32X16 }, { TX_16X16, TX_16X16 } },
|
|
},
|
|
{
|
|
// BLOCK_32X64
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_16X16, TX_16X16 }, { TX_16X16, TX_16X16 } },
|
|
{ { TX_32X32, TX_32X32 }, { TX_16X16, TX_16X16 } },
|
|
{ { TX_4X8, TX_4X8 }, { TX_4X8, TX_4X8 } },
|
|
{ { TX_8X4, TX_8X4 }, { TX_8X4, TX_8X4 } },
|
|
{ { TX_8X16, TX_8X16 }, { TX_8X16, TX_8X16 } },
|
|
{ { TX_16X8, TX_16X8 }, { TX_16X8, TX_16X8 } },
|
|
{ { TX_16X32, TX_16X32 }, { TX_16X16, TX_16X16 } },
|
|
{ { TX_32X16, TX_32X16 }, { TX_16X16, TX_16X16 } },
|
|
},
|
|
{
|
|
// BLOCK_64X32
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_16X16, TX_16X16 }, { TX_16X16, TX_16X16 } },
|
|
{ { TX_32X32, TX_16X16 }, { TX_32X32, TX_16X16 } },
|
|
{ { TX_4X8, TX_4X8 }, { TX_4X8, TX_4X8 } },
|
|
{ { TX_8X4, TX_8X4 }, { TX_8X4, TX_8X4 } },
|
|
{ { TX_8X16, TX_8X16 }, { TX_8X16, TX_8X16 } },
|
|
{ { TX_16X8, TX_16X8 }, { TX_16X8, TX_16X8 } },
|
|
{ { TX_16X32, TX_16X16 }, { TX_16X32, TX_16X16 } },
|
|
{ { TX_32X16, TX_16X16 }, { TX_32X16, TX_16X16 } },
|
|
},
|
|
{
|
|
// BLOCK_64X64
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_16X16, TX_16X16 }, { TX_16X16, TX_16X16 } },
|
|
{ { TX_32X32, TX_32X32 }, { TX_32X32, TX_32X32 } },
|
|
#if CONFIG_EXT_PARTITION
|
|
},
|
|
{
|
|
// BLOCK_64X128
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_16X16, TX_16X16 }, { TX_16X16, TX_16X16 } },
|
|
{ { TX_32X32, TX_32X32 }, { TX_32X32, TX_32X32 } },
|
|
{ { TX_4X8, TX_4X8 }, { TX_4X8, TX_4X8 } },
|
|
{ { TX_8X4, TX_8X4 }, { TX_8X4, TX_8X4 } },
|
|
{ { TX_8X16, TX_8X16 }, { TX_8X16, TX_8X16 } },
|
|
{ { TX_16X8, TX_16X8 }, { TX_16X8, TX_16X8 } },
|
|
{ { TX_16X32, TX_16X32 }, { TX_16X32, TX_16X32 } },
|
|
{ { TX_32X16, TX_32X16 }, { TX_32X16, TX_32X16 } },
|
|
},
|
|
{
|
|
// BLOCK_128X64
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_16X16, TX_16X16 }, { TX_16X16, TX_16X16 } },
|
|
{ { TX_32X32, TX_32X32 }, { TX_32X32, TX_32X32 } },
|
|
{ { TX_4X8, TX_4X8 }, { TX_4X8, TX_4X8 } },
|
|
{ { TX_8X4, TX_8X4 }, { TX_8X4, TX_8X4 } },
|
|
{ { TX_8X16, TX_8X16 }, { TX_8X16, TX_8X16 } },
|
|
{ { TX_16X8, TX_16X8 }, { TX_16X8, TX_16X8 } },
|
|
{ { TX_16X32, TX_16X32 }, { TX_16X32, TX_16X32 } },
|
|
{ { TX_32X16, TX_32X16 }, { TX_32X16, TX_32X16 } },
|
|
},
|
|
{
|
|
// BLOCK_128X128
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_8X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_16X16, TX_16X16 }, { TX_16X16, TX_16X16 } },
|
|
{ { TX_32X32, TX_32X32 }, { TX_32X32, TX_32X32 } },
|
|
{ { TX_4X8, TX_4X8 }, { TX_4X8, TX_4X8 } },
|
|
{ { TX_8X4, TX_8X4 }, { TX_8X4, TX_8X4 } },
|
|
{ { TX_8X16, TX_8X16 }, { TX_8X16, TX_8X16 } },
|
|
{ { TX_16X8, TX_16X8 }, { TX_16X8, TX_16X8 } },
|
|
{ { TX_16X32, TX_16X32 }, { TX_16X32, TX_16X32 } },
|
|
{ { TX_32X16, TX_32X16 }, { TX_32X16, TX_32X16 } },
|
|
#endif // CONFIG_EXT_PARTITION
|
|
},
|
|
};
|
|
|
|
// Generates 4 bit field in which each bit set to 1 represents
|
|
// a blocksize partition 1111 means we split 64x64, 32x32, 16x16
|
|
// and 8x8. 1000 means we just split the 64x64 to 32x32
|
|
static const struct {
|
|
PARTITION_CONTEXT above;
|
|
PARTITION_CONTEXT left;
|
|
} partition_context_lookup[BLOCK_SIZES] = {
|
|
#if CONFIG_EXT_PARTITION
|
|
{ 31, 31 }, // 4X4 - {0b11111, 0b11111}
|
|
{ 31, 30 }, // 4X8 - {0b11111, 0b11110}
|
|
{ 30, 31 }, // 8X4 - {0b11110, 0b11111}
|
|
{ 30, 30 }, // 8X8 - {0b11110, 0b11110}
|
|
{ 30, 28 }, // 8X16 - {0b11110, 0b11100}
|
|
{ 28, 30 }, // 16X8 - {0b11100, 0b11110}
|
|
{ 28, 28 }, // 16X16 - {0b11100, 0b11100}
|
|
{ 28, 24 }, // 16X32 - {0b11100, 0b11000}
|
|
{ 24, 28 }, // 32X16 - {0b11000, 0b11100}
|
|
{ 24, 24 }, // 32X32 - {0b11000, 0b11000}
|
|
{ 24, 16 }, // 32X64 - {0b11000, 0b10000}
|
|
{ 16, 24 }, // 64X32 - {0b10000, 0b11000}
|
|
{ 16, 16 }, // 64X64 - {0b10000, 0b10000}
|
|
{ 16, 0 }, // 64X128- {0b10000, 0b00000}
|
|
{ 0, 16 }, // 128X64- {0b00000, 0b10000}
|
|
{ 0, 0 }, // 128X128-{0b00000, 0b00000}
|
|
#else
|
|
{ 15, 15 }, // 4X4 - {0b1111, 0b1111}
|
|
{ 15, 14 }, // 4X8 - {0b1111, 0b1110}
|
|
{ 14, 15 }, // 8X4 - {0b1110, 0b1111}
|
|
{ 14, 14 }, // 8X8 - {0b1110, 0b1110}
|
|
{ 14, 12 }, // 8X16 - {0b1110, 0b1100}
|
|
{ 12, 14 }, // 16X8 - {0b1100, 0b1110}
|
|
{ 12, 12 }, // 16X16 - {0b1100, 0b1100}
|
|
{ 12, 8 }, // 16X32 - {0b1100, 0b1000}
|
|
{ 8, 12 }, // 32X16 - {0b1000, 0b1100}
|
|
{ 8, 8 }, // 32X32 - {0b1000, 0b1000}
|
|
{ 8, 0 }, // 32X64 - {0b1000, 0b0000}
|
|
{ 0, 8 }, // 64X32 - {0b0000, 0b1000}
|
|
{ 0, 0 }, // 64X64 - {0b0000, 0b0000}
|
|
#endif // CONFIG_EXT_PARTITION
|
|
};
|
|
|
|
#if CONFIG_SUPERTX
|
|
static const TX_SIZE uvsupertx_size_lookup[TX_SIZES][2][2] = {
|
|
// ss_x == 0 ss_x == 0 ss_x == 1 ss_x == 1
|
|
// ss_y == 0 ss_y == 1 ss_y == 0 ss_y == 1
|
|
#if CONFIG_CB4X4
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
#endif
|
|
{ { TX_4X4, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_8X8, TX_4X4 }, { TX_4X4, TX_4X4 } },
|
|
{ { TX_16X16, TX_8X8 }, { TX_8X8, TX_8X8 } },
|
|
{ { TX_32X32, TX_16X16 }, { TX_16X16, TX_16X16 } },
|
|
};
|
|
|
|
#if CONFIG_EXT_PARTITION_TYPES
|
|
static const int partition_supertx_context_lookup[EXT_PARTITION_TYPES] = {
|
|
-1, 0, 0, 1, 0, 0, 0, 0
|
|
};
|
|
|
|
#else
|
|
static const int partition_supertx_context_lookup[PARTITION_TYPES] = { -1, 0, 0,
|
|
1 };
|
|
#endif // CONFIG_EXT_PARTITION_TYPES
|
|
#endif // CONFIG_SUPERTX
|
|
|
|
#ifdef __cplusplus
|
|
} // extern "C"
|
|
#endif
|
|
|
|
#endif // AV1_COMMON_COMMON_DATA_H_
|