155 lines
5.9 KiB
C
155 lines
5.9 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 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
|
|
TX_TYPE_INVALID = 255, // Invalid transform type
|
|
} 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_
|