tf2/tf2_src/public/datamodel/dmvar.h

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose: 
//
//=============================================================================

#ifndef DMVAR_H
#define DMVAR_H
#ifdef _WIN32
#pragma once
#endif


class CDmAttribute;

//-----------------------------------------------------------------------------
// Helper template for external attributes
//-----------------------------------------------------------------------------
template< typename T >
class CDmaVar
{
	typedef typename CDmAttributeInfo< T >::StorageType_t D;

public:
	CDmaVar( );

	// Setup to be used in OnConstruction methods of DmElements
	void Init( CDmElement *pOwner, const char *pAttributeName, int flags = 0 );
	void InitAndSet( CDmElement *pOwner, const char *pAttributeName, const T &value, int flags = 0 );

	// Set/get
	const T& Set( const T &val );
	const T& Get() const;

	// Cast operators
	operator const T&() const;
	const T* operator->() const;

	// Assignment operator
	const CDmaVar<T>& operator=( const CDmaVar<T>& src );

	// Math utility operations
	const T& operator=( const T &val );
	const T& operator+=( const T &val ); 
	const T& operator-=( const T &val ); 
	const T& operator/=( const T &val ); 
	const T& operator*=( const T &val ); 
	const T& operator^=( const T &val ); 
	const T& operator|=( const T &val ); 
	const T& operator&=( const T &val ); 
	T operator++();
	T operator--();
	T operator++( int ); // postfix version..
	T operator--( int ); // postfix version..

	// Returns the attribute associated with the var
	CDmAttribute *GetAttribute();
	const CDmAttribute *GetAttribute() const;

	// Is the attribute dirty?
	bool IsDirty() const;

protected:
	const T& Value() const;
	T& Value();
	const D& Storage() const;
	D& Storage();

private:
	D m_Storage;

protected:
	CDmAttribute *m_pAttribute;
};

//-----------------------------------------------------------------------------
// Specialization for string
//-----------------------------------------------------------------------------
class CDmaString : public CDmaVar< CUtlString >
{
public:
	const char *Get( ) const;
	operator const char*() const;

	void Set( const char *pValue );
	CDmaString &operator=( const char *src );
	const CDmaString& operator=( const CDmaString& src );

	// Returns strlen
	int	Length() const;
};

#endif // DMVAR_H