hl2sdk/public/tier1/stringpool.h

//========= Copyright (c) 1996-2005, Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
// $NoKeywords: $
//=============================================================================//

#ifndef STRINGPOOL_H
#define STRINGPOOL_H

#if defined( _WIN32 )
#pragma once
#endif

#include "utlrbtree.h"
#include "utlvector.h"
#include "utlbuffer.h"

//-----------------------------------------------------------------------------
// Purpose: Allocates memory for strings, checking for duplicates first,
//			reusing exising strings if duplicate found.
//-----------------------------------------------------------------------------

enum StringPoolCase_t
{
	StringPoolCaseInsensitive,
	StringPoolCaseSensitive
};

class CStringPool
{
public:
	DLL_CLASS_IMPORT CStringPool( StringPoolCase_t caseSensitivity = StringPoolCaseInsensitive );
	DLL_CLASS_IMPORT ~CStringPool();

	DLL_CLASS_IMPORT unsigned int Count() const;

	DLL_CLASS_IMPORT const char * Allocate( const char *pszValue );
	DLL_CLASS_IMPORT void FreeAll();

	// searches for a string already in the pool
	DLL_CLASS_IMPORT const char * Find( const char *pszValue );

protected:
	typedef CUtlRBTree<const char *, unsigned short> CStrSet;

	CThreadFastMutex m_Mutex;
	CStrSet m_Strings;
};

//-----------------------------------------------------------------------------
// Purpose: A reference counted string pool.  
//
// Elements are stored more efficiently than in the conventional string pool, 
// quicker to look up, and storage is tracked via reference counts.  
//
// At some point this should replace CStringPool
//-----------------------------------------------------------------------------
class CCountedStringPool_CI
{
public: // HACK, hash_item_t structure should not be public.

	struct hash_item_t
	{
		char*			pString;
		unsigned short	nNextElement;
		unsigned char	nReferenceCount;
		unsigned char	pad;
	};

	enum
	{
		INVALID_ELEMENT = 0,
		MAX_REFERENCE   = 0xFF,
		HASH_TABLE_SIZE = 1024
	};

	CUtlVector<unsigned short>	m_HashTable;	// Points to each element
	CUtlVector<hash_item_t>		m_Elements;
	unsigned short				m_FreeListStart;
	StringPoolCase_t 			m_caseSensitivity;

public:
	DLL_CLASS_IMPORT CCountedStringPool_CI();
	DLL_CLASS_IMPORT virtual ~CCountedStringPool_CI();

	DLL_CLASS_IMPORT void			FreeAll();

	DLL_CLASS_IMPORT const char		*FindString( const char* pIntrinsic ) const;
	DLL_CLASS_IMPORT const char		*ReferenceString( const char* pIntrinsic );
	DLL_CLASS_IMPORT void			DereferenceString( const char* pIntrinsic );

	// These are only reliable if there are less than 64k strings in your string pool
	DLL_CLASS_IMPORT unsigned short	FindStringHandle( const char* pIntrinsic ) const; 
	DLL_CLASS_IMPORT unsigned short	ReferenceStringHandle( const char* pIntrinsic );
	DLL_CLASS_IMPORT const char		*HandleToString( unsigned short handle ) const;
	DLL_CLASS_IMPORT void			SpewStrings() const;
	DLL_CLASS_IMPORT unsigned int	Hash( const char *pszKey ) const;

	DLL_CLASS_IMPORT bool			SaveToBuffer( CUtlBuffer &buffer ) const;
	DLL_CLASS_IMPORT bool			RestoreFromBuffer( CUtlBuffer &buffer );
};

#endif // STRINGPOOL_H