git.sesse.net Git - x264/blob - common/cpu.c

   1 /*****************************************************************************
   2  * cpu.c: h264 encoder library
   3  *****************************************************************************
   4  * Copyright (C) 2003 Laurent Aimar
   5  * $Id: cpu.c,v 1.1 2004/06/03 19:27:06 fenrir Exp $
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  22  *****************************************************************************/
  23
  24 #ifdef HAVE_STDINT_H
  25 #include <stdint.h>
  26 #else
  27 #include <inttypes.h>
  28 #endif
  29 #include <stdlib.h>
  30 #include <string.h>
  31 #include <stdarg.h>
  32
  33 #include "x264.h"
  34 #include "cpu.h"
  35
  36 #if defined(ARCH_X86) || defined(ARCH_X86_64)
  37 extern int  x264_cpu_cpuid_test( void );
  38 extern uint32_t  x264_cpu_cpuid( uint32_t op, uint32_t *eax, uint32_t *ebx, uint32_t *ecx, uint32_t *edx );
  39 extern void x264_emms( void );
  40
  41 uint32_t x264_cpu_detect( void )
  42 {
  43     uint32_t cpu = 0;
  44
  45     uint32_t eax, ebx, ecx, edx;
  46     int      b_amd;
  47
  48
  49     if( !x264_cpu_cpuid_test() )
  50     {
  51         /* No cpuid */
  52         return 0;
  53     }
  54
  55     x264_cpu_cpuid( 0, &eax, &ebx, &ecx, &edx);
  56     if( eax == 0 )
  57     {
  58         return 0;
  59     }
  60     b_amd   = (ebx == 0x68747541) && (ecx == 0x444d4163) && (edx == 0x69746e65);
  61
  62     x264_cpu_cpuid( 1, &eax, &ebx, &ecx, &edx );
  63     if( (edx&0x00800000) == 0 )
  64     {
  65         /* No MMX */
  66         return 0;
  67     }
  68     cpu = X264_CPU_MMX;
  69     if( (edx&0x02000000) )
  70     {
  71         /* SSE - identical to AMD MMX extensions */
  72         cpu |= X264_CPU_MMXEXT|X264_CPU_SSE;
  73     }
  74     if( (edx&0x04000000) )
  75     {
  76         /* Is it OK ? */
  77         cpu |= X264_CPU_SSE2;
  78     }
  79
  80     x264_cpu_cpuid( 0x80000000, &eax, &ebx, &ecx, &edx );
  81     if( eax < 0x80000001 )
  82     {
  83         /* no extended capabilities */
  84         return cpu;
  85     }
  86
  87     x264_cpu_cpuid( 0x80000001, &eax, &ebx, &ecx, &edx );
  88     if( edx&0x80000000 )
  89     {
  90         cpu |= X264_CPU_3DNOW;
  91     }
  92     if( b_amd && (edx&0x00400000) )
  93     {
  94         /* AMD MMX extensions */
  95         cpu |= X264_CPU_MMXEXT;
  96     }
  97
  98     return cpu;
  99 }
 100
 101 void     x264_cpu_restore( uint32_t cpu )
 102 {
 103     if( cpu&(X264_CPU_MMX|X264_CPU_MMXEXT|X264_CPU_3DNOW|X264_CPU_3DNOWEXT) )
 104     {
 105         x264_emms();
 106     }
 107 }
 108
 109 #elif defined( ARCH_PPC )
 110
 111 #ifdef SYS_MACOSX
 112 #include <sys/sysctl.h>
 113 uint32_t x264_cpu_detect( void )
 114 {
 115     /* Thank you VLC */
 116     uint32_t cpu = 0;
 117     int      selectors[2] = { CTL_HW, HW_VECTORUNIT };
 118     int      has_altivec = 0;
 119     size_t   length = sizeof( has_altivec );
 120     int      error = sysctl( selectors, 2, &has_altivec, &length, NULL, 0 );
 121
 122     if( error == 0 && has_altivec != 0 )
 123     {
 124         cpu |= X264_CPU_ALTIVEC;
 125     }
 126
 127     return cpu;
 128 }
 129
 130 #elif defined( SYS_LINUX )
 131 uint32_t x264_cpu_detect( void )
 132 {
 133     /* FIXME (Linux PPC) */
 134     return X264_CPU_ALTIVEC;
 135 }
 136 #endif
 137
 138 void     x264_cpu_restore( uint32_t cpu )
 139 {
 140 }
 141
 142 #else
 143
 144 uint32_t x264_cpu_detect( void )
 145 {
 146     return 0;
 147 }
 148
 149 void     x264_cpu_restore( uint32_t cpu )
 150 {
 151 }
 152
 153 #endif