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

   1 /*****************************************************************************
   2  * cpu.c: h264 encoder library
   3  *****************************************************************************
   4  * Copyright (C) 2003-2008 x264 project
   5  *
   6  * Authors: Loren Merritt <lorenm@u.washington.edu>
   7  *          Laurent Aimar <fenrir@via.ecp.fr>
   8  *          Fiona Glaser <fiona@x264.com>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
  23  *****************************************************************************/
  24
  25 #define _GNU_SOURCE // for sched_getaffinity
  26 #include "common.h"
  27 #include "cpu.h"
  28
  29 #if defined(HAVE_PTHREAD) && defined(SYS_LINUX)
  30 #include <sched.h>
  31 #endif
  32 #ifdef SYS_BEOS
  33 #include <kernel/OS.h>
  34 #endif
  35 #if defined(SYS_MACOSX) || defined(SYS_FREEBSD)
  36 #include <sys/types.h>
  37 #include <sys/sysctl.h>
  38 #endif
  39 #ifdef SYS_OPENBSD
  40 #include <sys/param.h>
  41 #include <sys/sysctl.h>
  42 #include <machine/cpu.h>
  43 #endif
  44
  45 const x264_cpu_name_t x264_cpu_names[] = {
  46     {"Altivec", X264_CPU_ALTIVEC},
  47 //  {"MMX",     X264_CPU_MMX}, // we don't support asm on mmx1 cpus anymore
  48     {"MMX2",    X264_CPU_MMX|X264_CPU_MMXEXT},
  49     {"MMXEXT",  X264_CPU_MMX|X264_CPU_MMXEXT},
  50 //  {"SSE",     X264_CPU_MMX|X264_CPU_MMXEXT|X264_CPU_SSE}, // there are no sse1 functions in x264
  51     {"SSE2Slow",X264_CPU_MMX|X264_CPU_MMXEXT|X264_CPU_SSE|X264_CPU_SSE2|X264_CPU_SSE2_IS_SLOW},
  52     {"SSE2",    X264_CPU_MMX|X264_CPU_MMXEXT|X264_CPU_SSE|X264_CPU_SSE2},
  53     {"SSE2Fast",X264_CPU_MMX|X264_CPU_MMXEXT|X264_CPU_SSE|X264_CPU_SSE2|X264_CPU_SSE2_IS_FAST},
  54     {"SSE3",    X264_CPU_MMX|X264_CPU_MMXEXT|X264_CPU_SSE|X264_CPU_SSE2|X264_CPU_SSE3},
  55     {"SSSE3",   X264_CPU_MMX|X264_CPU_MMXEXT|X264_CPU_SSE|X264_CPU_SSE2|X264_CPU_SSE3|X264_CPU_SSSE3},
  56     {"FastShuffle",   X264_CPU_MMX|X264_CPU_MMXEXT|X264_CPU_SSE|X264_CPU_SSE2|X264_CPU_SHUFFLE_IS_FAST},
  57     {"SSE4.1",  X264_CPU_MMX|X264_CPU_MMXEXT|X264_CPU_SSE|X264_CPU_SSE2|X264_CPU_SSE3|X264_CPU_SSSE3|X264_CPU_SSE4},
  58     {"SSE4.2",  X264_CPU_MMX|X264_CPU_MMXEXT|X264_CPU_SSE|X264_CPU_SSE2|X264_CPU_SSE3|X264_CPU_SSSE3|X264_CPU_SSE4|X264_CPU_SSE42},
  59     {"Cache32", X264_CPU_CACHELINE_32},
  60     {"Cache64", X264_CPU_CACHELINE_64},
  61     {"SSEMisalign", X264_CPU_SSE_MISALIGN},
  62     {"LZCNT", X264_CPU_LZCNT},
  63     {"Slow_mod4_stack", X264_CPU_STACK_MOD4},
  64     {"ARMv6", X264_CPU_ARMV6},
  65     {"NEON",  X264_CPU_NEON},
  66     {"Fast_NEON_MRC",  X264_CPU_FAST_NEON_MRC},
  67     {"", 0},
  68 };
  69
  70 #if (defined(ARCH_PPC) && defined(SYS_LINUX)) || (defined(ARCH_ARM) && !defined(HAVE_NEON))
  71 #include <signal.h>
  72 #include <setjmp.h>
  73 static sigjmp_buf jmpbuf;
  74 static volatile sig_atomic_t canjump = 0;
  75
  76 static void sigill_handler( int sig )
  77 {
  78     if( !canjump )
  79     {
  80         signal( sig, SIG_DFL );
  81         raise( sig );
  82     }
  83
  84     canjump = 0;
  85     siglongjmp( jmpbuf, 1 );
  86 }
  87 #endif
  88
  89 #ifdef HAVE_MMX
  90 extern int  x264_cpu_cpuid_test( void );
  91 extern uint32_t  x264_cpu_cpuid( uint32_t op, uint32_t *eax, uint32_t *ebx, uint32_t *ecx, uint32_t *edx );
  92
  93 uint32_t x264_cpu_detect( void )
  94 {
  95     uint32_t cpu = 0;
  96     uint32_t eax, ebx, ecx, edx;
  97     uint32_t vendor[4] = {0};
  98     int max_extended_cap;
  99     int cache;
 100
 101 #ifndef ARCH_X86_64
 102     if( !x264_cpu_cpuid_test() )
 103         return 0;
 104 #endif
 105
 106     x264_cpu_cpuid( 0, &eax, vendor+0, vendor+2, vendor+1 );
 107     if( eax == 0 )
 108         return 0;
 109
 110     x264_cpu_cpuid( 1, &eax, &ebx, &ecx, &edx );
 111     if( edx&0x00800000 )
 112         cpu |= X264_CPU_MMX;
 113     else
 114         return 0;
 115     if( edx&0x02000000 )
 116         cpu |= X264_CPU_MMXEXT|X264_CPU_SSE;
 117     if( edx&0x04000000 )
 118         cpu |= X264_CPU_SSE2;
 119     if( ecx&0x00000001 )
 120         cpu |= X264_CPU_SSE3;
 121     if( ecx&0x00000200 )
 122         cpu |= X264_CPU_SSSE3;
 123     if( ecx&0x00080000 )
 124         cpu |= X264_CPU_SSE4;
 125     if( ecx&0x00100000 )
 126         cpu |= X264_CPU_SSE42;
 127
 128     if( cpu & X264_CPU_SSSE3 )
 129         cpu |= X264_CPU_SSE2_IS_FAST;
 130     if( cpu & X264_CPU_SSE4 )
 131         cpu |= X264_CPU_SHUFFLE_IS_FAST;
 132
 133     x264_cpu_cpuid( 0x80000000, &eax, &ebx, &ecx, &edx );
 134     max_extended_cap = eax;
 135
 136     if( !strcmp((char*)vendor, "AuthenticAMD") && max_extended_cap >= 0x80000001 )
 137     {
 138         x264_cpu_cpuid( 0x80000001, &eax, &ebx, &ecx, &edx );
 139         if( edx&0x00400000 )
 140             cpu |= X264_CPU_MMXEXT;
 141         if( cpu & X264_CPU_SSE2 )
 142         {
 143             if( ecx&0x00000040 ) /* SSE4a */
 144             {
 145                 cpu |= X264_CPU_SSE2_IS_FAST;
 146                 cpu |= X264_CPU_LZCNT;
 147                 cpu |= X264_CPU_SHUFFLE_IS_FAST;
 148             }
 149             else
 150                 cpu |= X264_CPU_SSE2_IS_SLOW;
 151
 152             if( ecx&0x00000080 ) /* Misalign SSE */
 153             {
 154                 cpu |= X264_CPU_SSE_MISALIGN;
 155                 x264_cpu_mask_misalign_sse();
 156             }
 157         }
 158     }
 159
 160     if( !strcmp((char*)vendor, "GenuineIntel") )
 161     {
 162         int family, model, stepping;
 163         x264_cpu_cpuid( 1, &eax, &ebx, &ecx, &edx );
 164         family = ((eax>>8)&0xf) + ((eax>>20)&0xff);
 165         model  = ((eax>>4)&0xf) + ((eax>>12)&0xf0);
 166         stepping = eax&0xf;
 167         /* 6/9 (pentium-m "banias"), 6/13 (pentium-m "dothan"), and 6/14 (core1 "yonah")
 168          * theoretically support sse2, but it's significantly slower than mmx for
 169          * almost all of x264's functions, so let's just pretend they don't. */
 170         if( family==6 && (model==9 || model==13 || model==14) )
 171         {
 172             cpu &= ~(X264_CPU_SSE2|X264_CPU_SSE3);
 173             assert(!(cpu&(X264_CPU_SSSE3|X264_CPU_SSE4)));
 174         }
 175     }
 176
 177     if( (!strcmp((char*)vendor, "GenuineIntel") || !strcmp((char*)vendor, "CyrixInstead")) && !(cpu&X264_CPU_SSE42))
 178     {
 179         /* cacheline size is specified in 3 places, any of which may be missing */
 180         x264_cpu_cpuid( 1, &eax, &ebx, &ecx, &edx );
 181         cache = (ebx&0xff00)>>5; // cflush size
 182         if( !cache && max_extended_cap >= 0x80000006 )
 183         {
 184             x264_cpu_cpuid( 0x80000006, &eax, &ebx, &ecx, &edx );
 185             cache = ecx&0xff; // cacheline size
 186         }
 187         if( !cache )
 188         {
 189             // Cache and TLB Information
 190             static const char cache32_ids[] = { 0x0a, 0x0c, 0x41, 0x42, 0x43, 0x44, 0x45, 0x82, 0x83, 0x84, 0x85, 0 };
 191             static const char cache64_ids[] = { 0x22, 0x23, 0x25, 0x29, 0x2c, 0x46, 0x47, 0x49, 0x60, 0x66, 0x67, 0x68, 0x78, 0x79, 0x7a, 0x7b, 0x7c, 0x7c, 0x7f, 0x86, 0x87, 0 };
 192             uint32_t buf[4];
 193             int max, i=0, j;
 194             do {
 195                 x264_cpu_cpuid( 2, buf+0, buf+1, buf+2, buf+3 );
 196                 max = buf[0]&0xff;
 197                 buf[0] &= ~0xff;
 198                 for(j=0; j<4; j++)
 199                     if( !(buf[j]>>31) )
 200                         while( buf[j] )
 201                         {
 202                             if( strchr( cache32_ids, buf[j]&0xff ) )
 203                                 cache = 32;
 204                             if( strchr( cache64_ids, buf[j]&0xff ) )
 205                                 cache = 64;
 206                             buf[j] >>= 8;
 207                         }
 208             } while( ++i < max );
 209         }
 210
 211         if( cache == 32 )
 212             cpu |= X264_CPU_CACHELINE_32;
 213         else if( cache == 64 )
 214             cpu |= X264_CPU_CACHELINE_64;
 215         else
 216             fprintf( stderr, "x264 [warning]: unable to determine cacheline size\n" );
 217     }
 218
 219 #ifdef BROKEN_STACK_ALIGNMENT
 220     cpu |= X264_CPU_STACK_MOD4;
 221 #endif
 222
 223     return cpu;
 224 }
 225
 226 #elif defined( ARCH_PPC )
 227
 228 #if defined(SYS_MACOSX) || defined(SYS_OPENBSD)
 229 #include <sys/sysctl.h>
 230 uint32_t x264_cpu_detect( void )
 231 {
 232     /* Thank you VLC */
 233     uint32_t cpu = 0;
 234 #ifdef SYS_OPENBSD
 235     int      selectors[2] = { CTL_MACHDEP, CPU_ALTIVEC };
 236 #else
 237     int      selectors[2] = { CTL_HW, HW_VECTORUNIT };
 238 #endif
 239     int      has_altivec = 0;
 240     size_t   length = sizeof( has_altivec );
 241     int      error = sysctl( selectors, 2, &has_altivec, &length, NULL, 0 );
 242
 243     if( error == 0 && has_altivec != 0 )
 244     {
 245         cpu |= X264_CPU_ALTIVEC;
 246     }
 247
 248     return cpu;
 249 }
 250
 251 #elif defined( SYS_LINUX )
 252
 253 uint32_t x264_cpu_detect( void )
 254 {
 255     static void (* oldsig)( int );
 256
 257     oldsig = signal( SIGILL, sigill_handler );
 258     if( sigsetjmp( jmpbuf, 1 ) )
 259     {
 260         signal( SIGILL, oldsig );
 261         return 0;
 262     }
 263
 264     canjump = 1;
 265     asm volatile( "mtspr 256, %0\n\t"
 266                   "vand 0, 0, 0\n\t"
 267                   :
 268                   : "r"(-1) );
 269     canjump = 0;
 270
 271     signal( SIGILL, oldsig );
 272
 273     return X264_CPU_ALTIVEC;
 274 }
 275 #endif
 276
 277 #elif defined( ARCH_ARM )
 278
 279 void x264_cpu_neon_test();
 280 int x264_cpu_fast_neon_mrc_test();
 281
 282 uint32_t x264_cpu_detect( void )
 283 {
 284     int flags = 0;
 285 #ifdef HAVE_ARMV6
 286     flags |= X264_CPU_ARMV6;
 287
 288     // don't do this hack if compiled with -mfpu=neon
 289 #ifndef HAVE_NEON
 290     static void (* oldsig)( int );
 291     oldsig = signal( SIGILL, sigill_handler );
 292     if( sigsetjmp( jmpbuf, 1 ) )
 293     {
 294         signal( SIGILL, oldsig );
 295         return flags;
 296     }
 297
 298     canjump = 1;
 299     x264_cpu_neon_test();
 300     canjump = 0;
 301     signal( SIGILL, oldsig );
 302 #endif
 303
 304     flags |= X264_CPU_NEON;
 305
 306     // fast neon -> arm (Cortex-A9) detection relies on user access to the
 307     // cycle counter; this assumes ARMv7 performance counters.
 308     // NEON requires at least ARMv7, ARMv8 may require changes here, but
 309     // hopefully this hacky detection method will have been replaced by then.
 310     // Note that there is potential for a race condition if another program or
 311     // x264 instance disables or reinits the counters while x264 is using them,
 312     // which may result in incorrect detection and the counters stuck enabled.
 313     flags |= x264_cpu_fast_neon_mrc_test() ? X264_CPU_FAST_NEON_MRC : 0;
 314     // TODO: write dual issue test? currently it's A8 (dual issue) vs. A9 (fast mrc)
 315 #endif
 316     return flags;
 317 }
 318
 319 #else
 320
 321 uint32_t x264_cpu_detect( void )
 322 {
 323     return 0;
 324 }
 325
 326 #endif
 327
 328 #ifndef HAVE_MMX
 329 void x264_emms( void )
 330 {
 331 }
 332 #endif
 333
 334
 335 int x264_cpu_num_processors( void )
 336 {
 337 #if !defined(HAVE_PTHREAD)
 338     return 1;
 339
 340 #elif defined(_WIN32)
 341     return pthread_num_processors_np();
 342
 343 #elif defined(SYS_LINUX)
 344     unsigned int bit;
 345     int np;
 346     cpu_set_t p_aff;
 347     memset( &p_aff, 0, sizeof(p_aff) );
 348     sched_getaffinity( 0, sizeof(p_aff), &p_aff );
 349     for( np = 0, bit = 0; bit < sizeof(p_aff); bit++ )
 350         np += (((uint8_t *)&p_aff)[bit / 8] >> (bit % 8)) & 1;
 351     return np;
 352
 353 #elif defined(SYS_BEOS)
 354     system_info info;
 355     get_system_info( &info );
 356     return info.cpu_count;
 357
 358 #elif defined(SYS_MACOSX) || defined(SYS_FREEBSD) || defined(SYS_OPENBSD)
 359     int numberOfCPUs;
 360     size_t length = sizeof( numberOfCPUs );
 361 #ifdef SYS_OPENBSD
 362     int mib[2] = { CTL_HW, HW_NCPU };
 363     if( sysctl(mib, 2, &numberOfCPUs, &length, NULL, 0) )
 364 #else
 365     if( sysctlbyname("hw.ncpu", &numberOfCPUs, &length, NULL, 0) )
 366 #endif
 367     {
 368         numberOfCPUs = 1;
 369     }
 370     return numberOfCPUs;
 371
 372 #else
 373     return 1;
 374 #endif
 375 }