Modified source engine (2017) developed by valve and leaked in 2020. Not for commercial purporses
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

97 lines
2.7 KiB

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// The copyright to the contents herein is the property of Valve, L.L.C.
// The contents may be used and/or copied only with the written permission of
// Valve, L.L.C., or in accordance with the terms and conditions stipulated in
// the agreement/contract under which the contents have been supplied.
//
// Purpose:
//
// $Workfile: $
// $Date: $
//
//-----------------------------------------------------------------------------
// $Log: $
//
// $NoKeywords: $
//=============================================================================
#ifndef TSMULTIMEMPOOL_H
#define TSMULTIMEMPOOL_H
#ifdef _WIN32
#pragma once
#endif
#include "tier1/utlmap.h"
#include "tier1/mempool.h"
#include "tier1/tsmempool.h"
//-----------------------------------------------------------------------------
// Purpose: A container of a range of mem pool sizes (for network buffers for example)
// and a raw alloc capability (for sizes greater than any contained mem pool).
//-----------------------------------------------------------------------------
class CThreadSafeMultiMemoryPool
{
public:
struct MemPoolConfig_t
{
uint32 m_cubBlockSize;
uint32 m_cubDefaultPoolSize;
};
CThreadSafeMultiMemoryPool( const MemPoolConfig_t *pnBlock, int cnMemPoolConfig, int nGrowMode = UTLMEMORYPOOL_GROW_FAST );
~CThreadSafeMultiMemoryPool();
// Allocate a block of at least nAllocSize bytes
void* Alloc( uint32 cubAlloc );
// Free a previously alloc'd block
void Free(void *pvMem);
// ReAllocate a previously allocated block to a new size
void* ReAlloc( void *pvMem, uint32 cubAlloc );
// Frees everything
void Clear();
// alloc size for this bit of alloc'd memory
int CubAllocSize( void *pvMem );
// prints details about our contained memory
void PrintStats();
// total number of alloc'd elements
int Count();
// Return the total size in MB allocated for this pool
int CMBPoolSize() const;
// Return the amount of memory in use
int CMBPoolSizeInUse() const;
#ifdef DBGFLAG_VALIDATE
void Validate( CValidator &validator, const char *pchName ); // Validate our internal structures
#endif // DBGFLAG_VALIDATE
private:
struct MemPoolRecord_t
{
CThreadSafeMemoryPool *m_pMemPool;
uint32 m_nBlockSize;
};
CUtlVector<MemPoolRecord_t> m_VecMemPool; // stores our list of mem pools
uint32 m_nBlockSizeMax;
CUtlVector<MemPoolRecord_t *> m_VecMemPoolLookup; // quick lookup table of mempools
struct RawAllocation_t
{
void *m_pvMem;
uint32 m_nBlockSize;
};
CUtlMap<void *,RawAllocation_t,int> m_MapRawAllocation; // stores our list of raw alloc'd mem
CThreadFastMutex m_mutexRawAllocations;
uint32 m_cubReallocedTotal;
};
#endif // TSMULTIMEMPOOL_H