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 171184aebb761cbeac9124c37189a80a1a58f04b6bhkuang#include "./vp9_rtcd.h" 18ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 191184aebb761cbeac9124c37189a80a1a58f04b6bhkuang#include "test/acm_random.h" 20ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang#include "vpx/vpx_integer.h" 21ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 22ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangusing libvpx_test::ACMRandom; 23ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 24ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangnamespace { 25ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 26ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang#ifdef _MSC_VER 27ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangstatic int round(double x) { 281184aebb761cbeac9124c37189a80a1a58f04b6bhkuang if (x < 0) 291184aebb761cbeac9124c37189a80a1a58f04b6bhkuang return static_cast<int>(ceil(x - 0.5)); 30ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang else 311184aebb761cbeac9124c37189a80a1a58f04b6bhkuang return static_cast<int>(floor(x + 0.5)); 32ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang} 33ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang#endif 34ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 35ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangvoid reference_dct_1d(double input[8], double output[8]) { 36ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang const double kPi = 3.141592653589793238462643383279502884; 37ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang const double kInvSqrt2 = 0.707106781186547524400844362104; 38ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int k = 0; k < 8; k++) { 39ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[k] = 0.0; 40ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int n = 0; n < 8; n++) 41ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[k] += input[n]*cos(kPi*(2*n+1)*k/16.0); 42ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang if (k == 0) 43ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[k] = output[k]*kInvSqrt2; 44ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 45ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang} 46ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 47ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangvoid reference_dct_2d(int16_t input[64], double output[64]) { 48ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang // First transform columns 49ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < 8; ++i) { 50ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang double temp_in[8], temp_out[8]; 51ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 52ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang temp_in[j] = input[j*8 + i]; 53ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang reference_dct_1d(temp_in, temp_out); 54ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 55ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[j*8 + i] = temp_out[j]; 56ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 57ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang // Then transform rows 58ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < 8; ++i) { 59ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang double temp_in[8], temp_out[8]; 60ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 61ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang temp_in[j] = output[j + i*8]; 62ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang reference_dct_1d(temp_in, temp_out); 63ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 64ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[j + i*8] = temp_out[j]; 65ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 66ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang // Scale by some magic number 67ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < 64; ++i) 68ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[i] *= 2; 69ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang} 70ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 71ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangvoid reference_idct_1d(double input[8], double output[8]) { 72ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang const double kPi = 3.141592653589793238462643383279502884; 73ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang const double kSqrt2 = 1.414213562373095048801688724209698; 74ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int k = 0; k < 8; k++) { 75ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[k] = 0.0; 76ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int n = 0; n < 8; n++) { 77ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[k] += input[n]*cos(kPi*(2*k+1)*n/16.0); 78ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang if (n == 0) 79ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[k] = output[k]/kSqrt2; 80ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 81ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 82ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang} 83ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 84ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangvoid reference_idct_2d(double input[64], int16_t output[64]) { 85ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang double out[64], out2[64]; 86ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang // First transform rows 87ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < 8; ++i) { 88ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang double temp_in[8], temp_out[8]; 89ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 90ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang temp_in[j] = input[j + i*8]; 91ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang reference_idct_1d(temp_in, temp_out); 92ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 93ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang out[j + i*8] = temp_out[j]; 94ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 95ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang // Then transform columns 96ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < 8; ++i) { 97ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang double temp_in[8], temp_out[8]; 98ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 99ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang temp_in[j] = out[j*8 + i]; 100ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang reference_idct_1d(temp_in, temp_out); 101ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int j = 0; j < 8; ++j) 102ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang out2[j*8 + i] = temp_out[j]; 103ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang } 104ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < 64; ++i) 105ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang output[i] = round(out2[i]/32); 106ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang} 107ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang 108ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuangTEST(VP9Idct8x8Test, AccuracyCheck) { 109ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang ACMRandom rnd(ACMRandom::DeterministicSeed()); 110ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang const int count_test_block = 10000; 111ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang for (int i = 0; i < count_test_block; ++i) { 112ba164dffc5a6795bce97fae02b51ccf3330e15e4hkuang int16_t input[64], 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]); 1275ae7ac49f08a179e4f054d99fcfc9dce78d26e58hkuang vp9_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