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