initial

movieobjects/dmeconnectionoperator.cpp

@@ -0,0 +1,188 @@
+//====== Copyright <20> 1996-2009, Valve Corporation, All rights reserved. =======
+// 
+// Purpose: Declaration of the CDmeConnectionOperator class, a CDmeOperator 
+// which copies one attribute value to another, providing similar functionality
+// to CDmeChannel, but does not store a log and is not effected by the 
+// recording mode.
+//
+//=============================================================================
+#include "movieobjects/dmeconnectionoperator.h"
+#include "datamodel/dmelementfactoryhelper.h"
+#include "tier1/fmtstr.h"
+
+
+// memdbgon must be the last include file in a .cpp file!!!
+#include "tier0/memdbgon.h"
+
+
+//-------------------------------------------------------------------------------------------------
+// Expose this class to the scene database 
+//-------------------------------------------------------------------------------------------------
+IMPLEMENT_ELEMENT_FACTORY( DmeConnectionOperator, CDmeConnectionOperator );
+
+//-------------------------------------------------------------------------------------------------
+// Purpose: Constructor, initializes attributes, create the embedded target
+//-------------------------------------------------------------------------------------------------
+void CDmeConnectionOperator::OnConstruction()
+{
+	m_Input.InitAndCreate( this, "input" );
+	m_Outputs.Init( this, "outputs" );
+}
+
+
+//-------------------------------------------------------------------------------------------------
+// Purpose: Perform destruction tasks, destroy the internal elements of the constraint.
+//-------------------------------------------------------------------------------------------------
+void CDmeConnectionOperator::OnDestruction()
+{
+	g_pDataModel->DestroyElement( m_Input.GetHandle() );
+
+	int nOutputs = m_Outputs.Count();
+	for ( int i = 0 ;i < nOutputs; ++i )
+	{
+		if ( m_Outputs[ i ] )
+		{
+			g_pDataModel->DestroyElement( m_Outputs[ i ]->GetHandle() );
+		}
+	}
+
+	m_Outputs.RemoveAll();
+}
+
+
+//-------------------------------------------------------------------------------------------------
+// Purpose: Run the operator, which copies the value from the source attribute to the destination 
+// attributes.
+//-------------------------------------------------------------------------------------------------
+void CDmeConnectionOperator::Operate()
+{
+	if ( !m_Input->IsValid() )
+		return;
+
+	int nOutputs = m_Outputs.Count();
+	if ( nOutputs == 0 )
+		return;
+			
+	DmAttributeType_t inputType = AT_UNKNOWN;
+	const void *pValue = m_Input->GetAttributeValue( inputType );
+
+	for ( int iOutput = 0; iOutput < nOutputs; ++iOutput )
+	{
+		m_Outputs[ iOutput ]->SetAttributeValue( pValue, inputType );
+	}
+}
+
+
+//-------------------------------------------------------------------------------------------------
+// Purpose: Determine if data has changed and the operator needs to be updated
+//-------------------------------------------------------------------------------------------------
+bool CDmeConnectionOperator::IsDirty()
+{
+	CDmAttribute* pAttr = m_Input->GetReferencedAttribute();
+
+	if ( pAttr )
+	{
+		return pAttr->IsFlagSet( FATTRIB_DIRTY );
+	}
+
+	return false;
+}
+
+
+//-------------------------------------------------------------------------------------------------
+// Purpose: Add the input attribute used by the operator to the provided list of attributes, This 
+// is generally used by the evaluation process to find the attributes an operator is dependent on.
+//-------------------------------------------------------------------------------------------------
+void CDmeConnectionOperator::GetInputAttributes( CUtlVector< CDmAttribute * > &attrs )
+{
+	CDmAttribute *pInputAttr = m_Input->GetReferencedAttribute();
+	if ( pInputAttr )
+	{
+		attrs.AddToTail( pInputAttr );
+	}
+}
+
+
+//-------------------------------------------------------------------------------------------------
+// Purpose: Add each of attributes the connection operator outputs to the provided list of 
+// attributes. This is generally used by the evaluation process to find out what attributes are 
+// written by the operator in order to determine what other operators are dependent on this 
+// operator.
+//-------------------------------------------------------------------------------------------------
+void CDmeConnectionOperator::GetOutputAttributes( CUtlVector< CDmAttribute * > &attrs )
+{
+	int nOutputs = m_Outputs.Count();
+	for ( int iOutput = 0; iOutput < nOutputs; ++iOutput )
+	{
+		CDmAttribute *pOutputAttr = m_Outputs[ iOutput ]->GetReferencedAttribute();
+		if ( pOutputAttr )
+		{
+			attrs.AddToTail( pOutputAttr );
+		}
+	}
+}
+
+
+//-------------------------------------------------------------------------------------------------
+// Purpose: Set the input attribute of the connection.
+//-------------------------------------------------------------------------------------------------
+void CDmeConnectionOperator::SetInput( CDmElement* pElement, const char* pchAttributeName, int index )
+{
+	m_Input->SetAttribute( pElement, pchAttributeName, index );
+}
+
+
+//-------------------------------------------------------------------------------------------------
+// Purpose: Add an attribute to be written to by the connection.
+//-------------------------------------------------------------------------------------------------
+void CDmeConnectionOperator::AddOutput( CDmElement* pElement, const char* pchAttributeName, int index )
+{
+	if ( ( pElement == NULL ) || ( pchAttributeName == NULL ) )
+		return;
+
+	CDmeAttributeReference *pAttrRef = CreateElement< CDmeAttributeReference >( CFmtStr( "%s_%s", pElement->GetName() , pchAttributeName ), GetFileId() );
+	if ( pAttrRef )
+	{
+		if ( pAttrRef->SetAttribute( pElement, pchAttributeName, index ) )
+		{
+			// Add the new reference to the list of outputs of the connection.
+			m_Outputs.AddToTail( pAttrRef );
+		}
+		else
+		{
+			// If the specified attribute was not valid destroy the reference.
+			g_pDataModel->DestroyElement( pAttrRef->GetHandle() );
+		}
+	}
+}
+
+
+//-------------------------------------------------------------------------------------------------
+// Purpose: Get the number of output attributes
+//-------------------------------------------------------------------------------------------------
+int CDmeConnectionOperator::NumOutputAttributes() const
+{
+	return m_Outputs.Count();
+}
+
+
+//-------------------------------------------------------------------------------------------------
+// Purpose: Get the specified output attribute
+//-------------------------------------------------------------------------------------------------
+CDmAttribute *CDmeConnectionOperator::GetOutputAttribute( int index ) const
+{
+	if ( index >= m_Outputs.Count() )
+		return NULL;
+
+	return m_Outputs[ index ]->GetReferencedAttribute();	
+}
+
+
+//-------------------------------------------------------------------------------------------------
+// Purpose: Get the input attribute
+//-------------------------------------------------------------------------------------------------
+CDmAttribute *CDmeConnectionOperator::GetInputAttribute()
+{
+	return m_Input.GetAttribute();
+}
+