test_libaom.cc 2.55 KB
Newer Older
1
/*
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
3
 *
4 5 6 7 8 9
 * This source code is subject to the terms of the BSD 2 Clause License and
 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
 * was not distributed with this source code in the LICENSE file, you can
 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
 * Media Patent License 1.0 was not distributed with this source code in the
 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
Johann's avatar
Johann committed
10
 */
11

Johann's avatar
Johann committed
12 13
#include <string.h>

14
#include <string>
15

16
#include "third_party/googletest/src/googletest/include/gtest/gtest.h"
17

18
#include "./aom_config.h"
19
#if ARCH_X86 || ARCH_X86_64
20
#include "aom_ports/x86.h"
21
#endif
22
extern "C" {
23 24 25 26 27
#if CONFIG_AV1
extern void av1_rtcd();
#endif  // CONFIG_AV1
extern void aom_dsp_rtcd();
extern void aom_scale_rtcd();
28 29
}

30
#if ARCH_X86 || ARCH_X86_64
31
static void append_negative_gtest_filter(const char *str) {
32
  std::string filter = ::testing::FLAGS_gtest_filter;
33 34
  // Negative patterns begin with one '-' followed by a ':' separated list.
  if (filter.find('-') == std::string::npos) filter += '-';
Johann's avatar
Johann committed
35 36 37 38 39 40 41 42 43 44 45
  // OPT.* matches TEST() functions
  // OPT/* matches TEST_P() functions
  // OPT_* matches tests which have been manually sharded.
  // We do not match OPT* because of SSE/SSE2 collisions.
  const char *search_terminators = "./_";
  for (size_t pos = 0; pos < strlen(search_terminators); ++pos) {
    filter += ":";
    filter += str;
    filter += search_terminators[pos];
    filter += "*";
  }
46 47
  ::testing::FLAGS_gtest_filter = filter;
}
48
#endif  // ARCH_X86 || ARCH_X86_64
49 50 51 52 53

int main(int argc, char **argv) {
  ::testing::InitGoogleTest(&argc, argv);

#if ARCH_X86 || ARCH_X86_64
54
  const int simd_caps = x86_simd_caps();
Johann's avatar
Johann committed
55 56 57 58 59 60 61 62
  if (!(simd_caps & HAS_MMX)) append_negative_gtest_filter("MMX");
  if (!(simd_caps & HAS_SSE)) append_negative_gtest_filter("SSE");
  if (!(simd_caps & HAS_SSE2)) append_negative_gtest_filter("SSE2");
  if (!(simd_caps & HAS_SSE3)) append_negative_gtest_filter("SSE3");
  if (!(simd_caps & HAS_SSSE3)) append_negative_gtest_filter("SSSE3");
  if (!(simd_caps & HAS_SSE4_1)) append_negative_gtest_filter("SSE4_1");
  if (!(simd_caps & HAS_AVX)) append_negative_gtest_filter("AVX");
  if (!(simd_caps & HAS_AVX2)) append_negative_gtest_filter("AVX2");
63
#endif  // ARCH_X86 || ARCH_X86_64
64

Johann's avatar
Johann committed
65 66
// Shared library builds don't support whitebox tests that exercise internal
// symbols.
67
#if !CONFIG_SHARED
68 69 70 71 72
#if CONFIG_AV1
  av1_rtcd();
#endif  // CONFIG_AV1
  aom_dsp_rtcd();
  aom_scale_rtcd();
73
#endif  // !CONFIG_SHARED
74

75 76
  return RUN_ALL_TESTS();
}