#include "tsystem.h"

#ifdef _WIN32

#include <wtypes.h></wtypes.h>

#include <winnt.h></winnt.h>

#include <emmintrin.h></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;

}

*/