git.sesse.net Git - ffmpeg/blob - libavcodec/i386/cputest.c

   1 /*
   2  * CPU detection code, extracted from mmx.h
   3  * (c)1997-99 by H. Dietz and R. Fisher
   4  * Converted to C and improved by Fabrice Bellard.
   5  *
   6  * This file is part of FFmpeg.
   7  *
   8  * FFmpeg is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * FFmpeg is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with FFmpeg; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21  */
  22
  23 #include <stdlib.h>
  24 #include "../dsputil.h"
  25
  26 #undef printf
  27
  28 #ifdef ARCH_X86_64
  29 #  define REG_b "rbx"
  30 #  define REG_S "rsi"
  31 #else
  32 #  define REG_b "ebx"
  33 #  define REG_S "esi"
  34 #endif
  35
  36 /* ebx saving is necessary for PIC. gcc seems unable to see it alone */
  37 #define cpuid(index,eax,ebx,ecx,edx)\
  38     __asm __volatile\
  39         ("mov %%"REG_b", %%"REG_S"\n\t"\
  40          "cpuid\n\t"\
  41          "xchg %%"REG_b", %%"REG_S\
  42          : "=a" (eax), "=S" (ebx),\
  43            "=c" (ecx), "=d" (edx)\
  44          : "0" (index));
  45
  46 /* Function to test if multimedia instructions are supported...  */
  47 int mm_support(void)
  48 {
  49     int rval = 0;
  50     int eax, ebx, ecx, edx;
  51     int max_std_level, max_ext_level, std_caps=0, ext_caps=0;
  52     long a, c;
  53
  54     __asm__ __volatile__ (
  55                           /* See if CPUID instruction is supported ... */
  56                           /* ... Get copies of EFLAGS into eax and ecx */
  57                           "pushf\n\t"
  58                           "pop %0\n\t"
  59                           "mov %0, %1\n\t"
  60
  61                           /* ... Toggle the ID bit in one copy and store */
  62                           /*     to the EFLAGS reg */
  63                           "xor $0x200000, %0\n\t"
  64                           "push %0\n\t"
  65                           "popf\n\t"
  66
  67                           /* ... Get the (hopefully modified) EFLAGS */
  68                           "pushf\n\t"
  69                           "pop %0\n\t"
  70                           : "=a" (a), "=c" (c)
  71                           :
  72                           : "cc"
  73                           );
  74
  75     if (a == c)
  76         return 0; /* CPUID not supported */
  77
  78     cpuid(0, max_std_level, ebx, ecx, edx);
  79
  80     if(max_std_level >= 1){
  81         cpuid(1, eax, ebx, ecx, std_caps);
  82         if (std_caps & (1<<23))
  83             rval |= MM_MMX;
  84         if (std_caps & (1<<25))
  85             rval |= MM_MMXEXT | MM_SSE;
  86         if (std_caps & (1<<26))
  87             rval |= MM_SSE2;
  88         if (ecx & 1)
  89             rval |= MM_SSE3;
  90     }
  91
  92     cpuid(0x80000000, max_ext_level, ebx, ecx, edx);
  93
  94     if(max_ext_level >= 0x80000001){
  95         cpuid(0x80000001, eax, ebx, ecx, ext_caps);
  96         if (ext_caps & (1<<31))
  97             rval |= MM_3DNOW;
  98         if (ext_caps & (1<<30))
  99             rval |= MM_3DNOWEXT;
 100         if (ext_caps & (1<<23))
 101             rval |= MM_MMX;
 102         if (ext_caps & (1<<22))
 103             rval |= MM_MMXEXT;
 104     }
 105
 106 #if 0
 107     av_log(NULL, AV_LOG_DEBUG, "%s%s%s%s%s%s\n",
 108         (rval&MM_MMX) ? "MMX ":"",
 109         (rval&MM_MMXEXT) ? "MMX2 ":"",
 110         (rval&MM_SSE) ? "SSE ":"",
 111         (rval&MM_SSE2) ? "SSE2 ":"",
 112         (rval&MM_3DNOW) ? "3DNow ":"",
 113         (rval&MM_3DNOWEXT) ? "3DNowExt ":"");
 114 #endif
 115     return rval;
 116 }
 117
 118 #ifdef __TEST__
 119 int main ( void )
 120 {
 121   int mm_flags;
 122   mm_flags = mm_support();
 123   printf("mm_support = 0x%08X\n",mm_flags);
 124   return 0;
 125 }
 126 #endif