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 $
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
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.
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.
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 *****************************************************************************/
24 #if defined(HAVE_PTHREAD) && defined(SYS_LINUX)
29 #include <kernel/OS.h>
32 #include <sys/types.h>
33 #include <sys/sysctl.h>
39 extern int x264_cpu_cpuid_test( void );
40 extern uint32_t x264_cpu_cpuid( uint32_t op, uint32_t *eax, uint32_t *ebx, uint32_t *ecx, uint32_t *edx );
41 extern void x264_emms( void );
43 uint32_t x264_cpu_detect( void )
47 uint32_t eax, ebx, ecx, edx;
50 if( !x264_cpu_cpuid_test() )
56 x264_cpu_cpuid( 0, &eax, &ebx, &ecx, &edx);
61 b_amd = (ebx == 0x68747541) && (ecx == 0x444d4163) && (edx == 0x69746e65);
63 x264_cpu_cpuid( 1, &eax, &ebx, &ecx, &edx );
64 if( (edx&0x00800000) == 0 )
70 if( (edx&0x02000000) )
72 /* SSE - identical to AMD MMX extensions */
73 cpu |= X264_CPU_MMXEXT|X264_CPU_SSE;
75 if( (edx&0x04000000) )
81 if( (ecx&0x00000001) )
85 if( (ecx&0x00000200) )
87 cpu |= X264_CPU_SSSE3;
91 x264_cpu_cpuid( 0x80000000, &eax, &ebx, &ecx, &edx );
92 if( eax < 0x80000001 )
94 /* no extended capabilities */
98 x264_cpu_cpuid( 0x80000001, &eax, &ebx, &ecx, &edx );
101 cpu |= X264_CPU_3DNOW;
103 if( b_amd && (edx&0x00400000) )
105 /* AMD MMX extensions */
106 cpu |= X264_CPU_MMXEXT;
112 void x264_cpu_restore( uint32_t cpu )
114 if( cpu&(X264_CPU_MMX|X264_CPU_MMXEXT|X264_CPU_3DNOW|X264_CPU_3DNOWEXT) )
120 #elif defined( ARCH_PPC )
123 #include <sys/sysctl.h>
124 uint32_t x264_cpu_detect( void )
128 int selectors[2] = { CTL_HW, HW_VECTORUNIT };
130 size_t length = sizeof( has_altivec );
131 int error = sysctl( selectors, 2, &has_altivec, &length, NULL, 0 );
133 if( error == 0 && has_altivec != 0 )
135 cpu |= X264_CPU_ALTIVEC;
141 #elif defined( SYS_LINUX )
144 static sigjmp_buf jmpbuf;
145 static volatile sig_atomic_t canjump = 0;
147 static void sigill_handler( int sig )
151 signal( sig, SIG_DFL );
156 siglongjmp( jmpbuf, 1 );
159 uint32_t x264_cpu_detect( void )
161 static void (* oldsig)( int );
163 oldsig = signal( SIGILL, sigill_handler );
164 if( sigsetjmp( jmpbuf, 1 ) )
166 signal( SIGILL, oldsig );
171 asm volatile( "mtspr 256, %0\n\t"
177 signal( SIGILL, oldsig );
179 return X264_CPU_ALTIVEC;
183 void x264_cpu_restore( uint32_t cpu )
189 uint32_t x264_cpu_detect( void )
194 void x264_cpu_restore( uint32_t cpu )
201 int x264_cpu_num_processors( void )
203 #if !defined(HAVE_PTHREAD)
207 return pthread_num_processors_np();
209 #elif defined(SYS_LINUX)
213 memset( &p_aff, 0, sizeof(p_aff) );
214 sched_getaffinity( 0, sizeof(p_aff), &p_aff );
215 for( np = 0, bit = 0; bit < sizeof(p_aff); bit++ )
216 np += (((uint8_t *)&p_aff)[bit / 8] >> (bit % 8)) & 1;
219 #elif defined(SYS_BEOS)
221 get_system_info( &info );
222 return info.cpu_count;
224 #elif defined(SYS_MACOSX)
226 size_t length = sizeof( numberOfCPUs );
227 if( sysctlbyname("hw.ncpu", &numberOfCPUs, &length, NULL, 0) )