1 files changed, 250 insertions, 0 deletions
diff --git a/game/shared/econ/econ_dynamic_recipe.cpp b/game/shared/econ/econ_dynamic_recipe.cpp
new file mode 100644
index 0000000..ee6fb17
--- /dev/null
+++ b/game/shared/econ/econ_dynamic_recipe.cpp
@@ -0,0 +1,250 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: Functions related to dynamic recipes
+//
+//=============================================================================
+
+
+#include "cbase.h"
+#include "econ_dynamic_recipe.h"
+#ifndef GC_DLL
+	#include "quest_objective_manager.h"
+#endif
+
+// This pattern was chosen to not be:
+//		- a valid string acceptable for user-input (ie., custom name)
+//		- a sensical float bit pattern
+//		- a common int bit pattern
+//		- meaningful Unicode data
+const char *g_pszAttrEncodeSeparator = "|\x01\x02\x01\x03|\x01\x02\x01\x03|";
+
+CRecipeComponentMatchingIterator::CRecipeComponentMatchingIterator( const IEconItemInterface *pSourceItem,
+																	const IEconItemInterface *pTargetItem )
+	: m_pSourceItem( pSourceItem )
+	, m_pTargetItem( pTargetItem )
+	, m_bIgnoreCompleted( true )
+	, m_nInputsTotal( 0 )
+	, m_nInputsFulfilled( 0 )
+	, m_nOutputsTotal( 0 )
+{}
+
+bool CRecipeComponentMatchingIterator::OnIterateAttributeValue( const CEconItemAttributeDefinition *pAttrDef, const CAttribute_DynamicRecipeComponent& value )
+{
+	// Don't count ourselves as a match!
+	if ( m_pSourceItem && m_pTargetItem && m_pSourceItem->GetID() == m_pTargetItem->GetID() )
+		return true;
+
+	// If this isn't a match and the item isn't NULL, we skip.  We consider NULL to mean
+	// that we want to tally ALL attributes of this type
+	if ( !DefinedItemAttribMatch( value, m_pTargetItem ) && m_pTargetItem != NULL )
+		return true;
+
+	// Dont let non-craftable items through
+	if ( m_pTargetItem && !m_pTargetItem->IsUsableInCrafting() )
+		return true;
+
+	// Is this an output?
+	if ( value.component_flags() & DYNAMIC_RECIPE_FLAG_IS_OUTPUT )
+	{
+		m_vecMatchingOutputs.AddToTail( pAttrDef );
+		m_nOutputsTotal += value.num_required();
+	}
+	else
+	{
+		m_vecMatchingInputs.AddToTail( pAttrDef );
+		m_nInputsTotal += value.num_required();
+		m_nInputsFulfilled += value.num_fulfilled();
+	}
+
+	return true;
+}
+
+
+bool DefinedItemAttribMatch( const CAttribute_DynamicRecipeComponent& attribValue,  const IEconItemInterface* pItem )
+{
+	if ( !pItem )
+		return false;
+
+	// If our fulfilled count is what our item count is, then we're done.  We dont want any more matches.
+	if ( attribValue.num_fulfilled() == attribValue.num_required() )
+		return false;
+
+	// If the item_def flag is set, and the item's item_def doesnt match then not a match
+	if ( ( attribValue.component_flags() & DYNAMIC_RECIPE_FLAG_PARAM_ITEM_DEF_SET ) &&
+		( attribValue.def_index() != (uint32)pItem->GetItemDefIndex() ) )
+		return false;
+
+	// If the quality flag is set, and the item's quality doesn't match, then not a match
+	if ( ( attribValue.component_flags() & DYNAMIC_RECIPE_FLAG_PARAM_QUALITY_SET ) &&
+		( attribValue.item_quality() != (uint32)pItem->GetQuality() ) )
+		return false;
+
+	// check if we have ALL required attributes
+	if ( attribValue.component_flags() & DYNAMIC_RECIPE_FLAG_PARAM_ATTRIBUTE_SET_ALL )
+	{
+		CUtlVector<CEconItem::attribute_t> vecAttribs;
+		if( !DecodeAttributeStringIntoAttributes( attribValue, vecAttribs ) )
+		{
+			AssertMsg2( 0, "%s: Unable to decode dynamic recipe attributes on item %llu", __FUNCTION__, pItem->GetID() );
+			return false;
+		}
+
+		FOR_EACH_VEC( vecAttribs, i )
+		{
+			const CEconItemAttributeDefinition *pAttr = GetItemSchema()->GetAttributeDefinition( vecAttribs[i].m_unDefinitionIndex );
+			Assert( pAttr );
+			uint32 itemAttributeValue;
+			if ( !pAttr || !pItem->FindAttribute( pAttr, &itemAttributeValue ) || itemAttributeValue != vecAttribs[i].m_value.asUint32 )
+			{
+				return false;
+			}
+		}
+	}
+	// check if we have ANY required attributes
+	else if ( attribValue.component_flags() & DYNAMIC_RECIPE_FLAG_PARAM_ATTRIBUTE_SET_ANY )
+	{
+		CUtlVector<CEconItem::attribute_t> vecAttribs;
+		if( !DecodeAttributeStringIntoAttributes( attribValue, vecAttribs ) )
+		{
+			AssertMsg2( 0, "%s: Unable to decode dynamic recipe attributes on item %llu", __FUNCTION__, pItem->GetID() );
+			return false;
+		}
+
+		bool bHasAnyMatchingAttributes = false;
+		FOR_EACH_VEC( vecAttribs, i )
+		{
+			const CEconItemAttributeDefinition *pAttr = GetItemSchema()->GetAttributeDefinition( vecAttribs[i].m_unDefinitionIndex );
+			Assert( pAttr );
+			uint32 itemAttributeValue;
+			if ( pAttr && pItem->FindAttribute( pAttr, &itemAttributeValue ) && itemAttributeValue == vecAttribs[i].m_value.asUint32 )
+			{
+				bHasAnyMatchingAttributes = true;
+				break;
+			}
+		}
+
+		if ( !bHasAnyMatchingAttributes )
+		{
+			return false;
+		}
+	}
+
+	return true;
+}
+
+bool DecodeAttributeStringIntoAttributes( const CAttribute_DynamicRecipeComponent& attribValue, CUtlVector<CEconItem::attribute_t>& vecAttribs )
+{
+	CUtlStringList vecAttributeStrings;	// Automatically free'd
+	V_SplitString( attribValue.attributes_string().c_str(), g_pszAttrEncodeSeparator, vecAttributeStrings );
+	
+	if( vecAttributeStrings.Count() % 2 != 0 )
+	{
+		AssertMsg1( 0, "%s: Uneven count of encoded attribute strings!", __FUNCTION__ );
+		return false;
+	}
+
+	for( int j = 0; j< vecAttributeStrings.Count(); j+=2 )
+	{
+		// Get the attribute definition that's stored in the string, and its type
+		attrib_definition_index_t index = Q_atoi( vecAttributeStrings[j] );
+		const CEconItemAttributeDefinition *pAttrDef = GEconItemSchema().GetAttributeDefinition( index );
+		if ( !pAttrDef )
+		{
+#ifdef GC
+			EmitError( SPEW_GC, __FUNCTION__ ": Unable to find attribute definition '%s' (index %d)!\n", vecAttributeStrings[j], j );
+#endif
+			return false;
+		}
+
+		CEconItem::attribute_t& attrib = vecAttribs[vecAttribs.AddToTail()];
+		attrib.m_unDefinitionIndex = pAttrDef->GetDefinitionIndex();
+		
+		// Now have the attribute read in the value stored in the string
+		const ISchemaAttributeType* pAttrType = pAttrDef->GetAttributeType();
+		pAttrType->InitializeNewEconAttributeValue( &attrib.m_value );
+
+		// Don't fail us now!
+		const char* pszAttribValue = vecAttributeStrings[j+1];
+		if ( !pAttrType->BConvertStringToEconAttributeValue( pAttrDef, pszAttribValue, &attrib.m_value ) )
+		{
+#ifdef GC
+			EmitError( SPEW_GC, __FUNCTION__ ": Unable to parse attribute value '%s' for attribute '%s'!\n", pszAttribValue, pAttrDef->GetDefinitionName() );
+#endif
+			return false;
+		}
+	}
+
+	return true;
+}
+
+bool DecodeItemFromEncodedAttributeString( const CAttribute_DynamicRecipeComponent& attribValue, CEconItem* pItem )
+{
+	// If the item_def flag is set, set that item def
+	if ( attribValue.component_flags() & DYNAMIC_RECIPE_FLAG_PARAM_ITEM_DEF_SET )
+	{
+		pItem->SetDefinitionIndex( attribValue.def_index() );
+	}
+	else
+	{
+		// If the flag is not set, then we want the item name to be generic.  In english, we want to just call it "item".
+		CAttribute_String attrStr;
+		attrStr.set_value( "#TF_ItemName_Item" );
+
+		static CSchemaAttributeDefHandle pAttrDef_ItemNameTextOverride( "item name text override" );
+		pItem->SetDynamicAttributeValue( pAttrDef_ItemNameTextOverride, attrStr );
+	}
+
+	// If the quality flag is set, take the quality
+	if ( attribValue.component_flags() & DYNAMIC_RECIPE_FLAG_PARAM_QUALITY_SET )
+	{
+		pItem->SetQuality( attribValue.item_quality() );
+
+		// If there's no item def set and the quality specified is "unique", we want to be explicit
+		// and have the item description actually say "unique item" so there's no confusion as to what
+		// item quality we want as an input.
+		if ( !( attribValue.component_flags() & DYNAMIC_RECIPE_FLAG_PARAM_ITEM_DEF_SET )
+			&& pItem->GetQuality() == AE_UNIQUE )
+		{
+			CAttribute_String attrStr;
+			attrStr.set_value( "#unique" );
+
+			static CSchemaAttributeDefHandle pAttrDef_QualityTextOverride( "quality text override" );
+			pItem->SetDynamicAttributeValue( pAttrDef_QualityTextOverride, attrStr );
+		}
+	}
+	else
+	{
+		// If no quality was specified, we want to explicity say that we'll accept ANY quality.
+		pItem->SetQuality( AE_UNIQUE );
+		CAttribute_String attrStr;
+		attrStr.set_value( "#TF_QualityText_Any" );
+
+		static CSchemaAttributeDefHandle pAttrDef_QualityTextOverride( "quality text override" );
+		pItem->SetDynamicAttributeValue( pAttrDef_QualityTextOverride, attrStr );
+	}
+	pItem->SetFlags( 0 );
+
+	// Get all the attributes encoded into the attribute
+	CUtlVector<CEconItem::attribute_t> vecAttribs;
+	if( !DecodeAttributeStringIntoAttributes( attribValue, vecAttribs ) )
+	{
+		AssertMsg1( 0, " %s : Unable to decode dynamic recipe attributes", __FUNCTION__ );
+		return false;
+	}
+
+	// Apply the attributes to the item
+	FOR_EACH_VEC( vecAttribs, j )
+	{
+		// We don't expect to get here with any missing attributes.
+		const CEconItemAttributeDefinition *pAttrDef = GetItemSchema()->GetAttributeDefinition( vecAttribs[j].m_unDefinitionIndex );
+		Assert( pAttrDef );
+
+		const ISchemaAttributeType *pAttrType = pAttrDef->GetAttributeType();
+		pAttrType->LoadEconAttributeValue( pItem, pAttrDef, vecAttribs[j].m_value );
+
+		// Free up our attribute memory now that we're done with it.
+		pAttrType->UnloadEconAttributeValue( &vecAttribs[j].m_value );
+	}
+
+	return true;
+}