|
|
//====== Copyright � 1996-2005, Valve Corporation, All rights reserved. =======//
//
// Purpose:
//
// $NoKeywords: $
//
// A growable array class that maintains a free list and keeps elements
// in the same location
//=============================================================================//
#ifndef UTLVECTOR_H
#define UTLVECTOR_H
#ifdef _WIN32
#pragma once
#endif
#include <string.h>
#include "tier0/platform.h"
#include "tier0/dbg.h"
#include "tier0/threadtools.h"
#include "tier1/utlmemory.h"
#include "tier1/utlblockmemory.h"
#include "tier1/strtools.h"
#define FOR_EACH_VEC( vecName, iteratorName ) \
for ( int iteratorName = 0; iteratorName < (vecName).Count(); iteratorName++ ) #define FOR_EACH_VEC_BACK( vecName, iteratorName ) \
for ( int iteratorName = (vecName).Count()-1; iteratorName >= 0; iteratorName-- )
//-----------------------------------------------------------------------------
// The CUtlVector class:
// A growable array class which doubles in size by default.
// It will always keep all elements consecutive in memory, and may move the
// elements around in memory (via a PvRealloc) when elements are inserted or
// removed. Clients should therefore refer to the elements of the vector
// by index (they should *never* maintain pointers to elements in the vector).
//-----------------------------------------------------------------------------
template< class T, class A = CUtlMemory<T> > class CUtlVector { typedef A CAllocator; public: typedef T ElemType_t;
// constructor, destructor
CUtlVector( int growSize = 0, int initSize = 0 ); CUtlVector( T* pMemory, int allocationCount, int numElements = 0 ); ~CUtlVector(); // Copy the array.
CUtlVector<T, A>& operator=( const CUtlVector<T, A> &other );
// element access
T& operator[]( int i ); const T& operator[]( int i ) const; T& Element( int i ); const T& Element( int i ) const; T& Head(); const T& Head() const; T& Tail(); const T& Tail() const;
// Gets the base address (can change when adding elements!)
T* Base() { return m_Memory.Base(); } const T* Base() const { return m_Memory.Base(); }
// Returns the number of elements in the vector
int Count() const;
// Is element index valid?
bool IsValidIndex( int i ) const; static int InvalidIndex();
// Adds an element, uses default constructor
int AddToHead(); int AddToTail(); int InsertBefore( int elem ); int InsertAfter( int elem );
// Adds an element, uses copy constructor
int AddToHead( const T& src ); int AddToTail( const T& src ); int InsertBefore( int elem, const T& src ); int InsertAfter( int elem, const T& src );
// Adds multiple elements, uses default constructor
int AddMultipleToHead( int num ); int AddMultipleToTail( int num ); int AddMultipleToTail( int num, const T *pToCopy ); int InsertMultipleBefore( int elem, int num ); int InsertMultipleBefore( int elem, int num, const T *pToCopy ); int InsertMultipleAfter( int elem, int num );
// Calls RemoveAll() then AddMultipleToTail.
void SetSize( int size ); void SetCount( int count ); void SetCountNonDestructively( int count ); //sets count by adding or removing elements to tail TODO: This should probably be the default behavior for SetCount
// Calls SetSize and copies each element.
void CopyArray( const T *pArray, int size );
// Fast swap
void Swap( CUtlVector< T, A > &vec ); // Add the specified array to the tail.
int AddVectorToTail( CUtlVector<T, A> const &src );
// Finds an element (element needs operator== defined)
int Find( const T& src ) const; void FillWithValue( const T& src );
bool HasElement( const T& src ) const;
// Makes sure we have enough memory allocated to store a requested # of elements
void EnsureCapacity( int num );
// Makes sure we have at least this many elements
void EnsureCount( int num );
// Element removal
void FastRemove( int elem ); // doesn't preserve order
void Remove( int elem ); // preserves order, shifts elements
bool FindAndRemove( const T& src ); // removes first occurrence of src, preserves order, shifts elements
bool FindAndFastRemove( const T& src ); // removes first occurrence of src, doesn't preserve order
void RemoveMultiple( int elem, int num ); // preserves order, shifts elements
void RemoveMultipleFromHead(int num); // removes num elements from tail
void RemoveMultipleFromTail(int num); // removes num elements from tail
void RemoveAll(); // doesn't deallocate memory
// Memory deallocation
void Purge();
// Purges the list and calls delete on each element in it.
void PurgeAndDeleteElements();
// Compacts the vector to the number of elements actually in use
void Compact();
// Set the size by which it grows when it needs to allocate more memory.
void SetGrowSize( int size ) { m_Memory.SetGrowSize( size ); }
int NumAllocated() const; // Only use this if you really know what you're doing!
void Sort( int (__cdecl *pfnCompare)(const T *, const T *) );
// Call this to quickly sort non-contiguously allocated vectors
void InPlaceQuickSort( int (__cdecl *pfnCompare)(const T *, const T *) );
#ifdef DBGFLAG_VALIDATE
void Validate( CValidator &validator, char *pchName ); // Validate our internal structures
#endif // DBGFLAG_VALIDATE
protected: // Can't copy this unless we explicitly do it!
CUtlVector( CUtlVector const& vec ) { Assert(0); }
// Grows the vector
void GrowVector( int num = 1 );
// Shifts elements....
void ShiftElementsRight( int elem, int num = 1 ); void ShiftElementsLeft( int elem, int num = 1 );
CAllocator m_Memory; int m_Size;
#ifndef _X360
// For easier access to the elements through the debugger
// it's in release builds so this can be used in libraries correctly
T *m_pElements;
inline void ResetDbgInfo() { m_pElements = Base(); } #else
inline void ResetDbgInfo() {} #endif
private: void InPlaceQuickSort_r( int (__cdecl *pfnCompare)(const T *, const T *), int nLeft, int nRight ); };
// this is kind of ugly, but until C++ gets templatized typedefs in C++0x, it's our only choice
template < class T > class CUtlBlockVector : public CUtlVector< T, CUtlBlockMemory< T, int > > { public: CUtlBlockVector( int growSize = 0, int initSize = 0 ) : CUtlVector< T, CUtlBlockMemory< T, int > >( growSize, initSize ) {} };
//-----------------------------------------------------------------------------
// The CUtlVectorMT class:
// An array class with spurious mutex protection. Nothing is actually protected
// unless you call Lock and Unlock. Also, the Mutex_t is actually not a type
// but a member which probably isn't used.
//-----------------------------------------------------------------------------
template< class BASE_UTLVECTOR, class MUTEX_TYPE = CThreadFastMutex > class CUtlVectorMT : public BASE_UTLVECTOR, public MUTEX_TYPE { typedef BASE_UTLVECTOR BaseClass; public: // MUTEX_TYPE Mutex_t;
// constructor, destructor
CUtlVectorMT( int growSize = 0, int initSize = 0 ) : BaseClass( growSize, initSize ) {} CUtlVectorMT( typename BaseClass::ElemType_t* pMemory, int numElements ) : BaseClass( pMemory, numElements ) {} };
//-----------------------------------------------------------------------------
// The CUtlVectorFixed class:
// A array class with a fixed allocation scheme
//-----------------------------------------------------------------------------
template< class T, size_t MAX_SIZE > class CUtlVectorFixed : public CUtlVector< T, CUtlMemoryFixed<T, MAX_SIZE > > { typedef CUtlVector< T, CUtlMemoryFixed<T, MAX_SIZE > > BaseClass; public:
// constructor, destructor
CUtlVectorFixed( int growSize = 0, int initSize = 0 ) : BaseClass( growSize, initSize ) {} CUtlVectorFixed( T* pMemory, int numElements ) : BaseClass( pMemory, numElements ) {} };
//-----------------------------------------------------------------------------
// The CUtlVectorFixedGrowable class:
// A array class with a fixed allocation scheme backed by a dynamic one
//-----------------------------------------------------------------------------
template< class T, size_t MAX_SIZE > class CUtlVectorFixedGrowable : public CUtlVector< T, CUtlMemoryFixedGrowable<T, MAX_SIZE > > { typedef CUtlVector< T, CUtlMemoryFixedGrowable<T, MAX_SIZE > > BaseClass;
public: // constructor, destructor
CUtlVectorFixedGrowable( int growSize = 0 ) : BaseClass( growSize, MAX_SIZE ) {} };
//-----------------------------------------------------------------------------
// The CUtlVectorConservative class:
// A array class with a conservative allocation scheme
//-----------------------------------------------------------------------------
template< class T > class CUtlVectorConservative : public CUtlVector< T, CUtlMemoryConservative<T> > { typedef CUtlVector< T, CUtlMemoryConservative<T> > BaseClass; public:
// constructor, destructor
CUtlVectorConservative( int growSize = 0, int initSize = 0 ) : BaseClass( growSize, initSize ) {} CUtlVectorConservative( T* pMemory, int numElements ) : BaseClass( pMemory, numElements ) {} };
//-----------------------------------------------------------------------------
// The CUtlVectorUltra Conservative class:
// A array class with a very conservative allocation scheme, with customizable allocator
// Especialy useful if you have a lot of vectors that are sparse, or if you're
// carefully packing holders of vectors
//-----------------------------------------------------------------------------
#pragma warning(push)
#pragma warning(disable : 4200) // warning C4200: nonstandard extension used : zero-sized array in struct/union
#pragma warning(disable : 4815 ) // warning C4815: 'staticData' : zero-sized array in stack object will have no elements
class CUtlVectorUltraConservativeAllocator { public: static void *Alloc( size_t nSize ) { return malloc( nSize ); }
static void *Realloc( void *pMem, size_t nSize ) { return realloc( pMem, nSize ); }
static void Free( void *pMem ) { free( pMem ); }
static size_t GetSize( void *pMem ) { return mallocsize( pMem ); }
};
template <typename T, typename A = CUtlVectorUltraConservativeAllocator > class CUtlVectorUltraConservative : private A { public: CUtlVectorUltraConservative() { m_pData = StaticData(); }
~CUtlVectorUltraConservative() { RemoveAll(); }
int Count() const { return m_pData->m_Size; }
static int InvalidIndex() { return -1; }
inline bool IsValidIndex( int i ) const { return (i >= 0) && (i < Count()); }
T& operator[]( int i ) { Assert( IsValidIndex( i ) ); return m_pData->m_Elements[i]; }
const T& operator[]( int i ) const { Assert( IsValidIndex( i ) ); return m_pData->m_Elements[i]; }
T& Element( int i ) { Assert( IsValidIndex( i ) ); return m_pData->m_Elements[i]; }
const T& Element( int i ) const { Assert( IsValidIndex( i ) ); return m_pData->m_Elements[i]; }
void EnsureCapacity( int num ) { int nCurCount = Count(); if ( num <= nCurCount ) { return; } if ( m_pData == StaticData() ) { m_pData = (Data_t *)A::Alloc( sizeof(int) + ( num * sizeof(T) ) ); m_pData->m_Size = 0; } else { int nNeeded = sizeof(int) + ( num * sizeof(T) ); int nHave = A::GetSize( m_pData ); if ( nNeeded > nHave ) { m_pData = (Data_t *)A::Realloc( m_pData, nNeeded ); } } }
int AddToTail( const T& src ) { int iNew = Count(); EnsureCapacity( Count() + 1 ); m_pData->m_Elements[iNew] = src; m_pData->m_Size++; return iNew; }
void RemoveAll() { if ( Count() ) { for (int i = m_pData->m_Size; --i >= 0; ) { Destruct(&m_pData->m_Elements[i]); } } if ( m_pData != StaticData() ) { A::Free( m_pData ); m_pData = StaticData();
} }
void PurgeAndDeleteElements() { if ( m_pData != StaticData() ) { for( int i=0; i < m_pData->m_Size; i++ ) { delete Element(i); } RemoveAll(); } }
void FastRemove( int elem ) { Assert( IsValidIndex(elem) );
Destruct( &Element(elem) ); if (Count() > 0) { if ( elem != m_pData->m_Size -1 ) memcpy( &Element(elem), &Element(m_pData->m_Size-1), sizeof(T) ); --m_pData->m_Size; } if ( !m_pData->m_Size ) { A::Free( m_pData ); m_pData = StaticData(); } }
void Remove( int elem ) { Destruct( &Element(elem) ); ShiftElementsLeft(elem); --m_pData->m_Size; if ( !m_pData->m_Size ) { A::Free( m_pData ); m_pData = StaticData(); } }
int Find( const T& src ) const { int nCount = Count(); for ( int i = 0; i < nCount; ++i ) { if (Element(i) == src) return i; } return -1; }
bool FindAndRemove( const T& src ) { int elem = Find( src ); if ( elem != -1 ) { Remove( elem ); return true; } return false; }
bool FindAndFastRemove( const T& src ) { int elem = Find( src ); if ( elem != -1 ) { FastRemove( elem ); return true; } return false; }
struct Data_t { int m_Size; T m_Elements[]; };
Data_t *m_pData; private: void ShiftElementsLeft( int elem, int num = 1 ) { int Size = Count(); Assert( IsValidIndex(elem) || ( Size == 0 ) || ( num == 0 )); int numToMove = Size - elem - num; if ((numToMove > 0) && (num > 0)) { V_memmove( &Element(elem), &Element(elem+num), numToMove * sizeof(T) );
#ifdef _DEBUG
V_memset( &Element(Size-num), 0xDD, num * sizeof(T) ); #endif
} }
static Data_t *StaticData() { static Data_t staticData; Assert( staticData.m_Size == 0 ); return &staticData; } };
#pragma warning(pop)
//-----------------------------------------------------------------------------
// The CCopyableUtlVector class:
// A array class that allows copy construction (so you can nest a CUtlVector inside of another one of our containers)
// WARNING - this class lets you copy construct which can be an expensive operation if you don't carefully control when it happens
// Only use this when nesting a CUtlVector() inside of another one of our container classes (i.e a CUtlMap)
//-----------------------------------------------------------------------------
template< class T > class CCopyableUtlVector : public CUtlVector< T, CUtlMemory<T> > { typedef CUtlVector< T, CUtlMemory<T> > BaseClass; public: CCopyableUtlVector( int growSize = 0, int initSize = 0 ) : BaseClass( growSize, initSize ) {} CCopyableUtlVector( T* pMemory, int numElements ) : BaseClass( pMemory, numElements ) {} virtual ~CCopyableUtlVector() {} CCopyableUtlVector( CCopyableUtlVector const& vec ) { CopyArray( vec.Base(), vec.Count() ); } };
// TODO (Ilya): It seems like all the functions in CUtlVector are simple enough that they should be inlined.
//-----------------------------------------------------------------------------
// constructor, destructor
//-----------------------------------------------------------------------------
template< typename T, class A > inline CUtlVector<T, A>::CUtlVector( int growSize, int initSize ) : m_Memory(growSize, initSize), m_Size(0) { ResetDbgInfo(); }
template< typename T, class A > inline CUtlVector<T, A>::CUtlVector( T* pMemory, int allocationCount, int numElements ) : m_Memory(pMemory, allocationCount), m_Size(numElements) { ResetDbgInfo(); }
template< typename T, class A > inline CUtlVector<T, A>::~CUtlVector() { Purge(); }
template< typename T, class A > inline CUtlVector<T, A>& CUtlVector<T, A>::operator=( const CUtlVector<T, A> &other ) { int nCount = other.Count(); SetSize( nCount ); for ( int i = 0; i < nCount; i++ ) { (*this)[ i ] = other[ i ]; } return *this; }
//-----------------------------------------------------------------------------
// element access
//-----------------------------------------------------------------------------
template< typename T, class A > inline T& CUtlVector<T, A>::operator[]( int i ) { Assert( i < m_Size ); return m_Memory[ i ]; }
template< typename T, class A > inline const T& CUtlVector<T, A>::operator[]( int i ) const { Assert( i < m_Size ); return m_Memory[ i ]; }
template< typename T, class A > inline T& CUtlVector<T, A>::Element( int i ) { Assert( i < m_Size ); return m_Memory[ i ]; }
template< typename T, class A > inline const T& CUtlVector<T, A>::Element( int i ) const { Assert( i < m_Size ); return m_Memory[ i ]; }
template< typename T, class A > inline T& CUtlVector<T, A>::Head() { Assert( m_Size > 0 ); return m_Memory[ 0 ]; }
template< typename T, class A > inline const T& CUtlVector<T, A>::Head() const { Assert( m_Size > 0 ); return m_Memory[ 0 ]; }
template< typename T, class A > inline T& CUtlVector<T, A>::Tail() { Assert( m_Size > 0 ); return m_Memory[ m_Size - 1 ]; }
template< typename T, class A > inline const T& CUtlVector<T, A>::Tail() const { Assert( m_Size > 0 ); return m_Memory[ m_Size - 1 ]; }
//-----------------------------------------------------------------------------
// Count
//-----------------------------------------------------------------------------
template< typename T, class A > inline int CUtlVector<T, A>::Count() const { return m_Size; }
//-----------------------------------------------------------------------------
// Is element index valid?
//-----------------------------------------------------------------------------
template< typename T, class A > inline bool CUtlVector<T, A>::IsValidIndex( int i ) const { return (i >= 0) && (i < m_Size); }
//-----------------------------------------------------------------------------
// Returns in invalid index
//-----------------------------------------------------------------------------
template< typename T, class A > inline int CUtlVector<T, A>::InvalidIndex() { return -1; }
//-----------------------------------------------------------------------------
// Grows the vector
//-----------------------------------------------------------------------------
template< typename T, class A > void CUtlVector<T, A>::GrowVector( int num ) { if (m_Size + num > m_Memory.NumAllocated()) { MEM_ALLOC_CREDIT_CLASS(); m_Memory.Grow( m_Size + num - m_Memory.NumAllocated() ); }
m_Size += num; ResetDbgInfo(); }
//-----------------------------------------------------------------------------
// Sorts the vector
//-----------------------------------------------------------------------------
template< typename T, class A > void CUtlVector<T, A>::Sort( int (__cdecl *pfnCompare)(const T *, const T *) ) { typedef int (__cdecl *QSortCompareFunc_t)(const void *, const void *); if ( Count() <= 1 ) return;
if ( Base() ) { qsort( Base(), Count(), sizeof(T), (QSortCompareFunc_t)(pfnCompare) ); } else { Assert( 0 ); // this path is untested
// if you want to sort vectors that use a non-sequential memory allocator,
// you'll probably want to patch in a quicksort algorithm here
// I just threw in this bubble sort to have something just in case...
for ( int i = m_Size - 1; i >= 0; --i ) { for ( int j = 1; j <= i; ++j ) { if ( pfnCompare( &Element( j - 1 ), &Element( j ) ) < 0 ) { V_swap( Element( j - 1 ), Element( j ) ); } } } } }
//----------------------------------------------------------------------------------------------
// Private function that does the in-place quicksort for non-contiguously allocated vectors.
//----------------------------------------------------------------------------------------------
template< typename T, class A > void CUtlVector<T, A>::InPlaceQuickSort_r( int (__cdecl *pfnCompare)(const T *, const T *), int nLeft, int nRight ) { int nPivot; int nLeftIdx = nLeft; int nRightIdx = nRight;
if ( nRight - nLeft > 0 ) { nPivot = ( nLeft + nRight ) / 2;
while ( ( nLeftIdx <= nPivot ) && ( nRightIdx >= nPivot ) ) { while ( ( pfnCompare( &Element( nLeftIdx ), &Element( nPivot ) ) < 0 ) && ( nLeftIdx <= nPivot ) ) { nLeftIdx++; }
while ( ( pfnCompare( &Element( nRightIdx ), &Element( nPivot ) ) > 0 ) && ( nRightIdx >= nPivot ) ) { nRightIdx--; }
V_swap( Element( nLeftIdx ), Element( nRightIdx ) );
nLeftIdx++; nRightIdx--;
if ( ( nLeftIdx - 1 ) == nPivot ) { nPivot = nRightIdx = nRightIdx + 1; } else if ( nRightIdx + 1 == nPivot ) { nPivot = nLeftIdx = nLeftIdx - 1; } }
InPlaceQuickSort_r( pfnCompare, nLeft, nPivot - 1 ); InPlaceQuickSort_r( pfnCompare, nPivot + 1, nRight ); } }
//----------------------------------------------------------------------------------------------
// Call this to quickly sort non-contiguously allocated vectors. Sort uses a slower bubble sort.
//----------------------------------------------------------------------------------------------
template< typename T, class A > void CUtlVector<T, A>::InPlaceQuickSort( int (__cdecl *pfnCompare)(const T *, const T *) ) { InPlaceQuickSort_r( pfnCompare, 0, Count() - 1 ); }
//-----------------------------------------------------------------------------
// Makes sure we have enough memory allocated to store a requested # of elements
//-----------------------------------------------------------------------------
template< typename T, class A > void CUtlVector<T, A>::EnsureCapacity( int num ) { MEM_ALLOC_CREDIT_CLASS(); m_Memory.EnsureCapacity(num); ResetDbgInfo(); }
//-----------------------------------------------------------------------------
// Makes sure we have at least this many elements
//-----------------------------------------------------------------------------
template< typename T, class A > void CUtlVector<T, A>::EnsureCount( int num ) { if (Count() < num) { AddMultipleToTail( num - Count() ); } }
//-----------------------------------------------------------------------------
// Shifts elements
//-----------------------------------------------------------------------------
template< typename T, class A > void CUtlVector<T, A>::ShiftElementsRight( int elem, int num ) { Assert( IsValidIndex(elem) || ( m_Size == 0 ) || ( num == 0 )); int numToMove = m_Size - elem - num; if ((numToMove > 0) && (num > 0)) V_memmove( &Element(elem+num), &Element(elem), numToMove * sizeof(T) ); }
template< typename T, class A > void CUtlVector<T, A>::ShiftElementsLeft( int elem, int num ) { Assert( IsValidIndex(elem) || ( m_Size == 0 ) || ( num == 0 )); int numToMove = m_Size - elem - num; if ((numToMove > 0) && (num > 0)) { V_memmove( &Element(elem), &Element(elem+num), numToMove * sizeof(T) );
#ifdef _DEBUG
V_memset( &Element(m_Size-num), 0xDD, num * sizeof(T) ); #endif
} }
//-----------------------------------------------------------------------------
// Adds an element, uses default constructor
//-----------------------------------------------------------------------------
template< typename T, class A > inline int CUtlVector<T, A>::AddToHead() { return InsertBefore(0); }
template< typename T, class A > inline int CUtlVector<T, A>::AddToTail() { return InsertBefore( m_Size ); }
template< typename T, class A > inline int CUtlVector<T, A>::InsertAfter( int elem ) { return InsertBefore( elem + 1 ); }
template< typename T, class A > int CUtlVector<T, A>::InsertBefore( int elem ) { // Can insert at the end
Assert( (elem == Count()) || IsValidIndex(elem) );
GrowVector(); ShiftElementsRight(elem); Construct( &Element(elem) ); return elem; }
//-----------------------------------------------------------------------------
// Adds an element, uses copy constructor
//-----------------------------------------------------------------------------
template< typename T, class A > inline int CUtlVector<T, A>::AddToHead( const T& src ) { // Can't insert something that's in the list... reallocation may hose us
Assert( (Base() == NULL) || (&src < Base()) || (&src >= (Base() + Count()) ) ); return InsertBefore( 0, src ); }
template< typename T, class A > inline int CUtlVector<T, A>::AddToTail( const T& src ) { // Can't insert something that's in the list... reallocation may hose us
Assert( (Base() == NULL) || (&src < Base()) || (&src >= (Base() + Count()) ) ); return InsertBefore( m_Size, src ); }
template< typename T, class A > inline int CUtlVector<T, A>::InsertAfter( int elem, const T& src ) { // Can't insert something that's in the list... reallocation may hose us
Assert( (Base() == NULL) || (&src < Base()) || (&src >= (Base() + Count()) ) ); return InsertBefore( elem + 1, src ); }
template< typename T, class A > int CUtlVector<T, A>::InsertBefore( int elem, const T& src ) { // Can't insert something that's in the list... reallocation may hose us
Assert( (Base() == NULL) || (&src < Base()) || (&src >= (Base() + Count()) ) );
// Can insert at the end
Assert( (elem == Count()) || IsValidIndex(elem) );
GrowVector(); ShiftElementsRight(elem); CopyConstruct( &Element(elem), src ); return elem; }
//-----------------------------------------------------------------------------
// Adds multiple elements, uses default constructor
//-----------------------------------------------------------------------------
template< typename T, class A > inline int CUtlVector<T, A>::AddMultipleToHead( int num ) { return InsertMultipleBefore( 0, num ); }
template< typename T, class A > inline int CUtlVector<T, A>::AddMultipleToTail( int num ) { return InsertMultipleBefore( m_Size, num ); }
template< typename T, class A > inline int CUtlVector<T, A>::AddMultipleToTail( int num, const T *pToCopy ) { // Can't insert something that's in the list... reallocation may hose us
Assert( (Base() == NULL) || !pToCopy || (pToCopy + num <= Base()) || (pToCopy >= (Base() + Count()) ) );
return InsertMultipleBefore( m_Size, num, pToCopy ); }
template< typename T, class A > int CUtlVector<T, A>::InsertMultipleAfter( int elem, int num ) { return InsertMultipleBefore( elem + 1, num ); }
template< typename T, class A > void CUtlVector<T, A>::SetCount( int count ) { RemoveAll(); AddMultipleToTail( count ); }
template< typename T, class A > inline void CUtlVector<T, A>::SetSize( int size ) { SetCount( size ); }
template< typename T, class A > void CUtlVector<T, A>::SetCountNonDestructively( int count ) { int delta = count - m_Size; if(delta > 0) AddMultipleToTail( delta ); else if(delta < 0) RemoveMultipleFromTail( -delta ); }
template< typename T, class A > void CUtlVector<T, A>::CopyArray( const T *pArray, int size ) { // Can't insert something that's in the list... reallocation may hose us
Assert( (Base() == NULL) || !pArray || (Base() >= (pArray + size)) || (pArray >= (Base() + Count()) ) );
SetSize( size ); for( int i=0; i < size; i++ ) { (*this)[i] = pArray[i]; } }
template< typename T, class A > void CUtlVector<T, A>::Swap( CUtlVector< T, A > &vec ) { m_Memory.Swap( vec.m_Memory ); V_swap( m_Size, vec.m_Size ); #ifndef _X360
V_swap( m_pElements, vec.m_pElements ); #endif
}
template< typename T, class A > int CUtlVector<T, A>::AddVectorToTail( CUtlVector const &src ) { Assert( &src != this );
int base = Count(); // Make space.
int nSrcCount = src.Count(); EnsureCapacity( base + nSrcCount );
// Copy the elements.
m_Size += nSrcCount; for ( int i=0; i < nSrcCount; i++ ) { CopyConstruct( &Element(base+i), src[i] ); } return base; }
template< typename T, class A > inline int CUtlVector<T, A>::InsertMultipleBefore( int elem, int num ) { if( num == 0 ) return elem;
// Can insert at the end
Assert( (elem == Count()) || IsValidIndex(elem) );
GrowVector(num); ShiftElementsRight( elem, num );
// Invoke default constructors
for (int i = 0; i < num; ++i ) { Construct( &Element( elem+i ) ); }
return elem; }
template< typename T, class A > inline int CUtlVector<T, A>::InsertMultipleBefore( int elem, int num, const T *pToInsert ) { if( num == 0 ) return elem; // Can insert at the end
Assert( (elem == Count()) || IsValidIndex(elem) );
GrowVector(num); ShiftElementsRight( elem, num );
// Invoke default constructors
if ( !pToInsert ) { for (int i = 0; i < num; ++i ) { Construct( &Element( elem+i ) ); } } else { for ( int i=0; i < num; i++ ) { CopyConstruct( &Element( elem+i ), pToInsert[i] ); } }
return elem; }
//-----------------------------------------------------------------------------
// Finds an element (element needs operator== defined)
//-----------------------------------------------------------------------------
template< typename T, class A > int CUtlVector<T, A>::Find( const T& src ) const { for ( int i = 0; i < Count(); ++i ) { if (Element(i) == src) return i; } return -1; }
template< typename T, class A > void CUtlVector<T, A>::FillWithValue( const T& src ) { for ( int i = 0; i < Count(); i++ ) { Element(i) = src; } }
template< typename T, class A > bool CUtlVector<T, A>::HasElement( const T& src ) const { return ( Find(src) >= 0 ); }
//-----------------------------------------------------------------------------
// Element removal
//-----------------------------------------------------------------------------
template< typename T, class A > void CUtlVector<T, A>::FastRemove( int elem ) { Assert( IsValidIndex(elem) );
Destruct( &Element(elem) ); if (m_Size > 0) { if ( elem != m_Size -1 ) memcpy( &Element(elem), &Element(m_Size-1), sizeof(T) ); --m_Size; } }
template< typename T, class A > void CUtlVector<T, A>::Remove( int elem ) { Destruct( &Element(elem) ); ShiftElementsLeft(elem); --m_Size; }
template< typename T, class A > bool CUtlVector<T, A>::FindAndRemove( const T& src ) { int elem = Find( src ); if ( elem != -1 ) { Remove( elem ); return true; } return false; }
template< typename T, class A > bool CUtlVector<T, A>::FindAndFastRemove( const T& src ) { int elem = Find( src ); if ( elem != -1 ) { FastRemove( elem ); return true; } return false; }
template< typename T, class A > void CUtlVector<T, A>::RemoveMultiple( int elem, int num ) { Assert( elem >= 0 ); Assert( elem + num <= Count() );
for (int i = elem + num; --i >= elem; ) Destruct(&Element(i));
ShiftElementsLeft(elem, num); m_Size -= num; }
template< typename T, class A > void CUtlVector<T, A>::RemoveMultipleFromHead( int num ) { Assert( num <= Count() );
for (int i = num; --i >= 0; ) Destruct(&Element(i));
ShiftElementsLeft(0, num); m_Size -= num; }
template< typename T, class A > void CUtlVector<T, A>::RemoveMultipleFromTail( int num ) { Assert( num <= Count() );
for (int i = m_Size-num; i < m_Size; i++) Destruct(&Element(i));
m_Size -= num; }
template< typename T, class A > void CUtlVector<T, A>::RemoveAll() { for (int i = m_Size; --i >= 0; ) { Destruct(&Element(i)); }
m_Size = 0; }
//-----------------------------------------------------------------------------
// Memory deallocation
//-----------------------------------------------------------------------------
template< typename T, class A > inline void CUtlVector<T, A>::Purge() { RemoveAll(); m_Memory.Purge(); ResetDbgInfo(); }
template< typename T, class A > inline void CUtlVector<T, A>::PurgeAndDeleteElements() { for( int i=0; i < m_Size; i++ ) { delete Element(i); } Purge(); }
template< typename T, class A > inline void CUtlVector<T, A>::Compact() { m_Memory.Purge(m_Size); }
template< typename T, class A > inline int CUtlVector<T, A>::NumAllocated() const { return m_Memory.NumAllocated(); }
//-----------------------------------------------------------------------------
// Data and memory validation
//-----------------------------------------------------------------------------
#ifdef DBGFLAG_VALIDATE
template< typename T, class A > void CUtlVector<T, A>::Validate( CValidator &validator, char *pchName ) { validator.Push( typeid(*this).name(), this, pchName );
m_Memory.Validate( validator, "m_Memory" );
validator.Pop(); } #endif // DBGFLAG_VALIDATE
// A vector class for storing pointers, so that the elements pointed to by the pointers are deleted
// on exit.
template<class T> class CUtlVectorAutoPurge : public CUtlVector< T, CUtlMemory< T, int> > { public: ~CUtlVectorAutoPurge( void ) { this->PurgeAndDeleteElements(); }
};
// easy string list class with dynamically allocated strings. For use with V_SplitString, etc.
// Frees the dynamic strings in destructor.
class CUtlStringList : public CUtlVectorAutoPurge< char *> { public: void CopyAndAddToTail( char const *pString ) // clone the string and add to the end
{ char *pNewStr = new char[1 + strlen( pString )]; V_strcpy( pNewStr, pString ); AddToTail( pNewStr ); }
static int __cdecl SortFunc( char * const * sz1, char * const * sz2 ) { return strcmp( *sz1, *sz2 ); }
};
// <Sergiy> placing it here a few days before Cert to minimize disruption to the rest of codebase
class CSplitString: public CUtlVector<char*, CUtlMemory<char*, int> > { public: CSplitString(const char *pString, const char *pSeparator); CSplitString(const char *pString, const char **pSeparators, int nSeparators); ~CSplitString(); //
// NOTE: If you want to make Construct() public and implement Purge() here, you'll have to free m_szBuffer there
//
private: void Construct(const char *pString, const char **pSeparators, int nSeparators); void PurgeAndDeleteElements(); private: char *m_szBuffer; // a copy of original string, with '\0' instead of separators
};
#endif // CCVECTOR_H
|