Clean up av_get_cpu_flag()
[ffmpeg.git] / libavutil / x86 / cpu.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 "libavutil/x86_cpu.h"
25 #include "libavutil/cpu.h"
26
27 /* ebx saving is necessary for PIC. gcc seems unable to see it alone */
28 #define cpuid(index,eax,ebx,ecx,edx)\
29     __asm__ volatile\
30         ("mov %%"REG_b", %%"REG_S"\n\t"\
31          "cpuid\n\t"\
32          "xchg %%"REG_b", %%"REG_S\
33          : "=a" (eax), "=S" (ebx),\
34            "=c" (ecx), "=d" (edx)\
35          : "0" (index));
36
37 /* Function to test if multimedia instructions are supported...  */
38 int ff_get_cpu_flags_x86(void)
39 {
40     int rval = 0;
41     int eax, ebx, ecx, edx;
42     int max_std_level, max_ext_level, std_caps=0, ext_caps=0;
43     int family=0, model=0;
44     union { int i[3]; char c[12]; } vendor;
45
46 #if ARCH_X86_32
47     x86_reg a, c;
48     __asm__ volatile (
49         /* See if CPUID instruction is supported ... */
50         /* ... Get copies of EFLAGS into eax and ecx */
51         "pushfl\n\t"
52         "pop %0\n\t"
53         "mov %0, %1\n\t"
54
55         /* ... Toggle the ID bit in one copy and store */
56         /*     to the EFLAGS reg */
57         "xor $0x200000, %0\n\t"
58         "push %0\n\t"
59         "popfl\n\t"
60
61         /* ... Get the (hopefully modified) EFLAGS */
62         "pushfl\n\t"
63         "pop %0\n\t"
64         : "=a" (a), "=c" (c)
65         :
66         : "cc"
67         );
68
69     if (a == c)
70         return 0; /* CPUID not supported */
71 #endif
72
73     cpuid(0, max_std_level, vendor.i[0], vendor.i[2], vendor.i[1]);
74
75     if(max_std_level >= 1){
76         cpuid(1, eax, ebx, ecx, std_caps);
77         family = ((eax>>8)&0xf) + ((eax>>20)&0xff);
78         model  = ((eax>>4)&0xf) + ((eax>>12)&0xf0);
79         if (std_caps & (1<<23))
80             rval |= AV_CPU_FLAG_MMX;
81         if (std_caps & (1<<25))
82             rval |= AV_CPU_FLAG_MMX2
83 #if HAVE_SSE
84                   | AV_CPU_FLAG_SSE;
85         if (std_caps & (1<<26))
86             rval |= AV_CPU_FLAG_SSE2;
87         if (ecx & 1)
88             rval |= AV_CPU_FLAG_SSE3;
89         if (ecx & 0x00000200 )
90             rval |= AV_CPU_FLAG_SSSE3;
91         if (ecx & 0x00080000 )
92             rval |= AV_CPU_FLAG_SSE4;
93         if (ecx & 0x00100000 )
94             rval |= AV_CPU_FLAG_SSE42;
95 #endif
96                   ;
97     }
98
99     cpuid(0x80000000, max_ext_level, ebx, ecx, edx);
100
101     if(max_ext_level >= 0x80000001){
102         cpuid(0x80000001, eax, ebx, ecx, ext_caps);
103         if (ext_caps & (1<<31))
104             rval |= AV_CPU_FLAG_3DNOW;
105         if (ext_caps & (1<<30))
106             rval |= AV_CPU_FLAG_3DNOWEXT;
107         if (ext_caps & (1<<23))
108             rval |= AV_CPU_FLAG_MMX;
109         if (ext_caps & (1<<22))
110             rval |= AV_CPU_FLAG_MMX2;
111     }
112
113     if (!strncmp(vendor.c, "GenuineIntel", 12) &&
114         family == 6 && (model == 9 || model == 13 || model == 14)) {
115         /* 6/9 (pentium-m "banias"), 6/13 (pentium-m "dothan"), and 6/14 (core1 "yonah")
116          * theoretically support sse2, but it's usually slower than mmx,
117          * so let's just pretend they don't. */
118         if (rval & AV_CPU_FLAG_SSE2) rval ^= AV_CPU_FLAG_SSE2SLOW|AV_CPU_FLAG_SSE2;
119         if (rval & AV_CPU_FLAG_SSE3) rval ^= AV_CPU_FLAG_SSE3SLOW|AV_CPU_FLAG_SSE3;
120     }
121
122     return rval;
123 }