/**

  * vim: set ts=4 :

  * =============================================================================

  * SourceMod

  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.

  * =============================================================================

  *

  * This program is free software; you can redistribute it and/or modify it under

  * the terms of the GNU General Public License, version 3.0, as published by the

  * Free Software Foundation.

  * 

  * This program 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

  * this program.  If not, see <http://www.gnu.org/licenses/>.

  *

  * As a special exception, AlliedModders LLC gives you permission to link the

  * code of this program (as well as its derivative works) to "Half-Life 2," the

  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software

  * by the Valve Corporation.  You must obey the GNU General Public License in

  * all respects for all other code used.  Additionally, AlliedModders LLC grants

  * this exception to all derivative works.  AlliedModders LLC defines further

  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),

  * or <http://www.sourcemod.net/license.php>.

  *

  * Version: $Id$

  */

 #include "sm_globals.h"

 #include <vector.h>

 #if SOURCE_ENGINE >= SE_ORANGEBOX

 #include <mathlib.h>

 #else

 #include <math_base.h>

 #endif

 #define SET_VECTOR(addr, vec) \

 	addr[0] = sp_ftoc(vec.x); \

 	addr[1] = sp_ftoc(vec.y); \

 	addr[2] = sp_ftoc(vec.z); 

 static cell_t GetVectorLength(IPluginContext *pContext, const cell_t *params)

 {

 	cell_t *addr;

 	pContext->LocalToPhysAddr(params[1], &addr);

 	Vector source(sp_ctof(addr[0]), sp_ctof(addr[1]), sp_ctof(addr[2]));

 	if (!params[2])

 	{

 		return sp_ftoc(source.Length());

 	} else {

 		return sp_ftoc(source.LengthSqr());

 	}

 }

 static cell_t GetVectorDistance(IPluginContext *pContext, const cell_t *params)

 {

 	cell_t *addr1, *addr2;

 	pContext->LocalToPhysAddr(params[1], &addr1);

 	pContext->LocalToPhysAddr(params[2], &addr2);

 	Vector source(sp_ctof(addr1[0]), sp_ctof(addr1[1]), sp_ctof(addr1[2]));

 	Vector dest(sp_ctof(addr2[0]), sp_ctof(addr2[1]), sp_ctof(addr2[2]));

 	if (!params[3])

 	{

 		return sp_ftoc(source.DistTo(dest));

 	} else {

 		return sp_ftoc(source.DistToSqr(dest));

 	}

 }

 static cell_t GetVectorDotProduct(IPluginContext *pContext, const cell_t *params)

 {

 	cell_t *addr1, *addr2;

 	pContext->LocalToPhysAddr(params[1], &addr1);

 	pContext->LocalToPhysAddr(params[2], &addr2);

 	Vector source(sp_ctof(addr1[0]), sp_ctof(addr1[1]), sp_ctof(addr1[2]));

 	Vector dest(sp_ctof(addr2[0]), sp_ctof(addr2[1]), sp_ctof(addr2[2]));

 	return sp_ftoc(source.Dot(dest));

 }

 static cell_t GetVectorCrossProduct(IPluginContext *pContext, const cell_t *params)

 {

 	cell_t *addr1, *addr2, *set;

 	pContext->LocalToPhysAddr(params[1], &addr1);

 	pContext->LocalToPhysAddr(params[2], &addr2);

 	pContext->LocalToPhysAddr(params[3], &set);

 	Vector vec1(sp_ctof(addr1[0]), sp_ctof(addr1[1]), sp_ctof(addr1[2]));

 	Vector vec2(sp_ctof(addr2[0]), sp_ctof(addr2[1]), sp_ctof(addr2[2]));

 	Vector vec3 = vec1.Cross(vec2);

 	SET_VECTOR(set, vec3);

 	return 1;

 }

 static cell_t GetAngleVectors(IPluginContext *pContext, const cell_t *params)

 {

 	cell_t *ang_addr;

 	pContext->LocalToPhysAddr(params[1], &ang_addr);

 	QAngle angle(sp_ctof(ang_addr[0]), sp_ctof(ang_addr[1]), sp_ctof(ang_addr[2]));

 	Vector fwd, right, up;

 	AngleVectors(angle, &fwd, &right, &up);

 	cell_t *addr_fwd, *addr_right, *addr_up;

 	pContext->LocalToPhysAddr(params[2], &addr_fwd);

 	pContext->LocalToPhysAddr(params[3], &addr_right);

 	pContext->LocalToPhysAddr(params[4], &addr_up);

 	SET_VECTOR(addr_fwd, fwd);

 	SET_VECTOR(addr_right, right);

 	SET_VECTOR(addr_up, up);

 	return 1;

 }

 static cell_t GetVectorAngles(IPluginContext *pContext, const cell_t *params)

 {

 	cell_t *vec_addr;

 	pContext->LocalToPhysAddr(params[1], &vec_addr);

 	Vector vec(sp_ctof(vec_addr[0]), sp_ctof(vec_addr[1]), sp_ctof(vec_addr[2]));

 	QAngle angle;

 	VectorAngles(vec, angle);

 	cell_t *ang_addr;

 	pContext->LocalToPhysAddr(params[2], &ang_addr);

 	SET_VECTOR(ang_addr, angle);

 	return 1;

 }

 static cell_t GetVectorVectors(IPluginContext *pContext, const cell_t *params)

 {

 	cell_t *vec_addr;

 	pContext->LocalToPhysAddr(params[1], &vec_addr);

 	Vector vec(sp_ctof(vec_addr[0]), sp_ctof(vec_addr[1]), sp_ctof(vec_addr[2]));

 	Vector right, up;

 	VectorVectors(vec, right, up);

 	cell_t *addr_right, *addr_up;

 	pContext->LocalToPhysAddr(params[2], &addr_right);

 	pContext->LocalToPhysAddr(params[3], &addr_up);

 	SET_VECTOR(addr_right, right);

 	SET_VECTOR(addr_up, up);

 	return 1;

 }

 static cell_t NormalizeVector(IPluginContext *pContext, const cell_t *params)

 {

 	cell_t *addr;

 	pContext->LocalToPhysAddr(params[1], &addr);

 	Vector source(sp_ctof(addr[0]), sp_ctof(addr[1]), sp_ctof(addr[2]));

 	float length = VectorNormalize(source);

 	pContext->LocalToPhysAddr(params[2], &addr);

 	SET_VECTOR(addr, source);

 	return sp_ftoc(length);

 }

 REGISTER_NATIVES(vectorNatives)

 {

 	{"GetAngleVectors",			GetAngleVectors},

 	{"GetVectorAngles",			GetVectorAngles},

 	{"GetVectorCrossProduct",	GetVectorCrossProduct},

 	{"GetVectorDistance",		GetVectorDistance},

 	{"GetVectorDotProduct",		GetVectorDotProduct},

 	{"GetVectorLength",			GetVectorLength},

 	{"GetVectorVectors",		GetVectorVectors},

 	{"NormalizeVector",			NormalizeVector},

 	{NULL,					NULL}

 };