dvander@1069: /**
dvander@1069:  * vim: set ts=4 :
ds@1330:  * =============================================================================
dvander@1231:  * SourceMod
ds@1947:  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
ds@1330:  * =============================================================================
dvander@1069:  *
ds@1330:  * This program is free software; you can redistribute it and/or modify it under
ds@1330:  * the terms of the GNU General Public License, version 3.0, as published by the
ds@1330:  * Free Software Foundation.
dvander@1231:  * 
ds@1330:  * This program is distributed in the hope that it will be useful, but WITHOUT
ds@1330:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
ds@1330:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
ds@1330:  * details.
dvander@1231:  *
ds@1330:  * You should have received a copy of the GNU General Public License along with
ds@1330:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@1231:  *
ds@1330:  * As a special exception, AlliedModders LLC gives you permission to link the
ds@1330:  * code of this program (as well as its derivative works) to "Half-Life 2," the
ds@1330:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
ds@1330:  * by the Valve Corporation.  You must obey the GNU General Public License in
ds@1330:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
ds@1330:  * this exception to all derivative works.  AlliedModders LLC defines further
ds@1330:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
ds@1330:  * or <http://www.sourcemod.net/license.php>.
dvander@1069:  *
dvander@1069:  * Version: $Id$
dvander@1069:  */
dvander@1069: 
dvander@1069: #include "sm_globals.h"
dvander@1069: #include <vector.h>
pred@1871: #if defined ORANGEBOX_BUILD
pred@1871: #include <mathlib.h>
pred@1871: #else
pred@1871: #include <math_base.h>
pred@1871: #endif
dvander@1069: 
dvander@1069: #define SET_VECTOR(addr, vec) \
dvander@1070: 	addr[0] = sp_ftoc(vec.x); \
dvander@1070: 	addr[1] = sp_ftoc(vec.y); \
dvander@1070: 	addr[2] = sp_ftoc(vec.z); 
dvander@1069: 
dvander@1069: static cell_t GetVectorLength(IPluginContext *pContext, const cell_t *params)
dvander@1069: {
dvander@1069: 	cell_t *addr;
dvander@1069: 
dvander@1069: 	pContext->LocalToPhysAddr(params[1], &addr);
dvander@1069: 
dvander@1069: 	Vector source(sp_ctof(addr[0]), sp_ctof(addr[1]), sp_ctof(addr[2]));
dvander@1069: 	
dvander@1069: 	if (!params[2])
dvander@1069: 	{
dvander@1069: 		return sp_ftoc(source.Length());
dvander@1069: 	} else {
dvander@1069: 		return sp_ftoc(source.LengthSqr());
dvander@1069: 	}
dvander@1069: }
dvander@1069: 
dvander@1069: static cell_t GetVectorDistance(IPluginContext *pContext, const cell_t *params)
dvander@1069: {
dvander@1069: 	cell_t *addr1, *addr2;
dvander@1069: 
dvander@1069: 	pContext->LocalToPhysAddr(params[1], &addr1);
dvander@1069: 	pContext->LocalToPhysAddr(params[2], &addr2);
dvander@1069: 
dvander@1069: 	Vector source(sp_ctof(addr1[0]), sp_ctof(addr1[1]), sp_ctof(addr1[2]));
dvander@1069: 	Vector dest(sp_ctof(addr2[0]), sp_ctof(addr2[1]), sp_ctof(addr2[2]));
dvander@1069: 
dvander@1069: 	if (!params[3])
dvander@1069: 	{
dvander@1069: 		return sp_ftoc(source.DistTo(dest));
dvander@1069: 	} else {
dvander@1069: 		return sp_ftoc(source.DistToSqr(dest));
dvander@1069: 	}
dvander@1069: }
dvander@1069: 
dvander@1069: static cell_t GetVectorDotProduct(IPluginContext *pContext, const cell_t *params)
dvander@1069: {
dvander@1069: 	cell_t *addr1, *addr2;
dvander@1069: 
dvander@1069: 	pContext->LocalToPhysAddr(params[1], &addr1);
dvander@1069: 	pContext->LocalToPhysAddr(params[2], &addr2);
dvander@1069: 
dvander@1069: 	Vector source(sp_ctof(addr1[0]), sp_ctof(addr1[1]), sp_ctof(addr1[2]));
dvander@1069: 	Vector dest(sp_ctof(addr2[0]), sp_ctof(addr2[1]), sp_ctof(addr2[2]));
dvander@1069: 
dvander@1069: 	return sp_ftoc(source.Dot(dest));
dvander@1069: }
dvander@1069: 
dvander@1069: static cell_t GetVectorCrossProduct(IPluginContext *pContext, const cell_t *params)
dvander@1069: {
dvander@1069: 	cell_t *addr1, *addr2, *set;
dvander@1069: 
dvander@1069: 	pContext->LocalToPhysAddr(params[1], &addr1);
dvander@1069: 	pContext->LocalToPhysAddr(params[2], &addr2);
dvander@1069: 	pContext->LocalToPhysAddr(params[3], &set);
dvander@1069: 
dvander@1069: 	Vector vec1(sp_ctof(addr1[0]), sp_ctof(addr1[1]), sp_ctof(addr1[2]));
dvander@1069: 	Vector vec2(sp_ctof(addr2[0]), sp_ctof(addr2[1]), sp_ctof(addr2[2]));
dvander@1069: 
dvander@1069: 	Vector vec3 = vec1.Cross(vec2);
dvander@1069: 
dvander@1069: 	SET_VECTOR(set, vec3);
dvander@1069: 
dvander@1069: 	return 1;
dvander@1069: }
dvander@1069: 
dvander@1069: static cell_t GetAngleVectors(IPluginContext *pContext, const cell_t *params)
dvander@1069: {
dvander@1069: 	cell_t *ang_addr;
dvander@1069: 
dvander@1069: 	pContext->LocalToPhysAddr(params[1], &ang_addr);
dvander@1069: 
dvander@1069: 	QAngle angle(sp_ctof(ang_addr[0]), sp_ctof(ang_addr[1]), sp_ctof(ang_addr[2]));
dvander@1069: 	Vector fwd, right, up;
dvander@1069: 
dvander@1069: 	AngleVectors(angle, &fwd, &right, &up);
dvander@1069: 
dvander@1069: 	cell_t *addr_fwd, *addr_right, *addr_up;
dvander@1069: 	pContext->LocalToPhysAddr(params[2], &addr_fwd);
dvander@1069: 	pContext->LocalToPhysAddr(params[3], &addr_right);
dvander@1069: 	pContext->LocalToPhysAddr(params[4], &addr_up);
dvander@1069: 
dvander@1069: 	SET_VECTOR(addr_fwd, fwd);
dvander@1069: 	SET_VECTOR(addr_right, right);
dvander@1069: 	SET_VECTOR(addr_up, up);
dvander@1069: 
dvander@1069: 	return 1;
dvander@1069: }
dvander@1069: 
dvander@1069: static cell_t GetVectorAngles(IPluginContext *pContext, const cell_t *params)
dvander@1069: {
dvander@1069: 	cell_t *vec_addr;
dvander@1069: 
dvander@1069: 	pContext->LocalToPhysAddr(params[1], &vec_addr);
dvander@1069: 
dvander@1069: 	Vector vec(sp_ctof(vec_addr[0]), sp_ctof(vec_addr[1]), sp_ctof(vec_addr[2]));
dvander@1069: 	QAngle angle;
dvander@1069: 
dvander@1069: 	VectorAngles(vec, angle);
dvander@1069: 
dvander@1069: 	cell_t *ang_addr;
dvander@1069: 	pContext->LocalToPhysAddr(params[2], &ang_addr);
dvander@1069: 
dvander@1069: 	SET_VECTOR(ang_addr, angle);
dvander@1069: 
dvander@1069: 	return 1;
dvander@1069: }
dvander@1069: 
dvander@1069: static cell_t GetVectorVectors(IPluginContext *pContext, const cell_t *params)
dvander@1069: {
dvander@1069: 	cell_t *vec_addr;
dvander@1069: 
dvander@1069: 	pContext->LocalToPhysAddr(params[1], &vec_addr);
dvander@1069: 
dvander@1069: 	Vector vec(sp_ctof(vec_addr[0]), sp_ctof(vec_addr[1]), sp_ctof(vec_addr[2]));
dvander@1069: 	Vector right, up;
dvander@1069: 
dvander@1069: 	VectorVectors(vec, right, up);
dvander@1069: 
dvander@1069: 	cell_t *addr_right, *addr_up;
dvander@1069: 	pContext->LocalToPhysAddr(params[2], &addr_right);
dvander@1069: 	pContext->LocalToPhysAddr(params[3], &addr_up);
dvander@1069: 
dvander@1069: 	SET_VECTOR(addr_right, right);
dvander@1069: 	SET_VECTOR(addr_up, up);
dvander@1069: 
dvander@1069: 	return 1;
dvander@1069: }
dvander@1069: 
dvander@1069: static cell_t NormalizeVector(IPluginContext *pContext, const cell_t *params)
dvander@1069: {
dvander@1069: 	cell_t *addr;
dvander@1069: 
dvander@1069: 	pContext->LocalToPhysAddr(params[1], &addr);
dvander@1069: 
dvander@1069: 	Vector source(sp_ctof(addr[0]), sp_ctof(addr[1]), sp_ctof(addr[2]));
dvander@1069: 	float length = VectorNormalize(source);
dvander@1069: 
dvander@1069: 	pContext->LocalToPhysAddr(params[2], &addr);
dvander@1069: 	SET_VECTOR(addr, source);
dvander@1069: 
dvander@1069: 	return sp_ftoc(length);
dvander@1069: }
dvander@1069: 
dvander@1069: REGISTER_NATIVES(vectorNatives)
dvander@1069: {
dvander@1069: 	{"GetAngleVectors",			GetAngleVectors},
dvander@1069: 	{"GetVectorAngles",			GetVectorAngles},
dvander@1069: 	{"GetVectorCrossProduct",	GetVectorCrossProduct},
dvander@1069: 	{"GetVectorDistance",		GetVectorDistance},
dvander@1069: 	{"GetVectorDotProduct",		GetVectorDotProduct},
dvander@1069: 	{"GetVectorLength",			GetVectorLength},
dvander@1069: 	{"GetVectorVectors",		GetVectorVectors},
dvander@1069: 	{"NormalizeVector",			NormalizeVector},
dvander@1069: 	{NULL,					NULL}
dvander@1069: };