diff options
Diffstat (limited to 'game/shared/econ/econ_entity_creation.cpp')
| -rw-r--r-- | game/shared/econ/econ_entity_creation.cpp | 180 |
1 files changed, 180 insertions, 0 deletions
diff --git a/game/shared/econ/econ_entity_creation.cpp b/game/shared/econ/econ_entity_creation.cpp new file mode 100644 index 0000000..2461005 --- /dev/null +++ b/game/shared/econ/econ_entity_creation.cpp @@ -0,0 +1,180 @@ +//========= Copyright Valve Corporation, All rights reserved. ============// +// +// Purpose: +// +//============================================================================= + +#include "cbase.h" +#include "econ_entity_creation.h" +#include "utldict.h" +#include "filesystem.h" +#include "gamestringpool.h" +#include "KeyValues.h" +#include "attribute_manager.h" +#include "vgui/ILocalize.h" +#include "tier3/tier3.h" +#include "util_shared.h" + +#ifdef TF_CLIENT_DLL +#include "c_tf_player.h" +#endif // TF_CLIENT_DLL + +//================================================================================== +// GENERATION SYSTEM +//================================================================================== +CItemGeneration g_ItemGenerationSystem; +CItemGeneration *ItemGeneration( void ) +{ + return &g_ItemGenerationSystem; +} + +//----------------------------------------------------------------------------- +// Constructor, destructor +//----------------------------------------------------------------------------- +CItemGeneration::CItemGeneration( void ) +{ +} + +//----------------------------------------------------------------------------- +// Purpose: Generate a random item matching the specified criteria +//----------------------------------------------------------------------------- +CBaseEntity *CItemGeneration::GenerateRandomItem( CItemSelectionCriteria *pCriteria, const Vector &vecOrigin, const QAngle &vecAngles ) +{ + entityquality_t iQuality; + int iChosenItem = ItemSystem()->GenerateRandomItem( pCriteria, &iQuality ); + if ( iChosenItem == INVALID_ITEM_DEF_INDEX ) + return NULL; + + return SpawnItem( iChosenItem, vecOrigin, vecAngles, pCriteria->GetItemLevel(), iQuality, NULL ); +} + +//----------------------------------------------------------------------------- +// Purpose: Generate a random item matching the specified definition index +//----------------------------------------------------------------------------- +CBaseEntity *CItemGeneration::GenerateItemFromDefIndex( int iDefIndex, const Vector &vecOrigin, const QAngle &vecAngles ) +{ + return SpawnItem( iDefIndex, vecOrigin, vecAngles, 1, AE_UNIQUE, NULL ); +} + +//----------------------------------------------------------------------------- +// Purpose: Generate an item from the specified item data +//----------------------------------------------------------------------------- +CBaseEntity *CItemGeneration::GenerateItemFromScriptData( const CEconItemView *pData, const Vector &vecOrigin, const QAngle &vecAngles, const char *pszOverrideClassName ) +{ + return SpawnItem( pData, vecOrigin, vecAngles, pszOverrideClassName ); +} + +//----------------------------------------------------------------------------- +// Purpose: Generate the base item for a class's loadout slot +//----------------------------------------------------------------------------- +CBaseEntity *CItemGeneration::GenerateBaseItem( struct baseitemcriteria_t *pCriteria ) +{ + int iChosenItem = ItemSystem()->GenerateBaseItem( pCriteria ); + if ( iChosenItem == INVALID_ITEM_DEF_INDEX ) + return NULL; + + return SpawnItem( iChosenItem, vec3_origin, vec3_angle, 1, AE_NORMAL, NULL ); +} + +//----------------------------------------------------------------------------- +// Purpose: Create a new instance of the chosen item +//----------------------------------------------------------------------------- +CBaseEntity *CItemGeneration::SpawnItem( int iChosenItem, const Vector &vecAbsOrigin, const QAngle &vecAbsAngles, int iItemLevel, entityquality_t entityQuality, const char *pszOverrideClassName ) +{ + CEconItemDefinition *pData = ItemSystem()->GetStaticDataForItemByDefIndex( iChosenItem ); + if ( !pData ) + return NULL; + + if ( !pszOverrideClassName ) + { + pszOverrideClassName = pData->GetItemClass(); + } + + if ( !pszOverrideClassName ) + return NULL; + + CBaseEntity *pItem = CreateEntityByName( pszOverrideClassName ); + if ( !pItem ) + return NULL; + + // Set the item level & quality + IHasAttributes *pItemInterface = GetAttribInterface( pItem ); + Assert( pItemInterface ); + if ( pItemInterface ) + { + // Setup the script item. Don't generate attributes here, because it'll be done during entity spawn. + CEconItemView *pScriptItem = pItemInterface->GetAttributeContainer()->GetItem(); + pScriptItem->Init( iChosenItem, entityQuality, iItemLevel, false ); + } + + return PostSpawnItem( pItem, pItemInterface, vecAbsOrigin, vecAbsAngles ); +} + +//----------------------------------------------------------------------------- +// Purpose: Create a base entity for the specified item data +//----------------------------------------------------------------------------- +CBaseEntity *CItemGeneration::SpawnItem( const CEconItemView *pData, const Vector &vecAbsOrigin, const QAngle &vecAbsAngles, const char *pszOverrideClassName ) +{ + if ( !pData->GetStaticData() ) + return NULL; + + if ( !pszOverrideClassName ) + { + pszOverrideClassName = pData->GetStaticData()->GetItemClass(); + } + + if ( !pszOverrideClassName ) + return NULL; + + CBaseEntity *pItem = CreateEntityByName( pszOverrideClassName ); + if ( !pItem ) + return NULL; + + // Set the item level & quality + IHasAttributes *pItemInterface = GetAttribInterface( pItem ); + Assert( pItemInterface ); + if ( pItemInterface ) + { + pItemInterface->GetAttributeContainer()->SetItem( pData ); + } + + return PostSpawnItem( pItem, pItemInterface, vecAbsOrigin, vecAbsAngles ); +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +CBaseEntity *CItemGeneration::PostSpawnItem( CBaseEntity *pItem, IHasAttributes *pItemInterface, const Vector &vecAbsOrigin, const QAngle &vecAbsAngles ) +{ +#ifdef CLIENT_DLL + const char *pszPlayerModel = NULL; + if ( pItemInterface ) + { + CEconItemView *pScriptItem = pItemInterface->GetAttributeContainer()->GetItem(); + + int iClass = 0; + int iTeam = 0; +#ifdef TF_CLIENT_DLL + C_TFPlayer *pTFPlayer = ToTFPlayer( GetPlayerByAccountID( pScriptItem->GetAccountID() ) ); + if ( pTFPlayer ) + { + iClass = pTFPlayer->GetPlayerClass()->GetClassIndex(); + iTeam = pTFPlayer->GetTeamNumber(); + } +#endif // TF_CLIENT_DLL + pszPlayerModel = pScriptItem->GetPlayerDisplayModel( iClass, iTeam ); + } + + // If we create a clientside item, we need to force it to initialize attributes + if ( pItem->InitializeAsClientEntity( pszPlayerModel, RENDER_GROUP_OPAQUE_ENTITY ) == false ) + return NULL; +#endif // CLIENT_DLL + + pItem->SetAbsOrigin( vecAbsOrigin ); + pItem->SetAbsAngles( vecAbsAngles ); + + pItem->Spawn(); + pItem->Activate(); + return pItem; +} + |