57b85ff9cee4b9b4da1cff72b86aa0da922ded9d
[ffmpeg.git] / 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 |= FF_MM_MMX;
84         if (std_caps & (1<<25))
85             rval |= FF_MM_MMXEXT
86 #if !defined(__GNUC__) || __GNUC__ > 2
87                   | FF_MM_SSE;
88         if (std_caps & (1<<26))
89             rval |= FF_MM_SSE2;
90         if (ecx & 1)
91             rval |= FF_MM_SSE3;
92         if (ecx & 0x00000200 )
93             rval |= FF_MM_SSSE3
94 #endif
95                   ;
96     }
97
98     cpuid(0x80000000, max_ext_level, ebx, ecx, edx);
99
100     if(max_ext_level >= 0x80000001){
101         cpuid(0x80000001, eax, ebx, ecx, ext_caps);
102         if (ext_caps & (1<<31))
103             rval |= FF_MM_3DNOW;
104         if (ext_caps & (1<<30))
105             rval |= FF_MM_3DNOWEXT;
106         if (ext_caps & (1<<23))
107             rval |= FF_MM_MMX;
108         if (ext_caps & (1<<22))
109             rval |= FF_MM_MMXEXT;
110     }
111
112 #if 0
113     av_log(NULL, AV_LOG_DEBUG, "%s%s%s%s%s%s%s%s\n",
114         (rval&FF_MM_MMX) ? "MMX ":"",
115         (rval&FF_MM_MMXEXT) ? "MMX2 ":"",
116         (rval&FF_MM_SSE) ? "SSE ":"",
117         (rval&FF_MM_SSE2) ? "SSE2 ":"",
118         (rval&FF_MM_SSE3) ? "SSE3 ":"",
119         (rval&FF_MM_SSSE3) ? "SSSE3 ":"",
120         (rval&FF_MM_3DNOW) ? "3DNow ":"",
121         (rval&FF_MM_3DNOWEXT) ? "3DNowExt ":"");
122 #endif
123     return rval;
124 }
125
126 #ifdef TEST
127 int main ( void )
128 {
129   int mm_flags;
130   mm_flags = mm_support();
131   printf("mm_support = 0x%08X\n",mm_flags);
132   return 0;
133 }
134 #endif