idct8x8_test.cc revision da49e34c1fb5e99681f4ad99c21d9cfd83eddb96
1ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang/* 2ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang * Copyright (c) 2012 The WebM project authors. All Rights Reserved. 3ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang * 4ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang * Use of this source code is governed by a BSD-style license 5ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang * that can be found in the LICENSE file in the root of the source 6ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang * tree. An additional intellectual property rights grant can be found 7ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang * in the file PATENTS. All contributing project authors may 8ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang * be found in the AUTHORS file in the root of the source tree. 9ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang */ 10ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 11ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang#include <math.h> 12ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang#include <stdlib.h> 13ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang#include <string.h> 14ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 15ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang#include "third_party/googletest/src/include/gtest/gtest.h" 16ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 17da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#include "./vpx_dsp_rtcd.h" 181184aebb761cbeac9124c37189a80a1a58f04b6bhkuang#include "test/acm_random.h" 19ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang#include "vpx/vpx_integer.h" 20ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 21ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangusing libvpx_test::ACMRandom; 22ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 23ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangnamespace { 24ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 25ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang#ifdef _MSC_VER 26ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangstatic int round(double x) { 271184aebb761cbeac9124c37189a80a1a58f04b6bhkuang if (x < 0) 281184aebb761cbeac9124c37189a80a1a58f04b6bhkuang return static_cast<int>(ceil(x - 0.5)); 29ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang else 301184aebb761cbeac9124c37189a80a1a58f04b6bhkuang return static_cast<int>(floor(x + 0.5)); 31ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang} 32ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang#endif 33ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 34ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangvoid reference_dct_1d(double input[8], double output[8]) { 35ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang const double kPi = 3.141592653589793238462643383279502884; 36ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang const double kInvSqrt2 = 0.707106781186547524400844362104; 37ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int k = 0; k < 8; k++) { 38ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[k] = 0.0; 39ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int n = 0; n < 8; n++) 40ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[k] += input[n]*cos(kPi*(2*n+1)*k/16.0); 41ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang if (k == 0) 42ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[k] = output[k]*kInvSqrt2; 43ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 44ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang} 45ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 46ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangvoid reference_dct_2d(int16_t input[64], double output[64]) { 47ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang // First transform columns 48ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < 8; ++i) { 49ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang double temp_in[8], temp_out[8]; 50ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 51ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang temp_in[j] = input[j*8 + i]; 52ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang reference_dct_1d(temp_in, temp_out); 53ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 54ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[j*8 + i] = temp_out[j]; 55ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 56ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang // Then transform rows 57ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < 8; ++i) { 58ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang double temp_in[8], temp_out[8]; 59ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 60ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang temp_in[j] = output[j + i*8]; 61ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang reference_dct_1d(temp_in, temp_out); 62ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 63ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[j + i*8] = temp_out[j]; 64ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 65ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang // Scale by some magic number 66ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < 64; ++i) 67ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[i] *= 2; 68ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang} 69ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 70ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangvoid reference_idct_1d(double input[8], double output[8]) { 71ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang const double kPi = 3.141592653589793238462643383279502884; 72ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang const double kSqrt2 = 1.414213562373095048801688724209698; 73ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int k = 0; k < 8; k++) { 74ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[k] = 0.0; 75ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int n = 0; n < 8; n++) { 76ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[k] += input[n]*cos(kPi*(2*k+1)*n/16.0); 77ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang if (n == 0) 78ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[k] = output[k]/kSqrt2; 79ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 80ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 81ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang} 82ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 83ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangvoid reference_idct_2d(double input[64], int16_t output[64]) { 84ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang double out[64], out2[64]; 85ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang // First transform rows 86ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < 8; ++i) { 87ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang double temp_in[8], temp_out[8]; 88ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 89ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang temp_in[j] = input[j + i*8]; 90ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang reference_idct_1d(temp_in, temp_out); 91ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 92ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang out[j + i*8] = temp_out[j]; 93ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 94ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang // Then transform columns 95ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < 8; ++i) { 96ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang double temp_in[8], temp_out[8]; 97ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 98ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang temp_in[j] = out[j*8 + i]; 99ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang reference_idct_1d(temp_in, temp_out); 100ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 101ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang out2[j*8 + i] = temp_out[j]; 102ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 103ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < 64; ++i) 104ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[i] = round(out2[i]/32); 105ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang} 106ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 107ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangTEST(VP9Idct8x8Test, AccuracyCheck) { 108ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang ACMRandom rnd(ACMRandom::DeterministicSeed()); 109ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang const int count_test_block = 10000; 110ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < count_test_block; ++i) { 111da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian int16_t input[64]; 112da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian tran_low_t coeff[64]; 113ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang double output_r[64]; 114ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang uint8_t dst[64], src[64]; 115ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 116ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 64; ++j) { 117ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang src[j] = rnd.Rand8(); 118ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang dst[j] = rnd.Rand8(); 119ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 120ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang // Initialize a test block with input range [-255, 255]. 121ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 64; ++j) 122ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang input[j] = src[j] - dst[j]; 123ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 124ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang reference_dct_2d(input, output_r); 125ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 64; ++j) 126ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang coeff[j] = round(output_r[j]); 127da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_idct8x8_64_add_c(coeff, dst, 8); 128ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 64; ++j) { 129ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang const int diff = dst[j] - src[j]; 130ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang const int error = diff * diff; 131ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang EXPECT_GE(1, error) 132ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang << "Error: 8x8 FDCT/IDCT has error " << error 133ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang << " at index " << j; 134ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 135ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 136ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang} 137ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 138ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang} // namespace 139