#ifndef _SIMPLEARRAY_H_ #define _SIMPLEARRAY_H_ #include #ifndef SIMPLEARRAY_STARTSIZE #define SIMPLEARRAY_STARTSIZE 1 #endif template class CSimpleArray { public: T* m_aT; int m_nSize; int m_nAllocSize; // Construction/destruction CSimpleArray() : m_aT(NULL), m_nSize(0), m_nAllocSize(0) { } ~CSimpleArray() { RemoveAll(); } // Operations int GetSize() const { return m_nSize; } BOOL Add(T& t) { if(m_nSize == m_nAllocSize) { T* aT; int nNewAllocSize = (m_nAllocSize == 0) ? SIMPLEARRAY_STARTSIZE : (m_nSize * 2); if (m_aT) aT = (T*)realloc(m_aT, nNewAllocSize * sizeof(T)); else aT = (T*)malloc(nNewAllocSize * sizeof(T)); if(aT == NULL) return FALSE; m_nAllocSize = nNewAllocSize; m_aT = aT; } m_nSize++; SetAtIndex(m_nSize - 1, t); return TRUE; } BOOL Remove(T& t) { int nIndex = Find(t); if(nIndex == -1) return FALSE; return RemoveAt(nIndex); } BOOL RemoveAt(int nIndex) { //---- always call the dtr ---- #if _MSC_VER >= 1200 m_aT[nIndex].~T(); #else T* MyT; MyT = &m_aT[nIndex]; MyT->~T(); #endif //---- if target entry is not at end, compact the array ---- if(nIndex != (m_nSize - 1)) { memmove((void*)&m_aT[nIndex], (void*)&m_aT[nIndex + 1], (m_nSize - (nIndex + 1)) * sizeof(T)); } m_nSize--; return TRUE; } void RemoveAll() { if(m_aT != NULL) { for(int i = 0; i < m_nSize; i++) { #if _MSC_VER >= 1200 m_aT[i].~T(); #else T* MyT; MyT = &m_aT[i]; MyT->~T(); #endif } free(m_aT); m_aT = NULL; } m_nSize = 0; m_nAllocSize = 0; } T& operator[] (int nIndex) const { DBG_ASSERT(nIndex >= 0 && nIndex < m_nSize); return m_aT[nIndex]; } T* GetData() const { return m_aT; } // Implementation class Wrapper { public: Wrapper(T& _t) : t(_t) { } template void *operator new(size_t, _Ty* p) { return p; } T t; }; void SetAtIndex(int nIndex, T& t) { DBG_ASSERT(nIndex >= 0 && nIndex < m_nSize); new(m_aT + nIndex) Wrapper(t); } int Find(T& t) const { for(int i = 0; i < m_nSize; i++) { if(m_aT[i] == t) return i; } return -1; // not found } }; #endif