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_TEST_AV1_TXFM_TEST_H_
13 #define AOM_TEST_AV1_TXFM_TEST_H_
14
15 #include <stdio.h>
16 #include <stdlib.h>
17 #ifdef _MSC_VER
18 #define _USE_MATH_DEFINES
19 #endif
20 #include <math.h>
21
22 #include "config/av1_rtcd.h"
23
24 #include "gtest/gtest.h"
25
26 #include "test/acm_random.h"
27 #include "av1/common/av1_txfm.h"
28 #include "av1/common/blockd.h"
29 #include "av1/common/enums.h"
30
31 namespace libaom_test {
32
33 extern const char *tx_type_name[];
34
35 enum {
36 TYPE_DCT = 0,
37 TYPE_ADST,
38 TYPE_IDTX,
39 TYPE_IDCT,
40 TYPE_IADST,
41 TYPE_LAST
42 } UENUM1BYTE(TYPE_TXFM);
43
44 int get_txfm1d_size(TX_SIZE tx_size);
45
46 void get_txfm1d_type(TX_TYPE txfm2d_type, TYPE_TXFM *type0, TYPE_TXFM *type1);
47
48 void reference_dct_1d(const double *in, double *out, int size);
49 void reference_idct_1d(const double *in, double *out, int size);
50
51 void reference_adst_1d(const double *in, double *out, int size);
52
53 void reference_hybrid_1d(double *in, double *out, int size, int type);
54
55 double get_amplification_factor(TX_TYPE tx_type, TX_SIZE tx_size);
56
57 void reference_hybrid_2d(double *in, double *out, TX_TYPE tx_type,
58 TX_SIZE tx_size);
59 template <typename Type1, typename Type2>
compute_avg_abs_error(const Type1 * a,const Type2 * b,const int size)60 static double compute_avg_abs_error(const Type1 *a, const Type2 *b,
61 const int size) {
62 double error = 0;
63 for (int i = 0; i < size; i++) {
64 error += fabs(static_cast<double>(a[i]) - static_cast<double>(b[i]));
65 }
66 error = error / size;
67 return error;
68 }
69
70 template <typename Type>
71 void fliplr(Type *dest, int width, int height, int stride);
72
73 template <typename Type>
74 void flipud(Type *dest, int width, int height, int stride);
75
76 template <typename Type>
77 void fliplrud(Type *dest, int width, int height, int stride);
78
79 typedef void (*TxfmFunc)(const int32_t *in, int32_t *out, const int8_t cos_bit,
80 const int8_t *range_bit);
81
82 typedef void (*InvTxfm2dFunc)(const int32_t *, uint16_t *, int, TX_TYPE, int);
83 typedef void (*LbdInvTxfm2dFunc)(const int32_t *, uint8_t *, int, TX_TYPE,
84 TX_SIZE, int);
85
86 static const int bd = 10;
87 static const int input_base = (1 << bd);
88
IsTxSizeTypeValid(TX_SIZE tx_size,TX_TYPE tx_type)89 static inline bool IsTxSizeTypeValid(TX_SIZE tx_size, TX_TYPE tx_type) {
90 const TX_SIZE tx_size_sqr_up = txsize_sqr_up_map[tx_size];
91 TxSetType tx_set_type;
92 if (tx_size_sqr_up > TX_32X32) {
93 tx_set_type = EXT_TX_SET_DCTONLY;
94 } else if (tx_size_sqr_up == TX_32X32) {
95 tx_set_type = EXT_TX_SET_DCT_IDTX;
96 } else {
97 tx_set_type = EXT_TX_SET_ALL16;
98 }
99 return av1_ext_tx_used[tx_set_type][tx_type] != 0;
100 }
101
102 #if CONFIG_AV1_ENCODER
103 #if !CONFIG_REALTIME_ONLY
104 static const FwdTxfm2dFunc fwd_txfm_func_ls[TX_SIZES_ALL] = {
105 av1_fwd_txfm2d_4x4_c, av1_fwd_txfm2d_8x8_c, av1_fwd_txfm2d_16x16_c,
106 av1_fwd_txfm2d_32x32_c, av1_fwd_txfm2d_64x64_c, av1_fwd_txfm2d_4x8_c,
107 av1_fwd_txfm2d_8x4_c, av1_fwd_txfm2d_8x16_c, av1_fwd_txfm2d_16x8_c,
108 av1_fwd_txfm2d_16x32_c, av1_fwd_txfm2d_32x16_c, av1_fwd_txfm2d_32x64_c,
109 av1_fwd_txfm2d_64x32_c, av1_fwd_txfm2d_4x16_c, av1_fwd_txfm2d_16x4_c,
110 av1_fwd_txfm2d_8x32_c, av1_fwd_txfm2d_32x8_c, av1_fwd_txfm2d_16x64_c,
111 av1_fwd_txfm2d_64x16_c,
112 };
113 #else
114 static const FwdTxfm2dFunc fwd_txfm_func_ls[TX_SIZES_ALL] = {
115 av1_fwd_txfm2d_4x4_c,
116 av1_fwd_txfm2d_8x8_c,
117 av1_fwd_txfm2d_16x16_c,
118 av1_fwd_txfm2d_32x32_c,
119 av1_fwd_txfm2d_64x64_c,
120 av1_fwd_txfm2d_4x8_c,
121 av1_fwd_txfm2d_8x4_c,
122 av1_fwd_txfm2d_8x16_c,
123 av1_fwd_txfm2d_16x8_c,
124 av1_fwd_txfm2d_16x32_c,
125 av1_fwd_txfm2d_32x16_c,
126 av1_fwd_txfm2d_32x64_c,
127 av1_fwd_txfm2d_64x32_c,
128 nullptr,
129 av1_fwd_txfm2d_16x4_c,
130 nullptr,
131 nullptr,
132 nullptr,
133 nullptr,
134 };
135 #endif
136 #endif
137
138 static const InvTxfm2dFunc inv_txfm_func_ls[TX_SIZES_ALL] = {
139 av1_inv_txfm2d_add_4x4_c, av1_inv_txfm2d_add_8x8_c,
140 av1_inv_txfm2d_add_16x16_c, av1_inv_txfm2d_add_32x32_c,
141 av1_inv_txfm2d_add_64x64_c, av1_inv_txfm2d_add_4x8_c,
142 av1_inv_txfm2d_add_8x4_c, av1_inv_txfm2d_add_8x16_c,
143 av1_inv_txfm2d_add_16x8_c, av1_inv_txfm2d_add_16x32_c,
144 av1_inv_txfm2d_add_32x16_c, av1_inv_txfm2d_add_32x64_c,
145 av1_inv_txfm2d_add_64x32_c, av1_inv_txfm2d_add_4x16_c,
146 av1_inv_txfm2d_add_16x4_c, av1_inv_txfm2d_add_8x32_c,
147 av1_inv_txfm2d_add_32x8_c, av1_inv_txfm2d_add_16x64_c,
148 av1_inv_txfm2d_add_64x16_c,
149 };
150
151 #define BD_NUM 3
152
153 extern int bd_arr[];
154 extern int8_t low_range_arr[];
155 extern int8_t high_range_arr[];
156
157 void txfm_stage_range_check(const int8_t *stage_range, int stage_num,
158 const int8_t cos_bit, int low_range,
159 int high_range);
160 } // namespace libaom_test
161 #endif // AOM_TEST_AV1_TXFM_TEST_H_
162