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 "libavcodec/dsputil.h"
26
27#undef printf
28
29/* ebx saving is necessary for PIC. gcc seems unable to see it alone */
30#define cpuid(index,eax,ebx,ecx,edx)\
31    __asm__ volatile\
32        ("mov %%"REG_b", %%"REG_S"\n\t"\
33         "cpuid\n\t"\
34         "xchg %%"REG_b", %%"REG_S\
35         : "=a" (eax), "=S" (ebx),\
36           "=c" (ecx), "=d" (edx)\
37         : "0" (index));
38
39/* Function to test if multimedia instructions are supported...  */
40int mm_support(void)
41{
42    int rval = 0;
43    int eax, ebx, ecx, edx;
44    int max_std_level, max_ext_level, std_caps=0, ext_caps=0;
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, ebx, ecx, edx);
74
75    if(max_std_level >= 1){
76        cpuid(1, eax, ebx, ecx, std_caps);
77        if (std_caps & (1<<23))
78            rval |= FF_MM_MMX;
79        if (std_caps & (1<<25))
80            rval |= FF_MM_MMXEXT
81#if HAVE_SSE
82                  | FF_MM_SSE;
83        if (std_caps & (1<<26))
84            rval |= FF_MM_SSE2;
85        if (ecx & 1)
86            rval |= FF_MM_SSE3;
87        if (ecx & 0x00000200 )
88            rval |= FF_MM_SSSE3
89#endif
90                  ;
91    }
92
93    cpuid(0x80000000, max_ext_level, ebx, ecx, edx);
94
95    if(max_ext_level >= 0x80000001){
96        cpuid(0x80000001, eax, ebx, ecx, ext_caps);
97        if (ext_caps & (1<<31))
98            rval |= FF_MM_3DNOW;
99        if (ext_caps & (1<<30))
100            rval |= FF_MM_3DNOWEXT;
101        if (ext_caps & (1<<23))
102            rval |= FF_MM_MMX;
103        if (ext_caps & (1<<22))
104            rval |= FF_MM_MMXEXT;
105    }
106
107#if 0
108    av_log(NULL, AV_LOG_DEBUG, "%s%s%s%s%s%s%s%s\n",
109        (rval&FF_MM_MMX) ? "MMX ":"",
110        (rval&FF_MM_MMXEXT) ? "MMX2 ":"",
111        (rval&FF_MM_SSE) ? "SSE ":"",
112        (rval&FF_MM_SSE2) ? "SSE2 ":"",
113        (rval&FF_MM_SSE3) ? "SSE3 ":"",
114        (rval&FF_MM_SSSE3) ? "SSSE3 ":"",
115        (rval&FF_MM_3DNOW) ? "3DNow ":"",
116        (rval&FF_MM_3DNOWEXT) ? "3DNowExt ":"");
117#endif
118    return rval;
119}
120
121#ifdef TEST
122int main ( void )
123{
124    int mm_flags;
125    mm_flags = mm_support();
126    printf("mm_support = 0x%08X\n",mm_flags);
127    return 0;
128}
129#endif
130