429 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			429 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /**
 | |
|  * vim: set ts=4 sw=4 tw=99 :
 | |
|  * =============================================================================
 | |
|  * 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 <math.h>
 | |
| #include <string.h>
 | |
| #include <stdlib.h>
 | |
| #include "common_logic.h"
 | |
| #include "MersenneTwister.h"
 | |
| #include <IPluginSys.h>
 | |
| #include <am-utility.h>
 | |
| 
 | |
| /****************************************
 | |
| *                                       *
 | |
| * FLOATING POINT NATIVE IMPLEMENTATIONS *
 | |
| *                                       *
 | |
| ****************************************/
 | |
| 
 | |
| 
 | |
| static cell_t sm_float(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = static_cast<float>(params[1]);
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatAbs(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	val = (val >= 0.0f) ? val : -val;
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatAdd(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]) + sp_ctof(params[2]);
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatSub(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]) - sp_ctof(params[2]);
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatMul(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]) * sp_ctof(params[2]);
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatDiv(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]) / sp_ctof(params[2]);
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatGt(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	return !!(sp_ctof(params[1]) > sp_ctof(params[2]));
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatGe(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	return !!(sp_ctof(params[1]) >= sp_ctof(params[2]));
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatLt(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	return !!(sp_ctof(params[1]) < sp_ctof(params[2]));
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatLe(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	return !!(sp_ctof(params[1]) <= sp_ctof(params[2]));
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatEq(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	return !!(sp_ctof(params[1]) == sp_ctof(params[2]));
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatNe(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	return !!(sp_ctof(params[1]) != sp_ctof(params[2]));
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatNot(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	if (ke::IsNaN(val))
 | |
| 		return 1;
 | |
| 	return val ? 0 : 1;
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatCompare(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val1 = sp_ctof(params[1]);
 | |
| 	float val2 = sp_ctof(params[2]);
 | |
| 
 | |
| 	if (val1 > val2)
 | |
| 	{
 | |
| 		return 1;
 | |
| 	} else if (val1 < val2) {
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| static cell_t sm_Logarithm(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	float base = sp_ctof(params[2]);
 | |
| 
 | |
| 	if ((val <= 0.0f) || (base <= 0.0f))
 | |
| 	{
 | |
| 		return pCtx->ThrowNativeError("Cannot evaluate the logarithm of zero or a negative number (val:%f base:%f)", val, base);
 | |
| 	}
 | |
| 	if (base == 10.0f)
 | |
| 	{
 | |
| 		val = log10(val);
 | |
| 	} else {
 | |
| 		val = log(val) / log(base);
 | |
| 	}
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_Exponential(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 
 | |
| 	return sp_ftoc(exp(val));
 | |
| }
 | |
| 
 | |
| static cell_t sm_Pow(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float base = sp_ctof(params[1]);
 | |
| 	float exponent = sp_ctof(params[2]);
 | |
| 
 | |
| 	return sp_ftoc(pow(base, exponent));
 | |
| }
 | |
| 
 | |
| static cell_t sm_SquareRoot(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 
 | |
| 	if (val < 0.0f)
 | |
| 	{
 | |
| 		return pCtx->ThrowNativeError("Cannot evaluate the square root of a negative number (val:%f)", val);
 | |
| 	}
 | |
| 
 | |
| 	return sp_ftoc(sqrt(val));
 | |
| }
 | |
| 
 | |
| static cell_t sm_RoundToNearest(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	val = (float)floor(val + 0.5f);
 | |
| 
 | |
| 	return static_cast<int>(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_RoundToFloor(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	val = floor(val);
 | |
| 
 | |
| 	return static_cast<int>(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_RoundToCeil(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	val = ceil(val);
 | |
| 
 | |
| 	return static_cast<int>(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_RoundToZero(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	if (val >= 0.0f)
 | |
| 	{
 | |
| 		val = floor(val);
 | |
| 	} else {
 | |
| 		val = ceil(val);
 | |
| 	}
 | |
| 
 | |
| 	return static_cast<int>(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_FloatFraction(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	val = val - floor(val);
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_Sine(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	val = sin(val);
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_Cosine(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	val = cos(val);
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_Tangent(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	val = tan(val);
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_ArcSine(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	val = asin(val);
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_ArcCosine(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	val = acos(val);
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_ArcTangent(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 	val = atan(val);
 | |
| 
 | |
| 	return sp_ftoc(val);
 | |
| }
 | |
| 
 | |
| static cell_t sm_ArcTangent2(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val1 = sp_ctof(params[1]);
 | |
| 	float val2 = sp_ctof(params[2]);
 | |
| 	val1 = atan2(val1, val2);
 | |
| 
 | |
| 	return sp_ftoc(val1);
 | |
| }
 | |
| 
 | |
| #if 0
 | |
| static cell_t sm_FloatRound(IPluginContext *pCtx, const cell_t *params)
 | |
| {
 | |
| 	float val = sp_ctof(params[1]);
 | |
| 
 | |
| 	switch (params[2])
 | |
| 	{
 | |
| 	case 1:
 | |
| 		{
 | |
| 			val = floor(val);
 | |
| 			break;
 | |
| 		}
 | |
| 	case 2:
 | |
| 		{
 | |
| 			val = ceil(val);
 | |
| 			break;
 | |
| 		}
 | |
| 	case 3:
 | |
| 		{
 | |
| 			if (val >= 0.0f)
 | |
| 			{
 | |
| 				val = floor(val);
 | |
| 			} else {
 | |
| 				val = ceil(val);
 | |
| 			}
 | |
| 			break;
 | |
| 		}
 | |
| 	default:
 | |
| 		{
 | |
| 			val = (float)floor(val + 0.5f);
 | |
| 			break;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return static_cast<int>(val);
 | |
| }
 | |
| #endif
 | |
| 
 | |
| class RandomHelpers :
 | |
| 	public SMGlobalClass,
 | |
| 	public IPluginsListener
 | |
| {
 | |
| public:
 | |
| 	void OnSourceModAllInitialized()
 | |
| 	{
 | |
| 		pluginsys->AddPluginsListener(this);
 | |
| 	}
 | |
| 
 | |
| 	void OnSourceModShutdown()
 | |
| 	{
 | |
| 		pluginsys->RemovePluginsListener(this);
 | |
| 	}
 | |
| 
 | |
| 	void OnPluginDestroyed(IPlugin *plugin)
 | |
| 	{
 | |
| 		MTRand *mtrand;
 | |
| 		if (plugin->GetProperty("core.logic.mtrand", (void**)&mtrand, true))
 | |
| 		{
 | |
| 			delete mtrand;
 | |
| 		}
 | |
| 	}
 | |
| 	
 | |
| 	MTRand *RandObjForPlugin(IPluginContext *ctx)
 | |
| 	{
 | |
| 		IPlugin *plugin = pluginsys->FindPluginByContext(ctx->GetContext());
 | |
| 		MTRand *mtrand;	
 | |
| 		if (!plugin->GetProperty("core.logic.mtrand", (void**)&mtrand))
 | |
| 		{
 | |
| 			mtrand = new MTRand();
 | |
| 			plugin->SetProperty("core.logic.mtrand", mtrand);
 | |
| 		}
 | |
| 		return mtrand;
 | |
| 	}
 | |
| } s_RandHelpers;
 | |
| 
 | |
| static cell_t GetURandomInt(IPluginContext *ctx, const cell_t *params)
 | |
| {
 | |
| 	MTRand *randobj = s_RandHelpers.RandObjForPlugin(ctx);
 | |
| 	/* Note the sign bit must be stripped off because cell_t is signed,
 | |
| 	 * and this guarantees a range of [0,max_int]
 | |
| 	 */
 | |
| 	return randobj->randInt() & 0x7FFFFFFF;
 | |
| }
 | |
| 
 | |
| static cell_t GetURandomFloat(IPluginContext *ctx, const cell_t *params)
 | |
| {
 | |
| 	MTRand *randobj = s_RandHelpers.RandObjForPlugin(ctx);
 | |
| 	return sp_ftoc((float)randobj->rand());
 | |
| }
 | |
| 
 | |
| static cell_t SetURandomSeed(IPluginContext *ctx, const cell_t *params)
 | |
| {
 | |
| 	MTRand *randobj = s_RandHelpers.RandObjForPlugin(ctx);
 | |
| 	cell_t *addr;
 | |
| 	ctx->LocalToPhysAddr(params[1], &addr);
 | |
| 	/* We're 32-bit only. */
 | |
| 	randobj->seed((MTRand::uint32*)addr, params[2]);
 | |
| 	return 1;
 | |
| }
 | |
| 
 | |
| REGISTER_NATIVES(floatnatives)
 | |
| {
 | |
| 	{"float",			sm_float},
 | |
| 	{"FloatMul",		sm_FloatMul},
 | |
| 	{"FloatDiv",		sm_FloatDiv},
 | |
| 	{"FloatAdd",		sm_FloatAdd},
 | |
| 	{"FloatSub",		sm_FloatSub},
 | |
| 	{"FloatFraction",	sm_FloatFraction},
 | |
| 	{"RoundToZero",		sm_RoundToZero},
 | |
| 	{"RoundToCeil",		sm_RoundToCeil},
 | |
| 	{"RoundToFloor",	sm_RoundToFloor},
 | |
| 	{"RoundToNearest",	sm_RoundToNearest},
 | |
| 	{"__FLOAT_GT__",	sm_FloatGt},
 | |
| 	{"__FLOAT_GE__",	sm_FloatGe},
 | |
| 	{"__FLOAT_LT__",	sm_FloatLt},
 | |
| 	{"__FLOAT_LE__",	sm_FloatLe},
 | |
| 	{"__FLOAT_EQ__",	sm_FloatEq},
 | |
| 	{"__FLOAT_NE__",	sm_FloatNe},
 | |
| 	{"__FLOAT_NOT__",	sm_FloatNot},
 | |
| 	{"FloatCompare",	sm_FloatCompare},
 | |
| 	{"SquareRoot",		sm_SquareRoot},
 | |
| 	{"Pow",				sm_Pow},
 | |
| 	{"Exponential",		sm_Exponential},
 | |
| 	{"Logarithm",		sm_Logarithm},
 | |
| 	{"Sine",			sm_Sine},
 | |
| 	{"Cosine",			sm_Cosine},
 | |
| 	{"Tangent",			sm_Tangent},
 | |
| 	{"FloatAbs",		sm_FloatAbs},
 | |
| 	{"ArcTangent",		sm_ArcTangent},
 | |
| 	{"ArcCosine",		sm_ArcCosine},
 | |
| 	{"ArcSine",			sm_ArcSine},
 | |
| 	{"ArcTangent2",		sm_ArcTangent2},
 | |
| 	{"GetURandomInt",	GetURandomInt},
 | |
| 	{"GetURandomFloat",	GetURandomFloat},
 | |
| 	{"SetURandomSeed",	SetURandomSeed},
 | |
| 	{NULL,				NULL}
 | |
| };
 | |
| 
 |