1HEADmaster

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;
+}
+