cpuid.c 3.76 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

46 47
#if ARCH_X86_32
    x86_reg a, c;
48
    __asm__ volatile (
49 50
        /* See if CPUID instruction is supported ... */
        /* ... Get copies of EFLAGS into eax and ecx */
51
        "pushfl\n\t"
52 53
        "pop %0\n\t"
        "mov %0, %1\n\t"
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"
59
        "popfl\n\t"
60

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

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

73 74 75 76 77
    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))
78
            rval |= FF_MM_MMX;
79
        if (std_caps & (1<<25))
80
            rval |= FF_MM_MMX2
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 89 90 91 92
            rval |= FF_MM_SSSE3;
        if (ecx & 0x00080000 )
            rval |= FF_MM_SSE4;
        if (ecx & 0x00100000 )
            rval |= FF_MM_SSE42;
93 94
#endif
                  ;
95 96 97 98 99 100 101
    }

    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))
102
            rval |= FF_MM_3DNOW;
103
        if (ext_caps & (1<<30))
104
            rval |= FF_MM_3DNOWEXT;
105
        if (ext_caps & (1<<23))
106
            rval |= FF_MM_MMX;
107
        if (ext_caps & (1<<22))
108
            rval |= FF_MM_MMX2;
Fabrice Bellard's avatar
Fabrice Bellard committed
109
    }
110

111
#if 0
112
    av_log(NULL, AV_LOG_DEBUG, "%s%s%s%s%s%s%s%s%s%s\n",
113
        (rval&FF_MM_MMX) ? "MMX ":"",
114
        (rval&FF_MM_MMX2) ? "MMX2 ":"",
115 116 117 118
        (rval&FF_MM_SSE) ? "SSE ":"",
        (rval&FF_MM_SSE2) ? "SSE2 ":"",
        (rval&FF_MM_SSE3) ? "SSE3 ":"",
        (rval&FF_MM_SSSE3) ? "SSSE3 ":"",
119 120
        (rval&FF_MM_SSE4) ? "SSE4.1 ":"",
        (rval&FF_MM_SSE42) ? "SSE4.2 ":"",
121 122
        (rval&FF_MM_3DNOW) ? "3DNow ":"",
        (rval&FF_MM_3DNOWEXT) ? "3DNowExt ":"");
123 124
#endif
    return rval;
Fabrice Bellard's avatar
Fabrice Bellard committed
125
}
126

127
#ifdef TEST
128 129
int main ( void )
{
130 131 132 133
    int mm_flags;
    mm_flags = mm_support();
    printf("mm_support = 0x%08X\n",mm_flags);
    return 0;
134 135
}
#endif