| /* |
| * 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 AOM_AV1_COMMON_COMMON_DATA_H_ |
| #define AOM_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 |
| |
| // Log 2 conversion lookup tables in units of mode info (4x4). |
| // The Mi_Width_Log2 table in the spec (Section 9.3. Conversion tables). |
| static const uint8_t mi_size_wide_log2[BLOCK_SIZES_ALL] = { |
| 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 0, 2, 1, 3, 2, 4, |
| #if CONFIG_FLEX_PARTITION |
| 0, 3, 1, 4, 0, 4, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| // The Mi_Height_Log2 table in the spec (Section 9.3. Conversion tables). |
| static const uint8_t mi_size_high_log2[BLOCK_SIZES_ALL] = { |
| 0, 1, 0, 1, 2, 1, 2, 3, 2, 3, 4, 3, 4, 5, 4, 5, 2, 0, 3, 1, 4, 2, |
| #if CONFIG_FLEX_PARTITION |
| 3, 0, 4, 1, 4, 0, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| // Width/height lookup tables in units of mode info (4x4). |
| // The Num_4x4_Blocks_Wide table in the spec (Section 9.3. Conversion tables). |
| static const uint8_t mi_size_wide[BLOCK_SIZES_ALL] = { |
| 1, 1, 2, 2, 2, 4, 4, 4, 8, 8, 8, 16, 16, 16, 32, 32, 1, 4, 2, 8, 4, 16, |
| #if CONFIG_FLEX_PARTITION |
| 1, 8, 2, 16, 1, 16, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| // The Num_4x4_Blocks_High table in the spec (Section 9.3. Conversion tables). |
| static const uint8_t mi_size_high[BLOCK_SIZES_ALL] = { |
| 1, 2, 1, 2, 4, 2, 4, 8, 4, 8, 16, 8, 16, 32, 16, 32, 4, 1, 8, 2, 16, 4, |
| #if CONFIG_FLEX_PARTITION |
| 8, 1, 16, 2, 16, 1, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| // Width/height lookup tables in units of samples. |
| // The Block_Width table in the spec (Section 9.3. Conversion tables). |
| static const uint8_t block_size_wide[BLOCK_SIZES_ALL] = { |
| 4, 4, 8, 8, 8, 16, 16, 16, 32, 32, 32, |
| 64, 64, 64, 128, 128, 4, 16, 8, 32, 16, 64, |
| #if CONFIG_FLEX_PARTITION |
| 4, 32, 8, 64, 4, 64, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| // The Block_Height table in the spec (Section 9.3. Conversion tables). |
| static const uint8_t block_size_high[BLOCK_SIZES_ALL] = { |
| 4, 8, 4, 8, 16, 8, 16, 32, 16, 32, 64, |
| 32, 64, 128, 64, 128, 16, 4, 32, 8, 64, 16, |
| #if CONFIG_FLEX_PARTITION |
| 32, 4, 64, 8, 64, 4, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| // Maps a block size to a context. |
| // The Size_Group table in the spec (Section 9.3. Conversion tables). |
| // AOMMIN(3, AOMMIN(mi_size_wide_log2(bsize), mi_size_high_log2(bsize))) |
| static const uint8_t size_group_lookup[BLOCK_SIZES_ALL] = { |
| 0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 0, 0, 1, 1, 2, 2, |
| #if CONFIG_FLEX_PARTITION |
| 1, 1, 2, 2, 2, 2, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| static const uint8_t num_pels_log2_lookup[BLOCK_SIZES_ALL] = { |
| 4, 5, 5, 6, 7, 7, 8, 9, 9, 10, 11, 11, 12, 13, 13, 14, 6, 6, 8, 8, 10, 10, |
| #if CONFIG_FLEX_PARTITION |
| 7, 7, 9, 9, 8, 8, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| #if CONFIG_EXT_RECUR_PARTITIONS |
| static const PARTITION_TYPE |
| partition_map_from_symbol_block_wgth[PARTITION_TYPES_REC] = { |
| PARTITION_NONE, PARTITION_VERT, PARTITION_VERT_3, PARTITION_HORZ, |
| #if CONFIG_FLEX_PARTITION |
| PARTITION_HORZ_3, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| static const PARTITION_TYPE_REC |
| symbol_map_from_partition_block_wgth[EXT_PARTITION_TYPES] = { |
| PARTITION_NONE_REC, PARTITION_SHORT_SIDE_2_REC, |
| PARTITION_LONG_SIDE_2_REC, |
| #if KEEP_PARTITION_SPLIT |
| PARTITION_INVALID_REC, |
| #endif // KEEP_PARTITION_SPLIT |
| #if CONFIG_FLEX_PARTITION |
| PARTITION_SHORT_SIDE_3_REC, |
| #else // CONFIG_FLEX_PARTITION |
| PARTITION_INVALID_REC, |
| #endif // CONFIG_FLEX_PARTITION |
| PARTITION_LONG_SIDE_3_REC, |
| }; |
| |
| static const PARTITION_TYPE |
| partition_map_from_symbol_block_hgtw[PARTITION_TYPES_REC] = { |
| PARTITION_NONE, PARTITION_HORZ, PARTITION_HORZ_3, PARTITION_VERT, |
| #if CONFIG_FLEX_PARTITION |
| PARTITION_VERT_3, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| static const PARTITION_TYPE_REC |
| symbol_map_from_partition_block_hgtw[EXT_PARTITION_TYPES] = { |
| PARTITION_NONE_REC, PARTITION_LONG_SIDE_2_REC, |
| PARTITION_SHORT_SIDE_2_REC, |
| #if KEEP_PARTITION_SPLIT |
| PARTITION_INVALID_REC, |
| #endif // KEEP_PARTITION_SPLIT |
| PARTITION_LONG_SIDE_3_REC, |
| #if CONFIG_FLEX_PARTITION |
| PARTITION_SHORT_SIDE_3_REC, |
| #else // CONFIG_FLEX_PARTITION |
| PARTITION_INVALID_REC, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| /* clang-format off */ |
| // This table covers all square blocks and 1:2/2:1 rectangular blocks |
| #if KEEP_PARTITION_SPLIT |
| static const BLOCK_SIZE subsize_lookup[EXT_PARTITION_TYPES][BLOCK_SIZES_ALL] = { |
| #else |
| static const BLOCK_SIZE |
| subsize_lookup[EXT_PARTITION_TYPES + 1][BLOCK_SIZES_ALL] = { |
| #endif // KEEP_PARTITION_SPLIT |
| { // PARTITION_NONE |
| BLOCK_4X4, BLOCK_4X8, BLOCK_8X4, BLOCK_8X8, BLOCK_8X16, BLOCK_16X8, |
| BLOCK_16X16, BLOCK_16X32, BLOCK_32X16, BLOCK_32X32, BLOCK_32X64, |
| BLOCK_64X32, BLOCK_64X64, BLOCK_64X128, BLOCK_128X64, BLOCK_128X128, |
| BLOCK_4X16, BLOCK_16X4, BLOCK_8X32, BLOCK_32X8, BLOCK_16X64, BLOCK_64X16, |
| #if CONFIG_FLEX_PARTITION |
| BLOCK_4X32, BLOCK_32X4, BLOCK_8X64, BLOCK_64X8, BLOCK_4X64, BLOCK_64X4, |
| #endif // CONFIG_FLEX_PARTITION |
| }, { // PARTITION_HORZ |
| BLOCK_INVALID, BLOCK_4X4, BLOCK_INVALID, BLOCK_8X4, BLOCK_8X8, BLOCK_16X4, |
| BLOCK_16X8, BLOCK_16X16, BLOCK_32X8, BLOCK_32X16, BLOCK_32X32, BLOCK_64X16, |
| BLOCK_64X32, BLOCK_64X64, BLOCK_INVALID, BLOCK_128X64, |
| #if CONFIG_FLEX_PARTITION |
| BLOCK_4X8, BLOCK_INVALID, BLOCK_8X16, BLOCK_32X4, BLOCK_16X32, BLOCK_64X8, |
| BLOCK_4X16, BLOCK_INVALID, BLOCK_8X32, BLOCK_64X4, BLOCK_4X32, |
| BLOCK_INVALID, |
| #else // CONFIG_FLEX_PARTITION |
| BLOCK_4X8, BLOCK_INVALID, BLOCK_8X16, BLOCK_INVALID, BLOCK_16X32, |
| BLOCK_INVALID, |
| #endif // CONFIG_FLEX_PARTITION |
| }, { // PARTITION_VERT |
| BLOCK_INVALID, BLOCK_INVALID, BLOCK_4X4, BLOCK_4X8, BLOCK_4X16, BLOCK_8X8, |
| BLOCK_8X16, BLOCK_8X32, BLOCK_16X16, BLOCK_16X32, BLOCK_16X64, BLOCK_32X32, |
| BLOCK_32X64, BLOCK_INVALID, BLOCK_64X64, BLOCK_64X128, |
| #if CONFIG_FLEX_PARTITION |
| BLOCK_INVALID, BLOCK_8X4, BLOCK_4X32, BLOCK_16X8, BLOCK_8X64, BLOCK_32X16, |
| BLOCK_INVALID, BLOCK_16X4, BLOCK_4X64, BLOCK_32X8, BLOCK_INVALID, |
| BLOCK_32X4, |
| #else // CONFIG_FLEX_PARTITION |
| BLOCK_INVALID, BLOCK_8X4, BLOCK_INVALID, BLOCK_16X8, BLOCK_INVALID, |
| BLOCK_32X16, |
| #endif // CONFIG_FLEX_PARTITION |
| #if KEEP_PARTITION_SPLIT |
| }, { // PARTITION_SPLIT |
| BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, BLOCK_4X4, BLOCK_INVALID, |
| BLOCK_INVALID, BLOCK_8X8, BLOCK_INVALID, BLOCK_INVALID, BLOCK_16X16, |
| BLOCK_INVALID, BLOCK_INVALID, BLOCK_32X32, BLOCK_INVALID, BLOCK_INVALID, |
| BLOCK_64X64, BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, |
| BLOCK_INVALID, BLOCK_INVALID, |
| #if CONFIG_FLEX_PARTITION |
| BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, |
| BLOCK_INVALID, |
| #endif // CONFIG_FLEX_PARTITION |
| #endif // KEEP_PARTITION_SPLIT |
| }, { // PARTITION_HORZ_3 |
| BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, BLOCK_8X4, |
| BLOCK_INVALID, BLOCK_16X4, BLOCK_16X8, |
| #if CONFIG_FLEX_PARTITION |
| BLOCK_32X4, |
| #else // CONFIG_FLEX_PARTITION |
| BLOCK_INVALID, |
| #endif // CONFIG_FLEX_PARTITION |
| BLOCK_32X8, BLOCK_32X16, |
| #if CONFIG_FLEX_PARTITION |
| BLOCK_64X8, |
| #else // CONFIG_FLEX_PARTITION |
| BLOCK_INVALID, |
| #endif // CONFIG_FLEX_PARTITION |
| BLOCK_64X16, BLOCK_64X32, BLOCK_INVALID, BLOCK_INVALID, |
| BLOCK_4X4, BLOCK_INVALID, BLOCK_8X8, BLOCK_INVALID, BLOCK_16X16, |
| #if CONFIG_FLEX_PARTITION |
| BLOCK_64X4, BLOCK_4X8, BLOCK_INVALID, BLOCK_8X16, BLOCK_INVALID, BLOCK_4X16, |
| BLOCK_INVALID, |
| #else // CONFIG_FLEX_PARTITION |
| BLOCK_INVALID, |
| #endif // CONFIG_FLEX_PARTITION |
| }, { // PARTITION_VERT_3 |
| BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, |
| BLOCK_4X8, BLOCK_4X16, |
| #if CONFIG_FLEX_PARTITION |
| BLOCK_4X32, |
| #else // CONFIG_FLEX_PARTITION |
| BLOCK_INVALID, |
| #endif // CONFIG_FLEX_PARTITION |
| BLOCK_8X16, BLOCK_8X32, |
| #if CONFIG_FLEX_PARTITION |
| BLOCK_8X64, |
| #else // CONFIG_FLEX_PARTITION |
| BLOCK_INVALID, |
| #endif // CONFIG_FLEX_PARTITION |
| BLOCK_16X32, BLOCK_16X64, BLOCK_INVALID, BLOCK_32X64, BLOCK_INVALID, |
| BLOCK_INVALID, BLOCK_4X4, BLOCK_INVALID, BLOCK_8X8, |
| #if CONFIG_FLEX_PARTITION |
| BLOCK_4X64, BLOCK_16X16, BLOCK_INVALID, BLOCK_8X4, BLOCK_INVALID, |
| BLOCK_16X8, BLOCK_INVALID, BLOCK_16X4, |
| #else // CONFIG_FLEX_PARTITION |
| BLOCK_INVALID, BLOCK_16X16, |
| #endif // CONFIG_FLEX_PARTITION |
| #if !KEEP_PARTITION_SPLIT |
| }, { // PARTITION_SPLIT |
| BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, BLOCK_4X4, BLOCK_INVALID, |
| BLOCK_INVALID, BLOCK_8X8, BLOCK_INVALID, BLOCK_INVALID, BLOCK_16X16, |
| BLOCK_INVALID, BLOCK_INVALID, BLOCK_32X32, BLOCK_INVALID, BLOCK_INVALID, |
| BLOCK_64X64, BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, |
| BLOCK_INVALID, BLOCK_INVALID, |
| #if CONFIG_FLEX_PARTITION |
| BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, BLOCK_INVALID, |
| BLOCK_INVALID, |
| #endif // CONFIG_FLEX_PARTITION |
| #endif // !KEEP_PARTITION_SPLIT |
| }, |
| }; |
| #else // CONFIG_EXT_RECUR_PARTITIONS |
| /* clang-format off */ |
| // A compressed version of the Partition_Subsize table in the spec (9.3. |
| // Conversion tables), for square block sizes only. |
| static const BLOCK_SIZE subsize_lookup[EXT_PARTITION_TYPES][SQR_BLOCK_SIZES] = { |
| { // PARTITION_NONE |
| BLOCK_4X4, BLOCK_8X8, BLOCK_16X16, |
| BLOCK_32X32, BLOCK_64X64, BLOCK_128X128 |
| }, { // PARTITION_HORZ |
| BLOCK_INVALID, BLOCK_8X4, BLOCK_16X8, |
| BLOCK_32X16, BLOCK_64X32, BLOCK_128X64 |
| }, { // PARTITION_VERT |
| BLOCK_INVALID, BLOCK_4X8, BLOCK_8X16, |
| BLOCK_16X32, BLOCK_32X64, BLOCK_64X128 |
| }, { // PARTITION_SPLIT |
| BLOCK_INVALID, BLOCK_4X4, BLOCK_8X8, |
| BLOCK_16X16, BLOCK_32X32, BLOCK_64X64 |
| }, { // PARTITION_HORZ_A |
| BLOCK_INVALID, BLOCK_8X4, BLOCK_16X8, |
| BLOCK_32X16, BLOCK_64X32, BLOCK_128X64 |
| }, { // PARTITION_HORZ_B |
| BLOCK_INVALID, BLOCK_8X4, BLOCK_16X8, |
| BLOCK_32X16, BLOCK_64X32, BLOCK_128X64 |
| }, { // PARTITION_VERT_A |
| BLOCK_INVALID, BLOCK_4X8, BLOCK_8X16, |
| BLOCK_16X32, BLOCK_32X64, BLOCK_64X128 |
| }, { // PARTITION_VERT_B |
| BLOCK_INVALID, BLOCK_4X8, BLOCK_8X16, |
| BLOCK_16X32, BLOCK_32X64, BLOCK_64X128 |
| }, { // PARTITION_HORZ_4 |
| BLOCK_INVALID, BLOCK_INVALID, BLOCK_16X4, |
| BLOCK_32X8, BLOCK_64X16, BLOCK_INVALID |
| }, { // PARTITION_VERT_4 |
| BLOCK_INVALID, BLOCK_INVALID, BLOCK_4X16, |
| BLOCK_8X32, BLOCK_16X64, BLOCK_INVALID |
| } |
| }; |
| #endif // CONFIG_EXT_RECUR_PARTITIONS |
| |
| static const TX_SIZE max_txsize_lookup[BLOCK_SIZES_ALL] = { |
| // 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, |
| TX_32X32, TX_32X32, |
| // 64X64 |
| TX_64X64, |
| // 64x128, 128x64, 128x128 |
| TX_64X64, TX_64X64, TX_64X64, |
| // 4x16, 16x4, 8x32 |
| TX_4X4, TX_4X4, TX_8X8, |
| // 32x8, 16x64, 64x16 |
| TX_8X8, TX_16X16, TX_16X16, |
| #if CONFIG_FLEX_PARTITION |
| // 4x32, 32x4, 8x64 |
| TX_4X4, TX_4X4, TX_8X8, |
| // 64x8, 4x64, 64x4 |
| TX_8X8, TX_4X4, TX_4X4, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| static const TX_SIZE max_txsize_rect_lookup[BLOCK_SIZES_ALL] = { |
| // 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, |
| TX_32X64, TX_64X32, |
| // 64X64 |
| TX_64X64, |
| // 64x128, 128x64, 128x128 |
| TX_64X64, TX_64X64, TX_64X64, |
| // 4x16, 16x4, |
| TX_4X16, TX_16X4, |
| // 8x32, 32x8 |
| TX_8X32, TX_32X8, |
| // 16x64, 64x16 |
| TX_16X64, TX_64X16, |
| #if CONFIG_FLEX_PARTITION |
| // 4x32, 32x4, |
| TX_4X32, TX_32X4, |
| // 8x64, 64x8 |
| TX_8X64, TX_64X8, |
| // 4x64, 64x4 |
| TX_4X64, TX_64X4 |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| static const TX_TYPE_1D vtx_tab[TX_TYPES] = { |
| DCT_1D, ADST_1D, DCT_1D, ADST_1D, |
| FLIPADST_1D, DCT_1D, FLIPADST_1D, ADST_1D, FLIPADST_1D, IDTX_1D, |
| DCT_1D, IDTX_1D, ADST_1D, IDTX_1D, FLIPADST_1D, IDTX_1D, |
| #if CONFIG_MODE_DEP_INTRA_TX |
| MDTX1_1D, MDTX1_1D, DCT_1D, |
| #if CONFIG_MODE_DEP_NONSEP_INTRA_TX |
| NSTX, |
| #endif // CONFIG_MODE_DEP_NONSEP_INTRA_TX |
| #endif // CONFIG_MODE_DEP_INTRA_TX |
| #if CONFIG_MODE_DEP_INTER_TX |
| MDTX1_1D, MDTX1_1D, DCT_1D, MDTX1_1D, MDTX1_1D, DCT_1D, |
| MDTX1_1D, MDTX1_1D, |
| #endif // CONFIG_MODE_DEP_INTER_TX |
| }; |
| |
| static const TX_TYPE_1D htx_tab[TX_TYPES] = { |
| DCT_1D, DCT_1D, ADST_1D, ADST_1D, |
| DCT_1D, FLIPADST_1D, FLIPADST_1D, FLIPADST_1D, ADST_1D, IDTX_1D, |
| IDTX_1D, DCT_1D, IDTX_1D, ADST_1D, IDTX_1D, FLIPADST_1D, |
| #if CONFIG_MODE_DEP_INTRA_TX |
| MDTX1_1D, DCT_1D, MDTX1_1D, |
| #if CONFIG_MODE_DEP_NONSEP_INTRA_TX |
| NSTX, |
| #endif // CONFIG_MODE_DEP_NONSEP_INTRA_TX |
| #endif // CONFIG_MODE_DEP_INTRA_TX |
| #if CONFIG_MODE_DEP_INTER_TX |
| MDTX1_1D, DCT_1D, MDTX1_1D, MDTX1_1D, DCT_1D, MDTX1_1D, |
| MDTX1_1D, MDTX1_1D, |
| #endif // CONFIG_MODE_DEP_INTER_TX |
| }; |
| |
| #define TXSIZE_CAT_INVALID (-1) |
| |
| /* clang-format on */ |
| |
| #if CONFIG_NEW_TX_PARTITION |
| // Smallest sub_tx size units. Used to compute the index in the |
| // tx type map. |
| static const TX_SIZE smallest_sub_tx_size_map[TX_SIZES_ALL] = { |
| TX_4X4, // TX_4X4 |
| TX_4X4, // TX_8X8 |
| TX_4X4, // TX_16X16 |
| TX_8X8, // TX_32X32 |
| TX_16X16, // TX_64X64 |
| TX_4X4, // TX_4X8 |
| TX_4X4, // TX_8X4 |
| TX_4X4, // TX_8X16 |
| TX_4X4, // TX_16X8 |
| TX_4X8, // TX_16X32 |
| TX_8X4, // TX_32X16 |
| TX_8X16, // TX_32X64 |
| TX_16X8, // TX_64X32 |
| TX_4X4, // TX_4X16 |
| TX_4X4, // TX_16X4 |
| TX_4X8, // TX_8X32 |
| TX_8X4, // TX_32X8 |
| TX_4X16, // TX_16X64 |
| TX_16X4, // TX_64X16 |
| #if CONFIG_FLEX_PARTITION |
| TX_4X8, // TX_4X32 |
| TX_8X4, // TX_32X4 |
| TX_4X16, // TX_8X64 |
| TX_16X4, // TX_64X8 |
| TX_4X16, // TX_4X64 |
| TX_16X4, // TX_64X4 |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| #endif |
| |
| static const TX_SIZE sub_tx_size_map[TX_SIZES_ALL] = { |
| TX_4X4, // TX_4X4 |
| TX_4X4, // TX_8X8 |
| TX_8X8, // TX_16X16 |
| TX_16X16, // TX_32X32 |
| TX_32X32, // TX_64X64 |
| TX_4X4, // TX_4X8 |
| TX_4X4, // TX_8X4 |
| TX_8X8, // TX_8X16 |
| TX_8X8, // TX_16X8 |
| TX_16X16, // TX_16X32 |
| TX_16X16, // TX_32X16 |
| TX_32X32, // TX_32X64 |
| TX_32X32, // TX_64X32 |
| TX_4X8, // TX_4X16 |
| TX_8X4, // TX_16X4 |
| TX_8X16, // TX_8X32 |
| TX_16X8, // TX_32X8 |
| TX_16X32, // TX_16X64 |
| TX_32X16, // TX_64X16 |
| #if CONFIG_FLEX_PARTITION |
| TX_4X16, // TX_4X32 |
| TX_16X4, // TX_32X4 |
| TX_8X32, // TX_8X64 |
| TX_32X8, // TX_64X8 |
| TX_4X32, // TX_4X64 |
| TX_32X4, // TX_64X4 |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| static const TX_SIZE txsize_horz_map[TX_SIZES_ALL] = { |
| TX_4X4, // TX_4X4 |
| TX_8X8, // TX_8X8 |
| TX_16X16, // TX_16X16 |
| TX_32X32, // TX_32X32 |
| TX_64X64, // TX_64X64 |
| TX_4X4, // TX_4X8 |
| TX_8X8, // TX_8X4 |
| TX_8X8, // TX_8X16 |
| TX_16X16, // TX_16X8 |
| TX_16X16, // TX_16X32 |
| TX_32X32, // TX_32X16 |
| TX_32X32, // TX_32X64 |
| TX_64X64, // TX_64X32 |
| TX_4X4, // TX_4X16 |
| TX_16X16, // TX_16X4 |
| TX_8X8, // TX_8X32 |
| TX_32X32, // TX_32X8 |
| TX_16X16, // TX_16X64 |
| TX_64X64, // TX_64X16 |
| #if CONFIG_FLEX_PARTITION |
| TX_4X4, // TX_4X32 |
| TX_32X32, // TX_32X4 |
| TX_8X8, // TX_8X64 |
| TX_64X64, // TX_64X8 |
| TX_4X4, // TX_4X64 |
| TX_64X64, // TX_64X4 |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| static const TX_SIZE txsize_vert_map[TX_SIZES_ALL] = { |
| TX_4X4, // TX_4X4 |
| TX_8X8, // TX_8X8 |
| TX_16X16, // TX_16X16 |
| TX_32X32, // TX_32X32 |
| TX_64X64, // TX_64X64 |
| TX_8X8, // TX_4X8 |
| TX_4X4, // TX_8X4 |
| TX_16X16, // TX_8X16 |
| TX_8X8, // TX_16X8 |
| TX_32X32, // TX_16X32 |
| TX_16X16, // TX_32X16 |
| TX_64X64, // TX_32X64 |
| TX_32X32, // TX_64X32 |
| TX_16X16, // TX_4X16 |
| TX_4X4, // TX_16X4 |
| TX_32X32, // TX_8X32 |
| TX_8X8, // TX_32X8 |
| TX_64X64, // TX_16X64 |
| TX_16X16, // TX_64X16 |
| #if CONFIG_FLEX_PARTITION |
| TX_32X32, // TX_4X32 |
| TX_4X4, // TX_32X4 |
| TX_64X64, // TX_8X64 |
| TX_8X8, // TX_64X8 |
| TX_64X64, // TX_4X64 |
| TX_4X4, // TX_64X4 |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| #if CONFIG_DSPL_RESIDUAL |
| /*! |
| * Mapping for halving transform size in both X and Y directions indexed by |
| * TX_SIZE. Note that transforms which have width or height equal to 4 cannot be |
| * halved because transforms for those sizes aren't available. |
| */ |
| static const TX_SIZE dspl_tx_size_map[TX_SIZES_ALL] = { |
| TX_INVALID, // TX_4X4 |
| TX_4X4, // TX_8X8 |
| TX_8X8, // TX_16X16 |
| TX_16X16, // TX_32X32 |
| TX_32X32, // TX_64X64 |
| TX_INVALID, // TX_4X8 |
| TX_INVALID, // TX_8X4 |
| TX_4X8, // TX_8X16 |
| TX_8X4, // TX_16X8 |
| TX_8X16, // TX_16X32 |
| TX_16X8, // TX_32X16 |
| TX_16X32, // TX_32X64 |
| TX_32X16, // TX_64X32 |
| TX_INVALID, // TX_4X16 |
| TX_INVALID, // TX_16X4 |
| TX_4X16, // TX_8X32 |
| TX_16X4, // TX_32X8 |
| TX_8X32, // TX_16X64 |
| TX_32X8, // TX_64X16 |
| }; |
| #endif // CONFIG_DSPL_RESIDUAL |
| |
| #define TX_SIZE_W_MIN 4 |
| |
| // Transform block width in pixels |
| static const int tx_size_wide[TX_SIZES_ALL] = { |
| 4, 8, 16, 32, 64, 4, 8, 8, 16, 16, 32, 32, 64, 4, 16, 8, 32, 16, 64, |
| #if CONFIG_FLEX_PARTITION |
| 4, 32, 8, 64, 4, 64, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| #define TX_SIZE_H_MIN 4 |
| |
| // Transform block height in pixels |
| static const int tx_size_high[TX_SIZES_ALL] = { |
| 4, 8, 16, 32, 64, 8, 4, 16, 8, 32, 16, 64, 32, 16, 4, 32, 8, 64, 16, |
| #if CONFIG_FLEX_PARTITION |
| 32, 4, 64, 8, 64, 4, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| // Transform block width in unit |
| static const int tx_size_wide_unit[TX_SIZES_ALL] = { |
| 1, 2, 4, 8, 16, 1, 2, 2, 4, 4, 8, 8, 16, 1, 4, 2, 8, 4, 16, |
| #if CONFIG_FLEX_PARTITION |
| 1, 8, 2, 16, 1, 16, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| // Transform block height in unit |
| static const int tx_size_high_unit[TX_SIZES_ALL] = { |
| 1, 2, 4, 8, 16, 2, 1, 4, 2, 8, 4, 16, 8, 4, 1, 8, 2, 16, 4, |
| #if CONFIG_FLEX_PARTITION |
| 8, 1, 16, 2, 16, 1, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| // Transform block width in log2 |
| static const int tx_size_wide_log2[TX_SIZES_ALL] = { |
| 2, 3, 4, 5, 6, 2, 3, 3, 4, 4, 5, 5, 6, 2, 4, 3, 5, 4, 6, |
| #if CONFIG_FLEX_PARTITION |
| 2, 5, 3, 6, 2, 6, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| // Transform block height in log2 |
| static const int tx_size_high_log2[TX_SIZES_ALL] = { |
| 2, 3, 4, 5, 6, 3, 2, 4, 3, 5, 4, 6, 5, 4, 2, 5, 3, 6, 4, |
| #if CONFIG_FLEX_PARTITION |
| 5, 2, 6, 3, 6, 2, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| static const int tx_size_2d[TX_SIZES_ALL + 1] = { |
| 16, 64, 256, 1024, 4096, 32, 32, 128, 128, 512, |
| 512, 2048, 2048, 64, 64, 256, 256, 1024, 1024, |
| #if CONFIG_FLEX_PARTITION |
| 128, 128, 512, 512, 256, 256, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| static const BLOCK_SIZE txsize_to_bsize[TX_SIZES_ALL] = { |
| BLOCK_4X4, // TX_4X4 |
| BLOCK_8X8, // TX_8X8 |
| BLOCK_16X16, // TX_16X16 |
| BLOCK_32X32, // TX_32X32 |
| BLOCK_64X64, // TX_64X64 |
| BLOCK_4X8, // TX_4X8 |
| BLOCK_8X4, // TX_8X4 |
| BLOCK_8X16, // TX_8X16 |
| BLOCK_16X8, // TX_16X8 |
| BLOCK_16X32, // TX_16X32 |
| BLOCK_32X16, // TX_32X16 |
| BLOCK_32X64, // TX_32X64 |
| BLOCK_64X32, // TX_64X32 |
| BLOCK_4X16, // TX_4X16 |
| BLOCK_16X4, // TX_16X4 |
| BLOCK_8X32, // TX_8X32 |
| BLOCK_32X8, // TX_32X8 |
| BLOCK_16X64, // TX_16X64 |
| BLOCK_64X16, // TX_64X16 |
| #if CONFIG_FLEX_PARTITION |
| BLOCK_4X32, // TX_4X32 |
| BLOCK_32X4, // TX_32X4 |
| BLOCK_8X64, // TX_8X64 |
| BLOCK_64X8, // TX_64X8 |
| BLOCK_4X64, // TX_4X64 |
| BLOCK_64X4, // TX_64X4 |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| static const TX_SIZE txsize_sqr_map[TX_SIZES_ALL] = { |
| TX_4X4, // TX_4X4 |
| TX_8X8, // TX_8X8 |
| TX_16X16, // TX_16X16 |
| TX_32X32, // TX_32X32 |
| TX_64X64, // TX_64X64 |
| TX_4X4, // TX_4X8 |
| TX_4X4, // TX_8X4 |
| TX_8X8, // TX_8X16 |
| TX_8X8, // TX_16X8 |
| TX_16X16, // TX_16X32 |
| TX_16X16, // TX_32X16 |
| TX_32X32, // TX_32X64 |
| TX_32X32, // TX_64X32 |
| TX_4X4, // TX_4X16 |
| TX_4X4, // TX_16X4 |
| TX_8X8, // TX_8X32 |
| TX_8X8, // TX_32X8 |
| TX_16X16, // TX_16X64 |
| TX_16X16, // TX_64X16 |
| #if CONFIG_FLEX_PARTITION |
| TX_4X4, // TX_4X32 |
| TX_4X4, // TX_32X4 |
| TX_8X8, // TX_8X64 |
| TX_8X8, // TX_64X8 |
| TX_4X4, // TX_4X64 |
| TX_4X4, // TX_64X4 |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| static const TX_SIZE txsize_sqr_up_map[TX_SIZES_ALL] = { |
| TX_4X4, // TX_4X4 |
| TX_8X8, // TX_8X8 |
| TX_16X16, // TX_16X16 |
| TX_32X32, // TX_32X32 |
| TX_64X64, // TX_64X64 |
| TX_8X8, // TX_4X8 |
| TX_8X8, // TX_8X4 |
| TX_16X16, // TX_8X16 |
| TX_16X16, // TX_16X8 |
| TX_32X32, // TX_16X32 |
| TX_32X32, // TX_32X16 |
| TX_64X64, // TX_32X64 |
| TX_64X64, // TX_64X32 |
| TX_16X16, // TX_4X16 |
| TX_16X16, // TX_16X4 |
| TX_32X32, // TX_8X32 |
| TX_32X32, // TX_32X8 |
| TX_64X64, // TX_16X64 |
| TX_64X64, // TX_64X16 |
| #if CONFIG_FLEX_PARTITION |
| TX_32X32, // TX_4X32 |
| TX_32X32, // TX_32X4 |
| TX_64X64, // TX_8X64 |
| TX_64X64, // TX_64X8 |
| TX_64X64, // TX_4X64 |
| TX_64X64, // TX_64X4 |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| static const int8_t txsize_log2_minus4[TX_SIZES_ALL] = { |
| 0, // TX_4X4 |
| 2, // TX_8X8 |
| 4, // TX_16X16 |
| 6, // TX_32X32 |
| 6, // TX_64X64 |
| 1, // TX_4X8 |
| 1, // TX_8X4 |
| 3, // TX_8X16 |
| 3, // TX_16X8 |
| 5, // TX_16X32 |
| 5, // TX_32X16 |
| 6, // TX_32X64 |
| 6, // TX_64X32 |
| 2, // TX_4X16 |
| 2, // TX_16X4 |
| 4, // TX_8X32 |
| 4, // TX_32X8 |
| 5, // TX_16X64 |
| 5, // TX_64X16 |
| #if CONFIG_FLEX_PARTITION |
| 3, // TX_4X32 |
| 3, // TX_32X4 |
| 4, // TX_8X64 |
| 4, // TX_64X8 |
| 3, // TX_4X64 |
| 3, // TX_64X4 |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| |
| /* clang-format off */ |
| static const TX_SIZE tx_mode_to_biggest_tx_size[TX_MODES] = { |
| TX_4X4, // ONLY_4X4 |
| TX_64X64, // TX_MODE_LARGEST |
| TX_64X64, // TX_MODE_SELECT |
| }; |
| |
| // The Subsampled_Size table in the spec (Section 5.11.38. Get plane residual |
| // size function). |
| static const BLOCK_SIZE ss_size_lookup[BLOCK_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, BLOCK_4X4 }, { BLOCK_4X4, BLOCK_4X4 } }, |
| { { BLOCK_4X8, BLOCK_4X4 }, { BLOCK_INVALID, BLOCK_4X4 } }, |
| { { BLOCK_8X4, BLOCK_INVALID }, { BLOCK_4X4, BLOCK_4X4 } }, |
| { { 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 } }, |
| { { BLOCK_64X128, BLOCK_64X64 }, { BLOCK_INVALID, BLOCK_32X64 } }, |
| { { BLOCK_128X64, BLOCK_INVALID }, { BLOCK_64X64, BLOCK_64X32 } }, |
| { { BLOCK_128X128, BLOCK_128X64 }, { BLOCK_64X128, BLOCK_64X64 } }, |
| { { BLOCK_4X16, BLOCK_4X8 }, { BLOCK_INVALID, BLOCK_4X8 } }, |
| { { BLOCK_16X4, BLOCK_INVALID }, { BLOCK_8X4, BLOCK_8X4 } }, |
| { { BLOCK_8X32, BLOCK_8X16 }, { BLOCK_INVALID, BLOCK_4X16 } }, |
| { { BLOCK_32X8, BLOCK_INVALID }, { BLOCK_16X8, BLOCK_16X4 } }, |
| { { BLOCK_16X64, BLOCK_16X32 }, { BLOCK_INVALID, BLOCK_8X32 } }, |
| { { BLOCK_64X16, BLOCK_INVALID }, { BLOCK_32X16, BLOCK_32X8 } }, |
| #if CONFIG_FLEX_PARTITION |
| { { BLOCK_4X32, BLOCK_4X16 }, { BLOCK_INVALID, BLOCK_4X16 } }, |
| { { BLOCK_32X4, BLOCK_INVALID }, { BLOCK_16X4, BLOCK_16X4 } }, |
| { { BLOCK_8X64, BLOCK_8X32 }, { BLOCK_4X64, BLOCK_4X32 } }, |
| { { BLOCK_64X8, BLOCK_64X4 }, { BLOCK_32X8, BLOCK_32X4 } }, |
| { { BLOCK_4X64, BLOCK_4X32 }, { BLOCK_INVALID, BLOCK_4X32 } }, |
| { { BLOCK_64X4, BLOCK_INVALID }, { BLOCK_32X4, BLOCK_32X4 } }, |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| /* clang-format on */ |
| |
| // Generates 5 bit field in which each bit set to 1 represents |
| // a blocksize partition 11111 means we split 128x128, 64x64, 32x32, 16x16 |
| // and 8x8. 10000 means we just split the 128x128 to 64x64 |
| /* clang-format off */ |
| static const struct { |
| PARTITION_CONTEXT above; |
| PARTITION_CONTEXT left; |
| } partition_context_lookup[BLOCK_SIZES_ALL] = { |
| { 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} |
| { 31, 28 }, // 4X16 - {0b11111, 0b11100} |
| { 28, 31 }, // 16X4 - {0b11100, 0b11111} |
| { 30, 24 }, // 8X32 - {0b11110, 0b11000} |
| { 24, 30 }, // 32X8 - {0b11000, 0b11110} |
| { 28, 16 }, // 16X64 - {0b11100, 0b10000} |
| { 16, 28 }, // 64X16 - {0b10000, 0b11100} |
| #if CONFIG_FLEX_PARTITION |
| { 31, 24 }, // 4X32 - {0b11111, 0b11000} |
| { 24, 31 }, // 32X4 - {0b11000, 0b11111} |
| { 30, 16 }, // 8X64 - {0b11110, 0b10000} |
| { 16, 30 }, // 64X8 - {0b10000, 0b11110} |
| { 31, 16 }, // 4X64 - {0b11110, 0b10000} |
| { 16, 31 }, // 64X4 - {0b10000, 0b11110} |
| #endif // CONFIG_FLEX_PARTITION |
| }; |
| /* clang-format on */ |
| |
| #if CONFIG_DERIVED_INTRA_MODE |
| static const int dr_mode_to_index[INTRA_MODES] = { |
| -1, 0, 1, 2, 3, 4, 5, 6, 7, -1, -1, -1, -1, |
| }; |
| |
| static const int dr_index_to_mode[DIRECTIONAL_MODES] = { |
| V_PRED, H_PRED, D45_PRED, D135_PRED, |
| D113_PRED, D157_PRED, D203_PRED, D67_PRED, |
| }; |
| |
| static const int none_dr_mode_to_index[INTRA_MODES] = { |
| 0, -1, -1, -1, -1, -1, -1, -1, -1, 1, 2, 3, 4, |
| }; |
| |
| static const int none_dr_index_to_mode[NONE_DIRECTIONAL_MODES] = { |
| DC_PRED, SMOOTH_PRED, SMOOTH_V_PRED, SMOOTH_H_PRED, PAETH_PRED |
| }; |
| |
| static const int intra_mode_context[INTRA_MODES] = { |
| 0, 1, 3, 1, 2, 2, 2, 3, 1, 4, 4, 4, 5, |
| }; |
| #else |
| static const int intra_mode_context[INTRA_MODES] = { |
| 0, 1, 2, 3, 4, 4, 4, 4, 3, 0, 1, 2, 0, |
| }; |
| #endif // CONFIG_DERIVED_INTRA_MODE |
| |
| // Note: this is also used in unit tests. So whenever one changes the table, |
| // the unit tests need to be changed accordingly. |
| static const int quant_dist_weight[4][2] = { |
| { 2, 3 }, { 2, 5 }, { 2, 7 }, { 1, MAX_FRAME_DISTANCE } |
| }; |
| static const int quant_dist_lookup_table[2][4][2] = { |
| { { 9, 7 }, { 11, 5 }, { 12, 4 }, { 13, 3 } }, |
| { { 7, 9 }, { 5, 11 }, { 4, 12 }, { 3, 13 } }, |
| }; |
| |
| #ifdef __cplusplus |
| } // extern "C" |
| #endif |
| |
| #endif // AOM_AV1_COMMON_COMMON_DATA_H_ |
