#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;

}

*/