#include "util.h"

void * GetObjectAddr(HookParamType type, unsigned int flags, void **params, size_t offset)
{
#ifdef  WIN32
	if (type == HookParamType_Object)
		return (void *)((intptr_t)params + offset);
#elif POSIX
	if (type == HookParamType_Object && !(flags & PASSFLAG_ODTOR)) //Objects are passed by rrefrence if they contain destructors.
		return (void *)((intptr_t)params + offset);
#endif
	return *(void **)((intptr_t)params + offset);

}

size_t GetStackParamOffset(HookParamsStruct *paramStruct, unsigned int index)
{
	assert(paramStruct->dg->params[index].custom_register == None);

	size_t offset = 0;
	for (unsigned int i = 0; i < index; i++)
	{
		// Only care for arguments on the stack before us.
		if (paramStruct->dg->params[i].custom_register != None)
			continue;

#ifndef WIN32
		if (paramStruct->dg->params[i].type == HookParamType_Object && (paramStruct->dg->params[i].flags & PASSFLAG_ODTOR)) //Passed by refrence
		{
			offset += sizeof(void *);
			continue;
		}
#endif
		offset += paramStruct->dg->params[i].size;
	}
	return offset;
}

size_t GetRegisterParamOffset(HookParamsStruct *paramStruct, unsigned int index)
{
	// TODO: Fix this up and get a pointer to the CDetour
	assert(paramStruct->dg->params[index].custom_register != None);

	// Need to get the size of the stack arguments first. Register arguments are stored after them in the buffer.
	size_t stackSize = 0;
	for (int i = paramStruct->dg->params.size() - 1; i >= 0; i--)
	{
		if (paramStruct->dg->params[i].custom_register == None)
			stackSize += paramStruct->dg->params[i].size;
	}

	size_t offset = stackSize;
	for (unsigned int i = 0; i < index; i++)
	{
		// Only care for arguments passed through a register as well before us.
		if (paramStruct->dg->params[i].custom_register == None)
			continue;

		offset += paramStruct->dg->params[i].size;
	}
	return offset;
}

size_t GetParamOffset(HookParamsStruct *paramStruct, unsigned int index)
{
	if (paramStruct->dg->params[index].custom_register == None)
		return GetStackParamOffset(paramStruct, index);
	else
		return GetRegisterParamOffset(paramStruct, index);
}

size_t GetParamTypeSize(HookParamType type)
{
	return sizeof(void *);
}

size_t GetParamsSize(DHooksCallback *dg)//Get the full size, this is for creating the STACK.
{
	size_t res = 0;

	for (int i = dg->params.size() - 1; i >= 0; i--)
	{
		res += dg->params.at(i).size;
	}

	return res;
}

DataType_t DynamicHooks_ConvertParamTypeFrom(HookParamType type)
{
	switch (type)
	{
	case HookParamType_Int:
		return DATA_TYPE_INT;
	case HookParamType_Bool:
		return DATA_TYPE_BOOL;
	case HookParamType_Float:
		return DATA_TYPE_FLOAT;
	case HookParamType_StringPtr:
	case HookParamType_CharPtr:
	case HookParamType_VectorPtr:
	case HookParamType_CBaseEntity:
	case HookParamType_ObjectPtr:
	case HookParamType_Edict:
		return DATA_TYPE_POINTER;
	case HookParamType_Object:
		return DATA_TYPE_OBJECT;
	default:
		smutils->LogError(myself, "Unhandled parameter type %d!", type);
	}

	return DATA_TYPE_POINTER;
}

DataType_t DynamicHooks_ConvertReturnTypeFrom(ReturnType type)
{
	switch (type)
	{
	case ReturnType_Void:
		return DATA_TYPE_VOID;
	case ReturnType_Int:
		return DATA_TYPE_INT;
	case ReturnType_Bool:
		return DATA_TYPE_BOOL;
	case ReturnType_Float:
		return DATA_TYPE_FLOAT;
	case ReturnType_StringPtr:
	case ReturnType_CharPtr:
	case ReturnType_VectorPtr:
	case ReturnType_CBaseEntity:
	case ReturnType_Edict:
		return DATA_TYPE_POINTER;
	case ReturnType_Vector:
		return DATA_TYPE_OBJECT;
	default:
		smutils->LogError(myself, "Unhandled return type %d!", type);
	}

	return DATA_TYPE_VOID;
}

Register_t DynamicHooks_ConvertRegisterFrom(PluginRegister reg)
{
	switch (reg)
	{
	case DHookRegister_Default:
		return None;

	case DHookRegister_AL:
		return AL;
	case DHookRegister_CL:
		return CL;
	case DHookRegister_DL:
		return DL;
	case DHookRegister_BL:
		return BL;
	case DHookRegister_AH:
		return AH;
	case DHookRegister_CH:
		return CH;
	case DHookRegister_DH:
		return DH;
	case DHookRegister_BH:
		return BH;

	case DHookRegister_EAX:
		return EAX;
	case DHookRegister_ECX:
		return ECX;
	case DHookRegister_EDX:
		return EDX;
	case DHookRegister_EBX:
		return EBX;
	case DHookRegister_ESP:
		return ESP;
	case DHookRegister_EBP:
		return EBP;
	case DHookRegister_ESI:
		return ESI;
	case DHookRegister_EDI:
		return EDI;

	case DHookRegister_XMM0:
		return XMM0;
	case DHookRegister_XMM1:
		return XMM1;
	case DHookRegister_XMM2:
		return XMM2;
	case DHookRegister_XMM3:
		return XMM3;
	case DHookRegister_XMM4:
		return XMM4;
	case DHookRegister_XMM5:
		return XMM5;
	case DHookRegister_XMM6:
		return XMM6;
	case DHookRegister_XMM7:
		return XMM7;

	case DHookRegister_ST0:
		return ST0;
	}

	return None;
}