granicus.if.org Git - libvpx/blob - test/test_intra_pred_speed.cc

   1 /*
   2  *  Copyright (c) 2015 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 //  Test and time VPX intra-predictor functions
  11
  12 #include <stdio.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 "test/clear_system_state.h"
  20 #include "test/md5_helper.h"
  21 #include "vpx/vpx_integer.h"
  22 #include "vpx_ports/mem.h"
  23 #include "vpx_ports/vpx_timer.h"
  24
  25 // -----------------------------------------------------------------------------
  26
  27 namespace {
  28
  29 typedef void (*VpxPredFunc)(uint8_t *dst, ptrdiff_t y_stride,
  30                             const uint8_t *above, const uint8_t *left);
  31
  32 const int kBPS = 32;
  33 const int kTotalPixels = 32 * kBPS;
  34 const int kNumVp9IntraPredFuncs = 13;
  35 const char *kVp9IntraPredNames[kNumVp9IntraPredFuncs] = {
  36   "DC_PRED",   "DC_LEFT_PRED", "DC_TOP_PRED", "DC_128_PRED", "V_PRED",
  37   "H_PRED",    "D45_PRED",     "D135_PRED",   "D117_PRED",   "D153_PRED",
  38   "D207_PRED", "D63_PRED",     "TM_PRED"
  39 };
  40
  41 template <typename Pixel>
  42 struct IntraPredTestMem {
  43   void Init(int block_size, int bd) {
  44     libvpx_test::ACMRandom rnd(libvpx_test::ACMRandom::DeterministicSeed());
  45     Pixel *const above = above_mem + 16;
  46     const int mask = (1 << bd) - 1;
  47     for (int i = 0; i < kTotalPixels; ++i) ref_src[i] = rnd.Rand16() & mask;
  48     for (int i = 0; i < kBPS; ++i) left[i] = rnd.Rand16() & mask;
  49     for (int i = -1; i < kBPS; ++i) above[i] = rnd.Rand16() & mask;
  50
  51     // some code assumes the top row has been extended:
  52     // d45/d63 C-code, for instance, but not the assembly.
  53     // TODO(jzern): this style of extension isn't strictly necessary.
  54     ASSERT_LE(block_size, kBPS);
  55     for (int i = block_size; i < 2 * kBPS; ++i) {
  56       above[i] = above[block_size - 1];
  57     }
  58   }
  59
  60   DECLARE_ALIGNED(16, Pixel, src[kTotalPixels]);
  61   DECLARE_ALIGNED(16, Pixel, ref_src[kTotalPixels]);
  62   DECLARE_ALIGNED(16, Pixel, left[kBPS]);
  63   DECLARE_ALIGNED(16, Pixel, above_mem[2 * kBPS + 16]);
  64 };
  65
  66 typedef IntraPredTestMem<uint8_t> Vp9IntraPredTestMem;
  67
  68 void CheckMd5Signature(const char name[], const char *const signatures[],
  69                        const void *data, size_t data_size, int elapsed_time,
  70                        int idx) {
  71   libvpx_test::MD5 md5;
  72   md5.Add(reinterpret_cast<const uint8_t *>(data), data_size);
  73   printf("Mode %s[%12s]: %5d ms     MD5: %s\n", name, kVp9IntraPredNames[idx],
  74          elapsed_time, md5.Get());
  75   EXPECT_STREQ(signatures[idx], md5.Get());
  76 }
  77
  78 void TestIntraPred(const char name[], VpxPredFunc const *pred_funcs,
  79                    const char *const signatures[], int block_size) {
  80   const int kNumTests = static_cast<int>(
  81       2.e10 / (block_size * block_size * kNumVp9IntraPredFuncs));
  82   Vp9IntraPredTestMem intra_pred_test_mem;
  83   const uint8_t *const above = intra_pred_test_mem.above_mem + 16;
  84
  85   intra_pred_test_mem.Init(block_size, 8);
  86
  87   for (int k = 0; k < kNumVp9IntraPredFuncs; ++k) {
  88     if (pred_funcs[k] == NULL) continue;
  89     memcpy(intra_pred_test_mem.src, intra_pred_test_mem.ref_src,
  90            sizeof(intra_pred_test_mem.src));
  91     vpx_usec_timer timer;
  92     vpx_usec_timer_start(&timer);
  93     for (int num_tests = 0; num_tests < kNumTests; ++num_tests) {
  94       pred_funcs[k](intra_pred_test_mem.src, kBPS, above,
  95                     intra_pred_test_mem.left);
  96     }
  97     libvpx_test::ClearSystemState();
  98     vpx_usec_timer_mark(&timer);
  99     const int elapsed_time =
 100         static_cast<int>(vpx_usec_timer_elapsed(&timer) / 1000);
 101     CheckMd5Signature(name, signatures, intra_pred_test_mem.src,
 102                       sizeof(intra_pred_test_mem.src), elapsed_time, k);
 103   }
 104 }
 105
 106 void TestIntraPred4(VpxPredFunc const *pred_funcs) {
 107   static const char *const kSignatures[kNumVp9IntraPredFuncs] = {
 108     "e7ed7353c3383fff942e500e9bfe82fe", "2a4a26fcc6ce005eadc08354d196c8a9",
 109     "269d92eff86f315d9c38fe7640d85b15", "ae2960eea9f71ee3dabe08b282ec1773",
 110     "6c1abcc44e90148998b51acd11144e9c", "f7bb3186e1ef8a2b326037ff898cad8e",
 111     "364c1f3fb2f445f935aec2a70a67eaa4", "141624072a4a56773f68fadbdd07c4a7",
 112     "7be49b08687a5f24df3a2c612fca3876", "459bb5d9fd5b238348179c9a22108cd6",
 113     "73edb8831bf1bdfce21ae8eaa43b1234", "2e2457f2009c701a355a8b25eb74fcda",
 114     "52ae4e8bdbe41494c1f43051d4dd7f0b"
 115   };
 116   TestIntraPred("Intra4", pred_funcs, kSignatures, 4);
 117 }
 118
 119 void TestIntraPred8(VpxPredFunc const *pred_funcs) {
 120   static const char *const kSignatures[kNumVp9IntraPredFuncs] = {
 121     "d8bbae5d6547cfc17e4f5f44c8730e88", "373bab6d931868d41a601d9d88ce9ac3",
 122     "6fdd5ff4ff79656c14747598ca9e3706", "d9661c2811d6a73674f40ffb2b841847",
 123     "7c722d10b19ccff0b8c171868e747385", "f81dd986eb2b50f750d3a7da716b7e27",
 124     "d500f2c8fc78f46a4c74e4dcf51f14fb", "0e3523f9cab2142dd37fd07ec0760bce",
 125     "79ac4efe907f0a0f1885d43066cfedee", "19ecf2432ac305057de3b6578474eec6",
 126     "4f985b61acc6dd5d2d2585fa89ea2e2d", "f1bb25a9060dd262f405f15a38f5f674",
 127     "209ea00801584829e9a0f7be7d4a74ba"
 128   };
 129   TestIntraPred("Intra8", pred_funcs, kSignatures, 8);
 130 }
 131
 132 void TestIntraPred16(VpxPredFunc const *pred_funcs) {
 133   static const char *const kSignatures[kNumVp9IntraPredFuncs] = {
 134     "50971c07ce26977d30298538fffec619", "527a6b9e0dc5b21b98cf276305432bef",
 135     "7eff2868f80ebc2c43a4f367281d80f7", "67cd60512b54964ef6aff1bd4816d922",
 136     "48371c87dc95c08a33b2048f89cf6468", "b0acf2872ee411d7530af6d2625a7084",
 137     "f32aafed4d8d3776ed58bcb6188756d5", "dae208f3dca583529cff49b73f7c4183",
 138     "7af66a2f4c8e0b4908e40f047e60c47c", "125e3ab6ab9bc961f183ec366a7afa88",
 139     "6b90f25b23983c35386b9fd704427622", "f8d6b11d710edc136a7c62c917435f93",
 140     "ed308f18614a362917f411c218aee532"
 141   };
 142   TestIntraPred("Intra16", pred_funcs, kSignatures, 16);
 143 }
 144
 145 void TestIntraPred32(VpxPredFunc const *pred_funcs) {
 146   static const char *const kSignatures[kNumVp9IntraPredFuncs] = {
 147     "a0a618c900e65ae521ccc8af789729f2", "985aaa7c72b4a6c2fb431d32100cf13a",
 148     "10662d09febc3ca13ee4e700120daeb5", "b3b01379ba08916ef6b1b35f7d9ad51c",
 149     "9f4261755795af97e34679c333ec7004", "bc2c9da91ad97ef0d1610fb0a9041657",
 150     "75c79b1362ad18abfcdb1aa0aacfc21d", "4039bb7da0f6860090d3c57b5c85468f",
 151     "b29fff7b61804e68383e3a609b33da58", "e1aa5e49067fd8dba66c2eb8d07b7a89",
 152     "4e042822909c1c06d3b10a88281df1eb", "72eb9d9e0e67c93f4c66b70348e9fef7",
 153     "a22d102bcb51ca798aac12ca4ae8f2e8"
 154   };
 155   TestIntraPred("Intra32", pred_funcs, kSignatures, 32);
 156 }
 157
 158 }  // namespace
 159
 160 // Defines a test case for |arch| (e.g., C, SSE2, ...) passing the predictors
 161 // to |test_func|. The test name is 'arch.test_func', e.g., C.TestIntraPred4.
 162 #define INTRA_PRED_TEST(arch, test_func, dc, dc_left, dc_top, dc_128, v, h,   \
 163                         d45, d135, d117, d153, d207, d63, tm)                 \
 164   TEST(arch, test_func) {                                                     \
 165     static const VpxPredFunc vpx_intra_pred[] = {                             \
 166       dc, dc_left, dc_top, dc_128, v, h, d45, d135, d117, d153, d207, d63, tm \
 167     };                                                                        \
 168     test_func(vpx_intra_pred);                                                \
 169   }
 170
 171 // -----------------------------------------------------------------------------
 172
 173 INTRA_PRED_TEST(C, TestIntraPred4, vpx_dc_predictor_4x4_c,
 174                 vpx_dc_left_predictor_4x4_c, vpx_dc_top_predictor_4x4_c,
 175                 vpx_dc_128_predictor_4x4_c, vpx_v_predictor_4x4_c,
 176                 vpx_h_predictor_4x4_c, vpx_d45_predictor_4x4_c,
 177                 vpx_d135_predictor_4x4_c, vpx_d117_predictor_4x4_c,
 178                 vpx_d153_predictor_4x4_c, vpx_d207_predictor_4x4_c,
 179                 vpx_d63_predictor_4x4_c, vpx_tm_predictor_4x4_c)
 180
 181 INTRA_PRED_TEST(C, TestIntraPred8, vpx_dc_predictor_8x8_c,
 182                 vpx_dc_left_predictor_8x8_c, vpx_dc_top_predictor_8x8_c,
 183                 vpx_dc_128_predictor_8x8_c, vpx_v_predictor_8x8_c,
 184                 vpx_h_predictor_8x8_c, vpx_d45_predictor_8x8_c,
 185                 vpx_d135_predictor_8x8_c, vpx_d117_predictor_8x8_c,
 186                 vpx_d153_predictor_8x8_c, vpx_d207_predictor_8x8_c,
 187                 vpx_d63_predictor_8x8_c, vpx_tm_predictor_8x8_c)
 188
 189 INTRA_PRED_TEST(C, TestIntraPred16, vpx_dc_predictor_16x16_c,
 190                 vpx_dc_left_predictor_16x16_c, vpx_dc_top_predictor_16x16_c,
 191                 vpx_dc_128_predictor_16x16_c, vpx_v_predictor_16x16_c,
 192                 vpx_h_predictor_16x16_c, vpx_d45_predictor_16x16_c,
 193                 vpx_d135_predictor_16x16_c, vpx_d117_predictor_16x16_c,
 194                 vpx_d153_predictor_16x16_c, vpx_d207_predictor_16x16_c,
 195                 vpx_d63_predictor_16x16_c, vpx_tm_predictor_16x16_c)
 196
 197 INTRA_PRED_TEST(C, TestIntraPred32, vpx_dc_predictor_32x32_c,
 198                 vpx_dc_left_predictor_32x32_c, vpx_dc_top_predictor_32x32_c,
 199                 vpx_dc_128_predictor_32x32_c, vpx_v_predictor_32x32_c,
 200                 vpx_h_predictor_32x32_c, vpx_d45_predictor_32x32_c,
 201                 vpx_d135_predictor_32x32_c, vpx_d117_predictor_32x32_c,
 202                 vpx_d153_predictor_32x32_c, vpx_d207_predictor_32x32_c,
 203                 vpx_d63_predictor_32x32_c, vpx_tm_predictor_32x32_c)
 204
 205 #if HAVE_SSE2
 206 INTRA_PRED_TEST(SSE2, TestIntraPred4, vpx_dc_predictor_4x4_sse2,
 207                 vpx_dc_left_predictor_4x4_sse2, vpx_dc_top_predictor_4x4_sse2,
 208                 vpx_dc_128_predictor_4x4_sse2, vpx_v_predictor_4x4_sse2,
 209                 vpx_h_predictor_4x4_sse2, vpx_d45_predictor_4x4_sse2, NULL,
 210                 NULL, NULL, vpx_d207_predictor_4x4_sse2, NULL,
 211                 vpx_tm_predictor_4x4_sse2)
 212
 213 INTRA_PRED_TEST(SSE2, TestIntraPred8, vpx_dc_predictor_8x8_sse2,
 214                 vpx_dc_left_predictor_8x8_sse2, vpx_dc_top_predictor_8x8_sse2,
 215                 vpx_dc_128_predictor_8x8_sse2, vpx_v_predictor_8x8_sse2,
 216                 vpx_h_predictor_8x8_sse2, vpx_d45_predictor_8x8_sse2, NULL,
 217                 NULL, NULL, NULL, NULL, vpx_tm_predictor_8x8_sse2)
 218
 219 INTRA_PRED_TEST(SSE2, TestIntraPred16, vpx_dc_predictor_16x16_sse2,
 220                 vpx_dc_left_predictor_16x16_sse2,
 221                 vpx_dc_top_predictor_16x16_sse2,
 222                 vpx_dc_128_predictor_16x16_sse2, vpx_v_predictor_16x16_sse2,
 223                 vpx_h_predictor_16x16_sse2, NULL, NULL, NULL, NULL, NULL, NULL,
 224                 vpx_tm_predictor_16x16_sse2)
 225
 226 INTRA_PRED_TEST(SSE2, TestIntraPred32, vpx_dc_predictor_32x32_sse2,
 227                 vpx_dc_left_predictor_32x32_sse2,
 228                 vpx_dc_top_predictor_32x32_sse2,
 229                 vpx_dc_128_predictor_32x32_sse2, vpx_v_predictor_32x32_sse2,
 230                 vpx_h_predictor_32x32_sse2, NULL, NULL, NULL, NULL, NULL, NULL,
 231                 vpx_tm_predictor_32x32_sse2)
 232 #endif  // HAVE_SSE2
 233
 234 #if HAVE_SSSE3
 235 INTRA_PRED_TEST(SSSE3, TestIntraPred4, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
 236                 NULL, NULL, vpx_d153_predictor_4x4_ssse3, NULL,
 237                 vpx_d63_predictor_4x4_ssse3, NULL)
 238 INTRA_PRED_TEST(SSSE3, TestIntraPred8, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
 239                 NULL, NULL, vpx_d153_predictor_8x8_ssse3,
 240                 vpx_d207_predictor_8x8_ssse3, vpx_d63_predictor_8x8_ssse3, NULL)
 241 INTRA_PRED_TEST(SSSE3, TestIntraPred16, NULL, NULL, NULL, NULL, NULL, NULL,
 242                 vpx_d45_predictor_16x16_ssse3, NULL, NULL,
 243                 vpx_d153_predictor_16x16_ssse3, vpx_d207_predictor_16x16_ssse3,
 244                 vpx_d63_predictor_16x16_ssse3, NULL)
 245 INTRA_PRED_TEST(SSSE3, TestIntraPred32, NULL, NULL, NULL, NULL, NULL, NULL,
 246                 vpx_d45_predictor_32x32_ssse3, NULL, NULL,
 247                 vpx_d153_predictor_32x32_ssse3, vpx_d207_predictor_32x32_ssse3,
 248                 vpx_d63_predictor_32x32_ssse3, NULL)
 249 #endif  // HAVE_SSSE3
 250
 251 #if HAVE_DSPR2
 252 INTRA_PRED_TEST(DSPR2, TestIntraPred4, vpx_dc_predictor_4x4_dspr2, NULL, NULL,
 253                 NULL, NULL, vpx_h_predictor_4x4_dspr2, NULL, NULL, NULL, NULL,
 254                 NULL, NULL, vpx_tm_predictor_4x4_dspr2)
 255 INTRA_PRED_TEST(DSPR2, TestIntraPred8, vpx_dc_predictor_8x8_dspr2, NULL, NULL,
 256                 NULL, NULL, vpx_h_predictor_8x8_dspr2, NULL, NULL, NULL, NULL,
 257                 NULL, NULL, vpx_tm_predictor_8x8_c)
 258 INTRA_PRED_TEST(DSPR2, TestIntraPred16, vpx_dc_predictor_16x16_dspr2, NULL,
 259                 NULL, NULL, NULL, vpx_h_predictor_16x16_dspr2, NULL, NULL, NULL,
 260                 NULL, NULL, NULL, NULL)
 261 #endif  // HAVE_DSPR2
 262
 263 #if HAVE_NEON
 264 INTRA_PRED_TEST(NEON, TestIntraPred4, vpx_dc_predictor_4x4_neon,
 265                 vpx_dc_left_predictor_4x4_neon, vpx_dc_top_predictor_4x4_neon,
 266                 vpx_dc_128_predictor_4x4_neon, vpx_v_predictor_4x4_neon,
 267                 vpx_h_predictor_4x4_neon, vpx_d45_predictor_4x4_neon,
 268                 vpx_d135_predictor_4x4_neon, NULL, NULL, NULL, NULL,
 269                 vpx_tm_predictor_4x4_neon)
 270 INTRA_PRED_TEST(NEON, TestIntraPred8, vpx_dc_predictor_8x8_neon,
 271                 vpx_dc_left_predictor_8x8_neon, vpx_dc_top_predictor_8x8_neon,
 272                 vpx_dc_128_predictor_8x8_neon, vpx_v_predictor_8x8_neon,
 273                 vpx_h_predictor_8x8_neon, vpx_d45_predictor_8x8_neon,
 274                 vpx_d135_predictor_8x8_neon, NULL, NULL, NULL, NULL,
 275                 vpx_tm_predictor_8x8_neon)
 276 INTRA_PRED_TEST(NEON, TestIntraPred16, vpx_dc_predictor_16x16_neon,
 277                 vpx_dc_left_predictor_16x16_neon,
 278                 vpx_dc_top_predictor_16x16_neon,
 279                 vpx_dc_128_predictor_16x16_neon, vpx_v_predictor_16x16_neon,
 280                 vpx_h_predictor_16x16_neon, vpx_d45_predictor_16x16_neon,
 281                 vpx_d135_predictor_16x16_neon, NULL, NULL, NULL, NULL,
 282                 vpx_tm_predictor_16x16_neon)
 283 INTRA_PRED_TEST(NEON, TestIntraPred32, vpx_dc_predictor_32x32_neon,
 284                 vpx_dc_left_predictor_32x32_neon,
 285                 vpx_dc_top_predictor_32x32_neon,
 286                 vpx_dc_128_predictor_32x32_neon, vpx_v_predictor_32x32_neon,
 287                 vpx_h_predictor_32x32_neon, vpx_d45_predictor_32x32_neon,
 288                 vpx_d135_predictor_32x32_neon, NULL, NULL, NULL, NULL,
 289                 vpx_tm_predictor_32x32_neon)
 290 #endif  // HAVE_NEON
 291
 292 #if HAVE_MSA
 293 INTRA_PRED_TEST(MSA, TestIntraPred4, vpx_dc_predictor_4x4_msa,
 294                 vpx_dc_left_predictor_4x4_msa, vpx_dc_top_predictor_4x4_msa,
 295                 vpx_dc_128_predictor_4x4_msa, vpx_v_predictor_4x4_msa,
 296                 vpx_h_predictor_4x4_msa, NULL, NULL, NULL, NULL, NULL, NULL,
 297                 vpx_tm_predictor_4x4_msa)
 298 INTRA_PRED_TEST(MSA, TestIntraPred8, vpx_dc_predictor_8x8_msa,
 299                 vpx_dc_left_predictor_8x8_msa, vpx_dc_top_predictor_8x8_msa,
 300                 vpx_dc_128_predictor_8x8_msa, vpx_v_predictor_8x8_msa,
 301                 vpx_h_predictor_8x8_msa, NULL, NULL, NULL, NULL, NULL, NULL,
 302                 vpx_tm_predictor_8x8_msa)
 303 INTRA_PRED_TEST(MSA, TestIntraPred16, vpx_dc_predictor_16x16_msa,
 304                 vpx_dc_left_predictor_16x16_msa, vpx_dc_top_predictor_16x16_msa,
 305                 vpx_dc_128_predictor_16x16_msa, vpx_v_predictor_16x16_msa,
 306                 vpx_h_predictor_16x16_msa, NULL, NULL, NULL, NULL, NULL, NULL,
 307                 vpx_tm_predictor_16x16_msa)
 308 INTRA_PRED_TEST(MSA, TestIntraPred32, vpx_dc_predictor_32x32_msa,
 309                 vpx_dc_left_predictor_32x32_msa, vpx_dc_top_predictor_32x32_msa,
 310                 vpx_dc_128_predictor_32x32_msa, vpx_v_predictor_32x32_msa,
 311                 vpx_h_predictor_32x32_msa, NULL, NULL, NULL, NULL, NULL, NULL,
 312                 vpx_tm_predictor_32x32_msa)
 313 #endif  // HAVE_MSA
 314
 315 #if HAVE_VSX
 316 INTRA_PRED_TEST(VSX, TestIntraPred4, NULL, NULL, NULL, NULL, NULL,
 317                 vpx_h_predictor_4x4_vsx, NULL, NULL, NULL, NULL, NULL, NULL,
 318                 vpx_tm_predictor_4x4_vsx)
 319
 320 INTRA_PRED_TEST(VSX, TestIntraPred8, vpx_dc_predictor_8x8_vsx, NULL, NULL, NULL,
 321                 NULL, vpx_h_predictor_8x8_vsx, vpx_d45_predictor_8x8_vsx, NULL,
 322                 NULL, NULL, NULL, vpx_d63_predictor_8x8_vsx,
 323                 vpx_tm_predictor_8x8_vsx)
 324
 325 INTRA_PRED_TEST(VSX, TestIntraPred16, vpx_dc_predictor_16x16_vsx,
 326                 vpx_dc_left_predictor_16x16_vsx, vpx_dc_top_predictor_16x16_vsx,
 327                 vpx_dc_128_predictor_16x16_vsx, vpx_v_predictor_16x16_vsx,
 328                 vpx_h_predictor_16x16_vsx, vpx_d45_predictor_16x16_vsx, NULL,
 329                 NULL, NULL, NULL, vpx_d63_predictor_16x16_vsx,
 330                 vpx_tm_predictor_16x16_vsx)
 331
 332 INTRA_PRED_TEST(VSX, TestIntraPred32, vpx_dc_predictor_32x32_vsx,
 333                 vpx_dc_left_predictor_32x32_vsx, vpx_dc_top_predictor_32x32_vsx,
 334                 vpx_dc_128_predictor_32x32_vsx, vpx_v_predictor_32x32_vsx,
 335                 vpx_h_predictor_32x32_vsx, vpx_d45_predictor_32x32_vsx, NULL,
 336                 NULL, NULL, NULL, vpx_d63_predictor_32x32_vsx,
 337                 vpx_tm_predictor_32x32_vsx)
 338 #endif  // HAVE_VSX
 339
 340 // -----------------------------------------------------------------------------
 341
 342 #if CONFIG_VP9_HIGHBITDEPTH
 343 namespace {
 344
 345 typedef void (*VpxHighbdPredFunc)(uint16_t *dst, ptrdiff_t y_stride,
 346                                   const uint16_t *above, const uint16_t *left,
 347                                   int bd);
 348
 349 typedef IntraPredTestMem<uint16_t> Vp9HighbdIntraPredTestMem;
 350
 351 void TestHighbdIntraPred(const char name[], VpxHighbdPredFunc const *pred_funcs,
 352                          const char *const signatures[], int block_size) {
 353   const int kNumTests = static_cast<int>(
 354       2.e10 / (block_size * block_size * kNumVp9IntraPredFuncs));
 355   Vp9HighbdIntraPredTestMem intra_pred_test_mem;
 356   const uint16_t *const above = intra_pred_test_mem.above_mem + 16;
 357
 358   intra_pred_test_mem.Init(block_size, 12);
 359
 360   for (int k = 0; k < kNumVp9IntraPredFuncs; ++k) {
 361     if (pred_funcs[k] == NULL) continue;
 362     memcpy(intra_pred_test_mem.src, intra_pred_test_mem.ref_src,
 363            sizeof(intra_pred_test_mem.src));
 364     vpx_usec_timer timer;
 365     vpx_usec_timer_start(&timer);
 366     for (int num_tests = 0; num_tests < kNumTests; ++num_tests) {
 367       pred_funcs[k](intra_pred_test_mem.src, kBPS, above,
 368                     intra_pred_test_mem.left, 12);
 369     }
 370     libvpx_test::ClearSystemState();
 371     vpx_usec_timer_mark(&timer);
 372     const int elapsed_time =
 373         static_cast<int>(vpx_usec_timer_elapsed(&timer) / 1000);
 374     CheckMd5Signature(name, signatures, intra_pred_test_mem.src,
 375                       sizeof(intra_pred_test_mem.src), elapsed_time, k);
 376   }
 377 }
 378
 379 void TestHighbdIntraPred4(VpxHighbdPredFunc const *pred_funcs) {
 380   static const char *const kSignatures[kNumVp9IntraPredFuncs] = {
 381     "11f74af6c5737df472f3275cbde062fa", "51bea056b6447c93f6eb8f6b7e8f6f71",
 382     "27e97f946766331795886f4de04c5594", "53ab15974b049111fb596c5168ec7e3f",
 383     "f0b640bb176fbe4584cf3d32a9b0320a", "729783ca909e03afd4b47111c80d967b",
 384     "fbf1c30793d9f32812e4d9f905d53530", "293fc903254a33754133314c6cdba81f",
 385     "f8074d704233e73dfd35b458c6092374", "aa6363d08544a1ec4da33d7a0be5640d",
 386     "462abcfdfa3d087bb33c9a88f2aec491", "863eab65d22550dd44a2397277c1ec71",
 387     "23d61df1574d0fa308f9731811047c4b"
 388   };
 389   TestHighbdIntraPred("Intra4", pred_funcs, kSignatures, 4);
 390 }
 391
 392 void TestHighbdIntraPred8(VpxHighbdPredFunc const *pred_funcs) {
 393   static const char *const kSignatures[kNumVp9IntraPredFuncs] = {
 394     "03da8829fe94663047fd108c5fcaa71d", "ecdb37b8120a2d3a4c706b016bd1bfd7",
 395     "1d4543ed8d2b9368cb96898095fe8a75", "f791c9a67b913cbd82d9da8ecede30e2",
 396     "065c70646f4dbaff913282f55a45a441", "51f87123616662ef7c35691497dfd0ba",
 397     "2a5b0131ef4716f098ee65e6df01e3dd", "9ffe186a6bc7db95275f1bbddd6f7aba",
 398     "a3258a2eae2e2bd55cb8f71351b22998", "8d909f0a2066e39b3216092c6289ece4",
 399     "d183abb30b9f24c886a0517e991b22c7", "702a42fe4c7d665dc561b2aeeb60f311",
 400     "7b5dbbbe7ae3a4ac2948731600bde5d6"
 401   };
 402   TestHighbdIntraPred("Intra8", pred_funcs, kSignatures, 8);
 403 }
 404
 405 void TestHighbdIntraPred16(VpxHighbdPredFunc const *pred_funcs) {
 406   static const char *const kSignatures[kNumVp9IntraPredFuncs] = {
 407     "e33cb3f56a878e2fddb1b2fc51cdd275", "c7bff6f04b6052c8ab335d726dbbd52d",
 408     "d0b0b47b654a9bcc5c6008110a44589b", "78f5da7b10b2b9ab39f114a33b6254e9",
 409     "c78e31d23831abb40d6271a318fdd6f3", "90d1347f4ec9198a0320daecb6ff90b8",
 410     "d2c623746cbb64a0c9e29c10f2c57041", "cf28bd387b81ad3e5f1a1c779a4b70a0",
 411     "24c304330431ddeaf630f6ce94af2eac", "91a329798036bf64e8e00a87b131b8b1",
 412     "d39111f22885307f920796a42084c872", "e2e702f7250ece98dd8f3f2854c31eeb",
 413     "e2fb05b01eb8b88549e85641d8ce5b59"
 414   };
 415   TestHighbdIntraPred("Intra16", pred_funcs, kSignatures, 16);
 416 }
 417
 418 void TestHighbdIntraPred32(VpxHighbdPredFunc const *pred_funcs) {
 419   static const char *const kSignatures[kNumVp9IntraPredFuncs] = {
 420     "a3e8056ba7e36628cce4917cd956fedd", "cc7d3024fe8748b512407edee045377e",
 421     "2aab0a0f330a1d3e19b8ecb8f06387a3", "a547bc3fb7b06910bf3973122a426661",
 422     "26f712514da95042f93d6e8dc8e431dc", "bb08c6e16177081daa3d936538dbc2e3",
 423     "8f031af3e2650e89620d8d2c3a843d8b", "42867c8553285e94ee8e4df7abafbda8",
 424     "6496bdee96100667833f546e1be3d640", "2ebfa25bf981377e682e580208504300",
 425     "3e8ae52fd1f607f348aa4cb436c71ab7", "3d4efe797ca82193613696753ea624c4",
 426     "cb8aab6d372278f3131e8d99efde02d9"
 427   };
 428   TestHighbdIntraPred("Intra32", pred_funcs, kSignatures, 32);
 429 }
 430
 431 }  // namespace
 432
 433 // Defines a test case for |arch| (e.g., C, SSE2, ...) passing the predictors
 434 // to |test_func|. The test name is 'arch.test_func', e.g., C.TestIntraPred4.
 435 #define HIGHBD_INTRA_PRED_TEST(arch, test_func, dc, dc_left, dc_top, dc_128,  \
 436                                v, h, d45, d135, d117, d153, d207, d63, tm)    \
 437   TEST(arch, test_func) {                                                     \
 438     static const VpxHighbdPredFunc vpx_intra_pred[] = {                       \
 439       dc, dc_left, dc_top, dc_128, v, h, d45, d135, d117, d153, d207, d63, tm \
 440     };                                                                        \
 441     test_func(vpx_intra_pred);                                                \
 442   }
 443
 444 // -----------------------------------------------------------------------------
 445
 446 HIGHBD_INTRA_PRED_TEST(
 447     C, TestHighbdIntraPred4, vpx_highbd_dc_predictor_4x4_c,
 448     vpx_highbd_dc_left_predictor_4x4_c, vpx_highbd_dc_top_predictor_4x4_c,
 449     vpx_highbd_dc_128_predictor_4x4_c, vpx_highbd_v_predictor_4x4_c,
 450     vpx_highbd_h_predictor_4x4_c, vpx_highbd_d45_predictor_4x4_c,
 451     vpx_highbd_d135_predictor_4x4_c, vpx_highbd_d117_predictor_4x4_c,
 452     vpx_highbd_d153_predictor_4x4_c, vpx_highbd_d207_predictor_4x4_c,
 453     vpx_highbd_d63_predictor_4x4_c, vpx_highbd_tm_predictor_4x4_c)
 454
 455 HIGHBD_INTRA_PRED_TEST(
 456     C, TestHighbdIntraPred8, vpx_highbd_dc_predictor_8x8_c,
 457     vpx_highbd_dc_left_predictor_8x8_c, vpx_highbd_dc_top_predictor_8x8_c,
 458     vpx_highbd_dc_128_predictor_8x8_c, vpx_highbd_v_predictor_8x8_c,
 459     vpx_highbd_h_predictor_8x8_c, vpx_highbd_d45_predictor_8x8_c,
 460     vpx_highbd_d135_predictor_8x8_c, vpx_highbd_d117_predictor_8x8_c,
 461     vpx_highbd_d153_predictor_8x8_c, vpx_highbd_d207_predictor_8x8_c,
 462     vpx_highbd_d63_predictor_8x8_c, vpx_highbd_tm_predictor_8x8_c)
 463
 464 HIGHBD_INTRA_PRED_TEST(
 465     C, TestHighbdIntraPred16, vpx_highbd_dc_predictor_16x16_c,
 466     vpx_highbd_dc_left_predictor_16x16_c, vpx_highbd_dc_top_predictor_16x16_c,
 467     vpx_highbd_dc_128_predictor_16x16_c, vpx_highbd_v_predictor_16x16_c,
 468     vpx_highbd_h_predictor_16x16_c, vpx_highbd_d45_predictor_16x16_c,
 469     vpx_highbd_d135_predictor_16x16_c, vpx_highbd_d117_predictor_16x16_c,
 470     vpx_highbd_d153_predictor_16x16_c, vpx_highbd_d207_predictor_16x16_c,
 471     vpx_highbd_d63_predictor_16x16_c, vpx_highbd_tm_predictor_16x16_c)
 472
 473 HIGHBD_INTRA_PRED_TEST(
 474     C, TestHighbdIntraPred32, vpx_highbd_dc_predictor_32x32_c,
 475     vpx_highbd_dc_left_predictor_32x32_c, vpx_highbd_dc_top_predictor_32x32_c,
 476     vpx_highbd_dc_128_predictor_32x32_c, vpx_highbd_v_predictor_32x32_c,
 477     vpx_highbd_h_predictor_32x32_c, vpx_highbd_d45_predictor_32x32_c,
 478     vpx_highbd_d135_predictor_32x32_c, vpx_highbd_d117_predictor_32x32_c,
 479     vpx_highbd_d153_predictor_32x32_c, vpx_highbd_d207_predictor_32x32_c,
 480     vpx_highbd_d63_predictor_32x32_c, vpx_highbd_tm_predictor_32x32_c)
 481
 482 #if HAVE_SSE2
 483 HIGHBD_INTRA_PRED_TEST(SSE2, TestHighbdIntraPred4,
 484                        vpx_highbd_dc_predictor_4x4_sse2, NULL, NULL, NULL,
 485                        vpx_highbd_v_predictor_4x4_sse2, NULL, NULL, NULL, NULL,
 486                        NULL, NULL, NULL, vpx_highbd_tm_predictor_4x4_c)
 487
 488 HIGHBD_INTRA_PRED_TEST(SSE2, TestHighbdIntraPred8,
 489                        vpx_highbd_dc_predictor_8x8_sse2, NULL, NULL, NULL,
 490                        vpx_highbd_v_predictor_8x8_sse2, NULL, NULL, NULL, NULL,
 491                        NULL, NULL, NULL, vpx_highbd_tm_predictor_8x8_sse2)
 492
 493 HIGHBD_INTRA_PRED_TEST(SSE2, TestHighbdIntraPred16,
 494                        vpx_highbd_dc_predictor_16x16_sse2, NULL, NULL, NULL,
 495                        vpx_highbd_v_predictor_16x16_sse2, NULL, NULL, NULL,
 496                        NULL, NULL, NULL, NULL,
 497                        vpx_highbd_tm_predictor_16x16_sse2)
 498
 499 HIGHBD_INTRA_PRED_TEST(SSE2, TestHighbdIntraPred32,
 500                        vpx_highbd_dc_predictor_32x32_sse2, NULL, NULL, NULL,
 501                        vpx_highbd_v_predictor_32x32_sse2, NULL, NULL, NULL,
 502                        NULL, NULL, NULL, NULL,
 503                        vpx_highbd_tm_predictor_32x32_sse2)
 504 #endif  // HAVE_SSE2
 505
 506 #if HAVE_NEON
 507 HIGHBD_INTRA_PRED_TEST(
 508     NEON, TestHighbdIntraPred4, vpx_highbd_dc_predictor_4x4_neon,
 509     vpx_highbd_dc_left_predictor_4x4_neon, vpx_highbd_dc_top_predictor_4x4_neon,
 510     vpx_highbd_dc_128_predictor_4x4_neon, vpx_highbd_v_predictor_4x4_neon,
 511     vpx_highbd_h_predictor_4x4_neon, vpx_highbd_d45_predictor_4x4_neon,
 512     vpx_highbd_d135_predictor_4x4_neon, NULL, NULL, NULL, NULL,
 513     vpx_highbd_tm_predictor_4x4_neon)
 514 HIGHBD_INTRA_PRED_TEST(
 515     NEON, TestHighbdIntraPred8, vpx_highbd_dc_predictor_8x8_neon,
 516     vpx_highbd_dc_left_predictor_8x8_neon, vpx_highbd_dc_top_predictor_8x8_neon,
 517     vpx_highbd_dc_128_predictor_8x8_neon, vpx_highbd_v_predictor_8x8_neon,
 518     vpx_highbd_h_predictor_8x8_neon, vpx_highbd_d45_predictor_8x8_neon,
 519     vpx_highbd_d135_predictor_8x8_neon, NULL, NULL, NULL, NULL,
 520     vpx_highbd_tm_predictor_8x8_neon)
 521 HIGHBD_INTRA_PRED_TEST(NEON, TestHighbdIntraPred16,
 522                        vpx_highbd_dc_predictor_16x16_neon,
 523                        vpx_highbd_dc_left_predictor_16x16_neon,
 524                        vpx_highbd_dc_top_predictor_16x16_neon,
 525                        vpx_highbd_dc_128_predictor_16x16_neon,
 526                        vpx_highbd_v_predictor_16x16_neon,
 527                        vpx_highbd_h_predictor_16x16_neon,
 528                        vpx_highbd_d45_predictor_16x16_neon,
 529                        vpx_highbd_d135_predictor_16x16_neon, NULL, NULL, NULL,
 530                        NULL, vpx_highbd_tm_predictor_16x16_neon)
 531 HIGHBD_INTRA_PRED_TEST(NEON, TestHighbdIntraPred32,
 532                        vpx_highbd_dc_predictor_32x32_neon,
 533                        vpx_highbd_dc_left_predictor_32x32_neon,
 534                        vpx_highbd_dc_top_predictor_32x32_neon,
 535                        vpx_highbd_dc_128_predictor_32x32_neon,
 536                        vpx_highbd_v_predictor_32x32_neon,
 537                        vpx_highbd_h_predictor_32x32_neon,
 538                        vpx_highbd_d45_predictor_32x32_neon,
 539                        vpx_highbd_d135_predictor_32x32_neon, NULL, NULL, NULL,
 540                        NULL, vpx_highbd_tm_predictor_32x32_neon)
 541 #endif  // HAVE_NEON
 542
 543 #endif  // CONFIG_VP9_HIGHBITDEPTH
 544
 545 #include "test/test_libvpx.cc"