289 lines
8.2 KiB
C++
289 lines
8.2 KiB
C++
/**
|
|
* vim: set ts=4 :
|
|
* =============================================================================
|
|
* SourceMod SDKTools Extension
|
|
* 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 "extension.h"
|
|
#include <datamap.h>
|
|
|
|
#define SIZEOF_VARIANT_T 20
|
|
|
|
ICallWrapper *g_pAcceptInput = NULL;
|
|
unsigned char g_Variant_t[SIZEOF_VARIANT_T] = {0};
|
|
|
|
#define ENTINDEX_TO_CBASEENTITY(index, buffer) \
|
|
pEdict = PEntityOfEntIndex(index); \
|
|
if (!pEdict || pEdict->IsFree()) \
|
|
{ \
|
|
return pContext->ThrowNativeError("Entity %d is not valid or is freed", index); \
|
|
} \
|
|
pUnk = pEdict->GetUnknown(); \
|
|
if (!pUnk) \
|
|
{ \
|
|
return pContext->ThrowNativeError("Entity %d is a not an IServerUnknown", index); \
|
|
} \
|
|
buffer = pUnk->GetBaseEntity(); \
|
|
if (!buffer) \
|
|
{ \
|
|
return pContext->ThrowNativeError("Entity %d is not a CBaseEntity", index); \
|
|
}
|
|
|
|
/* Hack to init the variant_t object for the first time */
|
|
class VariantFirstTimeInit
|
|
{
|
|
public:
|
|
VariantFirstTimeInit()
|
|
{
|
|
*(unsigned int *)(&g_Variant_t[12]) = INVALID_EHANDLE_INDEX;
|
|
}
|
|
} g_VariantFirstTimeInit;
|
|
|
|
inline void _init_variant_t()
|
|
{
|
|
unsigned char *vptr = g_Variant_t;
|
|
|
|
*(int *)vptr = 0;
|
|
vptr += sizeof(int)*3;
|
|
*(unsigned long *)vptr = INVALID_EHANDLE_INDEX;
|
|
vptr += sizeof(unsigned long);
|
|
*(fieldtype_t *)vptr = FIELD_VOID;
|
|
}
|
|
|
|
static cell_t AcceptEntityInput(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
if (!g_pAcceptInput)
|
|
{
|
|
int offset;
|
|
if (!g_pGameConf->GetOffset("AcceptInput", &offset))
|
|
{
|
|
return pContext->ThrowNativeError("\"AcceptEntityInput\" not supported by this mod");
|
|
}
|
|
|
|
PassInfo pass[6];
|
|
pass[0].type = PassType_Basic;
|
|
pass[0].flags = PASSFLAG_BYVAL;
|
|
pass[0].size = sizeof(const char *);
|
|
pass[1].type = pass[2].type = PassType_Basic;
|
|
pass[1].flags = pass[2].flags = PASSFLAG_BYVAL;
|
|
pass[1].size = pass[2].size = sizeof(CBaseEntity *);
|
|
pass[3].type = PassType_Object;
|
|
pass[3].flags = PASSFLAG_BYVAL|PASSFLAG_OCTOR|PASSFLAG_ODTOR|PASSFLAG_OASSIGNOP;
|
|
pass[3].size = SIZEOF_VARIANT_T;
|
|
pass[4].type = PassType_Basic;
|
|
pass[4].flags = PASSFLAG_BYVAL;
|
|
pass[4].size = sizeof(int);
|
|
pass[5].type = PassType_Basic;
|
|
pass[5].flags = PASSFLAG_BYVAL;
|
|
pass[5].size = sizeof(bool);
|
|
|
|
if (!(g_pAcceptInput=g_pBinTools->CreateVCall(offset, 0, 0, &pass[5], pass, 5)))
|
|
{
|
|
pContext->ThrowNativeError("\"AcceptEntityInput\" wrapper failed to initialized");
|
|
}
|
|
}
|
|
|
|
CBaseEntity *pActivator, *pCaller, *pDest;
|
|
edict_t *pEdict;
|
|
IServerUnknown *pUnk;
|
|
|
|
char *inputname;
|
|
unsigned char vstk[sizeof(void *) + sizeof(const char *) + sizeof(CBaseEntity *)*2 + SIZEOF_VARIANT_T + sizeof(int)];
|
|
unsigned char *vptr = vstk;
|
|
|
|
ENTINDEX_TO_CBASEENTITY(params[1], pDest);
|
|
pContext->LocalToString(params[2], &inputname);
|
|
if (params[3] == -1)
|
|
{
|
|
pActivator = NULL;
|
|
} else {
|
|
ENTINDEX_TO_CBASEENTITY(params[3], pActivator);
|
|
}
|
|
if (params[4] == -1)
|
|
{
|
|
pCaller = NULL;
|
|
} else {
|
|
ENTINDEX_TO_CBASEENTITY(params[4], pCaller);
|
|
}
|
|
|
|
*(void **)vptr = pDest;
|
|
vptr += sizeof(void *);
|
|
*(const char **)vptr = inputname;
|
|
vptr += sizeof(const char *);
|
|
*(CBaseEntity **)vptr = pActivator;
|
|
vptr += sizeof(CBaseEntity *);
|
|
*(CBaseEntity **)vptr = pCaller;
|
|
vptr += sizeof(CBaseEntity *);
|
|
memcpy(vptr, g_Variant_t, SIZEOF_VARIANT_T);
|
|
vptr += SIZEOF_VARIANT_T;
|
|
*(int *)vptr = params[5];
|
|
|
|
bool ret;
|
|
g_pAcceptInput->Execute(vstk, &ret);
|
|
|
|
_init_variant_t();
|
|
|
|
return (ret) ? 1 : 0;
|
|
}
|
|
|
|
static cell_t SetVariantBool(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
unsigned char *vptr = g_Variant_t;
|
|
|
|
*(bool *)vptr = (params[1]) ? true : false;
|
|
vptr += sizeof(int)*3 + sizeof(unsigned long);
|
|
*(fieldtype_t *)vptr = FIELD_BOOLEAN;
|
|
|
|
return 1;
|
|
}
|
|
|
|
static cell_t SetVariantString(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
char *str;
|
|
unsigned char *vptr = g_Variant_t;
|
|
|
|
pContext->LocalToString(params[1], &str);
|
|
|
|
*(string_t *)vptr = MAKE_STRING(str);
|
|
vptr += sizeof(int)*3 + sizeof(unsigned long);
|
|
*(fieldtype_t *)vptr = FIELD_STRING;
|
|
|
|
return 1;
|
|
}
|
|
|
|
static cell_t SetVariantInt(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
unsigned char *vptr = g_Variant_t;
|
|
|
|
*(int *)vptr = params[1];
|
|
vptr += sizeof(int)*3 + sizeof(unsigned long);
|
|
*(fieldtype_t *)vptr = FIELD_INTEGER;
|
|
|
|
return 1;
|
|
}
|
|
|
|
static cell_t SetVariantFloat(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
unsigned char *vptr = g_Variant_t;
|
|
|
|
*(float *)vptr = sp_ctof(params[1]);
|
|
vptr += sizeof(int)*3 + sizeof(unsigned long);
|
|
*(fieldtype_t *)vptr = FIELD_FLOAT;
|
|
|
|
return 1;
|
|
}
|
|
|
|
static cell_t SetVariantVector3D(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
cell_t *val;
|
|
unsigned char *vptr = g_Variant_t;
|
|
|
|
pContext->LocalToPhysAddr(params[1], &val);
|
|
|
|
*(float *)vptr = sp_ctof(val[0]);
|
|
vptr += sizeof(float);
|
|
*(float *)vptr = sp_ctof(val[1]);
|
|
vptr += sizeof(float);
|
|
*(float *)vptr = sp_ctof(val[2]);
|
|
vptr += sizeof(float) + sizeof(unsigned long);
|
|
*(fieldtype_t *)vptr = FIELD_VECTOR;
|
|
|
|
return 1;
|
|
}
|
|
|
|
static cell_t SetVariantPosVector3D(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
cell_t *val;
|
|
unsigned char *vptr = g_Variant_t;
|
|
|
|
pContext->LocalToPhysAddr(params[1], &val);
|
|
|
|
*(float *)vptr = sp_ctof(val[0]);
|
|
vptr += sizeof(float);
|
|
*(float *)vptr = sp_ctof(val[1]);
|
|
vptr += sizeof(float);
|
|
*(float *)vptr = sp_ctof(val[2]);
|
|
vptr += sizeof(float) + sizeof(unsigned long);
|
|
*(fieldtype_t *)vptr = FIELD_POSITION_VECTOR;
|
|
|
|
return 1;
|
|
}
|
|
|
|
static cell_t SetVariantColor(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
cell_t *val;
|
|
unsigned char *vptr = g_Variant_t;
|
|
|
|
pContext->LocalToPhysAddr(params[1], &val);
|
|
|
|
*(unsigned char *)vptr = val[0];
|
|
vptr += sizeof(unsigned char);
|
|
*(unsigned char *)vptr = val[1];
|
|
vptr += sizeof(unsigned char);
|
|
*(unsigned char *)vptr = val[2];
|
|
vptr += sizeof(unsigned char);
|
|
*(unsigned char *)vptr = val[3];
|
|
vptr += sizeof(unsigned char) + sizeof(int)*2 + sizeof(unsigned long);
|
|
*(fieldtype_t *)vptr = FIELD_COLOR32;
|
|
|
|
return 1;
|
|
}
|
|
|
|
static cell_t SetVariantEntity(IPluginContext *pContext, const cell_t *params)
|
|
{
|
|
CBaseEntity *pEntity;
|
|
edict_t *pEdict;
|
|
IServerUnknown *pUnk;
|
|
unsigned char *vptr = g_Variant_t;
|
|
CBaseHandle bHandle;
|
|
|
|
ENTINDEX_TO_CBASEENTITY(params[1], pEntity);
|
|
bHandle = reinterpret_cast<IHandleEntity *>(pEntity)->GetRefEHandle();
|
|
|
|
vptr += sizeof(int)*3;
|
|
*(unsigned long *)vptr = (unsigned long)(bHandle.ToInt());
|
|
vptr += sizeof(unsigned long);
|
|
*(fieldtype_t *)vptr = FIELD_EHANDLE;
|
|
|
|
return 1;
|
|
}
|
|
|
|
sp_nativeinfo_t g_EntInputNatives[] =
|
|
{
|
|
{"AcceptEntityInput", AcceptEntityInput},
|
|
{"SetVariantBool", SetVariantBool},
|
|
{"SetVariantString", SetVariantString},
|
|
{"SetVariantInt", SetVariantInt},
|
|
{"SetVariantFloat", SetVariantFloat},
|
|
{"SetVariantVector3D", SetVariantVector3D},
|
|
{"SetVariantPosVector3D", SetVariantPosVector3D},
|
|
{"SetVariantColor", SetVariantColor},
|
|
{"SetVariantEntity", SetVariantEntity},
|
|
{NULL, NULL},
|
|
};
|