Files
UnrealEngineUWP/Engine/Source/Editor/AnimGraph/Private/AnimGraphNode_PoseHandler.cpp
thomas sarkanen e9cdd12510 Fixed ensure when expanding asset override tree for anim BPs that contain pose handlers
#rb Jurre.deBaare
#preflight 627d27d84a05ef0394cc3ebb

#ROBOMERGE-AUTHOR: thomas.sarkanen
#ROBOMERGE-SOURCE: CL 20178510 via CL 20181518 via CL 20181734 via CL 20181818 via CL 20181861
#ROBOMERGE-BOT: UE5 (Release-Engine-Staging -> Main) (v943-19904690)

[CL 20183656 by thomas sarkanen in ue5-main branch]
2022-05-13 13:24:41 -04:00

64 lines
2.0 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#include "AnimGraphNode_PoseHandler.h"
#include "Animation/PoseAsset.h"
#include "AnimNodes/AnimNode_PoseHandler.h"
#include "Kismet2/CompilerResultsLog.h"
#include "IAnimBlueprintNodeOverrideAssetsContext.h"
UAnimGraphNode_PoseHandler::UAnimGraphNode_PoseHandler(const FObjectInitializer& ObjectInitializer)
:Super(ObjectInitializer)
{
}
void UAnimGraphNode_PoseHandler::ValidateAnimNodeDuringCompilation(USkeleton* ForSkeleton, FCompilerResultsLog& MessageLog)
{
Super::ValidateAnimNodeDuringCompilation(ForSkeleton, MessageLog);
ValidateAnimNodeDuringCompilationHelper(ForSkeleton, MessageLog, GetPoseHandlerNode()->PoseAsset, UPoseAsset::StaticClass(), FindPin(GET_MEMBER_NAME_STRING_CHECKED(FAnimNode_PoseHandler, PoseAsset)), GET_MEMBER_NAME_CHECKED(FAnimNode_PoseHandler, PoseAsset));
}
void UAnimGraphNode_PoseHandler::SetAnimationAsset(UAnimationAsset* Asset)
{
if (UPoseAsset* PoseAsset = Cast<UPoseAsset>(Asset))
{
GetPoseHandlerNode()->PoseAsset = PoseAsset;
}
}
void UAnimGraphNode_PoseHandler::OnOverrideAssets(IAnimBlueprintNodeOverrideAssetsContext& InContext) const
{
if(InContext.GetAssets().Num() > 0)
{
if (UPoseAsset* PoseAsset = Cast<UPoseAsset>(InContext.GetAssets()[0]))
{
FAnimNode_PoseHandler& AnimNode = InContext.GetAnimNode<FAnimNode_PoseHandler>();
AnimNode.SetPoseAsset(PoseAsset);
}
}
}
void UAnimGraphNode_PoseHandler::PreloadRequiredAssets()
{
PreloadObject(GetPoseHandlerNode()->PoseAsset);
Super::PreloadRequiredAssets();
}
UAnimationAsset* UAnimGraphNode_PoseHandler::GetAnimationAsset() const
{
UPoseAsset* PoseAsset = GetPoseHandlerNode()->PoseAsset;
UEdGraphPin* PoseAssetPin = FindPin(GET_MEMBER_NAME_STRING_CHECKED(FAnimNode_PoseHandler, PoseAsset));
if (PoseAssetPin != nullptr && PoseAsset == nullptr)
{
PoseAsset = Cast<UPoseAsset>(PoseAssetPin->DefaultObject);
}
return PoseAsset;
}
TSubclassOf<UAnimationAsset> UAnimGraphNode_PoseHandler::GetAnimationAssetClass() const
{
return UPoseAsset::StaticClass();
}