OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright (c) 2011 The LibYuv 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 "third_party/libyuv/include/libyuv/cpu_id.h" | |
12 | |
13 #ifdef _MSC_VER | |
14 #include <intrin.h> | |
15 #endif | |
16 #ifdef __ANDROID__ | |
17 #include <cpu-features.h> | |
18 #endif | |
19 | |
20 #include "third_party/libyuv/include/libyuv/basic_types.h" // for CPU_X86 | |
21 | |
22 // TODO(fbarchard): Use cpuid.h when gcc 4.4 is used on OSX and Linux. | |
23 #if (defined(__pic__) || defined(__APPLE__)) && defined(__i386__) | |
24 static inline void __cpuid(int cpu_info[4], int info_type) { | |
25 asm volatile ( | |
26 "mov %%ebx, %%edi \n" | |
27 "cpuid \n" | |
28 "xchg %%edi, %%ebx \n" | |
29 : "=a"(cpu_info[0]), "=D"(cpu_info[1]), "=c"(cpu_info[2]), "=d"(cpu_info[3]) | |
30 : "a"(info_type) | |
31 ); | |
32 } | |
33 #elif defined(__i386__) || defined(__x86_64__) | |
34 static inline void __cpuid(int cpu_info[4], int info_type) { | |
35 asm volatile ( | |
36 "cpuid \n" | |
37 : "=a"(cpu_info[0]), "=b"(cpu_info[1]), "=c"(cpu_info[2]), "=d"(cpu_info[3]) | |
38 : "a"(info_type) | |
39 ); | |
40 } | |
41 #endif | |
42 | |
43 #ifdef __cplusplus | |
44 namespace libyuv { | |
45 extern "C" { | |
46 #endif | |
47 | |
48 // CPU detect function for SIMD instruction sets. | |
49 int cpu_info_ = 0; | |
50 | |
51 int InitCpuFlags() { | |
52 #ifdef CPU_X86 | |
53 int cpu_info[4]; | |
54 __cpuid(cpu_info, 1); | |
55 cpu_info_ = (cpu_info[3] & 0x04000000 ? kCpuHasSSE2 : 0) | | |
56 (cpu_info[2] & 0x00000200 ? kCpuHasSSSE3 : 0) | | |
57 kCpuInitialized; | |
58 #elif defined(__ANDROID__) && defined(__ARM_NEON__) | |
59 uint64_t features = android_getCpuFeatures(); | |
60 cpu_info_ = ((features & ANDROID_CPU_ARM_FEATURE_NEON) ? kCpuHasNEON : 0) | | |
61 kCpuInitialized; | |
62 #elif defined(__ARM_NEON__) | |
63 // gcc -mfpu=neon defines __ARM_NEON__ | |
64 // Enable Neon if you want support for Neon and Arm, and use MaskCpuFlags | |
65 // to disable Neon on devices that do not have it. | |
66 cpu_info_ = kCpuHasNEON | kCpuInitialized; | |
67 #else | |
68 cpu_info_ = kCpuInitialized; | |
69 #endif | |
70 return cpu_info_; | |
71 } | |
72 | |
73 void MaskCpuFlags(int enable_flags) { | |
74 InitCpuFlags(); | |
75 cpu_info_ = (cpu_info_ & enable_flags) | kCpuInitialized; | |
76 } | |
77 | |
78 #ifdef __cplusplus | |
79 } // extern "C" | |
80 } // namespace libyuv | |
81 #endif | |
OLD | NEW |