1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
#include "Util.h"
#if defined(_MSC_VER) && (_MSC_VER >= 1310)
#include <intrin.h>
#endif
std::string StringReplace(const std::string& text,
const std::string& from,
const std::string& to)
{
std::string working = text;
std::string::size_type pos = 0;
while (true) {
pos = working.find(from, pos);
if (pos == std::string::npos) {
break;
}
std::string tmp = working.substr(0, pos);
tmp += to;
tmp += working.substr(pos + from.size(), std::string::npos);
pos += to.size();
working = tmp;
}
return working;
}
namespace proc {
#if defined(__GNUC__)
// function inlining breaks this
__attribute__((__noinline__))
void ExecCPUID(unsigned int* a, unsigned int* b, unsigned int* c, unsigned int* d)
{
#ifndef __APPLE__
__asm__ __volatile__(
"cpuid"
: "=a" (*a), "=b" (*b), "=c" (*c), "=d" (*d)
: "0" (*a)
);
#else
#ifdef __x86_64__
__asm__ __volatile__(
"pushq %%rbx\n\t"
"cpuid\n\t"
"movl %%ebx, %1\n\t"
"popq %%rbx"
: "=a" (*a), "=r" (*b), "=c" (*c), "=d" (*d)
: "0" (*a)
);
#else
__asm__ __volatile__(
"pushl %%ebx\n\t"
"cpuid\n\t"
"movl %%ebx, %1\n\t"
"popl %%ebx"
: "=a" (*a), "=r" (*b), "=c" (*c), "=d" (*d)
: "0" (*a)
);
#endif
#endif
}
#elif defined(_MSC_VER) && (_MSC_VER >= 1310)
void ExecCPUID(unsigned int* a, unsigned int* b, unsigned int* c, unsigned int* d)
{
int features[4];
__cpuid(features, *a);
*a=features[0];
*b=features[1];
*c=features[2];
*d=features[3];
}
#else
// no-op on other compilers
void ExecCPUID(unsigned int* a, unsigned int* b, unsigned int* c, unsigned int* d)
{
}
#endif
unsigned int GetProcMaxStandardLevel()
{
unsigned int rEAX = 0x00000000;
unsigned int rEBX = 0;
unsigned int rECX = 0;
unsigned int rEDX = 0;
ExecCPUID(&rEAX, &rEBX, &rECX, &rEDX);
return rEAX;
}
unsigned int GetProcMaxExtendedLevel()
{
unsigned int rEAX = 0x80000000;
unsigned int rEBX = 0;
unsigned int rECX = 0;
unsigned int rEDX = 0;
ExecCPUID(&rEAX, &rEBX, &rECX, &rEDX);
return rEAX;
}
unsigned int GetProcSSEBits()
{
unsigned int rEAX = 0;
unsigned int rEBX = 0;
unsigned int rECX = 0;
unsigned int rEDX = 0;
unsigned int bits = 0;
if (GetProcMaxStandardLevel() >= 0x00000001U) {
rEAX = 0x00000001U; ExecCPUID(&rEAX, &rEBX, &rECX, &rEDX);
int SSE42 = (rECX >> 20) & 1; bits |= ( SSE42 << 0); // SSE 4.2
int SSE41 = (rECX >> 19) & 1; bits |= ( SSE41 << 1); // SSE 4.1
int SSSE30 = (rECX >> 9) & 1; bits |= (SSSE30 << 2); // Supplemental SSE 3.0
int SSE30 = (rECX >> 0) & 1; bits |= ( SSE30 << 3); // SSE 3.0
int SSE20 = (rEDX >> 26) & 1; bits |= ( SSE20 << 4); // SSE 2.0
int SSE10 = (rEDX >> 25) & 1; bits |= ( SSE10 << 5); // SSE 1.0
int MMX = (rEDX >> 23) & 1; bits |= ( MMX << 6); // MMX
}
if (GetProcMaxExtendedLevel() >= 0x80000001U) {
rEAX = 0x80000001U; ExecCPUID(&rEAX, &rEBX, &rECX, &rEDX);
int SSE50A = (rECX >> 11) & 1; bits |= (SSE50A << 7); // SSE 5.0A
int SSE40A = (rECX >> 6) & 1; bits |= (SSE40A << 8); // SSE 4.0A
int MSSE = (rECX >> 7) & 1; bits |= (MSSE << 9); // Misaligned SSE
}
return bits;
}
}
|
