/*  PCSX2 - PS2 Emulator for PCs
 *  Copyright (C) 2002-2010  PCSX2 Dev Team
 *
 *  PCSX2 is free software: you can redistribute it and/or modify it under the terms
 *  of the GNU Lesser General Public License as published by the Free Software Found-
 *  ation, either version 3 of the License, or (at your option) any later version.
 *
 *  PCSX2 is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 *  without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
 *  PURPOSE.  See the GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License along with PCSX2.
 *  If not, see <http://www.gnu.org/licenses/>.
 */

#pragma once

#include "x86emitter.h"

enum x86VendorType
{
	x86Vendor_Intel=0,
	x86Vendor_AMD,
	x86Vendor_Unknown,
};

// --------------------------------------------------------------------------------------
//  x86capabilities
// --------------------------------------------------------------------------------------
class x86capabilities
{
public:
	bool isIdentified;
	u32 LogicalCoresPerPhysicalCPU;
	u32 PhysicalCoresPerPhysicalCPU;

public:
	x86VendorType VendorID;

	uint FamilyID;		// Processor Family
	uint Model;			// Processor Model
	uint TypeID;		// Processor Type
	uint StepID;		// Stepping ID

	u32 Flags;			// Feature Flags
	u32 Flags2;			// More Feature Flags
	u32 EFlags;			// Extended Feature Flags
	u32 EFlags2;		// Extended Feature Flags pg2

	char VendorName[16];	// Vendor/Creator ID
	char FamilyName[50];    // the original cpu name

	// ----------------------------------------------------------------------------
	//   x86 CPU Capabilities Section (all boolean flags!)
	// ----------------------------------------------------------------------------

	u32 hasFloatingPointUnit						:1;
	u32 hasVirtual8086ModeEnhancements				:1;
	u32 hasDebuggingExtensions						:1;
	u32 hasPageSizeExtensions						:1;
	u32 hasTimeStampCounter							:1;
	u32 hasModelSpecificRegisters					:1;
	u32 hasPhysicalAddressExtension					:1;
	u32 hasCOMPXCHG8BInstruction					:1;
	u32 hasAdvancedProgrammableInterruptController	:1;
	u32 hasSEPFastSystemCall						:1;
	u32 hasMemoryTypeRangeRegisters					:1;
	u32 hasPTEGlobalFlag							:1;
	u32 hasMachineCheckArchitecture					:1;
	u32 hasConditionalMoveAndCompareInstructions	:1;
	u32 hasFGPageAttributeTable						:1;
	u32 has36bitPageSizeExtension					:1;
	u32 hasProcessorSerialNumber					:1;
	u32 hasCFLUSHInstruction						:1;
	u32 hasDebugStore								:1;
	u32 hasACPIThermalMonitorAndClockControl		:1;
	u32 hasMultimediaExtensions						:1;
	u32 hasFastStreamingSIMDExtensionsSaveRestore	:1;
	u32 hasStreamingSIMDExtensions					:1;
	u32 hasStreamingSIMD2Extensions					:1;
	u32 hasSelfSnoop								:1;

	// is TRUE for both multi-core and Hyperthreaded CPUs.
	u32 hasMultiThreading							:1;

	u32 hasThermalMonitor							:1;
	u32 hasIntel64BitArchitecture					:1;
	u32 hasStreamingSIMD3Extensions					:1;
	u32 hasSupplementalStreamingSIMD3Extensions		:1;
	u32 hasStreamingSIMD4Extensions					:1;
	u32 hasStreamingSIMD4Extensions2				:1;

	// AMD-specific CPU Features
	u32 hasMultimediaExtensionsExt					:1;
	u32 hasAMD64BitArchitecture						:1;
	u32 has3DNOWInstructionExtensionsExt			:1;
	u32 has3DNOWInstructionExtensions				:1;
	u32 hasStreamingSIMD4ExtensionsA				:1;

	// Core Counts!
	u32 PhysicalCores;
	u32 LogicalCores;

public:
	x86capabilities()
	{
		isIdentified = false;
		VendorID = x86Vendor_Unknown;
		LogicalCoresPerPhysicalCPU = 1;
		PhysicalCoresPerPhysicalCPU = 1;
	}

	void Identify();
	void CountCores();
	wxString GetTypeName() const;

	u32 CalculateMHz() const;

	void SIMD_ExceptionTest();
	void SIMD_EstablishMXCSRmask();

protected:
	s64 _CPUSpeedHz( u64 time ) const;
	void CountLogicalCores();
};

enum SSE_RoundMode
{
	SSE_RoundMode_FIRST = 0,
	SSEround_Nearest = 0,
	SSEround_NegInf,
	SSEround_PosInf,
	SSEround_Chop,
	SSE_RoundMode_COUNT
};

ImplementEnumOperators( SSE_RoundMode );

// --------------------------------------------------------------------------------------
//  SSE_MXCSR  -  Control/Status Register (bitfield)
// --------------------------------------------------------------------------------------
// Bits 0-5 are exception flags; used only if SSE exceptions have been enabled.
//   Bits in this field are "sticky" and, once an exception has occured, must be manually
//   cleared using LDMXCSR or FXRSTOR.
//
// Bits 7-12 are the masks for disabling the exceptions in bits 0-5.  Cleared bits allow
//   exceptions, set bits mask exceptions from being raised.
//
union SSE_MXCSR
{
	u32		bitmask;
	struct
	{
		u32
			InvalidOpFlag		:1,
			DenormalFlag		:1,
			DivideByZeroFlag	:1,
			OverflowFlag		:1,
			UnderflowFlag		:1,
			PrecisionFlag		:1,

			// This bit is supported only on SSE2 or better CPUs.  Setting it to 1 on
			// SSE1 cpus will result in an invalid instruction exception when executing
			// LDMXSCR.
			DenormalsAreZero	:1,

			InvalidOpMask		:1,
			DenormalMask		:1,
			DivideByZeroMask	:1,
			OverflowMask		:1,
			UnderflowMask		:1,
			PrecisionMask		:1,

			RoundingControl		:2,
			FlushToZero			:1;
	};

	SSE_RoundMode GetRoundMode() const;
	SSE_MXCSR& SetRoundMode( SSE_RoundMode mode );
	SSE_MXCSR& ClearExceptionFlags();
	SSE_MXCSR& EnableExceptions();
	SSE_MXCSR& DisableExceptions();

	SSE_MXCSR& ApplyReserveMask();

	bool operator ==( const SSE_MXCSR& right ) const
	{
		return bitmask == right.bitmask;
	}

	bool operator !=( const SSE_MXCSR& right ) const
	{
		return bitmask != right.bitmask;
	}

	operator x86Emitter::xIndirect32() const;
};

extern SSE_MXCSR	MXCSR_Mask;


extern __aligned16 x86capabilities x86caps;