dvander@1156: /**
dvander@1156: * vim: set ts=4 :
ds@1330: * =============================================================================
dvander@1231: * SourceMod
dvander@1231: * Copyright (C) 2004-2007 AlliedModders LLC. All rights reserved.
ds@1330: * =============================================================================
dvander@1156: *
ds@1330: * This program is free software; you can redistribute it and/or modify it under
ds@1330: * the terms of the GNU General Public License, version 3.0, as published by the
ds@1330: * Free Software Foundation.
dvander@1231: *
ds@1330: * This program is distributed in the hope that it will be useful, but WITHOUT
ds@1330: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
ds@1330: * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
ds@1330: * details.
dvander@1231: *
ds@1330: * You should have received a copy of the GNU General Public License along with
ds@1330: * this program. If not, see .
dvander@1231: *
ds@1330: * As a special exception, AlliedModders LLC gives you permission to link the
ds@1330: * code of this program (as well as its derivative works) to "Half-Life 2," the
ds@1330: * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
ds@1330: * by the Valve Corporation. You must obey the GNU General Public License in
ds@1330: * all respects for all other code used. Additionally, AlliedModders LLC grants
ds@1330: * this exception to all derivative works. AlliedModders LLC defines further
ds@1330: * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
ds@1330: * or .
dvander@1156: *
dvander@1156: * Version: $Id$
dvander@1156: */
dvander@1156:
dvander@1156: #include
dvander@1156: #include "sm_globals.h"
dvander@1156: #include "sm_stringutil.h"
dvander@1156: #include "CellArray.h"
dvander@1156: #include "HandleSys.h"
dvander@1156:
dvander@1156: HandleType_t htCellArray;
dvander@1156:
dvander@1156: class CellArrayHelpers :
dvander@1156: public SMGlobalClass,
dvander@1156: public IHandleTypeDispatch
dvander@1156: {
dvander@1156: public: //SMGlobalClass
dvander@1156: void OnSourceModAllInitialized()
dvander@1156: {
dvander@1156: htCellArray = g_HandleSys.CreateType("CellArray", this, 0, NULL, NULL, g_pCoreIdent, NULL);
dvander@1156: }
dvander@1156: void OnSourceModShutdown()
dvander@1156: {
dvander@1156: g_HandleSys.RemoveType(htCellArray, g_pCoreIdent);
dvander@1156: }
dvander@1156: public: //IHandleTypeDispatch
dvander@1156: void OnHandleDestroy(HandleType_t type, void *object)
dvander@1156: {
dvander@1156: CellArray *array = (CellArray *)object;
dvander@1156: delete array;
dvander@1156: }
dvander@1156: } s_CellArrayHelpers;
dvander@1156:
dvander@1156: static cell_t CreateArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: if (!params[1])
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid block size (must be > 0)");
dvander@1156: }
dvander@1156:
dvander@1156: CellArray *array = new CellArray(params[1]);
dvander@1156:
dvander@1156: if (params[2])
dvander@1156: {
dvander@1156: array->resize(params[2]);
dvander@1156: }
dvander@1156:
dvander@1156: Handle_t hndl = g_HandleSys.CreateHandle(htCellArray, array, pContext->GetIdentity(), g_pCoreIdent, NULL);
dvander@1156: if (!hndl)
dvander@1156: {
dvander@1156: delete array;
dvander@1156: }
dvander@1156:
dvander@1156: return hndl;
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t ClearArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: array->clear();
dvander@1156:
dvander@1156: return 1;
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t ResizeArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: if (!array->resize(params[2]))
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Unable to resize array to \"%u\"", params[2]);
dvander@1156: }
dvander@1156:
dvander@1156: return 1;
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t GetArraySize(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: return (cell_t)array->size();
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t PushArrayCell(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: cell_t *blk = array->push();
dvander@1156: if (!blk)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Failed to grow array");
dvander@1156: }
dvander@1156:
dvander@1156: *blk = params[2];
dvander@1156:
dvander@1156: return (cell_t)(array->size() - 1);
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t PushArrayString(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: cell_t *blk = array->push();
dvander@1156: if (!blk)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Failed to grow array");
dvander@1156: }
dvander@1156:
dvander@1156: char *str;
dvander@1156: pContext->LocalToString(params[2], &str);
dvander@1156:
dvander@1156: strncopy((char *)blk, str, array->blocksize() * sizeof(cell_t));
dvander@1156:
dvander@1156: return (cell_t)(array->size() - 1);
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t PushArrayArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: cell_t *blk = array->push();
dvander@1156: if (!blk)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Failed to grow array");
dvander@1156: }
dvander@1156:
dvander@1156: cell_t *addr;
dvander@1156: pContext->LocalToPhysAddr(params[2], &addr);
dvander@1156:
dvander@1156: size_t indexes = array->blocksize();
dvander@1156: if (params[3] != -1 && (size_t)params[3] <= array->blocksize())
dvander@1156: {
dvander@1156: indexes = params[3];
dvander@1156: }
dvander@1156:
dvander@1156: memcpy(blk, addr, sizeof(cell_t) * indexes);
dvander@1156:
dvander@1156: return (cell_t)(array->size() - 1);
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t GetArrayCell(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: size_t idx = (size_t)params[2];
dvander@1156: if (idx >= array->size())
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: }
dvander@1156:
dvander@1156: cell_t *blk = array->at(idx);
dvander@1156:
dvander@1156: idx = (size_t)params[3];
dvander@1159: if (params[4] == 0)
dvander@1156: {
dvander@1156: if (idx >= array->blocksize())
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid block %d (blocksize: %d)", idx, array->blocksize());
dvander@1156: }
dvander@1156: return blk[idx];
dvander@1156: } else {
dvander@1156: if (idx >= array->blocksize() * 4)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid byte %d (blocksize: %d bytes)", idx, array->blocksize() * 4);
dvander@1156: }
dvander@1156: return (cell_t)*((char *)blk + idx);
dvander@1156: }
dvander@1156:
dvander@1156: return 0;
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t GetArrayString(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: size_t idx = (size_t)params[2];
dvander@1156: if (idx >= array->size())
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: }
dvander@1156:
dvander@1156: cell_t *blk = array->at(idx);
dvander@1156: size_t numWritten = 0;
dvander@1156:
dvander@1156: pContext->StringToLocalUTF8(params[3], params[4], (char *)blk, &numWritten);
dvander@1156:
dvander@1156: return numWritten;
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t GetArrayArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: size_t idx = (size_t)params[2];
dvander@1156: if (idx >= array->size())
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: }
dvander@1156:
dvander@1156: cell_t *blk = array->at(idx);
dvander@1156: size_t indexes = array->blocksize();
dvander@1156: if (params[4] != -1 && (size_t)params[4] <= array->blocksize())
dvander@1156: {
dvander@1156: indexes = params[4];
dvander@1156: }
dvander@1156:
dvander@1156: cell_t *addr;
dvander@1156: pContext->LocalToPhysAddr(params[3], &addr);
dvander@1156:
dvander@1156: memcpy(addr, blk, sizeof(cell_t) * indexes);
dvander@1156:
dvander@1156: return indexes;
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t SetArrayCell(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: size_t idx = (size_t)params[2];
dvander@1156: if (idx >= array->size())
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: }
dvander@1156:
dvander@1156: cell_t *blk = array->at(idx);
dvander@1156:
dvander@1156: idx = (size_t)params[4];
dvander@1159: if (params[5] == 0)
dvander@1156: {
dvander@1156: if (idx >= array->blocksize())
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid block %d (blocksize: %d)", idx, array->blocksize());
dvander@1156: }
dvander@1156: blk[idx] = params[3];
dvander@1156: } else {
dvander@1156: if (idx >= array->blocksize() * 4)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid byte %d (blocksize: %d bytes)", idx, array->blocksize() * 4);
dvander@1156: }
dvander@1156: *((char *)blk + idx) = (char)params[3];
dvander@1156: }
dvander@1156:
dvander@1156: return 1;
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t SetArrayString(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: size_t idx = (size_t)params[2];
dvander@1156: if (idx >= array->size())
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: }
dvander@1156:
dvander@1156: cell_t *blk = array->at(idx);
dvander@1156:
dvander@1156: char *str;
dvander@1156: pContext->LocalToString(params[3], &str);
dvander@1156:
dvander@1156: return strncopy((char *)blk, str, array->blocksize() * sizeof(cell_t));
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t SetArrayArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: size_t idx = (size_t)params[2];
dvander@1156: if (idx >= array->size())
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: }
dvander@1156:
dvander@1156: cell_t *blk = array->at(idx);
dvander@1156: size_t indexes = array->blocksize();
dvander@1156: if (params[4] != -1 && (size_t)params[4] <= array->blocksize())
dvander@1156: {
dvander@1156: indexes = params[4];
dvander@1156: }
dvander@1156:
dvander@1156: cell_t *addr;
dvander@1156: pContext->LocalToPhysAddr(params[3], &addr);
dvander@1156:
dvander@1156: memcpy(blk, addr, sizeof(cell_t) * indexes);
dvander@1156:
dvander@1156: return indexes;
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t ShiftArrayUp(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: size_t idx = (size_t)params[2];
dvander@1156: if (idx >= array->size())
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: }
dvander@1156:
dvander@1156: array->insert_at(idx);
dvander@1156:
dvander@1156: return 1;
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t RemoveFromArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: size_t idx = (size_t)params[2];
dvander@1156: if (idx >= array->size())
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: }
dvander@1156:
dvander@1156: array->remove(idx);
dvander@1156:
dvander@1156: return 1;
dvander@1156: }
dvander@1156:
dvander@1156: static cell_t SwapArrayItems(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: CellArray *array;
dvander@1156: HandleError err;
dvander@1156: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156:
dvander@1156: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array))
dvander@1156: != HandleError_None)
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: }
dvander@1156:
dvander@1156: size_t idx1 = (size_t)params[2];
dvander@1156: size_t idx2 = (size_t)params[3];
dvander@1156: if (idx1 >= array->size())
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx1, array->size());
dvander@1156: }
dvander@1156: if (idx2 >= array->size())
dvander@1156: {
dvander@1156: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx2, array->size());
dvander@1156: }
dvander@1156:
dvander@1156: array->swap(idx1, idx2);
dvander@1156:
dvander@1156: return 1;
dvander@1156: }
dvander@1156:
dvander@1156: REGISTER_NATIVES(cellArrayNatives)
dvander@1156: {
dvander@1156: {"ClearArray", ClearArray},
dvander@1156: {"CreateArray", CreateArray},
dvander@1156: {"GetArrayArray", GetArrayArray},
dvander@1156: {"GetArrayCell", GetArrayCell},
dvander@1156: {"GetArraySize", GetArraySize},
dvander@1156: {"GetArrayString", GetArrayString},
dvander@1156: {"ResizeArray", ResizeArray},
dvander@1156: {"PushArrayArray", PushArrayArray},
dvander@1156: {"PushArrayCell", PushArrayCell},
dvander@1156: {"PushArrayString", PushArrayString},
dvander@1156: {"RemoveFromArray", RemoveFromArray},
dvander@1156: {"SetArrayCell", SetArrayCell},
dvander@1156: {"SetArrayString", SetArrayString},
dvander@1156: {"SetArrayArray", SetArrayArray},
dvander@1156: {"ShiftArrayUp", ShiftArrayUp},
dvander@1156: {"SwapArrayItems", SwapArrayItems},
dvander@1156: {NULL, NULL},
dvander@1156: };