blob: 3f6b738e5dc95a38da1e28a01c8a4d381f824ec1 [file] [log] [blame]
Daniel Kang58156f12012-06-26 18:11:33 -07001/*
2 * Copyright (c) 2012 The WebM project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11#include <math.h>
12#include <stdlib.h>
13#include <string.h>
14
15#include "third_party/googletest/src/include/gtest/gtest.h"
Jingning Han097d59c2015-07-29 14:51:36 -070016
17#include "./vp9_rtcd.h"
18#include "./vpx_dsp_rtcd.h"
Jingning Han30d4c5e2013-11-12 12:47:32 -080019#include "test/acm_random.h"
20#include "test/clear_system_state.h"
21#include "test/register_state_check.h"
22#include "test/util.h"
James Zern002ad402014-01-18 13:03:31 -080023#include "vp9/common/vp9_entropy.h"
Deb Mukherjee10783d42014-09-02 16:34:09 -070024#include "vpx/vpx_codec.h"
James Zern002ad402014-01-18 13:03:31 -080025#include "vpx/vpx_integer.h"
Johann1d7ccd52015-05-11 19:09:22 -070026#include "vpx_ports/mem.h"
James Zern002ad402014-01-18 13:03:31 -080027
Daniel Kang26641c72012-06-28 16:26:31 -070028using libvpx_test::ACMRandom;
29
Daniel Kang58156f12012-06-26 18:11:33 -070030namespace {
Jingning Han30d4c5e2013-11-12 12:47:32 -080031const int kNumCoeffs = 16;
Deb Mukherjee10783d42014-09-02 16:34:09 -070032typedef void (*FdctFunc)(const int16_t *in, tran_low_t *out, int stride);
33typedef void (*IdctFunc)(const tran_low_t *in, uint8_t *out, int stride);
34typedef void (*FhtFunc)(const int16_t *in, tran_low_t *out, int stride,
James Zernb8b3dd92014-07-16 18:55:40 -070035 int tx_type);
Deb Mukherjee10783d42014-09-02 16:34:09 -070036typedef void (*IhtFunc)(const tran_low_t *in, uint8_t *out, int stride,
James Zernb8b3dd92014-07-16 18:55:40 -070037 int tx_type);
Jingning Han30d4c5e2013-11-12 12:47:32 -080038
Deb Mukherjee10783d42014-09-02 16:34:09 -070039typedef std::tr1::tuple<FdctFunc, IdctFunc, int, vpx_bit_depth_t> Dct4x4Param;
40typedef std::tr1::tuple<FhtFunc, IhtFunc, int, vpx_bit_depth_t> Ht4x4Param;
Joshua Litt51490e52013-11-18 17:07:55 -080041
Deb Mukherjee10783d42014-09-02 16:34:09 -070042void fdct4x4_ref(const int16_t *in, tran_low_t *out, int stride,
43 int tx_type) {
Jingning Han4b5109c2015-07-28 15:57:40 -070044 vpx_fdct4x4_c(in, out, stride);
Jingning Han362809d2013-06-18 10:46:33 -070045}
Jingning Han30d4c5e2013-11-12 12:47:32 -080046
Deb Mukherjee10783d42014-09-02 16:34:09 -070047void fht4x4_ref(const int16_t *in, tran_low_t *out, int stride, int tx_type) {
Dmitry Kovalev005fc692014-02-06 11:54:15 -080048 vp9_fht4x4_c(in, out, stride, tx_type);
Jingning Han362809d2013-06-18 10:46:33 -070049}
Daniel Kang58156f12012-06-26 18:11:33 -070050
Deb Mukherjee10783d42014-09-02 16:34:09 -070051void fwht4x4_ref(const int16_t *in, tran_low_t *out, int stride,
52 int tx_type) {
Alex Converse9f9f87c2014-05-05 13:50:12 -070053 vp9_fwht4x4_c(in, out, stride);
54}
55
Deb Mukherjee10783d42014-09-02 16:34:09 -070056#if CONFIG_VP9_HIGHBITDEPTH
57void idct4x4_10(const tran_low_t *in, uint8_t *out, int stride) {
Jingning Han08a453b2015-08-03 14:51:10 -070058 vpx_highbd_idct4x4_16_add_c(in, out, stride, 10);
Deb Mukherjee10783d42014-09-02 16:34:09 -070059}
60
61void idct4x4_12(const tran_low_t *in, uint8_t *out, int stride) {
Jingning Han08a453b2015-08-03 14:51:10 -070062 vpx_highbd_idct4x4_16_add_c(in, out, stride, 12);
Deb Mukherjee10783d42014-09-02 16:34:09 -070063}
64
65void iht4x4_10(const tran_low_t *in, uint8_t *out, int stride, int tx_type) {
Deb Mukherjee1929c9b2014-10-08 12:43:22 -070066 vp9_highbd_iht4x4_16_add_c(in, out, stride, tx_type, 10);
Deb Mukherjee10783d42014-09-02 16:34:09 -070067}
68
69void iht4x4_12(const tran_low_t *in, uint8_t *out, int stride, int tx_type) {
Deb Mukherjee1929c9b2014-10-08 12:43:22 -070070 vp9_highbd_iht4x4_16_add_c(in, out, stride, tx_type, 12);
Deb Mukherjee10783d42014-09-02 16:34:09 -070071}
72
73void iwht4x4_10(const tran_low_t *in, uint8_t *out, int stride) {
Jingning Han08a453b2015-08-03 14:51:10 -070074 vpx_highbd_iwht4x4_16_add_c(in, out, stride, 10);
Deb Mukherjee10783d42014-09-02 16:34:09 -070075}
76
77void iwht4x4_12(const tran_low_t *in, uint8_t *out, int stride) {
Jingning Han08a453b2015-08-03 14:51:10 -070078 vpx_highbd_iwht4x4_16_add_c(in, out, stride, 12);
Deb Mukherjee10783d42014-09-02 16:34:09 -070079}
Peter de Rivaz7e40a552014-10-24 08:48:02 +010080
81#if HAVE_SSE2
82void idct4x4_10_sse2(const tran_low_t *in, uint8_t *out, int stride) {
Jingning Han08a453b2015-08-03 14:51:10 -070083 vpx_highbd_idct4x4_16_add_sse2(in, out, stride, 10);
Peter de Rivaz7e40a552014-10-24 08:48:02 +010084}
85
86void idct4x4_12_sse2(const tran_low_t *in, uint8_t *out, int stride) {
Jingning Han08a453b2015-08-03 14:51:10 -070087 vpx_highbd_idct4x4_16_add_sse2(in, out, stride, 12);
Peter de Rivaz7e40a552014-10-24 08:48:02 +010088}
89#endif // HAVE_SSE2
90#endif // CONFIG_VP9_HIGHBITDEPTH
Deb Mukherjee10783d42014-09-02 16:34:09 -070091
Jingning Han30d4c5e2013-11-12 12:47:32 -080092class Trans4x4TestBase {
Jingning Han362809d2013-06-18 10:46:33 -070093 public:
Jingning Han30d4c5e2013-11-12 12:47:32 -080094 virtual ~Trans4x4TestBase() {}
Jingning Han362809d2013-06-18 10:46:33 -070095
96 protected:
Deb Mukherjee10783d42014-09-02 16:34:09 -070097 virtual void RunFwdTxfm(const int16_t *in, tran_low_t *out, int stride) = 0;
Jingning Han30d4c5e2013-11-12 12:47:32 -080098
Deb Mukherjee10783d42014-09-02 16:34:09 -070099 virtual void RunInvTxfm(const tran_low_t *out, uint8_t *dst, int stride) = 0;
Jingning Han30d4c5e2013-11-12 12:47:32 -0800100
Alex Converse9f9f87c2014-05-05 13:50:12 -0700101 void RunAccuracyCheck(int limit) {
Jingning Han30d4c5e2013-11-12 12:47:32 -0800102 ACMRandom rnd(ACMRandom::DeterministicSeed());
103 uint32_t max_error = 0;
104 int64_t total_error = 0;
105 const int count_test_block = 10000;
106 for (int i = 0; i < count_test_block; ++i) {
James Zernfd3658b2015-05-02 13:24:16 -0700107 DECLARE_ALIGNED(16, int16_t, test_input_block[kNumCoeffs]);
108 DECLARE_ALIGNED(16, tran_low_t, test_temp_block[kNumCoeffs]);
109 DECLARE_ALIGNED(16, uint8_t, dst[kNumCoeffs]);
110 DECLARE_ALIGNED(16, uint8_t, src[kNumCoeffs]);
Deb Mukherjee10783d42014-09-02 16:34:09 -0700111#if CONFIG_VP9_HIGHBITDEPTH
James Zernfd3658b2015-05-02 13:24:16 -0700112 DECLARE_ALIGNED(16, uint16_t, dst16[kNumCoeffs]);
113 DECLARE_ALIGNED(16, uint16_t, src16[kNumCoeffs]);
Deb Mukherjee10783d42014-09-02 16:34:09 -0700114#endif
Jingning Han30d4c5e2013-11-12 12:47:32 -0800115
116 // Initialize a test block with input range [-255, 255].
117 for (int j = 0; j < kNumCoeffs; ++j) {
Deb Mukherjee10783d42014-09-02 16:34:09 -0700118 if (bit_depth_ == VPX_BITS_8) {
119 src[j] = rnd.Rand8();
120 dst[j] = rnd.Rand8();
121 test_input_block[j] = src[j] - dst[j];
122#if CONFIG_VP9_HIGHBITDEPTH
123 } else {
124 src16[j] = rnd.Rand16() & mask_;
125 dst16[j] = rnd.Rand16() & mask_;
126 test_input_block[j] = src16[j] - dst16[j];
127#endif
128 }
Jingning Han30d4c5e2013-11-12 12:47:32 -0800129 }
130
James Zern29e1b1a2014-07-09 21:02:02 -0700131 ASM_REGISTER_STATE_CHECK(RunFwdTxfm(test_input_block,
132 test_temp_block, pitch_));
Deb Mukherjee10783d42014-09-02 16:34:09 -0700133 if (bit_depth_ == VPX_BITS_8) {
134 ASM_REGISTER_STATE_CHECK(RunInvTxfm(test_temp_block, dst, pitch_));
135#if CONFIG_VP9_HIGHBITDEPTH
136 } else {
137 ASM_REGISTER_STATE_CHECK(RunInvTxfm(test_temp_block,
138 CONVERT_TO_BYTEPTR(dst16), pitch_));
139#endif
140 }
Jingning Han30d4c5e2013-11-12 12:47:32 -0800141
142 for (int j = 0; j < kNumCoeffs; ++j) {
Deb Mukherjee10783d42014-09-02 16:34:09 -0700143#if CONFIG_VP9_HIGHBITDEPTH
144 const uint32_t diff =
145 bit_depth_ == VPX_BITS_8 ? dst[j] - src[j] : dst16[j] - src16[j];
146#else
James Zernfd3658b2015-05-02 13:24:16 -0700147 ASSERT_EQ(VPX_BITS_8, bit_depth_);
Jingning Han30d4c5e2013-11-12 12:47:32 -0800148 const uint32_t diff = dst[j] - src[j];
Deb Mukherjee10783d42014-09-02 16:34:09 -0700149#endif
Jingning Han30d4c5e2013-11-12 12:47:32 -0800150 const uint32_t error = diff * diff;
151 if (max_error < error)
152 max_error = error;
153 total_error += error;
154 }
155 }
156
Alex Converse9f9f87c2014-05-05 13:50:12 -0700157 EXPECT_GE(static_cast<uint32_t>(limit), max_error)
158 << "Error: 4x4 FHT/IHT has an individual round trip error > "
159 << limit;
Jingning Han30d4c5e2013-11-12 12:47:32 -0800160
Alex Converse9f9f87c2014-05-05 13:50:12 -0700161 EXPECT_GE(count_test_block * limit, total_error)
162 << "Error: 4x4 FHT/IHT has average round trip error > " << limit
163 << " per block";
Jingning Han362809d2013-06-18 10:46:33 -0700164 }
165
Jingning Han30d4c5e2013-11-12 12:47:32 -0800166 void RunCoeffCheck() {
167 ACMRandom rnd(ACMRandom::DeterministicSeed());
168 const int count_test_block = 5000;
James Zernfd3658b2015-05-02 13:24:16 -0700169 DECLARE_ALIGNED(16, int16_t, input_block[kNumCoeffs]);
170 DECLARE_ALIGNED(16, tran_low_t, output_ref_block[kNumCoeffs]);
171 DECLARE_ALIGNED(16, tran_low_t, output_block[kNumCoeffs]);
Jingning Han30d4c5e2013-11-12 12:47:32 -0800172
173 for (int i = 0; i < count_test_block; ++i) {
Deb Mukherjee10783d42014-09-02 16:34:09 -0700174 // Initialize a test block with input range [-mask_, mask_].
Jingning Han30d4c5e2013-11-12 12:47:32 -0800175 for (int j = 0; j < kNumCoeffs; ++j)
Deb Mukherjee10783d42014-09-02 16:34:09 -0700176 input_block[j] = (rnd.Rand16() & mask_) - (rnd.Rand16() & mask_);
Jingning Han30d4c5e2013-11-12 12:47:32 -0800177
178 fwd_txfm_ref(input_block, output_ref_block, pitch_, tx_type_);
James Zern29e1b1a2014-07-09 21:02:02 -0700179 ASM_REGISTER_STATE_CHECK(RunFwdTxfm(input_block, output_block, pitch_));
Jingning Han30d4c5e2013-11-12 12:47:32 -0800180
181 // The minimum quant value is 4.
182 for (int j = 0; j < kNumCoeffs; ++j)
183 EXPECT_EQ(output_block[j], output_ref_block[j]);
184 }
Jingning Han362809d2013-06-18 10:46:33 -0700185 }
186
Jingning Han30d4c5e2013-11-12 12:47:32 -0800187 void RunMemCheck() {
188 ACMRandom rnd(ACMRandom::DeterministicSeed());
189 const int count_test_block = 5000;
James Zernfd3658b2015-05-02 13:24:16 -0700190 DECLARE_ALIGNED(16, int16_t, input_extreme_block[kNumCoeffs]);
191 DECLARE_ALIGNED(16, tran_low_t, output_ref_block[kNumCoeffs]);
192 DECLARE_ALIGNED(16, tran_low_t, output_block[kNumCoeffs]);
Jingning Han30d4c5e2013-11-12 12:47:32 -0800193
194 for (int i = 0; i < count_test_block; ++i) {
Deb Mukherjee10783d42014-09-02 16:34:09 -0700195 // Initialize a test block with input range [-mask_, mask_].
Jingning Han30d4c5e2013-11-12 12:47:32 -0800196 for (int j = 0; j < kNumCoeffs; ++j) {
Deb Mukherjee10783d42014-09-02 16:34:09 -0700197 input_extreme_block[j] = rnd.Rand8() % 2 ? mask_ : -mask_;
Jingning Han30d4c5e2013-11-12 12:47:32 -0800198 }
Jingning Han5c2696c2014-06-02 16:40:01 -0700199 if (i == 0) {
Jingning Han30d4c5e2013-11-12 12:47:32 -0800200 for (int j = 0; j < kNumCoeffs; ++j)
Deb Mukherjee10783d42014-09-02 16:34:09 -0700201 input_extreme_block[j] = mask_;
Jingning Han5c2696c2014-06-02 16:40:01 -0700202 } else if (i == 1) {
Jingning Han30d4c5e2013-11-12 12:47:32 -0800203 for (int j = 0; j < kNumCoeffs; ++j)
Deb Mukherjee10783d42014-09-02 16:34:09 -0700204 input_extreme_block[j] = -mask_;
Jingning Han5c2696c2014-06-02 16:40:01 -0700205 }
Jingning Han30d4c5e2013-11-12 12:47:32 -0800206
207 fwd_txfm_ref(input_extreme_block, output_ref_block, pitch_, tx_type_);
James Zern29e1b1a2014-07-09 21:02:02 -0700208 ASM_REGISTER_STATE_CHECK(RunFwdTxfm(input_extreme_block,
209 output_block, pitch_));
Jingning Han30d4c5e2013-11-12 12:47:32 -0800210
211 // The minimum quant value is 4.
212 for (int j = 0; j < kNumCoeffs; ++j) {
213 EXPECT_EQ(output_block[j], output_ref_block[j]);
Deb Mukherjee10783d42014-09-02 16:34:09 -0700214 EXPECT_GE(4 * DCT_MAX_VALUE << (bit_depth_ - 8), abs(output_block[j]))
215 << "Error: 4x4 FDCT has coefficient larger than 4*DCT_MAX_VALUE";
Jingning Han30d4c5e2013-11-12 12:47:32 -0800216 }
217 }
218 }
219
Alex Converse9f9f87c2014-05-05 13:50:12 -0700220 void RunInvAccuracyCheck(int limit) {
Jingning Han30d4c5e2013-11-12 12:47:32 -0800221 ACMRandom rnd(ACMRandom::DeterministicSeed());
222 const int count_test_block = 1000;
James Zernfd3658b2015-05-02 13:24:16 -0700223 DECLARE_ALIGNED(16, int16_t, in[kNumCoeffs]);
224 DECLARE_ALIGNED(16, tran_low_t, coeff[kNumCoeffs]);
225 DECLARE_ALIGNED(16, uint8_t, dst[kNumCoeffs]);
226 DECLARE_ALIGNED(16, uint8_t, src[kNumCoeffs]);
Deb Mukherjee10783d42014-09-02 16:34:09 -0700227#if CONFIG_VP9_HIGHBITDEPTH
James Zernfd3658b2015-05-02 13:24:16 -0700228 DECLARE_ALIGNED(16, uint16_t, dst16[kNumCoeffs]);
229 DECLARE_ALIGNED(16, uint16_t, src16[kNumCoeffs]);
Deb Mukherjee10783d42014-09-02 16:34:09 -0700230#endif
Jingning Han30d4c5e2013-11-12 12:47:32 -0800231
232 for (int i = 0; i < count_test_block; ++i) {
Deb Mukherjee10783d42014-09-02 16:34:09 -0700233 // Initialize a test block with input range [-mask_, mask_].
Jingning Han30d4c5e2013-11-12 12:47:32 -0800234 for (int j = 0; j < kNumCoeffs; ++j) {
Deb Mukherjee10783d42014-09-02 16:34:09 -0700235 if (bit_depth_ == VPX_BITS_8) {
236 src[j] = rnd.Rand8();
237 dst[j] = rnd.Rand8();
238 in[j] = src[j] - dst[j];
239#if CONFIG_VP9_HIGHBITDEPTH
240 } else {
241 src16[j] = rnd.Rand16() & mask_;
242 dst16[j] = rnd.Rand16() & mask_;
243 in[j] = src16[j] - dst16[j];
244#endif
245 }
Jingning Han30d4c5e2013-11-12 12:47:32 -0800246 }
247
248 fwd_txfm_ref(in, coeff, pitch_, tx_type_);
249
Deb Mukherjee10783d42014-09-02 16:34:09 -0700250 if (bit_depth_ == VPX_BITS_8) {
251 ASM_REGISTER_STATE_CHECK(RunInvTxfm(coeff, dst, pitch_));
252#if CONFIG_VP9_HIGHBITDEPTH
253 } else {
254 ASM_REGISTER_STATE_CHECK(RunInvTxfm(coeff, CONVERT_TO_BYTEPTR(dst16),
255 pitch_));
256#endif
257 }
Jingning Han30d4c5e2013-11-12 12:47:32 -0800258
259 for (int j = 0; j < kNumCoeffs; ++j) {
Deb Mukherjee10783d42014-09-02 16:34:09 -0700260#if CONFIG_VP9_HIGHBITDEPTH
261 const uint32_t diff =
262 bit_depth_ == VPX_BITS_8 ? dst[j] - src[j] : dst16[j] - src16[j];
263#else
Jingning Han30d4c5e2013-11-12 12:47:32 -0800264 const uint32_t diff = dst[j] - src[j];
Deb Mukherjee10783d42014-09-02 16:34:09 -0700265#endif
Jingning Han30d4c5e2013-11-12 12:47:32 -0800266 const uint32_t error = diff * diff;
Alex Converse9f9f87c2014-05-05 13:50:12 -0700267 EXPECT_GE(static_cast<uint32_t>(limit), error)
268 << "Error: 4x4 IDCT has error " << error
Jingning Han30d4c5e2013-11-12 12:47:32 -0800269 << " at index " << j;
270 }
271 }
272 }
273
274 int pitch_;
Jingning Hanab362622013-06-21 11:45:47 -0700275 int tx_type_;
James Zernb8b3dd92014-07-16 18:55:40 -0700276 FhtFunc fwd_txfm_ref;
Deb Mukherjee10783d42014-09-02 16:34:09 -0700277 vpx_bit_depth_t bit_depth_;
278 int mask_;
Jingning Han362809d2013-06-18 10:46:33 -0700279};
280
Jingning Han30d4c5e2013-11-12 12:47:32 -0800281class Trans4x4DCT
282 : public Trans4x4TestBase,
James Zernb8b3dd92014-07-16 18:55:40 -0700283 public ::testing::TestWithParam<Dct4x4Param> {
Jingning Han30d4c5e2013-11-12 12:47:32 -0800284 public:
285 virtual ~Trans4x4DCT() {}
Daniel Kang58156f12012-06-26 18:11:33 -0700286
Jingning Han30d4c5e2013-11-12 12:47:32 -0800287 virtual void SetUp() {
288 fwd_txfm_ = GET_PARAM(0);
289 inv_txfm_ = GET_PARAM(1);
290 tx_type_ = GET_PARAM(2);
291 pitch_ = 4;
292 fwd_txfm_ref = fdct4x4_ref;
Deb Mukherjee10783d42014-09-02 16:34:09 -0700293 bit_depth_ = GET_PARAM(3);
294 mask_ = (1 << bit_depth_) - 1;
Jingning Han30d4c5e2013-11-12 12:47:32 -0800295 }
296 virtual void TearDown() { libvpx_test::ClearSystemState(); }
Daniel Kang58156f12012-06-26 18:11:33 -0700297
Jingning Han30d4c5e2013-11-12 12:47:32 -0800298 protected:
Deb Mukherjee10783d42014-09-02 16:34:09 -0700299 void RunFwdTxfm(const int16_t *in, tran_low_t *out, int stride) {
Jingning Han30d4c5e2013-11-12 12:47:32 -0800300 fwd_txfm_(in, out, stride);
301 }
Deb Mukherjee10783d42014-09-02 16:34:09 -0700302 void RunInvTxfm(const tran_low_t *out, uint8_t *dst, int stride) {
Jingning Han30d4c5e2013-11-12 12:47:32 -0800303 inv_txfm_(out, dst, stride);
Daniel Kang58156f12012-06-26 18:11:33 -0700304 }
305
James Zernb8b3dd92014-07-16 18:55:40 -0700306 FdctFunc fwd_txfm_;
307 IdctFunc inv_txfm_;
Jingning Han30d4c5e2013-11-12 12:47:32 -0800308};
Daniel Kang58156f12012-06-26 18:11:33 -0700309
Jingning Han30d4c5e2013-11-12 12:47:32 -0800310TEST_P(Trans4x4DCT, AccuracyCheck) {
Alex Converse9f9f87c2014-05-05 13:50:12 -0700311 RunAccuracyCheck(1);
Jingning Han362809d2013-06-18 10:46:33 -0700312}
Daniel Kang58156f12012-06-26 18:11:33 -0700313
Jingning Han30d4c5e2013-11-12 12:47:32 -0800314TEST_P(Trans4x4DCT, CoeffCheck) {
315 RunCoeffCheck();
Jingning Han362809d2013-06-18 10:46:33 -0700316}
Daniel Kang58156f12012-06-26 18:11:33 -0700317
Jingning Han30d4c5e2013-11-12 12:47:32 -0800318TEST_P(Trans4x4DCT, MemCheck) {
319 RunMemCheck();
320}
321
322TEST_P(Trans4x4DCT, InvAccuracyCheck) {
Alex Converse9f9f87c2014-05-05 13:50:12 -0700323 RunInvAccuracyCheck(1);
Jingning Han30d4c5e2013-11-12 12:47:32 -0800324}
325
326class Trans4x4HT
327 : public Trans4x4TestBase,
James Zernb8b3dd92014-07-16 18:55:40 -0700328 public ::testing::TestWithParam<Ht4x4Param> {
Jingning Han30d4c5e2013-11-12 12:47:32 -0800329 public:
330 virtual ~Trans4x4HT() {}
331
332 virtual void SetUp() {
333 fwd_txfm_ = GET_PARAM(0);
334 inv_txfm_ = GET_PARAM(1);
335 tx_type_ = GET_PARAM(2);
336 pitch_ = 4;
337 fwd_txfm_ref = fht4x4_ref;
Deb Mukherjee10783d42014-09-02 16:34:09 -0700338 bit_depth_ = GET_PARAM(3);
339 mask_ = (1 << bit_depth_) - 1;
Jingning Han30d4c5e2013-11-12 12:47:32 -0800340 }
341 virtual void TearDown() { libvpx_test::ClearSystemState(); }
342
343 protected:
Deb Mukherjee10783d42014-09-02 16:34:09 -0700344 void RunFwdTxfm(const int16_t *in, tran_low_t *out, int stride) {
Jingning Han30d4c5e2013-11-12 12:47:32 -0800345 fwd_txfm_(in, out, stride, tx_type_);
346 }
347
Deb Mukherjee10783d42014-09-02 16:34:09 -0700348 void RunInvTxfm(const tran_low_t *out, uint8_t *dst, int stride) {
Jingning Han30d4c5e2013-11-12 12:47:32 -0800349 inv_txfm_(out, dst, stride, tx_type_);
350 }
351
James Zernb8b3dd92014-07-16 18:55:40 -0700352 FhtFunc fwd_txfm_;
353 IhtFunc inv_txfm_;
Jingning Han30d4c5e2013-11-12 12:47:32 -0800354};
355
356TEST_P(Trans4x4HT, AccuracyCheck) {
Alex Converse9f9f87c2014-05-05 13:50:12 -0700357 RunAccuracyCheck(1);
Jingning Han30d4c5e2013-11-12 12:47:32 -0800358}
359
360TEST_P(Trans4x4HT, CoeffCheck) {
361 RunCoeffCheck();
362}
363
364TEST_P(Trans4x4HT, MemCheck) {
365 RunMemCheck();
366}
367
368TEST_P(Trans4x4HT, InvAccuracyCheck) {
Alex Converse9f9f87c2014-05-05 13:50:12 -0700369 RunInvAccuracyCheck(1);
Jingning Han30d4c5e2013-11-12 12:47:32 -0800370}
371
Alex Converse9f9f87c2014-05-05 13:50:12 -0700372class Trans4x4WHT
373 : public Trans4x4TestBase,
James Zernb8b3dd92014-07-16 18:55:40 -0700374 public ::testing::TestWithParam<Dct4x4Param> {
Alex Converse9f9f87c2014-05-05 13:50:12 -0700375 public:
376 virtual ~Trans4x4WHT() {}
377
378 virtual void SetUp() {
379 fwd_txfm_ = GET_PARAM(0);
380 inv_txfm_ = GET_PARAM(1);
381 tx_type_ = GET_PARAM(2);
382 pitch_ = 4;
383 fwd_txfm_ref = fwht4x4_ref;
Deb Mukherjee10783d42014-09-02 16:34:09 -0700384 bit_depth_ = GET_PARAM(3);
385 mask_ = (1 << bit_depth_) - 1;
Alex Converse9f9f87c2014-05-05 13:50:12 -0700386 }
387 virtual void TearDown() { libvpx_test::ClearSystemState(); }
388
389 protected:
Deb Mukherjee10783d42014-09-02 16:34:09 -0700390 void RunFwdTxfm(const int16_t *in, tran_low_t *out, int stride) {
Alex Converse9f9f87c2014-05-05 13:50:12 -0700391 fwd_txfm_(in, out, stride);
392 }
Deb Mukherjee10783d42014-09-02 16:34:09 -0700393 void RunInvTxfm(const tran_low_t *out, uint8_t *dst, int stride) {
Alex Converse9f9f87c2014-05-05 13:50:12 -0700394 inv_txfm_(out, dst, stride);
395 }
396
James Zernb8b3dd92014-07-16 18:55:40 -0700397 FdctFunc fwd_txfm_;
398 IdctFunc inv_txfm_;
Alex Converse9f9f87c2014-05-05 13:50:12 -0700399};
400
401TEST_P(Trans4x4WHT, AccuracyCheck) {
402 RunAccuracyCheck(0);
403}
404
405TEST_P(Trans4x4WHT, CoeffCheck) {
406 RunCoeffCheck();
407}
408
409TEST_P(Trans4x4WHT, MemCheck) {
410 RunMemCheck();
411}
412
413TEST_P(Trans4x4WHT, InvAccuracyCheck) {
414 RunInvAccuracyCheck(0);
415}
Jingning Han30d4c5e2013-11-12 12:47:32 -0800416using std::tr1::make_tuple;
417
Deb Mukherjee10783d42014-09-02 16:34:09 -0700418#if CONFIG_VP9_HIGHBITDEPTH
Jingning Han30d4c5e2013-11-12 12:47:32 -0800419INSTANTIATE_TEST_CASE_P(
420 C, Trans4x4DCT,
421 ::testing::Values(
Jingning Han4b5109c2015-07-28 15:57:40 -0700422 make_tuple(&vpx_highbd_fdct4x4_c, &idct4x4_10, 0, VPX_BITS_10),
423 make_tuple(&vpx_highbd_fdct4x4_c, &idct4x4_12, 0, VPX_BITS_12),
Jingning Han08a453b2015-08-03 14:51:10 -0700424 make_tuple(&vpx_fdct4x4_c, &vpx_idct4x4_16_add_c, 0, VPX_BITS_8)));
Deb Mukherjee10783d42014-09-02 16:34:09 -0700425#else
426INSTANTIATE_TEST_CASE_P(
427 C, Trans4x4DCT,
428 ::testing::Values(
Jingning Han08a453b2015-08-03 14:51:10 -0700429 make_tuple(&vpx_fdct4x4_c, &vpx_idct4x4_16_add_c, 0, VPX_BITS_8)));
Peter de Rivaz7e40a552014-10-24 08:48:02 +0100430#endif // CONFIG_VP9_HIGHBITDEPTH
Deb Mukherjee10783d42014-09-02 16:34:09 -0700431
432#if CONFIG_VP9_HIGHBITDEPTH
Jingning Han30d4c5e2013-11-12 12:47:32 -0800433INSTANTIATE_TEST_CASE_P(
434 C, Trans4x4HT,
435 ::testing::Values(
Deb Mukherjee1929c9b2014-10-08 12:43:22 -0700436 make_tuple(&vp9_highbd_fht4x4_c, &iht4x4_10, 0, VPX_BITS_10),
437 make_tuple(&vp9_highbd_fht4x4_c, &iht4x4_10, 1, VPX_BITS_10),
438 make_tuple(&vp9_highbd_fht4x4_c, &iht4x4_10, 2, VPX_BITS_10),
439 make_tuple(&vp9_highbd_fht4x4_c, &iht4x4_10, 3, VPX_BITS_10),
440 make_tuple(&vp9_highbd_fht4x4_c, &iht4x4_12, 0, VPX_BITS_12),
441 make_tuple(&vp9_highbd_fht4x4_c, &iht4x4_12, 1, VPX_BITS_12),
442 make_tuple(&vp9_highbd_fht4x4_c, &iht4x4_12, 2, VPX_BITS_12),
443 make_tuple(&vp9_highbd_fht4x4_c, &iht4x4_12, 3, VPX_BITS_12),
Deb Mukherjee10783d42014-09-02 16:34:09 -0700444 make_tuple(&vp9_fht4x4_c, &vp9_iht4x4_16_add_c, 0, VPX_BITS_8),
445 make_tuple(&vp9_fht4x4_c, &vp9_iht4x4_16_add_c, 1, VPX_BITS_8),
446 make_tuple(&vp9_fht4x4_c, &vp9_iht4x4_16_add_c, 2, VPX_BITS_8),
447 make_tuple(&vp9_fht4x4_c, &vp9_iht4x4_16_add_c, 3, VPX_BITS_8)));
448#else
449INSTANTIATE_TEST_CASE_P(
450 C, Trans4x4HT,
451 ::testing::Values(
452 make_tuple(&vp9_fht4x4_c, &vp9_iht4x4_16_add_c, 0, VPX_BITS_8),
453 make_tuple(&vp9_fht4x4_c, &vp9_iht4x4_16_add_c, 1, VPX_BITS_8),
454 make_tuple(&vp9_fht4x4_c, &vp9_iht4x4_16_add_c, 2, VPX_BITS_8),
455 make_tuple(&vp9_fht4x4_c, &vp9_iht4x4_16_add_c, 3, VPX_BITS_8)));
Peter de Rivaz7e40a552014-10-24 08:48:02 +0100456#endif // CONFIG_VP9_HIGHBITDEPTH
Deb Mukherjee10783d42014-09-02 16:34:09 -0700457
458#if CONFIG_VP9_HIGHBITDEPTH
Alex Converse9f9f87c2014-05-05 13:50:12 -0700459INSTANTIATE_TEST_CASE_P(
460 C, Trans4x4WHT,
461 ::testing::Values(
Deb Mukherjee1929c9b2014-10-08 12:43:22 -0700462 make_tuple(&vp9_highbd_fwht4x4_c, &iwht4x4_10, 0, VPX_BITS_10),
463 make_tuple(&vp9_highbd_fwht4x4_c, &iwht4x4_12, 0, VPX_BITS_12),
Jingning Han08a453b2015-08-03 14:51:10 -0700464 make_tuple(&vp9_fwht4x4_c, &vpx_iwht4x4_16_add_c, 0, VPX_BITS_8)));
Deb Mukherjee10783d42014-09-02 16:34:09 -0700465#else
466INSTANTIATE_TEST_CASE_P(
467 C, Trans4x4WHT,
468 ::testing::Values(
Jingning Han08a453b2015-08-03 14:51:10 -0700469 make_tuple(&vp9_fwht4x4_c, &vpx_iwht4x4_16_add_c, 0, VPX_BITS_8)));
Peter de Rivaz7e40a552014-10-24 08:48:02 +0100470#endif // CONFIG_VP9_HIGHBITDEPTH
Jingning Han30d4c5e2013-11-12 12:47:32 -0800471
Deb Mukherjeed50716f2014-10-02 15:43:27 -0700472#if HAVE_NEON_ASM && !CONFIG_VP9_HIGHBITDEPTH && !CONFIG_EMULATE_HARDWARE
James Zern08c31802014-02-25 23:11:49 -0800473INSTANTIATE_TEST_CASE_P(
474 NEON, Trans4x4DCT,
475 ::testing::Values(
Jingning Han4b5109c2015-07-28 15:57:40 -0700476 make_tuple(&vpx_fdct4x4_c,
Jingning Han08a453b2015-08-03 14:51:10 -0700477 &vpx_idct4x4_16_add_neon, 0, VPX_BITS_8)));
James Yu6b710132014-01-27 18:38:35 +0800478#endif // HAVE_NEON_ASM && !CONFIG_VP9_HIGHBITDEPTH && !CONFIG_EMULATE_HARDWARE
479
480#if HAVE_NEON && !CONFIG_VP9_HIGHBITDEPTH && !CONFIG_EMULATE_HARDWARE
James Zern08c31802014-02-25 23:11:49 -0800481INSTANTIATE_TEST_CASE_P(
James Yu6b710132014-01-27 18:38:35 +0800482 NEON, Trans4x4HT,
James Zern08c31802014-02-25 23:11:49 -0800483 ::testing::Values(
Deb Mukherjee10783d42014-09-02 16:34:09 -0700484 make_tuple(&vp9_fht4x4_c, &vp9_iht4x4_16_add_neon, 0, VPX_BITS_8),
485 make_tuple(&vp9_fht4x4_c, &vp9_iht4x4_16_add_neon, 1, VPX_BITS_8),
486 make_tuple(&vp9_fht4x4_c, &vp9_iht4x4_16_add_neon, 2, VPX_BITS_8),
487 make_tuple(&vp9_fht4x4_c, &vp9_iht4x4_16_add_neon, 3, VPX_BITS_8)));
James Yu6b710132014-01-27 18:38:35 +0800488#endif // HAVE_NEON && !CONFIG_VP9_HIGHBITDEPTH && !CONFIG_EMULATE_HARDWARE
James Zern08c31802014-02-25 23:11:49 -0800489
Deb Mukherjeed50716f2014-10-02 15:43:27 -0700490#if CONFIG_USE_X86INC && HAVE_MMX && !CONFIG_VP9_HIGHBITDEPTH && \
491 !CONFIG_EMULATE_HARDWARE
Alex Converseb5422fa2014-05-07 12:51:11 -0700492INSTANTIATE_TEST_CASE_P(
493 MMX, Trans4x4WHT,
494 ::testing::Values(
Jingning Han08a453b2015-08-03 14:51:10 -0700495 make_tuple(&vp9_fwht4x4_mmx, &vpx_iwht4x4_16_add_c, 0, VPX_BITS_8)));
Alex Converseb5422fa2014-05-07 12:51:11 -0700496#endif
497
Alex Conversed8426d62015-07-13 11:12:45 -0700498#if CONFIG_USE_X86INC && HAVE_SSE2 && !CONFIG_VP9_HIGHBITDEPTH && \
499 !CONFIG_EMULATE_HARDWARE
500INSTANTIATE_TEST_CASE_P(
501 SSE2, Trans4x4WHT,
502 ::testing::Values(
Jingning Han08a453b2015-08-03 14:51:10 -0700503 make_tuple(&vp9_fwht4x4_c, &vpx_iwht4x4_16_add_sse2, 0, VPX_BITS_8)));
Alex Conversed8426d62015-07-13 11:12:45 -0700504#endif
505
Deb Mukherjeed50716f2014-10-02 15:43:27 -0700506#if HAVE_SSE2 && !CONFIG_VP9_HIGHBITDEPTH && !CONFIG_EMULATE_HARDWARE
Jingning Han30d4c5e2013-11-12 12:47:32 -0800507INSTANTIATE_TEST_CASE_P(
508 SSE2, Trans4x4DCT,
509 ::testing::Values(
Jingning Han4b5109c2015-07-28 15:57:40 -0700510 make_tuple(&vpx_fdct4x4_sse2,
Jingning Han08a453b2015-08-03 14:51:10 -0700511 &vpx_idct4x4_16_add_sse2, 0, VPX_BITS_8)));
Jingning Han30d4c5e2013-11-12 12:47:32 -0800512INSTANTIATE_TEST_CASE_P(
513 SSE2, Trans4x4HT,
514 ::testing::Values(
Deb Mukherjee10783d42014-09-02 16:34:09 -0700515 make_tuple(&vp9_fht4x4_sse2, &vp9_iht4x4_16_add_sse2, 0, VPX_BITS_8),
516 make_tuple(&vp9_fht4x4_sse2, &vp9_iht4x4_16_add_sse2, 1, VPX_BITS_8),
517 make_tuple(&vp9_fht4x4_sse2, &vp9_iht4x4_16_add_sse2, 2, VPX_BITS_8),
518 make_tuple(&vp9_fht4x4_sse2, &vp9_iht4x4_16_add_sse2, 3, VPX_BITS_8)));
Peter de Rivaz7e40a552014-10-24 08:48:02 +0100519#endif // HAVE_SSE2 && !CONFIG_VP9_HIGHBITDEPTH && !CONFIG_EMULATE_HARDWARE
Jingning Han30d4c5e2013-11-12 12:47:32 -0800520
Peter de Rivaz7e40a552014-10-24 08:48:02 +0100521#if HAVE_SSE2 && CONFIG_VP9_HIGHBITDEPTH && !CONFIG_EMULATE_HARDWARE
522INSTANTIATE_TEST_CASE_P(
523 SSE2, Trans4x4DCT,
524 ::testing::Values(
Jingning Han4b5109c2015-07-28 15:57:40 -0700525 make_tuple(&vpx_highbd_fdct4x4_c, &idct4x4_10_sse2, 0, VPX_BITS_10),
526 make_tuple(&vpx_highbd_fdct4x4_sse2, &idct4x4_10_sse2, 0, VPX_BITS_10),
527 make_tuple(&vpx_highbd_fdct4x4_c, &idct4x4_12_sse2, 0, VPX_BITS_12),
528 make_tuple(&vpx_highbd_fdct4x4_sse2, &idct4x4_12_sse2, 0, VPX_BITS_12),
Jingning Han08a453b2015-08-03 14:51:10 -0700529 make_tuple(&vpx_fdct4x4_sse2, &vpx_idct4x4_16_add_c, 0,
Peter de Rivaz7e40a552014-10-24 08:48:02 +0100530 VPX_BITS_8)));
531
532INSTANTIATE_TEST_CASE_P(
533 SSE2, Trans4x4HT,
534 ::testing::Values(
Peter de Rivaz7e40a552014-10-24 08:48:02 +0100535 make_tuple(&vp9_fht4x4_sse2, &vp9_iht4x4_16_add_c, 0, VPX_BITS_8),
536 make_tuple(&vp9_fht4x4_sse2, &vp9_iht4x4_16_add_c, 1, VPX_BITS_8),
537 make_tuple(&vp9_fht4x4_sse2, &vp9_iht4x4_16_add_c, 2, VPX_BITS_8),
538 make_tuple(&vp9_fht4x4_sse2, &vp9_iht4x4_16_add_c, 3, VPX_BITS_8)));
539#endif // HAVE_SSE2 && CONFIG_VP9_HIGHBITDEPTH && !CONFIG_EMULATE_HARDWARE
Parag Salasakar54a6f732015-06-02 12:16:28 +0530540
Jingning Han9aaf5232015-07-22 11:53:21 -0700541#if HAVE_MSA && !CONFIG_VP9_HIGHBITDEPTH && !CONFIG_EMULATE_HARDWARE
Parag Salasakar54a6f732015-06-02 12:16:28 +0530542INSTANTIATE_TEST_CASE_P(
543 MSA, Trans4x4DCT,
544 ::testing::Values(
Jingning Han08a453b2015-08-03 14:51:10 -0700545 make_tuple(&vpx_fdct4x4_msa, &vpx_idct4x4_16_add_msa, 0, VPX_BITS_8)));
Parag Salasakar54a6f732015-06-02 12:16:28 +0530546INSTANTIATE_TEST_CASE_P(
547 MSA, Trans4x4HT,
548 ::testing::Values(
Parag Salasakarbc949992015-06-22 14:30:24 +0530549 make_tuple(&vp9_fht4x4_msa, &vp9_iht4x4_16_add_msa, 0, VPX_BITS_8),
550 make_tuple(&vp9_fht4x4_msa, &vp9_iht4x4_16_add_msa, 1, VPX_BITS_8),
551 make_tuple(&vp9_fht4x4_msa, &vp9_iht4x4_16_add_msa, 2, VPX_BITS_8),
552 make_tuple(&vp9_fht4x4_msa, &vp9_iht4x4_16_add_msa, 3, VPX_BITS_8)));
Parag Salasakar54a6f732015-06-02 12:16:28 +0530553#endif // HAVE_MSA && !CONFIG_VP9_HIGHBITDEPTH && !CONFIG_EMULATE_HARDWARE
Daniel Kang58156f12012-06-26 18:11:33 -0700554} // namespace