Files
UnrealEngineUWP/Engine/Plugins/Animation/AnimationWarping/Source/Runtime/Private/AnimationWarpingRuntimeModule.cpp
koray hagen 1ccdd667ca #jira UE-131665
#rb aaron.cox

Migrating the Animation Warping plugin from Experimental to Beta.
#preflight 61ab2d10003ada0304ad2b6a

#ROBOMERGE-AUTHOR: koray.hagen
#ROBOMERGE-SOURCE: CL 18376688 in //UE5/Release-5.0/... via CL 18376689 via CL 18376691
#ROBOMERGE-BOT: STARSHIP (Release-Engine-Test -> Main) (v895-18170469)

[CL 18376692 by koray hagen in ue5-main branch]
2021-12-04 04:19:01 -05:00

75 lines
3.1 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#include "AnimationWarpingRuntimeModule.h"
#include "Animation/AnimRootMotionProvider.h"
#include "Animation/AnimNodeBase.h"
#include "Animation/AnimSequence.h"
#include "Animation/BuiltInAttributeTypes.h"
#include "Modules/ModuleManager.h"
class FAnimationWarpingRuntimeModule : public IAnimationWarpingRuntimeModule
{
};
namespace UE { namespace AnimationWarping {
// Root motion animation attribute data will always be accessible on the root bone's attribute container
static const UE::Anim::FAttributeId RootMotionAttributeId = { UE::Anim::IAnimRootMotionProvider::AttributeName , FCompactPoseBoneIndex(0) };
class FModule : public IModuleInterface, public UE::Anim::IAnimRootMotionProvider
{
public:
virtual void StartupModule() override;
virtual void ShutdownModule() override;
public:
virtual void SampleRootMotion(const FDeltaTimeRecord& SampleRange, const UAnimSequence& Sequence, bool bLoopingSequence, UE::Anim::FStackAttributeContainer& OutAttributes) const override;
virtual bool OverrideRootMotion(const FTransform& RootMotionDelta, UE::Anim::FStackAttributeContainer& OutAttributes) const override;
virtual bool ExtractRootMotion(const UE::Anim::FStackAttributeContainer& Attributes, FTransform& OutRootMotionDelta) const override;
virtual bool HasRootMotion(const UE::Anim::FStackAttributeContainer& Attributes) const override;
};
void FModule::StartupModule()
{
IModularFeatures::Get().RegisterModularFeature(UE::Anim::IAnimRootMotionProvider::ModularFeatureName, this);
}
void FModule::ShutdownModule()
{
IModularFeatures::Get().UnregisterModularFeature(UE::Anim::IAnimRootMotionProvider::ModularFeatureName, this);
}
void FModule::SampleRootMotion(const FDeltaTimeRecord& SampleRange, const UAnimSequence& Sequence, bool bLoopingSequence, UE::Anim::FStackAttributeContainer& OutAttributes) const
{
const FTransform RootMotionTransform = Sequence.ExtractRootMotion(SampleRange.GetPrevious(), SampleRange.Delta, bLoopingSequence);
FTransformAnimationAttribute* RootMotionAttribute = OutAttributes.FindOrAdd<FTransformAnimationAttribute>(RootMotionAttributeId);
RootMotionAttribute->Value = RootMotionTransform;
}
bool FModule::OverrideRootMotion(const FTransform& RootMotionDelta, UE::Anim::FStackAttributeContainer& OutAttributes) const
{
if (FTransformAnimationAttribute* RootMotionAttribute = OutAttributes.Find<FTransformAnimationAttribute>(RootMotionAttributeId))
{
RootMotionAttribute->Value = RootMotionDelta;
return true;
}
return false;
}
bool FModule::ExtractRootMotion(const UE::Anim::FStackAttributeContainer& Attributes, FTransform& OutRootMotionDelta) const
{
const FTransformAnimationAttribute* RootMotionAttribute = Attributes.Find<FTransformAnimationAttribute>(RootMotionAttributeId);
OutRootMotionDelta = RootMotionAttribute ? RootMotionAttribute->Value : FTransform::Identity;
return !!RootMotionAttribute;
}
bool FModule::HasRootMotion(const UE::Anim::FStackAttributeContainer& Attributes) const
{
return !!Attributes.Find<FTransformAnimationAttribute>(RootMotionAttributeId);
}
}}
IMPLEMENT_MODULE(UE::AnimationWarping::FModule, AnimationWarpingRuntime)