dvander@1156: /** dvander@1156: * vim: set ts=4 : dvander@1231: * ================================================================ dvander@1231: * SourceMod dvander@1231: * Copyright (C) 2004-2007 AlliedModders LLC. All rights reserved. dvander@1231: * ================================================================ dvander@1156: * dvander@1231: * This program is free software; you can redistribute it and/or dvander@1231: * modify it under the terms of the GNU General Public License, dvander@1231: * version 3.0, as published by the Free Software Foundation. dvander@1231: * dvander@1231: * This program is distributed in the hope that it will be useful, dvander@1231: * but WITHOUT ANY WARRANTY; without even the implied warranty of dvander@1231: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the dvander@1231: * GNU General Public License for more details. dvander@1231: * dvander@1231: * You should have received a copy of the GNU General Public License dvander@1231: * along with this program. If not, see . dvander@1231: * dvander@1231: * As a special exception, AlliedModders LLC gives you permission to dvander@1231: * link the code of this program (as well as its derivative works) to dvander@1231: * "Half-Life 2," the "Source Engine," the "SourcePawn JIT," and any dvander@1231: * Game MODs that run on software by the Valve Corporation. You must dvander@1231: * obey the GNU General Public License in all respects for all other dvander@1231: * code used. Additionally, AlliedModders LLC grants this exception dvander@1231: * to all derivative works. AlliedModders LLC defines further dvander@1231: * exceptions, found in LICENSE.txt (as of this writing, version dvander@1231: * JULY-31-2007), or . dvander@1156: * dvander@1156: * Version: $Id$ dvander@1156: */ dvander@1156: dvander@1156: #include dvander@1156: #include dvander@1156: dvander@1156: class CellArray dvander@1156: { dvander@1156: public: dvander@1156: CellArray(size_t blocksize) : m_Data(NULL), m_BlockSize(blocksize), m_AllocSize(0), m_Size(0) dvander@1156: { dvander@1156: } dvander@1156: ~CellArray() dvander@1156: { dvander@1156: free(m_Data); dvander@1156: } dvander@1156: size_t size() const dvander@1156: { dvander@1156: return m_Size; dvander@1156: } dvander@1156: cell_t *push() dvander@1156: { dvander@1156: if (!GrowIfNeeded(1)) dvander@1156: { dvander@1156: return NULL; dvander@1156: } dvander@1156: cell_t *arr = &m_Data[m_Size * m_BlockSize]; dvander@1156: m_Size++; dvander@1156: return arr; dvander@1156: } dvander@1156: cell_t *at(size_t b) const dvander@1156: { dvander@1156: return &m_Data[b * m_BlockSize]; dvander@1156: } dvander@1156: size_t blocksize() const dvander@1156: { dvander@1156: return m_BlockSize; dvander@1156: } dvander@1156: void clear() dvander@1156: { dvander@1156: m_Size = 0; dvander@1156: } dvander@1156: bool swap(size_t item1, size_t item2) dvander@1156: { dvander@1156: /* Make sure there is extra space available */ dvander@1156: if (!GrowIfNeeded(1)) dvander@1156: { dvander@1156: return false; dvander@1156: } dvander@1156: dvander@1156: cell_t *pri = at(item1); dvander@1156: cell_t *alt = at(item2); dvander@1156: dvander@1156: /* Get our temporary array 1 after the limit */ dvander@1156: cell_t *temp = &m_Data[m_Size * m_BlockSize]; dvander@1156: dvander@1156: memcpy(temp, pri, sizeof(cell_t) * m_BlockSize); dvander@1156: memcpy(pri, alt, sizeof(cell_t) * m_BlockSize); dvander@1156: memcpy(alt, temp, sizeof(cell_t) * m_BlockSize); dvander@1156: dvander@1156: return true; dvander@1156: } dvander@1156: void remove(size_t index) dvander@1156: { dvander@1156: /* If we're at the end, take the easy way out */ dvander@1156: if (index == m_Size - 1) dvander@1156: { dvander@1156: m_Size--; dvander@1156: return; dvander@1156: } dvander@1156: dvander@1156: /* Otherwise, it's time to move stuff! */ dvander@1156: size_t remaining_indexes = (m_Size - 1) - index; dvander@1156: cell_t *src = at(index + 1); dvander@1156: cell_t *dest = at(index); dvander@1156: memmove(dest, src, sizeof(cell_t) * m_BlockSize * remaining_indexes); dvander@1156: dvander@1156: m_Size--; dvander@1156: } dvander@1156: cell_t *insert_at(size_t index) dvander@1156: { dvander@1156: /* Make sure it'll fit */ dvander@1156: if (!GrowIfNeeded(1)) dvander@1156: { dvander@1156: return NULL; dvander@1156: } dvander@1156: dvander@1156: /* move everything up */ dvander@1156: cell_t *src = at(index); dvander@1156: cell_t *dst = at(index + 1); dvander@1156: memmove(dst, src, sizeof(cell_t) * m_BlockSize * m_Size); dvander@1156: dvander@1156: m_Size++; dvander@1156: dvander@1156: return src; dvander@1156: } dvander@1156: bool resize(size_t count) dvander@1156: { dvander@1156: if (count <= m_Size) dvander@1156: { dvander@1156: m_Size = count; dvander@1156: return true; dvander@1156: } teame06@1198: teame06@1198: if(!GrowIfNeeded(count - m_Size)) teame06@1198: { teame06@1198: return false; teame06@1198: } teame06@1198: teame06@1198: m_Size = count; teame06@1198: return true; dvander@1156: } dvander@1156: private: dvander@1156: bool GrowIfNeeded(size_t count) dvander@1156: { dvander@1156: /* Shortcut out if we can store this */ dvander@1156: if (m_Size + count <= m_AllocSize) dvander@1156: { dvander@1156: return true; dvander@1156: } dvander@1156: /* Set a base allocation size of 8 items */ dvander@1156: if (!m_AllocSize) dvander@1156: { dvander@1156: m_AllocSize = 8; dvander@1156: } dvander@1156: /* If it's not enough, keep doubling */ dvander@1156: while (m_Size + count > m_AllocSize) dvander@1156: { dvander@1156: m_AllocSize *= 2; dvander@1156: } dvander@1156: /* finally, allocate the new block */ dvander@1156: if (m_Data) dvander@1156: { dvander@1156: m_Data = (cell_t *)realloc(m_Data, sizeof(cell_t) * m_BlockSize * m_AllocSize); dvander@1156: } else { dvander@1156: m_Data = (cell_t *)malloc(sizeof(cell_t) * m_BlockSize * m_AllocSize); dvander@1156: } dvander@1156: return (m_Data != NULL); dvander@1156: } dvander@1156: private: dvander@1156: cell_t *m_Data; dvander@1156: size_t m_BlockSize; dvander@1156: size_t m_AllocSize; dvander@1156: size_t m_Size; dvander@1156: };