1 /*
2 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
3 *
4 * This source code is subject to the terms of the BSD 2 Clause License and
5 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
6 * was not distributed with this source code in the LICENSE file, you can
7 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
8 * Media Patent License 1.0 was not distributed with this source code in the
9 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
10 */
11
12 #ifndef AOM_AOM_DSP_TXFM_COMMON_H_
13 #define AOM_AOM_DSP_TXFM_COMMON_H_
14
15 #include "aom_dsp/aom_dsp_common.h"
16
17 // Constants and Macros used by all idct/dct functions
18 #define DCT_CONST_BITS 14
19 #define DCT_CONST_ROUNDING (1 << (DCT_CONST_BITS - 1))
20
21 #define UNIT_QUANT_SHIFT 2
22 #define UNIT_QUANT_FACTOR (1 << UNIT_QUANT_SHIFT)
23
24 // block transform size
25 enum {
26 TX_4X4, // 4x4 transform
27 TX_8X8, // 8x8 transform
28 TX_16X16, // 16x16 transform
29 TX_32X32, // 32x32 transform
30 TX_64X64, // 64x64 transform
31 TX_4X8, // 4x8 transform
32 TX_8X4, // 8x4 transform
33 TX_8X16, // 8x16 transform
34 TX_16X8, // 16x8 transform
35 TX_16X32, // 16x32 transform
36 TX_32X16, // 32x16 transform
37 TX_32X64, // 32x64 transform
38 TX_64X32, // 64x32 transform
39 TX_4X16, // 4x16 transform
40 TX_16X4, // 16x4 transform
41 TX_8X32, // 8x32 transform
42 TX_32X8, // 32x8 transform
43 TX_16X64, // 16x64 transform
44 TX_64X16, // 64x16 transform
45 TX_SIZES_ALL, // Includes rectangular transforms
46 TX_SIZES = TX_4X8, // Does NOT include rectangular transforms
47 TX_SIZES_LARGEST = TX_64X64,
48 TX_INVALID = 255 // Invalid transform size
49 } UENUM1BYTE(TX_SIZE);
50
51 enum {
52 DCT_DCT, // DCT in both horizontal and vertical
53 ADST_DCT, // ADST in vertical, DCT in horizontal
54 DCT_ADST, // DCT in vertical, ADST in horizontal
55 ADST_ADST, // ADST in both directions
56 FLIPADST_DCT, // FLIPADST in vertical, DCT in horizontal
57 DCT_FLIPADST, // DCT in vertical, FLIPADST in horizontal
58 FLIPADST_FLIPADST, // FLIPADST in both directions
59 ADST_FLIPADST, // ADST in vertical, FLIPADST in horizontal
60 FLIPADST_ADST, // FLIPADST in vertical, ADST in horizontal
61 IDTX, // Identity in both directions
62 V_DCT, // DCT in vertical, identity in horizontal
63 H_DCT, // Identity in vertical, DCT in horizontal
64 V_ADST, // ADST in vertical, identity in horizontal
65 H_ADST, // Identity in vertical, ADST in horizontal
66 V_FLIPADST, // FLIPADST in vertical, identity in horizontal
67 H_FLIPADST, // Identity in vertical, FLIPADST in horizontal
68 TX_TYPES,
69 DCT_ADST_TX_MASK = 0x000F, // Either DCT or ADST in each direction
70 TX_TYPE_INVALID = 255, // Invalid transform type
71 } UENUM1BYTE(TX_TYPE);
72
73 enum {
74 // DCT only
75 EXT_TX_SET_DCTONLY,
76 // DCT + Identity only
77 EXT_TX_SET_DCT_IDTX,
78 // Discrete Trig transforms w/o flip (4) + Identity (1)
79 EXT_TX_SET_DTT4_IDTX,
80 // Discrete Trig transforms w/o flip (4) + Identity (1) + 1D Hor/vert DCT (2)
81 EXT_TX_SET_DTT4_IDTX_1DDCT,
82 // Discrete Trig transforms w/ flip (9) + Identity (1) + 1D Hor/Ver DCT (2)
83 EXT_TX_SET_DTT9_IDTX_1DDCT,
84 // Discrete Trig transforms w/ flip (9) + Identity (1) + 1D Hor/Ver (6)
85 EXT_TX_SET_ALL16,
86 EXT_TX_SET_TYPES
87 } UENUM1BYTE(TxSetType);
88
89 typedef struct txfm_param {
90 // for both forward and inverse transforms
91 TX_TYPE tx_type;
92 TX_SIZE tx_size;
93 int lossless;
94 int bd;
95 // are the pixel buffers octets or shorts? This should collapse to
96 // bd==8 implies !is_hbd, but that's not certain right now.
97 int is_hbd;
98 TxSetType tx_set_type;
99 // for inverse transforms only
100 int eob;
101 } TxfmParam;
102
103 // Constants:
104 // for (int i = 1; i< 32; ++i)
105 // printf("static const int cospi_%d_64 = %.0f;\n", i,
106 // round(16384 * cos(i*PI/64)));
107 // Note: sin(k*Pi/64) = cos((32-k)*Pi/64)
108 static const tran_high_t cospi_1_64 = 16364;
109 static const tran_high_t cospi_2_64 = 16305;
110 static const tran_high_t cospi_3_64 = 16207;
111 static const tran_high_t cospi_4_64 = 16069;
112 static const tran_high_t cospi_5_64 = 15893;
113 static const tran_high_t cospi_6_64 = 15679;
114 static const tran_high_t cospi_7_64 = 15426;
115 static const tran_high_t cospi_8_64 = 15137;
116 static const tran_high_t cospi_9_64 = 14811;
117 static const tran_high_t cospi_10_64 = 14449;
118 static const tran_high_t cospi_11_64 = 14053;
119 static const tran_high_t cospi_12_64 = 13623;
120 static const tran_high_t cospi_13_64 = 13160;
121 static const tran_high_t cospi_14_64 = 12665;
122 static const tran_high_t cospi_15_64 = 12140;
123 static const tran_high_t cospi_16_64 = 11585;
124 static const tran_high_t cospi_17_64 = 11003;
125 static const tran_high_t cospi_18_64 = 10394;
126 static const tran_high_t cospi_19_64 = 9760;
127 static const tran_high_t cospi_20_64 = 9102;
128 static const tran_high_t cospi_21_64 = 8423;
129 static const tran_high_t cospi_22_64 = 7723;
130 static const tran_high_t cospi_23_64 = 7005;
131 static const tran_high_t cospi_24_64 = 6270;
132 static const tran_high_t cospi_25_64 = 5520;
133 static const tran_high_t cospi_26_64 = 4756;
134 static const tran_high_t cospi_27_64 = 3981;
135 static const tran_high_t cospi_28_64 = 3196;
136 static const tran_high_t cospi_29_64 = 2404;
137 static const tran_high_t cospi_30_64 = 1606;
138 static const tran_high_t cospi_31_64 = 804;
139
140 // 16384 * sqrt(2) * sin(kPi/9) * 2 / 3
141 static const tran_high_t sinpi_1_9 = 5283;
142 static const tran_high_t sinpi_2_9 = 9929;
143 static const tran_high_t sinpi_3_9 = 13377;
144 static const tran_high_t sinpi_4_9 = 15212;
145
146 // 16384 * sqrt(2)
147 static const tran_high_t Sqrt2 = 23170;
148 static const tran_high_t InvSqrt2 = 11585;
149
fdct_round_shift(tran_high_t input)150 static INLINE tran_high_t fdct_round_shift(tran_high_t input) {
151 tran_high_t rv = ROUND_POWER_OF_TWO(input, DCT_CONST_BITS);
152 return rv;
153 }
154
155 #endif // AOM_AOM_DSP_TXFM_COMMON_H_
156