1
This commit is contained in:
FluorescentCIAAfricanAmerican
275
sfmobjects/exportfacialanimation.cpp
Normal file
275
sfmobjects/exportfacialanimation.cpp
Normal file
@ -0,0 +1,275 @@
|
||||
//========= Copyright Valve Corporation, All rights reserved. ============//
|
||||
//
|
||||
// A class representing session state for the SFM
|
||||
//
|
||||
//=============================================================================
|
||||
|
||||
#include "sfmobjects/exportfacialanimation.h"
|
||||
#include "movieobjects/dmeclip.h"
|
||||
#include "movieobjects/dmeanimationset.h"
|
||||
#include "movieobjects/dmegamemodel.h"
|
||||
#include "movieobjects/dmetrackgroup.h"
|
||||
#include "movieobjects/dmetrack.h"
|
||||
#include "movieobjects/dmesound.h"
|
||||
#include "movieobjects/dmelog.h"
|
||||
#include "movieobjects/dmechannel.h"
|
||||
|
||||
|
||||
//-----------------------------------------------------------------------------
|
||||
// Contains export information
|
||||
//-----------------------------------------------------------------------------
|
||||
struct ExportInfo_t
|
||||
{
|
||||
CDmeFilmClip *m_pMovie;
|
||||
CDmeFilmClip *m_pShot;
|
||||
CDmeAnimationSet *m_pAnimationSet;
|
||||
DmeTime_t m_tExportStart;
|
||||
DmeTime_t m_tExportEnd;
|
||||
};
|
||||
|
||||
|
||||
//-----------------------------------------------------------------------------
|
||||
// Used to transform channel data into export time
|
||||
//-----------------------------------------------------------------------------
|
||||
static void ComputeExportChannelScaleBias( double *pScale, DmeTime_t *pBias, ExportInfo_t &info, CDmeChannel *pChannel )
|
||||
{
|
||||
DmeClipStack_t channelToGlobal;
|
||||
if ( pChannel->BuildClipStack( &channelToGlobal, info.m_pMovie, info.m_pShot ) )
|
||||
{
|
||||
DmeTime_t tOffset = CDmeClip::FromChildMediaTime( channelToGlobal, DMETIME_ZERO, false );
|
||||
DmeTime_t tScale = CDmeClip::FromChildMediaTime( channelToGlobal, DmeTime_t( 1.0f ), false );
|
||||
*pBias = tOffset - info.m_pShot->GetStartTime();
|
||||
*pScale = ( tScale - tOffset ).GetSeconds();
|
||||
}
|
||||
}
|
||||
|
||||
static void GetExportTimeRange( DmeTime_t *pExportStart, DmeTime_t *pExportEnd, CDmeFilmClip *pShot )
|
||||
{
|
||||
*pExportStart = DMETIME_ZERO;
|
||||
*pExportEnd = pShot->GetDuration();
|
||||
}
|
||||
|
||||
|
||||
//-----------------------------------------------------------------------------
|
||||
// Adds a log layer to the list of logs for export
|
||||
//-----------------------------------------------------------------------------
|
||||
static void AddLogLayerForExport( ExportInfo_t &info, CDmElement *pRoot, const char *pControlName, CDmeChannel *pChannel )
|
||||
{
|
||||
CDmeLog *pLog = pChannel->GetLog();
|
||||
if ( !pLog || pLog->GetNumLayers() == 0 )
|
||||
return;
|
||||
|
||||
CDmrElementArray<> animations( pRoot, "animations" );
|
||||
|
||||
DmeTime_t tBias;
|
||||
double flScale;
|
||||
ComputeExportChannelScaleBias( &flScale, &tBias, info, pChannel );
|
||||
|
||||
// Only export the base layer
|
||||
CDmeLogLayer* pLogLayer = pLog->GetLayer( 0 )->Copy();
|
||||
pLogLayer->SetName( pControlName );
|
||||
pLogLayer->ScaleBiasKeyTimes( flScale, tBias );
|
||||
|
||||
// Forcibly add keys @ the start + end time
|
||||
DmeTime_t tStartTime = ( info.m_tExportStart - tBias ) / flScale;
|
||||
DmeTime_t tEndTime = ( info.m_tExportEnd - tBias ) / flScale;
|
||||
pLogLayer->InsertKeyFromLayer( info.m_tExportStart, pLog->GetLayer(0), tStartTime );
|
||||
pLogLayer->InsertKeyFromLayer( info.m_tExportEnd, pLog->GetLayer(0), tEndTime );
|
||||
|
||||
pLogLayer->RemoveKeysOutsideRange( info.m_tExportStart, info.m_tExportEnd );
|
||||
animations.AddToTail( pLogLayer );
|
||||
}
|
||||
|
||||
|
||||
//-----------------------------------------------------------------------------
|
||||
// Exports animations
|
||||
//-----------------------------------------------------------------------------
|
||||
static void ExportAnimations( ExportInfo_t &info, CDmElement *pRoot )
|
||||
{
|
||||
CDmrElementArray<> animations( pRoot, "animations", true );
|
||||
|
||||
// Build a list of all controls
|
||||
const CDmaElementArray< CDmElement > &controls = info.m_pAnimationSet->GetControls();
|
||||
int nControlCount = controls.Count();
|
||||
for ( int i = 0; i < nControlCount; ++i )
|
||||
{
|
||||
CDmElement *pControl = controls[i];
|
||||
if ( !pControl || pControl->GetValue<bool>( "transform" ) )
|
||||
continue;
|
||||
bool bIsStereo = pControl->GetValue<bool>( "combo" );
|
||||
if ( bIsStereo )
|
||||
{
|
||||
CDmeChannel *pValueChannel = pControl->GetValueElement<CDmeChannel>( "valuechannel" );
|
||||
CDmeChannel *pBalanceChannel = pControl->GetValueElement<CDmeChannel>( "balancechannel" );
|
||||
|
||||
CDmeLog *pValueLog = pValueChannel->GetLog();
|
||||
CDmeLog *pBalanceLog = pBalanceChannel->GetLog();
|
||||
if ( pValueLog && pBalanceLog && pValueLog->GetNumLayers() != 0 && pBalanceLog->GetNumLayers() != 0 )
|
||||
{
|
||||
DmeTime_t tValueBias, tBalanceBias;
|
||||
double flValueScale, flBalanceScale;
|
||||
ComputeExportChannelScaleBias( &flValueScale, &tValueBias, info, pValueChannel );
|
||||
ComputeExportChannelScaleBias( &flBalanceScale, &tBalanceBias, info, pBalanceChannel );
|
||||
|
||||
// Make copy to maintain log layer types
|
||||
CDmeLogLayer *pValueLogLayer = pValueLog->GetLayer( 0 )->Copy();
|
||||
CDmeLogLayer *pBalanceLogLayer = pBalanceLog->GetLayer( 0 )->Copy();
|
||||
pValueLogLayer->ScaleBiasKeyTimes( flValueScale, tValueBias );
|
||||
pBalanceLogLayer->ScaleBiasKeyTimes( flBalanceScale, tBalanceBias );
|
||||
|
||||
// Forcibly insert keys @ start + end times.
|
||||
DmeTime_t tStartTime = ( info.m_tExportStart - tValueBias ) / flValueScale;
|
||||
DmeTime_t tEndTime = ( info.m_tExportEnd - tValueBias ) / flValueScale;
|
||||
pValueLogLayer->InsertKeyFromLayer( info.m_tExportStart, pValueLog->GetLayer(0), tStartTime );
|
||||
pValueLogLayer->InsertKeyFromLayer( info.m_tExportEnd, pValueLog->GetLayer(0), tEndTime );
|
||||
|
||||
tStartTime = ( info.m_tExportStart - tBalanceBias ) / flBalanceScale;
|
||||
tEndTime = ( info.m_tExportEnd - tBalanceBias ) / flBalanceScale;
|
||||
pBalanceLogLayer->InsertKeyFromLayer( info.m_tExportStart, pBalanceLog->GetLayer(0), tStartTime );
|
||||
pBalanceLogLayer->InsertKeyFromLayer( info.m_tExportEnd, pBalanceLog->GetLayer(0), tEndTime );
|
||||
|
||||
pValueLogLayer->RemoveKeysOutsideRange( info.m_tExportStart, info.m_tExportEnd );
|
||||
pBalanceLogLayer->RemoveKeysOutsideRange( info.m_tExportStart, info.m_tExportEnd );
|
||||
|
||||
char pControlName[512];
|
||||
Q_snprintf( pControlName, sizeof(pControlName), "value_%s", pControl->GetName() );
|
||||
pValueLogLayer->SetName( pControlName );
|
||||
animations.AddToTail( pValueLogLayer );
|
||||
|
||||
Q_snprintf( pControlName, sizeof(pControlName), "balance_%s", pControl->GetName() );
|
||||
pBalanceLogLayer->SetName( pControlName );
|
||||
animations.AddToTail( pBalanceLogLayer );
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
CDmeChannel *pChannel = pControl->GetValueElement<CDmeChannel>( "channel" );
|
||||
AddLogLayerForExport( info, pRoot, pControl->GetName(), pChannel );
|
||||
}
|
||||
|
||||
if ( pControl->GetValue<bool>( "multi" ) )
|
||||
{
|
||||
char pControlName[512];
|
||||
Q_snprintf( pControlName, sizeof(pControlName), "multi_%s", pControl->GetName() );
|
||||
CDmeChannel *pMultiChannel = pControl->GetValueElement<CDmeChannel>( "multilevelchannel" );
|
||||
AddLogLayerForExport( info, pRoot, pControlName, pMultiChannel );
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
//-----------------------------------------------------------------------------
|
||||
// Helper to export sounds
|
||||
//-----------------------------------------------------------------------------
|
||||
static void ExportSounds( ExportInfo_t &info, CDmElement *pRoot, CDmeClip *pClip, DmeTime_t tOffset )
|
||||
{
|
||||
CDmrElementArray<> sounds( pRoot, "sounds", true );
|
||||
|
||||
DmeClipStack_t soundToGlobal;
|
||||
int gc = pClip->GetTrackGroupCount();
|
||||
for ( int i = 0; i < gc; ++i )
|
||||
{
|
||||
CDmeTrackGroup *pTrackGroup = pClip->GetTrackGroup( i );
|
||||
DMETRACKGROUP_FOREACH_CLIP_TYPE_START( CDmeSoundClip, pTrackGroup, pTrack, pSoundClip )
|
||||
|
||||
const char *pGameSoundName = pSoundClip->m_Sound->m_GameSoundName;
|
||||
if ( !pGameSoundName || !pGameSoundName[0] )
|
||||
continue;
|
||||
|
||||
if ( pSoundClip->IsMute() )
|
||||
continue;
|
||||
|
||||
if ( !pSoundClip->BuildClipStack( &soundToGlobal, info.m_pMovie, pClip ) )
|
||||
continue;
|
||||
|
||||
DmeTime_t tStart = CDmeClip::FromChildMediaTime( soundToGlobal, DMETIME_ZERO, false );
|
||||
DmeTime_t tEnd = CDmeClip::FromChildMediaTime( soundToGlobal, pSoundClip->GetDuration(), false );
|
||||
tStart -= tOffset;
|
||||
tEnd -= tOffset;
|
||||
if ( tStart >= info.m_tExportEnd || tEnd <= info.m_tExportStart )
|
||||
continue;
|
||||
|
||||
const char *pName = pSoundClip->GetName();
|
||||
CDmElement *pSoundEvent = CreateElement<CDmElement>( pName );
|
||||
pSoundEvent->SetValue( "start", tStart.GetTenthsOfMS() );
|
||||
pSoundEvent->SetValue( "end", tEnd.GetTenthsOfMS() );
|
||||
pSoundEvent->SetValue( "gamesound", pGameSoundName );
|
||||
sounds.AddToTail( pSoundEvent );
|
||||
|
||||
DMETRACKGROUP_FOREACH_CLIP_TYPE_END()
|
||||
}
|
||||
}
|
||||
|
||||
static void ExportSounds_R( ExportInfo_t &info, CDmElement *pRoot, CDmeClip *pClip, DmeTime_t tOffset )
|
||||
{
|
||||
ExportSounds( info, pRoot, pClip, tOffset );
|
||||
|
||||
// Recurse
|
||||
DmeClipStack_t childToGlobal;
|
||||
int gc = pClip->GetTrackGroupCount();
|
||||
for ( int i = 0; i < gc; ++i )
|
||||
{
|
||||
CDmeTrackGroup *pTrackGroup = pClip->GetTrackGroup( i );
|
||||
DMETRACKGROUP_FOREACH_CLIP_START( pTrackGroup, pTrack, pChild )
|
||||
|
||||
if ( !pChild->BuildClipStack( &childToGlobal, info.m_pMovie, pClip ) )
|
||||
continue;
|
||||
|
||||
DmeTime_t tStart = CDmeClip::FromChildMediaTime( childToGlobal, DMETIME_ZERO, false );
|
||||
DmeTime_t tEnd = CDmeClip::FromChildMediaTime( childToGlobal, pChild->GetDuration(), false );
|
||||
tStart -= tOffset;
|
||||
tEnd -= tOffset;
|
||||
if ( tStart >= info.m_tExportEnd || tEnd <= info.m_tExportStart )
|
||||
continue;
|
||||
|
||||
ExportSounds_R( info, pRoot, pChild, tOffset );
|
||||
|
||||
DMETRACKGROUP_FOREACH_CLIP_END()
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
//-----------------------------------------------------------------------------
|
||||
// Exports sounds, default implementation
|
||||
//-----------------------------------------------------------------------------
|
||||
static void ExportSounds( ExportInfo_t &info, CDmElement *pRoot )
|
||||
{
|
||||
DmeTime_t tOffset = info.m_pShot->GetStartTime();
|
||||
ExportSounds( info, pRoot, info.m_pMovie, tOffset );
|
||||
ExportSounds_R( info, pRoot, info.m_pShot, tOffset );
|
||||
}
|
||||
|
||||
|
||||
//-----------------------------------------------------------------------------
|
||||
// Exports an .fac file
|
||||
//-----------------------------------------------------------------------------
|
||||
bool ExportFacialAnimation( const char *pFileName, CDmeFilmClip *pMovie, CDmeFilmClip *pShot, CDmeAnimationSet *pAnimationSet )
|
||||
{
|
||||
if ( !pMovie || !pShot || !pAnimationSet )
|
||||
return false;
|
||||
|
||||
const char *pFileFormat = "facial_animation";
|
||||
CDmElement *pRoot = CreateElement< CDmElement >( pAnimationSet->GetName() );
|
||||
|
||||
ExportInfo_t info;
|
||||
info.m_pMovie = pMovie;
|
||||
info.m_pShot = pShot;
|
||||
info.m_pAnimationSet = pAnimationSet;
|
||||
GetExportTimeRange( &info.m_tExportStart, &info.m_tExportEnd, pShot );
|
||||
|
||||
CDmeGameModel *pGameModel = pAnimationSet->GetValueElement<CDmeGameModel>( "gameModel" );
|
||||
if ( pGameModel )
|
||||
{
|
||||
pRoot->SetValue( "gamemodel", pGameModel->GetModelName() );
|
||||
}
|
||||
ExportAnimations( info, pRoot );
|
||||
ExportSounds( info, pRoot );
|
||||
|
||||
pRoot->SetFileId( DMFILEID_INVALID, TD_DEEP );
|
||||
const char *pEncoding = "keyvalues2_flat";
|
||||
bool bOk = g_pDataModel->SaveToFile( pFileName, NULL, pEncoding, pFileFormat, pRoot );
|
||||
DestroyElement( pRoot, TD_DEEP );
|
||||
return bOk;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user