intern/cycles/util/util_system.cpp

   1 /*
   2  * Copyright 2011, Blender Foundation.
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License
   6  * as published by the Free Software Foundation; either version 2
   7  * of the License, or (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software Foundation,
  16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  17  */
  18
  19 #include "util_system.h"
  20 #include "util_types.h"
  21
  22 #ifdef _WIN32
  23 #if(!defined(FREE_WINDOWS))
  24 #include <intrin.h>
  25 #endif
  26 #include <windows.h>
  27 #elif defined(__APPLE__)
  28 #include <sys/sysctl.h>
  29 #include <sys/types.h>
  30 #else
  31 #include <unistd.h>
  32 #endif
  33
  34 CCL_NAMESPACE_BEGIN
  35
  36 int system_cpu_thread_count()
  37 {
  38         static uint count = 0;
  39
  40         if(count > 0)
  41                 return count;
  42
  43 #ifdef _WIN32
  44         SYSTEM_INFO info;
  45         GetSystemInfo(&info);
  46         count = (uint)info.dwNumberOfProcessors;
  47 #elif defined(__APPLE__)
  48         size_t len = sizeof(count);
  49         int mib[2] = { CTL_HW, HW_NCPU };
  50
  51         sysctl(mib, 2, &count, &len, NULL, 0);
  52 #else
  53         count = (uint)sysconf(_SC_NPROCESSORS_ONLN);
  54 #endif
  55
  56         if(count < 1)
  57                 count = 1;
  58
  59         return count;
  60 }
  61
  62 #if !defined(_WIN32) || defined(FREE_WINDOWS)
  63 static void __cpuid(int data[4], int selector)
  64 {
  65 #ifdef __x86_64__
  66         asm("cpuid" : "=a" (data[0]), "=b" (data[1]), "=c" (data[2]), "=d" (data[3]) : "a"(selector));
  67 #else
  68 #ifdef __i386__
  69         asm("pushl %%ebx    \n\t"
  70                 "cpuid          \n\t"
  71                 "movl %%ebx, %1 \n\t"
  72                 "popl %%ebx     \n\t" : "=a" (data[0]), "=r" (data[1]), "=c" (data[2]), "=d" (data[3]) : "a"(selector));
  73 #else
  74         data[0] = data[1] = data[2] = data[3] = 0;
  75 #endif
  76 #endif
  77 }
  78 #endif
  79
  80 static void replace_string(string& haystack, const string& needle, const string& other)
  81 {
  82         size_t i;
  83
  84         while((i = haystack.find(needle)) != string::npos)
  85                 haystack.replace(i, needle.length(), other);
  86 }
  87
  88 string system_cpu_brand_string()
  89 {
  90         char buf[48];
  91         int result[4];
  92
  93         __cpuid(result, 0x80000000);
  94
  95         if(result[0] >= (int)0x80000004) {
  96                 __cpuid((int*)(buf+0), 0x80000002);
  97                 __cpuid((int*)(buf+16), 0x80000003);
  98                 __cpuid((int*)(buf+32), 0x80000004);
  99
 100                 string brand = buf;
 101
 102                 /* make it a bit more presentable */
 103                 replace_string(brand, "(TM)", "");
 104                 replace_string(brand, "(R)", "");
 105
 106                 size_t i;
 107                 if((i = brand.find("  ")) != string::npos)
 108                         brand = brand.substr(0, i);
 109
 110                 return brand;
 111         }
 112
 113         return "Unknown CPU";
 114 }
 115
 116 int system_cpu_bits()
 117 {
 118         return (sizeof(void*)*8);
 119 }
 120
 121 #if defined(__x86_64__) || defined(_M_X64) || defined(i386) || defined(_M_IX86)
 122
 123 struct CPUCapabilities {
 124         bool x64;
 125         bool mmx;
 126         bool sse;
 127         bool sse2;
 128         bool sse3;
 129         bool ssse3;
 130         bool sse41;
 131         bool sse42;
 132         bool sse4a;
 133         bool avx;
 134         bool xop;
 135         bool fma3;
 136         bool fma4;
 137 };
 138
 139 bool system_cpu_support_optimized()
 140 {
 141         static CPUCapabilities caps;
 142         static bool caps_init = false;
 143
 144         if(!caps_init) {
 145                 int result[4], num, num_ex;
 146
 147                 memset(&caps, 0, sizeof(caps));
 148
 149                 __cpuid(result, 0);
 150                 num = result[0];
 151
 152                 __cpuid(result, 0x80000000);
 153                 num_ex = result[0];
 154
 155                 if(num >= 1){
 156                         __cpuid(result, 0x00000001);
 157                         caps.mmx = (result[3] & ((int)1 << 23)) != 0;
 158                         caps.sse = (result[3] & ((int)1 << 25)) != 0;
 159                         caps.sse2 = (result[3] & ((int)1 << 26)) != 0;
 160                         caps.sse3 = (result[2] & ((int)1 <<  0)) != 0;
 161
 162                         caps.ssse3 = (result[2] & ((int)1 <<  9)) != 0;
 163                         caps.sse41 = (result[2] & ((int)1 << 19)) != 0;
 164                         caps.sse42 = (result[2] & ((int)1 << 20)) != 0;
 165
 166                         caps.avx = (result[2] & ((int)1 << 28)) != 0;
 167                         caps.fma3 = (result[2] & ((int)1 << 12)) != 0;
 168                 }
 169
 170                 /*if(num_ex >= 0x80000001){
 171                         __cpuid(result, 0x80000001);
 172                         caps.x64 = (result[3] & ((int)1 << 29)) != 0;
 173                         caps.sse4a = (result[2] & ((int)1 <<  6)) != 0;
 174                         caps.fma4 = (result[2] & ((int)1 << 16)) != 0;
 175                         caps.xop = (result[2] & ((int)1 << 11)) != 0;
 176                 }*/
 177
 178                 caps_init = true;
 179         }
 180
 181         /* optimization flags use these */
 182         return caps.sse && caps.sse2 && caps.sse3;
 183 }
 184
 185 #else
 186
 187 bool system_cpu_support_optimized()
 188 {
 189         return false;
 190 }
 191
 192 #endif
 193
 194 CCL_NAMESPACE_END
 195