#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;
}
*/