Home | History | Annotate | Download | only in libyuv
      1 /*
      2  *  Copyright 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 #ifndef INCLUDE_LIBYUV_CPU_ID_H_  // NOLINT
     12 #define INCLUDE_LIBYUV_CPU_ID_H_
     13 
     14 #include "libyuv/basic_types.h"
     15 
     16 #ifdef __cplusplus
     17 namespace libyuv {
     18 extern "C" {
     19 #endif
     20 
     21 // TODO(fbarchard): Consider overlapping bits for different architectures.
     22 // Internal flag to indicate cpuid requires initialization.
     23 #define kCpuInit 0x1
     24 
     25 // These flags are only valid on ARM processors.
     26 static const int kCpuHasARM = 0x2;
     27 static const int kCpuHasNEON = 0x4;
     28 // 0x8 reserved for future ARM flag.
     29 
     30 // These flags are only valid on x86 processors.
     31 static const int kCpuHasX86 = 0x10;
     32 static const int kCpuHasSSE2 = 0x20;
     33 static const int kCpuHasSSSE3 = 0x40;
     34 static const int kCpuHasSSE41 = 0x80;
     35 static const int kCpuHasSSE42 = 0x100;
     36 static const int kCpuHasAVX = 0x200;
     37 static const int kCpuHasAVX2 = 0x400;
     38 static const int kCpuHasERMS = 0x800;
     39 static const int kCpuHasFMA3 = 0x1000;
     40 // 0x2000, 0x4000, 0x8000 reserved for future X86 flags.
     41 
     42 // These flags are only valid on MIPS processors.
     43 static const int kCpuHasMIPS = 0x10000;
     44 static const int kCpuHasMIPS_DSP = 0x20000;
     45 static const int kCpuHasMIPS_DSPR2 = 0x40000;
     46 
     47 // Internal function used to auto-init.
     48 LIBYUV_API
     49 int InitCpuFlags(void);
     50 
     51 // Internal function for parsing /proc/cpuinfo.
     52 LIBYUV_API
     53 int ArmCpuCaps(const char* cpuinfo_name);
     54 
     55 // Detect CPU has SSE2 etc.
     56 // Test_flag parameter should be one of kCpuHas constants above.
     57 // returns non-zero if instruction set is detected
     58 static __inline int TestCpuFlag(int test_flag) {
     59   LIBYUV_API extern int cpu_info_;
     60   return (cpu_info_ == kCpuInit ? InitCpuFlags() : cpu_info_) & test_flag;
     61 }
     62 
     63 // For testing, allow CPU flags to be disabled.
     64 // ie MaskCpuFlags(~kCpuHasSSSE3) to disable SSSE3.
     65 // MaskCpuFlags(-1) to enable all cpu specific optimizations.
     66 // MaskCpuFlags(0) to disable all cpu specific optimizations.
     67 LIBYUV_API
     68 void MaskCpuFlags(int enable_flags);
     69 
     70 // Low level cpuid for X86. Returns zeros on other CPUs.
     71 // eax is the info type that you want.
     72 // ecx is typically the cpu number, and should normally be zero.
     73 LIBYUV_API
     74 void CpuId(uint32 eax, uint32 ecx, uint32* cpu_info);
     75 
     76 #ifdef __cplusplus
     77 }  // extern "C"
     78 }  // namespace libyuv
     79 #endif
     80 
     81 #endif  // INCLUDE_LIBYUV_CPU_ID_H_  NOLINT
     82