179 lines
3.2 KiB
C++
179 lines
3.2 KiB
C++
|
|
||
|
|
||
|
#include "tsystem.h"
|
||
|
#ifdef WIN32
|
||
|
#include <wtypes.h>
|
||
|
#include <winnt.h>
|
||
|
#include <emmintrin.h>
|
||
|
#endif
|
||
|
|
||
|
using namespace TSystem;
|
||
|
|
||
|
#ifdef x64
|
||
|
long TSystem::getCPUExtensions()
|
||
|
{
|
||
|
return TSystem::CpuSupportsSse | TSystem::CpuSupportsSse2; //TSystem::CPUExtensionsNone
|
||
|
}
|
||
|
|
||
|
#else
|
||
|
#ifndef WIN32
|
||
|
long TSystem::getCPUExtensions()
|
||
|
{
|
||
|
return TSystem::CPUExtensionsNone;
|
||
|
}
|
||
|
#else
|
||
|
namespace
|
||
|
{
|
||
|
|
||
|
long CPUExtensionsAvailable = TSystem::CPUExtensionsNone;
|
||
|
bool CPUExtensionsEnabled = true;
|
||
|
bool FistTime = true;
|
||
|
|
||
|
//#ifdef WIN32
|
||
|
|
||
|
//------------------------------------------------------------------------------
|
||
|
|
||
|
long CPUCheckForSSESupport()
|
||
|
{
|
||
|
__try {
|
||
|
// __asm andps xmm0,xmm0
|
||
|
|
||
|
__asm _emit 0x0f __asm _emit 0x54 __asm _emit 0xc0
|
||
|
|
||
|
} __except (EXCEPTION_EXECUTE_HANDLER) {
|
||
|
if (_exception_code() == STATUS_ILLEGAL_INSTRUCTION)
|
||
|
CPUExtensionsAvailable &= ~(CpuSupportsSse | CpuSupportsSse2);
|
||
|
}
|
||
|
|
||
|
return CPUExtensionsAvailable;
|
||
|
}
|
||
|
|
||
|
//------------------------------------------------------------------------------
|
||
|
|
||
|
long __declspec(naked) CPUCheckForExtensions()
|
||
|
{
|
||
|
__asm {
|
||
|
push ebp
|
||
|
push edi
|
||
|
push esi
|
||
|
push ebx
|
||
|
|
||
|
xor ebp,ebp ;cpu flags - if we don't have CPUID, we probably
|
||
|
;won't want to try FPU optimizations.
|
||
|
|
||
|
;check for CPUID.
|
||
|
|
||
|
pushfd ;flags -> EAX
|
||
|
pop eax
|
||
|
or eax,00200000h ;set the ID bit
|
||
|
push eax ;EAX -> flags
|
||
|
popfd
|
||
|
pushfd ;flags -> EAX
|
||
|
pop eax
|
||
|
and eax,00200000h ;ID bit set?
|
||
|
jz done ;nope...
|
||
|
|
||
|
;CPUID exists, check for features register.
|
||
|
|
||
|
mov ebp,00000003h
|
||
|
xor eax,eax
|
||
|
cpuid
|
||
|
or eax,eax
|
||
|
jz done ;no features register?!?
|
||
|
|
||
|
;features register exists, look for MMX, SSE, SSE2.
|
||
|
|
||
|
mov eax,1
|
||
|
cpuid
|
||
|
mov ebx,edx
|
||
|
and ebx,00800000h ;MMX is bit 23
|
||
|
shr ebx,21
|
||
|
or ebp,ebx ;set bit 2 if MMX exists
|
||
|
|
||
|
mov ebx,edx
|
||
|
and edx,02000000h ;SSE is bit 25
|
||
|
shr edx,25
|
||
|
neg edx
|
||
|
and edx,00000018h ;set bits 3 and 4 if SSE exists
|
||
|
or ebp,edx
|
||
|
|
||
|
and ebx,04000000h ;SSE2 is bit 26
|
||
|
shr ebx,21
|
||
|
and ebx,00000020h ;set bit 5
|
||
|
or ebp,ebx
|
||
|
|
||
|
;check for vendor feature register (K6/Athlon).
|
||
|
|
||
|
mov eax,80000000h
|
||
|
cpuid
|
||
|
mov ecx,80000001h
|
||
|
cmp eax,ecx
|
||
|
jb done
|
||
|
|
||
|
;vendor feature register exists, look for 3DNow! and Athlon extensions
|
||
|
|
||
|
mov eax,ecx
|
||
|
cpuid
|
||
|
|
||
|
mov eax,edx
|
||
|
and edx,80000000h ;3DNow! is bit 31
|
||
|
shr edx,25
|
||
|
or ebp,edx ;set bit 6
|
||
|
|
||
|
mov edx,eax
|
||
|
and eax,40000000h ;3DNow!2 is bit 30
|
||
|
shr eax,23
|
||
|
or ebp,eax ;set bit 7
|
||
|
|
||
|
and edx,00400000h ;AMD MMX extensions (integer SSE) is bit 22
|
||
|
shr edx,19
|
||
|
or ebp,edx
|
||
|
|
||
|
done:
|
||
|
mov eax,ebp
|
||
|
mov CPUExtensionsAvailable, ebp
|
||
|
|
||
|
;Full SSE and SSE-2 require OS support for the xmm* registers.
|
||
|
|
||
|
test eax,00000030h
|
||
|
jz nocheck
|
||
|
call CPUCheckForSSESupport
|
||
|
nocheck:
|
||
|
pop ebx
|
||
|
pop esi
|
||
|
pop edi
|
||
|
pop ebp
|
||
|
ret
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//#endif
|
||
|
//#endif
|
||
|
} // anonymous namespace
|
||
|
|
||
|
//------------------------------------------------------------------------------
|
||
|
|
||
|
long TSystem::getCPUExtensions()
|
||
|
{
|
||
|
|
||
|
if (FistTime) {
|
||
|
CPUCheckForExtensions();
|
||
|
FistTime = false;
|
||
|
}
|
||
|
|
||
|
if (CPUExtensionsEnabled)
|
||
|
return CPUExtensionsAvailable;
|
||
|
else
|
||
|
return TSystem::CPUExtensionsNone;
|
||
|
}
|
||
|
|
||
|
#endif
|
||
|
#endif
|
||
|
//------------------------------------------------------------------------------
|
||
|
/*
|
||
|
void TSystem::enableCPUExtensions(bool on)
|
||
|
{
|
||
|
CPUExtensionsEnabled = on;
|
||
|
}
|
||
|
*/
|