dvander@2726: /**
dvander@2833:  * vim: set ts=4 sw=4 tw=99 :
dvander@2726:  * =============================================================================
dvander@2726:  * SourceMod
dvander@2726:  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
dvander@2726:  * =============================================================================
dvander@2726:  *
dvander@2726:  * This program is free software; you can redistribute it and/or modify it under
dvander@2726:  * the terms of the GNU General Public License, version 3.0, as published by the
dvander@2726:  * Free Software Foundation.
dvander@2726:  * 
dvander@2726:  * This program is distributed in the hope that it will be useful, but WITHOUT
dvander@2726:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
dvander@2726:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
dvander@2726:  * details.
dvander@2726:  *
dvander@2726:  * You should have received a copy of the GNU General Public License along with
dvander@2726:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@2726:  *
dvander@2726:  * As a special exception, AlliedModders LLC gives you permission to link the
dvander@2726:  * code of this program (as well as its derivative works) to "Half-Life 2," the
dvander@2726:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
dvander@2726:  * by the Valve Corporation.  You must obey the GNU General Public License in
dvander@2726:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
dvander@2726:  * this exception to all derivative works.  AlliedModders LLC defines further
dvander@2726:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
dvander@2726:  * or <http://www.sourcemod.net/license.php>.
dvander@2726:  *
dvander@2726:  * Version: $Id$
dvander@2726:  */
dvander@2726: 
dvander@2726: #include <math.h>
dvander@2726: #include <string.h>
dvander@2726: #include <stdlib.h>
dvander@2726: #include "common_logic.h"
dvander@2833: #include "MersenneTwister.h"
dvander@2833: #include <IPluginSys.h>
dvander@2726: 
dvander@2726: /****************************************
dvander@2726: *                                       *
dvander@2726: * FLOATING POINT NATIVE IMPLEMENTATIONS *
dvander@2726: *                                       *
dvander@2726: ****************************************/
dvander@2726: 
dvander@2726: 
dvander@2726: static cell_t sm_float(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = static_cast<float>(params[1]);
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_FloatAbs(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 	val = (val >= 0.0f) ? val : -val;
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_FloatAdd(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]) + sp_ctof(params[2]);
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_FloatSub(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]) - sp_ctof(params[2]);
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_FloatMul(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]) * sp_ctof(params[2]);
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_FloatDiv(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]) / sp_ctof(params[2]);
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_FloatCompare(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val1 = sp_ctof(params[1]);
dvander@2726: 	float val2 = sp_ctof(params[2]);
dvander@2726: 
dvander@2726: 	if (val1 > val2)
dvander@2726: 	{
dvander@2726: 		return 1;
dvander@2726: 	} else if (val1 < val2) {
dvander@2726: 		return -1;
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	return 0;
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_Logarithm(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 	float base = sp_ctof(params[2]);
dvander@2726: 
dvander@2726: 	if ((val <= 0.0f) || (base <= 0.0f))
dvander@2726: 	{
dvander@2726: 		return pCtx->ThrowNativeError("Cannot evaluate the logarithm of zero or a negative number (val:%f base:%f)", val, base);
dvander@2726: 	}
dvander@2726: 	if (base == 10.0f)
dvander@2726: 	{
dvander@2726: 		val = log10(val);
dvander@2726: 	} else {
dvander@2726: 		val = log(val) / log(base);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_Exponential(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 
dvander@2726: 	return sp_ftoc(exp(val));
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_Pow(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float base = sp_ctof(params[1]);
dvander@2726: 	float exponent = sp_ctof(params[2]);
dvander@2726: 
dvander@2726: 	return sp_ftoc(pow(base, exponent));
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_SquareRoot(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 
dvander@2726: 	if (val < 0.0f)
dvander@2726: 	{
dvander@2726: 		return pCtx->ThrowNativeError("Cannot evaluate the square root of a negative number (val:%f)", val);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	return sp_ftoc(sqrt(val));
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_RoundToNearest(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 	val = (float)floor(val + 0.5f);
dvander@2726: 
dvander@2726: 	return static_cast<int>(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_RoundToFloor(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 	val = floor(val);
dvander@2726: 
dvander@2726: 	return static_cast<int>(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_RoundToCeil(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 	val = ceil(val);
dvander@2726: 
dvander@2726: 	return static_cast<int>(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_RoundToZero(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 	if (val >= 0.0f)
dvander@2726: 	{
dvander@2726: 		val = floor(val);
dvander@2726: 	} else {
dvander@2726: 		val = ceil(val);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	return static_cast<int>(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_FloatFraction(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 	val = val - floor(val);
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_Sine(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 	val = sin(val);
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_Cosine(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 	val = cos(val);
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_Tangent(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 	val = tan(val);
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_ArcSine(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 	val = asin(val);
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_ArcCosine(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 	val = acos(val);
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_ArcTangent(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 	val = atan(val);
dvander@2726: 
dvander@2726: 	return sp_ftoc(val);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_ArcTangent2(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val1 = sp_ctof(params[1]);
dvander@2726: 	float val2 = sp_ctof(params[2]);
dvander@2726: 	val1 = atan2(val1, val2);
dvander@2726: 
dvander@2726: 	return sp_ftoc(val1);
dvander@2726: }
dvander@2726: 
dvander@2726: #if 0
dvander@2726: static cell_t sm_FloatRound(IPluginContext *pCtx, const cell_t *params)
dvander@2726: {
dvander@2726: 	float val = sp_ctof(params[1]);
dvander@2726: 
dvander@2726: 	switch (params[2])
dvander@2726: 	{
dvander@2726: 	case 1:
dvander@2726: 		{
dvander@2726: 			val = floor(val);
dvander@2726: 			break;
dvander@2726: 		}
dvander@2726: 	case 2:
dvander@2726: 		{
dvander@2726: 			val = ceil(val);
dvander@2726: 			break;
dvander@2726: 		}
dvander@2726: 	case 3:
dvander@2726: 		{
dvander@2726: 			if (val >= 0.0f)
dvander@2726: 			{
dvander@2726: 				val = floor(val);
dvander@2726: 			} else {
dvander@2726: 				val = ceil(val);
dvander@2726: 			}
dvander@2726: 			break;
dvander@2726: 		}
dvander@2726: 	default:
dvander@2726: 		{
dvander@2726: 			val = (float)floor(val + 0.5f);
dvander@2726: 			break;
dvander@2726: 		}
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	return static_cast<int>(val);
dvander@2726: }
dvander@2726: #endif
dvander@2726: 
dvander@2833: class RandomHelpers :
dvander@2833: 	public SMGlobalClass,
dvander@2833: 	public IPluginsListener
dvander@2833: {
dvander@2833: public:
dvander@2833: 	void OnSourceModAllInitialized()
dvander@2833: 	{
dvander@2833: 		pluginsys->AddPluginsListener(this);
dvander@2833: 	}
dvander@2833: 
dvander@2833: 	void OnSourceModShutdown()
dvander@2833: 	{
dvander@2833: 		pluginsys->RemovePluginsListener(this);
dvander@2833: 	}
dvander@2833: 
dvander@2833: 	void OnPluginDestroyed(IPlugin *plugin)
dvander@2833: 	{
dvander@2833: 		MTRand *mtrand;
dvander@2833: 		if (plugin->GetProperty("core.logic.mtrand", (void**)&mtrand, true))
dvander@2833: 		{
dvander@2833: 			delete mtrand;
dvander@2833: 		}
dvander@2833: 	}
dvander@2833: 	
dvander@2833: 	MTRand *RandObjForPlugin(IPluginContext *ctx)
dvander@2833: 	{
dvander@2833: 		IPlugin *plugin = pluginsys->FindPluginByContext(ctx->GetContext());
dvander@2833: 		MTRand *mtrand;	
dvander@2833: 		if (!plugin->GetProperty("core.logic.mtrand", (void**)&mtrand))
dvander@2833: 		{
dvander@2833: 			mtrand = new MTRand();
dvander@2833: 			plugin->SetProperty("core.logic.mtrand", mtrand);
dvander@2833: 		}
dvander@2833: 		return mtrand;
dvander@2833: 	}
dvander@2833: } s_RandHelpers;
dvander@2833: 
dvander@2833: static cell_t GetURandomInt(IPluginContext *ctx, const cell_t *params)
dvander@2833: {
dvander@2833: 	MTRand *randobj = s_RandHelpers.RandObjForPlugin(ctx);
dvander@2833: 	/* Note the sign bit must be stripped off because cell_t is signed,
dvander@2833: 	 * and this guarantees a range of [0,max_int]
dvander@2833: 	 */
dvander@2833: 	return randobj->randInt() & 0x7FFFFFFF;
dvander@2833: }
dvander@2833: 
dvander@2833: static cell_t GetURandomFloat(IPluginContext *ctx, const cell_t *params)
dvander@2833: {
dvander@2833: 	MTRand *randobj = s_RandHelpers.RandObjForPlugin(ctx);
dvander@2833: 	return sp_ftoc((float)randobj->rand());
dvander@2833: }
dvander@2833: 
dvander@2833: static cell_t SetURandomSeed(IPluginContext *ctx, const cell_t *params)
dvander@2833: {
dvander@2833: 	MTRand *randobj = s_RandHelpers.RandObjForPlugin(ctx);
dvander@2833: 	cell_t *addr;
dvander@2833: 	ctx->LocalToPhysAddr(params[1], &addr);
dvander@2833: 	/* We're 32-bit only. */
dvander@2833: 	randobj->seed((MTRand::uint32*)addr, params[2]);
dvander@2833: 	return 1;
dvander@2833: }
dvander@2833: 
dvander@2726: REGISTER_NATIVES(floatnatives)
dvander@2726: {
dvander@2726: 	{"float",			sm_float},
dvander@2726: 	{"FloatMul",		sm_FloatMul},
dvander@2726: 	{"FloatDiv",		sm_FloatDiv},
dvander@2726: 	{"FloatAdd",		sm_FloatAdd},
dvander@2726: 	{"FloatSub",		sm_FloatSub},
dvander@2726: 	{"FloatFraction",	sm_FloatFraction},
dvander@2726: 	{"RoundToZero",		sm_RoundToZero},
dvander@2726: 	{"RoundToCeil",		sm_RoundToCeil},
dvander@2726: 	{"RoundToFloor",	sm_RoundToFloor},
dvander@2726: 	{"RoundToNearest",	sm_RoundToNearest},
dvander@2726: 	{"FloatCompare",	sm_FloatCompare},
dvander@2726: 	{"SquareRoot",		sm_SquareRoot},
dvander@2726: 	{"Pow",				sm_Pow},
dvander@2726: 	{"Exponential",		sm_Exponential},
dvander@2726: 	{"Logarithm",		sm_Logarithm},
dvander@2726: 	{"Sine",			sm_Sine},
dvander@2726: 	{"Cosine",			sm_Cosine},
dvander@2726: 	{"Tangent",			sm_Tangent},
dvander@2726: 	{"FloatAbs",		sm_FloatAbs},
dvander@2726: 	{"ArcTangent",		sm_ArcTangent},
dvander@2726: 	{"ArcCosine",		sm_ArcCosine},
dvander@2726: 	{"ArcSine",			sm_ArcSine},
dvander@2726: 	{"ArcTangent2",		sm_ArcTangent2},
dvander@2833: 	{"GetURandomInt",	GetURandomInt},
dvander@2833: 	{"GetURandomFloat",	GetURandomFloat},
dvander@2833: 	{"SetURandomSeed",	SetURandomSeed},
dvander@2726: 	{NULL,				NULL}
dvander@2726: };
dvander@2833: