Files
UnrealEngine/Engine/Source/Editor/AnimGraph/Public/AnimGraphNode_PoseDriver.h
2025-05-18 13:04:45 +08:00

133 lines
4.7 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
#include "CoreMinimal.h"
#include "UObject/ObjectMacros.h"
#include "AnimNodes/AnimNode_PoseDriver.h"
#include "AnimGraphNode_PoseHandler.h"
#include "EdGraph/EdGraphNodeUtils.h"
#include "AnimGraphNode_PoseDriver.generated.h"
class FCompilerResultsLog;
UCLASS(BlueprintType)
class ANIMGRAPH_API UAnimGraphNode_PoseDriver : public UAnimGraphNode_PoseHandler
{
GENERATED_UCLASS_BODY()
UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = Settings)
FAnimNode_PoseDriver Node;
/** Length of axis in world units used for debug drawing */
UPROPERTY(EditAnywhere, Category = Debugging, meta = (DefaultValue = "20.0", UIMin = "1.0", UIMax = "100.0"))
float AxisLength;
/** Number of subdivisions / lines used when debug drawing a cone */
UPROPERTY(EditAnywhere, Category = Debugging, meta = (DefaultValue = "32", UIMin = "6", UIMax = "128"))
int32 ConeSubdivision;
/** If checked the cones will be drawn in 3d for debugging */
UPROPERTY(EditAnywhere, Category = Debugging, meta = (DefaultValue = "True"))
bool bDrawDebugCones;
/** Used to indicate selected target to edit mode drawing */
int32 SelectedTargetIndex;
/** Delegate to call when selection changes */
FSimpleMulticastDelegate SelectedTargetChangeDelegate;
public:
/** Get the current preview node instance */
FAnimNode_PoseDriver* GetPreviewPoseDriverNode() const;
/** Util to replace current contents of PoseTargets with info from assigned PoseAsset */
UFUNCTION(BlueprintCallable, Category=PoseDriver)
void CopyTargetsFromPoseAsset();
/** Sets the pose-driver its source bones by name */
UFUNCTION(BlueprintCallable, Category=PoseDriver)
void SetSourceBones(const TArray<FName>& BoneNames);
/** Returns the pose-driver its source bones by name */
UFUNCTION(BlueprintPure, Category=PoseDriver)
void GetSourceBoneNames(TArray<FName>& BoneNames);
/** Set the pose-driver its driven bones by name */
UFUNCTION(BlueprintCallable, Category=PoseDriver)
void SetDrivingBones(const TArray<FName>& BoneNames);
/** Returns the pose-driver its driven bones by name */
UFUNCTION(BlueprintPure, Category=PoseDriver)
void GetDrivingBoneNames(TArray<FName>& BoneNames);
/** */
UFUNCTION(BlueprintCallable, Category=PoseDriver)
void SetRBFParameters(FRBFParams Parameters);
/** */
UFUNCTION(BlueprintPure, Category=PoseDriver)
FRBFParams& GetRBFParameters();
/** */
UFUNCTION(BlueprintCallable, Category=PoseDriver)
void SetPoseDriverSource(EPoseDriverSource DriverSource);
/** */
UFUNCTION(BlueprintPure, Category=PoseDriver)
EPoseDriverSource& GetPoseDriverSource();
/** */
UFUNCTION(BlueprintCallable, Category=PoseDriver)
void SetPoseDriverOutput(EPoseDriverOutput DriverOutput);
/** */
UFUNCTION(BlueprintPure, Category=PoseDriver)
EPoseDriverOutput& GetPoseDriverOutput();
/** Automatically modify TargetScale for each PoseTarget, based on distance to nearest neighbor */
void AutoSetTargetScales(float& OutMaxDistance);
/** Adds a new target, reallocating transforms array appropriately */
void AddNewTarget();
/** Reallocates transforms arrays as necessary to accommodate source bones */
void ReserveTargetTransforms();
/** Return the color for a given weight. Used for Details and EditMode */
FLinearColor GetColorFromWeight(float InWeight);
/** Used to refer back to preview instance in anim tools */
UPROPERTY(Transient)
TObjectPtr<USkeletalMeshComponent> LastPreviewComponent;
// Begin UObject Interface.
virtual void PostLoad() override;
virtual void Serialize(FArchive& Ar) override;
// End UObject Interface.
// UEdGraphNode interface
virtual FText GetNodeTitle(ENodeTitleType::Type TitleType) const override;
virtual FText GetTooltipText() const override;
virtual FText GetMenuCategory() const override;
virtual FLinearColor GetNodeBodyTintColor() const override;
// End of UEdGraphNode interface
// UAnimGraphNode_Base interface
virtual void ValidateAnimNodeDuringCompilation(USkeleton* ForSkeleton, FCompilerResultsLog& MessageLog) override;
virtual FEditorModeID GetEditorMode() const override;
virtual EAnimAssetHandlerType SupportsAssetClass(const UClass* AssetClass) const override;
virtual void CopyNodeDataToPreviewNode(FAnimNode_Base* InPreviewNode) override;
// End of UAnimGraphNode_Base interface
protected:
// UAnimGraphNode_PoseHandler interface
virtual bool IsPoseAssetRequired() override { return false; }
virtual FAnimNode_PoseHandler* GetPoseHandlerNode() override { return &Node; }
virtual const FAnimNode_PoseHandler* GetPoseHandlerNode() const override { return &Node; }
// End of UAnimGraphNode_PoseHandler interface
/** Constructing FText strings can be costly, so we cache the node's title */
FNodeTitleTextTable CachedNodeTitles;
};