faluco@505: /**
faluco@505: * vim: set ts=4 :
faluco@505: * ===============================================================
faluco@505: * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.
faluco@505: * ===============================================================
faluco@505: *
faluco@505: * This file is not open source and may not be copied without explicit
faluco@505: * written permission of AlliedModders LLC.  This file may not be redistributed 
faluco@505: * in whole or significant part.
faluco@505: * For information, see LICENSE.txt or http://www.sourcemod.net/license.php
faluco@505: *
faluco@505: * Version: $Id$
faluco@505: */
faluco@505: 
faluco@505: #include <bitbuf.h>
faluco@505: #include <vector.h>
faluco@505: #include "sourcemod.h"
faluco@505: #include "HandleSys.h"
faluco@505: 
faluco@505: static cell_t smn_BfWriteBool(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	pBitBuf->WriteOneBit(params[2]);
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: static cell_t smn_BfWriteByte(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	pBitBuf->WriteByte(params[2]);
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: static cell_t smn_BfWriteChar(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	pBitBuf->WriteChar(params[2]);
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: static cell_t smn_BfWriteShort(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	pBitBuf->WriteShort(params[2]);
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: static cell_t smn_BfWriteWord(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	pBitBuf->WriteWord(params[2]);
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: static cell_t smn_BfWriteNum(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	pBitBuf->WriteLong(static_cast<long>(params[2]));
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: static cell_t smn_BfWriteFloat(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	pBitBuf->WriteFloat(sp_ctof(params[2]));
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: static cell_t smn_BfWriteString(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 	int err;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	char *str;
faluco@505: 	if ((err=pCtx->LocalToString(params[2], &str)) != SP_ERROR_NONE)
faluco@505: 	{
faluco@505: 		pCtx->ThrowNativeErrorEx(err, NULL);
faluco@505: 		return 0;
faluco@505: 	}
faluco@505: 
faluco@505: 	pBitBuf->WriteString(str);
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: static cell_t smn_BfWriteEntity(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	pBitBuf->WriteShort(params[2]);
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: static cell_t smn_BfWriteAngle(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	pBitBuf->WriteBitAngle(sp_ctof(params[2]), params[3]);
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: static cell_t smn_BfWriteCoord(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	pBitBuf->WriteBitCoord(sp_ctof(params[2]));
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: static cell_t smn_BfWriteVecCoord(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	cell_t *pVec;
faluco@505: 	pCtx->LocalToPhysAddr(params[2], &pVec);
faluco@505: 	Vector vec(sp_ctof(pVec[0]), sp_ctof(pVec[1]), sp_ctof(pVec[2]));
faluco@505: 	pBitBuf->WriteBitVec3Coord(vec);
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: static cell_t smn_BfWriteVecNormal(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	cell_t *pVec;
faluco@505: 	pCtx->LocalToPhysAddr(params[2], &pVec);
faluco@505: 	Vector vec(sp_ctof(pVec[0]), sp_ctof(pVec[1]), sp_ctof(pVec[2]));
faluco@505: 	pBitBuf->WriteBitVec3Normal(vec);
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: static cell_t smn_BfWriteAngles(IPluginContext *pCtx, const cell_t *params)
faluco@505: {
faluco@505: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
faluco@505: 	HandleError herr;
faluco@505: 	HandleSecurity sec;
faluco@505: 	bf_write *pBitBuf;
faluco@505: 
faluco@505: 	sec.pOwner = NULL;
faluco@505: 	sec.pIdentity = g_pCoreIdent;
faluco@505: 
faluco@505: 	if ((herr=g_HandleSys.ReadHandle(hndl, g_WrBitBufType, &sec, (void **)&pBitBuf))
faluco@505: 		!= HandleError_None)
faluco@505: 	{
faluco@505: 		return pCtx->ThrowNativeError("Invalid bit buffer handle %x (error %d)", hndl, herr);
faluco@505: 	}
faluco@505: 
faluco@505: 	cell_t *pAng;
faluco@505: 	pCtx->LocalToPhysAddr(params[2], &pAng);
faluco@505: 	QAngle ang(sp_ctof(pAng[0]), sp_ctof(pAng[1]), sp_ctof(pAng[2]));
faluco@505: 	pBitBuf->WriteBitAngles(ang);
faluco@505: 
faluco@505: 	return 1;
faluco@505: }
faluco@505: 
faluco@505: REGISTER_NATIVES(wrbitbufnatives)
faluco@505: {
faluco@505: 	{"BfWriteBool",				smn_BfWriteBool},
faluco@505: 	{"BfWriteByte",				smn_BfWriteByte},
faluco@505: 	{"BfWriteChar",				smn_BfWriteChar},
faluco@505: 	{"BfWriteShort",			smn_BfWriteShort},
faluco@505: 	{"BfWriteWord",				smn_BfWriteWord},
faluco@505: 	{"BfWriteNum",				smn_BfWriteNum},
faluco@505: 	{"BfWriteFloat",			smn_BfWriteFloat},
faluco@505: 	{"BfWriteString",			smn_BfWriteString},
faluco@505: 	{"BfWriteEntity",			smn_BfWriteEntity},
faluco@505: 	{"BfWriteAngle",			smn_BfWriteAngle},
faluco@505: 	{"BfWriteCoord",			smn_BfWriteCoord},
faluco@505: 	{"BfWriteVecCoord",			smn_BfWriteVecCoord},
faluco@505: 	{"BfWriteVecNormal",		smn_BfWriteVecNormal},
faluco@505: 	{"BfWriteAngles",			smn_BfWriteAngles},
faluco@505: 	{NULL,						NULL}
faluco@505: };