source-sdk-2013-mapbase/mp/src/public/dt_utlvector_common.h
2013-12-02 19:46:31 -08:00

86 lines
2.7 KiB
C++

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
//=============================================================================//
#ifndef DT_UTLVECTOR_COMMON_H
#define DT_UTLVECTOR_COMMON_H
#ifdef _WIN32
#pragma once
#endif
#include "utlvector.h"
typedef void (*EnsureCapacityFn)( void *pVoid, int offsetToUtlVector, int len );
typedef void (*ResizeUtlVectorFn)( void *pVoid, int offsetToUtlVector, int len );
template< class T >
void UtlVector_InitializeAllocatedElements( T *pBase, int count )
{
memset( reinterpret_cast<void*>( pBase ), 0, count * sizeof( T ) );
}
template< class T, class A >
class UtlVectorTemplate
{
public:
static void ResizeUtlVector( void *pStruct, int offsetToUtlVector, int len )
{
CUtlVector<T,A> *pVec = (CUtlVector<T,A>*)((char*)pStruct + offsetToUtlVector);
if ( pVec->Count() < len )
pVec->AddMultipleToTail( len - pVec->Count() );
else if ( pVec->Count() > len )
pVec->RemoveMultiple( len, pVec->Count()-len );
// Ensure capacity
pVec->EnsureCapacity( len );
int nNumAllocated = pVec->NumAllocated();
// This is important to do because EnsureCapacity doesn't actually call the constructors
// on the elements, but we need them to be initialized, otherwise it'll have out-of-range
// values which will piss off the datatable encoder.
UtlVector_InitializeAllocatedElements( pVec->Base() + pVec->Count(), nNumAllocated - pVec->Count() );
}
static void EnsureCapacity( void *pStruct, int offsetToUtlVector, int len )
{
CUtlVector<T,A> *pVec = (CUtlVector<T,A>*)((char*)pStruct + offsetToUtlVector);
pVec->EnsureCapacity( len );
int nNumAllocated = pVec->NumAllocated();
// This is important to do because EnsureCapacity doesn't actually call the constructors
// on the elements, but we need them to be initialized, otherwise it'll have out-of-range
// values which will piss off the datatable encoder.
UtlVector_InitializeAllocatedElements( pVec->Base() + pVec->Count(), nNumAllocated - pVec->Count() );
}
};
template< class T, class A >
inline ResizeUtlVectorFn GetResizeUtlVectorTemplate( CUtlVector<T,A> &vec )
{
return &UtlVectorTemplate<T,A>::ResizeUtlVector;
}
template< class T, class A >
inline EnsureCapacityFn GetEnsureCapacityTemplate( CUtlVector<T,A> &vec )
{
return &UtlVectorTemplate<T,A>::EnsureCapacity;
}
// Format and allocate a string.
char* AllocateStringHelper( PRINTF_FORMAT_STRING const char *pFormat, ... );
// Allocates a string for a data table name. Data table names must be unique, so this will
// assert if you try to allocate a duplicate.
char* AllocateUniqueDataTableName( bool bSendTable, PRINTF_FORMAT_STRING const char *pFormat, ... );
#endif // DT_UTLVECTOR_COMMON_H