Team Fortress 2 Source Code as on 22/4/2020
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.
 
 
 
 
 
 

138 lines
3.8 KiB

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
// $NoKeywords: $
//===========================================================================//
#ifndef DMECONTROLS_UTILS_H
#define DMECONTROLS_UTILS_H
#ifdef _WIN32
#pragma once
#endif
#include "tier1/KeyValues.h"
#include "datamodel/dmelement.h"
#include "datamodel/dmattribute.h"
#include "datamodel/dmattributevar.h"
#include "movieobjects/timeutils.h"
//-----------------------------------------------------------------------------
// Helper method to insert + extract DmElement handles into keyvalues
//-----------------------------------------------------------------------------
inline void SetElementKeyValue( KeyValues *pKeyValues, const char *pName, CDmElement *pElement )
{
pKeyValues->SetInt( pName, pElement ? pElement->GetHandle() : DMELEMENT_HANDLE_INVALID );
}
template< class T >
T* GetElementKeyValue( KeyValues *pKeyValues, const char *pName )
{
DmElementHandle_t h = (DmElementHandle_t)pKeyValues->GetInt( pName, DMELEMENT_HANDLE_INVALID );
return GetElement<T>( h );
}
inline KeyValues *CreateElementKeyValues( const char *pName, const char *pKey, CDmElement *pElement )
{
return new KeyValues( pName, pKey, pElement ? ( int )pElement->GetHandle() : DMELEMENT_HANDLE_INVALID );
}
inline void AddStandardElementKeys( KeyValues *pKeyValues, CDmElement *pElement )
{
SetElementKeyValue( pKeyValues, "dmeelement", pElement );
if ( pElement )
{
char buf[ 256 ];
UniqueIdToString( pElement->GetId(), buf, sizeof( buf ) );
pKeyValues->SetString( "text", buf );
pKeyValues->SetString( "type", pElement->GetTypeString() );
}
}
//-----------------------------------------------------------------------------
// Helper method to insert + extract DmeTime_t into keyvalues
//-----------------------------------------------------------------------------
inline void SetDmeTimeKeyValue( KeyValues *pKeyValues, const char *pName, DmeTime_t t )
{
pKeyValues->SetInt( pName, t.GetTenthsOfMS() );
}
inline DmeTime_t GetDmeTimeKeyValue( KeyValues *pKeyValues, const char *pName, DmeTime_t defaultTime = DMETIME_ZERO )
{
return DmeTime_t( pKeyValues->GetInt( pName, defaultTime.GetTenthsOfMS() ) );
}
inline bool ElementTree_IsArrayItem( KeyValues *itemData )
{
return !itemData->IsEmpty( "arrayIndex" );
}
inline CDmAttribute *ElementTree_GetAttribute( KeyValues *itemData )
{
CDmElement *pOwner = GetElementKeyValue< CDmElement >( itemData, "ownerelement" );
if ( !pOwner )
return NULL;
const char *pAttributeName = itemData->GetString( "attributeName", "" );
return pOwner->GetAttribute( pAttributeName );
}
inline DmAttributeType_t ElementTree_GetAttributeType( KeyValues *itemData )
{
CDmElement *pOwner = GetElementKeyValue< CDmElement >( itemData, "ownerelement" );
if ( !pOwner )
return AT_UNKNOWN;
const char *pAttributeName = itemData->GetString( "attributeName", "" );
CDmAttribute *pAttribute = pOwner->GetAttribute( pAttributeName );
if ( !pAttribute )
return AT_UNKNOWN;
return pAttribute->GetType();
}
inline bool ElementTree_GetDroppableItems( CUtlVector< KeyValues * >& msglist, const char *elementType, CUtlVector< CDmElement * >& list )
{
int c = msglist.Count();
for ( int i = 0; i < c; ++i )
{
KeyValues *data = msglist[ i ];
CDmElement *e = GetElementKeyValue<CDmElement>( data, "dmeelement" );
if ( !e )
{
continue;
}
//if ( !e->IsA( elementType ) )
//{
// continue;
//}
list.AddToTail( e );
}
return list.Count() != 0;
}
inline void ElementTree_RemoveListFromArray( CDmAttribute *pArrayAttribute, CUtlVector< CDmElement * >& list )
{
CDmrElementArray<> array( pArrayAttribute );
int c = array.Count();
for ( int i = c - 1 ; i >= 0 ; --i )
{
CDmElement *element = array[ i ];
if ( list.Find( element ) != list.InvalidIndex() )
{
array.Remove( i );
}
}
}
#endif // DMECONTROLS_UTILS_H