source-engine/movieobjects/dmecombinationoperator.cpp

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose: 
//
//=============================================================================
#include "movieobjects/dmecombinationoperator.h"
#include "movieobjects_interfaces.h"
#include "datamodel/dmelementfactoryhelper.h"
#include "datamodel/dmattribute.h"
#include "movieobjects/dmechannel.h"
#include "movieobjects/dmemodel.h"
#include "movieobjects/dmeshape.h"
#include "movieobjects/dmedag.h"
#include "movieobjects/dmeclip.h"
#include "movieobjects/dmelog.h"
#include "movieobjects/dmevertexdata.h"
#include "movieobjects/dmemesh.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( DmeCombinationInputControl, CDmeCombinationInputControl );


//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void CDmeCombinationInputControl::OnConstruction()
{
	m_RawControlNames.Init( this, "rawControlNames" );
	m_bIsStereo.Init( this, "stereo" );
	m_bIsEyelid.InitAndSet( this, "eyelid", false );
	m_WrinkleScales.Init( this, "wrinkleScales" );
}

void CDmeCombinationInputControl::OnDestruction()
{
}


//-----------------------------------------------------------------------------
// Backward compat
//-----------------------------------------------------------------------------
void CDmeCombinationInputControl::OnElementUnserialized()
{
	BaseClass::OnElementUnserialized();

	int nWrinkleCount = m_WrinkleScales.Count();
	int nControlCount = m_RawControlNames.Count();
	if ( nWrinkleCount < nControlCount )
	{
		for ( int i = nWrinkleCount; i < nControlCount; ++i )
		{
			m_WrinkleScales.AddToTail( 0.0f );
		}
	}
	else if ( nWrinkleCount > nControlCount )
	{
		m_WrinkleScales.RemoveMultiple( nControlCount, nWrinkleCount - nControlCount );
	}
}


//-----------------------------------------------------------------------------
// Adds a control, returns the control index
//-----------------------------------------------------------------------------
bool CDmeCombinationInputControl::AddRawControl( const char *pRawControlName )
{
	Assert( !strchr( pRawControlName, '_' ) && !strchr( pRawControlName, ' ' ) );
	int nCount = m_RawControlNames.Count();
	for ( int i = 0; i < nCount; ++i )
	{
		if ( !Q_stricmp( pRawControlName, m_RawControlNames[i] ) )
			return false;
	}

	m_RawControlNames.AddToTail( pRawControlName );
	m_WrinkleScales.AddToTail( 0.0f );
	return true;
}


//-----------------------------------------------------------------------------
// Finds a raw control by name
//-----------------------------------------------------------------------------
int CDmeCombinationInputControl::FindRawControl( const char *pRawControlName )
{
	Assert( !strchr( pRawControlName, '_' ) && !strchr( pRawControlName, ' ' ) );
	int nCount = m_RawControlNames.Count();
	for ( int i = 0; i < nCount; ++i )
	{
		if ( !Q_stricmp( pRawControlName, m_RawControlNames[i] ) )
			return i;
	}
	return -1;
}


//-----------------------------------------------------------------------------
// Removes controls
//-----------------------------------------------------------------------------
bool CDmeCombinationInputControl::RemoveRawControl( const char *pRawControlName )
{
	int i = FindRawControl( pRawControlName );
	if ( i >= 0 )
	{
		m_RawControlNames.FastRemove( i );
		m_WrinkleScales.FastRemove( i );
		return true;
	}

	return false;
}

void CDmeCombinationInputControl::RemoveAllRawControls()
{
	m_RawControlNames.RemoveAll();
	m_WrinkleScales.RemoveAll( );
}


//-----------------------------------------------------------------------------
// Iterates remapped controls
//-----------------------------------------------------------------------------
int CDmeCombinationInputControl::RawControlCount() const
{
	return m_RawControlNames.Count();
}

const char *CDmeCombinationInputControl::RawControlName( int nIndex ) const
{
	return m_RawControlNames[ nIndex ];
}


//-----------------------------------------------------------------------------
// Is this control a stereo control?
//-----------------------------------------------------------------------------
bool CDmeCombinationInputControl::IsStereo() const
{
	return m_bIsStereo;
}

void CDmeCombinationInputControl::SetStereo( bool bStereo )
{
	m_bIsStereo = bStereo;
}


//-----------------------------------------------------------------------------
// Is this control an eyelid control?
//-----------------------------------------------------------------------------
bool CDmeCombinationInputControl::IsEyelid() const
{
	return m_bIsEyelid;
}

void CDmeCombinationInputControl::SetEyelid( bool bEyelid )
{
	m_bIsEyelid = bEyelid;
}


//-----------------------------------------------------------------------------
// Reordering controls
//-----------------------------------------------------------------------------
void CDmeCombinationInputControl::MoveRawControlUp( const char *pRawControlName )
{
	int nIndex = FindRawControl( pRawControlName );
	if ( nIndex > 0 )
	{
		m_RawControlNames.Swap( nIndex, nIndex - 1 );
		m_WrinkleScales.Swap( nIndex, nIndex - 1 );
	}
}

void CDmeCombinationInputControl::MoveRawControlDown( const char *pRawControlName )
{
	int nIndex = FindRawControl( pRawControlName );
	int nLastIndex = m_RawControlNames.Count() - 1;
	if ( nIndex >= 0 && nIndex < nLastIndex )
	{
		m_RawControlNames.Swap( nIndex, nIndex + 1 );
		m_WrinkleScales.Swap( nIndex, nIndex + 1 );
	}
}


//-----------------------------------------------------------------------------
// Returns the wrinkle scale for a particular control
//-----------------------------------------------------------------------------
float CDmeCombinationInputControl::WrinkleScale( const char *pRawControlName )
{
	int nIndex = FindRawControl( pRawControlName );
	return WrinkleScale( nIndex );
}

float CDmeCombinationInputControl::WrinkleScale( int nIndex )
{
	if ( nIndex < 0 || ( nIndex >= m_WrinkleScales.Count() ) )
		return 0.0f;
	return m_WrinkleScales[ nIndex ];
}

void CDmeCombinationInputControl::SetWrinkleScale( const char *pRawControlName, float flWrinkleScale )
{
	int nIndex = FindRawControl( pRawControlName );
	if ( nIndex < 0 || ( nIndex >= m_WrinkleScales.Count() ) )
		return;
	m_WrinkleScales.Set( nIndex, flWrinkleScale );
}


//-----------------------------------------------------------------------------
// The default value of an input control is the value the UI has by default
//-----------------------------------------------------------------------------
float CDmeCombinationInputControl::GetDefaultValue() const
{
	return RawControlCount() == 2 ? 0.5f : 0.0f;
}


//-----------------------------------------------------------------------------
// The base value of an input control will set the data to the base state
// i.e. The state upon which the deltas are relative.  Normally this is the
//      same as GetDefaultValue() except for EyeLid controls
//-----------------------------------------------------------------------------
float CDmeCombinationInputControl::GetBaseValue() const
{
	if ( IsEyelid() )
		return 0.0f;

	return GetDefaultValue();
}


//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const char *CDmeCombinationInputControl::GetEyesUpDownFlexName() const
{
	const CDmAttribute *pEyesUpDownFlexAttr = GetAttribute( "eyesUpDownFlex", AT_STRING );

	if ( pEyesUpDownFlexAttr )
		return pEyesUpDownFlexAttr->GetValueString();

	return NULL;
}


//-----------------------------------------------------------------------------
// Expose this class to the scene database 
//-----------------------------------------------------------------------------
IMPLEMENT_ELEMENT_FACTORY( DmeCombinationDominationRule, CDmeCombinationDominationRule );


//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void CDmeCombinationDominationRule::OnConstruction()
{
	m_Dominators.Init( this, "dominators", FATTRIB_HAS_CALLBACK | FATTRIB_HAS_ARRAY_CALLBACK );
	m_Suppressed.Init( this, "suppressed", FATTRIB_HAS_CALLBACK | FATTRIB_HAS_ARRAY_CALLBACK );
}

void CDmeCombinationDominationRule::OnDestruction()
{
}


//-----------------------------------------------------------------------------
// Notify parent that one of our attributes has changed
//-----------------------------------------------------------------------------
void CDmeCombinationDominationRule::OnAttributeChanged( CDmAttribute *pAttribute )
{
	BaseClass::OnAttributeChanged( pAttribute );

	if ( pAttribute == m_Dominators.GetAttribute() || pAttribute == m_Suppressed.GetAttribute() )
	{
		InvokeOnAttributeChangedOnReferrers( GetHandle(), pAttribute );
	}
}


//-----------------------------------------------------------------------------
// Do we have this string already?
//-----------------------------------------------------------------------------
bool CDmeCombinationDominationRule::HasString( const char *pString, const CDmaStringArray& attr )
{
	int nCount = attr.Count();
	for ( int i = 0; i < nCount; ++i )
	{
		if ( !Q_stricmp( pString, attr[i] ) )
			return true;
	}
	return false;
}


//-----------------------------------------------------------------------------
// Adds a dominating control
//-----------------------------------------------------------------------------
void CDmeCombinationDominationRule::AddDominator( const char *pDominatorControl )
{
	if ( HasString( pDominatorControl, m_Dominators ) )
	{
		Warning( "Domination rule already contains dominating control %s\n", pDominatorControl );
		return;
	}
	if ( HasString( pDominatorControl, m_Suppressed ) )
	{
		Warning( "Attemped to add a control as both dominator + suppressed %s\n", pDominatorControl );
		return;
	}
	m_Dominators.AddToTail( pDominatorControl );
}


//-----------------------------------------------------------------------------
// Add a suppressed control
//-----------------------------------------------------------------------------
void CDmeCombinationDominationRule::AddSuppressed( const char *pSuppressedControl )
{
	if ( HasString( pSuppressedControl, m_Suppressed ) )
	{
		Warning( "Domination rule already contains suppressed control %s\n", pSuppressedControl );
		return;
	}
	if ( HasString( pSuppressedControl, m_Dominators ) )
	{
		Warning( "Attemped to add a control as both dominator + suppressed %s\n", pSuppressedControl );
		return;
	}
	m_Suppressed.AddToTail( pSuppressedControl );
}


//-----------------------------------------------------------------------------
// Remove all dominatior + suppressed controls
//-----------------------------------------------------------------------------
void CDmeCombinationDominationRule::RemoveAllDominators()
{
	m_Dominators.RemoveAll();
}

void CDmeCombinationDominationRule::RemoveAllSuppressed()
{
	m_Suppressed.RemoveAll();
}


//-----------------------------------------------------------------------------
// Iteration
//-----------------------------------------------------------------------------
int CDmeCombinationDominationRule::DominatorCount() const
{
	return m_Dominators.Count();
}

const char *CDmeCombinationDominationRule::GetDominator( int i ) const
{
	return m_Dominators[i];
}

int CDmeCombinationDominationRule::SuppressedCount() const
{
	return m_Suppressed.Count();
}

const char *CDmeCombinationDominationRule::GetSuppressed( int i ) const
{
	return m_Suppressed[i];
}


//-----------------------------------------------------------------------------
// Search
//-----------------------------------------------------------------------------
bool CDmeCombinationDominationRule::HasDominatorControl( const char *pDominatorControl ) const
{
	int nCount = DominatorCount();
	for ( int i = 0; i < nCount; ++i )
	{
		if ( !Q_stricmp( GetDominator(i), pDominatorControl ) )
			return true;
	}
	return false;
}

bool CDmeCombinationDominationRule::HasSuppressedControl( const char *pSuppressedControl ) const
{
	int nCount = SuppressedCount();
	for ( int i = 0; i < nCount; ++i )
	{
		if ( !Q_stricmp( GetSuppressed(i), pSuppressedControl ) )
			return true;
	}
	return false;
}


//-----------------------------------------------------------------------------
// Expose this class to the scene database 
//-----------------------------------------------------------------------------
IMPLEMENT_ELEMENT_FACTORY( DmeCombinationOperator, CDmeCombinationOperator );


//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::OnConstruction()
{
	m_InputControls.Init( this, "controls" );
	m_ControlValues[COMBO_CONTROL_NORMAL].Init( this, "controlValues" );
	m_ControlValues[COMBO_CONTROL_LAGGED].Init( this, "controlValuesLagged" );
	m_bSpecifyingLaggedData.Init( this, "usesLaggedValues" );

	m_Dominators.Init( this, "dominators", FATTRIB_HAS_CALLBACK | FATTRIB_HAS_ARRAY_CALLBACK );

	m_Targets.Init( this, "targets" );
	m_flLastLaggedComputationTime = FLT_MIN;
}

void CDmeCombinationOperator::OnDestruction()
{
	m_RawControlInfo.RemoveAll();
	m_CombinationInfo.RemoveAll();
	m_DominatorInfo.RemoveAll();
}


//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::OnAttributeChanged( CDmAttribute *pAttribute )
{
	m_Dominators.GetAttribute()->AddFlag( FATTRIB_DIRTY );
}


//-----------------------------------------------------------------------------
// Finds the index of the input control with the specified name (and creates one if necessary) 
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::OnElementUnserialized()
{
	BaseClass::OnElementUnserialized();

	// Mark all of the input as not being in their default state since we read it from a file
	int nCount = m_InputControls.Count();
	m_IsDefaultValue.SetCount( nCount );
	for ( int i = 0; i < nCount; ++i ) 
	{
		m_IsDefaultValue[i] = false;
	}
}


//-----------------------------------------------------------------------------
// Finds the index of the input control with the specified name (and creates one if necessary) 
//-----------------------------------------------------------------------------
ControlIndex_t CDmeCombinationOperator::FindOrCreateControl( const char *pControlName, bool bStereo, bool bAutoAddRawControl )
{
	Assert( !strchr( pControlName, '_' ) && !strchr( pControlName, ' ' ) );
	int nCount = m_InputControls.Count();
	for ( int i = 0; i < nCount; ++i )
	{
		if ( !m_InputControls[i] )
			continue;

		if ( !Q_stricmp( pControlName, m_InputControls[i]->GetName() ) )
			return i;
	}

	// NOTE: the y coordinate of the control value is -1 if it's not a stereo control
	CDmeCombinationInputControl *pInputControl = CreateElement< CDmeCombinationInputControl >( pControlName, GetFileId() );
	pInputControl->SetStereo( bStereo );
	int nIndex = m_InputControls.AddToTail( pInputControl );
	m_ControlValues[COMBO_CONTROL_NORMAL].AddToTail( Vector( 0.0f, 0.5f, 0.5f ) );
	m_ControlValues[COMBO_CONTROL_LAGGED].AddToTail( Vector( 0.0f, 0.5f, 0.5f ) );
	m_IsDefaultValue.AddToTail( true );
	Assert( m_InputControls.Count() == m_ControlValues[COMBO_CONTROL_NORMAL].Count() );
	Assert( m_InputControls.Count() == m_ControlValues[COMBO_CONTROL_LAGGED].Count() );
	Assert( m_InputControls.Count() == m_IsDefaultValue.Count() );

	if ( bAutoAddRawControl )
	{
		AddRawControl( nIndex, pControlName );
	}

	return nIndex;
}


//-----------------------------------------------------------------------------
// Finds the index of the input control with the specified name
//-----------------------------------------------------------------------------
ControlIndex_t CDmeCombinationOperator::FindControlIndex( const char *pControlName )
{
	int nCount = m_InputControls.Count();
	for ( int i = 0; i < nCount; ++i )
	{
		if ( !m_InputControls[i] )
			continue;

		if ( !Q_stricmp( pControlName, m_InputControls[i]->GetName() ) )
			return i;
	}
	return -1;
}


//-----------------------------------------------------------------------------
// Removes a control
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::RemoveControl( const char *pControlName )
{
	ControlIndex_t nIndex = FindControlIndex( pControlName );
	if ( nIndex >= 0 )
	{
		DestroyElement( m_InputControls[nIndex] );
		m_InputControls.FastRemove( nIndex );
		m_ControlValues[COMBO_CONTROL_NORMAL].FastRemove( nIndex );
		m_ControlValues[COMBO_CONTROL_LAGGED].FastRemove( nIndex );
		m_IsDefaultValue.FastRemove( nIndex );

		Assert( m_InputControls.Count() == m_ControlValues[COMBO_CONTROL_NORMAL].Count() );
		Assert( m_InputControls.Count() == m_ControlValues[COMBO_CONTROL_LAGGED].Count() );
		Assert( m_InputControls.Count() == m_IsDefaultValue.Count() );

		RebuildRawControlList();
	}
}

void CDmeCombinationOperator::RemoveAllControls()
{
	int nCount = m_InputControls.Count();
	for ( int i = 0; i < nCount; ++i )
	{
		CDmeCombinationInputControl *pInputControl = m_InputControls[i];
		m_InputControls.Set( i, NULL );
		DestroyElement( pInputControl );
	}
	m_InputControls.RemoveAll();
	m_ControlValues[COMBO_CONTROL_NORMAL].RemoveAll( );
	m_ControlValues[COMBO_CONTROL_LAGGED].RemoveAll( );
	m_IsDefaultValue.RemoveAll( );

	RebuildRawControlList();
}


//-----------------------------------------------------------------------------
// Changes a control's name
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::SetControlName( ControlIndex_t nControl, const char *pControlName )
{
	ControlIndex_t nFoundIndex = FindControlIndex( pControlName );
	if ( nFoundIndex >= 0 && nFoundIndex != nControl )
	{
		Warning( "A control with name \"%s\" already exists!\n", pControlName );
		return;
	}
	m_InputControls[nControl]->SetName( pControlName );
}


//-----------------------------------------------------------------------------
// Updates the default value associated with a control
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::UpdateDefaultValue( ControlIndex_t nControlIndex )
{
	if ( m_IsDefaultValue[ nControlIndex ] )
	{
		float flDefaultValue = GetRawControlCount( nControlIndex ) == 2 ? 0.5f : 0.0f;
		const Vector& vec = m_ControlValues[COMBO_CONTROL_NORMAL][nControlIndex];
		m_ControlValues[COMBO_CONTROL_NORMAL].Set( nControlIndex, Vector( flDefaultValue, vec.y, vec.z ) );
		const Vector& vec2 = m_ControlValues[COMBO_CONTROL_LAGGED][nControlIndex];
		m_ControlValues[COMBO_CONTROL_LAGGED].Set( nControlIndex, Vector( flDefaultValue, vec2.y, vec2.z ) );
	}
}


//-----------------------------------------------------------------------------
// Adds an output control to a input control
// If force is true, removes the raw control from any existing controls
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::AddRawControl( ControlIndex_t nControl, const char *pRawControlName )
{
	if ( FindRawControlIndex( pRawControlName, true ) >= 0 )
	{
		Warning( "Attempted to add the same remapped control \"%s\" twice!\n", pRawControlName );
		return;
	}

	CDmeCombinationInputControl *pInputControl = m_InputControls[nControl];
	Assert( pInputControl );
	if ( pInputControl->AddRawControl( pRawControlName ) )
	{
		UpdateDefaultValue( nControl );
		RebuildRawControlList();
	}
}


//-----------------------------------------------------------------------------
// Removes an output control from an input control
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::RemoveRawControl( ControlIndex_t nControl, const char *pRawControlName )
{
	CDmeCombinationInputControl *pInputControl = m_InputControls[nControl];
	Assert( pInputControl );
	if ( pInputControl->RemoveRawControl( pRawControlName ) )
	{
		UpdateDefaultValue( nControl );
		RebuildRawControlList();
	}
}

void CDmeCombinationOperator::RemoveAllRawControls( ControlIndex_t nControl )
{
	CDmeCombinationInputControl *pInputControl = m_InputControls[nControl];
	Assert( pInputControl );
	pInputControl->RemoveAllRawControls( );

	int nCount = m_InputControls.Count();
	for ( int i = 0; i < nCount; ++i )
	{
		UpdateDefaultValue( i );
	}

	RebuildRawControlList();
}


//-----------------------------------------------------------------------------
// Iterates remapped controls
//-----------------------------------------------------------------------------
int CDmeCombinationOperator::GetRawControlCount( ControlIndex_t nControl ) const
{
	CDmeCombinationInputControl *pInputControl = m_InputControls[nControl];
	Assert( pInputControl );
	return pInputControl->RawControlCount();
}

const char *CDmeCombinationOperator::GetRawControlName( ControlIndex_t nControl, int nIndex ) const
{
	CDmeCombinationInputControl *pInputControl = m_InputControls[nControl];
	Assert( pInputControl );
	return pInputControl->RawControlName( nIndex );
}

float CDmeCombinationOperator::GetRawControlWrinkleScale( ControlIndex_t nControl, int nIndex ) const
{
	CDmeCombinationInputControl *pInputControl = m_InputControls[nControl];
	Assert( pInputControl );
	return pInputControl->WrinkleScale( nIndex );
}

float CDmeCombinationOperator::GetRawControlWrinkleScale( ControlIndex_t nControl, const char *pRawControlName ) const
{
	CDmeCombinationInputControl *pInputControl = m_InputControls[nControl];
	Assert( pInputControl );
	return pInputControl->WrinkleScale( pRawControlName );
}


//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
float CDmeCombinationOperator::GetControlDefaultValue( ControlIndex_t nControl ) const
{
	CDmeCombinationInputControl *pInputControl = m_InputControls[nControl];
	Assert( pInputControl );
	return pInputControl->GetDefaultValue();
}


//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
float CDmeCombinationOperator::GetControlBaseValue( ControlIndex_t nControl ) const
{
	CDmeCombinationInputControl *pInputControl = m_InputControls[nControl];
	Assert( pInputControl );
	return pInputControl->GetBaseValue();
}


//-----------------------------------------------------------------------------
// Iterates a global list of output controls
//-----------------------------------------------------------------------------
int CDmeCombinationOperator::GetRawControlCount( ) const
{
	return m_RawControlInfo.Count();
}

const char *CDmeCombinationOperator::GetRawControlName( int nIndex ) const
{
	return m_RawControlInfo[nIndex].m_Name;
}

float CDmeCombinationOperator::GetRawControlWrinkleScale( int nIndex ) const
{
	return m_RawControlInfo[nIndex].m_flWrinkleScale;
}


//-----------------------------------------------------------------------------
// Sets the wrinkle scale for a particular raw control
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::SetWrinkleScale( ControlIndex_t nControl, const char *pRawControlName, float flWrinkleScale )
{
	CDmeCombinationInputControl *pInputControl = m_InputControls[nControl];
	Assert( pInputControl );
	pInputControl->SetWrinkleScale( pRawControlName, flWrinkleScale );
	RebuildRawControlList();
}


//-----------------------------------------------------------------------------
// Reordering controls
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::MoveControlUp( const char *pControlName )
{

	int nIndex = FindControlIndex( pControlName );
	if ( nIndex > 0 )
	{
		m_InputControls.Swap( nIndex, nIndex - 1 );
		for ( int i = 0; i < COMBO_CONTROL_TYPE_COUNT; ++i )
		{
			m_ControlValues[ i ].Swap( nIndex, nIndex - 1 );
		}
		RebuildRawControlList();
	}
}


//-----------------------------------------------------------------------------
// Reordering controls
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::MoveControlDown( const char *pControlName )
{
	int nIndex = FindControlIndex( pControlName );
	int nLastIndex = m_InputControls.Count() - 1;
	if ( nIndex >= 0 && nIndex < nLastIndex )
	{
		m_InputControls.Swap( nIndex, nIndex + 1 );
		for ( int i = 0; i < COMBO_CONTROL_TYPE_COUNT; ++i )
		{
			m_ControlValues[ i ].Swap( nIndex, nIndex + 1 );
		}
		RebuildRawControlList();
	}
}


//-----------------------------------------------------------------------------
// Reordering controls
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::MoveControlBefore( const char *pDragControlName, const char *pDropControlName )
{
	int pDragIndex( FindControlIndex( pDragControlName ) );
	int pDropIndex( FindControlIndex( pDropControlName ) );

	// Have to copy because InsertAfter may reallocate memory before doing the copy and therefore might be referencing garabage
	CDmeCombinationInputControl *inputControlCopy( m_InputControls[ pDragIndex ] );
	m_InputControls.InsertBefore( pDropIndex, inputControlCopy );
	if ( pDragIndex <= pDropIndex )
	{
		m_InputControls.Remove( pDragIndex );
		for ( int i = 0; i < COMBO_CONTROL_TYPE_COUNT; ++i )
		{
			const Vector controlValueCopy( m_ControlValues[ i ][ pDragIndex ] );
			m_ControlValues[ i ].InsertBefore( pDropIndex, controlValueCopy );
			m_ControlValues[ i ].Remove( pDragIndex );
		}
	}
	else
	{
		m_InputControls.Remove( pDragIndex + 1 );
		for ( int i = 0; i < COMBO_CONTROL_TYPE_COUNT; ++i )
		{
			const Vector controlValueCopy( m_ControlValues[ i ][ pDragIndex ] );
			m_ControlValues[ i ].InsertBefore( pDropIndex, controlValueCopy );
			m_ControlValues[ i ].Remove( pDragIndex + 1 );
		}
	}
}


//-----------------------------------------------------------------------------
// Reordering controls
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::MoveControlAfter( const char *pDragControlName, const char *pDropControlName )
{
	int nDragIndex = FindControlIndex( pDragControlName );
	int nDropIndex = FindControlIndex( pDropControlName );

	// Have to copy because InsertAfter may reallocate memory before doing the copy and therefore might be referencing garabage
	CDmeCombinationInputControl *inputControlCopy( m_InputControls[ nDragIndex ] );
	m_InputControls.InsertBefore( nDropIndex + 1, inputControlCopy );
	if ( nDragIndex < nDropIndex )
	{
		m_InputControls.Remove( nDragIndex );
		for ( int i = 0; i < COMBO_CONTROL_TYPE_COUNT; ++i )
		{
			const Vector controlValueCopy( m_ControlValues[ i ][ nDragIndex ] );
			m_ControlValues[ i ].InsertBefore( nDropIndex + 1, controlValueCopy );
			m_ControlValues[ i ].Remove( nDragIndex );
		}
	}
	else
	{
		m_InputControls.Remove( nDragIndex + 1 );
		for ( int i = 0; i < COMBO_CONTROL_TYPE_COUNT; ++i )
		{
			const Vector controlValueCopy( m_ControlValues[ i ][ nDragIndex ] );
			m_ControlValues[ i ].InsertBefore( nDropIndex + 1, controlValueCopy );
			m_ControlValues[ i ].Remove( nDragIndex + 1 );
		}
	}
}


//-----------------------------------------------------------------------------
// Reordering controls
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::MoveRawControlUp( ControlIndex_t nControlIndex, const char *pRawControlName )
{
	m_InputControls[nControlIndex]->MoveRawControlUp( pRawControlName );
}


//-----------------------------------------------------------------------------
// Reordering controls
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::MoveRawControlDown( ControlIndex_t nControlIndex, const char *pRawControlName )
{
	m_InputControls[nControlIndex]->MoveRawControlDown( pRawControlName );
}


//-----------------------------------------------------------------------------
// Returns true if a control is a stereo control
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::SetStereoControl( ControlIndex_t nControlIndex, bool bIsStereo )
{
	m_InputControls[nControlIndex]->SetStereo( bIsStereo );
}

bool CDmeCombinationOperator::IsStereoControl( ControlIndex_t nControlIndex ) const
{
	return m_InputControls[nControlIndex]->IsStereo();
}

bool CDmeCombinationOperator::IsStereoRawControl( int nIndex ) const
{
	return IsStereoControl( m_RawControlInfo[nIndex].m_InputControl );
}

void CDmeCombinationOperator::SetEyelidControl( ControlIndex_t nControlIndex, bool bIsEyelid )
{
	m_InputControls[nControlIndex]->SetEyelid( bIsEyelid );
}

bool CDmeCombinationOperator::IsEyelidControl( ControlIndex_t nControlIndex ) const
{
	return m_InputControls[nControlIndex]->IsEyelid();
}

bool CDmeCombinationOperator::IsEyelidRawControl( int nIndex ) const
{
	return IsEyelidControl( m_RawControlInfo[nIndex].m_InputControl );
}


//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
const char *CDmeCombinationOperator::GetEyesUpDownFlexName( ControlIndex_t nControlIndex ) const
{
	return m_InputControls[ nControlIndex ]->GetEyesUpDownFlexName();
}


//-----------------------------------------------------------------------------
// Sets the value of a control
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::SetControlValue( ControlIndex_t nControlIndex, float flValue, CombinationControlType_t type )
{
	m_IsDefaultValue[ nControlIndex ] = false;
	float flMultiLevel = m_ControlValues[type][nControlIndex].z;
	m_ControlValues[type].Set( nControlIndex, Vector( flValue, flValue, flMultiLevel ) );
}

void CDmeCombinationOperator::SetControlValue( ControlIndex_t nControlIndex, float flLevel, float flBalance, CombinationControlType_t type )
{
	Assert( IsStereoControl( nControlIndex ) );
	m_IsDefaultValue[ nControlIndex ] = false;
	float flMultiLevel = m_ControlValues[type][nControlIndex].z;
	m_ControlValues[type].Set( nControlIndex, Vector( flLevel, flBalance, flMultiLevel ) );
}

void CDmeCombinationOperator::SetControlValue( ControlIndex_t nControlIndex, const Vector2D& vec, CombinationControlType_t type )
{
	Assert( IsStereoControl( nControlIndex ) );
	m_IsDefaultValue[ nControlIndex ] = false;
	float flMultiLevel = m_ControlValues[type][nControlIndex].z;
	m_ControlValues[type].Set( nControlIndex, Vector( vec.x, vec.y, flMultiLevel ) );
}


//-----------------------------------------------------------------------------
// Sets the value of a control
//-----------------------------------------------------------------------------
float CDmeCombinationOperator::GetControlValue( ControlIndex_t nControlIndex, CombinationControlType_t type )	const
{
	Assert( !IsStereoControl( nControlIndex ) );
	return m_ControlValues[type].Get( nControlIndex ).x;
}


//-----------------------------------------------------------------------------
// Sets the value of a control
//-----------------------------------------------------------------------------
const Vector2D& CDmeCombinationOperator::GetStereoControlValue( ControlIndex_t nControlIndex, CombinationControlType_t type ) const
{
	return m_ControlValues[type].Get( nControlIndex ).AsVector2D();
}


//-----------------------------------------------------------------------------
// Sets the level of a control (only used by controls w/ 3 or more remappings)
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::SetMultiControlLevel( ControlIndex_t nControlIndex, float flMultiLevel, CombinationControlType_t type )
{
	m_IsDefaultValue[ nControlIndex ] = false;
	const Vector2D &value = m_ControlValues[type][nControlIndex].AsVector2D();
	m_ControlValues[type].Set( nControlIndex, Vector( value.x, value.y, flMultiLevel ) );
}

float CDmeCombinationOperator::GetMultiControlLevel( ControlIndex_t nControlIndex, CombinationControlType_t type ) const
{
	Assert( IsMultiControl( nControlIndex ) );
	return m_ControlValues[type][nControlIndex].z;
}


//-----------------------------------------------------------------------------
// Returns true if a control is a multi control (a control w/ 3 or more remappings)
//-----------------------------------------------------------------------------
bool CDmeCombinationOperator::IsMultiControl( ControlIndex_t nControlIndex ) const
{
	return m_InputControls[nControlIndex]->RawControlCount() >= 3 || IsEyelidControl( nControlIndex );
}


//-----------------------------------------------------------------------------
// Iterates controls
//-----------------------------------------------------------------------------
int CDmeCombinationOperator::GetControlCount() const
{
	return m_InputControls.Count();
}

const char *CDmeCombinationOperator::GetControlName( ControlIndex_t i ) const
{
	return m_InputControls[i]->GetName();
}


//-----------------------------------------------------------------------------
// Do we have a raw control?
//-----------------------------------------------------------------------------
bool CDmeCombinationOperator::HasRawControl( const char *pRawControlName ) const
{
	return FindRawControlIndex( pRawControlName ) >= 0;
}


//-----------------------------------------------------------------------------
// Finds the index of the remapped control with the specified name
//-----------------------------------------------------------------------------
CDmeCombinationOperator::RawControlIndex_t CDmeCombinationOperator::FindRawControlIndex( const char *pControlName, bool bIgnoreDefaultControls ) const
{
	int nRawControlCount = m_RawControlInfo.Count();
	for ( int i = 0; i < nRawControlCount; ++i )
	{
		if ( bIgnoreDefaultControls && m_RawControlInfo[i].m_bIsDefaultControl )
			continue;

		if ( !Q_stricmp( pControlName, m_RawControlInfo[i].m_Name ) )
			return i;
	}
	return -1;
}


//-----------------------------------------------------------------------------
// Is a particular remapped control stereo?
//-----------------------------------------------------------------------------
bool CDmeCombinationOperator::IsRawControlStereo( const char *pRawControlName )
{
	RawControlIndex_t nIndex = FindRawControlIndex( pRawControlName );
	if ( nIndex < 0 )
		return false;

	CDmeCombinationInputControl *pInputControl = m_InputControls[ m_RawControlInfo[nIndex].m_InputControl ];
	return pInputControl->IsStereo();
}


//-----------------------------------------------------------------------------
// Would a particular delta state attached to this combination operator end up stereo?
//-----------------------------------------------------------------------------
bool CDmeCombinationOperator::IsDeltaStateStereo( const char *pDeltaStateName )
{
	int *pTemp = (int*)_alloca( m_RawControlInfo.Count() * sizeof(int) );
	int nCount = ParseDeltaName( pDeltaStateName, pTemp );
	for ( int i = 0; i < nCount; ++i )
	{
		CDmeCombinationInputControl *pInputControl = m_InputControls[ m_RawControlInfo[ pTemp[i] ].m_InputControl ];
		if ( pInputControl->IsStereo() )
			return true;
	}
	return false;
}


//-----------------------------------------------------------------------------
// Does one of the targets we refer to contain a particular delta state?
//-----------------------------------------------------------------------------
bool CDmeCombinationOperator::DoesTargetContainDeltaState( const char *pSearchName )
{
	int nTargetCount = m_Targets.Count();
	for ( int i = 0; i < nTargetCount; ++i )
	{
		const CDmrElementArray<> deltaArray( m_Targets[i], "deltaStates" );
		if ( !deltaArray.IsValid() )
			continue;

		int nDeltaCount = deltaArray.Count();
		for ( int j = 0; j < nDeltaCount; ++j )
		{
			if ( !deltaArray[j] )
				continue;

			char pBuf[512];
			Q_strncpy( pBuf, deltaArray[j]->GetName(), sizeof(pBuf) );

			char *pEnd;
			for ( char *pName = pBuf; *pName; pName = pEnd )
			{
				pEnd = strchr( pName, '_' );
				if ( !pEnd )
				{
					pEnd = pName + Q_strlen( pName );
				}
				else
				{
					// Null-terminate
					*pEnd = 0;
					++pEnd;
				}

				if ( !Q_stricmp( pSearchName, pName ) )
					return true;
			}
		}
	}
	return false;
}


//-----------------------------------------------------------------------------
// Computes list of all remapped controls
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::RebuildRawControlList()
{
	m_RawControlInfo.RemoveAll();

	int nControlCount = m_InputControls.Count();
	for ( int i = 0; i < nControlCount; ++i )
	{
		CDmeCombinationInputControl *pInputControl = m_InputControls[i];
		Assert( pInputControl );
		int nRemapCount = pInputControl->RawControlCount();

		const bool bIsEyelid = pInputControl->IsEyelid();

		float flStep = ( nRemapCount > 2 ) ? 1.0f / ( nRemapCount - 1 ) : 0.0f;
		for ( int j = 0; j < nRemapCount; ++j )
		{
			int k = m_RawControlInfo.AddToTail( );
			RawControlInfo_t &info = m_RawControlInfo[k];
			
			info.m_Name = pInputControl->RawControlName( j );
			info.m_InputControl = i;
			info.m_bIsDefaultControl = false;
			info.m_flWrinkleScale = pInputControl->WrinkleScale( j );
			info.m_bLowerEyelid = false;

			if ( bIsEyelid )
			{
				info.m_FilterRamp.Init( 0.0f, 1.0f, 10.0f, 11.0f );
				// TODO: Right now it's implicit that the lower eyelid is first...
				if ( j == 0 )
				{
					// Close Lower Lid
					info.m_bLowerEyelid = true;
				}
				continue;
			}

			switch( nRemapCount )
			{
			case 1:
				info.m_FilterRamp.Init( 0.0f, 1.0f, 10.0f, 11.0f );
				break;

			case 2:
				if ( j == 0 )
				{
					info.m_FilterRamp.Init( -11.0f, -10.0f, 0.0f, 0.5f );
				}
				else
				{
					info.m_FilterRamp.Init( 0.5f, 1.0f, 10.0f, 11.0f );
				}
				break;

			default:
				{
					if ( j == 0 )
					{
						info.m_FilterRamp.Init( -11.0f, -10.0f,	0, flStep );
					}
					else if ( j == nRemapCount-1 )
					{
						info.m_FilterRamp.Init( 1.0f - flStep, 1.0f, 10.0f, 11.0f );
					}
					else
					{
						float flPeak = j * flStep;
						info.m_FilterRamp.Init( flPeak - flStep, flPeak, flPeak, flPeak + flStep );
					}
				}
				break;
			}
		}
	}

	RebuildDominatorInfo();
}


//-----------------------------------------------------------------------------
// Computes lists of dominators and suppressors
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::RebuildDominatorInfo()
{
	m_DominatorInfo.RemoveAll();
	int nCount = m_Dominators.Count();
	int *pDominators = (int*)_alloca( m_RawControlInfo.Count() * sizeof(int) );
	int *pSuppressed = (int*)_alloca( m_RawControlInfo.Count() * sizeof(int) );
	for ( int i = 0; i < nCount; ++i )
	{
		CDmeCombinationDominationRule *pRule = m_Dominators[i];
		bool bRuleOk = true;

		int nDominatorCount = pRule->DominatorCount();
		int nSuppressedCount = pRule->SuppressedCount();
		if ( ( nDominatorCount == 0 ) || ( nSuppressedCount == 0 ) )
			continue;

		for ( int j = 0; j < nDominatorCount; ++j )
		{
			int nControlIndex = FindRawControlIndex( pRule->GetDominator(j) );
			if ( nControlIndex < 0 )
			{
				bRuleOk = false;
				break;
			}
			pDominators[j] = nControlIndex;
		}

		for ( int j = 0; j < nSuppressedCount; ++j )
		{
			int nControlIndex = FindRawControlIndex( pRule->GetSuppressed(j) );
			if ( nControlIndex < 0 )
			{
				bRuleOk = false;
				break;
			}
			pSuppressed[j] = nControlIndex;
		}

		if ( !bRuleOk )
			continue;

		int k = m_DominatorInfo.AddToTail();
		m_DominatorInfo[k].m_DominantIndices.AddMultipleToTail( nDominatorCount, pDominators );
		m_DominatorInfo[k].m_SuppressedIndices.AddMultipleToTail( nSuppressedCount, pSuppressed );
	}
}


//-----------------------------------------------------------------------------
// Adds a dominator. Dominators are specified using raw control names
//-----------------------------------------------------------------------------
CDmeCombinationDominationRule *CDmeCombinationOperator::AddDominationRule( )
{
	CDmeCombinationDominationRule *pDominationRule = CreateElement< CDmeCombinationDominationRule >( "rule", GetFileId() );
	m_Dominators.AddToTail( pDominationRule );
	return pDominationRule;
}


//-----------------------------------------------------------------------------
// Adds a dominator. Dominators are specified using raw control names
//-----------------------------------------------------------------------------
CDmeCombinationDominationRule *CDmeCombinationOperator::AddDominationRule( CDmeCombinationDominationRule *pSrcRule )
{
	CDmeCombinationDominationRule *pDestRule = pSrcRule->Copy( );
	pDestRule->SetFileId( GetFileId(), TD_DEEP );
	m_Dominators.AddToTail( pDestRule );
	return pDestRule;
}


//-----------------------------------------------------------------------------
// Adds a dominator. Dominators are specified using raw control names
//-----------------------------------------------------------------------------
CDmeCombinationDominationRule *CDmeCombinationOperator::AddDominationRule( int nDominatorCount, const char **ppDominatorControlNames, int nSuppressedCount, const char **ppSuppressedControlNames )
{
	CDmeCombinationDominationRule *pDominationRule = AddDominationRule();
	for ( int i = 0; i < nDominatorCount; ++i )
	{
		pDominationRule->AddDominator( ppDominatorControlNames[i] );
	}
	for ( int i = 0; i < nSuppressedCount; ++i )
	{
		pDominationRule->AddSuppressed( ppSuppressedControlNames[i] );
	}
	return pDominationRule;
}


//-----------------------------------------------------------------------------
// Adds a dominator. Dominators are specified using raw control names
//-----------------------------------------------------------------------------
CDmeCombinationDominationRule *CDmeCombinationOperator::AddDominationRule( const CUtlVector< const char * > dominators, const CUtlVector< const char * > suppressed )
{
	return AddDominationRule( dominators.Count(), (const char **)dominators.Base(), suppressed.Count(), (const char **)suppressed.Base() );
}


//-----------------------------------------------------------------------------
// Removes a domination rule
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::RemoveDominationRule( int nIndex )
{
	CDmeCombinationDominationRule *pRule = m_Dominators[nIndex];
	m_Dominators.Remove( nIndex );
	DestroyElement( pRule );
}

void CDmeCombinationOperator::RemoveDominationRule( CDmeCombinationDominationRule *pRule )
{
	int nCount = m_Dominators.Count();
	for ( int i = 0; i < nCount; ++i )
	{
		if ( m_Dominators[i] == pRule )
		{
			RemoveDominationRule( i );
			break;
		}
	}
}

void CDmeCombinationOperator::RemoveAllDominationRules()
{
	int nCount = m_Dominators.Count();
	for ( int i = nCount; --i >= 0; )
	{
		RemoveDominationRule( i );
	}
}


//-----------------------------------------------------------------------------
// Iteration
//-----------------------------------------------------------------------------
int CDmeCombinationOperator::DominationRuleCount() const
{
	return m_Dominators.Count();
}

CDmeCombinationDominationRule *CDmeCombinationOperator::GetDominationRule( int i )
{
	return m_Dominators[i];
}


//-----------------------------------------------------------------------------
// Finds a domination rule
//-----------------------------------------------------------------------------
int CDmeCombinationOperator::FindDominationRule( CDmeCombinationDominationRule *pRule )
{
	int nCount = m_Dominators.Count();
	for ( int i = 0; i < nCount; ++i )
	{
		if ( m_Dominators[i] == pRule )
			return i;
	}
	return -1;
}


//-----------------------------------------------------------------------------
// Rule reordering
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::MoveDominationRuleUp( CDmeCombinationDominationRule* pRule )
{
	int nIndex = FindDominationRule( pRule );
	if ( nIndex > 0 )
	{
		m_Dominators.Swap( nIndex, nIndex - 1 );
		return;
	}
}

void CDmeCombinationOperator::MoveDominationRuleDown( CDmeCombinationDominationRule* pRule )
{
	int nIndex = FindDominationRule( pRule );
	if ( nIndex >= 0 && nIndex < m_Dominators.Count() - 1 )
	{
		m_Dominators.Swap( nIndex, nIndex + 1 );
		return;
	}
}


//-----------------------------------------------------------------------------
// Attaches a channel to an input
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::AttachChannelToControlValue( ControlIndex_t nControlIndex, CombinationControlType_t type, CDmeChannel *pChannel )
{
	pChannel->SetOutput( m_ControlValues[type].GetAttribute(), nControlIndex );
}


//-----------------------------------------------------------------------------
// Determines the weighting of input controls based on the deltaState name
//-----------------------------------------------------------------------------
int CDmeCombinationOperator::FindDeltaStateIndex( CDmAttribute *pDeltaArray, const char *pDeltaStateName )
{
	const CDmrElementArray<> deltas( pDeltaArray );
	int nDeltaArrayCount = deltas.Count();
	for ( int i = 0; i < nDeltaArrayCount; ++i )
	{
		CDmElement *pDeltaElement = deltas[i];
		if ( pDeltaElement && !Q_stricmp( pDeltaElement->GetName(), pDeltaStateName ) )
			return i;
	}

	return -1;
}


//-----------------------------------------------------------------------------
// Determines which combination to use based on the deltaState name
//-----------------------------------------------------------------------------
int CDmeCombinationOperator::ParseDeltaName(const char *pDeltaStateName, int *pControlIndices )
{
	char pBuf[512];
	Q_strncpy( pBuf, pDeltaStateName, sizeof(pBuf) );

	int nComboCount = 0;
	char *pEnd;
	for ( char *pName = pBuf; *pName; pName = pEnd )
	{
		pEnd = strchr( pName, '_' );
		if ( !pEnd )
		{
			pEnd = pName + Q_strlen( pName );
		}
		else
		{
			// Null-terminate
			*pEnd = 0;
			++pEnd;
		}

		int nControlIndex = FindRawControlIndex( pName );
		if ( nControlIndex < 0 )
			return 0;

		pControlIndices[ nComboCount++ ] = nControlIndex;
	}

	return nComboCount;
}


//-----------------------------------------------------------------------------
// Finds dominators
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::FindDominators( CombinationOperation_t& op )
{
	// Dominators are sets of inputs, which, when set to 1, will 
	// supress another set of inputs. Only combinations which contain *all*
	// dominators will suppress any combinatation that contains *all* suppressors
	int nDominatorCount = m_DominatorInfo.Count();
	for ( int i = 0; i < nDominatorCount; ++i )
	{
		DominatorInfo_t &info = m_DominatorInfo[i];

		// Look for suppressor indices in the control indices list
		int nCount = info.m_SuppressedIndices.Count();
		int j;
		for ( j = 0; j < nCount; ++j )
		{
			if ( op.m_ControlIndices.Find( info.m_SuppressedIndices[j] ) < 0 )
				break;
		}
		if ( j != nCount )
			continue;

		op.m_DominatorIndices.AddToTail( i );
	}
}


//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::ComputeCombinationInfo( int nIndex )
{
	CleanUpCombinationInfo( nIndex );

	int nCurrentCount = m_CombinationInfo.Count();
	while ( nIndex >= nCurrentCount )
	{
		int j = m_CombinationInfo.AddToTail();
		CombinationInfo_t &info = m_CombinationInfo[j];
		for ( int k = 0; k < COMBO_CONTROL_TYPE_COUNT; ++k )
		{
			info.m_hDestAttribute[k] = DMATTRIBUTE_HANDLE_INVALID;
		}
		++nCurrentCount;
	}

	CombinationInfo_t &info = m_CombinationInfo[nIndex];

	CDmElement *pSource = m_Targets[ nIndex ];
	if ( !pSource )
		return;

	CDmrElementArray<> deltas( pSource, "deltaStates" );
	if ( !deltas.IsValid() )
		return;

	CDmrArray<Vector2D> weights( pSource, "deltaStateWeights" );
	if ( !weights.IsValid() )
		return;

	CDmrArray<Vector2D> weightsLagged( pSource, "deltaStateWeightsLagged" );

	// This is an error state
	if ( deltas.Count() > weights.Count() )
		return;

	// This is an error state
	if ( weightsLagged.IsValid() && deltas.Count() > weightsLagged.Count() )
		return;

	info.m_hDestAttribute[COMBO_CONTROL_NORMAL] = weights.GetAttribute()->GetHandle();
	info.m_hDestAttribute[COMBO_CONTROL_LAGGED] = weightsLagged.IsValid() ? weightsLagged.GetAttribute()->GetHandle() : DMATTRIBUTE_HANDLE_INVALID;

	int *pTemp = (int*)_alloca( m_RawControlInfo.Count() * sizeof(int) );
	int nDeltaCount = deltas.Count();
	for ( int i = 0; i < nDeltaCount; ++i )
	{
		CDmElement *pDeltaElement = deltas[i];
		if ( !pDeltaElement )
			continue;

		int nControlCount = ParseDeltaName( pDeltaElement->GetName(), pTemp );
		if ( nControlCount == 0 )
			continue;

		int j = info.m_Outputs.AddToTail();
		CombinationOperation_t &op = info.m_Outputs[j];
		op.m_nDeltaStateIndex = i;
		op.m_ControlIndices.AddMultipleToTail( nControlCount, pTemp );

		// Find dominators
		FindDominators( op );
	}
}


//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::ComputeCombinationInfo()
{
	int nTargetCount = m_Targets.Count();
	for ( int i = 0; i < nTargetCount; ++i )
	{
		ComputeCombinationInfo( i );
	}
}


//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::CleanUpCombinationInfo( int nIndex )
{
	if ( nIndex >= m_CombinationInfo.Count() )
		return;

	CombinationInfo_t &info = m_CombinationInfo[ nIndex ];
	for ( int i = 0; i < COMBO_CONTROL_TYPE_COUNT; ++i )
	{
		info.m_hDestAttribute[i] = DMATTRIBUTE_HANDLE_INVALID;
	}
	info.m_Outputs.RemoveAll();
}


void CDmeCombinationOperator::CleanUpCombinationInfo( )
{
	int nCount = m_CombinationInfo.Count();
	for ( int i = 0; i < nCount; ++i )
	{
		CleanUpCombinationInfo( i );
	}
	m_CombinationInfo.RemoveAll();
}


//-----------------------------------------------------------------------------
// helpers for finding corrector names
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::Resolve()
{
	BaseClass::Resolve();

	if ( m_InputControls.IsDirty() || m_Dominators.IsDirty() )
	{
		RebuildRawControlList();
		ComputeCombinationInfo();
	}
}


//-----------------------------------------------------------------------------
// Adds a target for the combination operator
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::AddTarget( CDmElement *pElement )
{
	if ( !pElement )
		return;

	int i = m_Targets.AddToTail( pElement );
	ComputeCombinationInfo( i );
}


//-----------------------------------------------------------------------------
// Sets targets
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::AddTarget( CDmeDag *pDag )
{
	if ( !pDag )
		return;

	CDmeShape *pShape = pDag->GetShape();
	AddTarget( pShape );
	
	int nChildCount = pDag->GetChildCount();
	for ( int i = 0; i < nChildCount; ++i )
	{
		CDmeDag *pChild = pDag->GetChild(i);

		// Do not traverse into models
		if ( CastElement< CDmeModel >( pChild ) )
			continue;

		AddTarget( pChild );
	}
}

 
//-----------------------------------------------------------------------------
// Remaps non-stereo -> stereo, stereo ->left/right, also adds multilevel + filter
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::ComputeInternalControlValue( RawControlIndex_t nRawControlIndex, CombinationControlType_t type, Vector2D &value )
{
	const RawControlInfo_t &info = m_RawControlInfo[nRawControlIndex];
	const Vector &vecControlValue = m_ControlValues[ type ][ info.m_InputControl ];

	const bool bMultiControl = IsMultiControl( info.m_InputControl );
	float flValue = bMultiControl ? vecControlValue.z : vecControlValue.x;

	// Apply multicontrol remapping
	if ( flValue <= info.m_FilterRamp.x || flValue >= info.m_FilterRamp.w )
	{
		flValue = 0.0f;
	}
	else if ( flValue < info.m_FilterRamp.y )
	{
		flValue = RemapVal( flValue, info.m_FilterRamp.x, info.m_FilterRamp.y, 0.0f, 1.0f );
	}
	else if ( flValue > info.m_FilterRamp.z )
	{
		flValue = RemapVal( flValue, info.m_FilterRamp.z, info.m_FilterRamp.w, 1.0f, 0.0f );
	}
	else
	{
		flValue = 1.0f;
	}

	if ( IsEyelidControl( info.m_InputControl ) )
	{
		if ( info.m_bLowerEyelid )
		{
			flValue = ( 1.0f - flValue ) * vecControlValue.x;
		}
		else
		{
			flValue *= vecControlValue.x;
		}
	}
	else if ( bMultiControl )
	{
		flValue *= vecControlValue.x;
	}

	value.x = value.y = flValue;
	if ( IsStereoControl( info.m_InputControl ) )
	{
		if ( vecControlValue.y < 0.5f )
		{
			float flRightAmount = RemapVal( vecControlValue.y, 0.0f, 0.5f, 0.0f, 1.0f );
			value.y *= flRightAmount;
		}
		else
		{
			float flLeftAmount = RemapVal( vecControlValue.y, 0.5f, 1.0f, 1.0f, 0.0f );
			value.x *= flLeftAmount;
		}
	}
}


//-----------------------------------------------------------------------------
// Computes lagged input values from non-lagged input
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::ComputeLaggedInputValues()
{
	if ( !m_bSpecifyingLaggedData )
		return;

	float t = Plat_FloatTime();
	if ( m_flLastLaggedComputationTime == FLT_MIN )
	{
		m_flLastLaggedComputationTime = t;
	}

	float dt = t - m_flLastLaggedComputationTime;
	m_flLastLaggedComputationTime = t;
	float flFactor = ExponentialDecay( 0.8, 0.033, dt );

	int nCount = m_ControlValues[COMBO_CONTROL_NORMAL].Count();
	for ( int i = 0; i < nCount; ++i )
	{
		Vector vecLerp;
		VectorLerp( m_ControlValues[COMBO_CONTROL_NORMAL][i], m_ControlValues[COMBO_CONTROL_LAGGED][i], flFactor, vecLerp );
		m_ControlValues[COMBO_CONTROL_LAGGED].Set( i, vecLerp );
	}
}


//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::Operate()
{
	ComputeLaggedInputValues();

	int nCount = m_CombinationInfo.Count();
	for ( int i = 0; i < nCount; ++i )
	{
		CombinationInfo_t &info = m_CombinationInfo[i];

		for ( CombinationControlType_t type = COMBO_CONTROL_FIRST; type < COMBO_CONTROL_TYPE_COUNT; type = (CombinationControlType_t)(type+1) )
		{
			if ( ( info.m_hDestAttribute[type] == DMATTRIBUTE_HANDLE_INVALID ) || !g_pDataModel->IsAttributeHandleValid( info.m_hDestAttribute[type] ) )
				continue;

			CDmAttribute* pAttribute = g_pDataModel->GetAttribute( info.m_hDestAttribute[type] );
			if ( !pAttribute || pAttribute->GetType() != AT_VECTOR2_ARRAY )
				continue;

			CDmrArray< Vector2D > vec2D( pAttribute );

			CombinationControlType_t useType = m_bSpecifyingLaggedData ? type : COMBO_CONTROL_NORMAL;
			int nOutputCount = info.m_Outputs.Count();
			for ( int j = 0; j < nOutputCount; ++j )
			{
				CombinationOperation_t &op = info.m_Outputs[j];

				// Compute the core combination
				Vector2D vecValue( 1.0f, 1.0f );
				int nCombinationCount = op.m_ControlIndices.Count();
				for ( int k = 0; k < nCombinationCount; ++k )
				{
					Vector2D v;
					ComputeInternalControlValue( op.m_ControlIndices[k], useType, v );
					vecValue *= v;
				}


				// Compute the dominators
				int nDominatorCount = op.m_DominatorIndices.Count();
				for ( int k = 0; k < nDominatorCount; ++k )
				{
					const CUtlVector< int > &dominantIndices = m_DominatorInfo[ op.m_DominatorIndices[k] ].m_DominantIndices;
					int nDominantCount = dominantIndices.Count();

					Vector2D suppressor( -1.0f, -1.0f );
					for ( int l = 0; l < nDominantCount; ++l )
					{
						Vector2D v;
						ComputeInternalControlValue( dominantIndices[l], useType, v );
						suppressor *= v;
					}
					suppressor.x += 1.0f; suppressor.y += 1.0f;

					vecValue *= suppressor;
				}

				vec2D.Set( op.m_nDeltaStateIndex, vecValue );
			}
		}
	}
}

void CDmeCombinationOperator::GetInputAttributes( CUtlVector< CDmAttribute * > &attrs )
{
	for ( int i = 0; i < COMBO_CONTROL_TYPE_COUNT; ++i )
	{
		attrs.AddToTail( m_ControlValues[i].GetAttribute() );
	}
}

void CDmeCombinationOperator::GetOutputAttributes( CUtlVector< CDmAttribute * > &attrs )
{
	int nCount = m_CombinationInfo.Count();
	for ( int i = 0; i < nCount; ++i )
	{
		CombinationInfo_t &info = m_CombinationInfo[i];
		for ( int j = 0; j < COMBO_CONTROL_TYPE_COUNT; ++j )
		{
			if ( ( info.m_hDestAttribute[j] == DMATTRIBUTE_HANDLE_INVALID ) || !g_pDataModel->IsAttributeHandleValid( info.m_hDestAttribute[j] ) )
				continue;

			CDmAttribute* pAttribute = g_pDataModel->GetAttribute( info.m_hDestAttribute[j] );
			if ( !pAttribute || pAttribute->GetType() != AT_VECTOR2_ARRAY )
				continue;

			attrs.AddToTail( pAttribute );
		}
	}
}


//-----------------------------------------------------------------------------
// Used by studiomdl to discover the various combination rules
//-----------------------------------------------------------------------------
int CDmeCombinationOperator::GetOperationTargetCount() const
{
	return m_Targets.Count();
}

CDmElement *CDmeCombinationOperator::GetOperationTarget( int nTargetIndex )
{
	return m_Targets[ nTargetIndex ];
}

int CDmeCombinationOperator::GetOperationCount( int nTargetIndex ) const
{
	return m_CombinationInfo[ nTargetIndex ].m_Outputs.Count();
}

CDmElement *CDmeCombinationOperator::GetOperationDeltaState( int nTargetIndex, int nOpIndex )
{
	CDmElement *pElement = GetOperationTarget( nTargetIndex );
	const CDmrElementArray<> deltaArray( pElement, "deltaStates" );
	if ( !deltaArray.IsValid() )
		return NULL;

	int nDeltaStateIndex = m_CombinationInfo[ nTargetIndex ].m_Outputs[ nOpIndex ].m_nDeltaStateIndex;
	return deltaArray[ nDeltaStateIndex ];
}

const CUtlVector< int > &CDmeCombinationOperator::GetOperationControls( int nTargetIndex, int nOpIndex ) const
{
	return m_CombinationInfo[ nTargetIndex ].m_Outputs[ nOpIndex ].m_ControlIndices;
}

int CDmeCombinationOperator::GetOperationDominatorCount( int nTargetIndex, int nOpIndex ) const
{
	return m_CombinationInfo[ nTargetIndex ].m_Outputs[ nOpIndex ].m_DominatorIndices.Count();
}

const CUtlVector< int > &CDmeCombinationOperator::GetOperationDominator( int nTargetIndex, int nOpIndex, int nDominatorIndex ) const
{
	int nIndex = m_CombinationInfo[ nTargetIndex ].m_Outputs[ nOpIndex ].m_DominatorIndices[ nDominatorIndex ];
	return m_DominatorInfo[ nIndex ].m_DominantIndices;
}

void CDmeCombinationOperator::CopyControls( CDmeCombinationOperator *pSrc )
{
	// Clean Me Out!
	RemoveAllControls();

	// Copy The Source Controls
	for ( int i = 0; i < pSrc->m_InputControls.Count(); ++i )
	{
		CDmeCombinationInputControl *pSrcInput = pSrc->m_InputControls[ i ];
		if ( pSrcInput )
		{
			CDmeCombinationInputControl *pDstInput = pSrcInput->Copy( );
			pDstInput->SetFileId( GetFileId(), TD_DEEP );
			m_InputControls.AddToTail( pDstInput );
		}
	}

	// Copy The Control Values
	m_ControlValues[ COMBO_CONTROL_NORMAL ].CopyArray(
		pSrc->m_ControlValues[ COMBO_CONTROL_NORMAL ].Base(),
		pSrc->m_ControlValues[ COMBO_CONTROL_NORMAL ].Count() );

	m_ControlValues[ COMBO_CONTROL_LAGGED ].CopyArray(
		pSrc->m_ControlValues[ COMBO_CONTROL_LAGGED ].Base(),
		pSrc->m_ControlValues[ COMBO_CONTROL_LAGGED ].Count() );

	RebuildRawControlList();

	m_Dominators.Purge();

	// Copy The Source Controls
	for ( int i = 0; i < pSrc->m_Dominators.Count(); ++i )
	{
		CDmeCombinationDominationRule *pSrcDom = pSrc->m_Dominators[ i ];
		if ( pSrcDom )
		{
			CDmeCombinationDominationRule *pDstDom = pSrcDom->Copy( );
			pDstDom->SetFileId( GetFileId(), TD_DEEP );
			m_Dominators.AddToTail( pDstDom );
		}
	}

	RebuildRawControlList();
}


//-----------------------------------------------------------------------------
// Generates wrinkle deltas for the uncorrected controls
// NOTE: This is only being used because we have no authoring path for wrinkle data yet
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::GenerateWrinkleDeltas( CDmeShape *pShape, bool bOverwrite )
{
	CDmeMesh* pMesh = CastElement< CDmeMesh >( pShape );
	CDmeVertexData *pBindState = pMesh ? pMesh->FindBaseState( "bind" ) : NULL;
	if ( pBindState )
	{
		int nRawControlCount = m_RawControlInfo.Count();
		for ( int i = 0; i < nRawControlCount; ++i )
		{
			CDmeVertexDeltaData* pDelta = pMesh->FindDeltaState( m_RawControlInfo[i].m_Name );
			if ( pDelta )
			{
				pDelta->GenerateWrinkleDelta( pBindState, m_RawControlInfo[i].m_flWrinkleScale, bOverwrite );
			}
		}
	}
}

void CDmeCombinationOperator::GenerateWrinkleDeltas( bool bOverwrite /* = true */ )
{
	int nTargetCount = m_Targets.Count();
	for ( int i = 0; i < nTargetCount; ++i )
	{
		GenerateWrinkleDeltas( CastElement< CDmeShape >( m_Targets[i] ), bOverwrite );
	}
}


//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::RemoveAllTargets()
{
	m_Targets.RemoveAll();

	CleanUpCombinationInfo();
}


//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::SetToDefault()
{
	const int nControlsCount = m_InputControls.Count();
	for ( int i = 0; i < nControlsCount; ++i )
	{
		m_IsDefaultValue[ i ] = true;
		UpdateDefaultValue( i );
	}
}


//-----------------------------------------------------------------------------
//
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::SetToBase()
{
	const int nControlsCount = m_InputControls.Count();
	for ( int i = 0; i < nControlsCount; ++i )
	{
		const float flBaseValue = GetControlBaseValue( i );
		for ( int j = 0; j < COMBO_CONTROL_TYPE_COUNT; ++j )
		{
			const Vector &v = m_ControlValues[ j ][ i ];
			m_ControlValues[ j ].Set( i, Vector( flBaseValue, v.y, v.z ) );
		}
	}
}


//-----------------------------------------------------------------------------
// * Remove all controls which refer to raw controls which refer to delta
//   states which do not exist in any target
// * Remove all dominators which refer to states which no longer exist
//-----------------------------------------------------------------------------
void CDmeCombinationOperator::Purge()
{
	bool bDelete = true;

	const int nTargetCount = m_Targets.Count();
	for ( int i = 0; i < nTargetCount; ++i )
	{
		CDmElement *pSource = m_Targets[ i ];
		if ( !pSource )
			continue;

		CDmrElementArray<> deltas( pSource, "deltaStates" );
		if ( !deltas.IsValid() )
			continue;

		const int nDeltaCount = deltas.Count();

		do
		{
			bDelete = false;

			const int nControlCount = GetControlCount();
			for ( int j = 0; j < nControlCount; ++j )
			{
				bDelete = true;
				const int nRawControlCount = GetRawControlCount( j );
				for ( int k = 0; bDelete && k < nRawControlCount; ++k )
				{
					for ( int l = 0; l < nDeltaCount; ++l )
					{
						if ( !Q_strcmp( GetRawControlName( j, k ), deltas[ l ]->GetName() ) )
						{
							bDelete = false;
							break;
						}
					}
				}

				if ( bDelete )
				{
					RemoveControl( GetControlName( j ) );
					break;
				}
			}
		} while ( bDelete );
	}

	do 
	{
		bDelete = false;

		const int nDomCount = DominationRuleCount();
		for ( int i = 0; i < nDomCount; ++i )
		{
			bDelete = true;
			CDmeCombinationDominationRule *pDom = GetDominationRule( i );

			const int nDCount = pDom->DominatorCount();
			for ( int j = 0; j < nDCount; ++j )
			{
				if ( HasRawControl( pDom->GetDominator( j ) ) )
				{
					bDelete = false;
					break;
				}
			}

			if ( !bDelete )
			{
				const int nSCount = pDom->SuppressedCount();
				for ( int j = 0; j < nSCount; ++j )
				{
					if ( HasRawControl( pDom->GetSuppressed( j ) ) )
					{
						bDelete = false;
						break;
					}
				}
			}

			if ( bDelete )
			{
				RemoveDominationRule( i );
				break;
			}
		}
	} while ( bDelete );
}


//-----------------------------------------------------------------------------
// Creates lagged log data from an input log
//-----------------------------------------------------------------------------
static void CreateLaggedLog( CDmeVector2Log *pLog, CDmeVector2Log *pLaggedLog, int nSamplesPerSec )
{
}


//-----------------------------------------------------------------------------
// Helper method to create a lagged version of channel data from source data
//-----------------------------------------------------------------------------
void CreateLaggedVertexAnimation( CDmeChannelsClip *pClip, int nSamplesPerSec )
{
	if ( !pClip )
		return;

	int nChannelCount = pClip->m_Channels.Count();
	for ( int i = 0; i < nChannelCount; ++i )
	{
		CDmeChannel *pChannel = pClip->m_Channels[i];
		CDmElement *pDest = pChannel->GetToElement();
		CDmAttribute *pDestAttr = pChannel->GetToAttribute();
		int nArrayIndex = pChannel->GetToArrayIndex();

		if ( pChannel->GetLog()->GetDataType() != AT_VECTOR2 )
			continue;

		if ( Q_stricmp( pDestAttr->GetName(), "controlValues" ) )
			continue;

		if ( !pDest->GetValue<bool>( "usesLaggedValues" ) )
			continue;

		CDmAttribute *pLaggedAttr = pDest->GetAttribute( "controlValuesLagged" );
		if ( !pLaggedAttr || pLaggedAttr->GetType() != AT_VECTOR2_ARRAY )
			continue;

		int nLen = Q_strlen( pChannel->GetName() );
		char *pNewChannelName = (char*)_alloca( nLen + 10 );
		memcpy( pNewChannelName, pChannel->GetName(), nLen+1 );
		Q_strncpy( &pNewChannelName[nLen], "_lagged", 10 );
		CDmeChannel *pNewChannel = CreateElement< CDmeChannel >( pNewChannelName, pClip->GetFileId() );
		pNewChannel->SetOutput( pLaggedAttr, nArrayIndex );
		CDmeVector2Log *pNewLog = pNewChannel->CreateLog< Vector2D >( );
		pClip->m_Channels.AddToTail( pNewChannel );

		CreateLaggedLog( static_cast<CDmeVector2Log*>( pChannel->GetLog() ), pNewLog, nSamplesPerSec );
	}
}


//-----------------------------------------------------------------------------
//
// A class used to edit combination operators in Maya.. doesn't connect to targets
//
//-----------------------------------------------------------------------------
IMPLEMENT_ELEMENT_FACTORY( DmeMayaCombinationOperator, CDmeMayaCombinationOperator );


//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void CDmeMayaCombinationOperator::OnConstruction()
{
	m_DeltaStates.Init( this, "deltaStates" );
	m_DeltaStateWeights[COMBO_CONTROL_NORMAL].Init( this, "deltaStateWeights" );
	m_DeltaStateWeights[COMBO_CONTROL_LAGGED].Init( this, "deltaStateWeightsLagged" );
	AddTarget( this );
}

void CDmeMayaCombinationOperator::OnDestruction()
{
}


//-----------------------------------------------------------------------------
// Add, remove delta states
//-----------------------------------------------------------------------------
void CDmeMayaCombinationOperator::AddDeltaState( const char *pDeltaStateName )
{
	m_DeltaStates.AddToTail( CreateElement<CDmElement>( pDeltaStateName ) );
	for ( int i = 0; i < COMBO_CONTROL_TYPE_COUNT; ++i )
	{
		m_DeltaStateWeights[i].AddToTail( Vector2D( 1.0f, 1.0f ) );
	}
	ComputeCombinationInfo( 0 );
}

void CDmeMayaCombinationOperator::RemoveDeltaState( const char *pDeltaStateName )
{
	int nIndex = FindDeltaState( pDeltaStateName );
	if ( nIndex >= 0 )
	{
		m_DeltaStates.Remove( nIndex );
		for ( int i = 0; i < COMBO_CONTROL_TYPE_COUNT; ++i )
		{
			m_DeltaStateWeights[i].Remove( nIndex );
		}
	}
	ComputeCombinationInfo( 0 );
}

void CDmeMayaCombinationOperator::RemoveAllDeltaStates()
{
	m_DeltaStates.RemoveAll( );
	for ( int i = 0; i < COMBO_CONTROL_TYPE_COUNT; ++i )
	{
		m_DeltaStateWeights[i].RemoveAll();
	}
	ComputeCombinationInfo( 0 );
}

int CDmeMayaCombinationOperator::FindDeltaState( const char *pDeltaStateName )
{
	int nCount = m_DeltaStates.Count();
	for ( int i = 0; i < nCount; ++i )
	{
		if ( !Q_stricmp( pDeltaStateName, m_DeltaStates[i]->GetName() ) )
			return i;
	}
	return -1;
}

int CDmeMayaCombinationOperator::DeltaStateCount() const
{
	return m_DeltaStates.Count();
}

const char *CDmeMayaCombinationOperator::GetDeltaState( int nIndex ) const
{
	return m_DeltaStates[nIndex]->GetName();
}

const Vector2D& CDmeMayaCombinationOperator::GetDeltaStateWeight( int nIndex, CombinationControlType_t type ) const
{
	return m_DeltaStateWeights[type][nIndex];
}