cpuid.c 3.55 KB
Newer Older
1 2 3 4 5
/*
 * CPU detection code, extracted from mmx.h
 * (c)1997-99 by H. Dietz and R. Fisher
 * Converted to C and improved by Fabrice Bellard.
 *
6 7 8
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
9 10
 * modify it under the terms of the GNU Lesser General Public
 * 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 15 16 17 18
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * 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 21
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */
Fabrice Bellard's avatar
Fabrice Bellard committed
22 23

#include <stdlib.h>
24 25
#include "libavutil/x86_cpu.h"
#include "libavcodec/dsputil.h"
Fabrice Bellard's avatar
Fabrice Bellard committed
26

27 28
#undef printf

29 30
/* ebx saving is necessary for PIC. gcc seems unable to see it alone */
#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 36
         : "=a" (eax), "=S" (ebx),\
           "=c" (ecx), "=d" (edx)\
37
         : "0" (index));
Fabrice Bellard's avatar
Fabrice Bellard committed
38 39 40 41

/* Function to test if multimedia instructions are supported...  */
int mm_support(void)
{
42
    int rval = 0;
Fabrice Bellard's avatar
Fabrice Bellard committed
43
    int eax, ebx, ecx, edx;
44
    int max_std_level, max_ext_level, std_caps=0, ext_caps=0;
45
    x86_reg a, c;
46

47
    asm volatile (
48 49 50 51 52
        /* See if CPUID instruction is supported ... */
        /* ... Get copies of EFLAGS into eax and ecx */
        "pushf\n\t"
        "pop %0\n\t"
        "mov %0, %1\n\t"
53

54 55 56 57 58
        /* ... Toggle the ID bit in one copy and store */
        /*     to the EFLAGS reg */
        "xor $0x200000, %0\n\t"
        "push %0\n\t"
        "popf\n\t"
59

60 61 62 63 64 65 66
        /* ... Get the (hopefully modified) EFLAGS */
        "pushf\n\t"
        "pop %0\n\t"
        : "=a" (a), "=c" (c)
        :
        : "cc"
        );
67

68
    if (a == c)
Fabrice Bellard's avatar
Fabrice Bellard committed
69 70
        return 0; /* CPUID not supported */

71 72 73 74 75
    cpuid(0, max_std_level, ebx, ecx, edx);

    if(max_std_level >= 1){
        cpuid(1, eax, ebx, ecx, std_caps);
        if (std_caps & (1<<23))
76
            rval |= FF_MM_MMX;
77
        if (std_caps & (1<<25))
78 79 80
            rval |= FF_MM_MMXEXT
#if !defined(__GNUC__) || __GNUC__ > 2
                  | FF_MM_SSE;
81
        if (std_caps & (1<<26))
82
            rval |= FF_MM_SSE2;
83
        if (ecx & 1)
84
            rval |= FF_MM_SSE3;
85
        if (ecx & 0x00000200 )
86 87 88
            rval |= FF_MM_SSSE3
#endif
                  ;
89 90 91 92 93 94 95
    }

    cpuid(0x80000000, max_ext_level, ebx, ecx, edx);

    if(max_ext_level >= 0x80000001){
        cpuid(0x80000001, eax, ebx, ecx, ext_caps);
        if (ext_caps & (1<<31))
96
            rval |= FF_MM_3DNOW;
97
        if (ext_caps & (1<<30))
98
            rval |= FF_MM_3DNOWEXT;
99
        if (ext_caps & (1<<23))
100
            rval |= FF_MM_MMX;
101
        if (ext_caps & (1<<22))
102
            rval |= FF_MM_MMXEXT;
Fabrice Bellard's avatar
Fabrice Bellard committed
103
    }
104

105
#if 0
106
    av_log(NULL, AV_LOG_DEBUG, "%s%s%s%s%s%s%s%s\n",
107 108 109 110 111 112 113 114
        (rval&FF_MM_MMX) ? "MMX ":"",
        (rval&FF_MM_MMXEXT) ? "MMX2 ":"",
        (rval&FF_MM_SSE) ? "SSE ":"",
        (rval&FF_MM_SSE2) ? "SSE2 ":"",
        (rval&FF_MM_SSE3) ? "SSE3 ":"",
        (rval&FF_MM_SSSE3) ? "SSSE3 ":"",
        (rval&FF_MM_3DNOW) ? "3DNow ":"",
        (rval&FF_MM_3DNOWEXT) ? "3DNowExt ":"");
115 116
#endif
    return rval;
Fabrice Bellard's avatar
Fabrice Bellard committed
117
}
118

119
#ifdef TEST
120 121
int main ( void )
{
122 123 124 125
    int mm_flags;
    mm_flags = mm_support();
    printf("mm_support = 0x%08X\n",mm_flags);
    return 0;
126 127
}
#endif