Chris@69: /* Copyright (c) 2014, Cisco Systems, INC
Chris@69:    Written by XiangMingZhu WeiZhou MinPeng YanWang
Chris@69: 
Chris@69:    Redistribution and use in source and binary forms, with or without
Chris@69:    modification, are permitted provided that the following conditions
Chris@69:    are met:
Chris@69: 
Chris@69:    - Redistributions of source code must retain the above copyright
Chris@69:    notice, this list of conditions and the following disclaimer.
Chris@69: 
Chris@69:    - Redistributions in binary form must reproduce the above copyright
Chris@69:    notice, this list of conditions and the following disclaimer in the
Chris@69:    documentation and/or other materials provided with the distribution.
Chris@69: 
Chris@69:    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
Chris@69:    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
Chris@69:    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
Chris@69:    A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
Chris@69:    OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
Chris@69:    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
Chris@69:    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
Chris@69:    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
Chris@69:    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
Chris@69:    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
Chris@69:    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
Chris@69: */
Chris@69: 
Chris@69: #ifdef HAVE_CONFIG_H
Chris@69: #include "config.h"
Chris@69: #endif
Chris@69: 
Chris@69: #include "cpu_support.h"
Chris@69: #include "macros.h"
Chris@69: #include "main.h"
Chris@69: #include "pitch.h"
Chris@69: #include "x86cpu.h"
Chris@69: 
Chris@69: #if (defined(OPUS_X86_MAY_HAVE_SSE) && !defined(OPUS_X86_PRESUME_SSE)) || \
Chris@69:   (defined(OPUS_X86_MAY_HAVE_SSE2) && !defined(OPUS_X86_PRESUME_SSE2)) || \
Chris@69:   (defined(OPUS_X86_MAY_HAVE_SSE4_1) && !defined(OPUS_X86_PRESUME_SSE4_1)) || \
Chris@69:   (defined(OPUS_X86_MAY_HAVE_AVX) && !defined(OPUS_X86_PRESUME_AVX))
Chris@69: 
Chris@69: 
Chris@69: #if defined(_MSC_VER)
Chris@69: 
Chris@69: #include <intrin.h>
Chris@69: static _inline void cpuid(unsigned int CPUInfo[4], unsigned int InfoType)
Chris@69: {
Chris@69:     __cpuid((int*)CPUInfo, InfoType);
Chris@69: }
Chris@69: 
Chris@69: #else
Chris@69: 
Chris@69: #if defined(CPU_INFO_BY_C)
Chris@69: #include <cpuid.h>
Chris@69: #endif
Chris@69: 
Chris@69: static void cpuid(unsigned int CPUInfo[4], unsigned int InfoType)
Chris@69: {
Chris@69: #if defined(CPU_INFO_BY_ASM)
Chris@69: #if defined(__i386__) && defined(__PIC__)
Chris@69: /* %ebx is PIC register in 32-bit, so mustn't clobber it. */
Chris@69:     __asm__ __volatile__ (
Chris@69:         "xchg %%ebx, %1\n"
Chris@69:         "cpuid\n"
Chris@69:         "xchg %%ebx, %1\n":
Chris@69:         "=a" (CPUInfo[0]),
Chris@69:         "=r" (CPUInfo[1]),
Chris@69:         "=c" (CPUInfo[2]),
Chris@69:         "=d" (CPUInfo[3]) :
Chris@69:         "0" (InfoType)
Chris@69:     );
Chris@69: #else
Chris@69:     __asm__ __volatile__ (
Chris@69:         "cpuid":
Chris@69:         "=a" (CPUInfo[0]),
Chris@69:         "=b" (CPUInfo[1]),
Chris@69:         "=c" (CPUInfo[2]),
Chris@69:         "=d" (CPUInfo[3]) :
Chris@69:         "0" (InfoType)
Chris@69:     );
Chris@69: #endif
Chris@69: #elif defined(CPU_INFO_BY_C)
Chris@69:     __get_cpuid(InfoType, &(CPUInfo[0]), &(CPUInfo[1]), &(CPUInfo[2]), &(CPUInfo[3]));
Chris@69: #endif
Chris@69: }
Chris@69: 
Chris@69: #endif
Chris@69: 
Chris@69: typedef struct CPU_Feature{
Chris@69:     /*  SIMD: 128-bit */
Chris@69:     int HW_SSE;
Chris@69:     int HW_SSE2;
Chris@69:     int HW_SSE41;
Chris@69:     /*  SIMD: 256-bit */
Chris@69:     int HW_AVX;
Chris@69: } CPU_Feature;
Chris@69: 
Chris@69: static void opus_cpu_feature_check(CPU_Feature *cpu_feature)
Chris@69: {
Chris@69:     unsigned int info[4] = {0};
Chris@69:     unsigned int nIds = 0;
Chris@69: 
Chris@69:     cpuid(info, 0);
Chris@69:     nIds = info[0];
Chris@69: 
Chris@69:     if (nIds >= 1){
Chris@69:         cpuid(info, 1);
Chris@69:         cpu_feature->HW_SSE = (info[3] & (1 << 25)) != 0;
Chris@69:         cpu_feature->HW_SSE2 = (info[3] & (1 << 26)) != 0;
Chris@69:         cpu_feature->HW_SSE41 = (info[2] & (1 << 19)) != 0;
Chris@69:         cpu_feature->HW_AVX = (info[2] & (1 << 28)) != 0;
Chris@69:     }
Chris@69:     else {
Chris@69:         cpu_feature->HW_SSE = 0;
Chris@69:         cpu_feature->HW_SSE2 = 0;
Chris@69:         cpu_feature->HW_SSE41 = 0;
Chris@69:         cpu_feature->HW_AVX = 0;
Chris@69:     }
Chris@69: }
Chris@69: 
Chris@69: int opus_select_arch(void)
Chris@69: {
Chris@69:     CPU_Feature cpu_feature;
Chris@69:     int arch;
Chris@69: 
Chris@69:     opus_cpu_feature_check(&cpu_feature);
Chris@69: 
Chris@69:     arch = 0;
Chris@69:     if (!cpu_feature.HW_SSE)
Chris@69:     {
Chris@69:        return arch;
Chris@69:     }
Chris@69:     arch++;
Chris@69: 
Chris@69:     if (!cpu_feature.HW_SSE2)
Chris@69:     {
Chris@69:        return arch;
Chris@69:     }
Chris@69:     arch++;
Chris@69: 
Chris@69:     if (!cpu_feature.HW_SSE41)
Chris@69:     {
Chris@69:         return arch;
Chris@69:     }
Chris@69:     arch++;
Chris@69: 
Chris@69:     if (!cpu_feature.HW_AVX)
Chris@69:     {
Chris@69:         return arch;
Chris@69:     }
Chris@69:     arch++;
Chris@69: 
Chris@69:     return arch;
Chris@69: }
Chris@69: 
Chris@69: #endif