Introduce DataPacks (bug 5885, r=ds)

amxmodx/datapacks.cpp

@@ -0,0 +1,250 @@
+/**
+ * vim: set ts=4 :
+ * =============================================================================
+ * 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 "CDataPack.h"
+
+CDataPackHandles g_DataPackHandles;
+
+static cell AMX_NATIVE_CALL CreateDataPack(AMX* amx, cell* params)
+{
+	return static_cast<cell>(g_DataPackHandles.create());
+}
+
+static cell AMX_NATIVE_CALL WritePackCell(AMX* amx, cell* params)
+{
+	CDataPack *d = g_DataPackHandles.lookup(params[1]);
+
+	if (d == NULL)
+	{
+		LogError(amx, AMX_ERR_NATIVE, "Invalid datapack handle provided (%d)", params[1]);
+		return 0;
+	}
+
+	d->PackCell(params[2]);
+
+	return 1;
+}
+
+static cell AMX_NATIVE_CALL WritePackFloat(AMX* amx, cell* params)
+{
+	CDataPack *d = g_DataPackHandles.lookup(params[1]);
+
+	if (d == NULL)
+	{
+		LogError(amx, AMX_ERR_NATIVE, "Invalid datapack handle provided (%d)", params[1]);
+		return 0;
+	}
+
+	d->PackFloat(amx_ctof(params[2]));
+
+	return 1;
+}
+
+static cell AMX_NATIVE_CALL WritePackString(AMX* amx, cell* params)
+{
+	CDataPack *d = g_DataPackHandles.lookup(params[1]);
+
+	if (d == NULL)
+	{
+		LogError(amx, AMX_ERR_NATIVE, "Invalid datapack handle provided (%d)", params[1]);
+		return 0;
+	}
+
+	int len;
+	const char *str = get_amxstring(amx, params[2], 0, len);
+
+	d->PackString(str);
+
+	return len;
+}
+
+static cell AMX_NATIVE_CALL ReadPackCell(AMX* amx, cell* params)
+{
+	CDataPack *d = g_DataPackHandles.lookup(params[1]);
+
+	if (d == NULL)
+	{
+		LogError(amx, AMX_ERR_NATIVE, "Invalid datapack handle provided (%d)", params[1]);
+		return 0;
+	}
+
+	if (!d->IsReadable(sizeof(size_t) + sizeof(cell)))
+	{
+		LogError(amx, AMX_ERR_NATIVE, "DataPack operation is out of bounds.");
+		return 0;
+	}
+
+	return d->ReadCell();
+}
+
+static cell AMX_NATIVE_CALL ReadPackFloat(AMX* amx, cell* params)
+{
+	CDataPack *d = g_DataPackHandles.lookup(params[1]);
+
+	if (d == NULL)
+	{
+		LogError(amx, AMX_ERR_NATIVE, "Invalid datapack handle provided (%d)", params[1]);
+		return 0;
+	}
+
+	if (!d->IsReadable(sizeof(size_t) + sizeof(float)))
+	{
+		LogError(amx, AMX_ERR_NATIVE, "DataPack operation is out of bounds.");
+		return 0;
+	}
+
+	float value = d->ReadFloat();
+
+	return amx_ftoc(value);
+}
+
+static cell AMX_NATIVE_CALL ReadPackString(AMX* amx, cell* params)
+{
+	CDataPack *d = g_DataPackHandles.lookup(params[1]);
+
+	if (d == NULL)
+	{
+		LogError(amx, AMX_ERR_NATIVE, "Invalid datapack handle provided (%d)", params[1]);
+		return 0;
+	}
+
+	const char *str;
+	if (!(str = d->ReadString(NULL)))
+	{
+		LogError(amx, AMX_ERR_NATIVE, "DataPack operation is out of bounds.");
+		return 0;
+	}
+
+	return set_amxstring(amx, params[2], str, params[3]);
+}
+
+static cell AMX_NATIVE_CALL ResetPack(AMX* amx, cell* params)
+{
+	CDataPack *d = g_DataPackHandles.lookup(params[1]);
+
+	if (d == NULL)
+	{
+		LogError(amx, AMX_ERR_NATIVE, "Invalid datapack handle provided (%d)", params[1]);
+		return 0;
+	}
+
+	d->Reset();
+
+	if (params[2])
+	{
+		d->ResetSize();
+	}
+
+	return 1;
+}
+
+static cell AMX_NATIVE_CALL GetPackPosition(AMX* amx, cell* params)
+{
+	CDataPack *d = g_DataPackHandles.lookup(params[1]);
+
+	if (d == NULL)
+	{
+		LogError(amx, AMX_ERR_NATIVE, "Invalid datapack handle provided (%d)", params[1]);
+		return 0;
+	}
+
+	return static_cast<cell>(d->GetPosition());
+}
+
+static cell AMX_NATIVE_CALL SetPackPosition(AMX* amx, cell* params)
+{
+	CDataPack *d = g_DataPackHandles.lookup(params[1]);
+
+	if (d == NULL)
+	{
+		LogError(amx, AMX_ERR_NATIVE, "Invalid datapack handle provided (%d)", params[1]);
+		return 0;
+	}
+
+	if (!d->SetPosition(params[2]))
+	{
+		LogError(amx, AMX_ERR_NATIVE, "Invalid DataPack position, %d is out of bounds", params[2]);
+		return 0;
+	}
+
+	return 1;
+}
+
+static cell AMX_NATIVE_CALL IsPackReadable(AMX* amx, cell* params)
+{
+	CDataPack *d = g_DataPackHandles.lookup(params[1]);
+
+	if (d == NULL)
+	{
+		LogError(amx, AMX_ERR_NATIVE, "Invalid datapack handle provided (%d)", params[1]);
+		return 0;
+	}
+
+	return d->IsReadable(params[2]) ? 1 : 0;
+}
+
+static cell AMX_NATIVE_CALL DestroyDataPack(AMX* amx, cell* params)
+{
+	cell *ptr = get_amxaddr(amx, params[1]);
+
+	CDataPack *d = g_DataPackHandles.lookup(*ptr);
+
+	if (d == NULL)
+	{
+		return 0;
+	}
+
+	if (g_DataPackHandles.destroy(*ptr))
+	{
+		*ptr = 0;
+		return 1;
+	}
+
+	return 0;
+}
+
+
+AMX_NATIVE_INFO g_DatapackNatives[] = 
+{
+	{ "CreateDataPack",				CreateDataPack },
+	{ "WritePackCell",				WritePackCell },
+	{ "WritePackFloat",				WritePackFloat },
+	{ "WritePackString",			WritePackString },
+	{ "ReadPackCell",				ReadPackCell },
+	{ "ReadPackFloat",				ReadPackFloat },
+	{ "ReadPackString",				ReadPackString },
+	{ "ResetPack",					ResetPack },
+	{ "GetPackPosition",			GetPackPosition },
+	{ "SetPackPosition",			SetPackPosition },
+	{ "IsPackReadable",				IsPackReadable },
+	{ "DestroyDataPack",			DestroyDataPack },
+	{NULL,							NULL}
+};