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 kNumVp9IntraPredFuncs = 13;
  33 const char *kVp9IntraPredNames[kNumVp9IntraPredFuncs] = {
  34   "DC_PRED", "DC_LEFT_PRED", "DC_TOP_PRED", "DC_128_PRED", "V_PRED", "H_PRED",
  35   "D45_PRED", "D135_PRED", "D117_PRED", "D153_PRED", "D207_PRED", "D63_PRED",
  36   "TM_PRED"
  37 };
  38
  39 void TestIntraPred(const char name[], VpxPredFunc const *pred_funcs,
  40                    const char *const pred_func_names[], int num_funcs,
  41                    const char *const signatures[], int block_size,
  42                    int num_pixels_per_test) {
  43   libvpx_test::ACMRandom rnd(libvpx_test::ACMRandom::DeterministicSeed());
  44   const int kBPS = 32;
  45   const int kTotalPixels = 32 * kBPS;
  46   DECLARE_ALIGNED(16, uint8_t, src[kTotalPixels]);
  47   DECLARE_ALIGNED(16, uint8_t, ref_src[kTotalPixels]);
  48   DECLARE_ALIGNED(16, uint8_t, left[kBPS]);
  49   DECLARE_ALIGNED(16, uint8_t, above_mem[2 * kBPS + 16]);
  50   uint8_t *const above = above_mem + 16;
  51   for (int i = 0; i < kTotalPixels; ++i) ref_src[i] = rnd.Rand8();
  52   for (int i = 0; i < kBPS; ++i) left[i] = rnd.Rand8();
  53   for (int i = -1; i < kBPS; ++i) above[i] = rnd.Rand8();
  54   const int kNumTests = static_cast<int>(2.e10 / num_pixels_per_test);
  55
  56   // some code assumes the top row has been extended:
  57   // d45/d63 C-code, for instance, but not the assembly.
  58   // TODO(jzern): this style of extension isn't strictly necessary.
  59   ASSERT_LE(block_size, kBPS);
  60   memset(above + block_size, above[block_size - 1], 2 * kBPS - block_size);
  61
  62   for (int k = 0; k < num_funcs; ++k) {
  63     if (pred_funcs[k] == NULL) continue;
  64     memcpy(src, ref_src, sizeof(src));
  65     vpx_usec_timer timer;
  66     vpx_usec_timer_start(&timer);
  67     for (int num_tests = 0; num_tests < kNumTests; ++num_tests) {
  68       pred_funcs[k](src, kBPS, above, left);
  69     }
  70     libvpx_test::ClearSystemState();
  71     vpx_usec_timer_mark(&timer);
  72     const int elapsed_time =
  73         static_cast<int>(vpx_usec_timer_elapsed(&timer) / 1000);
  74     libvpx_test::MD5 md5;
  75     md5.Add(src, sizeof(src));
  76     printf("Mode %s[%12s]: %5d ms     MD5: %s\n", name, pred_func_names[k],
  77            elapsed_time, md5.Get());
  78     EXPECT_STREQ(signatures[k], md5.Get());
  79   }
  80 }
  81
  82 void TestIntraPred4(VpxPredFunc const *pred_funcs) {
  83   static const int kNumVp9IntraFuncs = 13;
  84   static const char *const kSignatures[kNumVp9IntraFuncs] = {
  85     "4334156168b34ab599d9b5b30f522fe9",
  86     "bc4649d5ba47c7ff178d92e475960fb0",
  87     "8d316e5933326dcac24e1064794b5d12",
  88     "a27270fed024eafd762c95de85f4da51",
  89     "c33dff000d4256c2b8f3bf9e9bab14d2",
  90     "44d8cddc2ad8f79b8ed3306051722b4f",
  91     "eb54839b2bad6699d8946f01ec041cd0",
  92     "ecb0d56ae5f677ea45127ce9d5c058e4",
  93     "0b7936841f6813da818275944895b574",
  94     "9117972ef64f91a58ff73e1731c81db2",
  95     "c56d5e8c729e46825f46dd5d3b5d508a",
  96     "c0889e2039bcf7bcb5d2f33cdca69adc",
  97     "309a618577b27c648f9c5ee45252bc8f",
  98   };
  99   TestIntraPred("Intra4", pred_funcs, kVp9IntraPredNames, kNumVp9IntraFuncs,
 100                 kSignatures, 4, 4 * 4 * kNumVp9IntraFuncs);
 101 }
 102
 103 void TestIntraPred8(VpxPredFunc const *pred_funcs) {
 104   static const int kNumVp9IntraFuncs = 13;
 105   static const char *const kSignatures[kNumVp9IntraFuncs] = {
 106     "7694ddeeefed887faf9d339d18850928",
 107     "7d726b1213591b99f736be6dec65065b",
 108     "19c5711281357a485591aaf9c96c0a67",
 109     "ba6b66877a089e71cd938e3b8c40caac",
 110     "802440c93317e0f8ba93fab02ef74265",
 111     "9e09a47a15deb0b9d8372824f9805080",
 112     "b7c2d8c662268c0c427da412d7b0311d",
 113     "78339c1c60bb1d67d248ab8c4da08b7f",
 114     "5c97d70f7d47de1882a6cd86c165c8a9",
 115     "8182bf60688b42205acd95e59e967157",
 116     "08323400005a297f16d7e57e7fe1eaac",
 117     "95f7bfc262329a5849eda66d8f7c68ce",
 118     "815b75c8e0d91cc1ae766dc5d3e445a3",
 119   };
 120   TestIntraPred("Intra8", pred_funcs, kVp9IntraPredNames, kNumVp9IntraFuncs,
 121                 kSignatures, 8, 8 * 8 * kNumVp9IntraFuncs);
 122 }
 123
 124 void TestIntraPred16(VpxPredFunc const *pred_funcs) {
 125   static const int kNumVp9IntraFuncs = 13;
 126   static const char *const kSignatures[kNumVp9IntraFuncs] = {
 127     "b40dbb555d5d16a043dc361e6694fe53",
 128     "fb08118cee3b6405d64c1fd68be878c6",
 129     "6c190f341475c837cc38c2e566b64875",
 130     "db5c34ccbe2c7f595d9b08b0dc2c698c",
 131     "a62cbfd153a1f0b9fed13e62b8408a7a",
 132     "143df5b4c89335e281103f610f5052e4",
 133     "d87feb124107cdf2cfb147655aa0bb3c",
 134     "7841fae7d4d47b519322e6a03eeed9dc",
 135     "f6ebed3f71cbcf8d6d0516ce87e11093",
 136     "3cc480297dbfeed01a1c2d78dd03d0c5",
 137     "b9f69fa6532b372c545397dcb78ef311",
 138     "a8fe1c70432f09d0c20c67bdb6432c4d",
 139     "b8a41aa968ec108af447af4217cba91b",
 140   };
 141   TestIntraPred("Intra16", pred_funcs, kVp9IntraPredNames, kNumVp9IntraFuncs,
 142                 kSignatures, 16, 16 * 16 * kNumVp9IntraFuncs);
 143 }
 144
 145 void TestIntraPred32(VpxPredFunc const *pred_funcs) {
 146   static const int kNumVp9IntraFuncs = 13;
 147   static const char *const kSignatures[kNumVp9IntraFuncs] = {
 148     "558541656d84f9ae7896db655826febe",
 149     "b3587a1f9a01495fa38c8cd3c8e2a1bf",
 150     "4c6501e64f25aacc55a2a16c7e8f0255",
 151     "b3b01379ba08916ef6b1b35f7d9ad51c",
 152     "0f1eb38b6cbddb3d496199ef9f329071",
 153     "911c06efb9ed1c3b4c104b232b55812f",
 154     "9225beb0ddfa7a1d24eaa1be430a6654",
 155     "0a6d584a44f8db9aa7ade2e2fdb9fc9e",
 156     "b01c9076525216925f3456f034fb6eee",
 157     "d267e20ad9e5cd2915d1a47254d3d149",
 158     "ed012a4a5da71f36c2393023184a0e59",
 159     "f162b51ed618d28b936974cff4391da5",
 160     "9e1370c6d42e08d357d9612c93a71cfc",
 161   };
 162   TestIntraPred("Intra32", pred_funcs, kVp9IntraPredNames, kNumVp9IntraFuncs,
 163                 kSignatures, 32, 32 * 32 * kNumVp9IntraFuncs);
 164 }
 165
 166 }  // namespace
 167
 168 // Defines a test case for |arch| (e.g., C, SSE2, ...) passing the predictors
 169 // to |test_func|. The test name is 'arch.test_func', e.g., C.TestIntraPred4.
 170 #define INTRA_PRED_TEST(arch, test_func, dc, dc_left, dc_top, dc_128, v, h, \
 171                         d45, d135, d117, d153, d207, d63, tm)               \
 172   TEST(arch, test_func) {                                                   \
 173     static const VpxPredFunc vpx_intra_pred[] = {                           \
 174         dc,   dc_left, dc_top, dc_128, v,   h, d45,                         \
 175         d135, d117,    d153,   d207,   d63, tm};                            \
 176     test_func(vpx_intra_pred);                                              \
 177   }
 178
 179 // -----------------------------------------------------------------------------
 180 // 4x4
 181
 182 INTRA_PRED_TEST(C, TestIntraPred4, vpx_dc_predictor_4x4_c,
 183                 vpx_dc_left_predictor_4x4_c, vpx_dc_top_predictor_4x4_c,
 184                 vpx_dc_128_predictor_4x4_c, vpx_v_predictor_4x4_c,
 185                 vpx_h_predictor_4x4_c, vpx_d45_predictor_4x4_c,
 186                 vpx_d135_predictor_4x4_c, vpx_d117_predictor_4x4_c,
 187                 vpx_d153_predictor_4x4_c, vpx_d207_predictor_4x4_c,
 188                 vpx_d63_predictor_4x4_c, vpx_tm_predictor_4x4_c)
 189
 190 #if HAVE_SSE2
 191 INTRA_PRED_TEST(SSE2, TestIntraPred4, vpx_dc_predictor_4x4_sse2,
 192                 vpx_dc_left_predictor_4x4_sse2, vpx_dc_top_predictor_4x4_sse2,
 193                 vpx_dc_128_predictor_4x4_sse2, vpx_v_predictor_4x4_sse2,
 194                 vpx_h_predictor_4x4_sse2, vpx_d45_predictor_4x4_sse2, NULL,
 195                 NULL, NULL, vpx_d207_predictor_4x4_sse2, NULL,
 196                 vpx_tm_predictor_4x4_sse2)
 197 #endif  // HAVE_SSE2
 198
 199 #if HAVE_SSSE3
 200 INTRA_PRED_TEST(SSSE3, TestIntraPred4, NULL, NULL, NULL, NULL, NULL,
 201                 NULL, NULL, NULL, NULL,
 202                 vpx_d153_predictor_4x4_ssse3, NULL,
 203                 vpx_d63_predictor_4x4_ssse3, NULL)
 204 #endif  // HAVE_SSSE3
 205
 206 #if HAVE_DSPR2
 207 INTRA_PRED_TEST(DSPR2, TestIntraPred4, vpx_dc_predictor_4x4_dspr2, NULL, NULL,
 208                 NULL, NULL, vpx_h_predictor_4x4_dspr2, NULL, NULL, NULL, NULL,
 209                 NULL, NULL, vpx_tm_predictor_4x4_dspr2)
 210 #endif  // HAVE_DSPR2
 211
 212 #if HAVE_NEON
 213 INTRA_PRED_TEST(NEON, TestIntraPred4, vpx_dc_predictor_4x4_neon,
 214                 vpx_dc_left_predictor_4x4_neon, vpx_dc_top_predictor_4x4_neon,
 215                 vpx_dc_128_predictor_4x4_neon, vpx_v_predictor_4x4_neon,
 216                 vpx_h_predictor_4x4_neon, vpx_d45_predictor_4x4_neon,
 217                 vpx_d135_predictor_4x4_neon, NULL, NULL, NULL, NULL,
 218                 vpx_tm_predictor_4x4_neon)
 219 #endif  // HAVE_NEON
 220
 221 #if HAVE_MSA
 222 INTRA_PRED_TEST(MSA, TestIntraPred4, vpx_dc_predictor_4x4_msa,
 223                 vpx_dc_left_predictor_4x4_msa, vpx_dc_top_predictor_4x4_msa,
 224                 vpx_dc_128_predictor_4x4_msa, vpx_v_predictor_4x4_msa,
 225                 vpx_h_predictor_4x4_msa, NULL, NULL, NULL, NULL, NULL,
 226                 NULL, vpx_tm_predictor_4x4_msa)
 227 #endif  // HAVE_MSA
 228
 229 // -----------------------------------------------------------------------------
 230 // 8x8
 231
 232 INTRA_PRED_TEST(C, TestIntraPred8, vpx_dc_predictor_8x8_c,
 233                 vpx_dc_left_predictor_8x8_c, vpx_dc_top_predictor_8x8_c,
 234                 vpx_dc_128_predictor_8x8_c, vpx_v_predictor_8x8_c,
 235                 vpx_h_predictor_8x8_c, vpx_d45_predictor_8x8_c,
 236                 vpx_d135_predictor_8x8_c, vpx_d117_predictor_8x8_c,
 237                 vpx_d153_predictor_8x8_c, vpx_d207_predictor_8x8_c,
 238                 vpx_d63_predictor_8x8_c, vpx_tm_predictor_8x8_c)
 239
 240 #if HAVE_SSE2
 241 INTRA_PRED_TEST(SSE2, TestIntraPred8, vpx_dc_predictor_8x8_sse2,
 242                 vpx_dc_left_predictor_8x8_sse2, vpx_dc_top_predictor_8x8_sse2,
 243                 vpx_dc_128_predictor_8x8_sse2, vpx_v_predictor_8x8_sse2,
 244                 vpx_h_predictor_8x8_sse2, vpx_d45_predictor_8x8_sse2, NULL,
 245                 NULL, NULL, NULL, NULL, vpx_tm_predictor_8x8_sse2)
 246 #endif  // HAVE_SSE2
 247
 248 #if HAVE_SSSE3
 249 INTRA_PRED_TEST(SSSE3, TestIntraPred8, NULL, NULL, NULL, NULL, NULL,
 250                 NULL, NULL, NULL, NULL,
 251                 vpx_d153_predictor_8x8_ssse3, vpx_d207_predictor_8x8_ssse3,
 252                 vpx_d63_predictor_8x8_ssse3, NULL)
 253 #endif  // HAVE_SSSE3
 254
 255 #if HAVE_DSPR2
 256 INTRA_PRED_TEST(DSPR2, TestIntraPred8, vpx_dc_predictor_8x8_dspr2, NULL, NULL,
 257                 NULL, NULL, vpx_h_predictor_8x8_dspr2, NULL, NULL, NULL, NULL,
 258                 NULL, NULL, vpx_tm_predictor_8x8_c)
 259 #endif  // HAVE_DSPR2
 260
 261 #if HAVE_NEON
 262 INTRA_PRED_TEST(NEON, TestIntraPred8, vpx_dc_predictor_8x8_neon,
 263                 vpx_dc_left_predictor_8x8_neon, vpx_dc_top_predictor_8x8_neon,
 264                 vpx_dc_128_predictor_8x8_neon, vpx_v_predictor_8x8_neon,
 265                 vpx_h_predictor_8x8_neon, vpx_d45_predictor_8x8_neon, NULL,
 266                 NULL, NULL, NULL, NULL, vpx_tm_predictor_8x8_neon)
 267
 268 #endif  // HAVE_NEON
 269
 270 #if HAVE_MSA
 271 INTRA_PRED_TEST(MSA, TestIntraPred8, vpx_dc_predictor_8x8_msa,
 272                 vpx_dc_left_predictor_8x8_msa, vpx_dc_top_predictor_8x8_msa,
 273                 vpx_dc_128_predictor_8x8_msa, vpx_v_predictor_8x8_msa,
 274                 vpx_h_predictor_8x8_msa, NULL, NULL, NULL, NULL, NULL,
 275                 NULL, vpx_tm_predictor_8x8_msa)
 276 #endif  // HAVE_MSA
 277
 278 // -----------------------------------------------------------------------------
 279 // 16x16
 280
 281 INTRA_PRED_TEST(C, TestIntraPred16, vpx_dc_predictor_16x16_c,
 282                 vpx_dc_left_predictor_16x16_c, vpx_dc_top_predictor_16x16_c,
 283                 vpx_dc_128_predictor_16x16_c, vpx_v_predictor_16x16_c,
 284                 vpx_h_predictor_16x16_c, vpx_d45_predictor_16x16_c,
 285                 vpx_d135_predictor_16x16_c, vpx_d117_predictor_16x16_c,
 286                 vpx_d153_predictor_16x16_c, vpx_d207_predictor_16x16_c,
 287                 vpx_d63_predictor_16x16_c, vpx_tm_predictor_16x16_c)
 288
 289 #if HAVE_SSE2
 290 INTRA_PRED_TEST(SSE2, TestIntraPred16, vpx_dc_predictor_16x16_sse2,
 291                 vpx_dc_left_predictor_16x16_sse2,
 292                 vpx_dc_top_predictor_16x16_sse2,
 293                 vpx_dc_128_predictor_16x16_sse2, vpx_v_predictor_16x16_sse2,
 294                 vpx_h_predictor_16x16_sse2, NULL, NULL, NULL, NULL, NULL, NULL,
 295                 vpx_tm_predictor_16x16_sse2)
 296 #endif  // HAVE_SSE2
 297
 298 #if HAVE_SSSE3
 299 INTRA_PRED_TEST(SSSE3, TestIntraPred16, NULL, NULL, NULL, NULL, NULL,
 300                 NULL, vpx_d45_predictor_16x16_ssse3,
 301                 NULL, NULL, vpx_d153_predictor_16x16_ssse3,
 302                 vpx_d207_predictor_16x16_ssse3, vpx_d63_predictor_16x16_ssse3,
 303                 NULL)
 304 #endif  // HAVE_SSSE3
 305
 306 #if HAVE_DSPR2
 307 INTRA_PRED_TEST(DSPR2, TestIntraPred16, vpx_dc_predictor_16x16_dspr2, NULL,
 308                 NULL, NULL, NULL, vpx_h_predictor_16x16_dspr2, NULL, NULL, NULL,
 309                 NULL, NULL, NULL, NULL)
 310 #endif  // HAVE_DSPR2
 311
 312 #if HAVE_NEON
 313 INTRA_PRED_TEST(NEON, TestIntraPred16, vpx_dc_predictor_16x16_neon,
 314                 vpx_dc_left_predictor_16x16_neon,
 315                 vpx_dc_top_predictor_16x16_neon,
 316                 vpx_dc_128_predictor_16x16_neon, vpx_v_predictor_16x16_neon,
 317                 vpx_h_predictor_16x16_neon, vpx_d45_predictor_16x16_neon, NULL,
 318                 NULL, NULL, NULL, NULL, vpx_tm_predictor_16x16_neon)
 319 #endif  // HAVE_NEON
 320
 321 #if HAVE_MSA
 322 INTRA_PRED_TEST(MSA, TestIntraPred16, vpx_dc_predictor_16x16_msa,
 323                 vpx_dc_left_predictor_16x16_msa, vpx_dc_top_predictor_16x16_msa,
 324                 vpx_dc_128_predictor_16x16_msa, vpx_v_predictor_16x16_msa,
 325                 vpx_h_predictor_16x16_msa, NULL, NULL, NULL, NULL, NULL,
 326                 NULL, vpx_tm_predictor_16x16_msa)
 327 #endif  // HAVE_MSA
 328
 329 // -----------------------------------------------------------------------------
 330 // 32x32
 331
 332 INTRA_PRED_TEST(C, TestIntraPred32, vpx_dc_predictor_32x32_c,
 333                 vpx_dc_left_predictor_32x32_c, vpx_dc_top_predictor_32x32_c,
 334                 vpx_dc_128_predictor_32x32_c, vpx_v_predictor_32x32_c,
 335                 vpx_h_predictor_32x32_c, vpx_d45_predictor_32x32_c,
 336                 vpx_d135_predictor_32x32_c, vpx_d117_predictor_32x32_c,
 337                 vpx_d153_predictor_32x32_c, vpx_d207_predictor_32x32_c,
 338                 vpx_d63_predictor_32x32_c, vpx_tm_predictor_32x32_c)
 339
 340 #if HAVE_SSE2
 341 INTRA_PRED_TEST(SSE2, TestIntraPred32, vpx_dc_predictor_32x32_sse2,
 342                 vpx_dc_left_predictor_32x32_sse2,
 343                 vpx_dc_top_predictor_32x32_sse2,
 344                 vpx_dc_128_predictor_32x32_sse2, vpx_v_predictor_32x32_sse2,
 345                 vpx_h_predictor_32x32_sse2, NULL, NULL, NULL, NULL, NULL,
 346                 NULL, vpx_tm_predictor_32x32_sse2)
 347 #endif  // HAVE_SSE2
 348
 349 #if HAVE_SSSE3
 350 INTRA_PRED_TEST(SSSE3, TestIntraPred32, NULL, NULL, NULL, NULL, NULL,
 351                 NULL, vpx_d45_predictor_32x32_ssse3, NULL, NULL,
 352                 vpx_d153_predictor_32x32_ssse3, vpx_d207_predictor_32x32_ssse3,
 353                 vpx_d63_predictor_32x32_ssse3, NULL)
 354 #endif  // HAVE_SSSE3
 355
 356 #if HAVE_NEON
 357 INTRA_PRED_TEST(NEON, TestIntraPred32, vpx_dc_predictor_32x32_neon,
 358                 vpx_dc_left_predictor_32x32_neon,
 359                 vpx_dc_top_predictor_32x32_neon,
 360                 vpx_dc_128_predictor_32x32_neon, vpx_v_predictor_32x32_neon,
 361                 vpx_h_predictor_32x32_neon, NULL, NULL, NULL, NULL, NULL, NULL,
 362                 vpx_tm_predictor_32x32_neon)
 363 #endif  // HAVE_NEON
 364
 365 #if HAVE_MSA
 366 INTRA_PRED_TEST(MSA, TestIntraPred32, vpx_dc_predictor_32x32_msa,
 367                 vpx_dc_left_predictor_32x32_msa, vpx_dc_top_predictor_32x32_msa,
 368                 vpx_dc_128_predictor_32x32_msa, vpx_v_predictor_32x32_msa,
 369                 vpx_h_predictor_32x32_msa, NULL, NULL, NULL, NULL, NULL,
 370                 NULL, vpx_tm_predictor_32x32_msa)
 371 #endif  // HAVE_MSA
 372
 373 #include "test/test_libvpx.cc"