| /* |
| * 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_AOM_DSP_TXFM_COMMON_H_ |
| #define AOM_AOM_DSP_TXFM_COMMON_H_ |
| |
| #include "aom_dsp/aom_dsp_common.h" |
| |
| // Constants and Macros used by all idct/dct functions |
| #define DCT_CONST_BITS 14 |
| #define DCT_CONST_ROUNDING (1 << (DCT_CONST_BITS - 1)) |
| |
| #define UNIT_QUANT_SHIFT 2 |
| #define UNIT_QUANT_FACTOR (1 << UNIT_QUANT_SHIFT) |
| |
| // block transform size |
| enum { |
| TX_4X4, // 4x4 transform |
| TX_8X8, // 8x8 transform |
| TX_16X16, // 16x16 transform |
| TX_32X32, // 32x32 transform |
| TX_64X64, // 64x64 transform |
| TX_4X8, // 4x8 transform |
| TX_8X4, // 8x4 transform |
| TX_8X16, // 8x16 transform |
| TX_16X8, // 16x8 transform |
| TX_16X32, // 16x32 transform |
| TX_32X16, // 32x16 transform |
| TX_32X64, // 32x64 transform |
| TX_64X32, // 64x32 transform |
| TX_4X16, // 4x16 transform |
| TX_16X4, // 16x4 transform |
| TX_8X32, // 8x32 transform |
| TX_32X8, // 32x8 transform |
| TX_16X64, // 16x64 transform |
| TX_64X16, // 64x16 transform |
| TX_SIZES_ALL, // Includes rectangular transforms |
| TX_SIZES = TX_4X8, // Does NOT include rectangular transforms |
| TX_SIZES_LARGEST = TX_64X64, |
| TX_INVALID = 255 // Invalid transform size |
| } UENUM1BYTE(TX_SIZE); |
| |
| enum { |
| DCT_DCT, // DCT in both horizontal and vertical |
| ADST_DCT, // ADST in vertical, DCT in horizontal |
| DCT_ADST, // DCT in vertical, ADST in horizontal |
| ADST_ADST, // ADST in both directions |
| FLIPADST_DCT, // FLIPADST in vertical, DCT in horizontal |
| DCT_FLIPADST, // DCT in vertical, FLIPADST in horizontal |
| FLIPADST_FLIPADST, // FLIPADST in both directions |
| ADST_FLIPADST, // ADST in vertical, FLIPADST in horizontal |
| FLIPADST_ADST, // FLIPADST in vertical, ADST in horizontal |
| IDTX, // Identity in both directions |
| V_DCT, // DCT in vertical, identity in horizontal |
| H_DCT, // Identity in vertical, DCT in horizontal |
| V_ADST, // ADST in vertical, identity in horizontal |
| H_ADST, // Identity in vertical, ADST in horizontal |
| V_FLIPADST, // FLIPADST in vertical, identity in horizontal |
| H_FLIPADST, // Identity in vertical, FLIPADST in horizontal |
| TX_TYPES, |
| DCT_ADST_TX_MASK = 0x000F, // Either DCT or ADST in each direction |
| } UENUM1BYTE(TX_TYPE); |
| |
| enum { |
| // DCT only |
| EXT_TX_SET_DCTONLY, |
| // DCT + Identity only |
| EXT_TX_SET_DCT_IDTX, |
| // Discrete Trig transforms w/o flip (4) + Identity (1) |
| EXT_TX_SET_DTT4_IDTX, |
| // Discrete Trig transforms w/o flip (4) + Identity (1) + 1D Hor/vert DCT (2) |
| EXT_TX_SET_DTT4_IDTX_1DDCT, |
| // Discrete Trig transforms w/ flip (9) + Identity (1) + 1D Hor/Ver DCT (2) |
| EXT_TX_SET_DTT9_IDTX_1DDCT, |
| // Discrete Trig transforms w/ flip (9) + Identity (1) + 1D Hor/Ver (6) |
| EXT_TX_SET_ALL16, |
| EXT_TX_SET_TYPES |
| } UENUM1BYTE(TxSetType); |
| |
| typedef struct txfm_param { |
| // for both forward and inverse transforms |
| TX_TYPE tx_type; |
| TX_SIZE tx_size; |
| int lossless; |
| int bd; |
| // are the pixel buffers octets or shorts? This should collapse to |
| // bd==8 implies !is_hbd, but that's not certain right now. |
| int is_hbd; |
| TxSetType tx_set_type; |
| // for inverse transforms only |
| int eob; |
| } TxfmParam; |
| |
| // Constants: |
| // for (int i = 1; i< 32; ++i) |
| // printf("static const int cospi_%d_64 = %.0f;\n", i, |
| // round(16384 * cos(i*PI/64))); |
| // Note: sin(k*Pi/64) = cos((32-k)*Pi/64) |
| static const tran_high_t cospi_1_64 = 16364; |
| static const tran_high_t cospi_2_64 = 16305; |
| static const tran_high_t cospi_3_64 = 16207; |
| static const tran_high_t cospi_4_64 = 16069; |
| static const tran_high_t cospi_5_64 = 15893; |
| static const tran_high_t cospi_6_64 = 15679; |
| static const tran_high_t cospi_7_64 = 15426; |
| static const tran_high_t cospi_8_64 = 15137; |
| static const tran_high_t cospi_9_64 = 14811; |
| static const tran_high_t cospi_10_64 = 14449; |
| static const tran_high_t cospi_11_64 = 14053; |
| static const tran_high_t cospi_12_64 = 13623; |
| static const tran_high_t cospi_13_64 = 13160; |
| static const tran_high_t cospi_14_64 = 12665; |
| static const tran_high_t cospi_15_64 = 12140; |
| static const tran_high_t cospi_16_64 = 11585; |
| static const tran_high_t cospi_17_64 = 11003; |
| static const tran_high_t cospi_18_64 = 10394; |
| static const tran_high_t cospi_19_64 = 9760; |
| static const tran_high_t cospi_20_64 = 9102; |
| static const tran_high_t cospi_21_64 = 8423; |
| static const tran_high_t cospi_22_64 = 7723; |
| static const tran_high_t cospi_23_64 = 7005; |
| static const tran_high_t cospi_24_64 = 6270; |
| static const tran_high_t cospi_25_64 = 5520; |
| static const tran_high_t cospi_26_64 = 4756; |
| static const tran_high_t cospi_27_64 = 3981; |
| static const tran_high_t cospi_28_64 = 3196; |
| static const tran_high_t cospi_29_64 = 2404; |
| static const tran_high_t cospi_30_64 = 1606; |
| static const tran_high_t cospi_31_64 = 804; |
| |
| // 16384 * sqrt(2) * sin(kPi/9) * 2 / 3 |
| static const tran_high_t sinpi_1_9 = 5283; |
| static const tran_high_t sinpi_2_9 = 9929; |
| static const tran_high_t sinpi_3_9 = 13377; |
| static const tran_high_t sinpi_4_9 = 15212; |
| |
| // 16384 * sqrt(2) |
| static const tran_high_t Sqrt2 = 23170; |
| static const tran_high_t InvSqrt2 = 11585; |
| |
| static INLINE tran_high_t fdct_round_shift(tran_high_t input) { |
| tran_high_t rv = ROUND_POWER_OF_TWO(input, DCT_CONST_BITS); |
| return rv; |
| } |
| |
| #endif // AOM_AOM_DSP_TXFM_COMMON_H_ |