ssse3 detection. x86_64 ssse3 satd and quant.
[x264/x264-sandbox.git] / 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 #if defined(HAVE_PTHREAD) && defined(SYS_LINUX)
25 #define _GNU_SOURCE
26 #include <sched.h>
27 #endif
28 #ifdef SYS_BEOS
29 #include <kernel/OS.h>
30 #endif
31 #ifdef SYS_MACOSX
32 #include <sys/types.h>
33 #include <sys/sysctl.h>
34 #endif
35
36 #include <string.h>
37
38 #include "common.h"
39
40 #if defined(ARCH_X86) || defined(ARCH_X86_64)
41 extern int  x264_cpu_cpuid_test( void );
42 extern uint32_t  x264_cpu_cpuid( uint32_t op, uint32_t *eax, uint32_t *ebx, uint32_t *ecx, uint32_t *edx );
43 extern void x264_emms( void );
44
45 uint32_t x264_cpu_detect( void )
46 {
47     uint32_t cpu = 0;
48
49     uint32_t eax, ebx, ecx, edx;
50     int      b_amd;
51
52
53     if( !x264_cpu_cpuid_test() )
54     {
55         /* No cpuid */
56         return 0;
57     }
58
59     x264_cpu_cpuid( 0, &eax, &ebx, &ecx, &edx);
60     if( eax == 0 )
61     {
62         return 0;
63     }
64     b_amd   = (ebx == 0x68747541) && (ecx == 0x444d4163) && (edx == 0x69746e65);
65
66     x264_cpu_cpuid( 1, &eax, &ebx, &ecx, &edx );
67     if( (edx&0x00800000) == 0 )
68     {
69         /* No MMX */
70         return 0;
71     }
72     cpu = X264_CPU_MMX;
73     if( (edx&0x02000000) )
74     {
75         /* SSE - identical to AMD MMX extensions */
76         cpu |= X264_CPU_MMXEXT|X264_CPU_SSE;
77     }
78     if( (edx&0x04000000) )
79     {
80         /* Is it OK ? */
81         cpu |= X264_CPU_SSE2;
82     }
83 #ifdef HAVE_SSE3
84     if( (ecx&0x00000001) )
85     {
86         cpu |= X264_CPU_SSE3;
87     }
88     if( (ecx&0x00000200) )
89     {
90         cpu |= X264_CPU_SSSE3;
91     }
92 #endif
93
94     x264_cpu_cpuid( 0x80000000, &eax, &ebx, &ecx, &edx );
95     if( eax < 0x80000001 )
96     {
97         /* no extended capabilities */
98         return cpu;
99     }
100
101     x264_cpu_cpuid( 0x80000001, &eax, &ebx, &ecx, &edx );
102     if( edx&0x80000000 )
103     {
104         cpu |= X264_CPU_3DNOW;
105     }
106     if( b_amd && (edx&0x00400000) )
107     {
108         /* AMD MMX extensions */
109         cpu |= X264_CPU_MMXEXT;
110     }
111
112     return cpu;
113 }
114
115 void     x264_cpu_restore( uint32_t cpu )
116 {
117     if( cpu&(X264_CPU_MMX|X264_CPU_MMXEXT|X264_CPU_3DNOW|X264_CPU_3DNOWEXT) )
118     {
119         x264_emms();
120     }
121 }
122
123 #elif defined( ARCH_PPC )
124
125 #ifdef SYS_MACOSX
126 #include <sys/sysctl.h>
127 uint32_t x264_cpu_detect( void )
128 {
129     /* Thank you VLC */
130     uint32_t cpu = 0;
131     int      selectors[2] = { CTL_HW, HW_VECTORUNIT };
132     int      has_altivec = 0;
133     size_t   length = sizeof( has_altivec );
134     int      error = sysctl( selectors, 2, &has_altivec, &length, NULL, 0 );
135
136     if( error == 0 && has_altivec != 0 )
137     {
138         cpu |= X264_CPU_ALTIVEC;
139     }
140
141     return cpu;
142 }
143
144 #elif defined( SYS_LINUX )
145 #include <signal.h>
146 #include <setjmp.h>
147 static sigjmp_buf jmpbuf;
148 static volatile sig_atomic_t canjump = 0;
149
150 static void sigill_handler( int sig )
151 {
152     if( !canjump )
153     {
154         signal( sig, SIG_DFL );
155         raise( sig );
156     }
157
158     canjump = 0;
159     siglongjmp( jmpbuf, 1 );
160 }
161
162 uint32_t x264_cpu_detect( void )
163 {
164     static void (* oldsig)( int );
165
166     oldsig = signal( SIGILL, sigill_handler );
167     if( sigsetjmp( jmpbuf, 1 ) )
168     {
169         signal( SIGILL, oldsig );
170         return 0;
171     }
172
173     canjump = 1;
174     asm volatile( "mtspr 256, %0\n\t"
175                   "vand 0, 0, 0\n\t"
176                   :
177                   : "r"(-1) );
178     canjump = 0;
179
180     signal( SIGILL, oldsig );
181
182     return X264_CPU_ALTIVEC;
183 }
184 #endif
185
186 void     x264_cpu_restore( uint32_t cpu )
187 {
188 }
189
190 #else
191
192 uint32_t x264_cpu_detect( void )
193 {
194     return 0;
195 }
196
197 void     x264_cpu_restore( uint32_t cpu )
198 {
199 }
200
201 #endif
202
203
204 int x264_cpu_num_processors( void )
205 {
206 #if !defined(HAVE_PTHREAD)
207     return 1;
208
209 #elif defined(WIN32)
210     return pthread_num_processors_np();
211
212 #elif defined(SYS_LINUX)
213     unsigned int bit;
214     int np;
215     cpu_set_t p_aff;
216     memset( &p_aff, 0, sizeof(p_aff) );
217     sched_getaffinity( 0, sizeof(p_aff), &p_aff );
218     for( np = 0, bit = 0; bit < sizeof(p_aff); bit++ )
219         np += (((uint8_t *)&p_aff)[bit / 8] >> (bit % 8)) & 1;
220     return np;
221
222 #elif defined(SYS_BEOS)
223     system_info info;
224     get_system_info( &info );
225     return info.cpu_count;
226
227 #elif defined(SYS_MACOSX)
228     int numberOfCPUs;
229     size_t length = sizeof( numberOfCPUs );
230     if( sysctlbyname("hw.ncpu", &numberOfCPUs, &length, NULL, 0) )
231     {
232         numberOfCPUs = 1;
233     }
234     return numberOfCPUs;
235
236 #else
237     return 1;
238 #endif
239 }