test/idct8x8_test.cc

   1 /*
   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"
  16
  17 #include "./vpx_dsp_rtcd.h"
  18 #include "test/acm_random.h"
  19 #include "vpx/vpx_integer.h"
  20 #include "vpx_ports/msvc.h"  // for round()
  21
  22 using libvpx_test::ACMRandom;
  23
  24 namespace {
  25
  26 void reference_dct_1d(double input[8], double output[8]) {
  27   const double kPi = 3.141592653589793238462643383279502884;
  28   const double kInvSqrt2 = 0.707106781186547524400844362104;
  29   for (int k = 0; k < 8; k++) {
  30     output[k] = 0.0;
  31     for (int n = 0; n < 8; n++) {
  32       output[k] += input[n] * cos(kPi * (2 * n + 1) * k / 16.0);
  33     }
  34     if (k == 0) output[k] = output[k] * kInvSqrt2;
  35   }
  36 }
  37
  38 void reference_dct_2d(int16_t input[64], double output[64]) {
  39   // First transform columns
  40   for (int i = 0; i < 8; ++i) {
  41     double temp_in[8], temp_out[8];
  42     for (int j = 0; j < 8; ++j) temp_in[j] = input[j * 8 + i];
  43     reference_dct_1d(temp_in, temp_out);
  44     for (int j = 0; j < 8; ++j) output[j * 8 + i] = temp_out[j];
  45   }
  46   // Then transform rows
  47   for (int i = 0; i < 8; ++i) {
  48     double temp_in[8], temp_out[8];
  49     for (int j = 0; j < 8; ++j) temp_in[j] = output[j + i * 8];
  50     reference_dct_1d(temp_in, temp_out);
  51     for (int j = 0; j < 8; ++j) output[j + i * 8] = temp_out[j];
  52   }
  53   // Scale by some magic number
  54   for (int i = 0; i < 64; ++i) output[i] *= 2;
  55 }
  56
  57 TEST(VP9Idct8x8Test, AccuracyCheck) {
  58   ACMRandom rnd(ACMRandom::DeterministicSeed());
  59   const int count_test_block = 10000;
  60   for (int i = 0; i < count_test_block; ++i) {
  61     int16_t input[64];
  62     tran_low_t coeff[64];
  63     double output_r[64];
  64     uint8_t dst[64], src[64];
  65
  66     for (int j = 0; j < 64; ++j) {
  67       src[j] = rnd.Rand8();
  68       dst[j] = rnd.Rand8();
  69     }
  70     // Initialize a test block with input range [-255, 255].
  71     for (int j = 0; j < 64; ++j) input[j] = src[j] - dst[j];
  72
  73     reference_dct_2d(input, output_r);
  74     for (int j = 0; j < 64; ++j) {
  75       coeff[j] = static_cast<tran_low_t>(round(output_r[j]));
  76     }
  77     vpx_idct8x8_64_add_c(coeff, dst, 8);
  78     for (int j = 0; j < 64; ++j) {
  79       const int diff = dst[j] - src[j];
  80       const int error = diff * diff;
  81       EXPECT_GE(1, error) << "Error: 8x8 FDCT/IDCT has error " << error
  82                           << " at index " << j;
  83     }
  84   }
  85 }
  86
  87 }  // namespace